1 /*
2  * Copyright (c) 2018 Chelsio Communications, Inc.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License version 2 as
6  * published by the Free Software Foundation.
7  */
8 
9 #ifndef __CHTLS_H__
10 #define __CHTLS_H__
11 
12 #include <crypto/aes.h>
13 #include <crypto/algapi.h>
14 #include <crypto/hash.h>
15 #include <crypto/sha.h>
16 #include <crypto/authenc.h>
17 #include <crypto/ctr.h>
18 #include <crypto/gf128mul.h>
19 #include <crypto/internal/aead.h>
20 #include <crypto/null.h>
21 #include <crypto/internal/skcipher.h>
22 #include <crypto/aead.h>
23 #include <crypto/scatterwalk.h>
24 #include <crypto/internal/hash.h>
25 #include <linux/tls.h>
26 #include <net/tls.h>
27 
28 #include "t4fw_api.h"
29 #include "t4_msg.h"
30 #include "cxgb4.h"
31 #include "cxgb4_uld.h"
32 #include "l2t.h"
33 #include "chcr_algo.h"
34 #include "chcr_core.h"
35 #include "chcr_crypto.h"
36 
37 #define MAX_IVS_PAGE			256
38 #define TLS_KEY_CONTEXT_SZ		64
39 #define CIPHER_BLOCK_SIZE		16
40 #define GCM_TAG_SIZE			16
41 #define KEY_ON_MEM_SZ			16
42 #define AEAD_EXPLICIT_DATA_SIZE		8
43 #define TLS_HEADER_LENGTH		5
44 #define SCMD_CIPH_MODE_AES_GCM		2
45 /* Any MFS size should work and come from openssl */
46 #define TLS_MFS				16384
47 
48 #define RSS_HDR sizeof(struct rss_header)
49 #define TLS_WR_CPL_LEN \
50 	(sizeof(struct fw_tlstx_data_wr) + sizeof(struct cpl_tx_tls_sfo))
51 
52 enum {
53 	CHTLS_KEY_CONTEXT_DSGL,
54 	CHTLS_KEY_CONTEXT_IMM,
55 	CHTLS_KEY_CONTEXT_DDR,
56 };
57 
58 enum {
59 	CHTLS_LISTEN_START,
60 	CHTLS_LISTEN_STOP,
61 };
62 
63 /* Flags for return value of CPL message handlers */
64 enum {
65 	CPL_RET_BUF_DONE =    1,   /* buffer processing done */
66 	CPL_RET_BAD_MSG =     2,   /* bad CPL message */
67 	CPL_RET_UNKNOWN_TID = 4    /* unexpected unknown TID */
68 };
69 
70 #define LISTEN_INFO_HASH_SIZE 32
71 #define RSPQ_HASH_BITS 5
72 struct listen_info {
73 	struct listen_info *next;  /* Link to next entry */
74 	struct sock *sk;           /* The listening socket */
75 	unsigned int stid;         /* The server TID */
76 };
77 
78 enum {
79 	T4_LISTEN_START_PENDING,
80 	T4_LISTEN_STARTED
81 };
82 
83 enum csk_flags {
84 	CSK_CALLBACKS_CHKD,	/* socket callbacks have been sanitized */
85 	CSK_ABORT_REQ_RCVD,	/* received one ABORT_REQ_RSS message */
86 	CSK_TX_MORE_DATA,	/* sending ULP data; don't set SHOVE bit */
87 	CSK_TX_WAIT_IDLE,	/* suspend Tx until in-flight data is ACKed */
88 	CSK_ABORT_SHUTDOWN,	/* shouldn't send more abort requests */
89 	CSK_ABORT_RPL_PENDING,	/* expecting an abort reply */
90 	CSK_CLOSE_CON_REQUESTED,/* we've sent a close_conn_req */
91 	CSK_TX_DATA_SENT,	/* sent a TX_DATA WR on this connection */
92 	CSK_TX_FAILOVER,	/* Tx traffic failing over */
93 	CSK_UPDATE_RCV_WND,	/* Need to update rcv window */
94 	CSK_RST_ABORTED,	/* outgoing RST was aborted */
95 	CSK_TLS_HANDSHK,	/* TLS Handshake */
96 	CSK_CONN_INLINE,	/* Connection on HW */
97 };
98 
99 enum chtls_cdev_state {
100 	CHTLS_CDEV_STATE_UP = 1
101 };
102 
103 struct listen_ctx {
104 	struct sock *lsk;
105 	struct chtls_dev *cdev;
106 	struct sk_buff_head synq;
107 	u32 state;
108 };
109 
110 struct key_map {
111 	unsigned long *addr;
112 	unsigned int start;
113 	unsigned int available;
114 	unsigned int size;
115 	spinlock_t lock; /* lock for key id request from map */
116 } __packed;
117 
118 struct tls_scmd {
119 	u32 seqno_numivs;
120 	u32 ivgen_hdrlen;
121 };
122 
123 struct chtls_dev {
124 	struct tls_device tlsdev;
125 	struct list_head list;
126 	struct cxgb4_lld_info *lldi;
127 	struct pci_dev *pdev;
128 	struct listen_info *listen_hash_tab[LISTEN_INFO_HASH_SIZE];
129 	spinlock_t listen_lock; /* lock for listen list */
130 	struct net_device **ports;
131 	struct tid_info *tids;
132 	unsigned int pfvf;
133 	const unsigned short *mtus;
134 
135 	struct idr hwtid_idr;
136 	struct idr stid_idr;
137 
138 	spinlock_t idr_lock ____cacheline_aligned_in_smp;
139 
140 	struct net_device *egr_dev[NCHAN * 2];
141 	struct sk_buff *rspq_skb_cache[1 << RSPQ_HASH_BITS];
142 	struct sk_buff *askb;
143 
144 	struct sk_buff_head deferq;
145 	struct work_struct deferq_task;
146 
147 	struct list_head list_node;
148 	struct list_head rcu_node;
149 	struct list_head na_node;
150 	unsigned int send_page_order;
151 	int max_host_sndbuf;
152 	struct key_map kmap;
153 	unsigned int cdev_state;
154 };
155 
156 struct chtls_hws {
157 	struct sk_buff_head sk_recv_queue;
158 	u8 txqid;
159 	u8 ofld;
160 	u16 type;
161 	u16 rstate;
162 	u16 keyrpl;
163 	u16 pldlen;
164 	u16 rcvpld;
165 	u16 compute;
166 	u16 expansion;
167 	u16 keylen;
168 	u16 pdus;
169 	u16 adjustlen;
170 	u16 ivsize;
171 	u16 txleft;
172 	u32 mfs;
173 	s32 txkey;
174 	s32 rxkey;
175 	u32 fcplenmax;
176 	u32 copied_seq;
177 	u64 tx_seq_no;
178 	struct tls_scmd scmd;
179 	struct tls12_crypto_info_aes_gcm_128 crypto_info;
180 };
181 
182 struct chtls_sock {
183 	struct sock *sk;
184 	struct chtls_dev *cdev;
185 	struct l2t_entry *l2t_entry;    /* pointer to the L2T entry */
186 	struct net_device *egress_dev;  /* TX_CHAN for act open retry */
187 
188 	struct sk_buff_head txq;
189 	struct sk_buff *wr_skb_head;
190 	struct sk_buff *wr_skb_tail;
191 	struct sk_buff *ctrl_skb_cache;
192 	struct sk_buff *txdata_skb_cache; /* abort path messages */
193 	struct kref kref;
194 	unsigned long flags;
195 	u32 opt2;
196 	u32 wr_credits;
197 	u32 wr_unacked;
198 	u32 wr_max_credits;
199 	u32 wr_nondata;
200 	u32 hwtid;               /* TCP Control Block ID */
201 	u32 txq_idx;
202 	u32 rss_qid;
203 	u32 tid;
204 	u32 idr;
205 	u32 mss;
206 	u32 ulp_mode;
207 	u32 tx_chan;
208 	u32 rx_chan;
209 	u32 sndbuf;
210 	u32 txplen_max;
211 	u32 mtu_idx;           /* MTU table index */
212 	u32 smac_idx;
213 	u8 port_id;
214 	u8 tos;
215 	u16 resv2;
216 	u32 delack_mode;
217 	u32 delack_seq;
218 
219 	void *passive_reap_next;        /* placeholder for passive */
220 	struct chtls_hws tlshws;
221 	struct synq {
222 		struct sk_buff *next;
223 		struct sk_buff *prev;
224 	} synq;
225 	struct listen_ctx *listen_ctx;
226 };
227 
228 struct tls_hdr {
229 	u8  type;
230 	u16 version;
231 	u16 length;
232 } __packed;
233 
234 struct tlsrx_cmp_hdr {
235 	u8  type;
236 	u16 version;
237 	u16 length;
238 
239 	u64 tls_seq;
240 	u16 reserved1;
241 	u8  res_to_mac_error;
242 } __packed;
243 
244 /* res_to_mac_error fields */
245 #define TLSRX_HDR_PKT_INT_ERROR_S   4
246 #define TLSRX_HDR_PKT_INT_ERROR_M   0x1
247 #define TLSRX_HDR_PKT_INT_ERROR_V(x) \
248 	((x) << TLSRX_HDR_PKT_INT_ERROR_S)
249 #define TLSRX_HDR_PKT_INT_ERROR_G(x) \
250 	(((x) >> TLSRX_HDR_PKT_INT_ERROR_S) & TLSRX_HDR_PKT_INT_ERROR_M)
251 #define TLSRX_HDR_PKT_INT_ERROR_F   TLSRX_HDR_PKT_INT_ERROR_V(1U)
252 
253 #define TLSRX_HDR_PKT_SPP_ERROR_S        3
254 #define TLSRX_HDR_PKT_SPP_ERROR_M        0x1
255 #define TLSRX_HDR_PKT_SPP_ERROR_V(x)     ((x) << TLSRX_HDR_PKT_SPP_ERROR)
256 #define TLSRX_HDR_PKT_SPP_ERROR_G(x)     \
257 	(((x) >> TLSRX_HDR_PKT_SPP_ERROR_S) & TLSRX_HDR_PKT_SPP_ERROR_M)
258 #define TLSRX_HDR_PKT_SPP_ERROR_F        TLSRX_HDR_PKT_SPP_ERROR_V(1U)
259 
260 #define TLSRX_HDR_PKT_CCDX_ERROR_S       2
261 #define TLSRX_HDR_PKT_CCDX_ERROR_M       0x1
262 #define TLSRX_HDR_PKT_CCDX_ERROR_V(x)    ((x) << TLSRX_HDR_PKT_CCDX_ERROR_S)
263 #define TLSRX_HDR_PKT_CCDX_ERROR_G(x)    \
264 	(((x) >> TLSRX_HDR_PKT_CCDX_ERROR_S) & TLSRX_HDR_PKT_CCDX_ERROR_M)
265 #define TLSRX_HDR_PKT_CCDX_ERROR_F       TLSRX_HDR_PKT_CCDX_ERROR_V(1U)
266 
267 #define TLSRX_HDR_PKT_PAD_ERROR_S        1
268 #define TLSRX_HDR_PKT_PAD_ERROR_M        0x1
269 #define TLSRX_HDR_PKT_PAD_ERROR_V(x)     ((x) << TLSRX_HDR_PKT_PAD_ERROR_S)
270 #define TLSRX_HDR_PKT_PAD_ERROR_G(x)     \
271 	(((x) >> TLSRX_HDR_PKT_PAD_ERROR_S) & TLSRX_HDR_PKT_PAD_ERROR_M)
272 #define TLSRX_HDR_PKT_PAD_ERROR_F        TLSRX_HDR_PKT_PAD_ERROR_V(1U)
273 
274 #define TLSRX_HDR_PKT_MAC_ERROR_S        0
275 #define TLSRX_HDR_PKT_MAC_ERROR_M        0x1
276 #define TLSRX_HDR_PKT_MAC_ERROR_V(x)     ((x) << TLSRX_HDR_PKT_MAC_ERROR)
277 #define TLSRX_HDR_PKT_MAC_ERROR_G(x)     \
278 	(((x) >> S_TLSRX_HDR_PKT_MAC_ERROR_S) & TLSRX_HDR_PKT_MAC_ERROR_M)
279 #define TLSRX_HDR_PKT_MAC_ERROR_F        TLSRX_HDR_PKT_MAC_ERROR_V(1U)
280 
281 #define TLSRX_HDR_PKT_ERROR_M           0x1F
282 #define CONTENT_TYPE_ERROR		0x7F
283 
284 struct ulp_mem_rw {
285 	__be32 cmd;
286 	__be32 len16;             /* command length */
287 	__be32 dlen;              /* data length in 32-byte units */
288 	__be32 lock_addr;
289 };
290 
291 struct tls_key_wr {
292 	__be32 op_to_compl;
293 	__be32 flowid_len16;
294 	__be32 ftid;
295 	u8   reneg_to_write_rx;
296 	u8   protocol;
297 	__be16 mfs;
298 };
299 
300 struct tls_key_req {
301 	struct tls_key_wr wr;
302 	struct ulp_mem_rw req;
303 	struct ulptx_idata sc_imm;
304 };
305 
306 /*
307  * This lives in skb->cb and is used to chain WRs in a linked list.
308  */
309 struct wr_skb_cb {
310 	struct l2t_skb_cb l2t;          /* reserve space for l2t CB */
311 	struct sk_buff *next_wr;        /* next write request */
312 };
313 
314 /* Per-skb backlog handler.  Run when a socket's backlog is processed. */
315 struct blog_skb_cb {
316 	void (*backlog_rcv)(struct sock *sk, struct sk_buff *skb);
317 	struct chtls_dev *cdev;
318 };
319 
320 /*
321  * Similar to tcp_skb_cb but with ULP elements added to support TLS,
322  * etc.
323  */
324 struct ulp_skb_cb {
325 	struct wr_skb_cb wr;		/* reserve space for write request */
326 	u16 flags;			/* TCP-like flags */
327 	u8 psh;
328 	u8 ulp_mode;			/* ULP mode/submode of sk_buff */
329 	u32 seq;			/* TCP sequence number */
330 	union { /* ULP-specific fields */
331 		struct {
332 			u8  type;
333 			u8  ofld;
334 			u8  iv;
335 		} tls;
336 	} ulp;
337 };
338 
339 #define ULP_SKB_CB(skb) ((struct ulp_skb_cb *)&((skb)->cb[0]))
340 #define BLOG_SKB_CB(skb) ((struct blog_skb_cb *)(skb)->cb)
341 
342 /*
343  * Flags for ulp_skb_cb.flags.
344  */
345 enum {
346 	ULPCB_FLAG_NEED_HDR  = 1 << 0,	/* packet needs a TX_DATA_WR header */
347 	ULPCB_FLAG_NO_APPEND = 1 << 1,	/* don't grow this skb */
348 	ULPCB_FLAG_BARRIER   = 1 << 2,	/* set TX_WAIT_IDLE after sending */
349 	ULPCB_FLAG_HOLD      = 1 << 3,	/* skb not ready for Tx yet */
350 	ULPCB_FLAG_COMPL     = 1 << 4,	/* request WR completion */
351 	ULPCB_FLAG_URG       = 1 << 5,	/* urgent data */
352 	ULPCB_FLAG_TLS_HDR   = 1 << 6,  /* payload with tls hdr */
353 	ULPCB_FLAG_NO_HDR    = 1 << 7,  /* not a ofld wr */
354 };
355 
356 /* The ULP mode/submode of an skbuff */
357 #define skb_ulp_mode(skb)  (ULP_SKB_CB(skb)->ulp_mode)
358 #define TCP_PAGE(sk)   (sk->sk_frag.page)
359 #define TCP_OFF(sk)    (sk->sk_frag.offset)
360 
to_chtls_dev(struct tls_device * tlsdev)361 static inline struct chtls_dev *to_chtls_dev(struct tls_device *tlsdev)
362 {
363 	return container_of(tlsdev, struct chtls_dev, tlsdev);
364 }
365 
csk_set_flag(struct chtls_sock * csk,enum csk_flags flag)366 static inline void csk_set_flag(struct chtls_sock *csk,
367 				enum csk_flags flag)
368 {
369 	__set_bit(flag, &csk->flags);
370 }
371 
csk_reset_flag(struct chtls_sock * csk,enum csk_flags flag)372 static inline void csk_reset_flag(struct chtls_sock *csk,
373 				  enum csk_flags flag)
374 {
375 	__clear_bit(flag, &csk->flags);
376 }
377 
csk_conn_inline(const struct chtls_sock * csk)378 static inline bool csk_conn_inline(const struct chtls_sock *csk)
379 {
380 	return test_bit(CSK_CONN_INLINE, &csk->flags);
381 }
382 
csk_flag(const struct sock * sk,enum csk_flags flag)383 static inline int csk_flag(const struct sock *sk, enum csk_flags flag)
384 {
385 	struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
386 
387 	if (!csk_conn_inline(csk))
388 		return 0;
389 	return test_bit(flag, &csk->flags);
390 }
391 
csk_flag_nochk(const struct chtls_sock * csk,enum csk_flags flag)392 static inline int csk_flag_nochk(const struct chtls_sock *csk,
393 				 enum csk_flags flag)
394 {
395 	return test_bit(flag, &csk->flags);
396 }
397 
cplhdr(struct sk_buff * skb)398 static inline void *cplhdr(struct sk_buff *skb)
399 {
400 	return skb->data;
401 }
402 
is_neg_adv(unsigned int status)403 static inline int is_neg_adv(unsigned int status)
404 {
405 	return status == CPL_ERR_RTX_NEG_ADVICE ||
406 	       status == CPL_ERR_KEEPALV_NEG_ADVICE ||
407 	       status == CPL_ERR_PERSIST_NEG_ADVICE;
408 }
409 
process_cpl_msg(void (* fn)(struct sock *,struct sk_buff *),struct sock * sk,struct sk_buff * skb)410 static inline void process_cpl_msg(void (*fn)(struct sock *, struct sk_buff *),
411 				   struct sock *sk,
412 				   struct sk_buff *skb)
413 {
414 	skb_reset_mac_header(skb);
415 	skb_reset_network_header(skb);
416 	skb_reset_transport_header(skb);
417 
418 	bh_lock_sock(sk);
419 	if (unlikely(sock_owned_by_user(sk))) {
420 		BLOG_SKB_CB(skb)->backlog_rcv = fn;
421 		__sk_add_backlog(sk, skb);
422 	} else {
423 		fn(sk, skb);
424 	}
425 	bh_unlock_sock(sk);
426 }
427 
chtls_sock_free(struct kref * ref)428 static inline void chtls_sock_free(struct kref *ref)
429 {
430 	struct chtls_sock *csk = container_of(ref, struct chtls_sock,
431 					      kref);
432 	kfree(csk);
433 }
434 
__chtls_sock_put(const char * fn,struct chtls_sock * csk)435 static inline void __chtls_sock_put(const char *fn, struct chtls_sock *csk)
436 {
437 	kref_put(&csk->kref, chtls_sock_free);
438 }
439 
__chtls_sock_get(const char * fn,struct chtls_sock * csk)440 static inline void __chtls_sock_get(const char *fn,
441 				    struct chtls_sock *csk)
442 {
443 	kref_get(&csk->kref);
444 }
445 
send_or_defer(struct sock * sk,struct tcp_sock * tp,struct sk_buff * skb,int through_l2t)446 static inline void send_or_defer(struct sock *sk, struct tcp_sock *tp,
447 				 struct sk_buff *skb, int through_l2t)
448 {
449 	struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
450 
451 	if (through_l2t) {
452 		/* send through L2T */
453 		cxgb4_l2t_send(csk->egress_dev, skb, csk->l2t_entry);
454 	} else {
455 		/* send directly */
456 		cxgb4_ofld_send(csk->egress_dev, skb);
457 	}
458 }
459 
460 typedef int (*chtls_handler_func)(struct chtls_dev *, struct sk_buff *);
461 extern chtls_handler_func chtls_handlers[NUM_CPL_CMDS];
462 void chtls_install_cpl_ops(struct sock *sk);
463 int chtls_init_kmap(struct chtls_dev *cdev, struct cxgb4_lld_info *lldi);
464 void chtls_listen_stop(struct chtls_dev *cdev, struct sock *sk);
465 int chtls_listen_start(struct chtls_dev *cdev, struct sock *sk);
466 void chtls_close(struct sock *sk, long timeout);
467 int chtls_disconnect(struct sock *sk, int flags);
468 void chtls_shutdown(struct sock *sk, int how);
469 void chtls_destroy_sock(struct sock *sk);
470 int chtls_sendmsg(struct sock *sk, struct msghdr *msg, size_t size);
471 int chtls_recvmsg(struct sock *sk, struct msghdr *msg,
472 		  size_t len, int nonblock, int flags, int *addr_len);
473 int chtls_sendpage(struct sock *sk, struct page *page,
474 		   int offset, size_t size, int flags);
475 int send_tx_flowc_wr(struct sock *sk, int compl,
476 		     u32 snd_nxt, u32 rcv_nxt);
477 void chtls_tcp_push(struct sock *sk, int flags);
478 int chtls_push_frames(struct chtls_sock *csk, int comp);
479 int chtls_set_tcb_tflag(struct sock *sk, unsigned int bit_pos, int val);
480 int chtls_setkey(struct chtls_sock *csk, u32 keylen, u32 mode);
481 void skb_entail(struct sock *sk, struct sk_buff *skb, int flags);
482 unsigned int keyid_to_addr(int start_addr, int keyid);
483 void free_tls_keyid(struct sock *sk);
484 #endif
485