/* A simple server in the internet domain using TCP
   The port number is passed as an argument  - Vijay, Anna */
#include <stdio.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>

#include "socket_rapids.h"
#include "rapids_api.h"
#include "utils.h"

void error(char *msg)
{
    perror(msg);
    exit(1);
}

int main(int argc, char *argv[])
{
     int sockfd, newsockfd, portno, clilen;
     long ctrl_seqno = 0;
     char buffer[256];
     struct sockaddr_in serv_addr, cli_addr;
     int n;

     if (argc < 2) {
         fprintf(stderr,"ERROR, no port provided\n");
         exit(1);
     }

     //printf("A\n");
     rapids_init("server");
     //printf("B\n");

     sockfd = socket(AF_INET, SOCK_STREAM, 0);
     if (sockfd < 0) 
        error("ERROR opening socket");
     bzero((char *) &serv_addr, sizeof(serv_addr));
     portno = atoi(argv[1]);
     serv_addr.sin_family = AF_INET;
     serv_addr.sin_addr.s_addr = INADDR_ANY;
     serv_addr.sin_port = htons(portno);
     if (bind(sockfd, (struct sockaddr *) &serv_addr,
              sizeof(serv_addr)) < 0) 
              error("ERROR on binding");
     listen(sockfd,5);
     
     while(1) {  // main accept() loop
       clilen = sizeof(cli_addr);
       newsockfd = accept(sockfd, 
			  (struct sockaddr *) &cli_addr, 
			  &clilen);
       if (newsockfd < 0) { 
	 error("ERROR on accept");
	 continue;
       }

       printf("Got connection from %s\n", inet_ntoa(cli_addr.sin_addr));
       if (!fork()) { // this is the child process
	 close(sockfd); // child doesn't need the listener
	
	 //while(1) {
	 bzero(buffer,256);

	 rapids_socket_read_mesg(newsockfd, ctrl_seqno++, "msg");
	 
	 n = read(newsockfd,buffer,255);
	 if (n < 0) error("ERROR reading from socket");
	 printf("Here is the message: %s\n",buffer);
	 n = write(newsockfd,"I got your message",18);
	 if (n < 0) error("ERROR writing to socket");
	 
	 rapids_socket_write_mesg(newsockfd, ctrl_seqno++, "msg");
	 //}

	 close(newsockfd);
	 exit(0);
       }
       close(newsockfd);  // parent doesn't need this
     }

     close(sockfd) ; 
     return 0; 
}