1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Loopback bridge driver for the Greybus loopback module.
4  *
5  * Copyright 2014 Google Inc.
6  * Copyright 2014 Linaro Ltd.
7  */
8 
9 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
10 
11 #include <linux/kernel.h>
12 #include <linux/module.h>
13 #include <linux/mutex.h>
14 #include <linux/slab.h>
15 #include <linux/kthread.h>
16 #include <linux/delay.h>
17 #include <linux/random.h>
18 #include <linux/sizes.h>
19 #include <linux/cdev.h>
20 #include <linux/fs.h>
21 #include <linux/kfifo.h>
22 #include <linux/debugfs.h>
23 #include <linux/list_sort.h>
24 #include <linux/spinlock.h>
25 #include <linux/workqueue.h>
26 #include <linux/atomic.h>
27 #include <linux/pm_runtime.h>
28 
29 #include <asm/div64.h>
30 
31 #include "greybus.h"
32 #include "connection.h"
33 
34 #define NSEC_PER_DAY 86400000000000ULL
35 
36 struct gb_loopback_stats {
37 	u32 min;
38 	u32 max;
39 	u64 sum;
40 	u32 count;
41 };
42 
43 struct gb_loopback_device {
44 	struct dentry *root;
45 	u32 count;
46 	size_t size_max;
47 
48 	/* We need to take a lock in atomic context */
49 	spinlock_t lock;
50 	struct list_head list;
51 	struct list_head list_op_async;
52 	wait_queue_head_t wq;
53 };
54 
55 static struct gb_loopback_device gb_dev;
56 
57 struct gb_loopback_async_operation {
58 	struct gb_loopback *gb;
59 	struct gb_operation *operation;
60 	ktime_t ts;
61 	int (*completion)(struct gb_loopback_async_operation *op_async);
62 };
63 
64 struct gb_loopback {
65 	struct gb_connection *connection;
66 
67 	struct dentry *file;
68 	struct kfifo kfifo_lat;
69 	struct mutex mutex;
70 	struct task_struct *task;
71 	struct list_head entry;
72 	struct device *dev;
73 	wait_queue_head_t wq;
74 	wait_queue_head_t wq_completion;
75 	atomic_t outstanding_operations;
76 
77 	/* Per connection stats */
78 	ktime_t ts;
79 	struct gb_loopback_stats latency;
80 	struct gb_loopback_stats throughput;
81 	struct gb_loopback_stats requests_per_second;
82 	struct gb_loopback_stats apbridge_unipro_latency;
83 	struct gb_loopback_stats gbphy_firmware_latency;
84 
85 	int type;
86 	int async;
87 	int id;
88 	u32 size;
89 	u32 iteration_max;
90 	u32 iteration_count;
91 	int us_wait;
92 	u32 error;
93 	u32 requests_completed;
94 	u32 requests_timedout;
95 	u32 timeout;
96 	u32 jiffy_timeout;
97 	u32 timeout_min;
98 	u32 timeout_max;
99 	u32 outstanding_operations_max;
100 	u32 lbid;
101 	u64 elapsed_nsecs;
102 	u32 apbridge_latency_ts;
103 	u32 gbphy_latency_ts;
104 
105 	u32 send_count;
106 };
107 
108 static struct class loopback_class = {
109 	.name		= "gb_loopback",
110 	.owner		= THIS_MODULE,
111 };
112 static DEFINE_IDA(loopback_ida);
113 
114 /* Min/max values in jiffies */
115 #define GB_LOOPBACK_TIMEOUT_MIN				1
116 #define GB_LOOPBACK_TIMEOUT_MAX				10000
117 
118 #define GB_LOOPBACK_FIFO_DEFAULT			8192
119 
120 static unsigned int kfifo_depth = GB_LOOPBACK_FIFO_DEFAULT;
121 module_param(kfifo_depth, uint, 0444);
122 
123 /* Maximum size of any one send data buffer we support */
124 #define MAX_PACKET_SIZE (PAGE_SIZE * 2)
125 
126 #define GB_LOOPBACK_US_WAIT_MAX				1000000
127 
128 /* interface sysfs attributes */
129 #define gb_loopback_ro_attr(field)				\
130 static ssize_t field##_show(struct device *dev,			\
131 			    struct device_attribute *attr,		\
132 			    char *buf)					\
133 {									\
134 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
135 	return sprintf(buf, "%u\n", gb->field);			\
136 }									\
137 static DEVICE_ATTR_RO(field)
138 
139 #define gb_loopback_ro_stats_attr(name, field, type)		\
140 static ssize_t name##_##field##_show(struct device *dev,	\
141 			    struct device_attribute *attr,		\
142 			    char *buf)					\
143 {									\
144 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
145 	/* Report 0 for min and max if no transfer successed */		\
146 	if (!gb->requests_completed)					\
147 		return sprintf(buf, "0\n");				\
148 	return sprintf(buf, "%"#type"\n", gb->name.field);	\
149 }									\
150 static DEVICE_ATTR_RO(name##_##field)
151 
152 #define gb_loopback_ro_avg_attr(name)			\
153 static ssize_t name##_avg_show(struct device *dev,		\
154 			    struct device_attribute *attr,		\
155 			    char *buf)					\
156 {									\
157 	struct gb_loopback_stats *stats;				\
158 	struct gb_loopback *gb;						\
159 	u64 avg, rem;							\
160 	u32 count;							\
161 	gb = dev_get_drvdata(dev);			\
162 	stats = &gb->name;					\
163 	count = stats->count ? stats->count : 1;			\
164 	avg = stats->sum + count / 2000000; /* round closest */		\
165 	rem = do_div(avg, count);					\
166 	rem *= 1000000;							\
167 	do_div(rem, count);						\
168 	return sprintf(buf, "%llu.%06u\n", avg, (u32)rem);		\
169 }									\
170 static DEVICE_ATTR_RO(name##_avg)
171 
172 #define gb_loopback_stats_attrs(field)				\
173 	gb_loopback_ro_stats_attr(field, min, u);		\
174 	gb_loopback_ro_stats_attr(field, max, u);		\
175 	gb_loopback_ro_avg_attr(field)
176 
177 #define gb_loopback_attr(field, type)					\
178 static ssize_t field##_show(struct device *dev,				\
179 			    struct device_attribute *attr,		\
180 			    char *buf)					\
181 {									\
182 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
183 	return sprintf(buf, "%"#type"\n", gb->field);			\
184 }									\
185 static ssize_t field##_store(struct device *dev,			\
186 			    struct device_attribute *attr,		\
187 			    const char *buf,				\
188 			    size_t len)					\
189 {									\
190 	int ret;							\
191 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
192 	mutex_lock(&gb->mutex);						\
193 	ret = sscanf(buf, "%"#type, &gb->field);			\
194 	if (ret != 1)							\
195 		len = -EINVAL;						\
196 	else								\
197 		gb_loopback_check_attr(gb, bundle);			\
198 	mutex_unlock(&gb->mutex);					\
199 	return len;							\
200 }									\
201 static DEVICE_ATTR_RW(field)
202 
203 #define gb_dev_loopback_ro_attr(field, conn)				\
204 static ssize_t field##_show(struct device *dev,		\
205 			    struct device_attribute *attr,		\
206 			    char *buf)					\
207 {									\
208 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
209 	return sprintf(buf, "%u\n", gb->field);				\
210 }									\
211 static DEVICE_ATTR_RO(field)
212 
213 #define gb_dev_loopback_rw_attr(field, type)				\
214 static ssize_t field##_show(struct device *dev,				\
215 			    struct device_attribute *attr,		\
216 			    char *buf)					\
217 {									\
218 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
219 	return sprintf(buf, "%"#type"\n", gb->field);			\
220 }									\
221 static ssize_t field##_store(struct device *dev,			\
222 			    struct device_attribute *attr,		\
223 			    const char *buf,				\
224 			    size_t len)					\
225 {									\
226 	int ret;							\
227 	struct gb_loopback *gb = dev_get_drvdata(dev);			\
228 	mutex_lock(&gb->mutex);						\
229 	ret = sscanf(buf, "%"#type, &gb->field);			\
230 	if (ret != 1)							\
231 		len = -EINVAL;						\
232 	else								\
233 		gb_loopback_check_attr(gb);		\
234 	mutex_unlock(&gb->mutex);					\
235 	return len;							\
236 }									\
237 static DEVICE_ATTR_RW(field)
238 
239 static void gb_loopback_reset_stats(struct gb_loopback *gb);
240 static void gb_loopback_check_attr(struct gb_loopback *gb)
241 {
242 	if (gb->us_wait > GB_LOOPBACK_US_WAIT_MAX)
243 		gb->us_wait = GB_LOOPBACK_US_WAIT_MAX;
244 	if (gb->size > gb_dev.size_max)
245 		gb->size = gb_dev.size_max;
246 	gb->requests_timedout = 0;
247 	gb->requests_completed = 0;
248 	gb->iteration_count = 0;
249 	gb->send_count = 0;
250 	gb->error = 0;
251 
252 	if (kfifo_depth < gb->iteration_max) {
253 		dev_warn(gb->dev,
254 			 "cannot log bytes %u kfifo_depth %u\n",
255 			 gb->iteration_max, kfifo_depth);
256 	}
257 	kfifo_reset_out(&gb->kfifo_lat);
258 
259 	switch (gb->type) {
260 	case GB_LOOPBACK_TYPE_PING:
261 	case GB_LOOPBACK_TYPE_TRANSFER:
262 	case GB_LOOPBACK_TYPE_SINK:
263 		gb->jiffy_timeout = usecs_to_jiffies(gb->timeout);
264 		if (!gb->jiffy_timeout)
265 			gb->jiffy_timeout = GB_LOOPBACK_TIMEOUT_MIN;
266 		else if (gb->jiffy_timeout > GB_LOOPBACK_TIMEOUT_MAX)
267 			gb->jiffy_timeout = GB_LOOPBACK_TIMEOUT_MAX;
268 		gb_loopback_reset_stats(gb);
269 		wake_up(&gb->wq);
270 		break;
271 	default:
272 		gb->type = 0;
273 		break;
274 	}
275 }
276 
277 /* Time to send and receive one message */
278 gb_loopback_stats_attrs(latency);
279 /* Number of requests sent per second on this cport */
280 gb_loopback_stats_attrs(requests_per_second);
281 /* Quantity of data sent and received on this cport */
282 gb_loopback_stats_attrs(throughput);
283 /* Latency across the UniPro link from APBridge's perspective */
284 gb_loopback_stats_attrs(apbridge_unipro_latency);
285 /* Firmware induced overhead in the GPBridge */
286 gb_loopback_stats_attrs(gbphy_firmware_latency);
287 
288 /* Number of errors encountered during loop */
289 gb_loopback_ro_attr(error);
290 /* Number of requests successfully completed async */
291 gb_loopback_ro_attr(requests_completed);
292 /* Number of requests timed out async */
293 gb_loopback_ro_attr(requests_timedout);
294 /* Timeout minimum in useconds */
295 gb_loopback_ro_attr(timeout_min);
296 /* Timeout minimum in useconds */
297 gb_loopback_ro_attr(timeout_max);
298 
299 /*
300  * Type of loopback message to send based on protocol type definitions
301  * 0 => Don't send message
302  * 2 => Send ping message continuously (message without payload)
303  * 3 => Send transfer message continuously (message with payload,
304  *					   payload returned in response)
305  * 4 => Send a sink message (message with payload, no payload in response)
306  */
307 gb_dev_loopback_rw_attr(type, d);
308 /* Size of transfer message payload: 0-4096 bytes */
309 gb_dev_loopback_rw_attr(size, u);
310 /* Time to wait between two messages: 0-1000 ms */
311 gb_dev_loopback_rw_attr(us_wait, d);
312 /* Maximum iterations for a given operation: 1-(2^32-1), 0 implies infinite */
313 gb_dev_loopback_rw_attr(iteration_max, u);
314 /* The current index of the for (i = 0; i < iteration_max; i++) loop */
315 gb_dev_loopback_ro_attr(iteration_count, false);
316 /* A flag to indicate synchronous or asynchronous operations */
317 gb_dev_loopback_rw_attr(async, u);
318 /* Timeout of an individual asynchronous request */
319 gb_dev_loopback_rw_attr(timeout, u);
320 /* Maximum number of in-flight operations before back-off */
321 gb_dev_loopback_rw_attr(outstanding_operations_max, u);
322 
323 static struct attribute *loopback_attrs[] = {
324 	&dev_attr_latency_min.attr,
325 	&dev_attr_latency_max.attr,
326 	&dev_attr_latency_avg.attr,
327 	&dev_attr_requests_per_second_min.attr,
328 	&dev_attr_requests_per_second_max.attr,
329 	&dev_attr_requests_per_second_avg.attr,
330 	&dev_attr_throughput_min.attr,
331 	&dev_attr_throughput_max.attr,
332 	&dev_attr_throughput_avg.attr,
333 	&dev_attr_apbridge_unipro_latency_min.attr,
334 	&dev_attr_apbridge_unipro_latency_max.attr,
335 	&dev_attr_apbridge_unipro_latency_avg.attr,
336 	&dev_attr_gbphy_firmware_latency_min.attr,
337 	&dev_attr_gbphy_firmware_latency_max.attr,
338 	&dev_attr_gbphy_firmware_latency_avg.attr,
339 	&dev_attr_type.attr,
340 	&dev_attr_size.attr,
341 	&dev_attr_us_wait.attr,
342 	&dev_attr_iteration_count.attr,
343 	&dev_attr_iteration_max.attr,
344 	&dev_attr_async.attr,
345 	&dev_attr_error.attr,
346 	&dev_attr_requests_completed.attr,
347 	&dev_attr_requests_timedout.attr,
348 	&dev_attr_timeout.attr,
349 	&dev_attr_outstanding_operations_max.attr,
350 	&dev_attr_timeout_min.attr,
351 	&dev_attr_timeout_max.attr,
352 	NULL,
353 };
354 ATTRIBUTE_GROUPS(loopback);
355 
356 static void gb_loopback_calculate_stats(struct gb_loopback *gb, bool error);
357 
358 static u32 gb_loopback_nsec_to_usec_latency(u64 elapsed_nsecs)
359 {
360 	do_div(elapsed_nsecs, NSEC_PER_USEC);
361 	return elapsed_nsecs;
362 }
363 
364 static u64 __gb_loopback_calc_latency(u64 t1, u64 t2)
365 {
366 	if (t2 > t1)
367 		return t2 - t1;
368 	else
369 		return NSEC_PER_DAY - t2 + t1;
370 }
371 
372 static u64 gb_loopback_calc_latency(ktime_t ts, ktime_t te)
373 {
374 	return __gb_loopback_calc_latency(ktime_to_ns(ts), ktime_to_ns(te));
375 }
376 
377 static int gb_loopback_operation_sync(struct gb_loopback *gb, int type,
378 				      void *request, int request_size,
379 				      void *response, int response_size)
380 {
381 	struct gb_operation *operation;
382 	ktime_t ts, te;
383 	int ret;
384 
385 	ts = ktime_get();
386 	operation = gb_operation_create(gb->connection, type, request_size,
387 					response_size, GFP_KERNEL);
388 	if (!operation)
389 		return -ENOMEM;
390 
391 	if (request_size)
392 		memcpy(operation->request->payload, request, request_size);
393 
394 	ret = gb_operation_request_send_sync(operation);
395 	if (ret) {
396 		dev_err(&gb->connection->bundle->dev,
397 			"synchronous operation failed: %d\n", ret);
398 		goto out_put_operation;
399 	} else {
400 		if (response_size == operation->response->payload_size) {
401 			memcpy(response, operation->response->payload,
402 			       response_size);
403 		} else {
404 			dev_err(&gb->connection->bundle->dev,
405 				"response size %zu expected %d\n",
406 				operation->response->payload_size,
407 				response_size);
408 			ret = -EINVAL;
409 			goto out_put_operation;
410 		}
411 	}
412 
413 	te = ktime_get();
414 
415 	/* Calculate the total time the message took */
416 	gb->elapsed_nsecs = gb_loopback_calc_latency(ts, te);
417 
418 out_put_operation:
419 	gb_operation_put(operation);
420 
421 	return ret;
422 }
423 
424 static void gb_loopback_async_wait_all(struct gb_loopback *gb)
425 {
426 	wait_event(gb->wq_completion,
427 		   !atomic_read(&gb->outstanding_operations));
428 }
429 
430 static void gb_loopback_async_operation_callback(struct gb_operation *operation)
431 {
432 	struct gb_loopback_async_operation *op_async;
433 	struct gb_loopback *gb;
434 	ktime_t te;
435 	int result;
436 
437 	te = ktime_get();
438 	result = gb_operation_result(operation);
439 	op_async = gb_operation_get_data(operation);
440 	gb = op_async->gb;
441 
442 	mutex_lock(&gb->mutex);
443 
444 	if (!result && op_async->completion)
445 		result = op_async->completion(op_async);
446 
447 	if (!result) {
448 		gb->elapsed_nsecs = gb_loopback_calc_latency(op_async->ts, te);
449 	} else {
450 		gb->error++;
451 		if (result == -ETIMEDOUT)
452 			gb->requests_timedout++;
453 	}
454 
455 	gb->iteration_count++;
456 	gb_loopback_calculate_stats(gb, result);
457 
458 	mutex_unlock(&gb->mutex);
459 
460 	dev_dbg(&gb->connection->bundle->dev, "complete operation %d\n",
461 		operation->id);
462 
463 	/* Wake up waiters */
464 	atomic_dec(&op_async->gb->outstanding_operations);
465 	wake_up(&gb->wq_completion);
466 
467 	/* Release resources */
468 	gb_operation_put(operation);
469 	kfree(op_async);
470 }
471 
472 static int gb_loopback_async_operation(struct gb_loopback *gb, int type,
473 				       void *request, int request_size,
474 				       int response_size,
475 				       void *completion)
476 {
477 	struct gb_loopback_async_operation *op_async;
478 	struct gb_operation *operation;
479 	int ret;
480 
481 	op_async = kzalloc(sizeof(*op_async), GFP_KERNEL);
482 	if (!op_async)
483 		return -ENOMEM;
484 
485 	operation = gb_operation_create(gb->connection, type, request_size,
486 					response_size, GFP_KERNEL);
487 	if (!operation) {
488 		kfree(op_async);
489 		return -ENOMEM;
490 	}
491 
492 	if (request_size)
493 		memcpy(operation->request->payload, request, request_size);
494 
495 	gb_operation_set_data(operation, op_async);
496 
497 	op_async->gb = gb;
498 	op_async->operation = operation;
499 	op_async->completion = completion;
500 
501 	op_async->ts = ktime_get();
502 
503 	atomic_inc(&gb->outstanding_operations);
504 	ret = gb_operation_request_send(operation,
505 					gb_loopback_async_operation_callback,
506 					jiffies_to_msecs(gb->jiffy_timeout),
507 					GFP_KERNEL);
508 	if (ret) {
509 		atomic_dec(&gb->outstanding_operations);
510 		gb_operation_put(operation);
511 		kfree(op_async);
512 	}
513 	return ret;
514 }
515 
516 static int gb_loopback_sync_sink(struct gb_loopback *gb, u32 len)
517 {
518 	struct gb_loopback_transfer_request *request;
519 	int retval;
520 
521 	request = kmalloc(len + sizeof(*request), GFP_KERNEL);
522 	if (!request)
523 		return -ENOMEM;
524 
525 	request->len = cpu_to_le32(len);
526 	retval = gb_loopback_operation_sync(gb, GB_LOOPBACK_TYPE_SINK,
527 					    request, len + sizeof(*request),
528 					    NULL, 0);
529 	kfree(request);
530 	return retval;
531 }
532 
533 static int gb_loopback_sync_transfer(struct gb_loopback *gb, u32 len)
534 {
535 	struct gb_loopback_transfer_request *request;
536 	struct gb_loopback_transfer_response *response;
537 	int retval;
538 
539 	gb->apbridge_latency_ts = 0;
540 	gb->gbphy_latency_ts = 0;
541 
542 	request = kmalloc(len + sizeof(*request), GFP_KERNEL);
543 	if (!request)
544 		return -ENOMEM;
545 	response = kmalloc(len + sizeof(*response), GFP_KERNEL);
546 	if (!response) {
547 		kfree(request);
548 		return -ENOMEM;
549 	}
550 
551 	memset(request->data, 0x5A, len);
552 
553 	request->len = cpu_to_le32(len);
554 	retval = gb_loopback_operation_sync(gb, GB_LOOPBACK_TYPE_TRANSFER,
555 					    request, len + sizeof(*request),
556 					    response, len + sizeof(*response));
557 	if (retval)
558 		goto gb_error;
559 
560 	if (memcmp(request->data, response->data, len)) {
561 		dev_err(&gb->connection->bundle->dev,
562 			"Loopback Data doesn't match\n");
563 		retval = -EREMOTEIO;
564 	}
565 	gb->apbridge_latency_ts = (u32)__le32_to_cpu(response->reserved0);
566 	gb->gbphy_latency_ts = (u32)__le32_to_cpu(response->reserved1);
567 
568 gb_error:
569 	kfree(request);
570 	kfree(response);
571 
572 	return retval;
573 }
574 
575 static int gb_loopback_sync_ping(struct gb_loopback *gb)
576 {
577 	return gb_loopback_operation_sync(gb, GB_LOOPBACK_TYPE_PING,
578 					  NULL, 0, NULL, 0);
579 }
580 
581 static int gb_loopback_async_sink(struct gb_loopback *gb, u32 len)
582 {
583 	struct gb_loopback_transfer_request *request;
584 	int retval;
585 
586 	request = kmalloc(len + sizeof(*request), GFP_KERNEL);
587 	if (!request)
588 		return -ENOMEM;
589 
590 	request->len = cpu_to_le32(len);
591 	retval = gb_loopback_async_operation(gb, GB_LOOPBACK_TYPE_SINK,
592 					     request, len + sizeof(*request),
593 					     0, NULL);
594 	kfree(request);
595 	return retval;
596 }
597 
598 static int gb_loopback_async_transfer_complete(
599 				struct gb_loopback_async_operation *op_async)
600 {
601 	struct gb_loopback *gb;
602 	struct gb_operation *operation;
603 	struct gb_loopback_transfer_request *request;
604 	struct gb_loopback_transfer_response *response;
605 	size_t len;
606 	int retval = 0;
607 
608 	gb = op_async->gb;
609 	operation = op_async->operation;
610 	request = operation->request->payload;
611 	response = operation->response->payload;
612 	len = le32_to_cpu(request->len);
613 
614 	if (memcmp(request->data, response->data, len)) {
615 		dev_err(&gb->connection->bundle->dev,
616 			"Loopback Data doesn't match operation id %d\n",
617 			operation->id);
618 		retval = -EREMOTEIO;
619 	} else {
620 		gb->apbridge_latency_ts =
621 			(u32)__le32_to_cpu(response->reserved0);
622 		gb->gbphy_latency_ts =
623 			(u32)__le32_to_cpu(response->reserved1);
624 	}
625 
626 	return retval;
627 }
628 
629 static int gb_loopback_async_transfer(struct gb_loopback *gb, u32 len)
630 {
631 	struct gb_loopback_transfer_request *request;
632 	int retval, response_len;
633 
634 	request = kmalloc(len + sizeof(*request), GFP_KERNEL);
635 	if (!request)
636 		return -ENOMEM;
637 
638 	memset(request->data, 0x5A, len);
639 
640 	request->len = cpu_to_le32(len);
641 	response_len = sizeof(struct gb_loopback_transfer_response);
642 	retval = gb_loopback_async_operation(gb, GB_LOOPBACK_TYPE_TRANSFER,
643 					     request, len + sizeof(*request),
644 					     len + response_len,
645 					     gb_loopback_async_transfer_complete);
646 	if (retval)
647 		goto gb_error;
648 
649 gb_error:
650 	kfree(request);
651 	return retval;
652 }
653 
654 static int gb_loopback_async_ping(struct gb_loopback *gb)
655 {
656 	return gb_loopback_async_operation(gb, GB_LOOPBACK_TYPE_PING,
657 					   NULL, 0, 0, NULL);
658 }
659 
660 static int gb_loopback_request_handler(struct gb_operation *operation)
661 {
662 	struct gb_connection *connection = operation->connection;
663 	struct gb_loopback_transfer_request *request;
664 	struct gb_loopback_transfer_response *response;
665 	struct device *dev = &connection->bundle->dev;
666 	size_t len;
667 
668 	/* By convention, the AP initiates the version operation */
669 	switch (operation->type) {
670 	case GB_LOOPBACK_TYPE_PING:
671 	case GB_LOOPBACK_TYPE_SINK:
672 		return 0;
673 	case GB_LOOPBACK_TYPE_TRANSFER:
674 		if (operation->request->payload_size < sizeof(*request)) {
675 			dev_err(dev, "transfer request too small (%zu < %zu)\n",
676 				operation->request->payload_size,
677 				sizeof(*request));
678 			return -EINVAL;	/* -EMSGSIZE */
679 		}
680 		request = operation->request->payload;
681 		len = le32_to_cpu(request->len);
682 		if (len > gb_dev.size_max) {
683 			dev_err(dev, "transfer request too large (%zu > %zu)\n",
684 				len, gb_dev.size_max);
685 			return -EINVAL;
686 		}
687 
688 		if (!gb_operation_response_alloc(operation,
689 				len + sizeof(*response), GFP_KERNEL)) {
690 			dev_err(dev, "error allocating response\n");
691 			return -ENOMEM;
692 		}
693 		response = operation->response->payload;
694 		response->len = cpu_to_le32(len);
695 		if (len)
696 			memcpy(response->data, request->data, len);
697 
698 		return 0;
699 	default:
700 		dev_err(dev, "unsupported request: %u\n", operation->type);
701 		return -EINVAL;
702 	}
703 }
704 
705 static void gb_loopback_reset_stats(struct gb_loopback *gb)
706 {
707 	struct gb_loopback_stats reset = {
708 		.min = U32_MAX,
709 	};
710 
711 	/* Reset per-connection stats */
712 	memcpy(&gb->latency, &reset,
713 	       sizeof(struct gb_loopback_stats));
714 	memcpy(&gb->throughput, &reset,
715 	       sizeof(struct gb_loopback_stats));
716 	memcpy(&gb->requests_per_second, &reset,
717 	       sizeof(struct gb_loopback_stats));
718 	memcpy(&gb->apbridge_unipro_latency, &reset,
719 	       sizeof(struct gb_loopback_stats));
720 	memcpy(&gb->gbphy_firmware_latency, &reset,
721 	       sizeof(struct gb_loopback_stats));
722 
723 	/* Should be initialized at least once per transaction set */
724 	gb->apbridge_latency_ts = 0;
725 	gb->gbphy_latency_ts = 0;
726 	gb->ts = ktime_set(0, 0);
727 }
728 
729 static void gb_loopback_update_stats(struct gb_loopback_stats *stats, u32 val)
730 {
731 	if (stats->min > val)
732 		stats->min = val;
733 	if (stats->max < val)
734 		stats->max = val;
735 	stats->sum += val;
736 	stats->count++;
737 }
738 
739 static void gb_loopback_update_stats_window(struct gb_loopback_stats *stats,
740 					    u64 val, u32 count)
741 {
742 	stats->sum += val;
743 	stats->count += count;
744 
745 	do_div(val, count);
746 	if (stats->min > val)
747 		stats->min = val;
748 	if (stats->max < val)
749 		stats->max = val;
750 }
751 
752 static void gb_loopback_requests_update(struct gb_loopback *gb, u32 latency)
753 {
754 	u64 req = gb->requests_completed * USEC_PER_SEC;
755 
756 	gb_loopback_update_stats_window(&gb->requests_per_second, req, latency);
757 }
758 
759 static void gb_loopback_throughput_update(struct gb_loopback *gb, u32 latency)
760 {
761 	u64 aggregate_size = sizeof(struct gb_operation_msg_hdr) * 2;
762 
763 	switch (gb->type) {
764 	case GB_LOOPBACK_TYPE_PING:
765 		break;
766 	case GB_LOOPBACK_TYPE_SINK:
767 		aggregate_size += sizeof(struct gb_loopback_transfer_request) +
768 				  gb->size;
769 		break;
770 	case GB_LOOPBACK_TYPE_TRANSFER:
771 		aggregate_size += sizeof(struct gb_loopback_transfer_request) +
772 				  sizeof(struct gb_loopback_transfer_response) +
773 				  gb->size * 2;
774 		break;
775 	default:
776 		return;
777 	}
778 
779 	aggregate_size *= gb->requests_completed;
780 	aggregate_size *= USEC_PER_SEC;
781 	gb_loopback_update_stats_window(&gb->throughput, aggregate_size,
782 					latency);
783 }
784 
785 static void gb_loopback_calculate_latency_stats(struct gb_loopback *gb)
786 {
787 	u32 lat;
788 
789 	/* Express latency in terms of microseconds */
790 	lat = gb_loopback_nsec_to_usec_latency(gb->elapsed_nsecs);
791 
792 	/* Log latency stastic */
793 	gb_loopback_update_stats(&gb->latency, lat);
794 
795 	/* Raw latency log on a per thread basis */
796 	kfifo_in(&gb->kfifo_lat, (unsigned char *)&lat, sizeof(lat));
797 
798 	/* Log the firmware supplied latency values */
799 	gb_loopback_update_stats(&gb->apbridge_unipro_latency,
800 				 gb->apbridge_latency_ts);
801 	gb_loopback_update_stats(&gb->gbphy_firmware_latency,
802 				 gb->gbphy_latency_ts);
803 }
804 
805 static void gb_loopback_calculate_stats(struct gb_loopback *gb, bool error)
806 {
807 	u64 nlat;
808 	u32 lat;
809 	ktime_t te;
810 
811 	if (!error) {
812 		gb->requests_completed++;
813 		gb_loopback_calculate_latency_stats(gb);
814 	}
815 
816 	te = ktime_get();
817 	nlat = gb_loopback_calc_latency(gb->ts, te);
818 	if (nlat >= NSEC_PER_SEC || gb->iteration_count == gb->iteration_max) {
819 		lat = gb_loopback_nsec_to_usec_latency(nlat);
820 
821 		gb_loopback_throughput_update(gb, lat);
822 		gb_loopback_requests_update(gb, lat);
823 
824 		if (gb->iteration_count != gb->iteration_max) {
825 			gb->ts = te;
826 			gb->requests_completed = 0;
827 		}
828 	}
829 }
830 
831 static void gb_loopback_async_wait_to_send(struct gb_loopback *gb)
832 {
833 	if (!(gb->async && gb->outstanding_operations_max))
834 		return;
835 	wait_event_interruptible(gb->wq_completion,
836 				 (atomic_read(&gb->outstanding_operations) <
837 				  gb->outstanding_operations_max) ||
838 				  kthread_should_stop());
839 }
840 
841 static int gb_loopback_fn(void *data)
842 {
843 	int error = 0;
844 	int us_wait = 0;
845 	int type;
846 	int ret;
847 	u32 size;
848 
849 	struct gb_loopback *gb = data;
850 	struct gb_bundle *bundle = gb->connection->bundle;
851 
852 	ret = gb_pm_runtime_get_sync(bundle);
853 	if (ret)
854 		return ret;
855 
856 	while (1) {
857 		if (!gb->type) {
858 			gb_pm_runtime_put_autosuspend(bundle);
859 			wait_event_interruptible(gb->wq, gb->type ||
860 						 kthread_should_stop());
861 			ret = gb_pm_runtime_get_sync(bundle);
862 			if (ret)
863 				return ret;
864 		}
865 
866 		if (kthread_should_stop())
867 			break;
868 
869 		/* Limit the maximum number of in-flight async operations */
870 		gb_loopback_async_wait_to_send(gb);
871 		if (kthread_should_stop())
872 			break;
873 
874 		mutex_lock(&gb->mutex);
875 
876 		/* Optionally terminate */
877 		if (gb->send_count == gb->iteration_max) {
878 			mutex_unlock(&gb->mutex);
879 
880 			/* Wait for synchronous and asynchronus completion */
881 			gb_loopback_async_wait_all(gb);
882 
883 			/* Mark complete unless user-space has poked us */
884 			mutex_lock(&gb->mutex);
885 			if (gb->iteration_count == gb->iteration_max) {
886 				gb->type = 0;
887 				gb->send_count = 0;
888 				sysfs_notify(&gb->dev->kobj,  NULL,
889 						"iteration_count");
890 				dev_dbg(&bundle->dev, "load test complete\n");
891 			} else {
892 				dev_dbg(&bundle->dev,
893 					"continuing on with new test set\n");
894 			}
895 			mutex_unlock(&gb->mutex);
896 			continue;
897 		}
898 		size = gb->size;
899 		us_wait = gb->us_wait;
900 		type = gb->type;
901 		if (ktime_to_ns(gb->ts) == 0)
902 			gb->ts = ktime_get();
903 
904 		/* Else operations to perform */
905 		if (gb->async) {
906 			if (type == GB_LOOPBACK_TYPE_PING)
907 				error = gb_loopback_async_ping(gb);
908 			else if (type == GB_LOOPBACK_TYPE_TRANSFER)
909 				error = gb_loopback_async_transfer(gb, size);
910 			else if (type == GB_LOOPBACK_TYPE_SINK)
911 				error = gb_loopback_async_sink(gb, size);
912 
913 			if (error) {
914 				gb->error++;
915 				gb->iteration_count++;
916 			}
917 		} else {
918 			/* We are effectively single threaded here */
919 			if (type == GB_LOOPBACK_TYPE_PING)
920 				error = gb_loopback_sync_ping(gb);
921 			else if (type == GB_LOOPBACK_TYPE_TRANSFER)
922 				error = gb_loopback_sync_transfer(gb, size);
923 			else if (type == GB_LOOPBACK_TYPE_SINK)
924 				error = gb_loopback_sync_sink(gb, size);
925 
926 			if (error)
927 				gb->error++;
928 			gb->iteration_count++;
929 			gb_loopback_calculate_stats(gb, !!error);
930 		}
931 		gb->send_count++;
932 		mutex_unlock(&gb->mutex);
933 
934 		if (us_wait) {
935 			if (us_wait < 20000)
936 				usleep_range(us_wait, us_wait + 100);
937 			else
938 				msleep(us_wait / 1000);
939 		}
940 	}
941 
942 	gb_pm_runtime_put_autosuspend(bundle);
943 
944 	return 0;
945 }
946 
947 static int gb_loopback_dbgfs_latency_show_common(struct seq_file *s,
948 						 struct kfifo *kfifo,
949 						 struct mutex *mutex)
950 {
951 	u32 latency;
952 	int retval;
953 
954 	if (kfifo_len(kfifo) == 0) {
955 		retval = -EAGAIN;
956 		goto done;
957 	}
958 
959 	mutex_lock(mutex);
960 	retval = kfifo_out(kfifo, &latency, sizeof(latency));
961 	if (retval > 0) {
962 		seq_printf(s, "%u", latency);
963 		retval = 0;
964 	}
965 	mutex_unlock(mutex);
966 done:
967 	return retval;
968 }
969 
970 static int gb_loopback_dbgfs_latency_show(struct seq_file *s, void *unused)
971 {
972 	struct gb_loopback *gb = s->private;
973 
974 	return gb_loopback_dbgfs_latency_show_common(s, &gb->kfifo_lat,
975 						     &gb->mutex);
976 }
977 
978 static int gb_loopback_latency_open(struct inode *inode, struct file *file)
979 {
980 	return single_open(file, gb_loopback_dbgfs_latency_show,
981 			   inode->i_private);
982 }
983 
984 static const struct file_operations gb_loopback_debugfs_latency_ops = {
985 	.open		= gb_loopback_latency_open,
986 	.read		= seq_read,
987 	.llseek		= seq_lseek,
988 	.release	= single_release,
989 };
990 
991 static int gb_loopback_bus_id_compare(void *priv, struct list_head *lha,
992 				      struct list_head *lhb)
993 {
994 	struct gb_loopback *a = list_entry(lha, struct gb_loopback, entry);
995 	struct gb_loopback *b = list_entry(lhb, struct gb_loopback, entry);
996 	struct gb_connection *ca = a->connection;
997 	struct gb_connection *cb = b->connection;
998 
999 	if (ca->bundle->intf->interface_id < cb->bundle->intf->interface_id)
1000 		return -1;
1001 	if (cb->bundle->intf->interface_id < ca->bundle->intf->interface_id)
1002 		return 1;
1003 	if (ca->bundle->id < cb->bundle->id)
1004 		return -1;
1005 	if (cb->bundle->id < ca->bundle->id)
1006 		return 1;
1007 	if (ca->intf_cport_id < cb->intf_cport_id)
1008 		return -1;
1009 	else if (cb->intf_cport_id < ca->intf_cport_id)
1010 		return 1;
1011 
1012 	return 0;
1013 }
1014 
1015 static void gb_loopback_insert_id(struct gb_loopback *gb)
1016 {
1017 	struct gb_loopback *gb_list;
1018 	u32 new_lbid = 0;
1019 
1020 	/* perform an insertion sort */
1021 	list_add_tail(&gb->entry, &gb_dev.list);
1022 	list_sort(NULL, &gb_dev.list, gb_loopback_bus_id_compare);
1023 	list_for_each_entry(gb_list, &gb_dev.list, entry) {
1024 		gb_list->lbid = 1 << new_lbid;
1025 		new_lbid++;
1026 	}
1027 }
1028 
1029 #define DEBUGFS_NAMELEN 32
1030 
1031 static int gb_loopback_probe(struct gb_bundle *bundle,
1032 			     const struct greybus_bundle_id *id)
1033 {
1034 	struct greybus_descriptor_cport *cport_desc;
1035 	struct gb_connection *connection;
1036 	struct gb_loopback *gb;
1037 	struct device *dev;
1038 	int retval;
1039 	char name[DEBUGFS_NAMELEN];
1040 	unsigned long flags;
1041 
1042 	if (bundle->num_cports != 1)
1043 		return -ENODEV;
1044 
1045 	cport_desc = &bundle->cport_desc[0];
1046 	if (cport_desc->protocol_id != GREYBUS_PROTOCOL_LOOPBACK)
1047 		return -ENODEV;
1048 
1049 	gb = kzalloc(sizeof(*gb), GFP_KERNEL);
1050 	if (!gb)
1051 		return -ENOMEM;
1052 
1053 	connection = gb_connection_create(bundle, le16_to_cpu(cport_desc->id),
1054 					  gb_loopback_request_handler);
1055 	if (IS_ERR(connection)) {
1056 		retval = PTR_ERR(connection);
1057 		goto out_kzalloc;
1058 	}
1059 
1060 	gb->connection = connection;
1061 	greybus_set_drvdata(bundle, gb);
1062 
1063 	init_waitqueue_head(&gb->wq);
1064 	init_waitqueue_head(&gb->wq_completion);
1065 	atomic_set(&gb->outstanding_operations, 0);
1066 	gb_loopback_reset_stats(gb);
1067 
1068 	/* Reported values to user-space for min/max timeouts */
1069 	gb->timeout_min = jiffies_to_usecs(GB_LOOPBACK_TIMEOUT_MIN);
1070 	gb->timeout_max = jiffies_to_usecs(GB_LOOPBACK_TIMEOUT_MAX);
1071 
1072 	if (!gb_dev.count) {
1073 		/* Calculate maximum payload */
1074 		gb_dev.size_max = gb_operation_get_payload_size_max(connection);
1075 		if (gb_dev.size_max <=
1076 			sizeof(struct gb_loopback_transfer_request)) {
1077 			retval = -EINVAL;
1078 			goto out_connection_destroy;
1079 		}
1080 		gb_dev.size_max -= sizeof(struct gb_loopback_transfer_request);
1081 	}
1082 
1083 	/* Create per-connection sysfs and debugfs data-points */
1084 	snprintf(name, sizeof(name), "raw_latency_%s",
1085 		 dev_name(&connection->bundle->dev));
1086 	gb->file = debugfs_create_file(name, S_IFREG | 0444, gb_dev.root, gb,
1087 				       &gb_loopback_debugfs_latency_ops);
1088 
1089 	gb->id = ida_simple_get(&loopback_ida, 0, 0, GFP_KERNEL);
1090 	if (gb->id < 0) {
1091 		retval = gb->id;
1092 		goto out_debugfs_remove;
1093 	}
1094 
1095 	retval = gb_connection_enable(connection);
1096 	if (retval)
1097 		goto out_ida_remove;
1098 
1099 	dev = device_create_with_groups(&loopback_class,
1100 					&connection->bundle->dev,
1101 					MKDEV(0, 0), gb, loopback_groups,
1102 					"gb_loopback%d", gb->id);
1103 	if (IS_ERR(dev)) {
1104 		retval = PTR_ERR(dev);
1105 		goto out_connection_disable;
1106 	}
1107 	gb->dev = dev;
1108 
1109 	/* Allocate kfifo */
1110 	if (kfifo_alloc(&gb->kfifo_lat, kfifo_depth * sizeof(u32),
1111 			  GFP_KERNEL)) {
1112 		retval = -ENOMEM;
1113 		goto out_conn;
1114 	}
1115 	/* Fork worker thread */
1116 	mutex_init(&gb->mutex);
1117 	gb->task = kthread_run(gb_loopback_fn, gb, "gb_loopback");
1118 	if (IS_ERR(gb->task)) {
1119 		retval = PTR_ERR(gb->task);
1120 		goto out_kfifo;
1121 	}
1122 
1123 	spin_lock_irqsave(&gb_dev.lock, flags);
1124 	gb_loopback_insert_id(gb);
1125 	gb_dev.count++;
1126 	spin_unlock_irqrestore(&gb_dev.lock, flags);
1127 
1128 	gb_connection_latency_tag_enable(connection);
1129 
1130 	gb_pm_runtime_put_autosuspend(bundle);
1131 
1132 	return 0;
1133 
1134 out_kfifo:
1135 	kfifo_free(&gb->kfifo_lat);
1136 out_conn:
1137 	device_unregister(dev);
1138 out_connection_disable:
1139 	gb_connection_disable(connection);
1140 out_ida_remove:
1141 	ida_simple_remove(&loopback_ida, gb->id);
1142 out_debugfs_remove:
1143 	debugfs_remove(gb->file);
1144 out_connection_destroy:
1145 	gb_connection_destroy(connection);
1146 out_kzalloc:
1147 	kfree(gb);
1148 
1149 	return retval;
1150 }
1151 
1152 static void gb_loopback_disconnect(struct gb_bundle *bundle)
1153 {
1154 	struct gb_loopback *gb = greybus_get_drvdata(bundle);
1155 	unsigned long flags;
1156 	int ret;
1157 
1158 	ret = gb_pm_runtime_get_sync(bundle);
1159 	if (ret)
1160 		gb_pm_runtime_get_noresume(bundle);
1161 
1162 	gb_connection_disable(gb->connection);
1163 
1164 	if (!IS_ERR_OR_NULL(gb->task))
1165 		kthread_stop(gb->task);
1166 
1167 	kfifo_free(&gb->kfifo_lat);
1168 	gb_connection_latency_tag_disable(gb->connection);
1169 	debugfs_remove(gb->file);
1170 
1171 	/*
1172 	 * FIXME: gb_loopback_async_wait_all() is redundant now, as connection
1173 	 * is disabled at the beginning and so we can't have any more
1174 	 * incoming/outgoing requests.
1175 	 */
1176 	gb_loopback_async_wait_all(gb);
1177 
1178 	spin_lock_irqsave(&gb_dev.lock, flags);
1179 	gb_dev.count--;
1180 	list_del(&gb->entry);
1181 	spin_unlock_irqrestore(&gb_dev.lock, flags);
1182 
1183 	device_unregister(gb->dev);
1184 	ida_simple_remove(&loopback_ida, gb->id);
1185 
1186 	gb_connection_destroy(gb->connection);
1187 	kfree(gb);
1188 }
1189 
1190 static const struct greybus_bundle_id gb_loopback_id_table[] = {
1191 	{ GREYBUS_DEVICE_CLASS(GREYBUS_CLASS_LOOPBACK) },
1192 	{ }
1193 };
1194 MODULE_DEVICE_TABLE(greybus, gb_loopback_id_table);
1195 
1196 static struct greybus_driver gb_loopback_driver = {
1197 	.name		= "loopback",
1198 	.probe		= gb_loopback_probe,
1199 	.disconnect	= gb_loopback_disconnect,
1200 	.id_table	= gb_loopback_id_table,
1201 };
1202 
1203 static int loopback_init(void)
1204 {
1205 	int retval;
1206 
1207 	INIT_LIST_HEAD(&gb_dev.list);
1208 	INIT_LIST_HEAD(&gb_dev.list_op_async);
1209 	spin_lock_init(&gb_dev.lock);
1210 	gb_dev.root = debugfs_create_dir("gb_loopback", NULL);
1211 
1212 	retval = class_register(&loopback_class);
1213 	if (retval)
1214 		goto err;
1215 
1216 	retval = greybus_register(&gb_loopback_driver);
1217 	if (retval)
1218 		goto err_unregister;
1219 
1220 	return 0;
1221 
1222 err_unregister:
1223 	class_unregister(&loopback_class);
1224 err:
1225 	debugfs_remove_recursive(gb_dev.root);
1226 	return retval;
1227 }
1228 module_init(loopback_init);
1229 
1230 static void __exit loopback_exit(void)
1231 {
1232 	debugfs_remove_recursive(gb_dev.root);
1233 	greybus_deregister(&gb_loopback_driver);
1234 	class_unregister(&loopback_class);
1235 	ida_destroy(&loopback_ida);
1236 }
1237 module_exit(loopback_exit);
1238 
1239 MODULE_LICENSE("GPL v2");
1240