Annotation of src/usr.bin/rcs/rcsutil.c, Revision 1.38
1.38 ! chl 1: /* $OpenBSD: rcsutil.c,v 1.37 2010/10/20 19:53:53 tobias Exp $ */
1.1 xsa 2: /*
3: * Copyright (c) 2005, 2006 Joris Vink <joris@openbsd.org>
4: * Copyright (c) 2006 Xavier Santolaria <xsa@openbsd.org>
5: * Copyright (c) 2006 Niall O'Higgins <niallo@openbsd.org>
6: * Copyright (c) 2006 Ray Lai <ray@openbsd.org>
7: * All rights reserved.
8: *
9: * Redistribution and use in source and binary forms, with or without
10: * modification, are permitted provided that the following conditions
11: * are met:
12: *
13: * 1. Redistributions of source code must retain the above copyright
14: * notice, this list of conditions and the following disclaimer.
15: * 2. The name of the author may not be used to endorse or promote products
16: * derived from this software without specific prior written permission.
17: *
18: * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
19: * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
20: * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
21: * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
22: * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
23: * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
24: * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
25: * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26: * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
27: * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28: */
29:
1.29 xsa 30: #include <sys/stat.h>
1.38 ! chl 31: #include <sys/time.h>
1.29 xsa 32:
33: #include <ctype.h>
34: #include <err.h>
35: #include <fcntl.h>
36: #include <stdio.h>
37: #include <string.h>
38: #include <unistd.h>
1.1 xsa 39:
40: #include "rcsprog.h"
41:
42: /*
43: * rcs_get_mtime()
44: *
45: * Get <filename> last modified time.
46: * Returns last modified time on success, or -1 on failure.
47: */
48: time_t
1.6 joris 49: rcs_get_mtime(RCSFILE *file)
1.1 xsa 50: {
51: struct stat st;
52: time_t mtime;
53:
1.37 tobias 54: if (file->rf_file == NULL)
55: return (-1);
56:
57: if (fstat(fileno(file->rf_file), &st) == -1) {
1.6 joris 58: warn("%s", file->rf_path);
1.1 xsa 59: return (-1);
60: }
1.6 joris 61:
1.1 xsa 62: mtime = (time_t)st.st_mtimespec.tv_sec;
63:
64: return (mtime);
65: }
66:
67: /*
68: * rcs_set_mtime()
69: *
70: * Set <filename> last modified time to <mtime> if it's not set to -1.
71: */
72: void
1.6 joris 73: rcs_set_mtime(RCSFILE *file, time_t mtime)
1.1 xsa 74: {
75: static struct timeval tv[2];
76:
1.37 tobias 77: if (file->rf_file == NULL || mtime == -1)
1.1 xsa 78: return;
79:
80: tv[0].tv_sec = mtime;
81: tv[1].tv_sec = tv[0].tv_sec;
82:
1.37 tobias 83: if (futimes(fileno(file->rf_file), tv) == -1)
1.3 xsa 84: err(1, "utimes");
1.1 xsa 85: }
86:
87: int
88: rcs_getopt(int argc, char **argv, const char *optstr)
89: {
90: char *a;
91: const char *c;
92: static int i = 1;
93: int opt, hasargument, ret;
94:
95: hasargument = 0;
96: rcs_optarg = NULL;
97:
98: if (i >= argc)
99: return (-1);
100:
101: a = argv[i++];
102: if (*a++ != '-')
103: return (-1);
104:
105: ret = 0;
106: opt = *a;
107: for (c = optstr; *c != '\0'; c++) {
108: if (*c == opt) {
109: a++;
110: ret = opt;
111:
112: if (*(c + 1) == ':') {
113: if (*(c + 2) == ':') {
114: if (*a != '\0')
115: hasargument = 1;
116: } else {
117: if (*a != '\0') {
118: hasargument = 1;
119: } else {
120: ret = 1;
121: break;
122: }
123: }
124: }
125:
126: if (hasargument == 1)
127: rcs_optarg = a;
128:
129: if (ret == opt)
130: rcs_optind++;
131: break;
132: }
133: }
134:
135: if (ret == 0)
136: warnx("unknown option -%c", opt);
137: else if (ret == 1)
138: warnx("missing argument for option -%c", opt);
139:
140: return (ret);
141: }
142:
143: /*
144: * rcs_choosefile()
145: *
146: * Given a relative filename, decide where the corresponding RCS file
147: * should be. Tries each extension until a file is found. If no file
148: * was found, returns a path with the first extension.
149: *
1.11 ray 150: * Opens and returns file descriptor to RCS file.
1.1 xsa 151: */
1.6 joris 152: int
153: rcs_choosefile(const char *filename, char *out, size_t len)
1.1 xsa 154: {
1.6 joris 155: int fd;
1.1 xsa 156: struct stat sb;
157: char *p, *ext, name[MAXPATHLEN], *next, *ptr, rcsdir[MAXPATHLEN],
1.10 ray 158: *suffixes, rcspath[MAXPATHLEN];
1.1 xsa 159:
1.6 joris 160: fd = -1;
161:
1.1 xsa 162: /*
163: * If `filename' contains a directory, `rcspath' contains that
164: * directory, including a trailing slash. Otherwise `rcspath'
165: * contains an empty string.
166: */
167: if (strlcpy(rcspath, filename, sizeof(rcspath)) >= sizeof(rcspath))
1.6 joris 168: errx(1, "rcs_choosefile: truncation");
169:
1.1 xsa 170: /* If `/' is found, end string after `/'. */
171: if ((ptr = strrchr(rcspath, '/')) != NULL)
172: *(++ptr) = '\0';
173: else
174: rcspath[0] = '\0';
175:
176: /* Append RCS/ to `rcspath' if it exists. */
177: if (strlcpy(rcsdir, rcspath, sizeof(rcsdir)) >= sizeof(rcsdir) ||
178: strlcat(rcsdir, RCSDIR, sizeof(rcsdir)) >= sizeof(rcsdir))
1.6 joris 179: errx(1, "rcs_choosefile: truncation");
180:
1.19 otto 181: if (stat(rcsdir, &sb) == 0 && S_ISDIR(sb.st_mode))
1.6 joris 182: if (strlcpy(rcspath, rcsdir, sizeof(rcspath))
183: >= sizeof(rcspath) ||
1.1 xsa 184: strlcat(rcspath, "/", sizeof(rcspath)) >= sizeof(rcspath))
1.6 joris 185: errx(1, "rcs_choosefile: truncation");
1.1 xsa 186:
187: /* Name of file without path. */
188: if ((ptr = strrchr(filename, '/')) == NULL) {
189: if (strlcpy(name, filename, sizeof(name)) >= sizeof(name))
1.6 joris 190: errx(1, "rcs_choosefile: truncation");
1.1 xsa 191: } else {
192: /* Skip `/'. */
193: if (strlcpy(name, ptr + 1, sizeof(name)) >= sizeof(name))
1.6 joris 194: errx(1, "rcs_choosefile: truncation");
1.1 xsa 195: }
196:
197: /* Name of RCS file without an extension. */
198: if (strlcat(rcspath, name, sizeof(rcspath)) >= sizeof(rcspath))
1.6 joris 199: errx(1, "rcs_choosefile: truncation");
1.1 xsa 200:
201: /*
202: * If only the empty suffix was given, use existing rcspath.
203: * This ensures that there is at least one suffix for strsep().
204: */
205: if (strcmp(rcs_suffixes, "") == 0) {
1.8 ray 206: if (strlcpy(out, rcspath, len) >= len)
207: errx(1, "rcs_choosefile: truncation");
1.22 niallo 208: fd = open(rcspath, O_RDONLY);
1.6 joris 209: return (fd);
1.1 xsa 210: }
211:
212: /*
213: * Cycle through slash-separated `rcs_suffixes', appending each
214: * extension to `rcspath' and testing if the file exists. If it
215: * does, return that string. Otherwise return path with first
216: * extension.
217: */
218: suffixes = xstrdup(rcs_suffixes);
1.10 ray 219: for (next = suffixes; (ext = strsep(&next, "/")) != NULL;) {
1.1 xsa 220: char fpath[MAXPATHLEN];
221:
222: if ((p = strrchr(rcspath, ',')) != NULL) {
223: if (!strcmp(p, ext)) {
1.6 joris 224: if ((fd = open(rcspath, O_RDONLY)) == -1)
225: continue;
226:
227: if (fstat(fd, &sb) == -1)
228: err(1, "%s", rcspath);
229:
230: if (strlcpy(out, rcspath, len) >= len)
231: errx(1, "rcs_choosefile; truncation");
232:
1.33 joris 233: xfree(suffixes);
1.6 joris 234: return (fd);
1.1 xsa 235: }
236:
237: continue;
238: }
239:
240: /* Construct RCS file path. */
241: if (strlcpy(fpath, rcspath, sizeof(fpath)) >= sizeof(fpath) ||
242: strlcat(fpath, ext, sizeof(fpath)) >= sizeof(fpath))
1.6 joris 243: errx(1, "rcs_choosefile: truncation");
1.1 xsa 244:
245: /* Don't use `filename' as RCS file. */
246: if (strcmp(fpath, filename) == 0)
247: continue;
248:
1.6 joris 249: if ((fd = open(fpath, O_RDONLY)) == -1)
250: continue;
251:
252: if (fstat(fd, &sb) == -1)
253: err(1, "%s", fpath);
254:
255: if (strlcpy(out, fpath, len) >= len)
256: errx(1, "rcs_choosefile: truncation");
257:
1.33 joris 258: xfree(suffixes);
1.6 joris 259: return (fd);
1.1 xsa 260: }
261:
262: /*
263: * `suffixes' should now be NUL separated, so the first
264: * extension can be read just by reading `suffixes'.
265: */
1.11 ray 266: if (strlcat(rcspath, suffixes, sizeof(rcspath)) >= sizeof(rcspath))
1.6 joris 267: errx(1, "rcs_choosefile: truncation");
1.1 xsa 268:
269: xfree(suffixes);
1.6 joris 270:
1.8 ray 271: if (strlcpy(out, rcspath, len) >= len)
272: errx(1, "rcs_choosefile: truncation");
1.22 niallo 273:
274: fd = open(rcspath, O_RDONLY);
1.1 xsa 275:
1.6 joris 276: return (fd);
1.1 xsa 277: }
278:
279: /*
280: * Set <str> to <new_str>. Print warning if <str> is redefined.
281: */
282: void
283: rcs_setrevstr(char **str, char *new_str)
284: {
285: if (new_str == NULL)
286: return;
287: if (*str != NULL)
288: warnx("redefinition of revision number");
289: *str = new_str;
290: }
291:
292: /*
293: * Set <str1> or <str2> to <new_str>, depending on which is not set.
294: * If both are set, error out.
295: */
296: void
297: rcs_setrevstr2(char **str1, char **str2, char *new_str)
298: {
299: if (new_str == NULL)
300: return;
301: if (*str1 == NULL)
302: *str1 = new_str;
303: else if (*str2 == NULL)
304: *str2 = new_str;
305: else
1.3 xsa 306: errx(1, "too many revision numbers");
1.1 xsa 307: }
308:
309: /*
310: * Get revision from file. The revision can be specified as a symbol or
311: * a revision number.
312: */
313: RCSNUM *
314: rcs_getrevnum(const char *rev_str, RCSFILE *file)
315: {
316: RCSNUM *rev;
317:
318: /* Search for symbol. */
319: rev = rcs_sym_getrev(file, rev_str);
320:
321: /* Search for revision number. */
322: if (rev == NULL)
323: rev = rcsnum_parse(rev_str);
324:
325: return (rev);
326: }
327:
328: /*
329: * Prompt for and store user's input in an allocated string.
330: *
331: * Returns the string's pointer.
332: */
333: char *
334: rcs_prompt(const char *prompt)
335: {
336: BUF *bp;
337: size_t len;
338: char *buf;
339:
1.35 ray 340: bp = buf_alloc(0);
1.1 xsa 341: if (isatty(STDIN_FILENO))
342: (void)fprintf(stderr, "%s", prompt);
343: if (isatty(STDIN_FILENO))
344: (void)fprintf(stderr, ">> ");
1.25 ray 345: clearerr(stdin);
1.1 xsa 346: while ((buf = fgetln(stdin, &len)) != NULL) {
347: /* The last line may not be EOL terminated. */
348: if (buf[0] == '.' && (len == 1 || buf[1] == '\n'))
349: break;
350: else
1.34 ray 351: buf_append(bp, buf, len);
1.1 xsa 352:
353: if (isatty(STDIN_FILENO))
354: (void)fprintf(stderr, ">> ");
355: }
1.34 ray 356: buf_putc(bp, '\0');
1.1 xsa 357:
1.34 ray 358: return (buf_release(bp));
1.1 xsa 359: }
360:
361: u_int
1.14 ray 362: rcs_rev_select(RCSFILE *file, const char *range)
1.1 xsa 363: {
364: int i;
365: u_int nrev;
366: char *ep;
367: char *lstr, *rstr;
368: struct rcs_delta *rdp;
1.5 joris 369: struct rcs_argvector *revargv, *revrange;
1.1 xsa 370: RCSNUM lnum, rnum;
371:
372: nrev = 0;
373: (void)memset(&lnum, 0, sizeof(lnum));
374: (void)memset(&rnum, 0, sizeof(rnum));
375:
376: if (range == NULL) {
377: TAILQ_FOREACH(rdp, &file->rf_delta, rd_list)
378: if (rcsnum_cmp(rdp->rd_num, file->rf_head, 0) == 0) {
379: rdp->rd_flags |= RCS_RD_SELECT;
380: return (1);
381: }
382: return (0);
383: }
384:
1.5 joris 385: revargv = rcs_strsplit(range, ",");
1.1 xsa 386: for (i = 0; revargv->argv[i] != NULL; i++) {
1.5 joris 387: revrange = rcs_strsplit(revargv->argv[i], ":");
1.1 xsa 388: if (revrange->argv[0] == NULL)
389: /* should not happen */
1.3 xsa 390: errx(1, "invalid revision range: %s", revargv->argv[i]);
1.1 xsa 391: else if (revrange->argv[1] == NULL)
392: lstr = rstr = revrange->argv[0];
393: else {
394: if (revrange->argv[2] != NULL)
1.3 xsa 395: errx(1, "invalid revision range: %s",
396: revargv->argv[i]);
1.1 xsa 397: lstr = revrange->argv[0];
398: rstr = revrange->argv[1];
399: if (strcmp(lstr, "") == 0)
400: lstr = NULL;
401: if (strcmp(rstr, "") == 0)
402: rstr = NULL;
403: }
404:
405: if (lstr == NULL)
406: lstr = RCS_HEAD_INIT;
407: if (rcsnum_aton(lstr, &ep, &lnum) == 0 || (*ep != '\0'))
1.3 xsa 408: errx(1, "invalid revision: %s", lstr);
1.1 xsa 409:
410: if (rstr != NULL) {
411: if (rcsnum_aton(rstr, &ep, &rnum) == 0 || (*ep != '\0'))
1.3 xsa 412: errx(1, "invalid revision: %s", rstr);
1.1 xsa 413: } else
414: rcsnum_cpy(file->rf_head, &rnum, 0);
415:
1.5 joris 416: rcs_argv_destroy(revrange);
1.1 xsa 417:
418: TAILQ_FOREACH(rdp, &file->rf_delta, rd_list)
419: if (rcsnum_cmp(rdp->rd_num, &lnum, 0) <= 0 &&
420: rcsnum_cmp(rdp->rd_num, &rnum, 0) >= 0 &&
421: !(rdp->rd_flags & RCS_RD_SELECT)) {
422: rdp->rd_flags |= RCS_RD_SELECT;
423: nrev++;
424: }
425: }
1.5 joris 426: rcs_argv_destroy(revargv);
1.1 xsa 427:
428: if (lnum.rn_id != NULL)
429: xfree(lnum.rn_id);
430: if (rnum.rn_id != NULL)
431: xfree(rnum.rn_id);
432:
433: return (nrev);
1.2 ray 434: }
435:
436: /*
437: * Load description from <in> to <file>.
438: * If <in> starts with a `-', <in> is taken as the description.
439: * Otherwise <in> is the name of the file containing the description.
440: * If <in> is NULL, the description is read from stdin.
1.18 ray 441: * Returns 0 on success, -1 on failure, setting errno.
1.2 ray 442: */
1.18 ray 443: int
1.2 ray 444: rcs_set_description(RCSFILE *file, const char *in)
445: {
446: BUF *bp;
447: char *content;
448: const char *prompt =
449: "enter description, terminated with single '.' or end of file:\n"
450: "NOTE: This is NOT the log message!\n";
451:
452: /* Description is in file <in>. */
453: if (in != NULL && *in != '-') {
1.35 ray 454: if ((bp = buf_load(in)) == NULL)
1.18 ray 455: return (-1);
1.34 ray 456: buf_putc(bp, '\0');
457: content = buf_release(bp);
1.2 ray 458: /* Description is in <in>. */
459: } else if (in != NULL)
460: /* Skip leading `-'. */
461: content = xstrdup(in + 1);
462: /* Get description from stdin. */
463: else
464: content = rcs_prompt(prompt);
465:
466: rcs_desc_set(file, content);
467: xfree(content);
1.18 ray 468: return (0);
1.7 xsa 469: }
470:
471: /*
472: * Split the contents of a file into a list of lines.
473: */
474: struct rcs_lines *
1.27 xsa 475: rcs_splitlines(u_char *data, size_t len)
1.7 xsa 476: {
1.15 niallo 477: u_char *c, *p;
1.7 xsa 478: struct rcs_lines *lines;
479: struct rcs_line *lp;
1.24 niallo 480: size_t i, tlen;
1.7 xsa 481:
482: lines = xmalloc(sizeof(*lines));
1.24 niallo 483: memset(lines, 0, sizeof(*lines));
1.7 xsa 484: TAILQ_INIT(&(lines->l_lines));
485:
486: lp = xmalloc(sizeof(*lp));
1.24 niallo 487: memset(lp, 0, sizeof(*lp));
1.7 xsa 488: TAILQ_INSERT_TAIL(&(lines->l_lines), lp, l_list);
489:
490:
1.24 niallo 491: p = c = data;
492: for (i = 0; i < len; i++) {
493: if (*p == '\n' || (i == len - 1)) {
1.26 niallo 494: tlen = p - c + 1;
1.15 niallo 495: lp = xmalloc(sizeof(*lp));
1.24 niallo 496: lp->l_line = c;
497: lp->l_len = tlen;
1.15 niallo 498: lp->l_lineno = ++(lines->l_nblines);
499: TAILQ_INSERT_TAIL(&(lines->l_lines), lp, l_list);
500: c = p + 1;
501: }
502: p++;
1.7 xsa 503: }
504:
505: return (lines);
506: }
507:
508: void
509: rcs_freelines(struct rcs_lines *lines)
510: {
511: struct rcs_line *lp;
512:
513: while ((lp = TAILQ_FIRST(&(lines->l_lines))) != NULL) {
514: TAILQ_REMOVE(&(lines->l_lines), lp, l_list);
515: xfree(lp);
516: }
517:
518: xfree(lines);
519: }
520:
521: BUF *
1.27 xsa 522: rcs_patchfile(u_char *data, size_t dlen, u_char *patch, size_t plen,
1.7 xsa 523: int (*p)(struct rcs_lines *, struct rcs_lines *))
524: {
525: struct rcs_lines *dlines, *plines;
526: struct rcs_line *lp;
527: BUF *res;
528:
1.24 niallo 529: dlines = rcs_splitlines(data, dlen);
530: plines = rcs_splitlines(patch, plen);
1.7 xsa 531:
532: if (p(dlines, plines) < 0) {
533: rcs_freelines(dlines);
534: rcs_freelines(plines);
535: return (NULL);
536: }
537:
1.35 ray 538: res = buf_alloc(1024);
1.7 xsa 539: TAILQ_FOREACH(lp, &dlines->l_lines, l_list) {
1.24 niallo 540: if (lp->l_line == NULL)
541: continue;
1.34 ray 542: buf_append(res, lp->l_line, lp->l_len);
1.7 xsa 543: }
544:
545: rcs_freelines(dlines);
546: rcs_freelines(plines);
547: return (res);
548: }
549:
550: /*
551: * rcs_yesno()
552: *
1.23 millert 553: * Read a char from standard input, returns defc if the
554: * user enters an equivalent to defc, else whatever char
555: * was entered. Converts input to lower case.
1.7 xsa 556: */
557: int
1.23 millert 558: rcs_yesno(int defc)
1.7 xsa 559: {
560: int c, ret;
561:
562: fflush(stderr);
563: fflush(stdout);
564:
1.25 ray 565: clearerr(stdin);
1.23 millert 566: if (isalpha(c = getchar()))
567: c = tolower(c);
568: if (c == defc || c == '\n' || (c == EOF && feof(stdin)))
569: ret = defc;
1.7 xsa 570: else
1.23 millert 571: ret = c;
572:
573: while (c != EOF && c != '\n')
574: c = getchar();
1.7 xsa 575:
576: return (ret);
577: }
578:
579: /*
580: * rcs_strsplit()
581: *
582: * Split a string <str> of <sep>-separated values and allocate
583: * an argument vector for the values found.
584: */
585: struct rcs_argvector *
1.14 ray 586: rcs_strsplit(const char *str, const char *sep)
1.7 xsa 587: {
588: struct rcs_argvector *av;
589: size_t i = 0;
590: char *cp, *p;
591:
592: cp = xstrdup(str);
593: av = xmalloc(sizeof(*av));
594: av->str = cp;
1.31 ray 595: av->argv = xmalloc(sizeof(*(av->argv)));
1.7 xsa 596:
597: while ((p = strsep(&cp, sep)) != NULL) {
598: av->argv[i++] = p;
1.30 ray 599: av->argv = xrealloc(av->argv,
1.7 xsa 600: i + 1, sizeof(*(av->argv)));
601: }
602: av->argv[i] = NULL;
603:
604: return (av);
605: }
606:
607: /*
608: * rcs_argv_destroy()
609: *
610: * Free an argument vector previously allocated by rcs_strsplit().
611: */
612: void
613: rcs_argv_destroy(struct rcs_argvector *av)
614: {
615: xfree(av->str);
616: xfree(av->argv);
617: xfree(av);
1.28 otto 618: }
619:
620: /*
621: * Strip suffix from filename.
622: */
623: void
624: rcs_strip_suffix(char *filename)
625: {
626: char *p, *suffixes, *next, *ext;
627:
628: if ((p = strrchr(filename, ',')) != NULL) {
629: suffixes = xstrdup(rcs_suffixes);
630: for (next = suffixes; (ext = strsep(&next, "/")) != NULL;) {
631: if (!strcmp(p, ext)) {
632: *p = '\0';
633: break;
634: }
635: }
636: xfree(suffixes);
637: }
1.1 xsa 638: }