Module Iobuf

A non-moving (in the GC sense) contiguous range of bytes, useful for I/O operations.

An iobuf consists of:

All iobuf operations are restricted to operate within the limits. Initially, the window of an iobuf is identical to its limits. A phantom type, the "seek" permission, controls whether or not code is allowed to change the limits and window. With seek permission, the limits can be narrowed, but can never be widened, and the window can be set to an arbitrary subrange of the limits.

A phantom type controls whether code can read and write bytes in the bigstring (within the limits) or can only read them.

To present a restricted view of an iobuf to a client, one can create a sub-iobuf or add a type constraint.

Functions operate on the window unless the documentation or naming indicates otherwise.

type nonrec seek = Iobuf__.Iobuf_intf.seek
val sexp_of_seek : seek -> Ppx_sexp_conv_lib.Sexp.t
type nonrec no_seek = Iobuf__.Iobuf_intf.no_seek
val sexp_of_no_seek : no_seek -> Ppx_sexp_conv_lib.Sexp.t
type t_repr

This type is a compiler witness that 'rw and 'seek do not affect layout; it enables wider use of unboxed GADTs.

type (-'data_perm_read_write, +'seek_permission) t = private t_repr

The first type parameter controls whether the iobuf can be written to. The second type parameter controls whether the window and limits can be changed.

See the Perms module for information on how the first type parameter is used.

To allow no_seek or seek access, a function's type uses _ rather than no_seek as the type argument to t. Using _ allows the function to be directly applied to either permission. Using a specific permission would require code to use coercion :>.

There is no t_of_sexp. One should use Iobuf.Hexdump.t_of_sexp or @sexp.opaque as desired.

type ('rw, 'seek) t_with_shallow_sexp = ('rw'seek) t

t_with_shallow_sexp has a sexp_of that shows the windows and limits of the underlying bigstring, but no data. We do this rather than deriving sexp_of on t because it is much more likely to be noise than useful information, and so callers should probably not display the iobuf at all.

val sexp_of_t_with_shallow_sexp : ('rw -> Ppx_sexp_conv_lib.Sexp.t) -> ('seek -> Ppx_sexp_conv_lib.Sexp.t) -> ('rw'seek) t_with_shallow_sexp -> Ppx_sexp_conv_lib.Sexp.t
include Core_kernel.Invariant.S2 with type ('rw, 'seek) t := ('rw'seek) t
type ('a, 'b) t
val invariant : 'a Base__.Invariant_intf.inv -> 'b Base__.Invariant_intf.inv -> ('a'b) t Base__.Invariant_intf.inv
module Window : Core_kernel.Hexdump.S2 with type ('rw, 'seek) t := ('rw'seek) t

Provides a Window.Hexdump submodule that renders the contents of t's window.

module Limits : Core_kernel.Hexdump.S2 with type ('rw, 'seek) t := ('rw'seek) t

Provides a Limits.Hexdump submodule that renders the contents of t's limits.

Provides a Hexdump submodule that renders the contents of t's window and limits using indices relative to the limits.

include Iobuf__.Iobuf_intf.Compound_hexdump with type ('rw, 'seek) t := ('rw'seek) t
type ('rw, 'seek) t
module Hexdump : sig ... end
module Debug : Iobuf__.Iobuf_intf.Compound_hexdump with type ('rw, 'seek) t := ('rw'seek) t

Provides a Debug.Hexdump submodule that renders the contents of t's window, limits, and underlying bigstring using indices relative to the bigstring.

Creation

val create : len:int -> (__) t

create ~len creates a new iobuf, backed by a bigstring of length len, with the limits and window set to the entire bigstring.

val of_bigstring : ?⁠pos:int -> ?⁠len:int -> Core_kernel.Bigstring.t -> ([< Core_kernel.read_write ]_) t

of_bigstring bigstring ~pos ~len returns an iobuf backed by bigstring, with the window and limits specified starting at pos and of length len.

forbid immutable to prevent aliasing

val of_string : string -> (__) t

of_string s returns a new iobuf whose contents are s.

val sub_shared : ?⁠pos:int -> ?⁠len:int -> ('d_) t -> ('d_) t

sub_shared t ~pos ~len returns a new iobuf with limits and window set to the subrange of t's window specified by pos and len. sub_shared preserves data permissions, but allows arbitrary seek permissions on the resulting iobuf.

val copy : (__) t -> (__) t

copy t returns a new iobuf whose contents are the same as those in the window of t.

val clone : (__) t -> (__) t

clone t returns a new iobuf that is a deep-copy of t including an exact copy of the underlying buffer and bounds. This means data outside the window is copied as well.

val set_bounds_and_buffer : src:([> Core_kernel.write ] as 'data, _) t -> dst:('dataseek) t -> unit

