]>
git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.c
1 /* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
4 * Purpose: GSocket main Unix file
6 * -------------------------------------------------------------------------
10 #include <sys/ioctl.h>
11 #include <sys/types.h>
15 #include <sys/socket.h>
19 #include <netinet/in.h>
20 #include <arpa/inet.h>
29 #include <sys/filio.h>
37 /* #include <features.h> */
40 #include "wx/gsocket.h"
47 # define SOCKLEN_T socklen_t
50 # define SOCKLEN_T int
55 /* Global initialisers */
61 void GSocket_Cleanup()
65 /* Constructors / Destructors */
67 GSocket
*GSocket_new()
72 socket
= (GSocket
*)malloc(sizeof(GSocket
));
75 for (i
=0;i
<GSOCK_MAX_EVENT
;i
++) {
76 socket
->m_fbacks
[i
] = NULL
;
77 socket
->m_iocalls
[i
] = FALSE
;
79 socket
->m_local
= NULL
;
80 socket
->m_peer
= NULL
;
81 socket
->m_error
= GSOCK_NOERROR
;
82 socket
->m_server
= FALSE
;
83 socket
->m_stream
= TRUE
;
84 socket
->m_gui_dependent
= NULL
;
85 socket
->m_blocking
= FALSE
;
87 _GSocket_GUI_Init(socket
);
92 void GSocket_destroy(GSocket
*socket
)
94 assert(socket
!= NULL
);
96 if (socket
->m_fd
!= -1)
97 GSocket_Shutdown(socket
);
99 _GSocket_GUI_Destroy(socket
);
102 GAddress_destroy(socket
->m_local
);
105 GAddress_destroy(socket
->m_peer
);
110 void GSocket_Shutdown(GSocket
*socket
)
114 assert(socket
!= NULL
);
116 if (socket
->m_fd
!= -1) {
117 shutdown(socket
->m_fd
, 2);
122 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
123 _GSocket_Uninstall_Fallback(socket
, evt
);
126 /* Address handling */
128 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
130 assert(socket
!= NULL
);
132 if ((socket
->m_fd
!= -1 && !socket
->m_server
))
133 return GSOCK_INVSOCK
;
135 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
)
136 return GSOCK_INVADDR
;
139 GAddress_destroy(socket
->m_local
);
141 socket
->m_local
= GAddress_copy(address
);
143 return GSOCK_NOERROR
;
146 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
148 assert(socket
!= NULL
);
150 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
151 socket
->m_error
= GSOCK_INVADDR
;
152 return GSOCK_INVADDR
;
156 GAddress_destroy(socket
->m_peer
);
158 socket
->m_peer
= GAddress_copy(address
);
160 return GSOCK_NOERROR
;
163 GAddress
*GSocket_GetLocal(GSocket
*socket
)
166 struct sockaddr addr
;
169 assert(socket
!= NULL
);
172 return GAddress_copy(socket
->m_local
);
174 if (socket
->m_fd
== -1) {
175 socket
->m_error
= GSOCK_INVSOCK
;
181 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
182 socket
->m_error
= GSOCK_IOERR
;
186 address
= GAddress_new();
187 _GAddress_translate_from(address
, &addr
, size
);
192 GAddress
*GSocket_GetPeer(GSocket
*socket
)
194 assert(socket
!= NULL
);
197 return GAddress_copy(socket
->m_peer
);
202 /* Server specific parts */
205 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
206 of GSocket. "Local" must be set by GSocket_SetLocal() before
207 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
208 In case, you haven't yet defined the local address, it returns GSOCK_INVADDR.
209 In the other cases it returns GSOCK_IOERR.
211 GSocketError
GSocket_SetServer(GSocket
*sck
)
217 if (sck
->m_fd
!= -1) {
218 sck
->m_error
= GSOCK_INVSOCK
;
219 return GSOCK_INVSOCK
;
223 sck
->m_error
= GSOCK_INVADDR
;
224 return GSOCK_INVADDR
;
232 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
234 if (sck
->m_fd
== -1) {
235 sck
->m_error
= GSOCK_IOERR
;
239 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
242 sck
->m_error
= GSOCK_IOERR
;
246 if (listen(sck
->m_fd
, 5) < 0) {
249 sck
->m_error
= GSOCK_IOERR
;
253 sck
->m_server
= TRUE
;
255 return GSOCK_NOERROR
;
260 GSocket_WaitConnection() waits for an incoming client connection.
262 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
266 assert(socket
!= NULL
);
268 if (socket
->m_fd
== -1 || !socket
->m_server
) {
269 socket
->m_error
= GSOCK_INVSOCK
;
273 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
275 connection
= GSocket_new();
277 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
278 if (connection
->m_fd
== -1) {
279 GSocket_destroy(connection
);
280 socket
->m_error
= GSOCK_IOERR
;
284 connection
->m_stream
= TRUE
;
285 connection
->m_server
= FALSE
;
286 connection
->m_oriented
= TRUE
;
291 /* Non oriented connections */
293 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
297 if (sck
->m_fd
!= -1) {
298 sck
->m_error
= GSOCK_INVSOCK
;
299 return GSOCK_INVSOCK
;
303 sck
->m_error
= GSOCK_INVADDR
;
304 return GSOCK_INVADDR
;
307 sck
->m_stream
= FALSE
;
308 sck
->m_server
= FALSE
;
309 sck
->m_oriented
= FALSE
;
311 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
313 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
316 sck
->m_error
= GSOCK_IOERR
;
320 return GSOCK_NOERROR
;
323 /* Client specific parts */
326 GSocket_Connect() establishes a client connection to a server using the "Peer"
327 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
328 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
330 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
336 if (sck
->m_fd
!= -1) {
337 sck
->m_error
= GSOCK_INVSOCK
;
338 return GSOCK_INVSOCK
;
342 sck
->m_error
= GSOCK_INVADDR
;
343 return GSOCK_INVADDR
;
346 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
347 sck
->m_oriented
= TRUE
;
354 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
356 if (sck
->m_fd
== -1) {
357 sck
->m_error
= GSOCK_IOERR
;
361 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
362 sck
->m_peer
->m_len
) != 0) {
365 sck
->m_error
= GSOCK_IOERR
;
369 sck
->m_server
= FALSE
;
371 return GSOCK_NOERROR
;
376 /* Like recv(), send(), ... */
377 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
379 assert(socket
!= NULL
);
381 if (socket
->m_fd
== -1 || socket
->m_server
) {
382 socket
->m_error
= GSOCK_INVSOCK
;
386 _GSocket_Enable(socket
, GSOCK_INPUT
);
388 if (socket
->m_oriented
)
389 return _GSocket_Recv_Stream(socket
, buffer
, size
);
391 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
394 int GSocket_Write(GSocket
*socket
, const char *buffer
,
397 assert(socket
!= NULL
);
399 if (socket
->m_fd
== -1 || socket
->m_server
) {
400 socket
->m_error
= GSOCK_INVSOCK
;
404 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
406 if (socket
->m_oriented
)
407 return _GSocket_Send_Stream(socket
, buffer
, size
);
409 return _GSocket_Send_Dgram(socket
, buffer
, size
);
412 bool GSocket_DataAvailable(GSocket
*socket
)
417 assert(socket
!= NULL
);
419 if (socket
->m_fd
== -1 || socket
->m_server
) {
420 socket
->m_error
= GSOCK_INVSOCK
;
425 FD_SET(socket
->m_fd
, &read_set
);
430 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
432 return FD_ISSET(socket
->m_fd
, &read_set
);
438 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
439 if we don't want to wait.
441 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
443 assert(socket
!= NULL
);
445 socket
->m_blocking
= block
;
447 if (socket
->m_fd
!= -1)
448 ioctl(socket
->m_fd
, FIONBIO
, &block
);
452 GSocket_GetError() returns the last error occured on the socket stream.
455 GSocketError
GSocket_GetError(GSocket
*socket
)
457 assert(socket
!= NULL
);
459 return socket
->m_error
;
465 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
466 INPUT: The function is called when there is at least a byte in the
468 OUTPUT: The function is called when the system is sure the next write call
470 CONNECTION: Two cases is possible:
471 Client socket -> the connection is established
472 Server socket -> a client request a connection
473 LOST: the connection is lost
475 SetFallback accepts a combination of these flags so a same callback can
476 receive different events.
478 An event is generated only once and its state is reseted when the relative
479 IO call is requested.
480 For example: INPUT -> GSocket_Read()
481 CONNECTION -> GSocket_Accept()
483 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
484 GSocketFallback fallback
, char *cdata
)
488 assert (socket
!= NULL
);
490 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
491 if ((event
& (1 << count
)) != 0) {
492 socket
->m_fbacks
[count
] = fallback
;
493 socket
->m_data
[count
] = cdata
;
495 _GSocket_Install_Fallback(socket
, count
);
496 _GSocket_Enable(socket
, count
);
502 UnsetFallback will disables all fallbacks specified by "event".
503 NOTE: event may be a combination of flags
505 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
509 assert(socket
!= NULL
);
511 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
512 if ((event
& (1 << count
)) != 0) {
513 _GSocket_Disable(socket
, count
);
514 socket
->m_fbacks
[count
] = NULL
;
515 _GSocket_Uninstall_Fallback(socket
, count
);
520 #define CALL_FALLBACK(socket, event) \
521 if (socket->m_iocalls[event] && \
522 socket->m_fbacks[event]) {\
523 _GSocket_Disable(socket, event); \
524 socket->m_fbacks[event](socket, event, \
525 socket->m_data[event]); \
528 #define MASK_SIGNAL() \
530 void (*old_handler)(int); \
532 old_handler = signal(SIGPIPE, SIG_IGN);
534 #define UNMASK_SIGNAL() \
535 signal(SIGPIPE, old_handler); \
538 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
540 socket
->m_iocalls
[event
] = TRUE
;
541 if (socket
->m_fbacks
[event
])
542 _GSocket_Install_Fallback(socket
, event
);
545 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
547 socket
->m_iocalls
[event
] = FALSE
;
548 if (socket
->m_fbacks
[event
])
549 _GSocket_Uninstall_Fallback(socket
, event
);
552 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
557 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
560 socket
->m_error
= GSOCK_IOERR
;
566 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
568 struct sockaddr from
;
573 fromlen
= sizeof(from
);
576 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
579 socket
->m_error
= GSOCK_IOERR
;
584 socket
->m_peer
= GAddress_new();
585 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
590 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
595 ret
= send(socket
->m_fd
, buffer
, size
, 0);
598 socket
->m_error
= GSOCK_IOERR
;
604 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
606 struct sockaddr
*addr
;
609 if (!socket
->m_peer
) {
610 socket
->m_error
= GSOCK_INVADDR
;
614 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
617 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
620 socket
->m_error
= GSOCK_IOERR
;
629 void _GSocket_Detected_Read(GSocket
*socket
)
634 if (socket
->m_stream
) {
635 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
637 if (ret
< 0 && socket
->m_server
) {
638 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
643 CALL_FALLBACK(socket
, GSOCK_INPUT
);
645 CALL_FALLBACK(socket
, GSOCK_LOST
);
650 void _GSocket_Detected_Write(GSocket
*socket
)
652 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
660 * -------------------------------------------------------------------------
662 * -------------------------------------------------------------------------
665 #define CHECK_ADDRESS(address, family, retval) \
667 if (address->m_family == GSOCK_NOFAMILY) \
668 _GAddress_Init_##family(address); \
669 if (address->m_family != GSOCK_##family) {\
670 address->m_error = GSOCK_INVADDR; \
675 GAddress
*GAddress_new()
679 address
= (GAddress
*)malloc(sizeof(GAddress
));
681 address
->m_family
= GSOCK_NOFAMILY
;
682 address
->m_addr
= NULL
;
688 GAddress
*GAddress_copy(GAddress
*address
)
692 assert(address
!= NULL
);
694 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
695 memcpy(addr2
, address
, sizeof(GAddress
));
697 if (address
->m_addr
) {
698 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
699 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
705 void GAddress_destroy(GAddress
*address
)
707 assert(address
!= NULL
);
712 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
714 assert(address
!= NULL
);
716 address
->m_family
= type
;
719 GAddressType
GAddress_GetFamily(GAddress
*address
)
721 assert(address
!= NULL
);
723 return address
->m_family
;
726 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
727 address
->m_realfamily
= addr
->sa_family
;
728 switch (addr
->sa_family
) {
730 address
->m_family
= GSOCK_INET
;
733 address
->m_family
= GSOCK_UNIX
;
737 address
->m_family
= GSOCK_INET6
;
747 free(address
->m_addr
);
749 address
->m_len
= len
;
750 address
->m_addr
= (struct sockaddr
*)malloc(len
);
751 memcpy(address
->m_addr
, addr
, len
);
754 void _GAddress_translate_to(GAddress
*address
,
755 struct sockaddr
**addr
, int *len
)
757 if (!address
->m_addr
) {
762 *len
= address
->m_len
;
763 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
764 memcpy(*addr
, address
->m_addr
, address
->m_len
);
768 * -------------------------------------------------------------------------
769 * Internet address family
770 * -------------------------------------------------------------------------
773 void _GAddress_Init_INET(GAddress
*address
)
775 address
->m_len
= sizeof(struct sockaddr_in
);
776 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
777 address
->m_family
= GSOCK_INET
;
778 address
->m_realfamily
= PF_INET
;
779 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
780 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
783 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
786 struct in_addr
*addr
;
788 assert(address
!= NULL
);
790 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
792 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
794 /* only name for the moment */
795 if (inet_aton(hostname
, addr
) == 0) {
796 struct in_addr
*array_addr
;
798 if ((he
= gethostbyname(hostname
)) == NULL
) {
799 address
->m_error
= GSOCK_NOHOST
;
802 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
803 addr
->s_addr
= array_addr
[0].s_addr
;
805 return GSOCK_NOERROR
;
808 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
809 unsigned long hostaddr
)
811 struct in_addr
*addr
;
813 assert(address
!= NULL
);
815 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
817 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
818 addr
->s_addr
= hostaddr
;
820 return GSOCK_NOERROR
;
823 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
,
824 const char *protocol
)
827 struct sockaddr_in
*addr
;
829 assert(address
!= NULL
);
830 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
833 address
->m_error
= GSOCK_INVPORT
;
837 /* TODO: TCP or UDP */
838 se
= getservbyname(port
, protocol
);
840 if (isdigit(port
[0])) {
843 port_int
= atoi(port
);
844 addr
= (struct sockaddr_in
*)address
->m_addr
;
845 addr
->sin_port
= htons(port_int
);
846 return GSOCK_NOERROR
;
849 address
->m_error
= GSOCK_INVPORT
;
850 return GSOCK_INVPORT
;
853 addr
= (struct sockaddr_in
*)address
->m_addr
;
854 addr
->sin_port
= se
->s_port
;
856 return GSOCK_NOERROR
;
859 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
861 struct sockaddr_in
*addr
;
863 assert(address
!= NULL
);
864 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
866 addr
= (struct sockaddr_in
*)address
->m_addr
;
867 addr
->sin_port
= htons(port
);
869 return GSOCK_NOERROR
;
872 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
876 struct sockaddr_in
*addr
;
878 assert(address
!= NULL
);
879 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
881 addr
= (struct sockaddr_in
*)address
->m_addr
;
882 addr_buf
= (char *)&(addr
->sin_addr
);
884 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
886 address
->m_error
= GSOCK_NOHOST
;
890 strncpy(hostname
, he
->h_name
, sbuf
);
892 return GSOCK_NOERROR
;
895 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
897 struct sockaddr_in
*addr
;
899 assert(address
!= NULL
);
900 CHECK_ADDRESS(address
, INET
, 0);
902 addr
= (struct sockaddr_in
*)address
->m_addr
;
904 return addr
->sin_addr
.s_addr
;
907 unsigned short GAddress_INET_GetPort(GAddress
*address
)
909 struct sockaddr_in
*addr
;
911 assert(address
!= NULL
);
912 CHECK_ADDRESS(address
, INET
, 0);
914 addr
= (struct sockaddr_in
*)address
->m_addr
;
915 return ntohs(addr
->sin_port
);
919 * -------------------------------------------------------------------------
920 * Unix address family
921 * -------------------------------------------------------------------------
924 void _GAddress_Init_UNIX(GAddress
*address
)
926 address
->m_len
= sizeof(struct sockaddr_un
);
927 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
928 address
->m_family
= GSOCK_UNIX
;
929 address
->m_realfamily
= PF_UNIX
;
930 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
931 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
934 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
936 struct sockaddr_un
*addr
;
938 assert(address
!= NULL
);
940 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
942 addr
= ((struct sockaddr_un
*)address
->m_addr
);
943 memcpy(addr
->sun_path
, path
, strlen(path
));
945 return GSOCK_NOERROR
;
948 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
950 struct sockaddr_un
*addr
;
952 assert(address
!= NULL
);
953 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
955 addr
= (struct sockaddr_un
*)address
->m_addr
;
957 strncpy(path
, addr
->sun_path
, sbuf
);
959 return GSOCK_NOERROR
;