videobuf-core.c 27 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211
  1. /*
  2. * generic helper functions for handling video4linux capture buffers
  3. *
  4. * (c) 2007 Mauro Carvalho Chehab, <mchehab@infradead.org>
  5. *
  6. * Highly based on video-buf written originally by:
  7. * (c) 2001,02 Gerd Knorr <kraxel@bytesex.org>
  8. * (c) 2006 Mauro Carvalho Chehab, <mchehab@infradead.org>
  9. * (c) 2006 Ted Walther and John Sokol
  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
  13. * the Free Software Foundation; either version 2
  14. */
  15. #include <linux/init.h>
  16. #include <linux/module.h>
  17. #include <linux/moduleparam.h>
  18. #include <linux/mm.h>
  19. #include <linux/sched.h>
  20. #include <linux/slab.h>
  21. #include <linux/interrupt.h>
  22. #include <media/videobuf-core.h>
  23. #define MAGIC_BUFFER 0x20070728
  24. #define MAGIC_CHECK(is, should) \
  25. do { \
  26. if (unlikely((is) != (should))) { \
  27. printk(KERN_ERR \
  28. "magic mismatch: %x (expected %x)\n", \
  29. is, should); \
  30. BUG(); \
  31. } \
  32. } while (0)
  33. static int debug;
  34. module_param(debug, int, 0644);
  35. MODULE_DESCRIPTION("helper module to manage video4linux buffers");
  36. MODULE_AUTHOR("Mauro Carvalho Chehab <mchehab@infradead.org>");
  37. MODULE_LICENSE("GPL");
  38. #define dprintk(level, fmt, arg...) \
  39. do { \
  40. if (debug >= level) \
  41. printk(KERN_DEBUG "vbuf: " fmt, ## arg); \
  42. } while (0)
  43. /* --------------------------------------------------------------------- */
  44. #define CALL(q, f, arg...) \
  45. ((q->int_ops->f) ? q->int_ops->f(arg) : 0)
  46. struct videobuf_buffer *videobuf_alloc_vb(struct videobuf_queue *q)
  47. {
  48. struct videobuf_buffer *vb;
  49. BUG_ON(q->msize < sizeof(*vb));
  50. if (!q->int_ops || !q->int_ops->alloc_vb) {
  51. printk(KERN_ERR "No specific ops defined!\n");
  52. BUG();
  53. }
  54. vb = q->int_ops->alloc_vb(q->msize);
  55. if (NULL != vb) {
  56. init_waitqueue_head(&vb->done);
  57. vb->magic = MAGIC_BUFFER;
  58. }
  59. return vb;
  60. }
  61. EXPORT_SYMBOL_GPL(videobuf_alloc_vb);
  62. #define WAITON_CONDITION (vb->state != VIDEOBUF_ACTIVE &&\
  63. vb->state != VIDEOBUF_QUEUED)
  64. int videobuf_waiton(struct videobuf_buffer *vb, int non_blocking, int intr)
  65. {
  66. MAGIC_CHECK(vb->magic, MAGIC_BUFFER);
  67. if (non_blocking) {
  68. if (WAITON_CONDITION)
  69. return 0;
  70. else
  71. return -EAGAIN;
  72. }
  73. if (intr)
  74. return wait_event_interruptible(vb->done, WAITON_CONDITION);
  75. else
  76. wait_event(vb->done, WAITON_CONDITION);
  77. return 0;
  78. }
  79. EXPORT_SYMBOL_GPL(videobuf_waiton);
  80. int videobuf_iolock(struct videobuf_queue *q, struct videobuf_buffer *vb,
  81. struct v4l2_framebuffer *fbuf)
  82. {
  83. MAGIC_CHECK(vb->magic, MAGIC_BUFFER);
  84. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  85. return CALL(q, iolock, q, vb, fbuf);
  86. }
  87. EXPORT_SYMBOL_GPL(videobuf_iolock);
  88. void *videobuf_queue_to_vaddr(struct videobuf_queue *q,
  89. struct videobuf_buffer *buf)
  90. {
  91. if (q->int_ops->vaddr)
  92. return q->int_ops->vaddr(buf);
  93. return NULL;
  94. }
  95. EXPORT_SYMBOL_GPL(videobuf_queue_to_vaddr);
  96. /* --------------------------------------------------------------------- */
  97. void videobuf_queue_core_init(struct videobuf_queue *q,
  98. const struct videobuf_queue_ops *ops,
  99. struct device *dev,
  100. spinlock_t *irqlock,
  101. enum v4l2_buf_type type,
  102. enum v4l2_field field,
  103. unsigned int msize,
  104. void *priv,
  105. struct videobuf_qtype_ops *int_ops)
  106. {
  107. BUG_ON(!q);
  108. memset(q, 0, sizeof(*q));
  109. q->irqlock = irqlock;
  110. q->dev = dev;
  111. q->type = type;
  112. q->field = field;
  113. q->msize = msize;
  114. q->ops = ops;
  115. q->priv_data = priv;
  116. q->int_ops = int_ops;
  117. /* All buffer operations are mandatory */
  118. BUG_ON(!q->ops->buf_setup);
  119. BUG_ON(!q->ops->buf_prepare);
  120. BUG_ON(!q->ops->buf_queue);
  121. BUG_ON(!q->ops->buf_release);
  122. /* Lock is mandatory for queue_cancel to work */
  123. BUG_ON(!irqlock);
  124. /* Having implementations for abstract methods are mandatory */
  125. BUG_ON(!q->int_ops);
  126. mutex_init(&q->vb_lock);
  127. init_waitqueue_head(&q->wait);
  128. INIT_LIST_HEAD(&q->stream);
  129. }
  130. EXPORT_SYMBOL_GPL(videobuf_queue_core_init);
  131. /* Locking: Only usage in bttv unsafe find way to remove */
  132. int videobuf_queue_is_busy(struct videobuf_queue *q)
  133. {
  134. int i;
  135. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  136. if (q->streaming) {
  137. dprintk(1, "busy: streaming active\n");
  138. return 1;
  139. }
  140. if (q->reading) {
  141. dprintk(1, "busy: pending read #1\n");
  142. return 1;
  143. }
  144. if (q->read_buf) {
  145. dprintk(1, "busy: pending read #2\n");
  146. return 1;
  147. }
  148. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  149. if (NULL == q->bufs[i])
  150. continue;
  151. if (q->bufs[i]->map) {
  152. dprintk(1, "busy: buffer #%d mapped\n", i);
  153. return 1;
  154. }
  155. if (q->bufs[i]->state == VIDEOBUF_QUEUED) {
  156. dprintk(1, "busy: buffer #%d queued\n", i);
  157. return 1;
  158. }
  159. if (q->bufs[i]->state == VIDEOBUF_ACTIVE) {
  160. dprintk(1, "busy: buffer #%d avtive\n", i);
  161. return 1;
  162. }
  163. }
  164. return 0;
  165. }
  166. EXPORT_SYMBOL_GPL(videobuf_queue_is_busy);
  167. /**
  168. * __videobuf_free() - free all the buffers and their control structures
  169. *
  170. * This function can only be called if streaming/reading is off, i.e. no buffers
  171. * are under control of the driver.
  172. */
  173. /* Locking: Caller holds q->vb_lock */
  174. static int __videobuf_free(struct videobuf_queue *q)
  175. {
  176. int i;
  177. dprintk(1, "%s\n", __func__);
  178. if (!q)
  179. return 0;
  180. if (q->streaming || q->reading) {
  181. dprintk(1, "Cannot free buffers when streaming or reading\n");
  182. return -EBUSY;
  183. }
  184. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  185. for (i = 0; i < VIDEO_MAX_FRAME; i++)
  186. if (q->bufs[i] && q->bufs[i]->map) {
  187. dprintk(1, "Cannot free mmapped buffers\n");
  188. return -EBUSY;
  189. }
  190. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  191. if (NULL == q->bufs[i])
  192. continue;
  193. q->ops->buf_release(q, q->bufs[i]);
  194. kfree(q->bufs[i]);
  195. q->bufs[i] = NULL;
  196. }
  197. return 0;
  198. }
  199. /* Locking: Caller holds q->vb_lock */
  200. void videobuf_queue_cancel(struct videobuf_queue *q)
  201. {
  202. unsigned long flags = 0;
  203. int i;
  204. q->streaming = 0;
  205. q->reading = 0;
  206. wake_up_interruptible_sync(&q->wait);
  207. /* remove queued buffers from list */
  208. spin_lock_irqsave(q->irqlock, flags);
  209. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  210. if (NULL == q->bufs[i])
  211. continue;
  212. if (q->bufs[i]->state == VIDEOBUF_QUEUED) {
  213. list_del(&q->bufs[i]->queue);
  214. q->bufs[i]->state = VIDEOBUF_ERROR;
  215. wake_up_all(&q->bufs[i]->done);
  216. }
  217. }
  218. spin_unlock_irqrestore(q->irqlock, flags);
  219. /* free all buffers + clear queue */
  220. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  221. if (NULL == q->bufs[i])
  222. continue;
  223. q->ops->buf_release(q, q->bufs[i]);
  224. }
  225. INIT_LIST_HEAD(&q->stream);
  226. }
  227. EXPORT_SYMBOL_GPL(videobuf_queue_cancel);
  228. /* --------------------------------------------------------------------- */
  229. /* Locking: Caller holds q->vb_lock */
  230. enum v4l2_field videobuf_next_field(struct videobuf_queue *q)
  231. {
  232. enum v4l2_field field = q->field;
  233. BUG_ON(V4L2_FIELD_ANY == field);
  234. if (V4L2_FIELD_ALTERNATE == field) {
  235. if (V4L2_FIELD_TOP == q->last) {
  236. field = V4L2_FIELD_BOTTOM;
  237. q->last = V4L2_FIELD_BOTTOM;
  238. } else {
  239. field = V4L2_FIELD_TOP;
  240. q->last = V4L2_FIELD_TOP;
  241. }
  242. }
  243. return field;
  244. }
  245. EXPORT_SYMBOL_GPL(videobuf_next_field);
  246. /* Locking: Caller holds q->vb_lock */
  247. static void videobuf_status(struct videobuf_queue *q, struct v4l2_buffer *b,
  248. struct videobuf_buffer *vb, enum v4l2_buf_type type)
  249. {
  250. MAGIC_CHECK(vb->magic, MAGIC_BUFFER);
  251. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  252. b->index = vb->i;
  253. b->type = type;
  254. b->memory = vb->memory;
  255. switch (b->memory) {
  256. case V4L2_MEMORY_MMAP:
  257. b->m.offset = vb->boff;
  258. b->length = vb->bsize;
  259. break;
  260. case V4L2_MEMORY_USERPTR:
  261. b->m.userptr = vb->baddr;
  262. b->length = vb->bsize;
  263. break;
  264. case V4L2_MEMORY_OVERLAY:
  265. b->m.offset = vb->boff;
  266. break;
  267. }
  268. b->flags = 0;
  269. if (vb->map)
  270. b->flags |= V4L2_BUF_FLAG_MAPPED;
  271. switch (vb->state) {
  272. case VIDEOBUF_PREPARED:
  273. case VIDEOBUF_QUEUED:
  274. case VIDEOBUF_ACTIVE:
  275. b->flags |= V4L2_BUF_FLAG_QUEUED;
  276. break;
  277. case VIDEOBUF_ERROR:
  278. b->flags |= V4L2_BUF_FLAG_ERROR;
  279. /* fall through */
  280. case VIDEOBUF_DONE:
  281. b->flags |= V4L2_BUF_FLAG_DONE;
  282. break;
  283. case VIDEOBUF_NEEDS_INIT:
  284. case VIDEOBUF_IDLE:
  285. /* nothing */
  286. break;
  287. }
  288. if (vb->input != UNSET) {
  289. b->flags |= V4L2_BUF_FLAG_INPUT;
  290. b->input = vb->input;
  291. }
  292. b->field = vb->field;
  293. b->timestamp = vb->ts;
  294. b->bytesused = vb->size;
  295. b->sequence = vb->field_count >> 1;
  296. }
  297. int videobuf_mmap_free(struct videobuf_queue *q)
  298. {
  299. int ret;
  300. mutex_lock(&q->vb_lock);
  301. ret = __videobuf_free(q);
  302. mutex_unlock(&q->vb_lock);
  303. return ret;
  304. }
  305. EXPORT_SYMBOL_GPL(videobuf_mmap_free);
  306. /* Locking: Caller holds q->vb_lock */
  307. int __videobuf_mmap_setup(struct videobuf_queue *q,
  308. unsigned int bcount, unsigned int bsize,
  309. enum v4l2_memory memory)
  310. {
  311. unsigned int i;
  312. int err;
  313. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  314. err = __videobuf_free(q);
  315. if (0 != err)
  316. return err;
  317. /* Allocate and initialize buffers */
  318. for (i = 0; i < bcount; i++) {
  319. q->bufs[i] = videobuf_alloc_vb(q);
  320. if (NULL == q->bufs[i])
  321. break;
  322. q->bufs[i]->i = i;
  323. q->bufs[i]->input = UNSET;
  324. q->bufs[i]->memory = memory;
  325. q->bufs[i]->bsize = bsize;
  326. switch (memory) {
  327. case V4L2_MEMORY_MMAP:
  328. q->bufs[i]->boff = PAGE_ALIGN(bsize) * i;
  329. break;
  330. case V4L2_MEMORY_USERPTR:
  331. case V4L2_MEMORY_OVERLAY:
  332. /* nothing */
  333. break;
  334. }
  335. }
  336. if (!i)
  337. return -ENOMEM;
  338. dprintk(1, "mmap setup: %d buffers, %d bytes each\n", i, bsize);
  339. return i;
  340. }
  341. EXPORT_SYMBOL_GPL(__videobuf_mmap_setup);
  342. int videobuf_mmap_setup(struct videobuf_queue *q,
  343. unsigned int bcount, unsigned int bsize,
  344. enum v4l2_memory memory)
  345. {
  346. int ret;
  347. mutex_lock(&q->vb_lock);
  348. ret = __videobuf_mmap_setup(q, bcount, bsize, memory);
  349. mutex_unlock(&q->vb_lock);
  350. return ret;
  351. }
  352. EXPORT_SYMBOL_GPL(videobuf_mmap_setup);
  353. int videobuf_reqbufs(struct videobuf_queue *q,
  354. struct v4l2_requestbuffers *req)
  355. {
  356. unsigned int size, count;
  357. int retval;
  358. if (req->count < 1) {
  359. dprintk(1, "reqbufs: count invalid (%d)\n", req->count);
  360. return -EINVAL;
  361. }
  362. if (req->memory != V4L2_MEMORY_MMAP &&
  363. req->memory != V4L2_MEMORY_USERPTR &&
  364. req->memory != V4L2_MEMORY_OVERLAY) {
  365. dprintk(1, "reqbufs: memory type invalid\n");
  366. return -EINVAL;
  367. }
  368. mutex_lock(&q->vb_lock);
  369. if (req->type != q->type) {
  370. dprintk(1, "reqbufs: queue type invalid\n");
  371. retval = -EINVAL;
  372. goto done;
  373. }
  374. if (q->streaming) {
  375. dprintk(1, "reqbufs: streaming already exists\n");
  376. retval = -EBUSY;
  377. goto done;
  378. }
  379. if (!list_empty(&q->stream)) {
  380. dprintk(1, "reqbufs: stream running\n");
  381. retval = -EBUSY;
  382. goto done;
  383. }
  384. count = req->count;
  385. if (count > VIDEO_MAX_FRAME)
  386. count = VIDEO_MAX_FRAME;
  387. size = 0;
  388. q->ops->buf_setup(q, &count, &size);
  389. dprintk(1, "reqbufs: bufs=%d, size=0x%x [%u pages total]\n",
  390. count, size,
  391. (unsigned int)((count * PAGE_ALIGN(size)) >> PAGE_SHIFT));
  392. retval = __videobuf_mmap_setup(q, count, size, req->memory);
  393. if (retval < 0) {
  394. dprintk(1, "reqbufs: mmap setup returned %d\n", retval);
  395. goto done;
  396. }
  397. req->count = retval;
  398. retval = 0;
  399. done:
  400. mutex_unlock(&q->vb_lock);
  401. return retval;
  402. }
  403. EXPORT_SYMBOL_GPL(videobuf_reqbufs);
  404. int videobuf_querybuf(struct videobuf_queue *q, struct v4l2_buffer *b)
  405. {
  406. int ret = -EINVAL;
  407. mutex_lock(&q->vb_lock);
  408. if (unlikely(b->type != q->type)) {
  409. dprintk(1, "querybuf: Wrong type.\n");
  410. goto done;
  411. }
  412. if (unlikely(b->index >= VIDEO_MAX_FRAME)) {
  413. dprintk(1, "querybuf: index out of range.\n");
  414. goto done;
  415. }
  416. if (unlikely(NULL == q->bufs[b->index])) {
  417. dprintk(1, "querybuf: buffer is null.\n");
  418. goto done;
  419. }
  420. videobuf_status(q, b, q->bufs[b->index], q->type);
  421. ret = 0;
  422. done:
  423. mutex_unlock(&q->vb_lock);
  424. return ret;
  425. }
  426. EXPORT_SYMBOL_GPL(videobuf_querybuf);
  427. int videobuf_qbuf(struct videobuf_queue *q, struct v4l2_buffer *b)
  428. {
  429. struct videobuf_buffer *buf;
  430. enum v4l2_field field;
  431. unsigned long flags = 0;
  432. int retval;
  433. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  434. if (b->memory == V4L2_MEMORY_MMAP)
  435. down_read(&current->mm->mmap_sem);
  436. mutex_lock(&q->vb_lock);
  437. retval = -EBUSY;
  438. if (q->reading) {
  439. dprintk(1, "qbuf: Reading running...\n");
  440. goto done;
  441. }
  442. retval = -EINVAL;
  443. if (b->type != q->type) {
  444. dprintk(1, "qbuf: Wrong type.\n");
  445. goto done;
  446. }
  447. if (b->index >= VIDEO_MAX_FRAME) {
  448. dprintk(1, "qbuf: index out of range.\n");
  449. goto done;
  450. }
  451. buf = q->bufs[b->index];
  452. if (NULL == buf) {
  453. dprintk(1, "qbuf: buffer is null.\n");
  454. goto done;
  455. }
  456. MAGIC_CHECK(buf->magic, MAGIC_BUFFER);
  457. if (buf->memory != b->memory) {
  458. dprintk(1, "qbuf: memory type is wrong.\n");
  459. goto done;
  460. }
  461. if (buf->state != VIDEOBUF_NEEDS_INIT && buf->state != VIDEOBUF_IDLE) {
  462. dprintk(1, "qbuf: buffer is already queued or active.\n");
  463. goto done;
  464. }
  465. if (b->flags & V4L2_BUF_FLAG_INPUT) {
  466. if (b->input >= q->inputs) {
  467. dprintk(1, "qbuf: wrong input.\n");
  468. goto done;
  469. }
  470. buf->input = b->input;
  471. } else {
  472. buf->input = UNSET;
  473. }
  474. switch (b->memory) {
  475. case V4L2_MEMORY_MMAP:
  476. if (0 == buf->baddr) {
  477. dprintk(1, "qbuf: mmap requested "
  478. "but buffer addr is zero!\n");
  479. goto done;
  480. }
  481. if (q->type == V4L2_BUF_TYPE_VIDEO_OUTPUT
  482. || q->type == V4L2_BUF_TYPE_VBI_OUTPUT
  483. || q->type == V4L2_BUF_TYPE_SLICED_VBI_OUTPUT) {
  484. buf->size = b->bytesused;
  485. buf->field = b->field;
  486. buf->ts = b->timestamp;
  487. }
  488. break;
  489. case V4L2_MEMORY_USERPTR:
  490. if (b->length < buf->bsize) {
  491. dprintk(1, "qbuf: buffer length is not enough\n");
  492. goto done;
  493. }
  494. if (VIDEOBUF_NEEDS_INIT != buf->state &&
  495. buf->baddr != b->m.userptr)
  496. q->ops->buf_release(q, buf);
  497. buf->baddr = b->m.userptr;
  498. break;
  499. case V4L2_MEMORY_OVERLAY:
  500. buf->boff = b->m.offset;
  501. break;
  502. default:
  503. dprintk(1, "qbuf: wrong memory type\n");
  504. goto done;
  505. }
  506. dprintk(1, "qbuf: requesting next field\n");
  507. field = videobuf_next_field(q);
  508. retval = q->ops->buf_prepare(q, buf, field);
  509. if (0 != retval) {
  510. dprintk(1, "qbuf: buffer_prepare returned %d\n", retval);
  511. goto done;
  512. }
  513. list_add_tail(&buf->stream, &q->stream);
  514. if (q->streaming) {
  515. spin_lock_irqsave(q->irqlock, flags);
  516. q->ops->buf_queue(q, buf);
  517. spin_unlock_irqrestore(q->irqlock, flags);
  518. }
  519. dprintk(1, "qbuf: succeeded\n");
  520. retval = 0;
  521. wake_up_interruptible_sync(&q->wait);
  522. done:
  523. mutex_unlock(&q->vb_lock);
  524. if (b->memory == V4L2_MEMORY_MMAP)
  525. up_read(&current->mm->mmap_sem);
  526. return retval;
  527. }
  528. EXPORT_SYMBOL_GPL(videobuf_qbuf);
  529. /* Locking: Caller holds q->vb_lock */
  530. static int stream_next_buffer_check_queue(struct videobuf_queue *q, int noblock)
  531. {
  532. int retval;
  533. checks:
  534. if (!q->streaming) {
  535. dprintk(1, "next_buffer: Not streaming\n");
  536. retval = -EINVAL;
  537. goto done;
  538. }
  539. if (list_empty(&q->stream)) {
  540. if (noblock) {
  541. retval = -EAGAIN;
  542. dprintk(2, "next_buffer: no buffers to dequeue\n");
  543. goto done;
  544. } else {
  545. dprintk(2, "next_buffer: waiting on buffer\n");
  546. /* Drop lock to avoid deadlock with qbuf */
  547. mutex_unlock(&q->vb_lock);
  548. /* Checking list_empty and streaming is safe without
  549. * locks because we goto checks to validate while
  550. * holding locks before proceeding */
  551. retval = wait_event_interruptible(q->wait,
  552. !list_empty(&q->stream) || !q->streaming);
  553. mutex_lock(&q->vb_lock);
  554. if (retval)
  555. goto done;
  556. goto checks;
  557. }
  558. }
  559. retval = 0;
  560. done:
  561. return retval;
  562. }
  563. /* Locking: Caller holds q->vb_lock */
  564. static int stream_next_buffer(struct videobuf_queue *q,
  565. struct videobuf_buffer **vb, int nonblocking)
  566. {
  567. int retval;
  568. struct videobuf_buffer *buf = NULL;
  569. retval = stream_next_buffer_check_queue(q, nonblocking);
  570. if (retval)
  571. goto done;
  572. buf = list_entry(q->stream.next, struct videobuf_buffer, stream);
  573. retval = videobuf_waiton(buf, nonblocking, 1);
  574. if (retval < 0)
  575. goto done;
  576. *vb = buf;
  577. done:
  578. return retval;
  579. }
  580. int videobuf_dqbuf(struct videobuf_queue *q,
  581. struct v4l2_buffer *b, int nonblocking)
  582. {
  583. struct videobuf_buffer *buf = NULL;
  584. int retval;
  585. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  586. memset(b, 0, sizeof(*b));
  587. mutex_lock(&q->vb_lock);
  588. retval = stream_next_buffer(q, &buf, nonblocking);
  589. if (retval < 0) {
  590. dprintk(1, "dqbuf: next_buffer error: %i\n", retval);
  591. goto done;
  592. }
  593. switch (buf->state) {
  594. case VIDEOBUF_ERROR:
  595. dprintk(1, "dqbuf: state is error\n");
  596. break;
  597. case VIDEOBUF_DONE:
  598. dprintk(1, "dqbuf: state is done\n");
  599. break;
  600. default:
  601. dprintk(1, "dqbuf: state invalid\n");
  602. retval = -EINVAL;
  603. goto done;
  604. }
  605. CALL(q, sync, q, buf);
  606. videobuf_status(q, b, buf, q->type);
  607. list_del(&buf->stream);
  608. buf->state = VIDEOBUF_IDLE;
  609. b->flags &= ~V4L2_BUF_FLAG_DONE;
  610. done:
  611. mutex_unlock(&q->vb_lock);
  612. return retval;
  613. }
  614. EXPORT_SYMBOL_GPL(videobuf_dqbuf);
  615. int videobuf_streamon(struct videobuf_queue *q)
  616. {
  617. struct videobuf_buffer *buf;
  618. unsigned long flags = 0;
  619. int retval;
  620. mutex_lock(&q->vb_lock);
  621. retval = -EBUSY;
  622. if (q->reading)
  623. goto done;
  624. retval = 0;
  625. if (q->streaming)
  626. goto done;
  627. q->streaming = 1;
  628. spin_lock_irqsave(q->irqlock, flags);
  629. list_for_each_entry(buf, &q->stream, stream)
  630. if (buf->state == VIDEOBUF_PREPARED)
  631. q->ops->buf_queue(q, buf);
  632. spin_unlock_irqrestore(q->irqlock, flags);
  633. wake_up_interruptible_sync(&q->wait);
  634. done:
  635. mutex_unlock(&q->vb_lock);
  636. return retval;
  637. }
  638. EXPORT_SYMBOL_GPL(videobuf_streamon);
  639. /* Locking: Caller holds q->vb_lock */
  640. static int __videobuf_streamoff(struct videobuf_queue *q)
  641. {
  642. if (!q->streaming)
  643. return -EINVAL;
  644. videobuf_queue_cancel(q);
  645. return 0;
  646. }
  647. int videobuf_streamoff(struct videobuf_queue *q)
  648. {
  649. int retval;
  650. mutex_lock(&q->vb_lock);
  651. retval = __videobuf_streamoff(q);
  652. mutex_unlock(&q->vb_lock);
  653. return retval;
  654. }
  655. EXPORT_SYMBOL_GPL(videobuf_streamoff);
  656. /* Locking: Caller holds q->vb_lock */
  657. static ssize_t videobuf_read_zerocopy(struct videobuf_queue *q,
  658. char __user *data,
  659. size_t count, loff_t *ppos)
  660. {
  661. enum v4l2_field field;
  662. unsigned long flags = 0;
  663. int retval;
  664. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  665. /* setup stuff */
  666. q->read_buf = videobuf_alloc_vb(q);
  667. if (NULL == q->read_buf)
  668. return -ENOMEM;
  669. q->read_buf->memory = V4L2_MEMORY_USERPTR;
  670. q->read_buf->baddr = (unsigned long)data;
  671. q->read_buf->bsize = count;
  672. field = videobuf_next_field(q);
  673. retval = q->ops->buf_prepare(q, q->read_buf, field);
  674. if (0 != retval)
  675. goto done;
  676. /* start capture & wait */
  677. spin_lock_irqsave(q->irqlock, flags);
  678. q->ops->buf_queue(q, q->read_buf);
  679. spin_unlock_irqrestore(q->irqlock, flags);
  680. retval = videobuf_waiton(q->read_buf, 0, 0);
  681. if (0 == retval) {
  682. CALL(q, sync, q, q->read_buf);
  683. if (VIDEOBUF_ERROR == q->read_buf->state)
  684. retval = -EIO;
  685. else
  686. retval = q->read_buf->size;
  687. }
  688. done:
  689. /* cleanup */
  690. q->ops->buf_release(q, q->read_buf);
  691. kfree(q->read_buf);
  692. q->read_buf = NULL;
  693. return retval;
  694. }
  695. static int __videobuf_copy_to_user(struct videobuf_queue *q,
  696. struct videobuf_buffer *buf,
  697. char __user *data, size_t count,
  698. int nonblocking)
  699. {
  700. void *vaddr = CALL(q, vaddr, buf);
  701. /* copy to userspace */
  702. if (count > buf->size - q->read_off)
  703. count = buf->size - q->read_off;
  704. if (copy_to_user(data, vaddr + q->read_off, count))
  705. return -EFAULT;
  706. return count;
  707. }
  708. static int __videobuf_copy_stream(struct videobuf_queue *q,
  709. struct videobuf_buffer *buf,
  710. char __user *data, size_t count, size_t pos,
  711. int vbihack, int nonblocking)
  712. {
  713. unsigned int *fc = CALL(q, vaddr, buf);
  714. if (vbihack) {
  715. /* dirty, undocumented hack -- pass the frame counter
  716. * within the last four bytes of each vbi data block.
  717. * We need that one to maintain backward compatibility
  718. * to all vbi decoding software out there ... */
  719. fc += (buf->size >> 2) - 1;
  720. *fc = buf->field_count >> 1;
  721. dprintk(1, "vbihack: %d\n", *fc);
  722. }
  723. /* copy stuff using the common method */
  724. count = __videobuf_copy_to_user(q, buf, data, count, nonblocking);
  725. if ((count == -EFAULT) && (pos == 0))
  726. return -EFAULT;
  727. return count;
  728. }
  729. ssize_t videobuf_read_one(struct videobuf_queue *q,
  730. char __user *data, size_t count, loff_t *ppos,
  731. int nonblocking)
  732. {
  733. enum v4l2_field field;
  734. unsigned long flags = 0;
  735. unsigned size = 0, nbufs = 1;
  736. int retval;
  737. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  738. mutex_lock(&q->vb_lock);
  739. q->ops->buf_setup(q, &nbufs, &size);
  740. if (NULL == q->read_buf &&
  741. count >= size &&
  742. !nonblocking) {
  743. retval = videobuf_read_zerocopy(q, data, count, ppos);
  744. if (retval >= 0 || retval == -EIO)
  745. /* ok, all done */
  746. goto done;
  747. /* fallback to kernel bounce buffer on failures */
  748. }
  749. if (NULL == q->read_buf) {
  750. /* need to capture a new frame */
  751. retval = -ENOMEM;
  752. q->read_buf = videobuf_alloc_vb(q);
  753. dprintk(1, "video alloc=0x%p\n", q->read_buf);
  754. if (NULL == q->read_buf)
  755. goto done;
  756. q->read_buf->memory = V4L2_MEMORY_USERPTR;
  757. q->read_buf->bsize = count; /* preferred size */
  758. field = videobuf_next_field(q);
  759. retval = q->ops->buf_prepare(q, q->read_buf, field);
  760. if (0 != retval) {
  761. kfree(q->read_buf);
  762. q->read_buf = NULL;
  763. goto done;
  764. }
  765. spin_lock_irqsave(q->irqlock, flags);
  766. q->ops->buf_queue(q, q->read_buf);
  767. spin_unlock_irqrestore(q->irqlock, flags);
  768. q->read_off = 0;
  769. }
  770. /* wait until capture is done */
  771. retval = videobuf_waiton(q->read_buf, nonblocking, 1);
  772. if (0 != retval)
  773. goto done;
  774. CALL(q, sync, q, q->read_buf);
  775. if (VIDEOBUF_ERROR == q->read_buf->state) {
  776. /* catch I/O errors */
  777. q->ops->buf_release(q, q->read_buf);
  778. kfree(q->read_buf);
  779. q->read_buf = NULL;
  780. retval = -EIO;
  781. goto done;
  782. }
  783. /* Copy to userspace */
  784. retval = __videobuf_copy_to_user(q, q->read_buf, data, count, nonblocking);
  785. if (retval < 0)
  786. goto done;
  787. q->read_off += retval;
  788. if (q->read_off == q->read_buf->size) {
  789. /* all data copied, cleanup */
  790. q->ops->buf_release(q, q->read_buf);
  791. kfree(q->read_buf);
  792. q->read_buf = NULL;
  793. }
  794. done:
  795. mutex_unlock(&q->vb_lock);
  796. return retval;
  797. }
  798. EXPORT_SYMBOL_GPL(videobuf_read_one);
  799. /* Locking: Caller holds q->vb_lock */
  800. static int __videobuf_read_start(struct videobuf_queue *q)
  801. {
  802. enum v4l2_field field;
  803. unsigned long flags = 0;
  804. unsigned int count = 0, size = 0;
  805. int err, i;
  806. q->ops->buf_setup(q, &count, &size);
  807. if (count < 2)
  808. count = 2;
  809. if (count > VIDEO_MAX_FRAME)
  810. count = VIDEO_MAX_FRAME;
  811. size = PAGE_ALIGN(size);
  812. err = __videobuf_mmap_setup(q, count, size, V4L2_MEMORY_USERPTR);
  813. if (err < 0)
  814. return err;
  815. count = err;
  816. for (i = 0; i < count; i++) {
  817. field = videobuf_next_field(q);
  818. err = q->ops->buf_prepare(q, q->bufs[i], field);
  819. if (err)
  820. return err;
  821. list_add_tail(&q->bufs[i]->stream, &q->stream);
  822. }
  823. spin_lock_irqsave(q->irqlock, flags);
  824. for (i = 0; i < count; i++)
  825. q->ops->buf_queue(q, q->bufs[i]);
  826. spin_unlock_irqrestore(q->irqlock, flags);
  827. q->reading = 1;
  828. return 0;
  829. }
  830. static void __videobuf_read_stop(struct videobuf_queue *q)
  831. {
  832. int i;
  833. videobuf_queue_cancel(q);
  834. __videobuf_free(q);
  835. INIT_LIST_HEAD(&q->stream);
  836. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  837. if (NULL == q->bufs[i])
  838. continue;
  839. kfree(q->bufs[i]);
  840. q->bufs[i] = NULL;
  841. }
  842. q->read_buf = NULL;
  843. }
  844. int videobuf_read_start(struct videobuf_queue *q)
  845. {
  846. int rc;
  847. mutex_lock(&q->vb_lock);
  848. rc = __videobuf_read_start(q);
  849. mutex_unlock(&q->vb_lock);
  850. return rc;
  851. }
  852. EXPORT_SYMBOL_GPL(videobuf_read_start);
  853. void videobuf_read_stop(struct videobuf_queue *q)
  854. {
  855. mutex_lock(&q->vb_lock);
  856. __videobuf_read_stop(q);
  857. mutex_unlock(&q->vb_lock);
  858. }
  859. EXPORT_SYMBOL_GPL(videobuf_read_stop);
  860. void videobuf_stop(struct videobuf_queue *q)
  861. {
  862. mutex_lock(&q->vb_lock);
  863. if (q->streaming)
  864. __videobuf_streamoff(q);
  865. if (q->reading)
  866. __videobuf_read_stop(q);
  867. mutex_unlock(&q->vb_lock);
  868. }
  869. EXPORT_SYMBOL_GPL(videobuf_stop);
  870. ssize_t videobuf_read_stream(struct videobuf_queue *q,
  871. char __user *data, size_t count, loff_t *ppos,
  872. int vbihack, int nonblocking)
  873. {
  874. int rc, retval;
  875. unsigned long flags = 0;
  876. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  877. dprintk(2, "%s\n", __func__);
  878. mutex_lock(&q->vb_lock);
  879. retval = -EBUSY;
  880. if (q->streaming)
  881. goto done;
  882. if (!q->reading) {
  883. retval = __videobuf_read_start(q);
  884. if (retval < 0)
  885. goto done;
  886. }
  887. retval = 0;
  888. while (count > 0) {
  889. /* get / wait for data */
  890. if (NULL == q->read_buf) {
  891. q->read_buf = list_entry(q->stream.next,
  892. struct videobuf_buffer,
  893. stream);
  894. list_del(&q->read_buf->stream);
  895. q->read_off = 0;
  896. }
  897. rc = videobuf_waiton(q->read_buf, nonblocking, 1);
  898. if (rc < 0) {
  899. if (0 == retval)
  900. retval = rc;
  901. break;
  902. }
  903. if (q->read_buf->state == VIDEOBUF_DONE) {
  904. rc = __videobuf_copy_stream(q, q->read_buf, data + retval, count,
  905. retval, vbihack, nonblocking);
  906. if (rc < 0) {
  907. retval = rc;
  908. break;
  909. }
  910. retval += rc;
  911. count -= rc;
  912. q->read_off += rc;
  913. } else {
  914. /* some error */
  915. q->read_off = q->read_buf->size;
  916. if (0 == retval)
  917. retval = -EIO;
  918. }
  919. /* requeue buffer when done with copying */
  920. if (q->read_off == q->read_buf->size) {
  921. list_add_tail(&q->read_buf->stream,
  922. &q->stream);
  923. spin_lock_irqsave(q->irqlock, flags);
  924. q->ops->buf_queue(q, q->read_buf);
  925. spin_unlock_irqrestore(q->irqlock, flags);
  926. q->read_buf = NULL;
  927. }
  928. if (retval < 0)
  929. break;
  930. }
  931. done:
  932. mutex_unlock(&q->vb_lock);
  933. return retval;
  934. }
  935. EXPORT_SYMBOL_GPL(videobuf_read_stream);
  936. unsigned int videobuf_poll_stream(struct file *file,
  937. struct videobuf_queue *q,
  938. poll_table *wait)
  939. {
  940. struct videobuf_buffer *buf = NULL;
  941. unsigned int rc = 0;
  942. mutex_lock(&q->vb_lock);
  943. if (q->streaming) {
  944. if (!list_empty(&q->stream))
  945. buf = list_entry(q->stream.next,
  946. struct videobuf_buffer, stream);
  947. } else {
  948. if (!q->reading)
  949. __videobuf_read_start(q);
  950. if (!q->reading) {
  951. rc = POLLERR;
  952. } else if (NULL == q->read_buf) {
  953. q->read_buf = list_entry(q->stream.next,
  954. struct videobuf_buffer,
  955. stream);
  956. list_del(&q->read_buf->stream);
  957. q->read_off = 0;
  958. }
  959. buf = q->read_buf;
  960. }
  961. if (!buf)
  962. rc = POLLERR;
  963. if (0 == rc) {
  964. poll_wait(file, &buf->done, wait);
  965. if (buf->state == VIDEOBUF_DONE ||
  966. buf->state == VIDEOBUF_ERROR) {
  967. switch (q->type) {
  968. case V4L2_BUF_TYPE_VIDEO_OUTPUT:
  969. case V4L2_BUF_TYPE_VBI_OUTPUT:
  970. case V4L2_BUF_TYPE_SLICED_VBI_OUTPUT:
  971. rc = POLLOUT | POLLWRNORM;
  972. break;
  973. default:
  974. rc = POLLIN | POLLRDNORM;
  975. break;
  976. }
  977. }
  978. }
  979. mutex_unlock(&q->vb_lock);
  980. return rc;
  981. }
  982. EXPORT_SYMBOL_GPL(videobuf_poll_stream);
  983. int videobuf_mmap_mapper(struct videobuf_queue *q, struct vm_area_struct *vma)
  984. {
  985. int rc = -EINVAL;
  986. int i;
  987. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  988. if (!(vma->vm_flags & VM_WRITE) || !(vma->vm_flags & VM_SHARED)) {
  989. dprintk(1, "mmap appl bug: PROT_WRITE and MAP_SHARED are required\n");
  990. return -EINVAL;
  991. }
  992. mutex_lock(&q->vb_lock);
  993. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  994. struct videobuf_buffer *buf = q->bufs[i];
  995. if (buf && buf->memory == V4L2_MEMORY_MMAP &&
  996. buf->boff == (vma->vm_pgoff << PAGE_SHIFT)) {
  997. rc = CALL(q, mmap_mapper, q, buf, vma);
  998. break;
  999. }
  1000. }
  1001. mutex_unlock(&q->vb_lock);
  1002. return rc;
  1003. }
  1004. EXPORT_SYMBOL_GPL(videobuf_mmap_mapper);
  1005. #ifdef CONFIG_VIDEO_V4L1_COMPAT
  1006. int videobuf_cgmbuf(struct videobuf_queue *q,
  1007. struct video_mbuf *mbuf, int count)
  1008. {
  1009. struct v4l2_requestbuffers req;
  1010. int rc, i;
  1011. MAGIC_CHECK(q->int_ops->magic, MAGIC_QTYPE_OPS);
  1012. memset(&req, 0, sizeof(req));
  1013. req.type = q->type;
  1014. req.count = count;
  1015. req.memory = V4L2_MEMORY_MMAP;
  1016. rc = videobuf_reqbufs(q, &req);
  1017. if (rc < 0)
  1018. return rc;
  1019. mbuf->frames = req.count;
  1020. mbuf->size = 0;
  1021. for (i = 0; i < mbuf->frames; i++) {
  1022. mbuf->offsets[i] = q->bufs[i]->boff;
  1023. mbuf->size += PAGE_ALIGN(q->bufs[i]->bsize);
  1024. }
  1025. return 0;
  1026. }
  1027. EXPORT_SYMBOL_GPL(videobuf_cgmbuf);
  1028. #endif