conditional alloc_ctx type based on kernel version
[vdummy] / vdummy.c
index d107e59..240d4f8 100644 (file)
--- a/vdummy.c
+++ b/vdummy.c
@@ -5,9 +5,9 @@
 #include <linux/slab.h>
 #include <linux/hrtimer.h>
 #include <linux/ktime.h>
+#include <linux/version.h>
 #include <asm/uaccess.h>
 #include <linux/videodev2.h>
-#include <media/videobuf2-v4l2.h>
 #include <media/videobuf2-vmalloc.h>
 #include <media/v4l2-dev.h>
 #include <media/v4l2-ioctl.h>
@@ -25,6 +25,7 @@ static void shutdown(void);
 
 static int ioctl_querycap(struct file *file, void *fh, struct v4l2_capability *cap);
 static int ioctl_enum_fmt(struct file *file, void *fh, struct v4l2_fmtdesc *fdesc);
+static int ioctl_enum_framesz(struct file *file, void *fh, struct v4l2_frmsizeenum *fsz);
 static int ioctl_get_fmt(struct file *file, void *fh, struct v4l2_format *fmt);
 static int ioctl_set_fmt(struct file *file, void *fh, struct v4l2_format *fmt);
 static int ioctl_try_fmt(struct file *file, void *fh, struct v4l2_format *fmt);
@@ -32,8 +33,15 @@ static int ioctl_get_parm(struct file *file, void *fh, struct v4l2_streamparm *f
 static int ioctl_set_parm(struct file *file, void *fh, struct v4l2_streamparm *fmt);
 static int ioctl_queryctl(struct file *file, void *fh, struct v4l2_queryctrl *ctl);
 
+/* TODO find out exact version where this switched over */
+#if LINUX_VERSION_CODE < KERNEL_VERSION(4,7,0)
+#define ALLOC_CTX_TYPE void
+#else
+#define ALLOC_CTX_TYPE struct device
+#endif
+
 static int queue_setup(struct vb2_queue *vbq, unsigned int *nbuf,
-               unsigned int *nplanes, unsigned int *sizes, void **alloc_ctx);
+               unsigned int *nplanes, unsigned int *sizes, ALLOC_CTX_TYPE **alloc_ctx);
 static int buf_prepare(struct vb2_buffer *vb);
 static void buf_queue(struct vb2_buffer *vb);
 static int start_streaming(struct vb2_queue *vbq, unsigned int count);
@@ -50,7 +58,7 @@ module_exit(shutdown);
 
 MODULE_LICENSE("GPL");
 MODULE_AUTHOR("John Tsiombikas");
-MODULE_DESCRIPTION("v4l2 test module");
+MODULE_DESCRIPTION("Dummy V4L2 device");
 
 static struct mutex mutex;
 static struct video_device *vdev;
@@ -132,6 +140,7 @@ static int init(void)
 
        iops.vidioc_querycap = ioctl_querycap;
        iops.vidioc_enum_fmt_vid_cap = ioctl_enum_fmt;
+       iops.vidioc_enum_framesizes = ioctl_enum_framesz;
        iops.vidioc_g_fmt_vid_cap = ioctl_get_fmt;
        iops.vidioc_s_fmt_vid_cap = ioctl_set_fmt;
        iops.vidioc_try_fmt_vid_cap = ioctl_try_fmt;
@@ -167,12 +176,13 @@ static int init(void)
                return res;
        }
 
