*dma_handle = 0;
 }
 
+static void unmap_urb_for_dma(struct usb_hcd *hcd, struct urb *urb)
+{
+       enum dma_data_direction dir;
+
+       if (urb->transfer_flags & URB_SETUP_MAP_SINGLE)
+               dma_unmap_single(hcd->self.controller,
+                               urb->setup_dma,
+                               sizeof(struct usb_ctrlrequest),
+                               DMA_TO_DEVICE);
+       else if (urb->transfer_flags & URB_SETUP_MAP_LOCAL)
+               hcd_free_coherent(urb->dev->bus,
+                               &urb->setup_dma,
+                               (void **) &urb->setup_packet,
+                               sizeof(struct usb_ctrlrequest),
+                               DMA_TO_DEVICE);
+
+       dir = usb_urb_dir_in(urb) ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
+       if (urb->transfer_flags & URB_DMA_MAP_SG)
+               dma_unmap_sg(hcd->self.controller,
+                               urb->sg->sg,
+                               urb->num_sgs,
+                               dir);
+       else if (urb->transfer_flags & URB_DMA_MAP_PAGE)
+               dma_unmap_page(hcd->self.controller,
+                               urb->transfer_dma,
+                               urb->transfer_buffer_length,
+                               dir);
+       else if (urb->transfer_flags & URB_DMA_MAP_SINGLE)
+               dma_unmap_single(hcd->self.controller,
+                               urb->transfer_dma,
+                               urb->transfer_buffer_length,
+                               dir);
+       else if (urb->transfer_flags & URB_MAP_LOCAL)
+               hcd_free_coherent(urb->dev->bus,
+                               &urb->transfer_dma,
+                               &urb->transfer_buffer,
+                               urb->transfer_buffer_length,
+                               dir);
+
+       /* Make it safe to call this routine more than once */
+       urb->transfer_flags &= ~(URB_SETUP_MAP_SINGLE | URB_SETUP_MAP_LOCAL |
+                       URB_DMA_MAP_SG | URB_DMA_MAP_PAGE |
+                       URB_DMA_MAP_SINGLE | URB_MAP_LOCAL);
+}
+
 static int map_urb_for_dma(struct usb_hcd *hcd, struct urb *urb,
                           gfp_t mem_flags)
 {
         * unless it uses pio or talks to another transport,
         * or uses the provided scatter gather list for bulk.
         */
-       if (is_root_hub(urb->dev))
-               return 0;
 
        if (usb_endpoint_xfer_control(&urb->ep->desc)
            && !(urb->transfer_flags & URB_NO_SETUP_DMA_MAP)) {
                        if (dma_mapping_error(hcd->self.controller,
                                                urb->setup_dma))
                                return -EAGAIN;
+                       urb->transfer_flags |= URB_SETUP_MAP_SINGLE;
                } else if (hcd->driver->flags & HCD_LOCAL_MEM)
                        ret = hcd_alloc_coherent(
                                        urb->dev->bus, mem_flags,
                                        (void **)&urb->setup_packet,
                                        sizeof(struct usb_ctrlrequest),
                                        DMA_TO_DEVICE);
+                       if (ret)
+                               return ret;
+                       urb->transfer_flags |= URB_SETUP_MAP_LOCAL;
        }
 
        dir = usb_urb_dir_in(urb) ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
