hc
2023-11-22 f743a7adbd6e230d66a6206fa115b59fec2d88eb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*
 * xfrm4_mode_transport.c - Transport mode encapsulation for IPv4.
 *
 * Copyright (c) 2004-2006 Herbert Xu <herbert@gondor.apana.org.au>
 */
 
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/skbuff.h>
#include <linux/stringify.h>
#include <net/dst.h>
#include <net/ip.h>
#include <net/xfrm.h>
#include <net/protocol.h>
 
/* Add encapsulation header.
 *
 * The IP header will be moved forward to make space for the encapsulation
 * header.
 */
static int xfrm4_transport_output(struct xfrm_state *x, struct sk_buff *skb)
{
   struct iphdr *iph = ip_hdr(skb);
   int ihl = iph->ihl * 4;
 
   skb_set_inner_transport_header(skb, skb_transport_offset(skb));
 
   skb_set_network_header(skb, -x->props.header_len);
   skb->mac_header = skb->network_header +
             offsetof(struct iphdr, protocol);
   skb->transport_header = skb->network_header + ihl;
   __skb_pull(skb, ihl);
   memmove(skb_network_header(skb), iph, ihl);
   return 0;
}
 
/* Remove encapsulation header.
 *
 * The IP header will be moved over the top of the encapsulation header.
 *
 * On entry, skb->h shall point to where the IP header should be and skb->nh
 * shall be set to where the IP header currently is.  skb->data shall point
 * to the start of the payload.
 */
static int xfrm4_transport_input(struct xfrm_state *x, struct sk_buff *skb)
{
   int ihl = skb->data - skb_transport_header(skb);
 
   if (skb->transport_header != skb->network_header) {
       memmove(skb_transport_header(skb),
           skb_network_header(skb), ihl);
       skb->network_header = skb->transport_header;
   }
   ip_hdr(skb)->tot_len = htons(skb->len + ihl);
   skb_reset_transport_header(skb);
   return 0;
}
 
static struct sk_buff *xfrm4_transport_gso_segment(struct xfrm_state *x,
                          struct sk_buff *skb,
                          netdev_features_t features)
{
   const struct net_offload *ops;
   struct sk_buff *segs = ERR_PTR(-EINVAL);
   struct xfrm_offload *xo = xfrm_offload(skb);
 
   skb->transport_header += x->props.header_len;
   ops = rcu_dereference(inet_offloads[xo->proto]);
   if (likely(ops && ops->callbacks.gso_segment))
       segs = ops->callbacks.gso_segment(skb, features);
 
   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 = {
   .input = xfrm4_transport_input,
   .output = xfrm4_transport_output,
   .gso_segment = xfrm4_transport_gso_segment,
   .xmit = xfrm4_transport_xmit,
   .owner = THIS_MODULE,
   .encap = XFRM_MODE_TRANSPORT,
};
 
static int __init xfrm4_transport_init(void)
{
   return xfrm_register_mode(&xfrm4_transport_mode, AF_INET);
}
 
static void __exit xfrm4_transport_exit(void)
{
   int err;
 
   err = xfrm_unregister_mode(&xfrm4_transport_mode, AF_INET);
   BUG_ON(err);
}
 
module_init(xfrm4_transport_init);
module_exit(xfrm4_transport_exit);
MODULE_LICENSE("GPL");
MODULE_ALIAS_XFRM_MODE(AF_INET, XFRM_MODE_TRANSPORT);