resources/net/arp.h

   1 #ifndef _ARP_H_
   2 #define _ARP_H_
   3
   4 #include "arch/low/general.h"
   5 #include "libs/buffer.h"
   6 #include "mac_addr.h"
   7 #include "arp_table.h"
   8 #include "internet_layer.h"
   9 #include "link_layer.h"
  10
  11 namespace net {
  12         class arp : public osi_layer {
  13         private:
  14                 p<link_layer> down;
  15
  16                 /* EtherType */
  17                 enum {
  18                         et_arp = 0x806
  19                 };
  20
  21                 enum {
  22                         ethernet = 1
  23                 };
  24                 enum protocol {
  25                         ipv4 = 0x0800
  26                 };
  27                 enum {
  28                         request = 1,
  29                         reply = 2
  30                 };
  31
  32                 mac_addr my_haddr;
  33                 ipv4_addr my_paddr;
  34
  35                 struct arp_packet {
  36                         u16 htype;
  37                         u16 ptype;
  38                         u8 hlen;
  39                         u8 plen;
  40                         u16 oper;
  41                         u32 sha0;
  42                         u16 sha1;
  43                         u32 spa;
  44                         u16 tha0;
  45                         u32 tha1;
  46                         u32 tpa;
  47                 } __attribute__((packed));
  48
  49
  50                 volatile bool received;
  51
  52                 arp_table table;
  53
  54                 void receive_data(const buffer &x);
  55         public:
  56                 void set_link_layer(p<link_layer> x) {
  57                         down = x;
  58                         my_haddr = x->get_haddr();
  59                         my_paddr = ipv4_addr::from_le(192 << 24 | 168 << 16 | 1 << 8 | 50);
  60
  61                         x->add_handler(et_arp, delegate<void, const buffer&>::method(this, &arp::receive_data));
  62                 }
  63
  64                 void set_ipv4(const ipv4_addr&);
  65
  66                 mac_addr get_haddr(const ipv4_addr &ipaddr);
  67                 void receive_packet(p<arp_packet> pkg);
  68
  69                 //static void register_type();
  70         };
  71 }
  72
  73 #endif