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