Main Page | Namespace List | Class Hierarchy | Alphabetical List | Class List | Directories | File List | Namespace Members | Class Members | File Members

InspSocket Class Reference

InspSocket is an extendable socket class which modules can use for TCP socket support. More...

#include <socket.h>

Collaboration diagram for InspSocket:

Collaboration graph
[legend]
List of all members.

Public Member Functions

 InspSocket ()
 The default constructor does nothing and should not be used.
 InspSocket (int newfd, char *ip)
 This constructor is used to associate an existing connecting with an InspSocket class.
 InspSocket (std::string host, int port, bool listening, unsigned long maxtime)
 This constructor is used to create a new socket, either listening for connections, or an outbound connection to another host.
virtual bool OnConnected ()
 This method is called when an outbound connection on your socket is completed.
virtual void OnError (InspSocketError e)
 This method is called when an error occurs.
virtual int OnDisconnect ()
 When an established connection is terminated, the OnDisconnect method is triggered.
virtual bool OnDataReady ()
 When there is data waiting to be read on a socket, the OnDataReady() method is called.
virtual void OnTimeout ()
 When an outbound connection fails, and the attempt times out, you will receive this event.
virtual void OnClose ()
 Whenever close() is called, OnClose() will be called first.
virtual char * Read ()
 Reads all pending bytes from the socket into a char* array which can be up to 16 kilobytes in length.
std::string GetIP ()
 Returns the IP address associated with this connection, or an empty string if no IP address exists.
bool Timeout (time_t current)
 This function checks if the socket has timed out yet, given the current time in the parameter.
virtual int Write (std::string data)
 Writes a std::string to the socket.
virtual int OnIncomingConnection (int newfd, char *ip)
 If your socket is a listening socket, when a new connection comes in on the socket this method will be called.
void SetState (InspSocketState s)
 Changes the socket's state.
InspSocketState GetState ()
 Returns the current socket state.
bool Poll ()
 Only the core should call this function.
int GetFd ()
 This method returns the socket's file descriptor as assigned by the operating system, or -1 if no descriptor has been assigned.
virtual void Close ()
 This method causes the socket to close, and may also be triggered by other methods such as OnTimeout and OnError.
virtual ~InspSocket ()
 The destructor may implicitly call OnClose(), and will close() and shutdown() the file descriptor used for this socket.

Private Attributes

int fd
 The file descriptor of this socket.
std::string host
 The hostname connected to.
int port
 The port connected to, or the port this socket is listening on.
InspSocketState state
 The state for this socket, either listening, connecting, connected or error.
sockaddr_in addr
 The host being connected to, in sockaddr form.
in_addr addy
 The host being connected to, in in_addr form.
time_t timeout_end
 When this time is reached, the socket times out if it is in the CONNECTING state.
bool timeout
 This value is true if the socket has timed out.
char ibuf [16384]
 Socket input buffer, used by read().
std::string IP
 The IP address being connected to stored in string form for easy retrieval by accessors.
sockaddr_in client
 Client sockaddr structure used by accept().
sockaddr_in server
 Server sockaddr structure used by accept().
socklen_t length
 Used by accept() to indicate the sizes of the sockaddr_in structures.

Detailed Description

InspSocket is an extendable socket class which modules can use for TCP socket support.

It is fully integrated into InspIRCds socket loop and attaches its sockets to the core's instance of the SocketEngine class, meaning that any sockets you create have the same power and abilities as a socket created by the core itself. To use InspSocket, you must inherit a class from it, and use the InspSocket constructors to establish connections and bindings.

Definition at line 47 of file socket.h.


Constructor & Destructor Documentation

InspSocket::InspSocket  ) 
 

The default constructor does nothing and should not be used.

Definition at line 49 of file socket.cpp.

References I_DISCONNECTED, and state.

00050 {
00051         this->state = I_DISCONNECTED;
00052 }

InspSocket::InspSocket int  newfd,
char *  ip
 

This constructor is used to associate an existing connecting with an InspSocket class.

The given file descriptor must be valid, and when initialized, the InspSocket will be set with the given IP address and placed in CONNECTED state.

Definition at line 54 of file socket.cpp.

References SocketEngine::AddFd(), fd, I_CONNECTED, IP, state, and X_ESTAB_MODULE.

00055 {
00056         this->fd = newfd;
00057         this->state = I_CONNECTED;
00058         this->IP = ip;
00059         SE->AddFd(this->fd,true,X_ESTAB_MODULE);
00060 }

InspSocket::InspSocket std::string  host,
int  port,
bool  listening,
unsigned long  maxtime
 

