/*

* 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