Fix [perl #24269] socket() call uses non-IFS providers
[p5sagit/p5-mst-13.2.git] / win32 / win32sck.c
1 /* win32sck.c
2  *
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.
6  *
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.
9  */
10
11 #define WIN32IO_IS_STDIO
12 #define WIN32SCK_IS_STDSCK
13 #define WIN32_LEAN_AND_MEAN
14 #define PERLIO_NOT_STDIO 0
15 #ifdef __GNUC__
16 #define Win32_Winsock
17 #endif
18 #include <windows.h>
19 #include <ws2spi.h>
20
21 #include "EXTERN.h"
22 #include "perl.h"
23
24 #include "Win32iop.h"
25 #include <sys/socket.h>
26 #include <fcntl.h>
27 #include <sys/stat.h>
28 #include <assert.h>
29 #include <io.h>
30
31 /* thanks to Beverly Brown      (beverly@datacube.com) */
32 #ifdef USE_SOCKETS_AS_HANDLES
33 #       define OPEN_SOCKET(x)   win32_open_osfhandle(x,O_RDWR|O_BINARY)
34 #       define TO_SOCKET(x)     _get_osfhandle(x)
35 #else
36 #       define OPEN_SOCKET(x)   (x)
37 #       define TO_SOCKET(x)     (x)
38 #endif  /* USE_SOCKETS_AS_HANDLES */
39
40 #if defined(USE_ITHREADS)
41 #define StartSockets() \
42     STMT_START {                                        \
43         if (!wsock_started)                             \
44             start_sockets();                            \
45         set_socktype();                                 \
46     } STMT_END
47 #else
48 #define StartSockets() \
49     STMT_START {                                        \
50         if (!wsock_started) {                           \
51             start_sockets();                            \
52             set_socktype();                             \
53         }                                               \
54     } STMT_END
55 #endif
56
57 #define SOCKET_TEST(x, y) \
58     STMT_START {                                        \
59         StartSockets();                                 \
60         if((x) == (y))                                  \
61             errno = WSAGetLastError();                  \
62     } STMT_END
63
64 #define SOCKET_TEST_ERROR(x) SOCKET_TEST(x, SOCKET_ERROR)
65
66 static struct servent* win32_savecopyservent(struct servent*d,
67                                              struct servent*s,
68                                              const char *proto);
69
70 static int wsock_started = 0;
71
72 EXTERN_C void
73 EndSockets(void)
74 {
75     if (wsock_started)
76         WSACleanup();
77 }
78
79 void
80 start_sockets(void) 
81 {
82     dTHX;
83     unsigned short version;
84     WSADATA retdata;
85     int ret;
86
87     /*
88      * initalize the winsock interface and insure that it is
89      * cleaned up at exit.
90      */
91     version = 0x2;
92     if(ret = WSAStartup(version, &retdata))
93         Perl_croak_nocontext("Unable to locate winsock library!\n");
94     if(retdata.wVersion != version)
95         Perl_croak_nocontext("Could not find version 2.0 of winsock dll\n");
96
97     /* atexit((void (*)(void)) EndSockets); */
98     wsock_started = 1;
99 }
100
101 void
102 set_socktype(void)
103 {
104 }
105
106
107 #ifndef USE_SOCKETS_AS_HANDLES
108 #undef fdopen
109 FILE *
110 my_fdopen(int fd, char *mode)
111 {
112     FILE *fp;
113     char sockbuf[256];
114     int optlen = sizeof(sockbuf);
115     int retval;
116
117     if (!wsock_started)
118         return(fdopen(fd, mode));
119
120     retval = getsockopt((SOCKET)fd, SOL_SOCKET, SO_TYPE, sockbuf, &optlen);
121     if(retval == SOCKET_ERROR && WSAGetLastError() == WSAENOTSOCK) {
122         return(fdopen(fd, mode));
123     }
124
125     /*
126      * If we get here, then fd is actually a socket.
127      */
128     Newz(1310, fp, 1, FILE);    /* XXX leak, good thing this code isn't used */
129     if(fp == NULL) {
130         errno = ENOMEM;
131         return NULL;
132     }
133
134     fp->_file = fd;
135     if(*mode == 'r')
136         fp->_flag = _IOREAD;
137     else
138         fp->_flag = _IOWRT;
139    
140     return fp;
141 }
142 #endif  /* USE_SOCKETS_AS_HANDLES */
143
144
145 u_long
146 win32_htonl(u_long hostlong)
147 {
148     StartSockets();
149     return htonl(hostlong);
150 }
151
152 u_short
153 win32_htons(u_short hostshort)
154 {
155     StartSockets();
156     return htons(hostshort);
157 }
158
159 u_long
160 win32_ntohl(u_long netlong)
161 {
162     StartSockets();
163     return ntohl(netlong);
164 }
165
166 u_short
167 win32_ntohs(u_short netshort)
168 {
169     StartSockets();
170     return ntohs(netshort);
171 }
172
173
174
175 SOCKET
176 win32_accept(SOCKET s, struct sockaddr *addr, int *addrlen)
177 {
178     SOCKET r;
179
180     SOCKET_TEST((r = accept(TO_SOCKET(s), addr, addrlen)), INVALID_SOCKET);
181     return OPEN_SOCKET(r);
182 }
183
184 int
185 win32_bind(SOCKET s, const struct sockaddr *addr, int addrlen)
186 {
187     int r;
188
189     SOCKET_TEST_ERROR(r = bind(TO_SOCKET(s), addr, addrlen));
190     return r;
191 }
192
193 int
194 win32_connect(SOCKET s, const struct sockaddr *addr, int addrlen)
195 {
196     int r;
197
198     SOCKET_TEST_ERROR(r = connect(TO_SOCKET(s), addr, addrlen));
199     return r;
200 }
201
202
203 int
204 win32_getpeername(SOCKET s, struct sockaddr *addr, int *addrlen)
205 {
206     int r;
207
208     SOCKET_TEST_ERROR(r = getpeername(TO_SOCKET(s), addr, addrlen));
209     return r;
210 }
211
212 int
213 win32_getsockname(SOCKET s, struct sockaddr *addr, int *addrlen)
214 {
215     int r;
216
217     SOCKET_TEST_ERROR(r = getsockname(TO_SOCKET(s), addr, addrlen));
218     return r;
219 }
220
221 int
222 win32_getsockopt(SOCKET s, int level, int optname, char *optval, int *optlen)
223 {
224     int r;
225
226     SOCKET_TEST_ERROR(r = getsockopt(TO_SOCKET(s), level, optname, optval, optlen));
227     return r;
228 }
229
230 int
231 win32_ioctlsocket(SOCKET s, long cmd, u_long *argp)
232 {
233     int r;
234
235     SOCKET_TEST_ERROR(r = ioctlsocket(TO_SOCKET(s), cmd, argp));
236     return r;
237 }
238
239 int
240 win32_listen(SOCKET s, int backlog)
241 {
242     int r;
243
244     SOCKET_TEST_ERROR(r = listen(TO_SOCKET(s), backlog));
245     return r;
246 }
247
248 int
249 win32_recv(SOCKET s, char *buf, int len, int flags)
250 {
251     int r;
252
253     SOCKET_TEST_ERROR(r = recv(TO_SOCKET(s), buf, len, flags));
254     return r;
255 }
256
257 int
258 win32_recvfrom(SOCKET s, char *buf, int len, int flags, struct sockaddr *from, int *fromlen)
259 {
260     int r;
261     int frombufsize = *fromlen;
262
263     SOCKET_TEST_ERROR(r = recvfrom(TO_SOCKET(s), buf, len, flags, from, fromlen));
264     /* Winsock's recvfrom() only returns a valid 'from' when the socket
265      * is connectionless.  Perl expects a valid 'from' for all types
266      * of sockets, so go the extra mile.
267      */
268     if (r != SOCKET_ERROR && frombufsize == *fromlen)
269         (void)win32_getpeername(s, from, fromlen);
270     return r;
271 }
272
273 /* select contributed by Vincent R. Slyngstad (vrs@ibeam.intel.com) */
274 int
275 win32_select(int nfds, Perl_fd_set* rd, Perl_fd_set* wr, Perl_fd_set* ex, const struct timeval* timeout)
276 {
277     int r;
278 #ifdef USE_SOCKETS_AS_HANDLES
279     Perl_fd_set dummy;
280     int i, fd, save_errno = errno;
281     FD_SET nrd, nwr, nex, *prd, *pwr, *pex;
282
283     /* winsock seems incapable of dealing with all three null fd_sets,
284      * so do the (millisecond) sleep as a special case
285      */
286     if (!(rd || wr || ex)) {
287         if (timeout)
288             Sleep(timeout->tv_sec  * 1000 +
289                   timeout->tv_usec / 1000);     /* do the best we can */
290         else
291             Sleep(UINT_MAX);
292         return 0;
293     }
294     StartSockets();
295     PERL_FD_ZERO(&dummy);
296     if (!rd)
297         rd = &dummy, prd = NULL;
298     else
299         prd = &nrd;
300     if (!wr)
301         wr = &dummy, pwr = NULL;
302     else
303         pwr = &nwr;
304     if (!ex)
305         ex = &dummy, pex = NULL;
306     else
307         pex = &nex;
308
309     FD_ZERO(&nrd);
310     FD_ZERO(&nwr);
311     FD_ZERO(&nex);
312     for (i = 0; i < nfds; i++) {
313         fd = TO_SOCKET(i);
314         if (PERL_FD_ISSET(i,rd))
315             FD_SET(fd, &nrd);
316         if (PERL_FD_ISSET(i,wr))
317             FD_SET(fd, &nwr);
318         if (PERL_FD_ISSET(i,ex))
319             FD_SET(fd, &nex);
320     }
321
322     errno = save_errno;
323     SOCKET_TEST_ERROR(r = select(nfds, prd, pwr, pex, timeout));
324     save_errno = errno;
325
326     for (i = 0; i < nfds; i++) {
327         fd = TO_SOCKET(i);
328         if (PERL_FD_ISSET(i,rd) && !FD_ISSET(fd, &nrd))
329             PERL_FD_CLR(i,rd);
330         if (PERL_FD_ISSET(i,wr) && !FD_ISSET(fd, &nwr))
331             PERL_FD_CLR(i,wr);
332         if (PERL_FD_ISSET(i,ex) && !FD_ISSET(fd, &nex))
333             PERL_FD_CLR(i,ex);
334     }
335     errno = save_errno;
336 #else
337     SOCKET_TEST_ERROR(r = select(nfds, rd, wr, ex, timeout));
338 #endif
339     return r;
340 }
341
342 int
343 win32_send(SOCKET s, const char *buf, int len, int flags)
344 {
345     int r;
346
347     SOCKET_TEST_ERROR(r = send(TO_SOCKET(s), buf, len, flags));
348     return r;
349 }
350
351 int
352 win32_sendto(SOCKET s, const char *buf, int len, int flags,
353              const struct sockaddr *to, int tolen)
354 {
355     int r;
356
357     SOCKET_TEST_ERROR(r = sendto(TO_SOCKET(s), buf, len, flags, to, tolen));
358     return r;
359 }
360
361 int
362 win32_setsockopt(SOCKET s, int level, int optname, const char *optval, int optlen)
363 {
364     int r;
365
366     SOCKET_TEST_ERROR(r = setsockopt(TO_SOCKET(s), level, optname, optval, optlen));
367     return r;
368 }
369     
370 int
371 win32_shutdown(SOCKET s, int how)
372 {
373     int r;
374
375     SOCKET_TEST_ERROR(r = shutdown(TO_SOCKET(s), how));
376     return r;
377 }
378
379 int
380 win32_closesocket(SOCKET s)
381 {
382     int r;
383
384     SOCKET_TEST_ERROR(r = closesocket(TO_SOCKET(s)));
385     return r;
386 }
387
388 #ifdef USE_SOCKETS_AS_HANDLES
389 #define WIN32_OPEN_SOCKET(af, type, protocol) open_ifs_socket(af, type, protocol)
390
391 void
392 convert_proto_info_w2a(WSAPROTOCOL_INFOW *in, WSAPROTOCOL_INFOA *out)
393 {
394     Copy(in, out, 1, WSAPROTOCOL_INFOA);
395     wcstombs(out->szProtocol, in->szProtocol, sizeof(out->szProtocol));
396 }
397
398 SOCKET
399 open_ifs_socket(int af, int type, int protocol)
400 {
401     unsigned long proto_buffers_len = 0;
402     int error_code;
403     SOCKET out = INVALID_SOCKET;
404
405     if (WSCEnumProtocols(NULL, NULL, &proto_buffers_len, &error_code) == SOCKET_ERROR
406         && error_code == WSAENOBUFS)
407     {
408         WSAPROTOCOL_INFOW *proto_buffers;
409         int protocols_available = 0;       
410  
411         New(1, proto_buffers, proto_buffers_len / sizeof(WSAPROTOCOL_INFOW),
412             WSAPROTOCOL_INFOW);
413
414         if ((protocols_available = WSCEnumProtocols(NULL, proto_buffers, 
415             &proto_buffers_len, &error_code)) != SOCKET_ERROR)
416         {
417             int i;
418             for (i = 0; i < protocols_available; i++)
419             {
420                 WSAPROTOCOL_INFOA proto_info;
421
422                 if ((af != AF_UNSPEC && af != proto_buffers[i].iAddressFamily)
423                     || (type != proto_buffers[i].iSocketType)
424                     || (protocol != 0 && protocol != proto_buffers[i].iProtocol))
425                     continue;
426
427                 if ((proto_buffers[i].dwServiceFlags1 & XP1_IFS_HANDLES) == 0)
428                     continue;
429
430                 convert_proto_info_w2a(&(proto_buffers[i]), &proto_info);
431
432                 out = WSASocket(af, type, protocol, &proto_info, 0, 0);
433                 break;
434             }
435         }
436
437         Safefree(proto_buffers);
438     }
439
440     return out;
441 }
442
443 #else
444 #define WIN32_OPEN_SOCKET(af, type, protocol) socket(af, type, protocol)
445 #endif
446
447 SOCKET
448 win32_socket(int af, int type, int protocol)
449 {
450     SOCKET s;
451
452 #ifndef USE_SOCKETS_AS_HANDLES
453     SOCKET_TEST(s = socket(af, type, protocol), INVALID_SOCKET);
454 #else
455     StartSockets();
456
457     if((s = WIN32_OPEN_SOCKET(af, type, protocol)) == INVALID_SOCKET)
458         errno = WSAGetLastError();
459     else
460         s = OPEN_SOCKET(s);
461 #endif  /* USE_SOCKETS_AS_HANDLES */
462
463     return s;
464 }
465
466 /*
467  * close RTL fd while respecting sockets
468  * added as temporary measure until PerlIO has real
469  * Win32 native layer
470  *   -- BKS, 11-11-2000
471 */
472
473 int my_close(int fd)
474 {
475     int osf;
476     if (!wsock_started)         /* No WinSock? */
477         return(close(fd));      /* Then not a socket. */
478     osf = TO_SOCKET(fd);/* Get it now before it's gone! */
479     if (osf != -1) {
480         int err;
481         err = closesocket(osf);
482         if (err == 0) {
483 #if defined(USE_FIXED_OSFHANDLE) || defined(PERL_MSVCRT_READFIX)
484             _set_osfhnd(fd, INVALID_HANDLE_VALUE);
485 #endif
486             (void)close(fd);    /* handle already closed, ignore error */
487             return 0;
488         }
489         else if (err == SOCKET_ERROR) {
490             err = WSAGetLastError();
491             if (err != WSAENOTSOCK) {
492                 (void)close(fd);
493                 errno = err;
494                 return EOF;
495             }
496         }
497     }
498     return close(fd);
499 }
500
501 #undef fclose
502 int
503 my_fclose (FILE *pf)
504 {
505     int osf;
506     if (!wsock_started)         /* No WinSock? */
507         return(fclose(pf));     /* Then not a socket. */
508     osf = TO_SOCKET(win32_fileno(pf));/* Get it now before it's gone! */
509     if (osf != -1) {
510         int err;
511         win32_fflush(pf);
512         err = closesocket(osf);
513         if (err == 0) {
514 #if defined(USE_FIXED_OSFHANDLE) || defined(PERL_MSVCRT_READFIX)
515             _set_osfhnd(win32_fileno(pf), INVALID_HANDLE_VALUE);
516 #endif
517             (void)fclose(pf);   /* handle already closed, ignore error */
518             return 0;
519         }
520         else if (err == SOCKET_ERROR) {
521             err = WSAGetLastError();
522             if (err != WSAENOTSOCK) {
523                 (void)fclose(pf);
524                 errno = err;
525                 return EOF;
526             }
527         }
528     }
529     return fclose(pf);
530 }
531
532 #undef fstat
533 int
534 my_fstat(int fd, Stat_t *sbufptr)
535 {
536     /* This fixes a bug in fstat() on Windows 9x.  fstat() uses the
537      * GetFileType() win32 syscall, which will fail on Windows 9x.
538      * So if we recognize a socket on Windows 9x, we return the
539      * same results as on Windows NT/2000.
540      * XXX this should be extended further to set S_IFSOCK on
541      * sbufptr->st_mode.
542      */
543     int osf;
544     if (!wsock_started || IsWinNT()) {
545 #if defined(WIN64) || defined(USE_LARGE_FILES)
546 #if defined(__BORLANDC__) /* buk */
547         return win32_fstat(fd, sbufptr );
548 #else
549         return _fstati64(fd, sbufptr);
550 #endif
551 #else
552         return fstat(fd, sbufptr);
553 #endif
554     }
555
556     osf = TO_SOCKET(fd);
557     if (osf != -1) {
558         char sockbuf[256];
559         int optlen = sizeof(sockbuf);
560         int retval;
561
562         retval = getsockopt((SOCKET)osf, SOL_SOCKET, SO_TYPE, sockbuf, &optlen);
563         if (retval != SOCKET_ERROR || WSAGetLastError() != WSAENOTSOCK) {
564 #if defined(__BORLANDC__)&&(__BORLANDC__<=0x520)
565             sbufptr->st_mode = S_IFIFO;
566 #else
567             sbufptr->st_mode = _S_IFIFO;
568 #endif
569             sbufptr->st_rdev = sbufptr->st_dev = (dev_t)fd;
570             sbufptr->st_nlink = 1;
571             sbufptr->st_uid = sbufptr->st_gid = sbufptr->st_ino = 0;
572             sbufptr->st_atime = sbufptr->st_mtime = sbufptr->st_ctime = 0;
573             sbufptr->st_size = (Off_t)0;
574             return 0;
575         }
576     }
577 #if defined(WIN64) || defined(USE_LARGE_FILES)
578 #if defined(__BORLANDC__) /* buk */
579     return win32_fstat(fd, sbufptr );
580 #else
581     return _fstati64(fd, sbufptr);
582 #endif
583 #else
584     return fstat(fd, sbufptr);
585 #endif
586 }
587
588 struct hostent *
589 win32_gethostbyaddr(const char *addr, int len, int type)
590 {
591     struct hostent *r;
592
593     SOCKET_TEST(r = gethostbyaddr(addr, len, type), NULL);
594     return r;
595 }
596
597 struct hostent *
598 win32_gethostbyname(const char *name)
599 {
600     struct hostent *r;
601
602     SOCKET_TEST(r = gethostbyname(name), NULL);
603     return r;
604 }
605
606 int
607 win32_gethostname(char *name, int len)
608 {
609     int r;
610
611     SOCKET_TEST_ERROR(r = gethostname(name, len));
612     return r;
613 }
614
615 struct protoent *
616 win32_getprotobyname(const char *name)
617 {
618     struct protoent *r;
619
620     SOCKET_TEST(r = getprotobyname(name), NULL);
621     return r;
622 }
623
624 struct protoent *
625 win32_getprotobynumber(int num)
626 {
627     struct protoent *r;
628
629     SOCKET_TEST(r = getprotobynumber(num), NULL);
630     return r;
631 }
632
633 struct servent *
634 win32_getservbyname(const char *name, const char *proto)
635 {
636     dTHX;    
637     struct servent *r;
638
639     SOCKET_TEST(r = getservbyname(name, proto), NULL);
640     if (r) {
641         r = win32_savecopyservent(&w32_servent, r, proto);
642     }
643     return r;
644 }
645
646 struct servent *
647 win32_getservbyport(int port, const char *proto)
648 {
649     dTHX; 
650     struct servent *r;
651
652     SOCKET_TEST(r = getservbyport(port, proto), NULL);
653     if (r) {
654         r = win32_savecopyservent(&w32_servent, r, proto);
655     }
656     return r;
657 }
658
659 int
660 win32_ioctl(int i, unsigned int u, char *data)
661 {
662     dTHX;
663     u_long argp = (u_long)data;
664     int retval;
665
666     if (!wsock_started) {
667         Perl_croak_nocontext("ioctl implemented only on sockets");
668         /* NOTREACHED */
669     }
670
671     retval = ioctlsocket(TO_SOCKET(i), (long)u, &argp);
672     if (retval == SOCKET_ERROR) {
673         if (WSAGetLastError() == WSAENOTSOCK) {
674             Perl_croak_nocontext("ioctl implemented only on sockets");
675             /* NOTREACHED */
676         }
677         errno = WSAGetLastError();
678     }
679     return retval;
680 }
681
682 char FAR *
683 win32_inet_ntoa(struct in_addr in)
684 {
685     StartSockets();
686     return inet_ntoa(in);
687 }
688
689 unsigned long
690 win32_inet_addr(const char FAR *cp)
691 {
692     StartSockets();
693     return inet_addr(cp);
694 }
695
696 /*
697  * Networking stubs
698  */
699
700 void
701 win32_endhostent() 
702 {
703     dTHX;
704     Perl_croak_nocontext("endhostent not implemented!\n");
705 }
706
707 void
708 win32_endnetent()
709 {
710     dTHX;
711     Perl_croak_nocontext("endnetent not implemented!\n");
712 }
713
714 void
715 win32_endprotoent()
716 {
717     dTHX;
718     Perl_croak_nocontext("endprotoent not implemented!\n");
719 }
720
721 void
722 win32_endservent()
723 {
724     dTHX;
725     Perl_croak_nocontext("endservent not implemented!\n");
726 }
727
728
729 struct netent *
730 win32_getnetent(void) 
731 {
732     dTHX;
733     Perl_croak_nocontext("getnetent not implemented!\n");
734     return (struct netent *) NULL;
735 }
736
737 struct netent *
738 win32_getnetbyname(char *name) 
739 {
740     dTHX;
741     Perl_croak_nocontext("getnetbyname not implemented!\n");
742     return (struct netent *)NULL;
743 }
744
745 struct netent *
746 win32_getnetbyaddr(long net, int type) 
747 {
748     dTHX;
749     Perl_croak_nocontext("getnetbyaddr not implemented!\n");
750     return (struct netent *)NULL;
751 }
752
753 struct protoent *
754 win32_getprotoent(void) 
755 {
756     dTHX;
757     Perl_croak_nocontext("getprotoent not implemented!\n");
758     return (struct protoent *) NULL;
759 }
760
761 struct servent *
762 win32_getservent(void) 
763 {
764     dTHX;
765     Perl_croak_nocontext("getservent not implemented!\n");
766     return (struct servent *) NULL;
767 }
768
769 void
770 win32_sethostent(int stayopen)
771 {
772     dTHX;
773     Perl_croak_nocontext("sethostent not implemented!\n");
774 }
775
776
777 void
778 win32_setnetent(int stayopen)
779 {
780     dTHX;
781     Perl_croak_nocontext("setnetent not implemented!\n");
782 }
783
784
785 void
786 win32_setprotoent(int stayopen)
787 {
788     dTHX;
789     Perl_croak_nocontext("setprotoent not implemented!\n");
790 }
791
792
793 void
794 win32_setservent(int stayopen)
795 {
796     dTHX;
797     Perl_croak_nocontext("setservent not implemented!\n");
798 }
799
800 static struct servent*
801 win32_savecopyservent(struct servent*d, struct servent*s, const char *proto)
802 {
803     d->s_name = s->s_name;
804     d->s_aliases = s->s_aliases;
805     d->s_port = s->s_port;
806 #ifndef __BORLANDC__    /* Buggy on Win95 and WinNT-with-Borland-WSOCK */
807     if (!IsWin95() && s->s_proto && strlen(s->s_proto))
808         d->s_proto = s->s_proto;
809     else
810 #endif
811     if (proto && strlen(proto))
812         d->s_proto = (char *)proto;
813     else
814         d->s_proto = "tcp";
815    
816     return d;
817 }
818
819