videobuf-core.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988
  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/slab.h>
  19. #include <linux/interrupt.h>
  20. #include <media/videobuf-core.h>
  21. #define MAGIC_BUFFER 0x20070728
  22. #define MAGIC_CHECK(is,should) if (unlikely((is) != (should))) \
  23. { printk(KERN_ERR "magic mismatch: %x (expected %x)\n",is,should); BUG(); }
  24. static int debug = 0;
  25. module_param(debug, int, 0644);
  26. MODULE_DESCRIPTION("helper module to manage video4linux buffers");
  27. MODULE_AUTHOR("Mauro Carvalho Chehab <mchehab@infradead.org>");
  28. MODULE_LICENSE("GPL");
  29. #define dprintk(level, fmt, arg...) if (debug >= level) \
  30. printk(KERN_DEBUG "vbuf: " fmt , ## arg)
  31. /* --------------------------------------------------------------------- */
  32. #define CALL(q, f, arg...) \
  33. ( (q->int_ops->f)? q->int_ops->f(arg) : 0)
  34. void* videobuf_alloc(struct videobuf_queue* q)
  35. {
  36. struct videobuf_buffer *vb;
  37. BUG_ON (q->msize<sizeof(*vb));
  38. if (!q->int_ops || !q->int_ops->alloc) {
  39. printk(KERN_ERR "No specific ops defined!\n");
  40. BUG();
  41. }
  42. vb = q->int_ops->alloc(q->msize);
  43. if (NULL != vb) {
  44. init_waitqueue_head(&vb->done);
  45. vb->magic = MAGIC_BUFFER;
  46. }
  47. return vb;
  48. }
  49. int videobuf_waiton(struct videobuf_buffer *vb, int non_blocking, int intr)
  50. {
  51. int retval = 0;
  52. DECLARE_WAITQUEUE(wait, current);
  53. MAGIC_CHECK(vb->magic,MAGIC_BUFFER);
  54. add_wait_queue(&vb->done, &wait);
  55. while (vb->state == STATE_ACTIVE || vb->state == STATE_QUEUED) {
  56. if (non_blocking) {
  57. retval = -EAGAIN;
  58. break;
  59. }
  60. set_current_state(intr ? TASK_INTERRUPTIBLE
  61. : TASK_UNINTERRUPTIBLE);
  62. if (vb->state == STATE_ACTIVE || vb->state == STATE_QUEUED)
  63. schedule();
  64. set_current_state(TASK_RUNNING);
  65. if (intr && signal_pending(current)) {
  66. dprintk(1,"buffer waiton: -EINTR\n");
  67. retval = -EINTR;
  68. break;
  69. }
  70. }
  71. remove_wait_queue(&vb->done, &wait);
  72. return retval;
  73. }
  74. int videobuf_iolock(struct videobuf_queue* q, struct videobuf_buffer *vb,
  75. struct v4l2_framebuffer *fbuf)
  76. {
  77. MAGIC_CHECK(vb->magic,MAGIC_BUFFER);
  78. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  79. return CALL(q,iolock,q,vb,fbuf);
  80. }
  81. /* --------------------------------------------------------------------- */
  82. void videobuf_queue_init(struct videobuf_queue* q,
  83. struct videobuf_queue_ops *ops,
  84. void *dev,
  85. spinlock_t *irqlock,
  86. enum v4l2_buf_type type,
  87. enum v4l2_field field,
  88. unsigned int msize,
  89. void *priv)
  90. {
  91. memset(q,0,sizeof(*q));
  92. q->irqlock = irqlock;
  93. q->dev = dev;
  94. q->type = type;
  95. q->field = field;
  96. q->msize = msize;
  97. q->ops = ops;
  98. q->priv_data = priv;
  99. /* All buffer operations are mandatory */
  100. BUG_ON (!q->ops->buf_setup);
  101. BUG_ON (!q->ops->buf_prepare);
  102. BUG_ON (!q->ops->buf_queue);
  103. BUG_ON (!q->ops->buf_release);
  104. mutex_init(&q->lock);
  105. INIT_LIST_HEAD(&q->stream);
  106. }
  107. int videobuf_queue_is_busy(struct videobuf_queue *q)
  108. {
  109. int i;
  110. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  111. if (q->streaming) {
  112. dprintk(1,"busy: streaming active\n");
  113. return 1;
  114. }
  115. if (q->reading) {
  116. dprintk(1,"busy: pending read #1\n");
  117. return 1;
  118. }
  119. if (q->read_buf) {
  120. dprintk(1,"busy: pending read #2\n");
  121. return 1;
  122. }
  123. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  124. if (NULL == q->bufs[i])
  125. continue;
  126. if (q->bufs[i]->map) {
  127. dprintk(1,"busy: buffer #%d mapped\n",i);
  128. return 1;
  129. }
  130. if (q->bufs[i]->state == STATE_QUEUED) {
  131. dprintk(1,"busy: buffer #%d queued\n",i);
  132. return 1;
  133. }
  134. if (q->bufs[i]->state == STATE_ACTIVE) {
  135. dprintk(1,"busy: buffer #%d avtive\n",i);
  136. return 1;
  137. }
  138. }
  139. return 0;
  140. }
  141. void videobuf_queue_cancel(struct videobuf_queue *q)
  142. {
  143. unsigned long flags=0;
  144. int i;
  145. /* remove queued buffers from list */
  146. if (q->irqlock)
  147. spin_lock_irqsave(q->irqlock,flags);
  148. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  149. if (NULL == q->bufs[i])
  150. continue;
  151. if (q->bufs[i]->state == STATE_QUEUED) {
  152. list_del(&q->bufs[i]->queue);
  153. q->bufs[i]->state = STATE_ERROR;
  154. }
  155. }
  156. if (q->irqlock)
  157. spin_unlock_irqrestore(q->irqlock,flags);
  158. /* free all buffers + clear queue */
  159. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  160. if (NULL == q->bufs[i])
  161. continue;
  162. q->ops->buf_release(q,q->bufs[i]);
  163. }
  164. INIT_LIST_HEAD(&q->stream);
  165. }
  166. /* --------------------------------------------------------------------- */
  167. enum v4l2_field videobuf_next_field(struct videobuf_queue *q)
  168. {
  169. enum v4l2_field field = q->field;
  170. BUG_ON(V4L2_FIELD_ANY == field);
  171. if (V4L2_FIELD_ALTERNATE == field) {
  172. if (V4L2_FIELD_TOP == q->last) {
  173. field = V4L2_FIELD_BOTTOM;
  174. q->last = V4L2_FIELD_BOTTOM;
  175. } else {
  176. field = V4L2_FIELD_TOP;
  177. q->last = V4L2_FIELD_TOP;
  178. }
  179. }
  180. return field;
  181. }
  182. static void videobuf_status(struct videobuf_queue *q, struct v4l2_buffer *b,
  183. struct videobuf_buffer *vb, enum v4l2_buf_type type)
  184. {
  185. MAGIC_CHECK(vb->magic,MAGIC_BUFFER);
  186. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  187. b->index = vb->i;
  188. b->type = type;
  189. b->memory = vb->memory;
  190. switch (b->memory) {
  191. case V4L2_MEMORY_MMAP:
  192. b->m.offset = vb->boff;
  193. b->length = vb->bsize;
  194. break;
  195. case V4L2_MEMORY_USERPTR:
  196. b->m.userptr = vb->baddr;
  197. b->length = vb->bsize;
  198. break;
  199. case V4L2_MEMORY_OVERLAY:
  200. b->m.offset = vb->boff;
  201. break;
  202. }
  203. b->flags = 0;
  204. if (vb->map)
  205. b->flags |= V4L2_BUF_FLAG_MAPPED;
  206. switch (vb->state) {
  207. case STATE_PREPARED:
  208. case STATE_QUEUED:
  209. case STATE_ACTIVE:
  210. b->flags |= V4L2_BUF_FLAG_QUEUED;
  211. break;
  212. case STATE_DONE:
  213. case STATE_ERROR:
  214. b->flags |= V4L2_BUF_FLAG_DONE;
  215. break;
  216. case STATE_NEEDS_INIT:
  217. case STATE_IDLE:
  218. /* nothing */
  219. break;
  220. }
  221. if (vb->input != UNSET) {
  222. b->flags |= V4L2_BUF_FLAG_INPUT;
  223. b->input = vb->input;
  224. }
  225. b->field = vb->field;
  226. b->timestamp = vb->ts;
  227. b->bytesused = vb->size;
  228. b->sequence = vb->field_count >> 1;
  229. }
  230. int videobuf_reqbufs(struct videobuf_queue *q,
  231. struct v4l2_requestbuffers *req)
  232. {
  233. unsigned int size,count;
  234. int retval;
  235. if (req->type != q->type) {
  236. dprintk(1,"reqbufs: queue type invalid\n");
  237. return -EINVAL;
  238. }
  239. if (req->count < 1) {
  240. dprintk(1,"reqbufs: count invalid (%d)\n",req->count);
  241. return -EINVAL;
  242. }
  243. if (req->memory != V4L2_MEMORY_MMAP &&
  244. req->memory != V4L2_MEMORY_USERPTR &&
  245. req->memory != V4L2_MEMORY_OVERLAY) {
  246. dprintk(1,"reqbufs: memory type invalid\n");
  247. return -EINVAL;
  248. }
  249. mutex_lock(&q->lock);
  250. if (q->streaming) {
  251. dprintk(1,"reqbufs: streaming already exists\n");
  252. retval = -EBUSY;
  253. goto done;
  254. }
  255. if (!list_empty(&q->stream)) {
  256. dprintk(1,"reqbufs: stream running\n");
  257. retval = -EBUSY;
  258. goto done;
  259. }
  260. count = req->count;
  261. if (count > VIDEO_MAX_FRAME)
  262. count = VIDEO_MAX_FRAME;
  263. size = 0;
  264. q->ops->buf_setup(q,&count,&size);
  265. size = PAGE_ALIGN(size);
  266. dprintk(1,"reqbufs: bufs=%d, size=0x%x [%d pages total]\n",
  267. count, size, (count*size)>>PAGE_SHIFT);
  268. retval = videobuf_mmap_setup(q,count,size,req->memory);
  269. if (retval < 0) {
  270. dprintk(1,"reqbufs: mmap setup returned %d\n",retval);
  271. goto done;
  272. }
  273. req->count = count;
  274. done:
  275. mutex_unlock(&q->lock);
  276. return retval;
  277. }
  278. int videobuf_querybuf(struct videobuf_queue *q, struct v4l2_buffer *b)
  279. {
  280. if (unlikely(b->type != q->type)) {
  281. dprintk(1,"querybuf: Wrong type.\n");
  282. return -EINVAL;
  283. }
  284. if (unlikely(b->index < 0 || b->index >= VIDEO_MAX_FRAME)) {
  285. dprintk(1,"querybuf: index out of range.\n");
  286. return -EINVAL;
  287. }
  288. if (unlikely(NULL == q->bufs[b->index])) {
  289. dprintk(1,"querybuf: buffer is null.\n");
  290. return -EINVAL;
  291. }
  292. videobuf_status(q,b,q->bufs[b->index],q->type);
  293. return 0;
  294. }
  295. int videobuf_qbuf(struct videobuf_queue *q,
  296. struct v4l2_buffer *b)
  297. {
  298. struct videobuf_buffer *buf;
  299. enum v4l2_field field;
  300. unsigned long flags=0;
  301. int retval;
  302. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  303. if (b->memory == V4L2_MEMORY_MMAP)
  304. down_read(&current->mm->mmap_sem);
  305. mutex_lock(&q->lock);
  306. retval = -EBUSY;
  307. if (q->reading) {
  308. dprintk(1,"qbuf: Reading running...\n");
  309. goto done;
  310. }
  311. retval = -EINVAL;
  312. if (b->type != q->type) {
  313. dprintk(1,"qbuf: Wrong type.\n");
  314. goto done;
  315. }
  316. if (b->index < 0 || b->index >= VIDEO_MAX_FRAME) {
  317. dprintk(1,"qbuf: index out of range.\n");
  318. goto done;
  319. }
  320. buf = q->bufs[b->index];
  321. if (NULL == buf) {
  322. dprintk(1,"qbuf: buffer is null.\n");
  323. goto done;
  324. }
  325. MAGIC_CHECK(buf->magic,MAGIC_BUFFER);
  326. if (buf->memory != b->memory) {
  327. dprintk(1,"qbuf: memory type is wrong.\n");
  328. goto done;
  329. }
  330. if (buf->state != STATE_NEEDS_INIT && buf->state != STATE_IDLE) {
  331. dprintk(1,"qbuf: buffer is already queued or active.\n");
  332. goto done;
  333. }
  334. if (b->flags & V4L2_BUF_FLAG_INPUT) {
  335. if (b->input >= q->inputs) {
  336. dprintk(1,"qbuf: wrong input.\n");
  337. goto done;
  338. }
  339. buf->input = b->input;
  340. } else {
  341. buf->input = UNSET;
  342. }
  343. switch (b->memory) {
  344. case V4L2_MEMORY_MMAP:
  345. if (0 == buf->baddr) {
  346. dprintk(1,"qbuf: mmap requested but buffer addr is zero!\n");
  347. goto done;
  348. }
  349. break;
  350. case V4L2_MEMORY_USERPTR:
  351. if (b->length < buf->bsize) {
  352. dprintk(1,"qbuf: buffer length is not enough\n");
  353. goto done;
  354. }
  355. if (STATE_NEEDS_INIT != buf->state && buf->baddr != b->m.userptr)
  356. q->ops->buf_release(q,buf);
  357. buf->baddr = b->m.userptr;
  358. break;
  359. case V4L2_MEMORY_OVERLAY:
  360. buf->boff = b->m.offset;
  361. break;
  362. default:
  363. dprintk(1,"qbuf: wrong memory type\n");
  364. goto done;
  365. }
  366. dprintk(1,"qbuf: requesting next field\n");
  367. field = videobuf_next_field(q);
  368. retval = q->ops->buf_prepare(q,buf,field);
  369. if (0 != retval) {
  370. dprintk(1,"qbuf: buffer_prepare returned %d\n",retval);
  371. goto done;
  372. }
  373. list_add_tail(&buf->stream,&q->stream);
  374. if (q->streaming) {
  375. if (q->irqlock)
  376. spin_lock_irqsave(q->irqlock,flags);
  377. q->ops->buf_queue(q,buf);
  378. if (q->irqlock)
  379. spin_unlock_irqrestore(q->irqlock,flags);
  380. }
  381. dprintk(1,"qbuf: succeded\n");
  382. retval = 0;
  383. done:
  384. mutex_unlock(&q->lock);
  385. if (b->memory == V4L2_MEMORY_MMAP)
  386. up_read(&current->mm->mmap_sem);
  387. return retval;
  388. }
  389. int videobuf_dqbuf(struct videobuf_queue *q,
  390. struct v4l2_buffer *b, int nonblocking)
  391. {
  392. struct videobuf_buffer *buf;
  393. int retval;
  394. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  395. mutex_lock(&q->lock);
  396. retval = -EBUSY;
  397. if (q->reading) {
  398. dprintk(1,"dqbuf: Reading running...\n");
  399. goto done;
  400. }
  401. retval = -EINVAL;
  402. if (b->type != q->type) {
  403. dprintk(1,"dqbuf: Wrong type.\n");
  404. goto done;
  405. }
  406. if (list_empty(&q->stream)) {
  407. dprintk(1,"dqbuf: stream running\n");
  408. goto done;
  409. }
  410. buf = list_entry(q->stream.next, struct videobuf_buffer, stream);
  411. retval = videobuf_waiton(buf, nonblocking, 1);
  412. if (retval < 0) {
  413. dprintk(1,"dqbuf: waiton returned %d\n",retval);
  414. goto done;
  415. }
  416. switch (buf->state) {
  417. case STATE_ERROR:
  418. dprintk(1,"dqbuf: state is error\n");
  419. retval = -EIO;
  420. CALL(q,sync,q, buf);
  421. buf->state = STATE_IDLE;
  422. break;
  423. case STATE_DONE:
  424. dprintk(1,"dqbuf: state is done\n");
  425. CALL(q,sync,q, buf);
  426. buf->state = STATE_IDLE;
  427. break;
  428. default:
  429. dprintk(1,"dqbuf: state invalid\n");
  430. retval = -EINVAL;
  431. goto done;
  432. }
  433. list_del(&buf->stream);
  434. memset(b,0,sizeof(*b));
  435. videobuf_status(q,b,buf,q->type);
  436. done:
  437. mutex_unlock(&q->lock);
  438. return retval;
  439. }
  440. int videobuf_streamon(struct videobuf_queue *q)
  441. {
  442. struct videobuf_buffer *buf;
  443. struct list_head *list;
  444. unsigned long flags=0;
  445. int retval;
  446. mutex_lock(&q->lock);
  447. retval = -EBUSY;
  448. if (q->reading)
  449. goto done;
  450. retval = 0;
  451. if (q->streaming)
  452. goto done;
  453. q->streaming = 1;
  454. if (q->irqlock)
  455. spin_lock_irqsave(q->irqlock,flags);
  456. list_for_each(list,&q->stream) {
  457. buf = list_entry(list, struct videobuf_buffer, stream);
  458. if (buf->state == STATE_PREPARED)
  459. q->ops->buf_queue(q,buf);
  460. }
  461. if (q->irqlock)
  462. spin_unlock_irqrestore(q->irqlock,flags);
  463. done:
  464. mutex_unlock(&q->lock);
  465. return retval;
  466. }
  467. int videobuf_streamoff(struct videobuf_queue *q)
  468. {
  469. int retval = -EINVAL;
  470. mutex_lock(&q->lock);
  471. if (!q->streaming)
  472. goto done;
  473. videobuf_queue_cancel(q);
  474. q->streaming = 0;
  475. retval = 0;
  476. done:
  477. mutex_unlock(&q->lock);
  478. return retval;
  479. }
  480. static ssize_t videobuf_read_zerocopy(struct videobuf_queue *q,
  481. char __user *data,
  482. size_t count, loff_t *ppos)
  483. {
  484. enum v4l2_field field;
  485. unsigned long flags=0;
  486. int retval;
  487. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  488. /* setup stuff */
  489. q->read_buf = videobuf_alloc(q);
  490. if (NULL == q->read_buf)
  491. return -ENOMEM;
  492. q->read_buf->memory = V4L2_MEMORY_USERPTR;
  493. q->read_buf->baddr = (unsigned long)data;
  494. q->read_buf->bsize = count;
  495. field = videobuf_next_field(q);
  496. retval = q->ops->buf_prepare(q,q->read_buf,field);
  497. if (0 != retval)
  498. goto done;
  499. /* start capture & wait */
  500. if (q->irqlock)
  501. spin_lock_irqsave(q->irqlock,flags);
  502. q->ops->buf_queue(q,q->read_buf);
  503. if (q->irqlock)
  504. spin_unlock_irqrestore(q->irqlock,flags);
  505. retval = videobuf_waiton(q->read_buf,0,0);
  506. if (0 == retval) {
  507. CALL(q,sync,q,q->read_buf);
  508. if (STATE_ERROR == q->read_buf->state)
  509. retval = -EIO;
  510. else
  511. retval = q->read_buf->size;
  512. }
  513. done:
  514. /* cleanup */
  515. q->ops->buf_release(q,q->read_buf);
  516. kfree(q->read_buf);
  517. q->read_buf = NULL;
  518. return retval;
  519. }
  520. ssize_t videobuf_read_one(struct videobuf_queue *q,
  521. char __user *data, size_t count, loff_t *ppos,
  522. int nonblocking)
  523. {
  524. enum v4l2_field field;
  525. unsigned long flags=0;
  526. unsigned size, nbufs;
  527. int retval;
  528. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  529. mutex_lock(&q->lock);
  530. nbufs = 1; size = 0;
  531. q->ops->buf_setup(q,&nbufs,&size);
  532. if (NULL == q->read_buf &&
  533. count >= size &&
  534. !nonblocking) {
  535. retval = videobuf_read_zerocopy(q,data,count,ppos);
  536. if (retval >= 0 || retval == -EIO)
  537. /* ok, all done */
  538. goto done;
  539. /* fallback to kernel bounce buffer on failures */
  540. }
  541. if (NULL == q->read_buf) {
  542. /* need to capture a new frame */
  543. retval = -ENOMEM;
  544. q->read_buf = videobuf_alloc(q);
  545. dprintk(1,"video alloc=0x%p\n", q->read_buf);
  546. if (NULL == q->read_buf)
  547. goto done;
  548. q->read_buf->memory = V4L2_MEMORY_USERPTR;
  549. q->read_buf->bsize = count; /* preferred size */
  550. field = videobuf_next_field(q);
  551. retval = q->ops->buf_prepare(q,q->read_buf,field);
  552. if (0 != retval) {
  553. kfree (q->read_buf);
  554. q->read_buf = NULL;
  555. goto done;
  556. }
  557. if (q->irqlock)
  558. spin_lock_irqsave(q->irqlock,flags);
  559. q->ops->buf_queue(q,q->read_buf);
  560. if (q->irqlock)
  561. spin_unlock_irqrestore(q->irqlock,flags);
  562. q->read_off = 0;
  563. }
  564. /* wait until capture is done */
  565. retval = videobuf_waiton(q->read_buf, nonblocking, 1);
  566. if (0 != retval)
  567. goto done;
  568. CALL(q,sync,q,q->read_buf);
  569. if (STATE_ERROR == q->read_buf->state) {
  570. /* catch I/O errors */
  571. q->ops->buf_release(q,q->read_buf);
  572. kfree(q->read_buf);
  573. q->read_buf = NULL;
  574. retval = -EIO;
  575. goto done;
  576. }
  577. /* Copy to userspace */
  578. retval=CALL(q,copy_to_user,q,data,count,nonblocking);
  579. if (retval<0)
  580. goto done;
  581. q->read_off += retval;
  582. if (q->read_off == q->read_buf->size) {
  583. /* all data copied, cleanup */
  584. q->ops->buf_release(q,q->read_buf);
  585. kfree(q->read_buf);
  586. q->read_buf = NULL;
  587. }
  588. done:
  589. mutex_unlock(&q->lock);
  590. return retval;
  591. }
  592. int videobuf_read_start(struct videobuf_queue *q)
  593. {
  594. enum v4l2_field field;
  595. unsigned long flags=0;
  596. int count = 0, size = 0;
  597. int err, i;
  598. q->ops->buf_setup(q,&count,&size);
  599. if (count < 2)
  600. count = 2;
  601. if (count > VIDEO_MAX_FRAME)
  602. count = VIDEO_MAX_FRAME;
  603. size = PAGE_ALIGN(size);
  604. err = videobuf_mmap_setup(q, count, size, V4L2_MEMORY_USERPTR);
  605. if (err)
  606. return err;
  607. for (i = 0; i < count; i++) {
  608. field = videobuf_next_field(q);
  609. err = q->ops->buf_prepare(q,q->bufs[i],field);
  610. if (err)
  611. return err;
  612. list_add_tail(&q->bufs[i]->stream, &q->stream);
  613. }
  614. if (q->irqlock)
  615. spin_lock_irqsave(q->irqlock,flags);
  616. for (i = 0; i < count; i++)
  617. q->ops->buf_queue(q,q->bufs[i]);
  618. if (q->irqlock)
  619. spin_unlock_irqrestore(q->irqlock,flags);
  620. q->reading = 1;
  621. return 0;
  622. }
  623. void videobuf_read_stop(struct videobuf_queue *q)
  624. {
  625. int i;
  626. videobuf_queue_cancel(q);
  627. videobuf_mmap_free(q);
  628. INIT_LIST_HEAD(&q->stream);
  629. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  630. if (NULL == q->bufs[i])
  631. continue;
  632. kfree(q->bufs[i]);
  633. q->bufs[i] = NULL;
  634. }
  635. q->read_buf = NULL;
  636. q->reading = 0;
  637. }
  638. ssize_t videobuf_read_stream(struct videobuf_queue *q,
  639. char __user *data, size_t count, loff_t *ppos,
  640. int vbihack, int nonblocking)
  641. {
  642. int rc, retval;
  643. unsigned long flags=0;
  644. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  645. dprintk(2,"%s\n",__FUNCTION__);
  646. mutex_lock(&q->lock);
  647. retval = -EBUSY;
  648. if (q->streaming)
  649. goto done;
  650. if (!q->reading) {
  651. retval = videobuf_read_start(q);
  652. if (retval < 0)
  653. goto done;
  654. }
  655. retval = 0;
  656. while (count > 0) {
  657. /* get / wait for data */
  658. if (NULL == q->read_buf) {
  659. q->read_buf = list_entry(q->stream.next,
  660. struct videobuf_buffer,
  661. stream);
  662. list_del(&q->read_buf->stream);
  663. q->read_off = 0;
  664. }
  665. rc = videobuf_waiton(q->read_buf, nonblocking, 1);
  666. if (rc < 0) {
  667. if (0 == retval)
  668. retval = rc;
  669. break;
  670. }
  671. if (q->read_buf->state == STATE_DONE) {
  672. rc = CALL (q,copy_stream, q, data, count,
  673. retval, vbihack, nonblocking);
  674. if (rc < 0) {
  675. retval = rc;
  676. break;
  677. }
  678. retval += rc;
  679. count -= rc;
  680. q->read_off += rc;
  681. } else {
  682. /* some error */
  683. q->read_off = q->read_buf->size;
  684. if (0 == retval)
  685. retval = -EIO;
  686. }
  687. /* requeue buffer when done with copying */
  688. if (q->read_off == q->read_buf->size) {
  689. list_add_tail(&q->read_buf->stream,
  690. &q->stream);
  691. if (q->irqlock)
  692. spin_lock_irqsave(q->irqlock,flags);
  693. q->ops->buf_queue(q,q->read_buf);
  694. if (q->irqlock)
  695. spin_unlock_irqrestore(q->irqlock,flags);
  696. q->read_buf = NULL;
  697. }
  698. if (retval < 0)
  699. break;
  700. }
  701. done:
  702. mutex_unlock(&q->lock);
  703. return retval;
  704. }
  705. unsigned int videobuf_poll_stream(struct file *file,
  706. struct videobuf_queue *q,
  707. poll_table *wait)
  708. {
  709. struct videobuf_buffer *buf = NULL;
  710. unsigned int rc = 0;
  711. mutex_lock(&q->lock);
  712. if (q->streaming) {
  713. if (!list_empty(&q->stream))
  714. buf = list_entry(q->stream.next,
  715. struct videobuf_buffer, stream);
  716. } else {
  717. if (!q->reading)
  718. videobuf_read_start(q);
  719. if (!q->reading) {
  720. rc = POLLERR;
  721. } else if (NULL == q->read_buf) {
  722. q->read_buf = list_entry(q->stream.next,
  723. struct videobuf_buffer,
  724. stream);
  725. list_del(&q->read_buf->stream);
  726. q->read_off = 0;
  727. }
  728. buf = q->read_buf;
  729. }
  730. if (!buf)
  731. rc = POLLERR;
  732. if (0 == rc) {
  733. poll_wait(file, &buf->done, wait);
  734. if (buf->state == STATE_DONE ||
  735. buf->state == STATE_ERROR)
  736. rc = POLLIN|POLLRDNORM;
  737. }
  738. mutex_unlock(&q->lock);
  739. return rc;
  740. }
  741. int videobuf_mmap_setup(struct videobuf_queue *q,
  742. unsigned int bcount, unsigned int bsize,
  743. enum v4l2_memory memory)
  744. {
  745. unsigned int i;
  746. int err;
  747. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  748. err = videobuf_mmap_free(q);
  749. if (0 != err)
  750. return err;
  751. /* Allocate and initialize buffers */
  752. for (i = 0; i < bcount; i++) {
  753. q->bufs[i] = videobuf_alloc(q);
  754. q->bufs[i]->i = i;
  755. q->bufs[i]->input = UNSET;
  756. q->bufs[i]->memory = memory;
  757. q->bufs[i]->bsize = bsize;
  758. switch (memory) {
  759. case V4L2_MEMORY_MMAP:
  760. q->bufs[i]->boff = bsize * i;
  761. break;
  762. case V4L2_MEMORY_USERPTR:
  763. case V4L2_MEMORY_OVERLAY:
  764. /* nothing */
  765. break;
  766. }
  767. }
  768. dprintk(1,"mmap setup: %d buffers, %d bytes each\n",
  769. bcount,bsize);
  770. return 0;
  771. }
  772. int videobuf_mmap_free(struct videobuf_queue *q)
  773. {
  774. int i;
  775. int rc;
  776. if (!q)
  777. return 0;
  778. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  779. rc = CALL(q,mmap_free,q);
  780. if (rc<0)
  781. return rc;
  782. for (i = 0; i < VIDEO_MAX_FRAME; i++) {
  783. if (NULL == q->bufs[i])
  784. continue;
  785. q->ops->buf_release(q,q->bufs[i]);
  786. kfree(q->bufs[i]);
  787. q->bufs[i] = NULL;
  788. }
  789. return rc;
  790. }
  791. int videobuf_mmap_mapper(struct videobuf_queue *q,
  792. struct vm_area_struct *vma)
  793. {
  794. int retval;
  795. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  796. mutex_lock(&q->lock);
  797. retval=CALL(q,mmap_mapper,q,vma);
  798. mutex_unlock(&q->lock);
  799. return retval;
  800. }
  801. #ifdef CONFIG_VIDEO_V4L1_COMPAT
  802. int videobuf_cgmbuf(struct videobuf_queue *q,
  803. struct video_mbuf *mbuf, int count)
  804. {
  805. struct v4l2_requestbuffers req;
  806. int rc,i;
  807. MAGIC_CHECK(q->int_ops->magic,MAGIC_QTYPE_OPS);
  808. memset(&req,0,sizeof(req));
  809. req.type = q->type;
  810. req.count = count;
  811. req.memory = V4L2_MEMORY_MMAP;
  812. rc = videobuf_reqbufs(q,&req);
  813. if (rc < 0)
  814. return rc;
  815. mbuf->frames = req.count;
  816. mbuf->size = 0;
  817. for (i = 0; i < mbuf->frames; i++) {
  818. mbuf->offsets[i] = q->bufs[i]->boff;
  819. mbuf->size += q->bufs[i]->bsize;
  820. }
  821. return 0;
  822. }
  823. #endif
  824. /* --------------------------------------------------------------------- */
  825. EXPORT_SYMBOL_GPL(videobuf_waiton);
  826. EXPORT_SYMBOL_GPL(videobuf_iolock);
  827. EXPORT_SYMBOL_GPL(videobuf_alloc);
  828. EXPORT_SYMBOL_GPL(videobuf_queue_init);
  829. EXPORT_SYMBOL_GPL(videobuf_queue_cancel);
  830. EXPORT_SYMBOL_GPL(videobuf_queue_is_busy);
  831. EXPORT_SYMBOL_GPL(videobuf_next_field);
  832. EXPORT_SYMBOL_GPL(videobuf_reqbufs);
  833. EXPORT_SYMBOL_GPL(videobuf_querybuf);
  834. EXPORT_SYMBOL_GPL(videobuf_qbuf);
  835. EXPORT_SYMBOL_GPL(videobuf_dqbuf);
  836. EXPORT_SYMBOL_GPL(videobuf_cgmbuf);
  837. EXPORT_SYMBOL_GPL(videobuf_streamon);
  838. EXPORT_SYMBOL_GPL(videobuf_streamoff);
  839. EXPORT_SYMBOL_GPL(videobuf_read_start);
  840. EXPORT_SYMBOL_GPL(videobuf_read_stop);
  841. EXPORT_SYMBOL_GPL(videobuf_read_stream);
  842. EXPORT_SYMBOL_GPL(videobuf_read_one);
  843. EXPORT_SYMBOL_GPL(videobuf_poll_stream);
  844. EXPORT_SYMBOL_GPL(videobuf_mmap_setup);
  845. EXPORT_SYMBOL_GPL(videobuf_mmap_free);
  846. EXPORT_SYMBOL_GPL(videobuf_mmap_mapper);
  847. /*
  848. * Local variables:
  849. * c-basic-offset: 8
  850. * End:
  851. */