iovec_import() has a safer calling convention than import_iovec().

Signed-off-by: David Laight <david.lai...@aculab.com>
---
 block/scsi_ioctl.c | 14 ++++++++------
 drivers/scsi/sg.c  | 14 +++++++-------
 2 files changed, 15 insertions(+), 13 deletions(-)

diff --git a/block/scsi_ioctl.c b/block/scsi_ioctl.c
index ef722f04f88a..0343918a84d3 100644
--- a/block/scsi_ioctl.c
+++ b/block/scsi_ioctl.c
@@ -331,20 +331,22 @@ static int sg_io(struct request_queue *q, struct gendisk 
*bd_disk,
        ret = 0;
        if (hdr->iovec_count) {
                struct iov_iter i;
-               struct iovec *iov = NULL;
+               struct iovec *iov;
 
 #ifdef CONFIG_COMPAT
                if (in_compat_syscall())
-                       ret = compat_import_iovec(rq_data_dir(rq),
+                       iov = compat_iovec_import(rq_data_dir(rq),
                                   hdr->dxferp, hdr->iovec_count,
-                                  0, &iov, &i);
+                                  NULL, &i);
                else
 #endif
-                       ret = import_iovec(rq_data_dir(rq),
+                       iov = iovec_import(rq_data_dir(rq),
                                   hdr->dxferp, hdr->iovec_count,
-                                  0, &iov, &i);
-               if (ret < 0)
+                                  NULL, &i);
+               if (IS_ERR(iov)) {
+                       ret = PTR_ERR(iov);
                        goto out_free_cdb;
+               }
 
                /* SG_IO howto says that the shorter of the two wins */
                iov_iter_truncate(&i, hdr->dxfer_len);
diff --git a/drivers/scsi/sg.c b/drivers/scsi/sg.c
index 20472aaaf630..1dbc0a74add5 100644
--- a/drivers/scsi/sg.c
+++ b/drivers/scsi/sg.c
@@ -1817,19 +1817,19 @@ sg_start_req(Sg_request *srp, unsigned char *cmd)
        }
 
        if (iov_count) {
-               struct iovec *iov = NULL;
+               struct iovec *iov;
                struct iov_iter i;
 
 #ifdef CONFIG_COMPAT
                if (in_compat_syscall())
-                       res = compat_import_iovec(rw, hp->dxferp, iov_count,
-                                                 0, &iov, &i);
+                       iov = compat_iovec_import(rw, hp->dxferp, iov_count,
+                                                 NULL, &i);
                else
 #endif
-                       res = import_iovec(rw, hp->dxferp, iov_count,
-                                          0, &iov, &i);
-               if (res < 0)
-                       return res;
+                       iov = iovec_import(rw, hp->dxferp, iov_count,
+                                          NULL, &i);
+               if (IS_ERR(iov))
+                       return PTR_ERR(iov);
 
                iov_iter_truncate(&i, hp->dxfer_len);
                if (!iov_iter_count(&i)) {
-- 
2.25.1

-
Registered Address Lakeside, Bramley Road, Mount Farm, Milton Keynes, MK1 1PT, 
UK
Registration No: 1397386 (Wales)

Reply via email to