Igor Skochinsky
Experimental HTTPClient with proxy support
api/TCPSocket.cpp@0:b56b6a05cad4, 2011-06-29 (annotated)
- Committer:
- igorsk
- Date:
- Wed Jun 29 16:01:58 2011 +0000
- Revision:
- 0:b56b6a05cad4
Who changed what in which revision?
| User | Revision | Line number | New contents of line |
|---|---|---|---|
| igorsk | 0:b56b6a05cad4 | 1 | |
| igorsk | 0:b56b6a05cad4 | 2 | /* |
| igorsk | 0:b56b6a05cad4 | 3 | Copyright (c) 2010 Donatien Garnier (donatiengar [at] gmail [dot] com) |
| igorsk | 0:b56b6a05cad4 | 4 | |
| igorsk | 0:b56b6a05cad4 | 5 | Permission is hereby granted, free of charge, to any person obtaining a copy |
| igorsk | 0:b56b6a05cad4 | 6 | of this software and associated documentation files (the "Software"), to deal |
| igorsk | 0:b56b6a05cad4 | 7 | in the Software without restriction, including without limitation the rights |
| igorsk | 0:b56b6a05cad4 | 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
| igorsk | 0:b56b6a05cad4 | 9 | copies of the Software, and to permit persons to whom the Software is |
| igorsk | 0:b56b6a05cad4 | 10 | furnished to do so, subject to the following conditions: |
| igorsk | 0:b56b6a05cad4 | 11 | |
| igorsk | 0:b56b6a05cad4 | 12 | The above copyright notice and this permission notice shall be included in |
| igorsk | 0:b56b6a05cad4 | 13 | all copies or substantial portions of the Software. |
| igorsk | 0:b56b6a05cad4 | 14 | |
| igorsk | 0:b56b6a05cad4 | 15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| igorsk | 0:b56b6a05cad4 | 16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| igorsk | 0:b56b6a05cad4 | 17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| igorsk | 0:b56b6a05cad4 | 18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| igorsk | 0:b56b6a05cad4 | 19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| igorsk | 0:b56b6a05cad4 | 20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
| igorsk | 0:b56b6a05cad4 | 21 | THE SOFTWARE. |
| igorsk | 0:b56b6a05cad4 | 22 | */ |
| igorsk | 0:b56b6a05cad4 | 23 | |
| igorsk | 0:b56b6a05cad4 | 24 | #include "TCPSocket.h" |
| igorsk | 0:b56b6a05cad4 | 25 | #include "if/net/nettcpsocket.h" |
| igorsk | 0:b56b6a05cad4 | 26 | |
| igorsk | 0:b56b6a05cad4 | 27 | TCPSocket::TCPSocket() : m_pNetTcpSocket(NULL), m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL) |
| igorsk | 0:b56b6a05cad4 | 28 | { |
| igorsk | 0:b56b6a05cad4 | 29 | |
| igorsk | 0:b56b6a05cad4 | 30 | } |
| igorsk | 0:b56b6a05cad4 | 31 | |
| igorsk | 0:b56b6a05cad4 | 32 | TCPSocket::TCPSocket(NetTcpSocket* pNetTcpSocket) : m_pNetTcpSocket(pNetTcpSocket), m_pCbItem(NULL), m_pCbMeth(NULL), m_pCb(NULL) |
| igorsk | 0:b56b6a05cad4 | 33 | { |
| igorsk | 0:b56b6a05cad4 | 34 | m_pNetTcpSocket->setOnEvent(this, &TCPSocket::onNetTcpSocketEvent); |
| igorsk | 0:b56b6a05cad4 | 35 | } |
| igorsk | 0:b56b6a05cad4 | 36 | |
| igorsk | 0:b56b6a05cad4 | 37 | TCPSocket::~TCPSocket() //close() |
| igorsk | 0:b56b6a05cad4 | 38 | { |
| igorsk | 0:b56b6a05cad4 | 39 | close(); |
| igorsk | 0:b56b6a05cad4 | 40 | } |
| igorsk | 0:b56b6a05cad4 | 41 | |
| igorsk | 0:b56b6a05cad4 | 42 | TCPSocketErr TCPSocket::bind(const Host& me) |
| igorsk | 0:b56b6a05cad4 | 43 | { |
| igorsk | 0:b56b6a05cad4 | 44 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 45 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 46 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 47 | return (TCPSocketErr) m_pNetTcpSocket->bind(me); |
| igorsk | 0:b56b6a05cad4 | 48 | } |
| igorsk | 0:b56b6a05cad4 | 49 | |
| igorsk | 0:b56b6a05cad4 | 50 | TCPSocketErr TCPSocket::listen() |
| igorsk | 0:b56b6a05cad4 | 51 | { |
| igorsk | 0:b56b6a05cad4 | 52 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 53 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 54 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 55 | return (TCPSocketErr) m_pNetTcpSocket->listen(); |
| igorsk | 0:b56b6a05cad4 | 56 | } |
| igorsk | 0:b56b6a05cad4 | 57 | |
| igorsk | 0:b56b6a05cad4 | 58 | TCPSocketErr TCPSocket::connect(const Host& host) |
| igorsk | 0:b56b6a05cad4 | 59 | { |
| igorsk | 0:b56b6a05cad4 | 60 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 61 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 62 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 63 | return (TCPSocketErr) m_pNetTcpSocket->connect(host); |
| igorsk | 0:b56b6a05cad4 | 64 | } |
| igorsk | 0:b56b6a05cad4 | 65 | |
| igorsk | 0:b56b6a05cad4 | 66 | TCPSocketErr TCPSocket::accept(Host* pClient, TCPSocket** ppNewTCPSocket) |
| igorsk | 0:b56b6a05cad4 | 67 | { |
| igorsk | 0:b56b6a05cad4 | 68 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 69 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 70 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 71 | NetTcpSocket* pNewNetTcpSocket; |
| igorsk | 0:b56b6a05cad4 | 72 | tcpSocketErr = (TCPSocketErr) m_pNetTcpSocket->accept(pClient, &pNewNetTcpSocket); |
| igorsk | 0:b56b6a05cad4 | 73 | if(pNewNetTcpSocket) |
| igorsk | 0:b56b6a05cad4 | 74 | *ppNewTCPSocket = new TCPSocket(pNewNetTcpSocket); |
| igorsk | 0:b56b6a05cad4 | 75 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 76 | } |
| igorsk | 0:b56b6a05cad4 | 77 | |
| igorsk | 0:b56b6a05cad4 | 78 | int /*if < 0 : TCPSocketErr*/ TCPSocket::send(const char* buf, int len) |
| igorsk | 0:b56b6a05cad4 | 79 | { |
| igorsk | 0:b56b6a05cad4 | 80 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 81 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 82 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 83 | return m_pNetTcpSocket->send(buf, len); |
| igorsk | 0:b56b6a05cad4 | 84 | } |
| igorsk | 0:b56b6a05cad4 | 85 | |
| igorsk | 0:b56b6a05cad4 | 86 | int /*if < 0 : TCPSocketErr*/ TCPSocket::recv(char* buf, int len) |
| igorsk | 0:b56b6a05cad4 | 87 | { |
| igorsk | 0:b56b6a05cad4 | 88 | TCPSocketErr tcpSocketErr = checkInst(); |
| igorsk | 0:b56b6a05cad4 | 89 | if(tcpSocketErr) |
| igorsk | 0:b56b6a05cad4 | 90 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 91 | return m_pNetTcpSocket->recv(buf, len); |
| igorsk | 0:b56b6a05cad4 | 92 | } |
| igorsk | 0:b56b6a05cad4 | 93 | |
| igorsk | 0:b56b6a05cad4 | 94 | TCPSocketErr TCPSocket::close() |
| igorsk | 0:b56b6a05cad4 | 95 | { |
| igorsk | 0:b56b6a05cad4 | 96 | if(!m_pNetTcpSocket) |
| igorsk | 0:b56b6a05cad4 | 97 | return TCPSOCKET_SETUP; |
| igorsk | 0:b56b6a05cad4 | 98 | m_pNetTcpSocket->resetOnEvent(); |
| igorsk | 0:b56b6a05cad4 | 99 | TCPSocketErr tcpSocketErr = (TCPSocketErr) m_pNetTcpSocket->close(); //Close (can already be closed) |
| igorsk | 0:b56b6a05cad4 | 100 | Net::releaseTcpSocket(m_pNetTcpSocket); //And release it so it can be freed when properly removed |
| igorsk | 0:b56b6a05cad4 | 101 | m_pNetTcpSocket = NULL; |
| igorsk | 0:b56b6a05cad4 | 102 | return tcpSocketErr; |
| igorsk | 0:b56b6a05cad4 | 103 | } |
| igorsk | 0:b56b6a05cad4 | 104 | |
| igorsk | 0:b56b6a05cad4 | 105 | //Callbacks |
| igorsk | 0:b56b6a05cad4 | 106 | void TCPSocket::setOnEvent( void (*pMethod)(TCPSocketEvent) ) |
| igorsk | 0:b56b6a05cad4 | 107 | { |
| igorsk | 0:b56b6a05cad4 | 108 | m_pCb = pMethod; |
| igorsk | 0:b56b6a05cad4 | 109 | } |
| igorsk | 0:b56b6a05cad4 | 110 | |
| igorsk | 0:b56b6a05cad4 | 111 | #if 0 //For info only |
| igorsk | 0:b56b6a05cad4 | 112 | template<class T> |
| igorsk | 0:b56b6a05cad4 | 113 | void TCPSocket::setOnEvent( T* pItem, void (T::*pMethod)(TCPSocketEvent) ) |
| igorsk | 0:b56b6a05cad4 | 114 | { |
| igorsk | 0:b56b6a05cad4 | 115 | m_pCbItem = (CDummy*) pItem; |
| igorsk | 0:b56b6a05cad4 | 116 | m_pCbMeth = (void (CDummy::*)(TCPSocketEvent)) pMethod; |
| igorsk | 0:b56b6a05cad4 | 117 | } |
| igorsk | 0:b56b6a05cad4 | 118 | #endif |
| igorsk | 0:b56b6a05cad4 | 119 | |
| igorsk | 0:b56b6a05cad4 | 120 | void TCPSocket::resetOnEvent() //Disable callback |
| igorsk | 0:b56b6a05cad4 | 121 | { |
| igorsk | 0:b56b6a05cad4 | 122 | m_pCb = NULL; |
| igorsk | 0:b56b6a05cad4 | 123 | m_pCbItem = NULL; |
| igorsk | 0:b56b6a05cad4 | 124 | m_pCbMeth = NULL; |
| igorsk | 0:b56b6a05cad4 | 125 | } |
| igorsk | 0:b56b6a05cad4 | 126 | |
| igorsk | 0:b56b6a05cad4 | 127 | void TCPSocket::onNetTcpSocketEvent(NetTcpSocketEvent e) |
| igorsk | 0:b56b6a05cad4 | 128 | { |
| igorsk | 0:b56b6a05cad4 | 129 | if(m_pCbItem && m_pCbMeth) |
| igorsk | 0:b56b6a05cad4 | 130 | (m_pCbItem->*m_pCbMeth)((TCPSocketEvent) e); |
| igorsk | 0:b56b6a05cad4 | 131 | else if(m_pCb) |
| igorsk | 0:b56b6a05cad4 | 132 | m_pCb((TCPSocketEvent) e); |
| igorsk | 0:b56b6a05cad4 | 133 | } |
| igorsk | 0:b56b6a05cad4 | 134 | |
| igorsk | 0:b56b6a05cad4 | 135 | TCPSocketErr TCPSocket::checkInst() |
| igorsk | 0:b56b6a05cad4 | 136 | { |
| igorsk | 0:b56b6a05cad4 | 137 | if(!m_pNetTcpSocket) |
| igorsk | 0:b56b6a05cad4 | 138 | { |
| igorsk | 0:b56b6a05cad4 | 139 | m_pNetTcpSocket = Net::tcpSocket(); |
| igorsk | 0:b56b6a05cad4 | 140 | if(!m_pNetTcpSocket) |
| igorsk | 0:b56b6a05cad4 | 141 | { |
| igorsk | 0:b56b6a05cad4 | 142 | return TCPSOCKET_IF; //Interface did not return a socket (usually because a default interface does not exist) |
| igorsk | 0:b56b6a05cad4 | 143 | } |
| igorsk | 0:b56b6a05cad4 | 144 | m_pNetTcpSocket->setOnEvent(this, &TCPSocket::onNetTcpSocketEvent); |
| igorsk | 0:b56b6a05cad4 | 145 | } |
| igorsk | 0:b56b6a05cad4 | 146 | return TCPSOCKET_OK; |
| igorsk | 0:b56b6a05cad4 | 147 | } |