#include "rslib.h"
#pragma hdrstop

#ifndef NO_INET_STREAM

#ifdef XP_WIN
#if !defined(_WINSOCK2API_) && !defined(_WINSOCKAPI_)
#include <winsock2.h>
#endif
#else
#include <sys/socket.h>
#endif

bool InternetServer::Startup()
{
 s = socket(AF_INET, SOCK_STREAM, 0); /* make Internet socket */
 if (s == INVALID_SOCKET) return false;

 myaddr.sin_family = AF_INET;       /* Internet address family */
 myaddr.sin_port = htons(port);     /* port */
 memset(&myaddr.sin_addr.s_addr,0, sizeof(IN_ADDR)); /* host addr */
 memset(myaddr.sin_zero,0,sizeof(myaddr.sin_zero));

 if (bind(s,(sockaddr*)&myaddr,sizeof(myaddr)) == SOCKET_ERROR)
   return false;

 int i = sizeof(myaddr);
 if (getsockname(s,(sockaddr*)&myaddr,&i) != SOCKET_ERROR)
 {
	  port = ntohs(myaddr.sin_port);
     hostent* addr;
     addr = gethostbyaddr ((char*)(void*)&myaddr.sin_addr,sizeof(IN_ADDR),AF_INET);
     if (addr)
     {
        hostinfo.Resize(strlen(addr->h_name) + 10);
        sprintf(hostinfo,"%s:%d",addr->h_name,port);
     }
 }
 if (listen(s,SOMAXCONN) == SOCKET_ERROR)
   return false; /* the destructor calls closesocket and shutdown */

 return true;
}

bool InternetServer::AnyoneWaiting()
{
  fd_set incoming;
  FD_ZERO(&incoming);
  FD_SET(s,&incoming);

  timeval t;
  t.tv_sec=0;
  t.tv_usec = 0;

  if (select(0,&incoming,0,0,&t) ==SOCKET_ERROR) return false;

  return FD_ISSET(s,&incoming) != 0;
}


InternetServer::InternetServer(int p)
{
 port = p;
 s = INVALID_SOCKET;
#ifdef XP_WIN
 WSADATA d;
 if (WSAStartup(0x0101, &d) != 0) /* init winsock lib. want v1.1 */
 {
  error = new xdb("Cannot start winsock","file",__FILE__,"line",__LINE__);
  return;
 }
#endif
 if (!Startup())
 {
  error = new xdb("Cannot listen on that port");
  return;
 }
}

InternetServer::~InternetServer()
 {

 if (s != INVALID_SOCKET) /* shutdown and close the socket if needed */
   { shutdown(s, 2);   /* shutdown both channels (read+write) */
     closesocket(s);
   }
#ifdef XP_WIN
   WSACleanup();        /* free winsock lib */
#endif
 }

InternetStream* InternetServer::Accept()
{
 unsigned int t;
 sockaddr_in clientaddr; /* now, get the host name out */
 int len = sizeof(clientaddr);
 t = accept( s,(SOCKADDR*)&clientaddr,&len);

 if (t == INVALID_SOCKET)
   return 0;

 int hostaddr = clientaddr.sin_addr.s_addr;
 InternetStream* r;
 try{
 r = new InternetStream(t,hostaddr);
 if (r->error) return 0;
  } catch(...) {return 0;}
 return r;
}

InternetStream::InternetStream(unsigned int socket, int host) : Stream()
{
	port = 0;
   s = socket;

#ifdef XP_WIN
   WSADATA d;
   if (WSAStartup(0x0101, &d) != 0) /* init winsock lib. want v1.1 */
   {
      error = new xdb("Cannot start winsock","file",__FILE__,"line",__LINE__);
      Type=NotOpen;
      return;
   }
#endif

   hostaddr = host;

   hostent* addr;

   addr = gethostbyaddr ((char*)(void*)&hostaddr,sizeof(IN_ADDR),AF_INET);

   if (addr)
    hostinfo = addr->h_name;

  Type = ReadWrite;
}

/* proxy server from
IE:
HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Internet Settings\ProxyServer

NN:


*/


