-
Notifications
You must be signed in to change notification settings - Fork 33
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #19 from c-cube/rfc-io
modular IO
- Loading branch information
Showing
1 changed file
with
248 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,248 @@ | ||
# modular IO channels in the stdlib | ||
|
||
This RFC proposes to update the stdlib's types `in_channel` and `out_channel` to make them user-definable and composable. | ||
|
||
Even though a lot of networked applications use lwt or async to achieve high levels of concurrency, classic blocking IO still has its uses. However, the standard OCaml channels suffer from some painful limitations: | ||
|
||
- they cannot be created outside the stdlib, which means the only thing we can manipulate through them is sockets and other unix file descriptors. | ||
- they cannot be composed. In other languages [such as Go](https://golang.org/pkg/io/#Reader), one can write reader or writer combinators which transform the bytestream | ||
that is written or read. Typical examples would include (de)compression and encryption. | ||
- there is some duplication in `Printf`; namely, the existence, and incompatibility, of `bprintf`, `sprintf`, and `fprintf`. This makes `Printf` printers artificially limited | ||
since their type determines what kind of output they can produce. In my own experience, `Format` is a better choice because a single `Format.formatter -> t -> unit` function | ||
can be used in more cases than any single `Printf` function, despite the overhead of formatters. | ||
- input channels provide an API that is unsuited to some forms of parsing such | ||
as reading line by line. The stdlib's `input_line` has to cheat and use | ||
a special C primitive to know how much input to consume (by looking for `'\n'` | ||
in the underlying buffer). | ||
|
||
As a consequence, many libraries have their own opaque channel types that | ||
are not compatible with the standard ones. That's a missed opportunity for | ||
code reuse and composability. | ||
|
||
## Proof of concept | ||
|
||
A proof of concept is being developped at https://github.com/c-cube/poc-modular-io . | ||
|
||
It features the improved interface from below, extensible `In.t` and `Out.t` | ||
types, and a bunch of combinators written against the *public* interface, | ||
including `read_line`, http1.1 chunked encoding, char-by-char mapping, etc. | ||
|
||
## Extensibility | ||
|
||
The current types are implemented in C and are opaque. I propose that it be changed for: | ||
|
||
```ocaml | ||
type in_channel = | ||
| IC_raw of old_in_channel (* implemented in C *) | ||
| IC_user of { | ||
read: bytes -> int -> int -> int; | ||
close: unit -> unit; | ||
} | ||
type out_channel = | ||
| OC_raw of old_out_channel (* implemented in C *) | ||
| OC_user of { | ||
write: bytes -> int -> int -> int; | ||
flush: unit -> unit; | ||
close: unit -> unit; | ||
} | ||
(* now doable in userland: *) | ||
val ic_of_string : string -> in_channel | ||
val unzip : in_channel -> in_channel | ||
val oc_of_buf : Buffer.t -> out_channel | ||
val zip : out_channel -> out_channel | ||
val encrypt_rot13: out_channel -> out_channel | ||
(* Write to both channels *) | ||
val tee : out_channel -> out_channel -> out_channel | ||
``` | ||
|
||
**Aside**: In fact, I'm not sure why the channels are still implemented in C. I think the base case could be | ||
a raw Unix file descriptor and a `bytes` buffer — but maybe this is needed for portability. | ||
|
||
This change would dramatically improve compositionality, as one could then: | ||
|
||
- implement `Printf.{b,s}printf` in terms of `Printf.fprintf` (which would be | ||
the most general one of the three); | ||
- compose transformations on channels, such as encoding, http chunking, encryption, | ||
compression; | ||
- use APIs that only operate on `in_channel` with strings. | ||
|
||
### Backward compatibility | ||
|
||
Functions in `Unix` that map file descriptors to/from channels would become | ||
partial (i.e. they wouldn't work on user-defined channels). The `Unix.seek` | ||
function would not work on user defined channels, but it | ||
is already partial anyway (because of sockets). | ||
|
||
A function `{in,out}_channel_has_descr : {in,out}_channel -> bool` would help | ||
know what channels correspond to unix file descriptors. Alternatively, **if** | ||
the sum type is made public, a mere pattern matching can do. | ||
|
||
## Interface improvement for `in_channel` | ||
|
||
The current interface of `in_channel` provides, roughly, `input : bytes -> int -> int -> int` | ||
which takes a byte slice and returns how many bytes were read, `0` indicating end of input. | ||
This interface doesn't expose the underlying buffer and instead imitates the lower level posix APIs. | ||
|
||
The problem of this interface is that it makes some functions quite awkward to write, | ||
and hurts compositionality. | ||
An interesting alternative is [rust's `BufRead` interface](https://doc.rust-lang.org/std/io/trait.BufRead.html). | ||
In OCaml, that corresponds roughly to: | ||
|
||
```ocaml | ||
module In_channel : sig | ||
type t | ||
(** Obtain a slice of the current buffer. Empty iff EOF was reached *) | ||
val fill_buf : t -> (bytes * int * int) | ||
(** Consume n bytes from the input. *) | ||
val consume : t -> int -> unit | ||
(** Close channel and release resources *) | ||
val close : t -> unit | ||
end | ||
``` | ||
|
||
The semantics of these operations is: | ||
|
||
- `fill_buf ic` ensures that the channel's internal buffer is non-empty, unless | ||
end-of-input was reached. Then, it exposes a view of the internal buffer. | ||
The important aspect of this is that successive calls to `fill_buf` return | ||
the same result; this doesn't consume input on a logical level. This function | ||
just exposes a slice of the input. | ||
- `consume ic n` eats `n` bytes of the input. It must only be called after | ||
`fill_buf` returned a slice of length at least `n`. If the whole slice | ||
exposed by `fill_buf` was consumed, then the channel will have to read more | ||
from its underlying stream at the next call to `fill_buf`. | ||
- `close ic` closes the channel and releases underlying resources. | ||
|
||
### Advantages | ||
|
||
This interface is easier to use than the current `input` interface, especially when | ||
parsing formats with non-trivial framing (e.g. http1.1). One typically wants | ||
to read a line to get headers and framing (content-length) information, followed | ||
by a read of `n` bytes. It is therefore important to read the line(s) efficiently | ||
but without consuming _too much_ from the input buffer as it's possibly part | ||
of the payload. | ||
|
||
Compare the stdlib's [`input_line` implementation](https://github.com/ocaml/ocaml/blob/f333db8b0f176b1d75e6fdb46a97a78995426ed7/stdlib/stdlib.ml#L439) | ||
which uses a magical external to look inside the C buffer, with this snippet | ||
(adapted from [tiny httpd](https://github.com/c-cube/tiny_httpd/blob/3ac5510e2d5dfcdf448a03a99c0c178b73afeabd/src/Tiny_httpd.ml#L159)): | ||
|
||
```ocaml | ||
let input_line (ic:In_channel.t) : String.t = | ||
let buf = Buffer.create 32 in | ||
let continue = ref true in | ||
while !continue do | ||
let s, i, len = In_channel.fill_buf ic in | ||
if len=0 then ( | ||
continue := false; | ||
if Buffer.length buf = 0 then raise End_of_file; | ||
); | ||
let j = ref i in | ||
(* look for ['\n'] in the input buffer *) | ||
while !j < i+len && Bytes.get s !j <> '\n' do | ||
incr j | ||
done; | ||
if !j-i < len then ( | ||
assert (Bytes.get s !j = '\n'); | ||
Buffer.add_bytes buf s i (!j-i); (* without '\n' *) | ||
In_channel.consume ic (!j-i+1); (* consume rest of line + '\n' *) | ||
continue := false | ||
) else ( | ||
Buffer.add_bytes buf s i len; | ||
In_channel consume ic len; | ||
) | ||
done; | ||
Buffer.contents buf | ||
``` | ||
|
||
In the stdlib implementation, the external | ||
[`input_scan_line`](https://github.com/ocaml/ocaml/blob/f333db8b0f176b1d75e6fdb46a97a78995426ed7/stdlib/stdlib.ml#L437) | ||
is used to peek inside the `in_channel`'s buffer, breaking the abstraction of `input`. | ||
This shows that peeking into the buffer without consuming it is quite necessary | ||
in practice. | ||
|
||
### Compatibility | ||
|
||
The current type of channels could retain its interface, for backward compatibility, | ||
in addition to the new interface which exposes `consume` and `fill_buf`, | ||
but implement `input`, in the general case, as follows | ||
(adapted from [tiny httpd](https://github.com/c-cube/tiny_httpd/blob/3ac5510e2d5dfcdf448a03a99c0c178b73afeabd/src/Tiny_httpd.ml#L146)): | ||
|
||
```ocaml | ||
let input (ic:In_channel.t) buf i len : int = | ||
let offset = ref 0 in | ||
let continue = ref true in | ||
while continue && !offset < len do | ||
let s, j, n = In_channel.fill_buf ic () in | ||
let n_read = min n (len - !offset) in | ||
Bytes.blit s j buf (i + !offset) n_read; | ||
offset := !offset + n_read; | ||
In_channel.consume ic n_read; | ||
if n_read=0 then continue := false; (* eof *) | ||
done; | ||
!offset | ||
``` | ||
|
||
In most cases this should only do one iteration if `n` is smaller than the | ||
underlying buffer's size. | ||
|
||
**Alternatively**, this can be considered the implementation of `really_input`, | ||
and have input be just: | ||
|
||
```ocaml | ||
let input (ic:In_channel.t) buf i len : int = | ||
let s, j, n = In_channel.fill_buf ic in | ||
let n_read = min n len in | ||
Bytes.blit s j buf i n_read; | ||
In_channel.consume ic n_read; | ||
n_read | ||
``` | ||
|
||
Here we see that the classic `input` is simply the successive application | ||
of `fill_buf` and `consume`. | ||
|
||
|
||
### Relation to the extensibility aspect | ||
|
||
This is compatible with the extensibility approach from above. The new API | ||
would be: | ||
|
||
```ocaml | ||
type in_channel = | ||
| IC_raw of old_in_channel (* implemented in C *) | ||
| IC_user of { | ||
fill_buf: unit -> (bytes * int * int); | ||
consume: int -> unit; | ||
close: unit -> unit; | ||
} | ||
``` | ||
|
||
### Out channel | ||
|
||
There is no equivalent need to modify the interface of `out_channel`. Buffered | ||
output is simpler as one doesn't need to look inside the buffer at all. | ||
|
||
### Summary | ||
|
||
This change would improve the API of input channels, making them more flexible | ||
for some use cases that involve dynamic framing of input. Examples include http1 | ||
(as well as its chunked encoding), the Redis protocol, and netstrings. | ||
|
||
## Related interfaces | ||
|
||
- Batteries has [BatIO](http://ocaml-batteries-team.github.io/batteries-included/hdoc2/BatIO.html) | ||
which also contains an extensible type for channels. | ||
However, it's not compatible with the stdlib's channels, so projects cannot | ||
export such channels in their APIs unless they force the batteries dependency. | ||
- ocamlnet has [Netchannels](http://projects.camlcity.org/projects/dl/ocamlnet-4.1.6/doc/html-main/Netchannels_tut.html) | ||
which is also extensible. | ||
|