bpf: allow bpf_csum_diff to feed bpf_l3_csum_replace as well

[karo-tx-linux.git] / net / core / filter.c

diff --git a/net/core/filter.c b/net/core/filter.c
index 69f4ffc0a282fbc57a37fb453a7123c280a309e2..356a251657a5c14fe10fa8197397e70aad092063 100644 (file)
--- a/net/core/filter.c
+++ b/net/core/filter.c
@@ -1447,6 +1447,12 @@ static u64 bpf_l3_csum_replace(u64 r1, u64 r2, u64 from, u64 to, u64 flags)
                return -EFAULT;
 
        switch (flags & BPF_F_HDR_FIELD_MASK) {
+       case 0:
+               if (unlikely(from != 0))
+                       return -EINVAL;
+
+               csum_replace_by_diff(ptr, to);
+               break;
        case 2:
                csum_replace2(ptr, from, to);
                break;