InternetStream::InternetStream(const char* url) : Stream()
{
   hostaddr = 0;
   s = INVALID_SOCKET;
#ifdef XP_WIN
   WSADATA d;
   if (WSAStartup(0x0101, &d) != 0)     /* init winsock lib. want v1.1 */
   {
      WSACleanup();
      error = new xdb("Cannot start winsock","file",__FILE__,"line",__LINE__);
      Type = NotOpen;
      return;
   }
#endif

 TStr Error;
 TStr service, user, password, host, filename, query;
 URLSplit(url, service, user, password, host, filename, query);
// hostinfo = url; //  service://user:password@host:port/filename/query

 if (!*host) host = "localhost";

 if (!*filename) filename = "/";

 if (!init(host,80,&Error)) return; //throw xdb(Error,"host",host,"port",80);

 bool IsProxy = strnicmp(filename,"http://",7) == 0;

 if (*query) filename = TStr(filename,"?",query); // http://proxy.com/http://server.com/page?query

 size_t l = strcspn(filename,"%+ \t\r\n");
 if (filename[l])
  {
   query=filename;
   filename = 0;
   URLEncodeURL(filename, query);
  }


 const char* c = filename;
 while (*c == '/') c++;

 MemoryStream Request;

 Request << "GET ";

 if (!IsProxy)
  Request << "/";

 Request << c << " HTTP/1.1\r\nHost: " << host << "\r\n"
         << "User-Agent: Raosoft/1.0 (windows; en-us; " __DATE__ ")\r\n\r\n";

 write((char*)Request,Request.size());
}

int InternetStream::SkipHeaders()
{
 TParameterList temp;
 return GetHeaders(temp);
}

int InternetStream::GetHeaders(TNameValueList &n) //returns the status
{
 TStr status;
 readline(status);
 char * c = strchr(status,' ');
 if (!c) return 0;
 int ret = atoi(c+1);
 ReadMIME(n);
 return ret;
}

InternetStream::InternetStream() : Stream()
{
  hostaddr = 0;
  port= 0;
   s = INVALID_SOCKET;
#ifdef XP_WIN
   WSADATA d;
   if (WSAStartup(0x0101, &d) != 0)     /* init winsock lib. want v1.1 */
   {
      WSACleanup();
      error = new xdb("Cannot start winsock","file",__FILE__,"line",__LINE__);
      Type = NotOpen;
      return;
   }
#endif
}

InternetStream::InternetStream(const char host[],int port) : Stream()
{
   hostaddr = 0;
   s = INVALID_SOCKET;
#ifdef XP_WIN
   WSADATA d;
   if (WSAStartup(0x0101, &d) != 0)     /* init winsock lib. want v1.1 */
   {
      WSACleanup();
      error = new  xdb("Cannot start winsock","file",__FILE__,"line",__LINE__);
      Type = NotOpen;
      return;
   }
#endif
 TStr Error;
 if (!init(host,port,&Error)) return;
}

/* connect to the specified host at the specified port. the host address */
/* can be in the form of "192.168.0.1" or "phoenix.ttdev.com". isnum     */
/* should be set to true if the numeric form is used.                    */
/* return: true  --- ok                                                  */
/*         false --- error                                               */

#ifdef XP_WIN
 int InternetStream::GetLastError()   { return WSAGetLastError();}
#else
 int InternetStream::GetLastError()   { return TWinSockDll::WSAGetLastError();}
#endif

bool InternetStream::init(const char *phost, int port, TStr* errmsg)
{
   if (s != INVALID_SOCKET) /* shutdown and close the socket if needed */
   { shutdown(s, 2);   /* shutdown both channels (read+write) */
     closesocket(s);
     s = INVALID_SOCKET;
   }

   int opt;
   /* host addr as a 32bit unsigned int */
   uint32 *p;
   HOSTENT *he;
   sockaddr_in svraddr;
   TStr host(phost);

   char * cp = strchr(host,':');
   if (cp)
    {
     int newport = atoi(cp+1);
     if (newport) port = newport;
     *cp=0;
    }

   hostinfo.Resize(strlen(host) + 10);
   sprintf(hostinfo,"%s:%d",(char*)host,port);
   //hostinfo = TStr(host,':',port);

   if (isdigit(host[0]))        /* something like 192.168.0.1 */
   {
      hostaddr = inet_addr(host);      /* convert to 32bit unsigned int */
   }
   else                         /* something like phoenix.ttdev.com */
   {
      he = gethostbyname(host); /* convert to 32bit unsigned int */
      if (!he) {return 0;}
      p = (uint32*)he->h_addr_list[0];
      hostaddr = *p;
   }

#ifdef XP_WIN // synchronous sockets are not defined for win16

#ifndef SO_OPENTYPE
#define SO_OPENTYPE 0x7008
#endif
#ifndef SO_SYNCHRONOUS_NONALERT
#define SO_SYNCHRONOUS_NONALERT 0x20
#endif

   int err;
   int len=sizeof(opt);
   err = getsockopt( /* Get the original setting of SO_OPENTYPE. */
        INVALID_SOCKET,
        SOL_SOCKET,
        SO_OPENTYPE,
        (char *)&opt,
        &len);

   opt = SO_SYNCHRONOUS_NONALERT;
   int originalOpt = opt;

   /* make sure we're gonna create a synchronous socket */
   err = setsockopt(INVALID_SOCKET,
                  SOL_SOCKET,
                  SO_OPENTYPE,
                  (char*)&opt,
                  sizeof(opt));

   if (err != NO_ERROR) return false;
#endif

   s = socket(AF_INET, SOCK_STREAM, 0); /* make Internet socket */

#ifdef XP_WIN
   opt = originalOpt; /* reset the option to default for async sockets */

   err=setsockopt(INVALID_SOCKET, //NT and 95 specific
              SOL_SOCKET,
              SO_OPENTYPE,
              (char*)&opt,
              sizeof(opt));

   if (err != NO_ERROR) return false;
#endif

   if (s == INVALID_SOCKET) return false;  /* did we create the sync socket? */

   svraddr.sin_family = AF_INET;       /* Internet address family */
   svraddr.sin_port = htons(port);     /* port */
   memcpy(&svraddr.sin_addr.s_addr, &hostaddr, sizeof(IN_ADDR)); /* host addr */
   memset(svraddr.sin_zero,0,sizeof(svraddr.sin_zero));

   if (connect(s, (SOCKADDR*)&svraddr, sizeof(svraddr)) != 0)
   {if (errmsg) {errmsg->Resize(128); sprintf(*errmsg,"%d",GetLastError()); }
    return false;
   }
 Type = ReadWrite;
 return true;
}