This constructor is used to create a new socket, either listening for connections, or an outbound connection to another host.

Parameters:
host The hostname to connect to, or bind to
port The port number to connect to, or bind to
listening true to listen on the given host:port pair, or false to connect to them
maxtime Number of seconds to wait, if connecting, before the connection times out and an OnTimeout() event is generated

Definition at line 62 of file socket.cpp.

References SocketEngine::AddFd(), addr, addy, Close(), DEBUG, fd, I_CONNECTING, I_ERR_BIND, I_ERR_CONNECT, I_ERR_SOCKET, I_ERROR, I_LISTENING, IP, OnError(), state, timeout, timeout_end, and X_ESTAB_MODULE.

00063 {
00064         if (listening) {
00065                 if ((this->fd = OpenTCPSocket()) == ERROR)
00066                 {
00067                         this->fd = -1;
00068                         this->state = I_ERROR;
00069                         this->OnError(I_ERR_SOCKET);
00070                         log(DEBUG,"OpenTCPSocket() error");
00071                         return;
00072                 }
00073                 else
00074                 {
00075                         if (BindSocket(this->fd,this->client,this->server,port,(char*)host.c_str()) == ERROR)
00076                         {
00077                                 this->Close();
00078                                 this->fd = -1;
00079                                 this->state = I_ERROR;
00080                                 this->OnError(I_ERR_BIND);
00081                                 log(DEBUG,"BindSocket() error %s",strerror(errno));
00082                                 return;
00083                         }
00084                         else
00085                         {
00086                                 this->state = I_LISTENING;
00087                                 SE->AddFd(this->fd,true,X_ESTAB_MODULE);
00088                                 log(DEBUG,"New socket now in I_LISTENING state");
00089                                 return;
00090                         }
00091                 }                       
00092         } else {
00093                 char* ip;
00094                 this->host = host;
00095                 hostent* hoste = gethostbyname(host.c_str());
00096                 if (!hoste) {
00097                         ip = (char*)host.c_str();
00098                 } else {
00099                         struct in_addr* ia = (in_addr*)hoste->h_addr;
00100                         ip = inet_ntoa(*ia);
00101                 }
00102 
00103                 this->IP = ip;
00104 
00105                 timeout_end = time(NULL)+maxtime;
00106                 timeout = false;
00107                 if ((this->fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
00108                 {
00109                         this->state = I_ERROR;
00110                         this->OnError(I_ERR_SOCKET);
00111                         return;
00112                 }
00113                 this->port = port;
00114                 inet_aton(ip,&addy);
00115                 addr.sin_family = AF_INET;
00116                 addr.sin_addr = addy;
00117                 addr.sin_port = htons(this->port);
00118 
00119                 int flags;
00120                 flags = fcntl(this->fd, F_GETFL, 0);
00121                 fcntl(this->fd, F_SETFL, flags | O_NONBLOCK);
00122 
00123                 if(connect(this->fd, (sockaddr*)&this->addr,sizeof(this->addr)) == -1)
00124                 {
00125                         if (errno != EINPROGRESS)
00126                         {
00127                                 this->Close();
00128                                 this->OnError(I_ERR_CONNECT);
00129                                 this->state = I_ERROR;
00130                                 return;
00131                         }
00132                 }
00133                 this->state = I_CONNECTING;
00134                 SE->AddFd(this->fd,false,X_ESTAB_MODULE);
00135                 return;
00136         }
00137 }

InspSocket::~InspSocket  )  [virtual]
 

The destructor may implicitly call OnClose(), and will close() and shutdown() the file descriptor used for this socket.

Definition at line 269 of file socket.cpp.

References Close().

00270 {
00271         this->Close();
00272 }


Member Function Documentation

void InspSocket::Close  )  [virtual]
 

This method causes the socket to close, and may also be triggered by other methods such as OnTimeout and OnError.

Definition at line 139 of file socket.cpp.

References fd, and OnClose().

Referenced by InspSocket(), and ~InspSocket().

00140 {
00141         if (this->fd != -1)
00142         {
00143                 this->OnClose();
00144                 shutdown(this->fd,2);
00145                 close(this->fd);
00146                 this->fd = -1;
00147         }
00148 }

int InspSocket::GetFd  ) 
 

This method returns the socket's file descriptor as assigned by the operating system, or -1 if no descriptor has been assigned.

Definition at line 256 of file socket.cpp.

References fd.

00257 {
00258         return this->fd;
00259 }

std::string InspSocket::GetIP  ) 
 

Returns the IP address associated with this connection, or an empty string if no IP address exists.

