]> git.saurik.com Git - wxWidgets.git/blame - src/unix/gsocket.c
Added wxWS_EX_BLOCK_EVENTS style
[wxWidgets.git] / src / unix / gsocket.c
CommitLineData
a324a7bc
GL
1/* -------------------------------------------------------------------------
2 * Project: GSocket (Generic Socket) for WX
3 * Name: gsocket.c
9bf10d6b
GRG
4 * Authors: Guilhem Lavaux,
5 * Guillermo Rodriguez Garcia <guille@iies.es> (maintainer)
7d354cd1 6 * Purpose: GSocket main Unix and OS/2 file
84969af7 7 * Licence: The wxWindows licence
a324a7bc 8 * CVSID: $Id$
a324a7bc
GL
9 * -------------------------------------------------------------------------
10 */
11
9bf10d6b
GRG
12/*
13 * PLEASE don't put C++ comments here - this is a C source file.
14 */
15
16#ifndef __GSOCKET_STANDALONE__
813c20a6 17#include "wx/setup.h"
9bf10d6b 18#endif
813c20a6 19
7d354cd1
SN
20#if defined(__VISAGECPP__)
21/* Seems to be needed by Visual Age C++, though I don't see how it manages
22 to not break on including a C++ header into a plain C source file */
23#include "wx/defs.h"
24#define BSD_SELECT /* use Berkley Sockets select */
25#endif
26
9bf10d6b 27#if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__)
813c20a6 28
a324a7bc 29#include <assert.h>
efee48a0 30#include <sys/types.h>
7d354cd1
SN
31#ifdef __VISAGECPP__
32#include <string.h>
33#include <sys/time.h>
34#include <types.h>
35#include <netinet/in.h>
36#endif
6a2e7225
VZ
37#include <netdb.h>
38#include <sys/ioctl.h>
9bf10d6b 39
af9d1662 40#ifdef __VMS__
a324a7bc 41#include <socket.h>
5c9eff30
GRG
42struct sockaddr_un
43{
9bf10d6b
GRG
44 u_char sun_len; /* sockaddr len including null */
45 u_char sun_family; /* AF_UNIX */
46 char sun_path[108]; /* path name (gag) */
af9d1662 47};
a324a7bc
GL
48#else
49#include <sys/socket.h>
a324a7bc 50#include <sys/un.h>
af9d1662 51#endif
9bf10d6b 52
7d354cd1 53#ifndef __VISAGECPP__
a324a7bc
GL
54#include <sys/time.h>
55#include <netinet/in.h>
56#include <arpa/inet.h>
9592b687 57#include <errno.h>
a324a7bc
GL
58#include <string.h>
59#include <unistd.h>
7d354cd1
SN
60#else
61#include <nerrno.h>
62# if __IBMCPP__ < 400
63#include <machine/endian.h>
64#include <socket.h>
65#include <ioctl.h>
66#include <select.h>
67#include <unistd.h>
68
69#define EBADF SOCEBADF
70
71# ifdef min
72# undef min
73# endif
74# else
75#include <sys/socket.h>
76#include <sys/ioctl.h>
77#include <sys/select.h>
78
79#define close(a) soclose(a)
80#define select(a,b,c,d,e) bsdselect(a,b,c,d,e)
81int _System bsdselect(int,
82 struct fd_set *,
83 struct fd_set *,
84 struct fd_set *,
85 struct timeval *);
86int _System soclose(int);
87# endif
88#endif
a324a7bc
GL
89#include <stdio.h>
90#include <stdlib.h>
552e8bf8 91#include <stddef.h>
9b61f868 92#include <ctype.h>
a324a7bc 93#ifdef sun
9bf10d6b 94# include <sys/filio.h>
a324a7bc 95#endif
a324a7bc 96#ifdef sgi
9bf10d6b 97# include <bstring.h>
a324a7bc 98#endif
a324a7bc 99#include <signal.h>
5a96d2f4 100
efee48a0 101#ifndef SOCKLEN_T
5a96d2f4 102
85806dc2
GRG
103#ifdef VMS
104# define SOCKLEN_T unsigned int
5a96d2f4 105#else
85806dc2
GRG
106# ifdef __GLIBC__
107# if __GLIBC__ == 2
aac67d4c 108# define SOCKLEN_T socklen_t
85806dc2
GRG
109# endif
110# else
111# define SOCKLEN_T int
112# endif
5a96d2f4
GL
113#endif
114
ddb56b38 115#endif /* SOCKLEN_T */
efee48a0 116
aac67d4c
GRG
117/*
118 * MSW defines this, Unices don't.
119 */
120#ifndef INVALID_SOCKET
121#define INVALID_SOCKET -1
122#endif
9bf10d6b 123
a5e3c24d
VZ
124/* UnixWare reportedly needs this for FIONBIO definition */
125#ifdef __UNIXWARE__
126#include <sys/filio.h>
127#endif
128
9bf10d6b
GRG
129/*
130 * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
aac67d4c 131 * on all systems. INADDR_BROADCAST should be fine to indicate an error.
9bf10d6b
GRG
132 */
133#ifndef INADDR_NONE
134#define INADDR_NONE INADDR_BROADCAST
135#endif
136
483249fc
GRG
137#define MASK_SIGNAL() \
138{ \
139 void (*old_handler)(int); \
140 \
aa6d9706
GL
141 old_handler = signal(SIGPIPE, SIG_IGN);
142
483249fc
GRG
143#define UNMASK_SIGNAL() \
144 signal(SIGPIPE, old_handler); \
aa6d9706
GL
145}
146
54e575f9 147
9bf10d6b 148#ifndef __GSOCKET_STANDALONE__
8e907a13
VZ
149# include "wx/unix/gsockunx.h"
150# include "wx/gsocket.h"
9bf10d6b 151#else
8e907a13
VZ
152# include "gsockunx.h"
153# include "gsocket.h"
9bf10d6b
GRG
154#endif /* __GSOCKET_STANDALONE__ */
155
8e907a13
VZ
156/* debugging helpers */
157#ifdef __GSOCKET_DEBUG__
032d5581 158# define GSocket_Debug(args) printf args
8e907a13 159#else
032d5581
GRG
160# define GSocket_Debug(args)
161#endif /* __GSOCKET_DEBUG__ */
8e907a13 162
38bb138f
VS
163/* Table of GUI-related functions. We must call them indirectly because
164 * of wxBase and GUI separation: */
165
166static struct GSocketGUIFunctionsTable *gs_gui_functions;
167
168#define USE_GUI() (gs_gui_functions != NULL)
169
170/* Define macros to simplify indirection: */
171#define _GSocket_GUI_Init() \
172 if (gs_gui_functions) gs_gui_functions->GUI_Init()
173#define _GSocket_GUI_Cleanup() \
174 if (gs_gui_functions) gs_gui_functions->GUI_Cleanup()
175#define _GSocket_GUI_Init_Socket(socket) \
176 (gs_gui_functions ? gs_gui_functions->GUI_Init_Socket(socket) : 1)
177#define _GSocket_GUI_Destroy_Socket(socket) \
178 if (gs_gui_functions) gs_gui_functions->GUI_Destroy_Socket(socket)
179#define _GSocket_Enable_Events(socket) \
180 if (gs_gui_functions) gs_gui_functions->Enable_Events(socket)
181#define _GSocket_Disable_Events(socket) \
182 if (gs_gui_functions) gs_gui_functions->Disable_Events(socket)
183#define _GSocket_Install_Callback(socket, event) \
184 if (gs_gui_functions) gs_gui_functions->Install_Callback(socket, event)
185#define _GSocket_Uninstall_Callback(socket, event) \
186 if (gs_gui_functions) gs_gui_functions->Uninstall_Callback(socket, event)
187
188static struct GSocketBaseFunctionsTable gs_base_functions =
189{
190 _GSocket_Detected_Read,
191 _GSocket_Detected_Write
192};
193
a58d5df4
GL
194/* Global initialisers */
195
38bb138f
VS
196void GSocket_SetGUIFunctions(struct GSocketGUIFunctionsTable *guifunc)
197{
198 gs_gui_functions = guifunc;
199}
200
5c9eff30 201int GSocket_Init(void)
a58d5df4 202{
38bb138f
VS
203 if (gs_gui_functions)
204 {
205 if ( !gs_gui_functions->GUI_Init() )
206 return 0;
207 }
5c9eff30 208 return 1;
a58d5df4
GL
209}
210
da051b23 211void GSocket_Cleanup(void)
a58d5df4 212{
38bb138f
VS
213 if (gs_gui_functions)
214 {
215 gs_gui_functions->GUI_Cleanup();
216 }
a58d5df4
GL
217}
218
9bf10d6b 219/* Constructors / Destructors for GSocket */
a324a7bc 220
da051b23 221GSocket *GSocket_new(void)
a324a7bc 222{
8e907a13 223 int i, success;
a324a7bc
GL
224 GSocket *socket;
225
226 socket = (GSocket *)malloc(sizeof(GSocket));
227
e00f35bb
GL
228 if (socket == NULL)
229 return NULL;
230
aac67d4c 231 socket->m_fd = INVALID_SOCKET;
483249fc
GRG
232 for (i=0;i<GSOCK_MAX_EVENT;i++)
233 {
98781fa3 234 socket->m_cbacks[i] = NULL;
a324a7bc 235 }
483249fc 236 socket->m_detected = 0;
a324a7bc
GL
237 socket->m_local = NULL;
238 socket->m_peer = NULL;
239 socket->m_error = GSOCK_NOERROR;
ef25e638
GRG
240 socket->m_server = FALSE;
241 socket->m_stream = TRUE;
242 socket->m_gui_dependent = NULL;
243 socket->m_non_blocking = FALSE;
39b91eca 244 socket->m_timeout = 10*60*1000;
ef25e638 245 /* 10 minutes * 60 sec * 1000 millisec */
aa6d9706 246 socket->m_establishing = FALSE;
a324a7bc 247
38bb138f
VS
248 socket->m_functions = &gs_base_functions;
249
da051b23 250 /* Per-socket GUI-specific initialization */
38bb138f 251 success = _GSocket_GUI_Init_Socket(socket);
8e907a13 252 if (!success)
da051b23
GRG
253 {
254 free(socket);
255 return NULL;
256 }
a324a7bc
GL
257
258 return socket;
259}
260
007c77ab
RL
261void GSocket_close(GSocket *socket)
262{
263 _GSocket_Disable_Events(socket);
264 close(socket->m_fd);
265 socket->m_fd = INVALID_SOCKET;
266}
267
a324a7bc
GL
268void GSocket_destroy(GSocket *socket)
269{
270 assert(socket != NULL);
271
1c7edd1d
DE
272 /* When using CFSocket we MUST invalidate before closing the fd */
273#ifdef __DARWIN__
274 /* Per-socket GUI-specific cleanup */
275 _GSocket_GUI_Destroy_Socket(socket);
276#endif
277
9bf10d6b 278 /* Check that the socket is really shutdowned */
aac67d4c 279 if (socket->m_fd != INVALID_SOCKET)
a324a7bc
GL
280 GSocket_Shutdown(socket);
281
1c7edd1d 282#ifndef __DARWIN__
f6f13cd4 283 /* Per-socket GUI-specific cleanup */
38bb138f 284 _GSocket_GUI_Destroy_Socket(socket);
1c7edd1d 285#endif
f6f13cd4 286
9bf10d6b 287 /* Destroy private addresses */
a324a7bc
GL
288 if (socket->m_local)
289 GAddress_destroy(socket->m_local);
290
291 if (socket->m_peer)
292 GAddress_destroy(socket->m_peer);
293
9bf10d6b 294 /* Destroy the socket itself */
a324a7bc
GL
295 free(socket);
296}
297
9bf10d6b
GRG
298/* GSocket_Shutdown:
299 * Disallow further read/write operations on this socket, close
300 * the fd and disable all callbacks.
301 */
a324a7bc
GL
302void GSocket_Shutdown(GSocket *socket)
303{
304 int evt;
305
306 assert(socket != NULL);
307
9bf10d6b 308 /* If socket has been created, shutdown it */
aac67d4c 309 if (socket->m_fd != INVALID_SOCKET)
483249fc 310 {
533c7161 311 shutdown(socket->m_fd, 2);
007c77ab 312 GSocket_close(socket);
a324a7bc
GL
313 }
314
9bf10d6b 315 /* Disable GUI callbacks */
483249fc
GRG
316 for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
317 socket->m_cbacks[evt] = NULL;
318
557e7011 319 socket->m_detected = GSOCK_LOST_FLAG;
a324a7bc
GL
320}
321
322/* Address handling */
323
9bf10d6b
GRG
324/* GSocket_SetLocal:
325 * GSocket_GetLocal:
326 * GSocket_SetPeer:
327 * GSocket_GetPeer:
328 * Set or get the local or peer address for this socket. The 'set'
329 * functions return GSOCK_NOERROR on success, an error code otherwise.
330 * The 'get' functions return a pointer to a GAddress object on success,
331 * or NULL otherwise, in which case they set the error code of the
332 * corresponding GSocket.
333 *
334 * Error codes:
335 * GSOCK_INVSOCK - the socket is not valid.
336 * GSOCK_INVADDR - the address is not valid.
337 */
a324a7bc
GL
338GSocketError GSocket_SetLocal(GSocket *socket, GAddress *address)
339{
5a96d2f4
GL
340 assert(socket != NULL);
341
9bf10d6b 342 /* the socket must be initialized, or it must be a server */
aac67d4c 343 if ((socket->m_fd != INVALID_SOCKET && !socket->m_server))
9bf10d6b 344 {
31989b0b 345 socket->m_error = GSOCK_INVSOCK;
a324a7bc 346 return GSOCK_INVSOCK;
31989b0b 347 }
a324a7bc 348
9bf10d6b
GRG
349 /* check address */
350 if (address == NULL || address->m_family == GSOCK_NOFAMILY)
351 {
31989b0b 352 socket->m_error = GSOCK_INVADDR;
a324a7bc 353 return GSOCK_INVADDR;
31989b0b 354 }
a324a7bc
GL
355
356 if (socket->m_local)
357 GAddress_destroy(socket->m_local);
358
359 socket->m_local = GAddress_copy(address);
360
361 return GSOCK_NOERROR;
362}
363
364GSocketError GSocket_SetPeer(GSocket *socket, GAddress *address)
365{
5a96d2f4 366 assert(socket != NULL);
a324a7bc 367
9bf10d6b
GRG
368 /* check address */
369 if (address == NULL || address->m_family == GSOCK_NOFAMILY)
370 {
a324a7bc
GL
371 socket->m_error = GSOCK_INVADDR;
372 return GSOCK_INVADDR;
373 }
374
375 if (socket->m_peer)
376 GAddress_destroy(socket->m_peer);
377
378 socket->m_peer = GAddress_copy(address);
379
380 return GSOCK_NOERROR;
381}
382
383GAddress *GSocket_GetLocal(GSocket *socket)
384{
385 GAddress *address;
386 struct sockaddr addr;
85806dc2 387 SOCKLEN_T size = sizeof(addr);
aa6d9706 388 GSocketError err;
a324a7bc
GL
389
390 assert(socket != NULL);
391
9bf10d6b 392 /* try to get it from the m_local var first */
a324a7bc
GL
393 if (socket->m_local)
394 return GAddress_copy(socket->m_local);
395
9bf10d6b 396 /* else, if the socket is initialized, try getsockname */
aac67d4c 397 if (socket->m_fd == INVALID_SOCKET)
9bf10d6b 398 {
a324a7bc
GL
399 socket->m_error = GSOCK_INVSOCK;
400 return NULL;
401 }
402
9bf10d6b
GRG
403 if (getsockname(socket->m_fd, &addr, (SOCKLEN_T *) &size) < 0)
404 {
a324a7bc
GL
405 socket->m_error = GSOCK_IOERR;
406 return NULL;
407 }
408
9bf10d6b 409 /* got a valid address from getsockname, create a GAddress object */
a324a7bc 410 address = GAddress_new();
9bf10d6b
GRG
411 if (address == NULL)
412 {
e00f35bb
GL
413 socket->m_error = GSOCK_MEMERR;
414 return NULL;
415 }
9bf10d6b
GRG
416
417 err = _GAddress_translate_from(address, &addr, size);
418 if (err != GSOCK_NOERROR)
419 {
e00f35bb 420 GAddress_destroy(address);
85806dc2 421 socket->m_error = err;
e00f35bb
GL
422 return NULL;
423 }
a324a7bc
GL
424
425 return address;
426}
427
428GAddress *GSocket_GetPeer(GSocket *socket)
429{
430 assert(socket != NULL);
431
9bf10d6b 432 /* try to get it from the m_peer var */
a324a7bc
GL
433 if (socket->m_peer)
434 return GAddress_copy(socket->m_peer);
435
436 return NULL;
437}
438
439/* Server specific parts */
440
2f7c2af5 441/* GSocket_SetServer:
9bf10d6b
GRG
442 * Sets up this socket as a server. The local address must have been
443 * set with GSocket_SetLocal() before GSocket_SetServer() is called.
444 * Returns GSOCK_NOERROR on success, one of the following otherwise:
445 *
446 * Error codes:
447 * GSOCK_INVSOCK - the socket is in use.
448 * GSOCK_INVADDR - the local address has not been set.
449 * GSOCK_IOERR - low-level error.
2f7c2af5 450 */
a324a7bc
GL
451GSocketError GSocket_SetServer(GSocket *sck)
452{
483249fc 453 int arg = 1;
a324a7bc
GL
454
455 assert(sck != NULL);
456
9bf10d6b 457 /* must not be in use */
aac67d4c 458 if (sck->m_fd != INVALID_SOCKET)
9bf10d6b 459 {
a324a7bc
GL
460 sck->m_error = GSOCK_INVSOCK;
461 return GSOCK_INVSOCK;
462 }
463
9bf10d6b
GRG
464 /* the local addr must have been set */
465 if (!sck->m_local)
466 {
a324a7bc
GL
467 sck->m_error = GSOCK_INVADDR;
468 return GSOCK_INVADDR;
469 }
470
9bf10d6b
GRG
471 /* Initialize all fields */
472 sck->m_stream = TRUE;
473 sck->m_server = TRUE;
474 sck->m_oriented = TRUE;
a324a7bc 475
f61815af
GL
476 /* Create the socket */
477 sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_STREAM, 0);
a324a7bc 478
aac67d4c 479 if (sck->m_fd == INVALID_SOCKET)
9bf10d6b 480 {
a324a7bc
GL
481 sck->m_error = GSOCK_IOERR;
482 return GSOCK_IOERR;
483 }
7d354cd1
SN
484#if defined(__EMX__) || defined(__VISAGECPP__)
485 ioctl(sck->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
486#else
483249fc 487 ioctl(sck->m_fd, FIONBIO, &arg);
7d354cd1 488#endif
483249fc
GRG
489 _GSocket_Enable_Events(sck);
490
9bf10d6b
GRG
491 /* Bind to the local address,
492 * retrieve the actual address bound,
493 * and listen up to 5 connections.
494 */
495 if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
496 (getsockname(sck->m_fd,
497 sck->m_local->m_addr,
498 (SOCKLEN_T *) &sck->m_local->m_len) != 0) ||
499 (listen(sck->m_fd, 5) != 0))
500 {
007c77ab 501 GSocket_close(sck);
a324a7bc
GL
502 sck->m_error = GSOCK_IOERR;
503 return GSOCK_IOERR;
504 }
505
a324a7bc 506 return GSOCK_NOERROR;
a324a7bc
GL
507}
508
2f7c2af5 509/* GSocket_WaitConnection:
9bf10d6b
GRG
510 * Waits for an incoming client connection. Returns a pointer to
511 * a GSocket object, or NULL if there was an error, in which case
512 * the last error field will be updated for the calling GSocket.
513 *
514 * Error codes (set in the calling GSocket)
515 * GSOCK_INVSOCK - the socket is not valid or not a server.
516 * GSOCK_TIMEDOUT - timeout, no incoming connections.
517 * GSOCK_WOULDBLOCK - the call would block and the socket is nonblocking.
518 * GSOCK_MEMERR - couldn't allocate memory.
519 * GSOCK_IOERR - low-level error.
2f7c2af5 520 */
a324a7bc
GL
521GSocket *GSocket_WaitConnection(GSocket *socket)
522{
85806dc2
GRG
523 struct sockaddr from;
524 SOCKLEN_T fromlen = sizeof(from);
a324a7bc 525 GSocket *connection;
bd046d42 526 GSocketError err;
483249fc 527 int arg = 1;
a324a7bc
GL
528
529 assert(socket != NULL);
530
483249fc
GRG
531 /* Reenable CONNECTION events */
532 _GSocket_Enable(socket, GSOCK_CONNECTION);
533
f61815af 534 /* If the socket has already been created, we exit immediately */
aac67d4c 535 if (socket->m_fd == INVALID_SOCKET || !socket->m_server)
483249fc 536 {
a324a7bc
GL
537 socket->m_error = GSOCK_INVSOCK;
538 return NULL;
539 }
540
f61815af 541 /* Create a GSocket object for the new connection */
a324a7bc 542 connection = GSocket_new();
9bf10d6b 543
483249fc
GRG
544 if (!connection)
545 {
85806dc2 546 socket->m_error = GSOCK_MEMERR;
aa6d9706
GL
547 return NULL;
548 }
a324a7bc 549
9bf10d6b 550 /* Wait for a connection (with timeout) */
483249fc
GRG
551 if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
552 {
553 GSocket_destroy(connection);
554 /* socket->m_error set by _GSocket_Input_Timeout */
555 return NULL;
556 }
557
85806dc2 558 connection->m_fd = accept(socket->m_fd, &from, (SOCKLEN_T *) &fromlen);
483249fc 559
aac67d4c 560 if (connection->m_fd == INVALID_SOCKET)
483249fc
GRG
561 {
562 if (errno == EWOULDBLOCK)
563 socket->m_error = GSOCK_WOULDBLOCK;
564 else
565 socket->m_error = GSOCK_IOERR;
566
a324a7bc 567 GSocket_destroy(connection);
a324a7bc
GL
568 return NULL;
569 }
570
f61815af 571 /* Initialize all fields */
a324a7bc 572 connection->m_server = FALSE;
483249fc 573 connection->m_stream = TRUE;
a324a7bc
GL
574 connection->m_oriented = TRUE;
575
9bf10d6b 576 /* Setup the peer address field */
85806dc2
GRG
577 connection->m_peer = GAddress_new();
578 if (!connection->m_peer)
579 {
580 GSocket_destroy(connection);
581 socket->m_error = GSOCK_MEMERR;
582 return NULL;
583 }
584 err = _GAddress_translate_from(connection->m_peer, &from, fromlen);
585 if (err != GSOCK_NOERROR)
586 {
587 GAddress_destroy(connection->m_peer);
588 GSocket_destroy(connection);
589 socket->m_error = err;
590 return NULL;
591 }
7d354cd1
SN
592#if defined(__EMX__) || defined(__VISAGECPP__)
593 ioctl(connection->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
594#else
483249fc 595 ioctl(connection->m_fd, FIONBIO, &arg);
7d354cd1 596#endif
483249fc 597 _GSocket_Enable_Events(connection);
2f7c2af5 598
a324a7bc
GL
599 return connection;
600}
601
a324a7bc
GL
602/* Client specific parts */
603
2f7c2af5 604/* GSocket_Connect:
9bf10d6b
GRG
605 * For stream (connection oriented) sockets, GSocket_Connect() tries
606 * to establish a client connection to a server using the peer address
607 * as established with GSocket_SetPeer(). Returns GSOCK_NOERROR if the
608 * connection has been succesfully established, or one of the error
609 * codes listed below. Note that for nonblocking sockets, a return
610 * value of GSOCK_WOULDBLOCK doesn't mean a failure. The connection
611 * request can be completed later; you should use GSocket_Select()
612 * to poll for GSOCK_CONNECTION | GSOCK_LOST, or wait for the
613 * corresponding asynchronous events.
614 *
615 * For datagram (non connection oriented) sockets, GSocket_Connect()
616 * just sets the peer address established with GSocket_SetPeer() as
617 * default destination.
618 *
619 * Error codes:
620 * GSOCK_INVSOCK - the socket is in use or not valid.
621 * GSOCK_INVADDR - the peer address has not been established.
622 * GSOCK_TIMEDOUT - timeout, the connection failed.
623 * GSOCK_WOULDBLOCK - connection in progress (nonblocking sockets only)
624 * GSOCK_MEMERR - couldn't allocate memory.
625 * GSOCK_IOERR - low-level error.
2f7c2af5 626 */
a324a7bc
GL
627GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
628{
9bf10d6b 629 int err, ret;
483249fc 630 int arg = 1;
a324a7bc
GL
631
632 assert(sck != NULL);
633
483249fc
GRG
634 /* Enable CONNECTION events (needed for nonblocking connections) */
635 _GSocket_Enable(sck, GSOCK_CONNECTION);
636
aac67d4c 637 if (sck->m_fd != INVALID_SOCKET)
483249fc 638 {
a324a7bc
GL
639 sck->m_error = GSOCK_INVSOCK;
640 return GSOCK_INVSOCK;
641 }
642
483249fc
GRG
643 if (!sck->m_peer)
644 {
a324a7bc
GL
645 sck->m_error = GSOCK_INVADDR;
646 return GSOCK_INVADDR;
647 }
648
9bf10d6b 649 /* Streamed or dgram socket? */
483249fc 650 sck->m_stream = (stream == GSOCK_STREAMED);
a324a7bc 651 sck->m_oriented = TRUE;
483249fc
GRG
652 sck->m_server = FALSE;
653 sck->m_establishing = FALSE;
a324a7bc 654
f61815af 655 /* Create the socket */
9bf10d6b
GRG
656 sck->m_fd = socket(sck->m_peer->m_realfamily,
657 sck->m_stream? SOCK_STREAM : SOCK_DGRAM, 0);
a324a7bc 658
aac67d4c 659 if (sck->m_fd == INVALID_SOCKET)
9bf10d6b 660 {
a324a7bc
GL
661 sck->m_error = GSOCK_IOERR;
662 return GSOCK_IOERR;
663 }
7d354cd1
SN
664#if defined(__EMX__) || defined(__VISAGECPP__)
665 ioctl(sck->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
666#else
483249fc 667 ioctl(sck->m_fd, FIONBIO, &arg);
7d354cd1 668#endif
483249fc 669 _GSocket_Enable_Events(sck);
aa6d9706 670
9bf10d6b 671 /* Connect it to the peer address, with a timeout (see below) */
483249fc
GRG
672 ret = connect(sck->m_fd, sck->m_peer->m_addr, sck->m_peer->m_len);
673
674 if (ret == -1)
675 {
676 err = errno;
677
678 /* If connect failed with EINPROGRESS and the GSocket object
679 * is in blocking mode, we select() for the specified timeout
680 * checking for writability to see if the connection request
681 * completes.
682 */
683 if ((err == EINPROGRESS) && (!sck->m_non_blocking))
684 {
685 if (_GSocket_Output_Timeout(sck) == GSOCK_TIMEDOUT)
686 {
007c77ab 687 GSocket_close(sck);
483249fc 688 /* sck->m_error is set in _GSocket_Output_Timeout */
483249fc
GRG
689 return GSOCK_TIMEDOUT;
690 }
691 else
692 {
9bf10d6b
GRG
693 int error;
694 SOCKLEN_T len = sizeof(error);
695
696 getsockopt(sck->m_fd, SOL_SOCKET, SO_ERROR, (void*) &error, &len);
697
698 if (!error)
699 return GSOCK_NOERROR;
483249fc
GRG
700 }
701 }
702
703 /* If connect failed with EINPROGRESS and the GSocket object
704 * is set to nonblocking, we set m_error to GSOCK_WOULDBLOCK
705 * (and return GSOCK_WOULDBLOCK) but we don't close the socket;
706 * this way if the connection completes, a GSOCK_CONNECTION
707 * event will be generated, if enabled.
708 */
709 if ((err == EINPROGRESS) && (sck->m_non_blocking))
710 {
483249fc 711 sck->m_establishing = TRUE;
9bf10d6b 712 sck->m_error = GSOCK_WOULDBLOCK;
483249fc
GRG
713 return GSOCK_WOULDBLOCK;
714 }
aa6d9706 715
483249fc
GRG
716 /* If connect failed with an error other than EINPROGRESS,
717 * then the call to GSocket_Connect has failed.
718 */
007c77ab 719 GSocket_close(sck);
aac67d4c
GRG
720 sck->m_error = GSOCK_IOERR;
721 return GSOCK_IOERR;
722 }
723
724 return GSOCK_NOERROR;
725}
726
727/* Datagram sockets */
728
729/* GSocket_SetNonOriented:
730 * Sets up this socket as a non-connection oriented (datagram) socket.
731 * Before using this function, the local address must have been set
732 * with GSocket_SetLocal(), or the call will fail. Returns GSOCK_NOERROR
733 * on success, or one of the following otherwise.
734 *
735 * Error codes:
736 * GSOCK_INVSOCK - the socket is in use.
737 * GSOCK_INVADDR - the local address has not been set.
738 * GSOCK_IOERR - low-level error.
739 */
740GSocketError GSocket_SetNonOriented(GSocket *sck)
741{
742 int arg = 1;
743
744 assert(sck != NULL);
745
746 if (sck->m_fd != INVALID_SOCKET)
747 {
748 sck->m_error = GSOCK_INVSOCK;
749 return GSOCK_INVSOCK;
750 }
751
752 if (!sck->m_local)
753 {
754 sck->m_error = GSOCK_INVADDR;
755 return GSOCK_INVADDR;
756 }
757
758 /* Initialize all fields */
759 sck->m_stream = FALSE;
760 sck->m_server = FALSE;
761 sck->m_oriented = FALSE;
762
763 /* Create the socket */
764 sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_DGRAM, 0);
765
766 if (sck->m_fd == INVALID_SOCKET)
767 {
768 sck->m_error = GSOCK_IOERR;
769 return GSOCK_IOERR;
770 }
7d354cd1
SN
771#if defined(__EMX__) || defined(__VISAGECPP__)
772 ioctl(sck->m_fd, FIONBIO, (char*)&arg, sizeof(arg));
773#else
aac67d4c 774 ioctl(sck->m_fd, FIONBIO, &arg);
7d354cd1 775#endif
aac67d4c
GRG
776 _GSocket_Enable_Events(sck);
777
778 /* Bind to the local address,
779 * and retrieve the actual address bound.
780 */
781 if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
782 (getsockname(sck->m_fd,
783 sck->m_local->m_addr,
784 (SOCKLEN_T *) &sck->m_local->m_len) != 0))
785 {
007c77ab 786 GSocket_close(sck);
483249fc 787 sck->m_error = GSOCK_IOERR;
a324a7bc
GL
788 return GSOCK_IOERR;
789 }
39b91eca 790
483249fc 791 return GSOCK_NOERROR;
a324a7bc
GL
792}
793
794/* Generic IO */
795
796/* Like recv(), send(), ... */
797int GSocket_Read(GSocket *socket, char *buffer, int size)
798{
483249fc
GRG
799 int ret;
800
a324a7bc
GL
801 assert(socket != NULL);
802
1c7edd1d
DE
803 /* When using CFSocket we MUST NOT reenable events until we finish reading */
804#ifndef __DARWIN__
483249fc
GRG
805 /* Reenable INPUT events */
806 _GSocket_Enable(socket, GSOCK_INPUT);
1c7edd1d 807#endif
483249fc 808
aac67d4c 809 if (socket->m_fd == INVALID_SOCKET || socket->m_server)
483249fc 810 {
a324a7bc
GL
811 socket->m_error = GSOCK_INVSOCK;
812 return -1;
813 }
814
9bf10d6b 815 /* If the socket is blocking, wait for data (with a timeout) */
483249fc
GRG
816 if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
817 return -1;
a324a7bc 818
9bf10d6b 819 /* Read the data */
f61815af 820 if (socket->m_stream)
483249fc 821 ret = _GSocket_Recv_Stream(socket, buffer, size);
a324a7bc 822 else
483249fc
GRG
823 ret = _GSocket_Recv_Dgram(socket, buffer, size);
824
825 if (ret == -1)
826 {
827 if (errno == EWOULDBLOCK)
828 socket->m_error = GSOCK_WOULDBLOCK;
829 else
830 socket->m_error = GSOCK_IOERR;
831 }
832
1c7edd1d
DE
833#ifdef __DARWIN__
834 /* Reenable INPUT events */
835 _GSocket_Enable(socket, GSOCK_INPUT);
836#endif
837
483249fc 838 return ret;
a324a7bc
GL
839}
840
9bf10d6b
GRG
841int GSocket_Write(GSocket *socket, const char *buffer, int size)
842{
483249fc
GRG
843 int ret;
844
a324a7bc 845 assert(socket != NULL);
856bc99a 846
8e907a13 847 GSocket_Debug(( "GSocket_Write #1, size %d\n", size ));
a324a7bc 848
aac67d4c 849 if (socket->m_fd == INVALID_SOCKET || socket->m_server)
483249fc 850 {
a324a7bc
GL
851 socket->m_error = GSOCK_INVSOCK;
852 return -1;
853 }
854
8e907a13 855 GSocket_Debug(( "GSocket_Write #2, size %d\n", size ));
856bc99a 856
9bf10d6b 857 /* If the socket is blocking, wait for writability (with a timeout) */
483249fc
GRG
858 if (_GSocket_Output_Timeout(socket) == GSOCK_TIMEDOUT)
859 return -1;
a324a7bc 860
8e907a13 861 GSocket_Debug(( "GSocket_Write #3, size %d\n", size ));
856bc99a 862
9bf10d6b 863 /* Write the data */
f61815af 864 if (socket->m_stream)
483249fc 865 ret = _GSocket_Send_Stream(socket, buffer, size);
a324a7bc 866 else
483249fc
GRG
867 ret = _GSocket_Send_Dgram(socket, buffer, size);
868
8e907a13 869 GSocket_Debug(( "GSocket_Write #4, size %d\n", size ));
856bc99a 870
483249fc
GRG
871 if (ret == -1)
872 {
873 if (errno == EWOULDBLOCK)
d5ab387d 874 {
483249fc 875 socket->m_error = GSOCK_WOULDBLOCK;
8e907a13 876 GSocket_Debug(( "GSocket_Write error WOULDBLOCK\n" ));
d5ab387d 877 }
483249fc 878 else
d5ab387d 879 {
483249fc 880 socket->m_error = GSOCK_IOERR;
8e907a13 881 GSocket_Debug(( "GSocket_Write error IOERR\n" ));
d5ab387d 882 }
483249fc
GRG
883
884 /* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
885 * in MSW). Once the first OUTPUT event is received, users can assume
886 * that the socket is writable until a read operation fails. Only then
887 * will further OUTPUT events be posted.
888 */
889 _GSocket_Enable(socket, GSOCK_OUTPUT);
9bf10d6b 890 return -1;
483249fc
GRG
891 }
892
8e907a13 893 GSocket_Debug(( "GSocket_Write #5, size %d ret %d\n", size, ret ));
856bc99a 894
483249fc 895 return ret;
a324a7bc
GL
896}
897
ef25e638
GRG
898/* GSocket_Select:
899 * Polls the socket to determine its status. This function will
900 * check for the events specified in the 'flags' parameter, and
901 * it will return a mask indicating which operations can be
902 * performed. This function won't block, regardless of the
9bf10d6b 903 * mode (blocking | nonblocking) of the socket.
ef25e638
GRG
904 */
905GSocketEventFlags GSocket_Select(GSocket *socket, GSocketEventFlags flags)
a324a7bc 906{
38bb138f
VS
907 if (!USE_GUI())
908 {
8e907a13 909
38bb138f
VS
910 GSocketEventFlags result = 0;
911 fd_set readfds;
912 fd_set writefds;
913 fd_set exceptfds;
914 struct timeval tv;
a324a7bc 915
38bb138f
VS
916 /* Do not use a static struct, Linux can garble it */
917 tv.tv_sec = 0;
918 tv.tv_usec = 0;
8e907a13 919
38bb138f 920 assert(socket != NULL);
aa6ca3dc 921
38bb138f
VS
922 FD_ZERO(&readfds);
923 FD_ZERO(&writefds);
924 FD_ZERO(&exceptfds);
925 FD_SET(socket->m_fd, &readfds);
926 FD_SET(socket->m_fd, &writefds);
927 FD_SET(socket->m_fd, &exceptfds);
aa6ca3dc 928
38bb138f
VS
929 /* Check 'sticky' CONNECTION flag first */
930 result |= (GSOCK_CONNECTION_FLAG & socket->m_detected);
8e907a13 931
38bb138f
VS
932 /* If we have already detected a LOST event, then don't try
933 * to do any further processing.
934 */
935 if ((socket->m_detected & GSOCK_LOST_FLAG) != 0)
936 {
937 socket->m_establishing = FALSE;
8e907a13 938
38bb138f
VS
939 return (GSOCK_LOST_FLAG & flags);
940 }
8e907a13 941
38bb138f
VS
942 /* Try select now */
943 if (select(socket->m_fd + 1, &readfds, &writefds, &exceptfds, &tv) <= 0)
8e907a13 944 {
38bb138f
VS
945 /* What to do here? */
946 return (result & flags);
8e907a13 947 }
38bb138f
VS
948
949 /* Check for readability */
950 if (FD_ISSET(socket->m_fd, &readfds))
8e907a13 951 {
38bb138f
VS
952 char c;
953
954 if (recv(socket->m_fd, &c, 1, MSG_PEEK) > 0)
8e907a13 955 {
38bb138f 956 result |= GSOCK_INPUT_FLAG;
8e907a13
VZ
957 }
958 else
959 {
38bb138f
VS
960 if (socket->m_server && socket->m_stream)
961 {
962 result |= GSOCK_CONNECTION_FLAG;
963 socket->m_detected |= GSOCK_CONNECTION_FLAG;
964 }
965 else
966 {
967 socket->m_detected = GSOCK_LOST_FLAG;
968 socket->m_establishing = FALSE;
969
970 /* LOST event: Abort any further processing */
971 return (GSOCK_LOST_FLAG & flags);
972 }
8e907a13
VZ
973 }
974 }
8e907a13 975
38bb138f
VS
976 /* Check for writability */
977 if (FD_ISSET(socket->m_fd, &writefds))
8e907a13 978 {
38bb138f
VS
979 if (socket->m_establishing && !socket->m_server)
980 {
981 int error;
982 SOCKLEN_T len = sizeof(error);
8e907a13 983
38bb138f 984 socket->m_establishing = FALSE;
8e907a13 985
38bb138f 986 getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, (void*)&error, &len);
8e907a13 987
38bb138f
VS
988 if (error)
989 {
990 socket->m_detected = GSOCK_LOST_FLAG;
aa6ca3dc 991
38bb138f
VS
992 /* LOST event: Abort any further processing */
993 return (GSOCK_LOST_FLAG & flags);
994 }
995 else
996 {
997 result |= GSOCK_CONNECTION_FLAG;
998 socket->m_detected |= GSOCK_CONNECTION_FLAG;
999 }
8e907a13
VZ
1000 }
1001 else
1002 {
38bb138f 1003 result |= GSOCK_OUTPUT_FLAG;
8e907a13
VZ
1004 }
1005 }
38bb138f
VS
1006
1007 /* Check for exceptions and errors (is this useful in Unices?) */
1008 if (FD_ISSET(socket->m_fd, &exceptfds))
8e907a13 1009 {
38bb138f
VS
1010 socket->m_establishing = FALSE;
1011 socket->m_detected = GSOCK_LOST_FLAG;
1012
1013 /* LOST event: Abort any further processing */
1014 return (GSOCK_LOST_FLAG & flags);
8e907a13 1015 }
8e907a13 1016
38bb138f 1017 return (result & flags);
aa6ca3dc 1018
8e907a13 1019 }
38bb138f
VS
1020 else
1021 {
8e907a13 1022
38bb138f
VS
1023 assert(socket != NULL);
1024 return flags & socket->m_detected;
8e907a13 1025
38bb138f 1026 }
a324a7bc
GL
1027}
1028
1029/* Flags */
1030
2f7c2af5 1031/* GSocket_SetNonBlocking:
9bf10d6b
GRG
1032 * Sets the socket to non-blocking mode. All IO calls will return
1033 * immediately.
2f7c2af5 1034 */
5c9eff30 1035void GSocket_SetNonBlocking(GSocket *socket, int non_block)
a324a7bc
GL
1036{
1037 assert(socket != NULL);
1038
609c60af
GRG
1039 GSocket_Debug( ("GSocket_SetNonBlocking: %d\n", (int)non_block) );
1040
dc26b37a 1041 socket->m_non_blocking = non_block;
a324a7bc
GL
1042}
1043
2f7c2af5 1044/* GSocket_SetTimeout:
9bf10d6b
GRG
1045 * Sets the timeout for blocking calls. Time is expressed in
1046 * milliseconds.
39b91eca
GL
1047 */
1048void GSocket_SetTimeout(GSocket *socket, unsigned long millisec)
1049{
1050 assert(socket != NULL);
1051
54e575f9 1052 socket->m_timeout = millisec;
39b91eca
GL
1053}
1054
2f7c2af5 1055/* GSocket_GetError:
9bf10d6b
GRG
1056 * Returns the last error occured for this socket. Note that successful
1057 * operations do not clear this back to GSOCK_NOERROR, so use it only
1058 * after an error.
2f7c2af5 1059 */
a324a7bc
GL
1060GSocketError GSocket_GetError(GSocket *socket)
1061{
1062 assert(socket != NULL);
1063
1064 return socket->m_error;
1065}
1066
1067/* Callbacks */
1068
9bf10d6b
GRG
1069/* GSOCK_INPUT:
1070 * There is data to be read in the input buffer. If, after a read
1071 * operation, there is still data available, the callback function will
1072 * be called again.
1073 * GSOCK_OUTPUT:
1074 * The socket is available for writing. That is, the next write call
1075 * won't block. This event is generated only once, when the connection is
1076 * first established, and then only if a call failed with GSOCK_WOULDBLOCK,
1077 * when the output buffer empties again. This means that the app should
1078 * assume that it can write since the first OUTPUT event, and no more
1079 * OUTPUT events will be generated unless an error occurs.
1080 * GSOCK_CONNECTION:
1081 * Connection succesfully established, for client sockets, or incoming
1082 * client connection, for server sockets. Wait for this event (also watch
1083 * out for GSOCK_LOST) after you issue a nonblocking GSocket_Connect() call.
1084 * GSOCK_LOST:
1085 * The connection is lost (or a connection request failed); this could
1086 * be due to a failure, or due to the peer closing it gracefully.
2f7c2af5
GRG
1087 */
1088
1089/* GSocket_SetCallback:
1090 * Enables the callbacks specified by 'flags'. Note that 'flags'
1091 * may be a combination of flags OR'ed toghether, so the same
1092 * callback function can be made to accept different events.
1093 * The callback function must have the following prototype:
1094 *
1095 * void function(GSocket *socket, GSocketEvent event, char *cdata)
1096 */
ef25e638 1097void GSocket_SetCallback(GSocket *socket, GSocketEventFlags flags,
9bf10d6b 1098 GSocketCallback callback, char *cdata)
a324a7bc
GL
1099{
1100 int count;
1101
483249fc 1102 assert(socket != NULL);
a324a7bc 1103
483249fc
GRG
1104 for (count = 0; count < GSOCK_MAX_EVENT; count++)
1105 {
1106 if ((flags & (1 << count)) != 0)
1107 {
98781fa3 1108 socket->m_cbacks[count] = callback;
a324a7bc 1109 socket->m_data[count] = cdata;
a324a7bc
GL
1110 }
1111 }
1112}
1113
2f7c2af5
GRG
1114/* GSocket_UnsetCallback:
1115 * Disables all callbacks specified by 'flags', which may be a
1116 * combination of flags OR'ed toghether.
1117 */
ef25e638 1118void GSocket_UnsetCallback(GSocket *socket, GSocketEventFlags flags)
a324a7bc 1119{
2f7c2af5 1120 int count;
a324a7bc
GL
1121
1122 assert(socket != NULL);
1123
483249fc
GRG
1124 for (count = 0; count < GSOCK_MAX_EVENT; count++)
1125 {
1126 if ((flags & (1 << count)) != 0)
1127 {
98781fa3 1128 socket->m_cbacks[count] = NULL;
483249fc 1129 socket->m_data[count] = NULL;
a324a7bc
GL
1130 }
1131 }
1132}
1133
9bf10d6b 1134
483249fc
GRG
1135#define CALL_CALLBACK(socket, event) { \
1136 _GSocket_Disable(socket, event); \
1137 if (socket->m_cbacks[event]) \
1138 socket->m_cbacks[event](socket, event, socket->m_data[event]); \
a324a7bc
GL
1139}
1140
483249fc 1141
a324a7bc
GL
1142void _GSocket_Enable(GSocket *socket, GSocketEvent event)
1143{
483249fc
GRG
1144 socket->m_detected &= ~(1 << event);
1145 _GSocket_Install_Callback(socket, event);
a324a7bc
GL
1146}
1147
1148void _GSocket_Disable(GSocket *socket, GSocketEvent event)
1149{
483249fc
GRG
1150 socket->m_detected |= (1 << event);
1151 _GSocket_Uninstall_Callback(socket, event);
a324a7bc
GL
1152}
1153
483249fc
GRG
1154/* _GSocket_Input_Timeout:
1155 * For blocking sockets, wait until data is available or
1156 * until timeout ellapses.
1157 */
1158GSocketError _GSocket_Input_Timeout(GSocket *socket)
2f7c2af5 1159{
483249fc
GRG
1160 struct timeval tv;
1161 fd_set readfds;
a8a0b892 1162 int ret;
483249fc
GRG
1163
1164 /* Linux select() will overwrite the struct on return */
1165 tv.tv_sec = (socket->m_timeout / 1000);
1166 tv.tv_usec = (socket->m_timeout % 1000) * 1000;
1167
1168 if (!socket->m_non_blocking)
1169 {
1170 FD_ZERO(&readfds);
1171 FD_SET(socket->m_fd, &readfds);
a8a0b892
RR
1172 ret = select(socket->m_fd + 1, &readfds, NULL, NULL, &tv);
1173 if (ret == 0)
1174 {
8e907a13 1175 GSocket_Debug(( "GSocket_Input_Timeout, select returned 0\n" ));
a8a0b892
RR
1176 socket->m_error = GSOCK_TIMEDOUT;
1177 return GSOCK_TIMEDOUT;
1178 }
1179 if (ret == -1)
483249fc 1180 {
8e907a13 1181 GSocket_Debug(( "GSocket_Input_Timeout, select returned -1\n" ));
d84afea9
GD
1182 if (errno == EBADF) { GSocket_Debug(( "Invalid file descriptor\n" )); }
1183 if (errno == EINTR) { GSocket_Debug(( "A non blocked signal was caught\n" )); }
1184 if (errno == EINVAL) { GSocket_Debug(( "The highest number descriptor is negative\n" )); }
1185 if (errno == ENOMEM) { GSocket_Debug(( "Not enough memory\n" )); }
483249fc
GRG
1186 socket->m_error = GSOCK_TIMEDOUT;
1187 return GSOCK_TIMEDOUT;
1188 }
1189 }
1190 return GSOCK_NOERROR;
1191}
1192
1193/* _GSocket_Output_Timeout:
1194 * For blocking sockets, wait until data can be sent without
1195 * blocking or until timeout ellapses.
1196 */
1197GSocketError _GSocket_Output_Timeout(GSocket *socket)
1198{
1199 struct timeval tv;
1200 fd_set writefds;
a8a0b892 1201 int ret;
483249fc
GRG
1202
1203 /* Linux select() will overwrite the struct on return */
1204 tv.tv_sec = (socket->m_timeout / 1000);
1205 tv.tv_usec = (socket->m_timeout % 1000) * 1000;
1206
609c60af
GRG
1207 GSocket_Debug( ("m_non_blocking has: %d\n", (int)socket->m_non_blocking) );
1208
483249fc
GRG
1209 if (!socket->m_non_blocking)
1210 {
1211 FD_ZERO(&writefds);
1212 FD_SET(socket->m_fd, &writefds);
a8a0b892
RR
1213 ret = select(socket->m_fd + 1, NULL, &writefds, NULL, &tv);
1214 if (ret == 0)
1215 {
8e907a13 1216 GSocket_Debug(( "GSocket_Output_Timeout, select returned 0\n" ));
a8a0b892
RR
1217 socket->m_error = GSOCK_TIMEDOUT;
1218 return GSOCK_TIMEDOUT;
1219 }
1220 if (ret == -1)
483249fc 1221 {
8e907a13 1222 GSocket_Debug(( "GSocket_Output_Timeout, select returned -1\n" ));
d84afea9
GD
1223 if (errno == EBADF) { GSocket_Debug(( "Invalid file descriptor\n" )); }
1224 if (errno == EINTR) { GSocket_Debug(( "A non blocked signal was caught\n" )); }
1225 if (errno == EINVAL) { GSocket_Debug(( "The highest number descriptor is negative\n" )); }
1226 if (errno == ENOMEM) { GSocket_Debug(( "Not enough memory\n" )); }
483249fc
GRG
1227 socket->m_error = GSOCK_TIMEDOUT;
1228 return GSOCK_TIMEDOUT;
2f7c2af5 1229 }
d84afea9
GD
1230 if ( ! FD_ISSET(socket->m_fd, &writefds) ) {
1231 GSocket_Debug(( "GSocket_Output_Timeout is buggy!\n" ));
1232 }
1233 else {
1234 GSocket_Debug(( "GSocket_Output_Timeout seems correct\n" ));
1235 }
2f7c2af5 1236 }
aac67d4c
GRG
1237 else
1238 {
8e3aec85 1239 GSocket_Debug(( "GSocket_Output_Timeout, didn't try select!\n" ));
aac67d4c
GRG
1240 }
1241
483249fc 1242 return GSOCK_NOERROR;
2f7c2af5
GRG
1243}
1244
a324a7bc
GL
1245int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
1246{
04e1eb03 1247 return recv(socket->m_fd, buffer, size, 0);
a324a7bc
GL
1248}
1249
1250int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size)
1251{
1252 struct sockaddr from;
85806dc2 1253 SOCKLEN_T fromlen = sizeof(from);
ca17eff3 1254 int ret;
aa6d9706 1255 GSocketError err;
a324a7bc
GL
1256
1257 fromlen = sizeof(from);
1258
85806dc2 1259 ret = recvfrom(socket->m_fd, buffer, size, 0, &from, (SOCKLEN_T *) &fromlen);
483249fc
GRG
1260
1261 if (ret == -1)
aa6d9706 1262 return -1;
a324a7bc 1263
f61815af 1264 /* Translate a system address into a GSocket address */
483249fc
GRG
1265 if (!socket->m_peer)
1266 {
a324a7bc 1267 socket->m_peer = GAddress_new();
483249fc
GRG
1268 if (!socket->m_peer)
1269 {
e00f35bb
GL
1270 socket->m_error = GSOCK_MEMERR;
1271 return -1;
1272 }
1273 }
aa6d9706 1274 err = _GAddress_translate_from(socket->m_peer, &from, fromlen);
483249fc
GRG
1275 if (err != GSOCK_NOERROR)
1276 {
aa6d9706
GL
1277 GAddress_destroy(socket->m_peer);
1278 socket->m_peer = NULL;
1279 socket->m_error = err;
e00f35bb 1280 return -1;
aa6d9706 1281 }
a324a7bc
GL
1282
1283 return ret;
1284}
1285
1286int _GSocket_Send_Stream(GSocket *socket, const char *buffer, int size)
1287{
1288 int ret;
1289
7d354cd1 1290#ifndef __VISAGECPP__
a324a7bc
GL
1291 MASK_SIGNAL();
1292 ret = send(socket->m_fd, buffer, size, 0);
1293 UNMASK_SIGNAL();
7d354cd1
SN
1294#else
1295 ret = send(socket->m_fd, (char *)buffer, size, 0);
1296#endif
483249fc 1297
a324a7bc
GL
1298 return ret;
1299}
1300
1301int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size)
1302{
1303 struct sockaddr *addr;
1304 int len, ret;
aa6d9706 1305 GSocketError err;
a324a7bc 1306
9bf10d6b
GRG
1307 if (!socket->m_peer)
1308 {
a324a7bc
GL
1309 socket->m_error = GSOCK_INVADDR;
1310 return -1;
1311 }
1312
aa6d9706 1313 err = _GAddress_translate_to(socket->m_peer, &addr, &len);
9bf10d6b
GRG
1314 if (err != GSOCK_NOERROR)
1315 {
aa6d9706 1316 socket->m_error = err;
e00f35bb
GL
1317 return -1;
1318 }
a324a7bc 1319
7d354cd1 1320#ifndef __VISAGECPP__
a324a7bc
GL
1321 MASK_SIGNAL();
1322 ret = sendto(socket->m_fd, buffer, size, 0, addr, len);
1323 UNMASK_SIGNAL();
7d354cd1
SN
1324#else
1325 ret = sendto(socket->m_fd, (char *)buffer, size, 0, addr, len);
1326#endif
a324a7bc 1327
f61815af 1328 /* Frees memory allocated from _GAddress_translate_to */
a324a7bc
GL
1329 free(addr);
1330
1331 return ret;
1332}
1333
1334void _GSocket_Detected_Read(GSocket *socket)
1335{
1336 char c;
9bf10d6b 1337
aa6ca3dc
GRG
1338 /* If we have already detected a LOST event, then don't try
1339 * to do any further processing.
1340 */
1341 if ((socket->m_detected & GSOCK_LOST_FLAG) != 0)
1342 {
1343 socket->m_establishing = FALSE;
1344
1345 CALL_CALLBACK(socket, GSOCK_LOST);
1346 GSocket_Shutdown(socket);
1347 return;
1348 }
1349
8e907a13 1350 if (recv(socket->m_fd, &c, 1, MSG_PEEK) > 0)
9bf10d6b
GRG
1351 {
1352 CALL_CALLBACK(socket, GSOCK_INPUT);
1353 }
1354 else
1355 {
8e907a13
VZ
1356 if (socket->m_server && socket->m_stream)
1357 {
1358 CALL_CALLBACK(socket, GSOCK_CONNECTION);
1359 }
1360 else
1361 {
1362 CALL_CALLBACK(socket, GSOCK_LOST);
aa6ca3dc 1363 GSocket_Shutdown(socket);
8e907a13 1364 }
a324a7bc
GL
1365 }
1366}
1367
1368void _GSocket_Detected_Write(GSocket *socket)
1369{
aa6ca3dc
GRG
1370 /* If we have already detected a LOST event, then don't try
1371 * to do any further processing.
1372 */
1373 if ((socket->m_detected & GSOCK_LOST_FLAG) != 0)
1374 {
1375 socket->m_establishing = FALSE;
1376
1377 CALL_CALLBACK(socket, GSOCK_LOST);
1378 GSocket_Shutdown(socket);
1379 return;
1380 }
1381
483249fc
GRG
1382 if (socket->m_establishing && !socket->m_server)
1383 {
9bf10d6b
GRG
1384 int error;
1385 SOCKLEN_T len = sizeof(error);
aa6d9706 1386
aa6d9706 1387 socket->m_establishing = FALSE;
aa6d9706 1388
8e907a13 1389 getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, (void*)&error, &len);
aa6d9706 1390
483249fc
GRG
1391 if (error)
1392 {
1393 CALL_CALLBACK(socket, GSOCK_LOST);
aa6ca3dc 1394 GSocket_Shutdown(socket);
483249fc
GRG
1395 }
1396 else
1397 {
1398 CALL_CALLBACK(socket, GSOCK_CONNECTION);
1399 /* We have to fire this event by hand because CONNECTION (for clients)
1400 * and OUTPUT are internally the same and we just disabled CONNECTION
1401 * events with the above macro.
1402 */
1403 CALL_CALLBACK(socket, GSOCK_OUTPUT);
1404 }
1405 }
1406 else
1407 {
aa6d9706 1408 CALL_CALLBACK(socket, GSOCK_OUTPUT);
483249fc 1409 }
aa6d9706 1410}
a324a7bc
GL
1411
1412/*
1413 * -------------------------------------------------------------------------
1414 * GAddress
1415 * -------------------------------------------------------------------------
1416 */
1417
032d5581
GRG
1418/* CHECK_ADDRESS verifies that the current address family is either
1419 * GSOCK_NOFAMILY or GSOCK_*family*, and if it is GSOCK_NOFAMILY, it
1420 * initalizes it to be a GSOCK_*family*. In other cases, it returns
1421 * an appropiate error code.
1422 *
1423 * CHECK_ADDRESS_RETVAL does the same but returning 'retval' on error.
f61815af 1424 */
032d5581 1425#define CHECK_ADDRESS(address, family) \
9bf10d6b
GRG
1426{ \
1427 if (address->m_family == GSOCK_NOFAMILY) \
1428 if (_GAddress_Init_##family(address) != GSOCK_NOERROR) \
1429 return address->m_error; \
1430 if (address->m_family != GSOCK_##family) \
032d5581
GRG
1431 { \
1432 address->m_error = GSOCK_INVADDR; \
1433 return GSOCK_INVADDR; \
1434 } \
1435}
1436
1437#define CHECK_ADDRESS_RETVAL(address, family, retval) \
1438{ \
1439 if (address->m_family == GSOCK_NOFAMILY) \
1440 if (_GAddress_Init_##family(address) != GSOCK_NOERROR) \
1441 return retval; \
1442 if (address->m_family != GSOCK_##family) \
9bf10d6b
GRG
1443 { \
1444 address->m_error = GSOCK_INVADDR; \
1445 return retval; \
1446 } \
a324a7bc
GL
1447}
1448
032d5581 1449
da051b23 1450GAddress *GAddress_new(void)
a324a7bc
GL
1451{
1452 GAddress *address;
1453
9bf10d6b 1454 if ((address = (GAddress *) malloc(sizeof(GAddress))) == NULL)
e00f35bb
GL
1455 return NULL;
1456
a324a7bc
GL
1457 address->m_family = GSOCK_NOFAMILY;
1458 address->m_addr = NULL;
1459 address->m_len = 0;
1460
1461 return address;
1462}
1463
1464GAddress *GAddress_copy(GAddress *address)
1465{
1466 GAddress *addr2;
1467
1468 assert(address != NULL);
1469
9bf10d6b 1470 if ((addr2 = (GAddress *) malloc(sizeof(GAddress))) == NULL)
e00f35bb
GL
1471 return NULL;
1472
a324a7bc
GL
1473 memcpy(addr2, address, sizeof(GAddress));
1474
007c77ab 1475 if (address->m_addr && address->m_len > 0)
9bf10d6b 1476 {
a324a7bc 1477 addr2->m_addr = (struct sockaddr *)malloc(addr2->m_len);
9bf10d6b
GRG
1478 if (addr2->m_addr == NULL)
1479 {
e00f35bb
GL
1480 free(addr2);
1481 return NULL;
1482 }
a324a7bc
GL
1483 memcpy(addr2->m_addr, address->m_addr, addr2->m_len);
1484 }
1485
1486 return addr2;
1487}
1488
1489void GAddress_destroy(GAddress *address)
1490{
1491 assert(address != NULL);
1492
29c25a8e
GRG
1493 if (address->m_addr)
1494 free(address->m_addr);
1495
a324a7bc
GL
1496 free(address);
1497}
1498
1499void GAddress_SetFamily(GAddress *address, GAddressType type)
1500{
1501 assert(address != NULL);
1502
1503 address->m_family = type;
1504}
1505
1506GAddressType GAddress_GetFamily(GAddress *address)
1507{
1508 assert(address != NULL);
1509
1510 return address->m_family;
1511}
1512
9bf10d6b
GRG
1513GSocketError _GAddress_translate_from(GAddress *address,
1514 struct sockaddr *addr, int len)
1515{
a324a7bc 1516 address->m_realfamily = addr->sa_family;
9bf10d6b
GRG
1517 switch (addr->sa_family)
1518 {
1519 case AF_INET:
1520 address->m_family = GSOCK_INET;
1521 break;
1522 case AF_UNIX:
1523 address->m_family = GSOCK_UNIX;
1524 break;
efee48a0 1525#ifdef AF_INET6
9bf10d6b
GRG
1526 case AF_INET6:
1527 address->m_family = GSOCK_INET6;
1528 break;
efee48a0 1529#endif
9bf10d6b 1530 default:
ca17eff3 1531 {
9bf10d6b
GRG
1532 address->m_error = GSOCK_INVOP;
1533 return GSOCK_INVOP;
ca17eff3 1534 }
a324a7bc
GL
1535 }
1536
1537 if (address->m_addr)
1538 free(address->m_addr);
1539
1540 address->m_len = len;
1541 address->m_addr = (struct sockaddr *)malloc(len);
9bf10d6b
GRG
1542
1543 if (address->m_addr == NULL)
1544 {
54e575f9
GL
1545 address->m_error = GSOCK_MEMERR;
1546 return GSOCK_MEMERR;
1547 }
a324a7bc 1548 memcpy(address->m_addr, addr, len);
e00f35bb 1549
54e575f9 1550 return GSOCK_NOERROR;
a324a7bc
GL
1551}
1552
54e575f9
GL
1553GSocketError _GAddress_translate_to(GAddress *address,
1554 struct sockaddr **addr, int *len)
a324a7bc 1555{
9bf10d6b
GRG
1556 if (!address->m_addr)
1557 {
54e575f9
GL
1558 address->m_error = GSOCK_INVADDR;
1559 return GSOCK_INVADDR;
a324a7bc
GL
1560 }
1561
1562 *len = address->m_len;
1563 *addr = (struct sockaddr *)malloc(address->m_len);
032d5581
GRG
1564 if (*addr == NULL)
1565 {
54e575f9
GL
1566 address->m_error = GSOCK_MEMERR;
1567 return GSOCK_MEMERR;
1568 }
e00f35bb 1569
a324a7bc 1570 memcpy(*addr, address->m_addr, address->m_len);
54e575f9 1571 return GSOCK_NOERROR;
a324a7bc
GL
1572}
1573
1574/*
1575 * -------------------------------------------------------------------------
1576 * Internet address family
1577 * -------------------------------------------------------------------------
1578 */
1579
54e575f9 1580GSocketError _GAddress_Init_INET(GAddress *address)
a324a7bc 1581{
f3e60952 1582 address->m_len = sizeof(struct sockaddr_in);
9bf10d6b 1583 address->m_addr = (struct sockaddr *) malloc(address->m_len);
f3e60952
GRG
1584 if (address->m_addr == NULL)
1585 {
54e575f9
GL
1586 address->m_error = GSOCK_MEMERR;
1587 return GSOCK_MEMERR;
1588 }
e00f35bb 1589
a324a7bc
GL
1590 address->m_family = GSOCK_INET;
1591 address->m_realfamily = PF_INET;
1592 ((struct sockaddr_in *)address->m_addr)->sin_family = AF_INET;
9bf10d6b 1593 ((struct sockaddr_in *)address->m_addr)->sin_addr.s_addr = INADDR_ANY;
e00f35bb 1594
cc345f78 1595 return GSOCK_NOERROR;
a324a7bc
GL
1596}
1597
1598GSocketError GAddress_INET_SetHostName(GAddress *address, const char *hostname)
1599{
1600 struct hostent *he;
1601 struct in_addr *addr;
1602
1603 assert(address != NULL);
1604
032d5581 1605 CHECK_ADDRESS(address, INET);
a324a7bc
GL
1606
1607 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1608
f61815af 1609 /* If it is a numeric host name, convert it now */
d5b08e7b 1610#if defined(HAVE_INET_ATON)
f3e60952
GRG
1611 if (inet_aton(hostname, addr) == 0)
1612 {
d5b08e7b 1613#elif defined(HAVE_INET_ADDR)
f3e60952
GRG
1614 if ( (addr->s_addr = inet_addr(hostname)) == -1 )
1615 {
e96ac54e
GL
1616#else
1617 /* Use gethostbyname by default */
7d354cd1
SN
1618 int val = 1; //VA doesn't like constants in conditional expressions at all
1619 if (val)
f3e60952 1620 {
e96ac54e 1621#endif
a324a7bc
GL
1622 struct in_addr *array_addr;
1623
f61815af 1624 /* It is a real name, we solve it */
f3e60952
GRG
1625 if ((he = gethostbyname(hostname)) == NULL)
1626 {
9bf10d6b
GRG
1627 /* Reset to invalid address */
1628 addr->s_addr = INADDR_NONE;
a324a7bc
GL
1629 address->m_error = GSOCK_NOHOST;
1630 return GSOCK_NOHOST;
1631 }
1632 array_addr = (struct in_addr *) *(he->h_addr_list);
1633 addr->s_addr = array_addr[0].s_addr;
1634 }
1635 return GSOCK_NOERROR;
1636}
1637
9bf10d6b
GRG
1638GSocketError GAddress_INET_SetAnyAddress(GAddress *address)
1639{
1640 return GAddress_INET_SetHostAddress(address, INADDR_ANY);
1641}
1642
a324a7bc
GL
1643GSocketError GAddress_INET_SetHostAddress(GAddress *address,
1644 unsigned long hostaddr)
1645{
1646 struct in_addr *addr;
1647
1648 assert(address != NULL);
1649
032d5581 1650 CHECK_ADDRESS(address, INET);
a324a7bc
GL
1651
1652 addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1653 addr->s_addr = hostaddr;
1654
1655 return GSOCK_NOERROR;
1656}
1657
5a96d2f4
GL
1658GSocketError GAddress_INET_SetPortName(GAddress *address, const char *port,
1659 const char *protocol)
a324a7bc
GL
1660{
1661 struct servent *se;
1662 struct sockaddr_in *addr;
1663
1664 assert(address != NULL);
032d5581 1665 CHECK_ADDRESS(address, INET);
a324a7bc 1666
f3e60952
GRG
1667 if (!port)
1668 {
a324a7bc 1669 address->m_error = GSOCK_INVPORT;
54e575f9 1670 return GSOCK_INVPORT;
a324a7bc
GL
1671 }
1672
5a96d2f4 1673 se = getservbyname(port, protocol);
9bf10d6b
GRG
1674 if (!se)
1675 {
72c71d39
VZ
1676 /* the cast to int suppresses compiler warnings about subscript having the
1677 type char */
1678 if (isdigit((int)port[0]))
f3e60952 1679 {
a324a7bc
GL
1680 int port_int;
1681
1682 port_int = atoi(port);
1683 addr = (struct sockaddr_in *)address->m_addr;
1684 addr->sin_port = htons(port_int);
1685 return GSOCK_NOERROR;
1686 }
1687
1688 address->m_error = GSOCK_INVPORT;
1689 return GSOCK_INVPORT;
1690 }
1691
1692 addr = (struct sockaddr_in *)address->m_addr;
1693 addr->sin_port = se->s_port;
1694
1695 return GSOCK_NOERROR;
1696}
1697
1698GSocketError GAddress_INET_SetPort(GAddress *address, unsigned short port)
1699{
1700 struct sockaddr_in *addr;
1701
1702 assert(address != NULL);
032d5581 1703 CHECK_ADDRESS(address, INET);
a324a7bc
GL
1704
1705 addr = (struct sockaddr_in *)address->m_addr;
1706 addr->sin_port = htons(port);
1707
1708 return GSOCK_NOERROR;
1709}
1710
1711GSocketError GAddress_INET_GetHostName(GAddress *address, char *hostname, size_t sbuf)
1712{
1713 struct hostent *he;
1714 char *addr_buf;
1715 struct sockaddr_in *addr;
1716
1717 assert(address != NULL);
032d5581 1718 CHECK_ADDRESS(address, INET);
a324a7bc
GL
1719
1720 addr = (struct sockaddr_in *)address->m_addr;
1721 addr_buf = (char *)&(addr->sin_addr);
1722
f3e60952
GRG
1723 he = gethostbyaddr(addr_buf, sizeof(addr->sin_addr), AF_INET);
1724 if (he == NULL)
1725 {
a324a7bc
GL
1726 address->m_error = GSOCK_NOHOST;
1727 return GSOCK_NOHOST;
1728 }
1729
1730 strncpy(hostname, he->h_name, sbuf);
1731
1732 return GSOCK_NOERROR;
1733}
1734
1735unsigned long GAddress_INET_GetHostAddress(GAddress *address)
1736{
1737 struct sockaddr_in *addr;
1738
1739 assert(address != NULL);
032d5581 1740 CHECK_ADDRESS_RETVAL(address, INET, 0);
a324a7bc
GL
1741
1742 addr = (struct sockaddr_in *)address->m_addr;
1743
1744 return addr->sin_addr.s_addr;
1745}
1746
1747unsigned short GAddress_INET_GetPort(GAddress *address)
1748{
1749 struct sockaddr_in *addr;
1750
1751 assert(address != NULL);
032d5581 1752 CHECK_ADDRESS_RETVAL(address, INET, 0);
a324a7bc
GL
1753
1754 addr = (struct sockaddr_in *)address->m_addr;
1755 return ntohs(addr->sin_port);
1756}
1757
1758/*
1759 * -------------------------------------------------------------------------
1760 * Unix address family
1761 * -------------------------------------------------------------------------
1762 */
1763
7d354cd1 1764#ifndef __VISAGECPP__
54e575f9 1765GSocketError _GAddress_Init_UNIX(GAddress *address)
a324a7bc 1766{
f3e60952 1767 address->m_len = sizeof(struct sockaddr_un);
a324a7bc 1768 address->m_addr = (struct sockaddr *)malloc(address->m_len);
f3e60952
GRG
1769 if (address->m_addr == NULL)
1770 {
54e575f9
GL
1771 address->m_error = GSOCK_MEMERR;
1772 return GSOCK_MEMERR;
1773 }
e00f35bb 1774
a324a7bc
GL
1775 address->m_family = GSOCK_UNIX;
1776 address->m_realfamily = PF_UNIX;
1777 ((struct sockaddr_un *)address->m_addr)->sun_family = AF_UNIX;
1778 ((struct sockaddr_un *)address->m_addr)->sun_path[0] = 0;
e00f35bb 1779
032d5581 1780 return GSOCK_NOERROR;
a324a7bc
GL
1781}
1782
70914290
VZ
1783#define UNIX_SOCK_PATHLEN (sizeof(addr->sun_path)/sizeof(addr->sun_path[0]))
1784
a324a7bc
GL
1785GSocketError GAddress_UNIX_SetPath(GAddress *address, const char *path)
1786{
1787 struct sockaddr_un *addr;
1788
1789 assert(address != NULL);
1790
032d5581 1791 CHECK_ADDRESS(address, UNIX);
a324a7bc
GL
1792
1793 addr = ((struct sockaddr_un *)address->m_addr);
70914290
VZ
1794 strncpy(addr->sun_path, path, UNIX_SOCK_PATHLEN);
1795 addr->sun_path[UNIX_SOCK_PATHLEN - 1] = '\0';
a324a7bc
GL
1796
1797 return GSOCK_NOERROR;
1798}
1799
1800GSocketError GAddress_UNIX_GetPath(GAddress *address, char *path, size_t sbuf)
1801{
1802 struct sockaddr_un *addr;
1803
1804 assert(address != NULL);
032d5581 1805 CHECK_ADDRESS(address, UNIX);
a324a7bc
GL
1806
1807 addr = (struct sockaddr_un *)address->m_addr;
1808
1809 strncpy(path, addr->sun_path, sbuf);
1810
1811 return GSOCK_NOERROR;
1812}
7d354cd1 1813#endif /* !defined(__VISAGECPP__) */
9bf10d6b 1814#endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */
85806dc2 1815