no real changes, just moved wxSocketImplUnix ctor inline
[wxWidgets.git] / src / unix / sockunix.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: src/unix/sockunix.cpp
3 // Purpose: wxSocketImpl implementation for Unix systems
4 // Authors: Guilhem Lavaux, Guillermo Rodriguez Garcia, David Elliott,
5 // Vadim Zeitlin
6 // Created: April 1997
7 // RCS-ID: $Id$
8 // Copyright: (c) 1997 Guilhem Lavaux
9 // (c) 2008 Vadim Zeitlin
10 // Licence: wxWindows licence
11 /////////////////////////////////////////////////////////////////////////////
12
13
14 #include "wx/wxprec.h"
15
16 #if wxUSE_SOCKETS
17
18 #include "wx/private/fd.h"
19 #include "wx/private/socket.h"
20 #include "wx/unix/private/sockunix.h"
21
22 #if defined(__VISAGECPP__)
23 #define BSD_SELECT /* use Berkeley Sockets select */
24 #endif
25
26 #if defined(__WATCOMC__)
27 #include <errno.h>
28 #include <nerrno.h>
29 #endif
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 /* UnixWare reportedly needs this for FIONBIO definition */
137 #ifdef __UNIXWARE__
138 #include <sys/filio.h>
139 #endif
140
141 /*
142 * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
143 * on all systems. INADDR_BROADCAST should be fine to indicate an error.
144 */
145 #ifndef INADDR_NONE
146 #define INADDR_NONE INADDR_BROADCAST
147 #endif
148
149 #if defined(__VISAGECPP__) || defined(__WATCOMC__)
150
151 #define MASK_SIGNAL() {
152 #define UNMASK_SIGNAL() }
153
154 #else
155 extern "C" { typedef void (*wxSigHandler)(int); }
156
157 #define MASK_SIGNAL() \
158 { \
159 wxSigHandler old_handler = signal(SIGPIPE, SIG_IGN);
160
161 #define UNMASK_SIGNAL() \
162 signal(SIGPIPE, old_handler); \
163 }
164
165 #endif
166
167 /* If a SIGPIPE is issued by a socket call on a remotely closed socket,
168 the program will "crash" unless it explicitly handles the SIGPIPE.
169 By using MSG_NOSIGNAL, the SIGPIPE is suppressed. Later, we will
170 use SO_NOSIGPIPE (if available), the BSD equivalent. */
171 #ifdef MSG_NOSIGNAL
172 # define GSOCKET_MSG_NOSIGNAL MSG_NOSIGNAL
173 #else /* MSG_NOSIGNAL not available (FreeBSD including OS X) */
174 # define GSOCKET_MSG_NOSIGNAL 0
175 #endif /* MSG_NOSIGNAL */
176
177 #if wxUSE_THREADS && (defined(HAVE_GETHOSTBYNAME) || defined(HAVE_GETSERVBYNAME))
178 # include "wx/thread.h"
179 #endif
180
181 #if defined(HAVE_GETHOSTBYNAME)
182 static struct hostent * deepCopyHostent(struct hostent *h,
183 const struct hostent *he,
184 char *buffer, int size, int *err)
185 {
186 /* copy old structure */
187 memcpy(h, he, sizeof(struct hostent));
188
189 /* copy name */
190 int len = strlen(h->h_name);
191 if (len > size)
192 {
193 *err = ENOMEM;
194 return NULL;
195 }
196 memcpy(buffer, h->h_name, len);
197 buffer[len] = '\0';
198 h->h_name = buffer;
199
200 /* track position in the buffer */
201 int pos = len + 1;
202
203 /* reuse len to store address length */
204 len = h->h_length;
205
206 /* ensure pointer alignment */
207 unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
208 if(misalign < sizeof(char *))
209 pos += misalign;
210
211 /* leave space for pointer list */
212 char **p = h->h_addr_list, **q;
213 char **h_addr_list = (char **)(buffer + pos);
214 while(*(p++) != 0)
215 pos += sizeof(char *);
216
217 /* copy addresses and fill new pointer list */
218 for (p = h->h_addr_list, q = h_addr_list; *p != 0; p++, q++)
219 {
220 if (size < pos + len)
221 {
222 *err = ENOMEM;
223 return NULL;
224 }
225 memcpy(buffer + pos, *p, len); /* copy content */
226 *q = buffer + pos; /* set copied pointer to copied content */
227 pos += len;
228 }
229 *++q = 0; /* null terminate the pointer list */
230 h->h_addr_list = h_addr_list; /* copy pointer to pointers */
231
232 /* ensure word alignment of pointers */
233 misalign = sizeof(char *) - pos%sizeof(char *);
234 if(misalign < sizeof(char *))
235 pos += misalign;
236
237 /* leave space for pointer list */
238 p = h->h_aliases;
239 char **h_aliases = (char **)(buffer + pos);
240 while(*(p++) != 0)
241 pos += sizeof(char *);
242
243 /* copy aliases and fill new pointer list */
244 for (p = h->h_aliases, q = h_aliases; *p != 0; p++, q++)
245 {
246 len = strlen(*p);
247 if (size <= pos + len)
248 {
249 *err = ENOMEM;
250 return NULL;
251 }
252 memcpy(buffer + pos, *p, len); /* copy content */
253 buffer[pos + len] = '\0';
254 *q = buffer + pos; /* set copied pointer to copied content */
255 pos += len + 1;
256 }
257 *++q = 0; /* null terminate the pointer list */
258 h->h_aliases = h_aliases; /* copy pointer to pointers */
259
260 return h;
261 }
262 #endif
263
264 #if defined(HAVE_GETHOSTBYNAME) && wxUSE_THREADS
265 static wxMutex nameLock;
266 #endif
267 struct hostent * wxGethostbyname_r(const char *hostname, struct hostent *h,
268 void *buffer, int size, int *err)
269
270 {
271 struct hostent *he = NULL;
272 *err = 0;
273 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_6)
274 if (gethostbyname_r(hostname, h, (char*)buffer, size, &he, err))
275 he = NULL;
276 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_5)
277 he = gethostbyname_r(hostname, h, (char*)buffer, size, err);
278 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
279 if (gethostbyname_r(hostname, h, (struct hostent_data*) buffer))
280 {
281 he = NULL;
282 *err = h_errno;
283 }
284 else
285 he = h;
286 #elif defined(HAVE_GETHOSTBYNAME)
287 #if wxUSE_THREADS
288 wxMutexLocker locker(nameLock);
289 #endif
290 he = gethostbyname(hostname);
291 if (!he)
292 *err = h_errno;
293 else
294 he = deepCopyHostent(h, he, (char*)buffer, size, err);
295 #endif
296 return he;
297 }
298
299 #if defined(HAVE_GETHOSTBYNAME) && wxUSE_THREADS
300 static wxMutex addrLock;
301 #endif
302 struct hostent * wxGethostbyaddr_r(const char *addr_buf, int buf_size,
303 int proto, struct hostent *h,
304 void *buffer, int size, int *err)
305 {
306 struct hostent *he = NULL;
307 *err = 0;
308 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_6)
309 if (gethostbyaddr_r(addr_buf, buf_size, proto, h,
310 (char*)buffer, size, &he, err))
311 he = NULL;
312 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_5)
313 he = gethostbyaddr_r(addr_buf, buf_size, proto, h, (char*)buffer, size, err);
314 #elif defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
315 if (gethostbyaddr_r(addr_buf, buf_size, proto, h,
316 (struct hostent_data*) buffer))
317 {
318 he = NULL;
319 *err = h_errno;
320 }
321 else
322 he = h;
323 #elif defined(HAVE_GETHOSTBYNAME)
324 #if wxUSE_THREADS
325 wxMutexLocker locker(addrLock);
326 #endif
327 he = gethostbyaddr(addr_buf, buf_size, proto);
328 if (!he)
329 *err = h_errno;
330 else
331 he = deepCopyHostent(h, he, (char*)buffer, size, err);
332 #endif
333 return he;
334 }
335
336 #if defined(HAVE_GETSERVBYNAME)
337 static struct servent * deepCopyServent(struct servent *s,
338 const struct servent *se,
339 char *buffer, int size)
340 {
341 /* copy plain old structure */
342 memcpy(s, se, sizeof(struct servent));
343
344 /* copy name */
345 int len = strlen(s->s_name);
346 if (len >= size)
347 {
348 return NULL;
349 }
350 memcpy(buffer, s->s_name, len);
351 buffer[len] = '\0';
352 s->s_name = buffer;
353
354 /* track position in the buffer */
355 int pos = len + 1;
356
357 /* copy protocol */
358 len = strlen(s->s_proto);
359 if (pos + len >= size)
360 {
361 return NULL;
362 }
363 memcpy(buffer + pos, s->s_proto, len);
364 buffer[pos + len] = '\0';
365 s->s_proto = buffer + pos;
366
367 /* track position in the buffer */
368 pos += len + 1;
369
370 /* ensure pointer alignment */
371 unsigned int misalign = sizeof(char *) - pos%sizeof(char *);
372 if(misalign < sizeof(char *))
373 pos += misalign;
374
375 /* leave space for pointer list */
376 char **p = s->s_aliases, **q;
377 char **s_aliases = (char **)(buffer + pos);
378 while(*(p++) != 0)
379 pos += sizeof(char *);
380
381 /* copy addresses and fill new pointer list */
382 for (p = s->s_aliases, q = s_aliases; *p != 0; p++, q++){
383 len = strlen(*p);
384 if (size <= pos + len)
385 {
386 return NULL;
387 }
388 memcpy(buffer + pos, *p, len); /* copy content */
389 buffer[pos + len] = '\0';
390 *q = buffer + pos; /* set copied pointer to copied content */
391 pos += len + 1;
392 }
393 *++q = 0; /* null terminate the pointer list */
394 s->s_aliases = s_aliases; /* copy pointer to pointers */
395 return s;
396 }
397 #endif
398
399 #if defined(HAVE_GETSERVBYNAME) && wxUSE_THREADS
400 static wxMutex servLock;
401 #endif
402 struct servent *wxGetservbyname_r(const char *port, const char *protocol,
403 struct servent *serv, void *buffer, int size)
404 {
405 struct servent *se = NULL;
406 #if defined(HAVE_FUNC_GETSERVBYNAME_R_6)
407 if (getservbyname_r(port, protocol, serv, (char*)buffer, size, &se))
408 se = NULL;
409 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_5)
410 se = getservbyname_r(port, protocol, serv, (char*)buffer, size);
411 #elif defined(HAVE_FUNC_GETSERVBYNAME_R_4)
412 if (getservbyname_r(port, protocol, serv, (struct servent_data*) buffer))
413 se = NULL;
414 else
415 se = serv;
416 #elif defined(HAVE_GETSERVBYNAME)
417 #if wxUSE_THREADS
418 wxMutexLocker locker(servLock);
419 #endif
420 se = getservbyname(port, protocol);
421 if (se)
422 se = deepCopyServent(serv, se, (char*)buffer, size);
423 #endif
424 return se;
425 }
426
427 /* debugging helpers */
428 #ifdef __GSOCKET_DEBUG__
429 # define SOCKET_DEBUG(args) printf args
430 #else
431 # define SOCKET_DEBUG(args)
432 #endif /* __GSOCKET_DEBUG__ */
433
434
435 /*
436 * Disallow further read/write operations on this socket, close
437 * the fd and disable all callbacks.
438 */
439 void wxSocketImplUnix::Shutdown()
440 {
441 /* Don't allow events to fire after socket has been closed */
442 DisableEvents();
443
444 wxSocketImpl::Shutdown();
445 }
446
447 /*
448 * Waits for an incoming client connection. Returns a pointer to
449 * a wxSocketImplUnix object, or NULL if there was an error, in which case
450 * the last error field will be updated for the calling wxSocketImplUnix.
451 *
452 * Error codes (set in the calling wxSocketImplUnix)
453 * wxSOCKET_INVSOCK - the socket is not valid or not a server.
454 * wxSOCKET_TIMEDOUT - timeout, no incoming connections.
455 * wxSOCKET_WOULDBLOCK - the call would block and the socket is nonblocking.
456 * wxSOCKET_MEMERR - couldn't allocate memory.
457 * wxSOCKET_IOERR - low-level error.
458 */
459 wxSocketImpl *wxSocketImplUnix::WaitConnection(wxSocketBase& wxsocket)
460 {
461 wxSockAddr from;
462 WX_SOCKLEN_T fromlen = sizeof(from);
463 wxSocketImpl *connection;
464 wxSocketError err;
465 int arg = 1;
466
467 /* If the socket has already been created, we exit immediately */
468 if (m_fd == INVALID_SOCKET || !m_server)
469 {
470 m_error = wxSOCKET_INVSOCK;
471 return NULL;
472 }
473
474 /* Create a wxSocketImplUnix object for the new connection */
475 connection = wxSocketImplUnix::Create(wxsocket);
476
477 if (!connection)
478 {
479 m_error = wxSOCKET_MEMERR;
480 return NULL;
481 }
482
483 /* Wait for a connection (with timeout) */
484 if (Input_Timeout() == wxSOCKET_TIMEDOUT)
485 {
486 delete connection;
487 /* m_error set by Input_Timeout */
488 return NULL;
489 }
490
491 connection->m_fd = accept(m_fd, (sockaddr*)&from, (WX_SOCKLEN_T *) &fromlen);
492
493 /* Reenable CONNECTION events */
494 EnableEvent(wxSOCKET_CONNECTION);
495
496 if (connection->m_fd == INVALID_SOCKET)
497 {
498 if (errno == EWOULDBLOCK)
499 m_error = wxSOCKET_WOULDBLOCK;
500 else
501 m_error = wxSOCKET_IOERR;
502
503 delete connection;
504 return NULL;
505 }
506
507 /* Initialize all fields */
508 connection->m_server = false;
509 connection->m_stream = true;
510
511 /* Setup the peer address field */
512 connection->m_peer = GAddress_new();
513 if (!connection->m_peer)
514 {
515 delete connection;
516 m_error = wxSOCKET_MEMERR;
517 return NULL;
518 }
519
520 err = _GAddress_translate_from(connection->m_peer, (sockaddr*)&from, fromlen);
521 if (err != wxSOCKET_NOERROR)
522 {
523 delete connection;
524 m_error = err;
525 return NULL;
526 }
527
528 #if defined(__EMX__) || defined(__VISAGECPP__)
529 ioctl(connection->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
530 #else
531 ioctl(connection->m_fd, FIONBIO, &arg);
532 #endif
533 if (m_use_events)
534 connection->Notify(true);
535
536 return connection;
537 }
538
539 void wxSocketImplUnix::Notify(bool flag)
540 {
541 if (flag == m_use_events)
542 return;
543 m_use_events = flag;
544 DoEnableEvents(flag);
545 }
546
547 void wxSocketImplUnix::DoEnableEvents(bool flag)
548 {
549 wxSocketManager * const manager = wxSocketManager::Get();
550 if ( flag )
551 {
552 manager->Install_Callback(this, wxSOCKET_INPUT);
553 manager->Install_Callback(this, wxSOCKET_OUTPUT);
554 }
555 else // off
556 {
557 manager->Uninstall_Callback(this, wxSOCKET_INPUT);
558 manager->Uninstall_Callback(this, wxSOCKET_OUTPUT);
559 }
560 }
561
562 wxSocketError wxSocketImplUnix::DoHandleConnect(int ret)
563 {
564 /* We only call EnableEvents() if we know we aren't shutting down the socket.
565 * NB: EnableEvents() needs to be called whether the socket is blocking or
566 * non-blocking, it just shouldn't be called prior to knowing there is a
567 * connection _if_ blocking sockets are being used.
568 * If connect above returns 0, we are already connected and need to make the
569 * call to EnableEvents() now.
570 */
571 if ( m_non_blocking || (ret == 0) )
572 EnableEvents();
573
574 if (ret == -1)
575 {
576 const int err = errno;
577
578 /* If connect failed with EINPROGRESS and the wxSocketImplUnix object
579 * is in blocking mode, we select() for the specified timeout
580 * checking for writability to see if the connection request
581 * completes.
582 */
583 if ((err == EINPROGRESS) && (!m_non_blocking))
584 {
585 if (Output_Timeout() == wxSOCKET_TIMEDOUT)
586 {
587 Close();
588 /* m_error is set in Output_Timeout */
589 return wxSOCKET_TIMEDOUT;
590 }
591 else
592 {
593 int error;
594 SOCKOPTLEN_T len = sizeof(error);
595
596 getsockopt(m_fd, SOL_SOCKET, SO_ERROR, (char*) &error, &len);
597 EnableEvents();
598
599 if (!error)
600 return wxSOCKET_NOERROR;
601 }
602 }
603
604 /* If connect failed with EINPROGRESS and the wxSocketImplUnix object
605 * is set to nonblocking, we set m_error to wxSOCKET_WOULDBLOCK
606 * (and return wxSOCKET_WOULDBLOCK) but we don't close the socket;
607 * this way if the connection completes, a wxSOCKET_CONNECTION
608 * event will be generated, if enabled.
609 */
610 if ((err == EINPROGRESS) && (m_non_blocking))
611 {
612 m_establishing = true;
613 m_error = wxSOCKET_WOULDBLOCK;
614 return wxSOCKET_WOULDBLOCK;
615 }
616
617 /* If connect failed with an error other than EINPROGRESS,
618 * then the call to Connect has failed.
619 */
620 Close();
621 m_error = wxSOCKET_IOERR;
622
623 return wxSOCKET_IOERR;
624 }
625
626 return wxSOCKET_NOERROR;
627 }
628
629 /* Generic IO */
630
631 /* Like recv(), send(), ... */
632 int wxSocketImplUnix::Read(char *buffer, int size)
633 {
634 int ret;
635
636 if (m_fd == INVALID_SOCKET || m_server)
637 {
638 m_error = wxSOCKET_INVSOCK;
639 return -1;
640 }
641
642 /* Disable events during query of socket status */
643 DisableEvent(wxSOCKET_INPUT);
644
645 /* If the socket is blocking, wait for data (with a timeout) */
646 if (Input_Timeout() == wxSOCKET_TIMEDOUT) {
647 m_error = wxSOCKET_TIMEDOUT;
648 /* Don't return here immediately, otherwise socket events would not be
649 * re-enabled! */
650 ret = -1;
651 }
652 else
653 {
654 /* Read the data */
655 if (m_stream)
656 ret = Recv_Stream(buffer, size);
657 else
658 ret = Recv_Dgram(buffer, size);
659
660 /*
661 * If recv returned zero for a TCP socket (if m_stream == NULL, it's an UDP
662 * socket and empty datagrams are possible), then the connection has been
663 * gracefully closed.
664 *
665 * Otherwise, recv has returned an error (-1), in which case we have lost
666 * the socket only if errno does _not_ indicate that there may be more data
667 * to read.
668 */
669 if ((ret == 0) && m_stream)
670 {
671 /* Make sure wxSOCKET_LOST event gets sent and shut down the socket */
672 if (m_use_events)
673 {
674 m_detected = wxSOCKET_LOST_FLAG;
675 Detected_Read();
676 return 0;
677 }
678 }
679 else if (ret == -1)
680 {
681 if ((errno == EWOULDBLOCK) || (errno == EAGAIN))
682 m_error = wxSOCKET_WOULDBLOCK;
683 else
684 m_error = wxSOCKET_IOERR;
685 }
686 }
687
688 /* Enable events again now that we are done processing */
689 EnableEvent(wxSOCKET_INPUT);
690
691 return ret;
692 }
693
694 int wxSocketImplUnix::Write(const char *buffer, int size)
695 {
696 int ret;
697
698 SOCKET_DEBUG(( "Write #1, size %d\n", size ));
699
700 if (m_fd == INVALID_SOCKET || m_server)
701 {
702 m_error = wxSOCKET_INVSOCK;
703 return -1;
704 }
705
706 SOCKET_DEBUG(( "Write #2, size %d\n", size ));
707
708 /* If the socket is blocking, wait for writability (with a timeout) */
709 if (Output_Timeout() == wxSOCKET_TIMEDOUT)
710 return -1;
711
712 SOCKET_DEBUG(( "Write #3, size %d\n", size ));
713
714 /* Write the data */
715 if (m_stream)
716 ret = Send_Stream(buffer, size);
717 else
718 ret = Send_Dgram(buffer, size);
719
720 SOCKET_DEBUG(( "Write #4, size %d\n", size ));
721
722 if (ret == -1)
723 {
724 if ((errno == EWOULDBLOCK) || (errno == EAGAIN))
725 {
726 m_error = wxSOCKET_WOULDBLOCK;
727 SOCKET_DEBUG(( "Write error WOULDBLOCK\n" ));
728 }
729 else
730 {
731 m_error = wxSOCKET_IOERR;
732 SOCKET_DEBUG(( "Write error IOERR\n" ));
733 }
734
735 /* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
736 * in MSW). Once the first OUTPUT event is received, users can assume
737 * that the socket is writable until a read operation fails. Only then
738 * will further OUTPUT events be posted.
739 */
740 EnableEvent(wxSOCKET_OUTPUT);
741
742 return -1;
743 }
744
745 SOCKET_DEBUG(( "Write #5, size %d ret %d\n", size, ret ));
746
747 return ret;
748 }
749
750 /* Flags */
751
752 void wxSocketImplUnix::EnableEvent(wxSocketNotify event)
753 {
754 if (m_use_events)
755 {
756 m_detected &= ~(1 << event);
757 wxSocketManager::Get()->Install_Callback(this, event);
758 }
759 }
760
761 void wxSocketImplUnix::DisableEvent(wxSocketNotify event)
762 {
763 if (m_use_events)
764 {
765 m_detected |= (1 << event);
766 wxSocketManager::Get()->Uninstall_Callback(this, event);
767 }
768 }
769
770 /*
771 * For blocking sockets, wait until data is available or
772 * until timeout ellapses.
773 */
774 wxSocketError wxSocketImplUnix::Input_Timeout()
775 {
776 fd_set readfds;
777 int ret;
778
779 // Linux select() will overwrite the struct on return so make a copy
780 struct timeval tv = m_timeout;
781
782 if (!m_non_blocking)
783 {
784 wxFD_ZERO(&readfds);
785 wxFD_SET(m_fd, &readfds);
786 ret = select(m_fd + 1, &readfds, NULL, NULL, &tv);
787 if (ret == 0)
788 {
789 SOCKET_DEBUG(( "Input_Timeout, select returned 0\n" ));
790 m_error = wxSOCKET_TIMEDOUT;
791 return wxSOCKET_TIMEDOUT;
792 }
793
794 if (ret == -1)
795 {
796 SOCKET_DEBUG(( "Input_Timeout, select returned -1\n" ));
797 if (errno == EBADF) { SOCKET_DEBUG(( "Invalid file descriptor\n" )); }
798 if (errno == EINTR) { SOCKET_DEBUG(( "A non blocked signal was caught\n" )); }
799 if (errno == EINVAL) { SOCKET_DEBUG(( "The highest number descriptor is negative\n" )); }
800 if (errno == ENOMEM) { SOCKET_DEBUG(( "Not enough memory\n" )); }
801 m_error = wxSOCKET_TIMEDOUT;
802 return wxSOCKET_TIMEDOUT;
803 }
804 }
805
806 return wxSOCKET_NOERROR;
807 }
808
809 /*
810 * For blocking sockets, wait until data can be sent without
811 * blocking or until timeout ellapses.
812 */
813 wxSocketError wxSocketImplUnix::Output_Timeout()
814 {
815 fd_set writefds;
816 int ret;
817
818 // Linux select() will overwrite the struct on return so make a copy
819 struct timeval tv = m_timeout;
820
821 SOCKET_DEBUG( ("m_non_blocking has: %d\n", (int)m_non_blocking) );
822
823 if (!m_non_blocking)
824 {
825 wxFD_ZERO(&writefds);
826 wxFD_SET(m_fd, &writefds);
827 ret = select(m_fd + 1, NULL, &writefds, NULL, &tv);
828 if (ret == 0)
829 {
830 SOCKET_DEBUG(( "Output_Timeout, select returned 0\n" ));
831 m_error = wxSOCKET_TIMEDOUT;
832 return wxSOCKET_TIMEDOUT;
833 }
834
835 if (ret == -1)
836 {
837 SOCKET_DEBUG(( "Output_Timeout, select returned -1\n" ));
838 if (errno == EBADF) { SOCKET_DEBUG(( "Invalid file descriptor\n" )); }
839 if (errno == EINTR) { SOCKET_DEBUG(( "A non blocked signal was caught\n" )); }
840 if (errno == EINVAL) { SOCKET_DEBUG(( "The highest number descriptor is negative\n" )); }
841 if (errno == ENOMEM) { SOCKET_DEBUG(( "Not enough memory\n" )); }
842 m_error = wxSOCKET_TIMEDOUT;
843 return wxSOCKET_TIMEDOUT;
844 }
845
846 if ( ! wxFD_ISSET(m_fd, &writefds) )
847 {
848 SOCKET_DEBUG(( "Output_Timeout is buggy!\n" ));
849 }
850 else
851 {
852 SOCKET_DEBUG(( "Output_Timeout seems correct\n" ));
853 }
854 }
855 else
856 {
857 SOCKET_DEBUG(( "Output_Timeout, didn't try select!\n" ));
858 }
859
860 return wxSOCKET_NOERROR;
861 }
862
863 int wxSocketImplUnix::Recv_Stream(char *buffer, int size)
864 {
865 int ret;
866 do
867 {
868 ret = recv(m_fd, buffer, size, GSOCKET_MSG_NOSIGNAL);
869 }
870 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
871
872 return ret;
873 }
874
875 int wxSocketImplUnix::Recv_Dgram(char *buffer, int size)
876 {
877 wxSockAddr from;
878 WX_SOCKLEN_T fromlen = sizeof(from);
879 int ret;
880 wxSocketError err;
881
882 fromlen = sizeof(from);
883
884 do
885 {
886 ret = recvfrom(m_fd, buffer, size, 0, (sockaddr*)&from, (WX_SOCKLEN_T *) &fromlen);
887 }
888 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
889
890 if (ret == -1)
891 return -1;
892
893 /* Translate a system address into a wxSocketImplUnix address */
894 if (!m_peer)
895 {
896 m_peer = GAddress_new();
897 if (!m_peer)
898 {
899 m_error = wxSOCKET_MEMERR;
900 return -1;
901 }
902 }
903
904 err = _GAddress_translate_from(m_peer, (sockaddr*)&from, fromlen);
905 if (err != wxSOCKET_NOERROR)
906 {
907 GAddress_destroy(m_peer);
908 m_peer = NULL;
909 m_error = err;
910 return -1;
911 }
912
913 return ret;
914 }
915
916 int wxSocketImplUnix::Send_Stream(const char *buffer, int size)
917 {
918 int ret;
919
920 MASK_SIGNAL();
921
922 do
923 {
924 ret = send(m_fd, (char *)buffer, size, GSOCKET_MSG_NOSIGNAL);
925 }
926 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
927
928 UNMASK_SIGNAL();
929
930 return ret;
931 }
932
933 int wxSocketImplUnix::Send_Dgram(const char *buffer, int size)
934 {
935 struct sockaddr *addr;
936 int len, ret;
937 wxSocketError err;
938
939 if (!m_peer)
940 {
941 m_error = wxSOCKET_INVADDR;
942 return -1;
943 }
944
945 err = _GAddress_translate_to(m_peer, &addr, &len);
946 if (err != wxSOCKET_NOERROR)
947 {
948 m_error = err;
949 return -1;
950 }
951
952 MASK_SIGNAL();
953
954 do
955 {
956 ret = sendto(m_fd, (char *)buffer, size, 0, addr, len);
957 }
958 while (ret == -1 && errno == EINTR); /* Loop until not interrupted */
959
960 UNMASK_SIGNAL();
961
962 /* Frees memory allocated from _GAddress_translate_to */
963 free(addr);
964
965 return ret;
966 }
967
968 void wxSocketImplUnix::OnStateChange(wxSocketNotify event)
969 {
970 DisableEvent(event);
971 NotifyOnStateChange(event);
972
973 if ( event == wxSOCKET_LOST )
974 Shutdown();
975 }
976
977 void wxSocketImplUnix::Detected_Read()
978 {
979 char c;
980
981 /* Safeguard against straggling call to Detected_Read */
982 if (m_fd == INVALID_SOCKET)
983 {
984 return;
985 }
986
987 /* If we have already detected a LOST event, then don't try
988 * to do any further processing.
989 */
990 if ((m_detected & wxSOCKET_LOST_FLAG) != 0)
991 {
992 m_establishing = false;
993
994 OnStateChange(wxSOCKET_LOST);
995 return;
996 }
997
998 int num = recv(m_fd, &c, 1, MSG_PEEK | GSOCKET_MSG_NOSIGNAL);
999
1000 if (num > 0)
1001 {
1002 OnStateChange(wxSOCKET_INPUT);
1003 }
1004 else
1005 {
1006 if (m_server && m_stream)
1007 {
1008 OnStateChange(wxSOCKET_CONNECTION);
1009 }
1010 else if (num == 0)
1011 {
1012 if (m_stream)
1013 {
1014 /* graceful shutdown */
1015 OnStateChange(wxSOCKET_LOST);
1016 }
1017 else
1018 {
1019 /* Empty datagram received */
1020 OnStateChange(wxSOCKET_INPUT);
1021 }
1022 }
1023 else
1024 {
1025 /* Do not throw a lost event in cases where the socket isn't really lost */
1026 if ((errno == EWOULDBLOCK) || (errno == EAGAIN) || (errno == EINTR))
1027 {
1028 OnStateChange(wxSOCKET_INPUT);
1029 }
1030 else
1031 {
1032 OnStateChange(wxSOCKET_LOST);
1033 }
1034 }
1035 }
1036 }
1037
1038 void wxSocketImplUnix::Detected_Write()
1039 {
1040 /* If we have already detected a LOST event, then don't try
1041 * to do any further processing.
1042 */
1043 if ((m_detected & wxSOCKET_LOST_FLAG) != 0)
1044 {
1045 m_establishing = false;
1046
1047 OnStateChange(wxSOCKET_LOST);
1048 return;
1049 }
1050
1051 if (m_establishing && !m_server)
1052 {
1053 int error;
1054 SOCKOPTLEN_T len = sizeof(error);
1055
1056 m_establishing = false;
1057
1058 getsockopt(m_fd, SOL_SOCKET, SO_ERROR, (char*)&error, &len);
1059
1060 if (error)
1061 {
1062 OnStateChange(wxSOCKET_LOST);
1063 }
1064 else
1065 {
1066 OnStateChange(wxSOCKET_CONNECTION);
1067 /* We have to fire this event by hand because CONNECTION (for clients)
1068 * and OUTPUT are internally the same and we just disabled CONNECTION
1069 * events with the above macro.
1070 */
1071 OnStateChange(wxSOCKET_OUTPUT);
1072 }
1073 }
1074 else
1075 {
1076 OnStateChange(wxSOCKET_OUTPUT);
1077 }
1078 }
1079
1080 /*
1081 * -------------------------------------------------------------------------
1082 * GAddress
1083 * -------------------------------------------------------------------------
1084 */
1085
1086 /* CHECK_ADDRESS verifies that the current address family is either
1087 * wxSOCKET_NOFAMILY or wxSOCKET_*family*, and if it is wxSOCKET_NOFAMILY, it
1088 * initalizes it to be a wxSOCKET_*family*. In other cases, it returns
1089 * an appropiate error code.
1090 *
1091 * CHECK_ADDRESS_RETVAL does the same but returning 'retval' on error.
1092 */
1093 #define CHECK_ADDRESS(address, family) \
1094 { \
1095 if (address->m_family == wxSOCKET_NOFAMILY) \
1096 if (_GAddress_Init_##family(address) != wxSOCKET_NOERROR) \
1097 return address->m_error; \
1098 if (address->m_family != wxSOCKET_##family) \
1099 { \
1100 address->m_error = wxSOCKET_INVADDR; \
1101 return wxSOCKET_INVADDR; \
1102 } \
1103 }
1104
1105 #define CHECK_ADDRESS_RETVAL(address, family, retval) \
1106 { \
1107 if (address->m_family == wxSOCKET_NOFAMILY) \
1108 if (_GAddress_Init_##family(address) != wxSOCKET_NOERROR) \
1109 return retval; \
1110 if (address->m_family != wxSOCKET_##family) \
1111 { \
1112 address->m_error = wxSOCKET_INVADDR; \
1113 return retval; \
1114 } \
1115 }
1116
1117
1118 GAddress *GAddress_new(void)
1119 {
1120 GAddress *address;
1121
1122 if ((address = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1123 return NULL;
1124
1125 address->m_family = wxSOCKET_NOFAMILY;
1126 address->m_addr = NULL;
1127 address->m_len = 0;
1128
1129 return address;
1130 }
1131
1132 GAddress *GAddress_copy(GAddress *address)
1133 {
1134 GAddress *addr2;
1135
1136 assert(address != NULL);
1137
1138 if ((addr2 = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1139 return NULL;
1140
1141 memcpy(addr2, address, sizeof(GAddress));
1142
1143 if (address->m_addr && address->m_len > 0)
1144 {
1145 addr2->m_addr = (struct sockaddr *)malloc(addr2->m_len);
1146 if (addr2->m_addr == NULL)
1147 {
1148 free(addr2);
1149 return NULL;
1150 }
1151 memcpy(addr2->m_addr, address->m_addr, addr2->m_len);
1152 }
1153
1154 return addr2;
1155 }
1156
1157 void GAddress_destroy(GAddress *address)
1158 {
1159 assert(address != NULL);
1160
1161 if (address->m_addr)
1162 free(address->m_addr);
1163
1164 free(address);
1165 }
1166
1167 void GAddress_SetFamily(GAddress *address, GAddressType type)
1168 {
1169 assert(address != NULL);
1170
1171 address->m_family = type;
1172 }
1173
1174 GAddressType GAddress_GetFamily(GAddress *address)
1175 {
1176 assert(address != NULL);
1177
1178 return address->m_family;
1179 }
1180
1181 wxSocketError _GAddress_translate_from(GAddress *address,
1182 struct sockaddr *addr, int len)
1183 {
1184 address->m_realfamily = addr->sa_family;
1185 switch (addr->sa_family)
1186 {
1187 case AF_INET:
1188 address->m_family = wxSOCKET_INET;
1189 break;
1190 case AF_UNIX:
1191 address->m_family = wxSOCKET_UNIX;
1192 break;
1193 #if wxUSE_IPV6
1194 case AF_INET6:
1195 address->m_family = wxSOCKET_INET6;
1196 break;
1197 #endif // wxUSE_IPV6
1198 default:
1199 {
1200 address->m_error = wxSOCKET_INVOP;
1201 return wxSOCKET_INVOP;
1202 }
1203 }
1204
1205 if (address->m_addr)
1206 free(address->m_addr);
1207
1208 address->m_len = len;
1209 address->m_addr = (struct sockaddr *)malloc(len);
1210
1211 if (address->m_addr == NULL)
1212 {
1213 address->m_error = wxSOCKET_MEMERR;
1214 return wxSOCKET_MEMERR;
1215 }
1216
1217 memcpy(address->m_addr, addr, len);
1218
1219 return wxSOCKET_NOERROR;
1220 }
1221
1222 wxSocketError _GAddress_translate_to(GAddress *address,
1223 struct sockaddr **addr, int *len)
1224 {
1225 if (!address->m_addr)
1226 {
1227 address->m_error = wxSOCKET_INVADDR;
1228 return wxSOCKET_INVADDR;
1229 }
1230
1231 *len = address->m_len;
1232 *addr = (struct sockaddr *)malloc(address->m_len);
1233 if (*addr == NULL)
1234 {
1235 address->m_error = wxSOCKET_MEMERR;
1236 return wxSOCKET_MEMERR;
1237 }
1238
1239 memcpy(*addr, address->m_addr, address->m_len);
1240 return wxSOCKET_NOERROR;
1241 }
1242
1243 /*
1244 * -------------------------------------------------------------------------
1245 * Internet address family
1246 * -------------------------------------------------------------------------
1247 */
1248
1249 wxSocketError _GAddress_Init_INET(GAddress *address)
1250 {
1251 address->m_len = sizeof(struct sockaddr_in);
1252 address->m_addr = (struct sockaddr *) malloc(address->m_len);
1253 if (address->m_addr == NULL)
1254 {
1255 address->m_error = wxSOCKET_MEMERR;
1256 return wxSOCKET_MEMERR;
1257 }
1258
1259 address->m_family = wxSOCKET_INET;
1260 address->m_realfamily = PF_INET;
1261 ((struct sockaddr_in *)address->m_addr)->sin_family = AF_INET;
1262 ((struct sockaddr_in *)address->m_addr)->sin_addr.s_addr = INADDR_ANY;
1263
1264 return wxSOCKET_NOERROR;
1265 }
1266
1267 wxSocketError GAddress_INET_SetHostName(GAddress *address, const char *hostname)
1268 {
1269 struct hostent *he;
1270 struct in_addr *addr;
1271
1272 assert(address != NULL);
1273
1274 CHECK_ADDRESS(address, INET);
1275
1276 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1277
1278 /* If it is a numeric host name, convert it now */
1279 #if defined(HAVE_INET_ATON)
1280 if (inet_aton(hostname, addr) == 0)
1281 {
1282 #elif defined(HAVE_INET_ADDR)
1283 if ( (addr->s_addr = inet_addr(hostname)) == (unsigned)-1 )
1284 {
1285 #else
1286 /* Use gethostbyname by default */
1287 #ifndef __WXMAC__
1288 int val = 1; /* VA doesn't like constants in conditional expressions */
1289 if (val)
1290 #endif
1291 {
1292 #endif
1293 struct in_addr *array_addr;
1294
1295 /* It is a real name, we solve it */
1296 struct hostent h;
1297 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
1298 struct hostent_data buffer;
1299 #else
1300 char buffer[1024];
1301 #endif
1302 int err;
1303 he = wxGethostbyname_r(hostname, &h, (void*)&buffer, sizeof(buffer), &err);
1304 if (he == NULL)
1305 {
1306 /* Reset to invalid address */
1307 addr->s_addr = INADDR_NONE;
1308 address->m_error = wxSOCKET_NOHOST;
1309 return wxSOCKET_NOHOST;
1310 }
1311
1312 array_addr = (struct in_addr *) *(he->h_addr_list);
1313 addr->s_addr = array_addr[0].s_addr;
1314 }
1315
1316 return wxSOCKET_NOERROR;
1317 }
1318
1319
1320 wxSocketError GAddress_INET_SetBroadcastAddress(GAddress *address)
1321 {
1322 return GAddress_INET_SetHostAddress(address, INADDR_BROADCAST);
1323 }
1324
1325 wxSocketError GAddress_INET_SetAnyAddress(GAddress *address)
1326 {
1327 return GAddress_INET_SetHostAddress(address, INADDR_ANY);
1328 }
1329
1330 wxSocketError GAddress_INET_SetHostAddress(GAddress *address,
1331 unsigned long hostaddr)
1332 {
1333 struct in_addr *addr;
1334
1335 assert(address != NULL);
1336
1337 CHECK_ADDRESS(address, INET);
1338
1339 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1340 addr->s_addr = htonl(hostaddr);
1341
1342 return wxSOCKET_NOERROR;
1343 }
1344
1345 wxSocketError GAddress_INET_SetPortName(GAddress *address, const char *port,
1346 const char *protocol)
1347 {
1348 struct servent *se;
1349 struct sockaddr_in *addr;
1350
1351 assert(address != NULL);
1352 CHECK_ADDRESS(address, INET);
1353
1354 if (!port)
1355 {
1356 address->m_error = wxSOCKET_INVPORT;
1357 return wxSOCKET_INVPORT;
1358 }
1359
1360 #if defined(HAVE_FUNC_GETSERVBYNAME_R_4)
1361 struct servent_data buffer;
1362 #else
1363 char buffer[1024];
1364 #endif
1365 struct servent serv;
1366 se = wxGetservbyname_r(port, protocol, &serv,
1367 (void*)&buffer, sizeof(buffer));
1368 if (!se)
1369 {
1370 /* the cast to int suppresses compiler warnings about subscript having the
1371 type char */
1372 if (isdigit((int)port[0]))
1373 {
1374 int port_int;
1375
1376 port_int = atoi(port);
1377 addr = (struct sockaddr_in *)address->m_addr;
1378 addr->sin_port = htons(port_int);
1379 return wxSOCKET_NOERROR;
1380 }
1381
1382 address->m_error = wxSOCKET_INVPORT;
1383 return wxSOCKET_INVPORT;
1384 }
1385
1386 addr = (struct sockaddr_in *)address->m_addr;
1387 addr->sin_port = se->s_port;
1388
1389 return wxSOCKET_NOERROR;
1390 }
1391
1392 wxSocketError GAddress_INET_SetPort(GAddress *address, unsigned short port)
1393 {
1394 struct sockaddr_in *addr;
1395
1396 assert(address != NULL);
1397 CHECK_ADDRESS(address, INET);
1398
1399 addr = (struct sockaddr_in *)address->m_addr;
1400 addr->sin_port = htons(port);
1401
1402 return wxSOCKET_NOERROR;
1403 }
1404
1405 wxSocketError GAddress_INET_GetHostName(GAddress *address, char *hostname, size_t sbuf)
1406 {
1407 struct hostent *he;
1408 char *addr_buf;
1409 struct sockaddr_in *addr;
1410
1411 assert(address != NULL);
1412 CHECK_ADDRESS(address, INET);
1413
1414 addr = (struct sockaddr_in *)address->m_addr;
1415 addr_buf = (char *)&(addr->sin_addr);
1416
1417 struct hostent temphost;
1418 #if defined(HAVE_FUNC_GETHOSTBYNAME_R_3)
1419 struct hostent_data buffer;
1420 #else
1421 char buffer[1024];
1422 #endif
1423 int err;
1424 he = wxGethostbyaddr_r(addr_buf, sizeof(addr->sin_addr), AF_INET, &temphost,
1425 (void*)&buffer, sizeof(buffer), &err);
1426 if (he == NULL)
1427 {
1428 address->m_error = wxSOCKET_NOHOST;
1429 return wxSOCKET_NOHOST;
1430 }
1431
1432 strncpy(hostname, he->h_name, sbuf);
1433
1434 return wxSOCKET_NOERROR;
1435 }
1436
1437 unsigned long GAddress_INET_GetHostAddress(GAddress *address)
1438 {
1439 struct sockaddr_in *addr;
1440
1441 assert(address != NULL);
1442 CHECK_ADDRESS_RETVAL(address, INET, 0);
1443
1444 addr = (struct sockaddr_in *)address->m_addr;
1445
1446 return ntohl(addr->sin_addr.s_addr);
1447 }
1448
1449 unsigned short GAddress_INET_GetPort(GAddress *address)
1450 {
1451 struct sockaddr_in *addr;
1452
1453 assert(address != NULL);
1454 CHECK_ADDRESS_RETVAL(address, INET, 0);
1455
1456 addr = (struct sockaddr_in *)address->m_addr;
1457 return ntohs(addr->sin_port);
1458 }
1459
1460 #if wxUSE_IPV6
1461 /*
1462 * -------------------------------------------------------------------------
1463 * Internet IPv6 address family
1464 * -------------------------------------------------------------------------
1465 */
1466
1467 wxSocketError _GAddress_Init_INET6(GAddress *address)
1468 {
1469 struct in6_addr any_address = IN6ADDR_ANY_INIT;
1470 address->m_len = sizeof(struct sockaddr_in6);
1471 address->m_addr = (struct sockaddr *) malloc(address->m_len);
1472 if (address->m_addr == NULL)
1473 {
1474 address->m_error = wxSOCKET_MEMERR;
1475 return wxSOCKET_MEMERR;
1476 }
1477 memset(address->m_addr,0,address->m_len);
1478
1479 address->m_family = wxSOCKET_INET6;
1480 address->m_realfamily = AF_INET6;
1481 ((struct sockaddr_in6 *)address->m_addr)->sin6_family = AF_INET6;
1482 ((struct sockaddr_in6 *)address->m_addr)->sin6_addr = any_address;
1483
1484 return wxSOCKET_NOERROR;
1485 }
1486
1487 wxSocketError GAddress_INET6_SetHostName(GAddress *address, const char *hostname)
1488 {
1489 assert(address != NULL);
1490 CHECK_ADDRESS(address, INET6);
1491
1492 addrinfo hints;
1493 memset( & hints, 0, sizeof( hints ) );
1494 hints.ai_family = AF_INET6;
1495 addrinfo * info = 0;
1496 if ( getaddrinfo( hostname, "0", & hints, & info ) || ! info )
1497 {
1498 address->m_error = wxSOCKET_NOHOST;
1499 return wxSOCKET_NOHOST;
1500 }
1501
1502 memcpy( address->m_addr, info->ai_addr, info->ai_addrlen );
1503 freeaddrinfo( info );
1504 return wxSOCKET_NOERROR;
1505 }
1506
1507 wxSocketError GAddress_INET6_SetAnyAddress(GAddress *address)
1508 {
1509 assert(address != NULL);
1510
1511 CHECK_ADDRESS(address, INET6);
1512
1513 struct in6_addr addr;
1514 memset( & addr, 0, sizeof( addr ) );
1515 return GAddress_INET6_SetHostAddress(address, addr);
1516 }
1517 wxSocketError GAddress_INET6_SetHostAddress(GAddress *address,
1518 struct in6_addr hostaddr)
1519 {
1520 assert(address != NULL);
1521
1522 CHECK_ADDRESS(address, INET6);
1523
1524 ((struct sockaddr_in6 *)address->m_addr)->sin6_addr = hostaddr;
1525
1526 return wxSOCKET_NOERROR;
1527 }
1528
1529 wxSocketError GAddress_INET6_SetPortName(GAddress *address, const char *port,
1530 const char *protocol)
1531 {
1532 struct servent *se;
1533 struct sockaddr_in6 *addr;
1534
1535 assert(address != NULL);
1536 CHECK_ADDRESS(address, INET6);
1537
1538 if (!port)
1539 {
1540 address->m_error = wxSOCKET_INVPORT;
1541 return wxSOCKET_INVPORT;
1542 }
1543
1544 se = getservbyname(port, protocol);
1545 if (!se)
1546 {
1547 if (isdigit(port[0]))
1548 {
1549 int port_int;
1550
1551 port_int = atoi(port);
1552 addr = (struct sockaddr_in6 *)address->m_addr;
1553 addr->sin6_port = htons((u_short) port_int);
1554 return wxSOCKET_NOERROR;
1555 }
1556
1557 address->m_error = wxSOCKET_INVPORT;
1558 return wxSOCKET_INVPORT;
1559 }
1560
1561 addr = (struct sockaddr_in6 *)address->m_addr;
1562 addr->sin6_port = se->s_port;
1563
1564 return wxSOCKET_NOERROR;
1565 }
1566
1567 wxSocketError GAddress_INET6_SetPort(GAddress *address, unsigned short port)
1568 {
1569 struct sockaddr_in6 *addr;
1570
1571 assert(address != NULL);
1572 CHECK_ADDRESS(address, INET6);
1573
1574 addr = (struct sockaddr_in6 *)address->m_addr;
1575 addr->sin6_port = htons(port);
1576
1577 return wxSOCKET_NOERROR;
1578 }
1579
1580 wxSocketError GAddress_INET6_GetHostName(GAddress *address, char *hostname, size_t sbuf)
1581 {
1582 struct hostent *he;
1583 char *addr_buf;
1584 struct sockaddr_in6 *addr;
1585
1586 assert(address != NULL);
1587 CHECK_ADDRESS(address, INET6);
1588
1589 addr = (struct sockaddr_in6 *)address->m_addr;
1590 addr_buf = (char *)&(addr->sin6_addr);
1591
1592 he = gethostbyaddr(addr_buf, sizeof(addr->sin6_addr), AF_INET6);
1593 if (he == NULL)
1594 {
1595 address->m_error = wxSOCKET_NOHOST;
1596 return wxSOCKET_NOHOST;
1597 }
1598
1599 strncpy(hostname, he->h_name, sbuf);
1600
1601 return wxSOCKET_NOERROR;
1602 }
1603
1604 wxSocketError GAddress_INET6_GetHostAddress(GAddress *address,struct in6_addr *hostaddr)
1605 {
1606 assert(address != NULL);
1607 assert(hostaddr != NULL);
1608 CHECK_ADDRESS_RETVAL(address, INET6, wxSOCKET_INVADDR);
1609 *hostaddr = ( (struct sockaddr_in6 *)address->m_addr )->sin6_addr;
1610 return wxSOCKET_NOERROR;
1611 }
1612
1613 unsigned short GAddress_INET6_GetPort(GAddress *address)
1614 {
1615 assert(address != NULL);
1616 CHECK_ADDRESS_RETVAL(address, INET6, 0);
1617
1618 return ntohs( ((struct sockaddr_in6 *)address->m_addr)->sin6_port );
1619 }
1620
1621 #endif // wxUSE_IPV6
1622
1623 /*
1624 * -------------------------------------------------------------------------
1625 * Unix address family
1626 * -------------------------------------------------------------------------
1627 */
1628
1629 #ifndef __VISAGECPP__
1630 wxSocketError _GAddress_Init_UNIX(GAddress *address)
1631 {
1632 address->m_len = sizeof(struct sockaddr_un);
1633 address->m_addr = (struct sockaddr *)malloc(address->m_len);
1634 if (address->m_addr == NULL)
1635 {
1636 address->m_error = wxSOCKET_MEMERR;
1637 return wxSOCKET_MEMERR;
1638 }
1639
1640 address->m_family = wxSOCKET_UNIX;
1641 address->m_realfamily = PF_UNIX;
1642 ((struct sockaddr_un *)address->m_addr)->sun_family = AF_UNIX;
1643 ((struct sockaddr_un *)address->m_addr)->sun_path[0] = 0;
1644
1645 return wxSOCKET_NOERROR;
1646 }
1647
1648 #define UNIX_SOCK_PATHLEN (sizeof(addr->sun_path)/sizeof(addr->sun_path[0]))
1649
1650 wxSocketError GAddress_UNIX_SetPath(GAddress *address, const char *path)
1651 {
1652 struct sockaddr_un *addr;
1653
1654 assert(address != NULL);
1655
1656 CHECK_ADDRESS(address, UNIX);
1657
1658 addr = ((struct sockaddr_un *)address->m_addr);
1659 strncpy(addr->sun_path, path, UNIX_SOCK_PATHLEN);
1660 addr->sun_path[UNIX_SOCK_PATHLEN - 1] = '\0';
1661
1662 return wxSOCKET_NOERROR;
1663 }
1664
1665 wxSocketError GAddress_UNIX_GetPath(GAddress *address, char *path, size_t sbuf)
1666 {
1667 struct sockaddr_un *addr;
1668
1669 assert(address != NULL);
1670 CHECK_ADDRESS(address, UNIX);
1671
1672 addr = (struct sockaddr_un *)address->m_addr;
1673
1674 strncpy(path, addr->sun_path, sbuf);
1675
1676 return wxSOCKET_NOERROR;
1677 }
1678 #endif /* !defined(__VISAGECPP__) */
1679 #endif /* wxUSE_SOCKETS */