/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            socket.cc
 *
 *  Mon Nov  8 10:49:33 CET 2004
 *  Copyright  2004 Bent Bisballe
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *    This file is part of MIaV.
 *
 *    MIaV is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    MIaV is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with MIaV; if not, write to the Free Software
 *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "socket.h"

#include <errno.h>

// for gethostbyname
#include <netdb.h>

#include <hugin.hpp>

Socket::Socket()
{
  connected = false;
  err = 0;
}

Socket::Socket(u_short port)
{
  connected = false;
  err = 0;

  // create socket
  ssocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP); 
  // PF_INET: ipv4, PF_INET6: ipv6
  // tcp: IPPROTO_TCP
  // upd: IPPROTO_UDP

  if (ssocket < 0) {
    err = 1;
    ERR(socket, "Socket: socket() failed!");
  }

  memset((char *) &socketaddr, 0, sizeof(socketaddr));
  socketaddr.sin_family = AF_INET; // Use "internet protocol" IP
  socketaddr.sin_port = htons(port);  // connect to that port
  socketaddr.sin_addr.s_addr = INADDR_ANY;
  // INADDR_ANY puts your IP address automatically
}


Socket::~Socket()
{
  //  if(err) perror("Socket: No socket to kill");
  //  printf("Socket: I'm melting...[%d]\n", ssocket);
  if(ssocket >= 0) close(ssocket);  // close server socket
}


Socket Socket::slisten()
{
  Socket s = Socket();

  if(err) {
    ERR(socket, "Socket: No socket present!");
    return s;
  }
  if(!connected) {
    // bind socket to address specified by "sa" parameter
    err = bind(ssocket, (struct sockaddr*)&socketaddr, sizeof(socketaddr));
    
    if (err) {
      ERR(socket, "Socket: bind() failed! %s", strerror(errno));
      return s;
    }
    
    // start listen for connection - kernel will accept connection 
    // requests (max 5 in queue)
    err = listen(ssocket, 5);
    if(err) {
      ERR(socket, "Socket: listen() failed! %s", strerror(errno));
      return s;
    }
  }

  // accept new connection and get its connection descriptor
  int csalen = sizeof(s.socketaddr);

  {
    fd_set rfds;
    FD_ZERO(&rfds);
    FD_SET(ssocket, &rfds);
    int retval = select(ssocket + 1, &rfds, NULL, NULL, NULL);

    if (retval == -1) {
      if(errno == EINTR) {
        connected = true;
        s.connected = true;
        return s;
      }

      ERR(socket, "select() failed! %s", strerror(errno));
    } else if(retval) {
      if(FD_ISSET(ssocket, &rfds)) {
        s.ssocket = accept(ssocket, 
                           (struct sockaddr*)&s.socketaddr, 
                           (socklen_t*)&csalen);
        ERR(socket, "accept() failed! %s", strerror(errno));
      }
    } else {
      // We don't do timeout
    }
  }

  if (s.ssocket < 0) {
    s.connected = false;
    err = 1;
    ERR(socket, "Socket: accept() failed! %s", strerror(errno));
    return s;
  }

  connected = true;
  s.connected = true;
  return s;
}


int Socket::sconnect(char *addr)
{
  if(err) {
    connected = false;
    ERR(socket, "Socket: No socket present!");
    return err;
  }

  // Do DNS lookup
  char *ip;
  struct in_addr **addr_list;
  struct hostent *he;
  he = gethostbyname(addr);
  if(!he || !he->h_length) {
    /*
    throw TCPConnectException(addr, toString(port),
       std::string("host lookup failed: ") + hstrerror(h_errno));
    */
  }

  addr_list = (struct in_addr **)he->h_addr_list;
  //  Get first value. We know for sure that there are at least one.
  ip = inet_ntoa(*addr_list[0]);

  socketaddr.sin_addr.s_addr = inet_addr(ip);
  
  err = connect(ssocket, (struct sockaddr*)&socketaddr, sizeof(socketaddr));
  if (err) {
    connected = false;
    ERR(socket, "Socket: connect() failed! %s", strerror(errno));
    return err;
  }
  //  fprintf(stderr, "Socket connected\n");
  connected = true;
  return 0;
}


bool Socket::isConnected()
{
  return connected;
}

bool Socket::hasError()
{
  return err != 0;
}