]> git.sesse.net Git - bcachefs-tools-debian/blobdiff - bcache.c
Merge branch 'master' of ssh://gits.daterainc.com:2984/project/2013.MAIN/bcache-tools
[bcachefs-tools-debian] / bcache.c
index 2c1f2322be4222480ddd60e9d6c1da24c9a77b4a..2bc32c035a20efb55cebbd4945178d869bd0f235 100644 (file)
--- a/bcache.c
+++ b/bcache.c
@@ -559,16 +559,6 @@ static unsigned min_bucket_size(int num_bucket_sizes, unsigned *bucket_sizes)
        return min;
 }
 
-static unsigned node_size(unsigned bucket_size) {
-
-       if (bucket_size <= 256)
-               return bucket_size;
-       else if (bucket_size <= 512)
-               return bucket_size / 2;
-       else
-               return bucket_size / 4;
-}
-
 void write_cache_sbs(int *fds, struct cache_sb *sb,
                            unsigned block_size, unsigned *bucket_sizes,
                                int num_bucket_sizes)
@@ -596,7 +586,8 @@ void write_cache_sbs(int *fds, struct cache_sb *sb,
                        sb->bucket_size = bucket_sizes[0];
                else
                        sb->bucket_size = bucket_sizes[i];
-               SET_CACHE_BTREE_NODE_SIZE(sb, node_size(min_size));
+               SET_CACHE_BTREE_NODE_SIZE(sb, min_size);
+
 
                sb->uuid = m->uuid;
                sb->nbuckets            = getblocks(fds[i]) / sb->bucket_size;
@@ -755,7 +746,7 @@ static void show_super_common(struct cache_sb *sb, bool force_csum)
                printf(" [match]\n");
        } else {
                printf(" [expected %" PRIX64 "]\n", expected_csum);
-               if (!force_csum) {
+               if (force_csum) {
                        fprintf(stderr, "Corrupt superblock (bad csum)\n");
                        exit(2);
                }
@@ -829,6 +820,7 @@ static void show_cache_member(struct cache_sb *sb, unsigned i)
        struct cache_member *m = ((struct cache_member *) sb->d) + i;
 
        printf("cache.state\t%s\n",             cache_state[CACHE_STATE(m)]);
+
        printf("cache.tier\t%llu\n",            CACHE_TIER(m));
 
        printf("cache.replication_set\t%llu\n", CACHE_REPLICATION_SET(m));
@@ -868,33 +860,76 @@ void show_super_cache(struct cache_sb *sb, bool force_csum)
        show_cache_member(sb, sb->nr_this_dev);
 }
 
+static int __sysfs_attr_type(char *attr, const char **attr_arr) {
+       int i, j;
+       for(i = 0; attr_arr[i] != NULL; i++)
+               if(!strcmp(attr, attr_arr[i]))
+                       return 1;
+       return 0;
+}
+
+enum sysfs_attr sysfs_attr_type(char *attr) {
+       int ret;
+       if(__sysfs_attr_type(attr, set_attrs))
+               return SET_ATTR;
+       if(__sysfs_attr_type(attr, cache_attrs))
+               return CACHE_ATTR;
+       if(__sysfs_attr_type(attr, internal_attrs))
+               return INTERNAL_ATTR;
+
+       printf("No attribute called %s, try --list to see options\n", attr);
+
+       return -1;
+}
+
+static void __sysfs_attr_list(const char **attr_arr) {
+       int i, j;
+       for (i = 0; attr_arr[i] != NULL; i++)
+               printf("%s\n", attr_arr[i]);
+}
+
+void sysfs_attr_list() {
+       __sysfs_attr_list(set_attrs);
+       __sysfs_attr_list(cache_attrs);
+       __sysfs_attr_list(internal_attrs);
+}
+
 struct cache_sb *query_dev(char *dev, bool force_csum,
                bool print_sb, bool uuid_only, char *dev_uuid)
 {
-       struct cache_sb sb_stack, *sb = &sb_stack;
-       size_t bytes = sizeof(*sb);
+       size_t bytes = 4096;
+       struct cache_sb *sb = aligned_alloc(bytes, bytes);
 
-       int fd = open(dev, O_RDONLY);
+       int fd = open(dev, O_RDONLY|O_DIRECT);
        if (fd < 0) {
                printf("Can't open dev %s: %s\n", dev, strerror(errno));
-               exit(2);
+               return NULL;
        }
 
-       if (pread(fd, sb, bytes, SB_START) != bytes) {
-               fprintf(stderr, "Couldn't read\n");
-               exit(2);
-       }
-
-       if (sb->keys) {
-               bytes = sizeof(*sb) + sb->keys * sizeof(uint64_t);
-               sb = malloc(bytes);
-
-               if (pread(fd, sb, bytes, SB_START) != bytes) {
-                       fprintf(stderr, "Couldn't read\n");
-                       exit(2);
+       while (true) {
+               int ret = pread(fd, sb, bytes, SB_START);
+               if (ret < 0) {
+                       fprintf(stderr, "Couldn't read superblock: %s\n",
+                                       strerror(errno));
+                       close(fd);
+                       free(sb);
+                       return NULL;
+               } else if (bytes > sizeof(sb) + sb->keys * sizeof(u64)) {
+                       /* We read the whole superblock */
+                       break;
                }
+
+               /*
+                * otherwise double the size of our dest
+                * and read again
+                */
+               free(sb);
+               bytes *= 2;
+               sb = aligned_alloc(4096, bytes);
        }
 
+       close(fd);
+
        if(uuid_only) {
                show_uuid_only(sb, dev_uuid);
                return sb;
@@ -910,7 +945,7 @@ struct cache_sb *query_dev(char *dev, bool force_csum,
        return sb;
 }
 
-static void dev_name(const char *ugly_path) {
+static char *dev_name(const char *ugly_path) {
        char buf[32];
        int i, end = strlen(ugly_path);
 
@@ -924,42 +959,55 @@ static void dev_name(const char *ugly_path) {
 
        // Is the dev guaranteed to be in /dev?
        // This is needed for finding the superblock with a query-dev
-       printf("/dev%s\n", buf);
+       return strdup(buf);
 }
 
 static void list_cacheset_devs(char *cset_dir, char *cset_name, bool parse_dev_name) {
        int i = 0;
-       DIR *cachedir;
+       DIR *cachedir, *dir;
        struct stat cache_stat;
-       char intbuf[4];
        char entry[MAX_PATH];
-
-       snprintf(entry, MAX_PATH, "%s/%s/cache0", cset_dir, cset_name);
-       snprintf(intbuf, 4, "%d", i);
-
-       while(true) {
-               char buf[MAX_PATH];
-               int len;
-
-               if((cachedir = opendir(entry)) == NULL)
-                       break;
-
-               if(stat(entry, &cache_stat))
-                       break;
-
-               if((len = readlink(entry, buf, sizeof(buf) - 1)) != -1) {
-                       buf[len] = '\0';
-                       if(parse_dev_name)
-                               dev_name(buf);
-                       else
-                               printf("\t%s\n", buf);
+       struct dirent *ent;
+       snprintf(entry, MAX_PATH, "%s/%s", cset_dir, cset_name);
+
+       if((dir = opendir(entry)) != NULL) {
+               while((ent = readdir(dir)) != NULL) {
+                       char buf[MAX_PATH];
+                       int len;
+                       char *tmp;
+
+                       /*
+                        * We are looking for all cache# directories
+                        * do a strlen < 9 to skip over other entries
+                        * that also start with "cache"
+                        */
+                       if(strncmp(ent->d_name, "cache", 5) ||
+                                       !(strlen(ent->d_name) < 9))
+                               continue;
+
+                       snprintf(entry, MAX_PATH, "%s/%s/%s",
+                                       cset_dir,
+                                       cset_name,
+                                       ent->d_name);
+
+                       if((cachedir = opendir(entry)) == NULL)
+                               continue;
+
+                       if(stat(entry, &cache_stat))
+                               continue;
+
+                       if((len = readlink(entry, buf, sizeof(buf) - 1)) !=
+                                       -1) {
+                               buf[len] = '\0';
+                               if(parse_dev_name) {
+                                       tmp = dev_name(buf);
+                                       printf("/dev%s\n", tmp);
+                                       free(tmp);
+                               } else {
+                                       printf("\t%s\n", buf);
+                               }
+                       }
                }
-
-               /* remove i from end and append i++ */
-               entry[strlen(entry)-strlen(intbuf)] = 0;
-               i++;
-               snprintf(intbuf, 4, "%d", i);
-               strcat(entry, intbuf);
        }
 }
 
@@ -995,6 +1043,7 @@ char *find_matching_uuid(char *stats_dir, char *subdir, const char *stats_dev_uu
                        buf[len] = '\0';
                        int i, end = strlen(buf);
                        char tmp[32], devname[32];
+                       struct cache_sb *sb;
 
                        /* Chop off "/bcache", then look for the
                         * next '/' from the end
@@ -1008,10 +1057,16 @@ char *find_matching_uuid(char *stats_dir, char *subdir, const char *stats_dev_uu
                        strcpy(devname, "/dev");
                        strcat(devname, tmp);
 
-                       query_dev(devname, false, false, true, dev_uuid);
+                       err = "Unable to open superblock";
+                       sb = query_dev(devname, false, false, true, dev_uuid);
+                       if(!sb)
+                               return err;
+                       else
+                               free(sb);
+
                        if(!strcmp(stats_dev_uuid, dev_uuid)) {
                                strcat(subdir, intbuf);
-                               return err;
+                               return NULL;
                        }
                }
 
@@ -1042,7 +1097,6 @@ char *list_cachesets(char *cset_dir, bool list_devs)
        while ((ent = readdir(dir)) != NULL) {
                struct stat statbuf;
                char entry[MAX_PATH];
-               struct dirent *cache_ent;
 
                if (!strcmp(ent->d_name, ".") || !strcmp(ent->d_name, ".."))
                        continue;
@@ -1118,6 +1172,62 @@ err:
        return err;
 }
 
+char *add_devices(char *const *devs, char *uuid)
+{
+       int ret, bcachefd;
+       char *err = NULL;
+
+       bcachefd = open("/dev/bcache", O_RDWR);
+       if (bcachefd < 0) {
+               err = "Can't open bcache device";
+               goto err;
+       }
+
+       struct bch_ioctl_add_disks ia;
+       ia.devs = devs;
+       ia.uuid = uuid;
+
+       ret = ioctl(bcachefd, BCH_IOCTL_ADD_DISKS, &ia);
+       if (ret < 0) {
+               char tmp[128];
+               snprintf(tmp, 128, "ioctl add disk error: %s\n",
+                               strerror(ret));
+               err = strdup(tmp);
+       }
+
+err:
+       close(bcachefd);
+       return err;
+}
+
+char *remove_device(const char *dev, bool force)
+{
+       int ret, bcachefd;
+       char *err = NULL;
+
+       bcachefd = open("/dev/bcache", O_RDWR);
+       if (bcachefd < 0) {
+               err = "Can't open bcache device";
+               goto err;
+       }
+
+       struct bch_ioctl_rm_disk ir;
+       ir.dev = dev;
+       ir.force = force ? 1 : 0;
+
+       ret = ioctl(bcachefd, BCH_IOCTL_RM_DISK, &ir);
+       if (ret < 0) {
+               char tmp[128];
+               snprintf(tmp, 128, "ioctl add disk error: %s\n",
+                               strerror(ret));
+               err = strdup(tmp);
+       }
+
+err:
+       close(bcachefd);
+       return err;
+}
+
 char *probe(char *dev, int udev)
 {
        struct cache_sb sb;
@@ -1180,19 +1290,7 @@ err:
        return err;
 }
 
-void sb_state(struct cache_sb *sb, char *dev)
-{
-       struct cache_member *m = ((struct cache_member *) sb->d) +
-               sb->nr_this_dev;
-
-       printf("device %s\n", dev);
-       printf("\tcache state\t%s\n",   cache_state[CACHE_STATE(m)]);
-       printf("\tcache_tier\t%llu\n", CACHE_TIER(m));
-       printf("\tseq#: \t%llu\n", sb->seq);
-
-}
-
-char *read_stat_dir(DIR *dir, char *stats_dir, char *stat_name, bool print_val)
+char *read_stat_dir(DIR *dir, char *stats_dir, char *stat_name, char *ret)
 {
        struct stat statbuf;
        char entry[MAX_PATH];
@@ -1217,14 +1315,114 @@ char *read_stat_dir(DIR *dir, char *stats_dir, char *stat_name, bool print_val)
                        return NULL;
                }
 
-               while(fgets(buf, MAX_PATH, fp));
+               while(fgets(ret, MAX_PATH, fp));
+               fclose(fp);
+       }
+err:
+       return err;
+}
+
+char *bcache_get_capacity(const char *cset_dir, const char *capacity_uuid,
+               bool show_devs)
+{
+       char *err = NULL;
+       char bucket_size_path[MAX_PATH];
+       char nbuckets_path[MAX_PATH];
+       char avail_buckets_path[MAX_PATH];
+       char cache_path[MAX_PATH];
+
+       double bucket_sizes[MAX_DEVS];
+       double nbuckets[MAX_DEVS];
+       double avail_buckets[MAX_DEVS];
+       char *dev_names[MAX_DEVS];
+       int dev_count = 0, i;
+       char intbuf[4];
+       double total_cap = 0, total_free = 0;
+       int precision = 2;
+
+       snprintf(intbuf, 4, "%d", i);
+       snprintf(bucket_size_path, MAX_PATH, "%s/%s/%s/%s", cset_dir,
+                       capacity_uuid, "cache0", "bucket_size_bytes");
+       snprintf(nbuckets_path, MAX_PATH, "%s/%s/%s/%s", cset_dir,
+                       capacity_uuid, "cache0", "nbuckets");
+       snprintf(avail_buckets_path, MAX_PATH, "%s/%s/%s/%s", cset_dir,
+                       capacity_uuid, "cache0", "available_buckets");
+       snprintf(cache_path, MAX_PATH, "%s/%s/%s", cset_dir, capacity_uuid,
+                       "cache0");
 
-               if(print_val)
-                       printf("%s\n", buf);
+       while(true) {
+               char buf[MAX_PATH];
+               int len;
+               DIR *cache_dir;
+
+               if((cache_dir = opendir(cache_path)) == NULL)
+                       break;
+
+               err = read_stat_dir(cache_dir, cache_path,
+                               "bucket_size_bytes", buf);
+               if (err)
+                       goto err;
                else
-                       printf("%s\n", stat_name);
-               fclose(fp);
+                       bucket_sizes[dev_count] = atof(buf);
+
+               err = read_stat_dir(cache_dir, cache_path,
+                               "nbuckets", buf);
+               if (err)
+                       goto err;
+               else
+                       nbuckets[dev_count] = atof(buf);
+
+               err = read_stat_dir(cache_dir, cache_path,
+                               "available_buckets", buf);
+               if (err)
+                       goto err;
+               else
+                       avail_buckets[dev_count] = atof(buf);
+
+               if((len = readlink(cache_path, buf, sizeof(buf) - 1)) != -1) {
+                       buf[len] = '\0';
+                       dev_names[dev_count] = dev_name(buf);
+               }
+
+               /* remove i/stat and append i++/stat */
+               bucket_size_path[strlen(cache_path) - strlen(intbuf)] = 0;
+               nbuckets_path[strlen(cache_path) - strlen(intbuf)] = 0;
+               avail_buckets_path[strlen(cache_path) - strlen(intbuf)] = 0;
+               cache_path[strlen(cache_path) - strlen(intbuf)] = 0;
+
+               dev_count++;
+
+               snprintf(intbuf, 4, "%d", dev_count);
+               strcat(cache_path, intbuf);
+               strcat(bucket_size_path, intbuf);
+               strcat(nbuckets_path, intbuf);
+               strcat(avail_buckets_path, intbuf);
        }
+
+       printf("%-15s%-25s%-25s\n", "Device Name", "Capacity (512 Blocks)", "Free (512 Blocks)");
+
+       if (show_devs) {
+               for (i = 0; i < dev_count; i++) {
+                       printf("%s%-11s%-25.*f%-25.*f\n", "/dev", dev_names[i],
+                               precision,
+                               (bucket_sizes[i] * nbuckets[i]) / 512,
+                               precision,
+                               (bucket_sizes[i] * avail_buckets[i]) / 512);
+               }
+       }
+
+       for (i = 0; i < dev_count; i++) {
+               total_cap += (bucket_sizes[i] * nbuckets[i]) / 512;
+               total_free += (bucket_sizes[i] * avail_buckets[i]) / 512;
+
+       }
+
+       printf("%-15s%-25.*f%-25.*f\n", "Total", precision, total_cap,
+                       precision, total_free);
+
 err:
+       for (i = 0; i < dev_count; i++)
+               if (dev_names[i])
+                       free(dev_names[i]);
        return err;
 }