]> git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.cpp
test for special keys first, before testing for alphanumeric ones as even keys such...
[wxWidgets.git] / src / unix / gsocket.cpp
1 /* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
3 * Name: gsocket.c
4 * Copyright: (c) Guilhem Lavaux
5 * Licence: wxWindows Licence
6 * Authors: David Elliott (C++ conversion, maintainer)
7 * Guilhem Lavaux,
8 * Guillermo Rodriguez Garcia <guille@iies.es>
9 * Purpose: GSocket main Unix and OS/2 file
10 * Licence: The wxWindows licence
11 * CVSID: $Id$
12 * -------------------------------------------------------------------------
13 */
14
15 #if defined(__WATCOMC__)
16 #include "wx/wxprec.h"
17 #include <errno.h>
18 #include <nerrno.h>
19 #endif
20
21 #ifndef __GSOCKET_STANDALONE__
22 #include "wx/defs.h"
23 #endif
24
25 #if defined(__VISAGECPP__)
26 #define BSD_SELECT /* use Berkeley Sockets select */
27 #endif
28
29 #if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__)
30
31 #include <assert.h>
32 #include <sys/types.h>
33 #ifdef __VISAGECPP__
34 #include <string.h>
35 #include <sys/time.h>
36 #include <types.h>
37 #include <netinet/in.h>
38 #endif
39 #include <netdb.h>
40 #include <sys/ioctl.h>
41
42 #ifdef HAVE_SYS_SELECT_H
43 # include <sys/select.h>
44 #endif
45
46 #ifdef __VMS__
47 #include <socket.h>
48 struct sockaddr_un
49 {
50 u_char sun_len; /* sockaddr len including null */
51 u_char sun_family; /* AF_UNIX */
52 char sun_path[108]; /* path name (gag) */
53 };
54 #else
55 #include <sys/socket.h>
56 #include <sys/un.h>
57 #endif
58
59 #ifndef __VISAGECPP__
60 #include <sys/time.h>
61 #include <netinet/in.h>
62 #include <arpa/inet.h>
63 #include <errno.h>
64 #include <string.h>
65 #include <unistd.h>
66 #else
67 #include <nerrno.h>
68 # if __IBMCPP__ < 400
69 #include <machine/endian.h>
70 #include <socket.h>
71 #include <ioctl.h>
72 #include <select.h>
73 #include <unistd.h>
74
75 #define EBADF SOCEBADF
76
77 # ifdef min
78 # undef min
79 # endif
80 # else
81 #include <sys/socket.h>
82 #include <sys/ioctl.h>
83 #include <sys/select.h>
84
85 #define close(a) soclose(a)
86 #define select(a,b,c,d,e) bsdselect(a,b,c,d,e)
87 int _System bsdselect(int,
88 struct fd_set *,
89 struct fd_set *,
90 struct fd_set *,
91 struct timeval *);
92 int _System soclose(int);
93 # endif
94 #endif
95 #ifdef __EMX__
96 #include <sys/select.h>
97 #endif
98
99 #include <stdio.h>
100 #include <stdlib.h>
101 #include <stddef.h>
102 #include <ctype.h>
103 #ifdef sun
104 # include <sys/filio.h>
105 #endif
106 #ifdef sgi
107 # include <bstring.h>
108 #endif
109 #ifdef _AIX
110 # include <strings.h>
111 #endif
112 #include <signal.h>
113
114 #ifndef WX_SOCKLEN_T
115
116 #ifdef VMS
117 # define WX_SOCKLEN_T unsigned int
118 #else
119 # ifdef __GLIBC__
120 # if __GLIBC__ == 2
121 # define WX_SOCKLEN_T socklen_t
122 # endif
123 # elif defined(__WXMAC__)
124 # define WX_SOCKLEN_T socklen_t
125 # else
126 # define WX_SOCKLEN_T int
127 # endif
128 #endif
129
130 #endif /* SOCKLEN_T */
131
132 #ifndef SOCKOPTLEN_T
133 #define SOCKOPTLEN_T WX_SOCKLEN_T
134 #endif
135
136 /*
137 * MSW defines this, Unices don't.
138 */
139 #ifndef INVALID_SOCKET
140 #define INVALID_SOCKET -1
141 #endif
142
143 /* UnixWare reportedly needs this for FIONBIO definition */
144 #ifdef __UNIXWARE__
145 #include <sys/filio.h>
146 #endif
147
148 /*
149 * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
150 * on all systems. INADDR_BROADCAST should be fine to indicate an error.
151 */
152 #ifndef INADDR_NONE
153 #define INADDR_NONE INADDR_BROADCAST
154 #endif
155
156 #if defined(__VISAGECPP__) || defined(__WATCOMC__)
157
158 #define MASK_SIGNAL() {
159 #define UNMASK_SIGNAL() }
160
161 #else
162 extern "C" { typedef void (*wxSigHandler)(int); }
163
164 #define MASK_SIGNAL() \
165 { \
166 wxSigHandler old_handler = signal(SIGPIPE, SIG_IGN);
167
168 #define UNMASK_SIGNAL() \
169 signal(SIGPIPE, old_handler); \
170 }
171
172 #endif
173
174 /* If a SIGPIPE is issued by a socket call on a remotely closed socket,
175 the program will "crash" unless it explicitly handles the SIGPIPE.
176 By using MSG_NOSIGNAL, the SIGPIPE is suppressed. Later, we will
177 use SO_NOSIGPIPE (if available), the BSD equivalent. */
178 #ifdef MSG_NOSIGNAL
179 # define GSOCKET_MSG_NOSIGNAL MSG_NOSIGNAL
180 #else /* MSG_NOSIGNAL not available (FreeBSD including OS X) */
181 # define GSOCKET_MSG_NOSIGNAL 0
182 #endif /* MSG_NOSIGNAL */
183
184 #ifndef __GSOCKET_STANDALONE__
185 # include "wx/unix/gsockunx.h"
186 # include "wx/unix/private.h"
187 # include "wx/gsocket.h"
188 #if wxUSE_THREADS && (defined(HAVE_GETHOSTBYNAME) || defined(HAVE_GETSERVBYNAME))
189 # include "wx/thread.h"
190 #endif
191 #else
192 # include "gsockunx.h"
193 # include "gsocket.h"
194 # ifndef WXUNUSED
195 # define WXUNUSED(x)
196 # endif
197 #endif /* __GSOCKET_STANDALONE__ */
198
199 #if defined(HAVE_GETHOSTBYNAME)
200 static struct hostent * deepCopyHostent(struct hostent *h,
201 const struct hostent *he,
202 char *buffer, int size, int *err)
203 {
204 /* copy old structure */
205 memcpy(h, he, sizeof(struct hostent));
206
207 /* copy name */
208 int len = strlen(h->h_name);
209 if (len > size)
210 {
211 *err = ENOMEM;
212 return NULL;
213 }
214 memcpy(buffer, h->h_name, len);
215 buffer[len] = '\0';
216 h->h_name = buffer;
217
218 /* track position in the buffer */
219 int pos = len + 1;
220
221 /* reuse len to store address length */
222 len = h->h_length;
223
224 /* ensure pointer alignment */
225 unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
226 if(misalign < sizeof(char *))
227 pos += misalign;
228
229 /* leave space for pointer list */
230 char **p = h->h_addr_list, **q;
231 char **h_addr_list = (char **)(buffer + pos);
232 while(*(p++) != 0)
233 pos += sizeof(char *);
234
235 /* copy addresses and fill new pointer list */
236 for (p = h->h_addr_list, q = h_addr_list; *p != 0; p++, q++)
237 {
238 if (size < pos + len)
239 {
240 *err = ENOMEM;
241 return NULL;
242 }
243 memcpy(buffer + pos, *p, len); /* copy content */
244 *q = buffer + pos; /* set copied pointer to copied content */
245 pos += len;
246 }
247 *++q = 0; /* null terminate the pointer list */
248 h->h_addr_list = h_addr_list; /* copy pointer to pointers */
249
250 /* ensure word alignment of pointers */
251 misalign = sizeof(char *) - pos%sizeof(char *);
252 if(misalign < sizeof(char *))
253 pos += misalign;
254
255 /* leave space for pointer list */
256 p = h->h_aliases;
257 char **h_aliases = (char **)(buffer + pos);
258 while(*(p++) != 0)
259 pos += sizeof(char *);
260
261 /* copy aliases and fill new pointer list */
262 for (p = h->h_aliases, q = h_aliases; *p != 0; p++, q++)
263 {
264 len = strlen(*p);
265 if (size <= pos + len)
266 {
267 *err = ENOMEM;
268 return NULL;
269 }
270 memcpy(buffer + pos, *p, len); /* copy content */
271 buffer[pos + len] = '\0';
272 *q = buffer + pos; /* set copied pointer to copied content */
273 pos += len + 1;
274 }
275 *++q = 0; /* null terminate the pointer list */
276 h->h_aliases = h_aliases; /* copy pointer to pointers */
277
278 return h;
279 }
280 #endif
281
282 #if defined(HAVE_GETHOSTBYNAME) && wxUSE_THREADS
283 static wxMutex nameLock;
284 #endif
285 struct hostent * wxGethostbyname_r(const char *hostname, struct hostent *h,
286 void *buffer, int size, int *err)
287
288 {
289 struct hostent *he = NULL;
290 *err = 0;
291 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_6)
292 if (gethostbyname_r(hostname, h, (char*)buffer, size, &he, err))
293 he = NULL;
294 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_5)
295 he = gethostbyname_r(hostname, h, (char*)buffer, size, err);
296 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
297 if (gethostbyname_r(hostname, h, (struct hostent_data*) buffer))
298 {
299 he = NULL;
300 *err = h_errno;
301 }
302 else
303 he = h;
304 #elif defined(HAVE_GETHOSTBYNAME)
305 #if wxUSE_THREADS
306 wxMutexLocker locker(nameLock);
307 #endif
308 he = gethostbyname(hostname);
309 if (!he)
310 *err = h_errno;
311 else
312 he = deepCopyHostent(h, he, (char*)buffer, size, err);
313 #endif
314 return he;
315 }
316
317 #if defined(HAVE_GETHOSTBYNAME) && wxUSE_THREADS
318 static wxMutex addrLock;
319 #endif
320 struct hostent * wxGethostbyaddr_r(const char *addr_buf, int buf_size,
321 int proto, struct hostent *h,
322 void *buffer, int size, int *err)
323 {
324 struct hostent *he = NULL;
325 *err = 0;
326 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_6)
327 if (gethostbyaddr_r(addr_buf, buf_size, proto, h,
328 (char*)buffer, size, &he, err))
329 he = NULL;
330 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_5)
331 he = gethostbyaddr_r(addr_buf, buf_size, proto, h, (char*)buffer, size, err);
332 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
333 if (gethostbyaddr_r(addr_buf, buf_size, proto, h,
334 (struct hostent_data*) buffer))
335 {
336 he = NULL;
337 *err = h_errno;
338 }
339 else
340 he = h;
341 #elif defined(HAVE_GETHOSTBYNAME)
342 #if wxUSE_THREADS
343 wxMutexLocker locker(addrLock);
344 #endif
345 he = gethostbyaddr(addr_buf, buf_size, proto);
346 if (!he)
347 *err = h_errno;
348 else
349 he = deepCopyHostent(h, he, (char*)buffer, size, err);
350 #endif
351 return he;
352 }
353
354 #if defined(HAVE_GETSERVBYNAME)
355 static struct servent * deepCopyServent(struct servent *s,
356 const struct servent *se,
357 char *buffer, int size)
358 {
359 /* copy plain old structure */
360 memcpy(s, se, sizeof(struct servent));
361
362 /* copy name */
363 int len = strlen(s->s_name);
364 if (len >= size)
365 {
366 return NULL;
367 }
368 memcpy(buffer, s->s_name, len);
369 buffer[len] = '\0';
370 s->s_name = buffer;
371
372 /* track position in the buffer */
373 int pos = len + 1;
374
375 /* copy protocol */
376 len = strlen(s->s_proto);
377 if (pos + len >= size)
378 {
379 return NULL;
380 }
381 memcpy(buffer + pos, s->s_proto, len);
382 buffer[pos + len] = '\0';
383 s->s_proto = buffer + pos;
384
385 /* track position in the buffer */
386 pos += len + 1;
387
388 /* ensure pointer alignment */
389 unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
390 if(misalign < sizeof(char *))
391 pos += misalign;
392
393 /* leave space for pointer list */
394 char **p = s->s_aliases, **q;
395 char **s_aliases = (char **)(buffer + pos);
396 while(*(p++) != 0)
397 pos += sizeof(char *);
398
399 /* copy addresses and fill new pointer list */
400 for (p = s->s_aliases, q = s_aliases; *p != 0; p++, q++){
401 len = strlen(*p);
402 if (size <= pos + len)
403 {
404 return NULL;
405 }
406 memcpy(buffer + pos, *p, len); /* copy content */
407 buffer[pos + len] = '\0';
408 *q = buffer + pos; /* set copied pointer to copied content */
409 pos += len + 1;
410 }
411 *++q = 0; /* null terminate the pointer list */
412 s->s_aliases = s_aliases; /* copy pointer to pointers */
413 return s;
414 }
415 #endif
416
417 #if defined(HAVE_GETSERVBYNAME) && wxUSE_THREADS
418 static wxMutex servLock;
419 #endif
420 struct servent *wxGetservbyname_r(const char *port, const char *protocol,
421 struct servent *serv, void *buffer, int size)
422 {
423 struct servent *se = NULL;
424 #if defined(HAVE_FUNC_GETSERVBYNAME_R_6)
425 if (getservbyname_r(port, protocol, serv, (char*)buffer, size, &se))
426 se = NULL;
427 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_5)
428 se = getservbyname_r(port, protocol, serv, (char*)buffer, size);
429 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_4)
430 if (getservbyname_r(port, protocol, serv, (struct servent_data*) buffer))
431 se = NULL;
432 else
433 se = serv;
434 #elif defined(HAVE_GETSERVBYNAME)
435 #if wxUSE_THREADS
436 wxMutexLocker locker(servLock);
437 #endif
438 se = getservbyname(port, protocol);
439 if (se)
440 se = deepCopyServent(serv, se, (char*)buffer, size);
441 #endif
442 return se;
443 }
444
445 /* debugging helpers */
446 #ifdef __GSOCKET_DEBUG__
447 # define GSocket_Debug(args) printf args
448 #else
449 # define GSocket_Debug(args)
450 #endif /* __GSOCKET_DEBUG__ */
451
452 /* Table of GUI-related functions. We must call them indirectly because
453 * of wxBase and GUI separation: */
454
455 static GSocketGUIFunctionsTable *gs_gui_functions;
456
457 class GSocketGUIFunctionsTableNull: public GSocketGUIFunctionsTable
458 {
459 public:
460 virtual bool OnInit();
461 virtual void OnExit();
462 virtual bool CanUseEventLoop();
463 virtual bool Init_Socket(GSocket *socket);
464 virtual void Destroy_Socket(GSocket *socket);
465 virtual void Install_Callback(GSocket *socket, GSocketEvent event);
466 virtual void Uninstall_Callback(GSocket *socket, GSocketEvent event);
467 virtual void Enable_Events(GSocket *socket);
468 virtual void Disable_Events(GSocket *socket);
469 };
470
471 bool GSocketGUIFunctionsTableNull::OnInit()
472 { return true; }
473 void GSocketGUIFunctionsTableNull::OnExit()
474 {}
475 bool GSocketGUIFunctionsTableNull::CanUseEventLoop()
476 { return false; }
477 bool GSocketGUIFunctionsTableNull::Init_Socket(GSocket *WXUNUSED(socket))
478 { return true; }
479 void GSocketGUIFunctionsTableNull::Destroy_Socket(GSocket *WXUNUSED(socket))
480 {}
481 void GSocketGUIFunctionsTableNull::Install_Callback(GSocket *WXUNUSED(socket), GSocketEvent WXUNUSED(event))
482 {}
483 void GSocketGUIFunctionsTableNull::Uninstall_Callback(GSocket *WXUNUSED(socket), GSocketEvent WXUNUSED(event))
484 {}
485 void GSocketGUIFunctionsTableNull::Enable_Events(GSocket *WXUNUSED(socket))
486 {}
487 void GSocketGUIFunctionsTableNull::Disable_Events(GSocket *WXUNUSED(socket))
488 {}
489 /* Global initialisers */
490
491 void GSocket_SetGUIFunctions(GSocketGUIFunctionsTable *guifunc)
492 {
493 gs_gui_functions = guifunc;
494 }
495
496 int GSocket_Init(void)
497 {
498 if (!gs_gui_functions)
499 {
500 static GSocketGUIFunctionsTableNull table;
501 gs_gui_functions = &table;
502 }
503 if ( !gs_gui_functions->OnInit() )
504 return 0;
505 return 1;
506 }
507
508 void GSocket_Cleanup(void)
509 {
510 if (gs_gui_functions)
511 {
512 gs_gui_functions->OnExit();
513 }
514 }
515
516 /* Constructors / Destructors for GSocket */
517
518 GSocket::GSocket()
519 {
520 int i;
521
522 m_fd = INVALID_SOCKET;
523 for (i=0;i<GSOCK_MAX_EVENT;i++)
524 {
525 m_cbacks[i] = NULL;
526 }
527 m_detected = 0;
528 m_local = NULL;
529 m_peer = NULL;
530 m_error = GSOCK_NOERROR;
531 m_server = false;
532 m_stream = true;
533 m_gui_dependent = NULL;
534 m_non_blocking = false;
535 m_reusable = false;
536 m_timeout = 10*60*1000;
537 /* 10 minutes * 60 sec * 1000 millisec */
538 m_establishing = false;
539
540 assert(gs_gui_functions);
541 /* Per-socket GUI-specific initialization */
542 m_ok = gs_gui_functions->Init_Socket(this);
543 }
544
545 void GSocket::Close()
546 {
547 gs_gui_functions->Disable_Events(this);
548 /* gsockosx.c calls CFSocketInvalidate which closes the socket for us */
549 #if !(defined(__DARWIN__) && (defined(__WXMAC__) || defined(__WXCOCOA__)))
550 close(m_fd);
551 #endif
552 m_fd = INVALID_SOCKET;
553 }
554
555 GSocket::~GSocket()
556 {
557 assert(this);
558
559 /* Check that the socket is really shutdowned */
560 if (m_fd != INVALID_SOCKET)
561 Shutdown();
562
563 /* Per-socket GUI-specific cleanup */
564 gs_gui_functions->Destroy_Socket(this);
565
566 /* Destroy private addresses */
567 if (m_local)
568 GAddress_destroy(m_local);
569
570 if (m_peer)
571 GAddress_destroy(m_peer);
572 }
573
574 /* GSocket_Shutdown:
575 * Disallow further read/write operations on this socket, close
576 * the fd and disable all callbacks.
577 */
578 void GSocket::Shutdown()
579 {
580 int evt;
581
582 assert(this);
583
584 /* Don't allow events to fire after socket has been closed */
585 gs_gui_functions->Disable_Events(this);
586
587 /* If socket has been created, shutdown it */
588 if (m_fd != INVALID_SOCKET)
589 {
590 shutdown(m_fd, 2);
591 Close();
592 }
593
594 /* Disable GUI callbacks */
595 for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
596 m_cbacks[evt] = NULL;
597
598 m_detected = GSOCK_LOST_FLAG;
599 }
600
601 /* Address handling */
602
603 /* GSocket_SetLocal:
604 * GSocket_GetLocal:
605 * GSocket_SetPeer:
606 * GSocket_GetPeer:
607 * Set or get the local or peer address for this socket. The 'set'
608 * functions return GSOCK_NOERROR on success, an error code otherwise.
609 * The 'get' functions return a pointer to a GAddress object on success,
610 * or NULL otherwise, in which case they set the error code of the
611 * corresponding GSocket.
612 *
613 * Error codes:
614 * GSOCK_INVSOCK - the socket is not valid.
615 * GSOCK_INVADDR - the address is not valid.
616 */
617 GSocketError GSocket::SetLocal(GAddress *address)
618 {
619 assert(this);
620
621 /* the socket must be initialized, or it must be a server */
622 if ((m_fd != INVALID_SOCKET && !m_server))
623 {
624 m_error = GSOCK_INVSOCK;
625 return GSOCK_INVSOCK;
626 }
627
628 /* check address */
629 if (address == NULL || address->m_family == GSOCK_NOFAMILY)
630 {
631 m_error = GSOCK_INVADDR;
632 return GSOCK_INVADDR;
633 }
634
635 if (m_local)
636 GAddress_destroy(m_local);
637
638 m_local = GAddress_copy(address);
639
640 return GSOCK_NOERROR;
641 }
642
643 GSocketError GSocket::SetPeer(GAddress *address)
644 {
645 assert(this);
646
647 /* check address */
648 if (address == NULL || address->m_family == GSOCK_NOFAMILY)
649 {
650 m_error = GSOCK_INVADDR;
651 return GSOCK_INVADDR;
652 }
653
654 if (m_peer)
655 GAddress_destroy(m_peer);
656
657 m_peer = GAddress_copy(address);
658
659 return GSOCK_NOERROR;
660 }
661
662 GAddress *GSocket::GetLocal()
663 {
664 GAddress *address;
665 struct sockaddr addr;
666 WX_SOCKLEN_T size = sizeof(addr);
667 GSocketError err;
668
669 assert(this);
670
671 /* try to get it from the m_local var first */
672 if (m_local)
673 return GAddress_copy(m_local);
674
675 /* else, if the socket is initialized, try getsockname */
676 if (m_fd == INVALID_SOCKET)
677 {
678 m_error = GSOCK_INVSOCK;
679 return NULL;
680 }
681
682 if (getsockname(m_fd, &addr, (WX_SOCKLEN_T *) &size) < 0)
683 {
684 m_error = GSOCK_IOERR;
685 return NULL;
686 }
687
688 /* got a valid address from getsockname, create a GAddress object */
689 address = GAddress_new();
690 if (address == NULL)
691 {
692 m_error = GSOCK_MEMERR;
693 return NULL;
694 }
695
696 err = _GAddress_translate_from(address, &addr, size);
697 if (err != GSOCK_NOERROR)
698 {
699 GAddress_destroy(address);
700 m_error = err;
701 return NULL;
702 }
703
704 return address;
705 }
706
707 GAddress *GSocket::GetPeer()
708 {
709 assert(this);
710
711 /* try to get it from the m_peer var */
712 if (m_peer)
713 return GAddress_copy(m_peer);
714
715 return NULL;
716 }
717
718 /* Server specific parts */
719
720 /* GSocket_SetServer:
721 * Sets up this socket as a server. The local address must have been
722 * set with GSocket_SetLocal() before GSocket_SetServer() is called.
723 * Returns GSOCK_NOERROR on success, one of the following otherwise:
724 *
725 * Error codes:
726 * GSOCK_INVSOCK - the socket is in use.
727 * GSOCK_INVADDR - the local address has not been set.
728 * GSOCK_IOERR - low-level error.
729 */
730 GSocketError GSocket::SetServer()
731 {
732 int arg = 1;
733
734 assert(this);
735
736 /* must not be in use */
737 if (m_fd != INVALID_SOCKET)
738 {
739 m_error = GSOCK_INVSOCK;
740 return GSOCK_INVSOCK;
741 }
742
743 /* the local addr must have been set */
744 if (!m_local)
745 {
746 m_error = GSOCK_INVADDR;
747 return GSOCK_INVADDR;
748 }
749
750 /* Initialize all fields */
751 m_stream = true;
752 m_server = true;
753
754 /* Create the socket */
755 m_fd = socket(m_local->m_realfamily, SOCK_STREAM, 0);
756
757 if (m_fd == INVALID_SOCKET)
758 {
759 m_error = GSOCK_IOERR;
760 return GSOCK_IOERR;
761 }
762
763 /* FreeBSD variants can't use MSG_NOSIGNAL, and instead use a socket option */
764 #ifdef SO_NOSIGPIPE
765 setsockopt(m_fd, SOL_SOCKET, SO_NOSIGPIPE, (const char*)&arg, sizeof(u_long));
766 #endif
767
768 ioctl(m_fd, FIONBIO, &arg);
769 gs_gui_functions->Enable_Events(this);
770
771 /* allow a socket to re-bind if the socket is in the TIME_WAIT
772 state after being previously closed.
773 */
774 if (m_reusable)
775 {
776 setsockopt(m_fd, SOL_SOCKET, SO_REUSEADDR, (const char*)&arg, sizeof(u_long));
777 #ifdef SO_REUSEPORT
778 setsockopt(m_fd, SOL_SOCKET, SO_REUSEPORT, (const char*)&arg, sizeof(u_long));
779 #endif
780 }
781
782 /* Bind to the local address,
783 * retrieve the actual address bound,
784 * and listen up to 5 connections.
785 */
786 if ((bind(m_fd, m_local->m_addr, m_local->m_len) != 0) ||
787 (getsockname(m_fd,
788 m_local->m_addr,
789 (WX_SOCKLEN_T *) &m_local->m_len) != 0) ||
790 (listen(m_fd, 5) != 0))
791 {
792 Close();
793 m_error = GSOCK_IOERR;
794 return GSOCK_IOERR;
795 }
796
797 return GSOCK_NOERROR;
798 }
799
800 /* GSocket_WaitConnection:
801 * Waits for an incoming client connection. Returns a pointer to
802 * a GSocket object, or NULL if there was an error, in which case
803 * the last error field will be updated for the calling GSocket.
804 *
805 * Error codes (set in the calling GSocket)
806 * GSOCK_INVSOCK - the socket is not valid or not a server.
807 * GSOCK_TIMEDOUT - timeout, no incoming connections.
808 * GSOCK_WOULDBLOCK - the call would block and the socket is nonblocking.
809 * GSOCK_MEMERR - couldn't allocate memory.
810 * GSOCK_IOERR - low-level error.
811 */
812 GSocket *GSocket::WaitConnection()
813 {
814 struct sockaddr from;
815 WX_SOCKLEN_T fromlen = sizeof(from);
816 GSocket *connection;
817 GSocketError err;
818 int arg = 1;
819
820 assert(this);
821
822 /* If the socket has already been created, we exit immediately */
823 if (m_fd == INVALID_SOCKET || !m_server)
824 {
825 m_error = GSOCK_INVSOCK;
826 return NULL;
827 }
828
829 /* Create a GSocket object for the new connection */
830 connection = GSocket_new();
831
832 if (!connection)
833 {
834 m_error = GSOCK_MEMERR;
835 return NULL;
836 }
837
838 /* Wait for a connection (with timeout) */
839 if (Input_Timeout() == GSOCK_TIMEDOUT)
840 {
841 delete connection;
842 /* m_error set by _GSocket_Input_Timeout */
843 return NULL;
844 }
845
846 connection->m_fd = accept(m_fd, &from, (WX_SOCKLEN_T *) &fromlen);
847
848 /* Reenable CONNECTION events */
849 Enable(GSOCK_CONNECTION);
850
851 if (connection->m_fd == INVALID_SOCKET)
852 {
853 if (errno == EWOULDBLOCK)
854 m_error = GSOCK_WOULDBLOCK;
855 else
856 m_error = GSOCK_IOERR;
857
858 delete connection;
859 return NULL;
860 }
861
862 /* Initialize all fields */
863 connection->m_server = false;
864 connection->m_stream = true;
865
866 /* Setup the peer address field */
867 connection->m_peer = GAddress_new();
868 if (!connection->m_peer)
869 {
870 delete connection;
871 m_error = GSOCK_MEMERR;
872 return NULL;
873 }
874
875 err = _GAddress_translate_from(connection->m_peer, &from, fromlen);
876 if (err != GSOCK_NOERROR)
877 {
878 delete connection;
879 m_error = err;
880 return NULL;
881 }
882
883 #if defined(__EMX__) || defined(__VISAGECPP__)
884 ioctl(connection->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
885 #else
886 ioctl(connection->m_fd, FIONBIO, &arg);
887 #endif
888 gs_gui_functions->Enable_Events(connection);
889
890 return connection;
891 }
892
893 bool GSocket::SetReusable()
894 {
895 /* socket must not be null, and must not be in use/already bound */
896 if (this && m_fd == INVALID_SOCKET)
897 {
898 m_reusable = true;
899
900 return true;
901 }
902
903 return false;
904 }
905
906 /* Client specific parts */
907
908 /* GSocket_Connect:
909 * For stream (connection oriented) sockets, GSocket_Connect() tries
910 * to establish a client connection to a server using the peer address
911 * as established with GSocket_SetPeer(). Returns GSOCK_NOERROR if the
912 * connection has been successfully established, or one of the error
913 * codes listed below. Note that for nonblocking sockets, a return
914 * value of GSOCK_WOULDBLOCK doesn't mean a failure. The connection
915 * request can be completed later; you should use GSocket_Select()
916 * to poll for GSOCK_CONNECTION | GSOCK_LOST, or wait for the
917 * corresponding asynchronous events.
918 *
919 * For datagram (non connection oriented) sockets, GSocket_Connect()
920 * just sets the peer address established with GSocket_SetPeer() as
921 * default destination.
922 *
923 * Error codes:
924 * GSOCK_INVSOCK - the socket is in use or not valid.
925 * GSOCK_INVADDR - the peer address has not been established.
926 * GSOCK_TIMEDOUT - timeout, the connection failed.
927 * GSOCK_WOULDBLOCK - connection in progress (nonblocking sockets only)
928 * GSOCK_MEMERR - couldn't allocate memory.
929 * GSOCK_IOERR - low-level error.
930 */
931 GSocketError GSocket::Connect(GSocketStream stream)
932 {
933 int err, ret;
934 int arg = 1;
935
936 assert(this);
937
938 /* Enable CONNECTION events (needed for nonblocking connections) */
939 Enable(GSOCK_CONNECTION);
940
941 if (m_fd != INVALID_SOCKET)
942 {
943 m_error = GSOCK_INVSOCK;
944 return GSOCK_INVSOCK;
945 }
946
947 if (!m_peer)
948 {
949 m_error = GSOCK_INVADDR;
950 return GSOCK_INVADDR;
951 }
952
953 /* Streamed or dgram socket? */
954 m_stream = (stream == GSOCK_STREAMED);
955 m_server = false;
956 m_establishing = false;
957
958 /* Create the socket */
959 m_fd = socket(m_peer->m_realfamily,
960 m_stream? SOCK_STREAM : SOCK_DGRAM, 0);
961
962 if (m_fd == INVALID_SOCKET)
963 {
964 m_error = GSOCK_IOERR;
965 return GSOCK_IOERR;
966 }
967
968 /* FreeBSD variants can't use MSG_NOSIGNAL, and instead use a socket option */
969 #ifdef SO_NOSIGPIPE
970 setsockopt(m_fd, SOL_SOCKET, SO_NOSIGPIPE, (const char*)&arg, sizeof(u_long));
971 #endif
972
973 #if defined(__EMX__) || defined(__VISAGECPP__)
974 ioctl(m_fd, FIONBIO, (char*)&arg, sizeof(arg));
975 #else
976 ioctl(m_fd, FIONBIO, &arg);
977 #endif
978
979 // If the reuse flag is set, use the applicable socket reuse flags(s)
980 if (m_reusable)
981 {
982 setsockopt(m_fd, SOL_SOCKET, SO_REUSEADDR, (const char*)&arg, sizeof(u_long));
983 #ifdef SO_REUSEPORT
984 setsockopt(m_fd, SOL_SOCKET, SO_REUSEPORT, (const char*)&arg, sizeof(u_long));
985 #endif
986 }
987
988 // If a local address has been set, then we need to bind to it before calling connect
989 if (m_local && m_local->m_addr)
990 {
991 bind(m_fd, m_local->m_addr, m_local->m_len);
992 }
993
994 /* Connect it to the peer address, with a timeout (see below) */
995 ret = connect(m_fd, m_peer->m_addr, m_peer->m_len);
996
997 /* We only call Enable_Events if we know we aren't shutting down the socket.
998 * NB: Enable_Events needs to be called whether the socket is blocking or
999 * non-blocking, it just shouldn't be called prior to knowing there is a
1000 * connection _if_ blocking sockets are being used.
1001 * If connect above returns 0, we are already connected and need to make the
1002 * call to Enable_Events now.
1003 */
1004
1005 if (m_non_blocking || ret == 0)
1006 gs_gui_functions->Enable_Events(this);
1007
1008 if (ret == -1)
1009 {
1010 err = errno;
1011
1012 /* If connect failed with EINPROGRESS and the GSocket object
1013 * is in blocking mode, we select() for the specified timeout
1014 * checking for writability to see if the connection request
1015 * completes.
1016 */
1017 if ((err == EINPROGRESS) && (!m_non_blocking))
1018 {
1019 if (Output_Timeout() == GSOCK_TIMEDOUT)
1020 {
1021 Close();
1022 /* m_error is set in _GSocket_Output_Timeout */
1023 return GSOCK_TIMEDOUT;
1024 }
1025 else
1026 {
1027 int error;
1028 SOCKOPTLEN_T len = sizeof(error);
1029
1030 getsockopt(m_fd, SOL_SOCKET, SO_ERROR, (char*) &error, &len);
1031
1032 gs_gui_functions->Enable_Events(this);
1033
1034 if (!error)
1035 return GSOCK_NOERROR;
1036 }
1037 }
1038
1039 /* If connect failed with EINPROGRESS and the GSocket object
1040 * is set to nonblocking, we set m_error to GSOCK_WOULDBLOCK
1041 * (and return GSOCK_WOULDBLOCK) but we don't close the socket;
1042 * this way if the connection completes, a GSOCK_CONNECTION
1043 * event will be generated, if enabled.
1044 */
1045 if ((err == EINPROGRESS) && (m_non_blocking))
1046 {
1047 m_establishing = true;
1048 m_error = GSOCK_WOULDBLOCK;
1049 return GSOCK_WOULDBLOCK;
1050 }
1051
1052 /* If connect failed with an error other than EINPROGRESS,
1053 * then the call to GSocket_Connect has failed.
1054 */
1055 Close();
1056 m_error = GSOCK_IOERR;
1057
1058 return GSOCK_IOERR;
1059 }
1060
1061 return GSOCK_NOERROR;
1062 }
1063
1064 /* Datagram sockets */
1065
1066 /* GSocket_SetNonOriented:
1067 * Sets up this socket as a non-connection oriented (datagram) socket.
1068 * Before using this function, the local address must have been set
1069 * with GSocket_SetLocal(), or the call will fail. Returns GSOCK_NOERROR
1070 * on success, or one of the following otherwise.
1071 *
1072 * Error codes:
1073 * GSOCK_INVSOCK - the socket is in use.
1074 * GSOCK_INVADDR - the local address has not been set.
1075 * GSOCK_IOERR - low-level error.
1076 */
1077 GSocketError GSocket::SetNonOriented()
1078 {
1079 int arg = 1;
1080
1081 assert(this);
1082
1083 if (m_fd != INVALID_SOCKET)
1084 {
1085 m_error = GSOCK_INVSOCK;
1086 return GSOCK_INVSOCK;
1087 }
1088
1089 if (!m_local)
1090 {
1091 m_error = GSOCK_INVADDR;
1092 return GSOCK_INVADDR;
1093 }
1094
1095 /* Initialize all fields */
1096 m_stream = false;
1097 m_server = false;
1098
1099 /* Create the socket */
1100 m_fd = socket(m_local->m_realfamily, SOCK_DGRAM, 0);
1101
1102 if (m_fd == INVALID_SOCKET)
1103 {
1104 m_error = GSOCK_IOERR;
1105 return GSOCK_IOERR;
1106 }
1107 #if defined(__EMX__) || defined(__VISAGECPP__)
1108 ioctl(m_fd, FIONBIO, (char*)&arg, sizeof(arg));
1109 #else
1110 ioctl(m_fd, FIONBIO, &arg);
1111 #endif
1112 gs_gui_functions->Enable_Events(this);
1113
1114 /* Bind to the local address,
1115 * and retrieve the actual address bound.
1116 */
1117 if ((bind(m_fd, m_local->m_addr, m_local->m_len) != 0) ||
1118 (getsockname(m_fd,
1119 m_local->m_addr,
1120 (WX_SOCKLEN_T *) &m_local->m_len) != 0))
1121 {
1122 Close();
1123 m_error = GSOCK_IOERR;
1124 return GSOCK_IOERR;
1125 }
1126
1127 return GSOCK_NOERROR;
1128 }
1129
1130 /* Generic IO */
1131
1132 /* Like recv(), send(), ... */
1133 int GSocket::Read(char *buffer, int size)
1134 {
1135 int ret;
1136
1137 assert(this);
1138
1139 if (m_fd == INVALID_SOCKET || m_server)
1140 {
1141 m_error = GSOCK_INVSOCK;
1142 return -1;
1143 }
1144
1145 /* Disable events during query of socket status */
1146 Disable(GSOCK_INPUT);
1147
1148 /* If the socket is blocking, wait for data (with a timeout) */
1149 if (Input_Timeout() == GSOCK_TIMEDOUT) {
1150 m_error = GSOCK_TIMEDOUT;
1151 /* Don't return here immediately, otherwise socket events would not be
1152 * re-enabled! */
1153 ret = -1;
1154 }
1155 else
1156 {
1157 /* Read the data */
1158 if (m_stream)
1159 ret = Recv_Stream(buffer, size);
1160 else
1161 ret = Recv_Dgram(buffer, size);
1162
1163 /* If recv returned zero, then the connection is lost, and errno is not set.
1164 * Otherwise, recv has returned an error (-1), in which case we have lost the
1165 * socket only if errno does _not_ indicate that there may be more data to read.
1166 */
1167 if (ret == 0)
1168 {
1169 m_error = GSOCK_IOERR;
1170 m_detected = GSOCK_LOST_FLAG;
1171 Close();
1172 // Signal an error for return
1173 return -1;
1174 }
1175 else if (ret == -1)
1176 {
1177 if ((errno == EWOULDBLOCK) || (errno == EAGAIN))
1178 m_error = GSOCK_WOULDBLOCK;
1179 else
1180 m_error = GSOCK_IOERR;
1181 }
1182 }
1183
1184 /* Enable events again now that we are done processing */
1185 Enable(GSOCK_INPUT);
1186
1187 return ret;
1188 }
1189
1190 int GSocket::Write(const char *buffer, int size)
1191 {
1192 int ret;
1193
1194 assert(this);
1195
1196 GSocket_Debug(( "GSocket_Write #1, size %d\n", size ));
1197
1198 if (m_fd == INVALID_SOCKET || m_server)
1199 {
1200 m_error = GSOCK_INVSOCK;
1201 return -1;
1202 }
1203
1204 GSocket_Debug(( "GSocket_Write #2, size %d\n", size ));
1205
1206 /* If the socket is blocking, wait for writability (with a timeout) */
1207 if (Output_Timeout() == GSOCK_TIMEDOUT)
1208 return -1;
1209
1210 GSocket_Debug(( "GSocket_Write #3, size %d\n", size ));
1211
1212 /* Write the data */
1213 if (m_stream)
1214 ret = Send_Stream(buffer, size);
1215 else
1216 ret = Send_Dgram(buffer, size);
1217
1218 GSocket_Debug(( "GSocket_Write #4, size %d\n", size ));
1219
1220 if (ret == -1)
1221 {
1222 if ((errno == EWOULDBLOCK) || (errno == EAGAIN))
1223 {
1224 m_error = GSOCK_WOULDBLOCK;
1225 GSocket_Debug(( "GSocket_Write error WOULDBLOCK\n" ));
1226 }
1227 else
1228 {
1229 m_error = GSOCK_IOERR;
1230 GSocket_Debug(( "GSocket_Write error IOERR\n" ));
1231 }
1232
1233 /* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
1234 * in MSW). Once the first OUTPUT event is received, users can assume
1235 * that the socket is writable until a read operation fails. Only then
1236 * will further OUTPUT events be posted.
1237 */
1238 Enable(GSOCK_OUTPUT);
1239
1240 return -1;
1241 }
1242
1243 GSocket_Debug(( "GSocket_Write #5, size %d ret %d\n", size, ret ));
1244
1245 return ret;
1246 }
1247
1248 /* GSocket_Select:
1249 * Polls the socket to determine its status. This function will
1250 * check for the events specified in the 'flags' parameter, and
1251 * it will return a mask indicating which operations can be
1252 * performed. This function won't block, regardless of the
1253 * mode (blocking | nonblocking) of the socket.
1254 */
1255 GSocketEventFlags GSocket::Select(GSocketEventFlags flags)
1256 {
1257 if (!gs_gui_functions->CanUseEventLoop())
1258 {
1259
1260 GSocketEventFlags result = 0;
1261 fd_set readfds;
1262 fd_set writefds;
1263 fd_set exceptfds;
1264 struct timeval tv;
1265
1266 assert(this);
1267
1268 if (m_fd == -1)
1269 return (GSOCK_LOST_FLAG & flags);
1270
1271 /* Do not use a static struct, Linux can garble it */
1272 tv.tv_sec = m_timeout / 1000;
1273 tv.tv_usec = (m_timeout % 1000) * 1000;
1274
1275 wxFD_ZERO(&readfds);
1276 wxFD_ZERO(&writefds);
1277 wxFD_ZERO(&exceptfds);
1278 wxFD_SET(m_fd, &readfds);
1279 if (flags & GSOCK_OUTPUT_FLAG || flags & GSOCK_CONNECTION_FLAG)
1280 wxFD_SET(m_fd, &writefds);
1281 wxFD_SET(m_fd, &exceptfds);
1282
1283 /* Check 'sticky' CONNECTION flag first */
1284 result |= (GSOCK_CONNECTION_FLAG & m_detected);
1285
1286 /* If we have already detected a LOST event, then don't try
1287 * to do any further processing.
1288 */
1289 if ((m_detected & GSOCK_LOST_FLAG) != 0)
1290 {
1291 m_establishing = false;
1292
1293 return (GSOCK_LOST_FLAG & flags);
1294 }
1295
1296 /* Try select now */
1297 if (select(m_fd + 1, &readfds, &writefds, &exceptfds, &tv) <= 0)
1298 {
1299 /* What to do here? */
1300 return (result & flags);
1301 }
1302
1303 /* Check for exceptions and errors */
1304 if (wxFD_ISSET(m_fd, &exceptfds))
1305 {
1306 m_establishing = false;
1307 m_detected = GSOCK_LOST_FLAG;
1308
1309 /* LOST event: Abort any further processing */
1310 return (GSOCK_LOST_FLAG & flags);
1311 }
1312
1313 /* Check for readability */
1314 if (wxFD_ISSET(m_fd, &readfds))
1315 {
1316 result |= GSOCK_INPUT_FLAG;
1317
1318 if (m_server && m_stream)
1319 {
1320 /* This is a TCP server socket that detected a connection.
1321 While the INPUT_FLAG is also set, it doesn't matter on
1322 this kind of sockets, as we can only Accept() from them. */
1323 result |= GSOCK_CONNECTION_FLAG;
1324 m_detected |= GSOCK_CONNECTION_FLAG;
1325 }
1326 }
1327
1328 /* Check for writability */
1329 if (wxFD_ISSET(m_fd, &writefds))
1330 {
1331 if (m_establishing && !m_server)
1332 {
1333 int error;
1334 SOCKOPTLEN_T len = sizeof(error);
1335
1336 m_establishing = false;
1337
1338 getsockopt(m_fd, SOL_SOCKET, SO_ERROR, (char*)&error, &len);
1339
1340 if (error)
1341 {
1342 m_detected = GSOCK_LOST_FLAG;
1343
1344 /* LOST event: Abort any further processing */
1345 return (GSOCK_LOST_FLAG & flags);
1346 }
1347 else
1348 {
1349 result |= GSOCK_CONNECTION_FLAG;
1350 m_detected |= GSOCK_CONNECTION_FLAG;
1351 }
1352 }
1353 else
1354 {
1355 result |= GSOCK_OUTPUT_FLAG;
1356 }
1357 }
1358
1359 return (result & flags);
1360
1361 }
1362 else
1363 {
1364 assert(this);
1365 return flags & m_detected;
1366 }
1367 }
1368
1369 /* Flags */
1370
1371 /* GSocket_SetNonBlocking:
1372 * Sets the socket to non-blocking mode. All IO calls will return
1373 * immediately.
1374 */
1375 void GSocket::SetNonBlocking(bool non_block)
1376 {
1377 assert(this);
1378
1379 GSocket_Debug( ("GSocket_SetNonBlocking: %d\n", (int)non_block) );
1380
1381 m_non_blocking = non_block;
1382 }
1383
1384 /* GSocket_SetTimeout:
1385 * Sets the timeout for blocking calls. Time is expressed in
1386 * milliseconds.
1387 */
1388 void GSocket::SetTimeout(unsigned long millisec)
1389 {
1390 assert(this);
1391
1392 m_timeout = millisec;
1393 }
1394
1395 /* GSocket_GetError:
1396 * Returns the last error occurred for this socket. Note that successful
1397 * operations do not clear this back to GSOCK_NOERROR, so use it only
1398 * after an error.
1399 */
1400 GSocketError WXDLLIMPEXP_NET GSocket::GetError()
1401 {
1402 assert(this);
1403
1404 return m_error;
1405 }
1406
1407 /* Callbacks */
1408
1409 /* GSOCK_INPUT:
1410 * There is data to be read in the input buffer. If, after a read
1411 * operation, there is still data available, the callback function will
1412 * be called again.
1413 * GSOCK_OUTPUT:
1414 * The socket is available for writing. That is, the next write call
1415 * won't block. This event is generated only once, when the connection is
1416 * first established, and then only if a call failed with GSOCK_WOULDBLOCK,
1417 * when the output buffer empties again. This means that the app should
1418 * assume that it can write since the first OUTPUT event, and no more
1419 * OUTPUT events will be generated unless an error occurs.
1420 * GSOCK_CONNECTION:
1421 * Connection successfully established, for client sockets, or incoming
1422 * client connection, for server sockets. Wait for this event (also watch
1423 * out for GSOCK_LOST) after you issue a nonblocking GSocket_Connect() call.
1424 * GSOCK_LOST:
1425 * The connection is lost (or a connection request failed); this could
1426 * be due to a failure, or due to the peer closing it gracefully.
1427 */
1428
1429 /* GSocket_SetCallback:
1430 * Enables the callbacks specified by 'flags'. Note that 'flags'
1431 * may be a combination of flags OR'ed toghether, so the same
1432 * callback function can be made to accept different events.
1433 * The callback function must have the following prototype:
1434 *
1435 * void function(GSocket *socket, GSocketEvent event, char *cdata)
1436 */
1437 void GSocket::SetCallback(GSocketEventFlags flags,
1438 GSocketCallback callback, char *cdata)
1439 {
1440 int count;
1441
1442 assert(this);
1443
1444 for (count = 0; count < GSOCK_MAX_EVENT; count++)
1445 {
1446 if ((flags & (1 << count)) != 0)
1447 {
1448 m_cbacks[count] = callback;
1449 m_data[count] = cdata;
1450 }
1451 }
1452 }
1453
1454 /* GSocket_UnsetCallback:
1455 * Disables all callbacks specified by 'flags', which may be a
1456 * combination of flags OR'ed toghether.
1457 */
1458 void GSocket::UnsetCallback(GSocketEventFlags flags)
1459 {
1460 int count;
1461
1462 assert(this);
1463
1464 for (count = 0; count < GSOCK_MAX_EVENT; count++)
1465 {
1466 if ((flags & (1 << count)) != 0)
1467 {
1468 m_cbacks[count] = NULL;
1469 m_data[count] = NULL;
1470 }
1471 }
1472 }
1473
1474 GSocketError GSocket::GetSockOpt(int level, int optname,
1475 void *optval, int *optlen)
1476 {
1477 if (getsockopt(m_fd, level, optname, (char*)optval, (SOCKOPTLEN_T*)optlen) == 0)
1478 return GSOCK_NOERROR;
1479
1480 return GSOCK_OPTERR;
1481 }
1482
1483 GSocketError GSocket::SetSockOpt(int level, int optname,
1484 const void *optval, int optlen)
1485 {
1486 if (setsockopt(m_fd, level, optname, (const char*)optval, optlen) == 0)
1487 return GSOCK_NOERROR;
1488
1489 return GSOCK_OPTERR;
1490 }
1491
1492 #define CALL_CALLBACK(socket, event) { \
1493 socket->Disable(event); \
1494 if (socket->m_cbacks[event]) \
1495 socket->m_cbacks[event](socket, event, socket->m_data[event]); \
1496 }
1497
1498
1499 void GSocket::Enable(GSocketEvent event)
1500 {
1501 m_detected &= ~(1 << event);
1502 gs_gui_functions->Install_Callback(this, event);
1503 }
1504
1505 void GSocket::Disable(GSocketEvent event)
1506 {
1507 m_detected |= (1 << event);
1508 gs_gui_functions->Uninstall_Callback(this, event);
1509 }
1510
1511 /* _GSocket_Input_Timeout:
1512 * For blocking sockets, wait until data is available or
1513 * until timeout ellapses.
1514 */
1515 GSocketError GSocket::Input_Timeout()
1516 {
1517 struct timeval tv;
1518 fd_set readfds;
1519 int ret;
1520
1521 /* Linux select() will overwrite the struct on return */
1522 tv.tv_sec = (m_timeout / 1000);
1523 tv.tv_usec = (m_timeout % 1000) * 1000;
1524
1525 if (!m_non_blocking)
1526 {
1527 wxFD_ZERO(&readfds);
1528 wxFD_SET(m_fd, &readfds);
1529 ret = select(m_fd + 1, &readfds, NULL, NULL, &tv);
1530 if (ret == 0)
1531 {
1532 GSocket_Debug(( "GSocket_Input_Timeout, select returned 0\n" ));
1533 m_error = GSOCK_TIMEDOUT;
1534 return GSOCK_TIMEDOUT;
1535 }
1536
1537 if (ret == -1)
1538 {
1539 GSocket_Debug(( "GSocket_Input_Timeout, select returned -1\n" ));
1540 if (errno == EBADF) { GSocket_Debug(( "Invalid file descriptor\n" )); }
1541 if (errno == EINTR) { GSocket_Debug(( "A non blocked signal was caught\n" )); }
1542 if (errno == EINVAL) { GSocket_Debug(( "The highest number descriptor is negative\n" )); }
1543 if (errno == ENOMEM) { GSocket_Debug(( "Not enough memory\n" )); }
1544 m_error = GSOCK_TIMEDOUT;
1545 return GSOCK_TIMEDOUT;
1546 }
1547 }
1548
1549 return GSOCK_NOERROR;
1550 }
1551
1552 /* _GSocket_Output_Timeout:
1553 * For blocking sockets, wait until data can be sent without
1554 * blocking or until timeout ellapses.
1555 */
1556 GSocketError GSocket::Output_Timeout()
1557 {
1558 struct timeval tv;
1559 fd_set writefds;
1560 int ret;
1561
1562 /* Linux select() will overwrite the struct on return */
1563 tv.tv_sec = (m_timeout / 1000);
1564 tv.tv_usec = (m_timeout % 1000) * 1000;
1565
1566 GSocket_Debug( ("m_non_blocking has: %d\n", (int)m_non_blocking) );
1567
1568 if (!m_non_blocking)
1569 {
1570 wxFD_ZERO(&writefds);
1571 wxFD_SET(m_fd, &writefds);
1572 ret = select(m_fd + 1, NULL, &writefds, NULL, &tv);
1573 if (ret == 0)
1574 {
1575 GSocket_Debug(( "GSocket_Output_Timeout, select returned 0\n" ));
1576 m_error = GSOCK_TIMEDOUT;
1577 return GSOCK_TIMEDOUT;
1578 }
1579
1580 if (ret == -1)
1581 {
1582 GSocket_Debug(( "GSocket_Output_Timeout, select returned -1\n" ));
1583 if (errno == EBADF) { GSocket_Debug(( "Invalid file descriptor\n" )); }
1584 if (errno == EINTR) { GSocket_Debug(( "A non blocked signal was caught\n" )); }
1585 if (errno == EINVAL) { GSocket_Debug(( "The highest number descriptor is negative\n" )); }
1586 if (errno == ENOMEM) { GSocket_Debug(( "Not enough memory\n" )); }
1587 m_error = GSOCK_TIMEDOUT;
1588 return GSOCK_TIMEDOUT;
1589 }
1590
1591 if ( ! wxFD_ISSET(m_fd, &writefds) )
1592 {
1593 GSocket_Debug(( "GSocket_Output_Timeout is buggy!\n" ));
1594 }
1595 else
1596 {
1597 GSocket_Debug(( "GSocket_Output_Timeout seems correct\n" ));
1598 }
1599 }
1600 else
1601 {
1602 GSocket_Debug(( "GSocket_Output_Timeout, didn't try select!\n" ));
1603 }
1604
1605 return GSOCK_NOERROR;
1606 }
1607
1608 int GSocket::Recv_Stream(char *buffer, int size)
1609 {
1610 int ret;
1611 do
1612 {
1613 ret = recv(m_fd, buffer, size, GSOCKET_MSG_NOSIGNAL);
1614 }
1615 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
1616
1617 return ret;
1618 }
1619
1620 int GSocket::Recv_Dgram(char *buffer, int size)
1621 {
1622 struct sockaddr from;
1623 WX_SOCKLEN_T fromlen = sizeof(from);
1624 int ret;
1625 GSocketError err;
1626
1627 fromlen = sizeof(from);
1628
1629 do
1630 {
1631 ret = recvfrom(m_fd, buffer, size, 0, &from, (WX_SOCKLEN_T *) &fromlen);
1632 }
1633 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
1634
1635 if (ret == -1)
1636 return -1;
1637
1638 /* Translate a system address into a GSocket address */
1639 if (!m_peer)
1640 {
1641 m_peer = GAddress_new();
1642 if (!m_peer)
1643 {
1644 m_error = GSOCK_MEMERR;
1645 return -1;
1646 }
1647 }
1648
1649 err = _GAddress_translate_from(m_peer, &from, fromlen);
1650 if (err != GSOCK_NOERROR)
1651 {
1652 GAddress_destroy(m_peer);
1653 m_peer = NULL;
1654 m_error = err;
1655 return -1;
1656 }
1657
1658 return ret;
1659 }
1660
1661 int GSocket::Send_Stream(const char *buffer, int size)
1662 {
1663 int ret;
1664
1665 MASK_SIGNAL();
1666
1667 do
1668 {
1669 ret = send(m_fd, (char *)buffer, size, GSOCKET_MSG_NOSIGNAL);
1670 }
1671 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
1672
1673 UNMASK_SIGNAL();
1674
1675 return ret;
1676 }
1677
1678 int GSocket::Send_Dgram(const char *buffer, int size)
1679 {
1680 struct sockaddr *addr;
1681 int len, ret;
1682 GSocketError err;
1683
1684 if (!m_peer)
1685 {
1686 m_error = GSOCK_INVADDR;
1687 return -1;
1688 }
1689
1690 err = _GAddress_translate_to(m_peer, &addr, &len);
1691 if (err != GSOCK_NOERROR)
1692 {
1693 m_error = err;
1694 return -1;
1695 }
1696
1697 MASK_SIGNAL();
1698
1699 do
1700 {
1701 ret = sendto(m_fd, (char *)buffer, size, 0, addr, len);
1702 }
1703 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
1704
1705 UNMASK_SIGNAL();
1706
1707 /* Frees memory allocated from _GAddress_translate_to */
1708 free(addr);
1709
1710 return ret;
1711 }
1712
1713 void GSocket::Detected_Read()
1714 {
1715 char c;
1716
1717 /* Safeguard against straggling call to Detected_Read */
1718 if (m_fd == INVALID_SOCKET)
1719 {
1720 return;
1721 }
1722
1723 /* If we have already detected a LOST event, then don't try
1724 * to do any further processing.
1725 */
1726 if ((m_detected & GSOCK_LOST_FLAG) != 0)
1727 {
1728 m_establishing = false;
1729
1730 CALL_CALLBACK(this, GSOCK_LOST);
1731 Shutdown();
1732 return;
1733 }
1734
1735 int num = recv(m_fd, &c, 1, MSG_PEEK | GSOCKET_MSG_NOSIGNAL);
1736
1737 if (num > 0)
1738 {
1739 CALL_CALLBACK(this, GSOCK_INPUT);
1740 }
1741 else
1742 {
1743 if (m_server && m_stream)
1744 {
1745 CALL_CALLBACK(this, GSOCK_CONNECTION);
1746 }
1747 else
1748 {
1749 /* Do not throw a lost event in cases where the socket isn't really lost */
1750 if ((errno == EWOULDBLOCK) || (errno == EAGAIN) || (errno == EINTR))
1751 {
1752 CALL_CALLBACK(this, GSOCK_INPUT);
1753 }
1754 else
1755 {
1756 CALL_CALLBACK(this, GSOCK_LOST);
1757 Shutdown();
1758 }
1759 }
1760 }
1761 }
1762
1763 void GSocket::Detected_Write()
1764 {
1765 /* If we have already detected a LOST event, then don't try
1766 * to do any further processing.
1767 */
1768 if ((m_detected & GSOCK_LOST_FLAG) != 0)
1769 {
1770 m_establishing = false;
1771
1772 CALL_CALLBACK(this, GSOCK_LOST);
1773 Shutdown();
1774 return;
1775 }
1776
1777 if (m_establishing && !m_server)
1778 {
1779 int error;
1780 SOCKOPTLEN_T len = sizeof(error);
1781
1782 m_establishing = false;
1783
1784 getsockopt(m_fd, SOL_SOCKET, SO_ERROR, (char*)&error, &len);
1785
1786 if (error)
1787 {
1788 CALL_CALLBACK(this, GSOCK_LOST);
1789 Shutdown();
1790 }
1791 else
1792 {
1793 CALL_CALLBACK(this, GSOCK_CONNECTION);
1794 /* We have to fire this event by hand because CONNECTION (for clients)
1795 * and OUTPUT are internally the same and we just disabled CONNECTION
1796 * events with the above macro.
1797 */
1798 CALL_CALLBACK(this, GSOCK_OUTPUT);
1799 }
1800 }
1801 else
1802 {
1803 CALL_CALLBACK(this, GSOCK_OUTPUT);
1804 }
1805 }
1806
1807 /* Compatibility functions for GSocket */
1808 GSocket *GSocket_new(void)
1809 {
1810 GSocket *newsocket = new GSocket();
1811 if (newsocket->IsOk())
1812 return newsocket;
1813
1814 delete newsocket;
1815
1816 return NULL;
1817 }
1818
1819 /*
1820 * -------------------------------------------------------------------------
1821 * GAddress
1822 * -------------------------------------------------------------------------
1823 */
1824
1825 /* CHECK_ADDRESS verifies that the current address family is either
1826 * GSOCK_NOFAMILY or GSOCK_*family*, and if it is GSOCK_NOFAMILY, it
1827 * initalizes it to be a GSOCK_*family*. In other cases, it returns
1828 * an appropiate error code.
1829 *
1830 * CHECK_ADDRESS_RETVAL does the same but returning 'retval' on error.
1831 */
1832 #define CHECK_ADDRESS(address, family) \
1833 { \
1834 if (address->m_family == GSOCK_NOFAMILY) \
1835 if (_GAddress_Init_##family(address) != GSOCK_NOERROR) \
1836 return address->m_error; \
1837 if (address->m_family != GSOCK_##family) \
1838 { \
1839 address->m_error = GSOCK_INVADDR; \
1840 return GSOCK_INVADDR; \
1841 } \
1842 }
1843
1844 #define CHECK_ADDRESS_RETVAL(address, family, retval) \
1845 { \
1846 if (address->m_family == GSOCK_NOFAMILY) \
1847 if (_GAddress_Init_##family(address) != GSOCK_NOERROR) \
1848 return retval; \
1849 if (address->m_family != GSOCK_##family) \
1850 { \
1851 address->m_error = GSOCK_INVADDR; \
1852 return retval; \
1853 } \
1854 }
1855
1856
1857 GAddress *GAddress_new(void)
1858 {
1859 GAddress *address;
1860
1861 if ((address = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1862 return NULL;
1863
1864 address->m_family = GSOCK_NOFAMILY;
1865 address->m_addr = NULL;
1866 address->m_len = 0;
1867
1868 return address;
1869 }
1870
1871 GAddress *GAddress_copy(GAddress *address)
1872 {
1873 GAddress *addr2;
1874
1875 assert(address != NULL);
1876
1877 if ((addr2 = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1878 return NULL;
1879
1880 memcpy(addr2, address, sizeof(GAddress));
1881
1882 if (address->m_addr && address->m_len > 0)
1883 {
1884 addr2->m_addr = (struct sockaddr *)malloc(addr2->m_len);
1885 if (addr2->m_addr == NULL)
1886 {
1887 free(addr2);
1888 return NULL;
1889 }
1890 memcpy(addr2->m_addr, address->m_addr, addr2->m_len);
1891 }
1892
1893 return addr2;
1894 }
1895
1896 void GAddress_destroy(GAddress *address)
1897 {
1898 assert(address != NULL);
1899
1900 if (address->m_addr)
1901 free(address->m_addr);
1902
1903 free(address);
1904 }
1905
1906 void GAddress_SetFamily(GAddress *address, GAddressType type)
1907 {
1908 assert(address != NULL);
1909
1910 address->m_family = type;
1911 }
1912
1913 GAddressType GAddress_GetFamily(GAddress *address)
1914 {
1915 assert(address != NULL);
1916
1917 return address->m_family;
1918 }
1919
1920 GSocketError _GAddress_translate_from(GAddress *address,
1921 struct sockaddr *addr, int len)
1922 {
1923 address->m_realfamily = addr->sa_family;
1924 switch (addr->sa_family)
1925 {
1926 case AF_INET:
1927 address->m_family = GSOCK_INET;
1928 break;
1929 case AF_UNIX:
1930 address->m_family = GSOCK_UNIX;
1931 break;
1932 #ifdef AF_INET6
1933 case AF_INET6:
1934 address->m_family = GSOCK_INET6;
1935 break;
1936 #endif
1937 default:
1938 {
1939 address->m_error = GSOCK_INVOP;
1940 return GSOCK_INVOP;
1941 }
1942 }
1943
1944 if (address->m_addr)
1945 free(address->m_addr);
1946
1947 address->m_len = len;
1948 address->m_addr = (struct sockaddr *)malloc(len);
1949
1950 if (address->m_addr == NULL)
1951 {
1952 address->m_error = GSOCK_MEMERR;
1953 return GSOCK_MEMERR;
1954 }
1955
1956 memcpy(address->m_addr, addr, len);
1957
1958 return GSOCK_NOERROR;
1959 }
1960
1961 GSocketError _GAddress_translate_to(GAddress *address,
1962 struct sockaddr **addr, int *len)
1963 {
1964 if (!address->m_addr)
1965 {
1966 address->m_error = GSOCK_INVADDR;
1967 return GSOCK_INVADDR;
1968 }
1969
1970 *len = address->m_len;
1971 *addr = (struct sockaddr *)malloc(address->m_len);
1972 if (*addr == NULL)
1973 {
1974 address->m_error = GSOCK_MEMERR;
1975 return GSOCK_MEMERR;
1976 }
1977
1978 memcpy(*addr, address->m_addr, address->m_len);
1979 return GSOCK_NOERROR;
1980 }
1981
1982 /*
1983 * -------------------------------------------------------------------------
1984 * Internet address family
1985 * -------------------------------------------------------------------------
1986 */
1987
1988 GSocketError _GAddress_Init_INET(GAddress *address)
1989 {
1990 address->m_len = sizeof(struct sockaddr_in);
1991 address->m_addr = (struct sockaddr *) malloc(address->m_len);
1992 if (address->m_addr == NULL)
1993 {
1994 address->m_error = GSOCK_MEMERR;
1995 return GSOCK_MEMERR;
1996 }
1997
1998 address->m_family = GSOCK_INET;
1999 address->m_realfamily = PF_INET;
2000 ((struct sockaddr_in *)address->m_addr)->sin_family = AF_INET;
2001 ((struct sockaddr_in *)address->m_addr)->sin_addr.s_addr = INADDR_ANY;
2002
2003 return GSOCK_NOERROR;
2004 }
2005
2006 GSocketError GAddress_INET_SetHostName(GAddress *address, const char *hostname)
2007 {
2008 struct hostent *he;
2009 struct in_addr *addr;
2010
2011 assert(address != NULL);
2012
2013 CHECK_ADDRESS(address, INET);
2014
2015 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
2016
2017 /* If it is a numeric host name, convert it now */
2018 #if defined(HAVE_INET_ATON)
2019 if (inet_aton(hostname, addr) == 0)
2020 {
2021 #elif defined(HAVE_INET_ADDR)
2022 if ( (addr->s_addr = inet_addr(hostname)) == (unsigned)-1 )
2023 {
2024 #else
2025 /* Use gethostbyname by default */
2026 #ifndef __WXMAC__
2027 int val = 1; /* VA doesn't like constants in conditional expressions */
2028 if (val)
2029 #endif
2030 {
2031 #endif
2032 struct in_addr *array_addr;
2033
2034 /* It is a real name, we solve it */
2035 struct hostent h;
2036 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
2037 struct hostent_data buffer;
2038 #else
2039 char buffer[1024];
2040 #endif
2041 int err;
2042 he = wxGethostbyname_r(hostname, &h, (void*)&buffer, sizeof(buffer), &err);
2043 if (he == NULL)
2044 {
2045 /* Reset to invalid address */
2046 addr->s_addr = INADDR_NONE;
2047 address->m_error = GSOCK_NOHOST;
2048 return GSOCK_NOHOST;
2049 }
2050
2051 array_addr = (struct in_addr *) *(he->h_addr_list);
2052 addr->s_addr = array_addr[0].s_addr;
2053 }
2054
2055 return GSOCK_NOERROR;
2056 }
2057
2058 GSocketError GAddress_INET_SetAnyAddress(GAddress *address)
2059 {
2060 return GAddress_INET_SetHostAddress(address, INADDR_ANY);
2061 }
2062
2063 GSocketError GAddress_INET_SetHostAddress(GAddress *address,
2064 unsigned long hostaddr)
2065 {
2066 struct in_addr *addr;
2067
2068 assert(address != NULL);
2069
2070 CHECK_ADDRESS(address, INET);
2071
2072 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
2073 addr->s_addr = htonl(hostaddr);
2074
2075 return GSOCK_NOERROR;
2076 }
2077
2078 GSocketError GAddress_INET_SetPortName(GAddress *address, const char *port,
2079 const char *protocol)
2080 {
2081 struct servent *se;
2082 struct sockaddr_in *addr;
2083
2084 assert(address != NULL);
2085 CHECK_ADDRESS(address, INET);
2086
2087 if (!port)
2088 {
2089 address->m_error = GSOCK_INVPORT;
2090 return GSOCK_INVPORT;
2091 }
2092
2093 #if defined(HAVE_FUNC_GETSERVBYNAME_R_4)
2094 struct servent_data buffer;
2095 #else
2096 char buffer[1024];
2097 #endif
2098 struct servent serv;
2099 se = wxGetservbyname_r(port, protocol, &serv,
2100 (void*)&buffer, sizeof(buffer));
2101 if (!se)
2102 {
2103 /* the cast to int suppresses compiler warnings about subscript having the
2104 type char */
2105 if (isdigit((int)port[0]))
2106 {
2107 int port_int;
2108
2109 port_int = atoi(port);
2110 addr = (struct sockaddr_in *)address->m_addr;
2111 addr->sin_port = htons(port_int);
2112 return GSOCK_NOERROR;
2113 }
2114
2115 address->m_error = GSOCK_INVPORT;
2116 return GSOCK_INVPORT;
2117 }
2118
2119 addr = (struct sockaddr_in *)address->m_addr;
2120 addr->sin_port = se->s_port;
2121
2122 return GSOCK_NOERROR;
2123 }
2124
2125 GSocketError GAddress_INET_SetPort(GAddress *address, unsigned short port)
2126 {
2127 struct sockaddr_in *addr;
2128
2129 assert(address != NULL);
2130 CHECK_ADDRESS(address, INET);
2131
2132 addr = (struct sockaddr_in *)address->m_addr;
2133 addr->sin_port = htons(port);
2134
2135 return GSOCK_NOERROR;
2136 }
2137
2138 GSocketError GAddress_INET_GetHostName(GAddress *address, char *hostname, size_t sbuf)
2139 {
2140 struct hostent *he;
2141 char *addr_buf;
2142 struct sockaddr_in *addr;
2143
2144 assert(address != NULL);
2145 CHECK_ADDRESS(address, INET);
2146
2147 addr = (struct sockaddr_in *)address->m_addr;
2148 addr_buf = (char *)&(addr->sin_addr);
2149
2150 struct hostent temphost;
2151 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
2152 struct hostent_data buffer;
2153 #else
2154 char buffer[1024];
2155 #endif
2156 int err;
2157 he = wxGethostbyaddr_r(addr_buf, sizeof(addr->sin_addr), AF_INET, &temphost,
2158 (void*)&buffer, sizeof(buffer), &err);
2159 if (he == NULL)
2160 {
2161 address->m_error = GSOCK_NOHOST;
2162 return GSOCK_NOHOST;
2163 }
2164
2165 strncpy(hostname, he->h_name, sbuf);
2166
2167 return GSOCK_NOERROR;
2168 }
2169
2170 unsigned long GAddress_INET_GetHostAddress(GAddress *address)
2171 {
2172 struct sockaddr_in *addr;
2173
2174 assert(address != NULL);
2175 CHECK_ADDRESS_RETVAL(address, INET, 0);
2176
2177 addr = (struct sockaddr_in *)address->m_addr;
2178
2179 return ntohl(addr->sin_addr.s_addr);
2180 }
2181
2182 unsigned short GAddress_INET_GetPort(GAddress *address)
2183 {
2184 struct sockaddr_in *addr;
2185
2186 assert(address != NULL);
2187 CHECK_ADDRESS_RETVAL(address, INET, 0);
2188
2189 addr = (struct sockaddr_in *)address->m_addr;
2190 return ntohs(addr->sin_port);
2191 }
2192
2193 /*
2194 * -------------------------------------------------------------------------
2195 * Unix address family
2196 * -------------------------------------------------------------------------
2197 */
2198
2199 #ifndef __VISAGECPP__
2200 GSocketError _GAddress_Init_UNIX(GAddress *address)
2201 {
2202 address->m_len = sizeof(struct sockaddr_un);
2203 address->m_addr = (struct sockaddr *)malloc(address->m_len);
2204 if (address->m_addr == NULL)
2205 {
2206 address->m_error = GSOCK_MEMERR;
2207 return GSOCK_MEMERR;
2208 }
2209
2210 address->m_family = GSOCK_UNIX;
2211 address->m_realfamily = PF_UNIX;
2212 ((struct sockaddr_un *)address->m_addr)->sun_family = AF_UNIX;
2213 ((struct sockaddr_un *)address->m_addr)->sun_path[0] = 0;
2214
2215 return GSOCK_NOERROR;
2216 }
2217
2218 #define UNIX_SOCK_PATHLEN (sizeof(addr->sun_path)/sizeof(addr->sun_path[0]))
2219
2220 GSocketError GAddress_UNIX_SetPath(GAddress *address, const char *path)
2221 {
2222 struct sockaddr_un *addr;
2223
2224 assert(address != NULL);
2225
2226 CHECK_ADDRESS(address, UNIX);
2227
2228 addr = ((struct sockaddr_un *)address->m_addr);
2229 strncpy(addr->sun_path, path, UNIX_SOCK_PATHLEN);
2230 addr->sun_path[UNIX_SOCK_PATHLEN - 1] = '\0';
2231
2232 return GSOCK_NOERROR;
2233 }
2234
2235 GSocketError GAddress_UNIX_GetPath(GAddress *address, char *path, size_t sbuf)
2236 {
2237 struct sockaddr_un *addr;
2238
2239 assert(address != NULL);
2240 CHECK_ADDRESS(address, UNIX);
2241
2242 addr = (struct sockaddr_un *)address->m_addr;
2243
2244 strncpy(path, addr->sun_path, sbuf);
2245
2246 return GSOCK_NOERROR;
2247 }
2248 #endif /* !defined(__VISAGECPP__) */
2249 #endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */