libbb/loop.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 1999-2005 by Erik Andersen <andersen@codepoet.org>
   6  *
   7  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
   8  */
   9
  10
  11 #include <features.h>
  12 #include <stdio.h>
  13 #include <errno.h>
  14 #include <fcntl.h>
  15 #include <string.h>
  16 #include <unistd.h>
  17 #include <sys/ioctl.h>
  18 #include "libbb.h"
  19
  20 /* For 2.6, use the cleaned up header to get the 64 bit API. */
  21 #include <linux/version.h>
  22 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,0)
  23 #include <linux/loop.h>
  24 typedef struct loop_info64 bb_loop_info;
  25 #define BB_LOOP_SET_STATUS LOOP_SET_STATUS64
  26 #define BB_LOOP_GET_STATUS LOOP_GET_STATUS64
  27
  28 /* For 2.4 and earlier, use the 32 bit API (and don't trust the headers) */
  29 #else
  30 /* Stuff stolen from linux/loop.h for 2.4 and earlier kernels*/
  31 #include <linux/posix_types.h>
  32 #define LO_NAME_SIZE        64
  33 #define LO_KEY_SIZE         32
  34 #define LOOP_SET_FD         0x4C00
  35 #define LOOP_CLR_FD         0x4C01
  36 #define BB_LOOP_SET_STATUS  0x4C02
  37 #define BB_LOOP_GET_STATUS  0x4C03
  38 typedef struct {
  39         int                lo_number;
  40         __kernel_dev_t     lo_device;
  41         unsigned long      lo_inode;
  42         __kernel_dev_t     lo_rdevice;
  43         int                lo_offset;
  44         int                lo_encrypt_type;
  45         int                lo_encrypt_key_size;
  46         int                lo_flags;
  47         char               lo_file_name[LO_NAME_SIZE];
  48         unsigned char      lo_encrypt_key[LO_KEY_SIZE];
  49         unsigned long      lo_init[2];
  50         char               reserved[4];
  51 } bb_loop_info;
  52 #endif
  53
  54 extern int del_loop(const char *device)
  55 {
  56         int fd,rc=0;
  57
  58         if ((fd = open(device, O_RDONLY)) < 0) rc=1;
  59         else {
  60                 if (ioctl(fd, LOOP_CLR_FD, 0) < 0) rc=1;
  61                 close(fd);
  62         }
  63         return rc;
  64 }
  65
  66 /* Returns 0 if mounted RW, 1 if mounted read-only, <0 for error.
  67    *device is loop device to use, or if *device==NULL finds a loop device to
  68    mount it on and sets *device to a strdup of that loop device name.  This
  69    search will re-use an existing loop device already bound to that
  70    file/offset if it finds one.
  71  */
  72 extern int set_loop(char **device, const char *file, int offset)
  73 {
  74         char dev[20];
  75         bb_loop_info loopinfo;
  76         struct stat statbuf;
  77         int i, dfd, ffd, mode, rc=1;
  78
  79         /* Open the file.  Barf if this doesn't work.  */
  80         if((ffd = open(file, mode=O_RDWR))<0 && (ffd = open(file,mode=O_RDONLY))<0)
  81                 return errno;
  82
  83         /* Find a loop device.  */
  84         for(i=0;rc;i++) {
  85                 sprintf(dev, LOOP_FORMAT, i++);
  86                 /* Ran out of block devices, return failure.  */
  87                 if(stat(*device ? *device : dev, &statbuf) ||
  88                                 !S_ISBLK(statbuf.st_mode)) {
  89                         rc=ENOENT;
  90                         break;
  91                 }
  92                 /* Open the sucker and check its loopiness.  */
  93                 if((dfd=open(dev, mode))<0 && errno==EROFS)
  94                         dfd=open(dev,mode=O_RDONLY);
  95                 if(dfd<0) continue;
  96
  97                 rc=ioctl(dfd, BB_LOOP_GET_STATUS, &loopinfo);
  98                 /* If device free, claim it.  */
  99                 if(rc && errno==ENXIO) {
 100                         memset(&loopinfo, 0, sizeof(loopinfo));
 101                         safe_strncpy(loopinfo.lo_file_name, file, LO_NAME_SIZE);
 102                         loopinfo.lo_offset = offset;
 103                         /* Associate free loop device with file.  */
 104                         if(!ioctl(dfd, LOOP_SET_FD, ffd) &&
 105                            !ioctl(dfd, BB_LOOP_SET_STATUS, &loopinfo)) rc=0;
 106                         else ioctl(dfd, LOOP_CLR_FD, 0);
 107                 /* If this block device already set up right, re-use it.
 108                    (Yes this is racy, but associating two loop devices with the same
 109                    file isn't pretty either.  In general, mounting the same file twice
 110                    without using losetup manually is problematic.)
 111                  */
 112                 } else if(strcmp(file,loopinfo.lo_file_name)
 113                                         || offset!=loopinfo.lo_offset) rc=1;
 114                 close(dfd);
 115                 if(*device) break;
 116         }
 117         close(ffd);
 118         if(!rc) {
 119                 if(!*device) *device=strdup(dev);
 120                 return mode==O_RDONLY ? 1 : 0;
 121         } else return rc;
 122 }
 123
 124 /* END CODE */
 125 /*
 126 Local Variables:
 127 c-file-style: "linux"
 128 c-basic-offset: 4
 129 tab-width: 4
 130 End:
 131 */