tools/mtd-utils: add EOF marker support to libubigen/ubinize
Signed-off-by: Gabor Juhos <juhosg@openwrt.org> SVN-Revision: 38682
This commit is contained in:
		@@ -0,0 +1,89 @@
 | 
				
			|||||||
 | 
					--- a/ubi-utils/src/libubigen.c
 | 
				
			||||||
 | 
					+++ b/ubi-utils/src/libubigen.c
 | 
				
			||||||
 | 
					@@ -122,8 +122,9 @@ int ubigen_add_volume(const struct ubige
 | 
				
			||||||
 | 
					 	return 0;
 | 
				
			||||||
 | 
					 }
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					-void ubigen_init_ec_hdr(const struct ubigen_info *ui,
 | 
				
			||||||
 | 
					-		        struct ubi_ec_hdr *hdr, long long ec)
 | 
				
			||||||
 | 
					+static void __ubigen_init_ec_hdr(const struct ubigen_info *ui,
 | 
				
			||||||
 | 
					+				 struct ubi_ec_hdr *hdr, long long ec,
 | 
				
			||||||
 | 
					+				 int eof)
 | 
				
			||||||
 | 
					 {
 | 
				
			||||||
 | 
					 	uint32_t crc;
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					@@ -136,10 +137,22 @@ void ubigen_init_ec_hdr(const struct ubi
 | 
				
			||||||
 | 
					 	hdr->data_offset = cpu_to_be32(ui->data_offs);
 | 
				
			||||||
 | 
					 	hdr->image_seq = cpu_to_be32(ui->image_seq);
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					+	if (eof) {
 | 
				
			||||||
 | 
					+		hdr->padding1[0] = 'E';
 | 
				
			||||||
 | 
					+		hdr->padding1[1] = 'O';
 | 
				
			||||||
 | 
					+		hdr->padding1[2] = 'F';
 | 
				
			||||||
 | 
					+	}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					 	crc = mtd_crc32(UBI_CRC32_INIT, hdr, UBI_EC_HDR_SIZE_CRC);
 | 
				
			||||||
 | 
					 	hdr->hdr_crc = cpu_to_be32(crc);
 | 
				
			||||||
 | 
					 }
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					+void ubigen_init_ec_hdr(const struct ubigen_info *ui,
 | 
				
			||||||
 | 
					+		        struct ubi_ec_hdr *hdr, long long ec)
 | 
				
			||||||
 | 
					+{
 | 
				
			||||||
 | 
					+	__ubigen_init_ec_hdr(ui, hdr, ec, 0);
 | 
				
			||||||
 | 
					+}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					 void ubigen_init_vid_hdr(const struct ubigen_info *ui,
 | 
				
			||||||
 | 
					 			 const struct ubigen_vol_info *vi,
 | 
				
			||||||
 | 
					 			 struct ubi_vid_hdr *hdr, int lnum,
 | 
				
			||||||
 | 
					@@ -307,6 +320,39 @@ int ubigen_write_layout_vol(const struct
 | 
				
			||||||
 | 
					 	}
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					 	free(outbuf);
 | 
				
			||||||
 | 
					+	return 0;
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+out_free:
 | 
				
			||||||
 | 
					+	free(outbuf);
 | 
				
			||||||
 | 
					+	return -1;
 | 
				
			||||||
 | 
					+}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+int ubigen_write_eof_markers(const struct ubigen_info *ui, long long ec,
 | 
				
			||||||
 | 
					+			     int count, int out_fd)
 | 
				
			||||||
 | 
					+{
 | 
				
			||||||
 | 
					+	char *outbuf;
 | 
				
			||||||
 | 
					+	int peb_size = ui->peb_size;
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+	outbuf = malloc(peb_size);
 | 
				
			||||||
 | 
					+	if (!outbuf) {
 | 
				
			||||||
 | 
					+		sys_errmsg("cannot allocate %d bytes of memory", peb_size);
 | 
				
			||||||
 | 
					+		return -1;
 | 
				
			||||||
 | 
					+	}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+	memset(outbuf, 0xFF, peb_size);
 | 
				
			||||||
 | 
					+	__ubigen_init_ec_hdr(ui, (struct ubi_ec_hdr *)outbuf, ec, 1);
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+	while (count) {
 | 
				
			||||||
 | 
					+		if (write(out_fd, outbuf, peb_size) != peb_size) {
 | 
				
			||||||
 | 
					+			sys_errmsg("cannot write %d bytes to the output file",
 | 
				
			||||||
 | 
					+				   peb_size);
 | 
				
			||||||
 | 
					+			goto out_free;
 | 
				
			||||||
 | 
					+		}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+		count--;
 | 
				
			||||||
 | 
					+	}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+	free(outbuf);
 | 
				
			||||||
 | 
					 	return 0;
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					 out_free:
 | 
				
			||||||
 | 
					--- a/ubi-utils/include/libubigen.h
 | 
				
			||||||
 | 
					+++ b/ubi-utils/include/libubigen.h
 | 
				
			||||||
 | 
					@@ -188,6 +188,9 @@ int ubigen_write_layout_vol(const struct
 | 
				
			||||||
 | 
					 			    long long ec1, long long ec2,
 | 
				
			||||||
 | 
					 			    struct ubi_vtbl_record *vtbl, int fd);
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					+int ubigen_write_eof_markers(const struct ubigen_info *ui, long long ec,
 | 
				
			||||||
 | 
					+			     int count, int out_fd);
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					 #ifdef __cplusplus
 | 
				
			||||||
 | 
					 }
 | 
				
			||||||
 | 
					 #endif
 | 
				
			||||||
@@ -0,0 +1,77 @@
 | 
				
			|||||||
 | 
					--- a/ubi-utils/src/ubinize.c
 | 
				
			||||||
 | 
					+++ b/ubi-utils/src/ubinize.c
 | 
				
			||||||
 | 
					@@ -70,6 +70,8 @@ static const char optionsstr[] =
 | 
				
			||||||
 | 
					 "                             (default is 1)\n"
 | 
				
			||||||
 | 
					 "-Q, --image-seq=<num>        32-bit UBI image sequence number to use\n"
 | 
				
			||||||
 | 
					 "                             (by default a random number is picked)\n"
 | 
				
			||||||
 | 
					+"-E, --eof-markers=<num>      number of eof-markers to put at the end of the\n"
 | 
				
			||||||
 | 
					+"                             output image\n"
 | 
				
			||||||
 | 
					 "-v, --verbose                be verbose\n"
 | 
				
			||||||
 | 
					 "-h, --help                   print help message\n"
 | 
				
			||||||
 | 
					 "-V, --version                print program version";
 | 
				
			||||||
 | 
					@@ -79,7 +81,7 @@ static const char usage[] =
 | 
				
			||||||
 | 
					 "\t\t[-x <num>] [-Q <num>] [-v] [-h] [-V] [--output=<filename>] [--peb-size=<bytes>]\n"
 | 
				
			||||||
 | 
					 "\t\t[--min-io-size=<bytes>] [--sub-page-size=<bytes>] [--vid-hdr-offset=<num>]\n"
 | 
				
			||||||
 | 
					 "\t\t[--erase-counter=<num>] [--ubi-ver=<num>] [--image-seq=<num>] [--verbose] [--help]\n"
 | 
				
			||||||
 | 
					-"\t\t[--version] ini-file\n"
 | 
				
			||||||
 | 
					+"\t\t[--eof-markers=<num>] [--version] ini-file\n"
 | 
				
			||||||
 | 
					 "Example: " PROGRAM_NAME " -o ubi.img -p 16KiB -m 512 -s 256 cfg.ini - create UBI image\n"
 | 
				
			||||||
 | 
					 "         'ubi.img' as described by configuration file 'cfg.ini'";
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					@@ -125,6 +127,7 @@ static const struct option long_options[
 | 
				
			||||||
 | 
					 	{ .name = "erase-counter",  .has_arg = 1, .flag = NULL, .val = 'e' },
 | 
				
			||||||
 | 
					 	{ .name = "ubi-ver",        .has_arg = 1, .flag = NULL, .val = 'x' },
 | 
				
			||||||
 | 
					 	{ .name = "image-seq",      .has_arg = 1, .flag = NULL, .val = 'Q' },
 | 
				
			||||||
 | 
					+	{ .name = "eof-markers",    .has_arg = 1, .flag = NULL, .val = 'E' },
 | 
				
			||||||
 | 
					 	{ .name = "verbose",        .has_arg = 0, .flag = NULL, .val = 'v' },
 | 
				
			||||||
 | 
					 	{ .name = "help",           .has_arg = 0, .flag = NULL, .val = 'h' },
 | 
				
			||||||
 | 
					 	{ .name = "version",        .has_arg = 0, .flag = NULL, .val = 'V' },
 | 
				
			||||||
 | 
					@@ -144,6 +147,7 @@ struct args {
 | 
				
			||||||
 | 
					 	uint32_t image_seq;
 | 
				
			||||||
 | 
					 	int verbose;
 | 
				
			||||||
 | 
					 	dictionary *dict;
 | 
				
			||||||
 | 
					+	int eof_markers;
 | 
				
			||||||
 | 
					 };
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					 static struct args args = {
 | 
				
			||||||
 | 
					@@ -162,7 +166,7 @@ static int parse_opt(int argc, char * co
 | 
				
			||||||
 | 
					 		int key, error = 0;
 | 
				
			||||||
 | 
					 		unsigned long int image_seq;
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					-		key = getopt_long(argc, argv, "o:p:m:s:O:e:x:Q:vhV", long_options, NULL);
 | 
				
			||||||
 | 
					+		key = getopt_long(argc, argv, "o:p:m:s:O:e:x:Q:E:vhV", long_options, NULL);
 | 
				
			||||||
 | 
					 		if (key == -1)
 | 
				
			||||||
 | 
					 			break;
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					@@ -222,6 +226,12 @@ static int parse_opt(int argc, char * co
 | 
				
			||||||
 | 
					 			args.image_seq = image_seq;
 | 
				
			||||||
 | 
					 			break;
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					+		case 'E':
 | 
				
			||||||
 | 
					+			args.eof_markers = simple_strtoul(optarg, &error);
 | 
				
			||||||
 | 
					+			if (error)
 | 
				
			||||||
 | 
					+				return errmsg("bad number of eof-markers: \"%s\"", optarg);
 | 
				
			||||||
 | 
					+			break;
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					 		case 'v':
 | 
				
			||||||
 | 
					 			args.verbose = 1;
 | 
				
			||||||
 | 
					 			break;
 | 
				
			||||||
 | 
					@@ -599,6 +609,18 @@ int main(int argc, char * const argv[])
 | 
				
			||||||
 | 
					 			printf("\n");
 | 
				
			||||||
 | 
					 	}
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					+	if (args.eof_markers) {
 | 
				
			||||||
 | 
					+		verbose(args.verbose, "writing %d eof-marker blocks",
 | 
				
			||||||
 | 
					+			args.eof_markers);
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					+		err = ubigen_write_eof_markers(&ui, args.ec, args.eof_markers,
 | 
				
			||||||
 | 
					+					       args.out_fd);
 | 
				
			||||||
 | 
					+		if (err) {
 | 
				
			||||||
 | 
					+			errmsg("cannot write eof-marker blocks");
 | 
				
			||||||
 | 
					+			goto out_free;
 | 
				
			||||||
 | 
					+		}
 | 
				
			||||||
 | 
					+	}
 | 
				
			||||||
 | 
					+
 | 
				
			||||||
 | 
					 	verbose(args.verbose, "writing layout volume");
 | 
				
			||||||
 | 
					 
 | 
				
			||||||
 | 
					 	err = ubigen_write_layout_vol(&ui, 0, 1, args.ec, args.ec, vtbl, args.out_fd);
 | 
				
			||||||
		Reference in New Issue
	
	Block a user