script_extractor argc and SCRIPT_SIZE patch

--- script_extractor.c.orig     2015-09-20 14:11:25.896712635 +0200
+++ script_extractor.c  2015-09-20 14:26:01.860734775 +0200
@@ -34,13 +34,13 @@
        fd = open("/dev/mem", O_RDONLY);

        size = SCRIPT_SIZE;
-       if (argc)
+       if (argc > 1)
                size = atoi(argv[1]);

        addr = (char *)mmap(NULL, size, PROT_READ, MAP_SHARED, fd, SCRIPT_START);
-       for (i = 0; i < SCRIPT_SIZE; i++)
+       for (i = 0; i < size; i++)
                putchar(addr[i]);
-       munmap(NULL, SCRIPT_SIZE);
+       munmap(NULL, size);
        close(fd);

        return 0;