ps,top: add an option to show threads. +260 bytes of code
[oweals/busybox.git] / editors / patch.c
index 2a2b130b973cea81fbc77ffc806239ac1895ac46..4a97151449751e5fc21c8d1369aaf28110ac679b 100644 (file)
@@ -21,7 +21,7 @@
 
 #include "libbb.h"
 
-static unsigned copy_lines(FILE *src_stream, FILE *dest_stream, unsigned lines_count)
+static unsigned copy_lines(FILE *src_stream, FILE *dst_stream, unsigned lines_count)
 {
        while (src_stream && lines_count) {
                char *line;
@@ -29,7 +29,7 @@ static unsigned copy_lines(FILE *src_stream, FILE *dest_stream, unsigned lines_c
                if (line == NULL) {
                        break;
                }
-               if (fputs(line, dest_stream) == EOF) {
+               if (fputs(line, dst_stream) == EOF) {
                        bb_perror_msg_and_die("error writing to new file");
                }
                free(line);
@@ -50,7 +50,7 @@ static char *extract_filename(char *line, int patch_level, const char *pat)
 
        if (strncmp(line, pat, 4) == 0) {
                /* Terminate string at end of source filename */
-               line[strcspn(line,"\t\n\r")] = '\0';
+               line[strcspn(line, "\t\n\r")] = '\0';
 
                /* Skip over (patch_level) number of leading directories */
                while (patch_level--) {
@@ -66,24 +66,32 @@ static char *extract_filename(char *line, int patch_level, const char *pat)
 }
 
 int patch_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
-int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
+int patch_main(int argc UNUSED_PARAM, char **argv)
 {
        struct stat saved_stat;
        char *patch_line;
        FILE *patch_file;
        int patch_level;
        int ret = 0;
+       char plus = '+';
+       unsigned opt;
+       enum {
+               OPT_R = (1 << 2),
+               OPT_N = (1 << 3),
+       };
 
        xfunc_error_retval = 2;
        {
                const char *p = "-1";
                const char *i = "-"; /* compat */
-               getopt32(argv, "p:i:", &p, &i);
+               opt = getopt32(argv, "p:i:RN", &p, &i);
+               if (opt & OPT_R)
+                       plus = '-';
                patch_level = xatoi(p); /* can be negative! */
                patch_file = xfopen_stdin(i);
        }
 
-       patch_line = xmalloc_getline(patch_file);
+       patch_line = xmalloc_fgetline(patch_file);
        while (patch_line) {
                FILE *src_stream;
                FILE *dst_stream;
@@ -91,8 +99,8 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                char *new_filename;
                char *backup_filename;
                unsigned src_cur_line = 1;
-               unsigned dest_cur_line = 0;
-               unsigned dest_beg_line;
+               unsigned dst_cur_line = 0;
+               unsigned dst_beg_line;
                unsigned bad_hunk_count = 0;
                unsigned hunk_count = 0;
                smallint copy_trailing_lines_flag = 0;
@@ -104,7 +112,7 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                        /* Extract the filename used before the patch was generated */
                        new_filename = extract_filename(patch_line, patch_level, "--- ");
                        // was old_filename above
-                       patch_line = xmalloc_getline(patch_file);
+                       patch_line = xmalloc_fgetline(patch_file);
                        if (!patch_line) goto quit;
                } while (!new_filename);
                free(new_filename); // "source" filename is irrelevant
@@ -129,9 +137,9 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                } else {
                        backup_filename = xasprintf("%s.orig", new_filename);
                        xrename(new_filename, backup_filename);
-                       src_stream = xfopen(backup_filename, "r");
+                       src_stream = xfopen_for_read(backup_filename);
                }
-               dst_stream = xfopen(new_filename, "w");
+               dst_stream = xfopen_for_write(new_filename);
                fchmod(fileno(dst_stream), saved_stat.st_mode);
 
                printf("patching file %s\n", new_filename);
@@ -143,15 +151,26 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                        unsigned src_beg_line;
                        unsigned hunk_offset_start;
                        unsigned src_last_line = 1;
+                       unsigned dst_last_line = 1;
 
-                       if ((sscanf(patch_line, "@@ -%d,%d +%d", &src_beg_line, &src_last_line, &dest_beg_line) != 3)
-                        && (sscanf(patch_line, "@@ -%d +%d", &src_beg_line, &dest_beg_line) != 2)
-                       ) { /* No more hunks for this file */
+                       if ((sscanf(patch_line, "@@ -%d,%d +%d,%d", &src_beg_line, &src_last_line, &dst_beg_line, &dst_last_line) < 3)
+                        && (sscanf(patch_line, "@@ -%d +%d,%d", &src_beg_line, &dst_beg_line, &dst_last_line) < 2)
+                       ) {
+                               /* No more hunks for this file */
                                break;
                        }
+                       if (plus != '+') {
+                               /* reverse patch */
+                               unsigned tmp = src_last_line;
+                               src_last_line = dst_last_line;
+                               dst_last_line = tmp;
+                               tmp = src_beg_line;
+                               src_beg_line = dst_beg_line;
+                               dst_beg_line = tmp;
+                       }
                        hunk_count++;
 
-                       if (src_beg_line && dest_beg_line) {
+                       if (src_beg_line && dst_beg_line) {
                                /* Copy unmodified lines upto start of hunk */
                                /* src_beg_line will be 0 if it's a new file */
                                count = src_beg_line - src_cur_line;
@@ -159,14 +178,15 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                                        bb_error_msg_and_die("bad src file");
                                }
                                src_cur_line += count;
-                               dest_cur_line += count;
+                               dst_cur_line += count;
                                copy_trailing_lines_flag = 1;
                        }
                        src_last_line += hunk_offset_start = src_cur_line;
+                       dst_last_line += dst_cur_line;
 
                        while (1) {
                                free(patch_line);
-                               patch_line = xmalloc_fgets(patch_file);
+                               patch_line = xmalloc_fgets(patch_file);
                                if (patch_line == NULL)
                                        break; /* EOF */
                                if ((*patch_line != '-') && (*patch_line != '+')
@@ -174,7 +194,7 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                                ) {
                                        break; /* End of hunk */
                                }
-                               if (*patch_line != '+') { /* '-', ' ' or '\n' */
+                               if (*patch_line != plus) { /* '-' or ' ' */
                                        char *src_line = NULL;
                                        if (src_cur_line == src_last_line)
                                                break;
@@ -184,20 +204,27 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                                                        int diff = strcmp(src_line, patch_line + 1);
                                                        src_cur_line++;
                                                        free(src_line);
-                                                       if (diff) src_line = NULL;
+                                                       if (diff)
+                                                               src_line = NULL;
                                                }
                                        }
+                                       /* Do not patch an already patched hunk with -N */
+                                       if (src_line == 0 && (opt & OPT_N)) {
+                                               continue;
+                                       }
                                        if (!src_line) {
                                                bb_error_msg("hunk #%u FAILED at %u", hunk_count, hunk_offset_start);
                                                bad_hunk_count++;
                                                break;
                                        }
-                                       if (*patch_line == '-') {
+                                       if (*patch_line != ' ') { /* '-' */
                                                continue;
                                        }
                                }
+                               if (dst_cur_line == dst_last_line)
+                                       break;
                                fputs(patch_line + 1, dst_stream);
-                               dest_cur_line++;
+                               dst_cur_line++;
                        } /* end of while loop handling one hunk */
                } /* end of while loop handling one file */
 
@@ -217,7 +244,7 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                        if (backup_filename) {
                                unlink(backup_filename);
                        }
-                       if ((dest_cur_line == 0) || (dest_beg_line == 0)) {
+                       if ((dst_cur_line == 0) || (dst_beg_line == 0)) {
                                /* The new patched file is empty, remove it */
                                xunlink(new_filename);
                                // /* old_filename and new_filename may be the same file */
@@ -228,8 +255,7 @@ int patch_main(int argc ATTRIBUTE_UNUSED, char **argv)
                //free(old_filename);
                free(new_filename);
        } /* end of "while there are patch lines" */
-quit:
-
+ quit:
        /* 0 = SUCCESS
         * 1 = Some hunks failed
         * 2 = More serious problems (exited earlier)