feature: add GTP utils, support overwrite message length of GTP header
This commit is contained in:
@@ -4,8 +4,10 @@
|
||||
#include "checksum.h"
|
||||
#include "tcp_utils.h"
|
||||
#include "udp_utils.h"
|
||||
#include "ipv4_utils.h"
|
||||
#include "ipv6_utils.h"
|
||||
#include "ip4_utils.h"
|
||||
#include "ip6_utils.h"
|
||||
#include "gtp1_utils.h"
|
||||
#include "gtp2_utils.h"
|
||||
#include "packet_def.h"
|
||||
#include "packet_utils.h"
|
||||
#include "packet_layer.h"
|
||||
@@ -15,7 +17,7 @@
|
||||
#define PACKET_BUILD_LOG_DEBUG(format, ...) LOG_DEBUG("packet build", format, ##__VA_ARGS__)
|
||||
#define PACKET_BUILD_LOG_ERROR(format, ...) LOG_ERROR("packet build", format, ##__VA_ARGS__)
|
||||
|
||||
struct packet_fingerprint
|
||||
struct fingerprint
|
||||
{
|
||||
// TODO
|
||||
uint16_t ip_id;
|
||||
@@ -23,7 +25,7 @@ struct packet_fingerprint
|
||||
uint16_t tcp_win;
|
||||
};
|
||||
|
||||
static inline void calc_packet_fingerprint(struct packet_fingerprint *finger)
|
||||
static inline void calc_packet_fingerprint(struct fingerprint *finger)
|
||||
{
|
||||
#define RANGE(rand, start, end) (start + rand % (end - start + 1)) // [start, end]
|
||||
struct timespec time;
|
||||
@@ -34,39 +36,143 @@ static inline void calc_packet_fingerprint(struct packet_fingerprint *finger)
|
||||
finger->tcp_win = (uint16_t)(RANGE(random, 1000, 1460));
|
||||
}
|
||||
|
||||
static void update_tcp_hdr(struct tcphdr *tcphdr, uint32_t seq, uint32_t ack, uint16_t win, uint8_t flags, uint16_t opts_len)
|
||||
static void update_tcp_hdr(struct tcphdr *tcp, uint32_t seq, uint32_t ack, uint16_t win, uint8_t flags, uint16_t opts_len)
|
||||
{
|
||||
tcp_hdr_set_seq(tcphdr, seq);
|
||||
tcp_hdr_set_ack(tcphdr, ack);
|
||||
tcp_hdr_set_hdr_len(tcphdr, sizeof(struct tcphdr) + opts_len);
|
||||
tcp_hdr_set_flags(tcphdr, flags);
|
||||
tcp_hdr_set_window(tcphdr, win);
|
||||
tcp_hdr_set_urg_ptr(tcphdr, 0);
|
||||
tcp_hdr_set_checksum(tcphdr, 0);
|
||||
tcp_hdr_set_seq(tcp, seq);
|
||||
tcp_hdr_set_ack(tcp, ack);
|
||||
tcp_hdr_set_hdr_len(tcp, sizeof(struct tcphdr) + opts_len);
|
||||
tcp_hdr_set_flags(tcp, flags);
|
||||
tcp_hdr_set_window(tcp, win);
|
||||
tcp_hdr_set_urg_ptr(tcp, 0);
|
||||
tcp_hdr_set_checksum(tcp, 0);
|
||||
}
|
||||
|
||||
static void update_udp_hdr(struct udphdr *udphdr, int trim)
|
||||
static void update_udp_hdr(struct udphdr *udp, int trim_len)
|
||||
{
|
||||
uint16_t total = udp_hdr_get_total_len(udphdr);
|
||||
udp_hdr_set_total_len(udphdr, total - trim);
|
||||
udp_hdr_set_checksum(udphdr, 0);
|
||||
uint16_t total = udp_hdr_get_total_len(udp);
|
||||
udp_hdr_set_total_len(udp, total - trim_len);
|
||||
udp_hdr_set_checksum(udp, 0);
|
||||
}
|
||||
|
||||
static void update_ip4_hdr(struct ip *iphdr, uint16_t ipid, uint8_t ttl, int trim)
|
||||
static void update_ip4_hdr(struct ip *ip, uint16_t ipid, uint8_t ttl, int trim_len)
|
||||
{
|
||||
int hdr_len = ipv4_hdr_get_hdr_len(iphdr);
|
||||
uint16_t total = ipv4_hdr_get_total_len(iphdr);
|
||||
ipv4_hdr_set_total_len(iphdr, total - trim);
|
||||
ipv4_hdr_set_ipid(iphdr, ipid);
|
||||
ipv4_hdr_set_ttl(iphdr, ttl);
|
||||
iphdr->ip_sum = 0;
|
||||
iphdr->ip_sum = checksum((const char *)iphdr, hdr_len);
|
||||
int hdr_len = ip4_hdr_get_hdr_len(ip);
|
||||
uint16_t total = ip4_hdr_get_total_len(ip);
|
||||
ip4_hdr_set_total_len(ip, total - trim_len);
|
||||
ip4_hdr_set_ipid(ip, ipid);
|
||||
ip4_hdr_set_ttl(ip, ttl);
|
||||
ip->ip_sum = 0;
|
||||
ip->ip_sum = checksum((const char *)ip, hdr_len);
|
||||
}
|
||||
|
||||
static void update_ip6_hdr(struct ip6_hdr *ip6hdr, int trim)
|
||||
static void update_ip6_hdr(struct ip6_hdr *ip6, int trim_len)
|
||||
{
|
||||
uint16_t len = ipv6_hdr_get_payload_len(ip6hdr);
|
||||
ipv6_hdr_set_payload_len(ip6hdr, len - trim);
|
||||
uint16_t len = ip6_hdr_get_payload_len(ip6);
|
||||
ip6_hdr_set_payload_len(ip6, len - trim_len);
|
||||
}
|
||||
|
||||
static void update_gtp1_hdr(struct gtp1_hdr *gtp, int trim_len)
|
||||
{
|
||||
uint16_t msg_len = gtp1_hdr_get_msg_len(gtp);
|
||||
gtp1_hdr_set_msg_len(gtp, msg_len - trim_len);
|
||||
if (gtp1_hdr_get_seq_flag(gtp) && gtp1_hdr_get_seq(gtp))
|
||||
{
|
||||
PACKET_BUILD_LOG_ERROR("imiated packets may be dropped by intermediate devices, the GTPv1 layer requires a sequence number");
|
||||
}
|
||||
}
|
||||
|
||||
static void update_gtp2_hdr(struct gtp2_hdr *gtp, int trim_len)
|
||||
{
|
||||
uint16_t msg_len = gtp2_hdr_get_msg_len(gtp);
|
||||
gtp2_hdr_set_msg_len(gtp, msg_len - trim_len);
|
||||
if (gtp2_hdr_get_seq(gtp))
|
||||
{
|
||||
PACKET_BUILD_LOG_ERROR("imiated packets may be dropped by intermediate devices, the GTPv2 layer requires a sequence number");
|
||||
}
|
||||
}
|
||||
|
||||
static void update_packet_hdr(const struct packet *origin_pkt,
|
||||
char *new_pkt_data, uint16_t new_pkt_len, int trim_len,
|
||||
uint32_t tcp_seq, uint32_t tcp_ack, uint8_t tcp_flags, uint16_t tcp_opts_len)
|
||||
{
|
||||
uint8_t gtp_version = 0;
|
||||
char *curr_hdr_ptr = NULL;
|
||||
struct tcphdr *tcp = NULL;
|
||||
struct udphdr *udp = NULL;
|
||||
struct ip *ip4 = NULL;
|
||||
struct ip6_hdr *ip6 = NULL;
|
||||
struct gtp1_hdr *gtp1 = NULL;
|
||||
struct gtp2_hdr *gtp2 = NULL;
|
||||
struct raw_layer *curr_layer = NULL;
|
||||
struct raw_layer *last_layer = NULL;
|
||||
struct fingerprint finger = {0};
|
||||
calc_packet_fingerprint(&finger);
|
||||
int count = packet_get_layer_count(origin_pkt);
|
||||
for (int i = count - 1; i >= 0; i--)
|
||||
{
|
||||
curr_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i);
|
||||
curr_hdr_ptr = new_pkt_data + curr_layer->hdr_offset;
|
||||
switch (curr_layer->proto)
|
||||
{
|
||||
case LAYER_PROTO_TCP:
|
||||
tcp = (struct tcphdr *)curr_hdr_ptr;
|
||||
update_tcp_hdr(tcp, tcp_seq, tcp_ack, finger.tcp_win, tcp_flags, tcp_opts_len);
|
||||
break;
|
||||
case LAYER_PROTO_UDP:
|
||||
udp = (struct udphdr *)curr_hdr_ptr;
|
||||
update_udp_hdr(udp, trim_len);
|
||||
break;
|
||||
case LAYER_PROTO_IPV4:
|
||||
ip4 = (struct ip *)curr_hdr_ptr;
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp = (struct tcphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
tcp->th_sum = checksum_v4(tcp, new_pkt_len - last_layer->hdr_offset, IPPROTO_TCP, &ip4->ip_src, &ip4->ip_dst);
|
||||
}
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp->uh_sum = checksum_v4(udp, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip4->ip_src, &ip4->ip_dst);
|
||||
}
|
||||
update_ip4_hdr(ip4, finger.ip_id, finger.ip_ttl, trim_len);
|
||||
break;
|
||||
case LAYER_PROTO_IPV6:
|
||||
ip6 = (struct ip6_hdr *)curr_hdr_ptr;
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp = (struct tcphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
tcp->th_sum = checksum_v6(tcp, new_pkt_len - last_layer->hdr_offset, IPPROTO_TCP, &ip6->ip6_src, &ip6->ip6_dst);
|
||||
}
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp->uh_sum = checksum_v6(udp, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip6->ip6_src, &ip6->ip6_dst);
|
||||
}
|
||||
update_ip6_hdr(ip6, trim_len);
|
||||
break;
|
||||
case LAYER_PROTO_GTP_C: /* fall through */
|
||||
case LAYER_PROTO_GTP_U:
|
||||
gtp_version = peek_gtp_version(new_pkt_data + curr_layer->hdr_offset, curr_layer->hdr_len);
|
||||
if (gtp_version == 1)
|
||||
{
|
||||
gtp1 = (struct gtp1_hdr *)curr_hdr_ptr;
|
||||
update_gtp1_hdr(gtp1, trim_len);
|
||||
}
|
||||
if (gtp_version == 2)
|
||||
{
|
||||
gtp2 = (struct gtp2_hdr *)curr_hdr_ptr;
|
||||
update_gtp2_hdr(gtp2, trim_len);
|
||||
}
|
||||
break;
|
||||
case LAYER_PROTO_GRE:
|
||||
// TODO
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
@@ -98,8 +204,8 @@ struct packet *imitate_tcp_packet(const struct packet *origin_pkt, uint32_t tcp_
|
||||
}
|
||||
|
||||
// calculate the new packet length
|
||||
int trim = tcp_layer->hdr_len + tcp_layer->pld_len - tcp_options_len - tcp_payload_len - sizeof(struct tcphdr);
|
||||
uint16_t new_pkt_len = origin_pkt->data_len - origin_pkt->trim_len - trim;
|
||||
int trim_len = tcp_layer->hdr_len + tcp_layer->pld_len - tcp_options_len - tcp_payload_len - sizeof(struct tcphdr);
|
||||
uint16_t new_pkt_len = origin_pkt->data_len - origin_pkt->trim_len - trim_len;
|
||||
struct packet *new_pkt = packet_new(new_pkt_len);
|
||||
if (new_pkt == NULL)
|
||||
{
|
||||
@@ -117,67 +223,7 @@ struct packet *imitate_tcp_packet(const struct packet *origin_pkt, uint32_t tcp_
|
||||
memcpy(new_pkt_data + tcp_layer->hdr_offset + sizeof(struct tcphdr) + tcp_options_len, tcp_payload, tcp_payload_len);
|
||||
|
||||
// update the headers of the new packet
|
||||
struct tcphdr *tcp_hdr = NULL;
|
||||
struct udphdr *udp_hdr = NULL;
|
||||
struct ip *ip4_hdr = NULL;
|
||||
struct ip6_hdr *ip6_hdr = NULL;
|
||||
struct raw_layer *curr_layer = NULL;
|
||||
struct raw_layer *last_layer = NULL;
|
||||
struct packet_fingerprint finger = {0};
|
||||
calc_packet_fingerprint(&finger);
|
||||
for (int i = layers - 1; i >= 0; i--)
|
||||
{
|
||||
curr_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i);
|
||||
switch (curr_layer->proto)
|
||||
{
|
||||
case LAYER_PROTO_TCP:
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
update_tcp_hdr(tcp_hdr, tcp_seq, tcp_ack, finger.tcp_win, tcp_flags, tcp_options_len);
|
||||
break;
|
||||
case LAYER_PROTO_UDP:
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
update_udp_hdr(udp_hdr, trim);
|
||||
break;
|
||||
case LAYER_PROTO_IPV4:
|
||||
ip4_hdr = (struct ip *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
tcp_hdr->th_sum = checksum_v4(tcp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_TCP, &ip4_hdr->ip_src, &ip4_hdr->ip_dst);
|
||||
}
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp_hdr->uh_sum = checksum_v4(udp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip4_hdr->ip_src, &ip4_hdr->ip_dst);
|
||||
}
|
||||
update_ip4_hdr(ip4_hdr, finger.ip_id, finger.ip_ttl, trim);
|
||||
break;
|
||||
case LAYER_PROTO_IPV6:
|
||||
ip6_hdr = (struct ip6_hdr *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
tcp_hdr->th_sum = checksum_v6(tcp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_TCP, &ip6_hdr->ip6_src, &ip6_hdr->ip6_dst);
|
||||
}
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp_hdr->uh_sum = checksum_v6(udp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip6_hdr->ip6_src, &ip6_hdr->ip6_dst);
|
||||
}
|
||||
update_ip6_hdr(ip6_hdr, trim);
|
||||
break;
|
||||
case LAYER_PROTO_GTP:
|
||||
// TODO
|
||||
break;
|
||||
case LAYER_PROTO_GRE:
|
||||
// TODO
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
update_packet_hdr(origin_pkt, new_pkt_data, new_pkt_len, trim_len, tcp_seq, tcp_ack, tcp_flags, tcp_options_len);
|
||||
|
||||
packet_parse(new_pkt, new_pkt_data, new_pkt_len);
|
||||
memcpy(&new_pkt->meta, &origin_pkt->meta, sizeof(struct metadata));
|
||||
@@ -205,8 +251,8 @@ struct packet *imitate_udp_packet(const struct packet *origin_pkt, const char *u
|
||||
}
|
||||
|
||||
// calculate the new packet length
|
||||
int trim = udp_layer->hdr_len + udp_layer->pld_len - udp_payload_len - sizeof(struct udphdr);
|
||||
uint16_t new_pkt_len = origin_pkt->data_len - origin_pkt->trim_len - trim;
|
||||
int trim_len = udp_layer->hdr_len + udp_layer->pld_len - udp_payload_len - sizeof(struct udphdr);
|
||||
uint16_t new_pkt_len = origin_pkt->data_len - origin_pkt->trim_len - trim_len;
|
||||
struct packet *new_pkt = packet_new(new_pkt_len);
|
||||
if (new_pkt == NULL)
|
||||
{
|
||||
@@ -220,52 +266,7 @@ struct packet *imitate_udp_packet(const struct packet *origin_pkt, const char *u
|
||||
memcpy(new_pkt_data + udp_layer->hdr_offset + sizeof(struct udphdr), udp_payload, udp_payload_len);
|
||||
|
||||
// update the headers of the new packet
|
||||
struct udphdr *udp_hdr = NULL;
|
||||
struct ip *ip4_hdr = NULL;
|
||||
struct ip6_hdr *ip6_hdr = NULL;
|
||||
struct raw_layer *curr_layer = NULL;
|
||||
struct raw_layer *last_layer = NULL;
|
||||
struct packet_fingerprint finger = {0};
|
||||
calc_packet_fingerprint(&finger);
|
||||
for (int i = layers - 1; i >= 0; i--)
|
||||
{
|
||||
curr_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i);
|
||||
switch (curr_layer->proto)
|
||||
{
|
||||
case LAYER_PROTO_UDP:
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
update_udp_hdr(udp_hdr, trim);
|
||||
break;
|
||||
case LAYER_PROTO_IPV4:
|
||||
ip4_hdr = (struct ip *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp_hdr->uh_sum = checksum_v4(udp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip4_hdr->ip_src, &ip4_hdr->ip_dst);
|
||||
}
|
||||
update_ip4_hdr(ip4_hdr, finger.ip_id, finger.ip_ttl, trim);
|
||||
break;
|
||||
case LAYER_PROTO_IPV6:
|
||||
ip6_hdr = (struct ip6_hdr *)(new_pkt_data + curr_layer->hdr_offset);
|
||||
last_layer = (struct raw_layer *)packet_get_raw_layer(origin_pkt, i + 1);
|
||||
if (last_layer->proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + last_layer->hdr_offset);
|
||||
udp_hdr->uh_sum = checksum_v6(udp_hdr, new_pkt_len - last_layer->hdr_offset, IPPROTO_UDP, &ip6_hdr->ip6_src, &ip6_hdr->ip6_dst);
|
||||
}
|
||||
update_ip6_hdr(ip6_hdr, trim);
|
||||
break;
|
||||
case LAYER_PROTO_GTP:
|
||||
// TODO
|
||||
break;
|
||||
case LAYER_PROTO_GRE:
|
||||
// TODO
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
update_packet_hdr(origin_pkt, new_pkt_data, new_pkt_len, trim_len, 0, 0, 0, 0);
|
||||
|
||||
packet_parse(new_pkt, new_pkt_data, new_pkt_len);
|
||||
memcpy(&new_pkt->meta, &origin_pkt->meta, sizeof(struct metadata));
|
||||
@@ -313,77 +314,97 @@ struct packet *craft_packet_from_scratch(const struct layer layers[], uint16_t l
|
||||
memcpy(new_pkt_data + offset, payload, payload_len);
|
||||
|
||||
// update the headers of the new packet
|
||||
struct tcphdr *tcp_hdr = NULL;
|
||||
struct udphdr *udp_hdr = NULL;
|
||||
struct ip *ip4_hdr = NULL;
|
||||
struct ip6_hdr *ip6_hdr = NULL;
|
||||
uint8_t gtp_version = 0;
|
||||
uint16_t curr_hdr_len = 0;
|
||||
char *curr_hdr_ptr = NULL;
|
||||
struct tcphdr *tcp = NULL;
|
||||
struct udphdr *udp = NULL;
|
||||
struct ip *ip4 = NULL;
|
||||
struct ip6_hdr *ip6 = NULL;
|
||||
struct gtp1_hdr *gtp1 = NULL;
|
||||
struct gtp2_hdr *gtp2 = NULL;
|
||||
// update checksums and lengths
|
||||
uint16_t curr_layer_payload_len = payload_len;
|
||||
uint16_t curr_payload_len = payload_len;
|
||||
for (int i = layer_count - 1; i >= 0; i--)
|
||||
{
|
||||
curr_hdr_len = layers[i].hdr_len;
|
||||
curr_hdr_ptr = new_pkt_data + new_pkt_len - curr_hdr_len - curr_payload_len;
|
||||
switch (layers[i].proto)
|
||||
{
|
||||
case LAYER_PROTO_TCP:
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + new_pkt_len - layers[i].hdr_len - curr_layer_payload_len);
|
||||
tcp = (struct tcphdr *)curr_hdr_ptr;
|
||||
// update the TCP header
|
||||
tcp_hdr_set_hdr_len(tcp_hdr, layers[i].hdr_len);
|
||||
tcp_hdr_set_checksum(tcp_hdr, 0);
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
tcp_hdr_set_hdr_len(tcp, curr_hdr_len);
|
||||
tcp_hdr_set_checksum(tcp, 0);
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
case LAYER_PROTO_UDP:
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + new_pkt_len - layers[i].hdr_len - curr_layer_payload_len);
|
||||
udp = (struct udphdr *)curr_hdr_ptr;
|
||||
// update the UDP header
|
||||
udp_hdr_set_total_len(udp_hdr, layers[i].hdr_len + curr_layer_payload_len);
|
||||
udp_hdr_set_checksum(udp_hdr, 0);
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
udp_hdr_set_total_len(udp, curr_hdr_len + curr_payload_len);
|
||||
udp_hdr_set_checksum(udp, 0);
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
case LAYER_PROTO_IPV4:
|
||||
ip4_hdr = (struct ip *)(new_pkt_data + new_pkt_len - layers[i].hdr_len - curr_layer_payload_len);
|
||||
ip4 = (struct ip *)curr_hdr_ptr;
|
||||
// update the checksums of the upper layer
|
||||
if (i + 1 < layer_count && layers[i + 1].proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + new_pkt_len - curr_layer_payload_len);
|
||||
tcp_hdr->th_sum = checksum_v4(tcp_hdr, curr_layer_payload_len, IPPROTO_TCP, &ip4_hdr->ip_src, &ip4_hdr->ip_dst);
|
||||
tcp = (struct tcphdr *)(new_pkt_data + new_pkt_len - curr_payload_len);
|
||||
tcp->th_sum = checksum_v4(tcp, curr_payload_len, IPPROTO_TCP, &ip4->ip_src, &ip4->ip_dst);
|
||||
}
|
||||
if (i + 1 < layer_count && layers[i + 1].proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + new_pkt_len - curr_layer_payload_len);
|
||||
udp_hdr->uh_sum = checksum_v4(udp_hdr, curr_layer_payload_len, IPPROTO_UDP, &ip4_hdr->ip_src, &ip4_hdr->ip_dst);
|
||||
udp = (struct udphdr *)(new_pkt_data + new_pkt_len - curr_payload_len);
|
||||
udp->uh_sum = checksum_v4(udp, curr_payload_len, IPPROTO_UDP, &ip4->ip_src, &ip4->ip_dst);
|
||||
}
|
||||
// update the IPv4 header
|
||||
ipv4_hdr_set_hdr_len(ip4_hdr, layers[i].hdr_len);
|
||||
ipv4_hdr_set_total_len(ip4_hdr, layers[i].hdr_len + curr_layer_payload_len);
|
||||
ip4_hdr->ip_sum = 0;
|
||||
ip4_hdr->ip_sum = checksum((const char *)ip4_hdr, layers[i].hdr_len);
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
ip4_hdr_set_hdr_len(ip4, curr_hdr_len);
|
||||
ip4_hdr_set_total_len(ip4, curr_hdr_len + curr_payload_len);
|
||||
ip4->ip_sum = 0;
|
||||
ip4->ip_sum = checksum((const char *)ip4, curr_hdr_len);
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
case LAYER_PROTO_IPV6:
|
||||
ip6_hdr = (struct ip6_hdr *)(new_pkt_data + new_pkt_len - layers[i].hdr_len - curr_layer_payload_len);
|
||||
ip6 = (struct ip6_hdr *)curr_hdr_ptr;
|
||||
// update the checksums of the upper layer
|
||||
if (i + 1 < layer_count && layers[i + 1].proto == LAYER_PROTO_TCP)
|
||||
{
|
||||
tcp_hdr = (struct tcphdr *)(new_pkt_data + new_pkt_len - curr_layer_payload_len);
|
||||
tcp_hdr->th_sum = checksum_v6(tcp_hdr, curr_layer_payload_len, IPPROTO_TCP, &ip6_hdr->ip6_src, &ip6_hdr->ip6_dst);
|
||||
tcp = (struct tcphdr *)(new_pkt_data + new_pkt_len - curr_payload_len);
|
||||
tcp->th_sum = checksum_v6(tcp, curr_payload_len, IPPROTO_TCP, &ip6->ip6_src, &ip6->ip6_dst);
|
||||
}
|
||||
if (i + 1 < layer_count && layers[i + 1].proto == LAYER_PROTO_UDP)
|
||||
{
|
||||
udp_hdr = (struct udphdr *)(new_pkt_data + new_pkt_len - curr_layer_payload_len);
|
||||
udp_hdr->uh_sum = checksum_v6(udp_hdr, curr_layer_payload_len, IPPROTO_UDP, &ip6_hdr->ip6_src, &ip6_hdr->ip6_dst);
|
||||
udp = (struct udphdr *)(new_pkt_data + new_pkt_len - curr_payload_len);
|
||||
udp->uh_sum = checksum_v6(udp, curr_payload_len, IPPROTO_UDP, &ip6->ip6_src, &ip6->ip6_dst);
|
||||
}
|
||||
// update the IPv6 header
|
||||
ipv6_hdr_set_payload_len(ip6_hdr, layers[i].hdr_len + curr_layer_payload_len - sizeof(struct ip6_hdr));
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
ip6_hdr_set_payload_len(ip6, curr_hdr_len + curr_payload_len - sizeof(struct ip6_hdr));
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
case LAYER_PROTO_GTP:
|
||||
// TODO
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
case LAYER_PROTO_GTP_C: /* fall through */
|
||||
case LAYER_PROTO_GTP_U:
|
||||
gtp_version = peek_gtp_version(curr_hdr_ptr, curr_hdr_len);
|
||||
if (gtp_version == 1)
|
||||
{
|
||||
gtp1 = (struct gtp1_hdr *)curr_hdr_ptr;
|
||||
// update the GTP header
|
||||
gtp1_hdr_set_msg_len(gtp1, curr_hdr_len + curr_payload_len - sizeof(struct gtp1_hdr));
|
||||
}
|
||||
if (gtp_version == 2)
|
||||
{
|
||||
gtp2 = (struct gtp2_hdr *)curr_hdr_ptr;
|
||||
// update the GTP header
|
||||
gtp2_hdr_set_msg_len(gtp2, curr_hdr_len + curr_payload_len - 4);
|
||||
}
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
case LAYER_PROTO_GRE:
|
||||
// TODO
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
default:
|
||||
curr_layer_payload_len += layers[i].hdr_len;
|
||||
curr_payload_len += curr_hdr_len;
|
||||
break;
|
||||
}
|
||||
}
|
||||
@@ -393,4 +414,4 @@ struct packet *craft_packet_from_scratch(const struct layer layers[], uint16_t l
|
||||
new_pkt->meta.origin_ctx = NULL;
|
||||
|
||||
return new_pkt;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user