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 48 of file socket.cpp.

References I_DISCONNECTED, and state.

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

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 53 of file socket.cpp.

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

00054 {
00055         this->fd = newfd;
00056         this->state = I_CONNECTED;
00057         this->IP = ip;
00058         SE->AddFd(this->fd,true,X_ESTAB_MODULE);
00059         socket_ref[this->fd] = this;
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, BindSocket(), Close(), DEBUG, fd, I_CONNECTING, I_ERR_BIND, I_ERR_CONNECT, I_ERR_SOCKET, I_ERROR, I_LISTENING, IP, OnError(), OpenTCPSocket(), 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                                 socket_ref[this->fd] = this;
00089                                 log(DEBUG,"New socket now in I_LISTENING state");
00090                                 return;
00091                         }
00092                 }                       
00093         } else {
00094                 char* ip;
00095                 this->host = host;
00096                 hostent* hoste = gethostbyname(host.c_str());
00097                 if (!hoste) {
00098                         ip = (char*)host.c_str();
00099                 } else {
00100                         struct in_addr* ia = (in_addr*)hoste->h_addr;
00101                         ip = inet_ntoa(*ia);
00102                 }
00103 
00104                 this->IP = ip;
00105 
00106                 timeout_end = time(NULL)+maxtime;
00107                 timeout = false;
00108                 if ((this->fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
00109                 {
00110                         this->state = I_ERROR;
00111                         this->OnError(I_ERR_SOCKET);
00112                         return;
00113                 }
00114                 this->port = port;
00115                 inet_aton(ip,&addy);
00116                 addr.sin_family = AF_INET;
00117                 addr.sin_addr = addy;
00118                 addr.sin_port = htons(this->port);
00119 
00120                 int flags;
00121                 flags = fcntl(this->fd, F_GETFL, 0);
00122                 fcntl(this->fd, F_SETFL, flags | O_NONBLOCK);
00123 
00124                 if(connect(this->fd, (sockaddr*)&this->addr,sizeof(this->addr)) == -1)
00125                 {
00126                         if (errno != EINPROGRESS)
00127                         {
00128                                 this->Close();
00129                                 this->OnError(I_ERR_CONNECT);
00130                                 this->state = I_ERROR;
00131                                 return;
00132                         }
00133                 }
00134                 this->state = I_CONNECTING;
00135                 SE->AddFd(this->fd,false,X_ESTAB_MODULE);
00136                 socket_ref[this->fd] = this;
00137                 return;
00138         }
00139 }

InspSocket::~InspSocket  )  [virtual]
 

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

Definition at line 272 of file socket.cpp.

References Close().

00273 {
00274         this->Close();
00275 }


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 141 of file socket.cpp.

References fd, and OnClose().

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

00142 {
00143         if (this->fd != -1)
00144         {
00145                 this->OnClose();
00146                 shutdown(this->fd,2);
00147                 close(this->fd);
00148                 socket_ref[this->fd] = NULL;
00149                 this->fd = -1;
00150         }
00151 }

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 259 of file socket.cpp.

References fd.

00260 {
00261         return this->fd;
00262 }

std::string InspSocket::GetIP  ) 
 

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

Definition at line 153 of file socket.cpp.

References IP.

00154 {
00155         return this->IP;
00156 }

InspSocketState InspSocket::GetState  ) 
 

Returns the current socket state.

Definition at line 254 of file socket.cpp.

References state.

00255 {
00256         return this->state;
00257 }

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 270 of file socket.cpp.

Referenced by Close().

00270 { 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 264 of file socket.cpp.

Referenced by Poll().

00264 { 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 268 of file socket.cpp.

Referenced by Poll().

00268 { return true; }

int InspSocket::OnDisconnect  )  [virtual]
 

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

Definition at line 266 of file socket.cpp.

00266 { 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 265 of file socket.cpp.

Referenced by InspSocket(), and Timeout().

00265 { 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 267 of file socket.cpp.

Referenced by Poll().

00267 { 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 269 of file socket.cpp.

Referenced by Timeout().

00269 { 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 217 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.

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

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 158 of file socket.cpp.

References DEBUG, and ibuf.

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

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 248 of file socket.cpp.

References DEBUG, and state.

Referenced by Poll().

00249 {
00250         log(DEBUG,"Socket state change");
00251         this->state = s;
00252 }

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 200 of file socket.cpp.

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

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

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 177 of file socket.cpp.

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


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 Wed Dec 14 14:49:31 2005 for InspIRCd by  doxygen 1.4.4-20050815