set_bounds_and_buffer ~src ~dst copies bounds metadata (i.e., limits and window) and shallowly copies the buffer (data pointer) from src to dst. It does not access data, but does allow access through dst. This makes dst an alias of src.

Because set_bounds_and_buffer creates an alias, we disallow immutable src and dst using [> write]. Otherwise, one of src or dst could be read_write :> read and the other immutable :> read, which would allow you to write the immutable alias's data through the read_write alias.

set_bounds_and_buffer is typically used with a frame iobuf that need only be allocated once. This frame can be updated repeatedly and handed to users, without further allocation. Allocation-sensitive applications need this.

val set_bounds_and_buffer_sub : pos:int -> len:int -> src:([> Core_kernel.write ] as 'data, _) t -> dst:('dataseek) t -> unit

set_bounds_and_buffer_sub ~pos ~len ~src ~dst is a more efficient version of set_bounds_and_buffer ~src:(Iobuf.sub_shared ~pos ~len src) ~dst.

set_bounds_and_buffer ~src ~dst is not the same as set_bounds_and_buffer_sub ~dst ~src ~len:(Iobuf.length src) because the limits are narrowed in the latter case.

~len and ~pos are mandatory for performance reasons, in concert with @@inline. If they were optional, allocation would be necessary when passing a non-default, non-constant value, which is an important use case.

val read_only : ([> Core_kernel.read ]'s) t -> (Core_kernel.read's) t

Generalization

One may wonder why you'd want to call no_seek, given that a cast is already possible, e.g., t : (_, seek) t :> (_, no_seek) t. It turns out that if you want to define some f : (_, _) t -> unit of your own that can be conveniently applied to seek iobufs without the user having to cast seek up, you need this no_seek function.

read_only is more of a historical convenience now that read_write is a polymorphic variant, as one can now explicitly specify the general type for an argument with something like t : (_ perms, _) t :> (read, _) t.

val no_seek : ('r_) t -> ('rno_seek) t

Accessors

val capacity : (__) t -> int

capacity t returns the size of t's limits subrange. The capacity of an iobuf can be reduced via narrow.

val length : (__) t -> int

length t returns the size of t's window.

val length_lo : (__) t -> int

length_lo t returns the length that t's window would have after calling flip_lo, without actually changing the window. This is the number of bytes between the lower limit and the start of the window.

When you're writing to the window, you can think of this as the number of bytes already written. When reading from the window, this can mean the number of bytes already consumed.

This is equivalent to:

Iobuf.Expert.(lo t - lo_min t)

.

val length_hi : (__) t -> int

length_hi t returns the length that t's window would have after calling flip_hi, without actually changing the window. This is the number of bytes between the end of the window and the upper limit of the buffer.

This is equivalent to:

Iobuf.Expert.(hi_max t - hi t) 

.

val is_empty : (__) t -> bool

is_empty t is length t = 0.

Changing the limits

val narrow : (_seek) t -> unit

narrow t sets t's limits to the current window.

val narrow_lo : (_seek) t -> unit

narrow_lo t sets t's lower limit to the beginning of the current window.

val narrow_hi : (_seek) t -> unit

narrow_hi t sets t's upper limit to the end of the current window.

Comparison

val memcmp : (__) t -> (__) t -> int

memcmp a b first compares the length of a and b's windows and then compares the bytes in the windows for equivalence.

Changing the window

module type Bound = Iobuf__.Iobuf_intf.Bound with type ('d, 'w) iobuf := ('d'w) t
module Lo_bound : Bound
module Hi_bound : Bound
val advance : (_seek) t -> int -> unit

advance t amount advances the lower bound of the window by amount. It is an error to advance past the upper bound of the window or the lower limit.

val unsafe_advance : (_seek) t -> int -> unit

unsafe_advance is like advance but with no bounds checking, so incorrect usage can easily cause segfaults.

val resize : (_seek) t -> len:int -> unit

resize t sets the length of t's window, provided it does not exceed limits.

val unsafe_resize : (_seek) t -> len:int -> unit

unsafe_resize is like resize but with no bounds checking, so incorrect usage can easily cause segfaults.

val rewind : (_seek) t -> unit

rewind t sets the lower bound of the window to the lower limit.

val reset : (_seek) t -> unit

reset t sets the window to the limits.

val flip_lo : (_seek) t -> unit

flip_lo t sets the window to range from the lower limit to the lower bound of the old window. This is typically called after a series of Fills, to reposition the window in preparation to Consume the newly written data.

The bounded version narrows the effective limit. This can preserve some data near the limit, such as a hypothetical packet header (in the case of bounded_flip_lo) or unfilled suffix of a buffer (in bounded_flip_hi).

val bounded_flip_lo : (_seek) t -> Lo_bound.t -> unit
val compact : (Core_kernel.read_writeseek) t -> unit

