#include <linux/wait.h>
 
 #include <media/v4l2-common.h>
+#include <media/videobuf2-dma-sg.h>
 #include <media/videobuf2-vmalloc.h>
 
 #include "uvc.h"
                return -ENODEV;
 
        buf->state = UVC_BUF_STATE_QUEUED;
-       buf->mem = vb2_plane_vaddr(vb, 0);
+       if (queue->use_sg) {
+               buf->sgt = vb2_dma_sg_plane_desc(vb, 0);
+               buf->sg = buf->sgt->sgl;
+       } else {
+               buf->mem = vb2_plane_vaddr(vb, 0);
+       }
        buf->length = vb2_plane_size(vb, 0);
        if (vb->type == V4L2_BUF_TYPE_VIDEO_CAPTURE)
                buf->bytesused = 0;
        .wait_finish = vb2_ops_wait_finish,
 };
 
-int uvcg_queue_init(struct uvc_video_queue *queue, enum v4l2_buf_type type,
+int uvcg_queue_init(struct uvc_video_queue *queue, struct device *dev, enum v4l2_buf_type type,
                    struct mutex *lock)
 {
+       struct uvc_video *video = container_of(queue, struct uvc_video, queue);
+       struct usb_composite_dev *cdev = video->uvc->func.config->cdev;
        int ret;
 
        queue->queue.type = type;
        queue->queue.buf_struct_size = sizeof(struct uvc_buffer);
        queue->queue.ops = &uvc_queue_qops;
        queue->queue.lock = lock;
-       queue->queue.mem_ops = &vb2_vmalloc_memops;
+       if (cdev->gadget->sg_supported) {
+               queue->queue.mem_ops = &vb2_dma_sg_memops;
+               queue->use_sg = 1;
+       } else {
+               queue->queue.mem_ops = &vb2_vmalloc_memops;
+       }
+
        queue->queue.timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC
                                     | V4L2_BUF_FLAG_TSTAMP_SRC_EOF;
+       queue->queue.dev = dev;
+
        ret = vb2_queue_init(&queue->queue);
        if (ret)
                return ret;
 
 uvc_video_encode_header(struct uvc_video *video, struct uvc_buffer *buf,
                u8 *data, int len)
 {
-       data[0] = 2;
+       data[0] = UVCG_REQUEST_HEADER_LEN;
        data[1] = UVC_STREAM_EOH | video->fid;
 
-       if (buf->bytesused - video->queue.buf_used <= len - 2)
+       if (buf->bytesused - video->queue.buf_used <= len - UVCG_REQUEST_HEADER_LEN)
                data[1] |= UVC_STREAM_EOF;
 
        return 2;
                video->payload_size = 0;
 }
 
+static void
+uvc_video_encode_isoc_sg(struct usb_request *req, struct uvc_video *video,
+               struct uvc_buffer *buf)
+{
+       unsigned int pending = buf->bytesused - video->queue.buf_used;
+       struct uvc_request *ureq = req->context;
+       struct scatterlist *sg, *iter;
+       unsigned int len = video->req_size;
+       unsigned int sg_left, part = 0;
+       unsigned int i;
+       int ret;
+
+       sg = ureq->sgt.sgl;
+       sg_init_table(sg, ureq->sgt.nents);
+
+       /* Init the header. */
+       ret = uvc_video_encode_header(video, buf, ureq->header,
+                                     video->req_size);
+       sg_set_buf(sg, ureq->header, UVCG_REQUEST_HEADER_LEN);
+       len -= ret;
+
+       if (pending <= len)
+               len = pending;
+
+       req->length = (len == pending) ?
+               len + UVCG_REQUEST_HEADER_LEN : video->req_size;
+
+       /* Init the pending sgs with payload */
+       sg = sg_next(sg);
+
+       for_each_sg(sg, iter, ureq->sgt.nents - 1, i) {
+               if (!len || !buf->sg)
+                       break;
+
+               sg_left = sg_dma_len(buf->sg) - buf->offset;
+               part = min_t(unsigned int, len, sg_left);
+
+               sg_set_page(iter, sg_page(buf->sg), part, buf->offset);
+
+               if (part == sg_left) {
+                       buf->offset = 0;
+                       buf->sg = sg_next(buf->sg);
+               } else {
+                       buf->offset += part;
+               }
+               len -= part;
+       }
+
+       /* Assign the video data with header. */
+       req->buf = NULL;
+       req->sg = ureq->sgt.sgl;
+       req->num_sgs = i + 1;
+
+       req->length -= len;
+       video->queue.buf_used += req->length - UVCG_REQUEST_HEADER_LEN;
+
+       if (buf->bytesused == video->queue.buf_used || !buf->sg) {
+               video->queue.buf_used = 0;
+               buf->state = UVC_BUF_STATE_DONE;
+               buf->offset = 0;
+               uvcg_queue_next_buffer(&video->queue, buf);
+               video->fid ^= UVC_STREAM_FID;
+       }
+}
+
 static void
 uvc_video_encode_isoc(struct usb_request *req, struct uvc_video *video,
                struct uvc_buffer *buf)
 
        if (video->ureq) {
                for (i = 0; i < video->uvc_num_requests; ++i) {
+                       sg_free_table(&video->ureq[i].sgt);
+
                        if (video->ureq[i].req) {
                                usb_ep_free_request(video->ep, video->ureq[i].req);
                                video->ureq[i].req = NULL;
                video->ureq[i].video = video;
 
                list_add_tail(&video->ureq[i].req->list, &video->req_free);
+               /* req_size/PAGE_SIZE + 1 for overruns and + 1 for header */
+               sg_alloc_table(&video->ureq[i].sgt,
+                              DIV_ROUND_UP(req_size - 2, PAGE_SIZE) + 2,
+                              GFP_KERNEL);
        }
 
        video->req_size = req_size;
                video->encode = uvc_video_encode_bulk;
                video->payload_size = 0;
        } else
-               video->encode = uvc_video_encode_isoc;
+               video->encode = video->queue.use_sg ?
+                       uvc_video_encode_isoc_sg : uvc_video_encode_isoc;
 
        schedule_work(&video->pump);
 
        video->imagesize = 320 * 240 * 2;
 
        /* Initialize the video buffers queue. */
-       uvcg_queue_init(&video->queue, V4L2_BUF_TYPE_VIDEO_OUTPUT,
-                       &video->mutex);
+       uvcg_queue_init(&video->queue, uvc->v4l2_dev.dev->parent,
+                       V4L2_BUF_TYPE_VIDEO_OUTPUT, &video->mutex);
        return 0;
 }