| // Copyright (C) 1999-2005 Open Source Telecom Corporation. |
| // Copyright (C) 2006-2010 David Sugar, Tycho Softworks. |
| // |
| // 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 2 of the License, 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
| // |
| // As a special exception, you may use this file as part of a free software |
| // library without restriction. Specifically, if other files instantiate |
| // templates or use macros or inline functions from this file, or you compile |
| // this file and link it with other files to produce an executable, this |
| // file does not by itself cause the resulting executable to be covered by |
| // the GNU General Public License. This exception does not however |
| // invalidate any other reasons why the executable file might be covered by |
| // the GNU General Public License. |
| // |
| // This exception applies only to the code released under the name GNU |
| // Common C++. If you copy code from other releases into a copy of GNU |
| // Common C++, as the General Public License permits, the exception does |
| // not apply to the code that you add in this way. To avoid misleading |
| // anyone as to the status of such modified files, you must delete |
| // this exception notice from them. |
| // |
| // If you write modifications of your own for GNU Common C++, it is your choice |
| // whether to permit this exception to apply to your modifications. |
| // If you do not wish that, delete this exception notice. |
| // |
| |
| #include <cc++/config.h> |
| #include <cc++/address.h> |
| #include <cc++/socket.h> |
| #include <cc++/file.h> |
| #include <cc++/thread.h> |
| #include <cc++/exception.h> |
| #include <cc++/export.h> |
| #include <cc++/unix.h> |
| |
| #include <cc++/unix.h> |
| #include <fcntl.h> |
| #include <cerrno> |
| #include <cstdlib> |
| #ifndef WIN32 |
| #include <netinet/tcp.h> |
| #include <sys/un.h> |
| #endif |
| |
| #ifdef WIN32 |
| #include <io.h> |
| #endif |
| |
| #ifdef CCXX_NAMESPACES |
| namespace ost { |
| using namespace std; |
| #endif |
| |
| #ifndef WIN32 |
| |
| UnixSocket::UnixSocket(const char* pathname, int backlog) : |
| Socket(AF_UNIX, SOCK_STREAM, 0) |
| { |
| struct sockaddr_un addr; |
| socklen_t len; |
| unsigned slen = strlen(pathname); |
| |
| if(slen > sizeof(addr.sun_path)) |
| slen = sizeof(addr.sun_path); |
| |
| path = NULL; |
| memset(&addr, 0, sizeof(addr)); |
| addr.sun_family = AF_UNIX; |
| memcpy( addr.sun_path, pathname, slen ); |
| |
| #ifdef __SUN_LEN |
| len = sizeof(addr.sun_len) + strlen(addr.sun_path) + sizeof(addr.sun_family) + 1; |
| addr.sun_len = len; |
| #else |
| len = strlen(addr.sun_path) + sizeof(addr.sun_family) + 1; |
| #endif |
| remove(pathname); |
| if(bind(so, (struct sockaddr *)&addr, len)) { |
| endSocket(); |
| error(errBindingFailed); |
| return; |
| } |
| |
| path = new char[slen + 1]; |
| strcpy(path, pathname); |
| |
| if(listen(so, backlog)) { |
| endSocket(); |
| error(errBindingFailed); |
| return; |
| } |
| state = BOUND; |
| } |
| |
| |
| UnixSocket::~UnixSocket() |
| { |
| close(); |
| } |
| |
| void UnixSocket::close(void) |
| { |
| endSocket(); |
| if(path) { |
| remove(path); |
| delete[] path; |
| path = NULL; |
| } |
| } |
| |
| UnixStream::UnixStream(UnixSocket &server, int size, bool throwflag, timeout_t to) : |
| Socket(accept(server.so, NULL, NULL)) ,streambuf() |
| #ifdef HAVE_OLD_IOSTREAM |
| ,iostream() |
| #else |
| ,iostream((streambuf *)this) |
| #endif |
| ,bufsize(0), gbuf(NULL), pbuf(NULL) |
| { |
| #ifdef HAVE_OLD_IOSTREAM |
| init((streambuf *)this); |
| #endif |
| |
| timeout = to; |
| setError(throwflag); |
| |
| allocate(size); |
| Socket::state = CONNECTED; |
| } |
| |
| UnixStream::UnixStream(const char* pathname, int size, bool throwflag, timeout_t to) : |
| Socket(AF_UNIX, SOCK_STREAM, 0), streambuf(), |
| #ifdef HAVE_OLD_IOSTREAM |
| iostream(), |
| #else |
| iostream((streambuf *)this), |
| #endif |
| bufsize(0), gbuf(NULL), pbuf(NULL) |
| { |
| #ifdef HAVE_OLD_IOSTREAM |
| init((streambuf *)this); |
| #endif |
| timeout = to; |
| setError(throwflag); |
| connect(pathname, size); |
| } |
| |
| UnixStream::~UnixStream() |
| { |
| endStream(); |
| } |
| |
| void UnixStream::connect(const char* pathname, int size) |
| { |
| struct sockaddr_un addr; |
| socklen_t len; |
| unsigned slen = strlen(pathname); |
| |
| if(slen > sizeof(addr.sun_path)) |
| slen = sizeof(addr.sun_path); |
| |
| memset(&addr, 0, sizeof(addr)); |
| addr.sun_family = AF_UNIX; |
| memcpy( addr.sun_path, pathname, slen ); |
| #ifdef __SUN_LEN |
| len = sizeof(addr.sun_len) + strlen(addr.sun_path) + sizeof(addr.sun_family) + 1; |
| addr.sun_len = len; |
| #else |
| len = strlen(addr.sun_path) + sizeof(addr.sun_family); |
| #endif |
| if(::connect(so, (struct sockaddr *)&addr, len) != 0) { |
| connectError(); |
| endSocket(); |
| return; |
| } |
| |
| allocate(size); |
| Socket::state = CONNECTED; |
| } |
| |
| UnixStream::UnixStream(bool throwflag) : |
| Socket(PF_UNIX, SOCK_STREAM, 0), streambuf(), |
| #ifdef HAVE_OLD_IOSTREAM |
| iostream(), |
| #else |
| iostream((streambuf *)this), |
| #endif |
| timeout(0), bufsize(0), gbuf(NULL), pbuf(NULL) |
| { |
| #ifdef HAVE_OLD_IOSTREAM |
| init((streambuf *)this); |
| #endif |
| setError(throwflag); |
| } |
| |
| UnixStream::UnixStream(const UnixStream &source) : |
| Socket(dup(source.so)), streambuf(), |
| #ifdef HAVE_OLD_IOSTREAM |
| iostream() |
| #else |
| iostream((streambuf *)this) |
| #endif |
| { |
| #ifdef HAVE_OLD_IOSTREAM |
| init((streambuf *)this); |
| #endif |
| bufsize = source.bufsize; |
| allocate(bufsize); |
| } |
| |
| void UnixStream::endStream(void) |
| { |
| if(bufsize) |
| sync(); |
| |
| if(gbuf) |
| delete[] gbuf; |
| if(pbuf) |
| delete[] pbuf; |
| gbuf = pbuf = NULL; |
| bufsize = 0; |
| endSocket(); |
| } |
| |
| void UnixStream::allocate(int size) |
| { |
| if(size < 2) { |
| bufsize = 1; |
| return; |
| } |
| |
| gbuf = new char[size]; |
| pbuf = new char[size]; |
| if(!pbuf || !gbuf) { |
| error(errResourceFailure); |
| return; |
| } |
| bufsize = size; |
| clear(); |
| |
| #if (defined(__GNUC__) && (__GNUC__ < 3)) && !defined(WIN32) && !defined(STLPORT) |
| setb(gbuf, gbuf + size, 0); |
| #endif |
| setg(gbuf, gbuf + size, gbuf + size); |
| setp(pbuf, pbuf + size); |
| } |
| |
| int UnixStream::doallocate() |
| { |
| if(bufsize) |
| return 0; |
| |
| allocate(1); |
| return 1; |
| } |
| |
| int UnixStream::uflow(void) |
| { |
| int ret = underflow(); |
| |
| if (ret == EOF) |
| return EOF; |
| |
| if (bufsize != 1) |
| gbump(1); |
| |
| return ret; |
| } |
| |
| int UnixStream::underflow(void) |
| { |
| int rlen = 1; |
| unsigned char ch; |
| |
| if(bufsize == 1) { |
| if(Socket::state == STREAM) |
| rlen = ::read(so, (char *)&ch, 1); |
| else if(timeout && !Socket::isPending(pendingInput, timeout)) { |
| clear(ios::failbit | rdstate()); |
| error(errTimeout); |
| return EOF; |
| } |
| else |
| rlen = ::recv(so, (char *)&ch, 1, 0); |
| if(rlen < 1) { |
| if(rlen < 0) { |
| clear(ios::failbit | rdstate()); |
| error(errInput); |
| } |
| return EOF; |
| } |
| return ch; |
| } |
| |
| if(!gptr()) |
| return EOF; |
| |
| if(gptr() < egptr()) |
| return (unsigned char)*gptr(); |
| |
| rlen = (gbuf + bufsize) - eback(); |
| if(Socket::state == STREAM) |
| rlen = ::read(so, (char *)eback(), rlen); |
| else if(timeout && !Socket::isPending(pendingInput, timeout)) { |
| clear(ios::failbit | rdstate()); |
| error(errTimeout); |
| return EOF; |
| } |
| else |
| rlen = ::recv(so, (char *)eback(), rlen, 0); |
| if(rlen < 1) { |
| if(rlen < 0) { |
| clear(ios::failbit | rdstate()); |
| error(errInput); |
| } |
| return EOF; |
| } |
| |
| setg(eback(), eback(), eback() + rlen); |
| return (unsigned char) *gptr(); |
| } |
| |
| bool UnixStream::isPending(Pending pending, timeout_t timeout) |
| { |
| if(pending == pendingInput && in_avail()) |
| return true; |
| else if(pending == pendingOutput) |
| flush(); |
| |
| return Socket::isPending(pending, timeout); |
| } |
| |
| int UnixStream::sync(void) |
| { |
| overflow(EOF); |
| setg(gbuf, gbuf + bufsize, gbuf + bufsize); |
| return 0; |
| } |
| |
| int UnixStream::overflow(int c) |
| { |
| unsigned char ch; |
| int rlen, req; |
| |
| if(bufsize == 1) { |
| if(c == EOF) |
| return 0; |
| |
| ch = (unsigned char)(c); |
| if(Socket::state == STREAM) |
| rlen = ::write(so, (const char *)&ch, 1); |
| else |
| rlen = ::send(so, (const char *)&ch, 1, 0); |
| if(rlen < 1) { |
| if(rlen < 0) { |
| clear(ios::failbit | rdstate()); |
| error(errOutput); |
| } |
| return EOF; |
| } |
| else |
| return c; |
| } |
| |
| if(!pbase()) |
| return EOF; |
| |
| req = pptr() - pbase(); |
| if(req) { |
| if(Socket::state == STREAM) |
| rlen = ::write(so, (const char *)pbase(), req); |
| else |
| rlen = ::send(so, (const char *)pbase(), req, 0); |
| if(rlen < 1) { |
| if(rlen < 0) { |
| clear(ios::failbit | rdstate()); |
| error(errOutput); |
| } |
| return EOF; |
| } |
| req -= rlen; |
| } |
| |
| // if write "partial", rebuffer remainder |
| |
| if(req) |
| memcpy(pptr(), pptr() + rlen, req); |
| setp(pbuf + req, pbuf + bufsize); |
| |
| if(c != EOF) { |
| *pptr() = (unsigned char)c; |
| pbump(1); |
| } |
| return c; |
| } |
| |
| unixstream::unixstream() : |
| UnixStream() |
| { |
| setError(false); /* no exceptions */ |
| } |
| |
| unixstream::unixstream(const char *pathname, int buf) : |
| UnixStream() |
| { |
| setError(false); |
| open(pathname, buf); |
| } |
| |
| unixstream::unixstream(UnixSocket &server, int buf) : |
| UnixStream() |
| { |
| setError(false); |
| open(server, buf); |
| } |
| |
| bool unixstream::operator!() const |
| { |
| return (Socket::state != CONNECTED) ? true : false; |
| } |
| |
| void unixstream::open(UnixSocket &unixsock, int buf) |
| { |
| endStream(); |
| so = accept(unixsock.so, NULL, NULL); |
| if(so == INVALID_SOCKET) |
| return; |
| |
| allocate(buf); |
| Socket::state = CONNECTED; |
| } |
| |
| void unixstream::close(void) |
| { |
| if(Socket::state == AVAILABLE) |
| return; |
| |
| endStream(); |
| so = socket(AF_UNIX, SOCK_STREAM, 0); |
| if(so != INVALID_SOCKET) |
| Socket::state = AVAILABLE; |
| } |
| |
| UnixSession::UnixSession(const char* pathname, int size, int pri, int stack) : |
| Thread(pri, stack), UnixStream() |
| { |
| struct sockaddr_un addr; |
| unsigned slen = strlen(pathname); |
| |
| if(slen > sizeof(addr.sun_path)) |
| slen = sizeof(addr.sun_path); |
| socklen_t len; |
| setCompletion(false); |
| setError(false); |
| allocate(size); |
| |
| memset(&addr, 0, sizeof(addr)); |
| addr.sun_family = AF_UNIX; |
| memcpy( addr.sun_path, pathname, slen ); |
| |
| #ifdef __SUN_LEN |
| len = sizeof(addr.sun_len) + strlen(addr.sun_path) + sizeof(addr.sun_family) + 1; |
| addr.sun_len = len; |
| #else |
| len = strlen(addr.sun_path) + sizeof(addr.sun_family); |
| #endif |
| if(::connect(so, (struct sockaddr *)&addr, len ) != 0) { |
| #ifdef WIN32 |
| if( WSAGetLastError() == WAEISCONN ) |
| #else |
| if( EINPROGRESS == errno ) |
| #endif |
| { |
| Socket::state = CONNECTING; |
| } |
| else { |
| endSocket(); |
| Socket::state = INITIAL; |
| } |
| return; |
| } |
| |
| setCompletion(true); |
| Socket::state = CONNECTED; |
| } |
| |
| |
| UnixSession::UnixSession(UnixSocket &s,int size, int pri, int stack) : |
| Thread(pri, stack), UnixStream(s, size) |
| { |
| setCompletion(true); |
| setError(false); |
| } |
| |
| UnixSession::~UnixSession() |
| { |
| terminate(); |
| endStream(); |
| } |
| |
| int UnixSession::waitConnection(timeout_t timeout) |
| { |
| long sockopt = 0; |
| socklen_t len = sizeof(sockopt); |
| |
| switch(Socket::state) { |
| case INITIAL: |
| return -1; |
| case CONNECTED: |
| break; |
| case CONNECTING: |
| if(!Socket::isPending(pendingOutput, timeout)) { |
| endSocket(); |
| Socket::state = INITIAL; |
| return -1; |
| } |
| |
| getsockopt(so, SOL_SOCKET, SO_ERROR, (char *)&sockopt, &len); |
| if(sockopt) { |
| endSocket(); |
| Socket::state = INITIAL; |
| return -1; |
| } |
| case AVAILABLE: |
| case BOUND: |
| case STREAM: |
| break; |
| } |
| Socket::state = CONNECTED; |
| return 0; |
| } |
| |
| void UnixSession::initial(void) |
| { |
| if(waitConnection(60000)) |
| exit(); |
| } |
| |
| #endif // ndef WIN32 |
| #ifdef CCXX_NAMESPACES |
| } |
| #endif |