compact t copies data from the window to the lower limit of the iobuf and sets the window to range from the end of the copied data to the upper limit. This is typically called after a series of Consumes to save unread data and prepare for the next series of Fills and flip_lo.

val bounded_compact : (Core_kernel.read_writeseek) t -> Lo_bound.t -> Hi_bound.t -> unit
val flip_hi : (_seek) t -> unit

flip_hi t sets the window to range from the the upper bound of the current window to the upper limit. This operation is dual to flip_lo and is typically called when the data in the current (narrowed) window has been processed and the window needs to be positioned over the remaining data in the buffer. For example:

(* ... determine initial_data_len ... *)
Iobuf.resize buf ~len:initial_data_len;
(* ... and process initial data ... *)
Iobuf.flip_hi buf;

Now the window of buf ranges over the remainder of the data.

val bounded_flip_hi : (_seek) t -> Hi_bound.t -> unit
val protect_window_and_bounds : ('rwno_seek) t -> f:(('rwseek) t -> 'a) -> 'a

protect_window_and_bounds t ~f calls f t with t's bounds set to its current window, and restores t's window, bounds, and buffer afterward.

val protect_window_and_bounds_1 : ('rwno_seek) t -> 'a -> f:(('rwseek) t -> 'a -> 'b) -> 'b

protect_window_and_bounds_1 t x ~f is a more efficient version of protect_window_and_bounds t ~f:(fun t -> f t x).

Getting and setting data

"consume" and "fill" functions access data at the lower bound of the window and advance the lower bound of the window. "peek" and "poke" functions access data but do not advance the window.

val to_string : ?⁠len:int -> ([> Core_kernel.read ]_) t -> string

to_string t returns the bytes in t as a string. It does not alter the window.

val to_string_hum : ?⁠max_lines:int -> (__) t -> string

Equivalent to Hexdump.to_string_hum. Renders t's windows and limits.

val to_bytes : ?⁠len:int -> (__) t -> Core_kernel.Bytes.t

to_bytes t returns the bytes in t as a bytes. It does not alter the window.

val of_bytes : Core_kernel.Bytes.t -> (__) t

of_bytes b returns a new iobuf whose contents is b.

module Consume : sig ... end

Consume.string t ~len reads len characters (all, by default) from t into a new string and advances the lower bound of the window accordingly.

module Fill : sig ... end

Fill.bin_prot X.bin_write_t t x writes x to t in bin-prot form, advancing past the bytes written.

module Peek : sig ... end

Peek and Poke functions access a value at pos from the lower bound of the window and do not advance.

module Poke : sig ... end

Poke.bin_prot X.bin_write_t t x writes x to the beginning of t in binary form without advancing. You can use X.bin_size_t to tell how long it was. X.bin_write_t is only allowed to write that portion of the buffer you have access to.

module Unsafe : sig ... end

Unsafe has submodules that are like their corresponding module, except with no range checks. Hence, mistaken uses can cause segfaults. Be careful!

val bin_prot_length_prefix_bytes : int

The number of bytes in the length prefix of consume_bin_prot and fill_bin_prot.

val fill_bin_prot : ([> Core_kernel.write ]seek) t -> 'a Bin_prot.Type_class.writer -> 'a -> unit Core_kernel.Or_error.t

fill_bin_prot writes a bin-prot value to the lower bound of the window, prefixed by its length, and advances by the amount written. fill_bin_prot returns an error if the window is too small to write the value.

consume_bin_prot t reader reads a bin-prot value from the lower bound of the window, which should have been written using fill_bin_prot, and advances the window by the amount read. consume_bin_prot returns an error if there is not a complete message in the window and in that case the window is left unchanged.

Don't use these without a good reason, as they are incompatible with similar functions in Reader and Writer. They use a 4-byte length rather than an 8-byte length.

val consume_bin_prot : ([> Core_kernel.read ]seek) t -> 'a Bin_prot.Type_class.reader -> 'a Core_kernel.Or_error.t
module Blit : sig ... end

Blit copies between iobufs and advances neither src nor dst.

module Blit_consume : sig ... end

Blit_consume copies between iobufs and advances src but does not advance dst.

module Blit_fill : sig ... end

Blit_fill copies between iobufs and advances dst but does not advance src.

module Blit_consume_and_fill : sig ... end

Blit_consume_and_fill copies between iobufs and advances both src and dst.

val memset : (Core_kernel.read_write_) t -> pos:int -> len:int -> char -> unit

memset t ~pos ~len c fills t with c within the range [pos, pos + len).

val zero : (Core_kernel.read_write_) t -> unit

memsets a buffer to zero.

Expert

module Expert : sig ... end

The Expert module is for building efficient out-of-module Iobuf abstractions.

type nonrec ('src, 'dst) consuming_blito = ('src'dst) Iobuf__.Iobuf_intf.consuming_blito