-#define _XOPEN_SOURCE 500
+#define _FILE_OFFSET_BITS 64
+#define __USE_FILE_OFFSET64
+#define _XOPEN_SOURCE 600
#include <fcntl.h>
#include <linux/fs.h>
+#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
-#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/stat.h>
+#include <unistd.h>
+#include <uuid/uuid.h>
-static const char bcache_magic[] = {
- 0xc6, 0x85, 0x73, 0xf6, 0x4e, 0x1a, 0x45, 0xca,
- 0x82, 0x65, 0xf5, 0x7f, 0x48, 0xba, 0x6d, 0x81 };
-
-struct cache_sb {
- uint8_t magic[16];
- uint32_t version;
- uint16_t block_size; /* sectors */
- uint16_t bucket_size; /* sectors */
- uint32_t journal_start; /* buckets */
- uint32_t first_bucket; /* start of data */
- uint64_t nbuckets; /* device size */
- uint64_t btree_root;
- uint16_t btree_level;
-};
-
-struct bucket_disk {
- uint16_t priority;
- uint8_t generation;
-} __attribute((packed));
+#include "bcache.h"
char zero[4096];
-int main(int argc, char **argv)
+uint64_t getblocks(int fd)
{
- long n;
- int fd, i;
+ uint64_t ret;
struct stat statbuf;
+ if (fstat(fd, &statbuf)) {
+ perror("stat error\n");
+ exit(EXIT_FAILURE);
+ }
+ ret = statbuf.st_size / 512;
+ if (S_ISBLK(statbuf.st_mode))
+ if (ioctl(fd, BLKGETSIZE, &ret)) {
+ perror("ioctl error");
+ exit(EXIT_FAILURE);
+ }
+ return ret;
+}
+
+uint64_t hatoi(const char *s)
+{
+ char *e;
+ long long i = strtoll(s, &e, 10);
+ switch (*e) {
+ case 't':
+ case 'T':
+ i *= 1024;
+ case 'g':
+ case 'G':
+ i *= 1024;
+ case 'm':
+ case 'M':
+ i *= 1024;
+ case 'k':
+ case 'K':
+ i *= 1024;
+ }
+ return i;
+}
+
+void usage()
+{
+ printf("Usage: make-bcache [options] device\n"
+ " -C Format a cache device\n"
+ " -B Format a backing device\n"
+ " -b bucket size\n"
+ " -w block size (hard sector size of SSD, often 2k)\n"
+ " -j journal size, in buckets\n"
+ " -U UUID\n"
+ " -S Set UUID\n");
+ exit(EXIT_FAILURE);
+}
+
+int main(int argc, char **argv)
+{
+ bool cache = false, backingdev = false;
+ int64_t nblocks, journal = 0;
+ int fd, i, c;
+ char uuid[40], set_uuid[40];
struct cache_sb sb;
- if (argc < 2) {
+ memset(&sb, 0, sizeof(struct cache_sb));
+
+ uuid_generate(sb.uuid);
+ uuid_generate(sb.set_uuid);
+
+ while ((c = getopt(argc, argv, "CBU:w:b:j:")) != -1)
+ switch (c) {
+ case 'C':
+ cache = true;
+ break;
+ case 'B':
+ backingdev = true;
+ break;
+ case 'b':
+ sb.bucket_size = hatoi(optarg) / 512;
+ break;
+ case 'w':
+ sb.block_size = hatoi(optarg) / 512;
+ break;
+ case 'j':
+ journal = atoi(optarg);
+ break;
+ case 'U':
+ if (uuid_parse(optarg, sb.uuid)) {
+ printf("Bad uuid\n");
+ exit(EXIT_FAILURE);
+ }
+ break;
+ case 'S':
+ if (uuid_parse(optarg, sb.set_uuid)) {
+ printf("Bad uuid\n");
+ exit(EXIT_FAILURE);
+ }
+ break;
+ }
+
+ if (!sb.block_size)
+ sb.block_size = 4;
+
+ if (!sb.bucket_size)
+ sb.bucket_size = cache ? 256 : 8192;
+
+ if (cache == backingdev) {
+ printf("Must specify one of -C or -B\n");
+ usage();
+ }
+
+ if (argc <= optind) {
printf("Please supply a device\n");
exit(EXIT_FAILURE);
}
- fd = open(argv[1], O_RDWR);
- if (!fd) {
+ fd = open(argv[optind], O_RDWR);
+ if (fd == -1) {
perror("Can't open dev\n");
exit(EXIT_FAILURE);
}
+ nblocks = getblocks(fd);
+ printf("device is %ju sectors\n", nblocks);
- if (fstat(fd, &statbuf)) {
- perror("stat error\n");
+ if (sb.bucket_size < sb.block_size ||
+ sb.bucket_size > nblocks / 8) {
+ printf("Bad bucket size %i\n", sb.bucket_size);
exit(EXIT_FAILURE);
}
- if (!S_ISBLK(statbuf.st_mode))
- n = statbuf.st_blocks;
- else
- if (ioctl(fd, BLKGETSIZE, &n)) {
- perror("ioctl error");
- exit(EXIT_FAILURE);
- }
memcpy(sb.magic, bcache_magic, 16);
- sb.version = 0;
- sb.block_size = 8;
- sb.bucket_size = 32;
- sb.nbuckets = n / sb.bucket_size;
+ sb.version = backingdev ? CACHE_BACKING_DEV : 0;
+ sb.nbuckets = nblocks / sb.bucket_size;
+ sb.nr_in_set = 1;
+ uuid_unparse(sb.uuid, uuid);
+ uuid_unparse(sb.set_uuid, set_uuid);
- do
- sb.first_bucket = ((--sb.nbuckets * sizeof(struct bucket_disk))
- + 4096 * 3) / (sb.bucket_size * 512) + 1;
- while ((sb.nbuckets + sb.first_bucket) * sb.bucket_size * 512
- > statbuf.st_size);
-
- sb.journal_start = sb.first_bucket;
-
- sb.btree_root = sb.first_bucket * sb.bucket_size;
- sb.btree_level = 0;
+ sb.journal_start = ((sb.nbuckets * sizeof(struct bucket_disk)) + (24 << 9)) / (sb.bucket_size << 9) + 1;
+ sb.first_bucket = sb.journal_start + journal;
printf("block_size: %u\n"
"bucket_size: %u\n"
"journal_start: %u\n"
"first_bucket: %u\n"
- "nbuckets: %ju\n",
+ "nbuckets: %ju\n"
+ "UUID: %s\n"
+ "Set UUID: %s\n",
sb.block_size,
sb.bucket_size,
sb.journal_start,
sb.first_bucket,
- sb.nbuckets);
-
- /* Zero out priorities */
- lseek(fd, 4096, SEEK_SET);
- for (i = 8; i < sb.first_bucket * sb.bucket_size; i++)
- if (write(fd, zero, 512) != 512)
- goto err;
+ sb.nbuckets,
+ uuid, set_uuid);
+
+ if (!backingdev) {
+ /* Zero out priorities */
+ lseek(fd, 4096, SEEK_SET);
+ for (i = 8; i < sb.first_bucket * sb.bucket_size; i++)
+ if (write(fd, zero, 512) != 512)
+ goto err;
+ }
if (pwrite(fd, &sb, sizeof(sb), 4096) != sizeof(sb))
goto err;
+ fsync(fd);
exit(EXIT_SUCCESS);
err:
perror("write error\n");