practicas/practica3/parte1/serverhandler.cpp

   1 /*
   2  * Servidor de hash
   3  */
   4
   5 #include "libtcp.h"
   6 #include "common.h"
   7 #include "protocol.h"
   8 #include <set>
   9 #include <string>
  10 #include <cstdlib>
  11
  12 int main(int argc, char *argv[])
  13 {
  14     char *pname;
  15     int sockfd;/* socket que sirve como template */
  16     pid_t pid; /* pid del server iterativo */
  17     char linea[256];
  18     char fin = 0, fin_cmd = 0;
  19     int result = 0;
  20     std::set< std::string > set;
  21
  22     pname = argv[0];
  23     pid = getpid();
  24
  25     /* Verifico parametros de linea de comando */
  26     if (argc > 1)
  27         sockfd = atoi(argv[1]);
  28     else
  29     {
  30         perror("Falta parametro con fd");
  31         exit (1);
  32     }
  33
  34     print_msg(stdout, "%s (%d): atendiendo a cliente por socket %d\n", pname, pid, sockfd);
  35
  36     while (!fin)
  37     {
  38         std::string buffer;
  39         while (!fin_cmd)
  40         {
  41             Protocol proto;
  42             int n = libtcp_receive_bin(sockfd, (char*) &proto, sizeof(Protocol));
  43             if (n < 0)
  44             {
  45                 print_msg(stdout, "%s (%d): error en recibir\n", pname, pid);
  46                 exit(1);
  47             }
  48             print_msg(stdout, "%s (%d): cliente %d envio operacion (%d, %d, %d)\n",
  49                     pname, pid, proto.client_id, proto.type, proto.end, proto.len);
  50
  51             n = libtcp_receive_bin(sockfd, linea, proto.len);
  52             if (n < 0)
  53             {
  54                 print_msg(stdout, "%s (%d): error en recibir\n", pname, pid);
  55                 exit(1);
  56             }
  57             linea[proto.len] = '\0';
  58             print_msg(stdout, "%s (%d): se recibieron %d bytes de payload (%s)\n",
  59                     pname, pid, proto.len, linea);
  60             buffer += linea;
  61             print_msg(stdout, "%s (%d): buffer: %s\n", pname, pid, buffer.c_str());
  62
  63             if (proto.end)
  64             {
  65                 switch (proto.type)
  66                 {
  67                     case Protocol::PUT:
  68                         if (set.find(buffer) == set.end())
  69                         {
  70                             set.insert(buffer);
  71                             result = Protocol::OK;
  72                         }
  73                         else
  74                             result = Protocol::EXISTS;
  75                         break;
  76                     case Protocol::FIND:
  77                         if (set.find(buffer) == set.end())
  78                             result = Protocol::NOT_FOUND;
  79                         else
  80                             result = Protocol::OK;
  81                         break;
  82                     case Protocol::DEL:
  83                         if (set.erase(buffer))
  84                             result = Protocol::OK;
  85                         else
  86                             result = Protocol::NOT_FOUND;
  87                         break;
  88                     case Protocol::QUIT:
  89                         print_msg(stdout, "%s (%d): Say no more\n", pname, pid);
  90                         fin = 1;
  91                         fin_cmd = 1;
  92                         break;
  93                     default:
  94                         print_msg(stderr, "%s (%d): Operacion no soportada\n", pname, pid);
  95                 }
  96                 buffer.clear();
  97                 fin_cmd = 1;
  98             }
  99         }
 100         fin_cmd = 0;
 101
 102         // Envío respuesta
 103         libtcp_send(sockfd, (char*) &result, sizeof(int));
 104         print_msg(stdout, "%s (%d) FIN (resultado = %d)\n", pname, pid, result);
 105     }
 106
 107     close(sockfd);
 108
 109 }
 110
 111 // vim: set et sw=4 sts=4 :