]>
git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.c
86711be59972c3ee6528988611280cbcc6a19cab
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>
39 #include "wx/gsocket.h"
46 # define SOCKLEN_T socklen_t
49 # define SOCKLEN_T int
54 /* 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 /* We initialize the GUI specific entries here */
88 _GSocket_GUI_Init(socket
);
93 void GSocket_destroy(GSocket
*socket
)
95 assert(socket
!= NULL
);
97 /* First, we check that the socket is really shutdowned */
98 if (socket
->m_fd
!= -1)
99 GSocket_Shutdown(socket
);
101 /* We destroy GUI specific variables */
102 _GSocket_GUI_Destroy(socket
);
104 /* We destroy private addresses */
106 GAddress_destroy(socket
->m_local
);
109 GAddress_destroy(socket
->m_peer
);
111 /* We destroy socket itself */
115 void GSocket_Shutdown(GSocket
*socket
)
119 assert(socket
!= NULL
);
121 /* If socket has been created, we shutdown it */
122 if (socket
->m_fd
!= -1) {
123 shutdown(socket
->m_fd
, 2);
128 /* We also disable GUI callbacks */
129 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
130 _GSocket_Uninstall_Fallback(socket
, evt
);
133 /* Address handling */
135 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
137 assert(socket
!= NULL
);
139 if ((socket
->m_fd
!= -1 && !socket
->m_server
)) {
140 socket
->m_error
= GSOCK_INVSOCK
;
141 return GSOCK_INVSOCK
;
144 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
145 socket
->m_error
= GSOCK_INVADDR
;
146 return GSOCK_INVADDR
;
150 GAddress_destroy(socket
->m_local
);
152 socket
->m_local
= GAddress_copy(address
);
154 return GSOCK_NOERROR
;
157 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
159 assert(socket
!= NULL
);
161 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
162 socket
->m_error
= GSOCK_INVADDR
;
163 return GSOCK_INVADDR
;
167 GAddress_destroy(socket
->m_peer
);
169 socket
->m_peer
= GAddress_copy(address
);
171 return GSOCK_NOERROR
;
174 GAddress
*GSocket_GetLocal(GSocket
*socket
)
177 struct sockaddr addr
;
180 assert(socket
!= NULL
);
183 return GAddress_copy(socket
->m_local
);
185 if (socket
->m_fd
== -1) {
186 socket
->m_error
= GSOCK_INVSOCK
;
192 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
193 socket
->m_error
= GSOCK_IOERR
;
197 address
= GAddress_new();
198 _GAddress_translate_from(address
, &addr
, size
);
203 GAddress
*GSocket_GetPeer(GSocket
*socket
)
205 assert(socket
!= NULL
);
208 return GAddress_copy(socket
->m_peer
);
213 /* Server specific parts */
216 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
217 of GSocket. "Local" must be set by GSocket_SetLocal() before
218 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
219 In case, you haven't yet defined the local address, it returns GSOCK_INVADDR.
220 In the other cases it returns GSOCK_IOERR.
222 GSocketError
GSocket_SetServer(GSocket
*sck
)
228 if (sck
->m_fd
!= -1) {
229 sck
->m_error
= GSOCK_INVSOCK
;
230 return GSOCK_INVSOCK
;
234 sck
->m_error
= GSOCK_INVADDR
;
235 return GSOCK_INVADDR
;
243 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
245 if (sck
->m_fd
== -1) {
246 sck
->m_error
= GSOCK_IOERR
;
250 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
253 sck
->m_error
= GSOCK_IOERR
;
257 if (listen(sck
->m_fd
, 5) < 0) {
260 sck
->m_error
= GSOCK_IOERR
;
264 sck
->m_server
= TRUE
;
266 return GSOCK_NOERROR
;
271 GSocket_WaitConnection() waits for an incoming client connection.
273 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
277 assert(socket
!= NULL
);
279 if (socket
->m_fd
== -1 || !socket
->m_server
) {
280 socket
->m_error
= GSOCK_INVSOCK
;
284 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
286 connection
= GSocket_new();
288 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
289 if (connection
->m_fd
== -1) {
290 GSocket_destroy(connection
);
291 socket
->m_error
= GSOCK_IOERR
;
295 connection
->m_stream
= TRUE
;
296 connection
->m_server
= FALSE
;
297 connection
->m_oriented
= TRUE
;
302 /* Non oriented connections */
304 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
308 if (sck
->m_fd
!= -1) {
309 sck
->m_error
= GSOCK_INVSOCK
;
310 return GSOCK_INVSOCK
;
314 sck
->m_error
= GSOCK_INVADDR
;
315 return GSOCK_INVADDR
;
318 sck
->m_stream
= FALSE
;
319 sck
->m_server
= FALSE
;
320 sck
->m_oriented
= FALSE
;
322 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
324 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
327 sck
->m_error
= GSOCK_IOERR
;
331 return GSOCK_NOERROR
;
334 /* Client specific parts */
337 GSocket_Connect() establishes a client connection to a server using the "Peer"
338 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
339 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
341 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
347 if (sck
->m_fd
!= -1) {
348 sck
->m_error
= GSOCK_INVSOCK
;
349 return GSOCK_INVSOCK
;
353 sck
->m_error
= GSOCK_INVADDR
;
354 return GSOCK_INVADDR
;
357 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
358 sck
->m_oriented
= TRUE
;
365 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
367 if (sck
->m_fd
== -1) {
368 sck
->m_error
= GSOCK_IOERR
;
372 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
373 sck
->m_peer
->m_len
) != 0) {
376 sck
->m_error
= GSOCK_IOERR
;
380 sck
->m_server
= FALSE
;
382 return GSOCK_NOERROR
;
387 /* Like recv(), send(), ... */
388 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
390 assert(socket
!= NULL
);
392 if (socket
->m_fd
== -1 || socket
->m_server
) {
393 socket
->m_error
= GSOCK_INVSOCK
;
397 _GSocket_Enable(socket
, GSOCK_INPUT
);
399 if (socket
->m_oriented
)
400 return _GSocket_Recv_Stream(socket
, buffer
, size
);
402 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
405 int GSocket_Write(GSocket
*socket
, const char *buffer
,
408 assert(socket
!= NULL
);
410 if (socket
->m_fd
== -1 || socket
->m_server
) {
411 socket
->m_error
= GSOCK_INVSOCK
;
415 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
417 if (socket
->m_oriented
)
418 return _GSocket_Send_Stream(socket
, buffer
, size
);
420 return _GSocket_Send_Dgram(socket
, buffer
, size
);
423 bool GSocket_DataAvailable(GSocket
*socket
)
428 assert(socket
!= NULL
);
430 if (socket
->m_fd
== -1 || socket
->m_server
) {
431 socket
->m_error
= GSOCK_INVSOCK
;
436 FD_SET(socket
->m_fd
, &read_set
);
441 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
443 return FD_ISSET(socket
->m_fd
, &read_set
);
449 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
450 if we don't want to wait.
452 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
454 assert(socket
!= NULL
);
456 socket
->m_blocking
= block
;
458 if (socket
->m_fd
!= -1)
459 ioctl(socket
->m_fd
, FIONBIO
, &block
);
463 GSocket_GetError() returns the last error occured on the socket stream.
466 GSocketError
GSocket_GetError(GSocket
*socket
)
468 assert(socket
!= NULL
);
470 return socket
->m_error
;
476 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
477 INPUT: The function is called when there is at least a byte in the
479 OUTPUT: The function is called when the system is sure the next write call
481 CONNECTION: Two cases is possible:
482 Client socket -> the connection is established
483 Server socket -> a client request a connection
484 LOST: the connection is lost
486 SetFallback accepts a combination of these flags so a same callback can
487 receive different events.
489 An event is generated only once and its state is reseted when the relative
490 IO call is requested.
491 For example: INPUT -> GSocket_Read()
492 CONNECTION -> GSocket_Accept()
494 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
495 GSocketFallback fallback
, char *cdata
)
499 assert (socket
!= NULL
);
501 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
502 if ((event
& (1 << count
)) != 0) {
503 socket
->m_fbacks
[count
] = fallback
;
504 socket
->m_data
[count
] = cdata
;
506 _GSocket_Install_Fallback(socket
, count
);
507 _GSocket_Enable(socket
, count
);
513 UnsetFallback will disables all fallbacks specified by "event".
514 NOTE: event may be a combination of flags
516 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
520 assert(socket
!= NULL
);
522 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
523 if ((event
& (1 << count
)) != 0) {
524 _GSocket_Disable(socket
, count
);
525 socket
->m_fbacks
[count
] = NULL
;
526 _GSocket_Uninstall_Fallback(socket
, count
);
531 #define CALL_FALLBACK(socket, event) \
532 if (socket->m_iocalls[event] && \
533 socket->m_fbacks[event]) {\
534 _GSocket_Disable(socket, event); \
535 socket->m_fbacks[event](socket, event, \
536 socket->m_data[event]); \
539 #define MASK_SIGNAL() \
541 void (*old_handler)(int); \
543 old_handler = signal(SIGPIPE, SIG_IGN);
545 #define UNMASK_SIGNAL() \
546 signal(SIGPIPE, old_handler); \
549 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
551 socket
->m_iocalls
[event
] = TRUE
;
552 if (socket
->m_fbacks
[event
])
553 _GSocket_Install_Fallback(socket
, event
);
556 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
558 socket
->m_iocalls
[event
] = FALSE
;
559 if (socket
->m_fbacks
[event
])
560 _GSocket_Uninstall_Fallback(socket
, event
);
563 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
568 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
571 socket
->m_error
= GSOCK_IOERR
;
577 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
579 struct sockaddr from
;
583 fromlen
= sizeof(from
);
586 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
589 socket
->m_error
= GSOCK_IOERR
;
594 socket
->m_peer
= GAddress_new();
595 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
600 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
605 ret
= send(socket
->m_fd
, buffer
, size
, 0);
608 socket
->m_error
= GSOCK_IOERR
;
614 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
616 struct sockaddr
*addr
;
619 if (!socket
->m_peer
) {
620 socket
->m_error
= GSOCK_INVADDR
;
624 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
627 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
630 socket
->m_error
= GSOCK_IOERR
;
639 void _GSocket_Detected_Read(GSocket
*socket
)
644 if (socket
->m_stream
) {
645 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
647 if (ret
< 0 && socket
->m_server
) {
648 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
653 CALL_FALLBACK(socket
, GSOCK_INPUT
);
655 CALL_FALLBACK(socket
, GSOCK_LOST
);
660 void _GSocket_Detected_Write(GSocket
*socket
)
662 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
670 * -------------------------------------------------------------------------
672 * -------------------------------------------------------------------------
675 #define CHECK_ADDRESS(address, family, retval) \
677 if (address->m_family == GSOCK_NOFAMILY) \
678 _GAddress_Init_##family(address); \
679 if (address->m_family != GSOCK_##family) {\
680 address->m_error = GSOCK_INVADDR; \
685 GAddress
*GAddress_new()
689 address
= (GAddress
*)malloc(sizeof(GAddress
));
691 address
->m_family
= GSOCK_NOFAMILY
;
692 address
->m_addr
= NULL
;
698 GAddress
*GAddress_copy(GAddress
*address
)
702 assert(address
!= NULL
);
704 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
705 memcpy(addr2
, address
, sizeof(GAddress
));
707 if (address
->m_addr
) {
708 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
709 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
715 void GAddress_destroy(GAddress
*address
)
717 assert(address
!= NULL
);
722 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
724 assert(address
!= NULL
);
726 address
->m_family
= type
;
729 GAddressType
GAddress_GetFamily(GAddress
*address
)
731 assert(address
!= NULL
);
733 return address
->m_family
;
736 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
737 address
->m_realfamily
= addr
->sa_family
;
738 switch (addr
->sa_family
) {
740 address
->m_family
= GSOCK_INET
;
743 address
->m_family
= GSOCK_UNIX
;
747 address
->m_family
= GSOCK_INET6
;
757 free(address
->m_addr
);
759 address
->m_len
= len
;
760 address
->m_addr
= (struct sockaddr
*)malloc(len
);
761 memcpy(address
->m_addr
, addr
, len
);
764 void _GAddress_translate_to(GAddress
*address
,
765 struct sockaddr
**addr
, int *len
)
767 if (!address
->m_addr
) {
772 *len
= address
->m_len
;
773 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
774 memcpy(*addr
, address
->m_addr
, address
->m_len
);
778 * -------------------------------------------------------------------------
779 * Internet address family
780 * -------------------------------------------------------------------------
783 void _GAddress_Init_INET(GAddress
*address
)
785 address
->m_len
= sizeof(struct sockaddr_in
);
786 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
787 address
->m_family
= GSOCK_INET
;
788 address
->m_realfamily
= PF_INET
;
789 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
790 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
793 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
796 struct in_addr
*addr
;
798 assert(address
!= NULL
);
800 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
802 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
804 /* only name for the moment */
805 if (inet_aton(hostname
, addr
) == 0) {
806 struct in_addr
*array_addr
;
808 if ((he
= gethostbyname(hostname
)) == NULL
) {
809 address
->m_error
= GSOCK_NOHOST
;
812 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
813 addr
->s_addr
= array_addr
[0].s_addr
;
815 return GSOCK_NOERROR
;
818 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
819 unsigned long hostaddr
)
821 struct in_addr
*addr
;
823 assert(address
!= NULL
);
825 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
827 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
828 addr
->s_addr
= hostaddr
;
830 return GSOCK_NOERROR
;
833 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
,
834 const char *protocol
)
837 struct sockaddr_in
*addr
;
839 assert(address
!= NULL
);
840 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
843 address
->m_error
= GSOCK_INVPORT
;
847 /* TODO: TCP or UDP */
848 se
= getservbyname(port
, protocol
);
850 if (isdigit(port
[0])) {
853 port_int
= atoi(port
);
854 addr
= (struct sockaddr_in
*)address
->m_addr
;
855 addr
->sin_port
= htons(port_int
);
856 return GSOCK_NOERROR
;
859 address
->m_error
= GSOCK_INVPORT
;
860 return GSOCK_INVPORT
;
863 addr
= (struct sockaddr_in
*)address
->m_addr
;
864 addr
->sin_port
= se
->s_port
;
866 return GSOCK_NOERROR
;
869 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
871 struct sockaddr_in
*addr
;
873 assert(address
!= NULL
);
874 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
876 addr
= (struct sockaddr_in
*)address
->m_addr
;
877 addr
->sin_port
= htons(port
);
879 return GSOCK_NOERROR
;
882 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
886 struct sockaddr_in
*addr
;
888 assert(address
!= NULL
);
889 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
891 addr
= (struct sockaddr_in
*)address
->m_addr
;
892 addr_buf
= (char *)&(addr
->sin_addr
);
894 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
896 address
->m_error
= GSOCK_NOHOST
;
900 strncpy(hostname
, he
->h_name
, sbuf
);
902 return GSOCK_NOERROR
;
905 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
907 struct sockaddr_in
*addr
;
909 assert(address
!= NULL
);
910 CHECK_ADDRESS(address
, INET
, 0);
912 addr
= (struct sockaddr_in
*)address
->m_addr
;
914 return addr
->sin_addr
.s_addr
;
917 unsigned short GAddress_INET_GetPort(GAddress
*address
)
919 struct sockaddr_in
*addr
;
921 assert(address
!= NULL
);
922 CHECK_ADDRESS(address
, INET
, 0);
924 addr
= (struct sockaddr_in
*)address
->m_addr
;
925 return ntohs(addr
->sin_port
);
929 * -------------------------------------------------------------------------
930 * Unix address family
931 * -------------------------------------------------------------------------
934 void _GAddress_Init_UNIX(GAddress
*address
)
936 address
->m_len
= sizeof(struct sockaddr_un
);
937 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
938 address
->m_family
= GSOCK_UNIX
;
939 address
->m_realfamily
= PF_UNIX
;
940 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
941 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
944 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
946 struct sockaddr_un
*addr
;
948 assert(address
!= NULL
);
950 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
952 addr
= ((struct sockaddr_un
*)address
->m_addr
);
953 memcpy(addr
->sun_path
, path
, strlen(path
));
955 return GSOCK_NOERROR
;
958 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
960 struct sockaddr_un
*addr
;
962 assert(address
!= NULL
);
963 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
965 addr
= (struct sockaddr_un
*)address
->m_addr
;
967 strncpy(path
, addr
->sun_path
, sbuf
);
969 return GSOCK_NOERROR
;