--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/hti/PC_Tools/HTIGateway/HtiGateway/src/Socket.cpp Tue Feb 02 01:57:15 2010 +0200
@@ -0,0 +1,275 @@
+/*
+* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
+* All rights reserved.
+* This component and the accompanying materials are made available
+* under the terms of "Eclipse Public License v1.0"
+* which accompanies this distribution, and is available
+* at the URL "http://www.eclipse.org/legal/epl-v10.html".
+*
+* Initial Contributors:
+* Nokia Corporation - initial contribution.
+*
+* Contributors:
+*
+* Description:
+* This file contains implemention of Socket, SocketClient, SocketServer and
+* SocketSelect classes.
+*/
+#include <iostream>
+#include <sstream>
+#include <errno.h>
+
+#include "Socket.h"
+#include "util.h"
+
+using namespace std;
+
+int Socket::nofSockets_= 0;
+
+void Socket::Start() {
+ if (!nofSockets_) {
+ WSADATA info;
+ if (WSAStartup(MAKEWORD(2,0), &info)) {
+ throw "Could not start WSA";
+ }
+ }
+ ++nofSockets_;
+}
+
+void Socket::End() {
+ WSACleanup();
+}
+
+
+Socket::Socket() : s_(0) {
+ Start();
+ // UDP: use SOCK_DGRAM instead of SOCK_STREAM
+ s_ = socket(AF_INET,SOCK_STREAM,0);
+
+ refCounter_ = new int(1);
+}
+
+Socket::Socket(SOCKET s) : s_(s) {
+ Start();
+ refCounter_ = new int(1);
+};
+
+
+Socket::~Socket() {
+ if (! --(*refCounter_)) {
+ Close();
+ delete refCounter_;
+ }
+
+ --nofSockets_;
+ if (!nofSockets_) End();
+}
+
+Socket::Socket(const Socket& o) {
+ refCounter_=o.refCounter_;
+ (*refCounter_)++;
+ s_ =o.s_;
+
+ nofSockets_++;
+}
+
+Socket& Socket::operator =(Socket& o) {
+ (*o.refCounter_)++;
+
+ refCounter_=o.refCounter_;
+ s_ =o.s_;
+
+ nofSockets_++;
+
+ return *this;
+}
+
+
+void Socket::Close() {
+ closesocket(s_);
+}
+
+
+String Socket::ReceiveBytes() {
+ String ret;
+ char buf[1024];
+ int rv;
+
+ while ((rv=recv(s_, buf, 1024,0)) == WSAEMSGSIZE) {
+
+ String t;
+ t.assign(buf,1024);
+ ret +=t;
+ }
+ String t;
+ t.assign(buf, 0, rv);
+ ret += t;
+
+ printf("t = %s, rv = %d", t.c_str(), rv);
+
+ return ret;
+}
+
+int Socket::ReceiveBytes(BYTE *data, int bytes)
+{
+ int rv = SOCKET_ERROR;
+ while (rv == SOCKET_ERROR)
+ {
+ rv = recv(s_, (char *)data, bytes, 0);
+ if (rv == 0 || rv == WSAECONNRESET)
+ return -1;
+ else
+ break;
+ }
+ return rv;
+}
+
+String Socket::ReceiveLine() {
+ String ret;
+ while (1) {
+ char r;
+
+ switch(recv(s_, &r, 1, 0)) {
+ case 0: // not connected anymore;
+ return "";
+ case -1:
+ if (errno == EAGAIN) {
+ return ret;
+ } else {
+ // not connected anymore
+ return "";
+ }
+ }
+
+ ret += r;
+ if (r == '\n') return ret;
+ }
+}
+
+void Socket::SendLine(String s) {
+ s += '\n';
+ send(s_,s.c_str(),s.length(),0);
+}
+
+void Socket::SendBytes(const String& s) {
+ send(s_,s.c_str(),s.length(),0);
+}
+
+int Socket::SendBytes(const BYTE* data, int bytes)
+{
+ if (bytes <= 0) return 0;
+ send(s_,(char *)data, bytes, 0);
+ return 0;
+}
+
+SocketServer::SocketServer() {
+ /*
+ // create a socket
+ s_ = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+ if (s_ == INVALID_SOCKET) {
+ throw "INVALID_SOCKET";
+ }
+ */
+}
+
+
+void SocketServer::Accept( Socket*& s,int port, int connections, String& remoteHost ) {
+
+ sockaddr_in sa;
+ memset(&sa, 0, sizeof(sa)); /* clear our address */
+ sa.sin_family = PF_INET;
+ sa.sin_port = htons(port);
+
+ // Use the given socket for listening
+ Socket* listenSocket = s;
+
+ // bind the socket to the internet address
+ if (bind(listenSocket->s_, (sockaddr *)&sa, sizeof(sockaddr_in)) ==
+ SOCKET_ERROR) {
+ throw "INVALID_SOCKET";
+ }
+
+ listen(listenSocket->s_, connections);
+
+ sockaddr_in remoteAddr;
+ int remoteAddrLen = sizeof(sockaddr_in);
+ SOCKET new_sock = accept(listenSocket->s_, (sockaddr*) &remoteAddr, &remoteAddrLen);
+ if (new_sock == INVALID_SOCKET)
+ {
+ throw "Invalid Socket";
+ }
+ // Return remote host and port to caller
+ stringstream msg;
+ msg<<inet_ntoa(remoteAddr.sin_addr)<<":"<<ntohs(remoteAddr.sin_port);
+ remoteHost.append( msg.str() );
+
+
+ s = new Socket(new_sock);
+ delete listenSocket;
+}
+
+void SocketServer::Connect(Socket* s, const char* remoteHost, int remotePort) {
+
+ Socket* connectSocket = s;
+ sockaddr_in sa;
+ memset(&sa, 0, sizeof(sa)); /* clear our address */
+ sa.sin_family = AF_INET;
+ if( isalpha(remoteHost[0]) )
+ {
+ HOSTENT* Hostent;
+ if( (Hostent = gethostbyname( remoteHost )) != 0 )
+ sa.sin_addr.s_addr = *(long*) Hostent->h_addr;
+ else
+ throw "Unknown host!";
+ }
+ else
+ sa.sin_addr.s_addr = inet_addr( remoteHost );
+
+ sa.sin_port = htons( remotePort );
+
+ // Connect to server.
+ if ( connect( connectSocket->s_, (SOCKADDR*) &sa, sizeof(sa) ) == SOCKET_ERROR) {
+ throw "SOCKET_ERROR";
+ }
+}
+
+
+SocketClient::SocketClient(const String& host, int port) : Socket() {
+ String error;
+
+ hostent *he;
+ if ((he = gethostbyname(host.c_str())) == NULL) {
+ error = strerror(errno);
+ throw error;
+ }
+
+ sockaddr_in addr;
+ addr.sin_family = AF_INET;
+ addr.sin_port = htons(port);
+ addr.sin_addr = *((in_addr *)he->h_addr);
+ memset(&(addr.sin_zero), 0, 8);
+
+ if (::connect(s_, (sockaddr *) &addr, sizeof(sockaddr))) {
+ error = strerror(WSAGetLastError());
+ throw error;
+ }
+}
+
+
+SocketSelect::SocketSelect(Socket const * const s1, Socket const * const s2) {
+ FD_ZERO(&fds_);
+ FD_SET(const_cast<Socket*>(s1)->s_,&fds_);
+ FD_SET(const_cast<Socket*>(s2)->s_,&fds_);
+
+ if (select (0, &fds_, (fd_set*) 0, (fd_set*) 0, (timeval*) 0)
+ == SOCKET_ERROR) throw "Error in select";
+}
+
+
+bool SocketSelect::Readable(Socket const * const s) {
+ if (FD_ISSET(s->s_,&fds_)) return true;
+ return false;
+}
+
+// End of the file
+