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         drop("No existe una ruta para el destino", iph);
  66         return false;
  67     }
  68     // No quieren fragmentar
  69     if (iph.df && (IPHeader::header_len() + data.size() > r->iface->mtu))
  70     {
  71         drop("Tamaño de paquete más grande que MTU y DF=1", iph);
  72         return false;
  73     }
  74     // Fragmenta (de ser necesario)
  75     int max_payload = r->iface->mtu - IPHeader::header_len();
  76     int cant_frag = data.size() / max_payload;
  77     if (data.size() % max_payload)
  78         ++cant_frag;
  79     for (int i = 0; i < cant_frag; ++i)
  80     {
  81         IPHeader iph2 = iph;
  82         if (i != (cant_frag - 1))
  83             iph2.mf = 1;
  84         iph2.offset += i * max_payload;
  85         iph2.total_len -= i * max_payload;
  86         iph2.do_checksum();
  87         std::string buf((char*) &iph2, sizeof(IPHeader));
  88         buf += data.substr(i * max_payload, max_payload);
  89 #ifdef DEBUG
  90         std::cout << "IPOut::send (" << ip << "): Fragmento " << i
  91                 << " => IPHeader: " << iph2 << "\n";
  92         std::string tmp = data.substr(i * max_payload, max_payload);
  93         std::cout << "\tdata (" << tmp.size() << ") = " << tmp << "\n";
  94 #endif
  95         r->iface->transmit(buf, r->gateway ? r->gateway : IPAddr(iph.dst));
  96     }
  97     return true;
  98 }
  99
 100 /// Obtiene un identificador para el paquete
 101 uint16_t IPOut::get_id() const
 102 {
 103     return time(NULL);
 104 }
 105
 106 /// Se fija si hay paquetes a forwardear (y devuelve cuantos hay)
 107 unsigned IPOut::to_forward()
 108 {
 109     struct msqid_ds minfo;
 110     msgctl(forward_que.que_id, IPC_STAT, &minfo);
 111     return minfo.msg_qnum;
 112 }
 113
 114 // vim: set et sw=4 sts=4 :