src/c/udp.c

   1 #include "ip.h"
   2 #include "udp.h"
   3
   4 uint16 udp_port_local;
   5
   6 uint16 udp_port_remote;
   7
   8 byte udp_dgram_len;
   9
  10 /* para calcular checksum */
  11 uint16 checksum;
  12
  13 /* agrega un word al checksum calculado */
  14 void udp_checksum_sum(uint16 w)
  15 {
  16         checksum += w;
  17         if (checksum < w) /* corrección de carry (hubo OV) */
  18                 ++checksum;
  19 }
  20
  21 /* indica si el checksum calculado está ok */
  22 bool udp_checksum_ok()
  23 {
  24         return ~checksum;
  25 }
  26
  27 bool udp_read_dgram_header()
  28 {
  29         /* reseteamos checksum */
  30         checksum = 0;
  31         /* el UDP tiene un checksum que incluye parte de la cabecera IP */
  32         /* ip de origen */
  33         udp_checksum_sum(WORD(ip_addr_remote[0], ip_addr_remote[1]));
  34         udp_checksum_sum(WORD(ip_addr_remote[2], ip_addr_remote[3]));
  35         /* ip de destino */
  36         udp_checksum_sum(WORD(ip_addr_local[0], ip_addr_local[1]));
  37         udp_checksum_sum(WORD(ip_addr_local[2], ip_addr_local[3]));
  38         /* protocolo expresado en 16 bits (0x11 es UDP) */
  39         udp_checksum_sum(0x11);
  40         /* tamaño del paquete UDP (sin las cabeceras que son 20 bytes) */
  41         udp_checksum_sum(ip_packet_len - 20);
  42         /* de ahora en más todos los datos del checksum corresponden a UDP */
  43         /* puerto origen (remoto) */
  44         udp_port_remote = net_getw();
  45         /* agregamos puerto de origen al checksum */
  46         udp_checksum_sum(udp_port_remote);
  47         /* sólo aceptamos datagramas a nuestro puerto */
  48         if (net_getw() != udp_port_local)
  49                 return false; /* drop */
  50         /* agregamos puerto de destino al checksum */
  51         udp_checksum_sum(udp_port_local);
  52         /* tamaño del datagrama */
  53         if (net_getb()) /* no soportamos más de 255 bytes */
  54                 return false; /* drop */
  55         udp_dgram_len = net_getb();
  56         /* agregamos tamaño al checksum */
  57         udp_checksum_sum(udp_dgram_len);
  58         /* agregamos checksum al checksum */
  59         udp_checksum_sum(net_getw());
  60         /* falta agregar el cuerpo del mensaje para verificar la suma
  61          * esto debe hacerlo el protocolo que sigue para poder seguir obteniendo
  62          * los datos de la placa de red byte a byte */
  63         return true;
  64 }
  65
  66 void udp_write_dgram_header()
  67 {
  68         /* puerto origen */
  69         net_putw(udp_port_local);
  70         /* puerto destino */
  71         net_putw(udp_port_remote);
  72         /* tamaño del datagrama */
  73         net_putb(0x00); /* parte alta en 0 porque no soportamos más de 255 */
  74         net_putb(udp_dgram_len);
  75         /* indicamos que no se usa checksum */
  76         net_putw(0x0000);
  77         return true;
  78 }
  79