From 67ab6171729cfcf7867cbb92a2099d88d86f6e6b Mon Sep 17 00:00:00 2001 From: David Gibson Date: Wed, 19 Oct 2022 11:43:45 +1100 Subject: Add csum_icmp4() helper for calculating ICMP checksums Although tap_ip_send() is currently the only place calculating ICMP checksums, create a helper function for symmetry with ICMPv6. For future flexibility it allows the ICMPv6 header and payload to be in separate buffers. Signed-off-by: David Gibson Signed-off-by: Stefano Brivio --- checksum.c | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'checksum.c') diff --git a/checksum.c b/checksum.c index 78c6960..f35c948 100644 --- a/checksum.c +++ b/checksum.c @@ -52,6 +52,7 @@ #include #include +#include #include /** @@ -107,6 +108,21 @@ uint16_t csum_unaligned(const void *buf, size_t len, uint32_t init) return (uint16_t)~csum_fold(sum_16b(buf, len) + init); } +/** + * csum_icmp4() - Calculate and set checksum for an ICMP packet + * @icmp4hr: ICMP header, initialised apart from checksum + * @payload: ICMP packet payload + * @len: Length of @payload (not including ICMP header) + */ +void csum_icmp4(struct icmphdr *icmp4hr, const void *payload, size_t len) +{ + /* Partial checksum for ICMP header alone */ + uint32_t psum = sum_16b(icmp4hr, sizeof(*icmp4hr)); + + icmp4hr->checksum = 0; + icmp4hr->checksum = csum_unaligned(payload, len, psum); +} + /** * csum_icmp6() - Calculate and set checksum for an ICMPv6 packet * @icmp6hr: ICMPv6 header, initialised apart from checksum -- cgit v1.2.3