961 lines
25 KiB
C
961 lines
25 KiB
C
|
/***************************************************************************
|
||
|
|
||
|
CSocket.c
|
||
|
|
||
|
Network component
|
||
|
|
||
|
(c) 2003-2004 Daniel Campos Fernández <danielcampos@netcourrier.com>
|
||
|
|
||
|
This is the implementation of Socket Gambas Class
|
||
|
|
||
|
This program is free software; you can redistribute it and/or modify
|
||
|
it under the terms of the GNU General Public License as published by
|
||
|
the Free Software Foundation; either version 1, or (at your option)
|
||
|
any later version.
|
||
|
|
||
|
This program is distributed in the hope that it will be useful,
|
||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
GNU General Public License for more details.
|
||
|
|
||
|
You should have received a copy of the GNU General Public License
|
||
|
along with this program; if not, write to the Free Software
|
||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||
|
|
||
|
***************************************************************************/
|
||
|
|
||
|
#define __CSOCKET_C
|
||
|
|
||
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <string.h>
|
||
|
#include <unistd.h>
|
||
|
#include <errno.h>
|
||
|
#include <sys/types.h>
|
||
|
#include <sys/socket.h>
|
||
|
#include <sys/ioctl.h>
|
||
|
#include <sys/poll.h>
|
||
|
#include <sys/un.h>
|
||
|
#include <netinet/in.h>
|
||
|
#include <arpa/inet.h>
|
||
|
#include <unistd.h>
|
||
|
#include <netdb.h>
|
||
|
#include <time.h>
|
||
|
|
||
|
#include "main.h"
|
||
|
#include "tools.h"
|
||
|
|
||
|
|
||
|
#include "CSocket.h"
|
||
|
#include "CServerSocket.h"
|
||
|
#include "CDnsClient.h"
|
||
|
#define MAX_CLIENT_BUFFER_SIZE 65536
|
||
|
#define UNIXPATHMAX 108
|
||
|
|
||
|
DECLARE_EVENT (SocketError);
|
||
|
DECLARE_EVENT (Closed);
|
||
|
DECLARE_EVENT (HostFound);
|
||
|
DECLARE_EVENT (Socket_Read);
|
||
|
DECLARE_EVENT (Connected);
|
||
|
|
||
|
|
||
|
GB_STREAM_DESC SocketStream = {
|
||
|
open: CSocket_stream_open,
|
||
|
close: CSocket_stream_close,
|
||
|
read: CSocket_stream_read,
|
||
|
write: CSocket_stream_write,
|
||
|
seek: CSocket_stream_seek,
|
||
|
tell: CSocket_stream_tell,
|
||
|
flush: CSocket_stream_flush,
|
||
|
eof: CSocket_stream_eof,
|
||
|
lof: CSocket_stream_lof,
|
||
|
handle: CSocket_stream_handle
|
||
|
};
|
||
|
/**********************************
|
||
|
Routines to call events
|
||
|
**********************************/
|
||
|
void CSocket_post_error(CSOCKET *mythis)
|
||
|
{
|
||
|
|
||
|
GB.Raise(mythis,SocketError,0);
|
||
|
GB.Unref(POINTER(&mythis));
|
||
|
}
|
||
|
|
||
|
void CSocket_post_closed(CSOCKET *mythis)
|
||
|
{
|
||
|
GB.Raise(mythis,Closed,0);
|
||
|
GB.Unref(POINTER(&mythis));
|
||
|
}
|
||
|
|
||
|
void CSocket_post_hostfound(CSOCKET *mythis)
|
||
|
{
|
||
|
GB.Raise(mythis,HostFound,0);
|
||
|
GB.Unref(POINTER(&mythis));
|
||
|
}
|
||
|
|
||
|
void CSocket_post_connected(CSOCKET *mythis)
|
||
|
{
|
||
|
GB.Raise(mythis,Connected,0);
|
||
|
GB.Unref(POINTER(&mythis));
|
||
|
}
|
||
|
|
||
|
void CSocket_post_data_available(CSOCKET *mythis)
|
||
|
{
|
||
|
if (mythis->iStatus==7) GB.Raise(mythis,Socket_Read,0);
|
||
|
GB.Unref(POINTER(&mythis));
|
||
|
}
|
||
|
|
||
|
/**************************************************
|
||
|
This function is called by DnsClient to inform
|
||
|
that it has finished its work
|
||
|
*************************************************/
|
||
|
void CSocket_CallBackFromDns(void *myobj)
|
||
|
{
|
||
|
int NoBlock=1;
|
||
|
int myval=0;
|
||
|
CSOCKET *mythis;
|
||
|
|
||
|
mythis=(CSOCKET*)myobj;
|
||
|
if ( mythis->iStatus != 5) return;
|
||
|
if ( !mythis->DnsTool->sHostIP)
|
||
|
{
|
||
|
mythis->iStatus=-6; /* error host not found */
|
||
|
dns_close_all(mythis->DnsTool);
|
||
|
GB.Unref(POINTER(&mythis->DnsTool));
|
||
|
mythis->DnsTool=NULL;
|
||
|
GB.Ref (mythis);
|
||
|
GB.Post (CSocket_post_error,(long)mythis);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
GB.FreeString (&mythis->sRemoteHostIP);
|
||
|
GB.NewString ( &mythis->sRemoteHostIP ,mythis->DnsTool->sHostIP,0);
|
||
|
/* Let's turn socket to async mode */
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
/* Third, we connect the socket */
|
||
|
mythis->Server.sin_family=AF_INET;
|
||
|
mythis->Server.sin_port=htons(mythis->iPort);
|
||
|
mythis->Server.sin_addr.s_addr =inet_addr(mythis->DnsTool->sHostIP);
|
||
|
bzero(&(mythis->Server.sin_zero),8);
|
||
|
myval=connect(mythis->Socket,(struct sockaddr*)&(mythis->Server), sizeof(struct sockaddr));
|
||
|
if (errno==EINPROGRESS) /* this is the good answer : connect in progress */
|
||
|
{
|
||
|
mythis->iStatus=6;
|
||
|
GB.Watch (mythis->Socket,GB_WATCH_WRITE,(void *)CSocket_CallBackConnecting,(long)mythis);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
GB.Watch (mythis->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
mythis->iStatus=0;
|
||
|
}
|
||
|
if (mythis->DnsTool)
|
||
|
{
|
||
|
dns_close_all(mythis->DnsTool);
|
||
|
GB.Unref(POINTER(&mythis->DnsTool));
|
||
|
mythis->DnsTool=NULL;
|
||
|
}
|
||
|
if ( mythis->iStatus<=0 )
|
||
|
{
|
||
|
mythis->iStatus=-3;
|
||
|
GB.Ref (mythis);
|
||
|
GB.Post (CSocket_post_error,(long)mythis);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
GB.Ref(mythis);
|
||
|
GB.Post(CSocket_post_hostfound,(long)mythis);
|
||
|
|
||
|
}
|
||
|
/*******************************************************************
|
||
|
This CallBack is used while waiting to finish a connection process
|
||
|
******************************************************************/
|
||
|
void CSocket_CallBackConnecting(int t_sock,int type,long lParam)
|
||
|
{
|
||
|
struct sockaddr_in myhost;
|
||
|
int mylen;
|
||
|
struct timespec mywait;
|
||
|
CSOCKET *mythis;
|
||
|
|
||
|
/* Just sleeping a little to reduce CPU waste */
|
||
|
mywait.tv_sec=0;
|
||
|
mywait.tv_nsec=1000000;
|
||
|
nanosleep(&mywait,NULL);
|
||
|
|
||
|
mythis=(CSOCKET*)lParam;
|
||
|
|
||
|
if (mythis->iStatus!=6) return;
|
||
|
/****************************************************
|
||
|
Checks if Connection was Stablished or there was
|
||
|
an error trying to connect
|
||
|
****************************************************/
|
||
|
mythis->iStatus=CheckConnection(mythis->Socket);
|
||
|
if (mythis->iStatus == 0)
|
||
|
{
|
||
|
GB.Watch (mythis->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
mythis->iStatus=-3;
|
||
|
GB.Ref (mythis);
|
||
|
GB.Post (CSocket_post_error,(long)mythis);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return;
|
||
|
}
|
||
|
if (mythis->iStatus != 7) return;
|
||
|
// we obtain local IP and host
|
||
|
mylen=sizeof(struct sockaddr);
|
||
|
getsockname (mythis->Socket,(struct sockaddr*)&myhost,(socklen_t *)&mylen);
|
||
|
mythis->iLocalPort=ntohs(myhost.sin_port);
|
||
|
GB.FreeString( &mythis->sLocalHostIP);
|
||
|
GB.NewString ( &mythis->sLocalHostIP ,inet_ntoa(myhost.sin_addr),0);
|
||
|
|
||
|
GB.Watch (mythis->Socket,GB_WATCH_NONE,(void *)CSocket_CallBack,(long)mythis);
|
||
|
GB.Watch (mythis->Socket,GB_WATCH_WRITE,(void *)CSocket_CallBack,(long)mythis);
|
||
|
|
||
|
mythis->stream.desc=&SocketStream;
|
||
|
mythis->stream._free[0]=(long)mythis;
|
||
|
GB.Ref(mythis);
|
||
|
GB.Post(CSocket_post_connected,(long)mythis);
|
||
|
|
||
|
|
||
|
|
||
|
}
|
||
|
/*******************************************************************
|
||
|
This CallBack is used while socket is connected to remote host
|
||
|
******************************************************************/
|
||
|
void CSocket_CallBack(int t_sock,int type,long lParam)
|
||
|
{
|
||
|
char buf[1];
|
||
|
struct pollfd mypoll;
|
||
|
int numpoll;
|
||
|
struct timespec mywait;
|
||
|
CSOCKET *mythis;
|
||
|
|
||
|
/* Just sleeping a little to reduce CPU waste */
|
||
|
mywait.tv_sec=0;
|
||
|
mywait.tv_nsec=100000;
|
||
|
nanosleep(&mywait,NULL);
|
||
|
/* is there data avilable or an error? */
|
||
|
mythis=(CSOCKET*)lParam;
|
||
|
|
||
|
if (mythis->iStatus!=7) return;
|
||
|
|
||
|
mypoll.fd=t_sock;
|
||
|
mypoll.events=POLLIN | POLLNVAL;
|
||
|
mypoll.revents=0;
|
||
|
numpoll=poll(&mypoll,1,0);
|
||
|
if (numpoll<=0) return;
|
||
|
/* there's data available */
|
||
|
USE_MSG_NOSIGNAL(numpoll=recv(t_sock,(void*)buf,sizeof(char),MSG_PEEK | MSG_NOSIGNAL));
|
||
|
if (!numpoll)
|
||
|
{ /* socket error, no valid data received */
|
||
|
GB.Watch (mythis->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(t_sock);
|
||
|
mythis->iStatus=0;
|
||
|
GB.Ref(mythis);
|
||
|
GB.Post(CSocket_post_closed,(long)mythis);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return;
|
||
|
}
|
||
|
/******************************************************
|
||
|
There's data available to read, so we'll raise event
|
||
|
Socket_Read
|
||
|
*******************************************************/
|
||
|
|
||
|
GB.Ref(mythis);
|
||
|
GB.Post(CSocket_post_data_available,(long)mythis);
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
void CSocket_stream_internal_error(CSOCKET *mythis,int ncode)
|
||
|
{
|
||
|
/* fatal socket error handling */
|
||
|
GB.Watch (mythis->Socket,GB_WATCH_NONE,(void *)CSocket_CallBack,0);
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
mythis->iStatus = ncode;
|
||
|
}
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////////
|
||
|
//################################################################################
|
||
|
/*********************************************************************************
|
||
|
"PUBLIC" C/C++ INTERFACE
|
||
|
**********************************************************************************/
|
||
|
//################################################################################
|
||
|
//////////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
/* not allowed methods */
|
||
|
int CSocket_stream_open(GB_STREAM *stream, const char *path, int mode, void *data){return -1;}
|
||
|
int CSocket_stream_seek(GB_STREAM *stream, long long pos, int whence){return -1;}
|
||
|
int CSocket_stream_tell(GB_STREAM *stream, long long *pos){return -1;}
|
||
|
int CSocket_stream_flush(GB_STREAM *stream)
|
||
|
{
|
||
|
return 0; /* OK */
|
||
|
}
|
||
|
int CSocket_stream_handle(GB_STREAM *stream)
|
||
|
{
|
||
|
return 0; /* OK */
|
||
|
}
|
||
|
int CSocket_stream_close(GB_STREAM *stream)
|
||
|
{
|
||
|
CSOCKET *mythis;
|
||
|
|
||
|
if (!(mythis=(CSOCKET*)stream->_free[0]) ) return -1;
|
||
|
|
||
|
if (mythis->DnsTool)
|
||
|
{
|
||
|
dns_close_all(mythis->DnsTool);
|
||
|
GB.Unref(POINTER(&mythis->DnsTool));
|
||
|
mythis->DnsTool=NULL;
|
||
|
}
|
||
|
if (mythis->iStatus > 0) /* if it's not connected, does nothing */
|
||
|
{
|
||
|
GB.Watch (mythis->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
stream->desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
mythis->iStatus=0;
|
||
|
}
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return 0;
|
||
|
}
|
||
|
int CSocket_stream_lof(GB_STREAM *stream, long long *len)
|
||
|
{
|
||
|
CSOCKET *mythis;
|
||
|
int bytes;
|
||
|
|
||
|
*len=0;
|
||
|
if (!(mythis=(CSOCKET*)stream->_free[0]) ) return -1;
|
||
|
|
||
|
if (ioctl(mythis->Socket,FIONREAD,&bytes))
|
||
|
{
|
||
|
CSocket_stream_internal_error(mythis,-4);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
*len=bytes;
|
||
|
return 0;
|
||
|
}
|
||
|
int CSocket_stream_eof(GB_STREAM *stream)
|
||
|
{
|
||
|
CSOCKET *mythis;
|
||
|
int bytes;
|
||
|
|
||
|
if (!(mythis=(CSOCKET*)stream->_free[0]) ) return -1;
|
||
|
|
||
|
if (ioctl(mythis->Socket,FIONREAD,&bytes))
|
||
|
{
|
||
|
CSocket_stream_internal_error(mythis,-4);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
if (!bytes) return -1;
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
int CSocket_stream_read(GB_STREAM *stream, char *buffer, long len)
|
||
|
{
|
||
|
CSOCKET *mythis;
|
||
|
int npos=-1;
|
||
|
int NoBlock=0;
|
||
|
int bytes;
|
||
|
|
||
|
if (!(mythis=(CSOCKET*)stream->_free[0]) ) return -1;
|
||
|
|
||
|
if (ioctl(mythis->Socket,FIONREAD,&bytes))
|
||
|
{
|
||
|
CSocket_stream_internal_error(mythis,-4);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
if (bytes < len) return -1;
|
||
|
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
USE_MSG_NOSIGNAL(npos=recv(mythis->Socket,(void*)buffer,len*sizeof(char),MSG_NOSIGNAL));
|
||
|
NoBlock++;
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
if (npos==len) return 0;
|
||
|
if (npos<0)
|
||
|
{
|
||
|
CSocket_stream_internal_error(mythis,-4);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
int CSocket_stream_write(GB_STREAM *stream, char *buffer, long len)
|
||
|
{
|
||
|
CSOCKET *mythis;
|
||
|
int npos=-1;
|
||
|
int NoBlock=0;
|
||
|
|
||
|
if (!(mythis=(CSOCKET*)stream->_free[0]) ) return -1;
|
||
|
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
USE_MSG_NOSIGNAL(npos=send(mythis->Socket,(void*)buffer,len*sizeof(char),MSG_NOSIGNAL));
|
||
|
NoBlock++;
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
if (npos>=0) return 0;
|
||
|
CSocket_stream_internal_error(mythis,-5);
|
||
|
if (mythis->OnClose) mythis->OnClose((void*)mythis);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
/**************************************************************************
|
||
|
To start a UNIX connection
|
||
|
**************************************************************************/
|
||
|
int CSocket_connect_unix(CSOCKET *mythis,char *sPath,int lenpath)
|
||
|
{
|
||
|
int NoBlock=1;
|
||
|
|
||
|
if ( mythis->iStatus > 0 ) return 1;
|
||
|
if (!sPath) return 7;
|
||
|
if ( (lenpath<1) || (lenpath>UNIXPATHMAX) ) return 7;
|
||
|
|
||
|
GB.FreeString(&mythis->sRemoteHostIP);
|
||
|
GB.FreeString(&mythis->sLocalHostIP);
|
||
|
|
||
|
mythis->UServer.sun_family=AF_UNIX;
|
||
|
strcpy(mythis->UServer.sun_path,sPath);
|
||
|
if ( (mythis->Socket=socket(AF_UNIX,SOCK_STREAM,0))==-1 )
|
||
|
{
|
||
|
mythis->iStatus=-2;
|
||
|
GB.Ref (mythis);
|
||
|
CSocket_post_error(mythis); /* Unable to create socket */
|
||
|
return 2;
|
||
|
}
|
||
|
|
||
|
|
||
|
GB.FreeString(&mythis->sPath);
|
||
|
GB.NewString ( &mythis->sPath , mythis->UServer.sun_path ,0);
|
||
|
mythis->conn_type=1;
|
||
|
if (connect(mythis->Socket,(struct sockaddr*)&mythis->UServer,sizeof(struct sockaddr_un))==0)
|
||
|
{
|
||
|
mythis->iStatus=7;
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
GB.Watch (mythis->Socket,GB_WATCH_WRITE,(void *)CSocket_CallBack,(long)mythis);
|
||
|
mythis->stream.desc=&SocketStream;
|
||
|
mythis->stream._free[0]=(long)mythis;
|
||
|
// $BM
|
||
|
if (mythis->Host) GB.FreeString(&mythis->Host);
|
||
|
if (mythis->Path) GB.FreeString(&mythis->Path);
|
||
|
|
||
|
GB.NewString(&mythis->Path,sPath,0);
|
||
|
GB.Ref (mythis);
|
||
|
CSocket_post_connected(mythis);
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
/* Error */
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
GB.FreeString(&mythis->sPath);
|
||
|
mythis->iStatus=-3;
|
||
|
|
||
|
GB.Ref (mythis);
|
||
|
CSocket_post_error(mythis); /* Unable to connect to remote host */
|
||
|
|
||
|
return 3;
|
||
|
|
||
|
}
|
||
|
/**************************************************************************
|
||
|
To start a TCP connection
|
||
|
**************************************************************************/
|
||
|
int CSocket_connect_socket(CSOCKET *mythis,char *sHost,int lenhost,int myport)
|
||
|
{
|
||
|
|
||
|
if ( mythis->iStatus > 0 ) return 1;
|
||
|
if (!lenhost) return 9;
|
||
|
if (!sHost) return 9;
|
||
|
if ( (myport<1) || (myport>65535) ) return 8;
|
||
|
|
||
|
GB.FreeString(&mythis->sRemoteHostIP);
|
||
|
GB.FreeString(&mythis->sLocalHostIP);
|
||
|
|
||
|
|
||
|
if ( (mythis->Socket=socket(AF_INET,SOCK_STREAM,0))==-1 )
|
||
|
{
|
||
|
mythis->iStatus=-2;
|
||
|
GB.Ref (mythis);
|
||
|
CSocket_post_error(mythis);
|
||
|
return 2;
|
||
|
}
|
||
|
|
||
|
mythis->iPort=myport;
|
||
|
mythis->conn_type=0;
|
||
|
/******************************************
|
||
|
Let's turn hostname into host IP
|
||
|
*******************************************/
|
||
|
if (!mythis->DnsTool)
|
||
|
{
|
||
|
GB.New(POINTER(&mythis->DnsTool),GB.FindClass("DnsClient"),NULL,NULL);
|
||
|
mythis->DnsTool->CliParent=(void*)mythis;
|
||
|
}
|
||
|
|
||
|
if (mythis->DnsTool->iStatus > 0 ) dns_close_all(mythis->DnsTool);
|
||
|
|
||
|
dns_set_async_mode(1,mythis->DnsTool);
|
||
|
GB.FreeString (&(mythis->DnsTool->sHostName));
|
||
|
GB.NewString(&mythis->DnsTool->sHostName,sHost,lenhost);
|
||
|
mythis->DnsTool->finished_callback=CSocket_CallBackFromDns;
|
||
|
/********************************************
|
||
|
We start DNS lookup, when it is finished
|
||
|
it will call to CSocket_CallBack_fromDns,
|
||
|
and we'll continue there connection proccess
|
||
|
********************************************/
|
||
|
mythis->iStatus=5; /* looking for IP */
|
||
|
dns_thread_getip(mythis->DnsTool);
|
||
|
mythis->stream.desc=&SocketStream;
|
||
|
mythis->stream._free[0]=(long)mythis;
|
||
|
mythis->iUsePort=mythis->iPort;
|
||
|
|
||
|
// $BM
|
||
|
if (mythis->Host) GB.FreeString(&mythis->Host);
|
||
|
if (mythis->Path) GB.FreeString(&mythis->Path);
|
||
|
|
||
|
GB.NewString(&mythis->Host,sHost,0);
|
||
|
return 0;
|
||
|
|
||
|
}
|
||
|
|
||
|
/**********************************************
|
||
|
This function is used to peek data from socket,
|
||
|
you have to pass 3 parameters:
|
||
|
|
||
|
mythis -> CSocket object
|
||
|
buf -> Data Buffer (you have to free it after using it!)
|
||
|
MaxLen -> 0 no limit, >0 max. data t read
|
||
|
***********************************************/
|
||
|
int CSocket_peek_data(CSOCKET *mythis,char **buf,int MaxLen)
|
||
|
{
|
||
|
int retval=0;
|
||
|
int NoBlock=0;
|
||
|
int nread=0;
|
||
|
int bytes=0;
|
||
|
|
||
|
(*buf)=NULL;
|
||
|
nread=ioctl(mythis->Socket,FIONREAD,&bytes); /* Is there anythig to read? */
|
||
|
if (nread)
|
||
|
retval=-1;
|
||
|
else
|
||
|
retval=0;
|
||
|
if ( (!retval) && (bytes) ) /* if there's anything to receive */
|
||
|
{
|
||
|
if (bytes > MAX_CLIENT_BUFFER_SIZE) bytes=MAX_CLIENT_BUFFER_SIZE;
|
||
|
if (MaxLen >0 ) bytes=MaxLen;
|
||
|
GB.Alloc((void**)buf,bytes);
|
||
|
(*buf)[0]='\0';
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
USE_MSG_NOSIGNAL(retval=recv(mythis->Socket,(void*)(*buf),bytes*sizeof(char),MSG_PEEK|MSG_NOSIGNAL));
|
||
|
NoBlock++;
|
||
|
ioctl(mythis->Socket,FIONBIO,&NoBlock);
|
||
|
}
|
||
|
|
||
|
if (retval==-1)
|
||
|
{
|
||
|
/* An error happened while trying to receive data : SOCKET ERROR */
|
||
|
if (*buf)
|
||
|
{
|
||
|
GB.Free(POINTER(buf));
|
||
|
buf=NULL;
|
||
|
}
|
||
|
GB.Watch (mythis->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
mythis->stream.desc=NULL;
|
||
|
close(mythis->Socket);
|
||
|
mythis->iStatus=-4;
|
||
|
GB.Ref (mythis);
|
||
|
CSocket_post_error(mythis);
|
||
|
return -1;
|
||
|
|
||
|
}
|
||
|
|
||
|
return retval;
|
||
|
|
||
|
}
|
||
|
/*****************************************************************************************
|
||
|
##########################################################################################
|
||
|
-------------------------SOCKET GAMBAS INTERFACE IMPLEMENTATION---------------------
|
||
|
##########################################################################################
|
||
|
******************************************************************************************/
|
||
|
/********************************************************************
|
||
|
Returns current Status of the socket (connected,connecting,closed)
|
||
|
*********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_Status )
|
||
|
|
||
|
GB.ReturnInteger(THIS->iStatus);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
/********************************************************************
|
||
|
Port to connect to. Can be 'Net.Local' for Unix sockets, or
|
||
|
1-65535 for TCP sockets
|
||
|
********************************************************************/
|
||
|
BEGIN_PROPERTY(CSOCKET_Port)
|
||
|
|
||
|
if (READ_PROPERTY)
|
||
|
{
|
||
|
GB.ReturnInteger(THIS->iUsePort);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (THIS->iStatus>0)
|
||
|
{
|
||
|
GB.Error("Port property can not be changed while working");
|
||
|
return;
|
||
|
}
|
||
|
if (VPROP(GB_INTEGER)<0) { THIS->iUsePort=0; return; }
|
||
|
if (VPROP(GB_INTEGER)>65535) { THIS->iUsePort=65535; return; }
|
||
|
THIS->iUsePort=VPROP(GB_INTEGER);
|
||
|
|
||
|
|
||
|
END_PROPERTY
|
||
|
/*********************************************************************
|
||
|
Host or Path to connect to. If 'Port' value is zero, this
|
||
|
will be a local path, else, a remote host name
|
||
|
**********************************************************************/
|
||
|
// $BM
|
||
|
|
||
|
BEGIN_PROPERTY (CSOCKET_Host)
|
||
|
|
||
|
if (READ_PROPERTY)
|
||
|
{
|
||
|
GB.ReturnNewString(THIS->Host,0);
|
||
|
return;
|
||
|
}
|
||
|
GB.StoreString(PROP(GB_STRING), &THIS->Host);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
BEGIN_PROPERTY (CSOCKET_Path)
|
||
|
|
||
|
if (READ_PROPERTY)
|
||
|
{
|
||
|
GB.ReturnNewString(THIS->Path,0);
|
||
|
return;
|
||
|
}
|
||
|
GB.StoreString(PROP(GB_STRING), &THIS->Path);
|
||
|
|
||
|
END_PROPERTY
|
||
|
/********************************************************************
|
||
|
Returns current TCP remote port (only when connected via TCP)
|
||
|
*********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_RemotePort )
|
||
|
|
||
|
if ( THIS->iStatus != 7 ) { GB.ReturnInteger(0); return; }
|
||
|
if ( THIS->conn_type != 0 ) { GB.ReturnInteger(0); return; }
|
||
|
GB.ReturnInteger(THIS->iPort);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
/********************************************************************
|
||
|
Returns current TCP local port (only when connected via TCP)
|
||
|
*********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_LocalPort )
|
||
|
|
||
|
if ( THIS->iStatus != 7 ) { GB.ReturnInteger(0); return; }
|
||
|
if ( THIS->conn_type != 0 ) { GB.ReturnInteger(0); return; }
|
||
|
GB.ReturnInteger(THIS->iLocalPort);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
/***********************************************************************
|
||
|
Returns current foreing host IP (only when connected via TCP)
|
||
|
***********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_RemoteHost )
|
||
|
|
||
|
if ( THIS->iStatus==7 && (THIS->conn_type==0) )
|
||
|
{
|
||
|
GB.ReturnString(THIS->sRemoteHostIP);
|
||
|
return;
|
||
|
}
|
||
|
GB.ReturnString(NULL);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
/***********************************************************************
|
||
|
Returns current local host IP (only when connected via TCP)
|
||
|
***********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_LocalHost )
|
||
|
|
||
|
if ( THIS->iStatus==7 && (THIS->conn_type==0) )
|
||
|
{
|
||
|
GB.ReturnString(THIS->sLocalHostIP);
|
||
|
return;
|
||
|
}
|
||
|
GB.ReturnString(NULL);
|
||
|
|
||
|
END_PROPERTY
|
||
|
|
||
|
// $BM
|
||
|
#if 0
|
||
|
/***********************************************************************
|
||
|
Returns current Path (only when connected via UNIX socket)
|
||
|
***********************************************************************/
|
||
|
BEGIN_PROPERTY ( CSOCKET_Path )
|
||
|
|
||
|
int mylen;
|
||
|
char numtmp[6];
|
||
|
char *strtmp;
|
||
|
|
||
|
if ( (THIS->iStatus==7) && (THIS->conn_type==1) )
|
||
|
{
|
||
|
GB.ReturnString(THIS->sPath);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if ( (THIS->iStatus==7) && (THIS->conn_type==0) )
|
||
|
{
|
||
|
snprintf(numtmp,6,"%d",THIS->iPort);
|
||
|
mylen=strlen(THIS->sRemoteHostIP);
|
||
|
mylen++;
|
||
|
mylen+=strlen(numtmp);
|
||
|
mylen++;
|
||
|
GB.Alloc((void**)&strtmp,sizeof(char)*mylen);
|
||
|
strcpy(strtmp,THIS->sRemoteHostIP);
|
||
|
strcat(strtmp,":");
|
||
|
strcat(strtmp,numtmp);
|
||
|
GB.ReturnNewString(strtmp,0);
|
||
|
GB.Free((void**)&strtmp);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
GB.ReturnString(NULL);
|
||
|
|
||
|
END_PROPERTY
|
||
|
#endif
|
||
|
|
||
|
/****************************************************
|
||
|
Gambas object "Constructor"
|
||
|
****************************************************/
|
||
|
//BEGIN_METHOD(CSOCKET_new,GB_STRING sPath;)
|
||
|
BEGIN_METHOD_VOID(CSOCKET_new)
|
||
|
|
||
|
//char *buf=NULL;
|
||
|
//int retval;
|
||
|
//int nport=0;
|
||
|
|
||
|
// $BM - The object is filled with zero at creation
|
||
|
|
||
|
//THIS->stream.desc=NULL;
|
||
|
//THIS->c_parent=NULL;
|
||
|
//THIS->DnsTool=NULL;
|
||
|
//THIS->conn_type=0;
|
||
|
//THIS->iStatus=0;
|
||
|
THIS->iUsePort=80;
|
||
|
//THIS->sLocalHostIP=NULL;
|
||
|
//THIS->sRemoteHostIP=NULL;
|
||
|
//THIS->HostOrPath=NULL;
|
||
|
//THIS->OnClose=NULL;
|
||
|
|
||
|
/*
|
||
|
if (MISSING (sPath) ) return;
|
||
|
if (!STRING (sPath) ) return;
|
||
|
|
||
|
retval=IsHostPath(STRING(sPath),&buf,&nport);
|
||
|
|
||
|
if (!retval)
|
||
|
{
|
||
|
GB.Error("Invalid Host / Path string");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (retval==2)
|
||
|
{
|
||
|
switch(CSocket_connect_unix(THIS,STRING(sPath),LENGTH(sPath)))
|
||
|
{
|
||
|
case 1: GB.Error("Socket is already connected. Close it first."); return;
|
||
|
case 2: GB.Error("Invalid Path length"); return;
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (retval==1)
|
||
|
{
|
||
|
if (!buf)
|
||
|
{
|
||
|
GB.Error("Invalid host name");
|
||
|
return;
|
||
|
}
|
||
|
if (nport < 1)
|
||
|
{
|
||
|
GB.Error("Invalid Port");
|
||
|
return;
|
||
|
}
|
||
|
retval=CSocket_connect_socket(THIS,buf,strlen(buf),nport );
|
||
|
GB.Free((void**)&buf);
|
||
|
switch(retval)
|
||
|
{
|
||
|
case 1: GB.Error("Socket is already connected. Close it first."); return;
|
||
|
case 8: GB.Error("Port value out of range."); return;
|
||
|
case 9: GB.Error("Invalid Host Name."); return;
|
||
|
}
|
||
|
return;
|
||
|
}
|
||
|
*/
|
||
|
|
||
|
END_METHOD
|
||
|
|
||
|
/**************************************************
|
||
|
Gambas object "Destructor"
|
||
|
**************************************************/
|
||
|
BEGIN_METHOD_VOID(CSOCKET_free)
|
||
|
|
||
|
if (THIS->DnsTool)
|
||
|
{
|
||
|
dns_close_all(THIS->DnsTool);
|
||
|
GB.Unref(POINTER(&THIS->DnsTool));
|
||
|
THIS->DnsTool=NULL;
|
||
|
}
|
||
|
if (THIS->iStatus>0)
|
||
|
{
|
||
|
GB.Watch (THIS->Socket , GB_WATCH_NONE , (void *)CSocket_CallBack,0);
|
||
|
THIS->stream.desc=NULL;
|
||
|
close(THIS->Socket);
|
||
|
if (THIS->OnClose) THIS->OnClose((void*)THIS);
|
||
|
THIS->iStatus=0;
|
||
|
|
||
|
}
|
||
|
GB.FreeString(&THIS->sPath);
|
||
|
GB.FreeString(&THIS->sLocalHostIP);
|
||
|
GB.FreeString(&THIS->sRemoteHostIP);
|
||
|
// $BM
|
||
|
GB.FreeString(&THIS->Host);
|
||
|
GB.FreeString(&THIS->Path);
|
||
|
|
||
|
END_METHOD
|
||
|
|
||
|
BEGIN_METHOD_VOID(CSOCKET_exit)
|
||
|
|
||
|
END_METHOD
|
||
|
|
||
|
|
||
|
/*************************************************************
|
||
|
To Peek data arrived from the other side of the socket
|
||
|
**************************************************************/
|
||
|
BEGIN_METHOD_VOID(CSOCKET_Peek)
|
||
|
|
||
|
char *buf=NULL;
|
||
|
int retval=0;
|
||
|
|
||
|
if (THIS->iStatus != 7) /* if socket is not connected we can't receive anything */
|
||
|
{
|
||
|
GB.Error("Socket is not active. Connect it first.");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
retval=CSocket_peek_data(THIS,&buf,0);
|
||
|
|
||
|
if (retval==-1)
|
||
|
{
|
||
|
/* An error happened while trying to receive data : SOCKET ERROR */
|
||
|
if (buf) GB.Free(POINTER(&buf));
|
||
|
GB.ReturnNewString(NULL,0);
|
||
|
return;
|
||
|
|
||
|
}
|
||
|
|
||
|
if ( retval>0)
|
||
|
GB.ReturnNewString (buf,retval);
|
||
|
else
|
||
|
GB.ReturnNewString (NULL,0);
|
||
|
|
||
|
if (buf) GB.Free(POINTER(&buf));
|
||
|
|
||
|
END_METHOD
|
||
|
|
||
|
|
||
|
|
||
|
/**************************************************************************
|
||
|
To start a TCP or UNIX connection
|
||
|
**************************************************************************/
|
||
|
|
||
|
// $BM
|
||
|
|
||
|
BEGIN_METHOD(CSOCKET_Connect,GB_STRING HostOrPath;GB_INTEGER Port;)
|
||
|
|
||
|
int port;
|
||
|
int err;
|
||
|
|
||
|
port = VARGOPT(Port, THIS->iUsePort);
|
||
|
|
||
|
if (!port)
|
||
|
{
|
||
|
if (MISSING(HostOrPath))
|
||
|
err = CSocket_connect_unix(THIS,THIS->Path,GB.StringLength(THIS->Path));
|
||
|
else
|
||
|
err = CSocket_connect_unix(THIS,STRING(HostOrPath),LENGTH(HostOrPath));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (MISSING(HostOrPath))
|
||
|
err = CSocket_connect_socket(THIS,THIS->Host,GB.StringLength(THIS->Host),port);
|
||
|
else
|
||
|
err = CSocket_connect_socket(THIS,STRING(HostOrPath),LENGTH(HostOrPath),port);
|
||
|
}
|
||
|
|
||
|
switch (err)
|
||
|
{
|
||
|
case 1: GB.Error("Socket is already connected"); return;
|
||
|
case 2: GB.Error("Invalid path length"); return;
|
||
|
case 8: GB.Error("Port value out of range."); return;
|
||
|
case 9: GB.Error("Invalid host name."); return;
|
||
|
}
|
||
|
|
||
|
END_METHOD
|
||
|
|
||
|
|
||
|
/**********************************************************
|
||
|
Here we declare public structure of Socket Class
|
||
|
***********************************************************/
|
||
|
|
||
|
GB_DESC CSocketDesc[] =
|
||
|
{
|
||
|
|
||
|
GB_DECLARE("Socket", sizeof(CSOCKET)),
|
||
|
|
||
|
GB_INHERITS("Stream"),
|
||
|
|
||
|
|
||
|
GB_EVENT("Error", NULL, NULL, &SocketError),
|
||
|
GB_EVENT("Read", NULL, NULL, &Socket_Read),
|
||
|
GB_EVENT("Ready", NULL, NULL, &Connected),
|
||
|
GB_EVENT("Closed", NULL, NULL, &Closed),
|
||
|
GB_EVENT("Found", NULL, NULL, &HostFound),
|
||
|
|
||
|
GB_STATIC_METHOD ( "_exit" , NULL , CSOCKET_exit , NULL ),
|
||
|
|
||
|
//GB_METHOD("_new", NULL, CSOCKET_new, "[(Path)s]"),
|
||
|
GB_METHOD("_new", NULL, CSOCKET_new, NULL),
|
||
|
GB_METHOD("_free", NULL, CSOCKET_free, NULL),
|
||
|
GB_METHOD("Peek", "s", CSOCKET_Peek, NULL),
|
||
|
GB_METHOD("Connect",NULL, CSOCKET_Connect,"[(HostOrPath)s(Port)i]"),
|
||
|
|
||
|
GB_PROPERTY_READ("Status", "i", CSOCKET_Status),
|
||
|
GB_PROPERTY_READ("RemotePort", "i", CSOCKET_RemotePort),
|
||
|
GB_PROPERTY_READ("LocalPort", "i", CSOCKET_LocalPort),
|
||
|
GB_PROPERTY_READ("RemoteHost", "s", CSOCKET_RemoteHost),
|
||
|
GB_PROPERTY_READ("LocalHost","s",CSOCKET_LocalHost),
|
||
|
|
||
|
GB_PROPERTY("Host","s",CSOCKET_Host),
|
||
|
GB_PROPERTY("Path","s",CSOCKET_Path),
|
||
|
GB_PROPERTY("Port","i",CSOCKET_Port),
|
||
|
|
||
|
GB_CONSTANT("_Properties", "s", "Host,Path,Port=80"),
|
||
|
GB_CONSTANT("_DefaultEvent", "s", "Read"),
|
||
|
|
||
|
GB_END_DECLARE
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
|