]>
git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.c
ab328de2e79076ea2911caa36cbd41f56d8c36a3
1 /* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
4 * Purpose: GSocket main Unix file
6 * -------------------------------------------------------------------------
14 #include <sys/ioctl.h>
15 #include <sys/types.h>
19 #include <sys/socket.h>
23 #include <netinet/in.h>
24 #include <arpa/inet.h>
33 #include <sys/filio.h>
42 #include "wx/gsocket.h"
49 # define SOCKLEN_T socklen_t
52 # define SOCKLEN_T int
57 /* Global initialisers */
64 void GSocket_Cleanup()
68 /* Constructors / Destructors */
70 GSocket
*GSocket_new()
75 socket
= (GSocket
*)malloc(sizeof(GSocket
));
78 for (i
=0;i
<GSOCK_MAX_EVENT
;i
++) {
79 socket
->m_fbacks
[i
] = NULL
;
80 socket
->m_iocalls
[i
] = FALSE
;
82 socket
->m_local
= NULL
;
83 socket
->m_peer
= NULL
;
84 socket
->m_error
= GSOCK_NOERROR
;
85 socket
->m_server
= FALSE
;
86 socket
->m_stream
= TRUE
;
87 socket
->m_gui_dependent
= NULL
;
88 socket
->m_blocking
= FALSE
;
90 /* We initialize the GUI specific entries here */
91 _GSocket_GUI_Init(socket
);
96 void GSocket_destroy(GSocket
*socket
)
98 assert(socket
!= NULL
);
100 /* First, we check that the socket is really shutdowned */
101 if (socket
->m_fd
!= -1)
102 GSocket_Shutdown(socket
);
104 /* We destroy GUI specific variables */
105 _GSocket_GUI_Destroy(socket
);
107 /* We destroy private addresses */
109 GAddress_destroy(socket
->m_local
);
112 GAddress_destroy(socket
->m_peer
);
114 /* We destroy socket itself */
118 void GSocket_Shutdown(GSocket
*socket
)
122 assert(socket
!= NULL
);
124 /* If socket has been created, we shutdown it */
125 if (socket
->m_fd
!= -1) {
126 shutdown(socket
->m_fd
, 2);
131 /* We also disable GUI callbacks */
132 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
133 _GSocket_Uninstall_Fallback(socket
, evt
);
136 /* Address handling */
138 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
140 assert(socket
!= NULL
);
142 if ((socket
->m_fd
!= -1 && !socket
->m_server
)) {
143 socket
->m_error
= GSOCK_INVSOCK
;
144 return GSOCK_INVSOCK
;
147 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
148 socket
->m_error
= GSOCK_INVADDR
;
149 return GSOCK_INVADDR
;
153 GAddress_destroy(socket
->m_local
);
155 socket
->m_local
= GAddress_copy(address
);
157 return GSOCK_NOERROR
;
160 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
162 assert(socket
!= NULL
);
164 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
165 socket
->m_error
= GSOCK_INVADDR
;
166 return GSOCK_INVADDR
;
170 GAddress_destroy(socket
->m_peer
);
172 socket
->m_peer
= GAddress_copy(address
);
174 return GSOCK_NOERROR
;
177 GAddress
*GSocket_GetLocal(GSocket
*socket
)
180 struct sockaddr addr
;
183 assert(socket
!= NULL
);
186 return GAddress_copy(socket
->m_local
);
188 if (socket
->m_fd
== -1) {
189 socket
->m_error
= GSOCK_INVSOCK
;
195 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
196 socket
->m_error
= GSOCK_IOERR
;
200 address
= GAddress_new();
201 _GAddress_translate_from(address
, &addr
, size
);
206 GAddress
*GSocket_GetPeer(GSocket
*socket
)
208 assert(socket
!= NULL
);
211 return GAddress_copy(socket
->m_peer
);
216 /* Server specific parts */
219 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
220 of GSocket. "Local" must be set by GSocket_SetLocal() before
221 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
222 In case, you haven't yet defined the local address, it returns GSOCK_INVADDR.
223 In the other cases it returns GSOCK_IOERR.
225 GSocketError
GSocket_SetServer(GSocket
*sck
)
231 if (sck
->m_fd
!= -1) {
232 sck
->m_error
= GSOCK_INVSOCK
;
233 return GSOCK_INVSOCK
;
237 sck
->m_error
= GSOCK_INVADDR
;
238 return GSOCK_INVADDR
;
246 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
248 if (sck
->m_fd
== -1) {
249 sck
->m_error
= GSOCK_IOERR
;
253 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
256 sck
->m_error
= GSOCK_IOERR
;
260 if (listen(sck
->m_fd
, 5) < 0) {
263 sck
->m_error
= GSOCK_IOERR
;
267 sck
->m_server
= TRUE
;
269 return GSOCK_NOERROR
;
274 GSocket_WaitConnection() waits for an incoming client connection.
276 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
280 assert(socket
!= NULL
);
282 if (socket
->m_fd
== -1 || !socket
->m_server
) {
283 socket
->m_error
= GSOCK_INVSOCK
;
287 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
289 connection
= GSocket_new();
291 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
292 if (connection
->m_fd
== -1) {
293 GSocket_destroy(connection
);
294 socket
->m_error
= GSOCK_IOERR
;
298 connection
->m_stream
= TRUE
;
299 connection
->m_server
= FALSE
;
300 connection
->m_oriented
= TRUE
;
305 /* Non oriented connections */
307 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
311 if (sck
->m_fd
!= -1) {
312 sck
->m_error
= GSOCK_INVSOCK
;
313 return GSOCK_INVSOCK
;
317 sck
->m_error
= GSOCK_INVADDR
;
318 return GSOCK_INVADDR
;
321 sck
->m_stream
= FALSE
;
322 sck
->m_server
= FALSE
;
323 sck
->m_oriented
= FALSE
;
325 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
327 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
330 sck
->m_error
= GSOCK_IOERR
;
334 return GSOCK_NOERROR
;
337 /* Client specific parts */
340 GSocket_Connect() establishes a client connection to a server using the "Peer"
341 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
342 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
344 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
350 if (sck
->m_fd
!= -1) {
351 sck
->m_error
= GSOCK_INVSOCK
;
352 return GSOCK_INVSOCK
;
356 sck
->m_error
= GSOCK_INVADDR
;
357 return GSOCK_INVADDR
;
360 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
361 sck
->m_oriented
= TRUE
;
368 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
370 if (sck
->m_fd
== -1) {
371 sck
->m_error
= GSOCK_IOERR
;
375 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
376 sck
->m_peer
->m_len
) != 0) {
379 sck
->m_error
= GSOCK_IOERR
;
383 sck
->m_server
= FALSE
;
385 return GSOCK_NOERROR
;
390 /* Like recv(), send(), ... */
391 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
393 assert(socket
!= NULL
);
395 if (socket
->m_fd
== -1 || socket
->m_server
) {
396 socket
->m_error
= GSOCK_INVSOCK
;
400 _GSocket_Enable(socket
, GSOCK_INPUT
);
402 if (socket
->m_oriented
)
403 return _GSocket_Recv_Stream(socket
, buffer
, size
);
405 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
408 int GSocket_Write(GSocket
*socket
, const char *buffer
,
411 assert(socket
!= NULL
);
413 if (socket
->m_fd
== -1 || socket
->m_server
) {
414 socket
->m_error
= GSOCK_INVSOCK
;
418 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
420 if (socket
->m_oriented
)
421 return _GSocket_Send_Stream(socket
, buffer
, size
);
423 return _GSocket_Send_Dgram(socket
, buffer
, size
);
426 bool GSocket_DataAvailable(GSocket
*socket
)
431 assert(socket
!= NULL
);
433 if (socket
->m_fd
== -1 || socket
->m_server
) {
434 socket
->m_error
= GSOCK_INVSOCK
;
439 FD_SET(socket
->m_fd
, &read_set
);
444 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
446 return FD_ISSET(socket
->m_fd
, &read_set
);
452 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
453 if we don't want to wait.
455 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
457 assert(socket
!= NULL
);
459 socket
->m_blocking
= block
;
461 if (socket
->m_fd
!= -1)
462 ioctl(socket
->m_fd
, FIONBIO
, &block
);
466 GSocket_GetError() returns the last error occured on the socket stream.
469 GSocketError
GSocket_GetError(GSocket
*socket
)
471 assert(socket
!= NULL
);
473 return socket
->m_error
;
479 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
480 INPUT: The function is called when there is at least a byte in the
482 OUTPUT: The function is called when the system is sure the next write call
484 CONNECTION: Two cases is possible:
485 Client socket -> the connection is established
486 Server socket -> a client request a connection
487 LOST: the connection is lost
489 SetFallback accepts a combination of these flags so a same callback can
490 receive different events.
492 An event is generated only once and its state is reseted when the relative
493 IO call is requested.
494 For example: INPUT -> GSocket_Read()
495 CONNECTION -> GSocket_Accept()
497 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
498 GSocketFallback fallback
, char *cdata
)
502 assert (socket
!= NULL
);
504 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
505 if ((event
& (1 << count
)) != 0) {
506 socket
->m_fbacks
[count
] = fallback
;
507 socket
->m_data
[count
] = cdata
;
509 _GSocket_Install_Fallback(socket
, count
);
510 _GSocket_Enable(socket
, count
);
516 UnsetFallback will disables all fallbacks specified by "event".
517 NOTE: event may be a combination of flags
519 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
523 assert(socket
!= NULL
);
525 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
526 if ((event
& (1 << count
)) != 0) {
527 _GSocket_Disable(socket
, count
);
528 socket
->m_fbacks
[count
] = NULL
;
529 _GSocket_Uninstall_Fallback(socket
, count
);
534 #define CALL_FALLBACK(socket, event) \
535 if (socket->m_iocalls[event] && \
536 socket->m_fbacks[event]) {\
537 _GSocket_Disable(socket, event); \
538 socket->m_fbacks[event](socket, event, \
539 socket->m_data[event]); \
542 #define MASK_SIGNAL() \
544 void (*old_handler)(int); \
546 old_handler = signal(SIGPIPE, SIG_IGN);
548 #define UNMASK_SIGNAL() \
549 signal(SIGPIPE, old_handler); \
552 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
554 socket
->m_iocalls
[event
] = TRUE
;
555 if (socket
->m_fbacks
[event
])
556 _GSocket_Install_Fallback(socket
, event
);
559 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
561 socket
->m_iocalls
[event
] = FALSE
;
562 if (socket
->m_fbacks
[event
])
563 _GSocket_Uninstall_Fallback(socket
, event
);
566 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
571 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
574 socket
->m_error
= GSOCK_IOERR
;
580 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
582 struct sockaddr from
;
586 fromlen
= sizeof(from
);
589 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
592 socket
->m_error
= GSOCK_IOERR
;
597 socket
->m_peer
= GAddress_new();
598 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
603 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
608 ret
= send(socket
->m_fd
, buffer
, size
, 0);
611 socket
->m_error
= GSOCK_IOERR
;
617 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
619 struct sockaddr
*addr
;
622 if (!socket
->m_peer
) {
623 socket
->m_error
= GSOCK_INVADDR
;
627 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
630 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
633 socket
->m_error
= GSOCK_IOERR
;
642 void _GSocket_Detected_Read(GSocket
*socket
)
647 if (socket
->m_stream
) {
648 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
650 if (ret
< 0 && socket
->m_server
) {
651 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
656 CALL_FALLBACK(socket
, GSOCK_INPUT
);
658 CALL_FALLBACK(socket
, GSOCK_LOST
);
663 void _GSocket_Detected_Write(GSocket
*socket
)
665 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
673 * -------------------------------------------------------------------------
675 * -------------------------------------------------------------------------
678 #define CHECK_ADDRESS(address, family, retval) \
680 if (address->m_family == GSOCK_NOFAMILY) \
681 _GAddress_Init_##family(address); \
682 if (address->m_family != GSOCK_##family) {\
683 address->m_error = GSOCK_INVADDR; \
688 GAddress
*GAddress_new()
692 address
= (GAddress
*)malloc(sizeof(GAddress
));
694 address
->m_family
= GSOCK_NOFAMILY
;
695 address
->m_addr
= NULL
;
701 GAddress
*GAddress_copy(GAddress
*address
)
705 assert(address
!= NULL
);
707 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
708 memcpy(addr2
, address
, sizeof(GAddress
));
710 if (address
->m_addr
) {
711 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
712 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
718 void GAddress_destroy(GAddress
*address
)
720 assert(address
!= NULL
);
725 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
727 assert(address
!= NULL
);
729 address
->m_family
= type
;
732 GAddressType
GAddress_GetFamily(GAddress
*address
)
734 assert(address
!= NULL
);
736 return address
->m_family
;
739 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
740 address
->m_realfamily
= addr
->sa_family
;
741 switch (addr
->sa_family
) {
743 address
->m_family
= GSOCK_INET
;
746 address
->m_family
= GSOCK_UNIX
;
750 address
->m_family
= GSOCK_INET6
;
760 free(address
->m_addr
);
762 address
->m_len
= len
;
763 address
->m_addr
= (struct sockaddr
*)malloc(len
);
764 memcpy(address
->m_addr
, addr
, len
);
767 void _GAddress_translate_to(GAddress
*address
,
768 struct sockaddr
**addr
, int *len
)
770 if (!address
->m_addr
) {
775 *len
= address
->m_len
;
776 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
777 memcpy(*addr
, address
->m_addr
, address
->m_len
);
781 * -------------------------------------------------------------------------
782 * Internet address family
783 * -------------------------------------------------------------------------
786 void _GAddress_Init_INET(GAddress
*address
)
788 address
->m_len
= sizeof(struct sockaddr_in
);
789 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
790 address
->m_family
= GSOCK_INET
;
791 address
->m_realfamily
= PF_INET
;
792 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
793 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
796 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
799 struct in_addr
*addr
;
801 assert(address
!= NULL
);
803 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
805 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
807 /* only name for the moment */
808 if (inet_aton(hostname
, addr
) == 0) {
809 struct in_addr
*array_addr
;
811 if ((he
= gethostbyname(hostname
)) == NULL
) {
812 address
->m_error
= GSOCK_NOHOST
;
815 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
816 addr
->s_addr
= array_addr
[0].s_addr
;
818 return GSOCK_NOERROR
;
821 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
822 unsigned long hostaddr
)
824 struct in_addr
*addr
;
826 assert(address
!= NULL
);
828 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
830 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
831 addr
->s_addr
= hostaddr
;
833 return GSOCK_NOERROR
;
836 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
,
837 const char *protocol
)
840 struct sockaddr_in
*addr
;
842 assert(address
!= NULL
);
843 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
846 address
->m_error
= GSOCK_INVPORT
;
850 /* TODO: TCP or UDP */
851 se
= getservbyname(port
, protocol
);
853 if (isdigit(port
[0])) {
856 port_int
= atoi(port
);
857 addr
= (struct sockaddr_in
*)address
->m_addr
;
858 addr
->sin_port
= htons(port_int
);
859 return GSOCK_NOERROR
;
862 address
->m_error
= GSOCK_INVPORT
;
863 return GSOCK_INVPORT
;
866 addr
= (struct sockaddr_in
*)address
->m_addr
;
867 addr
->sin_port
= se
->s_port
;
869 return GSOCK_NOERROR
;
872 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
874 struct sockaddr_in
*addr
;
876 assert(address
!= NULL
);
877 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
879 addr
= (struct sockaddr_in
*)address
->m_addr
;
880 addr
->sin_port
= htons(port
);
882 return GSOCK_NOERROR
;
885 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
889 struct sockaddr_in
*addr
;
891 assert(address
!= NULL
);
892 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
894 addr
= (struct sockaddr_in
*)address
->m_addr
;
895 addr_buf
= (char *)&(addr
->sin_addr
);
897 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
899 address
->m_error
= GSOCK_NOHOST
;
903 strncpy(hostname
, he
->h_name
, sbuf
);
905 return GSOCK_NOERROR
;
908 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
910 struct sockaddr_in
*addr
;
912 assert(address
!= NULL
);
913 CHECK_ADDRESS(address
, INET
, 0);
915 addr
= (struct sockaddr_in
*)address
->m_addr
;
917 return addr
->sin_addr
.s_addr
;
920 unsigned short GAddress_INET_GetPort(GAddress
*address
)
922 struct sockaddr_in
*addr
;
924 assert(address
!= NULL
);
925 CHECK_ADDRESS(address
, INET
, 0);
927 addr
= (struct sockaddr_in
*)address
->m_addr
;
928 return ntohs(addr
->sin_port
);
932 * -------------------------------------------------------------------------
933 * Unix address family
934 * -------------------------------------------------------------------------
937 void _GAddress_Init_UNIX(GAddress
*address
)
939 address
->m_len
= sizeof(struct sockaddr_un
);
940 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
941 address
->m_family
= GSOCK_UNIX
;
942 address
->m_realfamily
= PF_UNIX
;
943 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
944 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
947 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
949 struct sockaddr_un
*addr
;
951 assert(address
!= NULL
);
953 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
955 addr
= ((struct sockaddr_un
*)address
->m_addr
);
956 memcpy(addr
->sun_path
, path
, strlen(path
));
958 return GSOCK_NOERROR
;
961 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
963 struct sockaddr_un
*addr
;
965 assert(address
!= NULL
);
966 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
968 addr
= (struct sockaddr_un
*)address
->m_addr
;
970 strncpy(path
, addr
->sun_path
, sbuf
);
972 return GSOCK_NOERROR
;
975 #endif // wxUSE_SOCKETS