Dependents:   TimeZoneDemo EthernetJackTestCode MMEx_Challenge ntp_mem ... more

Committer:
segundo
Date:
Tue Nov 09 20:54:15 2010 +0000
Revision:
0:ac1725ba162c

        

Who changed what in which revision?

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