forked from mirrors/linux
		
	smc: connection data control (CDC)
send and receive CDC messages (via IB message send and CQE) Signed-off-by: Ursula Braun <ubraun@linux.vnet.ibm.com> Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
		
							parent
							
								
									9bf9abead2
								
							
						
					
					
						commit
						5f08318f61
					
				
					 8 changed files with 641 additions and 0 deletions
				
			
		|  | @ -1,2 +1,3 @@ | ||||||
| obj-$(CONFIG_SMC)	+= smc.o | obj-$(CONFIG_SMC)	+= smc.o | ||||||
| smc-y := af_smc.o smc_pnet.o smc_ib.o smc_clc.o smc_core.o smc_wr.o smc_llc.o | smc-y := af_smc.o smc_pnet.o smc_ib.o smc_clc.o smc_core.o smc_wr.o smc_llc.o | ||||||
|  | smc-y += smc_cdc.o | ||||||
|  |  | ||||||
|  | @ -26,12 +26,14 @@ | ||||||
| #include <linux/socket.h> | #include <linux/socket.h> | ||||||
| #include <linux/inetdevice.h> | #include <linux/inetdevice.h> | ||||||
| #include <linux/workqueue.h> | #include <linux/workqueue.h> | ||||||
|  | #include <linux/in.h> | ||||||
| #include <net/sock.h> | #include <net/sock.h> | ||||||
| #include <net/tcp.h> | #include <net/tcp.h> | ||||||
| 
 | 
 | ||||||