Definition at line 150 of file socket.cpp.

References IP.

00151 {
00152         return this->IP;
00153 }

InspSocketState InspSocket::GetState  ) 
 

Returns the current socket state.

Definition at line 251 of file socket.cpp.

References state.

00252 {
00253         return this->state;
00254 }

void InspSocket::OnClose  )  [virtual]
 

Whenever close() is called, OnClose() will be called first.

Please note that this means OnClose will be called alongside OnError(), OnTimeout(), and Close(), and also when cancelling a listening socket by calling the destructor indirectly.

Definition at line 267 of file socket.cpp.

Referenced by Close().

00267 { return; }

bool InspSocket::OnConnected  )  [virtual]
 

This method is called when an outbound connection on your socket is completed.

Returns:
false to abort the connection, true to continue

Definition at line 261 of file socket.cpp.

Referenced by Poll().

00261 { return true; }

bool InspSocket::OnDataReady  )  [virtual]
 

When there is data waiting to be read on a socket, the OnDataReady() method is called.

Within this method, you *MUST* call the Read() method to read any pending data. At its lowest level, this event is signalled by the core via the socket engine. If you return false from this function, the core removes your socket from its list and erases it from the socket engine, then calls InspSocket::Close() and deletes it.

Returns:
false to close the socket

Definition at line 265 of file socket.cpp.

Referenced by Poll().

00265 { return true; }

int InspSocket::OnDisconnect  )  [virtual]
 

When an established connection is terminated, the OnDisconnect method is triggered.

Definition at line 263 of file socket.cpp.

00263 { return 0; }

void InspSocket::OnError InspSocketError  e  )  [virtual]
 

This method is called when an error occurs.

A closed socket in itself is not an error, however errors also generate close events.

Parameters:
e The error type which occured

Definition at line 262 of file socket.cpp.

Referenced by InspSocket(), and Timeout().

00262 { return; }

int InspSocket::OnIncomingConnection int  newfd,
char *  ip
[virtual]
 

If your socket is a listening socket, when a new connection comes in on the socket this method will be called.

Given the new file descriptor in the parameters, and the IP, it is recommended you copy them to a new instance of your socket class, e.g.:

MySocket* newsocket = new MySocket(newfd,ip);

Once you have done this, you can then associate the new socket with the core using Server::AddSocket().

Definition at line 264 of file socket.cpp.

Referenced by Poll().

00264 { return 0; }

void InspSocket::OnTimeout  )  [virtual]
 

When an outbound connection fails, and the attempt times out, you will receive this event.

The mthod will trigger once maxtime secons are reached (as given in the constructor) just before the socket's descriptor is closed.

Definition at line 266 of file socket.cpp.

Referenced by Timeout().

00266 { return; }

bool InspSocket::Poll  ) 
 

Only the core should call this function.

When called, it is assumed the socket is ready to read data, and the method call routes the event to the various methods of InspSocket for you to handle. This can also cause the socket's state to change.

Definition at line 214 of file socket.cpp.

References SocketEngine::AddFd(), client, SocketEngine::DelFd(), I_CONNECTED, I_CONNECTING, I_LISTENING, length, OnConnected(), OnDataReady(), OnIncomingConnection(), SetState(), and X_ESTAB_MODULE.

00215 {
00216         int incoming = -1;
00217         
00218         switch (this->state)
00219         {
00220                 case I_CONNECTING:
00221                         this->SetState(I_CONNECTED);
00222                         /* Our socket was in write-state, so delete it and re-add it
00223                          * in read-state.
00224                          */
00225                         SE->DelFd(this->fd);
00226                         SE->AddFd(this->fd,true,X_ESTAB_MODULE);
00227                         return this->OnConnected();
00228                 break;
00229                 case I_LISTENING:
00230                         length = sizeof (client);
00231                         incoming = accept (this->fd, (sockaddr*)&client,&length);
00232                         this->OnIncomingConnection(incoming,inet_ntoa(client.sin_addr));
00233                         return true;
00234                 break;
00235                 case I_CONNECTED:
00236                         return this->OnDataReady();
00237                 break;
00238                 default:
00239                 break;
00240         }
00241 
00242         return true;
00243 }

char * InspSocket::Read  )  [virtual]
 

Reads all pending bytes from the socket into a char* array which can be up to 16 kilobytes in length.

Definition at line 155 of file socket.cpp.

References DEBUG, and ibuf.

00156 {
00157         int n = recv(this->fd,this->ibuf,sizeof(this->ibuf),0);
00158         if (n > 0)
00159         {
00160                 ibuf[n] = 0;
00161                 return ibuf;
00162         }
00163         else
00164         {
00165                 log(DEBUG,"EOF or error on socket");
00166                 return NULL;
00167         }
00168 }

