1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Memory-to-memory device framework for Video for Linux 2 and vb2.
4  *
5  * Helper functions for devices that use vb2 buffers for both their
6  * source and destination.
7  *
8  * Copyright (c) 2009-2010 Samsung Electronics Co., Ltd.
9  * Pawel Osciak, <pawel@osciak.com>
10  * Marek Szyprowski, <m.szyprowski@samsung.com>
11  */
12 #include <linux/module.h>
13 #include <linux/sched.h>
14 #include <linux/slab.h>
15 
16 #include <media/media-device.h>
17 #include <media/videobuf2-v4l2.h>
18 #include <media/v4l2-mem2mem.h>
19 #include <media/v4l2-dev.h>
20 #include <media/v4l2-device.h>
21 #include <media/v4l2-fh.h>
22 #include <media/v4l2-event.h>
23 
24 MODULE_DESCRIPTION("Mem to mem device framework for vb2");
25 MODULE_AUTHOR("Pawel Osciak, <pawel@osciak.com>");
26 MODULE_LICENSE("GPL");
27 
28 static bool debug;
29 module_param(debug, bool, 0644);
30 
31 #define dprintk(fmt, arg...)						\
32 	do {								\
33 		if (debug)						\
34 			printk(KERN_DEBUG "%s: " fmt, __func__, ## arg);\
35 	} while (0)
36 
37 
38 /* Instance is already queued on the job_queue */
39 #define TRANS_QUEUED		(1 << 0)
40 /* Instance is currently running in hardware */
41 #define TRANS_RUNNING		(1 << 1)
42 /* Instance is currently aborting */
43 #define TRANS_ABORT		(1 << 2)
44 
45 
46 /* The job queue is not running new jobs */
47 #define QUEUE_PAUSED		(1 << 0)
48 
49 
50 /* Offset base for buffers on the destination queue - used to distinguish
51  * between source and destination buffers when mmapping - they receive the same
52  * offsets but for different queues */
53 #define DST_QUEUE_OFF_BASE	(1 << 30)
54 
55 enum v4l2_m2m_entity_type {
56 	MEM2MEM_ENT_TYPE_SOURCE,
57 	MEM2MEM_ENT_TYPE_SINK,
58 	MEM2MEM_ENT_TYPE_PROC
59 };
60 
61 static const char * const m2m_entity_name[] = {
62 	"source",
63 	"sink",
64 	"proc"
65 };
66 
67 /**
68  * struct v4l2_m2m_dev - per-device context
69  * @source:		&struct media_entity pointer with the source entity
70  *			Used only when the M2M device is registered via
71  *			v4l2_m2m_register_media_controller().
72  * @source_pad:		&struct media_pad with the source pad.
73  *			Used only when the M2M device is registered via
74  *			v4l2_m2m_register_media_controller().
75  * @sink:		&struct media_entity pointer with the sink entity
76  *			Used only when the M2M device is registered via
77  *			v4l2_m2m_register_media_controller().
78  * @sink_pad:		&struct media_pad with the sink pad.
79  *			Used only when the M2M device is registered via
80  *			v4l2_m2m_register_media_controller().
81  * @proc:		&struct media_entity pointer with the M2M device itself.
82  * @proc_pads:		&struct media_pad with the @proc pads.
83  *			Used only when the M2M device is registered via
84  *			v4l2_m2m_unregister_media_controller().
85  * @intf_devnode:	&struct media_intf devnode pointer with the interface
86  *			with controls the M2M device.
87  * @curr_ctx:		currently running instance
88  * @job_queue:		instances queued to run
89  * @job_spinlock:	protects job_queue
90  * @job_work:		worker to run queued jobs.
91  * @job_queue_flags:	flags of the queue status, %QUEUE_PAUSED.
92  * @m2m_ops:		driver callbacks
93  */
94 struct v4l2_m2m_dev {
95 	struct v4l2_m2m_ctx	*curr_ctx;
96 #ifdef CONFIG_MEDIA_CONTROLLER
97 	struct media_entity	*source;
98 	struct media_pad	source_pad;
99 	struct media_entity	sink;
100 	struct media_pad	sink_pad;
101 	struct media_entity	proc;
102 	struct media_pad	proc_pads[2];
103 	struct media_intf_devnode *intf_devnode;
104 #endif
105 
106 	struct list_head	job_queue;
107 	spinlock_t		job_spinlock;
108 	struct work_struct	job_work;
109 	unsigned long		job_queue_flags;
110 
111 	const struct v4l2_m2m_ops *m2m_ops;
112 };
113 
114 static struct v4l2_m2m_queue_ctx *get_queue_ctx(struct v4l2_m2m_ctx *m2m_ctx,
115 						enum v4l2_buf_type type)
116 {
117 	if (V4L2_TYPE_IS_OUTPUT(type))
118 		return &m2m_ctx->out_q_ctx;
119 	else
120 		return &m2m_ctx->cap_q_ctx;
121 }
122 
123 struct vb2_queue *v4l2_m2m_get_vq(struct v4l2_m2m_ctx *m2m_ctx,
124 				       enum v4l2_buf_type type)
125 {
126 	struct v4l2_m2m_queue_ctx *q_ctx;
127 
128 	q_ctx = get_queue_ctx(m2m_ctx, type);
129 	if (!q_ctx)
130 		return NULL;
131 
132 	return &q_ctx->q;
133 }
134 EXPORT_SYMBOL(v4l2_m2m_get_vq);
135 
136 struct vb2_v4l2_buffer *v4l2_m2m_next_buf(struct v4l2_m2m_queue_ctx *q_ctx)
137 {
138 	struct v4l2_m2m_buffer *b;
139 	unsigned long flags;
140 
141 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
142 
143 	if (list_empty(&q_ctx->rdy_queue)) {
144 		spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
145 		return NULL;
146 	}
147 
148 	b = list_first_entry(&q_ctx->rdy_queue, struct v4l2_m2m_buffer, list);
149 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
150 	return &b->vb;
151 }
152 EXPORT_SYMBOL_GPL(v4l2_m2m_next_buf);
153 
154 struct vb2_v4l2_buffer *v4l2_m2m_last_buf(struct v4l2_m2m_queue_ctx *q_ctx)
155 {
156 	struct v4l2_m2m_buffer *b;
157 	unsigned long flags;
158 
159 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
160 
161 	if (list_empty(&q_ctx->rdy_queue)) {
162 		spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
163 		return NULL;
164 	}
165 
166 	b = list_last_entry(&q_ctx->rdy_queue, struct v4l2_m2m_buffer, list);
167 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
168 	return &b->vb;
169 }
170 EXPORT_SYMBOL_GPL(v4l2_m2m_last_buf);
171 
172 struct vb2_v4l2_buffer *v4l2_m2m_buf_remove(struct v4l2_m2m_queue_ctx *q_ctx)
173 {
174 	struct v4l2_m2m_buffer *b;
175 	unsigned long flags;
176 
177 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
178 	if (list_empty(&q_ctx->rdy_queue)) {
179 		spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
180 		return NULL;
181 	}
182 	b = list_first_entry(&q_ctx->rdy_queue, struct v4l2_m2m_buffer, list);
183 	list_del(&b->list);
184 	q_ctx->num_rdy--;
185 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
186 
187 	return &b->vb;
188 }
189 EXPORT_SYMBOL_GPL(v4l2_m2m_buf_remove);
190 
191 void v4l2_m2m_buf_remove_by_buf(struct v4l2_m2m_queue_ctx *q_ctx,
192 				struct vb2_v4l2_buffer *vbuf)
193 {
194 	struct v4l2_m2m_buffer *b;
195 	unsigned long flags;
196 
197 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
198 	b = container_of(vbuf, struct v4l2_m2m_buffer, vb);
199 	list_del(&b->list);
200 	q_ctx->num_rdy--;
201 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
202 }
203 EXPORT_SYMBOL_GPL(v4l2_m2m_buf_remove_by_buf);
204 
205 struct vb2_v4l2_buffer *
206 v4l2_m2m_buf_remove_by_idx(struct v4l2_m2m_queue_ctx *q_ctx, unsigned int idx)
207 
208 {
209 	struct v4l2_m2m_buffer *b, *tmp;
210 	struct vb2_v4l2_buffer *ret = NULL;
211 	unsigned long flags;
212 
213 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
214 	list_for_each_entry_safe(b, tmp, &q_ctx->rdy_queue, list) {
215 		if (b->vb.vb2_buf.index == idx) {
216 			list_del(&b->list);
217 			q_ctx->num_rdy--;
218 			ret = &b->vb;
219 			break;
220 		}
221 	}
222 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
223 
224 	return ret;
225 }
226 EXPORT_SYMBOL_GPL(v4l2_m2m_buf_remove_by_idx);
227 
228 /*
229  * Scheduling handlers
230  */
231 
232 void *v4l2_m2m_get_curr_priv(struct v4l2_m2m_dev *m2m_dev)
233 {
234 	unsigned long flags;
235 	void *ret = NULL;
236 
237 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
238 	if (m2m_dev->curr_ctx)
239 		ret = m2m_dev->curr_ctx->priv;
240 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
241 
242 	return ret;
243 }
244 EXPORT_SYMBOL(v4l2_m2m_get_curr_priv);
245 
246 /**
247  * v4l2_m2m_try_run() - select next job to perform and run it if possible
248  * @m2m_dev: per-device context
249  *
250  * Get next transaction (if present) from the waiting jobs list and run it.
251  *
252  * Note that this function can run on a given v4l2_m2m_ctx context,
253  * but call .device_run for another context.
254  */
255 static void v4l2_m2m_try_run(struct v4l2_m2m_dev *m2m_dev)
256 {
257 	unsigned long flags;
258 
259 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
260 	if (NULL != m2m_dev->curr_ctx) {
261 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
262 		dprintk("Another instance is running, won't run now\n");
263 		return;
264 	}
265 
266 	if (list_empty(&m2m_dev->job_queue)) {
267 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
268 		dprintk("No job pending\n");
269 		return;
270 	}
271 
272 	if (m2m_dev->job_queue_flags & QUEUE_PAUSED) {
273 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
274 		dprintk("Running new jobs is paused\n");
275 		return;
276 	}
277 
278 	m2m_dev->curr_ctx = list_first_entry(&m2m_dev->job_queue,
279 				   struct v4l2_m2m_ctx, queue);
280 	m2m_dev->curr_ctx->job_flags |= TRANS_RUNNING;
281 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
282 
283 	dprintk("Running job on m2m_ctx: %p\n", m2m_dev->curr_ctx);
284 	m2m_dev->m2m_ops->device_run(m2m_dev->curr_ctx->priv);
285 }
286 
287 /*
288  * __v4l2_m2m_try_queue() - queue a job
289  * @m2m_dev: m2m device
290  * @m2m_ctx: m2m context
291  *
292  * Check if this context is ready to queue a job.
293  *
294  * This function can run in interrupt context.
295  */
296 static void __v4l2_m2m_try_queue(struct v4l2_m2m_dev *m2m_dev,
297 				 struct v4l2_m2m_ctx *m2m_ctx)
298 {
299 	unsigned long flags_job;
300 	struct vb2_v4l2_buffer *dst, *src;
301 
302 	dprintk("Trying to schedule a job for m2m_ctx: %p\n", m2m_ctx);
303 
304 	if (!m2m_ctx->out_q_ctx.q.streaming
305 	    || !m2m_ctx->cap_q_ctx.q.streaming) {
306 		dprintk("Streaming needs to be on for both queues\n");
307 		return;
308 	}
309 
310 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags_job);
311 
312 	/* If the context is aborted then don't schedule it */
313 	if (m2m_ctx->job_flags & TRANS_ABORT) {
314 		dprintk("Aborted context\n");
315 		goto job_unlock;
316 	}
317 
318 	if (m2m_ctx->job_flags & TRANS_QUEUED) {
319 		dprintk("On job queue already\n");
320 		goto job_unlock;
321 	}
322 
323 	src = v4l2_m2m_next_src_buf(m2m_ctx);
324 	dst = v4l2_m2m_next_dst_buf(m2m_ctx);
325 	if (!src && !m2m_ctx->out_q_ctx.buffered) {
326 		dprintk("No input buffers available\n");
327 		goto job_unlock;
328 	}
329 	if (!dst && !m2m_ctx->cap_q_ctx.buffered) {
330 		dprintk("No output buffers available\n");
331 		goto job_unlock;
332 	}
333 
334 	m2m_ctx->new_frame = true;
335 
336 	if (src && dst && dst->is_held &&
337 	    dst->vb2_buf.copied_timestamp &&
338 	    dst->vb2_buf.timestamp != src->vb2_buf.timestamp) {
339 		dprintk("Timestamp mismatch, returning held capture buffer\n");
340 		dst->is_held = false;
341 		v4l2_m2m_dst_buf_remove(m2m_ctx);
342 		v4l2_m2m_buf_done(dst, VB2_BUF_STATE_DONE);
343 		dst = v4l2_m2m_next_dst_buf(m2m_ctx);
344 
345 		if (!dst && !m2m_ctx->cap_q_ctx.buffered) {
346 			dprintk("No output buffers available after returning held buffer\n");
347 			goto job_unlock;
348 		}
349 	}
350 
351 	if (src && dst && (m2m_ctx->out_q_ctx.q.subsystem_flags &
352 			   VB2_V4L2_FL_SUPPORTS_M2M_HOLD_CAPTURE_BUF))
353 		m2m_ctx->new_frame = !dst->vb2_buf.copied_timestamp ||
354 			dst->vb2_buf.timestamp != src->vb2_buf.timestamp;
355 
356 	if (m2m_ctx->has_stopped) {
357 		dprintk("Device has stopped\n");
358 		goto job_unlock;
359 	}
360 
361 	if (m2m_dev->m2m_ops->job_ready
362 		&& (!m2m_dev->m2m_ops->job_ready(m2m_ctx->priv))) {
363 		dprintk("Driver not ready\n");
364 		goto job_unlock;
365 	}
366 
367 	list_add_tail(&m2m_ctx->queue, &m2m_dev->job_queue);
368 	m2m_ctx->job_flags |= TRANS_QUEUED;
369 
370 job_unlock:
371 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
372 }
373 
374 /**
375  * v4l2_m2m_try_schedule() - schedule and possibly run a job for any context
376  * @m2m_ctx: m2m context
377  *
378  * Check if this context is ready to queue a job. If suitable,
379  * run the next queued job on the mem2mem device.
380  *
381  * This function shouldn't run in interrupt context.
382  *
383  * Note that v4l2_m2m_try_schedule() can schedule one job for this context,
384  * and then run another job for another context.
385  */
386 void v4l2_m2m_try_schedule(struct v4l2_m2m_ctx *m2m_ctx)
387 {
388 	struct v4l2_m2m_dev *m2m_dev = m2m_ctx->m2m_dev;
389 
390 	__v4l2_m2m_try_queue(m2m_dev, m2m_ctx);
391 	v4l2_m2m_try_run(m2m_dev);
392 }
393 EXPORT_SYMBOL_GPL(v4l2_m2m_try_schedule);
394 
395 /**
396  * v4l2_m2m_device_run_work() - run pending jobs for the context
397  * @work: Work structure used for scheduling the execution of this function.
398  */
399 static void v4l2_m2m_device_run_work(struct work_struct *work)
400 {
401 	struct v4l2_m2m_dev *m2m_dev =
402 		container_of(work, struct v4l2_m2m_dev, job_work);
403 
404 	v4l2_m2m_try_run(m2m_dev);
405 }
406 
407 /**
408  * v4l2_m2m_cancel_job() - cancel pending jobs for the context
409  * @m2m_ctx: m2m context with jobs to be canceled
410  *
411  * In case of streamoff or release called on any context,
412  * 1] If the context is currently running, then abort job will be called
413  * 2] If the context is queued, then the context will be removed from
414  *    the job_queue
415  */
416 static void v4l2_m2m_cancel_job(struct v4l2_m2m_ctx *m2m_ctx)
417 {
418 	struct v4l2_m2m_dev *m2m_dev;
419 	unsigned long flags;
420 
421 	m2m_dev = m2m_ctx->m2m_dev;
422 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
423 
424 	m2m_ctx->job_flags |= TRANS_ABORT;
425 	if (m2m_ctx->job_flags & TRANS_RUNNING) {
426 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
427 		if (m2m_dev->m2m_ops->job_abort)
428 			m2m_dev->m2m_ops->job_abort(m2m_ctx->priv);
429 		dprintk("m2m_ctx %p running, will wait to complete\n", m2m_ctx);
430 		wait_event(m2m_ctx->finished,
431 				!(m2m_ctx->job_flags & TRANS_RUNNING));
432 	} else if (m2m_ctx->job_flags & TRANS_QUEUED) {
433 		list_del(&m2m_ctx->queue);
434 		m2m_ctx->job_flags &= ~(TRANS_QUEUED | TRANS_RUNNING);
435 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
436 		dprintk("m2m_ctx: %p had been on queue and was removed\n",
437 			m2m_ctx);
438 	} else {
439 		/* Do nothing, was not on queue/running */
440 		spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
441 	}
442 }
443 
444 /*
445  * Schedule the next job, called from v4l2_m2m_job_finish() or
446  * v4l2_m2m_buf_done_and_job_finish().
447  */
448 static void v4l2_m2m_schedule_next_job(struct v4l2_m2m_dev *m2m_dev,
449 				       struct v4l2_m2m_ctx *m2m_ctx)
450 {
451 	/*
452 	 * This instance might have more buffers ready, but since we do not
453 	 * allow more than one job on the job_queue per instance, each has
454 	 * to be scheduled separately after the previous one finishes.
455 	 */
456 	__v4l2_m2m_try_queue(m2m_dev, m2m_ctx);
457 
458 	/*
459 	 * We might be running in atomic context,
460 	 * but the job must be run in non-atomic context.
461 	 */
462 	schedule_work(&m2m_dev->job_work);
463 }
464 
465 /*
466  * Assumes job_spinlock is held, called from v4l2_m2m_job_finish() or
467  * v4l2_m2m_buf_done_and_job_finish().
468  */
469 static bool _v4l2_m2m_job_finish(struct v4l2_m2m_dev *m2m_dev,
470 				 struct v4l2_m2m_ctx *m2m_ctx)
471 {
472 	if (!m2m_dev->curr_ctx || m2m_dev->curr_ctx != m2m_ctx) {
473 		dprintk("Called by an instance not currently running\n");
474 		return false;
475 	}
476 
477 	list_del(&m2m_dev->curr_ctx->queue);
478 	m2m_dev->curr_ctx->job_flags &= ~(TRANS_QUEUED | TRANS_RUNNING);
479 	wake_up(&m2m_dev->curr_ctx->finished);
480 	m2m_dev->curr_ctx = NULL;
481 	return true;
482 }
483 
484 void v4l2_m2m_job_finish(struct v4l2_m2m_dev *m2m_dev,
485 			 struct v4l2_m2m_ctx *m2m_ctx)
486 {
487 	unsigned long flags;
488 	bool schedule_next;
489 
490 	/*
491 	 * This function should not be used for drivers that support
492 	 * holding capture buffers. Those should use
493 	 * v4l2_m2m_buf_done_and_job_finish() instead.
494 	 */
495 	WARN_ON(m2m_ctx->out_q_ctx.q.subsystem_flags &
496 		VB2_V4L2_FL_SUPPORTS_M2M_HOLD_CAPTURE_BUF);
497 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
498 	schedule_next = _v4l2_m2m_job_finish(m2m_dev, m2m_ctx);
499 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
500 
501 	if (schedule_next)
502 		v4l2_m2m_schedule_next_job(m2m_dev, m2m_ctx);
503 }
504 EXPORT_SYMBOL(v4l2_m2m_job_finish);
505 
506 void v4l2_m2m_buf_done_and_job_finish(struct v4l2_m2m_dev *m2m_dev,
507 				      struct v4l2_m2m_ctx *m2m_ctx,
508 				      enum vb2_buffer_state state)
509 {
510 	struct vb2_v4l2_buffer *src_buf, *dst_buf;
511 	bool schedule_next = false;
512 	unsigned long flags;
513 
514 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
515 	src_buf = v4l2_m2m_src_buf_remove(m2m_ctx);
516 	dst_buf = v4l2_m2m_next_dst_buf(m2m_ctx);
517 
518 	if (WARN_ON(!src_buf || !dst_buf))
519 		goto unlock;
520 	dst_buf->is_held = src_buf->flags & V4L2_BUF_FLAG_M2M_HOLD_CAPTURE_BUF;
521 	if (!dst_buf->is_held) {
522 		v4l2_m2m_dst_buf_remove(m2m_ctx);
523 		v4l2_m2m_buf_done(dst_buf, state);
524 	}
525 	/*
526 	 * If the request API is being used, returning the OUTPUT
527 	 * (src) buffer will wake-up any process waiting on the
528 	 * request file descriptor.
529 	 *
530 	 * Therefore, return the CAPTURE (dst) buffer first,
531 	 * to avoid signalling the request file descriptor
532 	 * before the CAPTURE buffer is done.
533 	 */
534 	v4l2_m2m_buf_done(src_buf, state);
535 	schedule_next = _v4l2_m2m_job_finish(m2m_dev, m2m_ctx);
536 unlock:
537 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
538 
539 	if (schedule_next)
540 		v4l2_m2m_schedule_next_job(m2m_dev, m2m_ctx);
541 }
542 EXPORT_SYMBOL(v4l2_m2m_buf_done_and_job_finish);
543 
544 void v4l2_m2m_suspend(struct v4l2_m2m_dev *m2m_dev)
545 {
546 	unsigned long flags;
547 	struct v4l2_m2m_ctx *curr_ctx;
548 
549 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
550 	m2m_dev->job_queue_flags |= QUEUE_PAUSED;
551 	curr_ctx = m2m_dev->curr_ctx;
552 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
553 
554 	if (curr_ctx)
555 		wait_event(curr_ctx->finished,
556 			   !(curr_ctx->job_flags & TRANS_RUNNING));
557 }
558 EXPORT_SYMBOL(v4l2_m2m_suspend);
559 
560 void v4l2_m2m_resume(struct v4l2_m2m_dev *m2m_dev)
561 {
562 	unsigned long flags;
563 
564 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
565 	m2m_dev->job_queue_flags &= ~QUEUE_PAUSED;
566 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
567 
568 	v4l2_m2m_try_run(m2m_dev);
569 }
570 EXPORT_SYMBOL(v4l2_m2m_resume);
571 
572 int v4l2_m2m_reqbufs(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
573 		     struct v4l2_requestbuffers *reqbufs)
574 {
575 	struct vb2_queue *vq;
576 	int ret;
577 
578 	vq = v4l2_m2m_get_vq(m2m_ctx, reqbufs->type);
579 	ret = vb2_reqbufs(vq, reqbufs);
580 	/* If count == 0, then the owner has released all buffers and he
581 	   is no longer owner of the queue. Otherwise we have an owner. */
582 	if (ret == 0)
583 		vq->owner = reqbufs->count ? file->private_data : NULL;
584 
585 	return ret;
586 }
587 EXPORT_SYMBOL_GPL(v4l2_m2m_reqbufs);
588 
589 static void v4l2_m2m_adjust_mem_offset(struct vb2_queue *vq,
590 				       struct v4l2_buffer *buf)
591 {
592 	/* Adjust MMAP memory offsets for the CAPTURE queue */
593 	if (buf->memory == V4L2_MEMORY_MMAP && V4L2_TYPE_IS_CAPTURE(vq->type)) {
594 		if (V4L2_TYPE_IS_MULTIPLANAR(vq->type)) {
595 			unsigned int i;
596 
597 			for (i = 0; i < buf->length; ++i)
598 				buf->m.planes[i].m.mem_offset
599 					+= DST_QUEUE_OFF_BASE;
600 		} else {
601 			buf->m.offset += DST_QUEUE_OFF_BASE;
602 		}
603 	}
604 }
605 
606 int v4l2_m2m_querybuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
607 		      struct v4l2_buffer *buf)
608 {
609 	struct vb2_queue *vq;
610 	int ret;
611 
612 	vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
613 	ret = vb2_querybuf(vq, buf);
614 	if (ret)
615 		return ret;
616 
617 	/* Adjust MMAP memory offsets for the CAPTURE queue */
618 	v4l2_m2m_adjust_mem_offset(vq, buf);
619 
620 	return 0;
621 }
622 EXPORT_SYMBOL_GPL(v4l2_m2m_querybuf);
623 
624 /*
625  * This will add the LAST flag and mark the buffer management
626  * state as stopped.
627  * This is called when the last capture buffer must be flagged as LAST
628  * in draining mode from the encoder/decoder driver buf_queue() callback
629  * or from v4l2_update_last_buf_state() when a capture buffer is available.
630  */
631 void v4l2_m2m_last_buffer_done(struct v4l2_m2m_ctx *m2m_ctx,
632 			       struct vb2_v4l2_buffer *vbuf)
633 {
634 	vbuf->flags |= V4L2_BUF_FLAG_LAST;
635 	vb2_buffer_done(&vbuf->vb2_buf, VB2_BUF_STATE_DONE);
636 
637 	v4l2_m2m_mark_stopped(m2m_ctx);
638 }
639 EXPORT_SYMBOL_GPL(v4l2_m2m_last_buffer_done);
640 
641 /* When stop command is issued, update buffer management state */
642 static int v4l2_update_last_buf_state(struct v4l2_m2m_ctx *m2m_ctx)
643 {
644 	struct vb2_v4l2_buffer *next_dst_buf;
645 
646 	if (m2m_ctx->is_draining)
647 		return -EBUSY;
648 
649 	if (m2m_ctx->has_stopped)
650 		return 0;
651 
652 	m2m_ctx->last_src_buf = v4l2_m2m_last_src_buf(m2m_ctx);
653 	m2m_ctx->is_draining = true;
654 
655 	/*
656 	 * The processing of the last output buffer queued before
657 	 * the STOP command is expected to mark the buffer management
658 	 * state as stopped with v4l2_m2m_mark_stopped().
659 	 */
660 	if (m2m_ctx->last_src_buf)
661 		return 0;
662 
663 	/*
664 	 * In case the output queue is empty, try to mark the last capture
665 	 * buffer as LAST.
666 	 */
667 	next_dst_buf = v4l2_m2m_dst_buf_remove(m2m_ctx);
668 	if (!next_dst_buf) {
669 		/*
670 		 * Wait for the next queued one in encoder/decoder driver
671 		 * buf_queue() callback using the v4l2_m2m_dst_buf_is_last()
672 		 * helper or in v4l2_m2m_qbuf() if encoder/decoder is not yet
673 		 * streaming.
674 		 */
675 		m2m_ctx->next_buf_last = true;
676 		return 0;
677 	}
678 
679 	v4l2_m2m_last_buffer_done(m2m_ctx, next_dst_buf);
680 
681 	return 0;
682 }
683 
684 /*
685  * Updates the encoding/decoding buffer management state, should
686  * be called from encoder/decoder drivers start_streaming()
687  */
688 void v4l2_m2m_update_start_streaming_state(struct v4l2_m2m_ctx *m2m_ctx,
689 					   struct vb2_queue *q)
690 {
691 	/* If start streaming again, untag the last output buffer */
692 	if (V4L2_TYPE_IS_OUTPUT(q->type))
693 		m2m_ctx->last_src_buf = NULL;
694 }
695 EXPORT_SYMBOL_GPL(v4l2_m2m_update_start_streaming_state);
696 
697 /*
698  * Updates the encoding/decoding buffer management state, should
699  * be called from encoder/decoder driver stop_streaming()
700  */
701 void v4l2_m2m_update_stop_streaming_state(struct v4l2_m2m_ctx *m2m_ctx,
702 					  struct vb2_queue *q)
703 {
704 	if (V4L2_TYPE_IS_OUTPUT(q->type)) {
705 		/*
706 		 * If in draining state, either mark next dst buffer as
707 		 * done or flag next one to be marked as done either
708 		 * in encoder/decoder driver buf_queue() callback using
709 		 * the v4l2_m2m_dst_buf_is_last() helper or in v4l2_m2m_qbuf()
710 		 * if encoder/decoder is not yet streaming
711 		 */
712 		if (m2m_ctx->is_draining) {
713 			struct vb2_v4l2_buffer *next_dst_buf;
714 
715 			m2m_ctx->last_src_buf = NULL;
716 			next_dst_buf = v4l2_m2m_dst_buf_remove(m2m_ctx);
717 			if (!next_dst_buf)
718 				m2m_ctx->next_buf_last = true;
719 			else
720 				v4l2_m2m_last_buffer_done(m2m_ctx,
721 							  next_dst_buf);
722 		}
723 	} else {
724 		v4l2_m2m_clear_state(m2m_ctx);
725 	}
726 }
727 EXPORT_SYMBOL_GPL(v4l2_m2m_update_stop_streaming_state);
728 
729 static void v4l2_m2m_force_last_buf_done(struct v4l2_m2m_ctx *m2m_ctx,
730 					 struct vb2_queue *q)
731 {
732 	struct vb2_buffer *vb;
733 	struct vb2_v4l2_buffer *vbuf;
734 	unsigned int i;
735 
736 	if (WARN_ON(q->is_output))
737 		return;
738 	if (list_empty(&q->queued_list))
739 		return;
740 
741 	vb = list_first_entry(&q->queued_list, struct vb2_buffer, queued_entry);
742 	for (i = 0; i < vb->num_planes; i++)
743 		vb2_set_plane_payload(vb, i, 0);
744 
745 	/*
746 	 * Since the buffer hasn't been queued to the ready queue,
747 	 * mark is active and owned before marking it LAST and DONE
748 	 */
749 	vb->state = VB2_BUF_STATE_ACTIVE;
750 	atomic_inc(&q->owned_by_drv_count);
751 
752 	vbuf = to_vb2_v4l2_buffer(vb);
753 	vbuf->field = V4L2_FIELD_NONE;
754 
755 	v4l2_m2m_last_buffer_done(m2m_ctx, vbuf);
756 }
757 
758 int v4l2_m2m_qbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
759 		  struct v4l2_buffer *buf)
760 {
761 	struct video_device *vdev = video_devdata(file);
762 	struct vb2_queue *vq;
763 	int ret;
764 
765 	vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
766 	if (V4L2_TYPE_IS_CAPTURE(vq->type) &&
767 	    (buf->flags & V4L2_BUF_FLAG_REQUEST_FD)) {
768 		dprintk("%s: requests cannot be used with capture buffers\n",
769 			__func__);
770 		return -EPERM;
771 	}
772 
773 	ret = vb2_qbuf(vq, vdev->v4l2_dev->mdev, buf);
774 	if (ret)
775 		return ret;
776 
777 	/* Adjust MMAP memory offsets for the CAPTURE queue */
778 	v4l2_m2m_adjust_mem_offset(vq, buf);
779 
780 	/*
781 	 * If the capture queue is streaming, but streaming hasn't started
782 	 * on the device, but was asked to stop, mark the previously queued
783 	 * buffer as DONE with LAST flag since it won't be queued on the
784 	 * device.
785 	 */
786 	if (V4L2_TYPE_IS_CAPTURE(vq->type) &&
787 	    vb2_is_streaming(vq) && !vb2_start_streaming_called(vq) &&
788 	   (v4l2_m2m_has_stopped(m2m_ctx) || v4l2_m2m_dst_buf_is_last(m2m_ctx)))
789 		v4l2_m2m_force_last_buf_done(m2m_ctx, vq);
790 	else if (!(buf->flags & V4L2_BUF_FLAG_IN_REQUEST))
791 		v4l2_m2m_try_schedule(m2m_ctx);
792 
793 	return 0;
794 }
795 EXPORT_SYMBOL_GPL(v4l2_m2m_qbuf);
796 
797 int v4l2_m2m_dqbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
798 		   struct v4l2_buffer *buf)
799 {
800 	struct vb2_queue *vq;
801 	int ret;
802 
803 	vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
804 	ret = vb2_dqbuf(vq, buf, file->f_flags & O_NONBLOCK);
805 	if (ret)
806 		return ret;
807 
808 	/* Adjust MMAP memory offsets for the CAPTURE queue */
809 	v4l2_m2m_adjust_mem_offset(vq, buf);
810 
811 	return 0;
812 }
813 EXPORT_SYMBOL_GPL(v4l2_m2m_dqbuf);
814 
815 int v4l2_m2m_prepare_buf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
816 			 struct v4l2_buffer *buf)
817 {
818 	struct video_device *vdev = video_devdata(file);
819 	struct vb2_queue *vq;
820 	int ret;
821 
822 	vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
823 	ret = vb2_prepare_buf(vq, vdev->v4l2_dev->mdev, buf);
824 	if (ret)
825 		return ret;
826 
827 	/* Adjust MMAP memory offsets for the CAPTURE queue */
828 	v4l2_m2m_adjust_mem_offset(vq, buf);
829 
830 	return 0;
831 }
832 EXPORT_SYMBOL_GPL(v4l2_m2m_prepare_buf);
833 
834 int v4l2_m2m_create_bufs(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
835 			 struct v4l2_create_buffers *create)
836 {
837 	struct vb2_queue *vq;
838 
839 	vq = v4l2_m2m_get_vq(m2m_ctx, create->format.type);
840 	return vb2_create_bufs(vq, create);
841 }
842 EXPORT_SYMBOL_GPL(v4l2_m2m_create_bufs);
843 
844 int v4l2_m2m_expbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
845 		  struct v4l2_exportbuffer *eb)
846 {
847 	struct vb2_queue *vq;
848 
849 	vq = v4l2_m2m_get_vq(m2m_ctx, eb->type);
850 	return vb2_expbuf(vq, eb);
851 }
852 EXPORT_SYMBOL_GPL(v4l2_m2m_expbuf);
853 
854 int v4l2_m2m_streamon(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
855 		      enum v4l2_buf_type type)
856 {
857 	struct vb2_queue *vq;
858 	int ret;
859 
860 	vq = v4l2_m2m_get_vq(m2m_ctx, type);
861 	ret = vb2_streamon(vq, type);
862 	if (!ret)
863 		v4l2_m2m_try_schedule(m2m_ctx);
864 
865 	return ret;
866 }
867 EXPORT_SYMBOL_GPL(v4l2_m2m_streamon);
868 
869 int v4l2_m2m_streamoff(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
870 		       enum v4l2_buf_type type)
871 {
872 	struct v4l2_m2m_dev *m2m_dev;
873 	struct v4l2_m2m_queue_ctx *q_ctx;
874 	unsigned long flags_job, flags;
875 	int ret;
876 
877 	/* wait until the current context is dequeued from job_queue */
878 	v4l2_m2m_cancel_job(m2m_ctx);
879 
880 	q_ctx = get_queue_ctx(m2m_ctx, type);
881 	ret = vb2_streamoff(&q_ctx->q, type);
882 	if (ret)
883 		return ret;
884 
885 	m2m_dev = m2m_ctx->m2m_dev;
886 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags_job);
887 	/* We should not be scheduled anymore, since we're dropping a queue. */
888 	if (m2m_ctx->job_flags & TRANS_QUEUED)
889 		list_del(&m2m_ctx->queue);
890 	m2m_ctx->job_flags = 0;
891 
892 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
893 	/* Drop queue, since streamoff returns device to the same state as after
894 	 * calling reqbufs. */
895 	INIT_LIST_HEAD(&q_ctx->rdy_queue);
896 	q_ctx->num_rdy = 0;
897 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
898 
899 	if (m2m_dev->curr_ctx == m2m_ctx) {
900 		m2m_dev->curr_ctx = NULL;
901 		wake_up(&m2m_ctx->finished);
902 	}
903 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
904 
905 	return 0;
906 }
907 EXPORT_SYMBOL_GPL(v4l2_m2m_streamoff);
908 
909 static __poll_t v4l2_m2m_poll_for_data(struct file *file,
910 				       struct v4l2_m2m_ctx *m2m_ctx,
911 				       struct poll_table_struct *wait)
912 {
913 	struct vb2_queue *src_q, *dst_q;
914 	__poll_t rc = 0;
915 	unsigned long flags;
916 
917 	src_q = v4l2_m2m_get_src_vq(m2m_ctx);
918 	dst_q = v4l2_m2m_get_dst_vq(m2m_ctx);
919 
920 	/*
921 	 * There has to be at least one buffer queued on each queued_list, which
922 	 * means either in driver already or waiting for driver to claim it
923 	 * and start processing.
924 	 */
925 	if ((!vb2_is_streaming(src_q) || src_q->error ||
926 	     list_empty(&src_q->queued_list)) &&
927 	    (!vb2_is_streaming(dst_q) || dst_q->error ||
928 	     (list_empty(&dst_q->queued_list) && !dst_q->last_buffer_dequeued)))
929 		return EPOLLERR;
930 
931 	spin_lock_irqsave(&src_q->done_lock, flags);
932 	if (!list_empty(&src_q->done_list))
933 		rc |= EPOLLOUT | EPOLLWRNORM;
934 	spin_unlock_irqrestore(&src_q->done_lock, flags);
935 
936 	spin_lock_irqsave(&dst_q->done_lock, flags);
937 	/*
938 	 * If the last buffer was dequeued from the capture queue, signal
939 	 * userspace. DQBUF(CAPTURE) will return -EPIPE.
940 	 */
941 	if (!list_empty(&dst_q->done_list) || dst_q->last_buffer_dequeued)
942 		rc |= EPOLLIN | EPOLLRDNORM;
943 	spin_unlock_irqrestore(&dst_q->done_lock, flags);
944 
945 	return rc;
946 }
947 
948 __poll_t v4l2_m2m_poll(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
949 		       struct poll_table_struct *wait)
950 {
951 	struct video_device *vfd = video_devdata(file);
952 	struct vb2_queue *src_q = v4l2_m2m_get_src_vq(m2m_ctx);
953 	struct vb2_queue *dst_q = v4l2_m2m_get_dst_vq(m2m_ctx);
954 	__poll_t req_events = poll_requested_events(wait);
955 	__poll_t rc = 0;
956 
957 	/*
958 	 * poll_wait() MUST be called on the first invocation on all the
959 	 * potential queues of interest, even if we are not interested in their
960 	 * events during this first call. Failure to do so will result in
961 	 * queue's events to be ignored because the poll_table won't be capable
962 	 * of adding new wait queues thereafter.
963 	 */
964 	poll_wait(file, &src_q->done_wq, wait);
965 	poll_wait(file, &dst_q->done_wq, wait);
966 
967 	if (req_events & (EPOLLOUT | EPOLLWRNORM | EPOLLIN | EPOLLRDNORM))
968 		rc = v4l2_m2m_poll_for_data(file, m2m_ctx, wait);
969 
970 	if (test_bit(V4L2_FL_USES_V4L2_FH, &vfd->flags)) {
971 		struct v4l2_fh *fh = file->private_data;
972 
973 		poll_wait(file, &fh->wait, wait);
974 		if (v4l2_event_pending(fh))
975 			rc |= EPOLLPRI;
976 	}
977 
978 	return rc;
979 }
980 EXPORT_SYMBOL_GPL(v4l2_m2m_poll);
981 
982 int v4l2_m2m_mmap(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
983 			 struct vm_area_struct *vma)
984 {
985 	unsigned long offset = vma->vm_pgoff << PAGE_SHIFT;
986 	struct vb2_queue *vq;
987 
988 	if (offset < DST_QUEUE_OFF_BASE) {
989 		vq = v4l2_m2m_get_src_vq(m2m_ctx);
990 	} else {
991 		vq = v4l2_m2m_get_dst_vq(m2m_ctx);
992 		vma->vm_pgoff -= (DST_QUEUE_OFF_BASE >> PAGE_SHIFT);
993 	}
994 
995 	return vb2_mmap(vq, vma);
996 }
997 EXPORT_SYMBOL(v4l2_m2m_mmap);
998 
999 #ifndef CONFIG_MMU
1000 unsigned long v4l2_m2m_get_unmapped_area(struct file *file, unsigned long addr,
1001 					 unsigned long len, unsigned long pgoff,
1002 					 unsigned long flags)
1003 {
1004 	struct v4l2_fh *fh = file->private_data;
1005 	unsigned long offset = pgoff << PAGE_SHIFT;
1006 	struct vb2_queue *vq;
1007 
1008 	if (offset < DST_QUEUE_OFF_BASE) {
1009 		vq = v4l2_m2m_get_src_vq(fh->m2m_ctx);
1010 	} else {
1011 		vq = v4l2_m2m_get_dst_vq(fh->m2m_ctx);
1012 		pgoff -= (DST_QUEUE_OFF_BASE >> PAGE_SHIFT);
1013 	}
1014 
1015 	return vb2_get_unmapped_area(vq, addr, len, pgoff, flags);
1016 }
1017 EXPORT_SYMBOL_GPL(v4l2_m2m_get_unmapped_area);
1018 #endif
1019 
1020 #if defined(CONFIG_MEDIA_CONTROLLER)
1021 void v4l2_m2m_unregister_media_controller(struct v4l2_m2m_dev *m2m_dev)
1022 {
1023 	media_remove_intf_links(&m2m_dev->intf_devnode->intf);
1024 	media_devnode_remove(m2m_dev->intf_devnode);
1025 
1026 	media_entity_remove_links(m2m_dev->source);
1027 	media_entity_remove_links(&m2m_dev->sink);
1028 	media_entity_remove_links(&m2m_dev->proc);
1029 	media_device_unregister_entity(m2m_dev->source);
1030 	media_device_unregister_entity(&m2m_dev->sink);
1031 	media_device_unregister_entity(&m2m_dev->proc);
1032 	kfree(m2m_dev->source->name);
1033 	kfree(m2m_dev->sink.name);
1034 	kfree(m2m_dev->proc.name);
1035 }
1036 EXPORT_SYMBOL_GPL(v4l2_m2m_unregister_media_controller);
1037 
1038 static int v4l2_m2m_register_entity(struct media_device *mdev,
1039 	struct v4l2_m2m_dev *m2m_dev, enum v4l2_m2m_entity_type type,
1040 	struct video_device *vdev, int function)
1041 {
1042 	struct media_entity *entity;
1043 	struct media_pad *pads;
1044 	char *name;
1045 	unsigned int len;
1046 	int num_pads;
1047 	int ret;
1048 
1049 	switch (type) {
1050 	case MEM2MEM_ENT_TYPE_SOURCE:
1051 		entity = m2m_dev->source;
1052 		pads = &m2m_dev->source_pad;
1053 		pads[0].flags = MEDIA_PAD_FL_SOURCE;
1054 		num_pads = 1;
1055 		break;
1056 	case MEM2MEM_ENT_TYPE_SINK:
1057 		entity = &m2m_dev->sink;
1058 		pads = &m2m_dev->sink_pad;
1059 		pads[0].flags = MEDIA_PAD_FL_SINK;
1060 		num_pads = 1;
1061 		break;
1062 	case MEM2MEM_ENT_TYPE_PROC:
1063 		entity = &m2m_dev->proc;
1064 		pads = m2m_dev->proc_pads;
1065 		pads[0].flags = MEDIA_PAD_FL_SINK;
1066 		pads[1].flags = MEDIA_PAD_FL_SOURCE;
1067 		num_pads = 2;
1068 		break;
1069 	default:
1070 		return -EINVAL;
1071 	}
1072 
1073 	entity->obj_type = MEDIA_ENTITY_TYPE_BASE;
1074 	if (type != MEM2MEM_ENT_TYPE_PROC) {
1075 		entity->info.dev.major = VIDEO_MAJOR;
1076 		entity->info.dev.minor = vdev->minor;
1077 	}
1078 	len = strlen(vdev->name) + 2 + strlen(m2m_entity_name[type]);
1079 	name = kmalloc(len, GFP_KERNEL);
1080 	if (!name)
1081 		return -ENOMEM;
1082 	snprintf(name, len, "%s-%s", vdev->name, m2m_entity_name[type]);
1083 	entity->name = name;
1084 	entity->function = function;
1085 
1086 	ret = media_entity_pads_init(entity, num_pads, pads);
1087 	if (ret)
1088 		return ret;
1089 	ret = media_device_register_entity(mdev, entity);
1090 	if (ret)
1091 		return ret;
1092 
1093 	return 0;
1094 }
1095 
1096 int v4l2_m2m_register_media_controller(struct v4l2_m2m_dev *m2m_dev,
1097 		struct video_device *vdev, int function)
1098 {
1099 	struct media_device *mdev = vdev->v4l2_dev->mdev;
1100 	struct media_link *link;
1101 	int ret;
1102 
1103 	if (!mdev)
1104 		return 0;
1105 
1106 	/* A memory-to-memory device consists in two
1107 	 * DMA engine and one video processing entities.
1108 	 * The DMA engine entities are linked to a V4L interface
1109 	 */
1110 
1111 	/* Create the three entities with their pads */
1112 	m2m_dev->source = &vdev->entity;
1113 	ret = v4l2_m2m_register_entity(mdev, m2m_dev,
1114 			MEM2MEM_ENT_TYPE_SOURCE, vdev, MEDIA_ENT_F_IO_V4L);
1115 	if (ret)
1116 		return ret;
1117 	ret = v4l2_m2m_register_entity(mdev, m2m_dev,
1118 			MEM2MEM_ENT_TYPE_PROC, vdev, function);
1119 	if (ret)
1120 		goto err_rel_entity0;
1121 	ret = v4l2_m2m_register_entity(mdev, m2m_dev,
1122 			MEM2MEM_ENT_TYPE_SINK, vdev, MEDIA_ENT_F_IO_V4L);
1123 	if (ret)
1124 		goto err_rel_entity1;
1125 
1126 	/* Connect the three entities */
1127 	ret = media_create_pad_link(m2m_dev->source, 0, &m2m_dev->proc, 0,
1128 			MEDIA_LNK_FL_IMMUTABLE | MEDIA_LNK_FL_ENABLED);
1129 	if (ret)
1130 		goto err_rel_entity2;
1131 
1132 	ret = media_create_pad_link(&m2m_dev->proc, 1, &m2m_dev->sink, 0,
1133 			MEDIA_LNK_FL_IMMUTABLE | MEDIA_LNK_FL_ENABLED);
1134 	if (ret)
1135 		goto err_rm_links0;
1136 
1137 	/* Create video interface */
1138 	m2m_dev->intf_devnode = media_devnode_create(mdev,
1139 			MEDIA_INTF_T_V4L_VIDEO, 0,
1140 			VIDEO_MAJOR, vdev->minor);
1141 	if (!m2m_dev->intf_devnode) {
1142 		ret = -ENOMEM;
1143 		goto err_rm_links1;
1144 	}
1145 
1146 	/* Connect the two DMA engines to the interface */
1147 	link = media_create_intf_link(m2m_dev->source,
1148 			&m2m_dev->intf_devnode->intf,
1149 			MEDIA_LNK_FL_IMMUTABLE | MEDIA_LNK_FL_ENABLED);
1150 	if (!link) {
1151 		ret = -ENOMEM;
1152 		goto err_rm_devnode;
1153 	}
1154 
1155 	link = media_create_intf_link(&m2m_dev->sink,
1156 			&m2m_dev->intf_devnode->intf,
1157 			MEDIA_LNK_FL_IMMUTABLE | MEDIA_LNK_FL_ENABLED);
1158 	if (!link) {
1159 		ret = -ENOMEM;
1160 		goto err_rm_intf_link;
1161 	}
1162 	return 0;
1163 
1164 err_rm_intf_link:
1165 	media_remove_intf_links(&m2m_dev->intf_devnode->intf);
1166 err_rm_devnode:
1167 	media_devnode_remove(m2m_dev->intf_devnode);
1168 err_rm_links1:
1169 	media_entity_remove_links(&m2m_dev->sink);
1170 err_rm_links0:
1171 	media_entity_remove_links(&m2m_dev->proc);
1172 	media_entity_remove_links(m2m_dev->source);
1173 err_rel_entity2:
1174 	media_device_unregister_entity(&m2m_dev->proc);
1175 	kfree(m2m_dev->proc.name);
1176 err_rel_entity1:
1177 	media_device_unregister_entity(&m2m_dev->sink);
1178 	kfree(m2m_dev->sink.name);
1179 err_rel_entity0:
1180 	media_device_unregister_entity(m2m_dev->source);
1181 	kfree(m2m_dev->source->name);
1182 	return ret;
1183 	return 0;
1184 }
1185 EXPORT_SYMBOL_GPL(v4l2_m2m_register_media_controller);
1186 #endif
1187 
1188 struct v4l2_m2m_dev *v4l2_m2m_init(const struct v4l2_m2m_ops *m2m_ops)
1189 {
1190 	struct v4l2_m2m_dev *m2m_dev;
1191 
1192 	if (!m2m_ops || WARN_ON(!m2m_ops->device_run))
1193 		return ERR_PTR(-EINVAL);
1194 
1195 	m2m_dev = kzalloc(sizeof *m2m_dev, GFP_KERNEL);
1196 	if (!m2m_dev)
1197 		return ERR_PTR(-ENOMEM);
1198 
1199 	m2m_dev->curr_ctx = NULL;
1200 	m2m_dev->m2m_ops = m2m_ops;
1201 	INIT_LIST_HEAD(&m2m_dev->job_queue);
1202 	spin_lock_init(&m2m_dev->job_spinlock);
1203 	INIT_WORK(&m2m_dev->job_work, v4l2_m2m_device_run_work);
1204 
1205 	return m2m_dev;
1206 }
1207 EXPORT_SYMBOL_GPL(v4l2_m2m_init);
1208 
1209 void v4l2_m2m_release(struct v4l2_m2m_dev *m2m_dev)
1210 {
1211 	kfree(m2m_dev);
1212 }
1213 EXPORT_SYMBOL_GPL(v4l2_m2m_release);
1214 
1215 struct v4l2_m2m_ctx *v4l2_m2m_ctx_init(struct v4l2_m2m_dev *m2m_dev,
1216 		void *drv_priv,
1217 		int (*queue_init)(void *priv, struct vb2_queue *src_vq, struct vb2_queue *dst_vq))
1218 {
1219 	struct v4l2_m2m_ctx *m2m_ctx;
1220 	struct v4l2_m2m_queue_ctx *out_q_ctx, *cap_q_ctx;
1221 	int ret;
1222 
1223 	m2m_ctx = kzalloc(sizeof *m2m_ctx, GFP_KERNEL);
1224 	if (!m2m_ctx)
1225 		return ERR_PTR(-ENOMEM);
1226 
1227 	m2m_ctx->priv = drv_priv;
1228 	m2m_ctx->m2m_dev = m2m_dev;
1229 	init_waitqueue_head(&m2m_ctx->finished);
1230 
1231 	out_q_ctx = &m2m_ctx->out_q_ctx;
1232 	cap_q_ctx = &m2m_ctx->cap_q_ctx;
1233 
1234 	INIT_LIST_HEAD(&out_q_ctx->rdy_queue);
1235 	INIT_LIST_HEAD(&cap_q_ctx->rdy_queue);
1236 	spin_lock_init(&out_q_ctx->rdy_spinlock);
1237 	spin_lock_init(&cap_q_ctx->rdy_spinlock);
1238 
1239 	INIT_LIST_HEAD(&m2m_ctx->queue);
1240 
1241 	ret = queue_init(drv_priv, &out_q_ctx->q, &cap_q_ctx->q);
1242 
1243 	if (ret)
1244 		goto err;
1245 	/*
1246 	 * Both queues should use same the mutex to lock the m2m context.
1247 	 * This lock is used in some v4l2_m2m_* helpers.
1248 	 */
1249 	if (WARN_ON(out_q_ctx->q.lock != cap_q_ctx->q.lock)) {
1250 		ret = -EINVAL;
1251 		goto err;
1252 	}
1253 	m2m_ctx->q_lock = out_q_ctx->q.lock;
1254 
1255 	return m2m_ctx;
1256 err:
1257 	kfree(m2m_ctx);
1258 	return ERR_PTR(ret);
1259 }
1260 EXPORT_SYMBOL_GPL(v4l2_m2m_ctx_init);
1261 
1262 void v4l2_m2m_ctx_release(struct v4l2_m2m_ctx *m2m_ctx)
1263 {
1264 	/* wait until the current context is dequeued from job_queue */
1265 	v4l2_m2m_cancel_job(m2m_ctx);
1266 
1267 	vb2_queue_release(&m2m_ctx->cap_q_ctx.q);
1268 	vb2_queue_release(&m2m_ctx->out_q_ctx.q);
1269 
1270 	kfree(m2m_ctx);
1271 }
1272 EXPORT_SYMBOL_GPL(v4l2_m2m_ctx_release);
1273 
1274 void v4l2_m2m_buf_queue(struct v4l2_m2m_ctx *m2m_ctx,
1275 		struct vb2_v4l2_buffer *vbuf)
1276 {
1277 	struct v4l2_m2m_buffer *b = container_of(vbuf,
1278 				struct v4l2_m2m_buffer, vb);
1279 	struct v4l2_m2m_queue_ctx *q_ctx;
1280 	unsigned long flags;
1281 
1282 	q_ctx = get_queue_ctx(m2m_ctx, vbuf->vb2_buf.vb2_queue->type);
1283 	if (!q_ctx)
1284 		return;
1285 
1286 	spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
1287 	list_add_tail(&b->list, &q_ctx->rdy_queue);
1288 	q_ctx->num_rdy++;
1289 	spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
1290 }
1291 EXPORT_SYMBOL_GPL(v4l2_m2m_buf_queue);
1292 
1293 void v4l2_m2m_buf_copy_metadata(const struct vb2_v4l2_buffer *out_vb,
1294 				struct vb2_v4l2_buffer *cap_vb,
1295 				bool copy_frame_flags)
1296 {
1297 	u32 mask = V4L2_BUF_FLAG_TIMECODE | V4L2_BUF_FLAG_TSTAMP_SRC_MASK;
1298 
1299 	if (copy_frame_flags)
1300 		mask |= V4L2_BUF_FLAG_KEYFRAME | V4L2_BUF_FLAG_PFRAME |
1301 			V4L2_BUF_FLAG_BFRAME;
1302 
1303 	cap_vb->vb2_buf.timestamp = out_vb->vb2_buf.timestamp;
1304 
1305 	if (out_vb->flags & V4L2_BUF_FLAG_TIMECODE)
1306 		cap_vb->timecode = out_vb->timecode;
1307 	cap_vb->field = out_vb->field;
1308 	cap_vb->flags &= ~mask;
1309 	cap_vb->flags |= out_vb->flags & mask;
1310 	cap_vb->vb2_buf.copied_timestamp = 1;
1311 }
1312 EXPORT_SYMBOL_GPL(v4l2_m2m_buf_copy_metadata);
1313 
1314 void v4l2_m2m_request_queue(struct media_request *req)
1315 {
1316 	struct media_request_object *obj, *obj_safe;
1317 	struct v4l2_m2m_ctx *m2m_ctx = NULL;
1318 
1319 	/*
1320 	 * Queue all objects. Note that buffer objects are at the end of the
1321 	 * objects list, after all other object types. Once buffer objects
1322 	 * are queued, the driver might delete them immediately (if the driver
1323 	 * processes the buffer at once), so we have to use
1324 	 * list_for_each_entry_safe() to handle the case where the object we
1325 	 * queue is deleted.
1326 	 */
1327 	list_for_each_entry_safe(obj, obj_safe, &req->objects, list) {
1328 		struct v4l2_m2m_ctx *m2m_ctx_obj;
1329 		struct vb2_buffer *vb;
1330 
1331 		if (!obj->ops->queue)
1332 			continue;
1333 
1334 		if (vb2_request_object_is_buffer(obj)) {
1335 			/* Sanity checks */
1336 			vb = container_of(obj, struct vb2_buffer, req_obj);
1337 			WARN_ON(!V4L2_TYPE_IS_OUTPUT(vb->vb2_queue->type));
1338 			m2m_ctx_obj = container_of(vb->vb2_queue,
1339 						   struct v4l2_m2m_ctx,
1340 						   out_q_ctx.q);
1341 			WARN_ON(m2m_ctx && m2m_ctx_obj != m2m_ctx);
1342 			m2m_ctx = m2m_ctx_obj;
1343 		}
1344 
1345 		/*
1346 		 * The buffer we queue here can in theory be immediately
1347 		 * unbound, hence the use of list_for_each_entry_safe()
1348 		 * above and why we call the queue op last.
1349 		 */
1350 		obj->ops->queue(obj);
1351 	}
1352 
1353 	WARN_ON(!m2m_ctx);
1354 
1355 	if (m2m_ctx)
1356 		v4l2_m2m_try_schedule(m2m_ctx);
1357 }
1358 EXPORT_SYMBOL_GPL(v4l2_m2m_request_queue);
1359 
1360 /* Videobuf2 ioctl helpers */
1361 
1362 int v4l2_m2m_ioctl_reqbufs(struct file *file, void *priv,
1363 				struct v4l2_requestbuffers *rb)
1364 {
1365 	struct v4l2_fh *fh = file->private_data;
1366 
1367 	return v4l2_m2m_reqbufs(file, fh->m2m_ctx, rb);
1368 }
1369 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_reqbufs);
1370 
1371 int v4l2_m2m_ioctl_create_bufs(struct file *file, void *priv,
1372 				struct v4l2_create_buffers *create)
1373 {
1374 	struct v4l2_fh *fh = file->private_data;
1375 
1376 	return v4l2_m2m_create_bufs(file, fh->m2m_ctx, create);
1377 }
1378 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_create_bufs);
1379 
1380 int v4l2_m2m_ioctl_querybuf(struct file *file, void *priv,
1381 				struct v4l2_buffer *buf)
1382 {
1383 	struct v4l2_fh *fh = file->private_data;
1384 
1385 	return v4l2_m2m_querybuf(file, fh->m2m_ctx, buf);
1386 }
1387 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_querybuf);
1388 
1389 int v4l2_m2m_ioctl_qbuf(struct file *file, void *priv,
1390 				struct v4l2_buffer *buf)
1391 {
1392 	struct v4l2_fh *fh = file->private_data;
1393 
1394 	return v4l2_m2m_qbuf(file, fh->m2m_ctx, buf);
1395 }
1396 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_qbuf);
1397 
1398 int v4l2_m2m_ioctl_dqbuf(struct file *file, void *priv,
1399 				struct v4l2_buffer *buf)
1400 {
1401 	struct v4l2_fh *fh = file->private_data;
1402 
1403 	return v4l2_m2m_dqbuf(file, fh->m2m_ctx, buf);
1404 }
1405 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_dqbuf);
1406 
1407 int v4l2_m2m_ioctl_prepare_buf(struct file *file, void *priv,
1408 			       struct v4l2_buffer *buf)
1409 {
1410 	struct v4l2_fh *fh = file->private_data;
1411 
1412 	return v4l2_m2m_prepare_buf(file, fh->m2m_ctx, buf);
1413 }
1414 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_prepare_buf);
1415 
1416 int v4l2_m2m_ioctl_expbuf(struct file *file, void *priv,
1417 				struct v4l2_exportbuffer *eb)
1418 {
1419 	struct v4l2_fh *fh = file->private_data;
1420 
1421 	return v4l2_m2m_expbuf(file, fh->m2m_ctx, eb);
1422 }
1423 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_expbuf);
1424 
1425 int v4l2_m2m_ioctl_streamon(struct file *file, void *priv,
1426 				enum v4l2_buf_type type)
1427 {
1428 	struct v4l2_fh *fh = file->private_data;
1429 
1430 	return v4l2_m2m_streamon(file, fh->m2m_ctx, type);
1431 }
1432 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_streamon);
1433 
1434 int v4l2_m2m_ioctl_streamoff(struct file *file, void *priv,
1435 				enum v4l2_buf_type type)
1436 {
1437 	struct v4l2_fh *fh = file->private_data;
1438 
1439 	return v4l2_m2m_streamoff(file, fh->m2m_ctx, type);
1440 }
1441 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_streamoff);
1442 
1443 int v4l2_m2m_ioctl_try_encoder_cmd(struct file *file, void *fh,
1444 				   struct v4l2_encoder_cmd *ec)
1445 {
1446 	if (ec->cmd != V4L2_ENC_CMD_STOP && ec->cmd != V4L2_ENC_CMD_START)
1447 		return -EINVAL;
1448 
1449 	ec->flags = 0;
1450 	return 0;
1451 }
1452 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_try_encoder_cmd);
1453 
1454 int v4l2_m2m_ioctl_try_decoder_cmd(struct file *file, void *fh,
1455 				   struct v4l2_decoder_cmd *dc)
1456 {
1457 	if (dc->cmd != V4L2_DEC_CMD_STOP && dc->cmd != V4L2_DEC_CMD_START)
1458 		return -EINVAL;
1459 
1460 	dc->flags = 0;
1461 
1462 	if (dc->cmd == V4L2_DEC_CMD_STOP) {
1463 		dc->stop.pts = 0;
1464 	} else if (dc->cmd == V4L2_DEC_CMD_START) {
1465 		dc->start.speed = 0;
1466 		dc->start.format = V4L2_DEC_START_FMT_NONE;
1467 	}
1468 	return 0;
1469 }
1470 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_try_decoder_cmd);
1471 
1472 /*
1473  * Updates the encoding state on ENC_CMD_STOP/ENC_CMD_START
1474  * Should be called from the encoder driver encoder_cmd() callback
1475  */
1476 int v4l2_m2m_encoder_cmd(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
1477 			 struct v4l2_encoder_cmd *ec)
1478 {
1479 	if (ec->cmd != V4L2_ENC_CMD_STOP && ec->cmd != V4L2_ENC_CMD_START)
1480 		return -EINVAL;
1481 
1482 	if (ec->cmd == V4L2_ENC_CMD_STOP)
1483 		return v4l2_update_last_buf_state(m2m_ctx);
1484 
1485 	if (m2m_ctx->is_draining)
1486 		return -EBUSY;
1487 
1488 	if (m2m_ctx->has_stopped)
1489 		m2m_ctx->has_stopped = false;
1490 
1491 	return 0;
1492 }
1493 EXPORT_SYMBOL_GPL(v4l2_m2m_encoder_cmd);
1494 
1495 /*
1496  * Updates the decoding state on DEC_CMD_STOP/DEC_CMD_START
1497  * Should be called from the decoder driver decoder_cmd() callback
1498  */
1499 int v4l2_m2m_decoder_cmd(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
1500 			 struct v4l2_decoder_cmd *dc)
1501 {
1502 	if (dc->cmd != V4L2_DEC_CMD_STOP && dc->cmd != V4L2_DEC_CMD_START)
1503 		return -EINVAL;
1504 
1505 	if (dc->cmd == V4L2_DEC_CMD_STOP)
1506 		return v4l2_update_last_buf_state(m2m_ctx);
1507 
1508 	if (m2m_ctx->is_draining)
1509 		return -EBUSY;
1510 
1511 	if (m2m_ctx->has_stopped)
1512 		m2m_ctx->has_stopped = false;
1513 
1514 	return 0;
1515 }
1516 EXPORT_SYMBOL_GPL(v4l2_m2m_decoder_cmd);
1517 
1518 int v4l2_m2m_ioctl_encoder_cmd(struct file *file, void *priv,
1519 			       struct v4l2_encoder_cmd *ec)
1520 {
1521 	struct v4l2_fh *fh = file->private_data;
1522 
1523 	return v4l2_m2m_encoder_cmd(file, fh->m2m_ctx, ec);
1524 }
1525 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_encoder_cmd);
1526 
1527 int v4l2_m2m_ioctl_decoder_cmd(struct file *file, void *priv,
1528 			       struct v4l2_decoder_cmd *dc)
1529 {
1530 	struct v4l2_fh *fh = file->private_data;
1531 
1532 	return v4l2_m2m_decoder_cmd(file, fh->m2m_ctx, dc);
1533 }
1534 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_decoder_cmd);
1535 
1536 int v4l2_m2m_ioctl_stateless_try_decoder_cmd(struct file *file, void *fh,
1537 					     struct v4l2_decoder_cmd *dc)
1538 {
1539 	if (dc->cmd != V4L2_DEC_CMD_FLUSH)
1540 		return -EINVAL;
1541 
1542 	dc->flags = 0;
1543 
1544 	return 0;
1545 }
1546 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_stateless_try_decoder_cmd);
1547 
1548 int v4l2_m2m_ioctl_stateless_decoder_cmd(struct file *file, void *priv,
1549 					 struct v4l2_decoder_cmd *dc)
1550 {
1551 	struct v4l2_fh *fh = file->private_data;
1552 	struct vb2_v4l2_buffer *out_vb, *cap_vb;
1553 	struct v4l2_m2m_dev *m2m_dev = fh->m2m_ctx->m2m_dev;
1554 	unsigned long flags;
1555 	int ret;
1556 
1557 	ret = v4l2_m2m_ioctl_stateless_try_decoder_cmd(file, priv, dc);
1558 	if (ret < 0)
1559 		return ret;
1560 
1561 	spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
1562 	out_vb = v4l2_m2m_last_src_buf(fh->m2m_ctx);
1563 	cap_vb = v4l2_m2m_last_dst_buf(fh->m2m_ctx);
1564 
1565 	/*
1566 	 * If there is an out buffer pending, then clear any HOLD flag.
1567 	 *
1568 	 * By clearing this flag we ensure that when this output
1569 	 * buffer is processed any held capture buffer will be released.
1570 	 */
1571 	if (out_vb) {
1572 		out_vb->flags &= ~V4L2_BUF_FLAG_M2M_HOLD_CAPTURE_BUF;
1573 	} else if (cap_vb && cap_vb->is_held) {
1574 		/*
1575 		 * If there were no output buffers, but there is a
1576 		 * capture buffer that is held, then release that
1577 		 * buffer.
1578 		 */
1579 		cap_vb->is_held = false;
1580 		v4l2_m2m_dst_buf_remove(fh->m2m_ctx);
1581 		v4l2_m2m_buf_done(cap_vb, VB2_BUF_STATE_DONE);
1582 	}
1583 	spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
1584 
1585 	return 0;
1586 }
1587 EXPORT_SYMBOL_GPL(v4l2_m2m_ioctl_stateless_decoder_cmd);
1588 
1589 /*
1590  * v4l2_file_operations helpers. It is assumed here same lock is used
1591  * for the output and the capture buffer queue.
1592  */
1593 
1594 int v4l2_m2m_fop_mmap(struct file *file, struct vm_area_struct *vma)
1595 {
1596 	struct v4l2_fh *fh = file->private_data;
1597 
1598 	return v4l2_m2m_mmap(file, fh->m2m_ctx, vma);
1599 }
1600 EXPORT_SYMBOL_GPL(v4l2_m2m_fop_mmap);
1601 
1602 __poll_t v4l2_m2m_fop_poll(struct file *file, poll_table *wait)
1603 {
1604 	struct v4l2_fh *fh = file->private_data;
1605 	struct v4l2_m2m_ctx *m2m_ctx = fh->m2m_ctx;
1606 	__poll_t ret;
1607 
1608 	if (m2m_ctx->q_lock)
1609 		mutex_lock(m2m_ctx->q_lock);
1610 
1611 	ret = v4l2_m2m_poll(file, m2m_ctx, wait);
1612 
1613 	if (m2m_ctx->q_lock)
1614 		mutex_unlock(m2m_ctx->q_lock);
1615 
1616 	return ret;
1617 }
1618 EXPORT_SYMBOL_GPL(v4l2_m2m_fop_poll);
1619 
1620