(* $Id: netsys_mem.mli 2195 2015-01-01 12:23:39Z gerd $ *)
(** Bigarrays as memory buffers *)
type memory =
Netsys_types.memory
(** We consider 1-dimensional bigarrays of chars as memory buffers.
They have the useful property that the garbage collector cannot
relocate them, i.e. the address is fixed. Also, one can mmap
a file, and connect the bigarray with shared memory.
*)
(** {2 General} *)
val blit_memory_to_string : memory -> int -> string -> int -> int -> unit
(** [blit_memory_to_string src srcoff dst dstoff len] copies [len] characters
from buffer [src], starting at character number [srcoff], to
string [dst], starting at character number [dstoff]
Raise [Invalid_argument] if [srcoff] and [len] do not
designate a valid subbuffer of [src], or if [dstoff] and [len]
do not designate a valid substring of [dst]. *)
external blit_memory_to_string_unsafe :
memory -> int -> string -> int -> int -> unit
= "netsys_blit_memory_to_string" "noalloc"
(** Unsafe version *)
val blit_string_to_memory : string -> int -> memory -> int -> int -> unit
(** [blit_string_to_memory src srcoff dst dstoff len] copies [len] characters
from string [src], starting at character number [srcoff], to
buffer [dst], starting at character number [dstoff]
Raise [Invalid_argument] if [srcoff] and [len] do not
designate a valid substring of [src], or if [dstoff] and [len]
do not designate a valid subbuffer of [dst]. *)
external blit_string_to_memory_unsafe :
string -> int -> memory -> int -> int -> unit
= "netsys_blit_string_to_memory" "noalloc"
(** Unsafe version *)
val memory_of_string : string -> memory
(** Return a new bigarray as a copy of the string *)
val string_of_memory : memory -> string
(** Return a new string as a copy of the bigarray *)
val memory_address : memory -> nativeint
(** Returns the start address of the buffer *)
val memory_of_bigarray : ('a, 'b, 'c) Bigarray.Genarray.t -> memory
val memory_of_bigarray_1 : ('a, 'b, 'c) Bigarray.Array1.t -> memory
val memory_of_bigarray_2 : ('a, 'b, 'c) Bigarray.Array2.t -> memory
val memory_of_bigarray_3 : ('a, 'b, 'c) Bigarray.Array3.t -> memory
(** These functions return an arbitrary bigarray as [memory].
Due to a bug in the OCaml runtime, this is for now limited to
arrays of up to 2G size (in bytes). (The bug exists at least
up to OCaml-3.12.1.)
*)
(** {2 Allocation and memory-mapping} *)
val getpagesize : unit -> int
(** Returns the size of a page as reported by [sysconf].
On many systems, a page has 4096 bytes, but this cannot be relied
upon.
This function is only available if the system has [sysconf].
*)
val pagesize : int
(** The best guess at the page size *)
val alloc_memory_pages : ?addr:nativeint -> ?exec:bool -> int -> memory
(** Allocates memory in units of pages. The memory buffer will start
on a page boundary.
The passed int is the requested number of {b bytes}. The size of
the buffer is rounded up so a whole number of pages is allocated.
Optionally, one can request a certain address [addr] (which must
be a multiple of the page size). There is, however, no guarantee
that this wish can be fulfilled. In any way, one should check with
[memory_address] what the start address really is.
If [exec], the memory region is marked as executable.
This function is only available if the system has [sysconf], [mmap],
and allows to allocate anonymous memory with [mmap] (outside POSIX
but common).
*)
val alloc_aligned_memory : int -> int -> memory
(** [alloc_aligned_memory alignment size]: Allocates a buffer of [size]
whose start address is a multiple of [alignment]. The [alignment]
must be a power of two, and at least [Sys.word_size/8].
Aligned memory can be useful for ensuring that the whole memory
block is in the same cache line. A cache line typically has
64 or 128 bytes - but this is very platform-specific. (Linux:
look at [/proc/cpuinfo].)
This function is only available if the system has [posix_memalign].
*)
val memory_map_file : Unix.file_descr ->
?pos:int64 ->
?addr:nativeint ->
bool -> int -> memory
(** [memory_map_file fd shared size]: Maps [size] bytes of the file
[fd] into memory, and returns the memory buffer like
[Bigarray.Array1.map_file]. [pos] and [shared] have the same
meaning as there. In [addr] one can suggest a start address.
There is, however, no guarantee that this wish can be fulfilled.
*)
val memory_unmap_file : memory -> unit
(** Unmaps the file. The memory block must have been allocated
with [memory_map_file] or with [Bigarray.Array1.map_file].
{b Note that the data pointer of the bigarray is set to NULL,
and that any further access of the array will trigger a
segmentation violation!} The intention of this function is to
control when the file mapping is removed. Normally, this is
done first when the GC finalizer is run.
It is required that there are no subarrays at the time of
calling this function. (If so, the function does nothing.)
*)
val zero_pages : memory -> int -> int -> unit
(** [zero_pages m pos len]: If possible, the memory pages in the
range [pos] to [pos+len-1] of [m] are allocated again, so that
they replace the previous pages.
It is required that the start address of the range is a
multiple of the page size, and the [len] is a multiple of
the page size. Fails with [Invalid_argument] if the requirements
are not met, or the function is otherwise unavailable.
Calling [zero_pages] is sometimes an optimization when old
memory pages can be dropped, and when the alternative of
overwriting these pages would imply a copy-on-write operation.
*)
val grab : nativeint -> int -> memory
(** [grab addr len]: Interprets the address range from [addr] to
[addr+len-1] as [memory] bigarray.
This function does not allocate! It assumes that the given
address range points to valid memory.
*)
(** {2 Interpreting memory as values} *)
val as_value : memory -> int -> 'a
(** [as_value mem offset]: Returns a pointer to [mem+offset]. There
must be a valid boxed value at this address (i.e. at
the word preceding [mem+offset] there must be a valid block
header, followed by a valid value of the right type). However,
this is not checked:
{b This is an unsafe function that may crash the program if used
in the wrong way!}
It is possible that the memory block is deallocated while the
returned value still exists. Any attempt to access the value will
result into undefined behavior (anything from funny results
to crashes may happen).
Some Ocaml primitives might not work on the returned values
(polymorphic equality, marshalling, hashing) unless
{!Netsys_mem.value_area} is called for the memory block.
*)
val as_obj : memory -> int -> Obj.t
(** Same as [as_value] but returns the value as [Obj.t] *)
val value_area : memory -> unit
(** Marks the memory block as value area. This enables that the
value primitives (polymorphic equality, marshalling, hashing)
return meaningful results. The memory area is automatically
unmarked when the finaliser for the memory block is run.
Be careful when marking sub arrays.
This function is first available since O'Caml 3.11.
*)
val obj_address : Obj.t -> nativeint
val hdr_address : Obj.t -> nativeint
(** These two functions return the address of the [Obj.t] and the
address of the header of the [Obj.t], respectively.
Note that this can only be relied upon if the input object
cannot be moved around by the garbage collector!
*)
val cmp_string : string -> string -> int
(** Compares two strings like [String.compare]. This also works
when the strings reside outside the O'Caml heap, e.g. in a
[memory] block.
*)
exception Out_of_space
val init_header : memory -> int -> int -> int -> unit
(** [init_header mem offset tag size]: Initializes the word at
[mem+offset] as an Ocaml value header with the given [tag]
and the given [size] (in words). The GC color is always set
to "white".
*)
val init_string : memory -> int -> int -> (int * int)
(** [let voffset, bytelen = init_string mem offset len]:
Initializes the memory at [offset]
and following bytes as Ocaml string with length [len].
Returns in [voffset] the offset where the value starts
(i.e. [offset] plus one word), and in [bytelen] the number
of bytes used in [mem].
[offset] must be a multiple of the word size in bytes.
The string can be accessed with
{[ let s = (as_value mem voffset : string) ]}
The function is useful for initializing shared memory as string
so that several processes can directly access the string.
The string has the GC color [White].
Raises [Out_of_space] if the memory block is too small.
*)
val init_string_bytelen : int -> int
(** Returns [bytelen] if [init_string] was called with the passed
[len].
*)
val init_array : memory -> int -> int -> (int * int)
(** [let voffset, bytelen = init_array mem offset size]:
Initializes the memory at [offset]
and following bytes as Ocaml array with [size] elements.
Returns in [voffset] the offset where the value starts
(i.e. [offset] plus one word), and in [bytelen] the number
of bytes used in [mem].
The array cannot be used as float array.
[offset] must be a multiple of the word size in bytes.
The array can be accessed with
{[ let a = (as_value mem voffset : _ array) ]}
The elements of the array have a value but it might not be
valid for the element type of the array. Because of this, it
is unwise to access the elements before setting them for the
first time.
The array has the GC color [White].
Raises [Out_of_space] if the memory block is too small.
*)
val init_float_array : memory -> int -> int -> (int * int)
(** Same for arrays of floats *)
val init_array_bytelen : int -> int
(** Returns [bytelen] if [init_array] was called with the passed
[size].
*)
val init_float_array_bytelen : int -> int
(** Same for arrays of floats *)
type custom_ops = nativeint
type init_value_flag =
| Copy_bigarray
| Copy_custom_int
| Copy_atom
| Copy_simulate
| Copy_conditionally
| Keep_atom
val init_value :
?targetaddr:nativeint ->
?target_custom_ops:(string * custom_ops) list ->
?cc:(nativeint * nativeint) list ->
memory -> int -> 'a -> init_value_flag list -> (int * int)
(** [let voffset, bytelen = init_value mem offset v flags]:
Initializes the memory at [offset] and following bytes as
copy of the boxed value [v].
Returns in [voffset] the offset where the value starts
(i.e. [offset] plus one word), and in [bytelen] the number
of bytes used in [mem].
The copied value can then be accessed with
{[ let v' = (as_value mem voffset : 'a) ]}
[offset] must be a multiple of the word size in bytes.
The input value [v] must be heap-allocated. Also, a number of
restrictions and caveats apply:
- Objects, closures, and lazy values are not supported
- Bigarrays are only supported if the [Copy_bigarray] flag
is given. In this case, a copy of the bigarray is also made
and appended to the value copy (i.e. it is also placed into
the buffer [mem]).
- Abstract and custom values need to be enabled. For
[int32], [int64], and [nativeint] the flag [Copy_custom_int]
enables copying, and for bigarrays the flag [Copy_bigarray].
Generally, there is a function pointer in such data blocks which
might be invalid when the memory buffer is loaded into a
different executable. This specific problem can be fixed
by passing [target_custom_ops] with the right pointers.
- Atoms (i.e. zero-sized blocks such as empty arrays) are only
supported if the [Copy_atom] or [Keep_atom] flags are present,
otherwise the function fails. [Keep_atom] means here to keep
atoms as-is. This is correct, but also keeps references to
the atom definitions which live outside [mem]. [Copy_atom] means to
create a copy of the atom as a zero-sized block outside the
atom table. This way the value in [mem] is self-contained,
but this unfortunately breaks some assumptions of the
OCaml code generator. In particular, comparisons like
[if array=[| |] then...] may yield wrong results.
- The input value may reside outside the Ocaml heap. This may break
badly written C wrappers that do not use abstract or custom
tags to mark foreign data.
The function raises [Out_of_space] if the memory block is too small.
Cyclic input values are supported, and value sharing is kept intact.
If the [Copy_simulate] flag is given, [mem] is not modified.
In simulation mode, it is pretended that [mem] is as large
as necessary to hold the value, no matter how large [mem] really
is. The returned values [voffset] and [bytelen] reflect how much
of [mem] would have been used.
If the [targetaddr] argument is passed, it is assumed that the
memory block is mapped at this address and not at the address it
is really mapped. This is useful for preparing memory that is going
to be mapped at a different address than it is right now.
The new value has the GC color [White].
If bigarrays are copied, the copy also includes the data part.
The data part is directly following the bigarray block, and is
represented in a special implementation-defined way.
If the [Copy_conditionally] flag is set, the condition [cc] is
evaluated for every block, and only if [cc] returns true, the block
is copied. [cc] is a list of addresses [(start,end)], and a block
is not copied if its address lies in any of these address ranges.
Otherwise the block is copied. As an exception of the foregoing,
the first block (i.e. [v]) is always copied.
*)
val get_custom_ops : 'a -> (string * custom_ops)
(** Returns custom ops for a sample value (or [Invalid_argument]) *)
val copy_value : init_value_flag list -> 'a -> 'a
(** [copy_value flags v]: Creates a deep copy of [v] and returns it.
The copy is allocated in the normal Ocaml heap.
Restrictions:
- Objects, closures, and lazy values are not supported (FIXME)
- Bigarrays are only supported if the [Copy_bigarray] flag
is given. In this case, a copy of bigarrays are also made,
and placed into additional buffers obtained via [stat_alloc].
- Abstract and custom values need to be enabled. For
[int32], [int64], and [nativeint] the flag [Copy_custom_int]
enables copying, and for bigarrays the flag [Copy_bigarray].
- Atoms are automatically fixed. [Copy_atoms] is ignored.
Cyclic input values are supported. [Copy_simulate] is ignored.
*)
type color = White | Gray | Blue | Black
(** GC colors *)
val color : Obj.t -> color
(** Return the GC color *)
val set_color : Obj.t -> color -> unit
(** Set the GC color *)
val is_bigarray : Obj.t -> bool
(** Checks whether the objects ia actually a bigarray *)
(** {2 I/O using [memory] as buffers} *)
val mem_read : Unix.file_descr -> memory -> int -> int -> int
(** A version of [Unix.read] that uses a [memory] buffer.
Some OS allow faster I/O when [memory] is page-aligned
(see [alloc_memory_pages]). Also, a copy in the stub function
can be avoided. Both effects can result in a considerable speedup.
*)
val mem_write : Unix.file_descr -> memory -> int -> int -> int
(** A version of [Unix.single_write] that uses a [memory] buffer. *)
val mem_recv : Unix.file_descr -> memory -> int -> int -> Unix.msg_flag list ->
int
val mem_send : Unix.file_descr -> memory -> int -> int -> Unix.msg_flag list ->
int
(** Versions of [Unix.recv], and [Unix.send]
using [memory] buffers.
*)
(* N.B. recvfrom, sendto missing because of difficulties accessing sockaddr
from C
*)
(** {2 Buffer pools} *)
type memory_pool
(** A pool of [memory] blocks that are all the same size and page-aligned
(if the OS supports this). The pool tries to bundle memory allocations
so that not for every block a system call is required. This reduces
the number of system calls, and the number of entries in the process
page table. Also, unused blocks are automatically returned to the
pool.
*)
val create_pool : int -> memory_pool
(** Create a new pool. The argument is the size of the memory blocks
(must be a multiple of the page size)
*)
val pool_alloc_memory : memory_pool -> memory
(** [let m = pool_alloc_memory p]:
Gets a memory block [m] from the pool [p]. If required, new blocks are
automatically allocated and added to the pool. This function is
thread-safe.
The memory block is automatically garbage-collected.
*)
val pool_alloc_memory2 : memory_pool -> (memory * (unit->unit))
(** [let m, free = pool_alloc_memory2 p]:
Gets a memory block [m] from the pool [p] like [pool_alloc_memory].
This function also returns the function [free] marking the
block as free again. The block can then be immediately recycled
for another use.
If [free] is not called, the block [m] is first recycled when it
is not referenced any more (like in [pool_alloc_memory]).
*)
val pool_reclaim : memory_pool -> unit
(** Reclaim as much memory as possible *)
val pool_block_size : memory_pool -> int
(** Returns the size of the memory blocks in bytes *)
val default_block_size : int
(** The default block size, normally 64 K (or better, 16 times the
page size)
*)
val default_pool : memory_pool
(** The default pool with the default block size. This pool is used
by Ocamlnet itself as much as possible
*)
val small_block_size : int
(** The block size of [small_pool], normally 4K (or better, the
page size)
*)
val small_pool : memory_pool
(** Another standard pool where the blocks are smaller than in
[default_pool].
*)
val pool_report : memory_pool -> string
(** Returns a report describing the memory allocation in the pool *)
