← Back to branch summary

~mmach/netext73/busybox

« back to all changes in this revision

Viewing changes to libbb/loop.c

  • Committer: mmach
  • Date: 2021-04-14 13:54:24 UTC
  • Revision ID: netbit73@gmail.com-20210414135424-8x3fxf716zs4wflb
1

Show diffs side-by-side

added added

removed removed

Lines of Context:
libbb/loop.c
 
1
/* vi: set sw=4 ts=4: */
 
2
/*
 
3
 * Utility routines.
 
4
 *
 
5
 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
 
6
 * Copyright (C) 2005 by Rob Landley <rob@landley.net>
 
7
 *
 
8
 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
 
9
 */
 
10
#include "libbb.h"
 
11
#include <linux/version.h>
 
12
 
 
13
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,0)
 
14
 
 
15
/* For 2.6, use the cleaned up header to get the 64 bit API. */
 
16
// Commented out per Rob's request
 
17
//# include "fix_u32.h" /* some old toolchains need __u64 for linux/loop.h */
 
18
# include <linux/loop.h>
 
19
typedef struct loop_info64 bb_loop_info;
 
20
# define BB_LOOP_SET_STATUS LOOP_SET_STATUS64
 
21
# define BB_LOOP_GET_STATUS LOOP_GET_STATUS64
 
22
 
 
23
#else
 
24
 
 
25
/* For 2.4 and earlier, use the 32 bit API (and don't trust the headers) */
 
26
/* Stuff stolen from linux/loop.h for 2.4 and earlier kernels */
 
27
# include <linux/posix_types.h>
 
28
# define LO_NAME_SIZE        64
 
29
# define LO_KEY_SIZE         32
 
30
# define LOOP_SET_FD         0x4C00
 
31
# define LOOP_CLR_FD         0x4C01
 
32
# define BB_LOOP_SET_STATUS  0x4C02
 
33
# define BB_LOOP_GET_STATUS  0x4C03
 
34
typedef struct {
 
35
        int                lo_number;
 
36
        __kernel_dev_t     lo_device;
 
37
        unsigned long      lo_inode;
 
38
        __kernel_dev_t     lo_rdevice;
 
39
        int                lo_offset;
 
40
        int                lo_encrypt_type;
 
41
        int                lo_encrypt_key_size;
 
42
        int                lo_flags;
 
43
        char               lo_file_name[LO_NAME_SIZE];
 
44
        unsigned char      lo_encrypt_key[LO_KEY_SIZE];
 
45
        unsigned long      lo_init[2];
 
46
        char               reserved[4];
 
47
} bb_loop_info;
 
48
#endif
 
49
 
 
50
char* FAST_FUNC query_loop(const char *device)
 
51
{
 
52
        int fd;
 
53
        bb_loop_info loopinfo;
 
54
        char *dev = NULL;
 
55
 
 
56
        fd = open(device, O_RDONLY);
 
57
        if (fd >= 0) {
 
58
                if (ioctl(fd, BB_LOOP_GET_STATUS, &loopinfo) == 0) {
 
59
                        dev = xasprintf("%"OFF_FMT"u %s", (off_t) loopinfo.lo_offset,
 
60
                                        (char *)loopinfo.lo_file_name);
 
61
                }
 
62
                close(fd);
 
63
        }
 
64
 
 
65
        return dev;
 
66
}
 
67
 
 
68
int FAST_FUNC del_loop(const char *device)
 
69
{
 
70
        int fd, rc;
 
71
 
 
72
        fd = open(device, O_RDONLY);
 
73
        if (fd < 0)
 
74
                return 1;
 
75
        rc = ioctl(fd, LOOP_CLR_FD, 0);
 
76
        close(fd);
 
77
 
 
78
        return rc;
 
79
}
 
80
 
 
81
/* Returns opened fd to the loop device, <0 on error.
 
82
 * *device is loop device to use, or if *device==NULL finds a loop device to
 
83
 * mount it on and sets *device to a strdup of that loop device name.  This
 
84
 * search will re-use an existing loop device already bound to that
 
85
 * file/offset if it finds one.
 
86
 */
 
87
int FAST_FUNC set_loop(char **device, const char *file, unsigned long long offset, unsigned flags)
 