void InternetStream::NoDelay()
{
   int opt = TRUE;

   /* no transmission delay */
   setsockopt(s,
                  IPPROTO_TCP,
                  TCP_NODELAY,
                  (char*)&opt,
                  sizeof(opt));
}                  

InternetStream::~InternetStream()
{ if (s != INVALID_SOCKET) /* shutdown and close the socket if needed */
   { shutdown(s, 2);   /* shutdown both channels (read+write) */
     closesocket(s);
   }
   WSACleanup();        /* free winsock lib */
}

int InternetStream::write(const char *b, int n)
{/* send a block of bytes. return # of bytes actually sent */
 if (Type == NotOpen) return 0;
 if (::send(s, (const char*)b, n, 0) == SOCKET_ERROR)
   Type = NotOpen;

}

bool InternetStream::canread()
{
 if (Type == NotOpen) return false;

  fd_set incoming;
  FD_ZERO(&incoming);
  FD_SET(s,&incoming);

  timeval t;
  t.tv_sec=0;
  t.tv_usec = 0;

  if (select(0,&incoming,0,0,&t) ==SOCKET_ERROR) return false;

  return FD_ISSET(s,&incoming);
}

bool InternetStream::eof()
{
 if (Type == NotOpen) return true;

  fd_set outgoing;
  FD_ZERO(&outgoing);
  FD_SET(s,&outgoing);

  timeval t;
  t.tv_sec=0;
  t.tv_usec = 0;

  if (select(0,0,0,&outgoing,&t) ==SOCKET_ERROR) return true;

  return FD_ISSET(s,&outgoing);
}


bool InternetStream::canwrite()
{
 if (Type == NotOpen) return false;

  fd_set outgoing;
  FD_ZERO(&outgoing);
  FD_SET(s,&outgoing);

  timeval t;
  t.tv_sec=0;
  t.tv_usec = 0;

  if (select(0,0,&outgoing,0,&t) ==SOCKET_ERROR) return false;

  return FD_ISSET(s,&outgoing);
}


int InternetStream::read(char *b, int n)
{/* receive a block of bytes. return # of bytes actually received */
 if (Type == NotOpen) return 0;
 int ret =0;
   if (start)
   {
    if (start->eof())
    {
     start = 0;
    }
    else
    {
     ret = start->read(b,n);
     b += ret;
     n -= ret;
     if (!n) return ret;
    }
   }
   ret += ::recv(s, (char*)b, n, 0);
   if (ret < 0) {Type = NotOpen; return 0;}
   return ret;
}

/* send a line of text. return true or false */
bool  InternetStream::sendln(const char s[])
{
 if (Type == NotOpen) return false;

   int l;
   l = strlen(s);            /* see if it ends with \n */
   if (l > 0 && s[l-1] == '\n') l--;  /* we don't want that */
   if (write(s, l) != l) return false;     /* send the string without trailing \n */
   if (write("\r\n", 2) != 2) return false;/* send \r\n now */
   return true;
}

/* receive a line of text. \r\n signals the end but */
/* it is not stored into s. return true or false    */
bool  InternetStream::recvln(char s[], int maxlen)
{
 if (Type == NotOpen) return false;
 bool ret = false;
   int i=0;
   char ch;
   s[0]=0;
   while (i < maxlen-1)
   {
      if (read(&ch,1)!=1) return false; /* receive a char */
      if (ch == '\n') {ret = true; break;}
      if (ch == '\r')     /* \r\n signals the end */
      { read(&ch,1); ret=true; break; }
      s[i] = ch;          /* save the char and proceed */
      i++;
   }
   s[i] = 0;              /* NULL terminate the string */
   return ret;
}

#endif
