Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net

[firefly-linux-kernel-4.4.55.git] / net / ipv4 / ip_gre.c

diff --git a/net/ipv4/ip_gre.c b/net/ipv4/ip_gre.c
index e5dfd2843f28abbab0c949922732871774433044..987a4e5e07e2d3d5352de4cd8115909449183015 100644 (file)
--- a/net/ipv4/ip_gre.c
+++ b/net/ipv4/ip_gre.c
@@ -159,14 +159,14 @@ static int ip_gre_calc_hlen(__be16 o_flags)
 static int parse_gre_header(struct sk_buff *skb, struct tnl_ptk_info *tpi,
                            bool *csum_err, int *hdr_len)
 {
-       struct iphdr *iph = ip_hdr(skb);
-       struct gre_base_hdr *greh;
+       unsigned int ip_hlen = ip_hdrlen(skb);
+       const struct gre_base_hdr *greh;
        __be32 *options;
 
        if (unlikely(!pskb_may_pull(skb, sizeof(struct gre_base_hdr))))
                return -EINVAL;
 
-       greh = (struct gre_base_hdr *)((u8 *)iph + (iph->ihl << 2));
+       greh = (struct gre_base_hdr *)(skb_network_header(skb) + ip_hlen);
        if (unlikely(greh->flags & (GRE_VERSION | GRE_ROUTING)))
                return -EINVAL;
 
@@ -176,6 +176,8 @@ static int parse_gre_header(struct sk_buff *skb, struct tnl_ptk_info *tpi,
        if (!pskb_may_pull(skb, *hdr_len))
                return -EINVAL;
 
+       greh = (struct gre_base_hdr *)(skb_network_header(skb) + ip_hlen);
+
        tpi->proto = greh->protocol;
 
        options = (__be32 *)(greh + 1);