]>
git.saurik.com Git - wxWidgets.git/blob - src/unix/gsocket.c
7d489ecdd05767c8692d5683dd3d559fd42273b1
1 /* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
4 * Purpose: GSocket main Unix file
7 * Log: Revision 1.2 1999/07/23 09:48:31 KB
8 * Log: Fixed stupid new bugs :-(
10 * Log: Revision 1.1 1999/07/22 17:51:54 GL
11 * Log: Added GSocket for Unix (only GTK for the moment)
12 * Log: Updated wxSocket to use GSocket API
13 * Log: Added a progress bar to client.cpp
14 * Log: Added CopyTo to wxMemoryOutputStream to copy the internal buffer to a specified buffer.
15 * Log: Various changes/fixes to the high-level protocols FTP/HTTP
16 * Log: Various Unicode fixes
17 * Log: Removed sckint.*
19 * Log: Revision 1.2 1999/07/18 15:52:34 guilhem
20 * Log: * Copyright, etc.
22 * -------------------------------------------------------------------------
26 #include <sys/ioctl.h>
27 #include <sys/types.h>
31 #include <sys/socket.h>
35 #include <netinet/in.h>
36 #include <arpa/inet.h>
45 #include <sys/filio.h>
54 #include <wx/gsocket.h>
58 # define SOCKLEN_T int
61 /* Constructors / Destructors */
63 GSocket
*GSocket_new()
68 socket
= (GSocket
*)malloc(sizeof(GSocket
));
71 for (i
=0;i
<GSOCK_MAX_EVENT
;i
++) {
72 socket
->m_fbacks
[i
] = NULL
;
73 socket
->m_iocalls
[i
] = FALSE
;
75 socket
->m_local
= NULL
;
76 socket
->m_peer
= NULL
;
77 socket
->m_error
= GSOCK_NOERROR
;
78 socket
->m_server
= FALSE
;
79 socket
->m_stream
= TRUE
;
80 socket
->m_gui_dependent
= NULL
;
81 socket
->m_blocking
= FALSE
;
83 _GSocket_GUI_Init(socket
);
88 void GSocket_destroy(GSocket
*socket
)
90 assert(socket
!= NULL
);
92 _GSocket_GUI_Destroy(socket
);
94 if (socket
->m_fd
!= -1)
95 GSocket_Shutdown(socket
);
98 GAddress_destroy(socket
->m_local
);
101 GAddress_destroy(socket
->m_peer
);
106 void GSocket_Shutdown(GSocket
*socket
)
110 assert(socket
!= NULL
);
112 if (socket
->m_fd
!= -1) {
113 shutdown(socket
->m_fd
, 2);
118 for (evt
=0;evt
<GSOCK_MAX_EVENT
;evt
++)
119 _GSocket_Uninstall_Fallback(socket
, evt
);
122 /* Address handling */
124 GSocketError
GSocket_SetLocal(GSocket
*socket
, GAddress
*address
)
126 if (socket
== NULL
|| (socket
->m_fd
!= -1 && !socket
->m_server
))
127 return GSOCK_INVSOCK
;
129 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
)
130 return GSOCK_INVADDR
;
133 GAddress_destroy(socket
->m_local
);
135 socket
->m_local
= GAddress_copy(address
);
137 return GSOCK_NOERROR
;
140 GSocketError
GSocket_SetPeer(GSocket
*socket
, GAddress
*address
)
143 return GSOCK_INVSOCK
;
145 if (address
== NULL
|| address
->m_family
== GSOCK_NOFAMILY
) {
146 socket
->m_error
= GSOCK_INVADDR
;
147 return GSOCK_INVADDR
;
151 GAddress_destroy(socket
->m_peer
);
153 socket
->m_peer
= GAddress_copy(address
);
155 return GSOCK_NOERROR
;
158 GAddress
*GSocket_GetLocal(GSocket
*socket
)
161 struct sockaddr addr
;
164 assert(socket
!= NULL
);
167 return GAddress_copy(socket
->m_local
);
169 if (socket
->m_fd
== -1) {
170 socket
->m_error
= GSOCK_INVSOCK
;
176 if (getsockname(socket
->m_fd
, &addr
, &size
) < 0) {
177 socket
->m_error
= GSOCK_IOERR
;
181 address
= GAddress_new();
182 _GAddress_translate_from(address
, &addr
, size
);
187 GAddress
*GSocket_GetPeer(GSocket
*socket
)
189 assert(socket
!= NULL
);
192 return GAddress_copy(socket
->m_peer
);
197 /* Server specific parts */
200 GSocket_SetServer() setup the socket as a server. It uses the "Local" field
201 of GSocket. "Local" must be set by GSocket_SetLocal() before
202 GSocket_SetServer() is called. GSOCK_INVSOCK if socket has been initialized.
203 In the other cases, it returns GSOCK_INVADDR.
205 GSocketError
GSocket_SetServer(GSocket
*sck
)
211 if (sck
->m_fd
!= -1) {
212 sck
->m_error
= GSOCK_INVSOCK
;
213 return GSOCK_INVSOCK
;
217 sck
->m_error
= GSOCK_INVADDR
;
218 return GSOCK_INVADDR
;
226 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, type
, 0);
228 if (sck
->m_fd
== -1) {
229 sck
->m_error
= GSOCK_IOERR
;
233 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
236 sck
->m_error
= GSOCK_IOERR
;
240 if (listen(sck
->m_fd
, 5) < 0) {
243 sck
->m_error
= GSOCK_IOERR
;
247 sck
->m_server
= TRUE
;
249 return GSOCK_NOERROR
;
254 GSocket_WaitConnection() waits for an incoming client connection.
256 GSocket
*GSocket_WaitConnection(GSocket
*socket
)
260 assert(socket
!= NULL
);
262 if (socket
->m_fd
== -1 || !socket
->m_server
) {
263 socket
->m_error
= GSOCK_INVSOCK
;
267 _GSocket_Enable(socket
, GSOCK_CONNECTION
);
269 connection
= GSocket_new();
271 connection
->m_fd
= accept(socket
->m_fd
, NULL
, NULL
);
272 if (connection
->m_fd
== -1) {
273 GSocket_destroy(connection
);
274 socket
->m_error
= GSOCK_IOERR
;
278 connection
->m_stream
= TRUE
;
279 connection
->m_server
= FALSE
;
280 connection
->m_oriented
= TRUE
;
285 /* Non oriented connections */
287 GSocketError
GSocket_SetNonOriented(GSocket
*sck
)
291 if (sck
->m_fd
!= -1) {
292 sck
->m_error
= GSOCK_INVSOCK
;
293 return GSOCK_INVSOCK
;
297 sck
->m_error
= GSOCK_INVADDR
;
298 return GSOCK_INVADDR
;
301 sck
->m_stream
= FALSE
;
302 sck
->m_server
= FALSE
;
303 sck
->m_oriented
= FALSE
;
305 sck
->m_fd
= socket(sck
->m_local
->m_realfamily
, SOCK_DGRAM
, 0);
307 if (bind(sck
->m_fd
, sck
->m_local
->m_addr
, sck
->m_local
->m_len
) < 0) {
310 sck
->m_error
= GSOCK_IOERR
;
314 return GSOCK_NOERROR
;
317 /* Client specific parts */
320 GSocket_Connect() establishes a client connection to a server using the "Peer"
321 field of GSocket. "Peer" must be set by GSocket_SetPeer() before
322 GSocket_Connect() is called. In the other case, it returns GSOCK_INVADDR.
324 GSocketError
GSocket_Connect(GSocket
*sck
, GSocketStream stream
)
330 if (sck
->m_fd
!= -1) {
331 sck
->m_error
= GSOCK_INVSOCK
;
332 return GSOCK_INVSOCK
;
336 sck
->m_error
= GSOCK_INVADDR
;
337 return GSOCK_INVADDR
;
340 sck
->m_stream
= (stream
== GSOCK_STREAMED
);
341 sck
->m_oriented
= TRUE
;
348 sck
->m_fd
= socket(sck
->m_peer
->m_realfamily
, type
, 0);
350 if (sck
->m_fd
== -1) {
351 sck
->m_error
= GSOCK_IOERR
;
355 if (connect(sck
->m_fd
, sck
->m_peer
->m_addr
,
356 sck
->m_peer
->m_len
) != 0) {
359 sck
->m_error
= GSOCK_IOERR
;
363 sck
->m_server
= FALSE
;
365 return GSOCK_NOERROR
;
370 /* Like recv(), send(), ... */
371 int GSocket_Read(GSocket
*socket
, char *buffer
, int size
)
373 assert(socket
!= NULL
);
375 if (socket
->m_fd
== -1 || socket
->m_server
) {
376 socket
->m_error
= GSOCK_INVSOCK
;
380 _GSocket_Enable(socket
, GSOCK_INPUT
);
382 if (socket
->m_oriented
)
383 return _GSocket_Recv_Stream(socket
, buffer
, size
);
385 return _GSocket_Recv_Dgram(socket
, buffer
, size
);
388 int GSocket_Write(GSocket
*socket
, const char *buffer
,
391 assert(socket
!= NULL
);
393 if (socket
->m_fd
== -1 || socket
->m_server
) {
394 socket
->m_error
= GSOCK_INVSOCK
;
398 _GSocket_Enable(socket
, GSOCK_OUTPUT
);
400 if (socket
->m_oriented
)
401 return _GSocket_Send_Stream(socket
, buffer
, size
);
403 return _GSocket_Send_Dgram(socket
, buffer
, size
);
406 bool GSocket_DataAvailable(GSocket
*socket
)
411 assert(socket
!= NULL
);
413 if (socket
->m_fd
== -1 || socket
->m_server
) {
414 socket
->m_error
= GSOCK_INVSOCK
;
419 FD_SET(socket
->m_fd
, &read_set
);
424 select(socket
->m_fd
+1, &read_set
, NULL
, NULL
, &tv
);
426 return FD_ISSET(socket
->m_fd
, &read_set
);
432 GSocket_SetBlocking() puts the socket in non-blocking mode. This is useful
433 if we don't want to wait.
435 void GSocket_SetBlocking(GSocket
*socket
, bool block
)
437 assert(socket
!= NULL
);
439 socket
->m_blocking
= block
;
441 if (socket
->m_fd
!= -1)
442 ioctl(socket
->m_fd
, FIONBIO
, &block
);
446 GSocket_GetError() returns the last error occured on the socket stream.
449 GSocketError
GSocket_GetError(GSocket
*socket
)
451 assert(socket
!= NULL
);
453 return socket
->m_error
;
459 Only one fallback is possible for each event (INPUT, OUTPUT, CONNECTION)
460 INPUT: The function is called when there is at least a byte in the
462 OUTPUT: The function is called when the system is sure the next write call
464 CONNECTION: Two cases is possible:
465 Client socket -> the connection is established
466 Server socket -> a client request a connection
467 LOST: the connection is lost
469 SetFallback accepts a combination of these flags so a same callback can
470 receive different events.
472 An event is generated only once and its state is reseted when the relative
473 IO call is requested.
474 For example: INPUT -> GSocket_Read()
475 CONNECTION -> GSocket_Accept()
477 void GSocket_SetFallback(GSocket
*socket
, GSocketEventFlags event
,
478 GSocketFallback fallback
, char *cdata
)
482 assert (socket
!= NULL
);
484 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
485 if ((event
& (1 << count
)) != 0) {
486 socket
->m_fbacks
[count
] = fallback
;
487 socket
->m_data
[count
] = cdata
;
489 _GSocket_Install_Fallback(socket
, count
);
490 _GSocket_Enable(socket
, count
);
496 UnsetFallback will disables all fallbacks specified by "event".
497 NOTE: event may be a combination of flags
499 void GSocket_UnsetFallback(GSocket
*socket
, GSocketEventFlags event
)
503 assert(socket
!= NULL
);
505 for (count
=0;count
<GSOCK_MAX_EVENT
;count
++) {
506 if ((event
& (1 << count
)) != 0) {
507 _GSocket_Disable(socket
, count
);
508 socket
->m_fbacks
[count
] = NULL
;
509 _GSocket_Uninstall_Fallback(socket
, count
);
514 #define CALL_FALLBACK(socket, event) \
515 if (socket->m_iocalls[event] && \
516 socket->m_fbacks[event]) {\
517 _GSocket_Disable(socket, event); \
518 socket->m_fbacks[event](socket, event, \
519 socket->m_data[event]); \
522 #define MASK_SIGNAL() \
524 void (*old_handler)(int); \
526 old_handler = signal(SIGPIPE, SIG_IGN);
528 #define UNMASK_SIGNAL() \
529 signal(SIGPIPE, old_handler); \
532 void _GSocket_Enable(GSocket
*socket
, GSocketEvent event
)
534 socket
->m_iocalls
[event
] = TRUE
;
535 if (socket
->m_fbacks
[event
])
536 _GSocket_Install_Fallback(socket
, event
);
539 void _GSocket_Disable(GSocket
*socket
, GSocketEvent event
)
541 socket
->m_iocalls
[event
] = FALSE
;
542 if (socket
->m_fbacks
[event
])
543 _GSocket_Uninstall_Fallback(socket
, event
);
546 int _GSocket_Recv_Stream(GSocket
*socket
, char *buffer
, int size
)
551 ret
= recv(socket
->m_fd
, buffer
, size
, 0);
554 socket
->m_error
= GSOCK_IOERR
;
560 int _GSocket_Recv_Dgram(GSocket
*socket
, char *buffer
, int size
)
562 struct sockaddr from
;
565 fromlen
= sizeof(from
);
568 ret
= recvfrom(socket
->m_fd
, buffer
, size
, 0, &from
, &fromlen
);
571 socket
->m_error
= GSOCK_IOERR
;
576 socket
->m_peer
= GAddress_new();
577 _GAddress_translate_from(socket
->m_peer
, &from
, fromlen
);
582 int _GSocket_Send_Stream(GSocket
*socket
, const char *buffer
, int size
)
587 ret
= send(socket
->m_fd
, buffer
, size
, 0);
590 socket
->m_error
= GSOCK_IOERR
;
596 int _GSocket_Send_Dgram(GSocket
*socket
, const char *buffer
, int size
)
598 struct sockaddr
*addr
;
601 if (!socket
->m_peer
) {
602 socket
->m_error
= GSOCK_INVADDR
;
606 _GAddress_translate_to(socket
->m_peer
, &addr
, &len
);
609 ret
= sendto(socket
->m_fd
, buffer
, size
, 0, addr
, len
);
612 socket
->m_error
= GSOCK_IOERR
;
621 void _GSocket_Detected_Read(GSocket
*socket
)
626 if (socket
->m_stream
) {
627 ret
= recv(socket
->m_fd
, &c
, 1, MSG_PEEK
);
629 if (ret
< 0 && socket
->m_server
) {
630 CALL_FALLBACK(socket
, GSOCK_CONNECTION
);
635 CALL_FALLBACK(socket
, GSOCK_INPUT
);
637 CALL_FALLBACK(socket
, GSOCK_LOST
);
642 void _GSocket_Detected_Write(GSocket
*socket
)
644 CALL_FALLBACK(socket
, GSOCK_OUTPUT
);
652 * -------------------------------------------------------------------------
654 * -------------------------------------------------------------------------
657 #define CHECK_ADDRESS(address, family, retval) \
659 if (address->m_family == GSOCK_NOFAMILY) \
660 _GAddress_Init_##family(address); \
661 if (address->m_family != GSOCK_##family) {\
662 address->m_error = GSOCK_INVADDR; \
667 GAddress
*GAddress_new()
671 address
= (GAddress
*)malloc(sizeof(GAddress
));
673 address
->m_family
= GSOCK_NOFAMILY
;
674 address
->m_addr
= NULL
;
680 GAddress
*GAddress_copy(GAddress
*address
)
684 assert(address
!= NULL
);
686 addr2
= (GAddress
*)malloc(sizeof(GAddress
));
687 memcpy(addr2
, address
, sizeof(GAddress
));
689 if (address
->m_addr
) {
690 addr2
->m_addr
= (struct sockaddr
*)malloc(addr2
->m_len
);
691 memcpy(addr2
->m_addr
, address
->m_addr
, addr2
->m_len
);
697 void GAddress_destroy(GAddress
*address
)
699 assert(address
!= NULL
);
704 void GAddress_SetFamily(GAddress
*address
, GAddressType type
)
706 assert(address
!= NULL
);
708 address
->m_family
= type
;
711 GAddressType
GAddress_GetFamily(GAddress
*address
)
713 assert(address
!= NULL
);
715 return address
->m_family
;
718 void _GAddress_translate_from(GAddress
*address
, struct sockaddr
*addr
, int len
){
719 address
->m_realfamily
= addr
->sa_family
;
720 switch (addr
->sa_family
) {
722 address
->m_family
= GSOCK_INET
;
725 address
->m_family
= GSOCK_UNIX
;
729 address
->m_family
= GSOCK_INET6
;
738 free(address
->m_addr
);
740 address
->m_len
= len
;
741 address
->m_addr
= (struct sockaddr
*)malloc(len
);
742 memcpy(address
->m_addr
, addr
, len
);
745 void _GAddress_translate_to(GAddress
*address
,
746 struct sockaddr
**addr
, int *len
)
748 if (!address
->m_addr
) {
753 *len
= address
->m_len
;
754 *addr
= (struct sockaddr
*)malloc(address
->m_len
);
755 memcpy(*addr
, address
->m_addr
, address
->m_len
);
759 * -------------------------------------------------------------------------
760 * Internet address family
761 * -------------------------------------------------------------------------
764 void _GAddress_Init_INET(GAddress
*address
)
766 address
->m_len
= sizeof(struct sockaddr_in
);
767 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
768 address
->m_family
= GSOCK_INET
;
769 address
->m_realfamily
= PF_INET
;
770 ((struct sockaddr_in
*)address
->m_addr
)->sin_family
= AF_INET
;
771 ((struct sockaddr_in
*)address
->m_addr
)->sin_addr
.s_addr
= INADDR_ANY
;
774 GSocketError
GAddress_INET_SetHostName(GAddress
*address
, const char *hostname
)
777 struct in_addr
*addr
;
779 assert(address
!= NULL
);
781 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
783 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
785 /* only name for the moment */
786 if (inet_aton(hostname
, addr
) == 0) {
787 struct in_addr
*array_addr
;
789 if ((he
= gethostbyname(hostname
)) == NULL
) {
790 address
->m_error
= GSOCK_NOHOST
;
793 array_addr
= (struct in_addr
*) *(he
->h_addr_list
);
794 addr
->s_addr
= array_addr
[0].s_addr
;
796 return GSOCK_NOERROR
;
799 GSocketError
GAddress_INET_SetHostAddress(GAddress
*address
,
800 unsigned long hostaddr
)
802 struct in_addr
*addr
;
804 assert(address
!= NULL
);
806 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
808 addr
= &(((struct sockaddr_in
*)address
->m_addr
)->sin_addr
);
809 addr
->s_addr
= hostaddr
;
811 return GSOCK_NOERROR
;
814 GSocketError
GAddress_INET_SetPortName(GAddress
*address
, const char *port
)
817 struct sockaddr_in
*addr
;
819 assert(address
!= NULL
);
820 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
823 address
->m_error
= GSOCK_INVPORT
;
827 /* TODO: TCP or UDP */
828 se
= getservbyname(port
, "tcp");
830 if (isdigit(port
[0])) {
833 port_int
= atoi(port
);
834 addr
= (struct sockaddr_in
*)address
->m_addr
;
835 addr
->sin_port
= htons(port_int
);
836 return GSOCK_NOERROR
;
839 address
->m_error
= GSOCK_INVPORT
;
840 return GSOCK_INVPORT
;
843 addr
= (struct sockaddr_in
*)address
->m_addr
;
844 addr
->sin_port
= se
->s_port
;
846 return GSOCK_NOERROR
;
849 GSocketError
GAddress_INET_SetPort(GAddress
*address
, unsigned short port
)
851 struct sockaddr_in
*addr
;
853 assert(address
!= NULL
);
854 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
856 addr
= (struct sockaddr_in
*)address
->m_addr
;
857 addr
->sin_port
= htons(port
);
859 return GSOCK_NOERROR
;
862 GSocketError
GAddress_INET_GetHostName(GAddress
*address
, char *hostname
, size_t sbuf
)
866 struct sockaddr_in
*addr
;
868 assert(address
!= NULL
);
869 CHECK_ADDRESS(address
, INET
, GSOCK_INVADDR
);
871 addr
= (struct sockaddr_in
*)address
->m_addr
;
872 addr_buf
= (char *)&(addr
->sin_addr
);
874 he
= gethostbyaddr(addr_buf
, sizeof(addr
->sin_addr
), AF_INET
);
876 address
->m_error
= GSOCK_NOHOST
;
880 strncpy(hostname
, he
->h_name
, sbuf
);
882 return GSOCK_NOERROR
;
885 unsigned long GAddress_INET_GetHostAddress(GAddress
*address
)
887 struct sockaddr_in
*addr
;
889 assert(address
!= NULL
);
890 CHECK_ADDRESS(address
, INET
, 0);
892 addr
= (struct sockaddr_in
*)address
->m_addr
;
894 return addr
->sin_addr
.s_addr
;
897 unsigned short GAddress_INET_GetPort(GAddress
*address
)
899 struct sockaddr_in
*addr
;
901 assert(address
!= NULL
);
902 CHECK_ADDRESS(address
, INET
, 0);
904 addr
= (struct sockaddr_in
*)address
->m_addr
;
905 return ntohs(addr
->sin_port
);
909 * -------------------------------------------------------------------------
910 * Unix address family
911 * -------------------------------------------------------------------------
914 void _GAddress_Init_UNIX(GAddress
*address
)
916 address
->m_len
= sizeof(struct sockaddr_un
);
917 address
->m_addr
= (struct sockaddr
*)malloc(address
->m_len
);
918 address
->m_family
= GSOCK_UNIX
;
919 address
->m_realfamily
= PF_UNIX
;
920 ((struct sockaddr_un
*)address
->m_addr
)->sun_family
= AF_UNIX
;
921 ((struct sockaddr_un
*)address
->m_addr
)->sun_path
[0] = 0;
924 GSocketError
GAddress_UNIX_SetPath(GAddress
*address
, const char *path
)
926 struct sockaddr_un
*addr
;
928 assert(address
!= NULL
);
930 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
932 addr
= ((struct sockaddr_un
*)address
->m_addr
);
933 memcpy(addr
->sun_path
, path
, strlen(path
));
935 return GSOCK_NOERROR
;
938 GSocketError
GAddress_UNIX_GetPath(GAddress
*address
, char *path
, size_t sbuf
)
940 struct sockaddr_un
*addr
;
942 assert(address
!= NULL
);
943 CHECK_ADDRESS(address
, UNIX
, GSOCK_INVADDR
);
945 addr
= (struct sockaddr_un
*)address
->m_addr
;
947 strncpy(path
, addr
->sun_path
, sbuf
);
949 return GSOCK_NOERROR
;