bugfixes
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@3772 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
This commit is contained in:
@@ -36,6 +36,7 @@ struct _GSocket
|
|||||||
GAddress *m_peer;
|
GAddress *m_peer;
|
||||||
GSocketError m_error;
|
GSocketError m_error;
|
||||||
|
|
||||||
|
/* Attributes */
|
||||||
bool m_non_blocking;
|
bool m_non_blocking;
|
||||||
bool m_server;
|
bool m_server;
|
||||||
bool m_stream;
|
bool m_stream;
|
||||||
@@ -43,6 +44,7 @@ struct _GSocket
|
|||||||
struct timeval m_timeout;
|
struct timeval m_timeout;
|
||||||
|
|
||||||
/* Callbacks */
|
/* Callbacks */
|
||||||
|
GSocketEventFlags m_detected;
|
||||||
GSocketCallback m_cbacks[GSOCK_MAX_EVENT];
|
GSocketCallback m_cbacks[GSOCK_MAX_EVENT];
|
||||||
char *m_data[GSOCK_MAX_EVENT];
|
char *m_data[GSOCK_MAX_EVENT];
|
||||||
int m_msgnumber;
|
int m_msgnumber;
|
||||||
@@ -72,7 +74,8 @@ int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size);
|
|||||||
|
|
||||||
/* Callbacks */
|
/* Callbacks */
|
||||||
|
|
||||||
void _GSocket_Configure_Callbacks(GSocket *socket);
|
void _GSocket_Enable_Events(GSocket *socket);
|
||||||
|
void _GSocket_Disable_Events(GSocket *socket);
|
||||||
LRESULT CALLBACK _GSocket_Internal_WinProc(HWND, UINT, WPARAM, LPARAM);
|
LRESULT CALLBACK _GSocket_Internal_WinProc(HWND, UINT, WPARAM, LPARAM);
|
||||||
|
|
||||||
/* GAddress */
|
/* GAddress */
|
||||||
|
@@ -23,6 +23,7 @@
|
|||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
// wxSocket headers (generic)
|
// wxSocket headers (generic)
|
||||||
// ---------------------------------------------------------------------------
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
#ifdef WXPREC
|
#ifdef WXPREC
|
||||||
# include "wx/wxprec.h"
|
# include "wx/wxprec.h"
|
||||||
#else
|
#else
|
||||||
@@ -63,9 +64,19 @@ typedef enum {
|
|||||||
wxSOCKET_INVPORT = GSOCK_INVPORT,
|
wxSOCKET_INVPORT = GSOCK_INVPORT,
|
||||||
wxSOCKET_WOULDBLOCK = GSOCK_WOULDBLOCK,
|
wxSOCKET_WOULDBLOCK = GSOCK_WOULDBLOCK,
|
||||||
wxSOCKET_TIMEDOUT = GSOCK_TIMEDOUT,
|
wxSOCKET_TIMEDOUT = GSOCK_TIMEDOUT,
|
||||||
wxSOCKET_MEMERR = GSOCK_MEMERR
|
wxSOCKET_MEMERR = GSOCK_MEMERR,
|
||||||
|
wxSOCKET_BUSY
|
||||||
} wxSocketError;
|
} wxSocketError;
|
||||||
|
|
||||||
|
enum {
|
||||||
|
wxSOCKET_NONE = 0,
|
||||||
|
wxSOCKET_NOWAIT = 1,
|
||||||
|
wxSOCKET_WAITALL = 2,
|
||||||
|
wxSOCKET_BLOCK = 4
|
||||||
|
};
|
||||||
|
|
||||||
|
typedef int wxSockFlags;
|
||||||
|
|
||||||
// ------------------------------------------------------------------------
|
// ------------------------------------------------------------------------
|
||||||
// wxSocket base
|
// wxSocket base
|
||||||
// ------------------------------------------------------------------------
|
// ------------------------------------------------------------------------
|
||||||
@@ -77,43 +88,64 @@ class WXDLLEXPORT wxSocketBase : public wxEvtHandler
|
|||||||
DECLARE_CLASS(wxSocketBase)
|
DECLARE_CLASS(wxSocketBase)
|
||||||
public:
|
public:
|
||||||
|
|
||||||
enum { NONE=0, NOWAIT=1, WAITALL=2, SPEED=4 };
|
enum {
|
||||||
typedef int wxSockFlags;
|
NONE = wxSOCKET_NONE,
|
||||||
|
NOWAIT = wxSOCKET_NOWAIT,
|
||||||
|
WAITALL = wxSOCKET_WAITALL,
|
||||||
|
SPEED = wxSOCKET_BLOCK
|
||||||
|
};
|
||||||
|
|
||||||
|
typedef ::wxSockFlags wxSockFlags;
|
||||||
|
|
||||||
// Type of request
|
// Type of request
|
||||||
|
|
||||||
enum wxSockType { SOCK_CLIENT, SOCK_SERVER, SOCK_INTERNAL, SOCK_UNINIT };
|
enum wxSockType { SOCK_CLIENT, SOCK_SERVER, SOCK_INTERNAL, SOCK_UNINIT };
|
||||||
typedef void (*wxSockCbk)(wxSocketBase& sock, wxSocketNotify evt, char *cdata);
|
typedef void (*wxSockCbk)(wxSocketBase& sock, wxSocketNotify evt, char *cdata);
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
GSocket *m_socket; // wxSocket socket
|
GSocket *m_socket; // GSocket
|
||||||
|
int m_id; // Socket id (for event handler)
|
||||||
|
|
||||||
|
// Attributes
|
||||||
wxSockFlags m_flags; // wxSocket flags
|
wxSockFlags m_flags; // wxSocket flags
|
||||||
wxSockType m_type; // wxSocket type
|
wxSockType m_type; // wxSocket type
|
||||||
wxSocketEventFlags m_neededreq; // Which events we are interested in
|
wxSocketEventFlags m_neededreq; // Event mask
|
||||||
wxUint32 m_lcount; // Last IO request size
|
bool m_notify_state; // Notify events to users?
|
||||||
|
bool m_connected; // Connected ?
|
||||||
|
bool m_establishing; // Establishing connection ?
|
||||||
|
bool m_reading; // Busy reading?
|
||||||
|
bool m_writing; // Busy writing?
|
||||||
|
bool m_error; // Did last IO call fail ?
|
||||||
|
wxUint32 m_lcount; // Last IO transaction size
|
||||||
unsigned long m_timeout; // IO timeout value
|
unsigned long m_timeout; // IO timeout value
|
||||||
|
wxList m_states; // Stack of states
|
||||||
|
|
||||||
char *m_unread; // Pushback buffer
|
char *m_unread; // Pushback buffer
|
||||||
wxUint32 m_unrd_size; // Pushback buffer size
|
wxUint32 m_unrd_size; // Pushback buffer size
|
||||||
wxUint32 m_unrd_cur; // Pushback pointer
|
wxUint32 m_unrd_cur; // Pushback pointer (index into buffer)
|
||||||
|
|
||||||
wxSockCbk m_cbk; // C callback
|
// Async IO variables
|
||||||
char *m_cdata; // C callback data
|
enum
|
||||||
|
{
|
||||||
bool m_connected; // Connected ?
|
NO_DEFER = 0,
|
||||||
bool m_establishing; // Pending connections ?
|
DEFER_READ = 1,
|
||||||
bool m_notify_state; // Notify state
|
DEFER_WRITE = 2
|
||||||
int m_id; // Socket id (for event handler)
|
|
||||||
|
|
||||||
// Defering variables
|
|
||||||
enum {
|
|
||||||
DEFER_READ, DEFER_WRITE, NO_DEFER
|
|
||||||
} m_defering; // Defering state
|
} m_defering; // Defering state
|
||||||
char *m_defer_buffer; // Defering target buffer
|
char *m_defer_buffer; // Defering target buffer
|
||||||
wxUint32 m_defer_nbytes; // Defering buffer size
|
wxUint32 m_defer_nbytes; // Defering buffer size
|
||||||
wxTimer *m_defer_timer; // Timer for defering mode
|
wxTimer *m_defer_timer; // Timer for defering mode
|
||||||
|
|
||||||
bool m_error; // Did an error occur in last IO call ?
|
/*
|
||||||
wxList m_states; // Stack of states
|
char *m_read_buffer; // Target buffer (read)
|
||||||
|
char *m_write_buffer; // Target buffer (write)
|
||||||
|
wxUint32 m_read_nbytes; // Buffer size (read)
|
||||||
|
wxUint32 m_write_nbytes; // Buffer size (write)
|
||||||
|
wxTimer *m_read_timer; // Timer (read)
|
||||||
|
wxTimer *m_write_timer; // Timer (write)
|
||||||
|
*/
|
||||||
|
|
||||||
|
wxSockCbk m_cbk; // C callback
|
||||||
|
char *m_cdata; // C callback data
|
||||||
|
|
||||||
public:
|
public:
|
||||||
wxSocketBase();
|
wxSocketBase();
|
||||||
@@ -127,7 +159,7 @@ public:
|
|||||||
wxSocketBase& Unread(const char *buffer, wxUint32 nbytes);
|
wxSocketBase& Unread(const char *buffer, wxUint32 nbytes);
|
||||||
wxSocketBase& ReadMsg(char *buffer, wxUint32 nbytes);
|
wxSocketBase& ReadMsg(char *buffer, wxUint32 nbytes);
|
||||||
wxSocketBase& WriteMsg(const char *buffer, wxUint32 nbytes);
|
wxSocketBase& WriteMsg(const char *buffer, wxUint32 nbytes);
|
||||||
void Discard();
|
wxSocketBase& Discard();
|
||||||
|
|
||||||
// Status
|
// Status
|
||||||
inline bool Ok() const { return (m_socket != NULL); };
|
inline bool Ok() const { return (m_socket != NULL); };
|
||||||
@@ -135,7 +167,7 @@ public:
|
|||||||
inline bool IsConnected() const { return m_connected; };
|
inline bool IsConnected() const { return m_connected; };
|
||||||
inline bool IsDisconnected() const { return !IsConnected(); };
|
inline bool IsDisconnected() const { return !IsConnected(); };
|
||||||
inline bool IsNoWait() const { return ((m_flags & NOWAIT) != 0); };
|
inline bool IsNoWait() const { return ((m_flags & NOWAIT) != 0); };
|
||||||
bool IsData() const;
|
inline bool IsData() { return WaitForRead(0, 0); };
|
||||||
inline wxUint32 LastCount() const { return m_lcount; }
|
inline wxUint32 LastCount() const { return m_lcount; }
|
||||||
inline wxSocketError LastError() const { return (wxSocketError)GSocket_GetError(m_socket); }
|
inline wxSocketError LastError() const { return (wxSocketError)GSocket_GetError(m_socket); }
|
||||||
inline wxSockType GetType() const { return m_type; }
|
inline wxSockType GetType() const { return m_type; }
|
||||||
@@ -145,9 +177,9 @@ public:
|
|||||||
virtual bool GetLocal(wxSockAddress& addr_man) const;
|
virtual bool GetLocal(wxSockAddress& addr_man) const;
|
||||||
|
|
||||||
// Set attributes and flags
|
// Set attributes and flags
|
||||||
void SetFlags(wxSockFlags _flags);
|
|
||||||
wxSockFlags GetFlags() const;
|
|
||||||
void SetTimeout(long seconds);
|
void SetTimeout(long seconds);
|
||||||
|
void SetFlags(wxSockFlags flags);
|
||||||
|
inline wxSockFlags GetFlags() const { return m_flags; };
|
||||||
|
|
||||||
// Wait functions
|
// Wait functions
|
||||||
// seconds = -1 means default timeout (change with SetTimeout)
|
// seconds = -1 means default timeout (change with SetTimeout)
|
||||||
@@ -178,9 +210,7 @@ public:
|
|||||||
// Public internal callback
|
// Public internal callback
|
||||||
virtual void OldOnNotify(wxSocketNotify WXUNUSED(evt));
|
virtual void OldOnNotify(wxSocketNotify WXUNUSED(evt));
|
||||||
|
|
||||||
// Do NOT use these functions; they should be protected!
|
// Do NOT use this function; it should be protected!
|
||||||
void CreatePushbackAfter(const char *buffer, wxUint32 size);
|
|
||||||
void CreatePushbackBefore(const char *buffer, wxUint32 size);
|
|
||||||
void OnRequest(wxSocketNotify req_evt);
|
void OnRequest(wxSocketNotify req_evt);
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
@@ -198,13 +228,17 @@ public:
|
|||||||
protected:
|
protected:
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
// Low level IO
|
||||||
|
wxUint32 _Read(char* buffer, wxUint32 nbytes);
|
||||||
|
wxUint32 _Write(const char *buffer, wxUint32 nbytes);
|
||||||
bool _Wait(long seconds, long milliseconds, wxSocketEventFlags flags);
|
bool _Wait(long seconds, long milliseconds, wxSocketEventFlags flags);
|
||||||
|
|
||||||
int DeferRead(char *buffer, wxUint32 nbytes);
|
wxUint32 DeferRead(char *buffer, wxUint32 nbytes);
|
||||||
int DeferWrite(const char *buffer, wxUint32 nbytes);
|
wxUint32 DeferWrite(const char *buffer, wxUint32 nbytes);
|
||||||
void DoDefer();
|
void DoDefer();
|
||||||
|
|
||||||
// Pushback library
|
// Pushbacks
|
||||||
|
void Pushback(const char *buffer, wxUint32 size);
|
||||||
wxUint32 GetPushback(char *buffer, wxUint32 size, bool peek);
|
wxUint32 GetPushback(char *buffer, wxUint32 size, bool peek);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@@ -5,14 +5,22 @@
|
|||||||
* CVSID: $Id$
|
* CVSID: $Id$
|
||||||
* -------------------------------------------------------------------------
|
* -------------------------------------------------------------------------
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef __GSOCK_UNX_H
|
#ifndef __GSOCK_UNX_H
|
||||||
#define __GSOCK_UNX_H
|
#define __GSOCK_UNX_H
|
||||||
|
|
||||||
|
#ifndef __GSOCKET_STANDALONE__
|
||||||
#include "wx/setup.h"
|
#include "wx/setup.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
#if wxUSE_SOCKETS
|
#if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__)
|
||||||
|
|
||||||
|
#ifndef __GSOCKET_STANDALONE__
|
||||||
#include "wx/gsocket.h"
|
#include "wx/gsocket.h"
|
||||||
|
#else
|
||||||
|
#include "gsocket.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
@@ -20,27 +28,31 @@ extern "C" {
|
|||||||
|
|
||||||
|
|
||||||
/* Definition of GSocket */
|
/* Definition of GSocket */
|
||||||
struct _GSocket {
|
struct _GSocket
|
||||||
|
{
|
||||||
int m_fd;
|
int m_fd;
|
||||||
GAddress *m_local, *m_peer;
|
GAddress *m_local;
|
||||||
|
GAddress *m_peer;
|
||||||
GSocketError m_error;
|
GSocketError m_error;
|
||||||
|
|
||||||
bool m_non_blocking, m_server, m_stream, m_oriented;
|
bool m_non_blocking;
|
||||||
|
bool m_server;
|
||||||
|
bool m_stream;
|
||||||
|
bool m_oriented;
|
||||||
bool m_establishing;
|
bool m_establishing;
|
||||||
unsigned long m_timeout;
|
unsigned long m_timeout;
|
||||||
|
|
||||||
/* Callbacks */
|
/* Callbacks */
|
||||||
|
GSocketEventFlags m_detected;
|
||||||
GSocketCallback m_cbacks[GSOCK_MAX_EVENT];
|
GSocketCallback m_cbacks[GSOCK_MAX_EVENT];
|
||||||
char *m_data[GSOCK_MAX_EVENT];
|
char *m_data[GSOCK_MAX_EVENT];
|
||||||
|
|
||||||
/* IO calls associated */
|
|
||||||
bool m_iocalls[GSOCK_MAX_EVENT];
|
|
||||||
|
|
||||||
char *m_gui_dependent;
|
char *m_gui_dependent;
|
||||||
};
|
};
|
||||||
|
|
||||||
/* Definition of GAddress */
|
/* Definition of GAddress */
|
||||||
struct _GAddress {
|
struct _GAddress
|
||||||
|
{
|
||||||
struct sockaddr *m_addr;
|
struct sockaddr *m_addr;
|
||||||
size_t m_len;
|
size_t m_len;
|
||||||
|
|
||||||
@@ -50,39 +62,43 @@ struct _GAddress {
|
|||||||
GSocketError m_error;
|
GSocketError m_error;
|
||||||
};
|
};
|
||||||
|
|
||||||
void _GSocket_Enable(GSocket *socket, GSocketEvent event);
|
/* Input / Output */
|
||||||
void _GSocket_Disable(GSocket *socket, GSocketEvent event);
|
|
||||||
void _GSocket_Configure_Callbacks(GSocket *socket);
|
GSocketError _GSocket_Input_Timeout(GSocket *socket);
|
||||||
|
GSocketError _GSocket_Output_Timeout(GSocket *socket);
|
||||||
int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size);
|
int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size);
|
||||||
int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size);
|
int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size);
|
||||||
int _GSocket_Send_Stream(GSocket *socket, const char *buffer, int size);
|
int _GSocket_Send_Stream(GSocket *socket, const char *buffer, int size);
|
||||||
int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size);
|
int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size);
|
||||||
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent count);
|
|
||||||
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent count);
|
/* Callbacks */
|
||||||
|
|
||||||
|
void _GSocket_Enable(GSocket *socket, GSocketEvent event);
|
||||||
|
void _GSocket_Disable(GSocket *socket, GSocketEvent event);
|
||||||
void _GSocket_Detected_Read(GSocket *socket);
|
void _GSocket_Detected_Read(GSocket *socket);
|
||||||
void _GSocket_Detected_Write(GSocket *socket);
|
void _GSocket_Detected_Write(GSocket *socket);
|
||||||
|
|
||||||
void _GSocket_GUI_Init(GSocket *socket);
|
void _GSocket_GUI_Init(GSocket *socket);
|
||||||
void _GSocket_GUI_Destroy(GSocket *socket);
|
void _GSocket_GUI_Destroy(GSocket *socket);
|
||||||
|
void _GSocket_Enable_Events(GSocket *socket);
|
||||||
|
void _GSocket_Disable_Events(GSocket *socket);
|
||||||
|
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event);
|
||||||
|
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event);
|
||||||
|
|
||||||
|
/* GAddress */
|
||||||
|
|
||||||
/* Translaters returns false when memory is exhausted */
|
|
||||||
GSocketError _GAddress_translate_from(GAddress *address,
|
GSocketError _GAddress_translate_from(GAddress *address,
|
||||||
struct sockaddr *addr, int len);
|
struct sockaddr *addr, int len);
|
||||||
GSocketError _GAddress_translate_to(GAddress *address,
|
GSocketError _GAddress_translate_to(GAddress *address,
|
||||||
struct sockaddr **addr, int *len);
|
struct sockaddr **addr, int *len);
|
||||||
|
|
||||||
/* Initialisers returns FALSE when an error happened in the initialisation */
|
|
||||||
|
|
||||||
/* Internet address family */
|
|
||||||
GSocketError _GAddress_Init_INET(GAddress *address);
|
GSocketError _GAddress_Init_INET(GAddress *address);
|
||||||
/* Local address family */
|
|
||||||
GSocketError _GAddress_Init_UNIX(GAddress *address);
|
GSocketError _GAddress_Init_UNIX(GAddress *address);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif /* __cplusplus */
|
#endif /* __cplusplus */
|
||||||
|
|
||||||
#endif
|
#endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */
|
||||||
/* wxUSE_SOCKETS */
|
|
||||||
|
|
||||||
#endif
|
#endif /* __GSOCK_UNX_H */
|
||||||
/* __GSOCK_UNX_H */
|
|
||||||
|
@@ -10,12 +10,16 @@
|
|||||||
#if wxUSE_SOCKETS
|
#if wxUSE_SOCKETS
|
||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
#include <stdio.h>
|
||||||
#include "gdk/gdk.h"
|
#include "gdk/gdk.h"
|
||||||
#include "glib.h"
|
#include "glib.h"
|
||||||
#include "wx/gsocket.h"
|
#include "wx/gsocket.h"
|
||||||
#include "wx/unix/gsockunx.h"
|
#include "wx/unix/gsockunx.h"
|
||||||
|
|
||||||
void _GSocket_GDK_Input(gpointer data, gint source, GdkInputCondition condition)
|
|
||||||
|
void _GSocket_GDK_Input(gpointer data,
|
||||||
|
gint source,
|
||||||
|
GdkInputCondition condition)
|
||||||
{
|
{
|
||||||
GSocket *socket = (GSocket *)data;
|
GSocket *socket = (GSocket *)data;
|
||||||
|
|
||||||
@@ -30,62 +34,55 @@ void _GSocket_GUI_Init(GSocket *socket)
|
|||||||
int i;
|
int i;
|
||||||
gint *m_id;
|
gint *m_id;
|
||||||
|
|
||||||
socket->m_gui_dependent = (char *)malloc(sizeof(gint)*3);
|
socket->m_gui_dependent = (char *)malloc(sizeof(gint)*2);
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
m_id = (gint *)(socket->m_gui_dependent);
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
m_id[0] = -1;
|
||||||
m_id[i] = -1;
|
m_id[1] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_GUI_Destroy(GSocket *socket)
|
void _GSocket_GUI_Destroy(GSocket *socket)
|
||||||
{
|
{
|
||||||
int i;
|
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
|
||||||
if (m_id[i] == -1)
|
|
||||||
gdk_input_remove(m_id[i]);
|
|
||||||
|
|
||||||
free(socket->m_gui_dependent);
|
free(socket->m_gui_dependent);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
GdkInputCondition flag;
|
gint *m_id = (gint *)(socket->m_gui_dependent);
|
||||||
int c;
|
int c;
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
if (socket->m_fd == -1)
|
||||||
|
return;
|
||||||
|
|
||||||
switch (event) {
|
switch (event)
|
||||||
case GSOCK_CONNECTION:
|
{
|
||||||
case GSOCK_LOST:
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_INPUT: c = 0; flag = GDK_INPUT_READ; break;
|
case GSOCK_INPUT: c = 0; break;
|
||||||
case GSOCK_OUTPUT: c = 1;flag = GDK_INPUT_WRITE; break;
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
default: return;
|
default: return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (m_id[c] != -1)
|
if (m_id[c] != -1)
|
||||||
gdk_input_remove(m_id[c]);
|
gdk_input_remove(m_id[c]);
|
||||||
|
|
||||||
m_id[c] = gdk_input_add(socket->m_fd, flag,
|
m_id[c] = gdk_input_add(socket->m_fd,
|
||||||
_GSocket_GDK_Input, (gpointer)socket);
|
(c ? GDK_INPUT_WRITE : GDK_INPUT_READ),
|
||||||
|
_GSocket_GDK_Input,
|
||||||
|
(gpointer)socket);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
|
gint *m_id = (gint *)(socket->m_gui_dependent);
|
||||||
int c;
|
int c;
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
switch (event)
|
||||||
|
{
|
||||||
switch (event) {
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_CONNECTION:
|
|
||||||
case GSOCK_LOST:
|
|
||||||
case GSOCK_INPUT: c = 0; break;
|
case GSOCK_INPUT: c = 0; break;
|
||||||
case GSOCK_OUTPUT: c = 1; break;
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
default: return;
|
default: return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -95,13 +92,16 @@ void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
|||||||
m_id[c] = -1;
|
m_id[c] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
unsigned long GSocket_GetEventID(GSocket *socket)
|
void _GSocket_Enable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
return 0;
|
_GSocket_Install_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Install_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
void GSocket_DoEvent(unsigned long evt_id)
|
void _GSocket_Disable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif /* wxUSE_SOCKETS */
|
#endif /* wxUSE_SOCKETS */
|
||||||
|
@@ -10,12 +10,16 @@
|
|||||||
#if wxUSE_SOCKETS
|
#if wxUSE_SOCKETS
|
||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
#include <stdio.h>
|
||||||
#include "gdk/gdk.h"
|
#include "gdk/gdk.h"
|
||||||
#include "glib.h"
|
#include "glib.h"
|
||||||
#include "wx/gsocket.h"
|
#include "wx/gsocket.h"
|
||||||
#include "wx/unix/gsockunx.h"
|
#include "wx/unix/gsockunx.h"
|
||||||
|
|
||||||
void _GSocket_GDK_Input(gpointer data, gint source, GdkInputCondition condition)
|
|
||||||
|
void _GSocket_GDK_Input(gpointer data,
|
||||||
|
gint source,
|
||||||
|
GdkInputCondition condition)
|
||||||
{
|
{
|
||||||
GSocket *socket = (GSocket *)data;
|
GSocket *socket = (GSocket *)data;
|
||||||
|
|
||||||
@@ -30,62 +34,55 @@ void _GSocket_GUI_Init(GSocket *socket)
|
|||||||
int i;
|
int i;
|
||||||
gint *m_id;
|
gint *m_id;
|
||||||
|
|
||||||
socket->m_gui_dependent = (char *)malloc(sizeof(gint)*3);
|
socket->m_gui_dependent = (char *)malloc(sizeof(gint)*2);
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
m_id = (gint *)(socket->m_gui_dependent);
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
m_id[0] = -1;
|
||||||
m_id[i] = -1;
|
m_id[1] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_GUI_Destroy(GSocket *socket)
|
void _GSocket_GUI_Destroy(GSocket *socket)
|
||||||
{
|
{
|
||||||
int i;
|
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
|
||||||
if (m_id[i] == -1)
|
|
||||||
gdk_input_remove(m_id[i]);
|
|
||||||
|
|
||||||
free(socket->m_gui_dependent);
|
free(socket->m_gui_dependent);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
GdkInputCondition flag;
|
gint *m_id = (gint *)(socket->m_gui_dependent);
|
||||||
int c;
|
int c;
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
if (socket->m_fd == -1)
|
||||||
|
return;
|
||||||
|
|
||||||
switch (event) {
|
switch (event)
|
||||||
case GSOCK_CONNECTION:
|
{
|
||||||
case GSOCK_LOST:
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_INPUT: c = 0; flag = GDK_INPUT_READ; break;
|
case GSOCK_INPUT: c = 0; break;
|
||||||
case GSOCK_OUTPUT: c = 1;flag = GDK_INPUT_WRITE; break;
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
default: return;
|
default: return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (m_id[c] != -1)
|
if (m_id[c] != -1)
|
||||||
gdk_input_remove(m_id[c]);
|
gdk_input_remove(m_id[c]);
|
||||||
|
|
||||||
m_id[c] = gdk_input_add(socket->m_fd, flag,
|
m_id[c] = gdk_input_add(socket->m_fd,
|
||||||
_GSocket_GDK_Input, (gpointer)socket);
|
(c ? GDK_INPUT_WRITE : GDK_INPUT_READ),
|
||||||
|
_GSocket_GDK_Input,
|
||||||
|
(gpointer)socket);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
|
gint *m_id = (gint *)(socket->m_gui_dependent);
|
||||||
int c;
|
int c;
|
||||||
gint *m_id;
|
|
||||||
|
|
||||||
m_id = (gint *)(socket->m_gui_dependent);
|
switch (event)
|
||||||
|
{
|
||||||
switch (event) {
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_CONNECTION:
|
|
||||||
case GSOCK_LOST:
|
|
||||||
case GSOCK_INPUT: c = 0; break;
|
case GSOCK_INPUT: c = 0; break;
|
||||||
case GSOCK_OUTPUT: c = 1; break;
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
default: return;
|
default: return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -95,13 +92,16 @@ void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
|||||||
m_id[c] = -1;
|
m_id[c] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
unsigned long GSocket_GetEventID(GSocket *socket)
|
void _GSocket_Enable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
return 0;
|
_GSocket_Install_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Install_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
void GSocket_DoEvent(unsigned long evt_id)
|
void _GSocket_Disable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif /* wxUSE_SOCKETS */
|
#endif /* wxUSE_SOCKETS */
|
||||||
|
@@ -11,9 +11,9 @@
|
|||||||
|
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <X11/Intrinsic.h>
|
#include <X11/Intrinsic.h>
|
||||||
#include <wx/gsocket.h>
|
#include "wx/gsocket.h"
|
||||||
#include <wx/app.h>
|
#include "wx/app.h"
|
||||||
#include <wx/unix/gsockunx.h>
|
#include "wx/unix/gsockunx.h"
|
||||||
|
|
||||||
#define wxAPP_CONTEXT ((XtAppContext)wxTheApp->GetAppContext())
|
#define wxAPP_CONTEXT ((XtAppContext)wxTheApp->GetAppContext())
|
||||||
|
|
||||||
@@ -38,69 +38,65 @@ void _GSocket_GUI_Init(GSocket *socket)
|
|||||||
int i;
|
int i;
|
||||||
int *m_id;
|
int *m_id;
|
||||||
|
|
||||||
socket->m_gui_dependent = (char *)malloc(sizeof(int)*3);
|
socket->m_gui_dependent = (char *)malloc(sizeof(int)*2);
|
||||||
m_id = (int *)(socket->m_gui_dependent);
|
m_id = (int *)(socket->m_gui_dependent);
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
m_id[0] = -1;
|
||||||
m_id[i] = -1;
|
m_id[1] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_GUI_Destroy(GSocket *socket)
|
void _GSocket_GUI_Destroy(GSocket *socket)
|
||||||
{
|
{
|
||||||
int i;
|
|
||||||
int *m_id;
|
|
||||||
|
|
||||||
m_id = (int *)(socket->m_gui_dependent);
|
|
||||||
|
|
||||||
for (i=0;i<3;i++)
|
|
||||||
if (m_id[i] == -1)
|
|
||||||
XtRemoveInput(m_id[i]);
|
|
||||||
|
|
||||||
free(socket->m_gui_dependent);
|
free(socket->m_gui_dependent);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Install_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
int *m_id;
|
int *m_id = (int *)(socket->m_gui_dependent);
|
||||||
|
int c;
|
||||||
|
|
||||||
m_id = (int *)(socket->m_gui_dependent);
|
if (socket->m_fd == -1)
|
||||||
|
return;
|
||||||
|
|
||||||
switch (event) {
|
switch (event)
|
||||||
case GSOCK_CONNECTION:
|
{
|
||||||
case GSOCK_LOST:
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_INPUT:
|
case GSOCK_INPUT: c = 0; break;
|
||||||
if (m_id[0] != -1)
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
XtRemoveInput(m_id[0]);
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
|
default: return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (m_id[c] != -1)
|
||||||
|
XtRemoveInput(m_id[c]);
|
||||||
|
|
||||||
|
if (c == 0)
|
||||||
|
{
|
||||||
m_id[0] = XtAppAddInput(wxAPP_CONTEXT, socket->m_fd,
|
m_id[0] = XtAppAddInput(wxAPP_CONTEXT, socket->m_fd,
|
||||||
(XtPointer *)XtInputReadMask,
|
(XtPointer *)XtInputReadMask,
|
||||||
(XtInputCallbackProc) _GSocket_Motif_Input,
|
(XtInputCallbackProc) _GSocket_Motif_Input,
|
||||||
(XtPointer) socket);
|
(XtPointer) socket);
|
||||||
break;
|
}
|
||||||
case GSOCK_OUTPUT:
|
else
|
||||||
if (m_id[1] != -1)
|
{
|
||||||
XtRemoveInput(m_id[1]);
|
|
||||||
m_id[1] = XtAppAddInput(wxAPP_CONTEXT, socket->m_fd,
|
m_id[1] = XtAppAddInput(wxAPP_CONTEXT, socket->m_fd,
|
||||||
(XtPointer *)XtInputWriteMask,
|
(XtPointer *)XtInputWriteMask,
|
||||||
(XtInputCallbackProc) _GSocket_Motif_Output,
|
(XtInputCallbackProc) _GSocket_Motif_Output,
|
||||||
(XtPointer) socket);
|
(XtPointer) socket);
|
||||||
break;
|
|
||||||
default: return;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
|
int *m_id = (int *)(socket->m_gui_dependent);
|
||||||
int c;
|
int c;
|
||||||
int *m_id;
|
|
||||||
|
|
||||||
m_id = (int *)(socket->m_gui_dependent);
|
switch (event)
|
||||||
|
{
|
||||||
switch (event) {
|
case GSOCK_LOST: /* fall-through */
|
||||||
case GSOCK_CONNECTION:
|
|
||||||
case GSOCK_LOST:
|
|
||||||
case GSOCK_INPUT: c = 0; break;
|
case GSOCK_INPUT: c = 0; break;
|
||||||
case GSOCK_OUTPUT: c = 1; break;
|
case GSOCK_OUTPUT: c = 1; break;
|
||||||
break;
|
case GSOCK_CONNECTION: c = ((socket->m_server) ? 0 : 1); break;
|
||||||
default: return;
|
default: return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,13 +106,16 @@ void _GSocket_Uninstall_Callback(GSocket *socket, GSocketEvent event)
|
|||||||
m_id[c] = -1;
|
m_id[c] = -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
unsigned long GSocket_GetEventID(GSocket *socket)
|
void _GSocket_Enable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
return 0;
|
_GSocket_Install_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Install_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
void GSocket_DoEvent(unsigned long evt_id)
|
void _GSocket_Disable_Events(GSocket *socket)
|
||||||
{
|
{
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_INPUT);
|
||||||
|
_GSocket_Uninstall_Callback(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // wxUSE_SOCKETS
|
#endif // wxUSE_SOCKETS
|
||||||
|
@@ -67,24 +67,6 @@
|
|||||||
signal(SIGPIPE, old_handler); \
|
signal(SIGPIPE, old_handler); \
|
||||||
}
|
}
|
||||||
|
|
||||||
#define ENABLE_TIMEOUT(socket) \
|
|
||||||
{ \
|
|
||||||
struct itimerval old_ival, new_ival; \
|
|
||||||
void (*old_timer_sig)(int); \
|
|
||||||
\
|
|
||||||
old_timer_sig = signal(SIGALRM, SIG_DFL); \
|
|
||||||
siginterrupt(SIGALRM, 1); \
|
|
||||||
new_ival.it_value.tv_sec = socket->m_timeout / 1000; \
|
|
||||||
new_ival.it_value.tv_usec = (socket->m_timeout % 1000) * 1000; \
|
|
||||||
new_ival.it_interval.tv_sec = 0; \
|
|
||||||
new_ival.it_interval.tv_usec = 0; \
|
|
||||||
setitimer(ITIMER_REAL, &new_ival, &old_ival);
|
|
||||||
|
|
||||||
#define DISABLE_TIMEOUT(socket) \
|
|
||||||
signal(SIGALRM, old_timer_sig); \
|
|
||||||
siginterrupt(SIGALRM, 0); \
|
|
||||||
setitimer(ITIMER_REAL, &old_ival, NULL); \
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Global initialisers */
|
/* Global initialisers */
|
||||||
|
|
||||||
@@ -110,10 +92,11 @@ GSocket *GSocket_new()
|
|||||||
return NULL;
|
return NULL;
|
||||||
|
|
||||||
socket->m_fd = -1;
|
socket->m_fd = -1;
|
||||||
for (i=0;i<GSOCK_MAX_EVENT;i++) {
|
for (i=0;i<GSOCK_MAX_EVENT;i++)
|
||||||
|
{
|
||||||
socket->m_cbacks[i] = NULL;
|
socket->m_cbacks[i] = NULL;
|
||||||
socket->m_iocalls[i] = FALSE;
|
|
||||||
}
|
}
|
||||||
|
socket->m_detected = 0;
|
||||||
socket->m_local = NULL;
|
socket->m_local = NULL;
|
||||||
socket->m_peer = NULL;
|
socket->m_peer = NULL;
|
||||||
socket->m_error = GSOCK_NOERROR;
|
socket->m_error = GSOCK_NOERROR;
|
||||||
@@ -160,7 +143,8 @@ void GSocket_Shutdown(GSocket *socket)
|
|||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
/* If socket has been created, we shutdown it */
|
/* If socket has been created, we shutdown it */
|
||||||
if (socket->m_fd != -1) {
|
if (socket->m_fd != -1)
|
||||||
|
{
|
||||||
shutdown(socket->m_fd, 2);
|
shutdown(socket->m_fd, 2);
|
||||||
close(socket->m_fd);
|
close(socket->m_fd);
|
||||||
socket->m_fd = -1;
|
socket->m_fd = -1;
|
||||||
@@ -168,7 +152,10 @@ void GSocket_Shutdown(GSocket *socket)
|
|||||||
|
|
||||||
/* We also disable GUI callbacks */
|
/* We also disable GUI callbacks */
|
||||||
for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
|
for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
|
||||||
_GSocket_Uninstall_Callback(socket, evt);
|
socket->m_cbacks[evt] = NULL;
|
||||||
|
|
||||||
|
socket->m_detected = 0;
|
||||||
|
_GSocket_Disable_Events(socket);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Address handling */
|
/* Address handling */
|
||||||
@@ -272,6 +259,7 @@ GAddress *GSocket_GetPeer(GSocket *socket)
|
|||||||
GSocketError GSocket_SetServer(GSocket *sck)
|
GSocketError GSocket_SetServer(GSocket *sck)
|
||||||
{
|
{
|
||||||
int type;
|
int type;
|
||||||
|
int arg = 1;
|
||||||
|
|
||||||
assert(sck != NULL);
|
assert(sck != NULL);
|
||||||
|
|
||||||
@@ -297,6 +285,9 @@ GSocketError GSocket_SetServer(GSocket *sck)
|
|||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ioctl(sck->m_fd, FIONBIO, &arg);
|
||||||
|
_GSocket_Enable_Events(sck);
|
||||||
|
|
||||||
/* Bind the socket to the LOCAL address */
|
/* Bind the socket to the LOCAL address */
|
||||||
if (bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) < 0) {
|
if (bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) < 0) {
|
||||||
close(sck->m_fd);
|
close(sck->m_fd);
|
||||||
@@ -313,10 +304,6 @@ GSocketError GSocket_SetServer(GSocket *sck)
|
|||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
_GSocket_Configure_Callbacks(sck);
|
|
||||||
GSocket_SetNonBlocking(sck, sck->m_non_blocking);
|
|
||||||
GSocket_SetTimeout(sck, sck->m_timeout);
|
|
||||||
|
|
||||||
return GSOCK_NOERROR;
|
return GSOCK_NOERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -326,54 +313,56 @@ GSocketError GSocket_SetServer(GSocket *sck)
|
|||||||
GSocket *GSocket_WaitConnection(GSocket *socket)
|
GSocket *GSocket_WaitConnection(GSocket *socket)
|
||||||
{
|
{
|
||||||
GSocket *connection;
|
GSocket *connection;
|
||||||
|
int arg = 1;
|
||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
|
/* Reenable CONNECTION events */
|
||||||
|
_GSocket_Enable(socket, GSOCK_CONNECTION);
|
||||||
|
|
||||||
/* If the socket has already been created, we exit immediately */
|
/* If the socket has already been created, we exit immediately */
|
||||||
if (socket->m_fd == -1 || !socket->m_server) {
|
if (socket->m_fd == -1 || !socket->m_server)
|
||||||
|
{
|
||||||
socket->m_error = GSOCK_INVSOCK;
|
socket->m_error = GSOCK_INVSOCK;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Reenable GSOCK_CONNECTION event */
|
|
||||||
_GSocket_Enable(socket, GSOCK_CONNECTION);
|
|
||||||
|
|
||||||
/* Create a GSocket object for the new connection */
|
/* Create a GSocket object for the new connection */
|
||||||
connection = GSocket_new();
|
connection = GSocket_new();
|
||||||
if (!connection) {
|
if (!connection)
|
||||||
|
{
|
||||||
connection->m_error = GSOCK_MEMERR;
|
connection->m_error = GSOCK_MEMERR;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Accept the incoming connection */
|
/* Accept the incoming connection */
|
||||||
ENABLE_TIMEOUT(connection);
|
if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
|
||||||
connection->m_fd = accept(socket->m_fd, NULL, NULL);
|
{
|
||||||
DISABLE_TIMEOUT(connection);
|
|
||||||
if (connection->m_fd == -1) {
|
|
||||||
GSocket_destroy(connection);
|
GSocket_destroy(connection);
|
||||||
switch(errno) {
|
/* socket->m_error set by _GSocket_Input_Timeout */
|
||||||
case EINTR:
|
return NULL;
|
||||||
case ETIMEDOUT:
|
|
||||||
connection->m_error = GSOCK_TIMEDOUT;
|
|
||||||
break;
|
|
||||||
case EWOULDBLOCK:
|
|
||||||
connection->m_error = GSOCK_WOULDBLOCK;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
connection->m_error = GSOCK_IOERR;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
connection->m_fd = accept(socket->m_fd, NULL, NULL);
|
||||||
|
|
||||||
|
if (connection->m_fd == -1)
|
||||||
|
{
|
||||||
|
if (errno == EWOULDBLOCK)
|
||||||
|
socket->m_error = GSOCK_WOULDBLOCK;
|
||||||
|
else
|
||||||
|
socket->m_error = GSOCK_IOERR;
|
||||||
|
|
||||||
|
GSocket_destroy(connection);
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Initialize all fields */
|
/* Initialize all fields */
|
||||||
connection->m_stream = TRUE;
|
|
||||||
connection->m_server = FALSE;
|
connection->m_server = FALSE;
|
||||||
|
connection->m_stream = TRUE;
|
||||||
connection->m_oriented = TRUE;
|
connection->m_oriented = TRUE;
|
||||||
|
|
||||||
_GSocket_Configure_Callbacks(connection);
|
ioctl(connection->m_fd, FIONBIO, &arg);
|
||||||
GSocket_SetNonBlocking(connection, connection->m_non_blocking);
|
_GSocket_Enable_Events(connection);
|
||||||
GSocket_SetTimeout(connection, connection->m_timeout);
|
|
||||||
|
|
||||||
return connection;
|
return connection;
|
||||||
}
|
}
|
||||||
@@ -382,6 +371,8 @@ GSocket *GSocket_WaitConnection(GSocket *socket)
|
|||||||
|
|
||||||
GSocketError GSocket_SetNonOriented(GSocket *sck)
|
GSocketError GSocket_SetNonOriented(GSocket *sck)
|
||||||
{
|
{
|
||||||
|
int arg = 1;
|
||||||
|
|
||||||
assert(sck != NULL);
|
assert(sck != NULL);
|
||||||
|
|
||||||
if (sck->m_fd != -1) {
|
if (sck->m_fd != -1) {
|
||||||
@@ -406,6 +397,9 @@ GSocketError GSocket_SetNonOriented(GSocket *sck)
|
|||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ioctl(sck->m_fd, FIONBIO, &arg);
|
||||||
|
_GSocket_Enable_Events(sck);
|
||||||
|
|
||||||
/* Bind it to the LOCAL address */
|
/* Bind it to the LOCAL address */
|
||||||
if (bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) < 0) {
|
if (bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) < 0) {
|
||||||
close(sck->m_fd);
|
close(sck->m_fd);
|
||||||
@@ -414,10 +408,6 @@ GSocketError GSocket_SetNonOriented(GSocket *sck)
|
|||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
_GSocket_Configure_Callbacks(sck);
|
|
||||||
GSocket_SetNonBlocking(sck, sck->m_non_blocking);
|
|
||||||
GSocket_SetTimeout(sck, sck->m_timeout);
|
|
||||||
|
|
||||||
return GSOCK_NOERROR;
|
return GSOCK_NOERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -434,16 +424,22 @@ GSocketError GSocket_SetNonOriented(GSocket *sck)
|
|||||||
*/
|
*/
|
||||||
GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
|
GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
|
||||||
{
|
{
|
||||||
int type, err;
|
int type, err, ret;
|
||||||
|
int arg = 1;
|
||||||
|
|
||||||
assert(sck != NULL);
|
assert(sck != NULL);
|
||||||
|
|
||||||
if (sck->m_fd != -1) {
|
/* Enable CONNECTION events (needed for nonblocking connections) */
|
||||||
|
_GSocket_Enable(sck, GSOCK_CONNECTION);
|
||||||
|
|
||||||
|
if (sck->m_fd != -1)
|
||||||
|
{
|
||||||
sck->m_error = GSOCK_INVSOCK;
|
sck->m_error = GSOCK_INVSOCK;
|
||||||
return GSOCK_INVSOCK;
|
return GSOCK_INVSOCK;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!sck->m_peer) {
|
if (!sck->m_peer)
|
||||||
|
{
|
||||||
sck->m_error = GSOCK_INVADDR;
|
sck->m_error = GSOCK_INVADDR;
|
||||||
return GSOCK_INVADDR;
|
return GSOCK_INVADDR;
|
||||||
}
|
}
|
||||||
@@ -451,6 +447,8 @@ GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
|
|||||||
/* Test whether we want the socket to be a stream (e.g. TCP) */
|
/* Test whether we want the socket to be a stream (e.g. TCP) */
|
||||||
sck->m_stream = (stream == GSOCK_STREAMED);
|
sck->m_stream = (stream == GSOCK_STREAMED);
|
||||||
sck->m_oriented = TRUE;
|
sck->m_oriented = TRUE;
|
||||||
|
sck->m_server = FALSE;
|
||||||
|
sck->m_establishing = FALSE;
|
||||||
|
|
||||||
if (sck->m_stream)
|
if (sck->m_stream)
|
||||||
type = SOCK_STREAM;
|
type = SOCK_STREAM;
|
||||||
@@ -465,35 +463,66 @@ GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
|
|||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
_GSocket_Configure_Callbacks(sck);
|
ioctl(sck->m_fd, FIONBIO, &arg);
|
||||||
GSocket_SetNonBlocking(sck, sck->m_non_blocking);
|
_GSocket_Enable_Events(sck);
|
||||||
GSocket_SetTimeout(sck, sck->m_timeout);
|
|
||||||
|
|
||||||
/* Connect it to the PEER address */
|
/* Connect it to the PEER address */
|
||||||
ENABLE_TIMEOUT(sck);
|
ret = connect(sck->m_fd, sck->m_peer->m_addr, sck->m_peer->m_len);
|
||||||
err = connect(sck->m_fd, sck->m_peer->m_addr, sck->m_peer->m_len);
|
|
||||||
DISABLE_TIMEOUT(sck);
|
|
||||||
|
|
||||||
if (err != 0 && errno != EINPROGRESS) {
|
if (ret == -1)
|
||||||
|
{
|
||||||
|
err = errno;
|
||||||
|
|
||||||
|
/* If connect failed with EINPROGRESS and the GSocket object
|
||||||
|
* is in blocking mode, we select() for the specified timeout
|
||||||
|
* checking for writability to see if the connection request
|
||||||
|
* completes.
|
||||||
|
*/
|
||||||
|
if ((err == EINPROGRESS) && (!sck->m_non_blocking))
|
||||||
|
{
|
||||||
|
if (_GSocket_Output_Timeout(sck) == GSOCK_TIMEDOUT)
|
||||||
|
{
|
||||||
close(sck->m_fd);
|
close(sck->m_fd);
|
||||||
sck->m_fd = -1;
|
sck->m_fd = -1;
|
||||||
switch (errno) {
|
/* sck->m_error is set in _GSocket_Output_Timeout */
|
||||||
case EINTR:
|
fprintf(stderr, "Blocking connect timeouts\n");
|
||||||
case ETIMEDOUT:
|
return GSOCK_TIMEDOUT;
|
||||||
sck->m_error = GSOCK_TIMEDOUT;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
sck->m_error = GSOCK_IOERR;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
fprintf(stderr, "Blocking connect OK\n");
|
||||||
|
return GSOCK_NOERROR;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* If connect failed with EINPROGRESS and the GSocket object
|
||||||
|
* is set to nonblocking, we set m_error to GSOCK_WOULDBLOCK
|
||||||
|
* (and return GSOCK_WOULDBLOCK) but we don't close the socket;
|
||||||
|
* this way if the connection completes, a GSOCK_CONNECTION
|
||||||
|
* event will be generated, if enabled.
|
||||||
|
*/
|
||||||
|
if ((err == EINPROGRESS) && (sck->m_non_blocking))
|
||||||
|
{
|
||||||
|
sck->m_error = GSOCK_WOULDBLOCK;
|
||||||
|
sck->m_establishing = TRUE;
|
||||||
|
fprintf(stderr, "Nonblocking connect in progress\n");
|
||||||
|
|
||||||
|
return GSOCK_WOULDBLOCK;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* If connect failed with an error other than EINPROGRESS,
|
||||||
|
* then the call to GSocket_Connect has failed.
|
||||||
|
*/
|
||||||
|
close(sck->m_fd);
|
||||||
|
sck->m_fd = -1;
|
||||||
|
sck->m_error = GSOCK_IOERR;
|
||||||
|
|
||||||
|
fprintf(stderr, "Connect failed (generic err)\n");
|
||||||
return GSOCK_IOERR;
|
return GSOCK_IOERR;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* It is not a server */
|
fprintf(stderr, "Connect OK\n");
|
||||||
sck->m_server = FALSE;
|
return GSOCK_NOERROR;
|
||||||
sck->m_establishing = (errno == EINPROGRESS);
|
|
||||||
|
|
||||||
return (sck->m_establishing) ? GSOCK_WOULDBLOCK : GSOCK_NOERROR;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Generic IO */
|
/* Generic IO */
|
||||||
@@ -501,38 +530,75 @@ GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
|
|||||||
/* Like recv(), send(), ... */
|
/* Like recv(), send(), ... */
|
||||||
int GSocket_Read(GSocket *socket, char *buffer, int size)
|
int GSocket_Read(GSocket *socket, char *buffer, int size)
|
||||||
{
|
{
|
||||||
|
int ret;
|
||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
if (socket->m_fd == -1 || socket->m_server) {
|
/* Reenable INPUT events */
|
||||||
|
_GSocket_Enable(socket, GSOCK_INPUT);
|
||||||
|
|
||||||
|
if (socket->m_fd == -1 || socket->m_server)
|
||||||
|
{
|
||||||
socket->m_error = GSOCK_INVSOCK;
|
socket->m_error = GSOCK_INVSOCK;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Reenable GSOCK_INPUT event */
|
if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
|
||||||
_GSocket_Enable(socket, GSOCK_INPUT);
|
return -1;
|
||||||
|
|
||||||
if (socket->m_stream)
|
if (socket->m_stream)
|
||||||
return _GSocket_Recv_Stream(socket, buffer, size);
|
ret = _GSocket_Recv_Stream(socket, buffer, size);
|
||||||
else
|
else
|
||||||
return _GSocket_Recv_Dgram(socket, buffer, size);
|
ret = _GSocket_Recv_Dgram(socket, buffer, size);
|
||||||
|
|
||||||
|
if (ret == -1)
|
||||||
|
{
|
||||||
|
if (errno == EWOULDBLOCK)
|
||||||
|
socket->m_error = GSOCK_WOULDBLOCK;
|
||||||
|
else
|
||||||
|
socket->m_error = GSOCK_IOERR;
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int GSocket_Write(GSocket *socket, const char *buffer,
|
int GSocket_Write(GSocket *socket, const char *buffer,
|
||||||
int size)
|
int size)
|
||||||
{
|
{
|
||||||
|
int ret;
|
||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
if (socket->m_fd == -1 || socket->m_server) {
|
if (socket->m_fd == -1 || socket->m_server)
|
||||||
|
{
|
||||||
socket->m_error = GSOCK_INVSOCK;
|
socket->m_error = GSOCK_INVSOCK;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
_GSocket_Enable(socket, GSOCK_OUTPUT);
|
if (_GSocket_Output_Timeout(socket) == GSOCK_TIMEDOUT)
|
||||||
|
return -1;
|
||||||
|
|
||||||
if (socket->m_stream)
|
if (socket->m_stream)
|
||||||
return _GSocket_Send_Stream(socket, buffer, size);
|
ret = _GSocket_Send_Stream(socket, buffer, size);
|
||||||
else
|
else
|
||||||
return _GSocket_Send_Dgram(socket, buffer, size);
|
ret = _GSocket_Send_Dgram(socket, buffer, size);
|
||||||
|
|
||||||
|
if (ret == -1)
|
||||||
|
{
|
||||||
|
if (errno == EWOULDBLOCK)
|
||||||
|
socket->m_error = GSOCK_WOULDBLOCK;
|
||||||
|
else
|
||||||
|
socket->m_error = GSOCK_IOERR;
|
||||||
|
|
||||||
|
/* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
|
||||||
|
* in MSW). Once the first OUTPUT event is received, users can assume
|
||||||
|
* that the socket is writable until a read operation fails. Only then
|
||||||
|
* will further OUTPUT events be posted.
|
||||||
|
*/
|
||||||
|
_GSocket_Enable(socket, GSOCK_OUTPUT);
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* GSocket_Select:
|
/* GSocket_Select:
|
||||||
@@ -544,62 +610,9 @@ int GSocket_Write(GSocket *socket, const char *buffer,
|
|||||||
*/
|
*/
|
||||||
GSocketEventFlags GSocket_Select(GSocket *socket, GSocketEventFlags flags)
|
GSocketEventFlags GSocket_Select(GSocket *socket, GSocketEventFlags flags)
|
||||||
{
|
{
|
||||||
fd_set readfds, writefds, exceptfds;
|
|
||||||
struct timeval tv;
|
|
||||||
GSocketEventFlags mask;
|
|
||||||
int error, len;
|
|
||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
if (socket->m_fd == -1)
|
return (flags & socket->m_detected);
|
||||||
{
|
|
||||||
socket->m_error = GSOCK_INVSOCK;
|
|
||||||
return FALSE;
|
|
||||||
}
|
|
||||||
|
|
||||||
FD_ZERO(&readfds);
|
|
||||||
FD_ZERO(&writefds);
|
|
||||||
FD_ZERO(&exceptfds);
|
|
||||||
FD_SET(socket->m_fd, &readfds);
|
|
||||||
FD_SET(socket->m_fd, &writefds);
|
|
||||||
FD_SET(socket->m_fd, &exceptfds);
|
|
||||||
|
|
||||||
tv.tv_sec = 0;
|
|
||||||
tv.tv_usec = 0;
|
|
||||||
select(socket->m_fd + 1, &readfds, &writefds, &exceptfds, &tv);
|
|
||||||
|
|
||||||
mask = 0;
|
|
||||||
|
|
||||||
/* If select() says that the socket is readable, then we have
|
|
||||||
* no way to distinguish if that means 'data available' (to
|
|
||||||
* recv) or 'incoming connection' (to accept). The same goes
|
|
||||||
* for writability: we cannot distinguish between 'you can
|
|
||||||
* send data' and 'connection request completed'. So we will
|
|
||||||
* assume the following: if the flag was set upon entry,
|
|
||||||
* that means that the event was possible.
|
|
||||||
*/
|
|
||||||
if (FD_ISSET(socket->m_fd, &readfds))
|
|
||||||
{
|
|
||||||
mask |= (flags & GSOCK_CONNECTION_FLAG);
|
|
||||||
mask |= (flags & GSOCK_INPUT_FLAG);
|
|
||||||
}
|
|
||||||
if (FD_ISSET(socket->m_fd, &writefds))
|
|
||||||
{
|
|
||||||
if (socket->m_establishing)
|
|
||||||
{
|
|
||||||
getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, &error, &len);
|
|
||||||
|
|
||||||
if (error)
|
|
||||||
mask |= (flags & GSOCK_LOST_FLAG);
|
|
||||||
else
|
|
||||||
mask |= (flags & GSOCK_CONNECTION_FLAG);
|
|
||||||
}
|
|
||||||
mask |= (flags & GSOCK_OUTPUT_FLAG);
|
|
||||||
}
|
|
||||||
if (FD_ISSET(socket->m_fd, &exceptfds))
|
|
||||||
mask |= (flags & GSOCK_LOST_FLAG);
|
|
||||||
|
|
||||||
return mask;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Flags */
|
/* Flags */
|
||||||
@@ -613,9 +626,6 @@ void GSocket_SetNonBlocking(GSocket *socket, bool non_block)
|
|||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
socket->m_non_blocking = non_block;
|
socket->m_non_blocking = non_block;
|
||||||
|
|
||||||
if (socket->m_fd != -1)
|
|
||||||
ioctl(socket->m_fd, FIONBIO, &non_block);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* GSocket_SetTimeout:
|
/* GSocket_SetTimeout:
|
||||||
@@ -672,15 +682,14 @@ void GSocket_SetCallback(GSocket *socket, GSocketEventFlags flags,
|
|||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
for (count=0;count<GSOCK_MAX_EVENT;count++) {
|
for (count = 0; count < GSOCK_MAX_EVENT; count++)
|
||||||
/* We test each flag and, if it is enabled, we enable the corresponding
|
{
|
||||||
event */
|
if ((flags & (1 << count)) != 0)
|
||||||
if ((flags & (1 << count)) != 0) {
|
{
|
||||||
socket->m_cbacks[count] = callback;
|
socket->m_cbacks[count] = callback;
|
||||||
socket->m_data[count] = cdata;
|
socket->m_data[count] = cdata;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_GSocket_Configure_Callbacks(socket);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* GSocket_UnsetCallback:
|
/* GSocket_UnsetCallback:
|
||||||
@@ -693,44 +702,85 @@ void GSocket_UnsetCallback(GSocket *socket, GSocketEventFlags flags)
|
|||||||
|
|
||||||
assert(socket != NULL);
|
assert(socket != NULL);
|
||||||
|
|
||||||
for (count=0;count<GSOCK_MAX_EVENT;count++) {
|
for (count = 0; count < GSOCK_MAX_EVENT; count++)
|
||||||
if ((flags & (1 << count)) != 0) {
|
{
|
||||||
|
if ((flags & (1 << count)) != 0)
|
||||||
|
{
|
||||||
socket->m_cbacks[count] = NULL;
|
socket->m_cbacks[count] = NULL;
|
||||||
|
socket->m_data[count] = NULL;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_GSocket_Configure_Callbacks(socket);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#define CALL_CALLBACK(socket, event) \
|
#define CALL_CALLBACK(socket, event) { \
|
||||||
if (socket->m_iocalls[event] && socket->m_cbacks[event]) { \
|
|
||||||
_GSocket_Disable(socket, event); \
|
_GSocket_Disable(socket, event); \
|
||||||
|
if (socket->m_cbacks[event]) \
|
||||||
socket->m_cbacks[event](socket, event, socket->m_data[event]); \
|
socket->m_cbacks[event](socket, event, socket->m_data[event]); \
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void _GSocket_Enable(GSocket *socket, GSocketEvent event)
|
void _GSocket_Enable(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
socket->m_iocalls[event] = TRUE;
|
socket->m_detected &= ~(1 << event);
|
||||||
if (socket->m_cbacks[event])
|
|
||||||
_GSocket_Install_Callback(socket, event);
|
_GSocket_Install_Callback(socket, event);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Disable(GSocket *socket, GSocketEvent event)
|
void _GSocket_Disable(GSocket *socket, GSocketEvent event)
|
||||||
{
|
{
|
||||||
socket->m_iocalls[event] = FALSE;
|
socket->m_detected |= (1 << event);
|
||||||
if (socket->m_cbacks[event])
|
|
||||||
_GSocket_Uninstall_Callback(socket, event);
|
_GSocket_Uninstall_Callback(socket, event);
|
||||||
}
|
}
|
||||||
|
|
||||||
void _GSocket_Configure_Callbacks(GSocket *socket)
|
/* _GSocket_Input_Timeout:
|
||||||
|
* For blocking sockets, wait until data is available or
|
||||||
|
* until timeout ellapses.
|
||||||
|
*/
|
||||||
|
GSocketError _GSocket_Input_Timeout(GSocket *socket)
|
||||||
{
|
{
|
||||||
int count;
|
struct timeval tv;
|
||||||
for (count=0;count<GSOCK_MAX_EVENT;count++) {
|
fd_set readfds;
|
||||||
if ((socket->m_cbacks[count]) != NULL) {
|
|
||||||
_GSocket_Enable(socket, count);
|
/* Linux select() will overwrite the struct on return */
|
||||||
} else {
|
tv.tv_sec = (socket->m_timeout / 1000);
|
||||||
_GSocket_Disable(socket, count);
|
tv.tv_usec = (socket->m_timeout % 1000) * 1000;
|
||||||
|
|
||||||
|
if (!socket->m_non_blocking)
|
||||||
|
{
|
||||||
|
FD_ZERO(&readfds);
|
||||||
|
FD_SET(socket->m_fd, &readfds);
|
||||||
|
if (select(socket->m_fd + 1, &readfds, NULL, NULL, &tv) == 0)
|
||||||
|
{
|
||||||
|
socket->m_error = GSOCK_TIMEDOUT;
|
||||||
|
return GSOCK_TIMEDOUT;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return GSOCK_NOERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* _GSocket_Output_Timeout:
|
||||||
|
* For blocking sockets, wait until data can be sent without
|
||||||
|
* blocking or until timeout ellapses.
|
||||||
|
*/
|
||||||
|
GSocketError _GSocket_Output_Timeout(GSocket *socket)
|
||||||
|
{
|
||||||
|
struct timeval tv;
|
||||||
|
fd_set writefds;
|
||||||
|
|
||||||
|
/* Linux select() will overwrite the struct on return */
|
||||||
|
tv.tv_sec = (socket->m_timeout / 1000);
|
||||||
|
tv.tv_usec = (socket->m_timeout % 1000) * 1000;
|
||||||
|
|
||||||
|
if (!socket->m_non_blocking)
|
||||||
|
{
|
||||||
|
FD_ZERO(&writefds);
|
||||||
|
FD_SET(socket->m_fd, &writefds);
|
||||||
|
if (select(socket->m_fd + 1, NULL, &writefds, NULL, &tv) == 0)
|
||||||
|
{
|
||||||
|
socket->m_error = GSOCK_TIMEDOUT;
|
||||||
|
return GSOCK_TIMEDOUT;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return GSOCK_NOERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
|
int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
|
||||||
@@ -738,24 +788,9 @@ int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
|
|||||||
int ret;
|
int ret;
|
||||||
|
|
||||||
MASK_SIGNAL();
|
MASK_SIGNAL();
|
||||||
ENABLE_TIMEOUT(socket);
|
|
||||||
ret = recv(socket->m_fd, buffer, size, 0);
|
ret = recv(socket->m_fd, buffer, size, 0);
|
||||||
DISABLE_TIMEOUT(socket);
|
|
||||||
UNMASK_SIGNAL();
|
UNMASK_SIGNAL();
|
||||||
|
|
||||||
if (ret == -1 &&
|
|
||||||
errno != ETIMEDOUT && errno != EWOULDBLOCK && errno != EINTR) {
|
|
||||||
socket->m_error = GSOCK_IOERR;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == EWOULDBLOCK) {
|
|
||||||
socket->m_error = GSOCK_WOULDBLOCK;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == ETIMEDOUT || errno == EINTR) {
|
|
||||||
socket->m_error = GSOCK_TIMEDOUT;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -769,33 +804,25 @@ int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size)
|
|||||||
fromlen = sizeof(from);
|
fromlen = sizeof(from);
|
||||||
|
|
||||||
MASK_SIGNAL();
|
MASK_SIGNAL();
|
||||||
ENABLE_TIMEOUT(socket);
|
|
||||||
ret = recvfrom(socket->m_fd, buffer, size, 0, &from, &fromlen);
|
ret = recvfrom(socket->m_fd, buffer, size, 0, &from, &fromlen);
|
||||||
DISABLE_TIMEOUT(socket);
|
|
||||||
UNMASK_SIGNAL();
|
UNMASK_SIGNAL();
|
||||||
if (ret == -1 && errno != EWOULDBLOCK && errno != EINTR && errno != ETIMEDOUT) {
|
|
||||||
socket->m_error = GSOCK_IOERR;
|
if (ret == -1)
|
||||||
return -1;
|
return -1;
|
||||||
}
|
|
||||||
if (errno == EWOULDBLOCK) {
|
|
||||||
socket->m_error = GSOCK_WOULDBLOCK;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == ETIMEDOUT || errno == EINTR) {
|
|
||||||
socket->m_error = GSOCK_TIMEDOUT;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Translate a system address into a GSocket address */
|
/* Translate a system address into a GSocket address */
|
||||||
if (!socket->m_peer) {
|
if (!socket->m_peer)
|
||||||
|
{
|
||||||
socket->m_peer = GAddress_new();
|
socket->m_peer = GAddress_new();
|
||||||
if (!socket->m_peer) {
|
if (!socket->m_peer)
|
||||||
|
{
|
||||||
socket->m_error = GSOCK_MEMERR;
|
socket->m_error = GSOCK_MEMERR;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
err = _GAddress_translate_from(socket->m_peer, &from, fromlen);
|
err = _GAddress_translate_from(socket->m_peer, &from, fromlen);
|
||||||
if (err != GSOCK_NOERROR) {
|
if (err != GSOCK_NOERROR)
|
||||||
|
{
|
||||||
GAddress_destroy(socket->m_peer);
|
GAddress_destroy(socket->m_peer);
|
||||||
socket->m_peer = NULL;
|
socket->m_peer = NULL;
|
||||||
socket->m_error = err;
|
socket->m_error = err;
|
||||||
@@ -811,22 +838,9 @@ int _GSocket_Send_Stream(GSocket *socket, const char *buffer, int size)
|
|||||||
GSocketError err;
|
GSocketError err;
|
||||||
|
|
||||||
MASK_SIGNAL();
|
MASK_SIGNAL();
|
||||||
ENABLE_TIMEOUT(socket);
|
|
||||||
ret = send(socket->m_fd, buffer, size, 0);
|
ret = send(socket->m_fd, buffer, size, 0);
|
||||||
DISABLE_TIMEOUT(socket);
|
|
||||||
UNMASK_SIGNAL();
|
UNMASK_SIGNAL();
|
||||||
if (ret == -1 && errno != EWOULDBLOCK && errno != ETIMEDOUT && errno != EINTR) {
|
|
||||||
socket->m_error = GSOCK_IOERR;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == EWOULDBLOCK) {
|
|
||||||
socket->m_error = GSOCK_WOULDBLOCK;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == ETIMEDOUT || errno == EINTR) {
|
|
||||||
socket->m_error = GSOCK_TIMEDOUT;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -848,23 +862,12 @@ int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size)
|
|||||||
}
|
}
|
||||||
|
|
||||||
MASK_SIGNAL();
|
MASK_SIGNAL();
|
||||||
ENABLE_TIMEOUT(socket);
|
|
||||||
ret = sendto(socket->m_fd, buffer, size, 0, addr, len);
|
ret = sendto(socket->m_fd, buffer, size, 0, addr, len);
|
||||||
DISABLE_TIMEOUT(socket);
|
|
||||||
UNMASK_SIGNAL();
|
UNMASK_SIGNAL();
|
||||||
|
|
||||||
/* Frees memory allocated from _GAddress_translate_to */
|
/* Frees memory allocated from _GAddress_translate_to */
|
||||||
free(addr);
|
free(addr);
|
||||||
|
|
||||||
if (ret == -1 && errno != EWOULDBLOCK) {
|
|
||||||
socket->m_error = GSOCK_IOERR;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
if (errno == EWOULDBLOCK) {
|
|
||||||
socket->m_error = GSOCK_WOULDBLOCK;
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -873,17 +876,22 @@ void _GSocket_Detected_Read(GSocket *socket)
|
|||||||
char c;
|
char c;
|
||||||
int ret;
|
int ret;
|
||||||
|
|
||||||
if (socket->m_stream) {
|
if (socket->m_stream)
|
||||||
|
{
|
||||||
ret = recv(socket->m_fd, &c, 1, MSG_PEEK);
|
ret = recv(socket->m_fd, &c, 1, MSG_PEEK);
|
||||||
|
|
||||||
if (ret < 0 && socket->m_server) {
|
if (ret < 0 && socket->m_server)
|
||||||
|
{
|
||||||
CALL_CALLBACK(socket, GSOCK_CONNECTION);
|
CALL_CALLBACK(socket, GSOCK_CONNECTION);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ret > 0) {
|
if (ret > 0)
|
||||||
|
{
|
||||||
CALL_CALLBACK(socket, GSOCK_INPUT);
|
CALL_CALLBACK(socket, GSOCK_INPUT);
|
||||||
} else {
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
CALL_CALLBACK(socket, GSOCK_LOST);
|
CALL_CALLBACK(socket, GSOCK_LOST);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -891,24 +899,34 @@ void _GSocket_Detected_Read(GSocket *socket)
|
|||||||
|
|
||||||
void _GSocket_Detected_Write(GSocket *socket)
|
void _GSocket_Detected_Write(GSocket *socket)
|
||||||
{
|
{
|
||||||
if (socket->m_establishing) {
|
if (socket->m_establishing && !socket->m_server)
|
||||||
|
{
|
||||||
int error, len;
|
int error, len;
|
||||||
|
|
||||||
len = sizeof(error);
|
|
||||||
|
|
||||||
socket->m_establishing = FALSE;
|
socket->m_establishing = FALSE;
|
||||||
|
|
||||||
|
len = sizeof(error);
|
||||||
getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, &error, &len);
|
getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, &error, &len);
|
||||||
|
|
||||||
if (error) {
|
if (error)
|
||||||
socket->m_error = GSOCK_IOERR;
|
{
|
||||||
GSocket_Shutdown(socket);
|
CALL_CALLBACK(socket, GSOCK_LOST);
|
||||||
} else
|
}
|
||||||
socket->m_error = GSOCK_NOERROR;
|
else
|
||||||
|
{
|
||||||
CALL_CALLBACK(socket, GSOCK_CONNECTION);
|
CALL_CALLBACK(socket, GSOCK_CONNECTION);
|
||||||
} else
|
/* We have to fire this event by hand because CONNECTION (for clients)
|
||||||
|
* and OUTPUT are internally the same and we just disabled CONNECTION
|
||||||
|
* events with the above macro.
|
||||||
|
*/
|
||||||
CALL_CALLBACK(socket, GSOCK_OUTPUT);
|
CALL_CALLBACK(socket, GSOCK_OUTPUT);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
CALL_CALLBACK(socket, GSOCK_OUTPUT);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* -------------------------------------------------------------------------
|
* -------------------------------------------------------------------------
|
||||||
|
Reference in New Issue
Block a user