]> 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 ec88ffc985e0c48de552822bd6e47a923614152f..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));
@@ -905,30 +897,39 @@ void sysfs_attr_list() {
 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);
-       }
-
-       if (pread(fd, sb, bytes, SB_START) != bytes) {
-               fprintf(stderr, "Couldn't read\n");
-               exit(2);
+               return NULL;
        }
 
-       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;
@@ -944,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);
 
@@ -958,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);
        }
 }
 
@@ -1029,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
@@ -1042,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;
                        }
                }
 
@@ -1269,7 +1290,7 @@ err:
        return err;
 }
 
-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];
@@ -1294,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;
+}
 
-               if(print_val)
-                       printf("%s\n", buf);
+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");
+
+       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;
 }