node2.c

#include "dataStruct.h"
#include "declarations.h"
#include "hashTable.h"
#include "headers.h"
#include "msgHandler.h"
#include "udp.h"
#include "utils.h"
int num=2; // have to check 
int main() {

  // initializes variables for sockets, socket addresses, buffers, and file
  // descriptors
  int opt = TRUE, TransferValue = 0;
  int master_socket, sock_udp;
  struct sockaddr_in address, server_addr_udp, client_addr;
  int addrlen, addr_len;
  int new_socket, maxfd;
  int loop, valread;
  // int num;
  char buffer[1025], replyBuffer[1024]; // data buffer of 1K
#if 0
	printf("Enter total number of nodes: ");
	scanf("%d",&N);
	printf("Enter the node number:");	
	scanf("%d",&num);
	printf("Enter current node IP:");
	scanf("%s",ip1);	
	printf("Enter next node IP:");
	scanf("%s",ip2);
#endif
  fd_set readfds; //	Socket file descriptors we want to wake up for, using
                  // select()
  // Sets up the current and next node details
  node[num].ip_address = ip1;
  node[num].tcpportno = 2000 + num * 2;
  node[num].udpportno = node[num].tcpportno + 1;
  node[(num + 1) % N].ip_address = ip2;
  node[(num + 1) % N].tcpportno = 2000 + (num + 1) % N * 2;
  node[(num + 1) % N].udpportno = node[(num + 1) % N].tcpportno + 1;

  // working with UDP
  // Clears the console and prints instructions.

  system("clear");
  printf("       INSTRUCTIONS \n\n  =================NODE "
         "%d=======================\n",
         num);
  puts("   1.'put' request format : PUT(<integer>,<integer>)\n");
  puts("   2.'get' request format : GET(<integer>)\n");
  puts("   3.To print Hash Table : 'r'\n");
  puts("-----------------------------------\n\nENTER GET/PUT REQUEST :");

  initialiseHashtable();
  if ((sock_udp = socket(AF_INET, SOCK_DGRAM, 0)) ==
      -1) // Creating a UDP Socket
  {

    perror("Socket");
    exit(1);
  }
  // current node address
  server_addr_udp.sin_family =
      AF_INET; // AF_INET represents the address family INET for IPv4 protocol
  server_addr_udp.sin_port = htons(
      node[num].udpportno); // convert port in host byte order to N/W byte order
  server_addr_udp.sin_addr.s_addr = INADDR_ANY;
  bzero(&(server_addr_udp.sin_zero),
        8); // bzero() sets all values in a buffer to zero.
            // UDP Socket Creation and Binding
  if (bind(sock_udp, (struct sockaddr *)&server_addr_udp,
           sizeof(struct sockaddr)) == -1) // Binding UDP socket
  {
    perror("Bind");
    exit(1);
  }
  addr_len = sizeof(struct sockaddr);
  // TCP Socket Creation and Binding
  if ((master_socket = socket(AF_INET, SOCK_STREAM, 0)) ==
      0) // Create TCP Socket
  {
    perror("Create master_socket");
    exit(EXIT_FAILURE);
  }

  // set master socket to allow multiple connections
  if (setsockopt(master_socket, SOL_SOCKET, SO_REUSEADDR, (char *)&opt,
                 sizeof(opt)) < 0) {

    perror("setsockopt");
    exit(EXIT_FAILURE);
  }
  // current node address for tcp
  address.sin_family = AF_INET;
  address.sin_addr.s_addr = INADDR_ANY;
  address.sin_port = htons(node[num].tcpportno);

  if (bind(master_socket, (struct sockaddr *)&address, sizeof(address)) <
      0) // bind the socket to port
  {
    perror("bind");
    exit(EXIT_FAILURE);
  }
  if (listen(master_socket, 5) <
      0) // Specify maximum of 5 pending connections for the master socket
  {
    perror("listen");
    exit(EXIT_FAILURE);
  }

  while (1) {

    FD_ZERO(
        &readfds); // clears out the fd_set-readfds, so that it doesn't
                   // contain any file descriptors.  Clears the file descriptor
                   // set and adds UDP, TCP sockets, and standard input to it.

    FD_SET(sock_udp, &readfds); // adds file descriptor "sock_udp" to the
                                // fd_set,select will return

    FD_SET(master_socket, &readfds); // adds file descriptor "master_socket" to
                                     // the fd_set,select will return

    FD_SET(0, &readfds); // to read from standard input

    if (master_socket > sock_udp)
      maxfd = master_socket;
    else
      maxfd = sock_udp;
    select(maxfd + 1, &readfds, NULL, NULL, NULL);
    //`select()` to multiplex between UDP socket (`sock_udp`), TCP socket
    //(`master_socket`), and standard input (`0`).
    // Waits for any activity on the file descriptors.

    // Checks if there's activity on the TCP socket.
    //  it processes incoming data
    if (FD_ISSET(master_socket, &readfds)) {

      /* Open the new socket as 'new_socket' */

      addrlen = sizeof(address);

      if ((new_socket = accept(master_socket, (struct sockaddr *)&address,
                               &addrlen)) < 0) {

        /* if accept failed to return a socket descriptor, display error and
         * exit */
        perror("accept");
        exit(EXIT_FAILURE);
      }

      /* inform user of socket number - used in send and receive commands */

      if (TransferValue != 0) { // means put request was made

        // means put request was made on this node , and this node need to
        // supply the value 					'TransferValue'
        // back to client

        itoa(TransferValue, replyBuffer); // as sooon as conection is made, TCP
                                          // server send value to client
        send(new_socket, replyBuffer, strlen(replyBuffer), 0);
        read(new_socket, buffer, 1024);
        printf("\n%s\n", buffer);
        TransferValue = 0;
        close(new_socket);
      } else {
        // means get request was made and tcp server is only need to recieve
        // value from client
        if ((valread = read(new_socket, buffer, 1024)) < 0)
          close(new_socket);

        else {
          /* set the terminating NULL byte on the end of the data read */
          buffer[valread] = 0;
          printf("\n%s\n", buffer); // m
        }
        close(new_socket);
      }
    }
    // data is ready to be read from the UDP socket
    if (FD_ISSET(sock_udp,
                 &readfds)) /*Check udp_socket has data available to be read */
    {
      char rec_buff[5000];
      int len = recvfrom(sock_udp, rec_buff, 5000, 0,
                         (struct sockaddr *)&client_addr, &addr_len);
      // recvfrom() function receives a message
      // Receives data from the UDP socket and stores it in `rec_buff
      rec_buff[len] = '\0';
      printf("\n--------\nUDP PACKET RECIEVED FROM (IP ADDRESS : %s , PORT NO "
             ": %d , NODE NO : %d)  : ",
             inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port),
             extractNodeno(rec_buff));

      // inet_ntoa() converts a n/w address in a struct in_addr to a
      // dots-and-numbers format string
      printf("%s\n", rec_buff);

      // Either get request is recieved or put request is recieved
      // Check whether the req is to be forwarded or to be processed by the
      // current node itself

      if (forYou(num, rec_buff) == 0) {

        forwardUDP(num + 1,
                   rec_buff); // forward the request as it is to the next node
        puts("\n---------------------------------------\nENTER NEW GET/PUT "
             "REQUEST:");
      } else {
        printf("\nPROCESSING THE REQUEST ON THE CURRENT NODE : \n");
        // this is the node where get/put request is to be processed, so
        // fetching extracting key from request

        int key = extractKeyFromPut(rec_buff), nodeno = extractNodeno(rec_buff);

        // establish TCP conectiom with server whether it is get Or put message

        int sock, bytes_recieved;
        char send_data[1024], recv_data[1024];
        char flag =
            rec_buff[strlen(rec_buff) - 2]; // extracted flag value from request
        struct hostent *host;
        struct sockaddr_in server_addr;
        host = gethostbyname(extractIpaddress(
            rec_buff, '[', ',')); // extracting originator IP from request

        if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {

          perror("Socket");
          exit(1);
        }
        // To which connection is to be made
        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(extractValueFromPut(rec_buff));
        // same format put(key,value) initially
        // now format put(key,port)
        server_addr.sin_addr = *((struct in_addr *)host->h_addr);
        bzero(&(server_addr.sin_zero), 8);

        if (connect(sock, (struct sockaddr *)&server_addr,
                    sizeof(struct sockaddr)) == -1) // Make Connec
        {

          perror("Connect");
          exit(1);
        }
        if (flag ==
            's') // Get request and the value is to be fetched from hash table
        {

          int valuefetched = fetchValueFromHT(key,num);
          if (valuefetched != 0) {

            strcpy(send_data, "value = ");
            char valuebuff[6], nodebuff[4];
            itoa(valuefetched, valuebuff);
            strcat(send_data, valuebuff);
            strcat(send_data, ". Value retrieved from node no : ");
            itoa(num, nodebuff);
            strcat(send_data, nodebuff);
            strcat(send_data,
                   "\n-------------------\nENTER NEW GET/PUT REQUEST:");
          } else {
            strcpy(send_data,
                   "RESULT: No hash entry to this key on node no : ");
            char nodebuff[4];
            itoa(num, nodebuff);
            strcat(send_data, nodebuff);
            strcat(send_data,
                   "\n--------------------\nENTER NEW GET/PUT REQUEST :");
          }
          send(sock, send_data, strlen(send_data), 0); // Send the fetched value
          printf("\n KEY RECIEVED %d ,\nREQUEST ORIGINALLY INVOKED ON NODE  %d "
                 ", flag recieved = %c . VALUE SUPPLIED BACK ON TCP "
                 "CONNECTION.\nENTER NEW GET/PUT REQUEST:",
                 key, nodeno, rec_buff[strlen(rec_buff) - 2]);
        } else {
          bytes_recieved = recv(sock, recv_data, 1024, 0); // Receiving Put
                                                           // value
          recv_data[bytes_recieved] = '\0';
          printf("\n KEY RECIEVED IS: %d ,\nREQUEST ORIGINALLY INVOKED ON NODE "
                 " %d , flag recieved = %c\n",
                 key, nodeno, rec_buff[strlen(rec_buff) - 2]);

          printf("\nVALUE RECIEVED (ON TCP CONNECTION) FROM NODE NO %d = %s ",
                 nodeno, recv_data);
          // now insert the key,value in hash table of this node and send
          // confirmation message back to parent node on which the request was
          // originallly invoked by the user

          if (addToHashtable(key, atoi(recv_data),num))
            strcpy(send_data, "RESULT: put operation has been done "
                              "successfully.  Value added on node no :");
          else
            strcpy(send_data, "RESULT: put operation failed. Maximum key limit "
                              "Exceeded on node number  ");
          char nodebuff[4];
          itoa(num, nodebuff);
          strcat(send_data, nodebuff);
          strcat(send_data, ".\n------------------------------------\nENTER "
                            "NEW GET/PUT REQUEST :");
          send(sock, send_data, strlen(send_data), 0);
        }
        close(sock);
        fflush(stdout);
      } // else ends
    }

    //  working with console

    if (FD_ISSET(0,
                 &readfds)) // FD_ISSET()Returns a non-zero value if the bit for
                            // the file descriptor 							'0' is set
                            // in the file descriptor set pointed to by readfds,
                            // and 0 otherwise.
    {

      char rec_buff[5000];
      gets(rec_buff); //  Reads the input into `rec_buff`
      // If the input is 'r' or 'R', it displays the hash table
      if (rec_buff[0] == 'r' || rec_buff[0] == 'R') {

        displayHtable(num);
      }

      /***
       If the request is to be forwarded (`forYou(num, rec_buff) == 0`), it
prepares the data to be forwarded and sends it to the next node.

If the request is to be processed locally, it checks whether it's a GET or PUT
request.

    - For a GET request, it fetches the value from the local hash table.

    - For a PUT request, it adds the key-value pair to the local hash table.

      ****/

      else if (forYou(num, rec_buff) ==
               0) // fun return 1 if for you;return 0 if not for you
      {
        char outputbuff[40], *out, flag;
        int i = 0;

        if (getOrPut(rec_buff) == 0) // fun() return 1 for get, 0 for put
        {
          // value from put to be transferred at last
          TransferValue = extractValueFromPut(rec_buff);

          flag = 'r'; // indicates that last node has to receive a value
        } else
          flag = 's'; // indicates that last node will send a value

        out = forwardedData(rec_buff,
                            flag,num); // fun() to prepare the data to be forwaded

        for (i = 0; i < strlen(out); i++)
          outputbuff[i] =
              *(out + i); // data to be forwaded is assigned to outputbuff

        outputbuff[i] = '\0';

        forwardUDP(num + 1, outputbuff); // forwading method called
        free(out);
      }

      else {
        printf("\nPROCESSING THE REQUEST HERE:\n-------------------\n");

        if (getOrPut(rec_buff) == 1) // fun() return 1 for get, 0 for put
        {

          // extract key from get request
          int key = extractKeyFromGet(rec_buff), value;
          int maxkeylimit =
              (tablesize - 1) * N + num; // Compute maximum key limit

          if (key <= maxkeylimit) {

            value = fetchValueFromHT(
                key,num); // call fun() to fetch value from hash table

            if (value == 0)

              printf("\nError - No value in Hash table for this key on Current "
                     "node. 						"
                     "		\n---------------------\nEnter NEW GET/PUT "
                     "REQUEST :");

            else
              printf(
                  "\n key = %d, value = %d on same node "
                  "\n----------------------- 				"
                  "				-\nEnter GET/PUT REQUEST :",
                  key, value);
          } else {
            printf("\n Result :  Error - value cannot be fetched , maximum key "
                   "is %d 						"
                   "	\n----------------------\nEnter NEW GET/PUT REQUEST :",
                   maxkeylimit);
          }
        } else {
          // processing put request on the same node
          addToHashtable(extractKeyFromPut(rec_buff),
                         extractValueFromPut(rec_buff),num);
        }

      } // else ends
      fflush(stdout);

    } // read from console ends here

  } // while ends

} // main ends