3 * (c) 1995 Microsoft Corporation. All rights reserved.
4 * Developed by hip communications inc., http://info.hip.com/info/
5 * Portions (c) 1993 Intergraph Corporation. All rights reserved.
7 * You may distribute under the terms of either the GNU General Public
8 * License or the Artistic License, as specified in the README file.
12 #define WIN32_LEAN_AND_MEAN
15 #include <sys/socket.h>
22 #ifdef USE_SOCKETS_AS_HANDLES
23 /* thanks to Beverly Brown (beverly@datacube.com) */
25 #define OPEN_SOCKET(x) _open_osfhandle(x,_O_RDWR|_O_BINARY)
26 #define TO_SOCKET(x) _get_osfhandle(x)
30 # define OPEN_SOCKET(x) (x)
31 # define TO_SOCKET(x) (x)
33 #endif /* USE_SOCKETS_AS_HANDLES */
35 static struct servent* win32_savecopyservent(struct servent*d,
38 #define SOCKETAPI PASCAL
40 typedef SOCKET (SOCKETAPI *LPSOCKACCEPT)(SOCKET, struct sockaddr *, int *);
41 typedef int (SOCKETAPI *LPSOCKBIND)(SOCKET, const struct sockaddr *, int);
42 typedef int (SOCKETAPI *LPSOCKCLOSESOCKET)(SOCKET);
43 typedef int (SOCKETAPI *LPSOCKCONNECT)(SOCKET, const struct sockaddr *, int);
44 typedef int (SOCKETAPI *LPSOCKIOCTLSOCKET)(SOCKET, long, u_long *);
45 typedef int (SOCKETAPI *LPSOCKGETPEERNAME)(SOCKET, struct sockaddr *, int *);
46 typedef int (SOCKETAPI *LPSOCKGETSOCKNAME)(SOCKET, struct sockaddr *, int *);
47 typedef int (SOCKETAPI *LPSOCKGETSOCKOPT)(SOCKET, int, int, char *, int *);
48 typedef u_long (SOCKETAPI *LPSOCKHTONL)(u_long);
49 typedef u_short (SOCKETAPI *LPSOCKHTONS)(u_short);
50 typedef int (SOCKETAPI *LPSOCKLISTEN)(SOCKET, int);
51 typedef u_long (SOCKETAPI *LPSOCKNTOHL)(u_long);
52 typedef u_short (SOCKETAPI *LPSOCKNTOHS)(u_short);
53 typedef int (SOCKETAPI *LPSOCKRECV)(SOCKET, char *, int, int);
54 typedef int (SOCKETAPI *LPSOCKRECVFROM)(SOCKET, char *, int, int, struct sockaddr *, int *);
55 typedef int (SOCKETAPI *LPSOCKSELECT)(int, fd_set *, fd_set *, fd_set *, const struct timeval *);
56 typedef int (SOCKETAPI *LPSOCKSEND)(SOCKET, const char *, int, int);
57 typedef int (SOCKETAPI *LPSOCKSENDTO)(SOCKET, const char *, int, int, const struct sockaddr *, int);
58 typedef int (SOCKETAPI *LPSOCKSETSOCKOPT)(SOCKET, int, int, const char *, int);
59 typedef int (SOCKETAPI *LPSOCKSHUTDOWN)(SOCKET, int);
60 typedef SOCKET (SOCKETAPI *LPSOCKSOCKET)(int, int, int);
61 typedef char FAR *(SOCKETAPI *LPSOCKINETNTOA)(struct in_addr in);
62 typedef unsigned long (SOCKETAPI *LPSOCKINETADDR)(const char FAR * cp);
65 /* Database function prototypes */
66 typedef struct hostent *(SOCKETAPI *LPSOCKGETHOSTBYADDR)(const char *, int, int);
67 typedef struct hostent *(SOCKETAPI *LPSOCKGETHOSTBYNAME)(const char *);
68 typedef int (SOCKETAPI *LPSOCKGETHOSTNAME)(char *, int);
69 typedef struct servent *(SOCKETAPI *LPSOCKGETSERVBYPORT)(int, const char *);
70 typedef struct servent *(SOCKETAPI *LPSOCKGETSERVBYNAME)(const char *, const char *);
71 typedef struct protoent *(SOCKETAPI *LPSOCKGETPROTOBYNUMBER)(int);
72 typedef struct protoent *(SOCKETAPI *LPSOCKGETPROTOBYNAME)(const char *);
74 /* Microsoft Windows Extension function prototypes */
75 typedef int (SOCKETAPI *LPSOCKWSASTARTUP)(unsigned short, LPWSADATA);
76 typedef int (SOCKETAPI *LPSOCKWSACLEANUP)(void);
77 typedef int (SOCKETAPI *LPSOCKWSAGETLASTERROR)(void);
78 typedef int (SOCKETAPI *LPWSAFDIsSet)(SOCKET, fd_set *);
80 static HINSTANCE hWinSockDll = 0;
81 /* extern CRITICAL_SECTION csSock; */
83 static LPSOCKACCEPT paccept = 0;
84 static LPSOCKBIND pbind = 0;
85 static LPSOCKCLOSESOCKET pclosesocket = 0;
86 static LPSOCKCONNECT pconnect = 0;
87 static LPSOCKIOCTLSOCKET pioctlsocket = 0;
88 static LPSOCKGETPEERNAME pgetpeername = 0;
89 static LPSOCKGETSOCKNAME pgetsockname = 0;
90 static LPSOCKGETSOCKOPT pgetsockopt = 0;
91 static LPSOCKHTONL phtonl = 0;
92 static LPSOCKHTONS phtons = 0;
93 static LPSOCKLISTEN plisten = 0;
94 static LPSOCKNTOHL pntohl = 0;
95 static LPSOCKNTOHS pntohs = 0;
96 static LPSOCKRECV precv = 0;
97 static LPSOCKRECVFROM precvfrom = 0;
98 static LPSOCKSELECT pselect = 0;
99 static LPSOCKSEND psend = 0;
100 static LPSOCKSENDTO psendto = 0;
101 static LPSOCKSETSOCKOPT psetsockopt = 0;
102 static LPSOCKSHUTDOWN pshutdown = 0;
103 static LPSOCKSOCKET psocket = 0;
104 static LPSOCKGETHOSTBYADDR pgethostbyaddr = 0;
105 static LPSOCKGETHOSTBYNAME pgethostbyname = 0;
106 static LPSOCKGETHOSTNAME pgethostname = 0;
107 static LPSOCKGETSERVBYPORT pgetservbyport = 0;
108 static LPSOCKGETSERVBYNAME pgetservbyname = 0;
109 static LPSOCKGETPROTOBYNUMBER pgetprotobynumber = 0;
110 static LPSOCKGETPROTOBYNAME pgetprotobyname = 0;
111 static LPSOCKWSASTARTUP pWSAStartup = 0;
112 static LPSOCKWSACLEANUP pWSACleanup = 0;
113 static LPSOCKWSAGETLASTERROR pWSAGetLastError = 0;
114 static LPWSAFDIsSet pWSAFDIsSet = 0;
115 static LPSOCKINETNTOA pinet_ntoa = 0;
116 static LPSOCKINETADDR pinet_addr = 0;
118 __declspec(thread) struct servent myservent;
122 GetAddress(HINSTANCE hInstance, char *lpFunctionName)
125 FARPROC proc = GetProcAddress(hInstance, lpFunctionName);
127 sprintf(buffer, "Unable to get address of %s in WSock32.dll", lpFunctionName);
136 /* EnterCriticalSection(&csSock); */
137 if(hWinSockDll == NULL) {
138 HINSTANCE hLib = LoadLibrary("WSock32.DLL");
140 CROAK("Could not load WSock32.dll\n");
142 paccept = (LPSOCKACCEPT)GetAddress(hLib, "accept");
143 pbind = (LPSOCKBIND)GetAddress(hLib, "bind");
144 pclosesocket = (LPSOCKCLOSESOCKET)GetAddress(hLib, "closesocket");
145 pconnect = (LPSOCKCONNECT)GetAddress(hLib, "connect");
146 pioctlsocket = (LPSOCKIOCTLSOCKET)GetAddress(hLib, "ioctlsocket");
147 pgetpeername = (LPSOCKGETPEERNAME)GetAddress(hLib, "getpeername");
148 pgetsockname = (LPSOCKGETSOCKNAME)GetAddress(hLib, "getsockname");
149 pgetsockopt = (LPSOCKGETSOCKOPT)GetAddress(hLib, "getsockopt");
150 phtonl = (LPSOCKHTONL)GetAddress(hLib, "htonl");
151 phtons = (LPSOCKHTONS)GetAddress(hLib, "htons");
152 plisten = (LPSOCKLISTEN)GetAddress(hLib, "listen");
153 pntohl = (LPSOCKNTOHL)GetAddress(hLib, "ntohl");
154 pntohs = (LPSOCKNTOHS)GetAddress(hLib, "ntohs");
155 precv = (LPSOCKRECV)GetAddress(hLib, "recv");
156 precvfrom = (LPSOCKRECVFROM)GetAddress(hLib, "recvfrom");
157 pselect = (LPSOCKSELECT)GetAddress(hLib, "select");
158 psend = (LPSOCKSEND)GetAddress(hLib, "send");
159 psendto = (LPSOCKSENDTO)GetAddress(hLib, "sendto");
160 psetsockopt = (LPSOCKSETSOCKOPT)GetAddress(hLib, "setsockopt");
161 pshutdown = (LPSOCKSHUTDOWN)GetAddress(hLib, "shutdown");
162 psocket = (LPSOCKSOCKET)GetAddress(hLib, "socket");
163 pgethostbyaddr = (LPSOCKGETHOSTBYADDR)GetAddress(hLib, "gethostbyaddr");
164 pgethostbyname = (LPSOCKGETHOSTBYNAME)GetAddress(hLib, "gethostbyname");
165 pgethostname = (LPSOCKGETHOSTNAME)GetAddress(hLib, "gethostname");
166 pgetservbyport = (LPSOCKGETSERVBYPORT)GetAddress(hLib, "getservbyport");
167 pgetservbyname = (LPSOCKGETSERVBYNAME)GetAddress(hLib, "getservbyname");
168 pgetprotobynumber = (LPSOCKGETPROTOBYNUMBER)GetAddress(hLib, "getprotobynumber");
169 pgetprotobyname = (LPSOCKGETPROTOBYNAME)GetAddress(hLib, "getprotobyname");
170 pWSAStartup = (LPSOCKWSASTARTUP)GetAddress(hLib, "WSAStartup");
171 pWSACleanup = (LPSOCKWSACLEANUP)GetAddress(hLib, "WSACleanup");
172 pWSAGetLastError = (LPSOCKWSAGETLASTERROR)GetAddress(hLib, "WSAGetLastError");
173 pWSAFDIsSet = (LPWSAFDIsSet)GetAddress(hLib, "__WSAFDIsSet");
174 pinet_addr = (LPSOCKINETADDR)GetAddress(hLib,"inet_addr");
175 pinet_ntoa = (LPSOCKINETNTOA)GetAddress(hLib,"inet_ntoa");
179 /* LeaveCriticalSection(&csSock); */
185 if(hWinSockDll != NULL) {
187 FreeLibrary(hWinSockDll);
195 unsigned short version;
198 int iSockOpt = SO_SYNCHRONOUS_NONALERT;
202 * initalize the winsock interface and insure that it is
203 * cleaned up at exit.
206 if(ret = pWSAStartup(version, &retdata))
207 CROAK("Unable to locate winsock library!\n");
208 if(retdata.wVersion != version)
209 CROAK("Could not find version 1.1 of winsock dll\n");
211 /* atexit((void (*)(void)) EndSockets); */
213 #ifdef USE_SOCKETS_AS_HANDLES
215 * Enable the use of sockets as filehandles
217 psetsockopt(INVALID_SOCKET, SOL_SOCKET, SO_OPENTYPE,
218 (char *)&iSockOpt, sizeof(iSockOpt));
219 #endif /* USE_SOCKETS_AS_HANDLES */
223 #ifndef USE_SOCKETS_AS_HANDLES
225 myfdopen(int fd, char *mode)
229 int optlen = sizeof(sockbuf);
232 if (hWinSockDll == 0)
235 retval = pgetsockopt((SOCKET)fd, SOL_SOCKET, SO_TYPE, sockbuf, &optlen);
236 if(retval == SOCKET_ERROR && pWSAGetLastError() == WSAENOTSOCK) {
237 return(_fdopen(fd, mode));
241 * If we get here, then fd is actually a socket.
243 Newz(1601, fp, 1, FILE);
257 #endif /* USE_SOCKETS_AS_HANDLES */
261 win32_htonl(u_long hostlong)
266 return phtonl(hostlong);
270 win32_htons(u_short hostshort)
275 return phtons(hostshort);
279 win32_ntohl(u_long netlong)
284 return pntohl(netlong);
288 win32_ntohs(u_short netshort)
293 return pntohs(netshort);
297 #define SOCKET_TEST(x, y) if(hWinSockDll == 0) StartSockets();\
298 if((x) == (y)) errno = pWSAGetLastError()
300 #define SOCKET_TEST_ERROR(x) SOCKET_TEST(x, SOCKET_ERROR)
303 win32_accept(SOCKET s, struct sockaddr *addr, int *addrlen)
307 SOCKET_TEST((r = paccept(TO_SOCKET(s), addr, addrlen)), INVALID_SOCKET);
308 return OPEN_SOCKET(r);
312 win32_bind(SOCKET s, const struct sockaddr *addr, int addrlen)
316 SOCKET_TEST_ERROR(r = pbind(TO_SOCKET(s), addr, addrlen));
321 win32_connect(SOCKET s, const struct sockaddr *addr, int addrlen)
325 SOCKET_TEST_ERROR(r = pconnect(TO_SOCKET(s), addr, addrlen));
331 win32_getpeername(SOCKET s, struct sockaddr *addr, int *addrlen)
335 SOCKET_TEST_ERROR(r = pgetpeername(TO_SOCKET(s), addr, addrlen));
340 win32_getsockname(SOCKET s, struct sockaddr *addr, int *addrlen)
344 SOCKET_TEST_ERROR(r = pgetsockname(TO_SOCKET(s), addr, addrlen));
349 win32_getsockopt(SOCKET s, int level, int optname, char *optval, int *optlen)
353 SOCKET_TEST_ERROR(r = pgetsockopt(TO_SOCKET(s), level, optname, optval, optlen));
358 win32_ioctlsocket(SOCKET s, long cmd, u_long *argp)
362 SOCKET_TEST_ERROR(r = pioctlsocket(TO_SOCKET(s), cmd, argp));
367 win32_listen(SOCKET s, int backlog)
371 SOCKET_TEST_ERROR(r = plisten(TO_SOCKET(s), backlog));
376 win32_recv(SOCKET s, char *buf, int len, int flags)
380 SOCKET_TEST_ERROR(r = precv(TO_SOCKET(s), buf, len, flags));
385 win32_recvfrom(SOCKET s, char *buf, int len, int flags, struct sockaddr *from, int *fromlen)
389 SOCKET_TEST_ERROR(r = precvfrom(TO_SOCKET(s), buf, len, flags, from, fromlen));
393 /* select contributed by Vincent R. Slyngstad (vrs@ibeam.intel.com) */
395 win32_select(int nfds, int* rd, int* wr, int* ex, const struct timeval* timeout)
399 int i, fd, bit, offset;
400 FD_SET nrd, nwr, nex,*prd,*pwr,*pex;
403 rd = &dummy, prd = NULL;
407 wr = &dummy, pwr = NULL;
411 ex = &dummy, pex = NULL;
418 for (i = 0; i < nfds; i++) {
420 bit = 1L<<(i % (sizeof(int)*8));
421 offset = i / (sizeof(int)*8);
422 if (rd[offset] & bit)
424 if (wr[offset] & bit)
426 if (ex[offset] & bit)
430 SOCKET_TEST_ERROR(r = pselect(nfds, prd, pwr, pex, timeout));
432 for (i = 0; i < nfds; i++) {
434 bit = 1L<<(i % (sizeof(int)*8));
435 offset = i / (sizeof(int)*8);
436 if (rd[offset] & bit) {
437 if (!pWSAFDIsSet(fd, &nrd))
440 if (wr[offset] & bit) {
441 if (!pWSAFDIsSet(fd, &nwr))
444 if (ex[offset] & bit) {
445 if (!pWSAFDIsSet(fd, &nex))
453 win32_send(SOCKET s, const char *buf, int len, int flags)
457 SOCKET_TEST_ERROR(r = psend(TO_SOCKET(s), buf, len, flags));
462 win32_sendto(SOCKET s, const char *buf, int len, int flags,
463 const struct sockaddr *to, int tolen)
467 SOCKET_TEST_ERROR(r = psendto(TO_SOCKET(s), buf, len, flags, to, tolen));
472 win32_setsockopt(SOCKET s, int level, int optname, const char *optval, int optlen)
476 SOCKET_TEST_ERROR(r = psetsockopt(TO_SOCKET(s), level, optname, optval, optlen));
481 win32_shutdown(SOCKET s, int how)
485 SOCKET_TEST_ERROR(r = pshutdown(TO_SOCKET(s), how));
490 win32_socket(int af, int type, int protocol)
494 #ifndef USE_SOCKETS_AS_HANDLES
495 SOCKET_TEST(s = psocket(af, type, protocol), INVALID_SOCKET);
500 if((s = psocket(af, type, protocol)) == INVALID_SOCKET)
501 errno = pWSAGetLastError();
504 #endif /* USE_SOCKETS_AS_HANDLES */
510 win32_gethostbyaddr(const char *addr, int len, int type)
514 SOCKET_TEST(r = pgethostbyaddr(addr, len, type), NULL);
519 win32_gethostbyname(const char *name)
523 SOCKET_TEST(r = pgethostbyname(name), NULL);
528 win32_gethostname(char *name, int len)
532 SOCKET_TEST_ERROR(r = pgethostname(name, len));
537 win32_getprotobyname(const char *name)
541 SOCKET_TEST(r = pgetprotobyname(name), NULL);
546 win32_getprotobynumber(int num)
550 SOCKET_TEST(r = pgetprotobynumber(num), NULL);
555 win32_getservbyname(const char *name, const char *proto)
559 SOCKET_TEST(r = pgetservbyname(name, proto), NULL);
561 r = win32_savecopyservent(&myservent, r, proto);
567 win32_getservbyport(int port, const char *proto)
571 SOCKET_TEST(r = pgetservbyport(port, proto), NULL);
573 r = win32_savecopyservent(&myservent, r, proto);
579 win32_inet_ntoa(struct in_addr in)
584 return pinet_ntoa(in);
588 win32_inet_addr(const char FAR *cp)
593 return pinet_addr(cp);
606 CROAK("endhostent not implemented!\n");
612 CROAK("endnetent not implemented!\n");
618 CROAK("endprotoent not implemented!\n");
624 CROAK("endservent not implemented!\n");
629 win32_getnetent(void)
631 CROAK("getnetent not implemented!\n");
632 return (struct netent *) NULL;
636 win32_getnetbyname(char *name)
638 CROAK("getnetbyname not implemented!\n");
639 return (struct netent *)NULL;
643 win32_getnetbyaddr(long net, int type)
645 CROAK("getnetbyaddr not implemented!\n");
646 return (struct netent *)NULL;
650 win32_getprotoent(void)
652 CROAK("getprotoent not implemented!\n");
653 return (struct protoent *) NULL;
657 win32_getservent(void)
659 CROAK("getservent not implemented!\n");
660 return (struct servent *) NULL;
664 win32_sethostent(int stayopen)
666 CROAK("sethostent not implemented!\n");
671 win32_setnetent(int stayopen)
673 CROAK("setnetent not implemented!\n");
678 win32_setprotoent(int stayopen)
680 CROAK("setprotoent not implemented!\n");
685 win32_setservent(int stayopen)
687 CROAK("setservent not implemented!\n");
690 #define WIN32IO_IS_STDIO
692 #include "win32iop.h"
694 static struct servent*
695 win32_savecopyservent(struct servent*d, struct servent*s, const char *proto)
697 d->s_name = s->s_name;
698 d->s_aliases = s->s_aliases;
699 d->s_port = s->s_port;
700 if (!IsWin95() && s->s_proto && strlen(s->s_proto))
701 d->s_proto = s->s_proto;
702 else if (proto && strlen(proto))
703 d->s_proto = (char *)proto;