1 5c860e29 2018-03-12 stsp /*
   2 f42b1b34 2018-03-12 stsp  * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
   3 5d56da81 2019-01-13 stsp  * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
   4 5c860e29 2018-03-12 stsp  *
   5 5c860e29 2018-03-12 stsp  * Permission to use, copy, modify, and distribute this software for any
   6 5c860e29 2018-03-12 stsp  * purpose with or without fee is hereby granted, provided that the above
   7 5c860e29 2018-03-12 stsp  * copyright notice and this permission notice appear in all copies.
   8 5c860e29 2018-03-12 stsp  *
   9 5c860e29 2018-03-12 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10 5c860e29 2018-03-12 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11 5c860e29 2018-03-12 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12 5c860e29 2018-03-12 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13 5c860e29 2018-03-12 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14 5c860e29 2018-03-12 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15 5c860e29 2018-03-12 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16 5c860e29 2018-03-12 stsp  */
  17 5c860e29 2018-03-12 stsp 
  18 f42b1b34 2018-03-12 stsp #include <sys/queue.h>
  19 64a96a6d 2018-04-01 stsp #include <sys/limits.h>
  20 c0768b0f 2018-06-10 stsp #include <sys/types.h>
  21 5de5890b 2018-10-18 stsp #include <sys/stat.h>
  22 f42b1b34 2018-03-12 stsp 
  23 5c860e29 2018-03-12 stsp #include <err.h>
  24 5c860e29 2018-03-12 stsp #include <errno.h>
  25 5c860e29 2018-03-12 stsp #include <locale.h>
  26 99437157 2018-11-11 stsp #include <signal.h>
  27 5c860e29 2018-03-12 stsp #include <stdio.h>
  28 5c860e29 2018-03-12 stsp #include <stdlib.h>
  29 5c860e29 2018-03-12 stsp #include <string.h>
  30 5c860e29 2018-03-12 stsp #include <unistd.h>
  31 c09a553d 2018-03-12 stsp #include <libgen.h>
  32 c0768b0f 2018-06-10 stsp #include <time.h>
  33 5c860e29 2018-03-12 stsp 
  34 f42b1b34 2018-03-12 stsp #include "got_error.h"
  35 f42b1b34 2018-03-12 stsp #include "got_object.h"
  36 5261c201 2018-04-01 stsp #include "got_reference.h"
  37 f42b1b34 2018-03-12 stsp #include "got_repository.h"
  38 c09a553d 2018-03-12 stsp #include "got_worktree.h"
  39 79109fed 2018-03-27 stsp #include "got_diff.h"
  40 372ccdbb 2018-06-10 stsp #include "got_commit_graph.h"
  41 404c43c4 2018-06-21 stsp #include "got_blame.h"
  42 63219cd2 2019-01-04 stsp #include "got_privsep.h"
  43 5c860e29 2018-03-12 stsp 
  44 5c860e29 2018-03-12 stsp #ifndef nitems
  45 5c860e29 2018-03-12 stsp #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
  46 5c860e29 2018-03-12 stsp #endif
  47 99437157 2018-11-11 stsp 
  48 99437157 2018-11-11 stsp static volatile sig_atomic_t sigint_received;
  49 99437157 2018-11-11 stsp static volatile sig_atomic_t sigpipe_received;
  50 99437157 2018-11-11 stsp 
  51 99437157 2018-11-11 stsp static void
  52 99437157 2018-11-11 stsp catch_sigint(int signo)
  53 99437157 2018-11-11 stsp {
  54 99437157 2018-11-11 stsp 	sigint_received = 1;
  55 99437157 2018-11-11 stsp }
  56 99437157 2018-11-11 stsp 
  57 99437157 2018-11-11 stsp static void
  58 99437157 2018-11-11 stsp catch_sigpipe(int signo)
  59 99437157 2018-11-11 stsp {
  60 99437157 2018-11-11 stsp 	sigpipe_received = 1;
  61 99437157 2018-11-11 stsp }
  62 5c860e29 2018-03-12 stsp 
  63 99437157 2018-11-11 stsp 
  64 5c860e29 2018-03-12 stsp struct cmd {
  65 5c860e29 2018-03-12 stsp 	const char	 *cmd_name;
  66 d7d4f210 2018-03-12 stsp 	const struct got_error *(*cmd_main)(int, char *[]);
  67 1b6b95a8 2018-03-12 stsp 	void		(*cmd_usage)(void);
  68 46a0db7d 2018-03-12 stsp 	const char	 *cmd_descr;
  69 5c860e29 2018-03-12 stsp };
  70 5c860e29 2018-03-12 stsp 
  71 4ed7e80c 2018-05-20 stsp __dead static void	usage(void);
  72 4ed7e80c 2018-05-20 stsp __dead static void	usage_checkout(void);
  73 507dc3bb 2018-12-29 stsp __dead static void	usage_update(void);
  74 4ed7e80c 2018-05-20 stsp __dead static void	usage_log(void);
  75 4ed7e80c 2018-05-20 stsp __dead static void	usage_diff(void);
  76 404c43c4 2018-06-21 stsp __dead static void	usage_blame(void);
  77 5de5890b 2018-10-18 stsp __dead static void	usage_tree(void);
  78 6bad629b 2019-02-04 stsp __dead static void	usage_status(void);
  79 d0eebce4 2019-03-11 stsp __dead static void	usage_ref(void);
  80 5c860e29 2018-03-12 stsp 
  81 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_checkout(int, char *[]);
  82 507dc3bb 2018-12-29 stsp static const struct got_error*		cmd_update(int, char *[]);
  83 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_log(int, char *[]);
  84 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_diff(int, char *[]);
  85 404c43c4 2018-06-21 stsp static const struct got_error*		cmd_blame(int, char *[]);
  86 5de5890b 2018-10-18 stsp static const struct got_error*		cmd_tree(int, char *[]);
  87 4ed7e80c 2018-05-20 stsp static const struct got_error*		cmd_status(int, char *[]);
  88 d0eebce4 2019-03-11 stsp static const struct got_error*		cmd_ref(int, char *[]);
  89 5c860e29 2018-03-12 stsp 
  90 4ed7e80c 2018-05-20 stsp static struct cmd got_commands[] = {
  91 c09a553d 2018-03-12 stsp 	{ "checkout",	cmd_checkout,	usage_checkout,
  92 0bb8a95e 2018-03-12 stsp 	    "check out a new work tree from a repository" },
  93 507dc3bb 2018-12-29 stsp 	{ "update",	cmd_update,	usage_update,
  94 507dc3bb 2018-12-29 stsp 	    "update a work tree to a different commit" },
  95 1b6b95a8 2018-03-12 stsp 	{ "log",	cmd_log,	usage_log,
  96 1b6b95a8 2018-03-12 stsp 	    "show repository history" },
  97 b00d56cd 2018-04-01 stsp 	{ "diff",	cmd_diff,	usage_diff,
  98 b00d56cd 2018-04-01 stsp 	    "compare files and directories" },
  99 404c43c4 2018-06-21 stsp 	{ "blame",	cmd_blame,	usage_blame,
 100 404c43c4 2018-06-21 stsp 	    " show when lines in a file were changed" },
 101 5de5890b 2018-10-18 stsp 	{ "tree",	cmd_tree,	usage_tree,
 102 5de5890b 2018-10-18 stsp 	    " list files and directories in repository" },
 103 1b6b95a8 2018-03-12 stsp 	{ "status",	cmd_status,	usage_status,
 104 1b6b95a8 2018-03-12 stsp 	    "show modification status of files" },
 105 d0eebce4 2019-03-11 stsp 	{ "ref",	cmd_ref,	usage_ref,
 106 d0eebce4 2019-03-11 stsp 	    "manage references in repository" },
 107 5c860e29 2018-03-12 stsp };
 108 5c860e29 2018-03-12 stsp 
 109 5c860e29 2018-03-12 stsp int
 110 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
 111 5c860e29 2018-03-12 stsp {
 112 5c860e29 2018-03-12 stsp 	struct cmd *cmd;
 113 5c860e29 2018-03-12 stsp 	unsigned int i;
 114 5c860e29 2018-03-12 stsp 	int ch;
 115 1b6b95a8 2018-03-12 stsp 	int hflag = 0;
 116 5c860e29 2018-03-12 stsp 
 117 289e3cbf 2019-02-04 stsp 	setlocale(LC_CTYPE, "");
 118 5c860e29 2018-03-12 stsp 
 119 1b6b95a8 2018-03-12 stsp 	while ((ch = getopt(argc, argv, "h")) != -1) {
 120 5c860e29 2018-03-12 stsp 		switch (ch) {
 121 1b6b95a8 2018-03-12 stsp 		case 'h':
 122 1b6b95a8 2018-03-12 stsp 			hflag = 1;
 123 1b6b95a8 2018-03-12 stsp 			break;
 124 5c860e29 2018-03-12 stsp 		default:
 125 5c860e29 2018-03-12 stsp 			usage();
 126 5c860e29 2018-03-12 stsp 			/* NOTREACHED */
 127 5c860e29 2018-03-12 stsp 		}
 128 5c860e29 2018-03-12 stsp 	}
 129 5c860e29 2018-03-12 stsp 
 130 5c860e29 2018-03-12 stsp 	argc -= optind;
 131 5c860e29 2018-03-12 stsp 	argv += optind;
 132 1e70621d 2018-03-27 stsp 	optind = 0;
 133 5c860e29 2018-03-12 stsp 
 134 5c860e29 2018-03-12 stsp 	if (argc <= 0)
 135 5c860e29 2018-03-12 stsp 		usage();
 136 5c860e29 2018-03-12 stsp 
 137 99437157 2018-11-11 stsp 	signal(SIGINT, catch_sigint);
 138 99437157 2018-11-11 stsp 	signal(SIGPIPE, catch_sigpipe);
 139 99437157 2018-11-11 stsp 
 140 5c860e29 2018-03-12 stsp 	for (i = 0; i < nitems(got_commands); i++) {
 141 d7d4f210 2018-03-12 stsp 		const struct got_error *error;
 142 d7d4f210 2018-03-12 stsp 
 143 5c860e29 2018-03-12 stsp 		cmd = &got_commands[i];
 144 5c860e29 2018-03-12 stsp 
 145 5c860e29 2018-03-12 stsp 		if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
 146 5c860e29 2018-03-12 stsp 			continue;
 147 5c860e29 2018-03-12 stsp 
 148 1b6b95a8 2018-03-12 stsp 		if (hflag)
 149 1b6b95a8 2018-03-12 stsp 			got_commands[i].cmd_usage();
 150 1b6b95a8 2018-03-12 stsp 
 151 d7d4f210 2018-03-12 stsp 		error = got_commands[i].cmd_main(argc, argv);
 152 80d5f134 2018-11-11 stsp 		if (error && !(sigint_received || sigpipe_received)) {
 153 d7d4f210 2018-03-12 stsp 			fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
 154 d7d4f210 2018-03-12 stsp 			return 1;
 155 d7d4f210 2018-03-12 stsp 		}
 156 d7d4f210 2018-03-12 stsp 
 157 d7d4f210 2018-03-12 stsp 		return 0;
 158 5c860e29 2018-03-12 stsp 	}
 159 5c860e29 2018-03-12 stsp 
 160 20ecf764 2018-03-12 stsp 	fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
 161 5c860e29 2018-03-12 stsp 	return 1;
 162 5c860e29 2018-03-12 stsp }
 163 5c860e29 2018-03-12 stsp 
 164 4ed7e80c 2018-05-20 stsp __dead static void
 165 5c860e29 2018-03-12 stsp usage(void)
 166 5c860e29 2018-03-12 stsp {
 167 46a0db7d 2018-03-12 stsp 	int i;
 168 46a0db7d 2018-03-12 stsp 
 169 987e94ba 2018-03-12 stsp 	fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
 170 987e94ba 2018-03-12 stsp 	    "Available commands:\n", getprogname());
 171 46a0db7d 2018-03-12 stsp 	for (i = 0; i < nitems(got_commands); i++) {
 172 46a0db7d 2018-03-12 stsp 		struct cmd *cmd = &got_commands[i];
 173 46a0db7d 2018-03-12 stsp 		fprintf(stderr, "    %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
 174 46a0db7d 2018-03-12 stsp 	}
 175 5c860e29 2018-03-12 stsp 	exit(1);
 176 5c860e29 2018-03-12 stsp }
 177 5c860e29 2018-03-12 stsp 
 178 0266afb7 2019-01-04 stsp static const struct got_error *
 179 d0eebce4 2019-03-11 stsp apply_unveil(const char *repo_path, int repo_read_only,
 180 d0eebce4 2019-03-11 stsp     const char *worktree_path)
 181 0266afb7 2019-01-04 stsp {
 182 0266afb7 2019-01-04 stsp 	const struct got_error *error;
 183 0266afb7 2019-01-04 stsp 
 184 d0eebce4 2019-03-11 stsp 	if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
 185 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 186 0266afb7 2019-01-04 stsp 
 187 0266afb7 2019-01-04 stsp 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
 188 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 189 0266afb7 2019-01-04 stsp 
 190 f12d0dbe 2019-01-04 stsp 	if (unveil("/tmp", "rwc") != 0)
 191 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 192 0266afb7 2019-01-04 stsp 
 193 0266afb7 2019-01-04 stsp 	error = got_privsep_unveil_exec_helpers();
 194 0266afb7 2019-01-04 stsp 	if (error != NULL)
 195 0266afb7 2019-01-04 stsp 		return error;
 196 0266afb7 2019-01-04 stsp 
 197 0266afb7 2019-01-04 stsp 	if (unveil(NULL, NULL) != 0)
 198 0266afb7 2019-01-04 stsp 		return got_error_from_errno();
 199 0266afb7 2019-01-04 stsp 
 200 0266afb7 2019-01-04 stsp 	return NULL;
 201 0266afb7 2019-01-04 stsp }
 202 0266afb7 2019-01-04 stsp 
 203 4ed7e80c 2018-05-20 stsp __dead static void
 204 c09a553d 2018-03-12 stsp usage_checkout(void)
 205 c09a553d 2018-03-12 stsp {
 206 0bb8a95e 2018-03-12 stsp 	fprintf(stderr, "usage: %s checkout [-p prefix] repository-path "
 207 0bb8a95e 2018-03-12 stsp 	    "[worktree-path]\n", getprogname());
 208 c09a553d 2018-03-12 stsp 	exit(1);
 209 92a684f4 2018-03-12 stsp }
 210 92a684f4 2018-03-12 stsp 
 211 92a684f4 2018-03-12 stsp static void
 212 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
 213 92a684f4 2018-03-12 stsp {
 214 92a684f4 2018-03-12 stsp 	char *worktree_path = arg;
 215 92a684f4 2018-03-12 stsp 
 216 92a684f4 2018-03-12 stsp 	while (path[0] == '/')
 217 92a684f4 2018-03-12 stsp 		path++;
 218 92a684f4 2018-03-12 stsp 
 219 d7b62c98 2018-12-27 stsp 	printf("%c  %s/%s\n", status, worktree_path, path);
 220 99437157 2018-11-11 stsp }
 221 99437157 2018-11-11 stsp 
 222 99437157 2018-11-11 stsp static const struct got_error *
 223 6bad629b 2019-02-04 stsp check_cancelled(void *arg)
 224 99437157 2018-11-11 stsp {
 225 99437157 2018-11-11 stsp 	if (sigint_received || sigpipe_received)
 226 99437157 2018-11-11 stsp 		return got_error(GOT_ERR_CANCELLED);
 227 99437157 2018-11-11 stsp 	return NULL;
 228 8069f636 2019-01-12 stsp }
 229 8069f636 2019-01-12 stsp 
 230 8069f636 2019-01-12 stsp static const struct got_error *
 231 8069f636 2019-01-12 stsp check_ancestry(struct got_worktree *worktree, struct got_object_id *commit_id,
 232 8069f636 2019-01-12 stsp     struct got_repository *repo)
 233 8069f636 2019-01-12 stsp {
 234 8069f636 2019-01-12 stsp 	const struct got_error *err;
 235 8069f636 2019-01-12 stsp 	struct got_reference *head_ref = NULL;
 236 8069f636 2019-01-12 stsp 	struct got_object_id *head_commit_id = NULL;
 237 8069f636 2019-01-12 stsp 	struct got_commit_graph *graph = NULL;
 238 8069f636 2019-01-12 stsp 
 239 8069f636 2019-01-12 stsp 	head_ref = got_worktree_get_head_ref(worktree);
 240 8069f636 2019-01-12 stsp 	if (head_ref == NULL)
 241 8069f636 2019-01-12 stsp 		return got_error_from_errno();
 242 8069f636 2019-01-12 stsp 
 243 8069f636 2019-01-12 stsp 	/* TODO: Check the reflog. The head ref may have been rebased. */
 244 8069f636 2019-01-12 stsp 	err = got_ref_resolve(&head_commit_id, repo, head_ref);
 245 8069f636 2019-01-12 stsp 	if (err)
 246 8069f636 2019-01-12 stsp 		goto done;
 247 8069f636 2019-01-12 stsp 
 248 8069f636 2019-01-12 stsp 	err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
 249 8069f636 2019-01-12 stsp 	if (err)
 250 8069f636 2019-01-12 stsp 		goto done;
 251 8069f636 2019-01-12 stsp 
 252 8069f636 2019-01-12 stsp 	err = got_commit_graph_iter_start(graph, head_commit_id, repo);
 253 8069f636 2019-01-12 stsp 	if (err)
 254 8069f636 2019-01-12 stsp 		goto done;
 255 8069f636 2019-01-12 stsp 	while (1) {
 256 8069f636 2019-01-12 stsp 		struct got_object_id *id;
 257 8069f636 2019-01-12 stsp 
 258 8069f636 2019-01-12 stsp 		if (sigint_received || sigpipe_received)
 259 8069f636 2019-01-12 stsp 			break;
 260 8069f636 2019-01-12 stsp 
 261 8069f636 2019-01-12 stsp 		err = got_commit_graph_iter_next(&id, graph);
 262 8069f636 2019-01-12 stsp 		if (err) {
 263 8069f636 2019-01-12 stsp 			if (err->code == GOT_ERR_ITER_COMPLETED) {
 264 8069f636 2019-01-12 stsp 				err = got_error(GOT_ERR_ANCESTRY);
 265 8069f636 2019-01-12 stsp 				break;
 266 8069f636 2019-01-12 stsp 			}
 267 8069f636 2019-01-12 stsp 			if (err->code != GOT_ERR_ITER_NEED_MORE)
 268 8069f636 2019-01-12 stsp 				break;
 269 8069f636 2019-01-12 stsp 			err = got_commit_graph_fetch_commits(graph, 1, repo);
 270 8069f636 2019-01-12 stsp 			if (err)
 271 8069f636 2019-01-12 stsp 				break;
 272 8069f636 2019-01-12 stsp 			else
 273 8069f636 2019-01-12 stsp 				continue;
 274 8069f636 2019-01-12 stsp 		}
 275 8069f636 2019-01-12 stsp 		if (id == NULL)
 276 8069f636 2019-01-12 stsp 			break;
 277 8069f636 2019-01-12 stsp 		if (got_object_id_cmp(id, commit_id) == 0)
 278 8069f636 2019-01-12 stsp 			break;
 279 8069f636 2019-01-12 stsp 	}
 280 8069f636 2019-01-12 stsp done:
 281 8069f636 2019-01-12 stsp 	if (head_ref)
 282 8069f636 2019-01-12 stsp 		got_ref_close(head_ref);
 283 8069f636 2019-01-12 stsp 	if (graph)
 284 8069f636 2019-01-12 stsp 		got_commit_graph_close(graph);
 285 8069f636 2019-01-12 stsp 	return err;
 286 c09a553d 2018-03-12 stsp }
 287 c09a553d 2018-03-12 stsp 
 288 8069f636 2019-01-12 stsp 
 289 4ed7e80c 2018-05-20 stsp static const struct got_error *
 290 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
 291 c09a553d 2018-03-12 stsp {
 292 c09a553d 2018-03-12 stsp 	const struct got_error *error = NULL;
 293 c09a553d 2018-03-12 stsp 	struct got_repository *repo = NULL;
 294 c09a553d 2018-03-12 stsp 	struct got_reference *head_ref = NULL;
 295 c09a553d 2018-03-12 stsp 	struct got_worktree *worktree = NULL;
 296 c09a553d 2018-03-12 stsp 	char *repo_path = NULL;
 297 c09a553d 2018-03-12 stsp 	char *worktree_path = NULL;
 298 0bb8a95e 2018-03-12 stsp 	const char *path_prefix = "";
 299 8069f636 2019-01-12 stsp 	char *commit_id_str = NULL;
 300 e5dc7198 2018-12-29 stsp 	int ch, same_path_prefix;
 301 c09a553d 2018-03-12 stsp 
 302 8069f636 2019-01-12 stsp 	while ((ch = getopt(argc, argv, "c:p:")) != -1) {
 303 0bb8a95e 2018-03-12 stsp 		switch (ch) {
 304 8069f636 2019-01-12 stsp 		case 'c':
 305 8069f636 2019-01-12 stsp 			commit_id_str = strdup(optarg);
 306 8069f636 2019-01-12 stsp 			if (commit_id_str == NULL)
 307 8069f636 2019-01-12 stsp 				return got_error_from_errno();
 308 8069f636 2019-01-12 stsp 			break;
 309 0bb8a95e 2018-03-12 stsp 		case 'p':
 310 0bb8a95e 2018-03-12 stsp 			path_prefix = optarg;
 311 0bb8a95e 2018-03-12 stsp 			break;
 312 0bb8a95e 2018-03-12 stsp 		default:
 313 2deda0b9 2019-03-07 stsp 			usage_checkout();
 314 0bb8a95e 2018-03-12 stsp 			/* NOTREACHED */
 315 0bb8a95e 2018-03-12 stsp 		}
 316 0bb8a95e 2018-03-12 stsp 	}
 317 0bb8a95e 2018-03-12 stsp 
 318 0bb8a95e 2018-03-12 stsp 	argc -= optind;
 319 0bb8a95e 2018-03-12 stsp 	argv += optind;
 320 0bb8a95e 2018-03-12 stsp 
 321 6715a751 2018-03-16 stsp #ifndef PROFILE
 322 68ed9ba5 2019-02-10 stsp 	if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
 323 68ed9ba5 2019-02-10 stsp 	    "unveil", NULL) == -1)
 324 c09a553d 2018-03-12 stsp 		err(1, "pledge");
 325 6715a751 2018-03-16 stsp #endif
 326 0bb8a95e 2018-03-12 stsp 	if (argc == 1) {
 327 c09a553d 2018-03-12 stsp 		char *cwd, *base, *dotgit;
 328 76089277 2018-04-01 stsp 		repo_path = realpath(argv[0], NULL);
 329 76089277 2018-04-01 stsp 		if (repo_path == NULL)
 330 76089277 2018-04-01 stsp 			return got_error_from_errno();
 331 c09a553d 2018-03-12 stsp 		cwd = getcwd(NULL, 0);
 332 76089277 2018-04-01 stsp 		if (cwd == NULL) {
 333 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 334 76089277 2018-04-01 stsp 			goto done;
 335 76089277 2018-04-01 stsp 		}
 336 5d7c1dab 2018-04-01 stsp 		if (path_prefix[0])
 337 5d7c1dab 2018-04-01 stsp 			base = basename(path_prefix);
 338 5d7c1dab 2018-04-01 stsp 		else
 339 5d7c1dab 2018-04-01 stsp 			base = basename(repo_path);
 340 76089277 2018-04-01 stsp 		if (base == NULL) {
 341 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 342 76089277 2018-04-01 stsp 			goto done;
 343 76089277 2018-04-01 stsp 		}
 344 c09a553d 2018-03-12 stsp 		dotgit = strstr(base, ".git");
 345 c09a553d 2018-03-12 stsp 		if (dotgit)
 346 c09a553d 2018-03-12 stsp 			*dotgit = '\0';
 347 c09a553d 2018-03-12 stsp 		if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
 348 0a585a0d 2018-03-17 stsp 			error = got_error_from_errno();
 349 c09a553d 2018-03-12 stsp 			free(cwd);
 350 76089277 2018-04-01 stsp 			goto done;
 351 c09a553d 2018-03-12 stsp 		}
 352 c09a553d 2018-03-12 stsp 		free(cwd);
 353 0bb8a95e 2018-03-12 stsp 	} else if (argc == 2) {
 354 76089277 2018-04-01 stsp 		repo_path = realpath(argv[0], NULL);
 355 76089277 2018-04-01 stsp 		if (repo_path == NULL) {
 356 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 357 76089277 2018-04-01 stsp 			goto done;
 358 76089277 2018-04-01 stsp 		}
 359 f7b38925 2018-04-01 stsp 		worktree_path = realpath(argv[1], NULL);
 360 76089277 2018-04-01 stsp 		if (worktree_path == NULL) {
 361 76089277 2018-04-01 stsp 			error = got_error_from_errno();
 362 76089277 2018-04-01 stsp 			goto done;
 363 76089277 2018-04-01 stsp 		}
 364 c09a553d 2018-03-12 stsp 	} else
 365 c09a553d 2018-03-12 stsp 		usage_checkout();
 366 63219cd2 2019-01-04 stsp 
 367 0cd1c46a 2019-03-11 stsp 	error = apply_unveil(repo_path, 0, worktree_path);
 368 0266afb7 2019-01-04 stsp 	if (error)
 369 63219cd2 2019-01-04 stsp 		goto done;
 370 c09a553d 2018-03-12 stsp 
 371 c09a553d 2018-03-12 stsp 	error = got_repo_open(&repo, repo_path);
 372 c09a553d 2018-03-12 stsp 	if (error != NULL)
 373 c09a553d 2018-03-12 stsp 		goto done;
 374 8069f636 2019-01-12 stsp 
 375 c09a553d 2018-03-12 stsp 	error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 376 c09a553d 2018-03-12 stsp 	if (error != NULL)
 377 c09a553d 2018-03-12 stsp 		goto done;
 378 c09a553d 2018-03-12 stsp 
 379 0bb8a95e 2018-03-12 stsp 	error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
 380 d70b8e30 2018-12-27 stsp 	if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
 381 c09a553d 2018-03-12 stsp 		goto done;
 382 c09a553d 2018-03-12 stsp 
 383 c09a553d 2018-03-12 stsp 	error = got_worktree_open(&worktree, worktree_path);
 384 c09a553d 2018-03-12 stsp 	if (error != NULL)
 385 c09a553d 2018-03-12 stsp 		goto done;
 386 c09a553d 2018-03-12 stsp 
 387 e5dc7198 2018-12-29 stsp 	error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
 388 e5dc7198 2018-12-29 stsp 	    path_prefix);
 389 e5dc7198 2018-12-29 stsp 	if (error != NULL)
 390 e5dc7198 2018-12-29 stsp 		goto done;
 391 e5dc7198 2018-12-29 stsp 	if (!same_path_prefix) {
 392 49520a32 2018-12-29 stsp 		error = got_error(GOT_ERR_PATH_PREFIX);
 393 49520a32 2018-12-29 stsp 		goto done;
 394 49520a32 2018-12-29 stsp 	}
 395 49520a32 2018-12-29 stsp 
 396 8069f636 2019-01-12 stsp 	if (commit_id_str) {
 397 8069f636 2019-01-12 stsp 		struct got_object_id *commit_id;
 398 8069f636 2019-01-12 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
 399 8069f636 2019-01-12 stsp 		    commit_id_str);
 400 8069f636 2019-01-12 stsp 		if (error != NULL)
 401 8069f636 2019-01-12 stsp 			goto done;
 402 8069f636 2019-01-12 stsp 		error = check_ancestry(worktree, commit_id, repo);
 403 8069f636 2019-01-12 stsp 		if (error != NULL) {
 404 8069f636 2019-01-12 stsp 			free(commit_id);
 405 8069f636 2019-01-12 stsp 			goto done;
 406 8069f636 2019-01-12 stsp 		}
 407 8069f636 2019-01-12 stsp 		error = got_worktree_set_base_commit_id(worktree, repo,
 408 8069f636 2019-01-12 stsp 		    commit_id);
 409 8069f636 2019-01-12 stsp 		free(commit_id);
 410 8069f636 2019-01-12 stsp 		if (error)
 411 8069f636 2019-01-12 stsp 			goto done;
 412 8069f636 2019-01-12 stsp 	}
 413 8069f636 2019-01-12 stsp 
 414 93a30277 2018-12-24 stsp 	error = got_worktree_checkout_files(worktree, repo,
 415 6bad629b 2019-02-04 stsp 	    checkout_progress, worktree_path, check_cancelled, NULL);
 416 c09a553d 2018-03-12 stsp 	if (error != NULL)
 417 c09a553d 2018-03-12 stsp 		goto done;
 418 c09a553d 2018-03-12 stsp 
 419 b65ae19a 2018-04-24 stsp 	printf("Now shut up and hack\n");
 420 c09a553d 2018-03-12 stsp 
 421 c09a553d 2018-03-12 stsp done:
 422 8069f636 2019-01-12 stsp 	free(commit_id_str);
 423 76089277 2018-04-01 stsp 	free(repo_path);
 424 507dc3bb 2018-12-29 stsp 	free(worktree_path);
 425 507dc3bb 2018-12-29 stsp 	return error;
 426 507dc3bb 2018-12-29 stsp }
 427 507dc3bb 2018-12-29 stsp 
 428 507dc3bb 2018-12-29 stsp __dead static void
 429 507dc3bb 2018-12-29 stsp usage_update(void)
 430 507dc3bb 2018-12-29 stsp {
 431 507dc3bb 2018-12-29 stsp 	fprintf(stderr, "usage: %s update [-c commit] [worktree-path]\n",
 432 507dc3bb 2018-12-29 stsp 	    getprogname());
 433 507dc3bb 2018-12-29 stsp 	exit(1);
 434 507dc3bb 2018-12-29 stsp }
 435 507dc3bb 2018-12-29 stsp 
 436 507dc3bb 2018-12-29 stsp static void
 437 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
 438 507dc3bb 2018-12-29 stsp {
 439 784955db 2019-01-12 stsp 	int *did_something = arg;
 440 784955db 2019-01-12 stsp 
 441 507dc3bb 2018-12-29 stsp 	if (status == GOT_STATUS_EXISTS)
 442 507dc3bb 2018-12-29 stsp 		return;
 443 507dc3bb 2018-12-29 stsp 
 444 1545c615 2019-02-10 stsp 	*did_something = 1;
 445 507dc3bb 2018-12-29 stsp 	while (path[0] == '/')
 446 507dc3bb 2018-12-29 stsp 		path++;
 447 507dc3bb 2018-12-29 stsp 	printf("%c  %s\n", status, path);
 448 be7061eb 2018-12-30 stsp }
 449 be7061eb 2018-12-30 stsp 
 450 be7061eb 2018-12-30 stsp static const struct got_error *
 451 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
 452 507dc3bb 2018-12-29 stsp {
 453 507dc3bb 2018-12-29 stsp 	const struct got_error *error = NULL;
 454 507dc3bb 2018-12-29 stsp 	struct got_repository *repo = NULL;
 455 507dc3bb 2018-12-29 stsp 	struct got_worktree *worktree = NULL;
 456 507dc3bb 2018-12-29 stsp 	char *worktree_path = NULL;
 457 507dc3bb 2018-12-29 stsp 	struct got_object_id *commit_id = NULL;
 458 9c4b8182 2019-01-02 stsp 	char *commit_id_str = NULL;
 459 784955db 2019-01-12 stsp 	int ch, did_something = 0;
 460 507dc3bb 2018-12-29 stsp 
 461 507dc3bb 2018-12-29 stsp 	while ((ch = getopt(argc, argv, "c:")) != -1) {
 462 507dc3bb 2018-12-29 stsp 		switch (ch) {
 463 507dc3bb 2018-12-29 stsp 		case 'c':
 464 9c4b8182 2019-01-02 stsp 			commit_id_str = strdup(optarg);
 465 9c4b8182 2019-01-02 stsp 			if (commit_id_str == NULL)
 466 9c4b8182 2019-01-02 stsp 				return got_error_from_errno();
 467 507dc3bb 2018-12-29 stsp 			break;
 468 507dc3bb 2018-12-29 stsp 		default:
 469 2deda0b9 2019-03-07 stsp 			usage_update();
 470 507dc3bb 2018-12-29 stsp 			/* NOTREACHED */
 471 507dc3bb 2018-12-29 stsp 		}
 472 507dc3bb 2018-12-29 stsp 	}
 473 507dc3bb 2018-12-29 stsp 
 474 507dc3bb 2018-12-29 stsp 	argc -= optind;
 475 507dc3bb 2018-12-29 stsp 	argv += optind;
 476 507dc3bb 2018-12-29 stsp 
 477 507dc3bb 2018-12-29 stsp #ifndef PROFILE
 478 68ed9ba5 2019-02-10 stsp 	if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
 479 68ed9ba5 2019-02-10 stsp 	    "unveil", NULL) == -1)
 480 507dc3bb 2018-12-29 stsp 		err(1, "pledge");
 481 507dc3bb 2018-12-29 stsp #endif
 482 507dc3bb 2018-12-29 stsp 	if (argc == 0) {
 483 507dc3bb 2018-12-29 stsp 		worktree_path = getcwd(NULL, 0);
 484 507dc3bb 2018-12-29 stsp 		if (worktree_path == NULL) {
 485 507dc3bb 2018-12-29 stsp 			error = got_error_from_errno();
 486 507dc3bb 2018-12-29 stsp 			goto done;
 487 507dc3bb 2018-12-29 stsp 		}
 488 507dc3bb 2018-12-29 stsp 	} else if (argc == 1) {
 489 507dc3bb 2018-12-29 stsp 		worktree_path = realpath(argv[0], NULL);
 490 507dc3bb 2018-12-29 stsp 		if (worktree_path == NULL) {
 491 507dc3bb 2018-12-29 stsp 			error = got_error_from_errno();
 492 507dc3bb 2018-12-29 stsp 			goto done;
 493 507dc3bb 2018-12-29 stsp 		}
 494 507dc3bb 2018-12-29 stsp 	} else
 495 507dc3bb 2018-12-29 stsp 		usage_update();
 496 507dc3bb 2018-12-29 stsp 
 497 507dc3bb 2018-12-29 stsp 	error = got_worktree_open(&worktree, worktree_path);
 498 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 499 507dc3bb 2018-12-29 stsp 		goto done;
 500 507dc3bb 2018-12-29 stsp 
 501 507dc3bb 2018-12-29 stsp 	error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
 502 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 503 507dc3bb 2018-12-29 stsp 		goto done;
 504 507dc3bb 2018-12-29 stsp 
 505 97430839 2019-03-11 stsp 	error = apply_unveil(got_repo_get_path(repo), 0,
 506 8c02d095 2019-02-05 stsp 	    got_worktree_get_root_path(worktree));
 507 0266afb7 2019-01-04 stsp 	if (error)
 508 0266afb7 2019-01-04 stsp 		goto done;
 509 0266afb7 2019-01-04 stsp 
 510 507dc3bb 2018-12-29 stsp 	if (commit_id_str == NULL) {
 511 507dc3bb 2018-12-29 stsp 		struct got_reference *head_ref;
 512 507dc3bb 2018-12-29 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 513 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 514 507dc3bb 2018-12-29 stsp 			goto done;
 515 507dc3bb 2018-12-29 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
 516 9c4b8182 2019-01-02 stsp 		if (error != NULL)
 517 9c4b8182 2019-01-02 stsp 			goto done;
 518 9c4b8182 2019-01-02 stsp 		error = got_object_id_str(&commit_id_str, commit_id);
 519 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 520 507dc3bb 2018-12-29 stsp 			goto done;
 521 507dc3bb 2018-12-29 stsp 	} else {
 522 507dc3bb 2018-12-29 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
 523 507dc3bb 2018-12-29 stsp 		    commit_id_str);
 524 507dc3bb 2018-12-29 stsp 		if (error != NULL)
 525 507dc3bb 2018-12-29 stsp 			goto done;
 526 507dc3bb 2018-12-29 stsp 	}
 527 35c965b2 2018-12-29 stsp 
 528 be7061eb 2018-12-30 stsp 	error = check_ancestry(worktree, commit_id, repo);
 529 be7061eb 2018-12-30 stsp 	if (error != NULL)
 530 be7061eb 2018-12-30 stsp 		goto done;
 531 507dc3bb 2018-12-29 stsp 
 532 507dc3bb 2018-12-29 stsp 	if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
 533 507dc3bb 2018-12-29 stsp 	    commit_id) != 0) {
 534 507dc3bb 2018-12-29 stsp 		error = got_worktree_set_base_commit_id(worktree, repo,
 535 507dc3bb 2018-12-29 stsp 		    commit_id);
 536 507dc3bb 2018-12-29 stsp 		if (error)
 537 507dc3bb 2018-12-29 stsp 			goto done;
 538 507dc3bb 2018-12-29 stsp 	}
 539 507dc3bb 2018-12-29 stsp 
 540 507dc3bb 2018-12-29 stsp 	error = got_worktree_checkout_files(worktree, repo,
 541 6bad629b 2019-02-04 stsp 	    update_progress, &did_something, check_cancelled, NULL);
 542 507dc3bb 2018-12-29 stsp 	if (error != NULL)
 543 507dc3bb 2018-12-29 stsp 		goto done;
 544 9c4b8182 2019-01-02 stsp 
 545 784955db 2019-01-12 stsp 	if (did_something)
 546 784955db 2019-01-12 stsp 		printf("Updated to commit %s\n", commit_id_str);
 547 784955db 2019-01-12 stsp 	else
 548 784955db 2019-01-12 stsp 		printf("Already up-to-date\n");
 549 507dc3bb 2018-12-29 stsp done:
 550 c09a553d 2018-03-12 stsp 	free(worktree_path);
 551 507dc3bb 2018-12-29 stsp 	free(commit_id);
 552 9c4b8182 2019-01-02 stsp 	free(commit_id_str);
 553 c09a553d 2018-03-12 stsp 	return error;
 554 c09a553d 2018-03-12 stsp }
 555 c09a553d 2018-03-12 stsp 
 556 f42b1b34 2018-03-12 stsp static const struct got_error *
 557 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
 558 c0cc5c62 2018-10-18 stsp     int diff_context, struct got_repository *repo)
 559 5c860e29 2018-03-12 stsp {
 560 f42b1b34 2018-03-12 stsp 	const struct got_error *err = NULL;
 561 79109fed 2018-03-27 stsp 	struct got_tree_object *tree1 = NULL, *tree2;
 562 79f35eb3 2018-06-11 stsp 	struct got_object_qid *qid;
 563 0f2b3dca 2018-12-22 stsp 	char *id_str1 = NULL, *id_str2;
 564 79109fed 2018-03-27 stsp 
 565 45d799e2 2018-12-23 stsp 	err = got_object_open_as_tree(&tree2, repo,
 566 45d799e2 2018-12-23 stsp 	    got_object_commit_get_tree_id(commit));
 567 79109fed 2018-03-27 stsp 	if (err)
 568 79109fed 2018-03-27 stsp 		return err;
 569 79109fed 2018-03-27 stsp 
 570 45d799e2 2018-12-23 stsp 	qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
 571 79f35eb3 2018-06-11 stsp 	if (qid != NULL) {
 572 79109fed 2018-03-27 stsp 		struct got_commit_object *pcommit;
 573 79109fed 2018-03-27 stsp 
 574 117e771c 2018-07-23 stsp 		err = got_object_open_as_commit(&pcommit, repo, qid->id);
 575 79109fed 2018-03-27 stsp 		if (err)
 576 79109fed 2018-03-27 stsp 			return err;
 577 79109fed 2018-03-27 stsp 
 578 45d799e2 2018-12-23 stsp 		err = got_object_open_as_tree(&tree1, repo,
 579 45d799e2 2018-12-23 stsp 		    got_object_commit_get_tree_id(pcommit));
 580 79109fed 2018-03-27 stsp 		got_object_commit_close(pcommit);
 581 0f2b3dca 2018-12-22 stsp 		if (err)
 582 0f2b3dca 2018-12-22 stsp 			return err;
 583 0f2b3dca 2018-12-22 stsp 
 584 0f2b3dca 2018-12-22 stsp 		err = got_object_id_str(&id_str1, qid->id);
 585 79109fed 2018-03-27 stsp 		if (err)
 586 79109fed 2018-03-27 stsp 			return err;
 587 79109fed 2018-03-27 stsp 	}
 588 79109fed 2018-03-27 stsp 
 589 0f2b3dca 2018-12-22 stsp 	err = got_object_id_str(&id_str2, id);
 590 0f2b3dca 2018-12-22 stsp 	if (err)
 591 0f2b3dca 2018-12-22 stsp 		goto done;
 592 0f2b3dca 2018-12-22 stsp 
 593 56765ebb 2018-12-23 stsp 	printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
 594 54156555 2018-12-24 stsp 	err = got_diff_tree(tree1, tree2, "", "", diff_context, repo, stdout);
 595 0f2b3dca 2018-12-22 stsp done:
 596 79109fed 2018-03-27 stsp 	if (tree1)
 597 79109fed 2018-03-27 stsp 		got_object_tree_close(tree1);
 598 79109fed 2018-03-27 stsp 	got_object_tree_close(tree2);
 599 0f2b3dca 2018-12-22 stsp 	free(id_str1);
 600 0f2b3dca 2018-12-22 stsp 	free(id_str2);
 601 79109fed 2018-03-27 stsp 	return err;
 602 79109fed 2018-03-27 stsp }
 603 79109fed 2018-03-27 stsp 
 604 4bb494d5 2018-06-16 stsp static char *
 605 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
 606 6c281f94 2018-06-11 stsp {
 607 6c281f94 2018-06-11 stsp 	char *p, *s = ctime_r(time, datebuf);
 608 6c281f94 2018-06-11 stsp 	p = strchr(s, '\n');
 609 6c281f94 2018-06-11 stsp 	if (p)
 610 6c281f94 2018-06-11 stsp 		*p = '\0';
 611 6c281f94 2018-06-11 stsp 	return s;
 612 6c281f94 2018-06-11 stsp }
 613 6c281f94 2018-06-11 stsp 
 614 79109fed 2018-03-27 stsp static const struct got_error *
 615 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
 616 199a4027 2019-02-02 stsp     struct got_repository *repo, int show_patch, int diff_context,
 617 199a4027 2019-02-02 stsp     struct got_reflist_head *refs)
 618 79109fed 2018-03-27 stsp {
 619 79109fed 2018-03-27 stsp 	const struct got_error *err = NULL;
 620 621015ac 2018-07-23 stsp 	char *id_str, *datestr, *logmsg0, *logmsg, *line;
 621 6c281f94 2018-06-11 stsp 	char datebuf[26];
 622 45d799e2 2018-12-23 stsp 	time_t committer_time;
 623 45d799e2 2018-12-23 stsp 	const char *author, *committer;
 624 199a4027 2019-02-02 stsp 	char *refs_str = NULL;
 625 199a4027 2019-02-02 stsp 	struct got_reflist_entry *re;
 626 5c860e29 2018-03-12 stsp 
 627 199a4027 2019-02-02 stsp 	SIMPLEQ_FOREACH(re, refs, entry) {
 628 199a4027 2019-02-02 stsp 		char *s;
 629 199a4027 2019-02-02 stsp 		const char *name;
 630 199a4027 2019-02-02 stsp 		if (got_object_id_cmp(re->id, id) != 0)
 631 199a4027 2019-02-02 stsp 			continue;
 632 199a4027 2019-02-02 stsp 		name = got_ref_get_name(re->ref);
 633 d9498b20 2019-02-05 stsp 		if (strcmp(name, GOT_REF_HEAD) == 0)
 634 d9498b20 2019-02-05 stsp 			continue;
 635 199a4027 2019-02-02 stsp 		if (strncmp(name, "refs/", 5) == 0)
 636 199a4027 2019-02-02 stsp 			name += 5;
 637 7143d404 2019-03-12 stsp 		if (strncmp(name, "got/", 4) == 0)
 638 7143d404 2019-03-12 stsp 			continue;
 639 e34f9ed6 2019-02-02 stsp 		if (strncmp(name, "heads/", 6) == 0)
 640 e34f9ed6 2019-02-02 stsp 			name += 6;
 641 141c2bff 2019-02-04 stsp 		if (strncmp(name, "remotes/", 8) == 0)
 642 141c2bff 2019-02-04 stsp 			name += 8;
 643 199a4027 2019-02-02 stsp 		s = refs_str;
 644 199a4027 2019-02-02 stsp 		if (asprintf(&refs_str, "%s%s%s", s ? s : "", s ? ", " : "",
 645 199a4027 2019-02-02 stsp 		    name) == -1) {
 646 199a4027 2019-02-02 stsp 			err = got_error_from_errno();
 647 199a4027 2019-02-02 stsp 			free(s);
 648 199a4027 2019-02-02 stsp 			break;
 649 199a4027 2019-02-02 stsp 		}
 650 199a4027 2019-02-02 stsp 		free(s);
 651 199a4027 2019-02-02 stsp 	}
 652 832c249c 2018-06-10 stsp 	err = got_object_id_str(&id_str, id);
 653 8bf5b3c9 2018-03-17 stsp 	if (err)
 654 8bf5b3c9 2018-03-17 stsp 		return err;
 655 788c352e 2018-06-16 stsp 
 656 33d869be 2018-12-22 stsp 	printf("-----------------------------------------------\n");
 657 199a4027 2019-02-02 stsp 	printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
 658 199a4027 2019-02-02 stsp 	    refs_str ? refs_str : "", refs_str ? ")" : "");
 659 832c249c 2018-06-10 stsp 	free(id_str);
 660 d3d493d7 2019-02-21 stsp 	id_str = NULL;
 661 d3d493d7 2019-02-21 stsp 	free(refs_str);
 662 d3d493d7 2019-02-21 stsp 	refs_str = NULL;
 663 45d799e2 2018-12-23 stsp 	printf("from: %s\n", got_object_commit_get_author(commit));
 664 45d799e2 2018-12-23 stsp 	committer_time = got_object_commit_get_committer_time(commit);
 665 45d799e2 2018-12-23 stsp 	datestr = get_datestr(&committer_time, datebuf);
 666 dab5fe87 2018-09-14 stsp 	printf("date: %s UTC\n", datestr);
 667 45d799e2 2018-12-23 stsp 	author = got_object_commit_get_author(commit);
 668 45d799e2 2018-12-23 stsp 	committer = got_object_commit_get_committer(commit);
 669 45d799e2 2018-12-23 stsp 	if (strcmp(author, committer) != 0)
 670 45d799e2 2018-12-23 stsp 		printf("via: %s\n", committer);
 671 45d799e2 2018-12-23 stsp 	if (got_object_commit_get_nparents(commit) > 1) {
 672 45d799e2 2018-12-23 stsp 		const struct got_object_id_queue *parent_ids;
 673 79f35eb3 2018-06-11 stsp 		struct got_object_qid *qid;
 674 3fe1abad 2018-06-10 stsp 		int n = 1;
 675 45d799e2 2018-12-23 stsp 		parent_ids = got_object_commit_get_parent_ids(commit);
 676 45d799e2 2018-12-23 stsp 		SIMPLEQ_FOREACH(qid, parent_ids, entry) {
 677 79f35eb3 2018-06-11 stsp 			err = got_object_id_str(&id_str, qid->id);
 678 a0603db2 2018-06-10 stsp 			if (err)
 679 a0603db2 2018-06-10 stsp 				return err;
 680 3fe1abad 2018-06-10 stsp 			printf("parent %d: %s\n", n++, id_str);
 681 a0603db2 2018-06-10 stsp 			free(id_str);
 682 a0603db2 2018-06-10 stsp 		}
 683 a0603db2 2018-06-10 stsp 	}
 684 832c249c 2018-06-10 stsp 
 685 45d799e2 2018-12-23 stsp 	logmsg0 = strdup(got_object_commit_get_logmsg(commit));
 686 621015ac 2018-07-23 stsp 	if (logmsg0 == NULL)
 687 832c249c 2018-06-10 stsp 		return got_error_from_errno();
 688 8bf5b3c9 2018-03-17 stsp 
 689 621015ac 2018-07-23 stsp 	logmsg = logmsg0;
 690 832c249c 2018-06-10 stsp 	do {
 691 832c249c 2018-06-10 stsp 		line = strsep(&logmsg, "\n");
 692 832c249c 2018-06-10 stsp 		if (line)
 693 832c249c 2018-06-10 stsp 			printf(" %s\n", line);
 694 832c249c 2018-06-10 stsp 	} while (line);
 695 621015ac 2018-07-23 stsp 	free(logmsg0);
 696 832c249c 2018-06-10 stsp 
 697 971751ac 2018-03-27 stsp 	if (show_patch) {
 698 c0cc5c62 2018-10-18 stsp 		err = print_patch(commit, id, diff_context, repo);
 699 971751ac 2018-03-27 stsp 		if (err == 0)
 700 971751ac 2018-03-27 stsp 			printf("\n");
 701 971751ac 2018-03-27 stsp 	}
 702 07862c20 2018-09-15 stsp 
 703 cbe7f848 2019-02-11 stsp 	if (fflush(stdout) != 0 && err == NULL)
 704 cbe7f848 2019-02-11 stsp 		err = got_error_from_errno();
 705 07862c20 2018-09-15 stsp 	return err;
 706 f42b1b34 2018-03-12 stsp }
 707 5c860e29 2018-03-12 stsp 
 708 f42b1b34 2018-03-12 stsp static const struct got_error *
 709 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
 710 15a94983 2018-12-23 stsp     char *path, int show_patch, int diff_context, int limit,
 711 199a4027 2019-02-02 stsp     int first_parent_traversal, struct got_reflist_head *refs)
 712 f42b1b34 2018-03-12 stsp {
 713 f42b1b34 2018-03-12 stsp 	const struct got_error *err;
 714 372ccdbb 2018-06-10 stsp 	struct got_commit_graph *graph;
 715 372ccdbb 2018-06-10 stsp 
 716 31cedeaf 2018-09-15 stsp 	err = got_commit_graph_open(&graph, root_id, path,
 717 31cedeaf 2018-09-15 stsp 	    first_parent_traversal, repo);
 718 f42b1b34 2018-03-12 stsp 	if (err)
 719 f42b1b34 2018-03-12 stsp 		return err;
 720 31cedeaf 2018-09-15 stsp 	err = got_commit_graph_iter_start(graph, root_id, repo);
 721 372ccdbb 2018-06-10 stsp 	if (err)
 722 fcc85cad 2018-07-22 stsp 		goto done;
 723 31cedeaf 2018-09-15 stsp 	while (1) {
 724 372ccdbb 2018-06-10 stsp 		struct got_commit_object *commit;
 725 372ccdbb 2018-06-10 stsp 		struct got_object_id *id;
 726 8bf5b3c9 2018-03-17 stsp 
 727 84453469 2018-11-11 stsp 		if (sigint_received || sigpipe_received)
 728 84453469 2018-11-11 stsp 			break;
 729 84453469 2018-11-11 stsp 
 730 b43fbaa0 2018-06-11 stsp 		err = got_commit_graph_iter_next(&id, graph);
 731 372ccdbb 2018-06-10 stsp 		if (err) {
 732 9ba79e04 2018-06-11 stsp 			if (err->code == GOT_ERR_ITER_COMPLETED) {
 733 9ba79e04 2018-06-11 stsp 				err = NULL;
 734 9ba79e04 2018-06-11 stsp 				break;
 735 9ba79e04 2018-06-11 stsp 			}
 736 372ccdbb 2018-06-10 stsp 			if (err->code != GOT_ERR_ITER_NEED_MORE)
 737 372ccdbb 2018-06-10 stsp 				break;
 738 31cedeaf 2018-09-15 stsp 			err = got_commit_graph_fetch_commits(graph, 1, repo);
 739 372ccdbb 2018-06-10 stsp 			if (err)
 740 372ccdbb 2018-06-10 stsp 				break;
 741 372ccdbb 2018-06-10 stsp 			else
 742 372ccdbb 2018-06-10 stsp 				continue;
 743 7e665116 2018-04-02 stsp 		}
 744 b43fbaa0 2018-06-11 stsp 		if (id == NULL)
 745 7e665116 2018-04-02 stsp 			break;
 746 b43fbaa0 2018-06-11 stsp 
 747 f8e900f3 2018-06-11 stsp 		err = got_object_open_as_commit(&commit, repo, id);
 748 b43fbaa0 2018-06-11 stsp 		if (err)
 749 fcc85cad 2018-07-22 stsp 			break;
 750 199a4027 2019-02-02 stsp 		err = print_commit(commit, id, repo, show_patch, diff_context,
 751 199a4027 2019-02-02 stsp 		    refs);
 752 b43fbaa0 2018-06-11 stsp 		got_object_commit_close(commit);
 753 372ccdbb 2018-06-10 stsp 		if (err || (limit && --limit == 0))
 754 7e665116 2018-04-02 stsp 			break;
 755 31cedeaf 2018-09-15 stsp 	}
 756 fcc85cad 2018-07-22 stsp done:
 757 372ccdbb 2018-06-10 stsp 	got_commit_graph_close(graph);
 758 f42b1b34 2018-03-12 stsp 	return err;
 759 f42b1b34 2018-03-12 stsp }
 760 5c860e29 2018-03-12 stsp 
 761 4ed7e80c 2018-05-20 stsp __dead static void
 762 6f3d1eb0 2018-03-12 stsp usage_log(void)
 763 6f3d1eb0 2018-03-12 stsp {
 764 c0cc5c62 2018-10-18 stsp 	fprintf(stderr, "usage: %s log [-c commit] [-C number] [-f] [ -l N ] [-p] "
 765 04ca23f4 2018-07-16 stsp 	    "[-r repository-path] [path]\n", getprogname());
 766 6f3d1eb0 2018-03-12 stsp 	exit(1);
 767 6f3d1eb0 2018-03-12 stsp }
 768 6f3d1eb0 2018-03-12 stsp 
 769 4ed7e80c 2018-05-20 stsp static const struct got_error *
 770 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
 771 f42b1b34 2018-03-12 stsp {
 772 f42b1b34 2018-03-12 stsp 	const struct got_error *error;
 773 04ca23f4 2018-07-16 stsp 	struct got_repository *repo = NULL;
 774 cffc0aa4 2019-02-05 stsp 	struct got_worktree *worktree = NULL;
 775 15a94983 2018-12-23 stsp 	struct got_commit_object *commit = NULL;
 776 3235492e 2018-04-01 stsp 	struct got_object_id *id = NULL;
 777 04ca23f4 2018-07-16 stsp 	char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
 778 d142fc45 2018-04-01 stsp 	char *start_commit = NULL;
 779 c0cc5c62 2018-10-18 stsp 	int diff_context = 3, ch;
 780 1fd6d7ea 2018-06-13 stsp 	int show_patch = 0, limit = 0, first_parent_traversal = 0;
 781 64a96a6d 2018-04-01 stsp 	const char *errstr;
 782 199a4027 2019-02-02 stsp 	struct got_reflist_head refs;
 783 5c860e29 2018-03-12 stsp 
 784 6715a751 2018-03-16 stsp #ifndef PROFILE
 785 6098196c 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
 786 6098196c 2019-01-04 stsp 	    NULL)
 787 ad242220 2018-09-08 stsp 	    == -1)
 788 f42b1b34 2018-03-12 stsp 		err(1, "pledge");
 789 6715a751 2018-03-16 stsp #endif
 790 79109fed 2018-03-27 stsp 
 791 c0cc5c62 2018-10-18 stsp 	while ((ch = getopt(argc, argv, "pc:C:l:fr:")) != -1) {
 792 79109fed 2018-03-27 stsp 		switch (ch) {
 793 79109fed 2018-03-27 stsp 		case 'p':
 794 79109fed 2018-03-27 stsp 			show_patch = 1;
 795 d142fc45 2018-04-01 stsp 			break;
 796 d142fc45 2018-04-01 stsp 		case 'c':
 797 d142fc45 2018-04-01 stsp 			start_commit = optarg;
 798 64a96a6d 2018-04-01 stsp 			break;
 799 c0cc5c62 2018-10-18 stsp 		case 'C':
 800 4a8520aa 2018-10-18 stsp 			diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
 801 4a8520aa 2018-10-18 stsp 			    &errstr);
 802 c0cc5c62 2018-10-18 stsp 			if (errstr != NULL)
 803 c0cc5c62 2018-10-18 stsp 				err(1, "-C option %s", errstr);
 804 c0cc5c62 2018-10-18 stsp 			break;
 805 64a96a6d 2018-04-01 stsp 		case 'l':
 806 64a96a6d 2018-04-01 stsp 			limit = strtonum(optarg, 1, INT_MAX, &errstr);
 807 64a96a6d 2018-04-01 stsp 			if (errstr != NULL)
 808 64a96a6d 2018-04-01 stsp 				err(1, "-l option %s", errstr);
 809 79109fed 2018-03-27 stsp 			break;
 810 0ed6ed4c 2018-06-13 stsp 		case 'f':
 811 0ed6ed4c 2018-06-13 stsp 			first_parent_traversal = 1;
 812 a0603db2 2018-06-10 stsp 			break;
 813 04ca23f4 2018-07-16 stsp 		case 'r':
 814 04ca23f4 2018-07-16 stsp 			repo_path = realpath(optarg, NULL);
 815 04ca23f4 2018-07-16 stsp 			if (repo_path == NULL)
 816 04ca23f4 2018-07-16 stsp 				err(1, "-r option");
 817 04ca23f4 2018-07-16 stsp 			break;
 818 79109fed 2018-03-27 stsp 		default:
 819 2deda0b9 2019-03-07 stsp 			usage_log();
 820 79109fed 2018-03-27 stsp 			/* NOTREACHED */
 821 79109fed 2018-03-27 stsp 		}
 822 79109fed 2018-03-27 stsp 	}
 823 79109fed 2018-03-27 stsp 
 824 79109fed 2018-03-27 stsp 	argc -= optind;
 825 79109fed 2018-03-27 stsp 	argv += optind;
 826 79109fed 2018-03-27 stsp 
 827 04ca23f4 2018-07-16 stsp 	if (argc == 0)
 828 04ca23f4 2018-07-16 stsp 		path = strdup("");
 829 04ca23f4 2018-07-16 stsp 	else if (argc == 1)
 830 04ca23f4 2018-07-16 stsp 		path = strdup(argv[0]);
 831 04ca23f4 2018-07-16 stsp 	else
 832 6f3d1eb0 2018-03-12 stsp 		usage_log();
 833 04ca23f4 2018-07-16 stsp 	if (path == NULL)
 834 04ca23f4 2018-07-16 stsp 		return got_error_from_errno();
 835 f42b1b34 2018-03-12 stsp 
 836 04ca23f4 2018-07-16 stsp 	cwd = getcwd(NULL, 0);
 837 04ca23f4 2018-07-16 stsp 	if (cwd == NULL) {
 838 04ca23f4 2018-07-16 stsp 		error = got_error_from_errno();
 839 04ca23f4 2018-07-16 stsp 		goto done;
 840 04ca23f4 2018-07-16 stsp 	}
 841 cffc0aa4 2019-02-05 stsp 
 842 cffc0aa4 2019-02-05 stsp 	error = got_worktree_open(&worktree, cwd);
 843 cffc0aa4 2019-02-05 stsp 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
 844 cffc0aa4 2019-02-05 stsp 		goto done;
 845 cffc0aa4 2019-02-05 stsp 	error = NULL;
 846 cffc0aa4 2019-02-05 stsp 
 847 cffc0aa4 2019-02-05 stsp 	repo_path = worktree ?
 848 cffc0aa4 2019-02-05 stsp 	    strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
 849 04ca23f4 2018-07-16 stsp 	if (repo_path == NULL) {
 850 cffc0aa4 2019-02-05 stsp 		error = got_error_from_errno();
 851 cffc0aa4 2019-02-05 stsp 		goto done;
 852 04ca23f4 2018-07-16 stsp 	}
 853 04ca23f4 2018-07-16 stsp 
 854 d0eebce4 2019-03-11 stsp 	error = apply_unveil(repo_path, 1,
 855 efcae1c6 2019-03-07 stsp 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
 856 6098196c 2019-01-04 stsp 	if (error)
 857 6098196c 2019-01-04 stsp 		goto done;
 858 6098196c 2019-01-04 stsp 
 859 f42b1b34 2018-03-12 stsp 	error = got_repo_open(&repo, repo_path);
 860 d7d4f210 2018-03-12 stsp 	if (error != NULL)
 861 04ca23f4 2018-07-16 stsp 		goto done;
 862 f42b1b34 2018-03-12 stsp 
 863 d142fc45 2018-04-01 stsp 	if (start_commit == NULL) {
 864 3235492e 2018-04-01 stsp 		struct got_reference *head_ref;
 865 3235492e 2018-04-01 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
 866 3235492e 2018-04-01 stsp 		if (error != NULL)
 867 3235492e 2018-04-01 stsp 			return error;
 868 3235492e 2018-04-01 stsp 		error = got_ref_resolve(&id, repo, head_ref);
 869 3235492e 2018-04-01 stsp 		got_ref_close(head_ref);
 870 3235492e 2018-04-01 stsp 		if (error != NULL)
 871 3235492e 2018-04-01 stsp 			return error;
 872 15a94983 2018-12-23 stsp 		error = got_object_open_as_commit(&commit, repo, id);
 873 3235492e 2018-04-01 stsp 	} else {
 874 d1f2edc9 2018-06-13 stsp 		struct got_reference *ref;
 875 d1f2edc9 2018-06-13 stsp 		error = got_ref_open(&ref, repo, start_commit);
 876 3235492e 2018-04-01 stsp 		if (error == NULL) {
 877 1caad61b 2019-02-01 stsp 			int obj_type;
 878 d1f2edc9 2018-06-13 stsp 			error = got_ref_resolve(&id, repo, ref);
 879 d1f2edc9 2018-06-13 stsp 			got_ref_close(ref);
 880 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 881 1caad61b 2019-02-01 stsp 				goto done;
 882 1caad61b 2019-02-01 stsp 			error = got_object_get_type(&obj_type, repo, id);
 883 1caad61b 2019-02-01 stsp 			if (error != NULL)
 884 1caad61b 2019-02-01 stsp 				goto done;
 885 1caad61b 2019-02-01 stsp 			if (obj_type == GOT_OBJ_TYPE_TAG) {
 886 1caad61b 2019-02-01 stsp 				struct got_tag_object *tag;
 887 1caad61b 2019-02-01 stsp 				error = got_object_open_as_tag(&tag, repo, id);
 888 1caad61b 2019-02-01 stsp 				if (error != NULL)
 889 1caad61b 2019-02-01 stsp 					goto done;
 890 1caad61b 2019-02-01 stsp 				if (got_object_tag_get_object_type(tag) !=
 891 1caad61b 2019-02-01 stsp 				    GOT_OBJ_TYPE_COMMIT) {
 892 1caad61b 2019-02-01 stsp 					got_object_tag_close(tag);
 893 1caad61b 2019-02-01 stsp 					error = got_error(GOT_ERR_OBJ_TYPE);
 894 1caad61b 2019-02-01 stsp 					goto done;
 895 1caad61b 2019-02-01 stsp 				}
 896 1caad61b 2019-02-01 stsp 				free(id);
 897 1caad61b 2019-02-01 stsp 				id = got_object_id_dup(
 898 1caad61b 2019-02-01 stsp 				    got_object_tag_get_object_id(tag));
 899 1caad61b 2019-02-01 stsp 				if (id == NULL)
 900 1caad61b 2019-02-01 stsp 					error = got_error_from_errno();
 901 1caad61b 2019-02-01 stsp 				got_object_tag_close(tag);
 902 1caad61b 2019-02-01 stsp 				if (error)
 903 1caad61b 2019-02-01 stsp 					goto done;
 904 1caad61b 2019-02-01 stsp 			} else if (obj_type != GOT_OBJ_TYPE_COMMIT) {
 905 1caad61b 2019-02-01 stsp 				error = got_error(GOT_ERR_OBJ_TYPE);
 906 1caad61b 2019-02-01 stsp 				goto done;
 907 1caad61b 2019-02-01 stsp 			}
 908 15a94983 2018-12-23 stsp 			error = got_object_open_as_commit(&commit, repo, id);
 909 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 910 1caad61b 2019-02-01 stsp 				goto done;
 911 d1f2edc9 2018-06-13 stsp 		}
 912 15a94983 2018-12-23 stsp 		if (commit == NULL) {
 913 15a94983 2018-12-23 stsp 			error = got_object_resolve_id_str(&id, repo,
 914 d1f2edc9 2018-06-13 stsp 			    start_commit);
 915 d1f2edc9 2018-06-13 stsp 			if (error != NULL)
 916 d1f2edc9 2018-06-13 stsp 				return error;
 917 3235492e 2018-04-01 stsp 		}
 918 3235492e 2018-04-01 stsp 	}
 919 d7d4f210 2018-03-12 stsp 	if (error != NULL)
 920 04ca23f4 2018-07-16 stsp 		goto done;
 921 04ca23f4 2018-07-16 stsp 
 922 23721109 2018-10-22 stsp 	error = got_repo_map_path(&in_repo_path, repo, path, 1);
 923 04ca23f4 2018-07-16 stsp 	if (error != NULL)
 924 04ca23f4 2018-07-16 stsp 		goto done;
 925 04ca23f4 2018-07-16 stsp 	if (in_repo_path) {
 926 04ca23f4 2018-07-16 stsp 		free(path);
 927 04ca23f4 2018-07-16 stsp 		path = in_repo_path;
 928 04ca23f4 2018-07-16 stsp 	}
 929 199a4027 2019-02-02 stsp 
 930 199a4027 2019-02-02 stsp 	SIMPLEQ_INIT(&refs);
 931 199a4027 2019-02-02 stsp 	error = got_ref_list(&refs, repo);
 932 199a4027 2019-02-02 stsp 	if (error)
 933 199a4027 2019-02-02 stsp 		goto done;
 934 04ca23f4 2018-07-16 stsp 
 935 15a94983 2018-12-23 stsp 	error = print_commits(id, repo, path, show_patch,
 936 199a4027 2019-02-02 stsp 	    diff_context, limit, first_parent_traversal, &refs);
 937 04ca23f4 2018-07-16 stsp done:
 938 04ca23f4 2018-07-16 stsp 	free(path);
 939 04ca23f4 2018-07-16 stsp 	free(repo_path);
 940 04ca23f4 2018-07-16 stsp 	free(cwd);
 941 f42b1b34 2018-03-12 stsp 	free(id);
 942 cffc0aa4 2019-02-05 stsp 	if (worktree)
 943 cffc0aa4 2019-02-05 stsp 		got_worktree_close(worktree);
 944 ad242220 2018-09-08 stsp 	if (repo) {
 945 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
 946 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
 947 ad242220 2018-09-08 stsp 		if (error == NULL)
 948 ad242220 2018-09-08 stsp 			error = repo_error;
 949 ad242220 2018-09-08 stsp 	}
 950 e2e879a0 2019-03-11 stsp 	got_ref_list_free(&refs);
 951 8bf5b3c9 2018-03-17 stsp 	return error;
 952 5c860e29 2018-03-12 stsp }
 953 5c860e29 2018-03-12 stsp 
 954 4ed7e80c 2018-05-20 stsp __dead static void
 955 3f8b7d6a 2018-04-01 stsp usage_diff(void)
 956 3f8b7d6a 2018-04-01 stsp {
 957 b72f483a 2019-02-05 stsp 	fprintf(stderr, "usage: %s diff [-C number] [-r repository-path] "
 958 927df6b7 2019-02-10 stsp 	    "[object1 object2 | path]\n", getprogname());
 959 3f8b7d6a 2018-04-01 stsp 	exit(1);
 960 b00d56cd 2018-04-01 stsp }
 961 b00d56cd 2018-04-01 stsp 
 962 b72f483a 2019-02-05 stsp struct print_diff_arg {
 963 b72f483a 2019-02-05 stsp 	struct got_repository *repo;
 964 b72f483a 2019-02-05 stsp 	struct got_worktree *worktree;
 965 b72f483a 2019-02-05 stsp 	int diff_context;
 966 3fc0c068 2019-02-10 stsp 	const char *id_str;
 967 3fc0c068 2019-02-10 stsp 	int header_shown;
 968 b72f483a 2019-02-05 stsp };
 969 b72f483a 2019-02-05 stsp 
 970 4ed7e80c 2018-05-20 stsp static const struct got_error *
 971 b72f483a 2019-02-05 stsp print_diff(void *arg, unsigned char status, const char *path,
 972 b72f483a 2019-02-05 stsp     struct got_object_id *id)
 973 b72f483a 2019-02-05 stsp {
 974 b72f483a 2019-02-05 stsp 	struct print_diff_arg *a = arg;
 975 b72f483a 2019-02-05 stsp 	const struct got_error *err = NULL;
 976 b72f483a 2019-02-05 stsp 	struct got_blob_object *blob1 = NULL;
 977 b72f483a 2019-02-05 stsp 	FILE *f2 = NULL;
 978 b72f483a 2019-02-05 stsp 	char *abspath = NULL;
 979 b72f483a 2019-02-05 stsp 	struct stat sb;
 980 b72f483a 2019-02-05 stsp 
 981 276262e8 2019-02-08 stsp 	if (status != GOT_STATUS_MODIFY)
 982 b72f483a 2019-02-05 stsp 		return NULL;
 983 3fc0c068 2019-02-10 stsp 
 984 3fc0c068 2019-02-10 stsp 	if (!a->header_shown) {
 985 3fc0c068 2019-02-10 stsp 		printf("diff %s %s\n", a->id_str,
 986 3fc0c068 2019-02-10 stsp 		    got_worktree_get_root_path(a->worktree));
 987 3fc0c068 2019-02-10 stsp 		a->header_shown = 1;
 988 3fc0c068 2019-02-10 stsp 	}
 989 b72f483a 2019-02-05 stsp 
 990 b72f483a 2019-02-05 stsp 	err = got_object_open_as_blob(&blob1, a->repo, id, 8192);
 991 b72f483a 2019-02-05 stsp 	if (err)
 992 b72f483a 2019-02-05 stsp 		goto done;
 993 b72f483a 2019-02-05 stsp 
 994 b72f483a 2019-02-05 stsp 	if (asprintf(&abspath, "%s/%s",
 995 b72f483a 2019-02-05 stsp 	    got_worktree_get_root_path(a->worktree), path) == -1) {
 996 b72f483a 2019-02-05 stsp 		err = got_error_from_errno();
 997 b72f483a 2019-02-05 stsp 		goto done;
 998 b72f483a 2019-02-05 stsp 	}
 999 b72f483a 2019-02-05 stsp 
1000 b72f483a 2019-02-05 stsp 	f2 = fopen(abspath, "r");
1001 b72f483a 2019-02-05 stsp 	if (f2 == NULL) {
1002 b72f483a 2019-02-05 stsp 		err = got_error_from_errno();
1003 b72f483a 2019-02-05 stsp 		goto done;
1004 b72f483a 2019-02-05 stsp 	}
1005 b72f483a 2019-02-05 stsp 	if (lstat(abspath, &sb) == -1) {
1006 b72f483a 2019-02-05 stsp 		err = got_error_from_errno();
1007 b72f483a 2019-02-05 stsp 		goto done;
1008 b72f483a 2019-02-05 stsp 	}
1009 b72f483a 2019-02-05 stsp 
1010 b72f483a 2019-02-05 stsp 	err = got_diff_blob_file(blob1, f2, sb.st_size, path, a->diff_context,
1011 b72f483a 2019-02-05 stsp 	    stdout);
1012 b72f483a 2019-02-05 stsp done:
1013 b72f483a 2019-02-05 stsp 	if (blob1)
1014 b72f483a 2019-02-05 stsp 		got_object_blob_close(blob1);
1015 fb43ecf1 2019-02-11 stsp 	if (f2 && fclose(f2) != 0 && err == NULL)
1016 fb43ecf1 2019-02-11 stsp 		err = got_error_from_errno();
1017 b72f483a 2019-02-05 stsp 	free(abspath);
1018 927df6b7 2019-02-10 stsp 	return err;
1019 927df6b7 2019-02-10 stsp }
1020 927df6b7 2019-02-10 stsp 
1021 927df6b7 2019-02-10 stsp static const struct got_error *
1022 927df6b7 2019-02-10 stsp get_status_path(char **status_path, struct got_worktree *worktree,
1023 927df6b7 2019-02-10 stsp     const char *arg)
1024 927df6b7 2019-02-10 stsp {
1025 927df6b7 2019-02-10 stsp 	const struct got_error *err = NULL;
1026 927df6b7 2019-02-10 stsp 	char *resolved, *path = NULL;
1027 927df6b7 2019-02-10 stsp 	size_t len;
1028 927df6b7 2019-02-10 stsp 
1029 927df6b7 2019-02-10 stsp 	*status_path = NULL;
1030 927df6b7 2019-02-10 stsp 
1031 927df6b7 2019-02-10 stsp 	resolved = realpath(arg, NULL);
1032 927df6b7 2019-02-10 stsp 	if (resolved == NULL)
1033 927df6b7 2019-02-10 stsp 		return got_error_from_errno();
1034 927df6b7 2019-02-10 stsp 
1035 927df6b7 2019-02-10 stsp 	if (strncmp(got_worktree_get_root_path(worktree), resolved,
1036 927df6b7 2019-02-10 stsp 	    strlen(got_worktree_get_root_path(worktree)))) {
1037 927df6b7 2019-02-10 stsp 		err = got_error(GOT_ERR_BAD_PATH);
1038 927df6b7 2019-02-10 stsp 		goto done;
1039 927df6b7 2019-02-10 stsp 	}
1040 927df6b7 2019-02-10 stsp 
1041 927df6b7 2019-02-10 stsp 	path = strdup(resolved + strlen(got_worktree_get_root_path(worktree)));
1042 927df6b7 2019-02-10 stsp 	if (path == NULL) {
1043 927df6b7 2019-02-10 stsp 		err = got_error_from_errno();
1044 927df6b7 2019-02-10 stsp 		goto done;
1045 927df6b7 2019-02-10 stsp 	}
1046 927df6b7 2019-02-10 stsp 
1047 927df6b7 2019-02-10 stsp 	/* XXX status walk can't deal with trailing slash! */
1048 927df6b7 2019-02-10 stsp 	len = strlen(path);
1049 927df6b7 2019-02-10 stsp 	while (path[len - 1] == '/') {
1050 927df6b7 2019-02-10 stsp 		path[len - 1] = '\0';
1051 927df6b7 2019-02-10 stsp 		len--;
1052 927df6b7 2019-02-10 stsp 	}
1053 927df6b7 2019-02-10 stsp done:
1054 927df6b7 2019-02-10 stsp 	free(resolved);
1055 927df6b7 2019-02-10 stsp 	if (err == NULL)
1056 927df6b7 2019-02-10 stsp 		*status_path = path;
1057 927df6b7 2019-02-10 stsp 	else
1058 927df6b7 2019-02-10 stsp 		free(path);
1059 b72f483a 2019-02-05 stsp 	return err;
1060 b72f483a 2019-02-05 stsp }
1061 b72f483a 2019-02-05 stsp 
1062 b72f483a 2019-02-05 stsp static const struct got_error *
1063 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
1064 b00d56cd 2018-04-01 stsp {
1065 b00d56cd 2018-04-01 stsp 	const struct got_error *error;
1066 b00d56cd 2018-04-01 stsp 	struct got_repository *repo = NULL;
1067 b72f483a 2019-02-05 stsp 	struct got_worktree *worktree = NULL;
1068 b72f483a 2019-02-05 stsp 	char *cwd = NULL, *repo_path = NULL;
1069 15a94983 2018-12-23 stsp 	struct got_object_id *id1 = NULL, *id2 = NULL;
1070 15a94983 2018-12-23 stsp 	char *id_str1 = NULL, *id_str2 = NULL;
1071 15a94983 2018-12-23 stsp 	int type1, type2;
1072 c0cc5c62 2018-10-18 stsp 	int diff_context = 3, ch;
1073 c0cc5c62 2018-10-18 stsp 	const char *errstr;
1074 927df6b7 2019-02-10 stsp 	char *path = NULL;
1075 b00d56cd 2018-04-01 stsp 
1076 b00d56cd 2018-04-01 stsp #ifndef PROFILE
1077 25eccc22 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1078 25eccc22 2019-01-04 stsp 	    NULL) == -1)
1079 b00d56cd 2018-04-01 stsp 		err(1, "pledge");
1080 b00d56cd 2018-04-01 stsp #endif
1081 b00d56cd 2018-04-01 stsp 
1082 b72f483a 2019-02-05 stsp 	while ((ch = getopt(argc, argv, "C:r:")) != -1) {
1083 b00d56cd 2018-04-01 stsp 		switch (ch) {
1084 c0cc5c62 2018-10-18 stsp 		case 'C':
1085 c0cc5c62 2018-10-18 stsp 			diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
1086 c0cc5c62 2018-10-18 stsp 			if (errstr != NULL)
1087 c0cc5c62 2018-10-18 stsp 				err(1, "-C option %s", errstr);
1088 c0cc5c62 2018-10-18 stsp 			break;
1089 b72f483a 2019-02-05 stsp 		case 'r':
1090 b72f483a 2019-02-05 stsp 			repo_path = realpath(optarg, NULL);
1091 b72f483a 2019-02-05 stsp 			if (repo_path == NULL)
1092 b72f483a 2019-02-05 stsp 				err(1, "-r option");
1093 b72f483a 2019-02-05 stsp 			break;
1094 b00d56cd 2018-04-01 stsp 		default:
1095 2deda0b9 2019-03-07 stsp 			usage_diff();
1096 b00d56cd 2018-04-01 stsp 			/* NOTREACHED */
1097 b00d56cd 2018-04-01 stsp 		}
1098 b00d56cd 2018-04-01 stsp 	}
1099 b00d56cd 2018-04-01 stsp 
1100 b00d56cd 2018-04-01 stsp 	argc -= optind;
1101 b00d56cd 2018-04-01 stsp 	argv += optind;
1102 b00d56cd 2018-04-01 stsp 
1103 b72f483a 2019-02-05 stsp 	cwd = getcwd(NULL, 0);
1104 b72f483a 2019-02-05 stsp 	if (cwd == NULL) {
1105 b72f483a 2019-02-05 stsp 		error = got_error_from_errno();
1106 b72f483a 2019-02-05 stsp 		goto done;
1107 b72f483a 2019-02-05 stsp 	}
1108 927df6b7 2019-02-10 stsp 	error = got_worktree_open(&worktree, cwd);
1109 927df6b7 2019-02-10 stsp 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
1110 927df6b7 2019-02-10 stsp 		goto done;
1111 927df6b7 2019-02-10 stsp 	if (argc <= 1) {
1112 927df6b7 2019-02-10 stsp 		if (worktree == NULL) {
1113 927df6b7 2019-02-10 stsp 			error = got_error(GOT_ERR_NOT_WORKTREE);
1114 927df6b7 2019-02-10 stsp 			goto done;
1115 927df6b7 2019-02-10 stsp 		}
1116 b72f483a 2019-02-05 stsp 		if (repo_path)
1117 b72f483a 2019-02-05 stsp 			errx(1,
1118 b72f483a 2019-02-05 stsp 			    "-r option can't be used when diffing a work tree");
1119 b72f483a 2019-02-05 stsp 		repo_path = strdup(got_worktree_get_repo_path(worktree));
1120 927df6b7 2019-02-10 stsp 		if (repo_path == NULL) {
1121 927df6b7 2019-02-10 stsp 			error = got_error_from_errno();
1122 927df6b7 2019-02-10 stsp 			goto done;
1123 927df6b7 2019-02-10 stsp 		}
1124 927df6b7 2019-02-10 stsp 		if (argc == 1) {
1125 927df6b7 2019-02-10 stsp 			error = get_status_path(&path, worktree, argv[0]);
1126 927df6b7 2019-02-10 stsp 			if (error)
1127 927df6b7 2019-02-10 stsp 				goto done;
1128 927df6b7 2019-02-10 stsp 		} else {
1129 927df6b7 2019-02-10 stsp 			path = strdup("");
1130 927df6b7 2019-02-10 stsp 			if (path == NULL) {
1131 927df6b7 2019-02-10 stsp 				error = got_error_from_errno();
1132 927df6b7 2019-02-10 stsp 				goto done;
1133 927df6b7 2019-02-10 stsp 			}
1134 927df6b7 2019-02-10 stsp 		}
1135 b72f483a 2019-02-05 stsp 	} else if (argc == 2) {
1136 15a94983 2018-12-23 stsp 		id_str1 = argv[0];
1137 15a94983 2018-12-23 stsp 		id_str2 = argv[1];
1138 b00d56cd 2018-04-01 stsp 	} else
1139 b00d56cd 2018-04-01 stsp 		usage_diff();
1140 25eccc22 2019-01-04 stsp 
1141 b72f483a 2019-02-05 stsp 	if (repo_path == NULL) {
1142 b72f483a 2019-02-05 stsp 		repo_path = getcwd(NULL, 0);
1143 b72f483a 2019-02-05 stsp 		if (repo_path == NULL)
1144 b72f483a 2019-02-05 stsp 			return got_error_from_errno();
1145 b72f483a 2019-02-05 stsp 	}
1146 b72f483a 2019-02-05 stsp 
1147 d0eebce4 2019-03-11 stsp 	error = apply_unveil(repo_path, 1,
1148 b72f483a 2019-02-05 stsp 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
1149 25eccc22 2019-01-04 stsp 	if (error)
1150 25eccc22 2019-01-04 stsp 		goto done;
1151 b00d56cd 2018-04-01 stsp 
1152 b00d56cd 2018-04-01 stsp 	error = got_repo_open(&repo, repo_path);
1153 76089277 2018-04-01 stsp 	free(repo_path);
1154 b00d56cd 2018-04-01 stsp 	if (error != NULL)
1155 b00d56cd 2018-04-01 stsp 		goto done;
1156 b00d56cd 2018-04-01 stsp 
1157 b72f483a 2019-02-05 stsp 	if (worktree) {
1158 b72f483a 2019-02-05 stsp 		struct print_diff_arg arg;
1159 b72f483a 2019-02-05 stsp 		char *id_str;
1160 b72f483a 2019-02-05 stsp 		error = got_object_id_str(&id_str,
1161 b72f483a 2019-02-05 stsp 		    got_worktree_get_base_commit_id(worktree));
1162 b72f483a 2019-02-05 stsp 		if (error)
1163 b72f483a 2019-02-05 stsp 			goto done;
1164 b72f483a 2019-02-05 stsp 		arg.repo = repo;
1165 b72f483a 2019-02-05 stsp 		arg.worktree = worktree;
1166 b72f483a 2019-02-05 stsp 		arg.diff_context = diff_context;
1167 3fc0c068 2019-02-10 stsp 		arg.id_str = id_str;
1168 3fc0c068 2019-02-10 stsp 		arg.header_shown = 0;
1169 b72f483a 2019-02-05 stsp 
1170 927df6b7 2019-02-10 stsp 		error = got_worktree_status(worktree, path, repo, print_diff,
1171 b72f483a 2019-02-05 stsp 		    &arg, check_cancelled, NULL);
1172 b72f483a 2019-02-05 stsp 		free(id_str);
1173 b72f483a 2019-02-05 stsp 		goto done;
1174 b72f483a 2019-02-05 stsp 	}
1175 b72f483a 2019-02-05 stsp 
1176 15a94983 2018-12-23 stsp 	error = got_object_resolve_id_str(&id1, repo, id_str1);
1177 6402fb3c 2018-09-15 stsp 	if (error)
1178 b00d56cd 2018-04-01 stsp 		goto done;
1179 b00d56cd 2018-04-01 stsp 
1180 15a94983 2018-12-23 stsp 	error = got_object_resolve_id_str(&id2, repo, id_str2);
1181 6402fb3c 2018-09-15 stsp 	if (error)
1182 b00d56cd 2018-04-01 stsp 		goto done;
1183 b00d56cd 2018-04-01 stsp 
1184 15a94983 2018-12-23 stsp 	error = got_object_get_type(&type1, repo, id1);
1185 15a94983 2018-12-23 stsp 	if (error)
1186 15a94983 2018-12-23 stsp 		goto done;
1187 15a94983 2018-12-23 stsp 
1188 15a94983 2018-12-23 stsp 	error = got_object_get_type(&type2, repo, id2);
1189 15a94983 2018-12-23 stsp 	if (error)
1190 15a94983 2018-12-23 stsp 		goto done;
1191 15a94983 2018-12-23 stsp 
1192 15a94983 2018-12-23 stsp 	if (type1 != type2) {
1193 b00d56cd 2018-04-01 stsp 		error = got_error(GOT_ERR_OBJ_TYPE);
1194 b00d56cd 2018-04-01 stsp 		goto done;
1195 b00d56cd 2018-04-01 stsp 	}
1196 b00d56cd 2018-04-01 stsp 
1197 15a94983 2018-12-23 stsp 	switch (type1) {
1198 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_BLOB:
1199 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
1200 54156555 2018-12-24 stsp 		    diff_context, repo, stdout);
1201 b00d56cd 2018-04-01 stsp 		break;
1202 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_TREE:
1203 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_trees(id1, id2, "", "",
1204 54156555 2018-12-24 stsp 		    diff_context, repo, stdout);
1205 b00d56cd 2018-04-01 stsp 		break;
1206 b00d56cd 2018-04-01 stsp 	case GOT_OBJ_TYPE_COMMIT:
1207 15a94983 2018-12-23 stsp 		printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null",
1208 15a94983 2018-12-23 stsp 		    id_str2);
1209 15a94983 2018-12-23 stsp 		error = got_diff_objects_as_commits(id1, id2, diff_context,
1210 c0cc5c62 2018-10-18 stsp 		    repo, stdout);
1211 b00d56cd 2018-04-01 stsp 		break;
1212 b00d56cd 2018-04-01 stsp 	default:
1213 b00d56cd 2018-04-01 stsp 		error = got_error(GOT_ERR_OBJ_TYPE);
1214 b00d56cd 2018-04-01 stsp 	}
1215 b00d56cd 2018-04-01 stsp 
1216 b00d56cd 2018-04-01 stsp done:
1217 15a94983 2018-12-23 stsp 	free(id1);
1218 15a94983 2018-12-23 stsp 	free(id2);
1219 927df6b7 2019-02-10 stsp 	free(path);
1220 b72f483a 2019-02-05 stsp 	if (worktree)
1221 b72f483a 2019-02-05 stsp 		got_worktree_close(worktree);
1222 ad242220 2018-09-08 stsp 	if (repo) {
1223 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
1224 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
1225 ad242220 2018-09-08 stsp 		if (error == NULL)
1226 ad242220 2018-09-08 stsp 			error = repo_error;
1227 ad242220 2018-09-08 stsp 	}
1228 b00d56cd 2018-04-01 stsp 	return error;
1229 404c43c4 2018-06-21 stsp }
1230 404c43c4 2018-06-21 stsp 
1231 404c43c4 2018-06-21 stsp __dead static void
1232 404c43c4 2018-06-21 stsp usage_blame(void)
1233 404c43c4 2018-06-21 stsp {
1234 9270e621 2019-02-05 stsp 	fprintf(stderr,
1235 9270e621 2019-02-05 stsp 	    "usage: %s blame [-c commit] [-r repository-path] path\n",
1236 404c43c4 2018-06-21 stsp 	    getprogname());
1237 404c43c4 2018-06-21 stsp 	exit(1);
1238 b00d56cd 2018-04-01 stsp }
1239 b00d56cd 2018-04-01 stsp 
1240 404c43c4 2018-06-21 stsp static const struct got_error *
1241 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
1242 404c43c4 2018-06-21 stsp {
1243 404c43c4 2018-06-21 stsp 	const struct got_error *error;
1244 404c43c4 2018-06-21 stsp 	struct got_repository *repo = NULL;
1245 0c06baac 2019-02-05 stsp 	struct got_worktree *worktree = NULL;
1246 66bea077 2018-08-02 stsp 	char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1247 404c43c4 2018-06-21 stsp 	struct got_object_id *commit_id = NULL;
1248 404c43c4 2018-06-21 stsp 	char *commit_id_str = NULL;
1249 404c43c4 2018-06-21 stsp 	int ch;
1250 404c43c4 2018-06-21 stsp 
1251 404c43c4 2018-06-21 stsp #ifndef PROFILE
1252 36e2fb66 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1253 36e2fb66 2019-01-04 stsp 	    NULL) == -1)
1254 404c43c4 2018-06-21 stsp 		err(1, "pledge");
1255 404c43c4 2018-06-21 stsp #endif
1256 404c43c4 2018-06-21 stsp 
1257 66bea077 2018-08-02 stsp 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1258 404c43c4 2018-06-21 stsp 		switch (ch) {
1259 404c43c4 2018-06-21 stsp 		case 'c':
1260 404c43c4 2018-06-21 stsp 			commit_id_str = optarg;
1261 404c43c4 2018-06-21 stsp 			break;
1262 66bea077 2018-08-02 stsp 		case 'r':
1263 66bea077 2018-08-02 stsp 			repo_path = realpath(optarg, NULL);
1264 66bea077 2018-08-02 stsp 			if (repo_path == NULL)
1265 66bea077 2018-08-02 stsp 				err(1, "-r option");
1266 66bea077 2018-08-02 stsp 			break;
1267 404c43c4 2018-06-21 stsp 		default:
1268 2deda0b9 2019-03-07 stsp 			usage_blame();
1269 404c43c4 2018-06-21 stsp 			/* NOTREACHED */
1270 404c43c4 2018-06-21 stsp 		}
1271 404c43c4 2018-06-21 stsp 	}
1272 404c43c4 2018-06-21 stsp 
1273 404c43c4 2018-06-21 stsp 	argc -= optind;
1274 404c43c4 2018-06-21 stsp 	argv += optind;
1275 404c43c4 2018-06-21 stsp 
1276 a39318fd 2018-08-02 stsp 	if (argc == 1)
1277 404c43c4 2018-06-21 stsp 		path = argv[0];
1278 a39318fd 2018-08-02 stsp 	else
1279 404c43c4 2018-06-21 stsp 		usage_blame();
1280 404c43c4 2018-06-21 stsp 
1281 66bea077 2018-08-02 stsp 	cwd = getcwd(NULL, 0);
1282 66bea077 2018-08-02 stsp 	if (cwd == NULL) {
1283 66bea077 2018-08-02 stsp 		error = got_error_from_errno();
1284 66bea077 2018-08-02 stsp 		goto done;
1285 66bea077 2018-08-02 stsp 	}
1286 66bea077 2018-08-02 stsp 	if (repo_path == NULL) {
1287 0c06baac 2019-02-05 stsp 		error = got_worktree_open(&worktree, cwd);
1288 0c06baac 2019-02-05 stsp 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
1289 66bea077 2018-08-02 stsp 			goto done;
1290 0c06baac 2019-02-05 stsp 		else
1291 0c06baac 2019-02-05 stsp 			error = NULL;
1292 0c06baac 2019-02-05 stsp 		if (worktree) {
1293 0c06baac 2019-02-05 stsp 			repo_path =
1294 0c06baac 2019-02-05 stsp 			    strdup(got_worktree_get_repo_path(worktree));
1295 0c06baac 2019-02-05 stsp 			if (repo_path == NULL)
1296 0c06baac 2019-02-05 stsp 				error = got_error_from_errno();
1297 0c06baac 2019-02-05 stsp 			if (error)
1298 0c06baac 2019-02-05 stsp 				goto done;
1299 0c06baac 2019-02-05 stsp 		} else {
1300 0c06baac 2019-02-05 stsp 			repo_path = strdup(cwd);
1301 0c06baac 2019-02-05 stsp 			if (repo_path == NULL) {
1302 0c06baac 2019-02-05 stsp 				error = got_error_from_errno();
1303 0c06baac 2019-02-05 stsp 				goto done;
1304 0c06baac 2019-02-05 stsp 			}
1305 66bea077 2018-08-02 stsp 		}
1306 66bea077 2018-08-02 stsp 	}
1307 36e2fb66 2019-01-04 stsp 
1308 d0eebce4 2019-03-11 stsp 	error = apply_unveil(repo_path, 1, NULL);
1309 36e2fb66 2019-01-04 stsp 	if (error)
1310 36e2fb66 2019-01-04 stsp 		goto done;
1311 66bea077 2018-08-02 stsp 
1312 404c43c4 2018-06-21 stsp 	error = got_repo_open(&repo, repo_path);
1313 404c43c4 2018-06-21 stsp 	if (error != NULL)
1314 404c43c4 2018-06-21 stsp 		goto done;
1315 404c43c4 2018-06-21 stsp 
1316 0c06baac 2019-02-05 stsp 	if (worktree) {
1317 6efaaa2d 2019-02-05 stsp 		const char *prefix = got_worktree_get_path_prefix(worktree);
1318 0c06baac 2019-02-05 stsp 		char *p, *worktree_subdir = cwd +
1319 0c06baac 2019-02-05 stsp 		    strlen(got_worktree_get_root_path(worktree));
1320 6efaaa2d 2019-02-05 stsp 		if (asprintf(&p, "%s%s%s%s%s",
1321 6efaaa2d 2019-02-05 stsp 		    prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
1322 6efaaa2d 2019-02-05 stsp 		    worktree_subdir, worktree_subdir[0] ? "/" : "",
1323 6efaaa2d 2019-02-05 stsp 		    path) == -1) {
1324 0c06baac 2019-02-05 stsp 			error = got_error_from_errno();
1325 0c06baac 2019-02-05 stsp 			goto done;
1326 0c06baac 2019-02-05 stsp 		}
1327 6efaaa2d 2019-02-05 stsp 		error = got_repo_map_path(&in_repo_path, repo, p, 0);
1328 0c06baac 2019-02-05 stsp 		free(p);
1329 0c06baac 2019-02-05 stsp 	} else {
1330 0c06baac 2019-02-05 stsp 		error = got_repo_map_path(&in_repo_path, repo, path, 1);
1331 0c06baac 2019-02-05 stsp 	}
1332 0c06baac 2019-02-05 stsp 	if (error)
1333 66bea077 2018-08-02 stsp 		goto done;
1334 66bea077 2018-08-02 stsp 
1335 404c43c4 2018-06-21 stsp 	if (commit_id_str == NULL) {
1336 404c43c4 2018-06-21 stsp 		struct got_reference *head_ref;
1337 404c43c4 2018-06-21 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1338 404c43c4 2018-06-21 stsp 		if (error != NULL)
1339 66bea077 2018-08-02 stsp 			goto done;
1340 404c43c4 2018-06-21 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
1341 404c43c4 2018-06-21 stsp 		got_ref_close(head_ref);
1342 404c43c4 2018-06-21 stsp 		if (error != NULL)
1343 66bea077 2018-08-02 stsp 			goto done;
1344 404c43c4 2018-06-21 stsp 	} else {
1345 15a94983 2018-12-23 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
1346 15a94983 2018-12-23 stsp 		    commit_id_str);
1347 404c43c4 2018-06-21 stsp 		if (error != NULL)
1348 66bea077 2018-08-02 stsp 			goto done;
1349 404c43c4 2018-06-21 stsp 	}
1350 404c43c4 2018-06-21 stsp 
1351 66bea077 2018-08-02 stsp 	error = got_blame(in_repo_path, commit_id, repo, stdout);
1352 404c43c4 2018-06-21 stsp done:
1353 66bea077 2018-08-02 stsp 	free(in_repo_path);
1354 66bea077 2018-08-02 stsp 	free(repo_path);
1355 66bea077 2018-08-02 stsp 	free(cwd);
1356 404c43c4 2018-06-21 stsp 	free(commit_id);
1357 0c06baac 2019-02-05 stsp 	if (worktree)
1358 0c06baac 2019-02-05 stsp 		got_worktree_close(worktree);
1359 ad242220 2018-09-08 stsp 	if (repo) {
1360 ad242220 2018-09-08 stsp 		const struct got_error *repo_error;
1361 ad242220 2018-09-08 stsp 		repo_error = got_repo_close(repo);
1362 ad242220 2018-09-08 stsp 		if (error == NULL)
1363 ad242220 2018-09-08 stsp 			error = repo_error;
1364 ad242220 2018-09-08 stsp 	}
1365 404c43c4 2018-06-21 stsp 	return error;
1366 5de5890b 2018-10-18 stsp }
1367 5de5890b 2018-10-18 stsp 
1368 5de5890b 2018-10-18 stsp __dead static void
1369 5de5890b 2018-10-18 stsp usage_tree(void)
1370 5de5890b 2018-10-18 stsp {
1371 c1669e2e 2019-01-09 stsp 	fprintf(stderr,
1372 c1669e2e 2019-01-09 stsp 	    "usage: %s tree [-c commit] [-r repository-path] [-iR] path\n",
1373 5de5890b 2018-10-18 stsp 	    getprogname());
1374 5de5890b 2018-10-18 stsp 	exit(1);
1375 5de5890b 2018-10-18 stsp }
1376 5de5890b 2018-10-18 stsp 
1377 c1669e2e 2019-01-09 stsp static void
1378 c1669e2e 2019-01-09 stsp print_entry(struct got_tree_entry *te, const char *id, const char *path,
1379 c1669e2e 2019-01-09 stsp     const char *root_path)
1380 c1669e2e 2019-01-09 stsp {
1381 c1669e2e 2019-01-09 stsp 	int is_root_path = (strcmp(path, root_path) == 0);
1382 5de5890b 2018-10-18 stsp 
1383 c1669e2e 2019-01-09 stsp 	path += strlen(root_path);
1384 c1669e2e 2019-01-09 stsp 	while (path[0] == '/')
1385 c1669e2e 2019-01-09 stsp 		path++;
1386 c1669e2e 2019-01-09 stsp 
1387 c1669e2e 2019-01-09 stsp 	printf("%s%s%s%s%s\n", id ? id : "", path,
1388 d6e648b4 2019-02-10 stsp 	    is_root_path ? "" : "/", te->name,
1389 d6e648b4 2019-02-10 stsp 	    S_ISDIR(te->mode) ? "/" : ((te->mode & S_IXUSR) ? "*" : ""));
1390 c1669e2e 2019-01-09 stsp }
1391 c1669e2e 2019-01-09 stsp 
1392 5de5890b 2018-10-18 stsp static const struct got_error *
1393 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
1394 c1669e2e 2019-01-09 stsp     int show_ids, int recurse, const char *root_path,
1395 c1669e2e 2019-01-09 stsp     struct got_repository *repo)
1396 5de5890b 2018-10-18 stsp {
1397 5de5890b 2018-10-18 stsp 	const struct got_error *err = NULL;
1398 5de5890b 2018-10-18 stsp 	struct got_object_id *tree_id = NULL;
1399 5de5890b 2018-10-18 stsp 	struct got_tree_object *tree = NULL;
1400 5de5890b 2018-10-18 stsp 	const struct got_tree_entries *entries;
1401 5de5890b 2018-10-18 stsp 	struct got_tree_entry *te;
1402 5de5890b 2018-10-18 stsp 
1403 5de5890b 2018-10-18 stsp 	err = got_object_id_by_path(&tree_id, repo, commit_id, path);
1404 5de5890b 2018-10-18 stsp 	if (err)
1405 5de5890b 2018-10-18 stsp 		goto done;
1406 5de5890b 2018-10-18 stsp 
1407 5de5890b 2018-10-18 stsp 	err = got_object_open_as_tree(&tree, repo, tree_id);
1408 5de5890b 2018-10-18 stsp 	if (err)
1409 5de5890b 2018-10-18 stsp 		goto done;
1410 5de5890b 2018-10-18 stsp 	entries = got_object_tree_get_entries(tree);
1411 5de5890b 2018-10-18 stsp 	te = SIMPLEQ_FIRST(&entries->head);
1412 5de5890b 2018-10-18 stsp 	while (te) {
1413 5de5890b 2018-10-18 stsp 		char *id = NULL;
1414 84453469 2018-11-11 stsp 
1415 84453469 2018-11-11 stsp 		if (sigint_received || sigpipe_received)
1416 84453469 2018-11-11 stsp 			break;
1417 84453469 2018-11-11 stsp 
1418 5de5890b 2018-10-18 stsp 		if (show_ids) {
1419 5de5890b 2018-10-18 stsp 			char *id_str;
1420 5de5890b 2018-10-18 stsp 			err = got_object_id_str(&id_str, te->id);
1421 5de5890b 2018-10-18 stsp 			if (err)
1422 5de5890b 2018-10-18 stsp 				goto done;
1423 5de5890b 2018-10-18 stsp 			if (asprintf(&id, "%s ", id_str) == -1) {
1424 5de5890b 2018-10-18 stsp 				err = got_error_from_errno();
1425 5de5890b 2018-10-18 stsp 				free(id_str);
1426 5de5890b 2018-10-18 stsp 				goto done;
1427 5de5890b 2018-10-18 stsp 			}
1428 5de5890b 2018-10-18 stsp 			free(id_str);
1429 5de5890b 2018-10-18 stsp 		}
1430 c1669e2e 2019-01-09 stsp 		print_entry(te, id, path, root_path);
1431 5de5890b 2018-10-18 stsp 		free(id);
1432 c1669e2e 2019-01-09 stsp 
1433 c1669e2e 2019-01-09 stsp 		if (recurse && S_ISDIR(te->mode)) {
1434 c1669e2e 2019-01-09 stsp 			char *child_path;
1435 c1669e2e 2019-01-09 stsp 			if (asprintf(&child_path, "%s%s%s", path,
1436 c1669e2e 2019-01-09 stsp 			    path[0] == '/' && path[1] == '\0' ? "" : "/",
1437 c1669e2e 2019-01-09 stsp 			    te->name) == -1) {
1438 c1669e2e 2019-01-09 stsp 				err = got_error_from_errno();
1439 c1669e2e 2019-01-09 stsp 				goto done;
1440 c1669e2e 2019-01-09 stsp 			}
1441 c1669e2e 2019-01-09 stsp 			err = print_tree(child_path, commit_id, show_ids, 1,
1442 c1669e2e 2019-01-09 stsp 			    root_path, repo);
1443 c1669e2e 2019-01-09 stsp 			free(child_path);
1444 c1669e2e 2019-01-09 stsp 			if (err)
1445 c1669e2e 2019-01-09 stsp 				goto done;
1446 c1669e2e 2019-01-09 stsp 		}
1447 c1669e2e 2019-01-09 stsp 
1448 c1669e2e 2019-01-09 stsp 		te = SIMPLEQ_NEXT(te, entry);
1449 5de5890b 2018-10-18 stsp 	}
1450 5de5890b 2018-10-18 stsp done:
1451 5de5890b 2018-10-18 stsp 	if (tree)
1452 5de5890b 2018-10-18 stsp 		got_object_tree_close(tree);
1453 5de5890b 2018-10-18 stsp 	free(tree_id);
1454 5de5890b 2018-10-18 stsp 	return err;
1455 404c43c4 2018-06-21 stsp }
1456 404c43c4 2018-06-21 stsp 
1457 5de5890b 2018-10-18 stsp static const struct got_error *
1458 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
1459 5de5890b 2018-10-18 stsp {
1460 5de5890b 2018-10-18 stsp 	const struct got_error *error;
1461 5de5890b 2018-10-18 stsp 	struct got_repository *repo = NULL;
1462 7a2c19d6 2019-02-05 stsp 	struct got_worktree *worktree = NULL;
1463 7a2c19d6 2019-02-05 stsp 	const char *path;
1464 7a2c19d6 2019-02-05 stsp 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1465 5de5890b 2018-10-18 stsp 	struct got_object_id *commit_id = NULL;
1466 5de5890b 2018-10-18 stsp 	char *commit_id_str = NULL;
1467 c1669e2e 2019-01-09 stsp 	int show_ids = 0, recurse = 0;
1468 5de5890b 2018-10-18 stsp 	int ch;
1469 5de5890b 2018-10-18 stsp 
1470 5de5890b 2018-10-18 stsp #ifndef PROFILE
1471 0f8d269b 2019-01-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1472 0f8d269b 2019-01-04 stsp 	    NULL) == -1)
1473 5de5890b 2018-10-18 stsp 		err(1, "pledge");
1474 5de5890b 2018-10-18 stsp #endif
1475 5de5890b 2018-10-18 stsp 
1476 c1669e2e 2019-01-09 stsp 	while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
1477 5de5890b 2018-10-18 stsp 		switch (ch) {
1478 5de5890b 2018-10-18 stsp 		case 'c':
1479 5de5890b 2018-10-18 stsp 			commit_id_str = optarg;
1480 5de5890b 2018-10-18 stsp 			break;
1481 5de5890b 2018-10-18 stsp 		case 'r':
1482 5de5890b 2018-10-18 stsp 			repo_path = realpath(optarg, NULL);
1483 5de5890b 2018-10-18 stsp 			if (repo_path == NULL)
1484 5de5890b 2018-10-18 stsp 				err(1, "-r option");
1485 5de5890b 2018-10-18 stsp 			break;
1486 5de5890b 2018-10-18 stsp 		case 'i':
1487 5de5890b 2018-10-18 stsp 			show_ids = 1;
1488 5de5890b 2018-10-18 stsp 			break;
1489 c1669e2e 2019-01-09 stsp 		case 'R':
1490 c1669e2e 2019-01-09 stsp 			recurse = 1;
1491 c1669e2e 2019-01-09 stsp 			break;
1492 5de5890b 2018-10-18 stsp 		default:
1493 2deda0b9 2019-03-07 stsp 			usage_tree();
1494 5de5890b 2018-10-18 stsp 			/* NOTREACHED */
1495 5de5890b 2018-10-18 stsp 		}
1496 5de5890b 2018-10-18 stsp 	}
1497 5de5890b 2018-10-18 stsp 
1498 5de5890b 2018-10-18 stsp 	argc -= optind;
1499 5de5890b 2018-10-18 stsp 	argv += optind;
1500 5de5890b 2018-10-18 stsp 
1501 5de5890b 2018-10-18 stsp 	if (argc == 1)
1502 5de5890b 2018-10-18 stsp 		path = argv[0];
1503 5de5890b 2018-10-18 stsp 	else if (argc > 1)
1504 5de5890b 2018-10-18 stsp 		usage_tree();
1505 5de5890b 2018-10-18 stsp 	else
1506 7a2c19d6 2019-02-05 stsp 		path = NULL;
1507 7a2c19d6 2019-02-05 stsp 
1508 9bf7a39b 2019-02-05 stsp 	cwd = getcwd(NULL, 0);
1509 9bf7a39b 2019-02-05 stsp 	if (cwd == NULL) {
1510 9bf7a39b 2019-02-05 stsp 		error = got_error_from_errno();
1511 9bf7a39b 2019-02-05 stsp 		goto done;
1512 9bf7a39b 2019-02-05 stsp 	}
1513 5de5890b 2018-10-18 stsp 	if (repo_path == NULL) {
1514 7a2c19d6 2019-02-05 stsp 		error = got_worktree_open(&worktree, cwd);
1515 8994de28 2019-02-05 stsp 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
1516 7a2c19d6 2019-02-05 stsp 			goto done;
1517 7a2c19d6 2019-02-05 stsp 		else
1518 7a2c19d6 2019-02-05 stsp 			error = NULL;
1519 7a2c19d6 2019-02-05 stsp 		if (worktree) {
1520 7a2c19d6 2019-02-05 stsp 			repo_path =
1521 7a2c19d6 2019-02-05 stsp 			    strdup(got_worktree_get_repo_path(worktree));
1522 7a2c19d6 2019-02-05 stsp 			if (repo_path == NULL)
1523 7a2c19d6 2019-02-05 stsp 				error = got_error_from_errno();
1524 7a2c19d6 2019-02-05 stsp 			if (error)
1525 7a2c19d6 2019-02-05 stsp 				goto done;
1526 7a2c19d6 2019-02-05 stsp 		} else {
1527 7a2c19d6 2019-02-05 stsp 			repo_path = strdup(cwd);
1528 7a2c19d6 2019-02-05 stsp 			if (repo_path == NULL) {
1529 7a2c19d6 2019-02-05 stsp 				error = got_error_from_errno();
1530 7a2c19d6 2019-02-05 stsp 				goto done;
1531 7a2c19d6 2019-02-05 stsp 			}
1532 5de5890b 2018-10-18 stsp 		}
1533 5de5890b 2018-10-18 stsp 	}
1534 5de5890b 2018-10-18 stsp 
1535 d0eebce4 2019-03-11 stsp 	error = apply_unveil(repo_path, 1, NULL);
1536 0f8d269b 2019-01-04 stsp 	if (error)
1537 0f8d269b 2019-01-04 stsp 		goto done;
1538 0f8d269b 2019-01-04 stsp 
1539 5de5890b 2018-10-18 stsp 	error = got_repo_open(&repo, repo_path);
1540 5de5890b 2018-10-18 stsp 	if (error != NULL)
1541 5de5890b 2018-10-18 stsp 		goto done;
1542 5de5890b 2018-10-18 stsp 
1543 9bf7a39b 2019-02-05 stsp 	if (path == NULL) {
1544 9bf7a39b 2019-02-05 stsp 		if (worktree) {
1545 9bf7a39b 2019-02-05 stsp 			char *p, *worktree_subdir = cwd +
1546 9bf7a39b 2019-02-05 stsp 			    strlen(got_worktree_get_root_path(worktree));
1547 9bf7a39b 2019-02-05 stsp 			if (asprintf(&p, "%s/%s",
1548 9bf7a39b 2019-02-05 stsp 			    got_worktree_get_path_prefix(worktree),
1549 9bf7a39b 2019-02-05 stsp 			    worktree_subdir) == -1) {
1550 9bf7a39b 2019-02-05 stsp 				error = got_error_from_errno();
1551 9bf7a39b 2019-02-05 stsp 				goto done;
1552 9bf7a39b 2019-02-05 stsp 			}
1553 9bf7a39b 2019-02-05 stsp 			error = got_repo_map_path(&in_repo_path, repo, p, 1);
1554 9bf7a39b 2019-02-05 stsp 			free(p);
1555 9bf7a39b 2019-02-05 stsp 			if (error)
1556 9bf7a39b 2019-02-05 stsp 				goto done;
1557 9bf7a39b 2019-02-05 stsp 		} else
1558 9bf7a39b 2019-02-05 stsp 			path = "/";
1559 9bf7a39b 2019-02-05 stsp 	}
1560 9bf7a39b 2019-02-05 stsp 	if (in_repo_path == NULL) {
1561 9bf7a39b 2019-02-05 stsp 		error = got_repo_map_path(&in_repo_path, repo, path, 1);
1562 9bf7a39b 2019-02-05 stsp 		if (error != NULL)
1563 9bf7a39b 2019-02-05 stsp 			goto done;
1564 9bf7a39b 2019-02-05 stsp 	}
1565 5de5890b 2018-10-18 stsp 
1566 5de5890b 2018-10-18 stsp 	if (commit_id_str == NULL) {
1567 5de5890b 2018-10-18 stsp 		struct got_reference *head_ref;
1568 5de5890b 2018-10-18 stsp 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1569 5de5890b 2018-10-18 stsp 		if (error != NULL)
1570 5de5890b 2018-10-18 stsp 			goto done;
1571 5de5890b 2018-10-18 stsp 		error = got_ref_resolve(&commit_id, repo, head_ref);
1572 5de5890b 2018-10-18 stsp 		got_ref_close(head_ref);
1573 5de5890b 2018-10-18 stsp 		if (error != NULL)
1574 5de5890b 2018-10-18 stsp 			goto done;
1575 5de5890b 2018-10-18 stsp 	} else {
1576 15a94983 2018-12-23 stsp 		error = got_object_resolve_id_str(&commit_id, repo,
1577 15a94983 2018-12-23 stsp 		    commit_id_str);
1578 5de5890b 2018-10-18 stsp 		if (error != NULL)
1579 5de5890b 2018-10-18 stsp 			goto done;
1580 5de5890b 2018-10-18 stsp 	}
1581 5de5890b 2018-10-18 stsp 
1582 c1669e2e 2019-01-09 stsp 	error = print_tree(in_repo_path, commit_id, show_ids, recurse,
1583 c1669e2e 2019-01-09 stsp 	    in_repo_path, repo);
1584 5de5890b 2018-10-18 stsp done:
1585 5de5890b 2018-10-18 stsp 	free(in_repo_path);
1586 5de5890b 2018-10-18 stsp 	free(repo_path);
1587 5de5890b 2018-10-18 stsp 	free(cwd);
1588 5de5890b 2018-10-18 stsp 	free(commit_id);
1589 7a2c19d6 2019-02-05 stsp 	if (worktree)
1590 7a2c19d6 2019-02-05 stsp 		got_worktree_close(worktree);
1591 5de5890b 2018-10-18 stsp 	if (repo) {
1592 5de5890b 2018-10-18 stsp 		const struct got_error *repo_error;
1593 5de5890b 2018-10-18 stsp 		repo_error = got_repo_close(repo);
1594 5de5890b 2018-10-18 stsp 		if (error == NULL)
1595 5de5890b 2018-10-18 stsp 			error = repo_error;
1596 5de5890b 2018-10-18 stsp 	}
1597 5de5890b 2018-10-18 stsp 	return error;
1598 5de5890b 2018-10-18 stsp }
1599 5de5890b 2018-10-18 stsp 
1600 6bad629b 2019-02-04 stsp __dead static void
1601 6bad629b 2019-02-04 stsp usage_status(void)
1602 6bad629b 2019-02-04 stsp {
1603 927df6b7 2019-02-10 stsp 	fprintf(stderr, "usage: %s status [path]\n", getprogname());
1604 6bad629b 2019-02-04 stsp 	exit(1);
1605 6bad629b 2019-02-04 stsp }
1606 5c860e29 2018-03-12 stsp 
1607 b72f483a 2019-02-05 stsp static const struct got_error *
1608 b72f483a 2019-02-05 stsp print_status(void *arg, unsigned char status, const char *path,
1609 b72f483a 2019-02-05 stsp     struct got_object_id *id)
1610 6bad629b 2019-02-04 stsp {
1611 6bad629b 2019-02-04 stsp 	printf("%c  %s\n", status, path);
1612 b72f483a 2019-02-05 stsp 	return NULL;
1613 6bad629b 2019-02-04 stsp }
1614 5c860e29 2018-03-12 stsp 
1615 6bad629b 2019-02-04 stsp static const struct got_error *
1616 6bad629b 2019-02-04 stsp cmd_status(int argc, char *argv[])
1617 6bad629b 2019-02-04 stsp {
1618 6bad629b 2019-02-04 stsp 	const struct got_error *error = NULL;
1619 6bad629b 2019-02-04 stsp 	struct got_repository *repo = NULL;
1620 6bad629b 2019-02-04 stsp 	struct got_worktree *worktree = NULL;
1621 927df6b7 2019-02-10 stsp 	char *cwd = NULL, *path = NULL;
1622 6bad629b 2019-02-04 stsp 	int ch;
1623 5c860e29 2018-03-12 stsp 
1624 6bad629b 2019-02-04 stsp 	while ((ch = getopt(argc, argv, "")) != -1) {
1625 6bad629b 2019-02-04 stsp 		switch (ch) {
1626 5c860e29 2018-03-12 stsp 		default:
1627 2deda0b9 2019-03-07 stsp 			usage_status();
1628 6bad629b 2019-02-04 stsp 			/* NOTREACHED */
1629 5c860e29 2018-03-12 stsp 		}
1630 5c860e29 2018-03-12 stsp 	}
1631 5c860e29 2018-03-12 stsp 
1632 6bad629b 2019-02-04 stsp 	argc -= optind;
1633 6bad629b 2019-02-04 stsp 	argv += optind;
1634 5c860e29 2018-03-12 stsp 
1635 6bad629b 2019-02-04 stsp #ifndef PROFILE
1636 6bad629b 2019-02-04 stsp 	if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1637 6bad629b 2019-02-04 stsp 	    NULL) == -1)
1638 6bad629b 2019-02-04 stsp 		err(1, "pledge");
1639 f42b1b34 2018-03-12 stsp #endif
1640 927df6b7 2019-02-10 stsp 	cwd = getcwd(NULL, 0);
1641 927df6b7 2019-02-10 stsp 	if (cwd == NULL) {
1642 927df6b7 2019-02-10 stsp 		error = got_error_from_errno();
1643 927df6b7 2019-02-10 stsp 		goto done;
1644 927df6b7 2019-02-10 stsp 	}
1645 927df6b7 2019-02-10 stsp 
1646 927df6b7 2019-02-10 stsp 	error = got_worktree_open(&worktree, cwd);
1647 927df6b7 2019-02-10 stsp 	if (error != NULL)
1648 927df6b7 2019-02-10 stsp 		goto done;
1649 927df6b7 2019-02-10 stsp 
1650 6bad629b 2019-02-04 stsp 	if (argc == 0) {
1651 927df6b7 2019-02-10 stsp 		path = strdup("");
1652 927df6b7 2019-02-10 stsp 		if (path == NULL) {
1653 6bad629b 2019-02-04 stsp 			error = got_error_from_errno();
1654 6bad629b 2019-02-04 stsp 			goto done;
1655 6bad629b 2019-02-04 stsp 		}
1656 6bad629b 2019-02-04 stsp 	} else if (argc == 1) {
1657 927df6b7 2019-02-10 stsp 		error = get_status_path(&path, worktree, argv[0]);
1658 927df6b7 2019-02-10 stsp 		if (error)
1659 6bad629b 2019-02-04 stsp 			goto done;
1660 6bad629b 2019-02-04 stsp 	} else
1661 6bad629b 2019-02-04 stsp 		usage_status();
1662 6bad629b 2019-02-04 stsp 
1663 6bad629b 2019-02-04 stsp 	error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1664 6bad629b 2019-02-04 stsp 	if (error != NULL)
1665 6bad629b 2019-02-04 stsp 		goto done;
1666 6bad629b 2019-02-04 stsp 
1667 d0eebce4 2019-03-11 stsp 	error = apply_unveil(got_repo_get_path(repo), 1,
1668 c7f4312f 2019-02-05 stsp 	    got_worktree_get_root_path(worktree));
1669 6bad629b 2019-02-04 stsp 	if (error)
1670 6bad629b 2019-02-04 stsp 		goto done;
1671 6bad629b 2019-02-04 stsp 
1672 927df6b7 2019-02-10 stsp 	error = got_worktree_status(worktree, path, repo, print_status, NULL,
1673 6bad629b 2019-02-04 stsp 	    check_cancelled, NULL);
1674 6bad629b 2019-02-04 stsp done:
1675 927df6b7 2019-02-10 stsp 	free(cwd);
1676 927df6b7 2019-02-10 stsp 	free(path);
1677 d0eebce4 2019-03-11 stsp 	return error;
1678 d0eebce4 2019-03-11 stsp }
1679 d0eebce4 2019-03-11 stsp 
1680 d0eebce4 2019-03-11 stsp __dead static void
1681 d0eebce4 2019-03-11 stsp usage_ref(void)
1682 d0eebce4 2019-03-11 stsp {
1683 d0eebce4 2019-03-11 stsp 	fprintf(stderr,
1684 d0eebce4 2019-03-11 stsp 	    "usage: %s ref [-r repository] -l | -d name | name object\n",
1685 d0eebce4 2019-03-11 stsp 	    getprogname());
1686 d0eebce4 2019-03-11 stsp 	exit(1);
1687 d0eebce4 2019-03-11 stsp }
1688 d0eebce4 2019-03-11 stsp 
1689 d0eebce4 2019-03-11 stsp static const struct got_error *
1690 d0eebce4 2019-03-11 stsp list_refs(struct got_repository *repo)
1691 d0eebce4 2019-03-11 stsp {
1692 d0eebce4 2019-03-11 stsp 	static const struct got_error *err = NULL;
1693 d0eebce4 2019-03-11 stsp 	struct got_reflist_head refs;
1694 d0eebce4 2019-03-11 stsp 	struct got_reflist_entry *re;
1695 d0eebce4 2019-03-11 stsp 
1696 d0eebce4 2019-03-11 stsp 	SIMPLEQ_INIT(&refs);
1697 d0eebce4 2019-03-11 stsp 	err = got_ref_list(&refs, repo);
1698 d0eebce4 2019-03-11 stsp 	if (err)
1699 d0eebce4 2019-03-11 stsp 		return err;
1700 d0eebce4 2019-03-11 stsp 
1701 d0eebce4 2019-03-11 stsp 	SIMPLEQ_FOREACH(re, &refs, entry) {
1702 d0eebce4 2019-03-11 stsp 		char *refstr;
1703 d0eebce4 2019-03-11 stsp 		refstr = got_ref_to_str(re->ref);
1704 d0eebce4 2019-03-11 stsp 		if (refstr == NULL)
1705 d0eebce4 2019-03-11 stsp 			return got_error_from_errno();
1706 d0eebce4 2019-03-11 stsp 		printf("%s: %s\n", got_ref_get_name(re->ref), refstr);
1707 d0eebce4 2019-03-11 stsp 		free(refstr);
1708 d0eebce4 2019-03-11 stsp 	}
1709 d0eebce4 2019-03-11 stsp 
1710 e2e879a0 2019-03-11 stsp 	got_ref_list_free(&refs);
1711 d0eebce4 2019-03-11 stsp 	return NULL;
1712 d0eebce4 2019-03-11 stsp }
1713 d0eebce4 2019-03-11 stsp 
1714 d0eebce4 2019-03-11 stsp static const struct got_error *
1715 d0eebce4 2019-03-11 stsp delete_ref(struct got_repository *repo, const char *refname)
1716 d0eebce4 2019-03-11 stsp {
1717 d0eebce4 2019-03-11 stsp 	const struct got_error *err = NULL;
1718 d0eebce4 2019-03-11 stsp 	struct got_reference *ref;
1719 d0eebce4 2019-03-11 stsp 
1720 d0eebce4 2019-03-11 stsp 	err = got_ref_open(&ref, repo, refname);
1721 d0eebce4 2019-03-11 stsp 	if (err)
1722 d0eebce4 2019-03-11 stsp 		return err;
1723 d0eebce4 2019-03-11 stsp 
1724 d0eebce4 2019-03-11 stsp 	err = got_ref_delete(ref, repo);
1725 d0eebce4 2019-03-11 stsp 	got_ref_close(ref);
1726 d0eebce4 2019-03-11 stsp 	return err;
1727 d0eebce4 2019-03-11 stsp }
1728 d0eebce4 2019-03-11 stsp 
1729 d0eebce4 2019-03-11 stsp static const struct got_error *
1730 d0eebce4 2019-03-11 stsp add_ref(struct got_repository *repo, const char *refname, const char *id_str)
1731 d0eebce4 2019-03-11 stsp {
1732 d0eebce4 2019-03-11 stsp 	const struct got_error *err = NULL;
1733 d0eebce4 2019-03-11 stsp 	struct got_object_id *id;
1734 d0eebce4 2019-03-11 stsp 	struct got_reference *ref = NULL;
1735 d0eebce4 2019-03-11 stsp 
1736 d0eebce4 2019-03-11 stsp 	err = got_object_resolve_id_str(&id, repo, id_str);
1737 d0eebce4 2019-03-11 stsp 	if (err)
1738 d0eebce4 2019-03-11 stsp 		return err;
1739 d0eebce4 2019-03-11 stsp 
1740 d0eebce4 2019-03-11 stsp 	err = got_ref_alloc(&ref, refname, id);
1741 d0eebce4 2019-03-11 stsp 	if (err)
1742 d0eebce4 2019-03-11 stsp 		goto done;
1743 d0eebce4 2019-03-11 stsp 
1744 d0eebce4 2019-03-11 stsp 	err = got_ref_write(ref, repo);
1745 d0eebce4 2019-03-11 stsp done:
1746 d0eebce4 2019-03-11 stsp 	if (ref)
1747 d0eebce4 2019-03-11 stsp 		got_ref_close(ref);
1748 d0eebce4 2019-03-11 stsp 	free(id);
1749 d0eebce4 2019-03-11 stsp 	return err;
1750 d0eebce4 2019-03-11 stsp }
1751 d0eebce4 2019-03-11 stsp 
1752 d0eebce4 2019-03-11 stsp static const struct got_error *
1753 d0eebce4 2019-03-11 stsp cmd_ref(int argc, char *argv[])
1754 d0eebce4 2019-03-11 stsp {
1755 d0eebce4 2019-03-11 stsp 	const struct got_error *error = NULL;
1756 d0eebce4 2019-03-11 stsp 	struct got_repository *repo = NULL;
1757 d0eebce4 2019-03-11 stsp 	struct got_worktree *worktree = NULL;
1758 d0eebce4 2019-03-11 stsp 	char *cwd = NULL, *repo_path = NULL;
1759 d0eebce4 2019-03-11 stsp 	int ch, do_list = 0;
1760 d0eebce4 2019-03-11 stsp 	const char *delref = NULL;
1761 d0eebce4 2019-03-11 stsp 
1762 d0eebce4 2019-03-11 stsp 	/* TODO: Add -s option for adding symbolic references. */
1763 d0eebce4 2019-03-11 stsp 	while ((ch = getopt(argc, argv, "d:r:l")) != -1) {
1764 d0eebce4 2019-03-11 stsp 		switch (ch) {
1765 d0eebce4 2019-03-11 stsp 		case 'd':
1766 d0eebce4 2019-03-11 stsp 			delref = optarg;
1767 d0eebce4 2019-03-11 stsp 			break;
1768 d0eebce4 2019-03-11 stsp 		case 'r':
1769 d0eebce4 2019-03-11 stsp 			repo_path = realpath(optarg, NULL);
1770 d0eebce4 2019-03-11 stsp 			if (repo_path == NULL)
1771 d0eebce4 2019-03-11 stsp 				err(1, "-r option");
1772 d0eebce4 2019-03-11 stsp 			break;
1773 d0eebce4 2019-03-11 stsp 		case 'l':
1774 d0eebce4 2019-03-11 stsp 			do_list = 1;
1775 d0eebce4 2019-03-11 stsp 			break;
1776 d0eebce4 2019-03-11 stsp 		default:
1777 d0eebce4 2019-03-11 stsp 			usage_ref();
1778 d0eebce4 2019-03-11 stsp 			/* NOTREACHED */
1779 d0eebce4 2019-03-11 stsp 		}
1780 d0eebce4 2019-03-11 stsp 	}
1781 d0eebce4 2019-03-11 stsp 
1782 d0eebce4 2019-03-11 stsp 	if (do_list && delref)
1783 d0eebce4 2019-03-11 stsp 		errx(1, "-l and -d options are mutually exclusive\n");
1784 d0eebce4 2019-03-11 stsp 
1785 d0eebce4 2019-03-11 stsp 	argc -= optind;
1786 d0eebce4 2019-03-11 stsp 	argv += optind;
1787 d0eebce4 2019-03-11 stsp 
1788 d0eebce4 2019-03-11 stsp 	if (do_list || delref) {
1789 d0eebce4 2019-03-11 stsp 		if (argc > 0)
1790 d0eebce4 2019-03-11 stsp 			usage_ref();
1791 d0eebce4 2019-03-11 stsp 	} else if (argc != 2)
1792 d0eebce4 2019-03-11 stsp 		usage_ref();
1793 d0eebce4 2019-03-11 stsp 
1794 d0eebce4 2019-03-11 stsp #ifndef PROFILE
1795 e0b57350 2019-03-12 stsp 	if (do_list) {
1796 e0b57350 2019-03-12 stsp 		if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
1797 e0b57350 2019-03-12 stsp 		    NULL) == -1)
1798 e0b57350 2019-03-12 stsp 			err(1, "pledge");
1799 e0b57350 2019-03-12 stsp 	} else {
1800 e0b57350 2019-03-12 stsp 		if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1801 e0b57350 2019-03-12 stsp 		    "sendfd unveil", NULL) == -1)
1802 e0b57350 2019-03-12 stsp 			err(1, "pledge");
1803 e0b57350 2019-03-12 stsp 	}
1804 d0eebce4 2019-03-11 stsp #endif
1805 d0eebce4 2019-03-11 stsp 	cwd = getcwd(NULL, 0);
1806 d0eebce4 2019-03-11 stsp 	if (cwd == NULL) {
1807 d0eebce4 2019-03-11 stsp 		error = got_error_from_errno();
1808 d0eebce4 2019-03-11 stsp 		goto done;
1809 d0eebce4 2019-03-11 stsp 	}
1810 d0eebce4 2019-03-11 stsp 
1811 d0eebce4 2019-03-11 stsp 	if (repo_path == NULL) {
1812 d0eebce4 2019-03-11 stsp 		error = got_worktree_open(&worktree, cwd);
1813 d0eebce4 2019-03-11 stsp 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
1814 d0eebce4 2019-03-11 stsp 			goto done;
1815 d0eebce4 2019-03-11 stsp 		else
1816 d0eebce4 2019-03-11 stsp 			error = NULL;
1817 d0eebce4 2019-03-11 stsp 		if (worktree) {
1818 d0eebce4 2019-03-11 stsp 			repo_path =
1819 d0eebce4 2019-03-11 stsp 			    strdup(got_worktree_get_repo_path(worktree));
1820 d0eebce4 2019-03-11 stsp 			if (repo_path == NULL)
1821 d0eebce4 2019-03-11 stsp 				error = got_error_from_errno();
1822 d0eebce4 2019-03-11 stsp 			if (error)
1823 d0eebce4 2019-03-11 stsp 				goto done;
1824 d0eebce4 2019-03-11 stsp 		} else {
1825 d0eebce4 2019-03-11 stsp 			repo_path = strdup(cwd);
1826 d0eebce4 2019-03-11 stsp 			if (repo_path == NULL) {
1827 d0eebce4 2019-03-11 stsp 				error = got_error_from_errno();
1828 d0eebce4 2019-03-11 stsp 				goto done;
1829 d0eebce4 2019-03-11 stsp 			}
1830 d0eebce4 2019-03-11 stsp 		}
1831 d0eebce4 2019-03-11 stsp 	}
1832 d0eebce4 2019-03-11 stsp 
1833 d0eebce4 2019-03-11 stsp 	error = apply_unveil(repo_path, do_list,
1834 d0eebce4 2019-03-11 stsp 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
1835 d0eebce4 2019-03-11 stsp 	if (error)
1836 d0eebce4 2019-03-11 stsp 		goto done;
1837 d0eebce4 2019-03-11 stsp 
1838 d0eebce4 2019-03-11 stsp 	error = got_repo_open(&repo, repo_path);
1839 d0eebce4 2019-03-11 stsp 	if (error != NULL)
1840 d0eebce4 2019-03-11 stsp 		goto done;
1841 d0eebce4 2019-03-11 stsp 
1842 d0eebce4 2019-03-11 stsp 	if (do_list)
1843 d0eebce4 2019-03-11 stsp 		error = list_refs(repo);
1844 d0eebce4 2019-03-11 stsp 	else if (delref)
1845 d0eebce4 2019-03-11 stsp 		error = delete_ref(repo, delref);
1846 d0eebce4 2019-03-11 stsp 	else
1847 d0eebce4 2019-03-11 stsp 		error = add_ref(repo, argv[0], argv[1]);
1848 d0eebce4 2019-03-11 stsp done:
1849 d0eebce4 2019-03-11 stsp 	if (repo)
1850 d0eebce4 2019-03-11 stsp 		got_repo_close(repo);
1851 d0eebce4 2019-03-11 stsp 	if (worktree)
1852 d0eebce4 2019-03-11 stsp 		got_worktree_close(worktree);
1853 d0eebce4 2019-03-11 stsp 	free(cwd);
1854 d0eebce4 2019-03-11 stsp 	free(repo_path);
1855 6bad629b 2019-02-04 stsp 	return error;
1856 6bad629b 2019-02-04 stsp }