#include<stdio.h>
#include<sys/socket.h>
#include<arpa/inet.h> //inet_addr
#include<unistd.h>
#include<string.h>
#include<stdlib.h>
#include<string.h>



int main(int argc , char *argv[])
{
    int socket_desc , new_socket , c, valread;
    struct sockaddr_in server, client ;
    char *message;
    char buffer[1024] = {0};
    char *hello = "Hello from server";
    int opt = 1;
    char *string;
    const char s[2] = ";";
    char *token;
    char *sid;
    char *name;
    char *host;
    char *port;
    char *proto;
    char *user;
    char *password;
    char *key;
    char *companyName;
    char *officeAddress;
    char *state;
    char *country;
    char *ladmin;
    char *phone;
    char *mobile;
    char *email;
    char *designation;
    char *rmanager;
    char *sip;
    char *polname;
    char *device;
    char *status;
    char *srczone;
    char *dstzone;
    char *srcaddr;
    char *dstaddr;
    char *srcuser;
    char *app;
    char *service;
    char *urlcategory;
    char *action;






 //Create socket
    socket_desc = socket(AF_INET , SOCK_STREAM , 0);
    if (socket_desc == 0)
    {
        printf("Could not create socket");
    }


    if (setsockopt(socket_desc, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt)))
    {
    perror("setsockopt");
    exit(EXIT_FAILURE);
    }


    //Prepare the sockaddr_in structure
    server.sin_family = AF_INET;
    server.sin_addr.s_addr = INADDR_ANY;
    server.sin_port = htons( 8888 );



    //Bind
    if( bind(socket_desc,(struct sockaddr *)&server , sizeof(server)) < 0)
    {
   puts("bind failed");
    }


     puts("bind done");




    //Listen
    listen(socket_desc , 3);
  while(1){
    //Accept and incoming connection
  puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);
    new_socket = accept(socket_desc, (struct sockaddr *)&client, (socklen_t*)&c);
    if (new_socket<0)
    {
        perror("accept failed");
    }

    puts("Connection acceptedn");




    valread = read(new_socket, buffer, 1024);
    printf("%sn", buffer);
    string = buffer;

//Macro that I am using to distinguish between clients and the work to be done
    token = strtok(string,s);

