1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Networking over Thunderbolt/USB4 cables using USB4NET protocol
4  * (formerly Apple ThunderboltIP).
5  *
6  * Copyright (C) 2017, Intel Corporation
7  * Authors: Amir Levy <amir.jer.levy@intel.com>
8  *          Michael Jamet <michael.jamet@intel.com>
9  *          Mika Westerberg <mika.westerberg@linux.intel.com>
10  */
11 
12 #include <linux/atomic.h>
13 #include <linux/highmem.h>
14 #include <linux/if_vlan.h>
15 #include <linux/jhash.h>
16 #include <linux/module.h>
17 #include <linux/etherdevice.h>
18 #include <linux/rtnetlink.h>
19 #include <linux/sizes.h>
20 #include <linux/thunderbolt.h>
21 #include <linux/uuid.h>
22 #include <linux/workqueue.h>
23 
24 #include <net/ip6_checksum.h>
25 
26 /* Protocol timeouts in ms */
27 #define TBNET_LOGIN_DELAY	4500
28 #define TBNET_LOGIN_TIMEOUT	500
29 #define TBNET_LOGOUT_TIMEOUT	1000
30 
31 #define TBNET_RING_SIZE		256
32 #define TBNET_LOGIN_RETRIES	60
33 #define TBNET_LOGOUT_RETRIES	10
34 #define TBNET_E2E		BIT(0)
35 #define TBNET_MATCH_FRAGS_ID	BIT(1)
36 #define TBNET_64K_FRAMES	BIT(2)
37 #define TBNET_MAX_MTU		SZ_64K
38 #define TBNET_FRAME_SIZE	SZ_4K
39 #define TBNET_MAX_PAYLOAD_SIZE	\
40 	(TBNET_FRAME_SIZE - sizeof(struct thunderbolt_ip_frame_header))
41 /* Rx packets need to hold space for skb_shared_info */
42 #define TBNET_RX_MAX_SIZE	\
43 	(TBNET_FRAME_SIZE + SKB_DATA_ALIGN(sizeof(struct skb_shared_info)))
44 #define TBNET_RX_PAGE_ORDER	get_order(TBNET_RX_MAX_SIZE)
45 #define TBNET_RX_PAGE_SIZE	(PAGE_SIZE << TBNET_RX_PAGE_ORDER)
46 
47 #define TBNET_L0_PORT_NUM(route) ((route) & GENMASK(5, 0))
48 
49 /**
50  * struct thunderbolt_ip_frame_header - Header for each Thunderbolt frame
51  * @frame_size: size of the data with the frame
52  * @frame_index: running index on the frames
53  * @frame_id: ID of the frame to match frames to specific packet
54  * @frame_count: how many frames assembles a full packet
55  *
56  * Each data frame passed to the high-speed DMA ring has this header. If
57  * the XDomain network directory announces that %TBNET_MATCH_FRAGS_ID is
58  * supported then @frame_id is filled, otherwise it stays %0.
59  */
60 struct thunderbolt_ip_frame_header {
61 	u32 frame_size;
62 	u16 frame_index;
63 	u16 frame_id;
64 	u32 frame_count;
65 };
66 
67 enum thunderbolt_ip_frame_pdf {
68 	TBIP_PDF_FRAME_START = 1,
69 	TBIP_PDF_FRAME_END,
70 };
71 
72 enum thunderbolt_ip_type {
73 	TBIP_LOGIN,
74 	TBIP_LOGIN_RESPONSE,
75 	TBIP_LOGOUT,
76 	TBIP_STATUS,
77 };
78 
79 struct thunderbolt_ip_header {
80 	u32 route_hi;
81 	u32 route_lo;
82 	u32 length_sn;
83 	uuid_t uuid;
84 	uuid_t initiator_uuid;
85 	uuid_t target_uuid;
86 	u32 type;
87 	u32 command_id;
88 };
89 
90 #define TBIP_HDR_LENGTH_MASK		GENMASK(5, 0)
91 #define TBIP_HDR_SN_MASK		GENMASK(28, 27)
92 #define TBIP_HDR_SN_SHIFT		27
93 
94 struct thunderbolt_ip_login {
95 	struct thunderbolt_ip_header hdr;
96 	u32 proto_version;
97 	u32 transmit_path;
98 	u32 reserved[4];
99 };
100 
101 #define TBIP_LOGIN_PROTO_VERSION	1
102 
103 struct thunderbolt_ip_login_response {
104 	struct thunderbolt_ip_header hdr;
105 	u32 status;
106 	u32 receiver_mac[2];
107 	u32 receiver_mac_len;
108 	u32 reserved[4];
109 };
110 
111 struct thunderbolt_ip_logout {
112 	struct thunderbolt_ip_header hdr;
113 };
114 
115 struct thunderbolt_ip_status {
116 	struct thunderbolt_ip_header hdr;
117 	u32 status;
118 };
119 
120 struct tbnet_stats {
121 	u64 tx_packets;
122 	u64 rx_packets;
123 	u64 tx_bytes;
124 	u64 rx_bytes;
125 	u64 rx_errors;
126 	u64 tx_errors;
127 	u64 rx_length_errors;
128 	u64 rx_over_errors;
129 	u64 rx_crc_errors;
130 	u64 rx_missed_errors;
131 };
132 
133 struct tbnet_frame {
134 	struct net_device *dev;
135 	struct page *page;
136 	struct ring_frame frame;
137 };
138 
139 struct tbnet_ring {
140 	struct tbnet_frame frames[TBNET_RING_SIZE];
141 	unsigned int cons;
142 	unsigned int prod;
143 	struct tb_ring *ring;
144 };
145 
146 /**
147  * struct tbnet - ThunderboltIP network driver private data
148  * @svc: XDomain service the driver is bound to
149  * @xd: XDomain the service blongs to
150  * @handler: ThunderboltIP configuration protocol handler
151  * @dev: Networking device
152  * @napi: NAPI structure for Rx polling
153  * @stats: Network statistics
154  * @skb: Network packet that is currently processed on Rx path
155  * @command_id: ID used for next configuration protocol packet
156  * @login_sent: ThunderboltIP login message successfully sent
157  * @login_received: ThunderboltIP login message received from the remote
158  *		    host
159  * @local_transmit_path: HopID we are using to send out packets
160  * @remote_transmit_path: HopID the other end is using to send packets to us
161  * @connection_lock: Lock serializing access to @login_sent,
162  *		     @login_received and @transmit_path.
163  * @login_retries: Number of login retries currently done
164  * @login_work: Worker to send ThunderboltIP login packets
165  * @connected_work: Worker that finalizes the ThunderboltIP connection
166  *		    setup and enables DMA paths for high speed data
167  *		    transfers
168  * @disconnect_work: Worker that handles tearing down the ThunderboltIP
169  *		     connection
170  * @rx_hdr: Copy of the currently processed Rx frame. Used when a
171  *	    network packet consists of multiple Thunderbolt frames.
172  *	    In host byte order.
173  * @rx_ring: Software ring holding Rx frames
174  * @frame_id: Frame ID use for next Tx packet
175  *            (if %TBNET_MATCH_FRAGS_ID is supported in both ends)
176  * @tx_ring: Software ring holding Tx frames
177  */
178 struct tbnet {
179 	const struct tb_service *svc;
180 	struct tb_xdomain *xd;
181 	struct tb_protocol_handler handler;
182 	struct net_device *dev;
183 	struct napi_struct napi;
184 	struct tbnet_stats stats;
185 	struct sk_buff *skb;
186 	atomic_t command_id;
187 	bool login_sent;
188 	bool login_received;
189 	int local_transmit_path;
190 	int remote_transmit_path;
191 	struct mutex connection_lock;
192 	int login_retries;
193 	struct delayed_work login_work;
194 	struct work_struct connected_work;
195 	struct work_struct disconnect_work;
196 	struct thunderbolt_ip_frame_header rx_hdr;
197 	struct tbnet_ring rx_ring;
198 	atomic_t frame_id;
199 	struct tbnet_ring tx_ring;
200 };
201 
202 /* Network property directory UUID: c66189ca-1cce-4195-bdb8-49592e5f5a4f */
203 static const uuid_t tbnet_dir_uuid =
204 	UUID_INIT(0xc66189ca, 0x1cce, 0x4195,
205 		  0xbd, 0xb8, 0x49, 0x59, 0x2e, 0x5f, 0x5a, 0x4f);
206 
207 /* ThunderboltIP protocol UUID: 798f589e-3616-8a47-97c6-5664a920c8dd */
208 static const uuid_t tbnet_svc_uuid =
209 	UUID_INIT(0x798f589e, 0x3616, 0x8a47,
210 		  0x97, 0xc6, 0x56, 0x64, 0xa9, 0x20, 0xc8, 0xdd);
211 
212 static struct tb_property_dir *tbnet_dir;
213 
214 static bool tbnet_e2e = true;
215 module_param_named(e2e, tbnet_e2e, bool, 0444);
216 MODULE_PARM_DESC(e2e, "USB4NET full end-to-end flow control (default: true)");
217 
tbnet_fill_header(struct thunderbolt_ip_header * hdr,u64 route,u8 sequence,const uuid_t * initiator_uuid,const uuid_t * target_uuid,enum thunderbolt_ip_type type,size_t size,u32 command_id)218 static void tbnet_fill_header(struct thunderbolt_ip_header *hdr, u64 route,
219 	u8 sequence, const uuid_t *initiator_uuid, const uuid_t *target_uuid,
220 	enum thunderbolt_ip_type type, size_t size, u32 command_id)
221 {
222 	u32 length_sn;
223 
224 	/* Length does not include route_hi/lo and length_sn fields */
225 	length_sn = (size - 3 * 4) / 4;
226 	length_sn |= (sequence << TBIP_HDR_SN_SHIFT) & TBIP_HDR_SN_MASK;
227 
228 	hdr->route_hi = upper_32_bits(route);
229 	hdr->route_lo = lower_32_bits(route);
230 	hdr->length_sn = length_sn;
231 	uuid_copy(&hdr->uuid, &tbnet_svc_uuid);
232 	uuid_copy(&hdr->initiator_uuid, initiator_uuid);
233 	uuid_copy(&hdr->target_uuid, target_uuid);
234 	hdr->type = type;
235 	hdr->command_id = command_id;
236 }
237 
tbnet_login_response(struct tbnet * net,u64 route,u8 sequence,u32 command_id)238 static int tbnet_login_response(struct tbnet *net, u64 route, u8 sequence,
239 				u32 command_id)
240 {
241 	struct thunderbolt_ip_login_response reply;
242 	struct tb_xdomain *xd = net->xd;
243 
244 	memset(&reply, 0, sizeof(reply));
245 	tbnet_fill_header(&reply.hdr, route, sequence, xd->local_uuid,
246 			  xd->remote_uuid, TBIP_LOGIN_RESPONSE, sizeof(reply),
247 			  command_id);
248 	memcpy(reply.receiver_mac, net->dev->dev_addr, ETH_ALEN);
249 	reply.receiver_mac_len = ETH_ALEN;
250 
251 	return tb_xdomain_response(xd, &reply, sizeof(reply),
252 				   TB_CFG_PKG_XDOMAIN_RESP);
253 }
254 
tbnet_login_request(struct tbnet * net,u8 sequence)255 static int tbnet_login_request(struct tbnet *net, u8 sequence)
256 {
257 	struct thunderbolt_ip_login_response reply;
258 	struct thunderbolt_ip_login request;
259 	struct tb_xdomain *xd = net->xd;
260 
261 	memset(&request, 0, sizeof(request));
262 	tbnet_fill_header(&request.hdr, xd->route, sequence, xd->local_uuid,
263 			  xd->remote_uuid, TBIP_LOGIN, sizeof(request),
264 			  atomic_inc_return(&net->command_id));
265 
266 	request.proto_version = TBIP_LOGIN_PROTO_VERSION;
267 	request.transmit_path = net->local_transmit_path;
268 
269 	return tb_xdomain_request(xd, &request, sizeof(request),
270 				  TB_CFG_PKG_XDOMAIN_RESP, &reply,
271 				  sizeof(reply), TB_CFG_PKG_XDOMAIN_RESP,
272 				  TBNET_LOGIN_TIMEOUT);
273 }
274 
tbnet_logout_response(struct tbnet * net,u64 route,u8 sequence,u32 command_id)275 static int tbnet_logout_response(struct tbnet *net, u64 route, u8 sequence,
276 				 u32 command_id)
277 {
278 	struct thunderbolt_ip_status reply;
279 	struct tb_xdomain *xd = net->xd;
280 
281 	memset(&reply, 0, sizeof(reply));
282 	tbnet_fill_header(&reply.hdr, route, sequence, xd->local_uuid,
283 			  xd->remote_uuid, TBIP_STATUS, sizeof(reply),
284 			  atomic_inc_return(&net->command_id));
285 	return tb_xdomain_response(xd, &reply, sizeof(reply),
286 				   TB_CFG_PKG_XDOMAIN_RESP);
287 }
288 
tbnet_logout_request(struct tbnet * net)289 static int tbnet_logout_request(struct tbnet *net)
290 {
291 	struct thunderbolt_ip_logout request;
292 	struct thunderbolt_ip_status reply;
293 	struct tb_xdomain *xd = net->xd;
294 
295 	memset(&request, 0, sizeof(request));
296 	tbnet_fill_header(&request.hdr, xd->route, 0, xd->local_uuid,
297 			  xd->remote_uuid, TBIP_LOGOUT, sizeof(request),
298 			  atomic_inc_return(&net->command_id));
299 
300 	return tb_xdomain_request(xd, &request, sizeof(request),
301 				  TB_CFG_PKG_XDOMAIN_RESP, &reply,
302 				  sizeof(reply), TB_CFG_PKG_XDOMAIN_RESP,
303 				  TBNET_LOGOUT_TIMEOUT);
304 }
305 
start_login(struct tbnet * net)306 static void start_login(struct tbnet *net)
307 {
308 	mutex_lock(&net->connection_lock);
309 	net->login_sent = false;
310 	net->login_received = false;
311 	mutex_unlock(&net->connection_lock);
312 
313 	queue_delayed_work(system_long_wq, &net->login_work,
314 			   msecs_to_jiffies(1000));
315 }
316 
stop_login(struct tbnet * net)317 static void stop_login(struct tbnet *net)
318 {
319 	cancel_delayed_work_sync(&net->login_work);
320 	cancel_work_sync(&net->connected_work);
321 }
322 
tbnet_frame_size(const struct tbnet_frame * tf)323 static inline unsigned int tbnet_frame_size(const struct tbnet_frame *tf)
324 {
325 	return tf->frame.size ? : TBNET_FRAME_SIZE;
326 }
327 
tbnet_free_buffers(struct tbnet_ring * ring)328 static void tbnet_free_buffers(struct tbnet_ring *ring)
329 {
330 	unsigned int i;
331 
332 	for (i = 0; i < TBNET_RING_SIZE; i++) {
333 		struct device *dma_dev = tb_ring_dma_device(ring->ring);
334 		struct tbnet_frame *tf = &ring->frames[i];
335 		enum dma_data_direction dir;
336 		unsigned int order;
337 		size_t size;
338 
339 		if (!tf->page)
340 			continue;
341 
342 		if (ring->ring->is_tx) {
343 			dir = DMA_TO_DEVICE;
344 			order = 0;
345 			size = TBNET_FRAME_SIZE;
346 		} else {
347 			dir = DMA_FROM_DEVICE;
348 			order = TBNET_RX_PAGE_ORDER;
349 			size = TBNET_RX_PAGE_SIZE;
350 		}
351 
352 		if (tf->frame.buffer_phy)
353 			dma_unmap_page(dma_dev, tf->frame.buffer_phy, size,
354 				       dir);
355 
356 		__free_pages(tf->page, order);
357 		tf->page = NULL;
358 	}
359 
360 	ring->cons = 0;
361 	ring->prod = 0;
362 }
363 
tbnet_tear_down(struct tbnet * net,bool send_logout)364 static void tbnet_tear_down(struct tbnet *net, bool send_logout)
365 {
366 	netif_carrier_off(net->dev);
367 	netif_stop_queue(net->dev);
368 
369 	stop_login(net);
370 
371 	mutex_lock(&net->connection_lock);
372 
373 	if (net->login_sent && net->login_received) {
374 		int ret, retries = TBNET_LOGOUT_RETRIES;
375 
376 		while (send_logout && retries-- > 0) {
377 			ret = tbnet_logout_request(net);
378 			if (ret != -ETIMEDOUT)
379 				break;
380 		}
381 
382 		tb_ring_stop(net->rx_ring.ring);
383 		tb_ring_stop(net->tx_ring.ring);
384 		tbnet_free_buffers(&net->rx_ring);
385 		tbnet_free_buffers(&net->tx_ring);
386 
387 		ret = tb_xdomain_disable_paths(net->xd,
388 					       net->local_transmit_path,
389 					       net->rx_ring.ring->hop,
390 					       net->remote_transmit_path,
391 					       net->tx_ring.ring->hop);
392 		if (ret)
393 			netdev_warn(net->dev, "failed to disable DMA paths\n");
394 
395 		tb_xdomain_release_in_hopid(net->xd, net->remote_transmit_path);
396 		net->remote_transmit_path = 0;
397 	}
398 
399 	net->login_retries = 0;
400 	net->login_sent = false;
401 	net->login_received = false;
402 
403 	mutex_unlock(&net->connection_lock);
404 }
405 
tbnet_handle_packet(const void * buf,size_t size,void * data)406 static int tbnet_handle_packet(const void *buf, size_t size, void *data)
407 {
408 	const struct thunderbolt_ip_login *pkg = buf;
409 	struct tbnet *net = data;
410 	u32 command_id;
411 	int ret = 0;
412 	u32 sequence;
413 	u64 route;
414 
415 	/* Make sure the packet is for us */
416 	if (size < sizeof(struct thunderbolt_ip_header))
417 		return 0;
418 	if (!uuid_equal(&pkg->hdr.initiator_uuid, net->xd->remote_uuid))
419 		return 0;
420 	if (!uuid_equal(&pkg->hdr.target_uuid, net->xd->local_uuid))
421 		return 0;
422 
423 	route = ((u64)pkg->hdr.route_hi << 32) | pkg->hdr.route_lo;
424 	route &= ~BIT_ULL(63);
425 	if (route != net->xd->route)
426 		return 0;
427 
428 	sequence = pkg->hdr.length_sn & TBIP_HDR_SN_MASK;
429 	sequence >>= TBIP_HDR_SN_SHIFT;
430 	command_id = pkg->hdr.command_id;
431 
432 	switch (pkg->hdr.type) {
433 	case TBIP_LOGIN:
434 		if (!netif_running(net->dev))
435 			break;
436 
437 		ret = tbnet_login_response(net, route, sequence,
438 					   pkg->hdr.command_id);
439 		if (!ret) {
440 			mutex_lock(&net->connection_lock);
441 			net->login_received = true;
442 			net->remote_transmit_path = pkg->transmit_path;
443 
444 			/* If we reached the number of max retries or
445 			 * previous logout, schedule another round of
446 			 * login retries
447 			 */
448 			if (net->login_retries >= TBNET_LOGIN_RETRIES ||
449 			    !net->login_sent) {
450 				net->login_retries = 0;
451 				queue_delayed_work(system_long_wq,
452 						   &net->login_work, 0);
453 			}
454 			mutex_unlock(&net->connection_lock);
455 
456 			queue_work(system_long_wq, &net->connected_work);
457 		}
458 		break;
459 
460 	case TBIP_LOGOUT:
461 		ret = tbnet_logout_response(net, route, sequence, command_id);
462 		if (!ret)
463 			queue_work(system_long_wq, &net->disconnect_work);
464 		break;
465 
466 	default:
467 		return 0;
468 	}
469 
470 	if (ret)
471 		netdev_warn(net->dev, "failed to send ThunderboltIP response\n");
472 
473 	return 1;
474 }
475 
tbnet_available_buffers(const struct tbnet_ring * ring)476 static unsigned int tbnet_available_buffers(const struct tbnet_ring *ring)
477 {
478 	return ring->prod - ring->cons;
479 }
480 
tbnet_alloc_rx_buffers(struct tbnet * net,unsigned int nbuffers)481 static int tbnet_alloc_rx_buffers(struct tbnet *net, unsigned int nbuffers)
482 {
483 	struct tbnet_ring *ring = &net->rx_ring;
484 	int ret;
485 
486 	while (nbuffers--) {
487 		struct device *dma_dev = tb_ring_dma_device(ring->ring);
488 		unsigned int index = ring->prod & (TBNET_RING_SIZE - 1);
489 		struct tbnet_frame *tf = &ring->frames[index];
490 		dma_addr_t dma_addr;
491 
492 		if (tf->page)
493 			break;
494 
495 		/* Allocate page (order > 0) so that it can hold maximum
496 		 * ThunderboltIP frame (4kB) and the additional room for
497 		 * SKB shared info required by build_skb().
498 		 */
499 		tf->page = dev_alloc_pages(TBNET_RX_PAGE_ORDER);
500 		if (!tf->page) {
501 			ret = -ENOMEM;
502 			goto err_free;
503 		}
504 
505 		dma_addr = dma_map_page(dma_dev, tf->page, 0,
506 					TBNET_RX_PAGE_SIZE, DMA_FROM_DEVICE);
507 		if (dma_mapping_error(dma_dev, dma_addr)) {
508 			ret = -ENOMEM;
509 			goto err_free;
510 		}
511 
512 		tf->frame.buffer_phy = dma_addr;
513 		tf->dev = net->dev;
514 
515 		tb_ring_rx(ring->ring, &tf->frame);
516 
517 		ring->prod++;
518 	}
519 
520 	return 0;
521 
522 err_free:
523 	tbnet_free_buffers(ring);
524 	return ret;
525 }
526 
tbnet_get_tx_buffer(struct tbnet * net)527 static struct tbnet_frame *tbnet_get_tx_buffer(struct tbnet *net)
528 {
529 	struct tbnet_ring *ring = &net->tx_ring;
530 	struct device *dma_dev = tb_ring_dma_device(ring->ring);
531 	struct tbnet_frame *tf;
532 	unsigned int index;
533 
534 	if (!tbnet_available_buffers(ring))
535 		return NULL;
536 
537 	index = ring->cons++ & (TBNET_RING_SIZE - 1);
538 
539 	tf = &ring->frames[index];
540 	tf->frame.size = 0;
541 
542 	dma_sync_single_for_cpu(dma_dev, tf->frame.buffer_phy,
543 				tbnet_frame_size(tf), DMA_TO_DEVICE);
544 
545 	return tf;
546 }
547 
tbnet_tx_callback(struct tb_ring * ring,struct ring_frame * frame,bool canceled)548 static void tbnet_tx_callback(struct tb_ring *ring, struct ring_frame *frame,
549 			      bool canceled)
550 {
551 	struct tbnet_frame *tf = container_of(frame, typeof(*tf), frame);
552 	struct tbnet *net = netdev_priv(tf->dev);
553 
554 	/* Return buffer to the ring */
555 	net->tx_ring.prod++;
556 
557 	if (tbnet_available_buffers(&net->tx_ring) >= TBNET_RING_SIZE / 2)
558 		netif_wake_queue(net->dev);
559 }
560 
tbnet_alloc_tx_buffers(struct tbnet * net)561 static int tbnet_alloc_tx_buffers(struct tbnet *net)
562 {
563 	struct tbnet_ring *ring = &net->tx_ring;
564 	struct device *dma_dev = tb_ring_dma_device(ring->ring);
565 	unsigned int i;
566 
567 	for (i = 0; i < TBNET_RING_SIZE; i++) {
568 		struct tbnet_frame *tf = &ring->frames[i];
569 		dma_addr_t dma_addr;
570 
571 		tf->page = alloc_page(GFP_KERNEL);
572 		if (!tf->page) {
573 			tbnet_free_buffers(ring);
574 			return -ENOMEM;
575 		}
576 
577 		dma_addr = dma_map_page(dma_dev, tf->page, 0, TBNET_FRAME_SIZE,
578 					DMA_TO_DEVICE);
579 		if (dma_mapping_error(dma_dev, dma_addr)) {
580 			__free_page(tf->page);
581 			tf->page = NULL;
582 			tbnet_free_buffers(ring);
583 			return -ENOMEM;
584 		}
585 
586 		tf->dev = net->dev;
587 		tf->frame.buffer_phy = dma_addr;
588 		tf->frame.callback = tbnet_tx_callback;
589 		tf->frame.sof = TBIP_PDF_FRAME_START;
590 		tf->frame.eof = TBIP_PDF_FRAME_END;
591 	}
592 
593 	ring->cons = 0;
594 	ring->prod = TBNET_RING_SIZE - 1;
595 
596 	return 0;
597 }
598 
tbnet_connected_work(struct work_struct * work)599 static void tbnet_connected_work(struct work_struct *work)
600 {
601 	struct tbnet *net = container_of(work, typeof(*net), connected_work);
602 	bool connected;
603 	int ret;
604 
605 	if (netif_carrier_ok(net->dev))
606 		return;
607 
608 	mutex_lock(&net->connection_lock);
609 	connected = net->login_sent && net->login_received;
610 	mutex_unlock(&net->connection_lock);
611 
612 	if (!connected)
613 		return;
614 
615 	ret = tb_xdomain_alloc_in_hopid(net->xd, net->remote_transmit_path);
616 	if (ret != net->remote_transmit_path) {
617 		netdev_err(net->dev, "failed to allocate Rx HopID\n");
618 		return;
619 	}
620 
621 	/* Both logins successful so enable the rings, high-speed DMA
622 	 * paths and start the network device queue.
623 	 *
624 	 * Note we enable the DMA paths last to make sure we have primed
625 	 * the Rx ring before any incoming packets are allowed to
626 	 * arrive.
627 	 */
628 	tb_ring_start(net->tx_ring.ring);
629 	tb_ring_start(net->rx_ring.ring);
630 
631 	ret = tbnet_alloc_rx_buffers(net, TBNET_RING_SIZE);
632 	if (ret)
633 		goto err_stop_rings;
634 
635 	ret = tbnet_alloc_tx_buffers(net);
636 	if (ret)
637 		goto err_free_rx_buffers;
638 
639 	ret = tb_xdomain_enable_paths(net->xd, net->local_transmit_path,
640 				      net->rx_ring.ring->hop,
641 				      net->remote_transmit_path,
642 				      net->tx_ring.ring->hop);
643 	if (ret) {
644 		netdev_err(net->dev, "failed to enable DMA paths\n");
645 		goto err_free_tx_buffers;
646 	}
647 
648 	netif_carrier_on(net->dev);
649 	netif_start_queue(net->dev);
650 	return;
651 
652 err_free_tx_buffers:
653 	tbnet_free_buffers(&net->tx_ring);
654 err_free_rx_buffers:
655 	tbnet_free_buffers(&net->rx_ring);
656 err_stop_rings:
657 	tb_ring_stop(net->rx_ring.ring);
658 	tb_ring_stop(net->tx_ring.ring);
659 	tb_xdomain_release_in_hopid(net->xd, net->remote_transmit_path);
660 }
661 
tbnet_login_work(struct work_struct * work)662 static void tbnet_login_work(struct work_struct *work)
663 {
664 	struct tbnet *net = container_of(work, typeof(*net), login_work.work);
665 	unsigned long delay = msecs_to_jiffies(TBNET_LOGIN_DELAY);
666 	int ret;
667 
668 	if (netif_carrier_ok(net->dev))
669 		return;
670 
671 	ret = tbnet_login_request(net, net->login_retries % 4);
672 	if (ret) {
673 		if (net->login_retries++ < TBNET_LOGIN_RETRIES) {
674 			queue_delayed_work(system_long_wq, &net->login_work,
675 					   delay);
676 		} else {
677 			netdev_info(net->dev, "ThunderboltIP login timed out\n");
678 		}
679 	} else {
680 		net->login_retries = 0;
681 
682 		mutex_lock(&net->connection_lock);
683 		net->login_sent = true;
684 		mutex_unlock(&net->connection_lock);
685 
686 		queue_work(system_long_wq, &net->connected_work);
687 	}
688 }
689 
tbnet_disconnect_work(struct work_struct * work)690 static void tbnet_disconnect_work(struct work_struct *work)
691 {
692 	struct tbnet *net = container_of(work, typeof(*net), disconnect_work);
693 
694 	tbnet_tear_down(net, false);
695 }
696 
tbnet_check_frame(struct tbnet * net,const struct tbnet_frame * tf,const struct thunderbolt_ip_frame_header * hdr)697 static bool tbnet_check_frame(struct tbnet *net, const struct tbnet_frame *tf,
698 			      const struct thunderbolt_ip_frame_header *hdr)
699 {
700 	u32 frame_id, frame_count, frame_size, frame_index;
701 	unsigned int size;
702 
703 	if (tf->frame.flags & RING_DESC_CRC_ERROR) {
704 		net->stats.rx_crc_errors++;
705 		return false;
706 	} else if (tf->frame.flags & RING_DESC_BUFFER_OVERRUN) {
707 		net->stats.rx_over_errors++;
708 		return false;
709 	}
710 
711 	/* Should be greater than just header i.e. contains data */
712 	size = tbnet_frame_size(tf);
713 	if (size <= sizeof(*hdr)) {
714 		net->stats.rx_length_errors++;
715 		return false;
716 	}
717 
718 	frame_count = le32_to_cpu(hdr->frame_count);
719 	frame_size = le32_to_cpu(hdr->frame_size);
720 	frame_index = le16_to_cpu(hdr->frame_index);
721 	frame_id = le16_to_cpu(hdr->frame_id);
722 
723 	if ((frame_size > size - sizeof(*hdr)) || !frame_size) {
724 		net->stats.rx_length_errors++;
725 		return false;
726 	}
727 
728 	/* In case we're in the middle of packet, validate the frame
729 	 * header based on first fragment of the packet.
730 	 */
731 	if (net->skb && net->rx_hdr.frame_count) {
732 		/* Check the frame count fits the count field */
733 		if (frame_count != net->rx_hdr.frame_count) {
734 			net->stats.rx_length_errors++;
735 			return false;
736 		}
737 
738 		/* Check the frame identifiers are incremented correctly,
739 		 * and id is matching.
740 		 */
741 		if (frame_index != net->rx_hdr.frame_index + 1 ||
742 		    frame_id != net->rx_hdr.frame_id) {
743 			net->stats.rx_missed_errors++;
744 			return false;
745 		}
746 
747 		if (net->skb->len + frame_size > TBNET_MAX_MTU) {
748 			net->stats.rx_length_errors++;
749 			return false;
750 		}
751 
752 		return true;
753 	}
754 
755 	/* Start of packet, validate the frame header */
756 	if (frame_count == 0 || frame_count > TBNET_RING_SIZE / 4) {
757 		net->stats.rx_length_errors++;
758 		return false;
759 	}
760 	if (frame_index != 0) {
761 		net->stats.rx_missed_errors++;
762 		return false;
763 	}
764 
765 	return true;
766 }
767 
tbnet_poll(struct napi_struct * napi,int budget)768 static int tbnet_poll(struct napi_struct *napi, int budget)
769 {
770 	struct tbnet *net = container_of(napi, struct tbnet, napi);
771 	unsigned int cleaned_count = tbnet_available_buffers(&net->rx_ring);
772 	struct device *dma_dev = tb_ring_dma_device(net->rx_ring.ring);
773 	unsigned int rx_packets = 0;
774 
775 	while (rx_packets < budget) {
776 		const struct thunderbolt_ip_frame_header *hdr;
777 		unsigned int hdr_size = sizeof(*hdr);
778 		struct sk_buff *skb = NULL;
779 		struct ring_frame *frame;
780 		struct tbnet_frame *tf;
781 		struct page *page;
782 		bool last = true;
783 		u32 frame_size;
784 
785 		/* Return some buffers to hardware, one at a time is too
786 		 * slow so allocate MAX_SKB_FRAGS buffers at the same
787 		 * time.
788 		 */
789 		if (cleaned_count >= MAX_SKB_FRAGS) {
790 			tbnet_alloc_rx_buffers(net, cleaned_count);
791 			cleaned_count = 0;
792 		}
793 
794 		frame = tb_ring_poll(net->rx_ring.ring);
795 		if (!frame)
796 			break;
797 
798 		dma_unmap_page(dma_dev, frame->buffer_phy,
799 			       TBNET_RX_PAGE_SIZE, DMA_FROM_DEVICE);
800 
801 		tf = container_of(frame, typeof(*tf), frame);
802 
803 		page = tf->page;
804 		tf->page = NULL;
805 		net->rx_ring.cons++;
806 		cleaned_count++;
807 
808 		hdr = page_address(page);
809 		if (!tbnet_check_frame(net, tf, hdr)) {
810 			__free_pages(page, TBNET_RX_PAGE_ORDER);
811 			dev_kfree_skb_any(net->skb);
812 			net->skb = NULL;
813 			continue;
814 		}
815 
816 		frame_size = le32_to_cpu(hdr->frame_size);
817 
818 		skb = net->skb;
819 		if (!skb) {
820 			skb = build_skb(page_address(page),
821 					TBNET_RX_PAGE_SIZE);
822 			if (!skb) {
823 				__free_pages(page, TBNET_RX_PAGE_ORDER);
824 				net->stats.rx_errors++;
825 				break;
826 			}
827 
828 			skb_reserve(skb, hdr_size);
829 			skb_put(skb, frame_size);
830 
831 			net->skb = skb;
832 		} else {
833 			skb_add_rx_frag(skb, skb_shinfo(skb)->nr_frags,
834 					page, hdr_size, frame_size,
835 					TBNET_RX_PAGE_SIZE - hdr_size);
836 		}
837 
838 		net->rx_hdr.frame_size = frame_size;
839 		net->rx_hdr.frame_count = le32_to_cpu(hdr->frame_count);
840 		net->rx_hdr.frame_index = le16_to_cpu(hdr->frame_index);
841 		net->rx_hdr.frame_id = le16_to_cpu(hdr->frame_id);
842 		last = net->rx_hdr.frame_index == net->rx_hdr.frame_count - 1;
843 
844 		rx_packets++;
845 		net->stats.rx_bytes += frame_size;
846 
847 		if (last) {
848 			skb->protocol = eth_type_trans(skb, net->dev);
849 			napi_gro_receive(&net->napi, skb);
850 			net->skb = NULL;
851 		}
852 	}
853 
854 	net->stats.rx_packets += rx_packets;
855 
856 	if (cleaned_count)
857 		tbnet_alloc_rx_buffers(net, cleaned_count);
858 
859 	if (rx_packets >= budget)
860 		return budget;
861 
862 	napi_complete_done(napi, rx_packets);
863 	/* Re-enable the ring interrupt */
864 	tb_ring_poll_complete(net->rx_ring.ring);
865 
866 	return rx_packets;
867 }
868 
tbnet_start_poll(void * data)869 static void tbnet_start_poll(void *data)
870 {
871 	struct tbnet *net = data;
872 
873 	napi_schedule(&net->napi);
874 }
875 
tbnet_open(struct net_device * dev)876 static int tbnet_open(struct net_device *dev)
877 {
878 	struct tbnet *net = netdev_priv(dev);
879 	struct tb_xdomain *xd = net->xd;
880 	u16 sof_mask, eof_mask;
881 	struct tb_ring *ring;
882 	unsigned int flags;
883 	int hopid;
884 
885 	netif_carrier_off(dev);
886 
887 	ring = tb_ring_alloc_tx(xd->tb->nhi, -1, TBNET_RING_SIZE,
888 				RING_FLAG_FRAME);
889 	if (!ring) {
890 		netdev_err(dev, "failed to allocate Tx ring\n");
891 		return -ENOMEM;
892 	}
893 	net->tx_ring.ring = ring;
894 
895 	hopid = tb_xdomain_alloc_out_hopid(xd, -1);
896 	if (hopid < 0) {
897 		netdev_err(dev, "failed to allocate Tx HopID\n");
898 		tb_ring_free(net->tx_ring.ring);
899 		net->tx_ring.ring = NULL;
900 		return hopid;
901 	}
902 	net->local_transmit_path = hopid;
903 
904 	sof_mask = BIT(TBIP_PDF_FRAME_START);
905 	eof_mask = BIT(TBIP_PDF_FRAME_END);
906 
907 	flags = RING_FLAG_FRAME;
908 	/* Only enable full E2E if the other end supports it too */
909 	if (tbnet_e2e && net->svc->prtcstns & TBNET_E2E)
910 		flags |= RING_FLAG_E2E;
911 
912 	ring = tb_ring_alloc_rx(xd->tb->nhi, -1, TBNET_RING_SIZE, flags,
913 				net->tx_ring.ring->hop, sof_mask,
914 				eof_mask, tbnet_start_poll, net);
915 	if (!ring) {
916 		netdev_err(dev, "failed to allocate Rx ring\n");
917 		tb_xdomain_release_out_hopid(xd, hopid);
918 		tb_ring_free(net->tx_ring.ring);
919 		net->tx_ring.ring = NULL;
920 		return -ENOMEM;
921 	}
922 	net->rx_ring.ring = ring;
923 
924 	napi_enable(&net->napi);
925 	start_login(net);
926 
927 	return 0;
928 }
929 
tbnet_stop(struct net_device * dev)930 static int tbnet_stop(struct net_device *dev)
931 {
932 	struct tbnet *net = netdev_priv(dev);
933 
934 	napi_disable(&net->napi);
935 
936 	cancel_work_sync(&net->disconnect_work);
937 	tbnet_tear_down(net, true);
938 
939 	tb_ring_free(net->rx_ring.ring);
940 	net->rx_ring.ring = NULL;
941 
942 	tb_xdomain_release_out_hopid(net->xd, net->local_transmit_path);
943 	tb_ring_free(net->tx_ring.ring);
944 	net->tx_ring.ring = NULL;
945 
946 	return 0;
947 }
948 
tbnet_xmit_csum_and_map(struct tbnet * net,struct sk_buff * skb,struct tbnet_frame ** frames,u32 frame_count)949 static bool tbnet_xmit_csum_and_map(struct tbnet *net, struct sk_buff *skb,
950 	struct tbnet_frame **frames, u32 frame_count)
951 {
952 	struct thunderbolt_ip_frame_header *hdr = page_address(frames[0]->page);
953 	struct device *dma_dev = tb_ring_dma_device(net->tx_ring.ring);
954 	__wsum wsum = htonl(skb->len - skb_transport_offset(skb));
955 	unsigned int i, len, offset = skb_transport_offset(skb);
956 	__be16 protocol = skb->protocol;
957 	void *data = skb->data;
958 	void *dest = hdr + 1;
959 	__sum16 *tucso;
960 
961 	if (skb->ip_summed != CHECKSUM_PARTIAL) {
962 		/* No need to calculate checksum so we just update the
963 		 * total frame count and sync the frames for DMA.
964 		 */
965 		for (i = 0; i < frame_count; i++) {
966 			hdr = page_address(frames[i]->page);
967 			hdr->frame_count = cpu_to_le32(frame_count);
968 			dma_sync_single_for_device(dma_dev,
969 				frames[i]->frame.buffer_phy,
970 				tbnet_frame_size(frames[i]), DMA_TO_DEVICE);
971 		}
972 
973 		return true;
974 	}
975 
976 	if (protocol == htons(ETH_P_8021Q)) {
977 		struct vlan_hdr *vhdr, vh;
978 
979 		vhdr = skb_header_pointer(skb, ETH_HLEN, sizeof(vh), &vh);
980 		if (!vhdr)
981 			return false;
982 
983 		protocol = vhdr->h_vlan_encapsulated_proto;
984 	}
985 
986 	/* Data points on the beginning of packet.
987 	 * Check is the checksum absolute place in the packet.
988 	 * ipcso will update IP checksum.
989 	 * tucso will update TCP/UPD checksum.
990 	 */
991 	if (protocol == htons(ETH_P_IP)) {
992 		__sum16 *ipcso = dest + ((void *)&(ip_hdr(skb)->check) - data);
993 
994 		*ipcso = 0;
995 		*ipcso = ip_fast_csum(dest + skb_network_offset(skb),
996 				      ip_hdr(skb)->ihl);
997 
998 		if (ip_hdr(skb)->protocol == IPPROTO_TCP)
999 			tucso = dest + ((void *)&(tcp_hdr(skb)->check) - data);
1000 		else if (ip_hdr(skb)->protocol == IPPROTO_UDP)
1001 			tucso = dest + ((void *)&(udp_hdr(skb)->check) - data);
1002 		else
1003 			return false;
1004 
1005 		*tucso = ~csum_tcpudp_magic(ip_hdr(skb)->saddr,
1006 					    ip_hdr(skb)->daddr, 0,
1007 					    ip_hdr(skb)->protocol, 0);
1008 	} else if (skb_is_gso_v6(skb)) {
1009 		tucso = dest + ((void *)&(tcp_hdr(skb)->check) - data);
1010 		*tucso = ~csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
1011 					  &ipv6_hdr(skb)->daddr, 0,
1012 					  IPPROTO_TCP, 0);
1013 		return false;
1014 	} else if (protocol == htons(ETH_P_IPV6)) {
1015 		tucso = dest + skb_checksum_start_offset(skb) + skb->csum_offset;
1016 		*tucso = ~csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
1017 					  &ipv6_hdr(skb)->daddr, 0,
1018 					  ipv6_hdr(skb)->nexthdr, 0);
1019 	} else {
1020 		return false;
1021 	}
1022 
1023 	/* First frame was headers, rest of the frames contain data.
1024 	 * Calculate checksum over each frame.
1025 	 */
1026 	for (i = 0; i < frame_count; i++) {
1027 		hdr = page_address(frames[i]->page);
1028 		dest = (void *)(hdr + 1) + offset;
1029 		len = le32_to_cpu(hdr->frame_size) - offset;
1030 		wsum = csum_partial(dest, len, wsum);
1031 		hdr->frame_count = cpu_to_le32(frame_count);
1032 
1033 		offset = 0;
1034 	}
1035 
1036 	*tucso = csum_fold(wsum);
1037 
1038 	/* Checksum is finally calculated and we don't touch the memory
1039 	 * anymore, so DMA sync the frames now.
1040 	 */
1041 	for (i = 0; i < frame_count; i++) {
1042 		dma_sync_single_for_device(dma_dev, frames[i]->frame.buffer_phy,
1043 			tbnet_frame_size(frames[i]), DMA_TO_DEVICE);
1044 	}
1045 
1046 	return true;
1047 }
1048 
tbnet_kmap_frag(struct sk_buff * skb,unsigned int frag_num,unsigned int * len)1049 static void *tbnet_kmap_frag(struct sk_buff *skb, unsigned int frag_num,
1050 			     unsigned int *len)
1051 {
1052 	const skb_frag_t *frag = &skb_shinfo(skb)->frags[frag_num];
1053 
1054 	*len = skb_frag_size(frag);
1055 	return kmap_atomic(skb_frag_page(frag)) + skb_frag_off(frag);
1056 }
1057 
tbnet_start_xmit(struct sk_buff * skb,struct net_device * dev)1058 static netdev_tx_t tbnet_start_xmit(struct sk_buff *skb,
1059 				    struct net_device *dev)
1060 {
1061 	struct tbnet *net = netdev_priv(dev);
1062 	struct tbnet_frame *frames[MAX_SKB_FRAGS];
1063 	u16 frame_id = atomic_read(&net->frame_id);
1064 	struct thunderbolt_ip_frame_header *hdr;
1065 	unsigned int len = skb_headlen(skb);
1066 	unsigned int data_len = skb->len;
1067 	unsigned int nframes, i;
1068 	unsigned int frag = 0;
1069 	void *src = skb->data;
1070 	u32 frame_index = 0;
1071 	bool unmap = false;
1072 	void *dest;
1073 
1074 	nframes = DIV_ROUND_UP(data_len, TBNET_MAX_PAYLOAD_SIZE);
1075 	if (tbnet_available_buffers(&net->tx_ring) < nframes) {
1076 		netif_stop_queue(net->dev);
1077 		return NETDEV_TX_BUSY;
1078 	}
1079 
1080 	frames[frame_index] = tbnet_get_tx_buffer(net);
1081 	if (!frames[frame_index])
1082 		goto err_drop;
1083 
1084 	hdr = page_address(frames[frame_index]->page);
1085 	dest = hdr + 1;
1086 
1087 	/* If overall packet is bigger than the frame data size */
1088 	while (data_len > TBNET_MAX_PAYLOAD_SIZE) {
1089 		unsigned int size_left = TBNET_MAX_PAYLOAD_SIZE;
1090 
1091 		hdr->frame_size = cpu_to_le32(TBNET_MAX_PAYLOAD_SIZE);
1092 		hdr->frame_index = cpu_to_le16(frame_index);
1093 		hdr->frame_id = cpu_to_le16(frame_id);
1094 
1095 		do {
1096 			if (len > size_left) {
1097 				/* Copy data onto Tx buffer data with
1098 				 * full frame size then break and go to
1099 				 * next frame
1100 				 */
1101 				memcpy(dest, src, size_left);
1102 				len -= size_left;
1103 				dest += size_left;
1104 				src += size_left;
1105 				break;
1106 			}
1107 
1108 			memcpy(dest, src, len);
1109 			size_left -= len;
1110 			dest += len;
1111 
1112 			if (unmap) {
1113 				kunmap_atomic(src);
1114 				unmap = false;
1115 			}
1116 
1117 			/* Ensure all fragments have been processed */
1118 			if (frag < skb_shinfo(skb)->nr_frags) {
1119 				/* Map and then unmap quickly */
1120 				src = tbnet_kmap_frag(skb, frag++, &len);
1121 				unmap = true;
1122 			} else if (unlikely(size_left > 0)) {
1123 				goto err_drop;
1124 			}
1125 		} while (size_left > 0);
1126 
1127 		data_len -= TBNET_MAX_PAYLOAD_SIZE;
1128 		frame_index++;
1129 
1130 		frames[frame_index] = tbnet_get_tx_buffer(net);
1131 		if (!frames[frame_index])
1132 			goto err_drop;
1133 
1134 		hdr = page_address(frames[frame_index]->page);
1135 		dest = hdr + 1;
1136 	}
1137 
1138 	hdr->frame_size = cpu_to_le32(data_len);
1139 	hdr->frame_index = cpu_to_le16(frame_index);
1140 	hdr->frame_id = cpu_to_le16(frame_id);
1141 
1142 	frames[frame_index]->frame.size = data_len + sizeof(*hdr);
1143 
1144 	/* In case the remaining data_len is smaller than a frame */
1145 	while (len < data_len) {
1146 		memcpy(dest, src, len);
1147 		data_len -= len;
1148 		dest += len;
1149 
1150 		if (unmap) {
1151 			kunmap_atomic(src);
1152 			unmap = false;
1153 		}
1154 
1155 		if (frag < skb_shinfo(skb)->nr_frags) {
1156 			src = tbnet_kmap_frag(skb, frag++, &len);
1157 			unmap = true;
1158 		} else if (unlikely(data_len > 0)) {
1159 			goto err_drop;
1160 		}
1161 	}
1162 
1163 	memcpy(dest, src, data_len);
1164 
1165 	if (unmap)
1166 		kunmap_atomic(src);
1167 
1168 	if (!tbnet_xmit_csum_and_map(net, skb, frames, frame_index + 1))
1169 		goto err_drop;
1170 
1171 	for (i = 0; i < frame_index + 1; i++)
1172 		tb_ring_tx(net->tx_ring.ring, &frames[i]->frame);
1173 
1174 	if (net->svc->prtcstns & TBNET_MATCH_FRAGS_ID)
1175 		atomic_inc(&net->frame_id);
1176 
1177 	net->stats.tx_packets++;
1178 	net->stats.tx_bytes += skb->len;
1179 
1180 	dev_consume_skb_any(skb);
1181 
1182 	return NETDEV_TX_OK;
1183 
1184 err_drop:
1185 	/* We can re-use the buffers */
1186 	net->tx_ring.cons -= frame_index;
1187 
1188 	dev_kfree_skb_any(skb);
1189 	net->stats.tx_errors++;
1190 
1191 	return NETDEV_TX_OK;
1192 }
1193 
tbnet_get_stats64(struct net_device * dev,struct rtnl_link_stats64 * stats)1194 static void tbnet_get_stats64(struct net_device *dev,
1195 			      struct rtnl_link_stats64 *stats)
1196 {
1197 	struct tbnet *net = netdev_priv(dev);
1198 
1199 	stats->tx_packets = net->stats.tx_packets;
1200 	stats->rx_packets = net->stats.rx_packets;
1201 	stats->tx_bytes = net->stats.tx_bytes;
1202 	stats->rx_bytes = net->stats.rx_bytes;
1203 	stats->rx_errors = net->stats.rx_errors + net->stats.rx_length_errors +
1204 		net->stats.rx_over_errors + net->stats.rx_crc_errors +
1205 		net->stats.rx_missed_errors;
1206 	stats->tx_errors = net->stats.tx_errors;
1207 	stats->rx_length_errors = net->stats.rx_length_errors;
1208 	stats->rx_over_errors = net->stats.rx_over_errors;
1209 	stats->rx_crc_errors = net->stats.rx_crc_errors;
1210 	stats->rx_missed_errors = net->stats.rx_missed_errors;
1211 }
1212 
1213 static const struct net_device_ops tbnet_netdev_ops = {
1214 	.ndo_open = tbnet_open,
1215 	.ndo_stop = tbnet_stop,
1216 	.ndo_start_xmit = tbnet_start_xmit,
1217 	.ndo_get_stats64 = tbnet_get_stats64,
1218 };
1219 
tbnet_generate_mac(struct net_device * dev)1220 static void tbnet_generate_mac(struct net_device *dev)
1221 {
1222 	const struct tbnet *net = netdev_priv(dev);
1223 	const struct tb_xdomain *xd = net->xd;
1224 	u8 addr[ETH_ALEN];
1225 	u8 phy_port;
1226 	u32 hash;
1227 
1228 	phy_port = tb_phy_port_from_link(TBNET_L0_PORT_NUM(xd->route));
1229 
1230 	/* Unicast and locally administered MAC */
1231 	addr[0] = phy_port << 4 | 0x02;
1232 	hash = jhash2((u32 *)xd->local_uuid, 4, 0);
1233 	memcpy(addr + 1, &hash, sizeof(hash));
1234 	hash = jhash2((u32 *)xd->local_uuid, 4, hash);
1235 	addr[5] = hash & 0xff;
1236 	eth_hw_addr_set(dev, addr);
1237 }
1238 
tbnet_probe(struct tb_service * svc,const struct tb_service_id * id)1239 static int tbnet_probe(struct tb_service *svc, const struct tb_service_id *id)
1240 {
1241 	struct tb_xdomain *xd = tb_service_parent(svc);
1242 	struct net_device *dev;
1243 	struct tbnet *net;
1244 	int ret;
1245 
1246 	dev = alloc_etherdev(sizeof(*net));
1247 	if (!dev)
1248 		return -ENOMEM;
1249 
1250 	SET_NETDEV_DEV(dev, &svc->dev);
1251 
1252 	net = netdev_priv(dev);
1253 	INIT_DELAYED_WORK(&net->login_work, tbnet_login_work);
1254 	INIT_WORK(&net->connected_work, tbnet_connected_work);
1255 	INIT_WORK(&net->disconnect_work, tbnet_disconnect_work);
1256 	mutex_init(&net->connection_lock);
1257 	atomic_set(&net->command_id, 0);
1258 	atomic_set(&net->frame_id, 0);
1259 	net->svc = svc;
1260 	net->dev = dev;
1261 	net->xd = xd;
1262 
1263 	tbnet_generate_mac(dev);
1264 
1265 	strcpy(dev->name, "thunderbolt%d");
1266 	dev->netdev_ops = &tbnet_netdev_ops;
1267 
1268 	/* ThunderboltIP takes advantage of TSO packets but instead of
1269 	 * segmenting them we just split the packet into Thunderbolt
1270 	 * frames (maximum payload size of each frame is 4084 bytes) and
1271 	 * calculate checksum over the whole packet here.
1272 	 *
1273 	 * The receiving side does the opposite if the host OS supports
1274 	 * LRO, otherwise it needs to split the large packet into MTU
1275 	 * sized smaller packets.
1276 	 *
1277 	 * In order to receive large packets from the networking stack,
1278 	 * we need to announce support for most of the offloading
1279 	 * features here.
1280 	 */
1281 	dev->hw_features = NETIF_F_SG | NETIF_F_ALL_TSO | NETIF_F_GRO |
1282 			   NETIF_F_IP_CSUM | NETIF_F_IPV6_CSUM;
1283 	dev->features = dev->hw_features | NETIF_F_HIGHDMA;
1284 	dev->hard_header_len += sizeof(struct thunderbolt_ip_frame_header);
1285 
1286 	netif_napi_add(dev, &net->napi, tbnet_poll);
1287 
1288 	/* MTU range: 68 - 65522 */
1289 	dev->min_mtu = ETH_MIN_MTU;
1290 	dev->max_mtu = TBNET_MAX_MTU - ETH_HLEN;
1291 
1292 	net->handler.uuid = &tbnet_svc_uuid;
1293 	net->handler.callback = tbnet_handle_packet;
1294 	net->handler.data = net;
1295 	tb_register_protocol_handler(&net->handler);
1296 
1297 	tb_service_set_drvdata(svc, net);
1298 
1299 	ret = register_netdev(dev);
1300 	if (ret) {
1301 		tb_unregister_protocol_handler(&net->handler);
1302 		free_netdev(dev);
1303 		return ret;
1304 	}
1305 
1306 	return 0;
1307 }
1308 
tbnet_remove(struct tb_service * svc)1309 static void tbnet_remove(struct tb_service *svc)
1310 {
1311 	struct tbnet *net = tb_service_get_drvdata(svc);
1312 
1313 	unregister_netdev(net->dev);
1314 	tb_unregister_protocol_handler(&net->handler);
1315 	free_netdev(net->dev);
1316 }
1317 
tbnet_shutdown(struct tb_service * svc)1318 static void tbnet_shutdown(struct tb_service *svc)
1319 {
1320 	tbnet_tear_down(tb_service_get_drvdata(svc), true);
1321 }
1322 
tbnet_suspend(struct device * dev)1323 static int __maybe_unused tbnet_suspend(struct device *dev)
1324 {
1325 	struct tb_service *svc = tb_to_service(dev);
1326 	struct tbnet *net = tb_service_get_drvdata(svc);
1327 
1328 	stop_login(net);
1329 	if (netif_running(net->dev)) {
1330 		netif_device_detach(net->dev);
1331 		tbnet_tear_down(net, true);
1332 	}
1333 
1334 	tb_unregister_protocol_handler(&net->handler);
1335 	return 0;
1336 }
1337 
tbnet_resume(struct device * dev)1338 static int __maybe_unused tbnet_resume(struct device *dev)
1339 {
1340 	struct tb_service *svc = tb_to_service(dev);
1341 	struct tbnet *net = tb_service_get_drvdata(svc);
1342 
1343 	tb_register_protocol_handler(&net->handler);
1344 
1345 	netif_carrier_off(net->dev);
1346 	if (netif_running(net->dev)) {
1347 		netif_device_attach(net->dev);
1348 		start_login(net);
1349 	}
1350 
1351 	return 0;
1352 }
1353 
1354 static const struct dev_pm_ops tbnet_pm_ops = {
1355 	SET_SYSTEM_SLEEP_PM_OPS(tbnet_suspend, tbnet_resume)
1356 };
1357 
1358 static const struct tb_service_id tbnet_ids[] = {
1359 	{ TB_SERVICE("network", 1) },
1360 	{ },
1361 };
1362 MODULE_DEVICE_TABLE(tbsvc, tbnet_ids);
1363 
1364 static struct tb_service_driver tbnet_driver = {
1365 	.driver = {
1366 		.owner = THIS_MODULE,
1367 		.name = "thunderbolt-net",
1368 		.pm = &tbnet_pm_ops,
1369 	},
1370 	.probe = tbnet_probe,
1371 	.remove = tbnet_remove,
1372 	.shutdown = tbnet_shutdown,
1373 	.id_table = tbnet_ids,
1374 };
1375 
tbnet_init(void)1376 static int __init tbnet_init(void)
1377 {
1378 	unsigned int flags;
1379 	int ret;
1380 
1381 	tbnet_dir = tb_property_create_dir(&tbnet_dir_uuid);
1382 	if (!tbnet_dir)
1383 		return -ENOMEM;
1384 
1385 	tb_property_add_immediate(tbnet_dir, "prtcid", 1);
1386 	tb_property_add_immediate(tbnet_dir, "prtcvers", 1);
1387 	tb_property_add_immediate(tbnet_dir, "prtcrevs", 1);
1388 
1389 	flags = TBNET_MATCH_FRAGS_ID | TBNET_64K_FRAMES;
1390 	if (tbnet_e2e)
1391 		flags |= TBNET_E2E;
1392 	tb_property_add_immediate(tbnet_dir, "prtcstns", flags);
1393 
1394 	ret = tb_register_property_dir("network", tbnet_dir);
1395 	if (ret)
1396 		goto err_free_dir;
1397 
1398 	ret = tb_register_service_driver(&tbnet_driver);
1399 	if (ret)
1400 		goto err_unregister;
1401 
1402 	return 0;
1403 
1404 err_unregister:
1405 	tb_unregister_property_dir("network", tbnet_dir);
1406 err_free_dir:
1407 	tb_property_free_dir(tbnet_dir);
1408 
1409 	return ret;
1410 }
1411 module_init(tbnet_init);
1412 
tbnet_exit(void)1413 static void __exit tbnet_exit(void)
1414 {
1415 	tb_unregister_service_driver(&tbnet_driver);
1416 	tb_unregister_property_dir("network", tbnet_dir);
1417 	tb_property_free_dir(tbnet_dir);
1418 }
1419 module_exit(tbnet_exit);
1420 
1421 MODULE_AUTHOR("Amir Levy <amir.jer.levy@intel.com>");
1422 MODULE_AUTHOR("Michael Jamet <michael.jamet@intel.com>");
1423 MODULE_AUTHOR("Mika Westerberg <mika.westerberg@linux.intel.com>");
1424 MODULE_DESCRIPTION("Thunderbolt/USB4 network driver");
1425 MODULE_LICENSE("GPL v2");
1426