-       if (ret == 0 && urb->transfer_buffer_length != 0
+       if (urb->transfer_buffer_length != 0
            && !(urb->transfer_flags & URB_NO_TRANSFER_DMA_MAP)) {
                if (hcd->self.uses_dma) {
-                       urb->transfer_dma = dma_map_single (
-                                       hcd->self.controller,
-                                       urb->transfer_buffer,
-                                       urb->transfer_buffer_length,
-                                       dir);
-                       if (dma_mapping_error(hcd->self.controller,
+                       if (urb->num_sgs) {
+                               int n = dma_map_sg(
+                                               hcd->self.controller,
+                                               urb->sg->sg,
+                                               urb->num_sgs,
+                                               dir);
+                               if (n <= 0)
+                                       ret = -EAGAIN;
+                               else
+                                       urb->transfer_flags |= URB_DMA_MAP_SG;
+                               if (n != urb->num_sgs) {
+                                       urb->num_sgs = n;
+                                       urb->transfer_flags |=
+                                                       URB_DMA_SG_COMBINED;
+                               }
+                       } else if (urb->sg) {
+                               struct scatterlist *sg;
+
+                               sg = (struct scatterlist *) urb->sg;
+                               urb->transfer_dma = dma_map_page(
+                                               hcd->self.controller,
+                                               sg_page(sg),
+                                               sg->offset,
+                                               urb->transfer_buffer_length,
+                                               dir);
+                               if (dma_mapping_error(hcd->self.controller,
                                                urb->transfer_dma))
-                               return -EAGAIN;
+                                       ret = -EAGAIN;
+                               else
+                                       urb->transfer_flags |= URB_DMA_MAP_PAGE;
+                       } else {
+                               urb->transfer_dma = dma_map_single(
+                                               hcd->self.controller,
+                                               urb->transfer_buffer,
+                                               urb->transfer_buffer_length,
+                                               dir);
+                               if (dma_mapping_error(hcd->self.controller,
+                                               urb->transfer_dma))
+                                       ret = -EAGAIN;
+                               else
+                                       urb->transfer_flags |= URB_DMA_MAP_SINGLE;
+                       }
                } else if (hcd->driver->flags & HCD_LOCAL_MEM) {
                        ret = hcd_alloc_coherent(
                                        urb->dev->bus, mem_flags,
                                        &urb->transfer_buffer,
                                        urb->transfer_buffer_length,
                                        dir);
-
-                       if (ret && usb_endpoint_xfer_control(&urb->ep->desc)
-                           && !(urb->transfer_flags & URB_NO_SETUP_DMA_MAP))
-                               hcd_free_coherent(urb->dev->bus,
-                                       &urb->setup_dma,
-                                       (void **)&urb->setup_packet,
-                                       sizeof(struct usb_ctrlrequest),
-                                       DMA_TO_DEVICE);
+                       if (ret == 0)
+                               urb->transfer_flags |= URB_MAP_LOCAL;
                }
+               if (ret && (urb->transfer_flags & (URB_SETUP_MAP_SINGLE |
+                               URB_SETUP_MAP_LOCAL)))
+                       unmap_urb_for_dma(hcd, urb);
        }
        return ret;
 }
 
-static void unmap_urb_for_dma(struct usb_hcd *hcd, struct urb *urb)
-{
-       enum dma_data_direction dir;
-
-       if (is_root_hub(urb->dev))
-               return;
-
-       if (usb_endpoint_xfer_control(&urb->ep->desc)
-           && !(urb->transfer_flags & URB_NO_SETUP_DMA_MAP)) {
-               if (hcd->self.uses_dma)
-                       dma_unmap_single(hcd->self.controller, urb->setup_dma,
-                                       sizeof(struct usb_ctrlrequest),
-                                       DMA_TO_DEVICE);
-               else if (hcd->driver->flags & HCD_LOCAL_MEM)
-                       hcd_free_coherent(urb->dev->bus, &urb->setup_dma,
-                                       (void **)&urb->setup_packet,
-                                       sizeof(struct usb_ctrlrequest),
-                                       DMA_TO_DEVICE);
-       }
-
-       dir = usb_urb_dir_in(urb) ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
-       if (urb->transfer_buffer_length != 0
-           && !(urb->transfer_flags & URB_NO_TRANSFER_DMA_MAP)) {
-               if (hcd->self.uses_dma)
-                       dma_unmap_single(hcd->self.controller,
-                                       urb->transfer_dma,
-                                       urb->transfer_buffer_length,
-                                       dir);
-               else if (hcd->driver->flags & HCD_LOCAL_MEM)
-                       hcd_free_coherent(urb->dev->bus, &urb->transfer_dma,
-                                       &urb->transfer_buffer,
-                                       urb->transfer_buffer_length,
-                                       dir);
-       }
-}
-
 /*-------------------------------------------------------------------------*/
 
 /* may be called in any context with a valid urb->dev usecount
         * URBs must be submitted in process context with interrupts
         * enabled.
         */
-       status = map_urb_for_dma(hcd, urb, mem_flags);
-       if (unlikely(status)) {
-               usbmon_urb_submit_error(&hcd->self, urb, status);
-               goto error;
-       }
 
-       if (is_root_hub(urb->dev))
+       if (is_root_hub(urb->dev)) {
                status = rh_urb_enqueue(hcd, urb);
-       else
-               status = hcd->driver->urb_enqueue(hcd, urb, mem_flags);
+       } else {
+               status = map_urb_for_dma(hcd, urb, mem_flags);
+               if (likely(status == 0)) {
+                       status = hcd->driver->urb_enqueue(hcd, urb, mem_flags);
+                       if (unlikely(status))
+                               unmap_urb_for_dma(hcd, urb);
+               }
+       }
 
        if (unlikely(status)) {
                usbmon_urb_submit_error(&hcd->self, urb, status);
-               unmap_urb_for_dma(hcd, urb);
- error:
                urb->hcpriv = NULL;
                INIT_LIST_HEAD(&urb->urb_list);
                atomic_dec(&urb->use_count);
 
                kfree(io->urbs);
                io->urbs = NULL;
        }
-       if (io->dev->dev.dma_mask != NULL)
-               usb_buffer_unmap_sg(io->dev, usb_pipein(io->pipe),
-                                   io->sg, io->nents);
        io->dev = NULL;
 }
 
 {
        int i;
        int urb_flags;
-       int dma;
        int use_sg;
 
        if (!io || !dev || !sg
        io->pipe = pipe;
        io->sg = sg;
        io->nents = nents;
-
-       /* not all host controllers use DMA (like the mainstream pci ones);
-        * they can use PIO (sl811) or be software over another transport.
-        */
-       dma = (dev->dev.dma_mask != NULL);
-       if (dma)
-               io->entries = usb_buffer_map_sg(dev, usb_pipein(pipe),
-                                               sg, nents);
-       else
-               io->entries = nents;
+       io->entries = nents;
 
        /* initialize all the urbs we'll use */
-       if (io->entries <= 0)
-               return io->entries;
-
        if (dev->bus->sg_tablesize > 0) {
                io->urbs = kmalloc(sizeof *io->urbs, mem_flags);
                use_sg = true;
                goto nomem;
 
        urb_flags = 0;
-       if (dma)
-               urb_flags |= URB_NO_TRANSFER_DMA_MAP;
        if (usb_pipein(pipe))
                urb_flags |= URB_SHORT_NOT_OK;
 
 
                io->urbs[0]->complete = sg_complete;
                io->urbs[0]->context = io;
+
                /* A length of zero means transfer the whole sg list */
                io->urbs[0]->transfer_buffer_length = length;
                if (length == 0) {
                        for_each_sg(sg, sg, io->entries, i) {
                                io->urbs[0]->transfer_buffer_length +=
-                                       sg_dma_len(sg);
+                                       sg->length;
                        }
                }
                io->urbs[0]->sg = io;
                        io->urbs[i]->context = io;
 
                        /*
-                        * Some systems need to revert to PIO when DMA is temporarily
-                        * unavailable.  For their sakes, both transfer_buffer and
-                        * transfer_dma are set when possible.
-                        *
-                        * Note that if IOMMU coalescing occurred, we cannot
-                        * trust sg_page anymore, so check if S/G list shrunk.
+                        * Some systems can't use DMA; they use PIO instead.
+                        * For their sakes, transfer_buffer is set whenever
+                        * possible.
                         */
-                       if (io->nents == io->entries && !PageHighMem(sg_page(sg)))
+                       if (!PageHighMem(sg_page(sg)))
                                io->urbs[i]->transfer_buffer = sg_virt(sg);
                        else
                                io->urbs[i]->transfer_buffer = NULL;
 
-                       if (dma) {
-                               io->urbs[i]->transfer_dma = sg_dma_address(sg);
-                               len = sg_dma_len(sg);
-                       } else {
-                               /* hc may use _only_ transfer_buffer */
-                               len = sg->length;
-                       }
-
+                       len = sg->length;
                        if (length) {
                                len = min_t(unsigned, len, length);
                                length -= len;
                                        io->entries = i + 1;
                        }
                        io->urbs[i]->transfer_buffer_length = len;
+
+                       io->urbs[i]->sg = (struct usb_sg_request *) sg;
                }
                io->urbs[--i]->transfer_flags &= ~URB_NO_INTERRUPT;
        }