diff options
Diffstat (limited to 'lib/network/socket.cpp')
| -rw-r--r-- | lib/network/socket.cpp | 294 |
1 files changed, 294 insertions, 0 deletions
diff --git a/lib/network/socket.cpp b/lib/network/socket.cpp new file mode 100644 index 00000000..9de0ca0a --- /dev/null +++ b/lib/network/socket.cpp @@ -0,0 +1,294 @@ +#include <sys/ioctl.h> +#include <asm/ioctls.h> +#include <unistd.h> +#include <errno.h> + +#include <lib/network/socket.h> + +void eSocket::close() +{ + if (writebuffer.empty()) + { + int wasconnected=(mystate==Connection) || (mystate==Closing); + delete rsn; + rsn=0; + ::close(socketdesc); + socketdesc=-1; + mystate=Idle; + if (wasconnected) + connectionClosed_(); + } else + { + mystate=Closing; + rsn->setRequested(rsn->getRequested()|eSocketNotifier::Write); + } +} + +void eSocket::enableRead() +{ + if (rsn) + rsn->setRequested(rsn->getRequested()|eSocketNotifier::Read); +} + +void eSocket::disableRead() +{ + if (rsn) + rsn->setRequested(rsn->getRequested()&~eSocketNotifier::Read); +} + +void eSocket::inject(const char *data, int len) +{ + readbuffer.write(data, len); + if (mystate == Connection) + readyRead_(); +} + +eString eSocket::readLine() +{ + int size=readbuffer.searchchr('\n'); + if (size == -1) + return eString(); + size++; // ich will auch das \n + char buffer[size+1]; + buffer[size]=0; + readbuffer.read(buffer, size); + return eString(buffer); +} + +bool eSocket::canReadLine() +{ + return readbuffer.searchchr('\n') != -1; +} + +int eSocket::bytesAvailable() +{ + return readbuffer.size(); +} + +int eSocket::readBlock(char *data, unsigned int maxlen) +{ + return readbuffer.read(data, maxlen); +} + +int eSocket::bytesToWrite() +{ + return writebuffer.size(); +} + +int eSocket::state() +{ + return mystate; +} + +int eSocket::setSocket(int s, int iss, eMainloop *ml) +{ + socketdesc=s; + issocket=iss; + fcntl(socketdesc, F_SETFL, O_NONBLOCK); + last_break = 0xFFFFFFFF; + + if (rsn) + delete rsn; + rsn=new eSocketNotifier(ml, getDescriptor(), + eSocketNotifier::Read|eSocketNotifier::Hungup); + CONNECT(rsn->activated, eSocket::notifier); + return 0; +} + +void eSocket::notifier(int what) +{ + if ((what & eSocketNotifier::Read) && (mystate == Connection)) + { + int bytesavail=256; + if (issocket) + if (ioctl(getDescriptor(), FIONREAD, &bytesavail)<0) + eDebug("FIONREAD failed.\n"); + + { + if (issocket) + { + if (!bytesavail) // does the REMOTE END has closed the connection? (no Hungup here!) + { + writebuffer.clear(); + close(); + return; + } + } else // when operating on terminals, check for break + { + // where is this struct defined? + struct async_icount { + unsigned long cts, dsr, rng, dcd, tx, rx; + unsigned long frame, parity, overrun, brk; + unsigned long buf_overrun; + } icount; + + if (!ioctl(getDescriptor(), TIOCGICOUNT, &icount)) + { + if (last_break == 0xFFFFFFFF) + last_break = icount.brk; + else if (last_break != icount.brk) + { + last_break = icount.brk; + readbuffer.fromfile(getDescriptor(), bytesavail); + readbuffer.clear(); + writebuffer.clear(); + rsn->setRequested(rsn->getRequested()&~eSocketNotifier::Write); + write(getDescriptor(), "BREAK!", 6); + hangup(); + return; + } + } + } + int r; + if ((r=readbuffer.fromfile(getDescriptor(), bytesavail)) != bytesavail) + if (issocket) + eDebug("fromfile failed!"); + readyRead_(); + } + } else if (what & eSocketNotifier::Write) + { + if ((mystate == Connection) || (mystate == Closing)) + { + if (!writebuffer.empty()) + { + bytesWritten_(writebuffer.tofile(getDescriptor(), 65536)); + if (writebuffer.empty()) + { + rsn->setRequested(rsn->getRequested()&~eSocketNotifier::Write); + if (mystate == Closing) + { + close(); // warning, we might get destroyed after close. + return; + } + } + } else + eDebug("got ready to write, but nothin in buffer. strange."); + if (mystate == Closing) + close(); + } else if (mystate == Connecting) + { + mystate=Connection; + rsn->setRequested(rsn->getRequested()&~eSocketNotifier::Write); + + int res; + socklen_t size=sizeof(res); + ::getsockopt(getDescriptor(), SOL_SOCKET, SO_ERROR, &res, &size); + if (!res) + connected_(); + else + { + close(); + error_(res); + } + } + } else if (what & eSocketNotifier::Hungup) + { + if (mystate == Connection) + { + writebuffer.clear(); + close(); + } else if (mystate == Connecting) + { + int res; + socklen_t size=sizeof(res); + ::getsockopt(getDescriptor(), SOL_SOCKET, SO_ERROR, &res, &size); + close(); + error_(res); + } + } +} + +int eSocket::writeBlock(const char *data, unsigned int len) +{ + int w=len; + if (issocket && writebuffer.empty()) + { + int tw=::send(getDescriptor(), data, len, MSG_NOSIGNAL); + if ((tw < 0) && (errno != EWOULDBLOCK)) + eDebug("write: %m"); + + if (tw < 0) + tw = 0; + data+=tw; + len-=tw; + } + if (len) + writebuffer.write(data, len); + + if (!writebuffer.empty()) + rsn->setRequested(rsn->getRequested()|eSocketNotifier::Write); + return w; +} + +int eSocket::getDescriptor() +{ + return socketdesc; +} + +int eSocket::connectToHost(eString hostname, int port) +{ + struct hostent *server; + int res; + + if(!socketdesc){ + error_(errno); + return(-1); + } + server=gethostbyname(hostname.c_str()); + if(server==NULL) + { + eDebug("can't resolve %s", hostname.c_str()); + error_(errno); + return(-2); + } + bzero( (char*)&serv_addr, sizeof(serv_addr)); + serv_addr.sin_family=AF_INET; + bcopy( (char*)server->h_addr, + (char*)&serv_addr.sin_addr.s_addr, + server->h_length); + serv_addr.sin_port=htons(port); + res=::connect(socketdesc, (const sockaddr*)&serv_addr, sizeof(serv_addr)); + if ((res < 0) && (errno != EINPROGRESS)) + { + eDebug("can't connect to host: %s", hostname.c_str()); + close(); + error_(errno); + return(-3); + } + if (res < 0) // EINPROGRESS + { + rsn->setRequested(rsn->getRequested()|eSocketNotifier::Write); + mystate=Connecting; + } else + { + mystate=Connection; + connected_(); + } + return(0); +} + +eSocket::eSocket(eMainloop *ml): readbuffer(32768), writebuffer(32768), rsn(0) +{ + int s=socket(AF_INET, SOCK_STREAM, 0); +#if 0 + eDebug("[SOCKET]: initalized socket %d", socketdesc); +#endif + mystate=Idle; + setSocket(s, 1, ml); +} + +eSocket::eSocket(int socket, int issocket, eMainloop *ml): readbuffer(32768), writebuffer(32768), rsn(0) +{ + setSocket(socket, issocket, ml); + mystate=Connection; +} + +eSocket::~eSocket() +{ + if (rsn) + delete rsn; + if(socketdesc>=0) + { + ::close(socketdesc); + } +} |