1 5993a8f2 2003-11-23 devnull #include <u.h>
2 5993a8f2 2003-11-23 devnull #include <libc.h>
3 5993a8f2 2003-11-23 devnull #include <bio.h>
4 5993a8f2 2003-11-23 devnull #include "diff.h"
6 5993a8f2 2003-11-23 devnull #define DIRECTORY(s) ((s)->qid.type&QTDIR)
7 5993a8f2 2003-11-23 devnull #define REGULAR_FILE(s) ((s)->type == 'M' && !DIRECTORY(s))
9 5993a8f2 2003-11-23 devnull Biobuf stdout;
11 5993a8f2 2003-11-23 devnull static char *tmp[] = {"/tmp/diff1", "/tmp/diff2"};
12 5993a8f2 2003-11-23 devnull static int whichtmp;
13 5993a8f2 2003-11-23 devnull static char *progname;
14 5993a8f2 2003-11-23 devnull static char usage[] = "diff [ -efmnbwr ] file1 ... file2\n";
16 5993a8f2 2003-11-23 devnull static void
17 5993a8f2 2003-11-23 devnull rmtmpfiles(void)
19 5993a8f2 2003-11-23 devnull while (whichtmp > 0) {
20 5993a8f2 2003-11-23 devnull whichtmp--;
21 5993a8f2 2003-11-23 devnull remove(tmp[whichtmp]);
26 5993a8f2 2003-11-23 devnull done(int status)
28 5993a8f2 2003-11-23 devnull rmtmpfiles();
29 8ad51794 2004-03-25 devnull Bflush(&stdout);
30 8ad51794 2004-03-25 devnull Bterm(&stdout);
31 5993a8f2 2003-11-23 devnull switch(status)
34 5993a8f2 2003-11-23 devnull exits("");
36 5993a8f2 2003-11-23 devnull exits("some");
38 5993a8f2 2003-11-23 devnull exits("error");
40 5993a8f2 2003-11-23 devnull /*NOTREACHED*/
44 5993a8f2 2003-11-23 devnull panic(int status, char *fmt, ...)
46 5993a8f2 2003-11-23 devnull va_list arg;
48 5993a8f2 2003-11-23 devnull Bflush(&stdout);
50 5993a8f2 2003-11-23 devnull fprint(2, "%s: ", progname);
51 5993a8f2 2003-11-23 devnull va_start(arg, fmt);
52 5993a8f2 2003-11-23 devnull vfprint(2, fmt, arg);
53 5993a8f2 2003-11-23 devnull va_end(arg);
54 5993a8f2 2003-11-23 devnull if (status)
55 5993a8f2 2003-11-23 devnull done(status);
56 5993a8f2 2003-11-23 devnull /*NOTREACHED*/
59 5993a8f2 2003-11-23 devnull static int
60 5993a8f2 2003-11-23 devnull catch(void *a, char *msg)
63 5993a8f2 2003-11-23 devnull panic(2, msg);
64 5993a8f2 2003-11-23 devnull return 1;
68 5993a8f2 2003-11-23 devnull mkpathname(char *pathname, char *path, char *name)
70 5993a8f2 2003-11-23 devnull if (strlen(path) + strlen(name) > MAXPATHLEN) {
71 5993a8f2 2003-11-23 devnull panic(0, "pathname %s/%s too long\n", path, name);
72 5993a8f2 2003-11-23 devnull return 1;
74 5993a8f2 2003-11-23 devnull sprint(pathname, "%s/%s", path, name);
75 5993a8f2 2003-11-23 devnull return 0;
78 5993a8f2 2003-11-23 devnull static char *
79 5993a8f2 2003-11-23 devnull mktmpfile(int input, Dir **sb)
81 5993a8f2 2003-11-23 devnull int fd, i;
83 5993a8f2 2003-11-23 devnull char buf[8192];
85 5993a8f2 2003-11-23 devnull atnotify(catch, 1);
86 5993a8f2 2003-11-23 devnull p = tmp[whichtmp++];
87 5993a8f2 2003-11-23 devnull fd = create(p, OWRITE, 0600);
88 5993a8f2 2003-11-23 devnull if (fd < 0) {
89 5993a8f2 2003-11-23 devnull panic(mflag ? 0: 2, "cannot create %s: %r\n", p);
90 5993a8f2 2003-11-23 devnull return 0;
92 5993a8f2 2003-11-23 devnull while ((i = read(input, buf, sizeof(buf))) > 0) {
93 5993a8f2 2003-11-23 devnull if ((i = write(fd, buf, i)) < 0)
96 5993a8f2 2003-11-23 devnull *sb = dirfstat(fd);
97 5993a8f2 2003-11-23 devnull close(fd);
98 5993a8f2 2003-11-23 devnull if (i < 0) {
99 5993a8f2 2003-11-23 devnull panic(mflag ? 0: 2, "cannot read/write %s: %r\n", p);
100 5993a8f2 2003-11-23 devnull return 0;
102 5993a8f2 2003-11-23 devnull return p;
105 5993a8f2 2003-11-23 devnull static char *
106 5993a8f2 2003-11-23 devnull statfile(char *file, Dir **sb)
108 5993a8f2 2003-11-23 devnull Dir *dir;
109 5993a8f2 2003-11-23 devnull int input;
111 5993a8f2 2003-11-23 devnull dir = dirstat(file);
112 5993a8f2 2003-11-23 devnull if(dir == nil) {
113 5993a8f2 2003-11-23 devnull if (strcmp(file, "-") || (dir = dirfstat(0)) == nil) {
114 5993a8f2 2003-11-23 devnull panic(mflag ? 0: 2, "cannot stat %s: %r\n", file);
115 5993a8f2 2003-11-23 devnull return 0;
117 5993a8f2 2003-11-23 devnull free(dir);
118 5993a8f2 2003-11-23 devnull return mktmpfile(0, sb);
120 5993a8f2 2003-11-23 devnull else if (!REGULAR_FILE(dir) && !DIRECTORY(dir)) {
121 5993a8f2 2003-11-23 devnull free(dir);
122 5993a8f2 2003-11-23 devnull if ((input = open(file, OREAD)) == -1) {
123 5993a8f2 2003-11-23 devnull panic(mflag ? 0: 2, "cannot open %s: %r\n", file);
124 5993a8f2 2003-11-23 devnull return 0;
126 5993a8f2 2003-11-23 devnull file = mktmpfile(input, sb);
127 5993a8f2 2003-11-23 devnull close(input);
130 5993a8f2 2003-11-23 devnull *sb = dir;
131 5993a8f2 2003-11-23 devnull return file;
135 5993a8f2 2003-11-23 devnull diff(char *f, char *t, int level)
137 5993a8f2 2003-11-23 devnull char *fp, *tp, *p, fb[MAXPATHLEN+1], tb[MAXPATHLEN+1];
138 5993a8f2 2003-11-23 devnull Dir *fsb, *tsb;
140 5993a8f2 2003-11-23 devnull if ((fp = statfile(f, &fsb)) == 0)
142 5993a8f2 2003-11-23 devnull if ((tp = statfile(t, &tsb)) == 0){
143 5993a8f2 2003-11-23 devnull free(fsb);
146 5993a8f2 2003-11-23 devnull if (DIRECTORY(fsb) && DIRECTORY(tsb)) {
147 5993a8f2 2003-11-23 devnull if (rflag || level == 0)
148 5993a8f2 2003-11-23 devnull diffdir(fp, tp, level);
150 5993a8f2 2003-11-23 devnull Bprint(&stdout, "Common subdirectories: %s and %s\n",
151 5993a8f2 2003-11-23 devnull fp, tp);
153 5993a8f2 2003-11-23 devnull else if (REGULAR_FILE(fsb) && REGULAR_FILE(tsb))
154 5993a8f2 2003-11-23 devnull diffreg(fp, tp);
156 5993a8f2 2003-11-23 devnull if (REGULAR_FILE(fsb)) {
157 5993a8f2 2003-11-23 devnull if ((p = utfrrune(f, '/')) == 0)
161 5993a8f2 2003-11-23 devnull if (mkpathname(tb, tp, p) == 0)
162 5993a8f2 2003-11-23 devnull diffreg(fp, tb);
165 5993a8f2 2003-11-23 devnull if ((p = utfrrune(t, '/')) == 0)
169 5993a8f2 2003-11-23 devnull if (mkpathname(fb, fp, p) == 0)
170 5993a8f2 2003-11-23 devnull diffreg(fb, tp);
173 5993a8f2 2003-11-23 devnull free(fsb);
174 5993a8f2 2003-11-23 devnull free(tsb);
178 5993a8f2 2003-11-23 devnull main(int argc, char *argv[])
180 5993a8f2 2003-11-23 devnull char *p;
182 5993a8f2 2003-11-23 devnull Dir *fsb, *tsb;
184 5993a8f2 2003-11-23 devnull Binit(&stdout, 1, OWRITE);
185 5993a8f2 2003-11-23 devnull progname = *argv;
186 5993a8f2 2003-11-23 devnull while (--argc && (*++argv)[0] == '-' && (*argv)[1]) {
187 5993a8f2 2003-11-23 devnull for (p = *argv+1; *p; p++) {
188 5993a8f2 2003-11-23 devnull switch (*p) {
190 5993a8f2 2003-11-23 devnull case 'e':
191 5993a8f2 2003-11-23 devnull case 'f':
192 5993a8f2 2003-11-23 devnull case 'n':
193 5993a8f2 2003-11-23 devnull mode = *p;
196 5993a8f2 2003-11-23 devnull case 'w':
197 5993a8f2 2003-11-23 devnull bflag = 2;
200 5993a8f2 2003-11-23 devnull case 'b':
201 5993a8f2 2003-11-23 devnull bflag = 1;
204 5993a8f2 2003-11-23 devnull case 'r':
205 5993a8f2 2003-11-23 devnull rflag = 1;
208 5993a8f2 2003-11-23 devnull case 'm':
209 5993a8f2 2003-11-23 devnull mflag = 1;
212 5993a8f2 2003-11-23 devnull case 'h':
213 5993a8f2 2003-11-23 devnull default:
214 5993a8f2 2003-11-23 devnull progname = "Usage";
215 5993a8f2 2003-11-23 devnull panic(2, usage);
219 5993a8f2 2003-11-23 devnull if (argc < 2)
220 5993a8f2 2003-11-23 devnull panic(2, usage, progname);
221 5993a8f2 2003-11-23 devnull if ((tsb = dirstat(argv[argc-1])) == nil)
222 5993a8f2 2003-11-23 devnull panic(2, "can't stat %s\n", argv[argc-1]);
223 5993a8f2 2003-11-23 devnull if (argc > 2) {
224 5993a8f2 2003-11-23 devnull if (!DIRECTORY(tsb))
225 5993a8f2 2003-11-23 devnull panic(2, usage, progname);
226 5993a8f2 2003-11-23 devnull mflag = 1;
229 5993a8f2 2003-11-23 devnull if ((fsb = dirstat(argv[0])) == nil)
230 5993a8f2 2003-11-23 devnull panic(2, "can't stat %s\n", argv[0]);
231 5993a8f2 2003-11-23 devnull if (DIRECTORY(fsb) && DIRECTORY(tsb))
232 5993a8f2 2003-11-23 devnull mflag = 1;
233 5993a8f2 2003-11-23 devnull free(fsb);
235 5993a8f2 2003-11-23 devnull free(tsb);
236 5993a8f2 2003-11-23 devnull for (i = 0; i < argc-1; i++) {
237 5993a8f2 2003-11-23 devnull diff(argv[i], argv[argc-1], 0);
238 5993a8f2 2003-11-23 devnull rmtmpfiles();
240 5993a8f2 2003-11-23 devnull done(anychange);
241 5993a8f2 2003-11-23 devnull /*NOTREACHED*/
244 5993a8f2 2003-11-23 devnull static char noroom[] = "out of memory - try diff -h\n";
247 5993a8f2 2003-11-23 devnull emalloc(unsigned n)
249 5993a8f2 2003-11-23 devnull register void *p;
251 5993a8f2 2003-11-23 devnull if ((p = malloc(n)) == 0)
252 5993a8f2 2003-11-23 devnull panic(2, noroom);
253 5993a8f2 2003-11-23 devnull return p;
257 5993a8f2 2003-11-23 devnull erealloc(void *p, unsigned n)
259 5993a8f2 2003-11-23 devnull register void *rp;
261 5993a8f2 2003-11-23 devnull if ((rp = realloc(p, n)) == 0)
262 5993a8f2 2003-11-23 devnull panic(2, noroom);
263 5993a8f2 2003-11-23 devnull return rp;