diff options
Diffstat (limited to 'btrfs-vol.c')
-rw-r--r-- | btrfs-vol.c | 145 |
1 files changed, 145 insertions, 0 deletions
diff --git a/btrfs-vol.c b/btrfs-vol.c new file mode 100644 index 00000000..0cc2e979 --- /dev/null +++ b/btrfs-vol.c @@ -0,0 +1,145 @@ +/* + * Copyright (C) 2007 Oracle. All rights reserved. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public + * License v2 as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * 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 021110-1307, USA. + */ + +#define _GNU_SOURCE +#ifndef __CHECKER__ +#include <sys/ioctl.h> +#include <sys/mount.h> +#include "ioctl.h" +#endif +#include <stdio.h> +#include <stdlib.h> +#include <getopt.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <unistd.h> +#include <dirent.h> +#include <uuid/uuid.h> +#include "kerncompat.h" +#include "ctree.h" +#include "transaction.h" +#include "utils.h" +#include "volumes.h" + +#ifdef __CHECKER__ +#define BLKGETSIZE64 0 +#define BTRFS_IOC_SNAP_CREATE 0 +#define BTRFS_IOC_ADD_DEV 0 +#define BTRFS_IOC_RM_DEV 0 +#define BTRFS_VOL_NAME_MAX 255 +struct btrfs_ioctl_vol_args { char name[BTRFS_VOL_NAME_MAX]; }; +static inline int ioctl(int fd, int define, void *arg) { return 0; } +#endif + +static void print_usage(void) +{ + fprintf(stderr, "usage: btrfs-vol [options] mount_point\n"); + fprintf(stderr, "\t-a device add one device\n"); + fprintf(stderr, "\t-b balance chunks across all devices\n"); + fprintf(stderr, "\t-r device remove one device\n"); + exit(1); +} + +static struct option long_options[] = { + /* { "byte-count", 1, NULL, 'b' }, */ + { "add", 1, NULL, 'a' }, + { "balance", 0, NULL, 'b' }, + { "remove", 1, NULL, 'r' }, + { 0, 0, 0, 0} +}; + +int main(int ac, char **av) +{ + struct stat st; + char *device = NULL; + char *mnt = NULL; + int ret; + int option_index = 0; + int cmd; + int fd; + int devfd; + DIR *dirstream; + struct btrfs_ioctl_vol_args args; + u64 dev_block_count = 0; + + while(1) { + int c; + c = getopt_long(ac, av, "a:br:", long_options, + &option_index); + if (c < 0) + break; + switch(c) { + case 'a': + device = strdup(optarg); + cmd = BTRFS_IOC_ADD_DEV; + break; + case 'b': + cmd = BTRFS_IOC_BALANCE; + break; + case 'r': + device = strdup(optarg); + cmd = BTRFS_IOC_RM_DEV; + break; + default: + print_usage(); + } + } + ac = ac - optind; + if (ac == 0) + print_usage(); + mnt = av[optind]; + + if (device) { + devfd = open(device, O_RDWR); + if (!devfd) { + fprintf(stderr, "Unable to open device %s\n", device); + } + ret = fstat(devfd, &st); + if (ret) { + fprintf(stderr, "Unable to stat %s\n", device); + exit(1); + } + if (!S_ISBLK(st.st_mode)) { + fprintf(stderr, "%s is not a block device\n", device); + exit(1); + } + } + dirstream = opendir(mnt); + if (!dirstream) { + fprintf(stderr, "Unable to open directory %s\n", mnt); + exit(1); + } + if (cmd == BTRFS_IOC_ADD_DEV) { + ret = btrfs_prepare_device(devfd, device, 1, &dev_block_count); + if (ret) { + fprintf(stderr, "Unable to init %s\n", device); + exit(1); + } + } + fd = dirfd(dirstream); + if (device) + strcpy(args.name, device); + else + args.name[0] = '\0'; + + ret = ioctl(fd, cmd, &args); + printf("ioctl returns %d\n", ret); + return 0; +} + |