Home | History | Annotate | Line # | Download | only in patch
patch.c revision 1.3
      1  1.1      cgd /* patch - a program to apply diffs to original files
      2  1.1      cgd  *
      3  1.1      cgd  * Copyright 1986, Larry Wall
      4  1.1      cgd  *
      5  1.1      cgd  * This program may be copied as long as you don't try to make any
      6  1.1      cgd  * money off of it, or pretend that you wrote it.
      7  1.1      cgd  */
      8  1.2  mycroft 
      9  1.2  mycroft #ifndef lint
     10  1.3  thorpej static char rcsid[] = "$Id: patch.c,v 1.3 1996/09/19 06:10:04 thorpej Exp $";
     11  1.2  mycroft #endif /* not lint */
     12  1.1      cgd 
     13  1.1      cgd #include "INTERN.h"
     14  1.1      cgd #include "common.h"
     15  1.1      cgd #include "EXTERN.h"
     16  1.1      cgd #include "version.h"
     17  1.1      cgd #include "util.h"
     18  1.1      cgd #include "pch.h"
     19  1.1      cgd #include "inp.h"
     20  1.1      cgd #include "backupfile.h"
     21  1.1      cgd 
     22  1.3  thorpej #include <stdlib.h>
     23  1.3  thorpej 
     24  1.1      cgd /* procedures */
     25  1.1      cgd 
     26  1.1      cgd void reinitialize_almost_everything();
     27  1.1      cgd void get_some_switches();
     28  1.1      cgd LINENUM locate_hunk();
     29  1.1      cgd void abort_hunk();
     30  1.1      cgd void apply_hunk();
     31  1.1      cgd void init_output();
     32  1.1      cgd void init_reject();
     33  1.1      cgd void copy_till();
     34  1.1      cgd void spew_output();
     35  1.1      cgd void dump_line();
     36  1.1      cgd bool patch_match();
     37  1.1      cgd bool similar();
     38  1.1      cgd void re_input();
     39  1.1      cgd void my_exit();
     40  1.1      cgd 
     41  1.1      cgd /* TRUE if -E was specified on command line.  */
     42  1.1      cgd static int remove_empty_files = FALSE;
     43  1.1      cgd 
     44  1.1      cgd /* TRUE if -R was specified on command line.  */
     45  1.1      cgd static int reverse_flag_specified = FALSE;
     46  1.1      cgd 
     47  1.1      cgd /* Apply a set of diffs as appropriate. */
     48  1.1      cgd 
     49  1.1      cgd int
     50  1.1      cgd main(argc,argv)
     51  1.1      cgd int argc;
     52  1.1      cgd char **argv;
     53  1.1      cgd {
     54  1.1      cgd     LINENUM where;
     55  1.1      cgd     LINENUM newwhere;
     56  1.1      cgd     LINENUM fuzz;
     57  1.1      cgd     LINENUM mymaxfuzz;
     58  1.1      cgd     int hunk = 0;
     59  1.1      cgd     int failed = 0;
     60  1.1      cgd     int failtotal = 0;
     61  1.1      cgd     int i;
     62  1.1      cgd 
     63  1.1      cgd     setbuf(stderr, serrbuf);
     64  1.1      cgd     for (i = 0; i<MAXFILEC; i++)
     65  1.1      cgd 	filearg[i] = Nullch;
     66  1.1      cgd 
     67  1.1      cgd     myuid = getuid();
     68  1.1      cgd 
     69  1.1      cgd     /* Cons up the names of the temporary files.  */
     70  1.1      cgd     {
     71  1.1      cgd       /* Directory for temporary files.  */
     72  1.1      cgd       char *tmpdir;
     73  1.1      cgd       int tmpname_len;
     74  1.1      cgd 
     75  1.1      cgd       tmpdir = getenv ("TMPDIR");
     76  1.1      cgd       if (tmpdir == NULL) {
     77  1.1      cgd 	tmpdir = "/tmp";
     78  1.1      cgd       }
     79  1.1      cgd       tmpname_len = strlen (tmpdir) + 20;
     80  1.1      cgd 
     81  1.1      cgd       TMPOUTNAME = (char *) malloc (tmpname_len);
     82  1.1      cgd       strcpy (TMPOUTNAME, tmpdir);
     83  1.1      cgd       strcat (TMPOUTNAME, "/patchoXXXXXX");
     84  1.1      cgd       Mktemp(TMPOUTNAME);
     85  1.1      cgd 
     86  1.1      cgd       TMPINNAME = (char *) malloc (tmpname_len);
     87  1.1      cgd       strcpy (TMPINNAME, tmpdir);
     88  1.1      cgd       strcat (TMPINNAME, "/patchiXXXXXX");
     89  1.1      cgd       Mktemp(TMPINNAME);
     90  1.1      cgd 
     91  1.1      cgd       TMPREJNAME = (char *) malloc (tmpname_len);
     92  1.1      cgd       strcpy (TMPREJNAME, tmpdir);
     93  1.1      cgd       strcat (TMPREJNAME, "/patchrXXXXXX");
     94  1.1      cgd       Mktemp(TMPREJNAME);
     95  1.1      cgd 
     96  1.1      cgd       TMPPATNAME = (char *) malloc (tmpname_len);
     97  1.1      cgd       strcpy (TMPPATNAME, tmpdir);
     98  1.1      cgd       strcat (TMPPATNAME, "/patchpXXXXXX");
     99  1.1      cgd       Mktemp(TMPPATNAME);
    100  1.1      cgd     }
    101  1.1      cgd 
    102  1.1      cgd     {
    103  1.1      cgd       char *v;
    104  1.1      cgd 
    105  1.1      cgd       v = getenv ("SIMPLE_BACKUP_SUFFIX");
    106  1.1      cgd       if (v)
    107  1.1      cgd 	simple_backup_suffix = v;
    108  1.1      cgd       else
    109  1.1      cgd 	simple_backup_suffix = ORIGEXT;
    110  1.1      cgd #ifndef NODIR
    111  1.1      cgd       v = getenv ("VERSION_CONTROL");
    112  1.1      cgd       backup_type = get_version (v); /* OK to pass NULL. */
    113  1.1      cgd #endif
    114  1.1      cgd     }
    115  1.1      cgd 
    116  1.1      cgd     /* parse switches */
    117  1.1      cgd     Argc = argc;
    118  1.1      cgd     Argv = argv;
    119  1.1      cgd     get_some_switches();
    120  1.1      cgd 
    121  1.1      cgd     /* make sure we clean up /tmp in case of disaster */
    122  1.1      cgd     set_signals(0);
    123  1.1      cgd 
    124  1.1      cgd     for (
    125  1.1      cgd 	open_patch_file(filearg[1]);
    126  1.1      cgd 	there_is_another_patch();
    127  1.1      cgd 	reinitialize_almost_everything()
    128  1.1      cgd     ) {					/* for each patch in patch file */
    129  1.1      cgd 
    130  1.1      cgd 	if (outname == Nullch)
    131  1.1      cgd 	    outname = savestr(filearg[0]);
    132  1.1      cgd 
    133  1.1      cgd 	/* for ed script just up and do it and exit */
    134  1.1      cgd 	if (diff_type == ED_DIFF) {
    135  1.1      cgd 	    do_ed_script();
    136  1.1      cgd 	    continue;
    137  1.1      cgd 	}
    138  1.1      cgd 
    139  1.1      cgd 	/* initialize the patched file */
    140  1.1      cgd 	if (!skip_rest_of_patch)
    141  1.1      cgd 	    init_output(TMPOUTNAME);
    142  1.1      cgd 
    143  1.1      cgd 	/* initialize reject file */
    144  1.1      cgd 	init_reject(TMPREJNAME);
    145  1.1      cgd 
    146  1.1      cgd 	/* find out where all the lines are */
    147  1.1      cgd 	if (!skip_rest_of_patch)
    148  1.1      cgd 	    scan_input(filearg[0]);
    149  1.1      cgd 
    150  1.1      cgd 	/* from here on, open no standard i/o files, because malloc */
    151  1.1      cgd 	/* might misfire and we can't catch it easily */
    152  1.1      cgd 
    153  1.1      cgd 	/* apply each hunk of patch */
    154  1.1      cgd 	hunk = 0;
    155  1.1      cgd 	failed = 0;
    156  1.1      cgd 	out_of_mem = FALSE;
    157  1.1      cgd 	while (another_hunk()) {
    158  1.1      cgd 	    hunk++;
    159  1.1      cgd 	    fuzz = Nulline;
    160  1.1      cgd 	    mymaxfuzz = pch_context();
    161  1.1      cgd 	    if (maxfuzz < mymaxfuzz)
    162  1.1      cgd 		mymaxfuzz = maxfuzz;
    163  1.1      cgd 	    if (!skip_rest_of_patch) {
    164  1.1      cgd 		do {
    165  1.1      cgd 		    where = locate_hunk(fuzz);
    166  1.1      cgd 		    if (hunk == 1 && where == Nulline && !force) {
    167  1.1      cgd 						/* dwim for reversed patch? */
    168  1.1      cgd 			if (!pch_swap()) {
    169  1.1      cgd 			    if (fuzz == Nulline)
    170  1.1      cgd 				say1(
    171  1.1      cgd "Not enough memory to try swapped hunk!  Assuming unswapped.\n");
    172  1.1      cgd 			    continue;
    173  1.1      cgd 			}
    174  1.1      cgd 			reverse = !reverse;
    175  1.1      cgd 			where = locate_hunk(fuzz);  /* try again */
    176  1.1      cgd 			if (where == Nulline) {	    /* didn't find it swapped */
    177  1.1      cgd 			    if (!pch_swap())         /* put it back to normal */
    178  1.1      cgd 				fatal1("lost hunk on alloc error!\n");
    179  1.1      cgd 			    reverse = !reverse;
    180  1.1      cgd 			}
    181  1.1      cgd 			else if (noreverse) {
    182  1.1      cgd 			    if (!pch_swap())         /* put it back to normal */
    183  1.1      cgd 				fatal1("lost hunk on alloc error!\n");
    184  1.1      cgd 			    reverse = !reverse;
    185  1.1      cgd 			    say1(
    186  1.1      cgd "Ignoring previously applied (or reversed) patch.\n");
    187  1.1      cgd 			    skip_rest_of_patch = TRUE;
    188  1.1      cgd 			}
    189  1.1      cgd 			else if (batch) {
    190  1.1      cgd 			    if (verbose)
    191  1.1      cgd 				say3(
    192  1.1      cgd "%seversed (or previously applied) patch detected!  %s -R.",
    193  1.1      cgd 				reverse ? "R" : "Unr",
    194  1.1      cgd 				reverse ? "Assuming" : "Ignoring");
    195  1.1      cgd 			}
    196  1.1      cgd 			else {
    197  1.1      cgd 			    ask3(
    198  1.1      cgd "%seversed (or previously applied) patch detected!  %s -R? [y] ",
    199  1.1      cgd 				reverse ? "R" : "Unr",
    200  1.1      cgd 				reverse ? "Assume" : "Ignore");
    201  1.1      cgd 			    if (*buf == 'n') {
    202  1.1      cgd 				ask1("Apply anyway? [n] ");
    203  1.1      cgd 				if (*buf != 'y')
    204  1.1      cgd 				    skip_rest_of_patch = TRUE;
    205  1.1      cgd 				where = Nulline;
    206  1.1      cgd 				reverse = !reverse;
    207  1.1      cgd 				if (!pch_swap())  /* put it back to normal */
    208  1.1      cgd 				    fatal1("lost hunk on alloc error!\n");
    209  1.1      cgd 			    }
    210  1.1      cgd 			}
    211  1.1      cgd 		    }
    212  1.1      cgd 		} while (!skip_rest_of_patch && where == Nulline &&
    213  1.1      cgd 		    ++fuzz <= mymaxfuzz);
    214  1.1      cgd 
    215  1.1      cgd 		if (skip_rest_of_patch) {		/* just got decided */
    216  1.1      cgd 		    Fclose(ofp);
    217  1.1      cgd 		    ofp = Nullfp;
    218  1.1      cgd 		}
    219  1.1      cgd 	    }
    220  1.1      cgd 
    221  1.1      cgd 	    newwhere = pch_newfirst() + last_offset;
    222  1.1      cgd 	    if (skip_rest_of_patch) {
    223  1.1      cgd 		abort_hunk();
    224  1.1      cgd 		failed++;
    225  1.1      cgd 		if (verbose)
    226  1.1      cgd 		    say3("Hunk #%d ignored at %ld.\n", hunk, newwhere);
    227  1.1      cgd 	    }
    228  1.1      cgd 	    else if (where == Nulline) {
    229  1.1      cgd 		abort_hunk();
    230  1.1      cgd 		failed++;
    231  1.1      cgd 		if (verbose)
    232  1.1      cgd 		    say3("Hunk #%d failed at %ld.\n", hunk, newwhere);
    233  1.1      cgd 	    }
    234  1.1      cgd 	    else {
    235  1.1      cgd 		apply_hunk(where);
    236  1.1      cgd 		if (verbose) {
    237  1.1      cgd 		    say3("Hunk #%d succeeded at %ld", hunk, newwhere);
    238  1.1      cgd 		    if (fuzz)
    239  1.1      cgd 			say2(" with fuzz %ld", fuzz);
    240  1.1      cgd 		    if (last_offset)
    241  1.1      cgd 			say3(" (offset %ld line%s)",
    242  1.1      cgd 			    last_offset, last_offset==1L?"":"s");
    243  1.1      cgd 		    say1(".\n");
    244  1.1      cgd 		}
    245  1.1      cgd 	    }
    246  1.1      cgd 	}
    247  1.1      cgd 
    248  1.1      cgd 	if (out_of_mem && using_plan_a) {
    249  1.1      cgd 	    Argc = Argc_last;
    250  1.1      cgd 	    Argv = Argv_last;
    251  1.1      cgd 	    say1("\n\nRan out of memory using Plan A--trying again...\n\n");
    252  1.1      cgd 	    if (ofp)
    253  1.1      cgd 	        Fclose(ofp);
    254  1.1      cgd 	    ofp = Nullfp;
    255  1.1      cgd 	    if (rejfp)
    256  1.1      cgd 	        Fclose(rejfp);
    257  1.1      cgd 	    rejfp = Nullfp;
    258  1.1      cgd 	    continue;
    259  1.1      cgd 	}
    260  1.1      cgd 
    261  1.1      cgd 	assert(hunk);
    262  1.1      cgd 
    263  1.1      cgd 	/* finish spewing out the new file */
    264  1.1      cgd 	if (!skip_rest_of_patch)
    265  1.1      cgd 	    spew_output();
    266  1.1      cgd 
    267  1.1      cgd 	/* and put the output where desired */
    268  1.1      cgd 	ignore_signals();
    269  1.1      cgd 	if (!skip_rest_of_patch) {
    270  1.1      cgd 	    struct stat statbuf;
    271  1.1      cgd 	    char *realout = outname;
    272  1.1      cgd 
    273  1.1      cgd 	    if (move_file(TMPOUTNAME, outname) < 0) {
    274  1.1      cgd 		toutkeep = TRUE;
    275  1.1      cgd 		realout = TMPOUTNAME;
    276  1.1      cgd 		chmod(TMPOUTNAME, filemode);
    277  1.1      cgd 	    }
    278  1.1      cgd 	    else
    279  1.1      cgd 		chmod(outname, filemode);
    280  1.1      cgd 
    281  1.1      cgd 	    if (remove_empty_files && stat(realout, &statbuf) == 0
    282  1.1      cgd 		&& statbuf.st_size == 0) {
    283  1.1      cgd 		if (verbose)
    284  1.1      cgd 		    say2("Removing %s (empty after patching).\n", realout);
    285  1.1      cgd 	        while (unlink(realout) >= 0) ; /* while is for Eunice.  */
    286  1.1      cgd 	    }
    287  1.1      cgd 	}
    288  1.1      cgd 	Fclose(rejfp);
    289  1.1      cgd 	rejfp = Nullfp;
    290  1.1      cgd 	if (failed) {
    291  1.1      cgd 	    failtotal += failed;
    292  1.1      cgd 	    if (!*rejname) {
    293  1.1      cgd 		Strcpy(rejname, outname);
    294  1.1      cgd #ifndef FLEXFILENAMES
    295  1.1      cgd 		{
    296  1.1      cgd 		    char *s = rindex(rejname,'/');
    297  1.1      cgd 
    298  1.1      cgd 		    if (!s)
    299  1.1      cgd 			s = rejname;
    300  1.1      cgd 		    if (strlen(s) > 13)
    301  1.1      cgd 			if (s[12] == '.')	/* try to preserve difference */
    302  1.1      cgd 			    s[12] = s[13];	/* between .h, .c, .y, etc. */
    303  1.1      cgd 			s[13] = '\0';
    304  1.1      cgd 		}
    305  1.1      cgd #endif
    306  1.1      cgd 		Strcat(rejname, REJEXT);
    307  1.1      cgd 	    }
    308  1.1      cgd 	    if (skip_rest_of_patch) {
    309  1.1      cgd 		say4("%d out of %d hunks ignored--saving rejects to %s\n",
    310  1.1      cgd 		    failed, hunk, rejname);
    311  1.1      cgd 	    }
    312  1.1      cgd 	    else {
    313  1.1      cgd 		say4("%d out of %d hunks failed--saving rejects to %s\n",
    314  1.1      cgd 		    failed, hunk, rejname);
    315  1.1      cgd 	    }
    316  1.1      cgd 	    if (move_file(TMPREJNAME, rejname) < 0)
    317  1.1      cgd 		trejkeep = TRUE;
    318  1.1      cgd 	}
    319  1.1      cgd 	set_signals(1);
    320  1.1      cgd     }
    321  1.1      cgd     my_exit(failtotal);
    322  1.1      cgd }
    323  1.1      cgd 
    324  1.1      cgd /* Prepare to find the next patch to do in the patch file. */
    325  1.1      cgd 
    326  1.1      cgd void
    327  1.1      cgd reinitialize_almost_everything()
    328  1.1      cgd {
    329  1.1      cgd     re_patch();
    330  1.1      cgd     re_input();
    331  1.1      cgd 
    332  1.1      cgd     input_lines = 0;
    333  1.1      cgd     last_frozen_line = 0;
    334  1.1      cgd 
    335  1.1      cgd     filec = 0;
    336  1.1      cgd     if (filearg[0] != Nullch && !out_of_mem) {
    337  1.1      cgd 	free(filearg[0]);
    338  1.1      cgd 	filearg[0] = Nullch;
    339  1.1      cgd     }
    340  1.1      cgd 
    341  1.1      cgd     if (outname != Nullch) {
    342  1.1      cgd 	free(outname);
    343  1.1      cgd 	outname = Nullch;
    344  1.1      cgd     }
    345  1.1      cgd 
    346  1.1      cgd     last_offset = 0;
    347  1.1      cgd 
    348  1.1      cgd     diff_type = 0;
    349  1.1      cgd 
    350  1.1      cgd     if (revision != Nullch) {
    351  1.1      cgd 	free(revision);
    352  1.1      cgd 	revision = Nullch;
    353  1.1      cgd     }
    354  1.1      cgd 
    355  1.1      cgd     reverse = reverse_flag_specified;
    356  1.1      cgd     skip_rest_of_patch = FALSE;
    357  1.1      cgd 
    358  1.1      cgd     get_some_switches();
    359  1.1      cgd 
    360  1.1      cgd     if (filec >= 2)
    361  1.1      cgd 	fatal1("you may not change to a different patch file\n");
    362  1.1      cgd }
    363  1.1      cgd 
    364  1.1      cgd static char *
    365  1.1      cgd nextarg()
    366  1.1      cgd {
    367  1.1      cgd     if (!--Argc)
    368  1.1      cgd 	fatal2("missing argument after `%s'\n", *Argv);
    369  1.1      cgd     return *++Argv;
    370  1.1      cgd }
    371  1.1      cgd 
    372  1.3  thorpej /* Module for handling of long options. */
    373  1.3  thorpej 
    374  1.3  thorpej struct option {
    375  1.3  thorpej     char *long_opt;
    376  1.3  thorpej     char short_opt;
    377  1.3  thorpej };
    378  1.3  thorpej 
    379  1.3  thorpej int
    380  1.3  thorpej optcmp(a, b)
    381  1.3  thorpej     struct option *a, *b;
    382  1.3  thorpej {
    383  1.3  thorpej     return strcmp (a->long_opt, b->long_opt);
    384  1.3  thorpej }
    385  1.3  thorpej 
    386  1.3  thorpej /* Decode Long options beginning with "--" to their short equivalents. */
    387  1.3  thorpej 
    388  1.3  thorpej char
    389  1.3  thorpej decode_long_option(opt)
    390  1.3  thorpej     char *opt;
    391  1.3  thorpej {
    392  1.3  thorpej     /*
    393  1.3  thorpej      * This table must be sorted on the first field.  We also decode
    394  1.3  thorpej      * unimplemented options as those will probably be dealt with
    395  1.3  thorpej      * later, anyhow.
    396  1.3  thorpej      */
    397  1.3  thorpej     static struct option options[] = {
    398  1.3  thorpej       { "batch",		't' },
    399  1.3  thorpej       { "check",		'C' },
    400  1.3  thorpej       { "context",		'c' },
    401  1.3  thorpej       { "debug",		'x' },
    402  1.3  thorpej       { "directory",		'd' },
    403  1.3  thorpej       { "ed",			'e' },
    404  1.3  thorpej       { "force",		'f' },
    405  1.3  thorpej       { "forward",		'N' },
    406  1.3  thorpej       { "fuzz",			'F' },
    407  1.3  thorpej       { "ifdef",		'D' },
    408  1.3  thorpej       { "ignore-whitespace",	'l' },
    409  1.3  thorpej       { "normal",		'n' },
    410  1.3  thorpej       { "output",		'o' },
    411  1.3  thorpej       { "prefix",		'B' },
    412  1.3  thorpej       { "quiet",		's' },
    413  1.3  thorpej       { "reject-file",		'r' },
    414  1.3  thorpej       { "remove-empty-files",	'E' },
    415  1.3  thorpej       { "reverse",		'R' },
    416  1.3  thorpej       { "silent",		's' },
    417  1.3  thorpej       { "skip",			'S' },
    418  1.3  thorpej       { "strip",		'p' },
    419  1.3  thorpej       { "suffix",		'b' },
    420  1.3  thorpej       { "unified",		'u' },
    421  1.3  thorpej       { "version",		'v' },
    422  1.3  thorpej       { "version-control",	'V' },
    423  1.3  thorpej     };
    424  1.3  thorpej     struct option key, *found;
    425  1.3  thorpej 
    426  1.3  thorpej     key.long_opt = opt;
    427  1.3  thorpej     found = (struct option *)bsearch(&key, options,
    428  1.3  thorpej 	sizeof(options) / sizeof(options[0]), sizeof(options[0]), optcmp);
    429  1.3  thorpej 
    430  1.3  thorpej     return found ? found->short_opt : '\0';
    431  1.3  thorpej }
    432  1.3  thorpej 
    433  1.1      cgd /* Process switches and filenames up to next '+' or end of list. */
    434  1.1      cgd 
    435  1.1      cgd void
    436  1.1      cgd get_some_switches()
    437  1.1      cgd {
    438  1.1      cgd     Reg1 char *s;
    439  1.1      cgd 
    440  1.1      cgd     rejname[0] = '\0';
    441  1.1      cgd     Argc_last = Argc;
    442  1.1      cgd     Argv_last = Argv;
    443  1.1      cgd     if (!Argc)
    444  1.1      cgd 	return;
    445  1.1      cgd     for (Argc--,Argv++; Argc; Argc--,Argv++) {
    446  1.1      cgd 	s = Argv[0];
    447  1.1      cgd 	if (strEQ(s, "+")) {
    448  1.1      cgd 	    return;			/* + will be skipped by for loop */
    449  1.1      cgd 	}
    450  1.1      cgd 	if (*s != '-' || !s[1]) {
    451  1.1      cgd 	    if (filec == MAXFILEC)
    452  1.1      cgd 		fatal1("too many file arguments\n");
    453  1.1      cgd 	    filearg[filec++] = savestr(s);
    454  1.1      cgd 	}
    455  1.1      cgd 	else {
    456  1.3  thorpej 	    char opt;
    457  1.3  thorpej 
    458  1.3  thorpej 	    if (*(s + 1) == '-') {
    459  1.3  thorpej 		opt = decode_long_option(s + 2);
    460  1.3  thorpej 		s += strlen(s) - 1;
    461  1.3  thorpej 	    }
    462  1.3  thorpej 	    else
    463  1.3  thorpej 		opt = *++s;
    464  1.3  thorpej 
    465  1.3  thorpej 	    switch (opt) {
    466  1.1      cgd 	    case 'b':
    467  1.1      cgd 		simple_backup_suffix = savestr(nextarg());
    468  1.1      cgd 		break;
    469  1.1      cgd 	    case 'B':
    470  1.1      cgd 		origprae = savestr(nextarg());
    471  1.1      cgd 		break;
    472  1.1      cgd 	    case 'c':
    473  1.1      cgd 		diff_type = CONTEXT_DIFF;
    474  1.1      cgd 		break;
    475  1.1      cgd 	    case 'd':
    476  1.1      cgd 		if (!*++s)
    477  1.1      cgd 		    s = nextarg();
    478  1.1      cgd 		if (chdir(s) < 0)
    479  1.1      cgd 		    pfatal2("can't cd to %s", s);
    480  1.1      cgd 		break;
    481  1.1      cgd 	    case 'D':
    482  1.1      cgd 	    	do_defines = TRUE;
    483  1.1      cgd 		if (!*++s)
    484  1.1      cgd 		    s = nextarg();
    485  1.1      cgd 		if (!isalpha(*s) && '_' != *s)
    486  1.1      cgd 		    fatal1("argument to -D is not an identifier\n");
    487  1.1      cgd 		Sprintf(if_defined, "#ifdef %s\n", s);
    488  1.1      cgd 		Sprintf(not_defined, "#ifndef %s\n", s);
    489  1.1      cgd 		Sprintf(end_defined, "#endif /* %s */\n", s);
    490  1.1      cgd 		break;
    491  1.1      cgd 	    case 'e':
    492  1.1      cgd 		diff_type = ED_DIFF;
    493  1.1      cgd 		break;
    494  1.1      cgd 	    case 'E':
    495  1.1      cgd 		remove_empty_files = TRUE;
    496  1.1      cgd 		break;
    497  1.1      cgd 	    case 'f':
    498  1.1      cgd 		force = TRUE;
    499  1.1      cgd 		break;
    500  1.1      cgd 	    case 'F':
    501  1.1      cgd 		if (*++s == '=')
    502  1.1      cgd 		    s++;
    503  1.1      cgd 		maxfuzz = atoi(s);
    504  1.1      cgd 		break;
    505  1.1      cgd 	    case 'l':
    506  1.1      cgd 		canonicalize = TRUE;
    507  1.1      cgd 		break;
    508  1.1      cgd 	    case 'n':
    509  1.1      cgd 		diff_type = NORMAL_DIFF;
    510  1.1      cgd 		break;
    511  1.1      cgd 	    case 'N':
    512  1.1      cgd 		noreverse = TRUE;
    513  1.1      cgd 		break;
    514  1.1      cgd 	    case 'o':
    515  1.1      cgd 		outname = savestr(nextarg());
    516  1.1      cgd 		break;
    517  1.1      cgd 	    case 'p':
    518  1.1      cgd 		if (*++s == '=')
    519  1.1      cgd 		    s++;
    520  1.1      cgd 		strippath = atoi(s);
    521  1.1      cgd 		break;
    522  1.1      cgd 	    case 'r':
    523  1.1      cgd 		Strcpy(rejname, nextarg());
    524  1.1      cgd 		break;
    525  1.1      cgd 	    case 'R':
    526  1.1      cgd 		reverse = TRUE;
    527  1.1      cgd 		reverse_flag_specified = TRUE;
    528  1.1      cgd 		break;
    529  1.1      cgd 	    case 's':
    530  1.1      cgd 		verbose = FALSE;
    531  1.1      cgd 		break;
    532  1.1      cgd 	    case 'S':
    533  1.1      cgd 		skip_rest_of_patch = TRUE;
    534  1.1      cgd 		break;
    535  1.1      cgd 	    case 't':
    536  1.1      cgd 		batch = TRUE;
    537  1.1      cgd 		break;
    538  1.1      cgd 	    case 'u':
    539  1.1      cgd 		diff_type = UNI_DIFF;
    540  1.1      cgd 		break;
    541  1.1      cgd 	    case 'v':
    542  1.1      cgd 		version();
    543  1.1      cgd 		break;
    544  1.1      cgd 	    case 'V':
    545  1.1      cgd #ifndef NODIR
    546  1.1      cgd 		backup_type = get_version (nextarg ());
    547  1.1      cgd #endif
    548  1.1      cgd 		break;
    549  1.1      cgd #ifdef DEBUGGING
    550  1.1      cgd 	    case 'x':
    551  1.1      cgd 		debug = atoi(s+1);
    552  1.1      cgd 		break;
    553  1.1      cgd #endif
    554  1.1      cgd 	    default:
    555  1.1      cgd 		fprintf(stderr, "patch: unrecognized option `%s'\n", Argv[0]);
    556  1.1      cgd 		fprintf(stderr, "\
    557  1.1      cgd Usage: patch [options] [origfile [patchfile]] [+ [options] [origfile]]...\n\
    558  1.1      cgd Options:\n\
    559  1.1      cgd        [-ceEflnNRsStuv] [-b backup-ext] [-B backup-prefix] [-d directory]\n\
    560  1.1      cgd        [-D symbol] [-Fmax-fuzz] [-o out-file] [-p[strip-count]]\n\
    561  1.1      cgd        [-r rej-name] [-V {numbered,existing,simple}]\n");
    562  1.1      cgd 		my_exit(1);
    563  1.1      cgd 	    }
    564  1.1      cgd 	}
    565  1.1      cgd     }
    566  1.1      cgd }
    567  1.1      cgd 
    568  1.1      cgd /* Attempt to find the right place to apply this hunk of patch. */
    569  1.1      cgd 
    570  1.1      cgd LINENUM
    571  1.1      cgd locate_hunk(fuzz)
    572  1.1      cgd LINENUM fuzz;
    573  1.1      cgd {
    574  1.1      cgd     Reg1 LINENUM first_guess = pch_first() + last_offset;
    575  1.1      cgd     Reg2 LINENUM offset;
    576  1.1      cgd     LINENUM pat_lines = pch_ptrn_lines();
    577  1.1      cgd     Reg3 LINENUM max_pos_offset = input_lines - first_guess
    578  1.1      cgd 				- pat_lines + 1;
    579  1.1      cgd     Reg4 LINENUM max_neg_offset = first_guess - last_frozen_line - 1
    580  1.1      cgd 				+ pch_context();
    581  1.1      cgd 
    582  1.1      cgd     if (!pat_lines)			/* null range matches always */
    583  1.1      cgd 	return first_guess;
    584  1.1      cgd     if (max_neg_offset >= first_guess)	/* do not try lines < 0 */
    585  1.1      cgd 	max_neg_offset = first_guess - 1;
    586  1.1      cgd     if (first_guess <= input_lines && patch_match(first_guess, Nulline, fuzz))
    587  1.1      cgd 	return first_guess;
    588  1.1      cgd     for (offset = 1; ; offset++) {
    589  1.1      cgd 	Reg5 bool check_after = (offset <= max_pos_offset);
    590  1.1      cgd 	Reg6 bool check_before = (offset <= max_neg_offset);
    591  1.1      cgd 
    592  1.1      cgd 	if (check_after && patch_match(first_guess, offset, fuzz)) {
    593  1.1      cgd #ifdef DEBUGGING
    594  1.1      cgd 	    if (debug & 1)
    595  1.1      cgd 		say3("Offset changing from %ld to %ld\n", last_offset, offset);
    596  1.1      cgd #endif
    597  1.1      cgd 	    last_offset = offset;
    598  1.1      cgd 	    return first_guess+offset;
    599  1.1      cgd 	}
    600  1.1      cgd 	else if (check_before && patch_match(first_guess, -offset, fuzz)) {
    601  1.1      cgd #ifdef DEBUGGING
    602  1.1      cgd 	    if (debug & 1)
    603  1.1      cgd 		say3("Offset changing from %ld to %ld\n", last_offset, -offset);
    604  1.1      cgd #endif
    605  1.1      cgd 	    last_offset = -offset;
    606  1.1      cgd 	    return first_guess-offset;
    607  1.1      cgd 	}
    608  1.1      cgd 	else if (!check_before && !check_after)
    609  1.1      cgd 	    return Nulline;
    610  1.1      cgd     }
    611  1.1      cgd }
    612  1.1      cgd 
    613  1.1      cgd /* We did not find the pattern, dump out the hunk so they can handle it. */
    614  1.1      cgd 
    615  1.1      cgd void
    616  1.1      cgd abort_hunk()
    617  1.1      cgd {
    618  1.1      cgd     Reg1 LINENUM i;
    619  1.1      cgd     Reg2 LINENUM pat_end = pch_end();
    620  1.1      cgd     /* add in last_offset to guess the same as the previous successful hunk */
    621  1.1      cgd     LINENUM oldfirst = pch_first() + last_offset;
    622  1.1      cgd     LINENUM newfirst = pch_newfirst() + last_offset;
    623  1.1      cgd     LINENUM oldlast = oldfirst + pch_ptrn_lines() - 1;
    624  1.1      cgd     LINENUM newlast = newfirst + pch_repl_lines() - 1;
    625  1.1      cgd     char *stars = (diff_type >= NEW_CONTEXT_DIFF ? " ****" : "");
    626  1.1      cgd     char *minuses = (diff_type >= NEW_CONTEXT_DIFF ? " ----" : " -----");
    627  1.1      cgd 
    628  1.1      cgd     fprintf(rejfp, "***************\n");
    629  1.1      cgd     for (i=0; i<=pat_end; i++) {
    630  1.1      cgd 	switch (pch_char(i)) {
    631  1.1      cgd 	case '*':
    632  1.1      cgd 	    if (oldlast < oldfirst)
    633  1.1      cgd 		fprintf(rejfp, "*** 0%s\n", stars);
    634  1.1      cgd 	    else if (oldlast == oldfirst)
    635  1.1      cgd 		fprintf(rejfp, "*** %ld%s\n", oldfirst, stars);
    636  1.1      cgd 	    else
    637  1.1      cgd 		fprintf(rejfp, "*** %ld,%ld%s\n", oldfirst, oldlast, stars);
    638  1.1      cgd 	    break;
    639  1.1      cgd 	case '=':
    640  1.1      cgd 	    if (newlast < newfirst)
    641  1.1      cgd 		fprintf(rejfp, "--- 0%s\n", minuses);
    642  1.1      cgd 	    else if (newlast == newfirst)
    643  1.1      cgd 		fprintf(rejfp, "--- %ld%s\n", newfirst, minuses);
    644  1.1      cgd 	    else
    645  1.1      cgd 		fprintf(rejfp, "--- %ld,%ld%s\n", newfirst, newlast, minuses);
    646  1.1      cgd 	    break;
    647  1.1      cgd 	case '\n':
    648  1.1      cgd 	    fprintf(rejfp, "%s", pfetch(i));
    649  1.1      cgd 	    break;
    650  1.1      cgd 	case ' ': case '-': case '+': case '!':
    651  1.1      cgd 	    fprintf(rejfp, "%c %s", pch_char(i), pfetch(i));
    652  1.1      cgd 	    break;
    653  1.1      cgd 	default:
    654  1.1      cgd 	    fatal1("fatal internal error in abort_hunk\n");
    655  1.1      cgd 	}
    656  1.1      cgd     }
    657  1.1      cgd }
    658  1.1      cgd 
    659  1.1      cgd /* We found where to apply it (we hope), so do it. */
    660  1.1      cgd 
    661  1.1      cgd void
    662  1.1      cgd apply_hunk(where)
    663  1.1      cgd LINENUM where;
    664  1.1      cgd {
    665  1.1      cgd     Reg1 LINENUM old = 1;
    666  1.1      cgd     Reg2 LINENUM lastline = pch_ptrn_lines();
    667  1.1      cgd     Reg3 LINENUM new = lastline+1;
    668  1.1      cgd #define OUTSIDE 0
    669  1.1      cgd #define IN_IFNDEF 1
    670  1.1      cgd #define IN_IFDEF 2
    671  1.1      cgd #define IN_ELSE 3
    672  1.1      cgd     Reg4 int def_state = OUTSIDE;
    673  1.1      cgd     Reg5 bool R_do_defines = do_defines;
    674  1.1      cgd     Reg6 LINENUM pat_end = pch_end();
    675  1.1      cgd 
    676  1.1      cgd     where--;
    677  1.1      cgd     while (pch_char(new) == '=' || pch_char(new) == '\n')
    678  1.1      cgd 	new++;
    679  1.1      cgd 
    680  1.1      cgd     while (old <= lastline) {
    681  1.1      cgd 	if (pch_char(old) == '-') {
    682  1.1      cgd 	    copy_till(where + old - 1);
    683  1.1      cgd 	    if (R_do_defines) {
    684  1.1      cgd 		if (def_state == OUTSIDE) {
    685  1.1      cgd 		    fputs(not_defined, ofp);
    686  1.1      cgd 		    def_state = IN_IFNDEF;
    687  1.1      cgd 		}
    688  1.1      cgd 		else if (def_state == IN_IFDEF) {
    689  1.1      cgd 		    fputs(else_defined, ofp);
    690  1.1      cgd 		    def_state = IN_ELSE;
    691  1.1      cgd 		}
    692  1.1      cgd 		fputs(pfetch(old), ofp);
    693  1.1      cgd 	    }
    694  1.1      cgd 	    last_frozen_line++;
    695  1.1      cgd 	    old++;
    696  1.1      cgd 	}
    697  1.1      cgd 	else if (new > pat_end) {
    698  1.1      cgd 	    break;
    699  1.1      cgd 	}
    700  1.1      cgd 	else if (pch_char(new) == '+') {
    701  1.1      cgd 	    copy_till(where + old - 1);
    702  1.1      cgd 	    if (R_do_defines) {
    703  1.1      cgd 		if (def_state == IN_IFNDEF) {
    704  1.1      cgd 		    fputs(else_defined, ofp);
    705  1.1      cgd 		    def_state = IN_ELSE;
    706  1.1      cgd 		}
    707  1.1      cgd 		else if (def_state == OUTSIDE) {
    708  1.1      cgd 		    fputs(if_defined, ofp);
    709  1.1      cgd 		    def_state = IN_IFDEF;
    710  1.1      cgd 		}
    711  1.1      cgd 	    }
    712  1.1      cgd 	    fputs(pfetch(new), ofp);
    713  1.1      cgd 	    new++;
    714  1.1      cgd 	}
    715  1.1      cgd 	else if (pch_char(new) != pch_char(old)) {
    716  1.1      cgd 	    say3("Out-of-sync patch, lines %ld,%ld--mangled text or line numbers, maybe?\n",
    717  1.1      cgd 		pch_hunk_beg() + old,
    718  1.1      cgd 		pch_hunk_beg() + new);
    719  1.1      cgd #ifdef DEBUGGING
    720  1.1      cgd 	    say3("oldchar = '%c', newchar = '%c'\n",
    721  1.1      cgd 		pch_char(old), pch_char(new));
    722  1.1      cgd #endif
    723  1.1      cgd 	    my_exit(1);
    724  1.1      cgd 	}
    725  1.1      cgd 	else if (pch_char(new) == '!') {
    726  1.1      cgd 	    copy_till(where + old - 1);
    727  1.1      cgd 	    if (R_do_defines) {
    728  1.1      cgd 	       fputs(not_defined, ofp);
    729  1.1      cgd 	       def_state = IN_IFNDEF;
    730  1.1      cgd 	    }
    731  1.1      cgd 	    while (pch_char(old) == '!') {
    732  1.1      cgd 		if (R_do_defines) {
    733  1.1      cgd 		    fputs(pfetch(old), ofp);
    734  1.1      cgd 		}
    735  1.1      cgd 		last_frozen_line++;
    736  1.1      cgd 		old++;
    737  1.1      cgd 	    }
    738  1.1      cgd 	    if (R_do_defines) {
    739  1.1      cgd 		fputs(else_defined, ofp);
    740  1.1      cgd 		def_state = IN_ELSE;
    741  1.1      cgd 	    }
    742  1.1      cgd 	    while (pch_char(new) == '!') {
    743  1.1      cgd 		fputs(pfetch(new), ofp);
    744  1.1      cgd 		new++;
    745  1.1      cgd 	    }
    746  1.1      cgd 	}
    747  1.1      cgd 	else {
    748  1.1      cgd 	    assert(pch_char(new) == ' ');
    749  1.1      cgd 	    old++;
    750  1.1      cgd 	    new++;
    751  1.1      cgd 	    if (R_do_defines && def_state != OUTSIDE) {
    752  1.1      cgd 		fputs(end_defined, ofp);
    753  1.1      cgd 		def_state = OUTSIDE;
    754  1.1      cgd 	    }
    755  1.1      cgd 	}
    756  1.1      cgd     }
    757  1.1      cgd     if (new <= pat_end && pch_char(new) == '+') {
    758  1.1      cgd 	copy_till(where + old - 1);
    759  1.1      cgd 	if (R_do_defines) {
    760  1.1      cgd 	    if (def_state == OUTSIDE) {
    761  1.1      cgd 	    	fputs(if_defined, ofp);
    762  1.1      cgd 		def_state = IN_IFDEF;
    763  1.1      cgd 	    }
    764  1.1      cgd 	    else if (def_state == IN_IFNDEF) {
    765  1.1      cgd 		fputs(else_defined, ofp);
    766  1.1      cgd 		def_state = IN_ELSE;
    767  1.1      cgd 	    }
    768  1.1      cgd 	}
    769  1.1      cgd 	while (new <= pat_end && pch_char(new) == '+') {
    770  1.1      cgd 	    fputs(pfetch(new), ofp);
    771  1.1      cgd 	    new++;
    772  1.1      cgd 	}
    773  1.1      cgd     }
    774  1.1      cgd     if (R_do_defines && def_state != OUTSIDE) {
    775  1.1      cgd 	fputs(end_defined, ofp);
    776  1.1      cgd     }
    777  1.1      cgd }
    778  1.1      cgd 
    779  1.1      cgd /* Open the new file. */
    780  1.1      cgd 
    781  1.1      cgd void
    782  1.1      cgd init_output(name)
    783  1.1      cgd char *name;
    784  1.1      cgd {
    785  1.1      cgd     ofp = fopen(name, "w");
    786  1.1      cgd     if (ofp == Nullfp)
    787  1.1      cgd 	pfatal2("can't create %s", name);
    788  1.1      cgd }
    789  1.1      cgd 
    790  1.1      cgd /* Open a file to put hunks we can't locate. */
    791  1.1      cgd 
    792  1.1      cgd void
    793  1.1      cgd init_reject(name)
    794  1.1      cgd char *name;
    795  1.1      cgd {
    796  1.1      cgd     rejfp = fopen(name, "w");
    797  1.1      cgd     if (rejfp == Nullfp)
    798  1.1      cgd 	pfatal2("can't create %s", name);
    799  1.1      cgd }
    800  1.1      cgd 
    801  1.1      cgd /* Copy input file to output, up to wherever hunk is to be applied. */
    802  1.1      cgd 
    803  1.1      cgd void
    804  1.1      cgd copy_till(lastline)
    805  1.1      cgd Reg1 LINENUM lastline;
    806  1.1      cgd {
    807  1.1      cgd     Reg2 LINENUM R_last_frozen_line = last_frozen_line;
    808  1.1      cgd 
    809  1.1      cgd     if (R_last_frozen_line > lastline)
    810  1.1      cgd 	fatal1("misordered hunks! output would be garbled\n");
    811  1.1      cgd     while (R_last_frozen_line < lastline) {
    812  1.1      cgd 	dump_line(++R_last_frozen_line);
    813  1.1      cgd     }
    814  1.1      cgd     last_frozen_line = R_last_frozen_line;
    815  1.1      cgd }
    816  1.1      cgd 
    817  1.1      cgd /* Finish copying the input file to the output file. */
    818  1.1      cgd 
    819  1.1      cgd void
    820  1.1      cgd spew_output()
    821  1.1      cgd {
    822  1.1      cgd #ifdef DEBUGGING
    823  1.1      cgd     if (debug & 256)
    824  1.1      cgd 	say3("il=%ld lfl=%ld\n",input_lines,last_frozen_line);
    825  1.1      cgd #endif
    826  1.1      cgd     if (input_lines)
    827  1.1      cgd 	copy_till(input_lines);		/* dump remainder of file */
    828  1.1      cgd     Fclose(ofp);
    829  1.1      cgd     ofp = Nullfp;
    830  1.1      cgd }
    831  1.1      cgd 
    832  1.1      cgd /* Copy one line from input to output. */
    833  1.1      cgd 
    834  1.1      cgd void
    835  1.1      cgd dump_line(line)
    836  1.1      cgd LINENUM line;
    837  1.1      cgd {
    838  1.1      cgd     Reg1 char *s;
    839  1.1      cgd     Reg2 char R_newline = '\n';
    840  1.1      cgd 
    841  1.1      cgd     /* Note: string is not null terminated. */
    842  1.1      cgd     for (s=ifetch(line, 0); putc(*s, ofp) != R_newline; s++) ;
    843  1.1      cgd }
    844  1.1      cgd 
    845  1.1      cgd /* Does the patch pattern match at line base+offset? */
    846  1.1      cgd 
    847  1.1      cgd bool
    848  1.1      cgd patch_match(base, offset, fuzz)
    849  1.1      cgd LINENUM base;
    850  1.1      cgd LINENUM offset;
    851  1.1      cgd LINENUM fuzz;
    852  1.1      cgd {
    853  1.1      cgd     Reg1 LINENUM pline = 1 + fuzz;
    854  1.1      cgd     Reg2 LINENUM iline;
    855  1.1      cgd     Reg3 LINENUM pat_lines = pch_ptrn_lines() - fuzz;
    856  1.1      cgd 
    857  1.1      cgd     for (iline=base+offset+fuzz; pline <= pat_lines; pline++,iline++) {
    858  1.1      cgd 	if (canonicalize) {
    859  1.1      cgd 	    if (!similar(ifetch(iline, (offset >= 0)),
    860  1.1      cgd 			 pfetch(pline),
    861  1.1      cgd 			 pch_line_len(pline) ))
    862  1.1      cgd 		return FALSE;
    863  1.1      cgd 	}
    864  1.1      cgd 	else if (strnNE(ifetch(iline, (offset >= 0)),
    865  1.1      cgd 		   pfetch(pline),
    866  1.1      cgd 		   pch_line_len(pline) ))
    867  1.1      cgd 	    return FALSE;
    868  1.1      cgd     }
    869  1.1      cgd     return TRUE;
    870  1.1      cgd }
    871  1.1      cgd 
    872  1.1      cgd /* Do two lines match with canonicalized white space? */
    873  1.1      cgd 
    874  1.1      cgd bool
    875  1.1      cgd similar(a,b,len)
    876  1.1      cgd Reg1 char *a;
    877  1.1      cgd Reg2 char *b;
    878  1.1      cgd Reg3 int len;
    879  1.1      cgd {
    880  1.1      cgd     while (len) {
    881  1.1      cgd 	if (isspace(*b)) {		/* whitespace (or \n) to match? */
    882  1.1      cgd 	    if (!isspace(*a))		/* no corresponding whitespace? */
    883  1.1      cgd 		return FALSE;
    884  1.1      cgd 	    while (len && isspace(*b) && *b != '\n')
    885  1.1      cgd 		b++,len--;		/* skip pattern whitespace */
    886  1.1      cgd 	    while (isspace(*a) && *a != '\n')
    887  1.1      cgd 		a++;			/* skip target whitespace */
    888  1.1      cgd 	    if (*a == '\n' || *b == '\n')
    889  1.1      cgd 		return (*a == *b);	/* should end in sync */
    890  1.1      cgd 	}
    891  1.1      cgd 	else if (*a++ != *b++)		/* match non-whitespace chars */
    892  1.1      cgd 	    return FALSE;
    893  1.1      cgd 	else
    894  1.1      cgd 	    len--;			/* probably not necessary */
    895  1.1      cgd     }
    896  1.1      cgd     return TRUE;			/* actually, this is not reached */
    897  1.1      cgd 					/* since there is always a \n */
    898  1.1      cgd }
    899  1.1      cgd 
    900  1.1      cgd /* Exit with cleanup. */
    901  1.1      cgd 
    902  1.1      cgd void
    903  1.1      cgd my_exit(status)
    904  1.1      cgd int status;
    905  1.1      cgd {
    906  1.1      cgd     Unlink(TMPINNAME);
    907  1.1      cgd     if (!toutkeep) {
    908  1.1      cgd 	Unlink(TMPOUTNAME);
    909  1.1      cgd     }
    910  1.1      cgd     if (!trejkeep) {
    911  1.1      cgd 	Unlink(TMPREJNAME);
    912  1.1      cgd     }
    913  1.1      cgd     Unlink(TMPPATNAME);
    914  1.1      cgd     exit(status);
    915  1.1      cgd }
    916