*/
        struct sk_buff *(*gso_segment)(struct xfrm_state *x, struct sk_buff *skb, netdev_features_t features);
 
-       /*
-        * Adjust pointers into the packet when IPsec is done at layer2.
-        */
-       void (*xmit)(struct xfrm_state *x, struct sk_buff *skb);
-
        struct xfrm_state_afinfo *afinfo;
        struct module *owner;
        u8 encap;
 
        return segs;
 }
 
-static void xfrm4_transport_xmit(struct xfrm_state *x, struct sk_buff *skb)
-{
-       struct xfrm_offload *xo = xfrm_offload(skb);
-
-       skb_reset_mac_len(skb);
-       pskb_pull(skb, skb->mac_len + sizeof(struct iphdr) + x->props.header_len);
-
-       if (xo->flags & XFRM_GSO_SEGMENT) {
-                skb_reset_transport_header(skb);
-                skb->transport_header -= x->props.header_len;
-       }
-}
-
 static struct xfrm_mode xfrm4_transport_mode = {
        .gso_segment = xfrm4_transport_gso_segment,
-       .xmit = xfrm4_transport_xmit,
        .owner = THIS_MODULE,
        .encap = XFRM_MODE_TRANSPORT,
        .family = AF_INET,
 
        return skb_mac_gso_segment(skb, features);
 }
 
-static void xfrm4_mode_tunnel_xmit(struct xfrm_state *x, struct sk_buff *skb)
-{
-       struct xfrm_offload *xo = xfrm_offload(skb);
-
-       if (xo->flags & XFRM_GSO_SEGMENT)
-               skb->transport_header = skb->network_header +
-                                       sizeof(struct iphdr);
-
-       skb_reset_mac_len(skb);
-       pskb_pull(skb, skb->mac_len + x->props.header_len);
-}
-
 static struct xfrm_mode xfrm4_tunnel_mode = {
        .input2 = xfrm4_mode_tunnel_input,
        .output2 = xfrm4_mode_tunnel_output,
        .gso_segment = xfrm4_mode_tunnel_gso_segment,
-       .xmit = xfrm4_mode_tunnel_xmit,
        .owner = THIS_MODULE,
        .encap = XFRM_MODE_TUNNEL,
        .flags = XFRM_MODE_FLAG_TUNNEL,
 
        return segs;
 }
 
-static void xfrm6_transport_xmit(struct xfrm_state *x, struct sk_buff *skb)
-{
-       struct xfrm_offload *xo = xfrm_offload(skb);
-
-       skb_reset_mac_len(skb);
-       pskb_pull(skb, skb->mac_len + sizeof(struct ipv6hdr) + x->props.header_len);
-
-       if (xo->flags & XFRM_GSO_SEGMENT) {
-                skb_reset_transport_header(skb);
-                skb->transport_header -= x->props.header_len;
-       }
-}
-
 static struct xfrm_mode xfrm6_transport_mode = {
        .gso_segment = xfrm4_transport_gso_segment,
-       .xmit = xfrm6_transport_xmit,
        .owner = THIS_MODULE,
        .encap = XFRM_MODE_TRANSPORT,
        .family = AF_INET6,
 
        return skb_mac_gso_segment(skb, features);
 }
 
-static void xfrm6_mode_tunnel_xmit(struct xfrm_state *x, struct sk_buff *skb)
-{
-       struct xfrm_offload *xo = xfrm_offload(skb);
-
-       if (xo->flags & XFRM_GSO_SEGMENT)
-               skb->transport_header = skb->network_header + sizeof(struct ipv6hdr);
-
-       skb_reset_mac_len(skb);
-       pskb_pull(skb, skb->mac_len + x->props.header_len);
-}
-
 static struct xfrm_mode xfrm6_tunnel_mode = {
        .input2 = xfrm6_mode_tunnel_input,
        .output2 = xfrm6_mode_tunnel_output,
        .gso_segment = xfrm6_mode_tunnel_gso_segment,
-       .xmit = xfrm6_mode_tunnel_xmit,
        .owner = THIS_MODULE,
        .encap = XFRM_MODE_TUNNEL,
        .flags = XFRM_MODE_FLAG_TUNNEL,
 
 #include <linux/notifier.h>
 
 #ifdef CONFIG_XFRM_OFFLOAD
+static void __xfrm_transport_prep(struct xfrm_state *x, struct sk_buff *skb,
+                                 unsigned int hsize)
+{
+       struct xfrm_offload *xo = xfrm_offload(skb);
+
+       skb_reset_mac_len(skb);
+       pskb_pull(skb, skb->mac_len + hsize + x->props.header_len);
+
+       if (xo->flags & XFRM_GSO_SEGMENT) {
+               skb_reset_transport_header(skb);
+               skb->transport_header -= x->props.header_len;
+       }
+}
+
+static void __xfrm_mode_tunnel_prep(struct xfrm_state *x, struct sk_buff *skb,
+                                   unsigned int hsize)
+
+{
+       struct xfrm_offload *xo = xfrm_offload(skb);
+
+       if (xo->flags & XFRM_GSO_SEGMENT)
+               skb->transport_header = skb->network_header + hsize;
+
+       skb_reset_mac_len(skb);
+       pskb_pull(skb, skb->mac_len + x->props.header_len);
+}
+
+/* Adjust pointers into the packet when IPsec is done at layer2 */
+static void xfrm_outer_mode_prep(struct xfrm_state *x, struct sk_buff *skb)
+{
+       switch (x->outer_mode->encap) {
+       case XFRM_MODE_TUNNEL:
+               if (x->outer_mode->family == AF_INET)
+                       return __xfrm_mode_tunnel_prep(x, skb,
+                                                      sizeof(struct iphdr));
+               if (x->outer_mode->family == AF_INET6)
+                       return __xfrm_mode_tunnel_prep(x, skb,
+                                                      sizeof(struct ipv6hdr));
+               break;
+       case XFRM_MODE_TRANSPORT:
+               if (x->outer_mode->family == AF_INET)
+                       return __xfrm_transport_prep(x, skb,
+                                                    sizeof(struct iphdr));
+               if (x->outer_mode->family == AF_INET6)
+                       return __xfrm_transport_prep(x, skb,
+                                                    sizeof(struct ipv6hdr));
+               break;
+       case XFRM_MODE_ROUTEOPTIMIZATION:
+       case XFRM_MODE_IN_TRIGGER:
+       case XFRM_MODE_BEET:
+               break;
+       }
+}
+
 struct sk_buff *validate_xmit_xfrm(struct sk_buff *skb, netdev_features_t features, bool *again)
 {
        int err;
 
        if (!skb->next) {
                esp_features |= skb->dev->gso_partial_features;
-               x->outer_mode->xmit(x, skb);
+               xfrm_outer_mode_prep(x, skb);
 
                xo->flags |= XFRM_DEV_RESUME;
 
                xo = xfrm_offload(skb2);
                xo->flags |= XFRM_DEV_RESUME;
 
-               x->outer_mode->xmit(x, skb2);
+               xfrm_outer_mode_prep(x, skb2);
 
                err = x->type_offload->xmit(x, skb2, esp_features);
                if (!err) {