1 /* 2 * videobuf2-v4l2.c - V4L2 driver helper framework 3 * 4 * Copyright (C) 2010 Samsung Electronics 5 * 6 * Author: Pawel Osciak <pawel@osciak.com> 7 * Marek Szyprowski <m.szyprowski@samsung.com> 8 * 9 * The vb2_thread implementation was based on code from videobuf-dvb.c: 10 * (c) 2004 Gerd Knorr <kraxel@bytesex.org> [SUSE Labs] 11 * 12 * This program is free software; you can redistribute it and/or modify 13 * it under the terms of the GNU General Public License as published by 14 * the Free Software Foundation. 15 */ 16 17 #include <linux/err.h> 18 #include <linux/kernel.h> 19 #include <linux/module.h> 20 #include <linux/mm.h> 21 #include <linux/poll.h> 22 #include <linux/slab.h> 23 #include <linux/sched.h> 24 #include <linux/freezer.h> 25 #include <linux/kthread.h> 26 27 #include <media/v4l2-dev.h> 28 #include <media/v4l2-device.h> 29 #include <media/v4l2-fh.h> 30 #include <media/v4l2-event.h> 31 #include <media/v4l2-common.h> 32 33 #include <media/videobuf2-v4l2.h> 34 35 static int debug; 36 module_param(debug, int, 0644); 37 38 #define dprintk(level, fmt, arg...) \ 39 do { \ 40 if (debug >= level) \ 41 pr_info("vb2-v4l2: %s: " fmt, __func__, ## arg); \ 42 } while (0) 43 44 /* Flags that are set by us */ 45 #define V4L2_BUFFER_MASK_FLAGS (V4L2_BUF_FLAG_MAPPED | V4L2_BUF_FLAG_QUEUED | \ 46 V4L2_BUF_FLAG_DONE | V4L2_BUF_FLAG_ERROR | \ 47 V4L2_BUF_FLAG_PREPARED | \ 48 V4L2_BUF_FLAG_IN_REQUEST | \ 49 V4L2_BUF_FLAG_REQUEST_FD | \ 50 V4L2_BUF_FLAG_TIMESTAMP_MASK) 51 /* Output buffer flags that should be passed on to the driver */ 52 #define V4L2_BUFFER_OUT_FLAGS (V4L2_BUF_FLAG_PFRAME | V4L2_BUF_FLAG_BFRAME | \ 53 V4L2_BUF_FLAG_KEYFRAME | V4L2_BUF_FLAG_TIMECODE) 54 55 /* 56 * __verify_planes_array() - verify that the planes array passed in struct 57 * v4l2_buffer from userspace can be safely used 58 */ 59 static int __verify_planes_array(struct vb2_buffer *vb, const struct v4l2_buffer *b) 60 { 61 if (!V4L2_TYPE_IS_MULTIPLANAR(b->type)) 62 return 0; 63 64 /* Is memory for copying plane information present? */ 65 if (b->m.planes == NULL) { 66 dprintk(1, "multi-planar buffer passed but planes array not provided\n"); 67 return -EINVAL; 68 } 69 70 if (b->length < vb->num_planes || b->length > VB2_MAX_PLANES) { 71 dprintk(1, "incorrect planes array length, expected %d, got %d\n", 72 vb->num_planes, b->length); 73 return -EINVAL; 74 } 75 76 return 0; 77 } 78 79 static int __verify_planes_array_core(struct vb2_buffer *vb, const void *pb) 80 { 81 return __verify_planes_array(vb, pb); 82 } 83 84 /* 85 * __verify_length() - Verify that the bytesused value for each plane fits in 86 * the plane length and that the data offset doesn't exceed the bytesused value. 87 */ 88 static int __verify_length(struct vb2_buffer *vb, const struct v4l2_buffer *b) 89 { 90 unsigned int length; 91 unsigned int bytesused; 92 unsigned int plane; 93 94 if (!V4L2_TYPE_IS_OUTPUT(b->type)) 95 return 0; 96 97 if (V4L2_TYPE_IS_MULTIPLANAR(b->type)) { 98 for (plane = 0; plane < vb->num_planes; ++plane) { 99 length = (b->memory == VB2_MEMORY_USERPTR || 100 b->memory == VB2_MEMORY_DMABUF) 101 ? b->m.planes[plane].length 102 : vb->planes[plane].length; 103 bytesused = b->m.planes[plane].bytesused 104 ? b->m.planes[plane].bytesused : length; 105 106 if (b->m.planes[plane].bytesused > length) 107 return -EINVAL; 108 109 if (b->m.planes[plane].data_offset > 0 && 110 b->m.planes[plane].data_offset >= bytesused) 111 return -EINVAL; 112 } 113 } else { 114 length = (b->memory == VB2_MEMORY_USERPTR) 115 ? b->length : vb->planes[0].length; 116 117 if (b->bytesused > length) 118 return -EINVAL; 119 } 120 121 return 0; 122 } 123 124 /* 125 * __init_v4l2_vb2_buffer() - initialize the v4l2_vb2_buffer struct 126 */ 127 static void __init_v4l2_vb2_buffer(struct vb2_buffer *vb) 128 { 129 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 130 131 vbuf->request_fd = -1; 132 } 133 134 static void __copy_timestamp(struct vb2_buffer *vb, const void *pb) 135 { 136 const struct v4l2_buffer *b = pb; 137 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 138 struct vb2_queue *q = vb->vb2_queue; 139 140 if (q->is_output) { 141 /* 142 * For output buffers copy the timestamp if needed, 143 * and the timecode field and flag if needed. 144 */ 145 if (q->copy_timestamp) 146 vb->timestamp = timeval_to_ns(&b->timestamp); 147 vbuf->flags |= b->flags & V4L2_BUF_FLAG_TIMECODE; 148 if (b->flags & V4L2_BUF_FLAG_TIMECODE) 149 vbuf->timecode = b->timecode; 150 } 151 }; 152 153 static void vb2_warn_zero_bytesused(struct vb2_buffer *vb) 154 { 155 static bool check_once; 156 157 if (check_once) 158 return; 159 160 check_once = true; 161 WARN_ON(1); 162 163 pr_warn("use of bytesused == 0 is deprecated and will be removed in the future,\n"); 164 if (vb->vb2_queue->allow_zero_bytesused) 165 pr_warn("use VIDIOC_DECODER_CMD(V4L2_DEC_CMD_STOP) instead.\n"); 166 else 167 pr_warn("use the actual size instead.\n"); 168 } 169 170 static int vb2_fill_vb2_v4l2_buffer(struct vb2_buffer *vb, struct v4l2_buffer *b) 171 { 172 struct vb2_queue *q = vb->vb2_queue; 173 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 174 struct vb2_plane *planes = vbuf->planes; 175 unsigned int plane; 176 int ret; 177 178 ret = __verify_length(vb, b); 179 if (ret < 0) { 180 dprintk(1, "plane parameters verification failed: %d\n", ret); 181 return ret; 182 } 183 if (b->field == V4L2_FIELD_ALTERNATE && q->is_output) { 184 /* 185 * If the format's field is ALTERNATE, then the buffer's field 186 * should be either TOP or BOTTOM, not ALTERNATE since that 187 * makes no sense. The driver has to know whether the 188 * buffer represents a top or a bottom field in order to 189 * program any DMA correctly. Using ALTERNATE is wrong, since 190 * that just says that it is either a top or a bottom field, 191 * but not which of the two it is. 192 */ 193 dprintk(1, "the field is incorrectly set to ALTERNATE for an output buffer\n"); 194 return -EINVAL; 195 } 196 vbuf->sequence = 0; 197 vbuf->request_fd = -1; 198 199 if (V4L2_TYPE_IS_MULTIPLANAR(b->type)) { 200 switch (b->memory) { 201 case VB2_MEMORY_USERPTR: 202 for (plane = 0; plane < vb->num_planes; ++plane) { 203 planes[plane].m.userptr = 204 b->m.planes[plane].m.userptr; 205 planes[plane].length = 206 b->m.planes[plane].length; 207 } 208 break; 209 case VB2_MEMORY_DMABUF: 210 for (plane = 0; plane < vb->num_planes; ++plane) { 211 planes[plane].m.fd = 212 b->m.planes[plane].m.fd; 213 planes[plane].length = 214 b->m.planes[plane].length; 215 } 216 break; 217 default: 218 for (plane = 0; plane < vb->num_planes; ++plane) { 219 planes[plane].m.offset = 220 vb->planes[plane].m.offset; 221 planes[plane].length = 222 vb->planes[plane].length; 223 } 224 break; 225 } 226 227 /* Fill in driver-provided information for OUTPUT types */ 228 if (V4L2_TYPE_IS_OUTPUT(b->type)) { 229 /* 230 * Will have to go up to b->length when API starts 231 * accepting variable number of planes. 232 * 233 * If bytesused == 0 for the output buffer, then fall 234 * back to the full buffer size. In that case 235 * userspace clearly never bothered to set it and 236 * it's a safe assumption that they really meant to 237 * use the full plane sizes. 238 * 239 * Some drivers, e.g. old codec drivers, use bytesused == 0 240 * as a way to indicate that streaming is finished. 241 * In that case, the driver should use the 242 * allow_zero_bytesused flag to keep old userspace 243 * applications working. 244 */ 245 for (plane = 0; plane < vb->num_planes; ++plane) { 246 struct vb2_plane *pdst = &planes[plane]; 247 struct v4l2_plane *psrc = &b->m.planes[plane]; 248 249 if (psrc->bytesused == 0) 250 vb2_warn_zero_bytesused(vb); 251 252 if (vb->vb2_queue->allow_zero_bytesused) 253 pdst->bytesused = psrc->bytesused; 254 else 255 pdst->bytesused = psrc->bytesused ? 256 psrc->bytesused : pdst->length; 257 pdst->data_offset = psrc->data_offset; 258 } 259 } 260 } else { 261 /* 262 * Single-planar buffers do not use planes array, 263 * so fill in relevant v4l2_buffer struct fields instead. 264 * In videobuf we use our internal V4l2_planes struct for 265 * single-planar buffers as well, for simplicity. 266 * 267 * If bytesused == 0 for the output buffer, then fall back 268 * to the full buffer size as that's a sensible default. 269 * 270 * Some drivers, e.g. old codec drivers, use bytesused == 0 as 271 * a way to indicate that streaming is finished. In that case, 272 * the driver should use the allow_zero_bytesused flag to keep 273 * old userspace applications working. 274 */ 275 switch (b->memory) { 276 case VB2_MEMORY_USERPTR: 277 planes[0].m.userptr = b->m.userptr; 278 planes[0].length = b->length; 279 break; 280 case VB2_MEMORY_DMABUF: 281 planes[0].m.fd = b->m.fd; 282 planes[0].length = b->length; 283 break; 284 default: 285 planes[0].m.offset = vb->planes[0].m.offset; 286 planes[0].length = vb->planes[0].length; 287 break; 288 } 289 290 planes[0].data_offset = 0; 291 if (V4L2_TYPE_IS_OUTPUT(b->type)) { 292 if (b->bytesused == 0) 293 vb2_warn_zero_bytesused(vb); 294 295 if (vb->vb2_queue->allow_zero_bytesused) 296 planes[0].bytesused = b->bytesused; 297 else 298 planes[0].bytesused = b->bytesused ? 299 b->bytesused : planes[0].length; 300 } else 301 planes[0].bytesused = 0; 302 303 } 304 305 /* Zero flags that we handle */ 306 vbuf->flags = b->flags & ~V4L2_BUFFER_MASK_FLAGS; 307 if (!vb->vb2_queue->copy_timestamp || !V4L2_TYPE_IS_OUTPUT(b->type)) { 308 /* 309 * Non-COPY timestamps and non-OUTPUT queues will get 310 * their timestamp and timestamp source flags from the 311 * queue. 312 */ 313 vbuf->flags &= ~V4L2_BUF_FLAG_TSTAMP_SRC_MASK; 314 } 315 316 if (V4L2_TYPE_IS_OUTPUT(b->type)) { 317 /* 318 * For output buffers mask out the timecode flag: 319 * this will be handled later in vb2_qbuf(). 320 * The 'field' is valid metadata for this output buffer 321 * and so that needs to be copied here. 322 */ 323 vbuf->flags &= ~V4L2_BUF_FLAG_TIMECODE; 324 vbuf->field = b->field; 325 } else { 326 /* Zero any output buffer flags as this is a capture buffer */ 327 vbuf->flags &= ~V4L2_BUFFER_OUT_FLAGS; 328 /* Zero last flag, this is a signal from driver to userspace */ 329 vbuf->flags &= ~V4L2_BUF_FLAG_LAST; 330 } 331 332 return 0; 333 } 334 335 static int vb2_queue_or_prepare_buf(struct vb2_queue *q, struct media_device *mdev, 336 struct v4l2_buffer *b, 337 const char *opname, 338 struct media_request **p_req) 339 { 340 struct media_request *req; 341 struct vb2_v4l2_buffer *vbuf; 342 struct vb2_buffer *vb; 343 int ret; 344 345 if (b->type != q->type) { 346 dprintk(1, "%s: invalid buffer type\n", opname); 347 return -EINVAL; 348 } 349 350 if (b->index >= q->num_buffers) { 351 dprintk(1, "%s: buffer index out of range\n", opname); 352 return -EINVAL; 353 } 354 355 if (q->bufs[b->index] == NULL) { 356 /* Should never happen */ 357 dprintk(1, "%s: buffer is NULL\n", opname); 358 return -EINVAL; 359 } 360 361 if (b->memory != q->memory) { 362 dprintk(1, "%s: invalid memory type\n", opname); 363 return -EINVAL; 364 } 365 366 vb = q->bufs[b->index]; 367 vbuf = to_vb2_v4l2_buffer(vb); 368 ret = __verify_planes_array(vb, b); 369 if (ret) 370 return ret; 371 372 if (!vb->prepared) { 373 /* Copy relevant information provided by the userspace */ 374 memset(vbuf->planes, 0, 375 sizeof(vbuf->planes[0]) * vb->num_planes); 376 ret = vb2_fill_vb2_v4l2_buffer(vb, b); 377 if (ret) 378 return ret; 379 } 380 381 if (!(b->flags & V4L2_BUF_FLAG_REQUEST_FD)) 382 return 0; 383 384 /* 385 * For proper locking when queueing a request you need to be able 386 * to lock access to the vb2 queue, so check that there is a lock 387 * that we can use. In addition p_req must be non-NULL. 388 */ 389 if (WARN_ON(!q->lock || !p_req)) 390 return -EINVAL; 391 392 /* 393 * Make sure this op is implemented by the driver. It's easy to forget 394 * this callback, but is it important when canceling a buffer in a 395 * queued request. 396 */ 397 if (WARN_ON(!q->ops->buf_request_complete)) 398 return -EINVAL; 399 400 if (vb->state != VB2_BUF_STATE_DEQUEUED) { 401 dprintk(1, "%s: buffer is not in dequeued state\n", opname); 402 return -EINVAL; 403 } 404 405 if (b->request_fd < 0) { 406 dprintk(1, "%s: request_fd < 0\n", opname); 407 return -EINVAL; 408 } 409 410 req = media_request_get_by_fd(mdev, b->request_fd); 411 if (IS_ERR(req)) { 412 dprintk(1, "%s: invalid request_fd\n", opname); 413 return PTR_ERR(req); 414 } 415 416 /* 417 * Early sanity check. This is checked again when the buffer 418 * is bound to the request in vb2_core_qbuf(). 419 */ 420 if (req->state != MEDIA_REQUEST_STATE_IDLE && 421 req->state != MEDIA_REQUEST_STATE_UPDATING) { 422 dprintk(1, "%s: request is not idle\n", opname); 423 media_request_put(req); 424 return -EBUSY; 425 } 426 427 *p_req = req; 428 vbuf->request_fd = b->request_fd; 429 430 return 0; 431 } 432 433 /* 434 * __fill_v4l2_buffer() - fill in a struct v4l2_buffer with information to be 435 * returned to userspace 436 */ 437 static void __fill_v4l2_buffer(struct vb2_buffer *vb, void *pb) 438 { 439 struct v4l2_buffer *b = pb; 440 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 441 struct vb2_queue *q = vb->vb2_queue; 442 unsigned int plane; 443 444 /* Copy back data such as timestamp, flags, etc. */ 445 b->index = vb->index; 446 b->type = vb->type; 447 b->memory = vb->memory; 448 b->bytesused = 0; 449 450 b->flags = vbuf->flags; 451 b->field = vbuf->field; 452 b->timestamp = ns_to_timeval(vb->timestamp); 453 b->timecode = vbuf->timecode; 454 b->sequence = vbuf->sequence; 455 b->reserved2 = 0; 456 b->request_fd = 0; 457 458 if (q->is_multiplanar) { 459 /* 460 * Fill in plane-related data if userspace provided an array 461 * for it. The caller has already verified memory and size. 462 */ 463 b->length = vb->num_planes; 464 for (plane = 0; plane < vb->num_planes; ++plane) { 465 struct v4l2_plane *pdst = &b->m.planes[plane]; 466 struct vb2_plane *psrc = &vb->planes[plane]; 467 468 pdst->bytesused = psrc->bytesused; 469 pdst->length = psrc->length; 470 if (q->memory == VB2_MEMORY_MMAP) 471 pdst->m.mem_offset = psrc->m.offset; 472 else if (q->memory == VB2_MEMORY_USERPTR) 473 pdst->m.userptr = psrc->m.userptr; 474 else if (q->memory == VB2_MEMORY_DMABUF) 475 pdst->m.fd = psrc->m.fd; 476 pdst->data_offset = psrc->data_offset; 477 memset(pdst->reserved, 0, sizeof(pdst->reserved)); 478 } 479 } else { 480 /* 481 * We use length and offset in v4l2_planes array even for 482 * single-planar buffers, but userspace does not. 483 */ 484 b->length = vb->planes[0].length; 485 b->bytesused = vb->planes[0].bytesused; 486 if (q->memory == VB2_MEMORY_MMAP) 487 b->m.offset = vb->planes[0].m.offset; 488 else if (q->memory == VB2_MEMORY_USERPTR) 489 b->m.userptr = vb->planes[0].m.userptr; 490 else if (q->memory == VB2_MEMORY_DMABUF) 491 b->m.fd = vb->planes[0].m.fd; 492 } 493 494 /* 495 * Clear any buffer state related flags. 496 */ 497 b->flags &= ~V4L2_BUFFER_MASK_FLAGS; 498 b->flags |= q->timestamp_flags & V4L2_BUF_FLAG_TIMESTAMP_MASK; 499 if (!q->copy_timestamp) { 500 /* 501 * For non-COPY timestamps, drop timestamp source bits 502 * and obtain the timestamp source from the queue. 503 */ 504 b->flags &= ~V4L2_BUF_FLAG_TSTAMP_SRC_MASK; 505 b->flags |= q->timestamp_flags & V4L2_BUF_FLAG_TSTAMP_SRC_MASK; 506 } 507 508 switch (vb->state) { 509 case VB2_BUF_STATE_QUEUED: 510 case VB2_BUF_STATE_ACTIVE: 511 b->flags |= V4L2_BUF_FLAG_QUEUED; 512 break; 513 case VB2_BUF_STATE_IN_REQUEST: 514 b->flags |= V4L2_BUF_FLAG_IN_REQUEST; 515 break; 516 case VB2_BUF_STATE_ERROR: 517 b->flags |= V4L2_BUF_FLAG_ERROR; 518 /* fall through */ 519 case VB2_BUF_STATE_DONE: 520 b->flags |= V4L2_BUF_FLAG_DONE; 521 break; 522 case VB2_BUF_STATE_PREPARING: 523 case VB2_BUF_STATE_DEQUEUED: 524 case VB2_BUF_STATE_REQUEUEING: 525 /* nothing */ 526 break; 527 } 528 529 if ((vb->state == VB2_BUF_STATE_DEQUEUED || 530 vb->state == VB2_BUF_STATE_IN_REQUEST) && 531 vb->synced && vb->prepared) 532 b->flags |= V4L2_BUF_FLAG_PREPARED; 533 534 if (vb2_buffer_in_use(q, vb)) 535 b->flags |= V4L2_BUF_FLAG_MAPPED; 536 if (vbuf->request_fd >= 0) { 537 b->flags |= V4L2_BUF_FLAG_REQUEST_FD; 538 b->request_fd = vbuf->request_fd; 539 } 540 541 if (!q->is_output && 542 b->flags & V4L2_BUF_FLAG_DONE && 543 b->flags & V4L2_BUF_FLAG_LAST) 544 q->last_buffer_dequeued = true; 545 } 546 547 /* 548 * __fill_vb2_buffer() - fill a vb2_buffer with information provided in a 549 * v4l2_buffer by the userspace. It also verifies that struct 550 * v4l2_buffer has a valid number of planes. 551 */ 552 static int __fill_vb2_buffer(struct vb2_buffer *vb, struct vb2_plane *planes) 553 { 554 struct vb2_v4l2_buffer *vbuf = to_vb2_v4l2_buffer(vb); 555 unsigned int plane; 556 557 if (!vb->vb2_queue->is_output || !vb->vb2_queue->copy_timestamp) 558 vb->timestamp = 0; 559 560 for (plane = 0; plane < vb->num_planes; ++plane) { 561 if (vb->vb2_queue->memory != VB2_MEMORY_MMAP) { 562 planes[plane].m = vbuf->planes[plane].m; 563 planes[plane].length = vbuf->planes[plane].length; 564 } 565 planes[plane].bytesused = vbuf->planes[plane].bytesused; 566 planes[plane].data_offset = vbuf->planes[plane].data_offset; 567 } 568 return 0; 569 } 570 571 static const struct vb2_buf_ops v4l2_buf_ops = { 572 .verify_planes_array = __verify_planes_array_core, 573 .init_buffer = __init_v4l2_vb2_buffer, 574 .fill_user_buffer = __fill_v4l2_buffer, 575 .fill_vb2_buffer = __fill_vb2_buffer, 576 .copy_timestamp = __copy_timestamp, 577 }; 578 579 /* 580 * vb2_querybuf() - query video buffer information 581 * @q: videobuf queue 582 * @b: buffer struct passed from userspace to vidioc_querybuf handler 583 * in driver 584 * 585 * Should be called from vidioc_querybuf ioctl handler in driver. 586 * This function will verify the passed v4l2_buffer structure and fill the 587 * relevant information for the userspace. 588 * 589 * The return values from this function are intended to be directly returned 590 * from vidioc_querybuf handler in driver. 591 */ 592 int vb2_querybuf(struct vb2_queue *q, struct v4l2_buffer *b) 593 { 594 struct vb2_buffer *vb; 595 int ret; 596 597 if (b->type != q->type) { 598 dprintk(1, "wrong buffer type\n"); 599 return -EINVAL; 600 } 601 602 if (b->index >= q->num_buffers) { 603 dprintk(1, "buffer index out of range\n"); 604 return -EINVAL; 605 } 606 vb = q->bufs[b->index]; 607 ret = __verify_planes_array(vb, b); 608 if (!ret) 609 vb2_core_querybuf(q, b->index, b); 610 return ret; 611 } 612 EXPORT_SYMBOL(vb2_querybuf); 613 614 int vb2_reqbufs(struct vb2_queue *q, struct v4l2_requestbuffers *req) 615 { 616 int ret = vb2_verify_memory_type(q, req->memory, req->type); 617 618 return ret ? ret : vb2_core_reqbufs(q, req->memory, &req->count); 619 } 620 EXPORT_SYMBOL_GPL(vb2_reqbufs); 621 622 int vb2_prepare_buf(struct vb2_queue *q, struct media_device *mdev, 623 struct v4l2_buffer *b) 624 { 625 int ret; 626 627 if (vb2_fileio_is_active(q)) { 628 dprintk(1, "file io in progress\n"); 629 return -EBUSY; 630 } 631 632 if (b->flags & V4L2_BUF_FLAG_REQUEST_FD) 633 return -EINVAL; 634 635 ret = vb2_queue_or_prepare_buf(q, mdev, b, "prepare_buf", NULL); 636 637 return ret ? ret : vb2_core_prepare_buf(q, b->index, b); 638 } 639 EXPORT_SYMBOL_GPL(vb2_prepare_buf); 640 641 int vb2_create_bufs(struct vb2_queue *q, struct v4l2_create_buffers *create) 642 { 643 unsigned requested_planes = 1; 644 unsigned requested_sizes[VIDEO_MAX_PLANES]; 645 struct v4l2_format *f = &create->format; 646 int ret = vb2_verify_memory_type(q, create->memory, f->type); 647 unsigned i; 648 649 create->index = q->num_buffers; 650 if (create->count == 0) 651 return ret != -EBUSY ? ret : 0; 652 653 switch (f->type) { 654 case V4L2_BUF_TYPE_VIDEO_CAPTURE_MPLANE: 655 case V4L2_BUF_TYPE_VIDEO_OUTPUT_MPLANE: 656 requested_planes = f->fmt.pix_mp.num_planes; 657 if (requested_planes == 0 || 658 requested_planes > VIDEO_MAX_PLANES) 659 return -EINVAL; 660 for (i = 0; i < requested_planes; i++) 661 requested_sizes[i] = 662 f->fmt.pix_mp.plane_fmt[i].sizeimage; 663 break; 664 case V4L2_BUF_TYPE_VIDEO_CAPTURE: 665 case V4L2_BUF_TYPE_VIDEO_OUTPUT: 666 requested_sizes[0] = f->fmt.pix.sizeimage; 667 break; 668 case V4L2_BUF_TYPE_VBI_CAPTURE: 669 case V4L2_BUF_TYPE_VBI_OUTPUT: 670 requested_sizes[0] = f->fmt.vbi.samples_per_line * 671 (f->fmt.vbi.count[0] + f->fmt.vbi.count[1]); 672 break; 673 case V4L2_BUF_TYPE_SLICED_VBI_CAPTURE: 674 case V4L2_BUF_TYPE_SLICED_VBI_OUTPUT: 675 requested_sizes[0] = f->fmt.sliced.io_size; 676 break; 677 case V4L2_BUF_TYPE_SDR_CAPTURE: 678 case V4L2_BUF_TYPE_SDR_OUTPUT: 679 requested_sizes[0] = f->fmt.sdr.buffersize; 680 break; 681 case V4L2_BUF_TYPE_META_CAPTURE: 682 requested_sizes[0] = f->fmt.meta.buffersize; 683 break; 684 default: 685 return -EINVAL; 686 } 687 for (i = 0; i < requested_planes; i++) 688 if (requested_sizes[i] == 0) 689 return -EINVAL; 690 return ret ? ret : vb2_core_create_bufs(q, create->memory, 691 &create->count, requested_planes, requested_sizes); 692 } 693 EXPORT_SYMBOL_GPL(vb2_create_bufs); 694 695 int vb2_qbuf(struct vb2_queue *q, struct media_device *mdev, 696 struct v4l2_buffer *b) 697 { 698 struct media_request *req = NULL; 699 int ret; 700 701 if (vb2_fileio_is_active(q)) { 702 dprintk(1, "file io in progress\n"); 703 return -EBUSY; 704 } 705 706 ret = vb2_queue_or_prepare_buf(q, mdev, b, "qbuf", &req); 707 if (ret) 708 return ret; 709 ret = vb2_core_qbuf(q, b->index, b, req); 710 if (req) 711 media_request_put(req); 712 return ret; 713 } 714 EXPORT_SYMBOL_GPL(vb2_qbuf); 715 716 int vb2_dqbuf(struct vb2_queue *q, struct v4l2_buffer *b, bool nonblocking) 717 { 718 int ret; 719 720 if (vb2_fileio_is_active(q)) { 721 dprintk(1, "file io in progress\n"); 722 return -EBUSY; 723 } 724 725 if (b->type != q->type) { 726 dprintk(1, "invalid buffer type\n"); 727 return -EINVAL; 728 } 729 730 ret = vb2_core_dqbuf(q, NULL, b, nonblocking); 731 732 /* 733 * After calling the VIDIOC_DQBUF V4L2_BUF_FLAG_DONE must be 734 * cleared. 735 */ 736 b->flags &= ~V4L2_BUF_FLAG_DONE; 737 738 return ret; 739 } 740 EXPORT_SYMBOL_GPL(vb2_dqbuf); 741 742 int vb2_streamon(struct vb2_queue *q, enum v4l2_buf_type type) 743 { 744 if (vb2_fileio_is_active(q)) { 745 dprintk(1, "file io in progress\n"); 746 return -EBUSY; 747 } 748 return vb2_core_streamon(q, type); 749 } 750 EXPORT_SYMBOL_GPL(vb2_streamon); 751 752 int vb2_streamoff(struct vb2_queue *q, enum v4l2_buf_type type) 753 { 754 if (vb2_fileio_is_active(q)) { 755 dprintk(1, "file io in progress\n"); 756 return -EBUSY; 757 } 758 return vb2_core_streamoff(q, type); 759 } 760 EXPORT_SYMBOL_GPL(vb2_streamoff); 761 762 int vb2_expbuf(struct vb2_queue *q, struct v4l2_exportbuffer *eb) 763 { 764 return vb2_core_expbuf(q, &eb->fd, eb->type, eb->index, 765 eb->plane, eb->flags); 766 } 767 EXPORT_SYMBOL_GPL(vb2_expbuf); 768 769 int vb2_queue_init(struct vb2_queue *q) 770 { 771 /* 772 * Sanity check 773 */ 774 if (WARN_ON(!q) || 775 WARN_ON(q->timestamp_flags & 776 ~(V4L2_BUF_FLAG_TIMESTAMP_MASK | 777 V4L2_BUF_FLAG_TSTAMP_SRC_MASK))) 778 return -EINVAL; 779 780 /* Warn that the driver should choose an appropriate timestamp type */ 781 WARN_ON((q->timestamp_flags & V4L2_BUF_FLAG_TIMESTAMP_MASK) == 782 V4L2_BUF_FLAG_TIMESTAMP_UNKNOWN); 783 784 /* Warn that vb2_memory should match with v4l2_memory */ 785 if (WARN_ON(VB2_MEMORY_MMAP != (int)V4L2_MEMORY_MMAP) 786 || WARN_ON(VB2_MEMORY_USERPTR != (int)V4L2_MEMORY_USERPTR) 787 || WARN_ON(VB2_MEMORY_DMABUF != (int)V4L2_MEMORY_DMABUF)) 788 return -EINVAL; 789 790 if (q->buf_struct_size == 0) 791 q->buf_struct_size = sizeof(struct vb2_v4l2_buffer); 792 793 q->buf_ops = &v4l2_buf_ops; 794 q->is_multiplanar = V4L2_TYPE_IS_MULTIPLANAR(q->type); 795 q->is_output = V4L2_TYPE_IS_OUTPUT(q->type); 796 q->copy_timestamp = (q->timestamp_flags & V4L2_BUF_FLAG_TIMESTAMP_MASK) 797 == V4L2_BUF_FLAG_TIMESTAMP_COPY; 798 /* 799 * For compatibility with vb1: if QBUF hasn't been called yet, then 800 * return EPOLLERR as well. This only affects capture queues, output 801 * queues will always initialize waiting_for_buffers to false. 802 */ 803 q->quirk_poll_must_check_waiting_for_buffers = true; 804 805 return vb2_core_queue_init(q); 806 } 807 EXPORT_SYMBOL_GPL(vb2_queue_init); 808 809 void vb2_queue_release(struct vb2_queue *q) 810 { 811 vb2_core_queue_release(q); 812 } 813 EXPORT_SYMBOL_GPL(vb2_queue_release); 814 815 __poll_t vb2_poll(struct vb2_queue *q, struct file *file, poll_table *wait) 816 { 817 struct video_device *vfd = video_devdata(file); 818 __poll_t req_events = poll_requested_events(wait); 819 __poll_t res = 0; 820 821 if (test_bit(V4L2_FL_USES_V4L2_FH, &vfd->flags)) { 822 struct v4l2_fh *fh = file->private_data; 823 824 if (v4l2_event_pending(fh)) 825 res = EPOLLPRI; 826 else if (req_events & EPOLLPRI) 827 poll_wait(file, &fh->wait, wait); 828 } 829 830 return res | vb2_core_poll(q, file, wait); 831 } 832 EXPORT_SYMBOL_GPL(vb2_poll); 833 834 /* 835 * The following functions are not part of the vb2 core API, but are helper 836 * functions that plug into struct v4l2_ioctl_ops, struct v4l2_file_operations 837 * and struct vb2_ops. 838 * They contain boilerplate code that most if not all drivers have to do 839 * and so they simplify the driver code. 840 */ 841 842 /* The queue is busy if there is a owner and you are not that owner. */ 843 static inline bool vb2_queue_is_busy(struct video_device *vdev, struct file *file) 844 { 845 return vdev->queue->owner && vdev->queue->owner != file->private_data; 846 } 847 848 /* vb2 ioctl helpers */ 849 850 int vb2_ioctl_reqbufs(struct file *file, void *priv, 851 struct v4l2_requestbuffers *p) 852 { 853 struct video_device *vdev = video_devdata(file); 854 int res = vb2_verify_memory_type(vdev->queue, p->memory, p->type); 855 856 if (res) 857 return res; 858 if (vb2_queue_is_busy(vdev, file)) 859 return -EBUSY; 860 res = vb2_core_reqbufs(vdev->queue, p->memory, &p->count); 861 /* If count == 0, then the owner has released all buffers and he 862 is no longer owner of the queue. Otherwise we have a new owner. */ 863 if (res == 0) 864 vdev->queue->owner = p->count ? file->private_data : NULL; 865 return res; 866 } 867 EXPORT_SYMBOL_GPL(vb2_ioctl_reqbufs); 868 869 int vb2_ioctl_create_bufs(struct file *file, void *priv, 870 struct v4l2_create_buffers *p) 871 { 872 struct video_device *vdev = video_devdata(file); 873 int res = vb2_verify_memory_type(vdev->queue, p->memory, 874 p->format.type); 875 876 p->index = vdev->queue->num_buffers; 877 /* 878 * If count == 0, then just check if memory and type are valid. 879 * Any -EBUSY result from vb2_verify_memory_type can be mapped to 0. 880 */ 881 if (p->count == 0) 882 return res != -EBUSY ? res : 0; 883 if (res) 884 return res; 885 if (vb2_queue_is_busy(vdev, file)) 886 return -EBUSY; 887 888 res = vb2_create_bufs(vdev->queue, p); 889 if (res == 0) 890 vdev->queue->owner = file->private_data; 891 return res; 892 } 893 EXPORT_SYMBOL_GPL(vb2_ioctl_create_bufs); 894 895 int vb2_ioctl_prepare_buf(struct file *file, void *priv, 896 struct v4l2_buffer *p) 897 { 898 struct video_device *vdev = video_devdata(file); 899 900 if (vb2_queue_is_busy(vdev, file)) 901 return -EBUSY; 902 return vb2_prepare_buf(vdev->queue, vdev->v4l2_dev->mdev, p); 903 } 904 EXPORT_SYMBOL_GPL(vb2_ioctl_prepare_buf); 905 906 int vb2_ioctl_querybuf(struct file *file, void *priv, struct v4l2_buffer *p) 907 { 908 struct video_device *vdev = video_devdata(file); 909 910 /* No need to call vb2_queue_is_busy(), anyone can query buffers. */ 911 return vb2_querybuf(vdev->queue, p); 912 } 913 EXPORT_SYMBOL_GPL(vb2_ioctl_querybuf); 914 915 int vb2_ioctl_qbuf(struct file *file, void *priv, struct v4l2_buffer *p) 916 { 917 struct video_device *vdev = video_devdata(file); 918 919 if (vb2_queue_is_busy(vdev, file)) 920 return -EBUSY; 921 return vb2_qbuf(vdev->queue, vdev->v4l2_dev->mdev, p); 922 } 923 EXPORT_SYMBOL_GPL(vb2_ioctl_qbuf); 924 925 int vb2_ioctl_dqbuf(struct file *file, void *priv, struct v4l2_buffer *p) 926 { 927 struct video_device *vdev = video_devdata(file); 928 929 if (vb2_queue_is_busy(vdev, file)) 930 return -EBUSY; 931 return vb2_dqbuf(vdev->queue, p, file->f_flags & O_NONBLOCK); 932 } 933 EXPORT_SYMBOL_GPL(vb2_ioctl_dqbuf); 934 935 int vb2_ioctl_streamon(struct file *file, void *priv, enum v4l2_buf_type i) 936 { 937 struct video_device *vdev = video_devdata(file); 938 939 if (vb2_queue_is_busy(vdev, file)) 940 return -EBUSY; 941 return vb2_streamon(vdev->queue, i); 942 } 943 EXPORT_SYMBOL_GPL(vb2_ioctl_streamon); 944 945 int vb2_ioctl_streamoff(struct file *file, void *priv, enum v4l2_buf_type i) 946 { 947 struct video_device *vdev = video_devdata(file); 948 949 if (vb2_queue_is_busy(vdev, file)) 950 return -EBUSY; 951 return vb2_streamoff(vdev->queue, i); 952 } 953 EXPORT_SYMBOL_GPL(vb2_ioctl_streamoff); 954 955 int vb2_ioctl_expbuf(struct file *file, void *priv, struct v4l2_exportbuffer *p) 956 { 957 struct video_device *vdev = video_devdata(file); 958 959 if (vb2_queue_is_busy(vdev, file)) 960 return -EBUSY; 961 return vb2_expbuf(vdev->queue, p); 962 } 963 EXPORT_SYMBOL_GPL(vb2_ioctl_expbuf); 964 965 /* v4l2_file_operations helpers */ 966 967 int vb2_fop_mmap(struct file *file, struct vm_area_struct *vma) 968 { 969 struct video_device *vdev = video_devdata(file); 970 971 return vb2_mmap(vdev->queue, vma); 972 } 973 EXPORT_SYMBOL_GPL(vb2_fop_mmap); 974 975 int _vb2_fop_release(struct file *file, struct mutex *lock) 976 { 977 struct video_device *vdev = video_devdata(file); 978 979 if (lock) 980 mutex_lock(lock); 981 if (file->private_data == vdev->queue->owner) { 982 vb2_queue_release(vdev->queue); 983 vdev->queue->owner = NULL; 984 } 985 if (lock) 986 mutex_unlock(lock); 987 return v4l2_fh_release(file); 988 } 989 EXPORT_SYMBOL_GPL(_vb2_fop_release); 990 991 int vb2_fop_release(struct file *file) 992 { 993 struct video_device *vdev = video_devdata(file); 994 struct mutex *lock = vdev->queue->lock ? vdev->queue->lock : vdev->lock; 995 996 return _vb2_fop_release(file, lock); 997 } 998 EXPORT_SYMBOL_GPL(vb2_fop_release); 999 1000 ssize_t vb2_fop_write(struct file *file, const char __user *buf, 1001 size_t count, loff_t *ppos) 1002 { 1003 struct video_device *vdev = video_devdata(file); 1004 struct mutex *lock = vdev->queue->lock ? vdev->queue->lock : vdev->lock; 1005 int err = -EBUSY; 1006 1007 if (!(vdev->queue->io_modes & VB2_WRITE)) 1008 return -EINVAL; 1009 if (lock && mutex_lock_interruptible(lock)) 1010 return -ERESTARTSYS; 1011 if (vb2_queue_is_busy(vdev, file)) 1012 goto exit; 1013 err = vb2_write(vdev->queue, buf, count, ppos, 1014 file->f_flags & O_NONBLOCK); 1015 if (vdev->queue->fileio) 1016 vdev->queue->owner = file->private_data; 1017 exit: 1018 if (lock) 1019 mutex_unlock(lock); 1020 return err; 1021 } 1022 EXPORT_SYMBOL_GPL(vb2_fop_write); 1023 1024 ssize_t vb2_fop_read(struct file *file, char __user *buf, 1025 size_t count, loff_t *ppos) 1026 { 1027 struct video_device *vdev = video_devdata(file); 1028 struct mutex *lock = vdev->queue->lock ? vdev->queue->lock : vdev->lock; 1029 int err = -EBUSY; 1030 1031 if (!(vdev->queue->io_modes & VB2_READ)) 1032 return -EINVAL; 1033 if (lock && mutex_lock_interruptible(lock)) 1034 return -ERESTARTSYS; 1035 if (vb2_queue_is_busy(vdev, file)) 1036 goto exit; 1037 err = vb2_read(vdev->queue, buf, count, ppos, 1038 file->f_flags & O_NONBLOCK); 1039 if (vdev->queue->fileio) 1040 vdev->queue->owner = file->private_data; 1041 exit: 1042 if (lock) 1043 mutex_unlock(lock); 1044 return err; 1045 } 1046 EXPORT_SYMBOL_GPL(vb2_fop_read); 1047 1048 __poll_t vb2_fop_poll(struct file *file, poll_table *wait) 1049 { 1050 struct video_device *vdev = video_devdata(file); 1051 struct vb2_queue *q = vdev->queue; 1052 struct mutex *lock = q->lock ? q->lock : vdev->lock; 1053 __poll_t res; 1054 void *fileio; 1055 1056 /* 1057 * If this helper doesn't know how to lock, then you shouldn't be using 1058 * it but you should write your own. 1059 */ 1060 WARN_ON(!lock); 1061 1062 if (lock && mutex_lock_interruptible(lock)) 1063 return EPOLLERR; 1064 1065 fileio = q->fileio; 1066 1067 res = vb2_poll(vdev->queue, file, wait); 1068 1069 /* If fileio was started, then we have a new queue owner. */ 1070 if (!fileio && q->fileio) 1071 q->owner = file->private_data; 1072 if (lock) 1073 mutex_unlock(lock); 1074 return res; 1075 } 1076 EXPORT_SYMBOL_GPL(vb2_fop_poll); 1077 1078 #ifndef CONFIG_MMU 1079 unsigned long vb2_fop_get_unmapped_area(struct file *file, unsigned long addr, 1080 unsigned long len, unsigned long pgoff, unsigned long flags) 1081 { 1082 struct video_device *vdev = video_devdata(file); 1083 1084 return vb2_get_unmapped_area(vdev->queue, addr, len, pgoff, flags); 1085 } 1086 EXPORT_SYMBOL_GPL(vb2_fop_get_unmapped_area); 1087 #endif 1088 1089 /* vb2_ops helpers. Only use if vq->lock is non-NULL. */ 1090 1091 void vb2_ops_wait_prepare(struct vb2_queue *vq) 1092 { 1093 mutex_unlock(vq->lock); 1094 } 1095 EXPORT_SYMBOL_GPL(vb2_ops_wait_prepare); 1096 1097 void vb2_ops_wait_finish(struct vb2_queue *vq) 1098 { 1099 mutex_lock(vq->lock); 1100 } 1101 EXPORT_SYMBOL_GPL(vb2_ops_wait_finish); 1102 1103 MODULE_DESCRIPTION("Driver helper framework for Video for Linux 2"); 1104 MODULE_AUTHOR("Pawel Osciak <pawel@osciak.com>, Marek Szyprowski"); 1105 MODULE_LICENSE("GPL"); 1106