aboutsummaryrefslogtreecommitdiff
path: root/toys/mke2fs.c
diff options
context:
space:
mode:
Diffstat (limited to 'toys/mke2fs.c')
-rw-r--r--toys/mke2fs.c149
1 files changed, 108 insertions, 41 deletions
diff --git a/toys/mke2fs.c b/toys/mke2fs.c
index bade3d22..7fb45e8c 100644
--- a/toys/mke2fs.c
+++ b/toys/mke2fs.c
@@ -31,12 +31,13 @@
#define INODES_RESERVED 10
// Calculate data blocks plus index blocks needed to hold a file.
-uint32_t blocks_used(uint64_t size)
+
+static uint32_t count_blocks_used(uint64_t size)
{
uint32_t dblocks = (uint32_t)((size+(TT.blocksize-1))/TT.blocksize);
uint32_t idx=TT.blocksize/4, iblocks=0, diblocks=0, tiblocks=0;
- // Account for direct, singly, doubly, and triply indiret index blocks
+ // Account for direct, singly, doubly, and triply indirect index blocks
if (dblocks > 12) {
iblocks = ((dblocks-13)/idx)+1;
@@ -46,10 +47,35 @@ uint32_t blocks_used(uint64_t size)
tiblocks = ((diblocks-2)/idx)+1;
}
}
-
+
return dblocks + iblocks + diblocks + tiblocks;
}
+// Calculate the number of blocks used by each inode. Returns blocks used,
+// assigns bytes used to *size. Writes total block count to TT.treeblocks
+// and inode count to TT.treeinodes.
+
+long check_treesize(struct dirtree *this, off_t *size)
+{
+ long blocks;
+
+ while (this) {
+ *size += sizeof(struct ext2_dentry) + strlen(this->name);
+
+ if (this->child)
+ this->st.st_blocks = check_treesize(this->child, &this->st.st_size);
+ else if (S_ISREG(this->st.st_mode)) {
+ this->st.st_blocks = count_blocks_used(this->st.st_size);
+ TT.treeblocks += this->st.st_blocks;
+ }
+ this = this->next;
+ }
+ TT.treeblocks += blocks = count_blocks_used(*size);
+ TT.treeinodes++;
+
+ return blocks;
+}
+
// According to http://www.opengroup.org/onlinepubs/9629399/apdxa.htm
// we should generate a uuid structure by reading a clock with 100 nanosecond
// precision, normalizing it to the start of the gregorian calendar in 1582,
@@ -60,7 +86,7 @@ uint32_t blocks_used(uint64_t size)
static void create_uuid(char *uuid)
{
- // Read 128 random bytes
+ // Read 128 random bits
int fd = xopen("/dev/urandom", O_RDONLY);
xreadall(fd, uuid, 16);
close(fd);
@@ -90,7 +116,6 @@ static void init_superblock(struct ext2_superblock *sb)
// Fill out blocks_count, r_blocks_count, first_data_block
sb->blocks_count = SWAP_LE32(TT.blocks);
-
if (!TT.reserved_percent) TT.reserved_percent = 5;
temp = (TT.blocks * (uint64_t)TT.reserved_percent) /100;
sb->r_blocks_count = SWAP_LE32(temp);
@@ -108,27 +133,27 @@ static void init_superblock(struct ext2_superblock *sb)
TT.groups = (TT.blocks)/temp;
if (TT.blocks & (temp-1)) TT.groups++;
- // Figure out how many inodes we need.
+ // Figure out how many total inodes we need.
- if (!TT.inodes) {
+ if (!TT.inodespg) {
if (!TT.bytes_per_inode) TT.bytes_per_inode = 8192;
- TT.inodes = (TT.blocks * (uint64_t)TT.blocksize) / TT.bytes_per_inode;
+ TT.inodespg = (TT.blocks * (uint64_t)TT.blocksize) / TT.bytes_per_inode;
}
// Figure out inodes per group, rounded up to block size.
// How many blocks of inodes total, rounded up
- temp = TT.inodes / (TT.blocksize/sizeof(struct ext2_inode));
- if (temp * (TT.blocksize/sizeof(struct ext2_inode)) != TT.inodes) temp++;
+ temp = TT.inodespg / (TT.blocksize/sizeof(struct ext2_inode));
+ if (temp * (TT.blocksize/sizeof(struct ext2_inode)) != TT.inodespg) temp++;
// How many blocks of inodes per group, again rounded up
- TT.inodes = temp / TT.groups;
- if (temp & (TT.groups-1)) TT.inodes++;
+ TT.inodespg = temp / TT.groups;
+ if (temp & (TT.groups-1)) TT.inodespg++;
// How many inodes per group is that?
- TT.inodes *= (TT.blocksize/sizeof(struct ext2_inode));
+ TT.inodespg *= (TT.blocksize/sizeof(struct ext2_inode));
// Set inodes_per_group and total inodes_count
- sb->inodes_per_group = SWAP_LE32(TT.inodes);
- sb->inodes_count = SWAP_LE32(TT.inodes *= TT.groups);
+ sb->inodes_per_group = SWAP_LE32(TT.inodespg);
+ sb->inodes_count = SWAP_LE32(TT.inodespg * TT.groups);
// Fill out the rest of the superblock.
sb->max_mnt_count=0xFFFF;
@@ -195,6 +220,43 @@ static void bits_set(char *array, int start, int len)
}
}
+// Seek past len bytes (to maintain sparse file), or write zeroes if output
+// not seekable
+static void put_zeroes(int len)
+{
+ if(TT.noseek || -1 == lseek(TT.fsfd, len, SEEK_SET)) {
+
+ TT.noseek=1;
+ memset(toybuf, 0, sizeof(toybuf));
+ while (len) {
+ int out = len > sizeof(toybuf) ? sizeof(toybuf) : len;
+ xwrite(TT.fsfd, toybuf, out);
+ len -= out;
+ }
+ }
+}
+
+static void fill_inode(struct ext2_inode *in, struct dirtree *this)
+{
+ memset(in,0,sizeof(struct ext2_inode));
+
+ // This works on Linux. S_ISREG/DIR/CHR/BLK/FIFO/LNK/SOCK(m)
+ in->mode = this->st.st_mode;
+
+ in->uid = this->st.st_uid & 0xFFFF;
+ in->uid_high = this->st.st_uid >> 16;
+ in->gid = this->st.st_gid & 0xFFFF;
+ in->gid_high = this->st.st_gid >> 16;
+ in->size = this->st.st_size & 0xFFFFFFFF;
+
+ in->atime = this->st.st_atime;
+ in->ctime = this->st.st_ctime;
+ in->mtime = this->st.st_mtime;
+
+ in->links_count = this->st.st_nlink; // TODO
+ in->blocks = this->st.st_blocks;
+}
+
int mke2fs_main(void)
{
int i, temp, blockbits;
@@ -219,7 +281,7 @@ int mke2fs_main(void)
TT.dt->st.st_ctime = TT.dt->st.st_mtime = time(NULL);
}
- // Calculate st_nlink for each node in tree.
+ // TODO: Calculate st_nlink for each node in tree.
// TODO: Check if filesystem is mounted here
@@ -233,18 +295,26 @@ int mke2fs_main(void)
if (!TT.blocks) TT.blocks = length/TT.blocksize;
if (!TT.blocks) error_exit("gene2fs is a TODO item");
- // Skip the first 1k to avoid the boot sector (if any). Use this to
- // figure out if this file is seekable.
- if(-1 == lseek(TT.fsfd, 1024, SEEK_SET)) {
- TT.noseek=1;
- xwrite(TT.fsfd, &TT.sb, 1024);
- }
-
- // Initialize superblock structure
+ // Skip the first 1k to avoid the boot sector (if any), then
+ // initialize superblock structure
+ put_zeroes(1024);
init_superblock(&TT.sb);
blockbits = 8*TT.blocksize;
+ // Figure out how much space is used
+ length = 0;
+ length = check_treesize(TT.dt, &length);
+ for (temp=i=0; i<TT.groups; i++) {
+ temp += group_superblock_used(i) + 2;
+ temp += TT.inodespg/(TT.blocksize/sizeof(struct ext2_inode));
+ }
+ TT.sb.free_blocks_count = SWAP_LE32(TT.blocks - TT.treeblocks - temp);
+ TT.sb.free_inodes_count = SWAP_LE32(TT.inodespg*TT.groups - INODES_RESERVED
+ - TT.treeinodes);
+
+ // Figure out how many inodes are used
+
// Loop through block groups.
for (i=0; i<TT.groups; i++) {
@@ -257,7 +327,7 @@ int mke2fs_main(void)
if ((i+1)*blockbits > TT.blocks) end = TT.blocks & (blockbits-1);
// Blocks used by inode table
- itable = ((TT.inodes/TT.groups)*sizeof(struct ext2_inode))/TT.blocksize;
+ itable = (TT.inodespg*sizeof(struct ext2_inode))/TT.blocksize;
// If a superblock goes here, write it out.
start = group_superblock_used(i);
@@ -286,18 +356,13 @@ int mke2fs_main(void)
memset(bg, 0, TT.blocksize);
}
- // sb.inodes_per_group is uint32_t, but group.free_inodes_count
- // is uint16_t. Add in endianness conversion and this little
- // dance is called for.
- temp = SWAP_LE32(TT.sb.inodes_per_group);
+ // How many free inodes in this group? (TODO)
+ temp = TT.inodespg;
if (!i) temp -= INODES_RESERVED;
bg[slot].free_inodes_count = SWAP_LE16(temp);
-
- // How many blocks will the inode table use?
- temp *= sizeof(struct ext2_inode);
- temp /= TT.blocksize;
- // How many does that leave? (TODO: fill it up)
+ // How many free blocks in this group? (TODO)
+ temp = TT.inodespg/(TT.blocksize/sizeof(struct ext2_inode)) + 2;
temp = end-used-temp;
bg[slot].free_blocks_count = SWAP_LE32(temp);
@@ -321,26 +386,28 @@ int mke2fs_main(void)
xwrite(TT.fsfd, toybuf, TT.blocksize);
// Write inode bitmap (TODO)
- temp = TT.inodes/TT.groups;
memset(toybuf, 0, TT.blocksize);
if (!i) bits_set(toybuf, 0, INODES_RESERVED);
- bits_set(toybuf, temp, blockbits-temp);
+ bits_set(toybuf, TT.inodespg, blockbits-TT.inodespg);
xwrite(TT.fsfd, toybuf, TT.blocksize);
+ start += 3;
+
// Write inode table for this group
- for (j = 0; j<temp; j++) {
+ for (j = 0; j<TT.inodespg; j++) {
slot = j % (TT.blocksize/sizeof(struct ext2_inode));
if (!slot) {
- if (j) xwrite(TT.fsfd, in, TT.blocksize);
+ if (j) {
+ xwrite(TT.fsfd, in, TT.blocksize);
+ start++;
+ }
memset(in, 0, TT.blocksize);
}
}
xwrite(TT.fsfd, in, TT.blocksize);
// Write empty data blocks
- memset(toybuf, 0, TT.blocksize);
- for (j = start; j < end; j++)
- xwrite(TT.fsfd, toybuf, TT.blocksize);
+ put_zeroes((end-start) * TT.blocksize);
}
return 0;