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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
/**************************************************************************/
/* */
/* OCaml */
/* */
/* Xavier Leroy and Pascal Cuoq, projet Cristal, INRIA Rocquencourt */
/* */
/* Copyright 1996 Institut National de Recherche en Informatique et */
/* en Automatique. */
/* */
/* All rights reserved. This file is distributed under the terms of */
/* the GNU Lesser General Public License version 2.1, with the */
/* special exception on linking described in the file LICENSE. */
/* */
/**************************************************************************/
#include <caml/mlvalues.h>
#include <caml/memory.h>
#include "unixsupport.h"
int caml_unix_socket_domain_table[] = {
PF_UNIX, PF_INET, PF_INET6
};
int caml_unix_socket_type_table[] = {
SOCK_STREAM, SOCK_DGRAM, SOCK_RAW, SOCK_SEQPACKET
};
SOCKET caml_win32_socket(int domain, int type, int protocol,
LPWSAPROTOCOL_INFO info,
BOOL inherit)
{
SOCKET s;
DWORD flags = WSA_FLAG_OVERLAPPED;
#ifndef WSA_FLAG_NO_HANDLE_INHERIT
#define WSA_FLAG_NO_HANDLE_INHERIT 0x80
#endif
if (! inherit)
flags |= WSA_FLAG_NO_HANDLE_INHERIT;
s = WSASocket(domain, type, protocol, info, 0, flags);
if (s == INVALID_SOCKET) {
if (! inherit && WSAGetLastError() == WSAEINVAL) {
/* WSASocket probably doesn't support WSA_FLAG_NO_HANDLE_INHERIT,
* retry without. */
flags &= ~(DWORD)WSA_FLAG_NO_HANDLE_INHERIT;
s = WSASocket(domain, type, protocol, info, 0, flags);
if (s == INVALID_SOCKET)
goto err;
caml_win32_set_inherit((HANDLE) s, FALSE);
return s;
}
goto err;
}
return s;
err:
caml_win32_maperr(WSAGetLastError());
return INVALID_SOCKET;
}
CAMLprim value caml_unix_socket(value cloexec, value domain, value type,
value proto)
{
CAMLparam4(cloexec, domain, type, proto);
CAMLlocal1(v_socket);
SOCKET s;
s = caml_win32_socket(caml_unix_socket_domain_table[Int_val(domain)],
caml_unix_socket_type_table[Int_val(type)],
Int_val(proto),
NULL,
! caml_unix_cloexec_p(cloexec));
if (s == INVALID_SOCKET)
caml_uerror("socket", Nothing);
v_socket = caml_win32_alloc_socket(s);
CAMLreturn(v_socket);
}
|
