]>
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>
40 #include <wx/gsocket.h>
47 # define SOCKLEN_T socklen_t
50 # define SOCKLEN_T int
55 /* Constructors / Destructors */
57 GSocket
*GSocket_new()
62 socket
= (GSocket
*)malloc(sizeof(GSocket
));
65 for (i
=0;i
<GSOCK_MAX_EVENT
;i
++) {
66 socket
->m_fbacks
[i
] = NULL
;
67 socket
->m_iocalls
[i
] = FALSE
;
69 socket
->m_local
= NULL
;
70 socket
->m_peer
= NULL
;
71 socket
->m_error
= GSOCK_NOERROR
;
72 socket
->m_server
= FALSE
;
73 socket
->m_stream
= TRUE
;
74 socket
->m_gui_dependent
= NULL
;
75 socket
->m_blocking
= FALSE
;
77 _GSocket_GUI_Init(socket
);
82 void GSocket_destroy(GSocket
*socket
)
84 assert(socket
!= NULL
);
86 if (socket
->m_fd
!= -1)
87 GSocket_Shutdown(socket
);
89 _GSocket_GUI_Destroy(socket
);
92 GAddress_destroy(socket
->m_local
);
95 GAddress_destroy(socket
->m_peer
);
100 void GSocket_Shutdown(GSocket
*socket
)
104 assert(socket
!= NULL
);
106 if (socket
->m_fd
!= -1) {
107 shutdown(socket
->m_fd
, 2);
112 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
113 _GSocket_Uninstall_Fallback(socket
, evt
);
116 /* Address handling */
118 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
120 assert(socket
!= NULL
);
122 if ((socket
->m_fd
!= -1 && !socket
->m_server
))
123 return GSOCK_INVSOCK
;
125 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
)
126 return GSOCK_INVADDR
;
129 GAddress_destroy(socket
->m_local
);
131 socket
->m_local
= GAddress_copy(address
);
133 return GSOCK_NOERROR
;
136 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
138 assert(socket
!= NULL
);
140 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
141 socket
->m_error
= GSOCK_INVADDR
;
142 return GSOCK_INVADDR
;
146 GAddress_destroy(socket
->m_peer
);
148 socket
->m_peer
= GAddress_copy(address
);
150 return GSOCK_NOERROR
;
153 GAddress
*GSocket_GetLocal(GSocket
*socket
)
156 struct sockaddr addr
;
159 assert(socket
!= NULL
);
162 return GAddress_copy(socket
->m_local
);
164 if (socket
->m_fd
== -1) {
165 socket
->m_error
= GSOCK_INVSOCK
;
171 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
172 socket
->m_error
= GSOCK_IOERR
;
176 address
= GAddress_new();
177 _GAddress_translate_from(address
, &addr
, size
);
182 GAddress
*GSocket_GetPeer(GSocket
*socket
)
184 assert(socket
!= NULL
);
187 return GAddress_copy(socket
->m_peer
);
192 /* Server specific parts */
195 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
196 of GSocket. "Local" must be set by GSocket_SetLocal() before
197 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
198 In case, you haven't yet defined the local address, it returns GSOCK_INVADDR.
199 In the other cases it returns GSOCK_IOERR.
201 GSocketError
GSocket_SetServer(GSocket
*sck
)
207 if (sck
->m_fd
!= -1) {
208 sck
->m_error
= GSOCK_INVSOCK
;
209 return GSOCK_INVSOCK
;
213 sck
->m_error
= GSOCK_INVADDR
;
214 return GSOCK_INVADDR
;
222 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
224 if (sck
->m_fd
== -1) {
225 sck
->m_error
= GSOCK_IOERR
;
229 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
232 sck
->m_error
= GSOCK_IOERR
;
236 if (listen(sck
->m_fd
, 5) < 0) {
239 sck
->m_error
= GSOCK_IOERR
;
243 sck
->m_server
= TRUE
;
245 return GSOCK_NOERROR
;
250 GSocket_WaitConnection() waits for an incoming client connection.
252 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
256 assert(socket
!= NULL
);
258 if (socket
->m_fd
== -1 || !socket
->m_server
) {
259 socket
->m_error
= GSOCK_INVSOCK
;
263 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
265 connection
= GSocket_new();
267 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
268 if (connection
->m_fd
== -1) {
269 GSocket_destroy(connection
);
270 socket
->m_error
= GSOCK_IOERR
;
274 connection
->m_stream
= TRUE
;
275 connection
->m_server
= FALSE
;
276 connection
->m_oriented
= TRUE
;
281 /* Non oriented connections */
283 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
287 if (sck
->m_fd
!= -1) {
288 sck
->m_error
= GSOCK_INVSOCK
;
289 return GSOCK_INVSOCK
;
293 sck
->m_error
= GSOCK_INVADDR
;
294 return GSOCK_INVADDR
;
297 sck
->m_stream
= FALSE
;
298 sck
->m_server
= FALSE
;
299 sck
->m_oriented
= FALSE
;
301 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
303 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
306 sck
->m_error
= GSOCK_IOERR
;
310 return GSOCK_NOERROR
;
313 /* Client specific parts */
316 GSocket_Connect() establishes a client connection to a server using the "Peer"
317 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
318 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
320 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
326 if (sck
->m_fd
!= -1) {
327 sck
->m_error
= GSOCK_INVSOCK
;
328 return GSOCK_INVSOCK
;
332 sck
->m_error
= GSOCK_INVADDR
;
333 return GSOCK_INVADDR
;
336 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
337 sck
->m_oriented
= TRUE
;
344 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
346 if (sck
->m_fd
== -1) {
347 sck
->m_error
= GSOCK_IOERR
;
351 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
352 sck
->m_peer
->m_len
) != 0) {
355 sck
->m_error
= GSOCK_IOERR
;
359 sck
->m_server
= FALSE
;
361 return GSOCK_NOERROR
;
366 /* Like recv(), send(), ... */
367 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
369 assert(socket
!= NULL
);
371 if (socket
->m_fd
== -1 || socket
->m_server
) {
372 socket
->m_error
= GSOCK_INVSOCK
;
376 _GSocket_Enable(socket
, GSOCK_INPUT
);
378 if (socket
->m_oriented
)
379 return _GSocket_Recv_Stream(socket
, buffer
, size
);
381 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
384 int GSocket_Write(GSocket
*socket
, const char *buffer
,
387 assert(socket
!= NULL
);
389 if (socket
->m_fd
== -1 || socket
->m_server
) {
390 socket
->m_error
= GSOCK_INVSOCK
;
394 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
396 if (socket
->m_oriented
)
397 return _GSocket_Send_Stream(socket
, buffer
, size
);
399 return _GSocket_Send_Dgram(socket
, buffer
, size
);
402 bool GSocket_DataAvailable(GSocket
*socket
)
407 assert(socket
!= NULL
);
409 if (socket
->m_fd
== -1 || socket
->m_server
) {
410 socket
->m_error
= GSOCK_INVSOCK
;
415 FD_SET(socket
->m_fd
, &read_set
);
420 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
422 return FD_ISSET(socket
->m_fd
, &read_set
);
428 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
429 if we don't want to wait.
431 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
433 assert(socket
!= NULL
);
435 socket
->m_blocking
= block
;
437 if (socket
->m_fd
!= -1)
438 ioctl(socket
->m_fd
, FIONBIO
, &block
);
442 GSocket_GetError() returns the last error occured on the socket stream.
445 GSocketError
GSocket_GetError(GSocket
*socket
)
447 assert(socket
!= NULL
);
449 return socket
->m_error
;
455 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
456 INPUT: The function is called when there is at least a byte in the
458 OUTPUT: The function is called when the system is sure the next write call
460 CONNECTION: Two cases is possible:
461 Client socket -> the connection is established
462 Server socket -> a client request a connection
463 LOST: the connection is lost
465 SetFallback accepts a combination of these flags so a same callback can
466 receive different events.
468 An event is generated only once and its state is reseted when the relative
469 IO call is requested.
470 For example: INPUT -> GSocket_Read()
471 CONNECTION -> GSocket_Accept()
473 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
474 GSocketFallback fallback
, char *cdata
)
478 assert (socket
!= NULL
);
480 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
481 if ((event
& (1 << count
)) != 0) {
482 socket
->m_fbacks
[count
] = fallback
;
483 socket
->m_data
[count
] = cdata
;
485 _GSocket_Install_Fallback(socket
, count
);
486 _GSocket_Enable(socket
, count
);
492 UnsetFallback will disables all fallbacks specified by "event".
493 NOTE: event may be a combination of flags
495 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
499 assert(socket
!= NULL
);
501 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
502 if ((event
& (1 << count
)) != 0) {
503 _GSocket_Disable(socket
, count
);
504 socket
->m_fbacks
[count
] = NULL
;
505 _GSocket_Uninstall_Fallback(socket
, count
);
510 #define CALL_FALLBACK(socket, event) \
511 if (socket->m_iocalls[event] && \
512 socket->m_fbacks[event]) {\
513 _GSocket_Disable(socket, event); \
514 socket->m_fbacks[event](socket, event, \
515 socket->m_data[event]); \
518 #define MASK_SIGNAL() \
520 void (*old_handler)(int); \
522 old_handler = signal(SIGPIPE, SIG_IGN);
524 #define UNMASK_SIGNAL() \
525 signal(SIGPIPE, old_handler); \
528 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
530 socket
->m_iocalls
[event
] = TRUE
;
531 if (socket
->m_fbacks
[event
])
532 _GSocket_Install_Fallback(socket
, event
);
535 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
537 socket
->m_iocalls
[event
] = FALSE
;
538 if (socket
->m_fbacks
[event
])
539 _GSocket_Uninstall_Fallback(socket
, event
);
542 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
547 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
550 socket
->m_error
= GSOCK_IOERR
;
556 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
558 struct sockaddr from
;
561 fromlen
= sizeof(from
);
564 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
567 socket
->m_error
= GSOCK_IOERR
;
572 socket
->m_peer
= GAddress_new();
573 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
578 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
583 ret
= send(socket
->m_fd
, buffer
, size
, 0);
586 socket
->m_error
= GSOCK_IOERR
;
592 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
594 struct sockaddr
*addr
;
597 if (!socket
->m_peer
) {
598 socket
->m_error
= GSOCK_INVADDR
;
602 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
605 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
608 socket
->m_error
= GSOCK_IOERR
;
617 void _GSocket_Detected_Read(GSocket
*socket
)
622 if (socket
->m_stream
) {
623 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
625 if (ret
< 0 && socket
->m_server
) {
626 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
631 CALL_FALLBACK(socket
, GSOCK_INPUT
);
633 CALL_FALLBACK(socket
, GSOCK_LOST
);
638 void _GSocket_Detected_Write(GSocket
*socket
)
640 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
648 * -------------------------------------------------------------------------
650 * -------------------------------------------------------------------------
653 #define CHECK_ADDRESS(address, family, retval) \
655 if (address->m_family == GSOCK_NOFAMILY) \
656 _GAddress_Init_##family(address); \
657 if (address->m_family != GSOCK_##family) {\
658 address->m_error = GSOCK_INVADDR; \
663 GAddress
*GAddress_new()
667 address
= (GAddress
*)malloc(sizeof(GAddress
));
669 address
->m_family
= GSOCK_NOFAMILY
;
670 address
->m_addr
= NULL
;
676 GAddress
*GAddress_copy(GAddress
*address
)
680 assert(address
!= NULL
);
682 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
683 memcpy(addr2
, address
, sizeof(GAddress
));
685 if (address
->m_addr
) {
686 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
687 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
693 void GAddress_destroy(GAddress
*address
)
695 assert(address
!= NULL
);
700 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
702 assert(address
!= NULL
);
704 address
->m_family
= type
;
707 GAddressType
GAddress_GetFamily(GAddress
*address
)
709 assert(address
!= NULL
);
711 return address
->m_family
;
714 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
715 address
->m_realfamily
= addr
->sa_family
;
716 switch (addr
->sa_family
) {
718 address
->m_family
= GSOCK_INET
;
721 address
->m_family
= GSOCK_UNIX
;
725 address
->m_family
= GSOCK_INET6
;
734 free(address
->m_addr
);
736 address
->m_len
= len
;
737 address
->m_addr
= (struct sockaddr
*)malloc(len
);
738 memcpy(address
->m_addr
, addr
, len
);
741 void _GAddress_translate_to(GAddress
*address
,
742 struct sockaddr
**addr
, int *len
)
744 if (!address
->m_addr
) {
749 *len
= address
->m_len
;
750 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
751 memcpy(*addr
, address
->m_addr
, address
->m_len
);
755 * -------------------------------------------------------------------------
756 * Internet address family
757 * -------------------------------------------------------------------------
760 void _GAddress_Init_INET(GAddress
*address
)
762 address
->m_len
= sizeof(struct sockaddr_in
);
763 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
764 address
->m_family
= GSOCK_INET
;
765 address
->m_realfamily
= PF_INET
;
766 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
767 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
770 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
773 struct in_addr
*addr
;
775 assert(address
!= NULL
);
777 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
779 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
781 /* only name for the moment */
782 if (inet_aton(hostname
, addr
) == 0) {
783 struct in_addr
*array_addr
;
785 if ((he
= gethostbyname(hostname
)) == NULL
) {
786 address
->m_error
= GSOCK_NOHOST
;
789 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
790 addr
->s_addr
= array_addr
[0].s_addr
;
792 return GSOCK_NOERROR
;
795 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
796 unsigned long hostaddr
)
798 struct in_addr
*addr
;
800 assert(address
!= NULL
);
802 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
804 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
805 addr
->s_addr
= hostaddr
;
807 return GSOCK_NOERROR
;
810 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
,
811 const char *protocol
)
814 struct sockaddr_in
*addr
;
816 assert(address
!= NULL
);
817 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
820 address
->m_error
= GSOCK_INVPORT
;
824 /* TODO: TCP or UDP */
825 se
= getservbyname(port
, protocol
);
827 if (isdigit(port
[0])) {
830 port_int
= atoi(port
);
831 addr
= (struct sockaddr_in
*)address
->m_addr
;
832 addr
->sin_port
= htons(port_int
);
833 return GSOCK_NOERROR
;
836 address
->m_error
= GSOCK_INVPORT
;
837 return GSOCK_INVPORT
;
840 addr
= (struct sockaddr_in
*)address
->m_addr
;
841 addr
->sin_port
= se
->s_port
;
843 return GSOCK_NOERROR
;
846 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
848 struct sockaddr_in
*addr
;
850 assert(address
!= NULL
);
851 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
853 addr
= (struct sockaddr_in
*)address
->m_addr
;
854 addr
->sin_port
= htons(port
);
856 return GSOCK_NOERROR
;
859 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
863 struct sockaddr_in
*addr
;
865 assert(address
!= NULL
);
866 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
868 addr
= (struct sockaddr_in
*)address
->m_addr
;
869 addr_buf
= (char *)&(addr
->sin_addr
);
871 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
873 address
->m_error
= GSOCK_NOHOST
;
877 strncpy(hostname
, he
->h_name
, sbuf
);
879 return GSOCK_NOERROR
;
882 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
884 struct sockaddr_in
*addr
;
886 assert(address
!= NULL
);
887 CHECK_ADDRESS(address
, INET
, 0);
889 addr
= (struct sockaddr_in
*)address
->m_addr
;
891 return addr
->sin_addr
.s_addr
;
894 unsigned short GAddress_INET_GetPort(GAddress
*address
)
896 struct sockaddr_in
*addr
;
898 assert(address
!= NULL
);
899 CHECK_ADDRESS(address
, INET
, 0);
901 addr
= (struct sockaddr_in
*)address
->m_addr
;
902 return ntohs(addr
->sin_port
);
906 * -------------------------------------------------------------------------
907 * Unix address family
908 * -------------------------------------------------------------------------
911 void _GAddress_Init_UNIX(GAddress
*address
)
913 address
->m_len
= sizeof(struct sockaddr_un
);
914 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
915 address
->m_family
= GSOCK_UNIX
;
916 address
->m_realfamily
= PF_UNIX
;
917 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
918 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
921 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
923 struct sockaddr_un
*addr
;
925 assert(address
!= NULL
);
927 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
929 addr
= ((struct sockaddr_un
*)address
->m_addr
);
930 memcpy(addr
->sun_path
, path
, strlen(path
));
932 return GSOCK_NOERROR
;
935 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
937 struct sockaddr_un
*addr
;
939 assert(address
!= NULL
);
940 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
942 addr
= (struct sockaddr_un
*)address
->m_addr
;
944 strncpy(path
, addr
->sun_path
, sbuf
);
946 return GSOCK_NOERROR
;