Edit

IABSD.fr/src/usr.bin/cvs/edit.c

Branch :

  • Show log

    Commit

  • Author : joris
    Date : 2017-06-01 08:08:24
    Hash : 4dcde513
    Message : Stop looking at current_cvsroot->cr_method to figure out if we're remote or not. Instead use cvsroot_is_local() and cvsroot_is_remote().

  • usr.bin/cvs/edit.c
  • /*	$OpenBSD: edit.c,v 1.53 2017/06/01 08:08:24 joris Exp $	*/
    /*
     * Copyright (c) 2006, 2007 Xavier Santolaria <xsa@openbsd.org>
     *
     * Permission to use, copy, modify, and distribute this software for any
     * purpose with or without fee is hereby granted, provided that the above
     * copyright notice and this permission notice appear in all copies.
     *
     * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
     * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
     * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
     * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
     * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
     * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
     * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
     */
    
    #include <sys/stat.h>
    
    #include <errno.h>
    #include <stdlib.h>
    #include <string.h>
    #include <time.h>
    #include <unistd.h>
    
    #include "cvs.h"
    #include "remote.h"
    
    #define E_COMMIT	0x01
    #define E_EDIT		0x02
    #define E_UNEDIT	0x04
    #define E_ALL		(E_EDIT|E_COMMIT|E_UNEDIT)
    
    #define BASE_ADD	0x01
    #define BASE_GET	0x02
    #define BASE_REMOVE	0x04
    
    static void	cvs_edit_local(struct cvs_file *);
    static void	cvs_editors_local(struct cvs_file *);
    static void	cvs_unedit_local(struct cvs_file *);
    
    static RCSNUM	*cvs_base_handle(struct cvs_file *, int);
    
    static int	edit_aflags = 0;
    
    struct cvs_cmd cvs_cmd_edit = {
    	CVS_OP_EDIT, CVS_USE_WDIR, "edit",
    	{ { 0 }, { 0 } },
    	"Get ready to edit a watched file",
    	"[-lR] [-a action] [file ...]",
    	"a:lR",
    	NULL,
    	cvs_edit
    };
    
    struct cvs_cmd cvs_cmd_editors = {
    	CVS_OP_EDITORS, CVS_USE_WDIR, "editors",
    	{ { 0 }, { 0 } },
    	"See who is editing a watched file",
    	"[-lR] [file ...]",
    	"lR",
    	NULL,
    	cvs_editors
    };
    
    struct cvs_cmd cvs_cmd_unedit = {
    	CVS_OP_UNEDIT, CVS_USE_WDIR, "unedit",
    	{ { 0 }, { 0 } },
    	"Undo an edit command",
    	"[-lR] [file ...]",
    	"lR",
    	NULL,
    	cvs_unedit
    };
    
    int
    cvs_edit(int argc, char **argv)
    {
    	int ch;
    	int flags;
    	struct cvs_recursion cr;
    
    	flags = CR_RECURSE_DIRS;
    
    	while ((ch = getopt(argc, argv, cvs_cmd_edit.cmd_opts)) != -1) {
    		switch (ch) {
    		case 'a':
    			if (strcmp(optarg, "edit") == 0)
    				edit_aflags |= E_EDIT;
    			else if (strcmp(optarg, "unedit") == 0)
    				edit_aflags |= E_UNEDIT;
    			else if (strcmp(optarg, "commit") == 0)
    				edit_aflags |= E_COMMIT;
    			else if (strcmp(optarg, "all") == 0)
    				edit_aflags |= E_ALL;
    			else if (strcmp(optarg, "none") == 0)
    				edit_aflags &= ~E_ALL;
    			else
    				fatal("%s", cvs_cmd_edit.cmd_synopsis);
    			break;
    		case 'l':
    			flags &= ~CR_RECURSE_DIRS;
    			break;
    		case 'R':
    			flags |= CR_RECURSE_DIRS;
    			break;
    		default:
    			fatal("%s", cvs_cmd_edit.cmd_synopsis);
    		}
    	}
    
    	argc -= optind;
    	argv += optind;
    
    	if (argc == 0)
    		fatal("%s", cvs_cmd_edit.cmd_synopsis);
    
    	if (edit_aflags == 0)
    		edit_aflags |= E_ALL;
    
    	cr.enterdir = NULL;
    	cr.leavedir = NULL;
    
    	if (cvsroot_is_remote()) {
    		cvs_client_connect_to_server();
    		cr.fileproc = cvs_client_sendfile;
    
    		if (!(flags & CR_RECURSE_DIRS))
    			cvs_client_send_request("Argument -l");
    	} else {
    		cr.fileproc = cvs_edit_local;
    	}
    
    	cr.flags = flags;
    
    	cvs_file_run(argc, argv, &cr);
    
    	if (cvsroot_is_remote()) {
    		cvs_client_send_files(argv, argc);
    		cvs_client_senddir(".");
    		cvs_client_send_request("edit");
    		cvs_client_get_responses();
    	}
    
    	return (0);
    }
    
    int
    cvs_editors(int argc, char **argv)
    {
    	int ch;
    	int flags;
    	struct cvs_recursion cr;
    
    	flags = CR_RECURSE_DIRS;
    
    	while ((ch = getopt(argc, argv, cvs_cmd_editors.cmd_opts)) != -1) {
    		switch (ch) {
    		case 'l':
    			flags &= ~CR_RECURSE_DIRS;
    			break;
    		case 'R':
    			flags |= CR_RECURSE_DIRS;
    			break;
    		default:
    			fatal("%s", cvs_cmd_editors.cmd_synopsis);
    		}
    	}
    
    	argc -= optind;
    	argv += optind;
    
    	if (argc == 0)
    		fatal("%s", cvs_cmd_editors.cmd_synopsis);
    
    	cr.enterdir = NULL;
    	cr.leavedir = NULL;
    
    	if (cvsroot_is_remote()) {
    		cvs_client_connect_to_server();
    		cr.fileproc = cvs_client_sendfile;
    
    		if (!(flags & CR_RECURSE_DIRS))
    			cvs_client_send_request("Argument -l");
    	} else {
    		cr.fileproc = cvs_editors_local;
    	}
    
    	cr.flags = flags;
    
    	cvs_file_run(argc, argv, &cr);
    
    	if (cvsroot_is_remote()) {
    		cvs_client_send_files(argv, argc);
    		cvs_client_senddir(".");
    		cvs_client_send_request("editors");
    		cvs_client_get_responses();
    	}
    
    	return (0);
    }
    
    int
    cvs_unedit(int argc, char **argv)
    {
    	int ch;
    	int flags;
    	struct cvs_recursion cr;
    
    	flags = CR_RECURSE_DIRS;
    
    	while ((ch = getopt(argc, argv, cvs_cmd_unedit.cmd_opts)) != -1) {
    		switch (ch) {
    		case 'l':
    			flags &= ~CR_RECURSE_DIRS;
    			break;
    		case 'R':
    			flags |= CR_RECURSE_DIRS;
    			break;
    		default:
    			fatal("%s", cvs_cmd_unedit.cmd_synopsis);
    		}
    	}
    
    	argc -= optind;
    	argv += optind;
    
    	if (argc == 0)
    		fatal("%s", cvs_cmd_unedit.cmd_synopsis);
    
    	cr.enterdir = NULL;
    	cr.leavedir = NULL;
    
    	if (cvsroot_is_remote()) {
    		cvs_client_connect_to_server();
    		cr.fileproc = cvs_client_sendfile;
    
    		if (!(flags & CR_RECURSE_DIRS))
    			cvs_client_send_request("Argument -l");
    	} else {
    		cr.fileproc = cvs_unedit_local;
    	}
    
    	cr.flags = flags;
    
    	cvs_file_run(argc, argv, &cr);
    
    	if (cvsroot_is_remote()) {
    		cvs_client_send_files(argv, argc);
    		cvs_client_senddir(".");
    		cvs_client_send_request("unedit");
    		cvs_client_get_responses();
    	}
    
    	return (0);
    }
    
    static void
    cvs_edit_local(struct cvs_file *cf)
    {
    	FILE *fp;
    	struct tm t;
    	time_t now;
    	char timebuf[CVS_TIME_BUFSZ], thishost[HOST_NAME_MAX+1];
    	char bfpath[PATH_MAX], wdir[PATH_MAX];
    
    	if (cvs_noexec == 1)
    		return;
    
    	cvs_log(LP_TRACE, "cvs_edit_local(%s)", cf->file_path);
    
    	cvs_file_classify(cf, cvs_directory_tag);
    
    	if ((fp = fopen(CVS_PATH_NOTIFY, "a")) == NULL)
    		fatal("cvs_edit_local: fopen: `%s': %s",
    		    CVS_PATH_NOTIFY, strerror(errno));
    
    	(void)time(&now);
    	gmtime_r(&now, &t);
    	asctime_r(&t, timebuf);
    	timebuf[strcspn(timebuf, "\n")] = '\0';
    
    	if (gethostname(thishost, sizeof(thishost)) == -1)
    		fatal("gethostname failed");
    
    	if (getcwd(wdir, sizeof(wdir)) == NULL)
    		fatal("getcwd failed");
    
    	(void)fprintf(fp, "E%s\t%s GMT\t%s\t%s\t",
    	    cf->file_name, timebuf, thishost, wdir);
    
    	if (edit_aflags & E_EDIT)
    		(void)fprintf(fp, "E");
    	if (edit_aflags & E_UNEDIT)
    		(void)fprintf(fp, "U");
    	if (edit_aflags & E_COMMIT)
    		(void)fprintf(fp, "C");
    
    	(void)fprintf(fp, "\n");
    
    	(void)fclose(fp);
    
    	if (fchmod(cf->fd, 0644) == -1)
    		fatal("cvs_edit_local: fchmod %s", strerror(errno));
    
    	(void)xsnprintf(bfpath, PATH_MAX, "%s/%s",
    	    CVS_PATH_BASEDIR, cf->file_name);
    
    	if (mkdir(CVS_PATH_BASEDIR, 0755) == -1 && errno != EEXIST)
    		fatal("cvs_edit_local: `%s': %s", CVS_PATH_BASEDIR,
    		    strerror(errno));
    
    	if (cvs_file_copy(cf->file_path, bfpath) == -1)
    		fatal("cvs_edit_local: cvs_file_copy failed");
    
    	(void)cvs_base_handle(cf, BASE_ADD);
    }
    
    static void
    cvs_editors_local(struct cvs_file *cf)
    {
    }
    
    static void
    cvs_unedit_local(struct cvs_file *cf)
    {
    	FILE *fp;
    	struct stat st;
    	struct tm t;
    	time_t now;
    	char bfpath[PATH_MAX], timebuf[64], thishost[HOST_NAME_MAX+1];
    	char wdir[PATH_MAX], sticky[CVS_ENT_MAXLINELEN];
    	RCSNUM *ba_rev;
    
    	cvs_log(LP_TRACE, "cvs_unedit_local(%s)", cf->file_path);
    
    	if (cvs_noexec == 1)
    		return;
    
    	cvs_file_classify(cf, cvs_directory_tag);
    
    	(void)xsnprintf(bfpath, PATH_MAX, "%s/%s",
    	    CVS_PATH_BASEDIR, cf->file_name);
    
    	if (stat(bfpath, &st) == -1)
    		return;
    
    	if (cvs_file_cmp(cf->file_path, bfpath) != 0) {
    		cvs_printf("%s has been modified; revert changes? ",
    		    cf->file_name);
    
    		if (cvs_yesno() == -1)
    			return;
    	}
    
    	cvs_rename(bfpath, cf->file_path);
    
    	if ((fp = fopen(CVS_PATH_NOTIFY, "a")) == NULL)
    		fatal("cvs_unedit_local: fopen: `%s': %s",
    		    CVS_PATH_NOTIFY, strerror(errno));
    
    	(void)time(&now);
    	gmtime_r(&now, &t);
    	asctime_r(&t, timebuf);
    	timebuf[strcspn(timebuf, "\n")] = '\0';
    
    	if (gethostname(thishost, sizeof(thishost)) == -1)
    		fatal("gethostname failed");
    
    	if (getcwd(wdir, sizeof(wdir)) == NULL)
    		fatal("getcwd failed");
    
    	(void)fprintf(fp, "U%s\t%s GMT\t%s\t%s\t\n",
    	    cf->file_name, timebuf, thishost, wdir);
    
    	(void)fclose(fp);
    
    	if ((ba_rev = cvs_base_handle(cf, BASE_GET)) == NULL) {
    		cvs_log(LP_ERR, "%s not mentioned in %s",
    		    cf->file_name, CVS_PATH_BASEREV);
    		return;
    	}
    
    	if (cf->file_ent != NULL) {
    		CVSENTRIES *entlist;
    		struct cvs_ent *ent;
    		char *entry, rbuf[CVS_REV_BUFSZ];
    
    		entlist = cvs_ent_open(cf->file_wd);
    
    		if ((ent = cvs_ent_get(entlist, cf->file_name)) == NULL)
    			fatal("cvs_unedit_local: cvs_ent_get failed");
    
    		(void)rcsnum_tostr(ba_rev, rbuf, sizeof(rbuf));
    
    		memset(timebuf, 0, sizeof(timebuf));
    		ctime_r(&cf->file_ent->ce_mtime, timebuf);
    		timebuf[strcspn(timebuf, "\n")] = '\0';
    
    		sticky[0] = '\0';
    		if (cf->file_ent->ce_tag != NULL)
    			(void)xsnprintf(sticky, sizeof(sticky), "T%s",
    			    cf->file_ent->ce_tag);
    
    		(void)xasprintf(&entry, "/%s/%s/%s/%s/%s",
    		    cf->file_name, rbuf, timebuf, cf->file_ent->ce_opts ? 
    		    cf->file_ent->ce_opts : "", sticky);
    
    		cvs_ent_add(entlist, entry);
    
    		cvs_ent_free(ent);
    
    		free(entry);
    	}
    
    	free(ba_rev);
    
    	(void)cvs_base_handle(cf, BASE_REMOVE);
    
    	if (fchmod(cf->fd, 0644) == -1)
    		fatal("cvs_unedit_local: fchmod %s", strerror(errno));
    }
    
    static RCSNUM *
    cvs_base_handle(struct cvs_file *cf, int flags)
    {
    	FILE *fp, *tfp;
    	RCSNUM *ba_rev;
    	int i;
    	char *dp, *sp;
    	char buf[PATH_MAX], *fields[2], rbuf[CVS_REV_BUFSZ];
    
    	cvs_log(LP_TRACE, "cvs_base_handle(%s)", cf->file_path);
    
    	tfp = NULL;
    	ba_rev = NULL;
    
    	if (((fp = fopen(CVS_PATH_BASEREV, "r")) == NULL) && errno != ENOENT) {
    		cvs_log(LP_ERRNO, "%s", CVS_PATH_BASEREV);
    		goto out;
    	}
    
    	if (flags & (BASE_ADD|BASE_REMOVE)) {
    		if ((tfp = fopen(CVS_PATH_BASEREVTMP, "w")) == NULL) {
    			cvs_log(LP_ERRNO, "%s", CVS_PATH_BASEREVTMP);
    			goto out;
    		}
    	}
    
    	if (fp != NULL) {
    		while (fgets(buf, sizeof(buf), fp)) {
    			buf[strcspn(buf, "\n")] = '\0';
    
    			if (buf[0] != 'B')
    				continue;
    
    			sp = buf + 1;
    			i = 0;
    			do {
    				if ((dp = strchr(sp, '/')) != NULL)
    					*(dp++) = '\0';
    				fields[i++] = sp;
    				sp = dp;
    			} while (dp != NULL && i < 2);
    
    			if (cvs_file_cmpname(fields[0], cf->file_path) == 0) {
    				if (flags & BASE_GET) {
    					ba_rev = rcsnum_parse(fields[1]);
    					if (ba_rev == NULL)
    						fatal("cvs_base_handle: "
    						    "rcsnum_parse");
    					goto got_rev;
    				}
    			} else {
    				if (flags & (BASE_ADD|BASE_REMOVE))
    					(void)fprintf(tfp, "%s\n", buf);
    			}
    		}
    	}
    
    got_rev:
    	if (flags & (BASE_ADD)) {
    		(void)rcsnum_tostr(cf->file_ent->ce_rev, rbuf, sizeof(rbuf));
    		(void)fprintf(tfp, "B%s/%s/\n", cf->file_path, rbuf);
    	}
    
    out:
    	if (fp != NULL)
    		(void)fclose(fp);
    
    	if (tfp != NULL) {
    		(void)fclose(tfp);
    		(void)cvs_rename(CVS_PATH_BASEREVTMP, CVS_PATH_BASEREV);
    	}
    
    	return (ba_rev);
    }