netdev_features_t features)
 {
        struct sk_buff *segs = ERR_PTR(-EINVAL);
+       u16 mac_offset = skb->mac_header;
        netdev_features_t mpls_features;
+       u16 mac_len = skb->mac_len;
        __be16 mpls_protocol;
+       unsigned int mpls_hlen;
+
+       skb_reset_network_header(skb);
+       mpls_hlen = skb_inner_network_header(skb) - skb_network_header(skb);
+       if (unlikely(!pskb_may_pull(skb, mpls_hlen)))
+               goto out;
 
        /* Setup inner SKB. */
        mpls_protocol = skb->protocol;
        skb->protocol = skb->inner_protocol;
 
-       /* Push back the mac header that skb_mac_gso_segment() has pulled.
-        * It will be re-pulled by the call to skb_mac_gso_segment() below
-        */
-       __skb_push(skb, skb->mac_len);
+       __skb_pull(skb, mpls_hlen);
+
+       skb->mac_len = 0;
+       skb_reset_mac_header(skb);
 
        /* Segment inner packet. */
        mpls_features = skb->dev->mpls_features & features;
        segs = skb_mac_gso_segment(skb, mpls_features);
+       if (IS_ERR_OR_NULL(segs)) {
+               skb_gso_error_unwind(skb, mpls_protocol, mpls_hlen, mac_offset,
+                                    mac_len);
+               goto out;
+       }
+       skb = segs;
+
+       mpls_hlen += mac_len;
+       do {
+               skb->mac_len = mac_len;
+               skb->protocol = mpls_protocol;
 
+               skb_reset_inner_network_header(skb);
 
-       /* Restore outer protocol. */
-       skb->protocol = mpls_protocol;
+               __skb_push(skb, mpls_hlen);
 
-       /* Re-pull the mac header that the call to skb_mac_gso_segment()
-        * above pulled.  It will be re-pushed after returning
-        * skb_mac_gso_segment(), an indirect caller of this function.
-        */
-       __skb_pull(skb, skb->data - skb_mac_header(skb));
+               skb_reset_mac_header(skb);
+               skb_set_network_header(skb, mac_len);
+       } while ((skb = skb->next));
 
+out:
        return segs;
 }
 
 
        if (skb_cow_head(skb, MPLS_HLEN) < 0)
                return -ENOMEM;
 
+       if (!skb->inner_protocol) {
+               skb_set_inner_network_header(skb, skb->mac_len);
+               skb_set_inner_protocol(skb, skb->protocol);
+       }
+
        skb_push(skb, MPLS_HLEN);
        memmove(skb_mac_header(skb) - MPLS_HLEN, skb_mac_header(skb),
                skb->mac_len);
        skb_reset_mac_header(skb);
+       skb_set_network_header(skb, skb->mac_len);
 
        new_mpls_lse = (__be32 *)skb_mpls_header(skb);
        *new_mpls_lse = mpls->mpls_lse;
        skb_postpush_rcsum(skb, new_mpls_lse, MPLS_HLEN);
 
        update_ethertype(skb, eth_hdr(skb), mpls->mpls_ethertype);
-       if (!skb->inner_protocol)
-               skb_set_inner_protocol(skb, skb->protocol);
        skb->protocol = mpls->mpls_ethertype;
 
        invalidate_flow_key(key);
 
        __skb_pull(skb, MPLS_HLEN);
        skb_reset_mac_header(skb);
+       skb_set_network_header(skb, skb->mac_len);
 
        /* skb_mpls_header() is used to locate the ethertype
         * field correctly in the presence of VLAN tags.