88
{
 
89
        char dev[LOOP_NAMESIZE];
 
90
        char *try;
 
91
        bb_loop_info loopinfo;
 
92
        struct stat statbuf;
 
93
        int i, dfd, ffd, mode, rc;
 
94
 
 
95
        rc = dfd = -1;
 
96
 
 
97
        /* Open the file.  Barf if this doesn't work.  */
 
98
        mode = (flags & BB_LO_FLAGS_READ_ONLY) ? O_RDONLY : O_RDWR;
 
99
 open_ffd:
 
100
        ffd = open(file, mode);
 
101
        if (ffd < 0) {
 
102
                if (mode != O_RDONLY) {
 
103
                        mode = O_RDONLY;
 
104
                        goto open_ffd;
 
105
                }
 
106
                return -errno;
 
107
        }
 
108
 
 
109
//TODO: use LOOP_CTL_GET_FREE instead of trying every loopN in sequence? a-la:
 
110
// fd = open("/dev/loop-control", O_RDWR);
 
111
// loopN = ioctl(fd, LOOP_CTL_GET_FREE);
 
112
//
 
113
        /* Find a loop device.  */
 
114
        try = *device ? *device : dev;
 
115
        /* 1048575 (0xfffff) is a max possible minor number in Linux circa 2010 */
 
116
        for (i = 0; rc && i < 1048576; i++) {
 
117
                sprintf(dev, LOOP_FORMAT, i);
 
118
 
 
119
                IF_FEATURE_MOUNT_LOOP_CREATE(errno = 0;)
 
120
                if (stat(try, &statbuf) != 0 || !S_ISBLK(statbuf.st_mode)) {
 
121
                        if (ENABLE_FEATURE_MOUNT_LOOP_CREATE
 
122
                         && errno == ENOENT
 
123
                         && try == dev
 
124
                        ) {
 
125
                                /* Node doesn't exist, try to create it.  */
 
126
                                if (mknod(dev, S_IFBLK|0644, makedev(7, i)) == 0)
 
127
                                        goto try_to_open;
 
128
                        }
 
129
                        /* Ran out of block devices, return failure.  */
 
130
                        rc = -1;
 
131
                        break;
 
132
                }
 
133
 try_to_open:
 
134
                /* Open the sucker and check its loopiness.  */
 
135
                dfd = open(try, mode);
 
136
                if (dfd < 0 && errno == EROFS) {
 
137
                        mode = O_RDONLY;
 
138
                        dfd = open(try, mode);
 
139
                }
 
140
                if (dfd < 0) {
 
141
                        if (errno == ENXIO) {
 
142
                                /* Happens if loop module is not loaded */
 
143
                                rc = -1;
 
144
                                break;
 
145
                        }
 
146
                        goto try_again;
 
147
                }
 
148
 
 
149
                rc = ioctl(dfd, BB_LOOP_GET_STATUS, &loopinfo);
 
150
 
 
151
                /* If device is free, claim it.  */
 
152
                if (rc && errno == ENXIO) {
 
153
                        /* Associate free loop device with file.  */
 
154
                        if (ioctl(dfd, LOOP_SET_FD, ffd) == 0) {
 
155
                                memset(&loopinfo, 0, sizeof(loopinfo));
 
156
                                safe_strncpy((char *)loopinfo.lo_file_name, file, LO_NAME_SIZE);
 
157
                                loopinfo.lo_offset = offset;
 
158
                                /*
 
159
                                 * Used by mount to set LO_FLAGS_AUTOCLEAR.
 
160
                                 * LO_FLAGS_READ_ONLY is not set because RO is controlled by open type of the file.
 
161
                                 * Note that closing LO_FLAGS_AUTOCLEARed dfd before mount
 
162
                                 * is wrong (would free the loop device!)
 
163
                                 */
 
164
                                loopinfo.lo_flags = (flags & ~BB_LO_FLAGS_READ_ONLY);
 
165
                                rc = ioctl(dfd, BB_LOOP_SET_STATUS, &loopinfo);
 
166
                                if (rc != 0 && (loopinfo.lo_flags & BB_LO_FLAGS_AUTOCLEAR)) {
 
167
                                        /* Old kernel, does not support LO_FLAGS_AUTOCLEAR? */
 
168
                                        /* (this code path is not tested) */
 
169
                                        loopinfo.lo_flags -= BB_LO_FLAGS_AUTOCLEAR;
 
170
                                        rc = ioctl(dfd, BB_LOOP_SET_STATUS, &loopinfo);
 
171
                                }
 
172
                                if (rc != 0) {
 
173
                                        ioctl(dfd, LOOP_CLR_FD, 0);
 
174
                                }
 
175
                        }
 
176
                } else {
 
177
                        rc = -1;
 
178
                }
 
179
                if (rc != 0) {
 
180
                        close(dfd);
 
181
                }
 
182
 try_again:
 
183
                if (*device) break;
 
184
        }
 
185
        close(ffd);
 
186
        if (rc == 0) {
 
187
                if (!*device)
 
188
                        *device = xstrdup(dev);
 
189
                return dfd;
 
190
        }
 
191
        return rc;
 
192
}