]>
git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.c
594c772a26018bb6696699735e1c48e26e9ecb56
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()
69 /* Constructors / Destructors */
71 GSocket
*GSocket_new()
76 socket
= (GSocket
*)malloc(sizeof(GSocket
));
79 for (i
=0;i
<GSOCK_MAX_EVENT
;i
++) {
80 socket
->m_fbacks
[i
] = NULL
;
81 socket
->m_iocalls
[i
] = FALSE
;
83 socket
->m_local
= NULL
;
84 socket
->m_peer
= NULL
;
85 socket
->m_error
= GSOCK_NOERROR
;
86 socket
->m_server
= FALSE
;
87 socket
->m_stream
= TRUE
;
88 socket
->m_gui_dependent
= NULL
;
89 socket
->m_blocking
= FALSE
;
91 _GSocket_GUI_Init(socket
);
96 void GSocket_destroy(GSocket
*socket
)
98 assert(socket
!= NULL
);
100 if (socket
->m_fd
!= -1)
101 GSocket_Shutdown(socket
);
103 _GSocket_GUI_Destroy(socket
);
106 GAddress_destroy(socket
->m_local
);
109 GAddress_destroy(socket
->m_peer
);
114 void GSocket_Shutdown(GSocket
*socket
)
118 assert(socket
!= NULL
);
120 if (socket
->m_fd
!= -1) {
121 shutdown(socket
->m_fd
, 2);
126 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
127 _GSocket_Uninstall_Fallback(socket
, evt
);
130 /* Address handling */
132 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
134 assert(socket
!= NULL
);
136 if ((socket
->m_fd
!= -1 && !socket
->m_server
))
137 return GSOCK_INVSOCK
;
139 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
)
140 return GSOCK_INVADDR
;
143 GAddress_destroy(socket
->m_local
);
145 socket
->m_local
= GAddress_copy(address
);
147 return GSOCK_NOERROR
;
150 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
152 assert(socket
!= NULL
);
154 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
155 socket
->m_error
= GSOCK_INVADDR
;
156 return GSOCK_INVADDR
;
160 GAddress_destroy(socket
->m_peer
);
162 socket
->m_peer
= GAddress_copy(address
);
164 return GSOCK_NOERROR
;
167 GAddress
*GSocket_GetLocal(GSocket
*socket
)
170 struct sockaddr addr
;
173 assert(socket
!= NULL
);
176 return GAddress_copy(socket
->m_local
);
178 if (socket
->m_fd
== -1) {
179 socket
->m_error
= GSOCK_INVSOCK
;
185 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
186 socket
->m_error
= GSOCK_IOERR
;
190 address
= GAddress_new();
191 _GAddress_translate_from(address
, &addr
, size
);
196 GAddress
*GSocket_GetPeer(GSocket
*socket
)
198 assert(socket
!= NULL
);
201 return GAddress_copy(socket
->m_peer
);
206 /* Server specific parts */
209 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
210 of GSocket. "Local" must be set by GSocket_SetLocal() before
211 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
212 In case, you haven't yet defined the local address, it returns GSOCK_INVADDR.
213 In the other cases it returns GSOCK_IOERR.
215 GSocketError
GSocket_SetServer(GSocket
*sck
)
221 if (sck
->m_fd
!= -1) {
222 sck
->m_error
= GSOCK_INVSOCK
;
223 return GSOCK_INVSOCK
;
227 sck
->m_error
= GSOCK_INVADDR
;
228 return GSOCK_INVADDR
;
236 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
238 if (sck
->m_fd
== -1) {
239 sck
->m_error
= GSOCK_IOERR
;
243 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
246 sck
->m_error
= GSOCK_IOERR
;
250 if (listen(sck
->m_fd
, 5) < 0) {
253 sck
->m_error
= GSOCK_IOERR
;
257 sck
->m_server
= TRUE
;
259 return GSOCK_NOERROR
;
264 GSocket_WaitConnection() waits for an incoming client connection.
266 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
270 assert(socket
!= NULL
);
272 if (socket
->m_fd
== -1 || !socket
->m_server
) {
273 socket
->m_error
= GSOCK_INVSOCK
;
277 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
279 connection
= GSocket_new();
281 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
282 if (connection
->m_fd
== -1) {
283 GSocket_destroy(connection
);
284 socket
->m_error
= GSOCK_IOERR
;
288 connection
->m_stream
= TRUE
;
289 connection
->m_server
= FALSE
;
290 connection
->m_oriented
= TRUE
;
295 /* Non oriented connections */
297 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
301 if (sck
->m_fd
!= -1) {
302 sck
->m_error
= GSOCK_INVSOCK
;
303 return GSOCK_INVSOCK
;
307 sck
->m_error
= GSOCK_INVADDR
;
308 return GSOCK_INVADDR
;
311 sck
->m_stream
= FALSE
;
312 sck
->m_server
= FALSE
;
313 sck
->m_oriented
= FALSE
;
315 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
317 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
320 sck
->m_error
= GSOCK_IOERR
;
324 return GSOCK_NOERROR
;
327 /* Client specific parts */
330 GSocket_Connect() establishes a client connection to a server using the "Peer"
331 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
332 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
334 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
340 if (sck
->m_fd
!= -1) {
341 sck
->m_error
= GSOCK_INVSOCK
;
342 return GSOCK_INVSOCK
;
346 sck
->m_error
= GSOCK_INVADDR
;
347 return GSOCK_INVADDR
;
350 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
351 sck
->m_oriented
= TRUE
;
358 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
360 if (sck
->m_fd
== -1) {
361 sck
->m_error
= GSOCK_IOERR
;
365 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
366 sck
->m_peer
->m_len
) != 0) {
369 sck
->m_error
= GSOCK_IOERR
;
373 sck
->m_server
= FALSE
;
375 return GSOCK_NOERROR
;
380 /* Like recv(), send(), ... */
381 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
383 assert(socket
!= NULL
);
385 if (socket
->m_fd
== -1 || socket
->m_server
) {
386 socket
->m_error
= GSOCK_INVSOCK
;
390 _GSocket_Enable(socket
, GSOCK_INPUT
);
392 if (socket
->m_oriented
)
393 return _GSocket_Recv_Stream(socket
, buffer
, size
);
395 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
398 int GSocket_Write(GSocket
*socket
, const char *buffer
,
401 assert(socket
!= NULL
);
403 if (socket
->m_fd
== -1 || socket
->m_server
) {
404 socket
->m_error
= GSOCK_INVSOCK
;
408 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
410 if (socket
->m_oriented
)
411 return _GSocket_Send_Stream(socket
, buffer
, size
);
413 return _GSocket_Send_Dgram(socket
, buffer
, size
);
416 bool GSocket_DataAvailable(GSocket
*socket
)
421 assert(socket
!= NULL
);
423 if (socket
->m_fd
== -1 || socket
->m_server
) {
424 socket
->m_error
= GSOCK_INVSOCK
;
429 FD_SET(socket
->m_fd
, &read_set
);
434 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
436 return FD_ISSET(socket
->m_fd
, &read_set
);
442 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
443 if we don't want to wait.
445 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
447 assert(socket
!= NULL
);
449 socket
->m_blocking
= block
;
451 if (socket
->m_fd
!= -1)
452 ioctl(socket
->m_fd
, FIONBIO
, &block
);
456 GSocket_GetError() returns the last error occured on the socket stream.
459 GSocketError
GSocket_GetError(GSocket
*socket
)
461 assert(socket
!= NULL
);
463 return socket
->m_error
;
469 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
470 INPUT: The function is called when there is at least a byte in the
472 OUTPUT: The function is called when the system is sure the next write call
474 CONNECTION: Two cases is possible:
475 Client socket -> the connection is established
476 Server socket -> a client request a connection
477 LOST: the connection is lost
479 SetFallback accepts a combination of these flags so a same callback can
480 receive different events.
482 An event is generated only once and its state is reseted when the relative
483 IO call is requested.
484 For example: INPUT -> GSocket_Read()
485 CONNECTION -> GSocket_Accept()
487 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
488 GSocketFallback fallback
, char *cdata
)
492 assert (socket
!= NULL
);
494 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
495 if ((event
& (1 << count
)) != 0) {
496 socket
->m_fbacks
[count
] = fallback
;
497 socket
->m_data
[count
] = cdata
;
499 _GSocket_Install_Fallback(socket
, count
);
500 _GSocket_Enable(socket
, count
);
506 UnsetFallback will disables all fallbacks specified by "event".
507 NOTE: event may be a combination of flags
509 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
513 assert(socket
!= NULL
);
515 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
516 if ((event
& (1 << count
)) != 0) {
517 _GSocket_Disable(socket
, count
);
518 socket
->m_fbacks
[count
] = NULL
;
519 _GSocket_Uninstall_Fallback(socket
, count
);
524 #define CALL_FALLBACK(socket, event) \
525 if (socket->m_iocalls[event] && \
526 socket->m_fbacks[event]) {\
527 _GSocket_Disable(socket, event); \
528 socket->m_fbacks[event](socket, event, \
529 socket->m_data[event]); \
532 #define MASK_SIGNAL() \
534 void (*old_handler)(int); \
536 old_handler = signal(SIGPIPE, SIG_IGN);
538 #define UNMASK_SIGNAL() \
539 signal(SIGPIPE, old_handler); \
542 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
544 socket
->m_iocalls
[event
] = TRUE
;
545 if (socket
->m_fbacks
[event
])
546 _GSocket_Install_Fallback(socket
, event
);
549 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
551 socket
->m_iocalls
[event
] = FALSE
;
552 if (socket
->m_fbacks
[event
])
553 _GSocket_Uninstall_Fallback(socket
, event
);
556 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
561 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
564 socket
->m_error
= GSOCK_IOERR
;
570 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
572 struct sockaddr from
;
575 fromlen
= sizeof(from
);
578 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
581 socket
->m_error
= GSOCK_IOERR
;
586 socket
->m_peer
= GAddress_new();
587 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
592 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
597 ret
= send(socket
->m_fd
, buffer
, size
, 0);
600 socket
->m_error
= GSOCK_IOERR
;
606 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
608 struct sockaddr
*addr
;
611 if (!socket
->m_peer
) {
612 socket
->m_error
= GSOCK_INVADDR
;
616 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
619 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
622 socket
->m_error
= GSOCK_IOERR
;
631 void _GSocket_Detected_Read(GSocket
*socket
)
636 if (socket
->m_stream
) {
637 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
639 if (ret
< 0 && socket
->m_server
) {
640 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
645 CALL_FALLBACK(socket
, GSOCK_INPUT
);
647 CALL_FALLBACK(socket
, GSOCK_LOST
);
652 void _GSocket_Detected_Write(GSocket
*socket
)
654 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
662 * -------------------------------------------------------------------------
664 * -------------------------------------------------------------------------
667 #define CHECK_ADDRESS(address, family, retval) \
669 if (address->m_family == GSOCK_NOFAMILY) \
670 _GAddress_Init_##family(address); \
671 if (address->m_family != GSOCK_##family) {\
672 address->m_error = GSOCK_INVADDR; \
677 GAddress
*GAddress_new()
681 address
= (GAddress
*)malloc(sizeof(GAddress
));
683 address
->m_family
= GSOCK_NOFAMILY
;
684 address
->m_addr
= NULL
;
690 GAddress
*GAddress_copy(GAddress
*address
)
694 assert(address
!= NULL
);
696 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
697 memcpy(addr2
, address
, sizeof(GAddress
));
699 if (address
->m_addr
) {
700 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
701 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
707 void GAddress_destroy(GAddress
*address
)
709 assert(address
!= NULL
);
714 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
716 assert(address
!= NULL
);
718 address
->m_family
= type
;
721 GAddressType
GAddress_GetFamily(GAddress
*address
)
723 assert(address
!= NULL
);
725 return address
->m_family
;
728 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
729 address
->m_realfamily
= addr
->sa_family
;
730 switch (addr
->sa_family
) {
732 address
->m_family
= GSOCK_INET
;
735 address
->m_family
= GSOCK_UNIX
;
739 address
->m_family
= GSOCK_INET6
;
748 free(address
->m_addr
);
750 address
->m_len
= len
;
751 address
->m_addr
= (struct sockaddr
*)malloc(len
);
752 memcpy(address
->m_addr
, addr
, len
);
755 void _GAddress_translate_to(GAddress
*address
,
756 struct sockaddr
**addr
, int *len
)
758 if (!address
->m_addr
) {
763 *len
= address
->m_len
;
764 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
765 memcpy(*addr
, address
->m_addr
, address
->m_len
);
769 * -------------------------------------------------------------------------
770 * Internet address family
771 * -------------------------------------------------------------------------
774 void _GAddress_Init_INET(GAddress
*address
)
776 address
->m_len
= sizeof(struct sockaddr_in
);
777 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
778 address
->m_family
= GSOCK_INET
;
779 address
->m_realfamily
= PF_INET
;
780 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
781 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
784 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
787 struct in_addr
*addr
;
789 assert(address
!= NULL
);
791 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
793 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
795 /* only name for the moment */
796 if (inet_aton(hostname
, addr
) == 0) {
797 struct in_addr
*array_addr
;
799 if ((he
= gethostbyname(hostname
)) == NULL
) {
800 address
->m_error
= GSOCK_NOHOST
;
803 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
804 addr
->s_addr
= array_addr
[0].s_addr
;
806 return GSOCK_NOERROR
;
809 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
810 unsigned long hostaddr
)
812 struct in_addr
*addr
;
814 assert(address
!= NULL
);
816 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
818 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
819 addr
->s_addr
= hostaddr
;
821 return GSOCK_NOERROR
;
824 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
,
825 const char *protocol
)
828 struct sockaddr_in
*addr
;
830 assert(address
!= NULL
);
831 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
834 address
->m_error
= GSOCK_INVPORT
;
838 /* TODO: TCP or UDP */
839 se
= getservbyname(port
, protocol
);
841 if (isdigit(port
[0])) {
844 port_int
= atoi(port
);
845 addr
= (struct sockaddr_in
*)address
->m_addr
;
846 addr
->sin_port
= htons(port_int
);
847 return GSOCK_NOERROR
;
850 address
->m_error
= GSOCK_INVPORT
;
851 return GSOCK_INVPORT
;
854 addr
= (struct sockaddr_in
*)address
->m_addr
;
855 addr
->sin_port
= se
->s_port
;
857 return GSOCK_NOERROR
;
860 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
862 struct sockaddr_in
*addr
;
864 assert(address
!= NULL
);
865 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
867 addr
= (struct sockaddr_in
*)address
->m_addr
;
868 addr
->sin_port
= htons(port
);
870 return GSOCK_NOERROR
;
873 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
877 struct sockaddr_in
*addr
;
879 assert(address
!= NULL
);
880 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
882 addr
= (struct sockaddr_in
*)address
->m_addr
;
883 addr_buf
= (char *)&(addr
->sin_addr
);
885 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
887 address
->m_error
= GSOCK_NOHOST
;
891 strncpy(hostname
, he
->h_name
, sbuf
);
893 return GSOCK_NOERROR
;
896 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
898 struct sockaddr_in
*addr
;
900 assert(address
!= NULL
);
901 CHECK_ADDRESS(address
, INET
, 0);
903 addr
= (struct sockaddr_in
*)address
->m_addr
;
905 return addr
->sin_addr
.s_addr
;
908 unsigned short GAddress_INET_GetPort(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
;
916 return ntohs(addr
->sin_port
);
920 * -------------------------------------------------------------------------
921 * Unix address family
922 * -------------------------------------------------------------------------
925 void _GAddress_Init_UNIX(GAddress
*address
)
927 address
->m_len
= sizeof(struct sockaddr_un
);
928 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
929 address
->m_family
= GSOCK_UNIX
;
930 address
->m_realfamily
= PF_UNIX
;
931 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
932 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
935 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
937 struct sockaddr_un
*addr
;
939 assert(address
!= NULL
);
941 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
943 addr
= ((struct sockaddr_un
*)address
->m_addr
);
944 memcpy(addr
->sun_path
, path
, strlen(path
));
946 return GSOCK_NOERROR
;
949 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
951 struct sockaddr_un
*addr
;
953 assert(address
!= NULL
);
954 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
956 addr
= (struct sockaddr_un
*)address
->m_addr
;
958 strncpy(path
, addr
->sun_path
, sbuf
);
960 return GSOCK_NOERROR
;