X-Git-Url: http://git.ozo.com/?a=blobdiff_plain;f=tools%2Fpatch-image%2Fsrc%2Fpatch-cmdline.c;h=8d1fce9172c48de14643240e99315eb1da030b29;hb=HEAD;hp=d3a5b119e446a542f3ac7bc09c755f8fe8e0d050;hpb=7eeb254cc49c4e9f837c980826ec112b9404f341;p=lede-git%2F.git

diff --git a/tools/patch-image/src/patch-cmdline.c b/tools/patch-image/src/patch-cmdline.c
index d3a5b119e4..8d1fce9172 100644
--- a/tools/patch-image/src/patch-cmdline.c
+++ b/tools/patch-image/src/patch-cmdline.c
@@ -35,10 +35,16 @@ int main(int argc, char **argv)
 {
 	int fd, found = 0, len, ret = -1;
 	char *ptr, *p;
+	unsigned int search_space;
 
-	if (argc != 3) {
-		fprintf(stderr, "Usage: %s <file> <cmdline>\n", argv[0]);
+	if (argc <= 2 || argc > 4) {
+		fprintf(stderr, "Usage: %s <file> <cmdline> [size]\n", argv[0]);
 		goto err1;
+	} else if (argc == 3) {
+		fprintf(stdout, "search space used is default of 16KB\n");
+		search_space = SEARCH_SPACE;
+	} else {
+		search_space = atoi(argv[3]);
 	}
 	len = strlen(argv[2]);
 	if (len + 9 > CMDLINE_MAX) {
@@ -47,12 +53,12 @@ int main(int argc, char **argv)
 	}
 	
 	if (((fd = open(argv[1], O_RDWR)) < 0) ||
-		(ptr = (char *) mmap(0, SEARCH_SPACE + CMDLINE_MAX, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0)) == (void *) (-1)) {
+		(ptr = (char *) mmap(0, search_space + CMDLINE_MAX, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0)) == (void *) (-1)) {
 		fprintf(stderr, "Could not open kernel image");
 		goto err2;
 	}
 	
-	for (p = ptr; p < (ptr + SEARCH_SPACE); p += 4) {
+	for (p = ptr; p < (ptr + search_space); p += 4) {
 		if (memcmp(p, "CMDLINE:", 8) == 0) {
 			found = 1;
 			p += 8;