1 files changed, 7 insertions, 4 deletions

diff --git a/drivers/scsi/lpfc/lpfc_scsi.c b/drivers/scsi/lpfc/lpfc_scsi.c
index 3d35eae01475..8f45bbc42126 100644
--- a/drivers/scsi/lpfc/lpfc_scsi.c
+++ b/drivers/scsi/lpfc/lpfc_scsi.c
@@ -332,8 +332,7 @@ lpfc_scsi_prep_dma_buf(struct lpfc_hba *phba, struct lpfc_scsi_buf *lpfc_cmd)
          * data bde entry.
          */
         bpl += 2;
-        nseg = scsi_dma_map(scsi_cmnd);
-        if (nseg > 0) {
+        if (scsi_sg_count(scsi_cmnd)) {
                 /*
                  * The driver stores the segment count returned from pci_map_sg
                  * because this a count of dma-mappings used to map the use_sg
@@ -341,6 +340,11 @@ lpfc_scsi_prep_dma_buf(struct lpfc_hba *phba, struct lpfc_scsi_buf *lpfc_cmd)
                  * architectures that implement an IOMMU.
                  */
 
+                nseg = dma_map_sg(&phba->pcidev->dev, scsi_sglist(scsi_cmnd),
+                                  scsi_sg_count(scsi_cmnd), datadir);
+                if (unlikely(!nseg))
+                        return 1;
+
                 lpfc_cmd->seg_cnt = nseg;
                 if (lpfc_cmd->seg_cnt > phba->cfg_sg_seg_cnt) {
                         printk(KERN_ERR "%s: Too many sg segments from "
@@ -370,8 +374,7 @@ lpfc_scsi_prep_dma_buf(struct lpfc_hba *phba, struct lpfc_scsi_buf *lpfc_cmd)
                         bpl++;
                         num_bde++;
                 }
-        } else if (nseg < 0)
-                return 1;
+        }
 
         /*
          * Finish initializing those IOCB fields that are dependent on the