blob: 3fec7ad6d876f641d691a4a44ee57cffe33e72ad [file] [log] [blame]
/* vi: set sw=4 ts=4: */
/*
* Utility routines.
*
* Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
* Copyright (C) 2005 by Rob Landley <rob@landley.net>
*
* Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
*/
#include "libbb.h"
#include <linux/version.h>
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,0)
/* For 2.6, use the cleaned up header to get the 64 bit API. */
/* linux/loop.h relies on __u64. Make sure we have that as a proper type
* until userspace is widely fixed. */
# if (defined __INTEL_COMPILER && !defined __GNUC__) \
|| (defined __GNUC__ && defined __STRICT_ANSI__)
__extension__ typedef long long __s64;
__extension__ typedef unsigned long long __u64;
# endif
# include <linux/loop.h>
typedef struct loop_info64 bb_loop_info;
# define BB_LOOP_SET_STATUS LOOP_SET_STATUS64
# define BB_LOOP_GET_STATUS LOOP_GET_STATUS64
#else
/* For 2.4 and earlier, use the 32 bit API (and don't trust the headers) */
/* Stuff stolen from linux/loop.h for 2.4 and earlier kernels */
# include <linux/posix_types.h>
# define LO_NAME_SIZE 64
# define LO_KEY_SIZE 32
# define LOOP_SET_FD 0x4C00
# define LOOP_CLR_FD 0x4C01
# define BB_LOOP_SET_STATUS 0x4C02
# define BB_LOOP_GET_STATUS 0x4C03
typedef struct {
int lo_number;
__kernel_dev_t lo_device;
unsigned long lo_inode;
__kernel_dev_t lo_rdevice;
int lo_offset;
int lo_encrypt_type;
int lo_encrypt_key_size;
int lo_flags;
char lo_file_name[LO_NAME_SIZE];
unsigned char lo_encrypt_key[LO_KEY_SIZE];
unsigned long lo_init[2];
char reserved[4];
} bb_loop_info;
#endif
char* FAST_FUNC query_loop(const char *device)
{
int fd;
bb_loop_info loopinfo;
char *dev = 0;
fd = open(device, O_RDONLY);
if (fd < 0) return 0;
if (!ioctl(fd, BB_LOOP_GET_STATUS, &loopinfo))
dev = xasprintf("%ld %s", (long) loopinfo.lo_offset,
(char *)loopinfo.lo_file_name);
close(fd);
return dev;
}
int FAST_FUNC del_loop(const char *device)
{
int fd, rc;
fd = open(device, O_RDONLY);
if (fd < 0) return 1;
rc = ioctl(fd, LOOP_CLR_FD, 0);
close(fd);
return rc;
}
/* Returns 0 if mounted RW, 1 if mounted read-only, <0 for error.
*device is loop device to use, or if *device==NULL finds a loop device to
mount it on and sets *device to a strdup of that loop device name. This
search will re-use an existing loop device already bound to that
file/offset if it finds one.
*/
int FAST_FUNC set_loop(char **device, const char *file, unsigned long long offset)
{
char dev[LOOP_NAMESIZE];
char *try;
bb_loop_info loopinfo;
struct stat statbuf;
int i, dfd, ffd, mode, rc = -1;
/* Open the file. Barf if this doesn't work. */
mode = O_RDWR;
ffd = open(file, mode);
if (ffd < 0) {
mode = O_RDONLY;
ffd = open(file, mode);
if (ffd < 0)
return -errno;
}
/* Find a loop device. */
try = *device ? *device : dev;
for (i = 0; rc; i++) {
sprintf(dev, LOOP_FORMAT, i);
/* Ran out of block devices, return failure. */
if (stat(try, &statbuf) || !S_ISBLK(statbuf.st_mode)) {
rc = -ENOENT;
break;
}
/* Open the sucker and check its loopiness. */
dfd = open(try, mode);
if (dfd < 0 && errno == EROFS) {
mode = O_RDONLY;
dfd = open(try, mode);
}
if (dfd < 0)
goto try_again;
rc = ioctl(dfd, BB_LOOP_GET_STATUS, &loopinfo);
/* If device is free, claim it. */
if (rc && errno == ENXIO) {
memset(&loopinfo, 0, sizeof(loopinfo));
safe_strncpy((char *)loopinfo.lo_file_name, file, LO_NAME_SIZE);
loopinfo.lo_offset = offset;
/* Associate free loop device with file. */
if (!ioctl(dfd, LOOP_SET_FD, ffd)) {
if (!ioctl(dfd, BB_LOOP_SET_STATUS, &loopinfo))
rc = 0;
else
ioctl(dfd, LOOP_CLR_FD, 0);
}
/* If this block device already set up right, re-use it.
(Yes this is racy, but associating two loop devices with the same
file isn't pretty either. In general, mounting the same file twice
without using losetup manually is problematic.)
*/
} else if (strcmp(file, (char *)loopinfo.lo_file_name) != 0
|| offset != loopinfo.lo_offset) {
rc = -1;
}
close(dfd);
try_again:
if (*device) break;
}
close(ffd);
if (!rc) {
if (!*device)
*device = xstrdup(dev);
return (mode == O_RDONLY); /* 1:ro, 0:rw */
}
return rc;
}