|
@@ -273,28 +273,36 @@ static int set_eth_addr(struct sk_buff *skb, struct sw_flow_key *flow_key,
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
-static void set_ip_addr(struct sk_buff *skb, struct iphdr *nh,
|
|
|
- __be32 *addr, __be32 new_addr)
|
|
|
+static void update_ip_l4_checksum(struct sk_buff *skb, struct iphdr *nh,
|
|
|
+ __be32 addr, __be32 new_addr)
|
|
|
{
|
|
|
int transport_len = skb->len - skb_transport_offset(skb);
|
|
|
|
|
|
+ if (nh->frag_off & htons(IP_OFFSET))
|
|
|
+ return;
|
|
|
+
|
|
|
if (nh->protocol == IPPROTO_TCP) {
|
|
|
if (likely(transport_len >= sizeof(struct tcphdr)))
|
|
|
inet_proto_csum_replace4(&tcp_hdr(skb)->check, skb,
|
|
|
- *addr, new_addr, 1);
|
|
|
+ addr, new_addr, 1);
|
|
|
} else if (nh->protocol == IPPROTO_UDP) {
|
|
|
if (likely(transport_len >= sizeof(struct udphdr))) {
|
|
|
struct udphdr *uh = udp_hdr(skb);
|
|
|
|
|
|
if (uh->check || skb->ip_summed == CHECKSUM_PARTIAL) {
|
|
|
inet_proto_csum_replace4(&uh->check, skb,
|
|
|
- *addr, new_addr, 1);
|
|
|
+ addr, new_addr, 1);
|
|
|
if (!uh->check)
|
|
|
uh->check = CSUM_MANGLED_0;
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+}
|
|
|
|
|
|
+static void set_ip_addr(struct sk_buff *skb, struct iphdr *nh,
|
|
|
+ __be32 *addr, __be32 new_addr)
|
|
|
+{
|
|
|
+ update_ip_l4_checksum(skb, nh, *addr, new_addr);
|
|
|
csum_replace4(&nh->check, *addr, new_addr);
|
|
|
skb_clear_hash(skb);
|
|
|
*addr = new_addr;
|