practicas/pipi/src/ipout.cpp

   1
   2 #include "ipout.h"
   3 #include "ipheader.h"
   4 #include <ctime>
   5 #include <unistd.h>
   6 #include <fcntl.h>
   7 #include <sys/types.h>
   8 #include <sys/ipc.h>
   9 #include <sys/msg.h>
  10 #ifdef DEBUG
  11 #include <iostream>
  12 #endif
  13
  14 /// Constructor
  15 IPOut::IPOut(const IPAddr& ip, RouteTable& rtable, Dev& forward_que, std::ostream& log):
  16     ip(ip), rtable(rtable), forward_que(forward_que), log(log)
  17 {
  18 }
  19
  20 void IPOut::drop(const std::string& msg, const std::string& buf)
  21 {
  22     log << "IPOut::drop (" << ip << "): " << msg << "\n\tBuffer: " << buf
  23             << "\n";
  24 }
  25
  26 void IPOut::drop(const std::string& msg, const IPHeader& iph)
  27 {
  28     log << "IPOut::drop (" << ip << "): " << msg << "\n\tIPHeader: " << iph
  29             << "\n";
  30 }
  31
  32 /// Envía un paquete IP
  33 bool IPOut::send(const std::string& data, uint8_t proto, IPAddr dst, IPAddr src,
  34         bool df, uint8_t ttl, uint16_t id)
  35     throw (std::runtime_error)
  36 {
  37     // Mando todo lo que tengo para forwardear
  38     while (to_forward())
  39     {
  40         std::string buf = forward_que.receive();
  41         IPHeader iph(buf);
  42 #ifdef DEBUG
  43         std::cout << "IPOut::send (" << ip << "): A forwardear\n";
  44 #endif
  45         send(iph, buf.substr(iph.header_len()));
  46     }
  47     // Mando el paquete en sí
  48     // Armamos cabecera
  49     if (!src)
  50         src = ip;
  51     if (!id)
  52         id = get_id();
  53     IPHeader iph(4, IPHeader::header_len() + data.size(), id, df, 0, 0,
  54             ttl, proto, src, dst);
  55     return send(iph, data);
  56 }
  57
  58 /// Envía un paquete IP
  59 bool IPOut::send(IPHeader iph, std::string data) throw (std::runtime_error)
  60 {
  61     // Buscamos ruta
  62     RouteTable::Route* r = rtable.get(iph.dst);
  63     if (!r)
  64     {
  65         // ICMP
  66         drop("No existe una ruta para el destino -> ICMP", iph);
  67         return false;
  68     }
  69     // No quieren fragmentar
  70     if (iph.df && (IPHeader::header_len() + data.size() > r->iface->mtu))
  71     {
  72         // Silencioso
  73         drop("Tamaño de paquete más grande que MTU y DF=1", iph);
  74         return false;
  75     }
  76     // Fragmenta (de ser necesario)
  77     int max_payload = r->iface->mtu - IPHeader::header_len();
  78     int cant_frag = data.size() / max_payload;
  79     if (data.size() % max_payload)
  80         ++cant_frag;
  81     for (int i = 0; i < cant_frag; ++i)
  82     {
  83         IPHeader iph2 = iph;
  84         if (i != (cant_frag - 1))
  85             iph2.mf = 1;
  86         iph2.offset += i * max_payload;
  87         iph2.total_len -= i * max_payload;
  88         iph2.do_checksum();
  89         std::string buf((char*) &iph2, sizeof(IPHeader));
  90         buf += data.substr(i * max_payload, max_payload);
  91 #ifdef DEBUG
  92         std::cout << "IPOut::send (" << ip << "): Fragmento " << i
  93                 << " => IPHeader: " << iph2 << "\n";
  94         std::string tmp = data.substr(i * max_payload, max_payload);
  95         std::cout << "\tdata (" << tmp.size() << ") = " << tmp << "\n";
  96 #endif
  97         r->iface->transmit(buf, r->gateway ? r->gateway : IPAddr(iph.dst));
  98     }
  99     return true;
 100 }
 101
 102 /// Obtiene un identificador para el paquete
 103 uint16_t IPOut::get_id() const
 104 {
 105     return time(NULL);
 106 }
 107
 108 /// Se fija si hay paquetes a forwardear (y devuelve cuantos hay)
 109 unsigned IPOut::to_forward()
 110 {
 111     struct msqid_ds minfo;
 112     msgctl(forward_que.que_id, IPC_STAT, &minfo);
 113     return minfo.msg_qnum;
 114 }
 115
 116 // vim: set et sw=4 sts=4 :