//If server is passed as the first token, following code to be executed
    printf("%s", token);
    if( token == "server" ){
    sid = strtok(NULL, s);
    printf("nSERVER-ID: %sn", sid);
    name = strtok(NULL, s);
    printf("nName: %sn", name);
    host = strtok(NULL,s);
    printf("nHost: %sn", host);
    port = strtok(NULL,s);
    printf("nPort: %sn", port);
    proto = strtok(NULL,s);
    printf("nProto: %sn", proto);
    user = strtok(NULL,s);
    printf("nUser: %sn", user);
    password = strtok(NULL, s);
    printf("nPassword: %sn", password);
    key = strtok(NULL,s);
    printf("nKey: %sn", key);
    companyName = strtok(NULL,s);
    printf("nCompanyName: %sn", companyName);
    officeAddress = strtok(NULL,s);
    printf("nOfficeAddress: %sn", officeAddress);
    state = strtok(NULL,s);
    printf("nState: %sn", state);
    country = strtok(NULL,s);
    printf("nCountry: %sn", country);
    email = strtok(NULL,s);
    printf("/nEmail: %sn", email);
    ladmin = strtok(NULL,s);
    printf("nLAdmin: %sn", ladmin);
    phone = strtok(NULL,s);
    printf("nPhone: %sn", phone);
    mobile = strtok(NULL,s);
    printf("nMobile: %sn", mobile);
    designation = strtok(NULL, s);
    printf("nDesignation: %sn", designation);
    rmanager = strtok(NULL, s);
    printf("nReportingManager: %sn", rmanager);

    FILE *xml_file;
    xml_file = fopen("/var/www/db/db_server_list.xml", "a+");
    fseek(xml_file, -8, SEEK_END);

    ftruncate(fileno(xml_file),ftello(xml_file));





    fprintf(xml_file, "n<server id='%s'>n", sid);
    fprintf(xml_file, "   <name>%s</name>n", name);
    fprintf(xml_file, "   <host>%s</host>n", host);
    fprintf(xml_file, "   <port>%s</port>n", port);
    fprintf(xml_file, "   <proto>%s</proto>n", proto);
    fprintf(xml_file, "   <user>%s</user>n", user);
    fprintf(xml_file, "   <password>%s</password>n", password);
    fprintf(xml_file, "   <key>%s</key>n", key);
    fprintf(xml_file, "   <companyName>%s</companyName>n", companyName);
    fprintf(xml_file, "   <address>n");
    fprintf(xml_file, "     <officeAddr>%s</officeAddr>n", officeAddress);
    fprintf(xml_file, "     <state>%s</state>n", state);
    fprintf(xml_file, "     <country>%s</country>n", country);
    fprintf(xml_file, "   </address>n");
    fprintf(xml_file, "   <contacts>n");
    fprintf(xml_file, "     <ladmin>%s</ladmin>n", ladmin);
    fprintf(xml_file, "     <phone>%s</phone>n", phone);
    fprintf(xml_file, "     <mobile>%s</mobile>n", mobile);
    fprintf(xml_file, "     <email>%s</email>n", email);
    fprintf(xml_file, "     <designation>%s</designation>n", designation);
    fprintf(xml_file, "     <rManager>%s</rManager>n", rmanager);
    fprintf(xml_file, "   </contacts>n");
    fprintf(xml_file, " </server>n", key);
    fprintf(xml_file, "</data>");

    fclose(xml_file);
 send(new_socket, hello, 17, 0);
    printf("nSocket: Sent data to the client!n");
}
else if (token == "policy"){
    sip = strtok(NULL, s);
    printf("nserverip: %sn", sip);
    polname = strtok(NULL, s);
    printf("nPolicy Name: %sn", polname);
    device = strtok(NULL,s);
    printf("nDevice Name: %sn", device);
    status = strtok(NULL,s);
    printf("nStatus: %sn", status);
    srczone = strtok(NULL,s);
    printf("nSrcZone: %sn", srczone);
    dstzone = strtok(NULL,s);
    printf("nDstZone: %sn", dstzone);
    srcaddr = strtok(NULL, s);
    printf("nSrcAddr: %sn", srcAddr);
    dstaddr = strtok(NULL,s);
    printf("nDstAddr: %sn", dstAddr);
    srcuser = strtok(NULL,s);
    printf("nSrcUser: %sn", srcuser);
    app = strtok(NULL,s);
    printf("nApplication: %sn", app);
    service = strtok(NULL,s);
    printf("nService: %sn", service);
    urlcategory = strtok(NULL,s);
    printf("nUrlCategory: %sn", urlcategory);
    action = strtok(NULL,s);
    printf("/nAction: %sn", action);

    FILE *xml_file;
    xml_file = fopen("/var/www/db/db_policy.xml", "a+");
    fseek(xml_file, -14, SEEK_END);

    ftruncate(fileno(xml_file),ftello(xml_file));





    fprintf(xml_file, "n<policyDecryption>n",);
    fprintf(xml_file, "   <policyName>%s</policyName>n", polname);
    fprintf(xml_file, "   <deviceName>%s</deviceName>n", device);
    fprintf(xml_file, "   <status>%s</status>n", status);
    fprintf(xml_file, "   <srcZone>%s</srcZone>n", srczone);
    fprintf(xml_file, "   <dstZone>%s</dstZone>n", dstzone);
    fprintf(xml_file, "   <srcAddr>%s</srcAddr>n", srcaddr);
    fprintf(xml_file, "   <dstAddr>%s</dstAddr>n", dstaddr);
    fprintf(xml_file, "   <srcUser>%s</srcUser>n", srcuser);
    fprintf(xml_file, "   <application>%s</application>n", app);
    fprintf(xml_file, "   <service>%s</service>n", service);
    fprintf(xml_file, "   <urlCategory>%s</urlCategory>n", urlcategory);
    fprintf(xml_file, "   <action>%s</action>n", action);
    fprintf(xml_file, " </policyDecryption>n");
    fprintf(xml_file, "</policyList>");

    fclose(xml_file);
  send(new_socket, hello, 17, 0);
    printf("nSocket: Sent data to the client!n");


}
}

    return 0;
}