blob: 87e779ed6884198f1758777927cf3da53a0771d7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
open Unix
let (@@) a b = a b
type id = int * int * int
let new_id () : id =
( Random.int 10000000,
Random.int 10000000,
Random.int 10000000 )
let id_str (a, b, c) = Format.sprintf "%d.%d.%d" a b c
(* make_addr : string -> int -> sockaddr *)
let make_addr host port =
let host = gethostbyname host in
ADDR_INET(host.h_addr_list.(Random.int (Array.length host.h_addr_list)), port)
(* Unmarshal ONE message from a file descriptor, and DO NOT buffer more data *)
let read_one_msg fd =
let hdr = String.create Marshal.header_size in
assert (read fd hdr 0 Marshal.header_size = Marshal.header_size);
let dlen = Marshal.data_size hdr 0 in
let data = String.create dlen in
let rl = ref 0 in
while !rl < dlen do
rl := !rl + read fd data !rl (dlen - !rl)
done;
Marshal.from_string (hdr ^ data) 0
|
