| // Copyright (C) 2002-2010 Wizzer Works. |
| // |
| // 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. |
| // |
| |
| /** |
| * @file simplesocket.cpp |
| * @brief Implementation of SimpleTCPStream methods. |
| * |
| * Implementation of SimpleTCPStream methods. |
| * |
| * @author Mark S. Millard (msm@wizzer.com) |
| * @date 2002-08-15 |
| */ |
| |
| // Include Common C++ Library header files. |
| #include <cc++/config.h> |
| #include <cc++/export.h> |
| #include <cc++/socket.h> |
| |
| #ifndef INADDR_LOOPBACK |
| #define INADDR_LOOPBACK (unsigned long)0x7f000001 |
| #endif |
| |
| #ifdef CCXX_NAMESPACES |
| namespace ost { |
| using namespace std; |
| #endif |
| |
| SimpleTCPStream::SimpleTCPStream(TCPSocket &server, size_t size) : |
| Socket(accept(server.getSocket(), NULL, NULL)) |
| { |
| tpport_t port; |
| IPV4Host host = getPeer(&port); |
| |
| if (! server.onAccept(host, port)) { |
| endSocket(); |
| error(errConnectRejected); |
| return; |
| } |
| |
| Socket::state = CONNECTED; |
| } |
| |
| SimpleTCPStream::SimpleTCPStream(const IPV4Host &host, tpport_t port, size_t size) : |
| Socket(AF_INET, SOCK_STREAM, IPPROTO_TCP) |
| { |
| Connect(host, port, size); |
| } |
| |
| SimpleTCPStream::~SimpleTCPStream() |
| { |
| endStream(); |
| } |
| |
| IPV4Host SimpleTCPStream::getSender(tpport_t *port) const |
| { |
| return IPV4Host(); |
| } |
| |
| void SimpleTCPStream::Connect(const IPV4Host &host, tpport_t port, size_t size) |
| { |
| size_t i; |
| |
| for (i = 0 ; i < host.getAddressCount(); i++) { |
| struct sockaddr_in addr; |
| memset(&addr, 0, sizeof(addr)); |
| addr.sin_family = AF_INET; |
| addr.sin_addr = host.getAddress(i); |
| addr.sin_port = htons(port); |
| |
| // Win32 will crash if you try to connect to INADDR_ANY. |
| if ( INADDR_ANY == addr.sin_addr.s_addr ) |
| addr.sin_addr.s_addr = INADDR_LOOPBACK; |
| if (::connect(so, (struct sockaddr *)&addr, (socklen_t)sizeof(addr)) == 0) |
| break; |
| } |
| |
| if (i == host.getAddressCount()) { |
| connectError(); |
| endSocket(); |
| return; |
| } |
| |
| Socket::state = CONNECTED; |
| } |
| |
| SimpleTCPStream::SimpleTCPStream() : |
| Socket(PF_INET, SOCK_STREAM, IPPROTO_TCP) {} |
| |
| SimpleTCPStream::SimpleTCPStream(const SimpleTCPStream &source) : |
| #ifdef WIN32 |
| Socket(source.so) |
| #else |
| Socket(dup(source.so)) |
| #endif |
| {} |
| |
| void SimpleTCPStream::endStream(void) |
| { |
| endSocket(); |
| } |
| |
| ssize_t SimpleTCPStream::read(char *bytes, size_t length, timeout_t timeout) |
| { |
| // Declare local variables. |
| ssize_t rlen = 0; |
| size_t totalrecv = 0; |
| char *currentpos = bytes; |
| |
| // Check for reasonable requested length. |
| if (length < 1) { |
| return (ssize_t)totalrecv; |
| } |
| |
| while (totalrecv < length) { |
| // Check for timeout condition. |
| if (timeout) { |
| if (! isPending(pendingInput, timeout)) { |
| error(errTimeout); |
| return -1; |
| } |
| } |
| |
| // Attempt to read data. |
| rlen = _IORET64 ::recv(so, (char *) currentpos, _IOLEN64 (length-totalrecv), 0); |
| if (rlen == 0 || rlen == -1) { |
| break; |
| } |
| // cout << "received " << rlen << " bytes, remaining " << length - totalrecv << flush; |
| |
| totalrecv += rlen; |
| currentpos += rlen; |
| } |
| |
| // Set error condition if necessary. |
| if (rlen == -1) { |
| error(errInput); |
| } |
| |
| // Return total number of bytes recieved. |
| return (ssize_t)totalrecv; |
| } |
| |
| ssize_t SimpleTCPStream::write(const char *bytes, size_t length, timeout_t timeout) |
| { |
| // Declare local variables. |
| ssize_t rlen = 0; |
| |
| // Check for reasonable requested length. |
| if (length < 1) { |
| return rlen; |
| } |
| |
| // Check for timeout condition. |
| if (timeout) { |
| if (! isPending(pendingOutput, timeout)) { |
| error(errTimeout); |
| return -1; |
| } |
| } |
| |
| // Attempt to write data. |
| rlen = _IORET64 ::send(so, (const char *)bytes, _IOLEN64 length, MSG_NOSIGNAL); |
| if (rlen == -1) { |
| error(errOutput); |
| } |
| |
| return rlen; |
| } |
| |
| ssize_t SimpleTCPStream::peek(char *bytes, size_t length, timeout_t timeout) |
| { |
| // Declare local variables. |
| ssize_t rlen = 0; |
| size_t totalrecv = 0; |
| char *currentpos = bytes; |
| |
| // Check for reasonable requested length. |
| if (length < 1) { |
| return (ssize_t)totalrecv; |
| } |
| |
| while (totalrecv < length) { |
| // Check for timeout condition. |
| if (timeout) { |
| if (! isPending(pendingInput, timeout)) { |
| error(errTimeout); |
| return -1; |
| } |
| } |
| |
| // Attempt to read data. |
| rlen = _IORET64 ::recv(so, (char *) currentpos, _IOLEN64 (length-totalrecv), MSG_PEEK); |
| if (rlen == 0 || rlen == -1) { |
| break; |
| } |
| // cout << "received " << rlen << " bytes, remaining " << length - totalrecv << flush; |
| |
| totalrecv += rlen; |
| currentpos += rlen; |
| } |
| |
| // Set error condition if necessary. |
| if (rlen == -1) { |
| error(errInput); |
| } |
| |
| // Return total number of bytes recieved. |
| return (ssize_t)totalrecv; |
| } |
| |
| bool SimpleTCPStream::isPending(Pending pending, timeout_t timeout) |
| { |
| return Socket::isPending(pending, timeout); |
| } |
| |
| #ifdef CCXX_NAMESPACES |
| } /* for ost */ |
| #endif |
| |
| /** EMACS ** |
| * Local variables: |
| * mode: c++ |
| * c-basic-offset: 4 |
| * End: |
| */ |