Merge branch 'upstream' into tizen
[platform/upstream/cryptsetup.git] / lib / utils_loop.c
index 123154b..6e67a6a 100644 (file)
@@ -1,11 +1,13 @@
 /*
  * loopback block device utilities
  *
- * Copyright (C) 2011, Red Hat, Inc. All rights reserved.
+ * Copyright (C) 2009-2023 Red Hat, Inc. All rights reserved.
+ * Copyright (C) 2009-2023 Milan Broz
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License
- * version 2 as published by the Free Software Foundation.
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
+#include <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 #include <unistd.h>
 #include <limits.h>
 #include <sys/ioctl.h>
 #include <sys/stat.h>
+#ifdef HAVE_SYS_SYSMACROS_H
+# include <sys/sysmacros.h>     /* for major, minor */
+#endif
+#include <linux/types.h>
 #include <linux/loop.h>
+#ifdef HAVE_SYS_SYSMACROS_H
+#include <sys/sysmacros.h>     /* for major, minor */
+#endif
 
 #include "utils_loop.h"
+#include "libcryptsetup_macros.h"
 
 #define LOOP_DEV_MAJOR 7
 
 #define LOOP_CTL_GET_FREE 0x4C82
 #endif
 
+#ifndef LOOP_SET_CAPACITY
+#define LOOP_SET_CAPACITY 0x4C07
+#endif
+
+#ifndef LOOP_SET_BLOCK_SIZE
+#define LOOP_SET_BLOCK_SIZE 0x4C09
+#endif
+
+#ifndef LOOP_CONFIGURE
+#define LOOP_CONFIGURE 0x4C0A
+struct loop_config {
+  __u32 fd;
+  __u32 block_size;
+  struct loop_info64 info;
+  __u64 __reserved[8];
+};
+#endif
+
 static char *crypt_loop_get_device_old(void)
 {
-       char dev[20];
+       char dev[64];
        int i, loop_fd;
-       struct stat st;
        struct loop_info64 lo64 = {0};
 
        for (i = 0; i < 256; i++) {
                sprintf(dev, "/dev/loop%d", i);
-               if (stat(dev, &st) || !S_ISBLK(st.st_mode))
-                       return NULL;
 
                loop_fd = open(dev, O_RDONLY);
                if (loop_fd < 0)
@@ -66,7 +92,7 @@ static char *crypt_loop_get_device_old(void)
        return NULL;
 }
 
-char *crypt_loop_get_device(void)
+static char *crypt_loop_get_device(void)
 {
        char dev[64];
        int i, loop_fd;
@@ -92,42 +118,99 @@ char *crypt_loop_get_device(void)
        return strdup(dev);
 }
 
-int crypt_loop_attach(const char *loop, const char *file, int offset,
-                     int autoclear, int *readonly)
+int crypt_loop_attach(char **loop, const char *file, int offset,
+                     int autoclear, int *readonly, size_t blocksize)
 {
-       struct loop_info64 lo64 = {0};
+       struct loop_config config = {0};
+       char *lo_file_name;
        int loop_fd = -1, file_fd = -1, r = 1;
+       int fallback = 0;
+
+       *loop = NULL;
 
        file_fd = open(file, (*readonly ? O_RDONLY : O_RDWR) | O_EXCL);
-       if (file_fd < 0 && errno == EROFS && !*readonly) {
+       if (file_fd < 0 && (errno == EROFS || errno == EACCES) && !*readonly) {
                *readonly = 1;
                file_fd = open(file, O_RDONLY | O_EXCL);
        }
        if (file_fd < 0)
                goto out;
 
-       loop_fd = open(loop, *readonly ? O_RDONLY : O_RDWR);
-       if (loop_fd < 0)
-               goto out;
+       config.fd = file_fd;
 
-       strncpy((char*)lo64.lo_file_name, file, LO_NAME_SIZE);
-       lo64.lo_offset = offset;
+       lo_file_name = (char*)config.info.lo_file_name;
+       lo_file_name[LO_NAME_SIZE-1] = '\0';
+       strncpy(lo_file_name, file, LO_NAME_SIZE-1);
+       config.info.lo_offset = offset;
        if (autoclear)
-               lo64.lo_flags |= LO_FLAGS_AUTOCLEAR;
+               config.info.lo_flags |= LO_FLAGS_AUTOCLEAR;
+       if (blocksize > SECTOR_SIZE)
+               config.block_size = blocksize;
 
-       if (ioctl(loop_fd, LOOP_SET_FD, file_fd) < 0)
-               goto out;
+       while (loop_fd < 0) {
+               *loop = crypt_loop_get_device();
+               if (!*loop)
+                       goto out;
 
-       if (ioctl(loop_fd, LOOP_SET_STATUS64, &lo64) < 0) {
-               (void)ioctl(loop_fd, LOOP_CLR_FD, 0);
-               goto out;
+               loop_fd = open(*loop, *readonly ? O_RDONLY : O_RDWR);
+               if (loop_fd < 0)
+                       goto out;
+               if (ioctl(loop_fd, LOOP_CONFIGURE, &config) < 0) {
+                       if (errno == EINVAL || errno == ENOTTY) {
+                               free(*loop);
+                               *loop = NULL;
+
+                               close(loop_fd);
+                               loop_fd = -1;
+
+                               /* kernel doesn't support LOOP_CONFIGURE */
+                               fallback = 1;
+                               break;
+                       }
+                       if (errno != EBUSY)
+                               goto out;
+                       free(*loop);
+                       *loop = NULL;
+
+                       close(loop_fd);
+                       loop_fd = -1;
+               }
+       }
+
+       if (fallback) {
+               while (loop_fd < 0) {
+                       *loop = crypt_loop_get_device();
+                       if (!*loop)
+                               goto out;
+
+                       loop_fd = open(*loop, *readonly ? O_RDONLY : O_RDWR);
+                       if (loop_fd < 0)
+                               goto out;
+                       if (ioctl(loop_fd, LOOP_SET_FD, file_fd) < 0) {
+                               if (errno != EBUSY)
+                                       goto out;
+                               free(*loop);
+                               *loop = NULL;
+
+                               close(loop_fd);
+                               loop_fd = -1;
+                       }
+               }
+
+               if (blocksize > SECTOR_SIZE)
+                       (void)ioctl(loop_fd, LOOP_SET_BLOCK_SIZE, (unsigned long)blocksize);
+
+               if (ioctl(loop_fd, LOOP_SET_STATUS64, &config.info) < 0) {
+                       (void)ioctl(loop_fd, LOOP_CLR_FD, 0);
+                       goto out;
+               }
        }
 
        /* Verify that autoclear is really set */
        if (autoclear) {
-               memset(&lo64, 0, sizeof(lo64));
-               if (ioctl(loop_fd, LOOP_GET_STATUS64, &lo64) < 0 ||
-                  !(lo64.lo_flags & LO_FLAGS_AUTOCLEAR)) {
+               memset(&config.info, 0, sizeof(config.info));
+               if (ioctl(loop_fd, LOOP_GET_STATUS64, &config.info) < 0 ||
+                  !(config.info.lo_flags & LO_FLAGS_AUTOCLEAR)) {
                (void)ioctl(loop_fd, LOOP_CLR_FD, 0);
                        goto out;
                }
@@ -139,6 +222,10 @@ out:
                close(loop_fd);
        if (file_fd >= 0)
                close(file_fd);
+       if (r && *loop) {
+               free(*loop);
+               *loop = NULL;
+       }
        return r ? -1 : loop_fd;
 }
 
@@ -157,6 +244,21 @@ int crypt_loop_detach(const char *loop)
        return r;
 }
 
+int crypt_loop_resize(const char *loop)
+{
+       int loop_fd = -1, r = 1;
+
+       loop_fd = open(loop, O_RDONLY);
+       if (loop_fd < 0)
+                return 1;
+
+       if (!ioctl(loop_fd, LOOP_SET_CAPACITY, 0))
+               r = 0;
+
+       close(loop_fd);
+       return r;
+}
+
 static char *_ioctl_backing_file(const char *loop)
 {
        struct loop_info64 lo64 = {0};
@@ -189,8 +291,9 @@ static char *_sysfs_backing_file(const char *loop)
        if (stat(loop, &st) || !S_ISBLK(st.st_mode))
                return NULL;
 
-       snprintf(buf, sizeof(buf), "/sys/dev/block/%d:%d/loop/backing_file",
-                major(st.st_rdev), minor(st.st_rdev));
+       if (snprintf(buf, sizeof(buf), "/sys/dev/block/%d:%d/loop/backing_file",
+                    major(st.st_rdev), minor(st.st_rdev)) < 0)
+               return NULL;
 
        fd = open(buf, O_RDONLY);
        if (fd < 0)
@@ -207,7 +310,12 @@ static char *_sysfs_backing_file(const char *loop)
 
 char *crypt_loop_backing_file(const char *loop)
 {
-       char *bf = _sysfs_backing_file(loop);
+       char *bf;
+
+       if (!crypt_loop_device(loop))
+               return NULL;
+
+       bf = _sysfs_backing_file(loop);
        return bf ?: _ioctl_backing_file(loop);
 }