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)
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;
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);
}
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));
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;
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);
// 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);
}
}
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
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;
}
}
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];
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;
}