practicas/pipi-2da-entrega/src/ipin.cpp

   1
   2 #include "ipin.h"
   3 #include "ipheader.h"
   4 #ifdef DEBUG
   5 #include <iostream>
   6 #endif
   7
   8 /// Constructor
   9 IPIn::IPIn(const IPAddr& ip, Dev& dev, Dev& forward_que, bool router,
  10         bool forward, std::ostream& log):
  11     ip(ip), dev(dev), forward_que(forward_que), router(router),
  12     forward(forward), log(log)
  13 {
  14     if (router) forward = true;
  15 }
  16
  17 void IPIn::drop(const std::string& msg, const std::string& buf)
  18 {
  19     log << "IPIn::drop (" << ip << "): " << msg << "\n\tBuffer: " << buf
  20             << "\n";
  21 }
  22
  23 void IPIn::drop(const std::string& msg, const IPHeader& iph)
  24 {
  25     log << "IPIn::drop (" << ip << "): " << msg << "\n\tIPHeader: " << iph
  26             << "\n";
  27 }
  28
  29 /// Recibe un paquete IP
  30 std::string IPIn::recv(uint8_t& proto, IPAddr& src, IPAddr& dst)
  31     throw (std::runtime_error)
  32 {
  33     while (true)
  34     {
  35         std::string buf = dev.receive();
  36         // No es siquiera IP
  37         if (buf.size() < IPHeader::header_len())
  38         {
  39             // Silencioso
  40             drop("Cabecera incompleta o no es IP", buf);
  41             continue;
  42         }
  43         IPHeader iph(buf);
  44 #ifdef DEBUG
  45         std::cout << "IPIn::recv (" << ip << "): IPHeader: " << iph << "\n";
  46         std::string tmp = buf.substr(iph.header_len());
  47         std::cout << "\tdata (" << tmp.size() << ") = " << tmp << "\n";
  48 #endif
  49         if (iph.version != 4)
  50         {
  51             // Silencioso
  52             drop("Versión IP incorrecta", iph);
  53             continue;
  54         }
  55         if (!iph.check_checksum())
  56         {
  57             // Silencioso
  58             drop("Mal checksum", iph);
  59             continue;
  60         }
  61         // Si el TTL se va a 0
  62         if (!--iph.ttl)
  63         {
  64             // ICMP
  65             drop("TTL == 0 -> ICMP", iph);
  66             continue;
  67         }
  68         // No es para nosotros y no forwardeamos
  69         if (iph.dst != ip && !forward)
  70         {
  71             // Silencioso
  72             drop("No es para nosotros y no hacemos forward", iph);
  73             continue;
  74         }
  75         // No es para nosotros pero forwardeamos
  76         else if (iph.dst != ip)
  77         {
  78             forward_que.transmit(buf, ip);
  79             continue;
  80         }
  81         // Es para nosotros pero somos router
  82         else if (router)
  83         {
  84             // Silencioso
  85             drop("Es para nosotros pero somos un router", iph);
  86             continue;
  87         }
  88         // Es para nosotros y somos un host
  89         // Guarda en buffer
  90         buffer[iph][iph.offset] = buf.substr(iph.header_len());
  91         // Si tiene más fragmentos, sigo
  92         if (iph.mf)
  93             continue;
  94         // No hay más fragmentos, reensamblamos (de ser necesario)
  95         std::string data;
  96         for (offsetmap_type::iterator i = buffer[iph].begin();
  97                 i != buffer[iph].end(); ++i)
  98         {
  99             //TODO chequear que los fragmentos estén todos
 100             data += i->second;
 101         }
 102 #ifdef DEBUG
 103         std::cout << "IPIn::recv (" << ip << "): Paquete completo: data = '"
 104                 << data << "'\n";
 105 #endif
 106         buffer.erase(iph);
 107         //TODO faltaría limpiar fragmentos viejos cada tanto (timer?)
 108         src = iph.src;
 109         dst = iph.dst;
 110         proto = iph.proto;
 111         return data;
 112     }
 113 }
 114
 115 // vim: set et sw=4 sts=4 :