v4l2-mem2mem.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742
  1. /*
  2. * Memory-to-memory device framework for Video for Linux 2 and videobuf.
  3. *
  4. * Helper functions for devices that use videobuf buffers for both their
  5. * source and destination.
  6. *
  7. * Copyright (c) 2009-2010 Samsung Electronics Co., Ltd.
  8. * Pawel Osciak, <pawel@osciak.com>
  9. * Marek Szyprowski, <m.szyprowski@samsung.com>
  10. *
  11. * This program is free software; you can redistribute it and/or modify
  12. * it under the terms of the GNU General Public License as published by the
  13. * Free Software Foundation; either version 2 of the License, or (at your
  14. * option) any later version.
  15. */
  16. #include <linux/module.h>
  17. #include <linux/sched.h>
  18. #include <linux/slab.h>
  19. #include <media/videobuf2-core.h>
  20. #include <media/v4l2-mem2mem.h>
  21. #include <media/v4l2-dev.h>
  22. #include <media/v4l2-fh.h>
  23. #include <media/v4l2-event.h>
  24. MODULE_DESCRIPTION("Mem to mem device framework for videobuf");
  25. MODULE_AUTHOR("Pawel Osciak, <pawel@osciak.com>");
  26. MODULE_LICENSE("GPL");
  27. static bool debug;
  28. module_param(debug, bool, 0644);
  29. #define dprintk(fmt, arg...) \
  30. do { \
  31. if (debug) \
  32. printk(KERN_DEBUG "%s: " fmt, __func__, ## arg);\
  33. } while (0)
  34. /* Instance is already queued on the job_queue */
  35. #define TRANS_QUEUED (1 << 0)
  36. /* Instance is currently running in hardware */
  37. #define TRANS_RUNNING (1 << 1)
  38. /* Instance is currently aborting */
  39. #define TRANS_ABORT (1 << 2)
  40. /* Offset base for buffers on the destination queue - used to distinguish
  41. * between source and destination buffers when mmapping - they receive the same
  42. * offsets but for different queues */
  43. #define DST_QUEUE_OFF_BASE (1 << 30)
  44. /**
  45. * struct v4l2_m2m_dev - per-device context
  46. * @curr_ctx: currently running instance
  47. * @job_queue: instances queued to run
  48. * @job_spinlock: protects job_queue
  49. * @m2m_ops: driver callbacks
  50. */
  51. struct v4l2_m2m_dev {
  52. struct v4l2_m2m_ctx *curr_ctx;
  53. struct list_head job_queue;
  54. spinlock_t job_spinlock;
  55. const struct v4l2_m2m_ops *m2m_ops;
  56. };
  57. static struct v4l2_m2m_queue_ctx *get_queue_ctx(struct v4l2_m2m_ctx *m2m_ctx,
  58. enum v4l2_buf_type type)
  59. {
  60. if (V4L2_TYPE_IS_OUTPUT(type))
  61. return &m2m_ctx->out_q_ctx;
  62. else
  63. return &m2m_ctx->cap_q_ctx;
  64. }
  65. /**
  66. * v4l2_m2m_get_vq() - return vb2_queue for the given type
  67. */
  68. struct vb2_queue *v4l2_m2m_get_vq(struct v4l2_m2m_ctx *m2m_ctx,
  69. enum v4l2_buf_type type)
  70. {
  71. struct v4l2_m2m_queue_ctx *q_ctx;
  72. q_ctx = get_queue_ctx(m2m_ctx, type);
  73. if (!q_ctx)
  74. return NULL;
  75. return &q_ctx->q;
  76. }
  77. EXPORT_SYMBOL(v4l2_m2m_get_vq);
  78. /**
  79. * v4l2_m2m_next_buf() - return next buffer from the list of ready buffers
  80. */
  81. void *v4l2_m2m_next_buf(struct v4l2_m2m_queue_ctx *q_ctx)
  82. {
  83. struct v4l2_m2m_buffer *b = NULL;
  84. unsigned long flags;
  85. spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
  86. if (list_empty(&q_ctx->rdy_queue)) {
  87. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  88. return NULL;
  89. }
  90. b = list_first_entry(&q_ctx->rdy_queue, struct v4l2_m2m_buffer, list);
  91. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  92. return &b->vb;
  93. }
  94. EXPORT_SYMBOL_GPL(v4l2_m2m_next_buf);
  95. /**
  96. * v4l2_m2m_buf_remove() - take off a buffer from the list of ready buffers and
  97. * return it
  98. */
  99. void *v4l2_m2m_buf_remove(struct v4l2_m2m_queue_ctx *q_ctx)
  100. {
  101. struct v4l2_m2m_buffer *b = NULL;
  102. unsigned long flags;
  103. spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
  104. if (list_empty(&q_ctx->rdy_queue)) {
  105. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  106. return NULL;
  107. }
  108. b = list_first_entry(&q_ctx->rdy_queue, struct v4l2_m2m_buffer, list);
  109. list_del(&b->list);
  110. q_ctx->num_rdy--;
  111. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  112. return &b->vb;
  113. }
  114. EXPORT_SYMBOL_GPL(v4l2_m2m_buf_remove);
  115. /*
  116. * Scheduling handlers
  117. */
  118. /**
  119. * v4l2_m2m_get_curr_priv() - return driver private data for the currently
  120. * running instance or NULL if no instance is running
  121. */
  122. void *v4l2_m2m_get_curr_priv(struct v4l2_m2m_dev *m2m_dev)
  123. {
  124. unsigned long flags;
  125. void *ret = NULL;
  126. spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
  127. if (m2m_dev->curr_ctx)
  128. ret = m2m_dev->curr_ctx->priv;
  129. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  130. return ret;
  131. }
  132. EXPORT_SYMBOL(v4l2_m2m_get_curr_priv);
  133. /**
  134. * v4l2_m2m_try_run() - select next job to perform and run it if possible
  135. *
  136. * Get next transaction (if present) from the waiting jobs list and run it.
  137. */
  138. static void v4l2_m2m_try_run(struct v4l2_m2m_dev *m2m_dev)
  139. {
  140. unsigned long flags;
  141. spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
  142. if (NULL != m2m_dev->curr_ctx) {
  143. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  144. dprintk("Another instance is running, won't run now\n");
  145. return;
  146. }
  147. if (list_empty(&m2m_dev->job_queue)) {
  148. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  149. dprintk("No job pending\n");
  150. return;
  151. }
  152. m2m_dev->curr_ctx = list_first_entry(&m2m_dev->job_queue,
  153. struct v4l2_m2m_ctx, queue);
  154. m2m_dev->curr_ctx->job_flags |= TRANS_RUNNING;
  155. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  156. m2m_dev->m2m_ops->device_run(m2m_dev->curr_ctx->priv);
  157. }
  158. /**
  159. * v4l2_m2m_try_schedule() - check whether an instance is ready to be added to
  160. * the pending job queue and add it if so.
  161. * @m2m_ctx: m2m context assigned to the instance to be checked
  162. *
  163. * There are three basic requirements an instance has to meet to be able to run:
  164. * 1) at least one source buffer has to be queued,
  165. * 2) at least one destination buffer has to be queued,
  166. * 3) streaming has to be on.
  167. *
  168. * If a queue is buffered (for example a decoder hardware ringbuffer that has
  169. * to be drained before doing streamoff), allow scheduling without v4l2 buffers
  170. * on that queue.
  171. *
  172. * There may also be additional, custom requirements. In such case the driver
  173. * should supply a custom callback (job_ready in v4l2_m2m_ops) that should
  174. * return 1 if the instance is ready.
  175. * An example of the above could be an instance that requires more than one
  176. * src/dst buffer per transaction.
  177. */
  178. static void v4l2_m2m_try_schedule(struct v4l2_m2m_ctx *m2m_ctx)
  179. {
  180. struct v4l2_m2m_dev *m2m_dev;
  181. unsigned long flags_job, flags_out, flags_cap;
  182. m2m_dev = m2m_ctx->m2m_dev;
  183. dprintk("Trying to schedule a job for m2m_ctx: %p\n", m2m_ctx);
  184. if (!m2m_ctx->out_q_ctx.q.streaming
  185. || !m2m_ctx->cap_q_ctx.q.streaming) {
  186. dprintk("Streaming needs to be on for both queues\n");
  187. return;
  188. }
  189. spin_lock_irqsave(&m2m_dev->job_spinlock, flags_job);
  190. /* If the context is aborted then don't schedule it */
  191. if (m2m_ctx->job_flags & TRANS_ABORT) {
  192. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  193. dprintk("Aborted context\n");
  194. return;
  195. }
  196. if (m2m_ctx->job_flags & TRANS_QUEUED) {
  197. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  198. dprintk("On job queue already\n");
  199. return;
  200. }
  201. spin_lock_irqsave(&m2m_ctx->out_q_ctx.rdy_spinlock, flags_out);
  202. if (list_empty(&m2m_ctx->out_q_ctx.rdy_queue)
  203. && !m2m_ctx->out_q_ctx.buffered) {
  204. spin_unlock_irqrestore(&m2m_ctx->out_q_ctx.rdy_spinlock,
  205. flags_out);
  206. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  207. dprintk("No input buffers available\n");
  208. return;
  209. }
  210. spin_lock_irqsave(&m2m_ctx->cap_q_ctx.rdy_spinlock, flags_cap);
  211. if (list_empty(&m2m_ctx->cap_q_ctx.rdy_queue)
  212. && !m2m_ctx->cap_q_ctx.buffered) {
  213. spin_unlock_irqrestore(&m2m_ctx->cap_q_ctx.rdy_spinlock,
  214. flags_cap);
  215. spin_unlock_irqrestore(&m2m_ctx->out_q_ctx.rdy_spinlock,
  216. flags_out);
  217. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  218. dprintk("No output buffers available\n");
  219. return;
  220. }
  221. spin_unlock_irqrestore(&m2m_ctx->cap_q_ctx.rdy_spinlock, flags_cap);
  222. spin_unlock_irqrestore(&m2m_ctx->out_q_ctx.rdy_spinlock, flags_out);
  223. if (m2m_dev->m2m_ops->job_ready
  224. && (!m2m_dev->m2m_ops->job_ready(m2m_ctx->priv))) {
  225. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  226. dprintk("Driver not ready\n");
  227. return;
  228. }
  229. list_add_tail(&m2m_ctx->queue, &m2m_dev->job_queue);
  230. m2m_ctx->job_flags |= TRANS_QUEUED;
  231. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  232. v4l2_m2m_try_run(m2m_dev);
  233. }
  234. /**
  235. * v4l2_m2m_cancel_job() - cancel pending jobs for the context
  236. *
  237. * In case of streamoff or release called on any context,
  238. * 1] If the context is currently running, then abort job will be called
  239. * 2] If the context is queued, then the context will be removed from
  240. * the job_queue
  241. */
  242. static void v4l2_m2m_cancel_job(struct v4l2_m2m_ctx *m2m_ctx)
  243. {
  244. struct v4l2_m2m_dev *m2m_dev;
  245. unsigned long flags;
  246. m2m_dev = m2m_ctx->m2m_dev;
  247. spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
  248. m2m_ctx->job_flags |= TRANS_ABORT;
  249. if (m2m_ctx->job_flags & TRANS_RUNNING) {
  250. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  251. m2m_dev->m2m_ops->job_abort(m2m_ctx->priv);
  252. dprintk("m2m_ctx %p running, will wait to complete", m2m_ctx);
  253. wait_event(m2m_ctx->finished,
  254. !(m2m_ctx->job_flags & TRANS_RUNNING));
  255. } else if (m2m_ctx->job_flags & TRANS_QUEUED) {
  256. list_del(&m2m_ctx->queue);
  257. m2m_ctx->job_flags &= ~(TRANS_QUEUED | TRANS_RUNNING);
  258. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  259. dprintk("m2m_ctx: %p had been on queue and was removed\n",
  260. m2m_ctx);
  261. } else {
  262. /* Do nothing, was not on queue/running */
  263. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  264. }
  265. }
  266. /**
  267. * v4l2_m2m_job_finish() - inform the framework that a job has been finished
  268. * and have it clean up
  269. *
  270. * Called by a driver to yield back the device after it has finished with it.
  271. * Should be called as soon as possible after reaching a state which allows
  272. * other instances to take control of the device.
  273. *
  274. * This function has to be called only after device_run() callback has been
  275. * called on the driver. To prevent recursion, it should not be called directly
  276. * from the device_run() callback though.
  277. */
  278. void v4l2_m2m_job_finish(struct v4l2_m2m_dev *m2m_dev,
  279. struct v4l2_m2m_ctx *m2m_ctx)
  280. {
  281. unsigned long flags;
  282. spin_lock_irqsave(&m2m_dev->job_spinlock, flags);
  283. if (!m2m_dev->curr_ctx || m2m_dev->curr_ctx != m2m_ctx) {
  284. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  285. dprintk("Called by an instance not currently running\n");
  286. return;
  287. }
  288. list_del(&m2m_dev->curr_ctx->queue);
  289. m2m_dev->curr_ctx->job_flags &= ~(TRANS_QUEUED | TRANS_RUNNING);
  290. wake_up(&m2m_dev->curr_ctx->finished);
  291. m2m_dev->curr_ctx = NULL;
  292. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags);
  293. /* This instance might have more buffers ready, but since we do not
  294. * allow more than one job on the job_queue per instance, each has
  295. * to be scheduled separately after the previous one finishes. */
  296. v4l2_m2m_try_schedule(m2m_ctx);
  297. v4l2_m2m_try_run(m2m_dev);
  298. }
  299. EXPORT_SYMBOL(v4l2_m2m_job_finish);
  300. /**
  301. * v4l2_m2m_reqbufs() - multi-queue-aware REQBUFS multiplexer
  302. */
  303. int v4l2_m2m_reqbufs(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  304. struct v4l2_requestbuffers *reqbufs)
  305. {
  306. struct vb2_queue *vq;
  307. vq = v4l2_m2m_get_vq(m2m_ctx, reqbufs->type);
  308. return vb2_reqbufs(vq, reqbufs);
  309. }
  310. EXPORT_SYMBOL_GPL(v4l2_m2m_reqbufs);
  311. /**
  312. * v4l2_m2m_querybuf() - multi-queue-aware QUERYBUF multiplexer
  313. *
  314. * See v4l2_m2m_mmap() documentation for details.
  315. */
  316. int v4l2_m2m_querybuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  317. struct v4l2_buffer *buf)
  318. {
  319. struct vb2_queue *vq;
  320. int ret = 0;
  321. unsigned int i;
  322. vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
  323. ret = vb2_querybuf(vq, buf);
  324. /* Adjust MMAP memory offsets for the CAPTURE queue */
  325. if (buf->memory == V4L2_MEMORY_MMAP && !V4L2_TYPE_IS_OUTPUT(vq->type)) {
  326. if (V4L2_TYPE_IS_MULTIPLANAR(vq->type)) {
  327. for (i = 0; i < buf->length; ++i)
  328. buf->m.planes[i].m.mem_offset
  329. += DST_QUEUE_OFF_BASE;
  330. } else {
  331. buf->m.offset += DST_QUEUE_OFF_BASE;
  332. }
  333. }
  334. return ret;
  335. }
  336. EXPORT_SYMBOL_GPL(v4l2_m2m_querybuf);
  337. /**
  338. * v4l2_m2m_qbuf() - enqueue a source or destination buffer, depending on
  339. * the type
  340. */
  341. int v4l2_m2m_qbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  342. struct v4l2_buffer *buf)
  343. {
  344. struct vb2_queue *vq;
  345. int ret;
  346. vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
  347. ret = vb2_qbuf(vq, buf);
  348. if (!ret)
  349. v4l2_m2m_try_schedule(m2m_ctx);
  350. return ret;
  351. }
  352. EXPORT_SYMBOL_GPL(v4l2_m2m_qbuf);
  353. /**
  354. * v4l2_m2m_dqbuf() - dequeue a source or destination buffer, depending on
  355. * the type
  356. */
  357. int v4l2_m2m_dqbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  358. struct v4l2_buffer *buf)
  359. {
  360. struct vb2_queue *vq;
  361. vq = v4l2_m2m_get_vq(m2m_ctx, buf->type);
  362. return vb2_dqbuf(vq, buf, file->f_flags & O_NONBLOCK);
  363. }
  364. EXPORT_SYMBOL_GPL(v4l2_m2m_dqbuf);
  365. /**
  366. * v4l2_m2m_create_bufs() - create a source or destination buffer, depending
  367. * on the type
  368. */
  369. int v4l2_m2m_create_bufs(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  370. struct v4l2_create_buffers *create)
  371. {
  372. struct vb2_queue *vq;
  373. vq = v4l2_m2m_get_vq(m2m_ctx, create->format.type);
  374. return vb2_create_bufs(vq, create);
  375. }
  376. EXPORT_SYMBOL_GPL(v4l2_m2m_create_bufs);
  377. /**
  378. * v4l2_m2m_expbuf() - export a source or destination buffer, depending on
  379. * the type
  380. */
  381. int v4l2_m2m_expbuf(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  382. struct v4l2_exportbuffer *eb)
  383. {
  384. struct vb2_queue *vq;
  385. vq = v4l2_m2m_get_vq(m2m_ctx, eb->type);
  386. return vb2_expbuf(vq, eb);
  387. }
  388. EXPORT_SYMBOL_GPL(v4l2_m2m_expbuf);
  389. /**
  390. * v4l2_m2m_streamon() - turn on streaming for a video queue
  391. */
  392. int v4l2_m2m_streamon(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  393. enum v4l2_buf_type type)
  394. {
  395. struct vb2_queue *vq;
  396. int ret;
  397. vq = v4l2_m2m_get_vq(m2m_ctx, type);
  398. ret = vb2_streamon(vq, type);
  399. if (!ret)
  400. v4l2_m2m_try_schedule(m2m_ctx);
  401. return ret;
  402. }
  403. EXPORT_SYMBOL_GPL(v4l2_m2m_streamon);
  404. /**
  405. * v4l2_m2m_streamoff() - turn off streaming for a video queue
  406. */
  407. int v4l2_m2m_streamoff(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  408. enum v4l2_buf_type type)
  409. {
  410. struct v4l2_m2m_dev *m2m_dev;
  411. struct v4l2_m2m_queue_ctx *q_ctx;
  412. unsigned long flags_job, flags;
  413. int ret;
  414. /* wait until the current context is dequeued from job_queue */
  415. v4l2_m2m_cancel_job(m2m_ctx);
  416. q_ctx = get_queue_ctx(m2m_ctx, type);
  417. ret = vb2_streamoff(&q_ctx->q, type);
  418. if (ret)
  419. return ret;
  420. m2m_dev = m2m_ctx->m2m_dev;
  421. spin_lock_irqsave(&m2m_dev->job_spinlock, flags_job);
  422. /* We should not be scheduled anymore, since we're dropping a queue. */
  423. if (m2m_ctx->job_flags & TRANS_QUEUED)
  424. list_del(&m2m_ctx->queue);
  425. m2m_ctx->job_flags = 0;
  426. spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
  427. /* Drop queue, since streamoff returns device to the same state as after
  428. * calling reqbufs. */
  429. INIT_LIST_HEAD(&q_ctx->rdy_queue);
  430. q_ctx->num_rdy = 0;
  431. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  432. if (m2m_dev->curr_ctx == m2m_ctx) {
  433. m2m_dev->curr_ctx = NULL;
  434. wake_up(&m2m_ctx->finished);
  435. }
  436. spin_unlock_irqrestore(&m2m_dev->job_spinlock, flags_job);
  437. return 0;
  438. }
  439. EXPORT_SYMBOL_GPL(v4l2_m2m_streamoff);
  440. /**
  441. * v4l2_m2m_poll() - poll replacement, for destination buffers only
  442. *
  443. * Call from the driver's poll() function. Will poll both queues. If a buffer
  444. * is available to dequeue (with dqbuf) from the source queue, this will
  445. * indicate that a non-blocking write can be performed, while read will be
  446. * returned in case of the destination queue.
  447. */
  448. unsigned int v4l2_m2m_poll(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  449. struct poll_table_struct *wait)
  450. {
  451. struct video_device *vfd = video_devdata(file);
  452. unsigned long req_events = poll_requested_events(wait);
  453. struct vb2_queue *src_q, *dst_q;
  454. struct vb2_buffer *src_vb = NULL, *dst_vb = NULL;
  455. unsigned int rc = 0;
  456. unsigned long flags;
  457. if (test_bit(V4L2_FL_USES_V4L2_FH, &vfd->flags)) {
  458. struct v4l2_fh *fh = file->private_data;
  459. if (v4l2_event_pending(fh))
  460. rc = POLLPRI;
  461. else if (req_events & POLLPRI)
  462. poll_wait(file, &fh->wait, wait);
  463. if (!(req_events & (POLLOUT | POLLWRNORM | POLLIN | POLLRDNORM)))
  464. return rc;
  465. }
  466. src_q = v4l2_m2m_get_src_vq(m2m_ctx);
  467. dst_q = v4l2_m2m_get_dst_vq(m2m_ctx);
  468. /*
  469. * There has to be at least one buffer queued on each queued_list, which
  470. * means either in driver already or waiting for driver to claim it
  471. * and start processing.
  472. */
  473. if ((!src_q->streaming || list_empty(&src_q->queued_list))
  474. && (!dst_q->streaming || list_empty(&dst_q->queued_list))) {
  475. rc |= POLLERR;
  476. goto end;
  477. }
  478. if (m2m_ctx->m2m_dev->m2m_ops->unlock)
  479. m2m_ctx->m2m_dev->m2m_ops->unlock(m2m_ctx->priv);
  480. if (list_empty(&src_q->done_list))
  481. poll_wait(file, &src_q->done_wq, wait);
  482. if (list_empty(&dst_q->done_list))
  483. poll_wait(file, &dst_q->done_wq, wait);
  484. if (m2m_ctx->m2m_dev->m2m_ops->lock)
  485. m2m_ctx->m2m_dev->m2m_ops->lock(m2m_ctx->priv);
  486. spin_lock_irqsave(&src_q->done_lock, flags);
  487. if (!list_empty(&src_q->done_list))
  488. src_vb = list_first_entry(&src_q->done_list, struct vb2_buffer,
  489. done_entry);
  490. if (src_vb && (src_vb->state == VB2_BUF_STATE_DONE
  491. || src_vb->state == VB2_BUF_STATE_ERROR))
  492. rc |= POLLOUT | POLLWRNORM;
  493. spin_unlock_irqrestore(&src_q->done_lock, flags);
  494. spin_lock_irqsave(&dst_q->done_lock, flags);
  495. if (!list_empty(&dst_q->done_list))
  496. dst_vb = list_first_entry(&dst_q->done_list, struct vb2_buffer,
  497. done_entry);
  498. if (dst_vb && (dst_vb->state == VB2_BUF_STATE_DONE
  499. || dst_vb->state == VB2_BUF_STATE_ERROR))
  500. rc |= POLLIN | POLLRDNORM;
  501. spin_unlock_irqrestore(&dst_q->done_lock, flags);
  502. end:
  503. return rc;
  504. }
  505. EXPORT_SYMBOL_GPL(v4l2_m2m_poll);
  506. /**
  507. * v4l2_m2m_mmap() - source and destination queues-aware mmap multiplexer
  508. *
  509. * Call from driver's mmap() function. Will handle mmap() for both queues
  510. * seamlessly for videobuffer, which will receive normal per-queue offsets and
  511. * proper videobuf queue pointers. The differentiation is made outside videobuf
  512. * by adding a predefined offset to buffers from one of the queues and
  513. * subtracting it before passing it back to videobuf. Only drivers (and
  514. * thus applications) receive modified offsets.
  515. */
  516. int v4l2_m2m_mmap(struct file *file, struct v4l2_m2m_ctx *m2m_ctx,
  517. struct vm_area_struct *vma)
  518. {
  519. unsigned long offset = vma->vm_pgoff << PAGE_SHIFT;
  520. struct vb2_queue *vq;
  521. if (offset < DST_QUEUE_OFF_BASE) {
  522. vq = v4l2_m2m_get_src_vq(m2m_ctx);
  523. } else {
  524. vq = v4l2_m2m_get_dst_vq(m2m_ctx);
  525. vma->vm_pgoff -= (DST_QUEUE_OFF_BASE >> PAGE_SHIFT);
  526. }
  527. return vb2_mmap(vq, vma);
  528. }
  529. EXPORT_SYMBOL(v4l2_m2m_mmap);
  530. /**
  531. * v4l2_m2m_init() - initialize per-driver m2m data
  532. *
  533. * Usually called from driver's probe() function.
  534. */
  535. struct v4l2_m2m_dev *v4l2_m2m_init(const struct v4l2_m2m_ops *m2m_ops)
  536. {
  537. struct v4l2_m2m_dev *m2m_dev;
  538. if (!m2m_ops || WARN_ON(!m2m_ops->device_run) ||
  539. WARN_ON(!m2m_ops->job_abort))
  540. return ERR_PTR(-EINVAL);
  541. m2m_dev = kzalloc(sizeof *m2m_dev, GFP_KERNEL);
  542. if (!m2m_dev)
  543. return ERR_PTR(-ENOMEM);
  544. m2m_dev->curr_ctx = NULL;
  545. m2m_dev->m2m_ops = m2m_ops;
  546. INIT_LIST_HEAD(&m2m_dev->job_queue);
  547. spin_lock_init(&m2m_dev->job_spinlock);
  548. return m2m_dev;
  549. }
  550. EXPORT_SYMBOL_GPL(v4l2_m2m_init);
  551. /**
  552. * v4l2_m2m_release() - cleans up and frees a m2m_dev structure
  553. *
  554. * Usually called from driver's remove() function.
  555. */
  556. void v4l2_m2m_release(struct v4l2_m2m_dev *m2m_dev)
  557. {
  558. kfree(m2m_dev);
  559. }
  560. EXPORT_SYMBOL_GPL(v4l2_m2m_release);
  561. /**
  562. * v4l2_m2m_ctx_init() - allocate and initialize a m2m context
  563. * @priv - driver's instance private data
  564. * @m2m_dev - a previously initialized m2m_dev struct
  565. * @vq_init - a callback for queue type-specific initialization function to be
  566. * used for initializing videobuf_queues
  567. *
  568. * Usually called from driver's open() function.
  569. */
  570. struct v4l2_m2m_ctx *v4l2_m2m_ctx_init(struct v4l2_m2m_dev *m2m_dev,
  571. void *drv_priv,
  572. int (*queue_init)(void *priv, struct vb2_queue *src_vq, struct vb2_queue *dst_vq))
  573. {
  574. struct v4l2_m2m_ctx *m2m_ctx;
  575. struct v4l2_m2m_queue_ctx *out_q_ctx, *cap_q_ctx;
  576. int ret;
  577. m2m_ctx = kzalloc(sizeof *m2m_ctx, GFP_KERNEL);
  578. if (!m2m_ctx)
  579. return ERR_PTR(-ENOMEM);
  580. m2m_ctx->priv = drv_priv;
  581. m2m_ctx->m2m_dev = m2m_dev;
  582. init_waitqueue_head(&m2m_ctx->finished);
  583. out_q_ctx = &m2m_ctx->out_q_ctx;
  584. cap_q_ctx = &m2m_ctx->cap_q_ctx;
  585. INIT_LIST_HEAD(&out_q_ctx->rdy_queue);
  586. INIT_LIST_HEAD(&cap_q_ctx->rdy_queue);
  587. spin_lock_init(&out_q_ctx->rdy_spinlock);
  588. spin_lock_init(&cap_q_ctx->rdy_spinlock);
  589. INIT_LIST_HEAD(&m2m_ctx->queue);
  590. ret = queue_init(drv_priv, &out_q_ctx->q, &cap_q_ctx->q);
  591. if (ret)
  592. goto err;
  593. return m2m_ctx;
  594. err:
  595. kfree(m2m_ctx);
  596. return ERR_PTR(ret);
  597. }
  598. EXPORT_SYMBOL_GPL(v4l2_m2m_ctx_init);
  599. /**
  600. * v4l2_m2m_ctx_release() - release m2m context
  601. *
  602. * Usually called from driver's release() function.
  603. */
  604. void v4l2_m2m_ctx_release(struct v4l2_m2m_ctx *m2m_ctx)
  605. {
  606. /* wait until the current context is dequeued from job_queue */
  607. v4l2_m2m_cancel_job(m2m_ctx);
  608. vb2_queue_release(&m2m_ctx->cap_q_ctx.q);
  609. vb2_queue_release(&m2m_ctx->out_q_ctx.q);
  610. kfree(m2m_ctx);
  611. }
  612. EXPORT_SYMBOL_GPL(v4l2_m2m_ctx_release);
  613. /**
  614. * v4l2_m2m_buf_queue() - add a buffer to the proper ready buffers list.
  615. *
  616. * Call from buf_queue(), videobuf_queue_ops callback.
  617. */
  618. void v4l2_m2m_buf_queue(struct v4l2_m2m_ctx *m2m_ctx, struct vb2_buffer *vb)
  619. {
  620. struct v4l2_m2m_buffer *b = container_of(vb, struct v4l2_m2m_buffer, vb);
  621. struct v4l2_m2m_queue_ctx *q_ctx;
  622. unsigned long flags;
  623. q_ctx = get_queue_ctx(m2m_ctx, vb->vb2_queue->type);
  624. if (!q_ctx)
  625. return;
  626. spin_lock_irqsave(&q_ctx->rdy_spinlock, flags);
  627. list_add_tail(&b->list, &q_ctx->rdy_queue);
  628. q_ctx->num_rdy++;
  629. spin_unlock_irqrestore(&q_ctx->rdy_spinlock, flags);
  630. }
  631. EXPORT_SYMBOL_GPL(v4l2_m2m_buf_queue);