1 #include <u.h> 2 #include <libc.h> 3 #include <bio.h> 4 #include "diff.h" 5 6 #define DIRECTORY(s) ((s)->qid.type&QTDIR) 7 #define REGULAR_FILE(s) ((s)->type == 'M' && !DIRECTORY(s)) 8 9 Biobuf stdout; 10 11 static char *tmp[] = {"/tmp/diff1", "/tmp/diff2"}; 12 static int whichtmp; 13 static char *progname; 14 static char usage[] = "diff [ -efmnbwr ] file1 ... file2\n"; 15 16 static void 17 rmtmpfiles(void) 18 { 19 while (whichtmp > 0) { 20 whichtmp--; 21 remove(tmp[whichtmp]); 22 } 23 } 24 25 void 26 done(int status) 27 { 28 rmtmpfiles(); 29 switch(status) 30 { 31 case 0: 32 exits(""); 33 case 1: 34 exits("some"); 35 default: 36 exits("error"); 37 } 38 /*NOTREACHED*/ 39 } 40 41 void 42 panic(int status, char *fmt, ...) 43 { 44 va_list arg; 45 46 Bflush(&stdout); 47 48 fprint(2, "%s: ", progname); 49 va_start(arg, fmt); 50 vfprint(2, fmt, arg); 51 va_end(arg); 52 if (status) 53 done(status); 54 /*NOTREACHED*/ 55 } 56 57 static int 58 catch(void *a, char *msg) 59 { 60 USED(a); 61 panic(2, msg); 62 return 1; 63 } 64 65 int 66 mkpathname(char *pathname, char *path, char *name) 67 { 68 if (strlen(path) + strlen(name) > MAXPATHLEN) { 69 panic(0, "pathname %s/%s too long\n", path, name); 70 return 1; 71 } 72 sprint(pathname, "%s/%s", path, name); 73 return 0; 74 } 75 76 static char * 77 mktmpfile(int input, Dir **sb) 78 { 79 int fd, i; 80 char *p; 81 char buf[8192]; 82 83 atnotify(catch, 1); 84 p = tmp[whichtmp++]; 85 fd = create(p, OWRITE, 0600); 86 if (fd < 0) { 87 panic(mflag ? 0: 2, "cannot create %s: %r\n", p); 88 return 0; 89 } 90 while ((i = read(input, buf, sizeof(buf))) > 0) { 91 if ((i = write(fd, buf, i)) < 0) 92 break; 93 } 94 *sb = dirfstat(fd); 95 close(fd); 96 if (i < 0) { 97 panic(mflag ? 0: 2, "cannot read/write %s: %r\n", p); 98 return 0; 99 } 100 return p; 101 } 102 103 static char * 104 statfile(char *file, Dir **sb) 105 { 106 Dir *dir; 107 int input; 108 109 dir = dirstat(file); 110 if(dir == nil) { 111 if (strcmp(file, "-") || (dir = dirfstat(0)) == nil) { 112 panic(mflag ? 0: 2, "cannot stat %s: %r\n", file); 113 return 0; 114 } 115 free(dir); 116 return mktmpfile(0, sb); 117 } 118 else if (!REGULAR_FILE(dir) && !DIRECTORY(dir)) { 119 free(dir); 120 if ((input = open(file, OREAD)) == -1) { 121 panic(mflag ? 0: 2, "cannot open %s: %r\n", file); 122 return 0; 123 } 124 file = mktmpfile(input, sb); 125 close(input); 126 } 127 else 128 *sb = dir; 129 return file; 130 } 131 132 void 133 diff(char *f, char *t, int level) 134 { 135 char *fp, *tp, *p, fb[MAXPATHLEN+1], tb[MAXPATHLEN+1]; 136 Dir *fsb, *tsb; 137 138 if ((fp = statfile(f, &fsb)) == 0) 139 return; 140 if ((tp = statfile(t, &tsb)) == 0){ 141 free(fsb); 142 return; 143 } 144 if (DIRECTORY(fsb) && DIRECTORY(tsb)) { 145 if (rflag || level == 0) 146 diffdir(fp, tp, level); 147 else 148 Bprint(&stdout, "Common subdirectories: %s and %s\n", 149 fp, tp); 150 } 151 else if (REGULAR_FILE(fsb) && REGULAR_FILE(tsb)) 152 diffreg(fp, tp); 153 else { 154 if (REGULAR_FILE(fsb)) { 155 if ((p = utfrrune(f, '/')) == 0) 156 p = f; 157 else 158 p++; 159 if (mkpathname(tb, tp, p) == 0) 160 diffreg(fp, tb); 161 } 162 else { 163 if ((p = utfrrune(t, '/')) == 0) 164 p = t; 165 else 166 p++; 167 if (mkpathname(fb, fp, p) == 0) 168 diffreg(fb, tp); 169 } 170 } 171 free(fsb); 172 free(tsb); 173 } 174 175 void 176 main(int argc, char *argv[]) 177 { 178 char *p; 179 int i; 180 Dir *fsb, *tsb; 181 182 Binit(&stdout, 1, OWRITE); 183 progname = *argv; 184 while (--argc && (*++argv)[0] == '-' && (*argv)[1]) { 185 for (p = *argv+1; *p; p++) { 186 switch (*p) { 187 188 case 'e': 189 case 'f': 190 case 'n': 191 mode = *p; 192 break; 193 194 case 'w': 195 bflag = 2; 196 break; 197 198 case 'b': 199 bflag = 1; 200 break; 201 202 case 'r': 203 rflag = 1; 204 break; 205 206 case 'm': 207 mflag = 1; 208 break; 209 210 case 'h': 211 default: 212 progname = "Usage"; 213 panic(2, usage); 214 } 215 } 216 } 217 if (argc < 2) 218 panic(2, usage, progname); 219 if ((tsb = dirstat(argv[argc-1])) == nil) 220 panic(2, "can't stat %s\n", argv[argc-1]); 221 if (argc > 2) { 222 if (!DIRECTORY(tsb)) 223 panic(2, usage, progname); 224 mflag = 1; 225 } 226 else { 227 if ((fsb = dirstat(argv[0])) == nil) 228 panic(2, "can't stat %s\n", argv[0]); 229 if (DIRECTORY(fsb) && DIRECTORY(tsb)) 230 mflag = 1; 231 free(fsb); 232 } 233 free(tsb); 234 for (i = 0; i < argc-1; i++) { 235 diff(argv[i], argv[argc-1], 0); 236 rmtmpfiles(); 237 } 238 done(anychange); 239 /*NOTREACHED*/ 240 } 241 242 static char noroom[] = "out of memory - try diff -h\n"; 243 244 void * 245 emalloc(unsigned n) 246 { 247 register void *p; 248 249 if ((p = malloc(n)) == 0) 250 panic(2, noroom); 251 return p; 252 } 253 254 void * 255 erealloc(void *p, unsigned n) 256 { 257 register void *rp; 258 259 if ((rp = realloc(p, n)) == 0) 260 panic(2, noroom); 261 return rp; 262 } 263