diff mbox series

[firmware-utils,v2,2/2] mkh3cvfs: add filesystem tool for H3C devices

Message ID 20220727193144.3782822-3-jan@3e8.eu
State Accepted
Headers show
Series add tools for H3C devices | expand

Commit Message

Jan Hoffmann July 27, 2022, 7:31 p.m. UTC
The bootloader can only boot images from a custom filesystem which
normally spans most of the flash on the device. This tool creates a
filesystem image containing a single file (which should be a firmware
image created using mkh3cimg).

The size of the filesystem is hardcoded in the bootloader. However, an
image that is only slightly larger than its content is sufficient, as
long as no write operations are performed in the bootloader menu. This
allows to append further data (i.e. rootfs) after the end of the
filesystem image.

Signed-off-by: Jan Hoffmann <jan@3e8.eu>
---
 CMakeLists.txt |   1 +
 src/mkh3cvfs.c | 426 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 427 insertions(+)
 create mode 100644 src/mkh3cvfs.c
diff mbox series

Patch

diff --git a/CMakeLists.txt b/CMakeLists.txt
index daf442681d66..9b70226b8dc1 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -66,6 +66,7 @@  FW_UTIL(mkedimaximg "" "" "")
 FW_UTIL(mkfwimage "" "-Wextra -D_FILE_OFFSET_BITS=64" "${ZLIB_LIBRARIES}")
 FW_UTIL(mkfwimage2 "" "" "${ZLIB_LIBRARIES}")
 FW_UTIL(mkh3cimg "" "" "")
+FW_UTIL(mkh3cvfs "" "" "")
 FW_UTIL(mkheader_gemtek "" "" "${ZLIB_LIBRARIES}")
 FW_UTIL(mkhilinkfw "" "" "${OPENSSL_CRYPTO_LIBRARIES}")
 FW_UTIL(mkmerakifw src/sha1.c "" "")