-       printk(KERN_INFO "vdummy device initialized\n");
+       printk(KERN_INFO "vdummy: initialized\n");
        return 0;
 }
 
 static void shutdown(void)
 {
+       hrtimer_cancel(&timer);
        video_unregister_device(vdev);
        v4l2_device_unregister(&v4l2_dev);
        kfree(frame);
@@ -181,7 +191,7 @@ static void shutdown(void)
 static int ioctl_querycap(struct file *file, void *fh, struct v4l2_capability *cap)
 {
        strcpy(cap->driver, KBUILD_MODNAME);
-       strcpy(cap->card, "Test V4L2 device");
+       strcpy(cap->card, "Dummy V4L2 device");
        strcpy(cap->bus_info, "nobus");
        cap->device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_READWRITE | V4L2_CAP_STREAMING;
        cap->capabilities = cap->device_caps | V4L2_CAP_DEVICE_CAPS;
@@ -199,6 +209,18 @@ static int ioctl_enum_fmt(struct file *file, void *fh, struct v4l2_fmtdesc *fdes
        return 0;
 }
 
+static int ioctl_enum_framesz(struct file *file, void *fh, struct v4l2_frmsizeenum *fsz)
+{
+       if(fsz->index != 0) {
+               return -EINVAL;
+       }
+       fsz->type = V4L2_FRMSIZE_TYPE_CONTINUOUS;
+       fsz->stepwise.min_width = fsz->stepwise.min_height = 1;
+       fsz->stepwise.max_width = fsz->stepwise.max_height = 8192;
+       fsz->stepwise.step_width = fsz->stepwise.step_height = 1;
+       return 0;
+}
+
 static int ioctl_get_fmt(struct file *file, void *fh, struct v4l2_format *fmt)
 {
        if(fmt->type != V4L2_BUF_TYPE_VIDEO_CAPTURE) {
@@ -287,7 +309,7 @@ static int ioctl_queryctl(struct file *file, void *fh, struct v4l2_queryctrl *ct
 }
 
 static int queue_setup(struct vb2_queue *vbq, unsigned int *nbuf,
-               unsigned int *nplanes, unsigned int *sizes, void **alloc_ctx)
+               unsigned int *nplanes, unsigned int *sizes, ALLOC_CTX_TYPE **alloc_ctx)
 {
        if(vbq->num_buffers + *nbuf < 2) {
                *nbuf = 2 - vbq->num_buffers;
@@ -313,6 +335,7 @@ static void buf_queue(struct vb2_buffer *vb)
        struct buffer *buf = (struct buffer*)((char*)vb - offsetof(struct buffer, vb));
 
        spin_lock(&blist_lock);
+       printk(KERN_INFO "vdummy: enqueue buffer\n");
        if(buflist) {
                buflist_tail->next = buf;
                buflist_tail = buf;
@@ -325,14 +348,20 @@ static void buf_queue(struct vb2_buffer *vb)
 
 static void clear_queue(enum vb2_buffer_state st)
 {
+       struct buffer *list;
+
+       printk(KERN_INFO "vdummy: clear_queue\n");
+
        spin_lock(&blist_lock);
-       while(buflist) {
-               struct buffer *buf = buflist;
-               buflist = buflist->next;
-               vb2_buffer_done(&buf->vb.vb2_buf, st);
-       }
+       list = buflist;
        buflist = buflist_tail = 0;
        spin_unlock(&blist_lock);
+
+       while(list) {
+               struct buffer *buf = list;
+               list = list->next;
+               vb2_buffer_done(&buf->vb.vb2_buf, st);
+       }
 }
 
 static int start_streaming(struct vb2_queue *vbq, unsigned int count)
@@ -349,13 +378,16 @@ static int start_streaming(struct vb2_queue *vbq, unsigned int count)
                frame_interval = ktime_set(0, 1000000000ul / (unsigned long)framerate);
        }
 
+       printk(KERN_INFO "vdummy: start streaming (interval: %ld ms)\n",
+                       (long)ktime_to_ms(frame_interval));
        hrtimer_start(&timer, frame_interval, HRTIMER_MODE_REL);
        return 0;
 }
 
 static void stop_streaming(struct vb2_queue *vbq)
 {
-       while(hrtimer_try_to_cancel(&timer) == -1);
+       printk(KERN_INFO "vdummy: stop streaming\n");
+       hrtimer_cancel(&timer);
 
        streaming = 0;
        clear_queue(VB2_BUF_STATE_ERROR);
@@ -400,21 +432,24 @@ static int update_frame(int xsz, int ysz)
 static int gen_frame(void)
 {
        struct buffer *buf = 0;
+       unsigned long flags;
 
-       spin_lock(&blist_lock);
+       printk(KERN_INFO "vdummy: gen_frame\n");
+
+       spin_lock_irqsave(&blist_lock, flags);
        if(buflist) {
                buf = buflist;
                buflist = buflist->next;
                if(!buflist) buflist_tail = 0;
        }
-       spin_unlock(&blist_lock);
+       spin_unlock_irqrestore(&blist_lock, flags);
 
        if(buf) {
-               printk(KERN_INFO "vdummy: copying frame: %d\n", seqno);
-
                memcpy(vb2_plane_vaddr(&buf->vb.vb2_buf, 0), frame, frame_size);
 
                buf->vb.vb2_buf.timestamp = ktime_get_ns();
+               printk(KERN_INFO "vdummy: copying frame: %d ts: %lu\n", seqno,
+                               (unsigned long)buf->vb.vb2_buf.timestamp);
                buf->vb.sequence = seqno++;
                buf->vb.field = V4L2_FIELD_NONE;
                vb2_buffer_done(&buf->vb.vb2_buf, VB2_BUF_STATE_DONE);
@@ -425,6 +460,7 @@ static int gen_frame(void)
 
 enum hrtimer_restart timer_func(struct hrtimer *timer)
 {
-       gen_frame();
+       printk(KERN_INFO "vdummy: timer!\n");
+       //gen_frame();
        return streaming ? HRTIMER_RESTART : HRTIMER_NORESTART;
 }