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