diff --git a/src/mkh3cvfs.c b/src/mkh3cvfs.c
new file mode 100644
index 000000000000..e5b3f353c920
--- /dev/null
+++ b/src/mkh3cvfs.c
@@ -0,0 +1,426 @@ 
+// SPDX-License-Identifier: GPL-2.0-only
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdint.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <byteswap.h>
+#include <endian.h>
+#include <getopt.h>
+
+
+#if !defined(__BYTE_ORDER)
+#error "Unknown byte order"
+#endif
+
+#if __BYTE_ORDER == __BIG_ENDIAN
+#define cpu_to_be16(x)  (x)
+#define cpu_to_be32(x)  (x)
+#define be16_to_cpu(x)  (x)
+#define be32_to_cpu(x)  (x)
+#elif __BYTE_ORDER == __LITTLE_ENDIAN
+#define cpu_to_be16(x)  bswap_16(x)
+#define cpu_to_be32(x)  bswap_32(x)
+#define be16_to_cpu(x)  bswap_16(x)
+#define be32_to_cpu(x)  bswap_32(x)
+#else
+#error "Unsupported endianness"
+#endif
+
+
+#define FAT_PTR_FLAGS_GET(x)     ((be32_to_cpu(x) & 0xff000000) >> 24)
+#define FAT_PTR_FLAGS_SET(x, y)  (x = cpu_to_be32((be32_to_cpu(x) & 0x00ffffff) | ((y & 0x000000ff) << 24)))
+
+#define FAT_PTR_VAL_GET(x)     (be32_to_cpu(x) & 0x00ffffff)
+#define FAT_PTR_VAL_SET(x, y)  (x = cpu_to_be32((be32_to_cpu(x) & 0xff000000) | (y & 0x00ffffff)))
+
+
+struct fat_entry {
+	/* first byte contains flags */
+	uint32_t previous;
+
+	/* first byte is reserved */
+	uint32_t next;
+} __attribute__ ((packed));
+
+struct file_entry {
+	uint8_t flags;
+
+	uint8_t res0[5];
+
+	uint16_t year;
+	uint8_t month;
+	uint8_t day;
+	uint8_t hour;
+	uint8_t minute;
+	uint8_t second;
+
+	uint8_t res1[3];
+
+	uint32_t length;
+
+	uint32_t parent_block;
+	uint16_t parent_index;
+
+	uint8_t res2[2];
+
+	uint32_t data_block;
+
+	char name[96];
+} __attribute__ ((packed));
+
+
+#define ERASEBLOCK_SIZE   0x10000
+#define BLOCK_SIZE        0x400
+
+#define BLOCKS_PER_ERASEBLOCK   (ERASEBLOCK_SIZE / BLOCK_SIZE)
+#define FILE_ENTRIES_PER_BLOCK  (BLOCK_SIZE / sizeof(struct file_entry))
+
+
+#define FLAG_FREE       0x80
+#define FLAG_VALID      0x40
+#define FLAG_INVALID    0x20
+#define FLAG_HIDE       0x10
+#define FLAG_DIRECTORY  0x02
+#define FLAG_READONLY   0x01
+
+
+static FILE *f;
+static size_t file_size = 0;
+
+static int dir_block = 1;
+static int dir_count = 0;
+
+static int next_data_block = 2;
+
+
+static inline size_t fat_entry_offset(int block) {
+	return ERASEBLOCK_SIZE * (block / (BLOCKS_PER_ERASEBLOCK-1))
+		+ sizeof(struct fat_entry) * (block % (BLOCKS_PER_ERASEBLOCK-1));
+}
+
+static inline size_t block_offset(int block) {
+	return ERASEBLOCK_SIZE * (block / (BLOCKS_PER_ERASEBLOCK-1))
+		+ BLOCK_SIZE * (1 + (block % (BLOCKS_PER_ERASEBLOCK-1)));
+}
+
+static int init_eraseblock(size_t offset) {
+	size_t end = offset - (offset % ERASEBLOCK_SIZE) + ERASEBLOCK_SIZE;
+	char *fill = "\xff";
+	int i;
+
+	while (file_size < end) {
+		if (fseek(f, file_size, SEEK_SET)) {
+			fprintf(stderr, "failed to seek to end\n");
+			return -1;
+		}
+
+		for (i = 0; i < ERASEBLOCK_SIZE; i++) {
+			if (fwrite(fill, 1, 1, f) != 1) {
+				fprintf(stderr, "failed to write eraseblock\n");
+				return -1;
+			}
+		}
+
+		file_size += ERASEBLOCK_SIZE;
+	}
+
+	return 0;
+}
+
+static inline void init_fat_entry(struct fat_entry *out) {
+	memset(out, '\xff', sizeof(struct fat_entry));
+}
+
+static int read_fat_entry(struct fat_entry *out, int block) {
+	size_t offset = fat_entry_offset(block);
+
+	if (init_eraseblock(offset)) {
+		return -1;
+	}
+
+	if (fseek(f, offset, SEEK_SET)) {
+		fprintf(stderr, "failed to seek to fat entry\n");
+		return -1;
+	}
+
+	if (fread(out, sizeof(struct fat_entry), 1, f) != 1) {
+		fprintf(stderr, "failed to read fat entry\n");
+		return -1;
+	}
+
+	return 0;
+}
+
+static int write_fat_entry(struct fat_entry *in, int block) {
+	size_t offset = fat_entry_offset(block);
+
+	if (init_eraseblock(offset)) {
+		return -1;
+	}
+
+	if (fseek(f, offset, SEEK_SET)) {
+		fprintf(stderr, "failed to seek to fat entry\n");
+		return -1;
+	}
+
+	if (fwrite(in, sizeof(struct fat_entry), 1, f) != 1) {
+		fprintf(stderr, "failed to write fat entry\n");
+		return -1;
+	}
+
+	return 0;
+}
+
+static inline void init_file_entry(struct file_entry *out) {
+	memset(out, '\xff', sizeof(struct file_entry));
+}
+
+static int write_file_entry(struct file_entry *in, int block, int index) {
+	size_t offset = block_offset(block) + sizeof(struct file_entry) * index;
+
+	if (init_eraseblock(offset)) {
+		return -1;
+	}
+
+	if (fseek(f, offset, SEEK_SET)) {
+		fprintf(stderr, "failed to seek to file entry\n");
+		return -1;
+	}
+
+	if (fwrite(in, sizeof(struct file_entry), 1, f) != 1) {
+		fprintf(stderr, "failed to write file entry\n");
+		return -1;
+	}
+
+	return 0;
+}
+
+static int write_block(void *in, size_t len, int block) {
+	size_t offset = block_offset(block);
+
+	if (init_eraseblock(offset)) {
+		return -1;
+	}
+
+	if (fseek(f, offset, SEEK_SET)) {
+		fprintf(stderr, "failed to seek to block\n");
+		return -1;
+	}
+
+	if (fwrite(in, len, 1, f) != 1) {
+		fprintf(stderr, "failed to write block\n");
+		return -1;
+	}
+
+	return 0;
+}
+
+static int create_root_directory() {
+	struct fat_entry fat;
+	struct file_entry file;
+
+	/* write format flag / FAT entry for block 0 (contains root file entry) */
+	init_fat_entry(&fat);
+	fat.previous = cpu_to_be32((ERASEBLOCK_SIZE << 12) | BLOCK_SIZE);
+	if (write_fat_entry(&fat, 0)) {
+		return -1;
+	}
+
+	/* write root file entry */
+	init_file_entry(&file);
+	file.flags = ~(FLAG_FREE | FLAG_VALID) & 0xff;
+	file.parent_block = cpu_to_be32(0);
+	file.data_block = cpu_to_be32(1);
+	if (write_file_entry(&file, 0, 0)) {
+		return -1;
+	}
+
+	/* write FAT entry for block 1 (contains first file entries of root directory) */
+	init_fat_entry(&fat);
+	FAT_PTR_FLAGS_SET(fat.previous, ~(FLAG_FREE | FLAG_VALID));
+	if (write_fat_entry(&fat, 1)) {
+		return -1;
+	}
+
+	return 0;
+}
+
+static int write_file(char *name, char *path) {
+	int ret = -1;
+	struct fat_entry fat;
+	struct file_entry file;
+	FILE *fin;
+	char buf[BLOCK_SIZE];
+	size_t len;
+	size_t total = 0;
+	int first_data_block = next_data_block;
+	int data_block = 0;
+
+	fin = fopen(path, "r");
+	if (fin == NULL) {
+		fprintf(stderr, "failed to open input file\n");
+		return ret;
+	}
+
+	while ((len = fread(buf, 1, BLOCK_SIZE, fin)) != 0 || !data_block) {
+		total += len;
+
+		/* update next pointer of previous FAT entry */
+		if (data_block) {
+			if (read_fat_entry(&fat, data_block)) {
+				goto err;
+			}
+			FAT_PTR_VAL_SET(fat.next, next_data_block);
+			if (write_fat_entry(&fat, data_block)) {
+				goto err;
+			}
+		}
+
+		/* write FAT entry for new block */
+		init_fat_entry(&fat);
+		FAT_PTR_FLAGS_SET(fat.previous, ~(FLAG_FREE | FLAG_VALID));
+		if (data_block) {
+			FAT_PTR_VAL_SET(fat.previous, data_block);
+		}
+		if (write_fat_entry(&fat, next_data_block)) {
+			goto err;
+		}
+
+		/* write data block */
+		if (write_block(buf, len, next_data_block)) {
+			goto err;
+		}
+
+		data_block = next_data_block;
+		next_data_block++;
+	}
+
+	/* create new file entries block if necessary */
+	if (dir_count == FILE_ENTRIES_PER_BLOCK) {
+		/* update next pointer of previous FAT entry */
+		if (read_fat_entry(&fat, dir_block)) {
+			goto err;
+		}
+		FAT_PTR_VAL_SET(fat.next, next_data_block);
+		if (write_fat_entry(&fat, dir_block)) {
+			goto err;
+		}
+
+		/* write FAT entry for new block */
+		init_fat_entry(&fat);
+		FAT_PTR_FLAGS_SET(fat.previous, ~(FLAG_FREE | FLAG_VALID));
+		FAT_PTR_VAL_SET(fat.previous, dir_block);
+		if (write_fat_entry(&fat, next_data_block)) {
+			goto err;
+		}
+
+		dir_block = next_data_block;
+		dir_count = 0;
+		next_data_block++;
+	}
+
+	/* write file entry */
+	init_file_entry(&file);
+
+	file.flags = ~(FLAG_FREE | FLAG_VALID) & 0xff;
+
+	file.year = cpu_to_be16(1970);
+	file.month = 1;
+	file.day = 1;
+	file.hour = 0;
+	file.minute = 0;
+	file.second = 0;
+
+	file.length = cpu_to_be32(total);
+
+	file.parent_block = cpu_to_be32(0);
+	file.parent_index = cpu_to_be16(0);
+
+	file.data_block = cpu_to_be32(first_data_block);
+
+	snprintf(file.name, sizeof(file.name), "%s", name);
+
+	if (write_file_entry(&file, dir_block, dir_count)) {
+		goto err;
+	}
+
+	dir_count++;
+
+	ret = 0;
+err:
+	fclose(fin);
+	return ret;
+}
+
+static void usage(char* argv[]) {
+	printf("Usage: %s [OPTIONS...]\n"
+	       "\n"
+	       "Options:\n"
+	       "  -f <filename>  filename in image\n"
+	       "  -i <file>      input filename\n"
+	       "  -o <file>      output filename\n"
+	       , argv[0]);
+}
+
+int main(int argc, char* argv[]) {
+	int ret = EXIT_FAILURE;
+
+	static char *filename = NULL;
+	static char *input_filename = NULL;
+	static char *output_filename = NULL;
+
+	while ( 1 ) {
+		int c;
+
+		c = getopt(argc, argv, "f:i:o:");
+		if (c == -1)
+			break;
+
+		switch (c) {
+		case 'f':
+			filename = optarg;
+			break;
+		case 'i':
+			input_filename = optarg;
+			break;
+		case 'o':
+			output_filename = optarg;
+			break;
+		default:
+			usage(argv);
+			goto err;
+		}
+	}
+
+	if (!filename || strlen(filename) == 0 ||
+		!input_filename || strlen(input_filename) == 0 ||
+		!output_filename || strlen(output_filename) == 0) {
+
+		usage(argv);
+		goto err;
+	}
+
+	f = fopen(output_filename, "w+");
+	if (f == NULL) {
+		fprintf(stderr, "failed to open output file\n");
+		goto err;
+	}
+
+	if (create_root_directory()) {
+		goto err_close;
+	}
+
+	if (write_file(filename, input_filename)) {
+		goto err_close;
+	}
+
+	ret = EXIT_SUCCESS;
+
+err_close:
+	fclose(f);
+err:
+	return ret;
+}