void InspSocket::SetState InspSocketState  s  ) 
 

Changes the socket's state.

The core uses this to change socket states, and you should not call it directly.

Definition at line 245 of file socket.cpp.

References DEBUG, and state.

Referenced by Poll().

00246 {
00247         log(DEBUG,"Socket state change");
00248         this->state = s;
00249 }

bool InspSocket::Timeout time_t  current  ) 
 

This function checks if the socket has timed out yet, given the current time in the parameter.

Returns:
true if timed out, false if not timed out

Definition at line 197 of file socket.cpp.

References I_CONNECTING, I_ERR_TIMEOUT, I_ERROR, OnError(), OnTimeout(), state, timeout, and timeout_end.

00198 {
00199         if ((this->state == I_CONNECTING) && (current > timeout_end))
00200         {
00201                 // for non-listening sockets, the timeout can occur
00202                 // which causes termination of the connection after
00203                 // the given number of seconds without a successful
00204                 // connection.
00205                 this->OnTimeout();
00206                 this->OnError(I_ERR_TIMEOUT);
00207                 timeout = true;
00208                 this->state = I_ERROR;
00209                 return true;
00210         }
00211         return false;
00212 }

int InspSocket::Write std::string  data  )  [virtual]
 

Writes a std::string to the socket.

No carriage returns or linefeeds are appended to the string.

Parameters:
data The data to send

Definition at line 174 of file socket.cpp.

00175 {
00176         char* d = (char*)data.c_str();
00177         unsigned int written = 0;
00178         int n = 0;
00179         int s = data.length();
00180         while ((written < data.length()) && (n >= 0))
00181         {
00182                 n = send(this->fd,d,s,0);
00183                 if (n > 0)
00184                 {
00185                         // If we didnt write everything, advance
00186                         // the pointers so that when we retry
00187                         // the next time around the loop, we try
00188                         // to write what we failed to write before.
00189                         written += n;
00190                         s -= n;
00191                         d += n;
00192                 }
00193         }
00194         return written;
00195 }


Member Data Documentation

sockaddr_in InspSocket::addr [private]
 

The host being connected to, in sockaddr form.

Definition at line 78 of file socket.h.

Referenced by InspSocket().

in_addr InspSocket::addy [private]
 

The host being connected to, in in_addr form.

Definition at line 84 of file socket.h.

Referenced by InspSocket().

sockaddr_in InspSocket::client [private]
 

Client sockaddr structure used by accept().

Definition at line 119 of file socket.h.

Referenced by Poll().

int InspSocket::fd [private]
 

The file descriptor of this socket.

Definition at line 54 of file socket.h.

Referenced by Close(), GetFd(), and InspSocket().

std::string InspSocket::host [private]
 

The hostname connected to.

Definition at line 59 of file socket.h.

char InspSocket::ibuf[16384] [private]
 

Socket input buffer, used by read().

The class which extends InspSocket is expected to implement an extendable buffer which can grow much larger than 16k, this buffer is just designed to be temporary storage. space.

Definition at line 106 of file socket.h.

Referenced by Read().

std::string InspSocket::IP [private]
 

The IP address being connected to stored in string form for easy retrieval by accessors.

Definition at line 113 of file socket.h.

Referenced by GetIP(), and InspSocket().

socklen_t InspSocket::length [private]
 

Used by accept() to indicate the sizes of the sockaddr_in structures.

Definition at line 131 of file socket.h.

Referenced by Poll().

int InspSocket::port [private]
 

The port connected to, or the port this socket is listening on.

Definition at line 65 of file socket.h.

sockaddr_in InspSocket::server [private]
 

Server sockaddr structure used by accept().

Definition at line 125 of file socket.h.

InspSocketState InspSocket::state [private]
 

The state for this socket, either listening, connecting, connected or error.

Definition at line 72 of file socket.h.

Referenced by GetState(), InspSocket(), SetState(), and Timeout().

bool InspSocket::timeout [private]
 

This value is true if the socket has timed out.

Definition at line 97 of file socket.h.

Referenced by InspSocket(), and Timeout().

time_t InspSocket::timeout_end [private]
 

When this time is reached, the socket times out if it is in the CONNECTING state.

Definition at line 91 of file socket.h.

Referenced by InspSocket(), and Timeout().


The documentation for this class was generated from the following files:
Generated on Mon Dec 12 18:31:02 2005 for InspIRCd by  doxygen 1.4.4-20050815