/*
 * socket.c
 *
 * $Id: socket.c,v 1.3 1996/03/29 16:51:01 athan Exp $
 *
 * $Log: socket.c,v $
 * Revision 1.3  1996/03/29 16:51:01  athan
 * General cleanup
 *
 * Revision 1.2  1996/03/28 20:02:07  athan
 * General cleanups
 *
 *
 */

#include <malloc.h>
#include <sys/types.h>
#include <ctype.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <netdb.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#if defined(SOLARIS)
#include <sys/filio.h>
#endif /* SOLARIS */

#include "config.h"

#define IPROTO   0
#define MAX_QUEUE   5

/* Extern Function Prototypes */

#if defined(SOLARIS)
extern int gethostname(char *name, int namelen);
#endif

/* Extern Variables */

extern int status;
extern int verbose;

/* Local Function Prototypes */

int accept_connection(void);
int open_main_socket(int port);
void shut_down(void);

/* Local Variables */

int mainsock;

int open_main_socket(int port)
{
   struct sockaddr_in sa;
   char *hostname;
   struct hostent *hp;
   int dummy = 1;

   hostname = (char *) malloc(101);
   memset((char *)&sa, 0, sizeof(struct sockaddr_in));
   gethostname(hostname, 100);

   hp = gethostbyname(hostname);
   if (hp == NULL)
   {
      fprintf(stderr, "Error: Host machine does not exist!\n");
      return -1;
   }

   sa.sin_family = hp->h_addrtype;
   sa.sin_port = htons(port);

   if (-1 == (mainsock = socket(AF_INET, SOCK_STREAM, 0)))
   {
      fprintf(stderr, "Couldn't open main socket!\n");
      return -1;
   }

   if (-1 == setsockopt(mainsock, SOL_SOCKET, SO_REUSEADDR, (char *)&dummy,
                  sizeof(dummy)))
   {
      fprintf(stderr, "Failed setsockopt\n");
      return -1;
   }

   if (-1 == ioctl(mainsock, (int)FIONBIO, (caddr_t)&dummy))
   {
      fprintf(stderr, "Couldn't set to non-blocking\n");
      return -1;
   }

   if (-1 == (bind(mainsock, (struct sockaddr*)&sa, sizeof(sa))))
   {
      close(mainsock);
      return -1;
   }

   if (listen(mainsock, MAX_QUEUE) < 0)
   {
      fprintf(stderr, "Listen refused\n");
      return -1;
   }

   return 0;
}

/* Accept a new connection */

int accept_connection(void)
{
   struct sockaddr_in add;
   int lengthadd;
   int dummy;
   int this_sock;
   
   lengthadd = sizeof(add);
   
   getsockname(mainsock, (struct sockaddr *)&add, &lengthadd);
   
   if (-1 == (this_sock = accept(mainsock, (struct sockaddr *)&add,
   				 &lengthadd)))
   {
      return -1;
   }
   
   if (-1 == ioctl(this_sock, (int)FIONBIO, (caddr_t)&dummy))
   {
      close(this_sock);
      return -1;
   }

   return this_sock;
}


void shut_down(void)
{
   if (verbose)
   {
      fprintf(stderr, "Shutting down NOW!\n");
   }
   shutdown(mainsock, 2);
   close(mainsock);
}