Annotation of src/usr.bin/rcs/co.c, Revision 1.52
1.52 ! xsa 1: /* $OpenBSD: co.c,v 1.51 2006/01/02 08:13:28 xsa Exp $ */
1.1 joris 2: /*
3: * Copyright (c) 2005 Joris Vink <joris@openbsd.org>
4: * All rights reserved.
5: *
6: * Redistribution and use in source and binary forms, with or without
7: * modification, are permitted provided that the following conditions
8: * are met:
9: *
10: * 1. Redistributions of source code must retain the above copyright
11: * notice, this list of conditions and the following disclaimer.
12: * 2. The name of the author may not be used to endorse or promote products
13: * derived from this software without specific prior written permission.
14: *
15: * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
16: * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
17: * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
18: * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19: * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20: * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
21: * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
22: * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
23: * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
24: * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25: */
26:
1.51 xsa 27: #include "includes.h"
1.1 joris 28:
29: #include "rcsprog.h"
30:
1.49 joris 31: #define CO_OPTSTRING "d:f::k:l::M::p::q::r::s:Tu::Vw::x:"
1.38 xsa 32:
1.43 xsa 33: static void checkout_err_nobranch(RCSFILE *, const char *, const char *,
34: const char *, int);
1.4 joris 35:
1.1 joris 36: int
37: checkout_main(int argc, char **argv)
38: {
1.33 xsa 39: int i, ch, flags, kflag;
1.4 joris 40: RCSNUM *frev, *rev;
1.1 joris 41: RCSFILE *file;
1.45 joris 42: char fpath[MAXPATHLEN];
1.49 joris 43: char *author, *username, *date;
1.38 xsa 44: const char *state;
1.37 xsa 45: time_t rcs_mtime = -1;
1.1 joris 46:
1.24 niallo 47: flags = 0;
1.33 xsa 48: kflag = RCS_KWEXP_ERR;
1.1 joris 49: rev = RCS_HEAD_REV;
1.6 joris 50: frev = NULL;
1.44 joris 51: state = NULL;
52: author = NULL;
1.49 joris 53: date = NULL;
1.4 joris 54:
1.38 xsa 55: while ((ch = rcs_getopt(argc, argv, CO_OPTSTRING)) != -1) {
1.1 joris 56: switch (ch) {
1.49 joris 57: case 'd':
58: date = xstrdup(rcs_optarg);
59: break;
1.18 joris 60: case 'f':
1.19 joris 61: rcs_set_rev(rcs_optarg, &rev);
1.24 niallo 62: flags |= FORCE;
1.18 joris 63: break;
1.33 xsa 64: case 'k':
65: kflag = rcs_kflag_get(rcs_optarg);
66: if (RCS_KWEXP_INVAL(kflag)) {
67: cvs_log(LP_ERR,
68: "invalid RCS keyword expansion mode");
69: (usage)();
70: exit(1);
71: }
72: break;
1.4 joris 73: case 'l':
1.19 joris 74: rcs_set_rev(rcs_optarg, &rev);
1.24 niallo 75: flags |= CO_LOCK;
1.26 niallo 76: break;
77: case 'M':
78: rcs_set_rev(rcs_optarg, &rev);
79: flags |= CO_REVDATE;
1.4 joris 80: break;
1.20 joris 81: case 'p':
82: rcs_set_rev(rcs_optarg, &rev);
83: pipeout = 1;
84: break;
1.3 joris 85: case 'q':
1.32 xsa 86: rcs_set_rev(rcs_optarg, &rev);
1.3 joris 87: verbose = 0;
88: break;
1.1 joris 89: case 'r':
1.19 joris 90: rcs_set_rev(rcs_optarg, &rev);
1.4 joris 91: break;
1.24 niallo 92: case 's':
1.47 joris 93: state = xstrdup(rcs_optarg);
1.24 niallo 94: flags |= CO_STATE;
1.34 xsa 95: break;
96: case 'T':
97: flags |= PRESERVETIME;
1.24 niallo 98: break;
1.4 joris 99: case 'u':
1.19 joris 100: rcs_set_rev(rcs_optarg, &rev);
1.24 niallo 101: flags |= CO_UNLOCK;
1.1 joris 102: break;
1.7 joris 103: case 'V':
104: printf("%s\n", rcs_version);
105: exit(0);
1.38 xsa 106: case 'w':
1.44 joris 107: /* if no argument, assume current user */
108: if (rcs_optarg == NULL) {
1.48 xsa 109: if ((author = getlogin()) == NULL)
110: fatal("getlogin failed");
1.47 joris 111: } else
112: author = xstrdup(rcs_optarg);
1.43 xsa 113: flags |= CO_AUTHOR;
1.38 xsa 114: break;
1.30 xsa 115: case 'x':
116: rcs_suffixes = rcs_optarg;
117: break;
1.1 joris 118: default:
119: (usage)();
120: exit(1);
121: }
122: }
123:
1.13 joris 124: argc -= rcs_optind;
125: argv += rcs_optind;
1.1 joris 126:
127: if (argc == 0) {
128: cvs_log(LP_ERR, "no input file");
129: (usage)();
130: exit (1);
131: }
1.11 deraadt 132:
1.44 joris 133: if ((username = getlogin()) == NULL) {
1.38 xsa 134: cvs_log(LP_ERRNO, "failed to get username");
135: exit (1);
136: }
137:
1.1 joris 138: for (i = 0; i < argc; i++) {
139: if (rcs_statfile(argv[i], fpath, sizeof(fpath)) < 0)
140: continue;
141:
1.21 niallo 142: if (verbose == 1)
1.22 xsa 143: printf("%s --> %s\n", fpath,
144: (pipeout == 1) ? "standard output" : argv[i]);
1.35 xsa 145:
146: if ((flags & CO_LOCK) && (kflag & RCS_KWEXP_VAL)) {
147: cvs_log(LP_ERR, "%s: cannot combine -kv and -l", fpath);
148: continue;
149: }
1.21 niallo 150:
1.50 niallo 151: if ((file = rcs_open(fpath, RCS_RDWR|RCS_PARSE_FULLY)) == NULL)
1.1 joris 152: continue;
153:
1.37 xsa 154: if (flags & PRESERVETIME)
155: rcs_mtime = rcs_get_mtime(file->rf_path);
156:
157: if (kflag != RCS_KWEXP_ERR)
158: rcs_kwexp_set(file, kflag);
159:
1.4 joris 160: if (rev == RCS_HEAD_REV)
161: frev = file->rf_head;
162: else
163: frev = rev;
1.17 joris 164:
1.44 joris 165: if (checkout_rev(file, frev, argv[i], flags,
1.49 joris 166: username, author, state, date) < 0) {
1.24 niallo 167: rcs_close(file);
168: continue;
1.8 niallo 169: }
1.17 joris 170:
1.1 joris 171: rcs_close(file);
1.37 xsa 172:
173: if (flags & PRESERVETIME)
174: rcs_set_mtime(fpath, rcs_mtime);
1.1 joris 175: }
176:
177: if (rev != RCS_HEAD_REV)
1.5 joris 178: rcsnum_free(frev);
1.1 joris 179:
180: return (0);
181: }
182:
183: void
184: checkout_usage(void)
185: {
1.11 deraadt 186: fprintf(stderr,
1.42 xsa 187: "usage: co [-TV] [-ddate] [-f[rev]] [-I[rev]] [-kmode] [-l[rev]]\n"
188: " [-M[rev]] [-p[rev]] [-q[rev]] [-r[rev]] [-sstate]\n"
189: " [-u[rev]] [-w[user]] [-xsuffixes] [-ztz] file ...\n");
1.1 joris 190: }
1.14 niallo 191:
192: /*
193: * Checkout revision <rev> from RCSFILE <file>, writing it to the path <dst>
1.29 xsa 194: * Currenly recognised <flags> are CO_LOCK, CO_UNLOCK and CO_REVDATE.
1.14 niallo 195: *
1.49 joris 196: * Looks up revision based upon <lockname>, <author>, <state> and <date>
1.44 joris 197: *
1.14 niallo 198: * Returns 0 on success, -1 on failure.
199: */
200: int
1.24 niallo 201: checkout_rev(RCSFILE *file, RCSNUM *frev, const char *dst, int flags,
1.49 joris 202: const char *lockname, const char *author, const char *state,
203: const char *date)
1.14 niallo 204: {
1.44 joris 205: BUF *bp;
206: int lcount;
1.18 joris 207: char buf[16], yn;
1.14 niallo 208: mode_t mode = 0444;
1.18 joris 209: struct stat st;
1.44 joris 210: struct rcs_delta *rdp;
211: struct rcs_lock *lkp;
1.49 joris 212: char *content, msg[128], *fdate;
213: time_t rcsdate, givendate;
214:
215: rcsdate = givendate = -1;
216: if (date != NULL)
217: givendate = cvs_date_parse(date);
1.14 niallo 218:
1.39 xsa 219: /* Check out the latest revision if <frev> is greater than HEAD */
1.15 niallo 220: if (rcsnum_cmp(frev, file->rf_head, 0) == -1)
1.23 xsa 221: frev = file->rf_head;
1.15 niallo 222:
1.44 joris 223: lcount = 0;
224: TAILQ_FOREACH(lkp, &(file->rf_locks), rl_list) {
225: if (!strcmp(lkp->rl_name, lockname))
226: lcount++;
227: }
228:
229: /*
230: * If the user didn't specify any revision, we cycle through
231: * revisions to lookup the first one that matches what he specified.
232: *
233: * If we cannot find one, we return an error.
234: */
235: rdp = NULL;
236: if (frev == file->rf_head) {
237: if (lcount > 1) {
238: cvs_log(LP_WARN,
239: "multiple revisions locked by %s; "
240: "please specify one", lockname);
241: return (-1);
242: }
243:
244: TAILQ_FOREACH(rdp, &file->rf_delta, rd_list) {
1.49 joris 245: if (date != NULL) {
246: fdate = asctime(&rdp->rd_date);
247: rcsdate = cvs_date_parse(fdate);
248: if (givendate <= rcsdate)
249: continue;
250: }
251:
1.44 joris 252: if ((author != NULL) &&
253: (strcmp(rdp->rd_author, author)))
254: continue;
1.49 joris 255:
1.44 joris 256: if ((state != NULL) &&
257: (strcmp(rdp->rd_state, state)))
258: continue;
259:
260: frev = rdp->rd_num;
261: break;
262: }
263: } else {
264: rdp = rcs_findrev(file, frev);
265: }
266:
267: if (rdp == NULL) {
1.49 joris 268: checkout_err_nobranch(file, author, date, state, flags);
1.44 joris 269: return (-1);
270: }
271:
1.15 niallo 272: rcsnum_tostr(frev, buf, sizeof(buf));
273:
1.44 joris 274: if (rdp->rd_locker != NULL) {
275: if (strcmp(lockname, rdp->rd_locker)) {
276: strlcpy(msg, "Revision %s is already locked by %s; ",
277: sizeof(msg));
278: if (flags & CO_UNLOCK)
279: strlcat(msg, "use co -r or rcs -u", sizeof(msg));
280: cvs_log(LP_ERR, msg, buf, rdp->rd_locker);
281: return (-1);
282: }
283: }
284:
1.18 joris 285: if (verbose == 1)
286: printf("revision %s", buf);
287:
1.44 joris 288:
1.14 niallo 289: if ((bp = rcs_getrev(file, frev)) == NULL) {
290: cvs_log(LP_ERR, "cannot find revision `%s'", buf);
291: return (-1);
292: }
293:
1.24 niallo 294: if (flags & CO_LOCK) {
1.44 joris 295: if ((lockname != NULL)
296: && (rcs_lock_add(file, lockname, frev) < 0)) {
297: if (rcs_errno != RCS_ERR_DUPENT)
298: return (-1);
1.14 niallo 299: }
1.18 joris 300:
1.14 niallo 301: mode = 0644;
1.18 joris 302: if (verbose == 1)
303: printf(" (locked)");
1.24 niallo 304: } else if (flags & CO_UNLOCK) {
1.46 joris 305: if (rcs_lock_remove(file, lockname, frev) < 0) {
306: if (rcs_errno != RCS_ERR_NOENT)
307: return (-1);
308: }
1.18 joris 309:
1.14 niallo 310: mode = 0444;
1.18 joris 311: if (verbose == 1)
312: printf(" (unlocked)");
313: }
314:
315: if (verbose == 1)
316: printf("\n");
317:
1.44 joris 318: if (flags & CO_LOCK) {
319: lcount++;
320: if (lcount > 1)
321: cvs_log(LP_WARN, "You now have %d locks.", lcount);
322: }
323:
1.41 xsa 324: if ((pipeout == 0) && (stat(dst, &st) == 0) && !(flags & FORCE)) {
1.18 joris 325: if (st.st_mode & S_IWUSR) {
326: yn = 0;
1.21 niallo 327: if (verbose == 0) {
328: cvs_log(LP_ERR,
1.40 xsa 329: "writable %s exists; checkout aborted",
1.21 niallo 330: dst);
331: return (-1);
332: }
1.41 xsa 333:
334: while ((yn != 'y') && (yn != 'n')) {
335: printf("writable %s exists%s; ", dst,
336: ((uid_t)getuid() == st.st_uid) ? "" :
337: ", and you do not own it");
1.21 niallo 338: printf("remove it? [ny](n): ");
1.18 joris 339: fflush(stdout);
340: yn = getchar();
341: }
342:
343: if (yn == 'n') {
1.21 niallo 344: cvs_log(LP_ERR, "checkout aborted");
1.18 joris 345: return (-1);
346: }
347: }
1.14 niallo 348: }
1.17 joris 349:
1.20 joris 350: if (pipeout == 1) {
351: cvs_buf_putc(bp, '\0');
352: content = cvs_buf_release(bp);
353: printf("%s", content);
1.47 joris 354: xfree(content);
1.20 joris 355: } else {
356: if (cvs_buf_write(bp, dst, mode) < 0) {
357: cvs_log(LP_ERR, "failed to write revision to file");
358: cvs_buf_free(bp);
359: return (-1);
360: }
1.14 niallo 361: cvs_buf_free(bp);
1.24 niallo 362: if (flags & CO_REVDATE) {
363: struct timeval tv[2];
1.36 xsa 364: memset(&tv, 0, sizeof(tv));
1.24 niallo 365: tv[0].tv_sec = (long)rcs_rev_getdate(file, frev);
366: tv[1].tv_sec = tv[0].tv_sec;
367: if (utimes(dst, (const struct timeval *)&tv) < 0)
368: cvs_log(LP_ERRNO, "error setting utimes");
369: }
1.20 joris 370:
371: if (verbose == 1)
372: printf("done\n");
1.14 niallo 373: }
1.17 joris 374:
1.14 niallo 375: return (0);
1.43 xsa 376: }
377:
378: /*
379: * checkout_err_nobranch()
380: *
381: * XXX - should handle the dates too.
382: */
383: static void
384: checkout_err_nobranch(RCSFILE *file, const char *author, const char *date,
385: const char *state, int flags)
386: {
387: if (!(flags & CO_AUTHOR))
388: author = NULL;
389: if (!(flags & CO_STATE))
390: state = NULL;
391:
392: cvs_log(LP_ERR, "%s: No revision on branch has%s%s%s%s%s%s.",
393: file->rf_path,
394: date ? " a date before " : "",
395: date ? date : "",
396: author ? " and author " + (date ? 0:4 ) : "",
397: author ? author : "",
398: state ? " and state " + (date || author ? 0:4) : "",
399: state ? state : "");
1.24 niallo 400: }