2 * netsniff-ng - the packet sniffing beast
3 * Subject to the GPL, version 2.
6 #include <linux/if_ether.h>
11 #include "trafgen_l2.h"
12 #include "trafgen_l3.h"
13 #include "trafgen_proto.h"
14 #include "trafgen_conf.h"
16 static struct proto_field ipv4_fields
[] = {
17 { .id
= IP4_VER
, .len
= 1, .offset
= 0, .shift
= 4, .mask
= 0xf0 },
18 { .id
= IP4_IHL
, .len
= 1, .offset
= 0, .shift
= 0, .mask
= 0x0f },
19 { .id
= IP4_DSCP
, .len
= 1, .offset
= 1, .shift
= 2, .mask
= 0xfc },
20 { .id
= IP4_ECN
, .len
= 1, .offset
= 1, .shift
= 0, .mask
= 0x03 },
21 { .id
= IP4_TOS
, .len
= 1, .offset
= 1 },
22 { .id
= IP4_LEN
, .len
= 2, .offset
= 2 },
23 { .id
= IP4_ID
, .len
= 2, .offset
= 4 },
24 { .id
= IP4_FLAGS
, .len
= 2, .offset
= 6, .shift
= 13, .mask
= 0xe000 },
25 { .id
= IP4_MF
, .len
= 2, .offset
= 6, .shift
= 13, .mask
= 0x2000 },
26 { .id
= IP4_DF
, .len
= 2, .offset
= 6, .shift
= 14, .mask
= 0x4000 },
27 { .id
= IP4_FRAG_OFFS
, .len
= 2, .offset
= 6, .shift
= 0, .mask
= 0x1fff },
28 { .id
= IP4_TTL
, .len
= 1, .offset
= 8 },
29 { .id
= IP4_PROTO
, .len
= 1, .offset
= 9 },
30 { .id
= IP4_CSUM
, .len
= 2, .offset
= 10 },
31 { .id
= IP4_SADDR
, .len
= 4, .offset
= 12 },
32 { .id
= IP4_DADDR
, .len
= 4, .offset
= 16 },
35 static void ipv4_header_init(struct proto_hdr
*hdr
)
37 proto_lower_default_add(hdr
, PROTO_ETH
);
39 proto_header_fields_add(hdr
, ipv4_fields
, array_size(ipv4_fields
));
41 proto_field_set_default_u8(hdr
, IP4_VER
, 4);
42 proto_field_set_default_u8(hdr
, IP4_IHL
, 5);
43 proto_field_set_default_dev_ipv4(hdr
, IP4_SADDR
);
46 static void ipv4_packet_finish(struct proto_hdr
*hdr
)
48 struct packet
*pkt
= current_packet();
51 total_len
= pkt
->len
- hdr
->pkt_offset
;
52 proto_field_set_default_be16(hdr
, IP4_LEN
, total_len
);
54 if (!proto_field_is_set(hdr
, IP4_CSUM
)) {
57 csum
= htons(calc_csum(&pkt
->payload
[hdr
->pkt_offset
], hdr
->len
));
58 proto_field_set_u16(hdr
, IP4_CSUM
, bswap_16(csum
));
62 static void ipv4_set_next_proto(struct proto_hdr
*hdr
, enum proto_id pid
)
68 ip_proto
= IPPROTO_IPIP
;
71 ip_proto
= IPPROTO_IPV6
;
74 ip_proto
= IPPROTO_ICMP
;
77 ip_proto
= IPPROTO_UDP
;
80 ip_proto
= IPPROTO_TCP
;
86 proto_field_set_default_u8(hdr
, IP4_PROTO
, ip_proto
);
89 static const struct proto_ops ipv4_proto_ops
= {
92 .header_init
= ipv4_header_init
,
93 .packet_finish
= ipv4_packet_finish
,
94 .set_next_proto
= ipv4_set_next_proto
,
97 static struct proto_field ipv6_fields
[] = {
98 { .id
= IP6_VER
, .len
= 4, .offset
= 0, .shift
= 28, .mask
= 0xf0000000 },
99 { .id
= IP6_CLASS
, .len
= 4, .offset
= 0, .shift
= 20, .mask
= 0x0ff00000 },
100 { .id
= IP6_FLOW_LBL
, .len
= 4, .offset
= 0, .shift
= 0, .mask
= 0x000fffff },
101 { .id
= IP6_LEN
, .len
= 2, .offset
= 4 },
102 { .id
= IP6_NEXT_HDR
, .len
= 1, .offset
= 6 },
103 { .id
= IP6_HOP_LIMIT
, .len
= 1, .offset
= 7 },
104 { .id
= IP6_SADDR
, .len
= 16, .offset
= 8 },
105 { .id
= IP6_DADDR
, .len
= 16, .offset
= 24 },
108 static void ipv6_header_init(struct proto_hdr
*hdr
)
110 proto_lower_default_add(hdr
, PROTO_ETH
);
112 proto_header_fields_add(hdr
, ipv6_fields
, array_size(ipv6_fields
));
114 proto_field_set_default_be32(hdr
, IP6_VER
, 6);
115 proto_field_set_default_dev_ipv6(hdr
, IP6_SADDR
);
118 #define IPV6_HDR_LEN 40
120 static void ipv6_packet_finish(struct proto_hdr
*hdr
)
122 struct packet
*pkt
= current_packet();
123 uint16_t total_len
= pkt
->len
- hdr
->pkt_offset
- IPV6_HDR_LEN
;
125 proto_field_set_default_be16(hdr
, IP6_LEN
, total_len
);
128 static void ipv6_set_next_proto(struct proto_hdr
*hdr
, enum proto_id pid
)
134 ip_proto
= IPPROTO_ICMPV6
;
137 ip_proto
= IPPROTO_UDP
;
140 ip_proto
= IPPROTO_TCP
;
146 proto_field_set_default_u8(hdr
, IP6_NEXT_HDR
, ip_proto
);
149 static const struct proto_ops ipv6_proto_ops
= {
152 .header_init
= ipv6_header_init
,
153 .packet_finish
= ipv6_packet_finish
,
154 .set_next_proto
= ipv6_set_next_proto
,
157 void protos_l3_init(void)
159 proto_ops_register(&ipv4_proto_ops
);
160 proto_ops_register(&ipv6_proto_ops
);