| #include "smc.h" | #include "smc.h" | ||||||
| #include "smc_clc.h" | #include "smc_clc.h" | ||||||
| #include "smc_llc.h" | #include "smc_llc.h" | ||||||
|  | #include "smc_cdc.h" | ||||||
| #include "smc_core.h" | #include "smc_core.h" | ||||||
| #include "smc_ib.h" | #include "smc_ib.h" | ||||||
| #include "smc_pnet.h" | #include "smc_pnet.h" | ||||||
|  | @ -285,6 +287,7 @@ static void smc_conn_save_peer_info(struct smc_sock *smc, | ||||||
| 				    struct smc_clc_msg_accept_confirm *clc) | 				    struct smc_clc_msg_accept_confirm *clc) | ||||||
| { | { | ||||||
| 	smc->conn.peer_conn_idx = clc->conn_idx; | 	smc->conn.peer_conn_idx = clc->conn_idx; | ||||||
|  | 	smc->conn.local_tx_ctrl.token = ntohl(clc->rmbe_alert_token); | ||||||
| 	smc->conn.peer_rmbe_size = smc_uncompress_bufsize(clc->rmbe_size); | 	smc->conn.peer_rmbe_size = smc_uncompress_bufsize(clc->rmbe_size); | ||||||
| 	atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size); | 	atomic_set(&smc->conn.peer_rmbe_space, smc->conn.peer_rmbe_size); | ||||||
| } | } | ||||||
|  | @ -1201,6 +1204,12 @@ static int __init smc_init(void) | ||||||
| 		goto out_pnet; | 		goto out_pnet; | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 	rc = smc_cdc_init(); | ||||||
|  | 	if (rc) { | ||||||
|  | 		pr_err("%s: smc_cdc_init fails with %d\n", __func__, rc); | ||||||
|  | 		goto out_pnet; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	rc = proto_register(&smc_proto, 1); | 	rc = proto_register(&smc_proto, 1); | ||||||
| 	if (rc) { | 	if (rc) { | ||||||
| 		pr_err("%s: proto_register fails with %d\n", __func__, rc); | 		pr_err("%s: proto_register fails with %d\n", __func__, rc); | ||||||
|  |  | ||||||
|  | @ -21,6 +21,10 @@ | ||||||
| 
 | 
 | ||||||
| #define SMC_MAX_PORTS		2	/* Max # of ports */ | #define SMC_MAX_PORTS		2	/* Max # of ports */ | ||||||
| 
 | 
 | ||||||
|  | #ifdef ATOMIC64_INIT | ||||||
|  | #define KERNEL_HAS_ATOMIC64 | ||||||
|  | #endif | ||||||
|  | 
 | ||||||
| enum smc_state {		/* possible states of an SMC socket */ | enum smc_state {		/* possible states of an SMC socket */ | ||||||
| 	SMC_ACTIVE	= 1, | 	SMC_ACTIVE	= 1, | ||||||
| 	SMC_INIT	= 2, | 	SMC_INIT	= 2, | ||||||
|  | @ -34,6 +38,67 @@ struct smc_wr_rx_hdr {	/* common prefix part of LLC and CDC to demultiplex */ | ||||||
| 	u8			type; | 	u8			type; | ||||||
| } __aligned(1); | } __aligned(1); | ||||||
| 
 | 
 | ||||||
|  | struct smc_cdc_conn_state_flags { | ||||||
|  | #if defined(__BIG_ENDIAN_BITFIELD) | ||||||
|  | 	u8	peer_done_writing : 1;	/* Sending done indicator */ | ||||||
|  | 	u8	peer_conn_closed : 1;	/* Peer connection closed indicator */ | ||||||
|  | 	u8	peer_conn_abort : 1;	/* Abnormal close indicator */ | ||||||
|  | 	u8	reserved : 5; | ||||||
|  | #elif defined(__LITTLE_ENDIAN_BITFIELD) | ||||||
|  | 	u8	reserved : 5; | ||||||
|  | 	u8	peer_conn_abort : 1; | ||||||
|  | 	u8	peer_conn_closed : 1; | ||||||
|  | 	u8	peer_done_writing : 1; | ||||||
|  | #endif | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | struct smc_cdc_producer_flags { | ||||||
|  | #if defined(__BIG_ENDIAN_BITFIELD) | ||||||
|  | 	u8	write_blocked : 1;	/* Writing Blocked, no rx buf space */ | ||||||
|  | 	u8	urg_data_pending : 1;	/* Urgent Data Pending */ | ||||||
|  | 	u8	urg_data_present : 1;	/* Urgent Data Present */ | ||||||
|  | 	u8	cons_curs_upd_req : 1;	/* cursor update requested */ | ||||||
|  | 	u8	failover_validation : 1;/* message replay due to failover */ | ||||||
|  | 	u8	reserved : 3; | ||||||
|  | #elif defined(__LITTLE_ENDIAN_BITFIELD) | ||||||
|  | 	u8	reserved : 3; | ||||||
|  | 	u8	failover_validation : 1; | ||||||
|  | 	u8	cons_curs_upd_req : 1; | ||||||
|  | 	u8	urg_data_present : 1; | ||||||
|  | 	u8	urg_data_pending : 1; | ||||||
|  | 	u8	write_blocked : 1; | ||||||
|  | #endif | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | /* in host byte order */ | ||||||
|  | union smc_host_cursor {	/* SMC cursor - an offset in an RMBE */ | ||||||
|  | 	struct { | ||||||
|  | 		u16	reserved; | ||||||
|  | 		u16	wrap;		/* window wrap sequence number */ | ||||||
|  | 		u32	count;		/* cursor (= offset) part */ | ||||||
|  | 	}; | ||||||
|  | #ifdef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	atomic64_t		acurs;	/* for atomic processing */ | ||||||
|  | #else | ||||||
|  | 	u64			acurs;	/* for atomic processing */ | ||||||
|  | #endif | ||||||
|  | } __aligned(8); | ||||||
|  | 
 | ||||||
|  | /* in host byte order, except for flag bitfields in network byte order */ | ||||||
|  | struct smc_host_cdc_msg {		/* Connection Data Control message */ | ||||||
|  | 	struct smc_wr_rx_hdr		common; /* .type = 0xFE */ | ||||||
|  | 	u8				len;	/* length = 44 */ | ||||||
|  | 	u16				seqno;	/* connection seq # */ | ||||||
|  | 	u32				token;	/* alert_token */ | ||||||
|  | 	union smc_host_cursor		prod;		/* producer cursor */ | ||||||
|  | 	union smc_host_cursor		cons;		/* consumer cursor,
 | ||||||
|  | 							 * piggy backed "ack" | ||||||
|  | 							 */ | ||||||
|  | 	struct smc_cdc_producer_flags	prod_flags;	/* conn. tx/rx status */ | ||||||
|  | 	struct smc_cdc_conn_state_flags	conn_state_flags; /* peer conn. status*/ | ||||||
|  | 	u8				reserved[18]; | ||||||
|  | } __aligned(8); | ||||||
|  | 
 | ||||||
| struct smc_connection { | struct smc_connection { | ||||||
| 	struct rb_node		alert_node; | 	struct rb_node		alert_node; | ||||||
| 	struct smc_link_group	*lgr;		/* link group of connection */ | 	struct smc_link_group	*lgr;		/* link group of connection */ | ||||||
|  | @ -50,6 +115,38 @@ struct smc_connection { | ||||||
| 	struct smc_buf_desc	*rmb_desc;	/* RMBE descriptor */ | 	struct smc_buf_desc	*rmb_desc;	/* RMBE descriptor */ | ||||||
| 	int			rmbe_size;	/* RMBE size <== sock rmem */ | 	int			rmbe_size;	/* RMBE size <== sock rmem */ | ||||||
| 	int			rmbe_size_short;/* compressed notation */ | 	int			rmbe_size_short;/* compressed notation */ | ||||||
|  | 
 | ||||||
|  | 	struct smc_host_cdc_msg	local_tx_ctrl;	/* host byte order staging
 | ||||||
|  | 						 * buffer for CDC msg send | ||||||
|  | 						 * .prod cf. TCP snd_nxt | ||||||
|  | 						 * .cons cf. TCP sends ack | ||||||
|  | 						 */ | ||||||
|  | 	union smc_host_cursor	tx_curs_prep;	/* tx - prepared data
 | ||||||
|  | 						 * snd_max..wmem_alloc | ||||||
|  | 						 */ | ||||||
|  | 	union smc_host_cursor	tx_curs_sent;	/* tx - sent data
 | ||||||
|  | 						 * snd_nxt ? | ||||||
|  | 						 */ | ||||||
|  | 	union smc_host_cursor	tx_curs_fin;	/* tx - confirmed by peer
 | ||||||
|  | 						 * snd-wnd-begin ? | ||||||
|  | 						 */ | ||||||
|  | 	atomic_t		sndbuf_space;	/* remaining space in sndbuf */ | ||||||
|  | 	u16			tx_cdc_seq;	/* sequence # for CDC send */ | ||||||
|  | 	spinlock_t		send_lock;	/* protect wr_sends */ | ||||||
|  | 
 | ||||||
|  | 	struct smc_host_cdc_msg	local_rx_ctrl;	/* filled during event_handl.
 | ||||||
|  | 						 * .prod cf. TCP rcv_nxt | ||||||
|  | 						 * .cons cf. TCP snd_una | ||||||
|  | 						 */ | ||||||
|  | 	union smc_host_cursor	rx_curs_confirmed; /* confirmed to peer
 | ||||||
|  | 						    * source of snd_una ? | ||||||
|  | 						    */ | ||||||
|  | 	atomic_t		bytes_to_rcv;	/* arrived data,
 | ||||||
|  | 						 * not yet received | ||||||
|  | 						 */ | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	spinlock_t		acurs_lock;	/* protect cursors */ | ||||||
|  | #endif | ||||||
| }; | }; | ||||||
| 
 | 
 | ||||||
| struct smc_sock {				/* smc sock container */ | struct smc_sock {				/* smc sock container */ | ||||||
|  |  | ||||||
							
								
								
									
										280
									
								
								net/smc/smc_cdc.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										280
									
								
								net/smc/smc_cdc.c
									
									
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,280 @@ | ||||||
|  | /*
 | ||||||
|  |  * Shared Memory Communications over RDMA (SMC-R) and RoCE | ||||||
|  |  * | ||||||
|  |  * Connection Data Control (CDC) | ||||||
|  |  * handles flow control | ||||||
|  |  * | ||||||
|  |  * Copyright IBM Corp. 2016 | ||||||
|  |  * | ||||||
|  |  * Author(s):  Ursula Braun <ubraun@linux.vnet.ibm.com> | ||||||
|  |  */ | ||||||
|  | 
 | ||||||
|  | #include <linux/spinlock.h> | ||||||
|  | 
 | ||||||
|  | #include "smc.h" | ||||||
|  | #include "smc_wr.h" | ||||||
|  | #include "smc_cdc.h" | ||||||
|  | 
 | ||||||
|  | /********************************** send *************************************/ | ||||||
|  | 
 | ||||||
|  | struct smc_cdc_tx_pend { | ||||||
|  | 	struct smc_connection	*conn;		/* socket connection */ | ||||||
|  | 	union smc_host_cursor	cursor;	/* tx sndbuf cursor sent */ | ||||||
|  | 	union smc_host_cursor	p_cursor;	/* rx RMBE cursor produced */ | ||||||
|  | 	u16			ctrl_seq;	/* conn. tx sequence # */ | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | /* handler for send/transmission completion of a CDC msg */ | ||||||
|  | static void smc_cdc_tx_handler(struct smc_wr_tx_pend_priv *pnd_snd, | ||||||
|  | 			       struct smc_link *link, | ||||||
|  | 			       enum ib_wc_status wc_status) | ||||||
|  | { | ||||||
|  | 	struct smc_cdc_tx_pend *cdcpend = (struct smc_cdc_tx_pend *)pnd_snd; | ||||||
|  | 	struct smc_sock *smc; | ||||||
|  | 	int diff; | ||||||
|  | 
 | ||||||
|  | 	if (!cdcpend->conn) | ||||||
|  | 		/* already dismissed */ | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	smc = container_of(cdcpend->conn, struct smc_sock, conn); | ||||||
|  | 	bh_lock_sock(&smc->sk); | ||||||
|  | 	if (!wc_status) { | ||||||
|  | 		diff = smc_curs_diff(cdcpend->conn->sndbuf_size, | ||||||
|  | 				     &cdcpend->conn->tx_curs_fin, | ||||||
|  | 				     &cdcpend->cursor); | ||||||
|  | 		/* sndbuf_space is decreased in smc_sendmsg */ | ||||||
|  | 		smp_mb__before_atomic(); | ||||||
|  | 		atomic_add(diff, &cdcpend->conn->sndbuf_space); | ||||||
|  | 		/* guarantee 0 <= sndbuf_space <= sndbuf_size */ | ||||||
|  | 		smp_mb__after_atomic(); | ||||||
|  | 		smc_curs_write(&cdcpend->conn->tx_curs_fin, | ||||||
|  | 			       smc_curs_read(&cdcpend->cursor, cdcpend->conn), | ||||||
|  | 			       cdcpend->conn); | ||||||
|  | 	} | ||||||
|  | 	/* subsequent patch: wake if send buffer space available */ | ||||||
|  | 	bh_unlock_sock(&smc->sk); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | int smc_cdc_get_free_slot(struct smc_link *link, | ||||||
|  | 			  struct smc_wr_buf **wr_buf, | ||||||
|  | 			  struct smc_cdc_tx_pend **pend) | ||||||
|  | { | ||||||
|  | 	return smc_wr_tx_get_free_slot(link, smc_cdc_tx_handler, wr_buf, | ||||||
|  | 				       (struct smc_wr_tx_pend_priv **)pend); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_cdc_add_pending_send(struct smc_connection *conn, | ||||||
|  | 					    struct smc_cdc_tx_pend *pend) | ||||||
|  | { | ||||||
|  | 	BUILD_BUG_ON_MSG( | ||||||
|  | 		sizeof(struct smc_cdc_msg) > SMC_WR_BUF_SIZE, | ||||||
|  | 		"must increase SMC_WR_BUF_SIZE to at least sizeof(struct smc_cdc_msg)"); | ||||||
|  | 	BUILD_BUG_ON_MSG( | ||||||
|  | 		offsetof(struct smc_cdc_msg, reserved) > SMC_WR_TX_SIZE, | ||||||
|  | 		"must adapt SMC_WR_TX_SIZE to sizeof(struct smc_cdc_msg); if not all smc_wr upper layer protocols use the same message size any more, must start to set link->wr_tx_sges[i].length on each individual smc_wr_tx_send()"); | ||||||
|  | 	BUILD_BUG_ON_MSG( | ||||||
|  | 		sizeof(struct smc_cdc_tx_pend) > SMC_WR_TX_PEND_PRIV_SIZE, | ||||||
|  | 		"must increase SMC_WR_TX_PEND_PRIV_SIZE to at least sizeof(struct smc_cdc_tx_pend)"); | ||||||
|  | 	pend->conn = conn; | ||||||
|  | 	pend->cursor = conn->tx_curs_sent; | ||||||
|  | 	pend->p_cursor = conn->local_tx_ctrl.prod; | ||||||
|  | 	pend->ctrl_seq = conn->tx_cdc_seq; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | int smc_cdc_msg_send(struct smc_connection *conn, | ||||||
|  | 		     struct smc_wr_buf *wr_buf, | ||||||
|  | 		     struct smc_cdc_tx_pend *pend) | ||||||
|  | { | ||||||
|  | 	struct smc_link *link; | ||||||
|  | 	int rc; | ||||||
|  | 
 | ||||||
|  | 	link = &conn->lgr->lnk[SMC_SINGLE_LINK]; | ||||||
|  | 
 | ||||||
|  | 	smc_cdc_add_pending_send(conn, pend); | ||||||
|  | 
 | ||||||
|  | 	conn->tx_cdc_seq++; | ||||||
|  | 	conn->local_tx_ctrl.seqno = conn->tx_cdc_seq; | ||||||
|  | 	smc_host_msg_to_cdc((struct smc_cdc_msg *)wr_buf, | ||||||
|  | 			    &conn->local_tx_ctrl, conn); | ||||||
|  | 	rc = smc_wr_tx_send(link, (struct smc_wr_tx_pend_priv *)pend); | ||||||
|  | 	if (!rc) | ||||||
|  | 		smc_curs_write(&conn->rx_curs_confirmed, | ||||||
|  | 			       smc_curs_read(&conn->local_tx_ctrl.cons, conn), | ||||||
|  | 			       conn); | ||||||
|  | 
 | ||||||
|  | 	return rc; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | int smc_cdc_get_slot_and_msg_send(struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	struct smc_cdc_tx_pend *pend; | ||||||
|  | 	struct smc_wr_buf *wr_buf; | ||||||
|  | 	int rc; | ||||||
|  | 
 | ||||||
|  | 	rc = smc_cdc_get_free_slot(&conn->lgr->lnk[SMC_SINGLE_LINK], &wr_buf, | ||||||
|  | 				   &pend); | ||||||
|  | 	if (rc) | ||||||
|  | 		return rc; | ||||||
|  | 
 | ||||||
|  | 	return smc_cdc_msg_send(conn, wr_buf, pend); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static bool smc_cdc_tx_filter(struct smc_wr_tx_pend_priv *tx_pend, | ||||||
|  | 			      unsigned long data) | ||||||
|  | { | ||||||
|  | 	struct smc_connection *conn = (struct smc_connection *)data; | ||||||
|  | 	struct smc_cdc_tx_pend *cdc_pend = | ||||||
|  | 		(struct smc_cdc_tx_pend *)tx_pend; | ||||||
|  | 
 | ||||||
|  | 	return cdc_pend->conn == conn; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static void smc_cdc_tx_dismisser(struct smc_wr_tx_pend_priv *tx_pend) | ||||||
|  | { | ||||||
|  | 	struct smc_cdc_tx_pend *cdc_pend = | ||||||
|  | 		(struct smc_cdc_tx_pend *)tx_pend; | ||||||
|  | 
 | ||||||
|  | 	cdc_pend->conn = NULL; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | void smc_cdc_tx_dismiss_slots(struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	struct smc_link *link = &conn->lgr->lnk[SMC_SINGLE_LINK]; | ||||||
|  | 
 | ||||||
|  | 	smc_wr_tx_dismiss_slots(link, SMC_CDC_MSG_TYPE, | ||||||
|  | 				smc_cdc_tx_filter, smc_cdc_tx_dismisser, | ||||||
|  | 				(unsigned long)conn); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | /********************************* receive ***********************************/ | ||||||
|  | 
 | ||||||
|  | static inline bool smc_cdc_before(u16 seq1, u16 seq2) | ||||||
|  | { | ||||||
|  | 	return (s16)(seq1 - seq2) < 0; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static void smc_cdc_msg_recv_action(struct smc_sock *smc, | ||||||
|  | 				    struct smc_link *link, | ||||||
|  | 				    struct smc_cdc_msg *cdc) | ||||||
|  | { | ||||||
|  | 	union smc_host_cursor cons_old, prod_old; | ||||||
|  | 	struct smc_connection *conn = &smc->conn; | ||||||
|  | 	int diff_cons, diff_prod; | ||||||
|  | 
 | ||||||
|  | 	if (!cdc->prod_flags.failover_validation) { | ||||||
|  | 		if (smc_cdc_before(ntohs(cdc->seqno), | ||||||
|  | 				   conn->local_rx_ctrl.seqno)) | ||||||
|  | 			/* received seqno is old */ | ||||||
|  | 			return; | ||||||
|  | 	} | ||||||
|  | 	smc_curs_write(&prod_old, | ||||||
|  | 		       smc_curs_read(&conn->local_rx_ctrl.prod, conn), | ||||||
|  | 		       conn); | ||||||
|  | 	smc_curs_write(&cons_old, | ||||||
|  | 		       smc_curs_read(&conn->local_rx_ctrl.cons, conn), | ||||||
|  | 		       conn); | ||||||
|  | 	smc_cdc_msg_to_host(&conn->local_rx_ctrl, cdc, conn); | ||||||
|  | 
 | ||||||
|  | 	diff_cons = smc_curs_diff(conn->peer_rmbe_size, &cons_old, | ||||||
|  | 				  &conn->local_rx_ctrl.cons); | ||||||
|  | 	if (diff_cons) { | ||||||
|  | 		/* peer_rmbe_space is decreased during data transfer with RDMA
 | ||||||
|  | 		 * write | ||||||
|  | 		 */ | ||||||
|  | 		smp_mb__before_atomic(); | ||||||
|  | 		atomic_add(diff_cons, &conn->peer_rmbe_space); | ||||||
|  | 		/* guarantee 0 <= peer_rmbe_space <= peer_rmbe_size */ | ||||||
|  | 		smp_mb__after_atomic(); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	diff_prod = smc_curs_diff(conn->rmbe_size, &prod_old, | ||||||
|  | 				  &conn->local_rx_ctrl.prod); | ||||||
|  | 	if (diff_prod) { | ||||||
|  | 		/* bytes_to_rcv is decreased in smc_recvmsg */ | ||||||
|  | 		smp_mb__before_atomic(); | ||||||
|  | 		atomic_add(diff_prod, &conn->bytes_to_rcv); | ||||||
|  | 		/* guarantee 0 <= bytes_to_rcv <= rmbe_size */ | ||||||
|  | 		smp_mb__after_atomic(); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if (conn->local_rx_ctrl.conn_state_flags.peer_conn_abort) | ||||||
|  | 		smc->sk.sk_err = ECONNRESET; | ||||||
|  | 	if (smc_cdc_rxed_any_close_or_senddone(conn)) | ||||||
|  | 		/* subsequent patch: terminate connection */ | ||||||
|  | 
 | ||||||
|  | 	/* piggy backed tx info */ | ||||||
|  | 	/* subsequent patch: wake receivers if receive buffer space available */ | ||||||
|  | 
 | ||||||
|  | 	/* subsequent patch: trigger socket release if connection closed */ | ||||||
|  | 
 | ||||||
|  | 	/* socket connected but not accepted */ | ||||||
|  | 	if (!smc->sk.sk_socket) | ||||||
|  | 		return; | ||||||
|  | 
 | ||||||
|  | 	/* data available */ | ||||||
|  | 	/* subsequent patch: send delayed ack, wake receivers */ | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | /* called under tasklet context */ | ||||||
|  | static inline void smc_cdc_msg_recv(struct smc_cdc_msg *cdc, | ||||||
|  | 				    struct smc_link *link, u64 wr_id) | ||||||
|  | { | ||||||
|  | 	struct smc_link_group *lgr = container_of(link, struct smc_link_group, | ||||||
|  | 						  lnk[SMC_SINGLE_LINK]); | ||||||
|  | 	struct smc_connection *connection; | ||||||
|  | 	struct smc_sock *smc; | ||||||
|  | 
 | ||||||
|  | 	/* lookup connection */ | ||||||
|  | 	read_lock_bh(&lgr->conns_lock); | ||||||
|  | 	connection = smc_lgr_find_conn(ntohl(cdc->token), lgr); | ||||||
|  | 	if (!connection) { | ||||||
|  | 		read_unlock_bh(&lgr->conns_lock); | ||||||
|  | 		return; | ||||||
|  | 	} | ||||||
|  | 	smc = container_of(connection, struct smc_sock, conn); | ||||||
|  | 	sock_hold(&smc->sk); | ||||||
|  | 	read_unlock_bh(&lgr->conns_lock); | ||||||
|  | 	bh_lock_sock(&smc->sk); | ||||||
|  | 	smc_cdc_msg_recv_action(smc, link, cdc); | ||||||
|  | 	bh_unlock_sock(&smc->sk); | ||||||
|  | 	sock_put(&smc->sk); /* no free sk in softirq-context */ | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | /***************************** init, exit, misc ******************************/ | ||||||
|  | 
 | ||||||
|  | static void smc_cdc_rx_handler(struct ib_wc *wc, void *buf) | ||||||
|  | { | ||||||
|  | 	struct smc_link *link = (struct smc_link *)wc->qp->qp_context; | ||||||
|  | 	struct smc_cdc_msg *cdc = buf; | ||||||
|  | 
 | ||||||
|  | 	if (wc->byte_len < offsetof(struct smc_cdc_msg, reserved)) | ||||||
|  | 		return; /* short message */ | ||||||
|  | 	if (cdc->len != sizeof(*cdc)) | ||||||
|  | 		return; /* invalid message */ | ||||||
|  | 	smc_cdc_msg_recv(cdc, link, wc->wr_id); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static struct smc_wr_rx_handler smc_cdc_rx_handlers[] = { | ||||||
|  | 	{ | ||||||
|  | 		.handler	= smc_cdc_rx_handler, | ||||||
|  | 		.type		= SMC_CDC_MSG_TYPE | ||||||
|  | 	}, | ||||||
|  | 	{ | ||||||
|  | 		.handler	= NULL, | ||||||
|  | 	} | ||||||
|  | }; | ||||||
|  | 
 | ||||||
|  | int __init smc_cdc_init(void) | ||||||
|  | { | ||||||
|  | 	struct smc_wr_rx_handler *handler; | ||||||
|  | 	int rc = 0; | ||||||
|  | 
 | ||||||
|  | 	for (handler = smc_cdc_rx_handlers; handler->handler; handler++) { | ||||||
|  | 		INIT_HLIST_NODE(&handler->list); | ||||||
|  | 		rc = smc_wr_rx_register_handler(handler); | ||||||
|  | 		if (rc) | ||||||
|  | 			break; | ||||||
|  | 	} | ||||||
|  | 	return rc; | ||||||
|  | } | ||||||
							
								
								
									
										217
									
								
								net/smc/smc_cdc.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										217
									
								
								net/smc/smc_cdc.h
									
									
									
									
									
										Normal file
									
								
							|  | @ -0,0 +1,217 @@ | ||||||
|  | /*
 | ||||||
|  |  * Shared Memory Communications over RDMA (SMC-R) and RoCE | ||||||
|  |  * | ||||||
|  |  * Connection Data Control (CDC) | ||||||
|  |  * | ||||||
|  |  * Copyright IBM Corp. 2016 | ||||||
|  |  * | ||||||
|  |  * Author(s):  Ursula Braun <ubraun@linux.vnet.ibm.com> | ||||||
|  |  */ | ||||||
|  | 
 | ||||||
|  | #ifndef SMC_CDC_H | ||||||
|  | #define SMC_CDC_H | ||||||
|  | 
 | ||||||
|  | #include <linux/kernel.h> /* max_t */ | ||||||
|  | #include <linux/atomic.h> | ||||||
|  | #include <linux/in.h> | ||||||
|  | #include <linux/compiler.h> | ||||||
|  | 
 | ||||||
|  | #include "smc.h" | ||||||
|  | #include "smc_core.h" | ||||||
|  | #include "smc_wr.h" | ||||||
|  | 
 | ||||||
|  | #define	SMC_CDC_MSG_TYPE		0xFE | ||||||
|  | 
 | ||||||
|  | /* in network byte order */ | ||||||
|  | union smc_cdc_cursor {		/* SMC cursor */ | ||||||
|  | 	struct { | ||||||
|  | 		__be16	reserved; | ||||||
|  | 		__be16	wrap; | ||||||
|  | 		__be32	count; | ||||||
|  | 	}; | ||||||
|  | #ifdef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	atomic64_t	acurs;		/* for atomic processing */ | ||||||
|  | #else | ||||||
|  | 	u64		acurs;		/* for atomic processing */ | ||||||
|  | #endif | ||||||
|  | } __aligned(8); | ||||||
|  | 
 | ||||||
|  | /* in network byte order */ | ||||||
|  | struct smc_cdc_msg { | ||||||
|  | 	struct smc_wr_rx_hdr		common; /* .type = 0xFE */ | ||||||
|  | 	u8				len;	/* 44 */ | ||||||
|  | 	__be16				seqno; | ||||||
|  | 	__be32				token; | ||||||
|  | 	union smc_cdc_cursor		prod; | ||||||
|  | 	union smc_cdc_cursor		cons;	/* piggy backed "ack" */ | ||||||
|  | 	struct smc_cdc_producer_flags	prod_flags; | ||||||
|  | 	struct smc_cdc_conn_state_flags	conn_state_flags; | ||||||
|  | 	u8				reserved[18]; | ||||||
|  | } __aligned(8); | ||||||
|  | 
 | ||||||
|  | static inline bool smc_cdc_rxed_any_close(struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	return conn->local_rx_ctrl.conn_state_flags.peer_conn_abort || | ||||||
|  | 	       conn->local_rx_ctrl.conn_state_flags.peer_conn_closed; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline bool smc_cdc_rxed_any_close_or_senddone( | ||||||
|  | 	struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	return smc_cdc_rxed_any_close(conn) || | ||||||
|  | 	       conn->local_rx_ctrl.conn_state_flags.peer_done_writing; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_curs_add(int size, union smc_host_cursor *curs, | ||||||
|  | 				int value) | ||||||
|  | { | ||||||
|  | 	curs->count += value; | ||||||
|  | 	if (curs->count >= size) { | ||||||
|  | 		curs->wrap++; | ||||||
|  | 		curs->count -= size; | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | /* SMC cursors are 8 bytes long and require atomic reading and writing */ | ||||||
|  | static inline u64 smc_curs_read(union smc_host_cursor *curs, | ||||||
|  | 				struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	unsigned long flags; | ||||||
|  | 	u64 ret; | ||||||
|  | 
 | ||||||
|  | 	spin_lock_irqsave(&conn->acurs_lock, flags); | ||||||
|  | 	ret = curs->acurs; | ||||||
|  | 	spin_unlock_irqrestore(&conn->acurs_lock, flags); | ||||||
|  | 	return ret; | ||||||
|  | #else | ||||||
|  | 	return atomic64_read(&curs->acurs); | ||||||
|  | #endif | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline u64 smc_curs_read_net(union smc_cdc_cursor *curs, | ||||||
|  | 				    struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	unsigned long flags; | ||||||
|  | 	u64 ret; | ||||||
|  | 
 | ||||||
|  | 	spin_lock_irqsave(&conn->acurs_lock, flags); | ||||||
|  | 	ret = curs->acurs; | ||||||
|  | 	spin_unlock_irqrestore(&conn->acurs_lock, flags); | ||||||
|  | 	return ret; | ||||||
|  | #else | ||||||
|  | 	return atomic64_read(&curs->acurs); | ||||||
|  | #endif | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_curs_write(union smc_host_cursor *curs, u64 val, | ||||||
|  | 				  struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	unsigned long flags; | ||||||
|  | 
 | ||||||
|  | 	spin_lock_irqsave(&conn->acurs_lock, flags); | ||||||
|  | 	curs->acurs = val; | ||||||
|  | 	spin_unlock_irqrestore(&conn->acurs_lock, flags); | ||||||
|  | #else | ||||||
|  | 	atomic64_set(&curs->acurs, val); | ||||||
|  | #endif | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_curs_write_net(union smc_cdc_cursor *curs, u64 val, | ||||||
|  | 				      struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	unsigned long flags; | ||||||
|  | 
 | ||||||
|  | 	spin_lock_irqsave(&conn->acurs_lock, flags); | ||||||
|  | 	curs->acurs = val; | ||||||
|  | 	spin_unlock_irqrestore(&conn->acurs_lock, flags); | ||||||
|  | #else | ||||||
|  | 	atomic64_set(&curs->acurs, val); | ||||||
|  | #endif | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | /* calculate cursor difference between old and new, where old <= new */ | ||||||
|  | static inline int smc_curs_diff(unsigned int size, | ||||||
|  | 				union smc_host_cursor *old, | ||||||
|  | 				union smc_host_cursor *new) | ||||||
|  | { | ||||||
|  | 	if (old->wrap != new->wrap) | ||||||
|  | 		return max_t(int, 0, | ||||||
|  | 			     ((size - old->count) + new->count)); | ||||||
|  | 
 | ||||||
|  | 	return max_t(int, 0, (new->count - old->count)); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_host_cursor_to_cdc(union smc_cdc_cursor *peer, | ||||||
|  | 					  union smc_host_cursor *local, | ||||||
|  | 					  struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	union smc_host_cursor temp; | ||||||
|  | 
 | ||||||
|  | 	smc_curs_write(&temp, smc_curs_read(local, conn), conn); | ||||||
|  | 	peer->count = htonl(temp.count); | ||||||
|  | 	peer->wrap = htons(temp.wrap); | ||||||
|  | 	/* peer->reserved = htons(0); must be ensured by caller */ | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_host_msg_to_cdc(struct smc_cdc_msg *peer, | ||||||
|  | 				       struct smc_host_cdc_msg *local, | ||||||
|  | 				       struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	peer->common.type = local->common.type; | ||||||
|  | 	peer->len = local->len; | ||||||
|  | 	peer->seqno = htons(local->seqno); | ||||||
|  | 	peer->token = htonl(local->token); | ||||||
|  | 	smc_host_cursor_to_cdc(&peer->prod, &local->prod, conn); | ||||||
|  | 	smc_host_cursor_to_cdc(&peer->cons, &local->cons, conn); | ||||||
|  | 	peer->prod_flags = local->prod_flags; | ||||||
|  | 	peer->conn_state_flags = local->conn_state_flags; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_cdc_cursor_to_host(union smc_host_cursor *local, | ||||||
|  | 					  union smc_cdc_cursor *peer, | ||||||
|  | 					  struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	union smc_host_cursor temp, old; | ||||||
|  | 	union smc_cdc_cursor net; | ||||||
|  | 
 | ||||||
|  | 	smc_curs_write(&old, smc_curs_read(local, conn), conn); | ||||||
|  | 	smc_curs_write_net(&net, smc_curs_read_net(peer, conn), conn); | ||||||
|  | 	temp.count = ntohl(net.count); | ||||||
|  | 	temp.wrap = ntohs(net.wrap); | ||||||
|  | 	if ((old.wrap > temp.wrap) && temp.wrap) | ||||||
|  | 		return; | ||||||
|  | 	if ((old.wrap == temp.wrap) && | ||||||
|  | 	    (old.count > temp.count)) | ||||||
|  | 		return; | ||||||
|  | 	smc_curs_write(local, smc_curs_read(&temp, conn), conn); | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | static inline void smc_cdc_msg_to_host(struct smc_host_cdc_msg *local, | ||||||
|  | 				       struct smc_cdc_msg *peer, | ||||||
|  | 				       struct smc_connection *conn) | ||||||
|  | { | ||||||
|  | 	local->common.type = peer->common.type; | ||||||
|  | 	local->len = peer->len; | ||||||
|  | 	local->seqno = ntohs(peer->seqno); | ||||||
|  | 	local->token = ntohl(peer->token); | ||||||
|  | 	smc_cdc_cursor_to_host(&local->prod, &peer->prod, conn); | ||||||
|  | 	smc_cdc_cursor_to_host(&local->cons, &peer->cons, conn); | ||||||
|  | 	local->prod_flags = peer->prod_flags; | ||||||
|  | 	local->conn_state_flags = peer->conn_state_flags; | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | struct smc_cdc_tx_pend; | ||||||
|  | 
 | ||||||
|  | int smc_cdc_get_free_slot(struct smc_link *link, struct smc_wr_buf **wr_buf, | ||||||
|  | 			  struct smc_cdc_tx_pend **pend); | ||||||
|  | void smc_cdc_tx_dismiss_slots(struct smc_connection *conn); | ||||||
|  | int smc_cdc_msg_send(struct smc_connection *conn, struct smc_wr_buf *wr_buf, | ||||||
|  | 		     struct smc_cdc_tx_pend *pend); | ||||||
|  | int smc_cdc_get_slot_and_msg_send(struct smc_connection *conn); | ||||||
|  | int smc_cdc_init(void) __init; | ||||||
|  | 
 | ||||||
|  | #endif /* SMC_CDC_H */ | ||||||
|  | @ -22,6 +22,7 @@ | ||||||
| #include "smc_ib.h" | #include "smc_ib.h" | ||||||
| #include "smc_wr.h" | #include "smc_wr.h" | ||||||
| #include "smc_llc.h" | #include "smc_llc.h" | ||||||
|  | #include "smc_cdc.h" | ||||||
| 
 | 
 | ||||||
| #define SMC_LGR_NUM_INCR	256 | #define SMC_LGR_NUM_INCR	256 | ||||||
| #define SMC_LGR_FREE_DELAY	(600 * HZ) | #define SMC_LGR_FREE_DELAY	(600 * HZ) | ||||||
|  | @ -228,6 +229,7 @@ void smc_conn_free(struct smc_connection *conn) | ||||||
| 
 | 
 | ||||||
| 	if (!lgr) | 	if (!lgr) | ||||||
| 		return; | 		return; | ||||||
|  | 	smc_cdc_tx_dismiss_slots(conn); | ||||||
| 	smc_lgr_unregister_conn(conn); | 	smc_lgr_unregister_conn(conn); | ||||||
| 	smc_rmb_unuse(conn); | 	smc_rmb_unuse(conn); | ||||||
| 	smc_sndbuf_unuse(conn); | 	smc_sndbuf_unuse(conn); | ||||||
|  | @ -435,6 +437,11 @@ int smc_conn_create(struct smc_sock *smc, __be32 peer_in_addr, | ||||||
| 		smc_lgr_register_conn(conn); /* add smc conn to lgr */ | 		smc_lgr_register_conn(conn); /* add smc conn to lgr */ | ||||||
| 		rc = smc_link_determine_gid(conn->lgr); | 		rc = smc_link_determine_gid(conn->lgr); | ||||||
| 	} | 	} | ||||||
|  | 	conn->local_tx_ctrl.common.type = SMC_CDC_MSG_TYPE; | ||||||
|  | 	conn->local_tx_ctrl.len = sizeof(struct smc_cdc_msg); | ||||||
|  | #ifndef KERNEL_HAS_ATOMIC64 | ||||||
|  | 	spin_lock_init(&conn->acurs_lock); | ||||||
|  | #endif | ||||||
| 
 | 
 | ||||||
| out: | out: | ||||||
| 	return rc ? rc : local_contact; | 	return rc ? rc : local_contact; | ||||||
|  | @ -535,6 +542,7 @@ int smc_sndbuf_create(struct smc_sock *smc) | ||||||
| 		conn->sndbuf_desc = sndbuf_desc; | 		conn->sndbuf_desc = sndbuf_desc; | ||||||
| 		conn->sndbuf_size = tmp_bufsize; | 		conn->sndbuf_size = tmp_bufsize; | ||||||
| 		smc->sk.sk_sndbuf = tmp_bufsize * 2; | 		smc->sk.sk_sndbuf = tmp_bufsize * 2; | ||||||
|  | 		atomic_set(&conn->sndbuf_space, tmp_bufsize); | ||||||
| 		return 0; | 		return 0; | ||||||
| 	} else { | 	} else { | ||||||
| 		return -ENOMEM; | 		return -ENOMEM; | ||||||
|  | @ -611,6 +619,7 @@ int smc_rmb_create(struct smc_sock *smc) | ||||||
| 		conn->rmbe_size = tmp_bufsize; | 		conn->rmbe_size = tmp_bufsize; | ||||||
| 		conn->rmbe_size_short = tmp_bufsize_short; | 		conn->rmbe_size_short = tmp_bufsize_short; | ||||||
| 		smc->sk.sk_rcvbuf = tmp_bufsize * 2; | 		smc->sk.sk_rcvbuf = tmp_bufsize * 2; | ||||||
|  | 		atomic_set(&conn->bytes_to_rcv, 0); | ||||||
| 		return 0; | 		return 0; | ||||||
| 	} else { | 	} else { | ||||||
| 		return -ENOMEM; | 		return -ENOMEM; | ||||||
|  |  | ||||||
|  | @ -237,6 +237,25 @@ int smc_wr_tx_send(struct smc_link *link, struct smc_wr_tx_pend_priv *priv) | ||||||
| 	return rc; | 	return rc; | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | void smc_wr_tx_dismiss_slots(struct smc_link *link, u8 wr_rx_hdr_type, | ||||||
|  | 			     smc_wr_tx_filter filter, | ||||||
|  | 			     smc_wr_tx_dismisser dismisser, | ||||||
|  | 			     unsigned long data) | ||||||
|  | { | ||||||
|  | 	struct smc_wr_tx_pend_priv *tx_pend; | ||||||
|  | 	struct smc_wr_rx_hdr *wr_rx; | ||||||
|  | 	int i; | ||||||
|  | 
 | ||||||
|  | 	for_each_set_bit(i, link->wr_tx_mask, link->wr_tx_cnt) { | ||||||
|  | 		wr_rx = (struct smc_wr_rx_hdr *)&link->wr_rx_bufs[i]; | ||||||
|  | 		if (wr_rx->type != wr_rx_hdr_type) | ||||||
|  | 			continue; | ||||||
|  | 		tx_pend = &link->wr_tx_pends[i].priv; | ||||||
|  | 		if (filter(tx_pend, data)) | ||||||
|  | 			dismisser(tx_pend); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  | 
 | ||||||
| /****************************** receive queue ********************************/ | /****************************** receive queue ********************************/ | ||||||
| 
 | 
 | ||||||
| int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler) | int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler) | ||||||
|  |  | ||||||
|  | @ -36,6 +36,11 @@ typedef void (*smc_wr_tx_handler)(struct smc_wr_tx_pend_priv *, | ||||||
| 				  struct smc_link *, | 				  struct smc_link *, | ||||||
| 				  enum ib_wc_status); | 				  enum ib_wc_status); | ||||||
| 
 | 
 | ||||||
|  | typedef bool (*smc_wr_tx_filter)(struct smc_wr_tx_pend_priv *, | ||||||
|  | 				 unsigned long); | ||||||
|  | 
 | ||||||
|  | typedef void (*smc_wr_tx_dismisser)(struct smc_wr_tx_pend_priv *); | ||||||
|  | 
 | ||||||
| struct smc_wr_rx_handler { | struct smc_wr_rx_handler { | ||||||
| 	struct hlist_node	list;	/* hash table collision resolution */ | 	struct hlist_node	list;	/* hash table collision resolution */ | ||||||
| 	void			(*handler)(struct ib_wc *, void *); | 	void			(*handler)(struct ib_wc *, void *); | ||||||
|  | @ -87,6 +92,10 @@ int smc_wr_tx_put_slot(struct smc_link *link, | ||||||
| int smc_wr_tx_send(struct smc_link *link, | int smc_wr_tx_send(struct smc_link *link, | ||||||
| 		   struct smc_wr_tx_pend_priv *wr_pend_priv); | 		   struct smc_wr_tx_pend_priv *wr_pend_priv); | ||||||
| void smc_wr_tx_cq_handler(struct ib_cq *ib_cq, void *cq_context); | void smc_wr_tx_cq_handler(struct ib_cq *ib_cq, void *cq_context); | ||||||
|  | void smc_wr_tx_dismiss_slots(struct smc_link *lnk, u8 wr_rx_hdr_type, | ||||||
|  | 			     smc_wr_tx_filter filter, | ||||||
|  | 			     smc_wr_tx_dismisser dismisser, | ||||||
|  | 			     unsigned long data); | ||||||
| 
 | 
 | ||||||
| int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler); | int smc_wr_rx_register_handler(struct smc_wr_rx_handler *handler); | ||||||
| int smc_wr_rx_post_init(struct smc_link *link); | int smc_wr_rx_post_init(struct smc_link *link); | ||||||
|  |  | ||||||
		Loading…
	
		Reference in a new issue
	
	 Ursula Braun
						Ursula Braun