vivi虚拟摄像头驱动程序

一、vivi虚拟摄像头驱动

基于V4L2（video for linux 2）摄像头驱动程序，我们减去不需要的ioctl_fops的函数，只增加ioctl函数增加的必要的摄像头流查询等函数；
 #include <linux/module.h>
 #include <linux/module.h>
 #include <linux/delay.h>
 #include <linux/errno.h>
 #include <linux/fs.h>
 #include <linux/kernel.h>
 #include <linux/slab.h>
 #include <linux/mm.h>
 #include <linux/ioport.h>
 #include <linux/init.h>
 #include <linux/sched.h>
 #include <linux/pci.h>
 #include <linux/random.h>
 #include <linux/version.h>
 #include <linux/mutex.h>
 #include <linux/videodev2.h>
 #include <linux/dma-mapping.h>
 #include <linux/interrupt.h>
 #include <linux/kthread.h>
 #include <linux/highmem.h>
 #include <linux/freezer.h>
 #include <media/videobuf-vmalloc.h>
 #include <media/v4l2-device.h>
 #include <media/v4l2-ioctl.h> static struct video_device *myvivi_device;
 static struct timer_list myvivi_timer;
 static struct list_head myvivi_vb_local_queue; static void myvivi_timer_function(unsigned long data)
 {
     struct videobuf_buffer *vb;
     void *vbuf;
     struct timeval ts;     /* 1. 构造数据: 从队列头部取出第1个videobuf到本地队列中, 填充数据
      */     /* 1.1 从本地队列取出第1个videobuf */
     if (list_empty(&myvivi_vb_local_queue)) {
         goto out;
     }     vb = list_entry(myvivi_vb_local_queue.next,
              struct videobuf_buffer, queue);     /* Nobody is waiting on this buffer, return */
     if (!waitqueue_active(&vb->done))
         goto out;     /* 1.2 填充数据 */
     vbuf = videobuf_to_vmalloc(vb);
     memset(vbuf, 0xff, vb->size);
     vb->field_count++;
     do_gettimeofday(&ts);
     vb->ts = ts;
     vb->state = VIDEOBUF_DONE;     /* 1.3 把videobuf从本地队列中删除 */
     list_del(&vb->queue);     /* 2. 唤醒进程: 唤醒videobuf->done上的进程 */
     wake_up(&vb->done); out:
     /* 3. 修改timer的超时时间 : 30fps, 1秒里有30帧数据
      *    每1/30 秒产生一帧数据
      */
     mod_timer(&myvivi_timer, jiffies + HZ/);
 }
 /* ------------------------------------------------------------------
     IOCTL vidioc handling
    ------------------------------------------------------------------*/
 static int myvivi_videoc_querycap(struct file *file, void  *priv,
                     struct v4l2_capability *cap)
 {
     //VIDIOC_QUERYCAP 命令通过结构 v4l2_capability 获取设备支持的操作模式：
     /*
         struct v4l2_capability {
         __u8    driver[16];         i.e. "bttv"
         __u8    card[32];        i.e. "Hauppauge WinTV"
         __u8    bus_info[32];    "PCI:" + pci_name(pci_dev)
         __u32   version;        should use KERNEL_VERSION()
         __u32    capabilities;     Device capabilities
         __u32    reserved[4];
         };
      *
      */
     strcpy(cap->driver, "myvivi");
     strcpy(cap->card, "myvivi");     cap->version = 0x0001;
     cap->capabilities =    V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING;        //V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_STREAMING表示一个视
                                                                             //频捕捉设备并且具有数据流控制模式     return ;
 } /* 用于列举、获得、测试、设置摄像头的数据的格式 */
 /* 列举支持哪种格式 */
 static int myvivi_vidioc_enum_fmt_vid_cap(struct file *file, void  *priv,
                     struct v4l2_fmtdesc *f)
 {
     /*
     *    F O R M A T   E N U M E R A T I O N     struct v4l2_fmtdesc {
         __u32            index;             // Format number      , 需要填充，从0开始，依次上升。
         enum v4l2_buf_type  type;          // buffer type       Camera，则填写V4L2_BUF_TYPE_VIDEO_CAPTURE
         __u32               flags;            //    如果压缩的，则Driver 填写：V4L2_FMT_FLAG_COMPRESSED，
         __u8            description[32];   // Description string ,image format的描述，如：YUV 4:2:2 (YUYV)
         __u32            pixelformat;       // Format fourcc ,所支持的格式。 如：V4L2_PIX_FMT_UYVY
         __u32            reserved[4];
     };
     */     if(f->index >= )
     {
         return -EINVAL;
     }
     strcpy(f->description, "4:2:2, packed, YUYV");     //从vivi_fmt结构体中可以看见：
     f->pixelformat = V4L2_PIX_FMT_YUYV;     return ;
 }
 struct v4l2_format myvivi_format;
 /* 返回当前所使用的格式 */
 static int myvivi_vidioc_g_fmt_vid_cap(struct file *file, void *priv,
                     struct v4l2_format *f)
 {     memcpy(f, &myvivi_format, sizeof(myvivi_format));     return ();
 } /* 测试驱动程序是否支持某种格式 */
 static int myvivi_vidioc_try_fmt_vid_cap(struct file *file, void *priv,
             struct v4l2_format *f)
 {
     unsigned int maxw, maxh;
     enum v4l2_field field;     if (f->fmt.pix.pixelformat != V4L2_PIX_FMT_YUYV)
         return -EINVAL;     field = f->fmt.pix.field;     if (field == V4L2_FIELD_ANY) {
         field = V4L2_FIELD_INTERLACED;
     } else if (V4L2_FIELD_INTERLACED != field) {
         return -EINVAL;
     }     maxw = ;
     maxh = ;     /*
      v4l2_format:
      struct v4l2_format {
         enum v4l2_buf_type type;
         union {
         struct v4l2_pix_format         pix;     // V4L2_BUF_TYPE_VIDEO_CAPTURE
         struct v4l2_window             win;     // V4L2_BUF_TYPE_VIDEO_OVERLAY
         struct v4l2_vbi_format         vbi;     // V4L2_BUF_TYPE_VBI_CAPTURE
         struct v4l2_sliced_vbi_format  sliced;  // V4L2_BUF_TYPE_SLICED_VBI_CAPTURE
         __u8   raw_data[200];                   // user-defined
     } fmt;
     };     其中
     enum v4l2_buf_type
     {
         V4L2_BUF_TYPE_VIDEO_CAPTURE        = 1,
         V4L2_BUF_TYPE_VIDEO_OUTPUT         = 2,
         V4L2_BUF_TYPE_VIDEO_OVERLAY        = 3,
         ...
         V4L2_BUF_TYPE_PRIVATE              = 0x80,
     };          struct v4l2_pix_format {
         __u32                   width;
         __u32                   height;
         __u32                   pixelformat;
         enum v4l2_field         field;
         __u32                   bytesperline;   // for padding, zero if unused
         __u32                   sizeimage;
         enum v4l2_colorspace    colorspace;
         __u32                   priv;           // private data, depends on pixelformat
     };       常见的捕获模式为 V4L2_BUF_TYPE_VIDEO_CAPTURE 即视频捕捉模式，在此模式下 fmt 联合体采用域 v4l2_pix_format：其中 width 为
     视频的宽、height 为视频的高、pixelformat 为视频数据格式（常见的值有 V4L2_PIX_FMT_YUV422P | V4L2_PIX_FMT_RGB565）、
     bytesperline 为一行图像占用的字节数、sizeimage 则为图像占用的总字节数、colorspace 指定设备的颜色空间。
     */
     //设置最小宽度和最大宽度等
     /* 调整format的width, height,
      * 计算bytesperline, sizeimage
      */
     v4l_bound_align_image(&f->fmt.pix.width, , maxw, , &f->fmt.pix.height, , maxh, , );     f->fmt.pix.bytesperline =
         (f->fmt.pix.width * ) >> ;        //颜色深度支持16
     f->fmt.pix.sizeimage =
         f->fmt.pix.height * f->fmt.pix.bytesperline;     return ;
 } static int myvivi_vidioc_s_fmt_vid_cap(struct file *file, void *priv,
                     struct v4l2_format *f)
 {
     int ret = myvivi_vidioc_try_fmt_vid_cap(file, NULL, f);
     if (ret < )
         return ret;     memcpy(&myvivi_format, f, sizeof(myvivi_format));     return ret;
 }
 /* 用于列举、获得、测试、设置摄像头的数据的格式 */ /* 队列操作1: 定义 */
 static struct videobuf_queue myvivi_vb_vidqueue;
 //自旋锁
 static spinlock_t myvivi_queue_slock; /* 参考documentations/video4linux/v4l2-framework.txt:
  *     drivers\media\video\videobuf-core.c
  ops->buf_setup   - calculates the size of the video buffers and avoid they
             to waste more than some maximum limit of RAM;
  ops->buf_prepare - fills the video buffer structs and calls
             videobuf_iolock() to alloc and prepare mmaped memory;
  ops->buf_queue   - advices the driver that another buffer were
             requested (by read() or by QBUF);
  ops->buf_release - frees any buffer that were allocated.  *
  */ /*****************buffer operations***********************/
 /* APP调用ioctl VIDIOC_REQBUFS(videobuf_reqbufs)时会导致此函数被调用,
  * 它重新调整count和size
  */
  static int myvivi_buffer_setup(struct videobuf_queue *vq, unsigned int *count, unsigned int *size)
 {
     //第一个参数为
     *size = myvivi_format.fmt.pix.sizeimage;     if ( == *count)
         *count = ;     return ;
 }
 /* APP调用ioctl VIDIOC_QBUF时导致此函数被调用,
  * 它会填充video_buffer结构体并调用videobuf_iolock来分配内存
  *
  */
 static int myvivi_buffer_prepare(struct videobuf_queue *vq, struct videobuf_buffer *vb,
                         enum v4l2_field field)
 {
     /* 1. 做些准备工作 */
      /* 0. 设置videobuf */
     vb->size = myvivi_format.fmt.pix.sizeimage;
     vb->bytesperline = myvivi_format.fmt.pix.bytesperline;
     vb->width  = myvivi_format.fmt.pix.width;
     vb->height = myvivi_format.fmt.pix.height;
     vb->field  = field;
 #if 0
     /* 2. 调用videobuf_iolock为类型为V4L2_MEMORY_USERPTR的videobuf分配内存 */
     if (VIDEOBUF_NEEDS_INIT == buf->vb.state) {
         rc = videobuf_iolock(vq, &buf->vb, NULL);
         if (rc < )
             goto fail;
     }
 #endif
     /* 3. 设置状态 */
     vb->state = VIDEOBUF_PREPARED;     return ;
 }
 /* APP调用ioctlVIDIOC_QBUF时:
  * 1. 先调用buf_prepare进行一些准备工作
  * 2. 把buf放入队列
  * 3. 调用buf_queue(起通知作用)通知应用程序正在请求调用
  */
 static void myvivi_buffer_queue(struct videobuf_queue *vq, struct videobuf_buffer *vb)
 {
     vb->state = VIDEOBUF_QUEUED;     /* 把videobuf放入本地一个队列尾部
      * 定时器处理函数就可以从本地队列取出videobuf，新增内核链表
      */
     list_add_tail(&vb->queue, &myvivi_vb_local_queue);
 }
 /* APP不再使用队列时, 用它来释放内存 */
 static void myvivi_buffer_release(struct videobuf_queue *vq,
                struct videobuf_buffer *vb)
 {
     videobuf_vmalloc_free(vb);
     vb->state = VIDEOBUF_NEEDS_INIT;
 }
 /*****************buffer operations***********************/ /* 缓冲区操作: 申请/查询/放入队列/取出队列 */
 static int myvivi_vidioc_reqbufs(struct file *file, void *priv,
               struct v4l2_requestbuffers *p)
 {
     return (videobuf_reqbufs(&myvivi_vb_vidqueue, p));
 }
 static int myvivi_vidioc_querybuf(struct file *file, void *priv, struct v4l2_buffer *p)
 {
     return (videobuf_querybuf(&myvivi_vb_vidqueue, p));
 }
 static int myvivi_vidioc_qbuf(struct file *file, void *priv, struct v4l2_buffer *p)
 {
     return (videobuf_qbuf(&myvivi_vb_vidqueue, p));
 }
 static int myvivi_vidioc_dqbuf(struct file *file, void *priv, struct v4l2_buffer *p)
 {
     return (videobuf_dqbuf(&myvivi_vb_vidqueue, p,
                 file->f_flags & O_NONBLOCK));
 }
 /* 缓冲区操作: 申请/查询/放入队列/取出队列 */
 /*------------------------------------------------------------------*/ /*-------------------开启与关闭--------------------------*/
 static int myvivi_vidioc_streamon(struct file *file, void *priv, enum v4l2_buf_type i)
 {
     return videobuf_streamon(&myvivi_vb_vidqueue);
 } static int myvivi_vidioc_streamoff(struct file *file, void *priv, enum v4l2_buf_type i)
 {
     videobuf_streamoff(&myvivi_vb_vidqueue);
     return ;
 }
 /*-------------------开启与关闭--------------------------*/ /* ------------------------------------------------------------------
     Videobuf operations
    ------------------------------------------------------------------*/
 static struct videobuf_queue_ops myvivi_video_qops = {
     .buf_setup      = myvivi_buffer_setup, /* 计算大小以免浪费 */
     .buf_prepare    = myvivi_buffer_prepare,
     .buf_queue      = myvivi_buffer_queue,
     .buf_release    = myvivi_buffer_release,
 };
 /* ------------------------------------------------------------------
     File operations for the device
    ------------------------------------------------------------------*/
 static int myvivi_open(struct file *file)
 {
     /* 队列操作2: 初始化 */
     videobuf_queue_vmalloc_init(&myvivi_vb_vidqueue, &myvivi_video_qops,
             NULL, &myvivi_queue_slock, V4L2_BUF_TYPE_VIDEO_CAPTURE, V4L2_FIELD_INTERLACED,
             sizeof(struct videobuf_buffer), NULL); /* V4L2_BUF_TYPE_VIDEO_CAPTURE用于视频捕获设备,倒数第2个参数是buffer的头部大小 */
     //myvivi_video_qops这个结构体我们怎么去处理它呢？     myvivi_timer.expires = jiffies + ;
     add_timer(&myvivi_timer);
     return ;
 }
 static int myvivi_close(struct file *file)
 {
     del_timer(&myvivi_timer);
     videobuf_stop(&myvivi_vb_vidqueue);        //stop the buf
     videobuf_mmap_free(&myvivi_vb_vidqueue);     return ;
 }
 static int myvivi_mmap(struct file *file, struct vm_area_struct *vma)
 {
     return videobuf_mmap_mapper(&myvivi_vb_vidqueue, vma);
 }
 static unsigned int myvivi_poll(struct file *file, struct poll_table_struct *wait)
 {
     return videobuf_poll_stream(file, &myvivi_vb_vidqueue, wait);
 }
 /*------------------------------------------------------------------*/ /*
 ------------------------------------------------------------
 struct video_device
     // device ops
     const struct v4l2_file_operations *fops;
 ------------------------------------------------------------
                 v4l2_file_operations=
                 {
                     struct module *owner;
                     long (*ioctl) (struct file *, unsigned int, unsigned long);
                 } ------------------------------------------------------------
     // callbacks
     void (*release)(struct video_device *vdev);
 ------------------------------------------------------------
 */
 static const struct v4l2_ioctl_ops myvivi_ioctl_ops =
 {
     //表示它是一个摄像头驱动
     .vidioc_querycap = myvivi_videoc_querycap,     /* 用于列举、获得、测试、设置摄像头的数据的格式 */
     .vidioc_enum_fmt_vid_cap  = myvivi_vidioc_enum_fmt_vid_cap,
     .vidioc_g_fmt_vid_cap     = myvivi_vidioc_g_fmt_vid_cap,
     .vidioc_try_fmt_vid_cap   = myvivi_vidioc_try_fmt_vid_cap,
     .vidioc_s_fmt_vid_cap     = myvivi_vidioc_s_fmt_vid_cap,     /* 缓冲区操作: 申请/查询/放入队列/取出队列 */
     .vidioc_reqbufs       = myvivi_vidioc_reqbufs,
     .vidioc_querybuf      = myvivi_vidioc_querybuf,
     .vidioc_qbuf          = myvivi_vidioc_qbuf,
     .vidioc_dqbuf         = myvivi_vidioc_dqbuf,     // 启动/停止
     .vidioc_streamon      = myvivi_vidioc_streamon,
     .vidioc_streamoff     = myvivi_vidioc_streamoff,   }; static const struct v4l2_file_operations myvivi_fops =
 {
     .owner = THIS_MODULE,
     .open = myvivi_open,
     .release = myvivi_close,
     .mmap = myvivi_mmap,
     .poll = myvivi_poll,        //提供了poll函数，还是会出现select timeout的情况，需要构造数据唤醒队列了，在poll_wait队列中休眠了
     .ioctl = video_ioctl2, /* v4l2 ioctl handler，在这个函数中设置了ioctl的用法 */
 };
 static void myvivi_release(struct video_device *vdev)
 { } static int myvivi_init(void)
 {
     int error=;     /* 1. 分配一个video_device函数，分配空间 */
     myvivi_device = video_device_alloc();     /* 2. 设置 */     /* 2.1 */
     myvivi_device->release = myvivi_release;
     /* 2.2 */
     myvivi_device->fops    = &myvivi_fops;
     /* 2.3 */
     myvivi_device->ioctl_ops = &myvivi_ioctl_ops;     /* 2.4 队列操作
     * a.定义/初始化一个队列（会用到一个自旋锁）
     */
     spin_lock_init(&myvivi_queue_slock);     /*
     #define VFL_TYPE_GRABBER    0        //表明是一个图像采集设备-包括摄像头、调谐器
     #define VFL_TYPE_VBI        1        //从视频消隐的时间段取得信息的设备
     #define VFL_TYPE_RADIO        2        //代表无线电设备
     #define VFL_TYPE_VTX        3        //代表视传设备
     #define VFL_TYPE_MAX        4
     @nr:   which device number (0 == /dev/video0, 1 == /dev/video1, ...
              -1 == first free)
     */
     /* 3.注册，第二个参数为VFL_TYPE_GRABBER， -1 为 first free */
     error = video_register_device(myvivi_device, VFL_TYPE_GRABBER, -);     //用定时器产生数据并唤醒进程
     init_timer(&myvivi_timer);
     myvivi_timer.function = myvivi_timer_function;     INIT_LIST_HEAD(&myvivi_vb_local_queue);
     return error;
 }; /*
  *    出口函数
  *
  */
 static void myvivi_exit(void)
 {
     video_unregister_device(myvivi_device);
     video_device_release(myvivi_device);
 } module_init(myvivi_init);
 module_exit(myvivi_exit);
 MODULE_LICENSE("GPL");