1 // SPDX-License-Identifier: GPL-2.0-only
2 #define pr_fmt(fmt) "IPsec: " fmt
4 #include <crypto/algapi.h>
5 #include <crypto/hash.h>
7 #include <linux/module.h>
8 #include <linux/slab.h>
12 #include <linux/crypto.h>
13 #include <linux/pfkeyv2.h>
14 #include <linux/scatterlist.h>
16 #include <net/protocol.h>
19 struct xfrm_skb_cb xfrm
;
23 #define AH_SKB_CB(__skb) ((struct ah_skb_cb *)&((__skb)->cb[0]))
25 static void *ah_alloc_tmp(struct crypto_ahash
*ahash
, int nfrags
,
30 len
= size
+ crypto_ahash_digestsize(ahash
) +
31 (crypto_ahash_alignmask(ahash
) &
32 ~(crypto_tfm_ctx_alignment() - 1));
34 len
= ALIGN(len
, crypto_tfm_ctx_alignment());
36 len
+= sizeof(struct ahash_request
) + crypto_ahash_reqsize(ahash
);
37 len
= ALIGN(len
, __alignof__(struct scatterlist
));
39 len
+= sizeof(struct scatterlist
) * nfrags
;
41 return kmalloc(len
, GFP_ATOMIC
);
44 static inline u8
*ah_tmp_auth(void *tmp
, unsigned int offset
)
49 static inline u8
*ah_tmp_icv(struct crypto_ahash
*ahash
, void *tmp
,
52 return PTR_ALIGN((u8
*)tmp
+ offset
, crypto_ahash_alignmask(ahash
) + 1);
55 static inline struct ahash_request
*ah_tmp_req(struct crypto_ahash
*ahash
,
58 struct ahash_request
*req
;
60 req
= (void *)PTR_ALIGN(icv
+ crypto_ahash_digestsize(ahash
),
61 crypto_tfm_ctx_alignment());
63 ahash_request_set_tfm(req
, ahash
);
68 static inline struct scatterlist
*ah_req_sg(struct crypto_ahash
*ahash
,
69 struct ahash_request
*req
)
71 return (void *)ALIGN((unsigned long)(req
+ 1) +
72 crypto_ahash_reqsize(ahash
),
73 __alignof__(struct scatterlist
));
76 /* Clear mutable options and find final destination to substitute
77 * into IP header for icv calculation. Options are already checked
78 * for validity, so paranoia is not required. */
80 static int ip_clear_mutable_options(const struct iphdr
*iph
, __be32
*daddr
)
82 unsigned char *optptr
= (unsigned char *)(iph
+1);
83 int l
= iph
->ihl
*4 - sizeof(struct iphdr
);
96 if (optlen
<2 || optlen
>l
)
100 case 0x85: /* Some "Extended Security" crap. */
103 case 0x80|21: /* RFC1770 */
109 memcpy(daddr
, optptr
+optlen
-4, 4);
112 memset(optptr
, 0, optlen
);
120 static void ah_output_done(struct crypto_async_request
*base
, int err
)
124 struct sk_buff
*skb
= base
->data
;
125 struct xfrm_state
*x
= skb_dst(skb
)->xfrm
;
126 struct ah_data
*ahp
= x
->data
;
127 struct iphdr
*top_iph
= ip_hdr(skb
);
128 struct ip_auth_hdr
*ah
= ip_auth_hdr(skb
);
129 int ihl
= ip_hdrlen(skb
);
131 iph
= AH_SKB_CB(skb
)->tmp
;
132 icv
= ah_tmp_icv(ahp
->ahash
, iph
, ihl
);
133 memcpy(ah
->auth_data
, icv
, ahp
->icv_trunc_len
);
135 top_iph
->tos
= iph
->tos
;
136 top_iph
->ttl
= iph
->ttl
;
137 top_iph
->frag_off
= iph
->frag_off
;
138 if (top_iph
->ihl
!= 5) {
139 top_iph
->daddr
= iph
->daddr
;
140 memcpy(top_iph
+1, iph
+1, top_iph
->ihl
*4 - sizeof(struct iphdr
));
143 kfree(AH_SKB_CB(skb
)->tmp
);
144 xfrm_output_resume(skb
, err
);
147 static int ah_output(struct xfrm_state
*x
, struct sk_buff
*skb
)
153 struct sk_buff
*trailer
;
154 struct crypto_ahash
*ahash
;
155 struct ahash_request
*req
;
156 struct scatterlist
*sg
;
157 struct iphdr
*iph
, *top_iph
;
158 struct ip_auth_hdr
*ah
;
163 struct scatterlist
*seqhisg
;
168 if ((err
= skb_cow_data(skb
, 0, &trailer
)) < 0)
172 skb_push(skb
, -skb_network_offset(skb
));
173 ah
= ip_auth_hdr(skb
);
174 ihl
= ip_hdrlen(skb
);
176 if (x
->props
.flags
& XFRM_STATE_ESN
) {
178 seqhi_len
= sizeof(*seqhi
);
181 iph
= ah_alloc_tmp(ahash
, nfrags
+ sglists
, ihl
+ seqhi_len
);
184 seqhi
= (__be32
*)((char *)iph
+ ihl
);
185 icv
= ah_tmp_icv(ahash
, seqhi
, seqhi_len
);
186 req
= ah_tmp_req(ahash
, icv
);
187 sg
= ah_req_sg(ahash
, req
);
188 seqhisg
= sg
+ nfrags
;
190 memset(ah
->auth_data
, 0, ahp
->icv_trunc_len
);
192 top_iph
= ip_hdr(skb
);
194 iph
->tos
= top_iph
->tos
;
195 iph
->ttl
= top_iph
->ttl
;
196 iph
->frag_off
= top_iph
->frag_off
;
198 if (top_iph
->ihl
!= 5) {
199 iph
->daddr
= top_iph
->daddr
;
200 memcpy(iph
+1, top_iph
+1, top_iph
->ihl
*4 - sizeof(struct iphdr
));
201 err
= ip_clear_mutable_options(top_iph
, &top_iph
->daddr
);
206 ah
->nexthdr
= *skb_mac_header(skb
);
207 *skb_mac_header(skb
) = IPPROTO_AH
;
210 top_iph
->tot_len
= htons(skb
->len
);
211 top_iph
->frag_off
= 0;
215 if (x
->props
.flags
& XFRM_STATE_ALIGN4
)
216 ah
->hdrlen
= (XFRM_ALIGN4(sizeof(*ah
) + ahp
->icv_trunc_len
) >> 2) - 2;
218 ah
->hdrlen
= (XFRM_ALIGN8(sizeof(*ah
) + ahp
->icv_trunc_len
) >> 2) - 2;
222 ah
->seq_no
= htonl(XFRM_SKB_CB(skb
)->seq
.output
.low
);
224 sg_init_table(sg
, nfrags
+ sglists
);
225 err
= skb_to_sgvec_nomark(skb
, sg
, 0, skb
->len
);
226 if (unlikely(err
< 0))
229 if (x
->props
.flags
& XFRM_STATE_ESN
) {
230 /* Attach seqhi sg right after packet payload */
231 *seqhi
= htonl(XFRM_SKB_CB(skb
)->seq
.output
.hi
);
232 sg_set_buf(seqhisg
, seqhi
, seqhi_len
);
234 ahash_request_set_crypt(req
, sg
, icv
, skb
->len
+ seqhi_len
);
235 ahash_request_set_callback(req
, 0, ah_output_done
, skb
);
237 AH_SKB_CB(skb
)->tmp
= iph
;
239 err
= crypto_ahash_digest(req
);
241 if (err
== -EINPROGRESS
)
249 memcpy(ah
->auth_data
, icv
, ahp
->icv_trunc_len
);
251 top_iph
->tos
= iph
->tos
;
252 top_iph
->ttl
= iph
->ttl
;
253 top_iph
->frag_off
= iph
->frag_off
;
254 if (top_iph
->ihl
!= 5) {
255 top_iph
->daddr
= iph
->daddr
;
256 memcpy(top_iph
+1, iph
+1, top_iph
->ihl
*4 - sizeof(struct iphdr
));
265 static void ah_input_done(struct crypto_async_request
*base
, int err
)
269 struct iphdr
*work_iph
;
270 struct sk_buff
*skb
= base
->data
;
271 struct xfrm_state
*x
= xfrm_input_state(skb
);
272 struct ah_data
*ahp
= x
->data
;
273 struct ip_auth_hdr
*ah
= ip_auth_hdr(skb
);
274 int ihl
= ip_hdrlen(skb
);
275 int ah_hlen
= (ah
->hdrlen
+ 2) << 2;
280 work_iph
= AH_SKB_CB(skb
)->tmp
;
281 auth_data
= ah_tmp_auth(work_iph
, ihl
);
282 icv
= ah_tmp_icv(ahp
->ahash
, auth_data
, ahp
->icv_trunc_len
);
284 err
= crypto_memneq(icv
, auth_data
, ahp
->icv_trunc_len
) ? -EBADMSG
: 0;
290 skb
->network_header
+= ah_hlen
;
291 memcpy(skb_network_header(skb
), work_iph
, ihl
);
292 __skb_pull(skb
, ah_hlen
+ ihl
);
294 if (x
->props
.mode
== XFRM_MODE_TUNNEL
)
295 skb_reset_transport_header(skb
);
297 skb_set_transport_header(skb
, -ihl
);
299 kfree(AH_SKB_CB(skb
)->tmp
);
300 xfrm_input_resume(skb
, err
);
303 static int ah_input(struct xfrm_state
*x
, struct sk_buff
*skb
)
311 struct sk_buff
*trailer
;
312 struct crypto_ahash
*ahash
;
313 struct ahash_request
*req
;
314 struct scatterlist
*sg
;
315 struct iphdr
*iph
, *work_iph
;
316 struct ip_auth_hdr
*ah
;
322 struct scatterlist
*seqhisg
;
324 if (!pskb_may_pull(skb
, sizeof(*ah
)))
327 ah
= (struct ip_auth_hdr
*)skb
->data
;
331 nexthdr
= ah
->nexthdr
;
332 ah_hlen
= (ah
->hdrlen
+ 2) << 2;
334 if (x
->props
.flags
& XFRM_STATE_ALIGN4
) {
335 if (ah_hlen
!= XFRM_ALIGN4(sizeof(*ah
) + ahp
->icv_full_len
) &&
336 ah_hlen
!= XFRM_ALIGN4(sizeof(*ah
) + ahp
->icv_trunc_len
))
339 if (ah_hlen
!= XFRM_ALIGN8(sizeof(*ah
) + ahp
->icv_full_len
) &&
340 ah_hlen
!= XFRM_ALIGN8(sizeof(*ah
) + ahp
->icv_trunc_len
))
344 if (!pskb_may_pull(skb
, ah_hlen
))
347 /* We are going to _remove_ AH header to keep sockets happy,
348 * so... Later this can change. */
349 if (skb_unclone(skb
, GFP_ATOMIC
))
352 skb
->ip_summed
= CHECKSUM_NONE
;
355 if ((err
= skb_cow_data(skb
, 0, &trailer
)) < 0)
359 ah
= (struct ip_auth_hdr
*)skb
->data
;
361 ihl
= ip_hdrlen(skb
);
363 if (x
->props
.flags
& XFRM_STATE_ESN
) {
365 seqhi_len
= sizeof(*seqhi
);
368 work_iph
= ah_alloc_tmp(ahash
, nfrags
+ sglists
, ihl
+
369 ahp
->icv_trunc_len
+ seqhi_len
);
375 seqhi
= (__be32
*)((char *)work_iph
+ ihl
);
376 auth_data
= ah_tmp_auth(seqhi
, seqhi_len
);
377 icv
= ah_tmp_icv(ahash
, auth_data
, ahp
->icv_trunc_len
);
378 req
= ah_tmp_req(ahash
, icv
);
379 sg
= ah_req_sg(ahash
, req
);
380 seqhisg
= sg
+ nfrags
;
382 memcpy(work_iph
, iph
, ihl
);
383 memcpy(auth_data
, ah
->auth_data
, ahp
->icv_trunc_len
);
384 memset(ah
->auth_data
, 0, ahp
->icv_trunc_len
);
390 if (ihl
> sizeof(*iph
)) {
392 err
= ip_clear_mutable_options(iph
, &dummy
);
399 sg_init_table(sg
, nfrags
+ sglists
);
400 err
= skb_to_sgvec_nomark(skb
, sg
, 0, skb
->len
);
401 if (unlikely(err
< 0))
404 if (x
->props
.flags
& XFRM_STATE_ESN
) {
405 /* Attach seqhi sg right after packet payload */
406 *seqhi
= XFRM_SKB_CB(skb
)->seq
.input
.hi
;
407 sg_set_buf(seqhisg
, seqhi
, seqhi_len
);
409 ahash_request_set_crypt(req
, sg
, icv
, skb
->len
+ seqhi_len
);
410 ahash_request_set_callback(req
, 0, ah_input_done
, skb
);
412 AH_SKB_CB(skb
)->tmp
= work_iph
;
414 err
= crypto_ahash_digest(req
);
416 if (err
== -EINPROGRESS
)
422 err
= crypto_memneq(icv
, auth_data
, ahp
->icv_trunc_len
) ? -EBADMSG
: 0;
426 skb
->network_header
+= ah_hlen
;
427 memcpy(skb_network_header(skb
), work_iph
, ihl
);
428 __skb_pull(skb
, ah_hlen
+ ihl
);
429 if (x
->props
.mode
== XFRM_MODE_TUNNEL
)
430 skb_reset_transport_header(skb
);
432 skb_set_transport_header(skb
, -ihl
);
442 static int ah4_err(struct sk_buff
*skb
, u32 info
)
444 struct net
*net
= dev_net(skb
->dev
);
445 const struct iphdr
*iph
= (const struct iphdr
*)skb
->data
;
446 struct ip_auth_hdr
*ah
= (struct ip_auth_hdr
*)(skb
->data
+(iph
->ihl
<<2));
447 struct xfrm_state
*x
;
449 switch (icmp_hdr(skb
)->type
) {
450 case ICMP_DEST_UNREACH
:
451 if (icmp_hdr(skb
)->code
!= ICMP_FRAG_NEEDED
)
459 x
= xfrm_state_lookup(net
, skb
->mark
, (const xfrm_address_t
*)&iph
->daddr
,
460 ah
->spi
, IPPROTO_AH
, AF_INET
);
464 if (icmp_hdr(skb
)->type
== ICMP_DEST_UNREACH
)
465 ipv4_update_pmtu(skb
, net
, info
, 0, IPPROTO_AH
);
467 ipv4_redirect(skb
, net
, 0, IPPROTO_AH
);
473 static int ah_init_state(struct xfrm_state
*x
)
475 struct ah_data
*ahp
= NULL
;
476 struct xfrm_algo_desc
*aalg_desc
;
477 struct crypto_ahash
*ahash
;
485 ahp
= kzalloc(sizeof(*ahp
), GFP_KERNEL
);
489 ahash
= crypto_alloc_ahash(x
->aalg
->alg_name
, 0, 0);
494 if (crypto_ahash_setkey(ahash
, x
->aalg
->alg_key
,
495 (x
->aalg
->alg_key_len
+ 7) / 8))
499 * Lookup the algorithm description maintained by xfrm_algo,
500 * verify crypto transform properties, and store information
501 * we need for AH processing. This lookup cannot fail here
502 * after a successful crypto_alloc_ahash().
504 aalg_desc
= xfrm_aalg_get_byname(x
->aalg
->alg_name
, 0);
507 if (aalg_desc
->uinfo
.auth
.icv_fullbits
/8 !=
508 crypto_ahash_digestsize(ahash
)) {
509 pr_info("%s: %s digestsize %u != %hu\n",
510 __func__
, x
->aalg
->alg_name
,
511 crypto_ahash_digestsize(ahash
),
512 aalg_desc
->uinfo
.auth
.icv_fullbits
/ 8);
516 ahp
->icv_full_len
= aalg_desc
->uinfo
.auth
.icv_fullbits
/8;
517 ahp
->icv_trunc_len
= x
->aalg
->alg_trunc_len
/8;
519 if (x
->props
.flags
& XFRM_STATE_ALIGN4
)
520 x
->props
.header_len
= XFRM_ALIGN4(sizeof(struct ip_auth_hdr
) +
523 x
->props
.header_len
= XFRM_ALIGN8(sizeof(struct ip_auth_hdr
) +
525 if (x
->props
.mode
== XFRM_MODE_TUNNEL
)
526 x
->props
.header_len
+= sizeof(struct iphdr
);
533 crypto_free_ahash(ahp
->ahash
);
539 static void ah_destroy(struct xfrm_state
*x
)
541 struct ah_data
*ahp
= x
->data
;
546 crypto_free_ahash(ahp
->ahash
);
550 static int ah4_rcv_cb(struct sk_buff
*skb
, int err
)
555 static const struct xfrm_type ah_type
=
557 .description
= "AH4",
558 .owner
= THIS_MODULE
,
560 .flags
= XFRM_TYPE_REPLAY_PROT
,
561 .init_state
= ah_init_state
,
562 .destructor
= ah_destroy
,
567 static struct xfrm4_protocol ah4_protocol
= {
568 .handler
= xfrm4_rcv
,
569 .input_handler
= xfrm_input
,
570 .cb_handler
= ah4_rcv_cb
,
571 .err_handler
= ah4_err
,
575 static int __init
ah4_init(void)
577 if (xfrm_register_type(&ah_type
, AF_INET
) < 0) {
578 pr_info("%s: can't add xfrm type\n", __func__
);
581 if (xfrm4_protocol_register(&ah4_protocol
, IPPROTO_AH
) < 0) {
582 pr_info("%s: can't add protocol\n", __func__
);
583 xfrm_unregister_type(&ah_type
, AF_INET
);
589 static void __exit
ah4_fini(void)
591 if (xfrm4_protocol_deregister(&ah4_protocol
, IPPROTO_AH
) < 0)
592 pr_info("%s: can't remove protocol\n", __func__
);
593 xfrm_unregister_type(&ah_type
, AF_INET
);
596 module_init(ah4_init
);
597 module_exit(ah4_fini
);
598 MODULE_LICENSE("GPL");
599 MODULE_ALIAS_XFRM_TYPE(AF_INET
, XFRM_PROTO_AH
);