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