Blame


1 24df9a28 2023-07-08 jrick /*
2 24df9a28 2023-07-08 jrick * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
3 24df9a28 2023-07-08 jrick * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
4 24df9a28 2023-07-08 jrick * Copyright (c) 2020 Ori Bernstein <ori@openbsd.org>
5 24df9a28 2023-07-08 jrick * Copyright (C) 2023 Josh Rickmar <jrick@zettaport.com>
6 24df9a28 2023-07-08 jrick *
7 24df9a28 2023-07-08 jrick * Permission to use, copy, modify, and distribute this software for any
8 24df9a28 2023-07-08 jrick * purpose with or without fee is hereby granted, provided that the above
9 24df9a28 2023-07-08 jrick * copyright notice and this permission notice appear in all copies.
10 24df9a28 2023-07-08 jrick *
11 24df9a28 2023-07-08 jrick * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12 24df9a28 2023-07-08 jrick * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 24df9a28 2023-07-08 jrick * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14 24df9a28 2023-07-08 jrick * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 24df9a28 2023-07-08 jrick * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 24df9a28 2023-07-08 jrick * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
17 24df9a28 2023-07-08 jrick * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 24df9a28 2023-07-08 jrick */
19 24df9a28 2023-07-08 jrick
20 24df9a28 2023-07-08 jrick #include <sys/queue.h>
21 24df9a28 2023-07-08 jrick #include <sys/time.h>
22 24df9a28 2023-07-08 jrick #include <sys/types.h>
23 24df9a28 2023-07-08 jrick #include <sys/stat.h>
24 24df9a28 2023-07-08 jrick #include <sys/wait.h>
25 24df9a28 2023-07-08 jrick
26 24df9a28 2023-07-08 jrick #include <err.h>
27 24df9a28 2023-07-08 jrick #include <errno.h>
28 24df9a28 2023-07-08 jrick #include <fcntl.h>
29 24df9a28 2023-07-08 jrick #include <limits.h>
30 24df9a28 2023-07-08 jrick #include <locale.h>
31 24df9a28 2023-07-08 jrick #include <ctype.h>
32 24df9a28 2023-07-08 jrick #include <sha1.h>
33 24df9a28 2023-07-08 jrick #include <sha2.h>
34 24df9a28 2023-07-08 jrick #include <signal.h>
35 24df9a28 2023-07-08 jrick #include <stdio.h>
36 24df9a28 2023-07-08 jrick #include <stdlib.h>
37 24df9a28 2023-07-08 jrick #include <string.h>
38 24df9a28 2023-07-08 jrick #include <unistd.h>
39 24df9a28 2023-07-08 jrick #include <libgen.h>
40 24df9a28 2023-07-08 jrick #include <time.h>
41 24df9a28 2023-07-08 jrick #include <paths.h>
42 24df9a28 2023-07-08 jrick #include <regex.h>
43 24df9a28 2023-07-08 jrick #include <getopt.h>
44 24df9a28 2023-07-08 jrick #include <util.h>
45 24df9a28 2023-07-08 jrick
46 24df9a28 2023-07-08 jrick #include "got_version.h"
47 24df9a28 2023-07-08 jrick #include "got_error.h"
48 24df9a28 2023-07-08 jrick #include "got_object.h"
49 24df9a28 2023-07-08 jrick #include "got_reference.h"
50 24df9a28 2023-07-08 jrick #include "got_repository.h"
51 24df9a28 2023-07-08 jrick #include "got_path.h"
52 24df9a28 2023-07-08 jrick #include "got_cancel.h"
53 24df9a28 2023-07-08 jrick #include "got_worktree.h"
54 24df9a28 2023-07-08 jrick #include "got_worktree_cvg.h"
55 24df9a28 2023-07-08 jrick #include "got_diff.h"
56 24df9a28 2023-07-08 jrick #include "got_commit_graph.h"
57 24df9a28 2023-07-08 jrick #include "got_fetch.h"
58 24df9a28 2023-07-08 jrick #include "got_send.h"
59 24df9a28 2023-07-08 jrick #include "got_blame.h"
60 24df9a28 2023-07-08 jrick #include "got_privsep.h"
61 24df9a28 2023-07-08 jrick #include "got_opentemp.h"
62 24df9a28 2023-07-08 jrick #include "got_gotconfig.h"
63 24df9a28 2023-07-08 jrick #include "got_dial.h"
64 24df9a28 2023-07-08 jrick #include "got_patch.h"
65 24df9a28 2023-07-08 jrick #include "got_sigs.h"
66 24df9a28 2023-07-08 jrick #include "got_date.h"
67 24df9a28 2023-07-08 jrick
68 24df9a28 2023-07-08 jrick #ifndef nitems
69 24df9a28 2023-07-08 jrick #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
70 24df9a28 2023-07-08 jrick #endif
71 24df9a28 2023-07-08 jrick
72 24df9a28 2023-07-08 jrick static volatile sig_atomic_t sigint_received;
73 24df9a28 2023-07-08 jrick static volatile sig_atomic_t sigpipe_received;
74 24df9a28 2023-07-08 jrick
75 24df9a28 2023-07-08 jrick static void
76 24df9a28 2023-07-08 jrick catch_sigint(int signo)
77 24df9a28 2023-07-08 jrick {
78 24df9a28 2023-07-08 jrick sigint_received = 1;
79 24df9a28 2023-07-08 jrick }
80 24df9a28 2023-07-08 jrick
81 24df9a28 2023-07-08 jrick static void
82 24df9a28 2023-07-08 jrick catch_sigpipe(int signo)
83 24df9a28 2023-07-08 jrick {
84 24df9a28 2023-07-08 jrick sigpipe_received = 1;
85 24df9a28 2023-07-08 jrick }
86 24df9a28 2023-07-08 jrick
87 24df9a28 2023-07-08 jrick
88 24df9a28 2023-07-08 jrick struct got_cmd {
89 24df9a28 2023-07-08 jrick const char *cmd_name;
90 24df9a28 2023-07-08 jrick const struct got_error *(*cmd_main)(int, char *[]);
91 24df9a28 2023-07-08 jrick void (*cmd_usage)(void);
92 24df9a28 2023-07-08 jrick const char *cmd_alias;
93 24df9a28 2023-07-08 jrick };
94 24df9a28 2023-07-08 jrick
95 24df9a28 2023-07-08 jrick __dead static void usage(int, int);
96 24df9a28 2023-07-08 jrick __dead static void usage_import(void);
97 24df9a28 2023-07-08 jrick __dead static void usage_clone(void);
98 24df9a28 2023-07-08 jrick __dead static void usage_checkout(void);
99 24df9a28 2023-07-08 jrick __dead static void usage_update(void);
100 24df9a28 2023-07-08 jrick __dead static void usage_log(void);
101 24df9a28 2023-07-08 jrick __dead static void usage_diff(void);
102 24df9a28 2023-07-08 jrick __dead static void usage_blame(void);
103 24df9a28 2023-07-08 jrick __dead static void usage_tree(void);
104 24df9a28 2023-07-08 jrick __dead static void usage_status(void);
105 24df9a28 2023-07-08 jrick __dead static void usage_tag(void);
106 24df9a28 2023-07-08 jrick __dead static void usage_add(void);
107 24df9a28 2023-07-08 jrick __dead static void usage_remove(void);
108 24df9a28 2023-07-08 jrick __dead static void usage_patch(void);
109 24df9a28 2023-07-08 jrick __dead static void usage_revert(void);
110 24df9a28 2023-07-08 jrick __dead static void usage_commit(void);
111 24df9a28 2023-07-08 jrick __dead static void usage_cherrypick(void);
112 24df9a28 2023-07-08 jrick __dead static void usage_backout(void);
113 24df9a28 2023-07-08 jrick __dead static void usage_cat(void);
114 24df9a28 2023-07-08 jrick __dead static void usage_info(void);
115 24df9a28 2023-07-08 jrick
116 24df9a28 2023-07-08 jrick static const struct got_error* cmd_import(int, char *[]);
117 24df9a28 2023-07-08 jrick static const struct got_error* cmd_clone(int, char *[]);
118 24df9a28 2023-07-08 jrick static const struct got_error* cmd_checkout(int, char *[]);
119 24df9a28 2023-07-08 jrick static const struct got_error* cmd_update(int, char *[]);
120 24df9a28 2023-07-08 jrick static const struct got_error* cmd_log(int, char *[]);
121 24df9a28 2023-07-08 jrick static const struct got_error* cmd_diff(int, char *[]);
122 24df9a28 2023-07-08 jrick static const struct got_error* cmd_blame(int, char *[]);
123 24df9a28 2023-07-08 jrick static const struct got_error* cmd_tree(int, char *[]);
124 24df9a28 2023-07-08 jrick static const struct got_error* cmd_status(int, char *[]);
125 24df9a28 2023-07-08 jrick static const struct got_error* cmd_tag(int, char *[]);
126 24df9a28 2023-07-08 jrick static const struct got_error* cmd_add(int, char *[]);
127 24df9a28 2023-07-08 jrick static const struct got_error* cmd_remove(int, char *[]);
128 24df9a28 2023-07-08 jrick static const struct got_error* cmd_patch(int, char *[]);
129 24df9a28 2023-07-08 jrick static const struct got_error* cmd_revert(int, char *[]);
130 24df9a28 2023-07-08 jrick static const struct got_error* cmd_commit(int, char *[]);
131 24df9a28 2023-07-08 jrick static const struct got_error* cmd_cherrypick(int, char *[]);
132 24df9a28 2023-07-08 jrick static const struct got_error* cmd_backout(int, char *[]);
133 24df9a28 2023-07-08 jrick static const struct got_error* cmd_cat(int, char *[]);
134 24df9a28 2023-07-08 jrick static const struct got_error* cmd_info(int, char *[]);
135 24df9a28 2023-07-08 jrick
136 24df9a28 2023-07-08 jrick static const struct got_cmd got_commands[] = {
137 24df9a28 2023-07-08 jrick { "import", cmd_import, usage_import, "im" },
138 24df9a28 2023-07-08 jrick { "clone", cmd_clone, usage_clone, "cl" },
139 24df9a28 2023-07-08 jrick { "checkout", cmd_checkout, usage_checkout, "co" },
140 24df9a28 2023-07-08 jrick { "update", cmd_update, usage_update, "up" },
141 24df9a28 2023-07-08 jrick { "log", cmd_log, usage_log, "" },
142 24df9a28 2023-07-08 jrick { "diff", cmd_diff, usage_diff, "di" },
143 24df9a28 2023-07-08 jrick { "blame", cmd_blame, usage_blame, "bl" },
144 24df9a28 2023-07-08 jrick { "tree", cmd_tree, usage_tree, "tr" },
145 24df9a28 2023-07-08 jrick { "status", cmd_status, usage_status, "st" },
146 24df9a28 2023-07-08 jrick { "tag", cmd_tag, usage_tag, "" },
147 24df9a28 2023-07-08 jrick { "add", cmd_add, usage_add, "" },
148 24df9a28 2023-07-08 jrick { "remove", cmd_remove, usage_remove, "rm" },
149 24df9a28 2023-07-08 jrick { "patch", cmd_patch, usage_patch, "pa" },
150 24df9a28 2023-07-08 jrick { "revert", cmd_revert, usage_revert, "rv" },
151 24df9a28 2023-07-08 jrick { "commit", cmd_commit, usage_commit, "ci" },
152 24df9a28 2023-07-08 jrick { "cherrypick", cmd_cherrypick, usage_cherrypick, "cy" },
153 24df9a28 2023-07-08 jrick { "backout", cmd_backout, usage_backout, "bo" },
154 24df9a28 2023-07-08 jrick { "cat", cmd_cat, usage_cat, "" },
155 24df9a28 2023-07-08 jrick { "info", cmd_info, usage_info, "" },
156 24df9a28 2023-07-08 jrick };
157 24df9a28 2023-07-08 jrick
158 24df9a28 2023-07-08 jrick static void
159 24df9a28 2023-07-08 jrick list_commands(FILE *fp)
160 24df9a28 2023-07-08 jrick {
161 24df9a28 2023-07-08 jrick size_t i;
162 24df9a28 2023-07-08 jrick
163 24df9a28 2023-07-08 jrick fprintf(fp, "commands:");
164 24df9a28 2023-07-08 jrick for (i = 0; i < nitems(got_commands); i++) {
165 24df9a28 2023-07-08 jrick const struct got_cmd *cmd = &got_commands[i];
166 24df9a28 2023-07-08 jrick fprintf(fp, " %s", cmd->cmd_name);
167 24df9a28 2023-07-08 jrick }
168 24df9a28 2023-07-08 jrick fputc('\n', fp);
169 24df9a28 2023-07-08 jrick }
170 24df9a28 2023-07-08 jrick
171 24df9a28 2023-07-08 jrick __dead static void
172 24df9a28 2023-07-08 jrick option_conflict(char a, char b)
173 24df9a28 2023-07-08 jrick {
174 24df9a28 2023-07-08 jrick errx(1, "-%c and -%c options are mutually exclusive", a, b);
175 24df9a28 2023-07-08 jrick }
176 24df9a28 2023-07-08 jrick
177 24df9a28 2023-07-08 jrick int
178 24df9a28 2023-07-08 jrick main(int argc, char *argv[])
179 24df9a28 2023-07-08 jrick {
180 24df9a28 2023-07-08 jrick const struct got_cmd *cmd;
181 24df9a28 2023-07-08 jrick size_t i;
182 24df9a28 2023-07-08 jrick int ch;
183 24df9a28 2023-07-08 jrick int hflag = 0, Vflag = 0;
184 24df9a28 2023-07-08 jrick static const struct option longopts[] = {
185 24df9a28 2023-07-08 jrick { "version", no_argument, NULL, 'V' },
186 24df9a28 2023-07-08 jrick { NULL, 0, NULL, 0 }
187 24df9a28 2023-07-08 jrick };
188 24df9a28 2023-07-08 jrick
189 24df9a28 2023-07-08 jrick setlocale(LC_CTYPE, "");
190 24df9a28 2023-07-08 jrick
191 24df9a28 2023-07-08 jrick while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
192 24df9a28 2023-07-08 jrick switch (ch) {
193 24df9a28 2023-07-08 jrick case 'h':
194 24df9a28 2023-07-08 jrick hflag = 1;
195 24df9a28 2023-07-08 jrick break;
196 24df9a28 2023-07-08 jrick case 'V':
197 24df9a28 2023-07-08 jrick Vflag = 1;
198 24df9a28 2023-07-08 jrick break;
199 24df9a28 2023-07-08 jrick default:
200 24df9a28 2023-07-08 jrick usage(hflag, 1);
201 24df9a28 2023-07-08 jrick /* NOTREACHED */
202 24df9a28 2023-07-08 jrick }
203 24df9a28 2023-07-08 jrick }
204 24df9a28 2023-07-08 jrick
205 24df9a28 2023-07-08 jrick argc -= optind;
206 24df9a28 2023-07-08 jrick argv += optind;
207 24df9a28 2023-07-08 jrick optind = 1;
208 24df9a28 2023-07-08 jrick optreset = 1;
209 24df9a28 2023-07-08 jrick
210 24df9a28 2023-07-08 jrick if (Vflag) {
211 24df9a28 2023-07-08 jrick got_version_print_str();
212 24df9a28 2023-07-08 jrick return 0;
213 24df9a28 2023-07-08 jrick }
214 24df9a28 2023-07-08 jrick
215 24df9a28 2023-07-08 jrick if (argc <= 0)
216 24df9a28 2023-07-08 jrick usage(hflag, hflag ? 0 : 1);
217 24df9a28 2023-07-08 jrick
218 24df9a28 2023-07-08 jrick signal(SIGINT, catch_sigint);
219 24df9a28 2023-07-08 jrick signal(SIGPIPE, catch_sigpipe);
220 24df9a28 2023-07-08 jrick
221 24df9a28 2023-07-08 jrick for (i = 0; i < nitems(got_commands); i++) {
222 24df9a28 2023-07-08 jrick const struct got_error *error;
223 24df9a28 2023-07-08 jrick
224 24df9a28 2023-07-08 jrick cmd = &got_commands[i];
225 24df9a28 2023-07-08 jrick
226 24df9a28 2023-07-08 jrick if (strcmp(cmd->cmd_name, argv[0]) != 0 &&
227 24df9a28 2023-07-08 jrick strcmp(cmd->cmd_alias, argv[0]) != 0)
228 24df9a28 2023-07-08 jrick continue;
229 24df9a28 2023-07-08 jrick
230 24df9a28 2023-07-08 jrick if (hflag)
231 24df9a28 2023-07-08 jrick cmd->cmd_usage();
232 24df9a28 2023-07-08 jrick
233 24df9a28 2023-07-08 jrick error = cmd->cmd_main(argc, argv);
234 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_CANCELLED &&
235 24df9a28 2023-07-08 jrick error->code != GOT_ERR_PRIVSEP_EXIT &&
236 24df9a28 2023-07-08 jrick !(sigpipe_received &&
237 24df9a28 2023-07-08 jrick error->code == GOT_ERR_ERRNO && errno == EPIPE) &&
238 24df9a28 2023-07-08 jrick !(sigint_received &&
239 24df9a28 2023-07-08 jrick error->code == GOT_ERR_ERRNO && errno == EINTR)) {
240 24df9a28 2023-07-08 jrick fflush(stdout);
241 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
242 24df9a28 2023-07-08 jrick return 1;
243 24df9a28 2023-07-08 jrick }
244 24df9a28 2023-07-08 jrick
245 24df9a28 2023-07-08 jrick return 0;
246 24df9a28 2023-07-08 jrick }
247 24df9a28 2023-07-08 jrick
248 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
249 24df9a28 2023-07-08 jrick list_commands(stderr);
250 24df9a28 2023-07-08 jrick return 1;
251 24df9a28 2023-07-08 jrick }
252 24df9a28 2023-07-08 jrick
253 24df9a28 2023-07-08 jrick __dead static void
254 24df9a28 2023-07-08 jrick usage(int hflag, int status)
255 24df9a28 2023-07-08 jrick {
256 24df9a28 2023-07-08 jrick FILE *fp = (status == 0) ? stdout : stderr;
257 24df9a28 2023-07-08 jrick
258 24df9a28 2023-07-08 jrick fprintf(fp, "usage: %s [-hV] command [arg ...]\n",
259 24df9a28 2023-07-08 jrick getprogname());
260 24df9a28 2023-07-08 jrick if (hflag)
261 24df9a28 2023-07-08 jrick list_commands(fp);
262 24df9a28 2023-07-08 jrick exit(status);
263 24df9a28 2023-07-08 jrick }
264 24df9a28 2023-07-08 jrick
265 24df9a28 2023-07-08 jrick static const struct got_error *
266 24df9a28 2023-07-08 jrick get_editor(char **abspath)
267 24df9a28 2023-07-08 jrick {
268 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
269 24df9a28 2023-07-08 jrick const char *editor;
270 24df9a28 2023-07-08 jrick
271 24df9a28 2023-07-08 jrick *abspath = NULL;
272 24df9a28 2023-07-08 jrick
273 24df9a28 2023-07-08 jrick editor = getenv("VISUAL");
274 24df9a28 2023-07-08 jrick if (editor == NULL)
275 24df9a28 2023-07-08 jrick editor = getenv("EDITOR");
276 24df9a28 2023-07-08 jrick
277 24df9a28 2023-07-08 jrick if (editor) {
278 24df9a28 2023-07-08 jrick err = got_path_find_prog(abspath, editor);
279 24df9a28 2023-07-08 jrick if (err)
280 24df9a28 2023-07-08 jrick return err;
281 24df9a28 2023-07-08 jrick }
282 24df9a28 2023-07-08 jrick
283 24df9a28 2023-07-08 jrick if (*abspath == NULL) {
284 24df9a28 2023-07-08 jrick *abspath = strdup("/usr/bin/vi");
285 24df9a28 2023-07-08 jrick if (*abspath == NULL)
286 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
287 24df9a28 2023-07-08 jrick }
288 24df9a28 2023-07-08 jrick
289 24df9a28 2023-07-08 jrick return NULL;
290 24df9a28 2023-07-08 jrick }
291 24df9a28 2023-07-08 jrick
292 24df9a28 2023-07-08 jrick static const struct got_error *
293 24df9a28 2023-07-08 jrick apply_unveil(const char *repo_path, int repo_read_only,
294 24df9a28 2023-07-08 jrick const char *worktree_path)
295 24df9a28 2023-07-08 jrick {
296 24df9a28 2023-07-08 jrick const struct got_error *err;
297 24df9a28 2023-07-08 jrick
298 24df9a28 2023-07-08 jrick #ifdef PROFILE
299 24df9a28 2023-07-08 jrick if (unveil("gmon.out", "rwc") != 0)
300 24df9a28 2023-07-08 jrick return got_error_from_errno2("unveil", "gmon.out");
301 24df9a28 2023-07-08 jrick #endif
302 24df9a28 2023-07-08 jrick if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
303 24df9a28 2023-07-08 jrick return got_error_from_errno2("unveil", repo_path);
304 24df9a28 2023-07-08 jrick
305 24df9a28 2023-07-08 jrick if (worktree_path && unveil(worktree_path, "rwc") != 0)
306 24df9a28 2023-07-08 jrick return got_error_from_errno2("unveil", worktree_path);
307 24df9a28 2023-07-08 jrick
308 24df9a28 2023-07-08 jrick if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
309 24df9a28 2023-07-08 jrick return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
310 24df9a28 2023-07-08 jrick
311 24df9a28 2023-07-08 jrick err = got_privsep_unveil_exec_helpers();
312 24df9a28 2023-07-08 jrick if (err != NULL)
313 24df9a28 2023-07-08 jrick return err;
314 24df9a28 2023-07-08 jrick
315 24df9a28 2023-07-08 jrick if (unveil(NULL, NULL) != 0)
316 24df9a28 2023-07-08 jrick return got_error_from_errno("unveil");
317 24df9a28 2023-07-08 jrick
318 24df9a28 2023-07-08 jrick return NULL;
319 24df9a28 2023-07-08 jrick }
320 24df9a28 2023-07-08 jrick
321 24df9a28 2023-07-08 jrick __dead static void
322 24df9a28 2023-07-08 jrick usage_import(void)
323 24df9a28 2023-07-08 jrick {
324 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s import [-b branch] [-I pattern] [-m message] "
325 24df9a28 2023-07-08 jrick "[-r repository-path] directory\n", getprogname());
326 24df9a28 2023-07-08 jrick exit(1);
327 24df9a28 2023-07-08 jrick }
328 24df9a28 2023-07-08 jrick
329 24df9a28 2023-07-08 jrick static int
330 24df9a28 2023-07-08 jrick spawn_editor(const char *editor, const char *file)
331 24df9a28 2023-07-08 jrick {
332 24df9a28 2023-07-08 jrick pid_t pid;
333 24df9a28 2023-07-08 jrick sig_t sighup, sigint, sigquit;
334 24df9a28 2023-07-08 jrick int st = -1;
335 24df9a28 2023-07-08 jrick
336 24df9a28 2023-07-08 jrick sighup = signal(SIGHUP, SIG_IGN);
337 24df9a28 2023-07-08 jrick sigint = signal(SIGINT, SIG_IGN);
338 24df9a28 2023-07-08 jrick sigquit = signal(SIGQUIT, SIG_IGN);
339 24df9a28 2023-07-08 jrick
340 24df9a28 2023-07-08 jrick switch (pid = fork()) {
341 24df9a28 2023-07-08 jrick case -1:
342 24df9a28 2023-07-08 jrick goto doneediting;
343 24df9a28 2023-07-08 jrick case 0:
344 24df9a28 2023-07-08 jrick execl(editor, editor, file, (char *)NULL);
345 24df9a28 2023-07-08 jrick _exit(127);
346 24df9a28 2023-07-08 jrick }
347 24df9a28 2023-07-08 jrick
348 24df9a28 2023-07-08 jrick while (waitpid(pid, &st, 0) == -1)
349 24df9a28 2023-07-08 jrick if (errno != EINTR)
350 24df9a28 2023-07-08 jrick break;
351 24df9a28 2023-07-08 jrick
352 24df9a28 2023-07-08 jrick doneediting:
353 24df9a28 2023-07-08 jrick (void)signal(SIGHUP, sighup);
354 24df9a28 2023-07-08 jrick (void)signal(SIGINT, sigint);
355 24df9a28 2023-07-08 jrick (void)signal(SIGQUIT, sigquit);
356 24df9a28 2023-07-08 jrick
357 24df9a28 2023-07-08 jrick if (!WIFEXITED(st)) {
358 24df9a28 2023-07-08 jrick errno = EINTR;
359 24df9a28 2023-07-08 jrick return -1;
360 24df9a28 2023-07-08 jrick }
361 24df9a28 2023-07-08 jrick
362 24df9a28 2023-07-08 jrick return WEXITSTATUS(st);
363 24df9a28 2023-07-08 jrick }
364 24df9a28 2023-07-08 jrick
365 24df9a28 2023-07-08 jrick static const struct got_error *
366 24df9a28 2023-07-08 jrick read_logmsg(char **logmsg, size_t *len, FILE *fp, size_t filesize)
367 24df9a28 2023-07-08 jrick {
368 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
369 24df9a28 2023-07-08 jrick char *line = NULL;
370 24df9a28 2023-07-08 jrick size_t linesize = 0;
371 24df9a28 2023-07-08 jrick
372 24df9a28 2023-07-08 jrick *logmsg = NULL;
373 24df9a28 2023-07-08 jrick *len = 0;
374 24df9a28 2023-07-08 jrick
375 24df9a28 2023-07-08 jrick if (fseeko(fp, 0L, SEEK_SET) == -1)
376 24df9a28 2023-07-08 jrick return got_error_from_errno("fseeko");
377 24df9a28 2023-07-08 jrick
378 24df9a28 2023-07-08 jrick *logmsg = malloc(filesize + 1);
379 24df9a28 2023-07-08 jrick if (*logmsg == NULL)
380 24df9a28 2023-07-08 jrick return got_error_from_errno("malloc");
381 24df9a28 2023-07-08 jrick (*logmsg)[0] = '\0';
382 24df9a28 2023-07-08 jrick
383 24df9a28 2023-07-08 jrick while (getline(&line, &linesize, fp) != -1) {
384 24df9a28 2023-07-08 jrick if (line[0] == '#' || (*len == 0 && line[0] == '\n'))
385 24df9a28 2023-07-08 jrick continue; /* remove comments and leading empty lines */
386 24df9a28 2023-07-08 jrick *len = strlcat(*logmsg, line, filesize + 1);
387 24df9a28 2023-07-08 jrick if (*len >= filesize + 1) {
388 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_NO_SPACE);
389 24df9a28 2023-07-08 jrick goto done;
390 24df9a28 2023-07-08 jrick }
391 24df9a28 2023-07-08 jrick }
392 24df9a28 2023-07-08 jrick if (ferror(fp)) {
393 24df9a28 2023-07-08 jrick err = got_ferror(fp, GOT_ERR_IO);
394 24df9a28 2023-07-08 jrick goto done;
395 24df9a28 2023-07-08 jrick }
396 24df9a28 2023-07-08 jrick
397 24df9a28 2023-07-08 jrick while (*len > 0 && (*logmsg)[*len - 1] == '\n') {
398 24df9a28 2023-07-08 jrick (*logmsg)[*len - 1] = '\0';
399 24df9a28 2023-07-08 jrick (*len)--;
400 24df9a28 2023-07-08 jrick }
401 24df9a28 2023-07-08 jrick done:
402 24df9a28 2023-07-08 jrick free(line);
403 24df9a28 2023-07-08 jrick if (err) {
404 24df9a28 2023-07-08 jrick free(*logmsg);
405 24df9a28 2023-07-08 jrick *logmsg = NULL;
406 24df9a28 2023-07-08 jrick *len = 0;
407 24df9a28 2023-07-08 jrick }
408 24df9a28 2023-07-08 jrick return err;
409 24df9a28 2023-07-08 jrick }
410 24df9a28 2023-07-08 jrick
411 24df9a28 2023-07-08 jrick static const struct got_error *
412 24df9a28 2023-07-08 jrick edit_logmsg(char **logmsg, const char *editor, const char *logmsg_path,
413 24df9a28 2023-07-08 jrick const char *initial_content, size_t initial_content_len,
414 24df9a28 2023-07-08 jrick int require_modification)
415 24df9a28 2023-07-08 jrick {
416 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
417 24df9a28 2023-07-08 jrick struct stat st, st2;
418 24df9a28 2023-07-08 jrick FILE *fp = NULL;
419 24df9a28 2023-07-08 jrick size_t logmsg_len;
420 24df9a28 2023-07-08 jrick
421 24df9a28 2023-07-08 jrick *logmsg = NULL;
422 24df9a28 2023-07-08 jrick
423 24df9a28 2023-07-08 jrick if (stat(logmsg_path, &st) == -1)
424 24df9a28 2023-07-08 jrick return got_error_from_errno2("stat", logmsg_path);
425 24df9a28 2023-07-08 jrick
426 24df9a28 2023-07-08 jrick if (spawn_editor(editor, logmsg_path) == -1)
427 24df9a28 2023-07-08 jrick return got_error_from_errno("failed spawning editor");
428 24df9a28 2023-07-08 jrick
429 24df9a28 2023-07-08 jrick if (require_modification) {
430 24df9a28 2023-07-08 jrick struct timespec timeout;
431 24df9a28 2023-07-08 jrick
432 24df9a28 2023-07-08 jrick timeout.tv_sec = 0;
433 24df9a28 2023-07-08 jrick timeout.tv_nsec = 1;
434 24df9a28 2023-07-08 jrick nanosleep(&timeout, NULL);
435 24df9a28 2023-07-08 jrick }
436 24df9a28 2023-07-08 jrick
437 24df9a28 2023-07-08 jrick if (stat(logmsg_path, &st2) == -1)
438 5a6c61ae 2023-07-15 naddy return got_error_from_errno2("stat", logmsg_path);
439 24df9a28 2023-07-08 jrick
440 24df9a28 2023-07-08 jrick if (require_modification && st.st_size == st2.st_size &&
441 24df9a28 2023-07-08 jrick timespeccmp(&st.st_mtim, &st2.st_mtim, ==))
442 24df9a28 2023-07-08 jrick return got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
443 24df9a28 2023-07-08 jrick "no changes made to commit message, aborting");
444 24df9a28 2023-07-08 jrick
445 24df9a28 2023-07-08 jrick fp = fopen(logmsg_path, "re");
446 24df9a28 2023-07-08 jrick if (fp == NULL) {
447 24df9a28 2023-07-08 jrick err = got_error_from_errno("fopen");
448 24df9a28 2023-07-08 jrick goto done;
449 24df9a28 2023-07-08 jrick }
450 24df9a28 2023-07-08 jrick
451 24df9a28 2023-07-08 jrick /* strip comments and leading/trailing newlines */
452 24df9a28 2023-07-08 jrick err = read_logmsg(logmsg, &logmsg_len, fp, st2.st_size);
453 24df9a28 2023-07-08 jrick if (err)
454 24df9a28 2023-07-08 jrick goto done;
455 24df9a28 2023-07-08 jrick if (logmsg_len == 0) {
456 24df9a28 2023-07-08 jrick err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
457 24df9a28 2023-07-08 jrick "commit message cannot be empty, aborting");
458 24df9a28 2023-07-08 jrick goto done;
459 24df9a28 2023-07-08 jrick }
460 24df9a28 2023-07-08 jrick done:
461 24df9a28 2023-07-08 jrick if (fp && fclose(fp) == EOF && err == NULL)
462 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
463 24df9a28 2023-07-08 jrick if (err) {
464 24df9a28 2023-07-08 jrick free(*logmsg);
465 24df9a28 2023-07-08 jrick *logmsg = NULL;
466 24df9a28 2023-07-08 jrick }
467 24df9a28 2023-07-08 jrick return err;
468 24df9a28 2023-07-08 jrick }
469 24df9a28 2023-07-08 jrick
470 24df9a28 2023-07-08 jrick static const struct got_error *
471 24df9a28 2023-07-08 jrick collect_import_msg(char **logmsg, char **logmsg_path, const char *editor,
472 24df9a28 2023-07-08 jrick const char *path_dir, const char *branch_name)
473 24df9a28 2023-07-08 jrick {
474 24df9a28 2023-07-08 jrick char *initial_content = NULL;
475 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
476 24df9a28 2023-07-08 jrick int initial_content_len;
477 24df9a28 2023-07-08 jrick int fd = -1;
478 24df9a28 2023-07-08 jrick
479 24df9a28 2023-07-08 jrick initial_content_len = asprintf(&initial_content,
480 24df9a28 2023-07-08 jrick "\n# %s to be imported to branch %s\n", path_dir,
481 24df9a28 2023-07-08 jrick branch_name);
482 24df9a28 2023-07-08 jrick if (initial_content_len == -1)
483 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
484 24df9a28 2023-07-08 jrick
485 24df9a28 2023-07-08 jrick err = got_opentemp_named_fd(logmsg_path, &fd,
486 24df9a28 2023-07-08 jrick GOT_TMPDIR_STR "/got-importmsg", "");
487 24df9a28 2023-07-08 jrick if (err)
488 24df9a28 2023-07-08 jrick goto done;
489 24df9a28 2023-07-08 jrick
490 24df9a28 2023-07-08 jrick if (write(fd, initial_content, initial_content_len) == -1) {
491 24df9a28 2023-07-08 jrick err = got_error_from_errno2("write", *logmsg_path);
492 24df9a28 2023-07-08 jrick goto done;
493 24df9a28 2023-07-08 jrick }
494 24df9a28 2023-07-08 jrick if (close(fd) == -1) {
495 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", *logmsg_path);
496 24df9a28 2023-07-08 jrick goto done;
497 24df9a28 2023-07-08 jrick }
498 24df9a28 2023-07-08 jrick fd = -1;
499 24df9a28 2023-07-08 jrick
500 24df9a28 2023-07-08 jrick err = edit_logmsg(logmsg, editor, *logmsg_path, initial_content,
501 24df9a28 2023-07-08 jrick initial_content_len, 1);
502 24df9a28 2023-07-08 jrick done:
503 24df9a28 2023-07-08 jrick if (fd != -1 && close(fd) == -1 && err == NULL)
504 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", *logmsg_path);
505 24df9a28 2023-07-08 jrick free(initial_content);
506 24df9a28 2023-07-08 jrick if (err) {
507 24df9a28 2023-07-08 jrick free(*logmsg_path);
508 24df9a28 2023-07-08 jrick *logmsg_path = NULL;
509 24df9a28 2023-07-08 jrick }
510 24df9a28 2023-07-08 jrick return err;
511 24df9a28 2023-07-08 jrick }
512 24df9a28 2023-07-08 jrick
513 24df9a28 2023-07-08 jrick static const struct got_error *
514 24df9a28 2023-07-08 jrick import_progress(void *arg, const char *path)
515 24df9a28 2023-07-08 jrick {
516 24df9a28 2023-07-08 jrick printf("A %s\n", path);
517 24df9a28 2023-07-08 jrick return NULL;
518 24df9a28 2023-07-08 jrick }
519 24df9a28 2023-07-08 jrick
520 24df9a28 2023-07-08 jrick static const struct got_error *
521 24df9a28 2023-07-08 jrick valid_author(const char *author)
522 24df9a28 2023-07-08 jrick {
523 24df9a28 2023-07-08 jrick const char *email = author;
524 24df9a28 2023-07-08 jrick
525 24df9a28 2023-07-08 jrick /*
526 24df9a28 2023-07-08 jrick * Git' expects the author (or committer) to be in the form
527 24df9a28 2023-07-08 jrick * "name <email>", which are mostly free form (see the
528 24df9a28 2023-07-08 jrick * "committer" description in git-fast-import(1)). We're only
529 24df9a28 2023-07-08 jrick * doing this to avoid git's object parser breaking on commits
530 24df9a28 2023-07-08 jrick * we create.
531 24df9a28 2023-07-08 jrick */
532 24df9a28 2023-07-08 jrick
533 24df9a28 2023-07-08 jrick while (*author && *author != '\n' && *author != '<' && *author != '>')
534 24df9a28 2023-07-08 jrick author++;
535 24df9a28 2023-07-08 jrick if (author != email && *author == '<' && *(author - 1) != ' ')
536 24df9a28 2023-07-08 jrick return got_error_fmt(GOT_ERR_COMMIT_BAD_AUTHOR, "%s: space "
537 24df9a28 2023-07-08 jrick "between author name and email required", email);
538 24df9a28 2023-07-08 jrick if (*author++ != '<')
539 24df9a28 2023-07-08 jrick return got_error_fmt(GOT_ERR_COMMIT_NO_EMAIL, "%s", email);
540 24df9a28 2023-07-08 jrick while (*author && *author != '\n' && *author != '<' && *author != '>')
541 24df9a28 2023-07-08 jrick author++;
542 24df9a28 2023-07-08 jrick if (strcmp(author, ">") != 0)
543 24df9a28 2023-07-08 jrick return got_error_fmt(GOT_ERR_COMMIT_NO_EMAIL, "%s", email);
544 24df9a28 2023-07-08 jrick return NULL;
545 24df9a28 2023-07-08 jrick }
546 24df9a28 2023-07-08 jrick
547 24df9a28 2023-07-08 jrick static const struct got_error *
548 24df9a28 2023-07-08 jrick get_author(char **author, struct got_repository *repo,
549 24df9a28 2023-07-08 jrick struct got_worktree *worktree)
550 24df9a28 2023-07-08 jrick {
551 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
552 24df9a28 2023-07-08 jrick const char *got_author = NULL, *name, *email;
553 24df9a28 2023-07-08 jrick const struct got_gotconfig *worktree_conf = NULL, *repo_conf = NULL;
554 24df9a28 2023-07-08 jrick
555 24df9a28 2023-07-08 jrick *author = NULL;
556 24df9a28 2023-07-08 jrick
557 24df9a28 2023-07-08 jrick if (worktree)
558 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
559 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
560 24df9a28 2023-07-08 jrick
561 24df9a28 2023-07-08 jrick /*
562 24df9a28 2023-07-08 jrick * Priority of potential author information sources, from most
563 24df9a28 2023-07-08 jrick * significant to least significant:
564 24df9a28 2023-07-08 jrick * 1) work tree's .got/got.conf file
565 24df9a28 2023-07-08 jrick * 2) repository's got.conf file
566 24df9a28 2023-07-08 jrick * 3) repository's git config file
567 24df9a28 2023-07-08 jrick * 4) environment variables
568 24df9a28 2023-07-08 jrick * 5) global git config files (in user's home directory or /etc)
569 24df9a28 2023-07-08 jrick */
570 24df9a28 2023-07-08 jrick
571 24df9a28 2023-07-08 jrick if (worktree_conf)
572 24df9a28 2023-07-08 jrick got_author = got_gotconfig_get_author(worktree_conf);
573 24df9a28 2023-07-08 jrick if (got_author == NULL)
574 24df9a28 2023-07-08 jrick got_author = got_gotconfig_get_author(repo_conf);
575 24df9a28 2023-07-08 jrick if (got_author == NULL) {
576 24df9a28 2023-07-08 jrick name = got_repo_get_gitconfig_author_name(repo);
577 24df9a28 2023-07-08 jrick email = got_repo_get_gitconfig_author_email(repo);
578 24df9a28 2023-07-08 jrick if (name && email) {
579 24df9a28 2023-07-08 jrick if (asprintf(author, "%s <%s>", name, email) == -1)
580 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
581 24df9a28 2023-07-08 jrick return NULL;
582 24df9a28 2023-07-08 jrick }
583 24df9a28 2023-07-08 jrick
584 24df9a28 2023-07-08 jrick got_author = getenv("GOT_AUTHOR");
585 24df9a28 2023-07-08 jrick if (got_author == NULL) {
586 24df9a28 2023-07-08 jrick name = got_repo_get_global_gitconfig_author_name(repo);
587 24df9a28 2023-07-08 jrick email = got_repo_get_global_gitconfig_author_email(
588 24df9a28 2023-07-08 jrick repo);
589 24df9a28 2023-07-08 jrick if (name && email) {
590 24df9a28 2023-07-08 jrick if (asprintf(author, "%s <%s>", name, email)
591 24df9a28 2023-07-08 jrick == -1)
592 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
593 24df9a28 2023-07-08 jrick return NULL;
594 24df9a28 2023-07-08 jrick }
595 24df9a28 2023-07-08 jrick /* TODO: Look up user in password database? */
596 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_COMMIT_NO_AUTHOR);
597 24df9a28 2023-07-08 jrick }
598 24df9a28 2023-07-08 jrick }
599 24df9a28 2023-07-08 jrick
600 24df9a28 2023-07-08 jrick *author = strdup(got_author);
601 24df9a28 2023-07-08 jrick if (*author == NULL)
602 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
603 24df9a28 2023-07-08 jrick
604 24df9a28 2023-07-08 jrick err = valid_author(*author);
605 24df9a28 2023-07-08 jrick if (err) {
606 24df9a28 2023-07-08 jrick free(*author);
607 24df9a28 2023-07-08 jrick *author = NULL;
608 24df9a28 2023-07-08 jrick }
609 24df9a28 2023-07-08 jrick return err;
610 24df9a28 2023-07-08 jrick }
611 24df9a28 2023-07-08 jrick
612 24df9a28 2023-07-08 jrick static const struct got_error *
613 24df9a28 2023-07-08 jrick get_allowed_signers(char **allowed_signers, struct got_repository *repo,
614 24df9a28 2023-07-08 jrick struct got_worktree *worktree)
615 24df9a28 2023-07-08 jrick {
616 24df9a28 2023-07-08 jrick const char *got_allowed_signers = NULL;
617 24df9a28 2023-07-08 jrick const struct got_gotconfig *worktree_conf = NULL, *repo_conf = NULL;
618 24df9a28 2023-07-08 jrick
619 24df9a28 2023-07-08 jrick *allowed_signers = NULL;
620 24df9a28 2023-07-08 jrick
621 24df9a28 2023-07-08 jrick if (worktree)
622 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
623 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
624 24df9a28 2023-07-08 jrick
625 24df9a28 2023-07-08 jrick /*
626 24df9a28 2023-07-08 jrick * Priority of potential author information sources, from most
627 24df9a28 2023-07-08 jrick * significant to least significant:
628 24df9a28 2023-07-08 jrick * 1) work tree's .got/got.conf file
629 24df9a28 2023-07-08 jrick * 2) repository's got.conf file
630 24df9a28 2023-07-08 jrick */
631 24df9a28 2023-07-08 jrick
632 24df9a28 2023-07-08 jrick if (worktree_conf)
633 24df9a28 2023-07-08 jrick got_allowed_signers = got_gotconfig_get_allowed_signers_file(
634 24df9a28 2023-07-08 jrick worktree_conf);
635 24df9a28 2023-07-08 jrick if (got_allowed_signers == NULL)
636 24df9a28 2023-07-08 jrick got_allowed_signers = got_gotconfig_get_allowed_signers_file(
637 24df9a28 2023-07-08 jrick repo_conf);
638 24df9a28 2023-07-08 jrick
639 24df9a28 2023-07-08 jrick if (got_allowed_signers) {
640 24df9a28 2023-07-08 jrick *allowed_signers = strdup(got_allowed_signers);
641 24df9a28 2023-07-08 jrick if (*allowed_signers == NULL)
642 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
643 24df9a28 2023-07-08 jrick }
644 24df9a28 2023-07-08 jrick return NULL;
645 24df9a28 2023-07-08 jrick }
646 24df9a28 2023-07-08 jrick
647 24df9a28 2023-07-08 jrick static const struct got_error *
648 24df9a28 2023-07-08 jrick get_revoked_signers(char **revoked_signers, struct got_repository *repo,
649 24df9a28 2023-07-08 jrick struct got_worktree *worktree)
650 24df9a28 2023-07-08 jrick {
651 24df9a28 2023-07-08 jrick const char *got_revoked_signers = NULL;
652 24df9a28 2023-07-08 jrick const struct got_gotconfig *worktree_conf = NULL, *repo_conf = NULL;
653 24df9a28 2023-07-08 jrick
654 24df9a28 2023-07-08 jrick *revoked_signers = NULL;
655 24df9a28 2023-07-08 jrick
656 24df9a28 2023-07-08 jrick if (worktree)
657 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
658 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
659 24df9a28 2023-07-08 jrick
660 24df9a28 2023-07-08 jrick /*
661 24df9a28 2023-07-08 jrick * Priority of potential author information sources, from most
662 24df9a28 2023-07-08 jrick * significant to least significant:
663 24df9a28 2023-07-08 jrick * 1) work tree's .got/got.conf file
664 24df9a28 2023-07-08 jrick * 2) repository's got.conf file
665 24df9a28 2023-07-08 jrick */
666 24df9a28 2023-07-08 jrick
667 24df9a28 2023-07-08 jrick if (worktree_conf)
668 24df9a28 2023-07-08 jrick got_revoked_signers = got_gotconfig_get_revoked_signers_file(
669 24df9a28 2023-07-08 jrick worktree_conf);
670 24df9a28 2023-07-08 jrick if (got_revoked_signers == NULL)
671 24df9a28 2023-07-08 jrick got_revoked_signers = got_gotconfig_get_revoked_signers_file(
672 24df9a28 2023-07-08 jrick repo_conf);
673 24df9a28 2023-07-08 jrick
674 24df9a28 2023-07-08 jrick if (got_revoked_signers) {
675 24df9a28 2023-07-08 jrick *revoked_signers = strdup(got_revoked_signers);
676 24df9a28 2023-07-08 jrick if (*revoked_signers == NULL)
677 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
678 24df9a28 2023-07-08 jrick }
679 24df9a28 2023-07-08 jrick return NULL;
680 24df9a28 2023-07-08 jrick }
681 24df9a28 2023-07-08 jrick
682 24df9a28 2023-07-08 jrick static const char *
683 24df9a28 2023-07-08 jrick get_signer_id(struct got_repository *repo, struct got_worktree *worktree)
684 24df9a28 2023-07-08 jrick {
685 24df9a28 2023-07-08 jrick const char *got_signer_id = NULL;
686 24df9a28 2023-07-08 jrick const struct got_gotconfig *worktree_conf = NULL, *repo_conf = NULL;
687 24df9a28 2023-07-08 jrick
688 24df9a28 2023-07-08 jrick if (worktree)
689 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
690 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
691 24df9a28 2023-07-08 jrick
692 24df9a28 2023-07-08 jrick /*
693 24df9a28 2023-07-08 jrick * Priority of potential author information sources, from most
694 24df9a28 2023-07-08 jrick * significant to least significant:
695 24df9a28 2023-07-08 jrick * 1) work tree's .got/got.conf file
696 24df9a28 2023-07-08 jrick * 2) repository's got.conf file
697 24df9a28 2023-07-08 jrick */
698 24df9a28 2023-07-08 jrick
699 24df9a28 2023-07-08 jrick if (worktree_conf)
700 24df9a28 2023-07-08 jrick got_signer_id = got_gotconfig_get_signer_id(worktree_conf);
701 24df9a28 2023-07-08 jrick if (got_signer_id == NULL)
702 24df9a28 2023-07-08 jrick got_signer_id = got_gotconfig_get_signer_id(repo_conf);
703 24df9a28 2023-07-08 jrick
704 24df9a28 2023-07-08 jrick return got_signer_id;
705 24df9a28 2023-07-08 jrick }
706 24df9a28 2023-07-08 jrick
707 24df9a28 2023-07-08 jrick static const struct got_error *
708 24df9a28 2023-07-08 jrick get_gitconfig_path(char **gitconfig_path)
709 24df9a28 2023-07-08 jrick {
710 24df9a28 2023-07-08 jrick const char *homedir = getenv("HOME");
711 24df9a28 2023-07-08 jrick
712 24df9a28 2023-07-08 jrick *gitconfig_path = NULL;
713 24df9a28 2023-07-08 jrick if (homedir) {
714 24df9a28 2023-07-08 jrick if (asprintf(gitconfig_path, "%s/.gitconfig", homedir) == -1)
715 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
716 24df9a28 2023-07-08 jrick
717 24df9a28 2023-07-08 jrick }
718 24df9a28 2023-07-08 jrick return NULL;
719 24df9a28 2023-07-08 jrick }
720 24df9a28 2023-07-08 jrick
721 24df9a28 2023-07-08 jrick static const struct got_error *
722 24df9a28 2023-07-08 jrick cmd_import(int argc, char *argv[])
723 24df9a28 2023-07-08 jrick {
724 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
725 24df9a28 2023-07-08 jrick char *path_dir = NULL, *repo_path = NULL, *logmsg = NULL;
726 24df9a28 2023-07-08 jrick char *gitconfig_path = NULL, *editor = NULL, *author = NULL;
727 24df9a28 2023-07-08 jrick const char *branch_name = NULL;
728 24df9a28 2023-07-08 jrick char *id_str = NULL, *logmsg_path = NULL;
729 24df9a28 2023-07-08 jrick char refname[PATH_MAX] = "refs/heads/";
730 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
731 24df9a28 2023-07-08 jrick struct got_reference *branch_ref = NULL, *head_ref = NULL;
732 24df9a28 2023-07-08 jrick struct got_object_id *new_commit_id = NULL;
733 24df9a28 2023-07-08 jrick int ch, n = 0;
734 24df9a28 2023-07-08 jrick struct got_pathlist_head ignores;
735 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
736 24df9a28 2023-07-08 jrick int preserve_logmsg = 0;
737 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
738 24df9a28 2023-07-08 jrick
739 24df9a28 2023-07-08 jrick TAILQ_INIT(&ignores);
740 24df9a28 2023-07-08 jrick
741 24df9a28 2023-07-08 jrick #ifndef PROFILE
742 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
743 24df9a28 2023-07-08 jrick "unveil",
744 24df9a28 2023-07-08 jrick NULL) == -1)
745 24df9a28 2023-07-08 jrick err(1, "pledge");
746 24df9a28 2023-07-08 jrick #endif
747 24df9a28 2023-07-08 jrick
748 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "b:I:m:r:")) != -1) {
749 24df9a28 2023-07-08 jrick switch (ch) {
750 24df9a28 2023-07-08 jrick case 'b':
751 24df9a28 2023-07-08 jrick branch_name = optarg;
752 24df9a28 2023-07-08 jrick break;
753 24df9a28 2023-07-08 jrick case 'I':
754 24df9a28 2023-07-08 jrick if (optarg[0] == '\0')
755 24df9a28 2023-07-08 jrick break;
756 24df9a28 2023-07-08 jrick error = got_pathlist_insert(&pe, &ignores, optarg,
757 24df9a28 2023-07-08 jrick NULL);
758 24df9a28 2023-07-08 jrick if (error)
759 24df9a28 2023-07-08 jrick goto done;
760 24df9a28 2023-07-08 jrick break;
761 24df9a28 2023-07-08 jrick case 'm':
762 24df9a28 2023-07-08 jrick logmsg = strdup(optarg);
763 24df9a28 2023-07-08 jrick if (logmsg == NULL) {
764 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
765 24df9a28 2023-07-08 jrick goto done;
766 24df9a28 2023-07-08 jrick }
767 24df9a28 2023-07-08 jrick break;
768 24df9a28 2023-07-08 jrick case 'r':
769 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
770 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
771 24df9a28 2023-07-08 jrick error = got_error_from_errno2("realpath",
772 24df9a28 2023-07-08 jrick optarg);
773 24df9a28 2023-07-08 jrick goto done;
774 24df9a28 2023-07-08 jrick }
775 24df9a28 2023-07-08 jrick break;
776 24df9a28 2023-07-08 jrick default:
777 24df9a28 2023-07-08 jrick usage_import();
778 24df9a28 2023-07-08 jrick /* NOTREACHED */
779 24df9a28 2023-07-08 jrick }
780 24df9a28 2023-07-08 jrick }
781 24df9a28 2023-07-08 jrick
782 24df9a28 2023-07-08 jrick argc -= optind;
783 24df9a28 2023-07-08 jrick argv += optind;
784 24df9a28 2023-07-08 jrick
785 24df9a28 2023-07-08 jrick if (argc != 1)
786 24df9a28 2023-07-08 jrick usage_import();
787 24df9a28 2023-07-08 jrick
788 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
789 24df9a28 2023-07-08 jrick repo_path = getcwd(NULL, 0);
790 24df9a28 2023-07-08 jrick if (repo_path == NULL)
791 24df9a28 2023-07-08 jrick return got_error_from_errno("getcwd");
792 24df9a28 2023-07-08 jrick }
793 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
794 24df9a28 2023-07-08 jrick error = get_gitconfig_path(&gitconfig_path);
795 24df9a28 2023-07-08 jrick if (error)
796 24df9a28 2023-07-08 jrick goto done;
797 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
798 24df9a28 2023-07-08 jrick if (error != NULL)
799 24df9a28 2023-07-08 jrick goto done;
800 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, gitconfig_path, pack_fds);
801 24df9a28 2023-07-08 jrick if (error)
802 24df9a28 2023-07-08 jrick goto done;
803 24df9a28 2023-07-08 jrick
804 24df9a28 2023-07-08 jrick error = get_author(&author, repo, NULL);
805 24df9a28 2023-07-08 jrick if (error)
806 24df9a28 2023-07-08 jrick return error;
807 24df9a28 2023-07-08 jrick
808 24df9a28 2023-07-08 jrick /*
809 24df9a28 2023-07-08 jrick * Don't let the user create a branch name with a leading '-'.
810 24df9a28 2023-07-08 jrick * While technically a valid reference name, this case is usually
811 24df9a28 2023-07-08 jrick * an unintended typo.
812 24df9a28 2023-07-08 jrick */
813 24df9a28 2023-07-08 jrick if (branch_name && branch_name[0] == '-')
814 24df9a28 2023-07-08 jrick return got_error_path(branch_name, GOT_ERR_REF_NAME_MINUS);
815 24df9a28 2023-07-08 jrick
816 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
817 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_REF)
818 24df9a28 2023-07-08 jrick goto done;
819 24df9a28 2023-07-08 jrick
820 24df9a28 2023-07-08 jrick if (branch_name)
821 24df9a28 2023-07-08 jrick n = strlcat(refname, branch_name, sizeof(refname));
822 24df9a28 2023-07-08 jrick else if (head_ref && got_ref_is_symbolic(head_ref))
823 24df9a28 2023-07-08 jrick n = strlcpy(refname, got_ref_get_symref_target(head_ref),
824 24df9a28 2023-07-08 jrick sizeof(refname));
825 24df9a28 2023-07-08 jrick else
826 24df9a28 2023-07-08 jrick n = strlcat(refname, "main", sizeof(refname));
827 24df9a28 2023-07-08 jrick if (n >= sizeof(refname)) {
828 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_NO_SPACE);
829 24df9a28 2023-07-08 jrick goto done;
830 24df9a28 2023-07-08 jrick }
831 24df9a28 2023-07-08 jrick
832 24df9a28 2023-07-08 jrick error = got_ref_open(&branch_ref, repo, refname, 0);
833 24df9a28 2023-07-08 jrick if (error) {
834 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_REF)
835 24df9a28 2023-07-08 jrick goto done;
836 24df9a28 2023-07-08 jrick } else {
837 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_BRANCH_EXISTS,
838 24df9a28 2023-07-08 jrick "import target branch already exists");
839 24df9a28 2023-07-08 jrick goto done;
840 24df9a28 2023-07-08 jrick }
841 24df9a28 2023-07-08 jrick
842 24df9a28 2023-07-08 jrick path_dir = realpath(argv[0], NULL);
843 24df9a28 2023-07-08 jrick if (path_dir == NULL) {
844 24df9a28 2023-07-08 jrick error = got_error_from_errno2("realpath", argv[0]);
845 24df9a28 2023-07-08 jrick goto done;
846 24df9a28 2023-07-08 jrick }
847 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(path_dir);
848 24df9a28 2023-07-08 jrick
849 24df9a28 2023-07-08 jrick /*
850 24df9a28 2023-07-08 jrick * unveil(2) traverses exec(2); if an editor is used we have
851 24df9a28 2023-07-08 jrick * to apply unveil after the log message has been written.
852 24df9a28 2023-07-08 jrick */
853 24df9a28 2023-07-08 jrick if (logmsg == NULL || *logmsg == '\0') {
854 24df9a28 2023-07-08 jrick error = get_editor(&editor);
855 24df9a28 2023-07-08 jrick if (error)
856 24df9a28 2023-07-08 jrick goto done;
857 24df9a28 2023-07-08 jrick free(logmsg);
858 24df9a28 2023-07-08 jrick error = collect_import_msg(&logmsg, &logmsg_path, editor,
859 24df9a28 2023-07-08 jrick path_dir, refname);
860 24df9a28 2023-07-08 jrick if (error) {
861 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_COMMIT_MSG_EMPTY &&
862 24df9a28 2023-07-08 jrick logmsg_path != NULL)
863 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
864 24df9a28 2023-07-08 jrick goto done;
865 24df9a28 2023-07-08 jrick }
866 24df9a28 2023-07-08 jrick }
867 24df9a28 2023-07-08 jrick
868 24df9a28 2023-07-08 jrick if (unveil(path_dir, "r") != 0) {
869 24df9a28 2023-07-08 jrick error = got_error_from_errno2("unveil", path_dir);
870 24df9a28 2023-07-08 jrick if (logmsg_path)
871 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
872 24df9a28 2023-07-08 jrick goto done;
873 24df9a28 2023-07-08 jrick }
874 24df9a28 2023-07-08 jrick
875 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0, NULL);
876 24df9a28 2023-07-08 jrick if (error) {
877 24df9a28 2023-07-08 jrick if (logmsg_path)
878 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
879 24df9a28 2023-07-08 jrick goto done;
880 24df9a28 2023-07-08 jrick }
881 24df9a28 2023-07-08 jrick
882 24df9a28 2023-07-08 jrick error = got_repo_import(&new_commit_id, path_dir, logmsg,
883 24df9a28 2023-07-08 jrick author, &ignores, repo, import_progress, NULL);
884 24df9a28 2023-07-08 jrick if (error) {
885 24df9a28 2023-07-08 jrick if (logmsg_path)
886 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
887 24df9a28 2023-07-08 jrick goto done;
888 24df9a28 2023-07-08 jrick }
889 24df9a28 2023-07-08 jrick
890 24df9a28 2023-07-08 jrick error = got_ref_alloc(&branch_ref, refname, new_commit_id);
891 24df9a28 2023-07-08 jrick if (error) {
892 24df9a28 2023-07-08 jrick if (logmsg_path)
893 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
894 24df9a28 2023-07-08 jrick goto done;
895 24df9a28 2023-07-08 jrick }
896 24df9a28 2023-07-08 jrick
897 24df9a28 2023-07-08 jrick error = got_ref_write(branch_ref, repo);
898 24df9a28 2023-07-08 jrick if (error) {
899 24df9a28 2023-07-08 jrick if (logmsg_path)
900 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
901 24df9a28 2023-07-08 jrick goto done;
902 24df9a28 2023-07-08 jrick }
903 24df9a28 2023-07-08 jrick
904 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str, new_commit_id);
905 24df9a28 2023-07-08 jrick if (error) {
906 24df9a28 2023-07-08 jrick if (logmsg_path)
907 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
908 24df9a28 2023-07-08 jrick goto done;
909 24df9a28 2023-07-08 jrick }
910 24df9a28 2023-07-08 jrick
911 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
912 24df9a28 2023-07-08 jrick if (error) {
913 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_REF) {
914 24df9a28 2023-07-08 jrick if (logmsg_path)
915 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
916 24df9a28 2023-07-08 jrick goto done;
917 24df9a28 2023-07-08 jrick }
918 24df9a28 2023-07-08 jrick
919 24df9a28 2023-07-08 jrick error = got_ref_alloc_symref(&head_ref, GOT_REF_HEAD,
920 24df9a28 2023-07-08 jrick branch_ref);
921 24df9a28 2023-07-08 jrick if (error) {
922 24df9a28 2023-07-08 jrick if (logmsg_path)
923 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
924 24df9a28 2023-07-08 jrick goto done;
925 24df9a28 2023-07-08 jrick }
926 24df9a28 2023-07-08 jrick
927 24df9a28 2023-07-08 jrick error = got_ref_write(head_ref, repo);
928 24df9a28 2023-07-08 jrick if (error) {
929 24df9a28 2023-07-08 jrick if (logmsg_path)
930 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
931 24df9a28 2023-07-08 jrick goto done;
932 24df9a28 2023-07-08 jrick }
933 24df9a28 2023-07-08 jrick }
934 24df9a28 2023-07-08 jrick
935 24df9a28 2023-07-08 jrick printf("Created branch %s with commit %s\n",
936 24df9a28 2023-07-08 jrick got_ref_get_name(branch_ref), id_str);
937 24df9a28 2023-07-08 jrick done:
938 24df9a28 2023-07-08 jrick if (pack_fds) {
939 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
940 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
941 24df9a28 2023-07-08 jrick if (error == NULL)
942 24df9a28 2023-07-08 jrick error = pack_err;
943 24df9a28 2023-07-08 jrick }
944 24df9a28 2023-07-08 jrick if (repo) {
945 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
946 24df9a28 2023-07-08 jrick if (error == NULL)
947 24df9a28 2023-07-08 jrick error = close_err;
948 24df9a28 2023-07-08 jrick }
949 24df9a28 2023-07-08 jrick if (preserve_logmsg) {
950 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: log message preserved in %s\n",
951 24df9a28 2023-07-08 jrick getprogname(), logmsg_path);
952 24df9a28 2023-07-08 jrick } else if (logmsg_path && unlink(logmsg_path) == -1 && error == NULL)
953 24df9a28 2023-07-08 jrick error = got_error_from_errno2("unlink", logmsg_path);
954 24df9a28 2023-07-08 jrick free(logmsg);
955 24df9a28 2023-07-08 jrick free(logmsg_path);
956 24df9a28 2023-07-08 jrick free(repo_path);
957 24df9a28 2023-07-08 jrick free(editor);
958 24df9a28 2023-07-08 jrick free(new_commit_id);
959 24df9a28 2023-07-08 jrick free(id_str);
960 24df9a28 2023-07-08 jrick free(author);
961 24df9a28 2023-07-08 jrick free(gitconfig_path);
962 24df9a28 2023-07-08 jrick if (branch_ref)
963 24df9a28 2023-07-08 jrick got_ref_close(branch_ref);
964 24df9a28 2023-07-08 jrick if (head_ref)
965 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
966 24df9a28 2023-07-08 jrick return error;
967 24df9a28 2023-07-08 jrick }
968 24df9a28 2023-07-08 jrick
969 24df9a28 2023-07-08 jrick __dead static void
970 24df9a28 2023-07-08 jrick usage_clone(void)
971 24df9a28 2023-07-08 jrick {
972 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s clone [-almqv] [-b branch] [-R reference] "
973 24df9a28 2023-07-08 jrick "repository-URL [directory]\n", getprogname());
974 24df9a28 2023-07-08 jrick exit(1);
975 24df9a28 2023-07-08 jrick }
976 24df9a28 2023-07-08 jrick
977 24df9a28 2023-07-08 jrick struct got_fetch_progress_arg {
978 24df9a28 2023-07-08 jrick char last_scaled_size[FMT_SCALED_STRSIZE];
979 24df9a28 2023-07-08 jrick int last_p_indexed;
980 24df9a28 2023-07-08 jrick int last_p_resolved;
981 24df9a28 2023-07-08 jrick int verbosity;
982 24df9a28 2023-07-08 jrick
983 24df9a28 2023-07-08 jrick struct got_repository *repo;
984 24df9a28 2023-07-08 jrick
985 24df9a28 2023-07-08 jrick int create_configs;
986 24df9a28 2023-07-08 jrick int configs_created;
987 24df9a28 2023-07-08 jrick struct {
988 24df9a28 2023-07-08 jrick struct got_pathlist_head *symrefs;
989 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_branches;
990 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_refs;
991 24df9a28 2023-07-08 jrick const char *proto;
992 24df9a28 2023-07-08 jrick const char *host;
993 24df9a28 2023-07-08 jrick const char *port;
994 24df9a28 2023-07-08 jrick const char *remote_repo_path;
995 24df9a28 2023-07-08 jrick const char *git_url;
996 24df9a28 2023-07-08 jrick int fetch_all_branches;
997 24df9a28 2023-07-08 jrick int mirror_references;
998 24df9a28 2023-07-08 jrick } config_info;
999 24df9a28 2023-07-08 jrick };
1000 24df9a28 2023-07-08 jrick
1001 24df9a28 2023-07-08 jrick /* XXX forward declaration */
1002 24df9a28 2023-07-08 jrick static const struct got_error *
1003 24df9a28 2023-07-08 jrick create_config_files(const char *proto, const char *host, const char *port,
1004 24df9a28 2023-07-08 jrick const char *remote_repo_path, const char *git_url, int fetch_all_branches,
1005 24df9a28 2023-07-08 jrick int mirror_references, struct got_pathlist_head *symrefs,
1006 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_branches,
1007 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_refs, struct got_repository *repo);
1008 24df9a28 2023-07-08 jrick
1009 24df9a28 2023-07-08 jrick static const struct got_error *
1010 24df9a28 2023-07-08 jrick fetch_progress(void *arg, const char *message, off_t packfile_size,
1011 24df9a28 2023-07-08 jrick int nobj_total, int nobj_indexed, int nobj_loose, int nobj_resolved)
1012 24df9a28 2023-07-08 jrick {
1013 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1014 24df9a28 2023-07-08 jrick struct got_fetch_progress_arg *a = arg;
1015 24df9a28 2023-07-08 jrick char scaled_size[FMT_SCALED_STRSIZE];
1016 24df9a28 2023-07-08 jrick int p_indexed, p_resolved;
1017 24df9a28 2023-07-08 jrick int print_size = 0, print_indexed = 0, print_resolved = 0;
1018 24df9a28 2023-07-08 jrick
1019 24df9a28 2023-07-08 jrick /*
1020 24df9a28 2023-07-08 jrick * In order to allow a failed clone to be resumed with 'got fetch'
1021 24df9a28 2023-07-08 jrick * we try to create configuration files as soon as possible.
1022 24df9a28 2023-07-08 jrick * Once the server has sent information about its default branch
1023 24df9a28 2023-07-08 jrick * we have all required information.
1024 24df9a28 2023-07-08 jrick */
1025 24df9a28 2023-07-08 jrick if (a->create_configs && !a->configs_created &&
1026 24df9a28 2023-07-08 jrick !TAILQ_EMPTY(a->config_info.symrefs)) {
1027 24df9a28 2023-07-08 jrick err = create_config_files(a->config_info.proto,
1028 24df9a28 2023-07-08 jrick a->config_info.host, a->config_info.port,
1029 24df9a28 2023-07-08 jrick a->config_info.remote_repo_path,
1030 24df9a28 2023-07-08 jrick a->config_info.git_url,
1031 24df9a28 2023-07-08 jrick a->config_info.fetch_all_branches,
1032 24df9a28 2023-07-08 jrick a->config_info.mirror_references,
1033 24df9a28 2023-07-08 jrick a->config_info.symrefs,
1034 24df9a28 2023-07-08 jrick a->config_info.wanted_branches,
1035 24df9a28 2023-07-08 jrick a->config_info.wanted_refs, a->repo);
1036 24df9a28 2023-07-08 jrick if (err)
1037 24df9a28 2023-07-08 jrick return err;
1038 24df9a28 2023-07-08 jrick a->configs_created = 1;
1039 24df9a28 2023-07-08 jrick }
1040 24df9a28 2023-07-08 jrick
1041 24df9a28 2023-07-08 jrick if (a->verbosity < 0)
1042 24df9a28 2023-07-08 jrick return NULL;
1043 24df9a28 2023-07-08 jrick
1044 24df9a28 2023-07-08 jrick if (message && message[0] != '\0') {
1045 24df9a28 2023-07-08 jrick printf("\rserver: %s", message);
1046 24df9a28 2023-07-08 jrick fflush(stdout);
1047 24df9a28 2023-07-08 jrick return NULL;
1048 24df9a28 2023-07-08 jrick }
1049 24df9a28 2023-07-08 jrick
1050 24df9a28 2023-07-08 jrick if (packfile_size > 0 || nobj_indexed > 0) {
1051 24df9a28 2023-07-08 jrick if (fmt_scaled(packfile_size, scaled_size) == 0 &&
1052 24df9a28 2023-07-08 jrick (a->last_scaled_size[0] == '\0' ||
1053 24df9a28 2023-07-08 jrick strcmp(scaled_size, a->last_scaled_size)) != 0) {
1054 24df9a28 2023-07-08 jrick print_size = 1;
1055 24df9a28 2023-07-08 jrick if (strlcpy(a->last_scaled_size, scaled_size,
1056 24df9a28 2023-07-08 jrick FMT_SCALED_STRSIZE) >= FMT_SCALED_STRSIZE)
1057 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_NO_SPACE);
1058 24df9a28 2023-07-08 jrick }
1059 24df9a28 2023-07-08 jrick if (nobj_indexed > 0) {
1060 24df9a28 2023-07-08 jrick p_indexed = (nobj_indexed * 100) / nobj_total;
1061 24df9a28 2023-07-08 jrick if (p_indexed != a->last_p_indexed) {
1062 24df9a28 2023-07-08 jrick a->last_p_indexed = p_indexed;
1063 24df9a28 2023-07-08 jrick print_indexed = 1;
1064 24df9a28 2023-07-08 jrick print_size = 1;
1065 24df9a28 2023-07-08 jrick }
1066 24df9a28 2023-07-08 jrick }
1067 24df9a28 2023-07-08 jrick if (nobj_resolved > 0) {
1068 24df9a28 2023-07-08 jrick p_resolved = (nobj_resolved * 100) /
1069 24df9a28 2023-07-08 jrick (nobj_total - nobj_loose);
1070 24df9a28 2023-07-08 jrick if (p_resolved != a->last_p_resolved) {
1071 24df9a28 2023-07-08 jrick a->last_p_resolved = p_resolved;
1072 24df9a28 2023-07-08 jrick print_resolved = 1;
1073 24df9a28 2023-07-08 jrick print_indexed = 1;
1074 24df9a28 2023-07-08 jrick print_size = 1;
1075 24df9a28 2023-07-08 jrick }
1076 24df9a28 2023-07-08 jrick }
1077 24df9a28 2023-07-08 jrick
1078 24df9a28 2023-07-08 jrick }
1079 24df9a28 2023-07-08 jrick if (print_size || print_indexed || print_resolved)
1080 24df9a28 2023-07-08 jrick printf("\r");
1081 24df9a28 2023-07-08 jrick if (print_size)
1082 24df9a28 2023-07-08 jrick printf("%*s fetched", FMT_SCALED_STRSIZE - 2, scaled_size);
1083 24df9a28 2023-07-08 jrick if (print_indexed)
1084 24df9a28 2023-07-08 jrick printf("; indexing %d%%", p_indexed);
1085 24df9a28 2023-07-08 jrick if (print_resolved)
1086 24df9a28 2023-07-08 jrick printf("; resolving deltas %d%%", p_resolved);
1087 24df9a28 2023-07-08 jrick if (print_size || print_indexed || print_resolved)
1088 24df9a28 2023-07-08 jrick fflush(stdout);
1089 24df9a28 2023-07-08 jrick
1090 24df9a28 2023-07-08 jrick return NULL;
1091 24df9a28 2023-07-08 jrick }
1092 24df9a28 2023-07-08 jrick
1093 24df9a28 2023-07-08 jrick static const struct got_error *
1094 24df9a28 2023-07-08 jrick create_symref(const char *refname, struct got_reference *target_ref,
1095 24df9a28 2023-07-08 jrick int verbosity, struct got_repository *repo)
1096 24df9a28 2023-07-08 jrick {
1097 24df9a28 2023-07-08 jrick const struct got_error *err;
1098 24df9a28 2023-07-08 jrick struct got_reference *head_symref;
1099 24df9a28 2023-07-08 jrick
1100 24df9a28 2023-07-08 jrick err = got_ref_alloc_symref(&head_symref, refname, target_ref);
1101 24df9a28 2023-07-08 jrick if (err)
1102 24df9a28 2023-07-08 jrick return err;
1103 24df9a28 2023-07-08 jrick
1104 24df9a28 2023-07-08 jrick err = got_ref_write(head_symref, repo);
1105 24df9a28 2023-07-08 jrick if (err == NULL && verbosity > 0) {
1106 24df9a28 2023-07-08 jrick printf("Created reference %s: %s\n", GOT_REF_HEAD,
1107 24df9a28 2023-07-08 jrick got_ref_get_name(target_ref));
1108 24df9a28 2023-07-08 jrick }
1109 24df9a28 2023-07-08 jrick got_ref_close(head_symref);
1110 24df9a28 2023-07-08 jrick return err;
1111 24df9a28 2023-07-08 jrick }
1112 24df9a28 2023-07-08 jrick
1113 24df9a28 2023-07-08 jrick static const struct got_error *
1114 24df9a28 2023-07-08 jrick list_remote_refs(struct got_pathlist_head *symrefs,
1115 24df9a28 2023-07-08 jrick struct got_pathlist_head *refs)
1116 24df9a28 2023-07-08 jrick {
1117 24df9a28 2023-07-08 jrick const struct got_error *err;
1118 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1119 24df9a28 2023-07-08 jrick
1120 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, symrefs, entry) {
1121 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1122 24df9a28 2023-07-08 jrick const char *targetref = pe->data;
1123 24df9a28 2023-07-08 jrick
1124 24df9a28 2023-07-08 jrick printf("%s: %s\n", refname, targetref);
1125 24df9a28 2023-07-08 jrick }
1126 24df9a28 2023-07-08 jrick
1127 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, refs, entry) {
1128 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1129 24df9a28 2023-07-08 jrick struct got_object_id *id = pe->data;
1130 24df9a28 2023-07-08 jrick char *id_str;
1131 24df9a28 2023-07-08 jrick
1132 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
1133 24df9a28 2023-07-08 jrick if (err)
1134 24df9a28 2023-07-08 jrick return err;
1135 24df9a28 2023-07-08 jrick printf("%s: %s\n", refname, id_str);
1136 24df9a28 2023-07-08 jrick free(id_str);
1137 24df9a28 2023-07-08 jrick }
1138 24df9a28 2023-07-08 jrick
1139 24df9a28 2023-07-08 jrick return NULL;
1140 24df9a28 2023-07-08 jrick }
1141 24df9a28 2023-07-08 jrick
1142 24df9a28 2023-07-08 jrick static const struct got_error *
1143 24df9a28 2023-07-08 jrick create_ref(const char *refname, struct got_object_id *id,
1144 24df9a28 2023-07-08 jrick int verbosity, struct got_repository *repo)
1145 24df9a28 2023-07-08 jrick {
1146 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1147 24df9a28 2023-07-08 jrick struct got_reference *ref;
1148 24df9a28 2023-07-08 jrick char *id_str;
1149 24df9a28 2023-07-08 jrick
1150 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
1151 24df9a28 2023-07-08 jrick if (err)
1152 24df9a28 2023-07-08 jrick return err;
1153 24df9a28 2023-07-08 jrick
1154 24df9a28 2023-07-08 jrick err = got_ref_alloc(&ref, refname, id);
1155 24df9a28 2023-07-08 jrick if (err)
1156 24df9a28 2023-07-08 jrick goto done;
1157 24df9a28 2023-07-08 jrick
1158 24df9a28 2023-07-08 jrick err = got_ref_write(ref, repo);
1159 24df9a28 2023-07-08 jrick got_ref_close(ref);
1160 24df9a28 2023-07-08 jrick
1161 24df9a28 2023-07-08 jrick if (err == NULL && verbosity >= 0)
1162 24df9a28 2023-07-08 jrick printf("Created reference %s: %s\n", refname, id_str);
1163 24df9a28 2023-07-08 jrick done:
1164 24df9a28 2023-07-08 jrick free(id_str);
1165 24df9a28 2023-07-08 jrick return err;
1166 24df9a28 2023-07-08 jrick }
1167 24df9a28 2023-07-08 jrick
1168 24df9a28 2023-07-08 jrick static int
1169 24df9a28 2023-07-08 jrick match_wanted_ref(const char *refname, const char *wanted_ref)
1170 24df9a28 2023-07-08 jrick {
1171 24df9a28 2023-07-08 jrick if (strncmp(refname, "refs/", 5) != 0)
1172 24df9a28 2023-07-08 jrick return 0;
1173 24df9a28 2023-07-08 jrick refname += 5;
1174 24df9a28 2023-07-08 jrick
1175 24df9a28 2023-07-08 jrick /*
1176 24df9a28 2023-07-08 jrick * Prevent fetching of references that won't make any
1177 24df9a28 2023-07-08 jrick * sense outside of the remote repository's context.
1178 24df9a28 2023-07-08 jrick */
1179 24df9a28 2023-07-08 jrick if (strncmp(refname, "got/", 4) == 0)
1180 24df9a28 2023-07-08 jrick return 0;
1181 24df9a28 2023-07-08 jrick if (strncmp(refname, "remotes/", 8) == 0)
1182 24df9a28 2023-07-08 jrick return 0;
1183 24df9a28 2023-07-08 jrick
1184 24df9a28 2023-07-08 jrick if (strncmp(wanted_ref, "refs/", 5) == 0)
1185 24df9a28 2023-07-08 jrick wanted_ref += 5;
1186 24df9a28 2023-07-08 jrick
1187 24df9a28 2023-07-08 jrick /* Allow prefix match. */
1188 24df9a28 2023-07-08 jrick if (got_path_is_child(refname, wanted_ref, strlen(wanted_ref)))
1189 24df9a28 2023-07-08 jrick return 1;
1190 24df9a28 2023-07-08 jrick
1191 24df9a28 2023-07-08 jrick /* Allow exact match. */
1192 24df9a28 2023-07-08 jrick return (strcmp(refname, wanted_ref) == 0);
1193 24df9a28 2023-07-08 jrick }
1194 24df9a28 2023-07-08 jrick
1195 24df9a28 2023-07-08 jrick static int
1196 24df9a28 2023-07-08 jrick is_wanted_ref(struct got_pathlist_head *wanted_refs, const char *refname)
1197 24df9a28 2023-07-08 jrick {
1198 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1199 24df9a28 2023-07-08 jrick
1200 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, wanted_refs, entry) {
1201 24df9a28 2023-07-08 jrick if (match_wanted_ref(refname, pe->path))
1202 24df9a28 2023-07-08 jrick return 1;
1203 24df9a28 2023-07-08 jrick }
1204 24df9a28 2023-07-08 jrick
1205 24df9a28 2023-07-08 jrick return 0;
1206 24df9a28 2023-07-08 jrick }
1207 24df9a28 2023-07-08 jrick
1208 24df9a28 2023-07-08 jrick static const struct got_error *
1209 24df9a28 2023-07-08 jrick create_wanted_ref(const char *refname, struct got_object_id *id,
1210 24df9a28 2023-07-08 jrick const char *remote_repo_name, int verbosity, struct got_repository *repo)
1211 24df9a28 2023-07-08 jrick {
1212 24df9a28 2023-07-08 jrick const struct got_error *err;
1213 24df9a28 2023-07-08 jrick char *remote_refname;
1214 24df9a28 2023-07-08 jrick
1215 24df9a28 2023-07-08 jrick if (strncmp("refs/", refname, 5) == 0)
1216 24df9a28 2023-07-08 jrick refname += 5;
1217 24df9a28 2023-07-08 jrick
1218 24df9a28 2023-07-08 jrick if (asprintf(&remote_refname, "refs/remotes/%s/%s",
1219 24df9a28 2023-07-08 jrick remote_repo_name, refname) == -1)
1220 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
1221 24df9a28 2023-07-08 jrick
1222 24df9a28 2023-07-08 jrick err = create_ref(remote_refname, id, verbosity, repo);
1223 24df9a28 2023-07-08 jrick free(remote_refname);
1224 24df9a28 2023-07-08 jrick return err;
1225 24df9a28 2023-07-08 jrick }
1226 24df9a28 2023-07-08 jrick
1227 24df9a28 2023-07-08 jrick static const struct got_error *
1228 24df9a28 2023-07-08 jrick create_gotconfig(const char *proto, const char *host, const char *port,
1229 24df9a28 2023-07-08 jrick const char *remote_repo_path, const char *default_branch,
1230 24df9a28 2023-07-08 jrick int fetch_all_branches, struct got_pathlist_head *wanted_branches,
1231 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_refs, int mirror_references,
1232 24df9a28 2023-07-08 jrick struct got_repository *repo)
1233 24df9a28 2023-07-08 jrick {
1234 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1235 24df9a28 2023-07-08 jrick char *gotconfig_path = NULL;
1236 24df9a28 2023-07-08 jrick char *gotconfig = NULL;
1237 24df9a28 2023-07-08 jrick FILE *gotconfig_file = NULL;
1238 24df9a28 2023-07-08 jrick const char *branchname = NULL;
1239 24df9a28 2023-07-08 jrick char *branches = NULL, *refs = NULL;
1240 24df9a28 2023-07-08 jrick ssize_t n;
1241 24df9a28 2023-07-08 jrick
1242 24df9a28 2023-07-08 jrick if (!fetch_all_branches && !TAILQ_EMPTY(wanted_branches)) {
1243 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1244 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, wanted_branches, entry) {
1245 24df9a28 2023-07-08 jrick char *s;
1246 24df9a28 2023-07-08 jrick branchname = pe->path;
1247 24df9a28 2023-07-08 jrick if (strncmp(branchname, "refs/heads/", 11) == 0)
1248 24df9a28 2023-07-08 jrick branchname += 11;
1249 24df9a28 2023-07-08 jrick if (asprintf(&s, "%s\"%s\" ",
1250 24df9a28 2023-07-08 jrick branches ? branches : "", branchname) == -1) {
1251 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1252 24df9a28 2023-07-08 jrick goto done;
1253 24df9a28 2023-07-08 jrick }
1254 24df9a28 2023-07-08 jrick free(branches);
1255 24df9a28 2023-07-08 jrick branches = s;
1256 24df9a28 2023-07-08 jrick }
1257 24df9a28 2023-07-08 jrick } else if (!fetch_all_branches && default_branch) {
1258 24df9a28 2023-07-08 jrick branchname = default_branch;
1259 24df9a28 2023-07-08 jrick if (strncmp(branchname, "refs/heads/", 11) == 0)
1260 24df9a28 2023-07-08 jrick branchname += 11;
1261 24df9a28 2023-07-08 jrick if (asprintf(&branches, "\"%s\" ", branchname) == -1) {
1262 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1263 24df9a28 2023-07-08 jrick goto done;
1264 24df9a28 2023-07-08 jrick }
1265 24df9a28 2023-07-08 jrick }
1266 24df9a28 2023-07-08 jrick if (!TAILQ_EMPTY(wanted_refs)) {
1267 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1268 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, wanted_refs, entry) {
1269 24df9a28 2023-07-08 jrick char *s;
1270 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1271 24df9a28 2023-07-08 jrick if (strncmp(refname, "refs/", 5) == 0)
1272 24df9a28 2023-07-08 jrick branchname += 5;
1273 24df9a28 2023-07-08 jrick if (asprintf(&s, "%s\"%s\" ",
1274 24df9a28 2023-07-08 jrick refs ? refs : "", refname) == -1) {
1275 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1276 24df9a28 2023-07-08 jrick goto done;
1277 24df9a28 2023-07-08 jrick }
1278 24df9a28 2023-07-08 jrick free(refs);
1279 24df9a28 2023-07-08 jrick refs = s;
1280 24df9a28 2023-07-08 jrick }
1281 24df9a28 2023-07-08 jrick }
1282 24df9a28 2023-07-08 jrick
1283 24df9a28 2023-07-08 jrick /* Create got.conf(5). */
1284 24df9a28 2023-07-08 jrick gotconfig_path = got_repo_get_path_gotconfig(repo);
1285 24df9a28 2023-07-08 jrick if (gotconfig_path == NULL) {
1286 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_repo_get_path_gotconfig");
1287 24df9a28 2023-07-08 jrick goto done;
1288 24df9a28 2023-07-08 jrick }
1289 24df9a28 2023-07-08 jrick gotconfig_file = fopen(gotconfig_path, "ae");
1290 24df9a28 2023-07-08 jrick if (gotconfig_file == NULL) {
1291 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fopen", gotconfig_path);
1292 24df9a28 2023-07-08 jrick goto done;
1293 24df9a28 2023-07-08 jrick }
1294 24df9a28 2023-07-08 jrick if (asprintf(&gotconfig,
1295 24df9a28 2023-07-08 jrick "remote \"%s\" {\n"
1296 24df9a28 2023-07-08 jrick "\tserver %s\n"
1297 24df9a28 2023-07-08 jrick "\tprotocol %s\n"
1298 24df9a28 2023-07-08 jrick "%s%s%s"
1299 24df9a28 2023-07-08 jrick "\trepository \"%s\"\n"
1300 24df9a28 2023-07-08 jrick "%s%s%s"
1301 24df9a28 2023-07-08 jrick "%s%s%s"
1302 24df9a28 2023-07-08 jrick "%s"
1303 24df9a28 2023-07-08 jrick "%s"
1304 24df9a28 2023-07-08 jrick "}\n",
1305 24df9a28 2023-07-08 jrick GOT_FETCH_DEFAULT_REMOTE_NAME, host, proto,
1306 24df9a28 2023-07-08 jrick port ? "\tport " : "", port ? port : "", port ? "\n" : "",
1307 24df9a28 2023-07-08 jrick remote_repo_path, branches ? "\tbranch { " : "",
1308 24df9a28 2023-07-08 jrick branches ? branches : "", branches ? "}\n" : "",
1309 24df9a28 2023-07-08 jrick refs ? "\treference { " : "", refs ? refs : "", refs ? "}\n" : "",
1310 24df9a28 2023-07-08 jrick mirror_references ? "\tmirror_references yes\n" : "",
1311 24df9a28 2023-07-08 jrick fetch_all_branches ? "\tfetch_all_branches yes\n" : "") == -1) {
1312 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1313 24df9a28 2023-07-08 jrick goto done;
1314 24df9a28 2023-07-08 jrick }
1315 24df9a28 2023-07-08 jrick n = fwrite(gotconfig, 1, strlen(gotconfig), gotconfig_file);
1316 24df9a28 2023-07-08 jrick if (n != strlen(gotconfig)) {
1317 24df9a28 2023-07-08 jrick err = got_ferror(gotconfig_file, GOT_ERR_IO);
1318 24df9a28 2023-07-08 jrick goto done;
1319 24df9a28 2023-07-08 jrick }
1320 24df9a28 2023-07-08 jrick
1321 24df9a28 2023-07-08 jrick done:
1322 24df9a28 2023-07-08 jrick if (gotconfig_file && fclose(gotconfig_file) == EOF && err == NULL)
1323 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fclose", gotconfig_path);
1324 24df9a28 2023-07-08 jrick free(gotconfig_path);
1325 24df9a28 2023-07-08 jrick free(branches);
1326 24df9a28 2023-07-08 jrick return err;
1327 24df9a28 2023-07-08 jrick }
1328 24df9a28 2023-07-08 jrick
1329 24df9a28 2023-07-08 jrick static const struct got_error *
1330 24df9a28 2023-07-08 jrick create_gitconfig(const char *git_url, const char *default_branch,
1331 24df9a28 2023-07-08 jrick int fetch_all_branches, struct got_pathlist_head *wanted_branches,
1332 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_refs, int mirror_references,
1333 24df9a28 2023-07-08 jrick struct got_repository *repo)
1334 24df9a28 2023-07-08 jrick {
1335 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1336 24df9a28 2023-07-08 jrick char *gitconfig_path = NULL;
1337 24df9a28 2023-07-08 jrick char *gitconfig = NULL;
1338 24df9a28 2023-07-08 jrick FILE *gitconfig_file = NULL;
1339 24df9a28 2023-07-08 jrick char *branches = NULL, *refs = NULL;
1340 24df9a28 2023-07-08 jrick const char *branchname;
1341 24df9a28 2023-07-08 jrick ssize_t n;
1342 24df9a28 2023-07-08 jrick
1343 24df9a28 2023-07-08 jrick /* Create a config file Git can understand. */
1344 24df9a28 2023-07-08 jrick gitconfig_path = got_repo_get_path_gitconfig(repo);
1345 24df9a28 2023-07-08 jrick if (gitconfig_path == NULL) {
1346 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_repo_get_path_gitconfig");
1347 24df9a28 2023-07-08 jrick goto done;
1348 24df9a28 2023-07-08 jrick }
1349 24df9a28 2023-07-08 jrick gitconfig_file = fopen(gitconfig_path, "ae");
1350 24df9a28 2023-07-08 jrick if (gitconfig_file == NULL) {
1351 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fopen", gitconfig_path);
1352 24df9a28 2023-07-08 jrick goto done;
1353 24df9a28 2023-07-08 jrick }
1354 24df9a28 2023-07-08 jrick if (fetch_all_branches) {
1355 24df9a28 2023-07-08 jrick if (mirror_references) {
1356 24df9a28 2023-07-08 jrick if (asprintf(&branches,
1357 24df9a28 2023-07-08 jrick "\tfetch = refs/heads/*:refs/heads/*\n") == -1) {
1358 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1359 24df9a28 2023-07-08 jrick goto done;
1360 24df9a28 2023-07-08 jrick }
1361 24df9a28 2023-07-08 jrick } else if (asprintf(&branches,
1362 24df9a28 2023-07-08 jrick "\tfetch = refs/heads/*:refs/remotes/%s/*\n",
1363 24df9a28 2023-07-08 jrick GOT_FETCH_DEFAULT_REMOTE_NAME) == -1) {
1364 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1365 24df9a28 2023-07-08 jrick goto done;
1366 24df9a28 2023-07-08 jrick }
1367 24df9a28 2023-07-08 jrick } else if (!TAILQ_EMPTY(wanted_branches)) {
1368 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1369 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, wanted_branches, entry) {
1370 24df9a28 2023-07-08 jrick char *s;
1371 24df9a28 2023-07-08 jrick branchname = pe->path;
1372 24df9a28 2023-07-08 jrick if (strncmp(branchname, "refs/heads/", 11) == 0)
1373 24df9a28 2023-07-08 jrick branchname += 11;
1374 24df9a28 2023-07-08 jrick if (mirror_references) {
1375 24df9a28 2023-07-08 jrick if (asprintf(&s,
1376 24df9a28 2023-07-08 jrick "%s\tfetch = refs/heads/%s:refs/heads/%s\n",
1377 24df9a28 2023-07-08 jrick branches ? branches : "",
1378 24df9a28 2023-07-08 jrick branchname, branchname) == -1) {
1379 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1380 24df9a28 2023-07-08 jrick goto done;
1381 24df9a28 2023-07-08 jrick }
1382 24df9a28 2023-07-08 jrick } else if (asprintf(&s,
1383 24df9a28 2023-07-08 jrick "%s\tfetch = refs/heads/%s:refs/remotes/%s/%s\n",
1384 24df9a28 2023-07-08 jrick branches ? branches : "",
1385 24df9a28 2023-07-08 jrick branchname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1386 24df9a28 2023-07-08 jrick branchname) == -1) {
1387 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1388 24df9a28 2023-07-08 jrick goto done;
1389 24df9a28 2023-07-08 jrick }
1390 24df9a28 2023-07-08 jrick free(branches);
1391 24df9a28 2023-07-08 jrick branches = s;
1392 24df9a28 2023-07-08 jrick }
1393 24df9a28 2023-07-08 jrick } else {
1394 24df9a28 2023-07-08 jrick /*
1395 24df9a28 2023-07-08 jrick * If the server specified a default branch, use just that one.
1396 24df9a28 2023-07-08 jrick * Otherwise fall back to fetching all branches on next fetch.
1397 24df9a28 2023-07-08 jrick */
1398 24df9a28 2023-07-08 jrick if (default_branch) {
1399 24df9a28 2023-07-08 jrick branchname = default_branch;
1400 24df9a28 2023-07-08 jrick if (strncmp(branchname, "refs/heads/", 11) == 0)
1401 24df9a28 2023-07-08 jrick branchname += 11;
1402 24df9a28 2023-07-08 jrick } else
1403 24df9a28 2023-07-08 jrick branchname = "*"; /* fall back to all branches */
1404 24df9a28 2023-07-08 jrick if (mirror_references) {
1405 24df9a28 2023-07-08 jrick if (asprintf(&branches,
1406 24df9a28 2023-07-08 jrick "\tfetch = refs/heads/%s:refs/heads/%s\n",
1407 24df9a28 2023-07-08 jrick branchname, branchname) == -1) {
1408 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1409 24df9a28 2023-07-08 jrick goto done;
1410 24df9a28 2023-07-08 jrick }
1411 24df9a28 2023-07-08 jrick } else if (asprintf(&branches,
1412 24df9a28 2023-07-08 jrick "\tfetch = refs/heads/%s:refs/remotes/%s/%s\n",
1413 24df9a28 2023-07-08 jrick branchname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1414 24df9a28 2023-07-08 jrick branchname) == -1) {
1415 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1416 24df9a28 2023-07-08 jrick goto done;
1417 24df9a28 2023-07-08 jrick }
1418 24df9a28 2023-07-08 jrick }
1419 24df9a28 2023-07-08 jrick if (!TAILQ_EMPTY(wanted_refs)) {
1420 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1421 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, wanted_refs, entry) {
1422 24df9a28 2023-07-08 jrick char *s;
1423 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1424 24df9a28 2023-07-08 jrick if (strncmp(refname, "refs/", 5) == 0)
1425 24df9a28 2023-07-08 jrick refname += 5;
1426 24df9a28 2023-07-08 jrick if (mirror_references) {
1427 24df9a28 2023-07-08 jrick if (asprintf(&s,
1428 24df9a28 2023-07-08 jrick "%s\tfetch = refs/%s:refs/%s\n",
1429 24df9a28 2023-07-08 jrick refs ? refs : "", refname, refname) == -1) {
1430 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1431 24df9a28 2023-07-08 jrick goto done;
1432 24df9a28 2023-07-08 jrick }
1433 24df9a28 2023-07-08 jrick } else if (asprintf(&s,
1434 24df9a28 2023-07-08 jrick "%s\tfetch = refs/%s:refs/remotes/%s/%s\n",
1435 24df9a28 2023-07-08 jrick refs ? refs : "",
1436 24df9a28 2023-07-08 jrick refname, GOT_FETCH_DEFAULT_REMOTE_NAME,
1437 24df9a28 2023-07-08 jrick refname) == -1) {
1438 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1439 24df9a28 2023-07-08 jrick goto done;
1440 24df9a28 2023-07-08 jrick }
1441 24df9a28 2023-07-08 jrick free(refs);
1442 24df9a28 2023-07-08 jrick refs = s;
1443 24df9a28 2023-07-08 jrick }
1444 24df9a28 2023-07-08 jrick }
1445 24df9a28 2023-07-08 jrick
1446 24df9a28 2023-07-08 jrick if (asprintf(&gitconfig,
1447 24df9a28 2023-07-08 jrick "[remote \"%s\"]\n"
1448 24df9a28 2023-07-08 jrick "\turl = %s\n"
1449 24df9a28 2023-07-08 jrick "%s"
1450 24df9a28 2023-07-08 jrick "%s"
1451 24df9a28 2023-07-08 jrick "\tfetch = refs/tags/*:refs/tags/*\n",
1452 24df9a28 2023-07-08 jrick GOT_FETCH_DEFAULT_REMOTE_NAME, git_url, branches ? branches : "",
1453 24df9a28 2023-07-08 jrick refs ? refs : "") == -1) {
1454 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
1455 24df9a28 2023-07-08 jrick goto done;
1456 24df9a28 2023-07-08 jrick }
1457 24df9a28 2023-07-08 jrick n = fwrite(gitconfig, 1, strlen(gitconfig), gitconfig_file);
1458 24df9a28 2023-07-08 jrick if (n != strlen(gitconfig)) {
1459 24df9a28 2023-07-08 jrick err = got_ferror(gitconfig_file, GOT_ERR_IO);
1460 24df9a28 2023-07-08 jrick goto done;
1461 24df9a28 2023-07-08 jrick }
1462 24df9a28 2023-07-08 jrick done:
1463 24df9a28 2023-07-08 jrick if (gitconfig_file && fclose(gitconfig_file) == EOF && err == NULL)
1464 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fclose", gitconfig_path);
1465 24df9a28 2023-07-08 jrick free(gitconfig_path);
1466 24df9a28 2023-07-08 jrick free(branches);
1467 24df9a28 2023-07-08 jrick return err;
1468 24df9a28 2023-07-08 jrick }
1469 24df9a28 2023-07-08 jrick
1470 24df9a28 2023-07-08 jrick static const struct got_error *
1471 24df9a28 2023-07-08 jrick create_config_files(const char *proto, const char *host, const char *port,
1472 24df9a28 2023-07-08 jrick const char *remote_repo_path, const char *git_url, int fetch_all_branches,
1473 24df9a28 2023-07-08 jrick int mirror_references, struct got_pathlist_head *symrefs,
1474 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_branches,
1475 24df9a28 2023-07-08 jrick struct got_pathlist_head *wanted_refs, struct got_repository *repo)
1476 24df9a28 2023-07-08 jrick {
1477 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1478 24df9a28 2023-07-08 jrick const char *default_branch = NULL;
1479 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1480 24df9a28 2023-07-08 jrick
1481 24df9a28 2023-07-08 jrick /*
1482 24df9a28 2023-07-08 jrick * If we asked for a set of wanted branches then use the first
1483 24df9a28 2023-07-08 jrick * one of those.
1484 24df9a28 2023-07-08 jrick */
1485 24df9a28 2023-07-08 jrick if (!TAILQ_EMPTY(wanted_branches)) {
1486 24df9a28 2023-07-08 jrick pe = TAILQ_FIRST(wanted_branches);
1487 24df9a28 2023-07-08 jrick default_branch = pe->path;
1488 24df9a28 2023-07-08 jrick } else {
1489 24df9a28 2023-07-08 jrick /* First HEAD ref listed by server is the default branch. */
1490 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, symrefs, entry) {
1491 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1492 24df9a28 2023-07-08 jrick const char *target = pe->data;
1493 24df9a28 2023-07-08 jrick
1494 24df9a28 2023-07-08 jrick if (strcmp(refname, GOT_REF_HEAD) != 0)
1495 24df9a28 2023-07-08 jrick continue;
1496 24df9a28 2023-07-08 jrick
1497 24df9a28 2023-07-08 jrick default_branch = target;
1498 24df9a28 2023-07-08 jrick break;
1499 24df9a28 2023-07-08 jrick }
1500 24df9a28 2023-07-08 jrick }
1501 24df9a28 2023-07-08 jrick
1502 24df9a28 2023-07-08 jrick /* Create got.conf(5). */
1503 24df9a28 2023-07-08 jrick err = create_gotconfig(proto, host, port, remote_repo_path,
1504 24df9a28 2023-07-08 jrick default_branch, fetch_all_branches, wanted_branches,
1505 24df9a28 2023-07-08 jrick wanted_refs, mirror_references, repo);
1506 24df9a28 2023-07-08 jrick if (err)
1507 24df9a28 2023-07-08 jrick return err;
1508 24df9a28 2023-07-08 jrick
1509 24df9a28 2023-07-08 jrick /* Create a config file Git can understand. */
1510 24df9a28 2023-07-08 jrick return create_gitconfig(git_url, default_branch, fetch_all_branches,
1511 24df9a28 2023-07-08 jrick wanted_branches, wanted_refs, mirror_references, repo);
1512 24df9a28 2023-07-08 jrick }
1513 24df9a28 2023-07-08 jrick
1514 24df9a28 2023-07-08 jrick static const struct got_error *
1515 24df9a28 2023-07-08 jrick cmd_clone(int argc, char *argv[])
1516 24df9a28 2023-07-08 jrick {
1517 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
1518 24df9a28 2023-07-08 jrick const char *uri, *dirname;
1519 24df9a28 2023-07-08 jrick char *proto, *host, *port, *repo_name, *server_path;
1520 24df9a28 2023-07-08 jrick char *default_destdir = NULL, *id_str = NULL;
1521 24df9a28 2023-07-08 jrick const char *repo_path;
1522 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
1523 24df9a28 2023-07-08 jrick struct got_pathlist_head refs, symrefs, wanted_branches, wanted_refs;
1524 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
1525 24df9a28 2023-07-08 jrick struct got_object_id *pack_hash = NULL;
1526 24df9a28 2023-07-08 jrick int ch, fetchfd = -1, fetchstatus;
1527 24df9a28 2023-07-08 jrick pid_t fetchpid = -1;
1528 24df9a28 2023-07-08 jrick struct got_fetch_progress_arg fpa;
1529 24df9a28 2023-07-08 jrick char *git_url = NULL;
1530 24df9a28 2023-07-08 jrick int verbosity = 0, fetch_all_branches = 0, mirror_references = 0;
1531 24df9a28 2023-07-08 jrick int bflag = 0, list_refs_only = 0;
1532 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
1533 24df9a28 2023-07-08 jrick
1534 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
1535 24df9a28 2023-07-08 jrick TAILQ_INIT(&symrefs);
1536 24df9a28 2023-07-08 jrick TAILQ_INIT(&wanted_branches);
1537 24df9a28 2023-07-08 jrick TAILQ_INIT(&wanted_refs);
1538 24df9a28 2023-07-08 jrick
1539 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "ab:lmqR:v")) != -1) {
1540 24df9a28 2023-07-08 jrick switch (ch) {
1541 24df9a28 2023-07-08 jrick case 'a':
1542 24df9a28 2023-07-08 jrick fetch_all_branches = 1;
1543 24df9a28 2023-07-08 jrick break;
1544 24df9a28 2023-07-08 jrick case 'b':
1545 24df9a28 2023-07-08 jrick error = got_pathlist_append(&wanted_branches,
1546 24df9a28 2023-07-08 jrick optarg, NULL);
1547 24df9a28 2023-07-08 jrick if (error)
1548 24df9a28 2023-07-08 jrick return error;
1549 24df9a28 2023-07-08 jrick bflag = 1;
1550 24df9a28 2023-07-08 jrick break;
1551 24df9a28 2023-07-08 jrick case 'l':
1552 24df9a28 2023-07-08 jrick list_refs_only = 1;
1553 24df9a28 2023-07-08 jrick break;
1554 24df9a28 2023-07-08 jrick case 'm':
1555 24df9a28 2023-07-08 jrick mirror_references = 1;
1556 24df9a28 2023-07-08 jrick break;
1557 24df9a28 2023-07-08 jrick case 'q':
1558 24df9a28 2023-07-08 jrick verbosity = -1;
1559 24df9a28 2023-07-08 jrick break;
1560 24df9a28 2023-07-08 jrick case 'R':
1561 24df9a28 2023-07-08 jrick error = got_pathlist_append(&wanted_refs,
1562 24df9a28 2023-07-08 jrick optarg, NULL);
1563 24df9a28 2023-07-08 jrick if (error)
1564 24df9a28 2023-07-08 jrick return error;
1565 24df9a28 2023-07-08 jrick break;
1566 24df9a28 2023-07-08 jrick case 'v':
1567 24df9a28 2023-07-08 jrick if (verbosity < 0)
1568 24df9a28 2023-07-08 jrick verbosity = 0;
1569 24df9a28 2023-07-08 jrick else if (verbosity < 3)
1570 24df9a28 2023-07-08 jrick verbosity++;
1571 24df9a28 2023-07-08 jrick break;
1572 24df9a28 2023-07-08 jrick default:
1573 24df9a28 2023-07-08 jrick usage_clone();
1574 24df9a28 2023-07-08 jrick break;
1575 24df9a28 2023-07-08 jrick }
1576 24df9a28 2023-07-08 jrick }
1577 24df9a28 2023-07-08 jrick argc -= optind;
1578 24df9a28 2023-07-08 jrick argv += optind;
1579 24df9a28 2023-07-08 jrick
1580 24df9a28 2023-07-08 jrick if (fetch_all_branches && !TAILQ_EMPTY(&wanted_branches))
1581 24df9a28 2023-07-08 jrick option_conflict('a', 'b');
1582 24df9a28 2023-07-08 jrick if (list_refs_only) {
1583 24df9a28 2023-07-08 jrick if (!TAILQ_EMPTY(&wanted_branches))
1584 24df9a28 2023-07-08 jrick option_conflict('l', 'b');
1585 24df9a28 2023-07-08 jrick if (fetch_all_branches)
1586 24df9a28 2023-07-08 jrick option_conflict('l', 'a');
1587 24df9a28 2023-07-08 jrick if (mirror_references)
1588 24df9a28 2023-07-08 jrick option_conflict('l', 'm');
1589 24df9a28 2023-07-08 jrick if (!TAILQ_EMPTY(&wanted_refs))
1590 24df9a28 2023-07-08 jrick option_conflict('l', 'R');
1591 24df9a28 2023-07-08 jrick }
1592 24df9a28 2023-07-08 jrick
1593 24df9a28 2023-07-08 jrick uri = argv[0];
1594 24df9a28 2023-07-08 jrick
1595 24df9a28 2023-07-08 jrick if (argc == 1)
1596 24df9a28 2023-07-08 jrick dirname = NULL;
1597 24df9a28 2023-07-08 jrick else if (argc == 2)
1598 24df9a28 2023-07-08 jrick dirname = argv[1];
1599 24df9a28 2023-07-08 jrick else
1600 24df9a28 2023-07-08 jrick usage_clone();
1601 24df9a28 2023-07-08 jrick
1602 24df9a28 2023-07-08 jrick error = got_dial_parse_uri(&proto, &host, &port, &server_path,
1603 24df9a28 2023-07-08 jrick &repo_name, uri);
1604 24df9a28 2023-07-08 jrick if (error)
1605 24df9a28 2023-07-08 jrick goto done;
1606 24df9a28 2023-07-08 jrick
1607 24df9a28 2023-07-08 jrick if (asprintf(&git_url, "%s://%s%s%s%s%s", proto,
1608 24df9a28 2023-07-08 jrick host, port ? ":" : "", port ? port : "",
1609 24df9a28 2023-07-08 jrick server_path[0] != '/' ? "/" : "", server_path) == -1) {
1610 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
1611 24df9a28 2023-07-08 jrick goto done;
1612 24df9a28 2023-07-08 jrick }
1613 24df9a28 2023-07-08 jrick
1614 24df9a28 2023-07-08 jrick if (strcmp(proto, "git") == 0) {
1615 24df9a28 2023-07-08 jrick #ifndef PROFILE
1616 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1617 24df9a28 2023-07-08 jrick "sendfd dns inet unveil", NULL) == -1)
1618 24df9a28 2023-07-08 jrick err(1, "pledge");
1619 24df9a28 2023-07-08 jrick #endif
1620 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "git+ssh") == 0 ||
1621 24df9a28 2023-07-08 jrick strcmp(proto, "ssh") == 0) {
1622 24df9a28 2023-07-08 jrick #ifndef PROFILE
1623 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1624 24df9a28 2023-07-08 jrick "sendfd unveil", NULL) == -1)
1625 24df9a28 2023-07-08 jrick err(1, "pledge");
1626 24df9a28 2023-07-08 jrick #endif
1627 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "http") == 0 ||
1628 24df9a28 2023-07-08 jrick strcmp(proto, "git+http") == 0) {
1629 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_NOT_IMPL);
1630 24df9a28 2023-07-08 jrick goto done;
1631 24df9a28 2023-07-08 jrick } else {
1632 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_BAD_PROTO);
1633 24df9a28 2023-07-08 jrick goto done;
1634 24df9a28 2023-07-08 jrick }
1635 24df9a28 2023-07-08 jrick if (dirname == NULL) {
1636 24df9a28 2023-07-08 jrick if (asprintf(&default_destdir, "%s.git", repo_name) == -1) {
1637 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
1638 24df9a28 2023-07-08 jrick goto done;
1639 24df9a28 2023-07-08 jrick }
1640 24df9a28 2023-07-08 jrick repo_path = default_destdir;
1641 24df9a28 2023-07-08 jrick } else
1642 24df9a28 2023-07-08 jrick repo_path = dirname;
1643 24df9a28 2023-07-08 jrick
1644 24df9a28 2023-07-08 jrick if (!list_refs_only) {
1645 24df9a28 2023-07-08 jrick error = got_path_mkdir(repo_path);
1646 24df9a28 2023-07-08 jrick if (error &&
1647 24df9a28 2023-07-08 jrick (!(error->code == GOT_ERR_ERRNO && errno == EISDIR) &&
1648 24df9a28 2023-07-08 jrick !(error->code == GOT_ERR_ERRNO && errno == EEXIST)))
1649 24df9a28 2023-07-08 jrick goto done;
1650 24df9a28 2023-07-08 jrick if (!got_path_dir_is_empty(repo_path)) {
1651 24df9a28 2023-07-08 jrick error = got_error_path(repo_path,
1652 24df9a28 2023-07-08 jrick GOT_ERR_DIR_NOT_EMPTY);
1653 24df9a28 2023-07-08 jrick goto done;
1654 24df9a28 2023-07-08 jrick }
1655 24df9a28 2023-07-08 jrick }
1656 24df9a28 2023-07-08 jrick
1657 24df9a28 2023-07-08 jrick error = got_dial_apply_unveil(proto);
1658 24df9a28 2023-07-08 jrick if (error)
1659 24df9a28 2023-07-08 jrick goto done;
1660 24df9a28 2023-07-08 jrick
1661 24df9a28 2023-07-08 jrick error = apply_unveil(repo_path, 0, NULL);
1662 24df9a28 2023-07-08 jrick if (error)
1663 24df9a28 2023-07-08 jrick goto done;
1664 24df9a28 2023-07-08 jrick
1665 24df9a28 2023-07-08 jrick if (verbosity >= 0)
1666 24df9a28 2023-07-08 jrick printf("Connecting to %s\n", git_url);
1667 24df9a28 2023-07-08 jrick
1668 24df9a28 2023-07-08 jrick error = got_fetch_connect(&fetchpid, &fetchfd, proto, host, port,
1669 24df9a28 2023-07-08 jrick server_path, verbosity);
1670 24df9a28 2023-07-08 jrick if (error)
1671 24df9a28 2023-07-08 jrick goto done;
1672 24df9a28 2023-07-08 jrick
1673 24df9a28 2023-07-08 jrick if (!list_refs_only) {
1674 24df9a28 2023-07-08 jrick error = got_repo_init(repo_path, NULL);
1675 24df9a28 2023-07-08 jrick if (error)
1676 24df9a28 2023-07-08 jrick goto done;
1677 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
1678 24df9a28 2023-07-08 jrick if (error != NULL)
1679 24df9a28 2023-07-08 jrick goto done;
1680 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
1681 24df9a28 2023-07-08 jrick if (error)
1682 24df9a28 2023-07-08 jrick goto done;
1683 24df9a28 2023-07-08 jrick }
1684 24df9a28 2023-07-08 jrick
1685 24df9a28 2023-07-08 jrick fpa.last_scaled_size[0] = '\0';
1686 24df9a28 2023-07-08 jrick fpa.last_p_indexed = -1;
1687 24df9a28 2023-07-08 jrick fpa.last_p_resolved = -1;
1688 24df9a28 2023-07-08 jrick fpa.verbosity = verbosity;
1689 24df9a28 2023-07-08 jrick fpa.create_configs = 1;
1690 24df9a28 2023-07-08 jrick fpa.configs_created = 0;
1691 24df9a28 2023-07-08 jrick fpa.repo = repo;
1692 24df9a28 2023-07-08 jrick fpa.config_info.symrefs = &symrefs;
1693 24df9a28 2023-07-08 jrick fpa.config_info.wanted_branches = &wanted_branches;
1694 24df9a28 2023-07-08 jrick fpa.config_info.wanted_refs = &wanted_refs;
1695 24df9a28 2023-07-08 jrick fpa.config_info.proto = proto;
1696 24df9a28 2023-07-08 jrick fpa.config_info.host = host;
1697 24df9a28 2023-07-08 jrick fpa.config_info.port = port;
1698 24df9a28 2023-07-08 jrick fpa.config_info.remote_repo_path = server_path;
1699 24df9a28 2023-07-08 jrick fpa.config_info.git_url = git_url;
1700 24df9a28 2023-07-08 jrick fpa.config_info.fetch_all_branches = fetch_all_branches;
1701 24df9a28 2023-07-08 jrick fpa.config_info.mirror_references = mirror_references;
1702 24df9a28 2023-07-08 jrick error = got_fetch_pack(&pack_hash, &refs, &symrefs,
1703 24df9a28 2023-07-08 jrick GOT_FETCH_DEFAULT_REMOTE_NAME, mirror_references,
1704 24df9a28 2023-07-08 jrick fetch_all_branches, &wanted_branches, &wanted_refs,
1705 24df9a28 2023-07-08 jrick list_refs_only, verbosity, fetchfd, repo, NULL, NULL, bflag,
1706 24df9a28 2023-07-08 jrick fetch_progress, &fpa);
1707 24df9a28 2023-07-08 jrick if (error)
1708 24df9a28 2023-07-08 jrick goto done;
1709 24df9a28 2023-07-08 jrick
1710 24df9a28 2023-07-08 jrick if (list_refs_only) {
1711 24df9a28 2023-07-08 jrick error = list_remote_refs(&symrefs, &refs);
1712 24df9a28 2023-07-08 jrick goto done;
1713 24df9a28 2023-07-08 jrick }
1714 24df9a28 2023-07-08 jrick
1715 24df9a28 2023-07-08 jrick if (pack_hash == NULL) {
1716 24df9a28 2023-07-08 jrick error = got_error_fmt(GOT_ERR_FETCH_FAILED, "%s",
1717 24df9a28 2023-07-08 jrick "server sent an empty pack file");
1718 24df9a28 2023-07-08 jrick goto done;
1719 24df9a28 2023-07-08 jrick }
1720 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str, pack_hash);
1721 24df9a28 2023-07-08 jrick if (error)
1722 24df9a28 2023-07-08 jrick goto done;
1723 24df9a28 2023-07-08 jrick if (verbosity >= 0)
1724 24df9a28 2023-07-08 jrick printf("\nFetched %s.pack\n", id_str);
1725 24df9a28 2023-07-08 jrick free(id_str);
1726 24df9a28 2023-07-08 jrick
1727 24df9a28 2023-07-08 jrick /* Set up references provided with the pack file. */
1728 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &refs, entry) {
1729 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1730 24df9a28 2023-07-08 jrick struct got_object_id *id = pe->data;
1731 24df9a28 2023-07-08 jrick char *remote_refname;
1732 24df9a28 2023-07-08 jrick
1733 24df9a28 2023-07-08 jrick if (is_wanted_ref(&wanted_refs, refname) &&
1734 24df9a28 2023-07-08 jrick !mirror_references) {
1735 24df9a28 2023-07-08 jrick error = create_wanted_ref(refname, id,
1736 24df9a28 2023-07-08 jrick GOT_FETCH_DEFAULT_REMOTE_NAME,
1737 24df9a28 2023-07-08 jrick verbosity - 1, repo);
1738 24df9a28 2023-07-08 jrick if (error)
1739 24df9a28 2023-07-08 jrick goto done;
1740 24df9a28 2023-07-08 jrick continue;
1741 24df9a28 2023-07-08 jrick }
1742 24df9a28 2023-07-08 jrick
1743 24df9a28 2023-07-08 jrick error = create_ref(refname, id, verbosity - 1, repo);
1744 24df9a28 2023-07-08 jrick if (error)
1745 24df9a28 2023-07-08 jrick goto done;
1746 24df9a28 2023-07-08 jrick
1747 24df9a28 2023-07-08 jrick if (mirror_references)
1748 24df9a28 2023-07-08 jrick continue;
1749 24df9a28 2023-07-08 jrick
1750 24df9a28 2023-07-08 jrick if (strncmp("refs/heads/", refname, 11) != 0)
1751 24df9a28 2023-07-08 jrick continue;
1752 24df9a28 2023-07-08 jrick
1753 24df9a28 2023-07-08 jrick if (asprintf(&remote_refname,
1754 24df9a28 2023-07-08 jrick "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1755 24df9a28 2023-07-08 jrick refname + 11) == -1) {
1756 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
1757 24df9a28 2023-07-08 jrick goto done;
1758 24df9a28 2023-07-08 jrick }
1759 24df9a28 2023-07-08 jrick error = create_ref(remote_refname, id, verbosity - 1, repo);
1760 24df9a28 2023-07-08 jrick free(remote_refname);
1761 24df9a28 2023-07-08 jrick if (error)
1762 24df9a28 2023-07-08 jrick goto done;
1763 24df9a28 2023-07-08 jrick }
1764 24df9a28 2023-07-08 jrick
1765 24df9a28 2023-07-08 jrick /* Set the HEAD reference if the server provided one. */
1766 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &symrefs, entry) {
1767 24df9a28 2023-07-08 jrick struct got_reference *target_ref;
1768 24df9a28 2023-07-08 jrick const char *refname = pe->path;
1769 24df9a28 2023-07-08 jrick const char *target = pe->data;
1770 24df9a28 2023-07-08 jrick char *remote_refname = NULL, *remote_target = NULL;
1771 24df9a28 2023-07-08 jrick
1772 24df9a28 2023-07-08 jrick if (strcmp(refname, GOT_REF_HEAD) != 0)
1773 24df9a28 2023-07-08 jrick continue;
1774 24df9a28 2023-07-08 jrick
1775 24df9a28 2023-07-08 jrick error = got_ref_open(&target_ref, repo, target, 0);
1776 24df9a28 2023-07-08 jrick if (error) {
1777 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_REF) {
1778 24df9a28 2023-07-08 jrick error = NULL;
1779 24df9a28 2023-07-08 jrick continue;
1780 24df9a28 2023-07-08 jrick }
1781 24df9a28 2023-07-08 jrick goto done;
1782 24df9a28 2023-07-08 jrick }
1783 24df9a28 2023-07-08 jrick
1784 24df9a28 2023-07-08 jrick error = create_symref(refname, target_ref, verbosity, repo);
1785 24df9a28 2023-07-08 jrick got_ref_close(target_ref);
1786 24df9a28 2023-07-08 jrick if (error)
1787 24df9a28 2023-07-08 jrick goto done;
1788 24df9a28 2023-07-08 jrick
1789 24df9a28 2023-07-08 jrick if (mirror_references)
1790 24df9a28 2023-07-08 jrick continue;
1791 24df9a28 2023-07-08 jrick
1792 24df9a28 2023-07-08 jrick if (strncmp("refs/heads/", target, 11) != 0)
1793 24df9a28 2023-07-08 jrick continue;
1794 24df9a28 2023-07-08 jrick
1795 24df9a28 2023-07-08 jrick if (asprintf(&remote_refname,
1796 24df9a28 2023-07-08 jrick "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1797 24df9a28 2023-07-08 jrick refname) == -1) {
1798 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
1799 24df9a28 2023-07-08 jrick goto done;
1800 24df9a28 2023-07-08 jrick }
1801 24df9a28 2023-07-08 jrick if (asprintf(&remote_target,
1802 24df9a28 2023-07-08 jrick "refs/remotes/%s/%s", GOT_FETCH_DEFAULT_REMOTE_NAME,
1803 24df9a28 2023-07-08 jrick target + 11) == -1) {
1804 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
1805 24df9a28 2023-07-08 jrick free(remote_refname);
1806 24df9a28 2023-07-08 jrick goto done;
1807 24df9a28 2023-07-08 jrick }
1808 24df9a28 2023-07-08 jrick error = got_ref_open(&target_ref, repo, remote_target, 0);
1809 24df9a28 2023-07-08 jrick if (error) {
1810 24df9a28 2023-07-08 jrick free(remote_refname);
1811 24df9a28 2023-07-08 jrick free(remote_target);
1812 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_REF) {
1813 24df9a28 2023-07-08 jrick error = NULL;
1814 24df9a28 2023-07-08 jrick continue;
1815 24df9a28 2023-07-08 jrick }
1816 24df9a28 2023-07-08 jrick goto done;
1817 24df9a28 2023-07-08 jrick }
1818 24df9a28 2023-07-08 jrick error = create_symref(remote_refname, target_ref,
1819 24df9a28 2023-07-08 jrick verbosity - 1, repo);
1820 24df9a28 2023-07-08 jrick free(remote_refname);
1821 24df9a28 2023-07-08 jrick free(remote_target);
1822 24df9a28 2023-07-08 jrick got_ref_close(target_ref);
1823 24df9a28 2023-07-08 jrick if (error)
1824 24df9a28 2023-07-08 jrick goto done;
1825 24df9a28 2023-07-08 jrick }
1826 24df9a28 2023-07-08 jrick if (pe == NULL) {
1827 24df9a28 2023-07-08 jrick /*
1828 24df9a28 2023-07-08 jrick * We failed to set the HEAD reference. If we asked for
1829 24df9a28 2023-07-08 jrick * a set of wanted branches use the first of one of those
1830 24df9a28 2023-07-08 jrick * which could be fetched instead.
1831 24df9a28 2023-07-08 jrick */
1832 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &wanted_branches, entry) {
1833 24df9a28 2023-07-08 jrick const char *target = pe->path;
1834 24df9a28 2023-07-08 jrick struct got_reference *target_ref;
1835 24df9a28 2023-07-08 jrick
1836 24df9a28 2023-07-08 jrick error = got_ref_open(&target_ref, repo, target, 0);
1837 24df9a28 2023-07-08 jrick if (error) {
1838 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_REF) {
1839 24df9a28 2023-07-08 jrick error = NULL;
1840 24df9a28 2023-07-08 jrick continue;
1841 24df9a28 2023-07-08 jrick }
1842 24df9a28 2023-07-08 jrick goto done;
1843 24df9a28 2023-07-08 jrick }
1844 24df9a28 2023-07-08 jrick
1845 24df9a28 2023-07-08 jrick error = create_symref(GOT_REF_HEAD, target_ref,
1846 24df9a28 2023-07-08 jrick verbosity, repo);
1847 24df9a28 2023-07-08 jrick got_ref_close(target_ref);
1848 24df9a28 2023-07-08 jrick if (error)
1849 24df9a28 2023-07-08 jrick goto done;
1850 24df9a28 2023-07-08 jrick break;
1851 24df9a28 2023-07-08 jrick }
1852 24df9a28 2023-07-08 jrick
1853 24df9a28 2023-07-08 jrick if (!fpa.configs_created && pe != NULL) {
1854 24df9a28 2023-07-08 jrick error = create_config_files(fpa.config_info.proto,
1855 24df9a28 2023-07-08 jrick fpa.config_info.host, fpa.config_info.port,
1856 24df9a28 2023-07-08 jrick fpa.config_info.remote_repo_path,
1857 24df9a28 2023-07-08 jrick fpa.config_info.git_url,
1858 24df9a28 2023-07-08 jrick fpa.config_info.fetch_all_branches,
1859 24df9a28 2023-07-08 jrick fpa.config_info.mirror_references,
1860 24df9a28 2023-07-08 jrick fpa.config_info.symrefs,
1861 24df9a28 2023-07-08 jrick fpa.config_info.wanted_branches,
1862 24df9a28 2023-07-08 jrick fpa.config_info.wanted_refs, fpa.repo);
1863 24df9a28 2023-07-08 jrick if (error)
1864 24df9a28 2023-07-08 jrick goto done;
1865 24df9a28 2023-07-08 jrick }
1866 24df9a28 2023-07-08 jrick }
1867 24df9a28 2023-07-08 jrick
1868 24df9a28 2023-07-08 jrick if (verbosity >= 0)
1869 24df9a28 2023-07-08 jrick printf("Created %s repository '%s'\n",
1870 24df9a28 2023-07-08 jrick mirror_references ? "mirrored" : "cloned", repo_path);
1871 24df9a28 2023-07-08 jrick done:
1872 24df9a28 2023-07-08 jrick if (pack_fds) {
1873 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
1874 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
1875 24df9a28 2023-07-08 jrick if (error == NULL)
1876 24df9a28 2023-07-08 jrick error = pack_err;
1877 24df9a28 2023-07-08 jrick }
1878 24df9a28 2023-07-08 jrick if (fetchpid > 0) {
1879 24df9a28 2023-07-08 jrick if (kill(fetchpid, SIGTERM) == -1)
1880 24df9a28 2023-07-08 jrick error = got_error_from_errno("kill");
1881 24df9a28 2023-07-08 jrick if (waitpid(fetchpid, &fetchstatus, 0) == -1 && error == NULL)
1882 24df9a28 2023-07-08 jrick error = got_error_from_errno("waitpid");
1883 24df9a28 2023-07-08 jrick }
1884 24df9a28 2023-07-08 jrick if (fetchfd != -1 && close(fetchfd) == -1 && error == NULL)
1885 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
1886 24df9a28 2023-07-08 jrick if (repo) {
1887 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
1888 24df9a28 2023-07-08 jrick if (error == NULL)
1889 24df9a28 2023-07-08 jrick error = close_err;
1890 24df9a28 2023-07-08 jrick }
1891 24df9a28 2023-07-08 jrick got_pathlist_free(&refs, GOT_PATHLIST_FREE_ALL);
1892 24df9a28 2023-07-08 jrick got_pathlist_free(&symrefs, GOT_PATHLIST_FREE_ALL);
1893 24df9a28 2023-07-08 jrick got_pathlist_free(&wanted_branches, GOT_PATHLIST_FREE_NONE);
1894 24df9a28 2023-07-08 jrick got_pathlist_free(&wanted_refs, GOT_PATHLIST_FREE_NONE);
1895 24df9a28 2023-07-08 jrick free(pack_hash);
1896 24df9a28 2023-07-08 jrick free(proto);
1897 24df9a28 2023-07-08 jrick free(host);
1898 24df9a28 2023-07-08 jrick free(port);
1899 24df9a28 2023-07-08 jrick free(server_path);
1900 24df9a28 2023-07-08 jrick free(repo_name);
1901 24df9a28 2023-07-08 jrick free(default_destdir);
1902 24df9a28 2023-07-08 jrick free(git_url);
1903 24df9a28 2023-07-08 jrick return error;
1904 24df9a28 2023-07-08 jrick }
1905 24df9a28 2023-07-08 jrick
1906 24df9a28 2023-07-08 jrick static const struct got_error *
1907 24df9a28 2023-07-08 jrick update_ref(struct got_reference *ref, struct got_object_id *new_id,
1908 24df9a28 2023-07-08 jrick int replace_tags, int verbosity, struct got_repository *repo)
1909 24df9a28 2023-07-08 jrick {
1910 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
1911 24df9a28 2023-07-08 jrick char *new_id_str = NULL;
1912 24df9a28 2023-07-08 jrick struct got_object_id *old_id = NULL;
1913 24df9a28 2023-07-08 jrick
1914 24df9a28 2023-07-08 jrick err = got_object_id_str(&new_id_str, new_id);
1915 24df9a28 2023-07-08 jrick if (err)
1916 24df9a28 2023-07-08 jrick goto done;
1917 24df9a28 2023-07-08 jrick
1918 24df9a28 2023-07-08 jrick if (!replace_tags &&
1919 24df9a28 2023-07-08 jrick strncmp(got_ref_get_name(ref), "refs/tags/", 10) == 0) {
1920 24df9a28 2023-07-08 jrick err = got_ref_resolve(&old_id, repo, ref);
1921 24df9a28 2023-07-08 jrick if (err)
1922 24df9a28 2023-07-08 jrick goto done;
1923 24df9a28 2023-07-08 jrick if (got_object_id_cmp(old_id, new_id) == 0)
1924 24df9a28 2023-07-08 jrick goto done;
1925 24df9a28 2023-07-08 jrick if (verbosity >= 0) {
1926 24df9a28 2023-07-08 jrick printf("Rejecting update of existing tag %s: %s\n",
1927 24df9a28 2023-07-08 jrick got_ref_get_name(ref), new_id_str);
1928 24df9a28 2023-07-08 jrick }
1929 24df9a28 2023-07-08 jrick goto done;
1930 24df9a28 2023-07-08 jrick }
1931 24df9a28 2023-07-08 jrick
1932 24df9a28 2023-07-08 jrick if (got_ref_is_symbolic(ref)) {
1933 24df9a28 2023-07-08 jrick if (verbosity >= 0) {
1934 24df9a28 2023-07-08 jrick printf("Replacing reference %s: %s\n",
1935 24df9a28 2023-07-08 jrick got_ref_get_name(ref),
1936 24df9a28 2023-07-08 jrick got_ref_get_symref_target(ref));
1937 24df9a28 2023-07-08 jrick }
1938 24df9a28 2023-07-08 jrick err = got_ref_change_symref_to_ref(ref, new_id);
1939 24df9a28 2023-07-08 jrick if (err)
1940 24df9a28 2023-07-08 jrick goto done;
1941 24df9a28 2023-07-08 jrick err = got_ref_write(ref, repo);
1942 24df9a28 2023-07-08 jrick if (err)
1943 24df9a28 2023-07-08 jrick goto done;
1944 24df9a28 2023-07-08 jrick } else {
1945 24df9a28 2023-07-08 jrick err = got_ref_resolve(&old_id, repo, ref);
1946 24df9a28 2023-07-08 jrick if (err)
1947 24df9a28 2023-07-08 jrick goto done;
1948 24df9a28 2023-07-08 jrick if (got_object_id_cmp(old_id, new_id) == 0)
1949 24df9a28 2023-07-08 jrick goto done;
1950 24df9a28 2023-07-08 jrick
1951 24df9a28 2023-07-08 jrick err = got_ref_change_ref(ref, new_id);
1952 24df9a28 2023-07-08 jrick if (err)
1953 24df9a28 2023-07-08 jrick goto done;
1954 24df9a28 2023-07-08 jrick err = got_ref_write(ref, repo);
1955 24df9a28 2023-07-08 jrick if (err)
1956 24df9a28 2023-07-08 jrick goto done;
1957 24df9a28 2023-07-08 jrick }
1958 24df9a28 2023-07-08 jrick
1959 24df9a28 2023-07-08 jrick if (verbosity >= 0)
1960 24df9a28 2023-07-08 jrick printf("Updated %s: %s\n", got_ref_get_name(ref),
1961 24df9a28 2023-07-08 jrick new_id_str);
1962 24df9a28 2023-07-08 jrick done:
1963 24df9a28 2023-07-08 jrick free(old_id);
1964 24df9a28 2023-07-08 jrick free(new_id_str);
1965 24df9a28 2023-07-08 jrick return err;
1966 24df9a28 2023-07-08 jrick }
1967 24df9a28 2023-07-08 jrick
1968 24df9a28 2023-07-08 jrick static const struct got_error *
1969 24df9a28 2023-07-08 jrick update_wanted_ref(const char *refname, struct got_object_id *id,
1970 24df9a28 2023-07-08 jrick const char *remote_repo_name, int verbosity, struct got_repository *repo)
1971 24df9a28 2023-07-08 jrick {
1972 24df9a28 2023-07-08 jrick const struct got_error *err, *unlock_err;
1973 24df9a28 2023-07-08 jrick char *remote_refname;
1974 24df9a28 2023-07-08 jrick struct got_reference *ref;
1975 24df9a28 2023-07-08 jrick
1976 24df9a28 2023-07-08 jrick if (strncmp("refs/", refname, 5) == 0)
1977 24df9a28 2023-07-08 jrick refname += 5;
1978 24df9a28 2023-07-08 jrick
1979 24df9a28 2023-07-08 jrick if (asprintf(&remote_refname, "refs/remotes/%s/%s",
1980 24df9a28 2023-07-08 jrick remote_repo_name, refname) == -1)
1981 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
1982 24df9a28 2023-07-08 jrick
1983 24df9a28 2023-07-08 jrick err = got_ref_open(&ref, repo, remote_refname, 1);
1984 24df9a28 2023-07-08 jrick if (err) {
1985 24df9a28 2023-07-08 jrick if (err->code != GOT_ERR_NOT_REF)
1986 24df9a28 2023-07-08 jrick goto done;
1987 24df9a28 2023-07-08 jrick err = create_ref(remote_refname, id, verbosity, repo);
1988 24df9a28 2023-07-08 jrick } else {
1989 24df9a28 2023-07-08 jrick err = update_ref(ref, id, 0, verbosity, repo);
1990 24df9a28 2023-07-08 jrick unlock_err = got_ref_unlock(ref);
1991 24df9a28 2023-07-08 jrick if (unlock_err && err == NULL)
1992 24df9a28 2023-07-08 jrick err = unlock_err;
1993 24df9a28 2023-07-08 jrick got_ref_close(ref);
1994 24df9a28 2023-07-08 jrick }
1995 24df9a28 2023-07-08 jrick done:
1996 24df9a28 2023-07-08 jrick free(remote_refname);
1997 24df9a28 2023-07-08 jrick return err;
1998 24df9a28 2023-07-08 jrick }
1999 24df9a28 2023-07-08 jrick
2000 24df9a28 2023-07-08 jrick __dead static void
2001 24df9a28 2023-07-08 jrick usage_checkout(void)
2002 24df9a28 2023-07-08 jrick {
2003 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s checkout [-Eq] [-b branch] [-c commit] "
2004 24df9a28 2023-07-08 jrick "[-p path-prefix] repository-path [work-tree-path]\n",
2005 24df9a28 2023-07-08 jrick getprogname());
2006 24df9a28 2023-07-08 jrick exit(1);
2007 24df9a28 2023-07-08 jrick }
2008 24df9a28 2023-07-08 jrick
2009 24df9a28 2023-07-08 jrick static void
2010 24df9a28 2023-07-08 jrick show_worktree_base_ref_warning(void)
2011 24df9a28 2023-07-08 jrick {
2012 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: warning: could not create a reference "
2013 24df9a28 2023-07-08 jrick "to the work tree's base commit; the commit could be "
2014 24df9a28 2023-07-08 jrick "garbage-collected by Git or 'gotadmin cleanup'; making the "
2015 24df9a28 2023-07-08 jrick "repository writable and running 'got update' will prevent this\n",
2016 24df9a28 2023-07-08 jrick getprogname());
2017 24df9a28 2023-07-08 jrick }
2018 24df9a28 2023-07-08 jrick
2019 24df9a28 2023-07-08 jrick struct got_checkout_progress_arg {
2020 24df9a28 2023-07-08 jrick const char *worktree_path;
2021 24df9a28 2023-07-08 jrick int had_base_commit_ref_error;
2022 24df9a28 2023-07-08 jrick int verbosity;
2023 24df9a28 2023-07-08 jrick };
2024 24df9a28 2023-07-08 jrick
2025 24df9a28 2023-07-08 jrick static const struct got_error *
2026 24df9a28 2023-07-08 jrick checkout_progress(void *arg, unsigned char status, const char *path)
2027 24df9a28 2023-07-08 jrick {
2028 24df9a28 2023-07-08 jrick struct got_checkout_progress_arg *a = arg;
2029 24df9a28 2023-07-08 jrick
2030 24df9a28 2023-07-08 jrick /* Base commit bump happens silently. */
2031 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_BUMP_BASE)
2032 24df9a28 2023-07-08 jrick return NULL;
2033 24df9a28 2023-07-08 jrick
2034 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_BASE_REF_ERR) {
2035 24df9a28 2023-07-08 jrick a->had_base_commit_ref_error = 1;
2036 24df9a28 2023-07-08 jrick return NULL;
2037 24df9a28 2023-07-08 jrick }
2038 24df9a28 2023-07-08 jrick
2039 24df9a28 2023-07-08 jrick while (path[0] == '/')
2040 24df9a28 2023-07-08 jrick path++;
2041 24df9a28 2023-07-08 jrick
2042 24df9a28 2023-07-08 jrick if (a->verbosity >= 0)
2043 24df9a28 2023-07-08 jrick printf("%c %s/%s\n", status, a->worktree_path, path);
2044 24df9a28 2023-07-08 jrick
2045 24df9a28 2023-07-08 jrick return NULL;
2046 24df9a28 2023-07-08 jrick }
2047 24df9a28 2023-07-08 jrick
2048 24df9a28 2023-07-08 jrick static const struct got_error *
2049 24df9a28 2023-07-08 jrick check_cancelled(void *arg)
2050 24df9a28 2023-07-08 jrick {
2051 24df9a28 2023-07-08 jrick if (sigint_received || sigpipe_received)
2052 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_CANCELLED);
2053 24df9a28 2023-07-08 jrick return NULL;
2054 24df9a28 2023-07-08 jrick }
2055 24df9a28 2023-07-08 jrick
2056 24df9a28 2023-07-08 jrick static const struct got_error *
2057 24df9a28 2023-07-08 jrick check_linear_ancestry(struct got_object_id *commit_id,
2058 24df9a28 2023-07-08 jrick struct got_object_id *base_commit_id, int allow_forwards_in_time_only,
2059 24df9a28 2023-07-08 jrick struct got_repository *repo)
2060 24df9a28 2023-07-08 jrick {
2061 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
2062 24df9a28 2023-07-08 jrick struct got_object_id *yca_id;
2063 24df9a28 2023-07-08 jrick
2064 24df9a28 2023-07-08 jrick err = got_commit_graph_find_youngest_common_ancestor(&yca_id,
2065 24df9a28 2023-07-08 jrick commit_id, base_commit_id, 1, repo, check_cancelled, NULL);
2066 24df9a28 2023-07-08 jrick if (err)
2067 24df9a28 2023-07-08 jrick return err;
2068 24df9a28 2023-07-08 jrick
2069 24df9a28 2023-07-08 jrick if (yca_id == NULL)
2070 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_ANCESTRY);
2071 24df9a28 2023-07-08 jrick
2072 24df9a28 2023-07-08 jrick /*
2073 24df9a28 2023-07-08 jrick * Require a straight line of history between the target commit
2074 24df9a28 2023-07-08 jrick * and the work tree's base commit.
2075 24df9a28 2023-07-08 jrick *
2076 24df9a28 2023-07-08 jrick * Non-linear situations such as this require a rebase:
2077 24df9a28 2023-07-08 jrick *
2078 24df9a28 2023-07-08 jrick * (commit) D F (base_commit)
2079 24df9a28 2023-07-08 jrick * \ /
2080 24df9a28 2023-07-08 jrick * C E
2081 24df9a28 2023-07-08 jrick * \ /
2082 24df9a28 2023-07-08 jrick * B (yca)
2083 24df9a28 2023-07-08 jrick * |
2084 24df9a28 2023-07-08 jrick * A
2085 24df9a28 2023-07-08 jrick *
2086 24df9a28 2023-07-08 jrick * 'got update' only handles linear cases:
2087 24df9a28 2023-07-08 jrick * Update forwards in time: A (base/yca) - B - C - D (commit)
2088 24df9a28 2023-07-08 jrick * Update backwards in time: D (base) - C - B - A (commit/yca)
2089 24df9a28 2023-07-08 jrick */
2090 24df9a28 2023-07-08 jrick if (allow_forwards_in_time_only) {
2091 24df9a28 2023-07-08 jrick if (got_object_id_cmp(base_commit_id, yca_id) != 0)
2092 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_ANCESTRY);
2093 24df9a28 2023-07-08 jrick } else if (got_object_id_cmp(commit_id, yca_id) != 0 &&
2094 24df9a28 2023-07-08 jrick got_object_id_cmp(base_commit_id, yca_id) != 0)
2095 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_ANCESTRY);
2096 24df9a28 2023-07-08 jrick
2097 24df9a28 2023-07-08 jrick free(yca_id);
2098 24df9a28 2023-07-08 jrick return NULL;
2099 24df9a28 2023-07-08 jrick }
2100 24df9a28 2023-07-08 jrick
2101 24df9a28 2023-07-08 jrick static const struct got_error *
2102 24df9a28 2023-07-08 jrick check_same_branch(struct got_object_id *commit_id,
2103 24df9a28 2023-07-08 jrick struct got_reference *head_ref, struct got_repository *repo)
2104 24df9a28 2023-07-08 jrick {
2105 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
2106 24df9a28 2023-07-08 jrick struct got_commit_graph *graph = NULL;
2107 24df9a28 2023-07-08 jrick struct got_object_id *head_commit_id = NULL;
2108 24df9a28 2023-07-08 jrick
2109 24df9a28 2023-07-08 jrick err = got_ref_resolve(&head_commit_id, repo, head_ref);
2110 24df9a28 2023-07-08 jrick if (err)
2111 24df9a28 2023-07-08 jrick goto done;
2112 24df9a28 2023-07-08 jrick
2113 24df9a28 2023-07-08 jrick if (got_object_id_cmp(head_commit_id, commit_id) == 0)
2114 24df9a28 2023-07-08 jrick goto done;
2115 24df9a28 2023-07-08 jrick
2116 24df9a28 2023-07-08 jrick err = got_commit_graph_open(&graph, "/", 1);
2117 24df9a28 2023-07-08 jrick if (err)
2118 24df9a28 2023-07-08 jrick goto done;
2119 24df9a28 2023-07-08 jrick
2120 24df9a28 2023-07-08 jrick err = got_commit_graph_iter_start(graph, head_commit_id, repo,
2121 24df9a28 2023-07-08 jrick check_cancelled, NULL);
2122 24df9a28 2023-07-08 jrick if (err)
2123 24df9a28 2023-07-08 jrick goto done;
2124 24df9a28 2023-07-08 jrick
2125 24df9a28 2023-07-08 jrick for (;;) {
2126 24df9a28 2023-07-08 jrick struct got_object_id id;
2127 24df9a28 2023-07-08 jrick
2128 24df9a28 2023-07-08 jrick err = got_commit_graph_iter_next(&id, graph, repo,
2129 24df9a28 2023-07-08 jrick check_cancelled, NULL);
2130 24df9a28 2023-07-08 jrick if (err) {
2131 24df9a28 2023-07-08 jrick if (err->code == GOT_ERR_ITER_COMPLETED)
2132 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_ANCESTRY);
2133 24df9a28 2023-07-08 jrick break;
2134 24df9a28 2023-07-08 jrick }
2135 24df9a28 2023-07-08 jrick
2136 24df9a28 2023-07-08 jrick if (got_object_id_cmp(&id, commit_id) == 0)
2137 24df9a28 2023-07-08 jrick break;
2138 24df9a28 2023-07-08 jrick }
2139 24df9a28 2023-07-08 jrick done:
2140 24df9a28 2023-07-08 jrick if (graph)
2141 24df9a28 2023-07-08 jrick got_commit_graph_close(graph);
2142 24df9a28 2023-07-08 jrick free(head_commit_id);
2143 24df9a28 2023-07-08 jrick return err;
2144 24df9a28 2023-07-08 jrick }
2145 24df9a28 2023-07-08 jrick
2146 24df9a28 2023-07-08 jrick static const struct got_error *
2147 24df9a28 2023-07-08 jrick checkout_ancestry_error(struct got_reference *ref, const char *commit_id_str)
2148 24df9a28 2023-07-08 jrick {
2149 24df9a28 2023-07-08 jrick static char msg[512];
2150 24df9a28 2023-07-08 jrick const char *branch_name;
2151 24df9a28 2023-07-08 jrick
2152 24df9a28 2023-07-08 jrick if (got_ref_is_symbolic(ref))
2153 24df9a28 2023-07-08 jrick branch_name = got_ref_get_symref_target(ref);
2154 24df9a28 2023-07-08 jrick else
2155 24df9a28 2023-07-08 jrick branch_name = got_ref_get_name(ref);
2156 24df9a28 2023-07-08 jrick
2157 24df9a28 2023-07-08 jrick if (strncmp("refs/heads/", branch_name, 11) == 0)
2158 24df9a28 2023-07-08 jrick branch_name += 11;
2159 24df9a28 2023-07-08 jrick
2160 24df9a28 2023-07-08 jrick snprintf(msg, sizeof(msg),
2161 24df9a28 2023-07-08 jrick "target commit is not contained in branch '%s'; "
2162 24df9a28 2023-07-08 jrick "the branch to use must be specified with -b; "
2163 24df9a28 2023-07-08 jrick "if necessary a new branch can be created for "
2164 24df9a28 2023-07-08 jrick "this commit with 'got branch -c %s BRANCH_NAME'",
2165 24df9a28 2023-07-08 jrick branch_name, commit_id_str);
2166 24df9a28 2023-07-08 jrick
2167 24df9a28 2023-07-08 jrick return got_error_msg(GOT_ERR_ANCESTRY, msg);
2168 24df9a28 2023-07-08 jrick }
2169 24df9a28 2023-07-08 jrick
2170 24df9a28 2023-07-08 jrick static const struct got_error *
2171 24df9a28 2023-07-08 jrick cmd_checkout(int argc, char *argv[])
2172 24df9a28 2023-07-08 jrick {
2173 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
2174 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
2175 24df9a28 2023-07-08 jrick struct got_reference *head_ref = NULL, *ref = NULL;
2176 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
2177 24df9a28 2023-07-08 jrick char *repo_path = NULL;
2178 24df9a28 2023-07-08 jrick char *worktree_path = NULL;
2179 24df9a28 2023-07-08 jrick const char *path_prefix = "";
2180 24df9a28 2023-07-08 jrick const char *branch_name = GOT_REF_HEAD, *refname = NULL;
2181 24df9a28 2023-07-08 jrick char *commit_id_str = NULL;
2182 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
2183 24df9a28 2023-07-08 jrick char *cwd = NULL;
2184 24df9a28 2023-07-08 jrick int ch, same_path_prefix, allow_nonempty = 0, verbosity = 0;
2185 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
2186 24df9a28 2023-07-08 jrick struct got_checkout_progress_arg cpa;
2187 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
2188 24df9a28 2023-07-08 jrick
2189 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
2190 24df9a28 2023-07-08 jrick
2191 24df9a28 2023-07-08 jrick #ifndef PROFILE
2192 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
2193 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
2194 24df9a28 2023-07-08 jrick err(1, "pledge");
2195 24df9a28 2023-07-08 jrick #endif
2196 24df9a28 2023-07-08 jrick
2197 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "b:c:Ep:q")) != -1) {
2198 24df9a28 2023-07-08 jrick switch (ch) {
2199 24df9a28 2023-07-08 jrick case 'b':
2200 24df9a28 2023-07-08 jrick branch_name = optarg;
2201 24df9a28 2023-07-08 jrick break;
2202 24df9a28 2023-07-08 jrick case 'c':
2203 24df9a28 2023-07-08 jrick commit_id_str = strdup(optarg);
2204 24df9a28 2023-07-08 jrick if (commit_id_str == NULL)
2205 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
2206 24df9a28 2023-07-08 jrick break;
2207 24df9a28 2023-07-08 jrick case 'E':
2208 24df9a28 2023-07-08 jrick allow_nonempty = 1;
2209 24df9a28 2023-07-08 jrick break;
2210 24df9a28 2023-07-08 jrick case 'p':
2211 24df9a28 2023-07-08 jrick path_prefix = optarg;
2212 24df9a28 2023-07-08 jrick break;
2213 24df9a28 2023-07-08 jrick case 'q':
2214 24df9a28 2023-07-08 jrick verbosity = -1;
2215 24df9a28 2023-07-08 jrick break;
2216 24df9a28 2023-07-08 jrick default:
2217 24df9a28 2023-07-08 jrick usage_checkout();
2218 24df9a28 2023-07-08 jrick /* NOTREACHED */
2219 24df9a28 2023-07-08 jrick }
2220 24df9a28 2023-07-08 jrick }
2221 24df9a28 2023-07-08 jrick
2222 24df9a28 2023-07-08 jrick argc -= optind;
2223 24df9a28 2023-07-08 jrick argv += optind;
2224 24df9a28 2023-07-08 jrick
2225 24df9a28 2023-07-08 jrick if (argc == 1) {
2226 24df9a28 2023-07-08 jrick char *base, *dotgit;
2227 24df9a28 2023-07-08 jrick const char *path;
2228 24df9a28 2023-07-08 jrick repo_path = realpath(argv[0], NULL);
2229 24df9a28 2023-07-08 jrick if (repo_path == NULL)
2230 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath", argv[0]);
2231 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
2232 24df9a28 2023-07-08 jrick if (cwd == NULL) {
2233 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
2234 24df9a28 2023-07-08 jrick goto done;
2235 24df9a28 2023-07-08 jrick }
2236 24df9a28 2023-07-08 jrick if (path_prefix[0])
2237 24df9a28 2023-07-08 jrick path = path_prefix;
2238 24df9a28 2023-07-08 jrick else
2239 24df9a28 2023-07-08 jrick path = repo_path;
2240 24df9a28 2023-07-08 jrick error = got_path_basename(&base, path);
2241 24df9a28 2023-07-08 jrick if (error)
2242 24df9a28 2023-07-08 jrick goto done;
2243 24df9a28 2023-07-08 jrick dotgit = strstr(base, ".git");
2244 24df9a28 2023-07-08 jrick if (dotgit)
2245 24df9a28 2023-07-08 jrick *dotgit = '\0';
2246 24df9a28 2023-07-08 jrick if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
2247 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
2248 24df9a28 2023-07-08 jrick free(base);
2249 24df9a28 2023-07-08 jrick goto done;
2250 24df9a28 2023-07-08 jrick }
2251 24df9a28 2023-07-08 jrick free(base);
2252 24df9a28 2023-07-08 jrick } else if (argc == 2) {
2253 24df9a28 2023-07-08 jrick repo_path = realpath(argv[0], NULL);
2254 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
2255 24df9a28 2023-07-08 jrick error = got_error_from_errno2("realpath", argv[0]);
2256 24df9a28 2023-07-08 jrick goto done;
2257 24df9a28 2023-07-08 jrick }
2258 24df9a28 2023-07-08 jrick worktree_path = realpath(argv[1], NULL);
2259 24df9a28 2023-07-08 jrick if (worktree_path == NULL) {
2260 24df9a28 2023-07-08 jrick if (errno != ENOENT) {
2261 24df9a28 2023-07-08 jrick error = got_error_from_errno2("realpath",
2262 24df9a28 2023-07-08 jrick argv[1]);
2263 24df9a28 2023-07-08 jrick goto done;
2264 24df9a28 2023-07-08 jrick }
2265 24df9a28 2023-07-08 jrick worktree_path = strdup(argv[1]);
2266 24df9a28 2023-07-08 jrick if (worktree_path == NULL) {
2267 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
2268 24df9a28 2023-07-08 jrick goto done;
2269 24df9a28 2023-07-08 jrick }
2270 24df9a28 2023-07-08 jrick }
2271 24df9a28 2023-07-08 jrick } else
2272 24df9a28 2023-07-08 jrick usage_checkout();
2273 24df9a28 2023-07-08 jrick
2274 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
2275 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(worktree_path);
2276 24df9a28 2023-07-08 jrick
2277 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
2278 24df9a28 2023-07-08 jrick if (error != NULL)
2279 24df9a28 2023-07-08 jrick goto done;
2280 24df9a28 2023-07-08 jrick
2281 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
2282 24df9a28 2023-07-08 jrick if (error != NULL)
2283 24df9a28 2023-07-08 jrick goto done;
2284 24df9a28 2023-07-08 jrick
2285 24df9a28 2023-07-08 jrick /* Pre-create work tree path for unveil(2) */
2286 24df9a28 2023-07-08 jrick error = got_path_mkdir(worktree_path);
2287 24df9a28 2023-07-08 jrick if (error) {
2288 24df9a28 2023-07-08 jrick if (!(error->code == GOT_ERR_ERRNO && errno == EISDIR) &&
2289 24df9a28 2023-07-08 jrick !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
2290 24df9a28 2023-07-08 jrick goto done;
2291 24df9a28 2023-07-08 jrick if (!allow_nonempty &&
2292 24df9a28 2023-07-08 jrick !got_path_dir_is_empty(worktree_path)) {
2293 24df9a28 2023-07-08 jrick error = got_error_path(worktree_path,
2294 24df9a28 2023-07-08 jrick GOT_ERR_DIR_NOT_EMPTY);
2295 24df9a28 2023-07-08 jrick goto done;
2296 24df9a28 2023-07-08 jrick }
2297 24df9a28 2023-07-08 jrick }
2298 24df9a28 2023-07-08 jrick
2299 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0, worktree_path);
2300 24df9a28 2023-07-08 jrick if (error)
2301 24df9a28 2023-07-08 jrick goto done;
2302 24df9a28 2023-07-08 jrick
2303 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo, branch_name, 0);
2304 24df9a28 2023-07-08 jrick if (error != NULL)
2305 24df9a28 2023-07-08 jrick goto done;
2306 24df9a28 2023-07-08 jrick
2307 24df9a28 2023-07-08 jrick error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
2308 24df9a28 2023-07-08 jrick if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
2309 24df9a28 2023-07-08 jrick goto done;
2310 24df9a28 2023-07-08 jrick
2311 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, worktree_path);
2312 24df9a28 2023-07-08 jrick if (error != NULL)
2313 24df9a28 2023-07-08 jrick goto done;
2314 24df9a28 2023-07-08 jrick
2315 24df9a28 2023-07-08 jrick error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
2316 24df9a28 2023-07-08 jrick path_prefix);
2317 24df9a28 2023-07-08 jrick if (error != NULL)
2318 24df9a28 2023-07-08 jrick goto done;
2319 24df9a28 2023-07-08 jrick if (!same_path_prefix) {
2320 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_PATH_PREFIX);
2321 24df9a28 2023-07-08 jrick goto done;
2322 24df9a28 2023-07-08 jrick }
2323 24df9a28 2023-07-08 jrick
2324 24df9a28 2023-07-08 jrick if (commit_id_str) {
2325 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
2326 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
2327 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
2328 24df9a28 2023-07-08 jrick NULL);
2329 24df9a28 2023-07-08 jrick if (error)
2330 24df9a28 2023-07-08 jrick goto done;
2331 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
2332 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
2333 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
2334 24df9a28 2023-07-08 jrick if (error)
2335 24df9a28 2023-07-08 jrick goto done;
2336 24df9a28 2023-07-08 jrick error = check_linear_ancestry(commit_id,
2337 24df9a28 2023-07-08 jrick got_worktree_get_base_commit_id(worktree), 0, repo);
2338 24df9a28 2023-07-08 jrick if (error != NULL) {
2339 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_ANCESTRY) {
2340 24df9a28 2023-07-08 jrick error = checkout_ancestry_error(
2341 24df9a28 2023-07-08 jrick head_ref, commit_id_str);
2342 24df9a28 2023-07-08 jrick }
2343 24df9a28 2023-07-08 jrick goto done;
2344 24df9a28 2023-07-08 jrick }
2345 24df9a28 2023-07-08 jrick error = check_same_branch(commit_id, head_ref, repo);
2346 24df9a28 2023-07-08 jrick if (error) {
2347 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_ANCESTRY) {
2348 24df9a28 2023-07-08 jrick error = checkout_ancestry_error(
2349 24df9a28 2023-07-08 jrick head_ref, commit_id_str);
2350 24df9a28 2023-07-08 jrick }
2351 24df9a28 2023-07-08 jrick goto done;
2352 24df9a28 2023-07-08 jrick }
2353 24df9a28 2023-07-08 jrick error = got_worktree_set_base_commit_id(worktree, repo,
2354 24df9a28 2023-07-08 jrick commit_id);
2355 24df9a28 2023-07-08 jrick if (error)
2356 24df9a28 2023-07-08 jrick goto done;
2357 24df9a28 2023-07-08 jrick /* Expand potentially abbreviated commit ID string. */
2358 24df9a28 2023-07-08 jrick free(commit_id_str);
2359 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
2360 24df9a28 2023-07-08 jrick if (error)
2361 24df9a28 2023-07-08 jrick goto done;
2362 24df9a28 2023-07-08 jrick } else {
2363 24df9a28 2023-07-08 jrick commit_id = got_object_id_dup(
2364 24df9a28 2023-07-08 jrick got_worktree_get_base_commit_id(worktree));
2365 24df9a28 2023-07-08 jrick if (commit_id == NULL) {
2366 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_object_id_dup");
2367 24df9a28 2023-07-08 jrick goto done;
2368 24df9a28 2023-07-08 jrick }
2369 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
2370 24df9a28 2023-07-08 jrick if (error)
2371 24df9a28 2023-07-08 jrick goto done;
2372 24df9a28 2023-07-08 jrick }
2373 24df9a28 2023-07-08 jrick
2374 24df9a28 2023-07-08 jrick error = got_pathlist_append(&paths, "", NULL);
2375 24df9a28 2023-07-08 jrick if (error)
2376 24df9a28 2023-07-08 jrick goto done;
2377 24df9a28 2023-07-08 jrick cpa.worktree_path = worktree_path;
2378 24df9a28 2023-07-08 jrick cpa.had_base_commit_ref_error = 0;
2379 24df9a28 2023-07-08 jrick cpa.verbosity = verbosity;
2380 24df9a28 2023-07-08 jrick error = got_worktree_checkout_files(worktree, &paths, repo,
2381 24df9a28 2023-07-08 jrick checkout_progress, &cpa, check_cancelled, NULL);
2382 24df9a28 2023-07-08 jrick if (error != NULL)
2383 24df9a28 2023-07-08 jrick goto done;
2384 24df9a28 2023-07-08 jrick
2385 24df9a28 2023-07-08 jrick if (got_ref_is_symbolic(head_ref)) {
2386 24df9a28 2023-07-08 jrick error = got_ref_resolve_symbolic(&ref, repo, head_ref);
2387 24df9a28 2023-07-08 jrick if (error)
2388 24df9a28 2023-07-08 jrick goto done;
2389 24df9a28 2023-07-08 jrick refname = got_ref_get_name(ref);
2390 24df9a28 2023-07-08 jrick } else
2391 24df9a28 2023-07-08 jrick refname = got_ref_get_name(head_ref);
2392 24df9a28 2023-07-08 jrick printf("Checked out %s: %s\n", refname, commit_id_str);
2393 24df9a28 2023-07-08 jrick printf("Now shut up and hack\n");
2394 24df9a28 2023-07-08 jrick if (cpa.had_base_commit_ref_error)
2395 24df9a28 2023-07-08 jrick show_worktree_base_ref_warning();
2396 24df9a28 2023-07-08 jrick done:
2397 24df9a28 2023-07-08 jrick if (pack_fds) {
2398 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
2399 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
2400 24df9a28 2023-07-08 jrick if (error == NULL)
2401 24df9a28 2023-07-08 jrick error = pack_err;
2402 24df9a28 2023-07-08 jrick }
2403 24df9a28 2023-07-08 jrick if (head_ref)
2404 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
2405 24df9a28 2023-07-08 jrick if (ref)
2406 24df9a28 2023-07-08 jrick got_ref_close(ref);
2407 24df9a28 2023-07-08 jrick if (repo) {
2408 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
2409 24df9a28 2023-07-08 jrick if (error == NULL)
2410 24df9a28 2023-07-08 jrick error = close_err;
2411 24df9a28 2023-07-08 jrick }
2412 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_NONE);
2413 24df9a28 2023-07-08 jrick free(commit_id_str);
2414 24df9a28 2023-07-08 jrick free(commit_id);
2415 24df9a28 2023-07-08 jrick free(repo_path);
2416 24df9a28 2023-07-08 jrick free(worktree_path);
2417 24df9a28 2023-07-08 jrick free(cwd);
2418 24df9a28 2023-07-08 jrick return error;
2419 24df9a28 2023-07-08 jrick }
2420 24df9a28 2023-07-08 jrick
2421 24df9a28 2023-07-08 jrick struct got_update_progress_arg {
2422 24df9a28 2023-07-08 jrick int did_something;
2423 24df9a28 2023-07-08 jrick int conflicts;
2424 24df9a28 2023-07-08 jrick int obstructed;
2425 24df9a28 2023-07-08 jrick int not_updated;
2426 24df9a28 2023-07-08 jrick int missing;
2427 24df9a28 2023-07-08 jrick int not_deleted;
2428 24df9a28 2023-07-08 jrick int unversioned;
2429 24df9a28 2023-07-08 jrick int verbosity;
2430 24df9a28 2023-07-08 jrick };
2431 24df9a28 2023-07-08 jrick
2432 24df9a28 2023-07-08 jrick static void
2433 24df9a28 2023-07-08 jrick print_update_progress_stats(struct got_update_progress_arg *upa)
2434 24df9a28 2023-07-08 jrick {
2435 24df9a28 2023-07-08 jrick if (!upa->did_something)
2436 24df9a28 2023-07-08 jrick return;
2437 24df9a28 2023-07-08 jrick
2438 24df9a28 2023-07-08 jrick if (upa->conflicts > 0)
2439 24df9a28 2023-07-08 jrick printf("Files with new merge conflicts: %d\n", upa->conflicts);
2440 24df9a28 2023-07-08 jrick if (upa->obstructed > 0)
2441 24df9a28 2023-07-08 jrick printf("File paths obstructed by a non-regular file: %d\n",
2442 24df9a28 2023-07-08 jrick upa->obstructed);
2443 24df9a28 2023-07-08 jrick if (upa->not_updated > 0)
2444 24df9a28 2023-07-08 jrick printf("Files not updated because of existing merge "
2445 24df9a28 2023-07-08 jrick "conflicts: %d\n", upa->not_updated);
2446 24df9a28 2023-07-08 jrick }
2447 24df9a28 2023-07-08 jrick
2448 24df9a28 2023-07-08 jrick /*
2449 24df9a28 2023-07-08 jrick * The meaning of some status codes differs between merge-style operations and
2450 24df9a28 2023-07-08 jrick * update operations. For example, the ! status code means "file was missing"
2451 24df9a28 2023-07-08 jrick * if changes were merged into the work tree, and "missing file was restored"
2452 24df9a28 2023-07-08 jrick * if the work tree was updated. This function should be used by any operation
2453 24df9a28 2023-07-08 jrick * which merges changes into the work tree without updating the work tree.
2454 24df9a28 2023-07-08 jrick */
2455 24df9a28 2023-07-08 jrick static void
2456 24df9a28 2023-07-08 jrick print_merge_progress_stats(struct got_update_progress_arg *upa)
2457 24df9a28 2023-07-08 jrick {
2458 24df9a28 2023-07-08 jrick if (!upa->did_something)
2459 24df9a28 2023-07-08 jrick return;
2460 24df9a28 2023-07-08 jrick
2461 24df9a28 2023-07-08 jrick if (upa->conflicts > 0)
2462 24df9a28 2023-07-08 jrick printf("Files with new merge conflicts: %d\n", upa->conflicts);
2463 24df9a28 2023-07-08 jrick if (upa->obstructed > 0)
2464 24df9a28 2023-07-08 jrick printf("File paths obstructed by a non-regular file: %d\n",
2465 24df9a28 2023-07-08 jrick upa->obstructed);
2466 24df9a28 2023-07-08 jrick if (upa->missing > 0)
2467 24df9a28 2023-07-08 jrick printf("Files which had incoming changes but could not be "
2468 24df9a28 2023-07-08 jrick "found in the work tree: %d\n", upa->missing);
2469 24df9a28 2023-07-08 jrick if (upa->not_deleted > 0)
2470 24df9a28 2023-07-08 jrick printf("Files not deleted due to differences in deleted "
2471 24df9a28 2023-07-08 jrick "content: %d\n", upa->not_deleted);
2472 24df9a28 2023-07-08 jrick if (upa->unversioned > 0)
2473 24df9a28 2023-07-08 jrick printf("Files not merged because an unversioned file was "
2474 24df9a28 2023-07-08 jrick "found in the work tree: %d\n", upa->unversioned);
2475 24df9a28 2023-07-08 jrick }
2476 24df9a28 2023-07-08 jrick
2477 24df9a28 2023-07-08 jrick __dead static void
2478 24df9a28 2023-07-08 jrick usage_update(void)
2479 24df9a28 2023-07-08 jrick {
2480 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s update [-qtvX] [-c commit] [-r remote] "
2481 24df9a28 2023-07-08 jrick "[path ...]\n", getprogname());
2482 24df9a28 2023-07-08 jrick exit(1);
2483 24df9a28 2023-07-08 jrick }
2484 24df9a28 2023-07-08 jrick
2485 24df9a28 2023-07-08 jrick static const struct got_error *
2486 24df9a28 2023-07-08 jrick update_progress(void *arg, unsigned char status, const char *path)
2487 24df9a28 2023-07-08 jrick {
2488 24df9a28 2023-07-08 jrick struct got_update_progress_arg *upa = arg;
2489 24df9a28 2023-07-08 jrick
2490 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_EXISTS ||
2491 24df9a28 2023-07-08 jrick status == GOT_STATUS_BASE_REF_ERR)
2492 24df9a28 2023-07-08 jrick return NULL;
2493 24df9a28 2023-07-08 jrick
2494 24df9a28 2023-07-08 jrick upa->did_something = 1;
2495 24df9a28 2023-07-08 jrick
2496 24df9a28 2023-07-08 jrick /* Base commit bump happens silently. */
2497 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_BUMP_BASE)
2498 24df9a28 2023-07-08 jrick return NULL;
2499 24df9a28 2023-07-08 jrick
2500 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_CONFLICT)
2501 24df9a28 2023-07-08 jrick upa->conflicts++;
2502 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_OBSTRUCTED)
2503 24df9a28 2023-07-08 jrick upa->obstructed++;
2504 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_CANNOT_UPDATE)
2505 24df9a28 2023-07-08 jrick upa->not_updated++;
2506 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_MISSING)
2507 24df9a28 2023-07-08 jrick upa->missing++;
2508 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_CANNOT_DELETE)
2509 24df9a28 2023-07-08 jrick upa->not_deleted++;
2510 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_UNVERSIONED)
2511 24df9a28 2023-07-08 jrick upa->unversioned++;
2512 24df9a28 2023-07-08 jrick
2513 24df9a28 2023-07-08 jrick while (path[0] == '/')
2514 24df9a28 2023-07-08 jrick path++;
2515 24df9a28 2023-07-08 jrick if (upa->verbosity >= 0)
2516 24df9a28 2023-07-08 jrick printf("%c %s\n", status, path);
2517 24df9a28 2023-07-08 jrick
2518 24df9a28 2023-07-08 jrick return NULL;
2519 24df9a28 2023-07-08 jrick }
2520 24df9a28 2023-07-08 jrick
2521 24df9a28 2023-07-08 jrick static const struct got_error *
2522 24df9a28 2023-07-08 jrick check_rebase_or_histedit_in_progress(struct got_worktree *worktree)
2523 24df9a28 2023-07-08 jrick {
2524 24df9a28 2023-07-08 jrick const struct got_error *err;
2525 24df9a28 2023-07-08 jrick int in_progress;
2526 24df9a28 2023-07-08 jrick
2527 24df9a28 2023-07-08 jrick err = got_worktree_rebase_in_progress(&in_progress, worktree);
2528 24df9a28 2023-07-08 jrick if (err)
2529 24df9a28 2023-07-08 jrick return err;
2530 24df9a28 2023-07-08 jrick if (in_progress)
2531 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_REBASING);
2532 24df9a28 2023-07-08 jrick
2533 24df9a28 2023-07-08 jrick err = got_worktree_histedit_in_progress(&in_progress, worktree);
2534 24df9a28 2023-07-08 jrick if (err)
2535 24df9a28 2023-07-08 jrick return err;
2536 24df9a28 2023-07-08 jrick if (in_progress)
2537 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_HISTEDIT_BUSY);
2538 24df9a28 2023-07-08 jrick
2539 24df9a28 2023-07-08 jrick return NULL;
2540 24df9a28 2023-07-08 jrick }
2541 24df9a28 2023-07-08 jrick
2542 24df9a28 2023-07-08 jrick static const struct got_error *
2543 24df9a28 2023-07-08 jrick check_merge_in_progress(struct got_worktree *worktree,
2544 24df9a28 2023-07-08 jrick struct got_repository *repo)
2545 24df9a28 2023-07-08 jrick {
2546 24df9a28 2023-07-08 jrick const struct got_error *err;
2547 24df9a28 2023-07-08 jrick int in_progress;
2548 24df9a28 2023-07-08 jrick
2549 24df9a28 2023-07-08 jrick err = got_worktree_merge_in_progress(&in_progress, worktree, repo);
2550 24df9a28 2023-07-08 jrick if (err)
2551 24df9a28 2023-07-08 jrick return err;
2552 24df9a28 2023-07-08 jrick if (in_progress)
2553 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_MERGE_BUSY);
2554 24df9a28 2023-07-08 jrick
2555 24df9a28 2023-07-08 jrick return NULL;
2556 24df9a28 2023-07-08 jrick }
2557 24df9a28 2023-07-08 jrick
2558 24df9a28 2023-07-08 jrick static const struct got_error *
2559 24df9a28 2023-07-08 jrick get_worktree_paths_from_argv(struct got_pathlist_head *paths, int argc,
2560 24df9a28 2023-07-08 jrick char *argv[], struct got_worktree *worktree)
2561 24df9a28 2023-07-08 jrick {
2562 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
2563 24df9a28 2023-07-08 jrick char *path;
2564 24df9a28 2023-07-08 jrick struct got_pathlist_entry *new;
2565 24df9a28 2023-07-08 jrick int i;
2566 24df9a28 2023-07-08 jrick
2567 24df9a28 2023-07-08 jrick if (argc == 0) {
2568 24df9a28 2023-07-08 jrick path = strdup("");
2569 24df9a28 2023-07-08 jrick if (path == NULL)
2570 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
2571 24df9a28 2023-07-08 jrick return got_pathlist_append(paths, path, NULL);
2572 24df9a28 2023-07-08 jrick }
2573 24df9a28 2023-07-08 jrick
2574 24df9a28 2023-07-08 jrick for (i = 0; i < argc; i++) {
2575 24df9a28 2023-07-08 jrick err = got_worktree_resolve_path(&path, worktree, argv[i]);
2576 24df9a28 2023-07-08 jrick if (err)
2577 24df9a28 2023-07-08 jrick break;
2578 24df9a28 2023-07-08 jrick err = got_pathlist_insert(&new, paths, path, NULL);
2579 24df9a28 2023-07-08 jrick if (err || new == NULL /* duplicate */) {
2580 24df9a28 2023-07-08 jrick free(path);
2581 24df9a28 2023-07-08 jrick if (err)
2582 24df9a28 2023-07-08 jrick break;
2583 24df9a28 2023-07-08 jrick }
2584 24df9a28 2023-07-08 jrick }
2585 24df9a28 2023-07-08 jrick
2586 24df9a28 2023-07-08 jrick return err;
2587 24df9a28 2023-07-08 jrick }
2588 24df9a28 2023-07-08 jrick
2589 24df9a28 2023-07-08 jrick static const struct got_error *
2590 24df9a28 2023-07-08 jrick wrap_not_worktree_error(const struct got_error *orig_err,
2591 24df9a28 2023-07-08 jrick const char *cmdname, const char *path)
2592 24df9a28 2023-07-08 jrick {
2593 24df9a28 2023-07-08 jrick const struct got_error *err;
2594 24df9a28 2023-07-08 jrick struct got_repository *repo;
2595 24df9a28 2023-07-08 jrick static char msg[512];
2596 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
2597 24df9a28 2023-07-08 jrick
2598 24df9a28 2023-07-08 jrick err = got_repo_pack_fds_open(&pack_fds);
2599 24df9a28 2023-07-08 jrick if (err)
2600 24df9a28 2023-07-08 jrick return err;
2601 24df9a28 2023-07-08 jrick
2602 24df9a28 2023-07-08 jrick err = got_repo_open(&repo, path, NULL, pack_fds);
2603 24df9a28 2023-07-08 jrick if (err)
2604 24df9a28 2023-07-08 jrick return orig_err;
2605 24df9a28 2023-07-08 jrick
2606 24df9a28 2023-07-08 jrick snprintf(msg, sizeof(msg),
2607 24df9a28 2023-07-08 jrick "'got %s' needs a work tree in addition to a git repository\n"
2608 24df9a28 2023-07-08 jrick "Work trees can be checked out from this Git repository with "
2609 24df9a28 2023-07-08 jrick "'got checkout'.\n"
2610 24df9a28 2023-07-08 jrick "The got(1) manual page contains more information.", cmdname);
2611 24df9a28 2023-07-08 jrick err = got_error_msg(GOT_ERR_NOT_WORKTREE, msg);
2612 24df9a28 2023-07-08 jrick if (repo) {
2613 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
2614 24df9a28 2023-07-08 jrick if (err == NULL)
2615 24df9a28 2023-07-08 jrick err = close_err;
2616 24df9a28 2023-07-08 jrick }
2617 24df9a28 2023-07-08 jrick if (pack_fds) {
2618 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
2619 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
2620 24df9a28 2023-07-08 jrick if (err == NULL)
2621 24df9a28 2023-07-08 jrick err = pack_err;
2622 24df9a28 2023-07-08 jrick }
2623 24df9a28 2023-07-08 jrick return err;
2624 24df9a28 2023-07-08 jrick }
2625 24df9a28 2023-07-08 jrick
2626 24df9a28 2023-07-08 jrick static const struct got_error *
2627 24df9a28 2023-07-08 jrick cmd_update(int argc, char *argv[])
2628 24df9a28 2023-07-08 jrick {
2629 24df9a28 2023-07-08 jrick const struct got_error *error = NULL, *unlock_err;
2630 24df9a28 2023-07-08 jrick char *worktree_path = NULL;
2631 24df9a28 2023-07-08 jrick const char *repo_path = NULL;
2632 24df9a28 2023-07-08 jrick const char *remote_name = NULL;
2633 24df9a28 2023-07-08 jrick char *proto = NULL, *host = NULL, *port = NULL;
2634 24df9a28 2023-07-08 jrick char *repo_name = NULL, *server_path = NULL;
2635 24df9a28 2023-07-08 jrick const struct got_remote_repo *remotes, *remote = NULL;
2636 24df9a28 2023-07-08 jrick int nremotes;
2637 24df9a28 2023-07-08 jrick char *id_str = NULL;
2638 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
2639 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
2640 24df9a28 2023-07-08 jrick const struct got_gotconfig *repo_conf = NULL, *worktree_conf = NULL;
2641 24df9a28 2023-07-08 jrick struct got_pathlist_head paths, refs, symrefs;
2642 24df9a28 2023-07-08 jrick struct got_pathlist_head wanted_branches, wanted_refs;
2643 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
2644 24df9a28 2023-07-08 jrick struct got_reflist_head remote_refs;
2645 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
2646 24df9a28 2023-07-08 jrick struct got_object_id *pack_hash = NULL;
2647 24df9a28 2023-07-08 jrick int i, ch, fetchfd = -1, fetchstatus;
2648 24df9a28 2023-07-08 jrick pid_t fetchpid = -1;
2649 24df9a28 2023-07-08 jrick struct got_fetch_progress_arg fpa;
2650 24df9a28 2023-07-08 jrick struct got_update_progress_arg upa;
2651 24df9a28 2023-07-08 jrick int verbosity = 0;
2652 24df9a28 2023-07-08 jrick int delete_remote = 0;
2653 24df9a28 2023-07-08 jrick int replace_tags = 0;
2654 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
2655 24df9a28 2023-07-08 jrick const char *remote_head = NULL, *worktree_branch = NULL;
2656 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
2657 24df9a28 2023-07-08 jrick char *commit_id_str = NULL;
2658 24df9a28 2023-07-08 jrick const char *refname;
2659 24df9a28 2023-07-08 jrick struct got_reference *head_ref = NULL;
2660 24df9a28 2023-07-08 jrick
2661 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
2662 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
2663 24df9a28 2023-07-08 jrick TAILQ_INIT(&symrefs);
2664 24df9a28 2023-07-08 jrick TAILQ_INIT(&remote_refs);
2665 24df9a28 2023-07-08 jrick TAILQ_INIT(&wanted_branches);
2666 24df9a28 2023-07-08 jrick TAILQ_INIT(&wanted_refs);
2667 24df9a28 2023-07-08 jrick
2668 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:qr:vX")) != -1) {
2669 24df9a28 2023-07-08 jrick switch (ch) {
2670 24df9a28 2023-07-08 jrick case 'c':
2671 24df9a28 2023-07-08 jrick commit_id_str = strdup(optarg);
2672 24df9a28 2023-07-08 jrick if (commit_id_str == NULL)
2673 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
2674 24df9a28 2023-07-08 jrick break;
2675 24df9a28 2023-07-08 jrick case 't':
2676 24df9a28 2023-07-08 jrick replace_tags = 1;
2677 24df9a28 2023-07-08 jrick break;
2678 24df9a28 2023-07-08 jrick case 'q':
2679 24df9a28 2023-07-08 jrick verbosity = -1;
2680 24df9a28 2023-07-08 jrick break;
2681 24df9a28 2023-07-08 jrick case 'r':
2682 24df9a28 2023-07-08 jrick remote_name = optarg;
2683 24df9a28 2023-07-08 jrick break;
2684 24df9a28 2023-07-08 jrick case 'v':
2685 24df9a28 2023-07-08 jrick if (verbosity < 0)
2686 24df9a28 2023-07-08 jrick verbosity = 0;
2687 24df9a28 2023-07-08 jrick else if (verbosity < 3)
2688 24df9a28 2023-07-08 jrick verbosity++;
2689 24df9a28 2023-07-08 jrick break;
2690 24df9a28 2023-07-08 jrick case 'X':
2691 24df9a28 2023-07-08 jrick delete_remote = 1;
2692 24df9a28 2023-07-08 jrick break;
2693 24df9a28 2023-07-08 jrick default:
2694 24df9a28 2023-07-08 jrick usage_update();
2695 24df9a28 2023-07-08 jrick break;
2696 24df9a28 2023-07-08 jrick }
2697 24df9a28 2023-07-08 jrick }
2698 24df9a28 2023-07-08 jrick argc -= optind;
2699 24df9a28 2023-07-08 jrick argv += optind;
2700 24df9a28 2023-07-08 jrick
2701 24df9a28 2023-07-08 jrick if (delete_remote) {
2702 24df9a28 2023-07-08 jrick if (replace_tags)
2703 24df9a28 2023-07-08 jrick option_conflict('X', 't');
2704 24df9a28 2023-07-08 jrick if (remote_name == NULL)
2705 24df9a28 2023-07-08 jrick errx(1, "-X option requires a remote name");
2706 24df9a28 2023-07-08 jrick }
2707 24df9a28 2023-07-08 jrick if (remote_name == NULL)
2708 24df9a28 2023-07-08 jrick remote_name = GOT_FETCH_DEFAULT_REMOTE_NAME;
2709 24df9a28 2023-07-08 jrick
2710 24df9a28 2023-07-08 jrick worktree_path = getcwd(NULL, 0);
2711 24df9a28 2023-07-08 jrick if (worktree_path == NULL) {
2712 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
2713 24df9a28 2023-07-08 jrick goto done;
2714 24df9a28 2023-07-08 jrick }
2715 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, worktree_path);
2716 24df9a28 2023-07-08 jrick if (error) {
2717 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
2718 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "update",
2719 24df9a28 2023-07-08 jrick worktree_path);
2720 24df9a28 2023-07-08 jrick goto done;
2721 24df9a28 2023-07-08 jrick }
2722 24df9a28 2023-07-08 jrick repo_path = got_worktree_get_repo_path(worktree);
2723 24df9a28 2023-07-08 jrick
2724 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
2725 24df9a28 2023-07-08 jrick if (error != NULL)
2726 24df9a28 2023-07-08 jrick goto done;
2727 24df9a28 2023-07-08 jrick
2728 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
2729 24df9a28 2023-07-08 jrick if (error)
2730 24df9a28 2023-07-08 jrick goto done;
2731 24df9a28 2023-07-08 jrick
2732 24df9a28 2023-07-08 jrick error = check_rebase_or_histedit_in_progress(worktree);
2733 24df9a28 2023-07-08 jrick if (error)
2734 24df9a28 2023-07-08 jrick goto done;
2735 24df9a28 2023-07-08 jrick error = check_merge_in_progress(worktree, repo);
2736 24df9a28 2023-07-08 jrick if (error)
2737 24df9a28 2023-07-08 jrick goto done;
2738 24df9a28 2023-07-08 jrick
2739 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
2740 24df9a28 2023-07-08 jrick if (error)
2741 24df9a28 2023-07-08 jrick goto done;
2742 24df9a28 2023-07-08 jrick
2743 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
2744 24df9a28 2023-07-08 jrick if (worktree_conf) {
2745 24df9a28 2023-07-08 jrick got_gotconfig_get_remotes(&nremotes, &remotes,
2746 24df9a28 2023-07-08 jrick worktree_conf);
2747 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
2748 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
2749 24df9a28 2023-07-08 jrick remote = &remotes[i];
2750 24df9a28 2023-07-08 jrick break;
2751 24df9a28 2023-07-08 jrick }
2752 24df9a28 2023-07-08 jrick }
2753 24df9a28 2023-07-08 jrick }
2754 24df9a28 2023-07-08 jrick
2755 24df9a28 2023-07-08 jrick if (remote == NULL) {
2756 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
2757 24df9a28 2023-07-08 jrick if (repo_conf) {
2758 24df9a28 2023-07-08 jrick got_gotconfig_get_remotes(&nremotes, &remotes,
2759 24df9a28 2023-07-08 jrick repo_conf);
2760 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
2761 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
2762 24df9a28 2023-07-08 jrick remote = &remotes[i];
2763 24df9a28 2023-07-08 jrick break;
2764 24df9a28 2023-07-08 jrick }
2765 24df9a28 2023-07-08 jrick }
2766 24df9a28 2023-07-08 jrick }
2767 24df9a28 2023-07-08 jrick }
2768 24df9a28 2023-07-08 jrick if (remote == NULL) {
2769 24df9a28 2023-07-08 jrick got_repo_get_gitconfig_remotes(&nremotes, &remotes, repo);
2770 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
2771 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
2772 24df9a28 2023-07-08 jrick remote = &remotes[i];
2773 24df9a28 2023-07-08 jrick break;
2774 24df9a28 2023-07-08 jrick }
2775 24df9a28 2023-07-08 jrick }
2776 24df9a28 2023-07-08 jrick }
2777 24df9a28 2023-07-08 jrick if (remote == NULL) {
2778 24df9a28 2023-07-08 jrick error = got_error_path(remote_name, GOT_ERR_NO_REMOTE);
2779 24df9a28 2023-07-08 jrick goto done;
2780 24df9a28 2023-07-08 jrick }
2781 24df9a28 2023-07-08 jrick
2782 24df9a28 2023-07-08 jrick error = got_dial_parse_uri(&proto, &host, &port, &server_path,
2783 24df9a28 2023-07-08 jrick &repo_name, remote->fetch_url);
2784 24df9a28 2023-07-08 jrick if (error)
2785 24df9a28 2023-07-08 jrick goto done;
2786 24df9a28 2023-07-08 jrick
2787 24df9a28 2023-07-08 jrick if (strcmp(proto, "git") == 0) {
2788 24df9a28 2023-07-08 jrick #ifndef PROFILE
2789 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2790 24df9a28 2023-07-08 jrick "sendfd dns inet unveil", NULL) == -1)
2791 24df9a28 2023-07-08 jrick err(1, "pledge");
2792 24df9a28 2023-07-08 jrick #endif
2793 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "git+ssh") == 0 ||
2794 24df9a28 2023-07-08 jrick strcmp(proto, "ssh") == 0) {
2795 24df9a28 2023-07-08 jrick #ifndef PROFILE
2796 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2797 24df9a28 2023-07-08 jrick "sendfd unveil", NULL) == -1)
2798 24df9a28 2023-07-08 jrick err(1, "pledge");
2799 24df9a28 2023-07-08 jrick #endif
2800 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "http") == 0 ||
2801 24df9a28 2023-07-08 jrick strcmp(proto, "git+http") == 0) {
2802 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_NOT_IMPL);
2803 24df9a28 2023-07-08 jrick goto done;
2804 24df9a28 2023-07-08 jrick } else {
2805 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_BAD_PROTO);
2806 24df9a28 2023-07-08 jrick goto done;
2807 24df9a28 2023-07-08 jrick }
2808 24df9a28 2023-07-08 jrick
2809 24df9a28 2023-07-08 jrick error = got_dial_apply_unveil(proto);
2810 24df9a28 2023-07-08 jrick if (error)
2811 24df9a28 2023-07-08 jrick goto done;
2812 24df9a28 2023-07-08 jrick
2813 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
2814 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
2815 24df9a28 2023-07-08 jrick if (error)
2816 24df9a28 2023-07-08 jrick goto done;
2817 24df9a28 2023-07-08 jrick
2818 24df9a28 2023-07-08 jrick if (verbosity >= 0) {
2819 24df9a28 2023-07-08 jrick printf("Connecting to \"%s\" %s://%s%s%s%s%s\n",
2820 24df9a28 2023-07-08 jrick remote->name, proto, host,
2821 24df9a28 2023-07-08 jrick port ? ":" : "", port ? port : "",
2822 24df9a28 2023-07-08 jrick *server_path == '/' ? "" : "/", server_path);
2823 24df9a28 2023-07-08 jrick }
2824 24df9a28 2023-07-08 jrick
2825 24df9a28 2023-07-08 jrick error = got_fetch_connect(&fetchpid, &fetchfd, proto, host, port,
2826 24df9a28 2023-07-08 jrick server_path, verbosity);
2827 24df9a28 2023-07-08 jrick if (error)
2828 24df9a28 2023-07-08 jrick goto done;
2829 24df9a28 2023-07-08 jrick
2830 24df9a28 2023-07-08 jrick /*
2831 24df9a28 2023-07-08 jrick * If set, get this remote's HEAD ref target so
2832 24df9a28 2023-07-08 jrick * if it has changed on the server we can fetch it.
2833 24df9a28 2023-07-08 jrick */
2834 24df9a28 2023-07-08 jrick error = got_ref_list(&remote_refs, repo, "refs/remotes",
2835 24df9a28 2023-07-08 jrick got_ref_cmp_by_name, repo);
2836 24df9a28 2023-07-08 jrick if (error)
2837 24df9a28 2023-07-08 jrick goto done;
2838 24df9a28 2023-07-08 jrick
2839 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &remote_refs, entry) {
2840 24df9a28 2023-07-08 jrick const char *remote_refname, *remote_target;
2841 24df9a28 2023-07-08 jrick size_t remote_name_len;
2842 24df9a28 2023-07-08 jrick
2843 24df9a28 2023-07-08 jrick if (!got_ref_is_symbolic(re->ref))
2844 24df9a28 2023-07-08 jrick continue;
2845 24df9a28 2023-07-08 jrick
2846 24df9a28 2023-07-08 jrick remote_name_len = strlen(remote->name);
2847 24df9a28 2023-07-08 jrick remote_refname = got_ref_get_name(re->ref);
2848 24df9a28 2023-07-08 jrick
2849 24df9a28 2023-07-08 jrick /* we only want refs/remotes/$remote->name/HEAD */
2850 24df9a28 2023-07-08 jrick if (strncmp(remote_refname + 13, remote->name,
2851 24df9a28 2023-07-08 jrick remote_name_len) != 0)
2852 24df9a28 2023-07-08 jrick continue;
2853 24df9a28 2023-07-08 jrick
2854 24df9a28 2023-07-08 jrick if (strcmp(remote_refname + remote_name_len + 14,
2855 24df9a28 2023-07-08 jrick GOT_REF_HEAD) != 0)
2856 24df9a28 2023-07-08 jrick continue;
2857 24df9a28 2023-07-08 jrick
2858 24df9a28 2023-07-08 jrick /*
2859 24df9a28 2023-07-08 jrick * Take the name itself because we already
2860 24df9a28 2023-07-08 jrick * only match with refs/heads/ in fetch_pack().
2861 24df9a28 2023-07-08 jrick */
2862 24df9a28 2023-07-08 jrick remote_target = got_ref_get_symref_target(re->ref);
2863 24df9a28 2023-07-08 jrick remote_head = remote_target + remote_name_len + 14;
2864 24df9a28 2023-07-08 jrick break;
2865 24df9a28 2023-07-08 jrick }
2866 24df9a28 2023-07-08 jrick
2867 24df9a28 2023-07-08 jrick refname = got_worktree_get_head_ref_name(worktree);
2868 24df9a28 2023-07-08 jrick if (strncmp(refname, "refs/heads/", 11) == 0)
2869 24df9a28 2023-07-08 jrick worktree_branch = refname;
2870 24df9a28 2023-07-08 jrick
2871 24df9a28 2023-07-08 jrick fpa.last_scaled_size[0] = '\0';
2872 24df9a28 2023-07-08 jrick fpa.last_p_indexed = -1;
2873 24df9a28 2023-07-08 jrick fpa.last_p_resolved = -1;
2874 24df9a28 2023-07-08 jrick fpa.verbosity = verbosity;
2875 24df9a28 2023-07-08 jrick fpa.repo = repo;
2876 24df9a28 2023-07-08 jrick fpa.create_configs = 0;
2877 24df9a28 2023-07-08 jrick fpa.configs_created = 0;
2878 24df9a28 2023-07-08 jrick memset(&fpa.config_info, 0, sizeof(fpa.config_info));
2879 24df9a28 2023-07-08 jrick
2880 24df9a28 2023-07-08 jrick error = got_fetch_pack(&pack_hash, &refs, &symrefs, remote->name,
2881 24df9a28 2023-07-08 jrick remote->mirror_references, 0, &wanted_branches, &wanted_refs,
2882 24df9a28 2023-07-08 jrick 0, verbosity, fetchfd, repo, worktree_branch, remote_head,
2883 24df9a28 2023-07-08 jrick 0, fetch_progress, &fpa);
2884 24df9a28 2023-07-08 jrick if (error)
2885 24df9a28 2023-07-08 jrick goto done;
2886 24df9a28 2023-07-08 jrick
2887 24df9a28 2023-07-08 jrick if (pack_hash != NULL && verbosity >= 0) {
2888 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str, pack_hash);
2889 24df9a28 2023-07-08 jrick if (error)
2890 24df9a28 2023-07-08 jrick goto done;
2891 24df9a28 2023-07-08 jrick printf("\nFetched %s.pack\n", id_str);
2892 24df9a28 2023-07-08 jrick free(id_str);
2893 24df9a28 2023-07-08 jrick id_str = NULL;
2894 24df9a28 2023-07-08 jrick }
2895 24df9a28 2023-07-08 jrick
2896 24df9a28 2023-07-08 jrick /* Update references provided with the pack file. */
2897 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &refs, entry) {
2898 24df9a28 2023-07-08 jrick const char *refname = pe->path;
2899 24df9a28 2023-07-08 jrick struct got_object_id *id = pe->data;
2900 24df9a28 2023-07-08 jrick struct got_reference *ref;
2901 24df9a28 2023-07-08 jrick
2902 24df9a28 2023-07-08 jrick if (is_wanted_ref(&wanted_refs, refname)) {
2903 24df9a28 2023-07-08 jrick error = update_wanted_ref(refname, id,
2904 24df9a28 2023-07-08 jrick remote->name, verbosity, repo);
2905 24df9a28 2023-07-08 jrick if (error)
2906 24df9a28 2023-07-08 jrick goto done;
2907 24df9a28 2023-07-08 jrick continue;
2908 24df9a28 2023-07-08 jrick }
2909 24df9a28 2023-07-08 jrick
2910 24df9a28 2023-07-08 jrick error = got_ref_open(&ref, repo, refname, 1);
2911 24df9a28 2023-07-08 jrick if (error) {
2912 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_REF)
2913 24df9a28 2023-07-08 jrick goto done;
2914 24df9a28 2023-07-08 jrick error = create_ref(refname, id, verbosity,
2915 24df9a28 2023-07-08 jrick repo);
2916 24df9a28 2023-07-08 jrick if (error)
2917 24df9a28 2023-07-08 jrick goto done;
2918 24df9a28 2023-07-08 jrick } else {
2919 24df9a28 2023-07-08 jrick error = update_ref(ref, id, replace_tags,
2920 24df9a28 2023-07-08 jrick verbosity-1, repo);
2921 24df9a28 2023-07-08 jrick unlock_err = got_ref_unlock(ref);
2922 24df9a28 2023-07-08 jrick if (unlock_err && error == NULL)
2923 24df9a28 2023-07-08 jrick error = unlock_err;
2924 24df9a28 2023-07-08 jrick got_ref_close(ref);
2925 24df9a28 2023-07-08 jrick if (error)
2926 24df9a28 2023-07-08 jrick goto done;
2927 24df9a28 2023-07-08 jrick }
2928 24df9a28 2023-07-08 jrick }
2929 24df9a28 2023-07-08 jrick
2930 24df9a28 2023-07-08 jrick /* Update worktree */
2931 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo,
2932 24df9a28 2023-07-08 jrick got_worktree_get_head_ref_name(worktree), 0);
2933 24df9a28 2023-07-08 jrick if (error != NULL)
2934 24df9a28 2023-07-08 jrick goto done;
2935 24df9a28 2023-07-08 jrick if (commit_id_str == NULL) {
2936 24df9a28 2023-07-08 jrick error = got_ref_resolve(&commit_id, repo, head_ref);
2937 24df9a28 2023-07-08 jrick if (error != NULL)
2938 24df9a28 2023-07-08 jrick goto done;
2939 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
2940 24df9a28 2023-07-08 jrick if (error != NULL)
2941 24df9a28 2023-07-08 jrick goto done;
2942 24df9a28 2023-07-08 jrick } else {
2943 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
2944 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
2945 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
2946 24df9a28 2023-07-08 jrick NULL);
2947 24df9a28 2023-07-08 jrick if (error)
2948 24df9a28 2023-07-08 jrick goto done;
2949 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
2950 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
2951 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
2952 24df9a28 2023-07-08 jrick free(commit_id_str);
2953 24df9a28 2023-07-08 jrick commit_id_str = NULL;
2954 24df9a28 2023-07-08 jrick if (error)
2955 24df9a28 2023-07-08 jrick goto done;
2956 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
2957 24df9a28 2023-07-08 jrick if (error)
2958 24df9a28 2023-07-08 jrick goto done;
2959 24df9a28 2023-07-08 jrick }
2960 24df9a28 2023-07-08 jrick
2961 24df9a28 2023-07-08 jrick
2962 24df9a28 2023-07-08 jrick error = check_linear_ancestry(commit_id,
2963 24df9a28 2023-07-08 jrick got_worktree_get_base_commit_id(worktree), 0, repo);
2964 24df9a28 2023-07-08 jrick if (error != NULL) {
2965 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_ANCESTRY)
2966 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_BRANCH_MOVED);
2967 24df9a28 2023-07-08 jrick goto done;
2968 24df9a28 2023-07-08 jrick }
2969 24df9a28 2023-07-08 jrick error = check_same_branch(commit_id, head_ref, repo);
2970 24df9a28 2023-07-08 jrick if (error)
2971 24df9a28 2023-07-08 jrick goto done;
2972 24df9a28 2023-07-08 jrick
2973 24df9a28 2023-07-08 jrick if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
2974 24df9a28 2023-07-08 jrick commit_id) != 0) {
2975 24df9a28 2023-07-08 jrick error = got_worktree_set_base_commit_id(worktree, repo,
2976 24df9a28 2023-07-08 jrick commit_id);
2977 24df9a28 2023-07-08 jrick if (error)
2978 24df9a28 2023-07-08 jrick goto done;
2979 24df9a28 2023-07-08 jrick }
2980 24df9a28 2023-07-08 jrick
2981 24df9a28 2023-07-08 jrick memset(&upa, 0, sizeof(upa));
2982 24df9a28 2023-07-08 jrick upa.verbosity = verbosity;
2983 24df9a28 2023-07-08 jrick error = got_worktree_checkout_files(worktree, &paths, repo,
2984 24df9a28 2023-07-08 jrick update_progress, &upa, check_cancelled, NULL);
2985 24df9a28 2023-07-08 jrick if (error != NULL)
2986 24df9a28 2023-07-08 jrick goto done;
2987 24df9a28 2023-07-08 jrick
2988 24df9a28 2023-07-08 jrick if (upa.did_something) {
2989 24df9a28 2023-07-08 jrick printf("Updated to %s: %s\n",
2990 24df9a28 2023-07-08 jrick got_worktree_get_head_ref_name(worktree), commit_id_str);
2991 24df9a28 2023-07-08 jrick } else
2992 24df9a28 2023-07-08 jrick printf("Already up-to-date\n");
2993 24df9a28 2023-07-08 jrick
2994 24df9a28 2023-07-08 jrick print_update_progress_stats(&upa);
2995 24df9a28 2023-07-08 jrick done:
2996 24df9a28 2023-07-08 jrick if (fetchpid > 0) {
2997 24df9a28 2023-07-08 jrick if (kill(fetchpid, SIGTERM) == -1)
2998 24df9a28 2023-07-08 jrick error = got_error_from_errno("kill");
2999 24df9a28 2023-07-08 jrick if (waitpid(fetchpid, &fetchstatus, 0) == -1 && error == NULL)
3000 24df9a28 2023-07-08 jrick error = got_error_from_errno("waitpid");
3001 24df9a28 2023-07-08 jrick }
3002 24df9a28 2023-07-08 jrick if (fetchfd != -1 && close(fetchfd) == -1 && error == NULL)
3003 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
3004 24df9a28 2023-07-08 jrick if (repo) {
3005 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
3006 24df9a28 2023-07-08 jrick if (error == NULL)
3007 24df9a28 2023-07-08 jrick error = close_err;
3008 24df9a28 2023-07-08 jrick }
3009 24df9a28 2023-07-08 jrick if (worktree)
3010 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
3011 24df9a28 2023-07-08 jrick if (pack_fds) {
3012 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
3013 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
3014 24df9a28 2023-07-08 jrick if (error == NULL)
3015 24df9a28 2023-07-08 jrick error = pack_err;
3016 24df9a28 2023-07-08 jrick }
3017 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
3018 24df9a28 2023-07-08 jrick got_pathlist_free(&refs, GOT_PATHLIST_FREE_ALL);
3019 24df9a28 2023-07-08 jrick got_pathlist_free(&symrefs, GOT_PATHLIST_FREE_ALL);
3020 24df9a28 2023-07-08 jrick got_pathlist_free(&wanted_branches, GOT_PATHLIST_FREE_NONE);
3021 24df9a28 2023-07-08 jrick got_pathlist_free(&wanted_refs, GOT_PATHLIST_FREE_NONE);
3022 24df9a28 2023-07-08 jrick got_ref_list_free(&remote_refs);
3023 24df9a28 2023-07-08 jrick free(id_str);
3024 24df9a28 2023-07-08 jrick free(worktree_path);
3025 24df9a28 2023-07-08 jrick free(pack_hash);
3026 24df9a28 2023-07-08 jrick free(proto);
3027 24df9a28 2023-07-08 jrick free(host);
3028 24df9a28 2023-07-08 jrick free(port);
3029 24df9a28 2023-07-08 jrick free(server_path);
3030 24df9a28 2023-07-08 jrick free(repo_name);
3031 24df9a28 2023-07-08 jrick free(commit_id);
3032 24df9a28 2023-07-08 jrick free(commit_id_str);
3033 24df9a28 2023-07-08 jrick return error;
3034 24df9a28 2023-07-08 jrick }
3035 24df9a28 2023-07-08 jrick
3036 24df9a28 2023-07-08 jrick static const struct got_error *
3037 24df9a28 2023-07-08 jrick diff_blobs(struct got_object_id *blob_id1, struct got_object_id *blob_id2,
3038 24df9a28 2023-07-08 jrick const char *path, int diff_context, int ignore_whitespace,
3039 24df9a28 2023-07-08 jrick int force_text_diff, struct got_diffstat_cb_arg *dsa,
3040 24df9a28 2023-07-08 jrick struct got_repository *repo, FILE *outfile)
3041 24df9a28 2023-07-08 jrick {
3042 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3043 24df9a28 2023-07-08 jrick struct got_blob_object *blob1 = NULL, *blob2 = NULL;
3044 24df9a28 2023-07-08 jrick FILE *f1 = NULL, *f2 = NULL;
3045 24df9a28 2023-07-08 jrick int fd1 = -1, fd2 = -1;
3046 24df9a28 2023-07-08 jrick
3047 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
3048 24df9a28 2023-07-08 jrick if (fd1 == -1)
3049 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentempfd");
3050 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
3051 24df9a28 2023-07-08 jrick if (fd2 == -1) {
3052 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
3053 24df9a28 2023-07-08 jrick goto done;
3054 24df9a28 2023-07-08 jrick }
3055 24df9a28 2023-07-08 jrick
3056 24df9a28 2023-07-08 jrick if (blob_id1) {
3057 24df9a28 2023-07-08 jrick err = got_object_open_as_blob(&blob1, repo, blob_id1, 8192,
3058 24df9a28 2023-07-08 jrick fd1);
3059 24df9a28 2023-07-08 jrick if (err)
3060 24df9a28 2023-07-08 jrick goto done;
3061 24df9a28 2023-07-08 jrick }
3062 24df9a28 2023-07-08 jrick
3063 24df9a28 2023-07-08 jrick err = got_object_open_as_blob(&blob2, repo, blob_id2, 8192, fd2);
3064 24df9a28 2023-07-08 jrick if (err)
3065 24df9a28 2023-07-08 jrick goto done;
3066 24df9a28 2023-07-08 jrick
3067 24df9a28 2023-07-08 jrick f1 = got_opentemp();
3068 24df9a28 2023-07-08 jrick if (f1 == NULL) {
3069 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3070 24df9a28 2023-07-08 jrick goto done;
3071 24df9a28 2023-07-08 jrick }
3072 24df9a28 2023-07-08 jrick f2 = got_opentemp();
3073 24df9a28 2023-07-08 jrick if (f2 == NULL) {
3074 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3075 24df9a28 2023-07-08 jrick goto done;
3076 24df9a28 2023-07-08 jrick }
3077 24df9a28 2023-07-08 jrick
3078 24df9a28 2023-07-08 jrick while (path[0] == '/')
3079 24df9a28 2023-07-08 jrick path++;
3080 24df9a28 2023-07-08 jrick err = got_diff_blob(NULL, NULL, blob1, blob2, f1, f2, path, path,
3081 24df9a28 2023-07-08 jrick GOT_DIFF_ALGORITHM_PATIENCE, diff_context, ignore_whitespace,
3082 24df9a28 2023-07-08 jrick force_text_diff, dsa, outfile);
3083 24df9a28 2023-07-08 jrick done:
3084 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && err == NULL)
3085 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3086 24df9a28 2023-07-08 jrick if (blob1)
3087 24df9a28 2023-07-08 jrick got_object_blob_close(blob1);
3088 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && err == NULL)
3089 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3090 24df9a28 2023-07-08 jrick if (blob2)
3091 24df9a28 2023-07-08 jrick got_object_blob_close(blob2);
3092 24df9a28 2023-07-08 jrick if (f1 && fclose(f1) == EOF && err == NULL)
3093 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3094 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && err == NULL)
3095 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3096 24df9a28 2023-07-08 jrick return err;
3097 24df9a28 2023-07-08 jrick }
3098 24df9a28 2023-07-08 jrick
3099 24df9a28 2023-07-08 jrick static const struct got_error *
3100 24df9a28 2023-07-08 jrick diff_trees(struct got_object_id *tree_id1, struct got_object_id *tree_id2,
3101 24df9a28 2023-07-08 jrick const char *path, int diff_context, int ignore_whitespace,
3102 24df9a28 2023-07-08 jrick int force_text_diff, struct got_diffstat_cb_arg *dsa,
3103 24df9a28 2023-07-08 jrick struct got_repository *repo, FILE *outfile)
3104 24df9a28 2023-07-08 jrick {
3105 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3106 24df9a28 2023-07-08 jrick struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3107 24df9a28 2023-07-08 jrick struct got_diff_blob_output_unidiff_arg arg;
3108 24df9a28 2023-07-08 jrick FILE *f1 = NULL, *f2 = NULL;
3109 24df9a28 2023-07-08 jrick int fd1 = -1, fd2 = -1;
3110 24df9a28 2023-07-08 jrick
3111 24df9a28 2023-07-08 jrick if (tree_id1) {
3112 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree1, repo, tree_id1);
3113 24df9a28 2023-07-08 jrick if (err)
3114 24df9a28 2023-07-08 jrick goto done;
3115 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
3116 24df9a28 2023-07-08 jrick if (fd1 == -1) {
3117 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
3118 24df9a28 2023-07-08 jrick goto done;
3119 24df9a28 2023-07-08 jrick }
3120 24df9a28 2023-07-08 jrick }
3121 24df9a28 2023-07-08 jrick
3122 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree2, repo, tree_id2);
3123 24df9a28 2023-07-08 jrick if (err)
3124 24df9a28 2023-07-08 jrick goto done;
3125 24df9a28 2023-07-08 jrick
3126 24df9a28 2023-07-08 jrick f1 = got_opentemp();
3127 24df9a28 2023-07-08 jrick if (f1 == NULL) {
3128 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3129 24df9a28 2023-07-08 jrick goto done;
3130 24df9a28 2023-07-08 jrick }
3131 24df9a28 2023-07-08 jrick
3132 24df9a28 2023-07-08 jrick f2 = got_opentemp();
3133 24df9a28 2023-07-08 jrick if (f2 == NULL) {
3134 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3135 24df9a28 2023-07-08 jrick goto done;
3136 24df9a28 2023-07-08 jrick }
3137 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
3138 24df9a28 2023-07-08 jrick if (fd2 == -1) {
3139 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
3140 24df9a28 2023-07-08 jrick goto done;
3141 24df9a28 2023-07-08 jrick }
3142 24df9a28 2023-07-08 jrick arg.diff_context = diff_context;
3143 24df9a28 2023-07-08 jrick arg.ignore_whitespace = ignore_whitespace;
3144 24df9a28 2023-07-08 jrick arg.force_text_diff = force_text_diff;
3145 24df9a28 2023-07-08 jrick arg.diffstat = dsa;
3146 24df9a28 2023-07-08 jrick arg.diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
3147 24df9a28 2023-07-08 jrick arg.outfile = outfile;
3148 24df9a28 2023-07-08 jrick arg.lines = NULL;
3149 24df9a28 2023-07-08 jrick arg.nlines = 0;
3150 24df9a28 2023-07-08 jrick while (path[0] == '/')
3151 24df9a28 2023-07-08 jrick path++;
3152 24df9a28 2023-07-08 jrick err = got_diff_tree(tree1, tree2, f1, f2, fd1, fd2, path, path, repo,
3153 24df9a28 2023-07-08 jrick got_diff_blob_output_unidiff, &arg, 1);
3154 24df9a28 2023-07-08 jrick done:
3155 24df9a28 2023-07-08 jrick if (tree1)
3156 24df9a28 2023-07-08 jrick got_object_tree_close(tree1);
3157 24df9a28 2023-07-08 jrick if (tree2)
3158 24df9a28 2023-07-08 jrick got_object_tree_close(tree2);
3159 24df9a28 2023-07-08 jrick if (f1 && fclose(f1) == EOF && err == NULL)
3160 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3161 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && err == NULL)
3162 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3163 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && err == NULL)
3164 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3165 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && err == NULL)
3166 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3167 24df9a28 2023-07-08 jrick return err;
3168 24df9a28 2023-07-08 jrick }
3169 24df9a28 2023-07-08 jrick
3170 24df9a28 2023-07-08 jrick static const struct got_error *
3171 24df9a28 2023-07-08 jrick get_changed_paths(struct got_pathlist_head *paths,
3172 24df9a28 2023-07-08 jrick struct got_commit_object *commit, struct got_repository *repo,
3173 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg *dsa)
3174 24df9a28 2023-07-08 jrick {
3175 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3176 24df9a28 2023-07-08 jrick struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
3177 24df9a28 2023-07-08 jrick struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3178 24df9a28 2023-07-08 jrick struct got_object_qid *qid;
3179 24df9a28 2023-07-08 jrick got_diff_blob_cb cb = got_diff_tree_collect_changed_paths;
3180 24df9a28 2023-07-08 jrick FILE *f1 = NULL, *f2 = NULL;
3181 24df9a28 2023-07-08 jrick int fd1 = -1, fd2 = -1;
3182 24df9a28 2023-07-08 jrick
3183 24df9a28 2023-07-08 jrick if (dsa) {
3184 24df9a28 2023-07-08 jrick cb = got_diff_tree_compute_diffstat;
3185 24df9a28 2023-07-08 jrick
3186 24df9a28 2023-07-08 jrick f1 = got_opentemp();
3187 24df9a28 2023-07-08 jrick if (f1 == NULL) {
3188 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3189 24df9a28 2023-07-08 jrick goto done;
3190 24df9a28 2023-07-08 jrick }
3191 24df9a28 2023-07-08 jrick f2 = got_opentemp();
3192 24df9a28 2023-07-08 jrick if (f2 == NULL) {
3193 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3194 24df9a28 2023-07-08 jrick goto done;
3195 24df9a28 2023-07-08 jrick }
3196 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
3197 24df9a28 2023-07-08 jrick if (fd1 == -1) {
3198 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
3199 24df9a28 2023-07-08 jrick goto done;
3200 24df9a28 2023-07-08 jrick }
3201 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
3202 24df9a28 2023-07-08 jrick if (fd2 == -1) {
3203 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
3204 24df9a28 2023-07-08 jrick goto done;
3205 24df9a28 2023-07-08 jrick }
3206 24df9a28 2023-07-08 jrick }
3207 24df9a28 2023-07-08 jrick
3208 24df9a28 2023-07-08 jrick qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3209 24df9a28 2023-07-08 jrick if (qid != NULL) {
3210 24df9a28 2023-07-08 jrick struct got_commit_object *pcommit;
3211 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&pcommit, repo,
3212 24df9a28 2023-07-08 jrick &qid->id);
3213 24df9a28 2023-07-08 jrick if (err)
3214 24df9a28 2023-07-08 jrick return err;
3215 24df9a28 2023-07-08 jrick
3216 24df9a28 2023-07-08 jrick tree_id1 = got_object_id_dup(
3217 24df9a28 2023-07-08 jrick got_object_commit_get_tree_id(pcommit));
3218 24df9a28 2023-07-08 jrick if (tree_id1 == NULL) {
3219 24df9a28 2023-07-08 jrick got_object_commit_close(pcommit);
3220 24df9a28 2023-07-08 jrick return got_error_from_errno("got_object_id_dup");
3221 24df9a28 2023-07-08 jrick }
3222 24df9a28 2023-07-08 jrick got_object_commit_close(pcommit);
3223 24df9a28 2023-07-08 jrick
3224 24df9a28 2023-07-08 jrick }
3225 24df9a28 2023-07-08 jrick
3226 24df9a28 2023-07-08 jrick if (tree_id1) {
3227 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree1, repo, tree_id1);
3228 24df9a28 2023-07-08 jrick if (err)
3229 24df9a28 2023-07-08 jrick goto done;
3230 24df9a28 2023-07-08 jrick }
3231 24df9a28 2023-07-08 jrick
3232 24df9a28 2023-07-08 jrick tree_id2 = got_object_commit_get_tree_id(commit);
3233 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree2, repo, tree_id2);
3234 24df9a28 2023-07-08 jrick if (err)
3235 24df9a28 2023-07-08 jrick goto done;
3236 24df9a28 2023-07-08 jrick
3237 24df9a28 2023-07-08 jrick err = got_diff_tree(tree1, tree2, f1, f2, fd1, fd2, "", "", repo,
3238 24df9a28 2023-07-08 jrick cb, dsa ? (void *)dsa : paths, dsa ? 1 : 0);
3239 24df9a28 2023-07-08 jrick done:
3240 24df9a28 2023-07-08 jrick if (tree1)
3241 24df9a28 2023-07-08 jrick got_object_tree_close(tree1);
3242 24df9a28 2023-07-08 jrick if (tree2)
3243 24df9a28 2023-07-08 jrick got_object_tree_close(tree2);
3244 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && err == NULL)
3245 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3246 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && err == NULL)
3247 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
3248 24df9a28 2023-07-08 jrick if (f1 && fclose(f1) == EOF && err == NULL)
3249 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3250 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && err == NULL)
3251 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3252 24df9a28 2023-07-08 jrick free(tree_id1);
3253 24df9a28 2023-07-08 jrick return err;
3254 24df9a28 2023-07-08 jrick }
3255 24df9a28 2023-07-08 jrick
3256 24df9a28 2023-07-08 jrick static const struct got_error *
3257 24df9a28 2023-07-08 jrick print_patch(struct got_commit_object *commit, struct got_object_id *id,
3258 24df9a28 2023-07-08 jrick const char *path, int diff_context, struct got_diffstat_cb_arg *dsa,
3259 24df9a28 2023-07-08 jrick struct got_repository *repo, FILE *outfile)
3260 24df9a28 2023-07-08 jrick {
3261 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3262 24df9a28 2023-07-08 jrick struct got_commit_object *pcommit = NULL;
3263 24df9a28 2023-07-08 jrick char *id_str1 = NULL, *id_str2 = NULL;
3264 24df9a28 2023-07-08 jrick struct got_object_id *obj_id1 = NULL, *obj_id2 = NULL;
3265 24df9a28 2023-07-08 jrick struct got_object_qid *qid;
3266 24df9a28 2023-07-08 jrick
3267 24df9a28 2023-07-08 jrick qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3268 24df9a28 2023-07-08 jrick if (qid != NULL) {
3269 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&pcommit, repo,
3270 24df9a28 2023-07-08 jrick &qid->id);
3271 24df9a28 2023-07-08 jrick if (err)
3272 24df9a28 2023-07-08 jrick return err;
3273 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str1, &qid->id);
3274 24df9a28 2023-07-08 jrick if (err)
3275 24df9a28 2023-07-08 jrick goto done;
3276 24df9a28 2023-07-08 jrick }
3277 24df9a28 2023-07-08 jrick
3278 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str2, id);
3279 24df9a28 2023-07-08 jrick if (err)
3280 24df9a28 2023-07-08 jrick goto done;
3281 24df9a28 2023-07-08 jrick
3282 24df9a28 2023-07-08 jrick if (path && path[0] != '\0') {
3283 24df9a28 2023-07-08 jrick int obj_type;
3284 24df9a28 2023-07-08 jrick err = got_object_id_by_path(&obj_id2, repo, commit, path);
3285 24df9a28 2023-07-08 jrick if (err)
3286 24df9a28 2023-07-08 jrick goto done;
3287 24df9a28 2023-07-08 jrick if (pcommit) {
3288 24df9a28 2023-07-08 jrick err = got_object_id_by_path(&obj_id1, repo,
3289 24df9a28 2023-07-08 jrick pcommit, path);
3290 24df9a28 2023-07-08 jrick if (err) {
3291 24df9a28 2023-07-08 jrick if (err->code != GOT_ERR_NO_TREE_ENTRY) {
3292 24df9a28 2023-07-08 jrick free(obj_id2);
3293 24df9a28 2023-07-08 jrick goto done;
3294 24df9a28 2023-07-08 jrick }
3295 24df9a28 2023-07-08 jrick }
3296 24df9a28 2023-07-08 jrick }
3297 24df9a28 2023-07-08 jrick err = got_object_get_type(&obj_type, repo, obj_id2);
3298 24df9a28 2023-07-08 jrick if (err) {
3299 24df9a28 2023-07-08 jrick free(obj_id2);
3300 24df9a28 2023-07-08 jrick goto done;
3301 24df9a28 2023-07-08 jrick }
3302 24df9a28 2023-07-08 jrick fprintf(outfile,
3303 24df9a28 2023-07-08 jrick "diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
3304 24df9a28 2023-07-08 jrick fprintf(outfile, "commit - %s\n",
3305 24df9a28 2023-07-08 jrick id_str1 ? id_str1 : "/dev/null");
3306 24df9a28 2023-07-08 jrick fprintf(outfile, "commit + %s\n", id_str2);
3307 24df9a28 2023-07-08 jrick switch (obj_type) {
3308 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_BLOB:
3309 24df9a28 2023-07-08 jrick err = diff_blobs(obj_id1, obj_id2, path, diff_context,
3310 24df9a28 2023-07-08 jrick 0, 0, dsa, repo, outfile);
3311 24df9a28 2023-07-08 jrick break;
3312 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TREE:
3313 24df9a28 2023-07-08 jrick err = diff_trees(obj_id1, obj_id2, path, diff_context,
3314 24df9a28 2023-07-08 jrick 0, 0, dsa, repo, outfile);
3315 24df9a28 2023-07-08 jrick break;
3316 24df9a28 2023-07-08 jrick default:
3317 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_OBJ_TYPE);
3318 24df9a28 2023-07-08 jrick break;
3319 24df9a28 2023-07-08 jrick }
3320 24df9a28 2023-07-08 jrick free(obj_id1);
3321 24df9a28 2023-07-08 jrick free(obj_id2);
3322 24df9a28 2023-07-08 jrick } else {
3323 24df9a28 2023-07-08 jrick obj_id2 = got_object_commit_get_tree_id(commit);
3324 24df9a28 2023-07-08 jrick if (pcommit)
3325 24df9a28 2023-07-08 jrick obj_id1 = got_object_commit_get_tree_id(pcommit);
3326 24df9a28 2023-07-08 jrick fprintf(outfile,
3327 24df9a28 2023-07-08 jrick "diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
3328 24df9a28 2023-07-08 jrick fprintf(outfile, "commit - %s\n",
3329 24df9a28 2023-07-08 jrick id_str1 ? id_str1 : "/dev/null");
3330 24df9a28 2023-07-08 jrick fprintf(outfile, "commit + %s\n", id_str2);
3331 24df9a28 2023-07-08 jrick err = diff_trees(obj_id1, obj_id2, "", diff_context, 0, 0,
3332 24df9a28 2023-07-08 jrick dsa, repo, outfile);
3333 24df9a28 2023-07-08 jrick }
3334 24df9a28 2023-07-08 jrick done:
3335 24df9a28 2023-07-08 jrick free(id_str1);
3336 24df9a28 2023-07-08 jrick free(id_str2);
3337 24df9a28 2023-07-08 jrick if (pcommit)
3338 24df9a28 2023-07-08 jrick got_object_commit_close(pcommit);
3339 24df9a28 2023-07-08 jrick return err;
3340 24df9a28 2023-07-08 jrick }
3341 24df9a28 2023-07-08 jrick
3342 24df9a28 2023-07-08 jrick static char *
3343 24df9a28 2023-07-08 jrick get_datestr(time_t *time, char *datebuf)
3344 24df9a28 2023-07-08 jrick {
3345 24df9a28 2023-07-08 jrick struct tm mytm, *tm;
3346 24df9a28 2023-07-08 jrick char *p, *s;
3347 24df9a28 2023-07-08 jrick
3348 24df9a28 2023-07-08 jrick tm = gmtime_r(time, &mytm);
3349 24df9a28 2023-07-08 jrick if (tm == NULL)
3350 24df9a28 2023-07-08 jrick return NULL;
3351 24df9a28 2023-07-08 jrick s = asctime_r(tm, datebuf);
3352 24df9a28 2023-07-08 jrick if (s == NULL)
3353 24df9a28 2023-07-08 jrick return NULL;
3354 24df9a28 2023-07-08 jrick p = strchr(s, '\n');
3355 24df9a28 2023-07-08 jrick if (p)
3356 24df9a28 2023-07-08 jrick *p = '\0';
3357 24df9a28 2023-07-08 jrick return s;
3358 24df9a28 2023-07-08 jrick }
3359 24df9a28 2023-07-08 jrick
3360 24df9a28 2023-07-08 jrick static const struct got_error *
3361 24df9a28 2023-07-08 jrick match_commit(int *have_match, struct got_object_id *id,
3362 24df9a28 2023-07-08 jrick struct got_commit_object *commit, regex_t *regex)
3363 24df9a28 2023-07-08 jrick {
3364 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3365 24df9a28 2023-07-08 jrick regmatch_t regmatch;
3366 24df9a28 2023-07-08 jrick char *id_str = NULL, *logmsg = NULL;
3367 24df9a28 2023-07-08 jrick
3368 24df9a28 2023-07-08 jrick *have_match = 0;
3369 24df9a28 2023-07-08 jrick
3370 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
3371 24df9a28 2023-07-08 jrick if (err)
3372 24df9a28 2023-07-08 jrick return err;
3373 24df9a28 2023-07-08 jrick
3374 24df9a28 2023-07-08 jrick err = got_object_commit_get_logmsg(&logmsg, commit);
3375 24df9a28 2023-07-08 jrick if (err)
3376 24df9a28 2023-07-08 jrick goto done;
3377 24df9a28 2023-07-08 jrick
3378 24df9a28 2023-07-08 jrick if (regexec(regex, got_object_commit_get_author(commit), 1,
3379 24df9a28 2023-07-08 jrick &regmatch, 0) == 0 ||
3380 24df9a28 2023-07-08 jrick regexec(regex, got_object_commit_get_committer(commit), 1,
3381 24df9a28 2023-07-08 jrick &regmatch, 0) == 0 ||
3382 24df9a28 2023-07-08 jrick regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
3383 24df9a28 2023-07-08 jrick regexec(regex, logmsg, 1, &regmatch, 0) == 0)
3384 24df9a28 2023-07-08 jrick *have_match = 1;
3385 24df9a28 2023-07-08 jrick done:
3386 24df9a28 2023-07-08 jrick free(id_str);
3387 24df9a28 2023-07-08 jrick free(logmsg);
3388 24df9a28 2023-07-08 jrick return err;
3389 24df9a28 2023-07-08 jrick }
3390 24df9a28 2023-07-08 jrick
3391 24df9a28 2023-07-08 jrick static void
3392 24df9a28 2023-07-08 jrick match_changed_paths(int *have_match, struct got_pathlist_head *changed_paths,
3393 24df9a28 2023-07-08 jrick regex_t *regex)
3394 24df9a28 2023-07-08 jrick {
3395 24df9a28 2023-07-08 jrick regmatch_t regmatch;
3396 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
3397 24df9a28 2023-07-08 jrick
3398 24df9a28 2023-07-08 jrick *have_match = 0;
3399 24df9a28 2023-07-08 jrick
3400 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, changed_paths, entry) {
3401 24df9a28 2023-07-08 jrick if (regexec(regex, pe->path, 1, &regmatch, 0) == 0) {
3402 24df9a28 2023-07-08 jrick *have_match = 1;
3403 24df9a28 2023-07-08 jrick break;
3404 24df9a28 2023-07-08 jrick }
3405 24df9a28 2023-07-08 jrick }
3406 24df9a28 2023-07-08 jrick }
3407 24df9a28 2023-07-08 jrick
3408 24df9a28 2023-07-08 jrick static const struct got_error *
3409 24df9a28 2023-07-08 jrick match_patch(int *have_match, struct got_commit_object *commit,
3410 24df9a28 2023-07-08 jrick struct got_object_id *id, const char *path, int diff_context,
3411 24df9a28 2023-07-08 jrick struct got_repository *repo, regex_t *regex, FILE *f)
3412 24df9a28 2023-07-08 jrick {
3413 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3414 24df9a28 2023-07-08 jrick char *line = NULL;
3415 24df9a28 2023-07-08 jrick size_t linesize = 0;
3416 24df9a28 2023-07-08 jrick regmatch_t regmatch;
3417 24df9a28 2023-07-08 jrick
3418 24df9a28 2023-07-08 jrick *have_match = 0;
3419 24df9a28 2023-07-08 jrick
3420 24df9a28 2023-07-08 jrick err = got_opentemp_truncate(f);
3421 24df9a28 2023-07-08 jrick if (err)
3422 24df9a28 2023-07-08 jrick return err;
3423 24df9a28 2023-07-08 jrick
3424 24df9a28 2023-07-08 jrick err = print_patch(commit, id, path, diff_context, NULL, repo, f);
3425 24df9a28 2023-07-08 jrick if (err)
3426 24df9a28 2023-07-08 jrick goto done;
3427 24df9a28 2023-07-08 jrick
3428 24df9a28 2023-07-08 jrick if (fseeko(f, 0L, SEEK_SET) == -1) {
3429 24df9a28 2023-07-08 jrick err = got_error_from_errno("fseeko");
3430 24df9a28 2023-07-08 jrick goto done;
3431 24df9a28 2023-07-08 jrick }
3432 24df9a28 2023-07-08 jrick
3433 24df9a28 2023-07-08 jrick while (getline(&line, &linesize, f) != -1) {
3434 24df9a28 2023-07-08 jrick if (regexec(regex, line, 1, &regmatch, 0) == 0) {
3435 24df9a28 2023-07-08 jrick *have_match = 1;
3436 24df9a28 2023-07-08 jrick break;
3437 24df9a28 2023-07-08 jrick }
3438 24df9a28 2023-07-08 jrick }
3439 24df9a28 2023-07-08 jrick done:
3440 24df9a28 2023-07-08 jrick free(line);
3441 24df9a28 2023-07-08 jrick return err;
3442 24df9a28 2023-07-08 jrick }
3443 24df9a28 2023-07-08 jrick
3444 24df9a28 2023-07-08 jrick #define GOT_COMMIT_SEP_STR "-----------------------------------------------\n"
3445 24df9a28 2023-07-08 jrick
3446 24df9a28 2023-07-08 jrick static const struct got_error*
3447 24df9a28 2023-07-08 jrick build_refs_str(char **refs_str, struct got_reflist_head *refs,
3448 24df9a28 2023-07-08 jrick struct got_object_id *id, struct got_repository *repo,
3449 24df9a28 2023-07-08 jrick int local_only)
3450 24df9a28 2023-07-08 jrick {
3451 24df9a28 2023-07-08 jrick static const struct got_error *err = NULL;
3452 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
3453 24df9a28 2023-07-08 jrick char *s;
3454 24df9a28 2023-07-08 jrick const char *name;
3455 24df9a28 2023-07-08 jrick
3456 24df9a28 2023-07-08 jrick *refs_str = NULL;
3457 24df9a28 2023-07-08 jrick
3458 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, refs, entry) {
3459 24df9a28 2023-07-08 jrick struct got_tag_object *tag = NULL;
3460 24df9a28 2023-07-08 jrick struct got_object_id *ref_id;
3461 24df9a28 2023-07-08 jrick int cmp;
3462 24df9a28 2023-07-08 jrick
3463 24df9a28 2023-07-08 jrick name = got_ref_get_name(re->ref);
3464 24df9a28 2023-07-08 jrick if (strcmp(name, GOT_REF_HEAD) == 0)
3465 24df9a28 2023-07-08 jrick continue;
3466 24df9a28 2023-07-08 jrick if (strncmp(name, "refs/", 5) == 0)
3467 24df9a28 2023-07-08 jrick name += 5;
3468 24df9a28 2023-07-08 jrick if (strncmp(name, "got/", 4) == 0)
3469 24df9a28 2023-07-08 jrick continue;
3470 24df9a28 2023-07-08 jrick if (strncmp(name, "heads/", 6) == 0)
3471 24df9a28 2023-07-08 jrick name += 6;
3472 24df9a28 2023-07-08 jrick if (strncmp(name, "remotes/", 8) == 0) {
3473 24df9a28 2023-07-08 jrick if (local_only)
3474 24df9a28 2023-07-08 jrick continue;
3475 24df9a28 2023-07-08 jrick name += 8;
3476 24df9a28 2023-07-08 jrick s = strstr(name, "/" GOT_REF_HEAD);
3477 24df9a28 2023-07-08 jrick if (s != NULL && strcmp(s, "/" GOT_REF_HEAD) == 0)
3478 24df9a28 2023-07-08 jrick continue;
3479 24df9a28 2023-07-08 jrick }
3480 24df9a28 2023-07-08 jrick err = got_ref_resolve(&ref_id, repo, re->ref);
3481 24df9a28 2023-07-08 jrick if (err)
3482 24df9a28 2023-07-08 jrick break;
3483 24df9a28 2023-07-08 jrick if (strncmp(name, "tags/", 5) == 0) {
3484 24df9a28 2023-07-08 jrick err = got_object_open_as_tag(&tag, repo, ref_id);
3485 24df9a28 2023-07-08 jrick if (err) {
3486 24df9a28 2023-07-08 jrick if (err->code != GOT_ERR_OBJ_TYPE) {
3487 24df9a28 2023-07-08 jrick free(ref_id);
3488 24df9a28 2023-07-08 jrick break;
3489 24df9a28 2023-07-08 jrick }
3490 24df9a28 2023-07-08 jrick /* Ref points at something other than a tag. */
3491 24df9a28 2023-07-08 jrick err = NULL;
3492 24df9a28 2023-07-08 jrick tag = NULL;
3493 24df9a28 2023-07-08 jrick }
3494 24df9a28 2023-07-08 jrick }
3495 24df9a28 2023-07-08 jrick cmp = got_object_id_cmp(tag ?
3496 24df9a28 2023-07-08 jrick got_object_tag_get_object_id(tag) : ref_id, id);
3497 24df9a28 2023-07-08 jrick free(ref_id);
3498 24df9a28 2023-07-08 jrick if (tag)
3499 24df9a28 2023-07-08 jrick got_object_tag_close(tag);
3500 24df9a28 2023-07-08 jrick if (cmp != 0)
3501 24df9a28 2023-07-08 jrick continue;
3502 24df9a28 2023-07-08 jrick s = *refs_str;
3503 24df9a28 2023-07-08 jrick if (asprintf(refs_str, "%s%s%s", s ? s : "",
3504 24df9a28 2023-07-08 jrick s ? ", " : "", name) == -1) {
3505 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
3506 24df9a28 2023-07-08 jrick free(s);
3507 24df9a28 2023-07-08 jrick *refs_str = NULL;
3508 24df9a28 2023-07-08 jrick break;
3509 24df9a28 2023-07-08 jrick }
3510 24df9a28 2023-07-08 jrick free(s);
3511 24df9a28 2023-07-08 jrick }
3512 24df9a28 2023-07-08 jrick
3513 24df9a28 2023-07-08 jrick return err;
3514 24df9a28 2023-07-08 jrick }
3515 24df9a28 2023-07-08 jrick
3516 24df9a28 2023-07-08 jrick static const struct got_error *
3517 24df9a28 2023-07-08 jrick print_commit_oneline(struct got_commit_object *commit, struct got_object_id *id,
3518 24df9a28 2023-07-08 jrick struct got_repository *repo, struct got_reflist_object_id_map *refs_idmap)
3519 24df9a28 2023-07-08 jrick {
3520 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3521 24df9a28 2023-07-08 jrick char *ref_str = NULL, *id_str = NULL, *logmsg0 = NULL;
3522 24df9a28 2023-07-08 jrick char *comma, *s, *nl;
3523 24df9a28 2023-07-08 jrick struct got_reflist_head *refs;
3524 24df9a28 2023-07-08 jrick char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
3525 24df9a28 2023-07-08 jrick struct tm tm;
3526 24df9a28 2023-07-08 jrick time_t committer_time;
3527 24df9a28 2023-07-08 jrick
3528 24df9a28 2023-07-08 jrick refs = got_reflist_object_id_map_lookup(refs_idmap, id);
3529 24df9a28 2023-07-08 jrick if (refs) {
3530 24df9a28 2023-07-08 jrick err = build_refs_str(&ref_str, refs, id, repo, 1);
3531 24df9a28 2023-07-08 jrick if (err)
3532 24df9a28 2023-07-08 jrick return err;
3533 24df9a28 2023-07-08 jrick
3534 24df9a28 2023-07-08 jrick /* Display the first matching ref only. */
3535 24df9a28 2023-07-08 jrick if (ref_str && (comma = strchr(ref_str, ',')) != NULL)
3536 24df9a28 2023-07-08 jrick *comma = '\0';
3537 24df9a28 2023-07-08 jrick }
3538 24df9a28 2023-07-08 jrick
3539 24df9a28 2023-07-08 jrick if (ref_str == NULL) {
3540 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
3541 24df9a28 2023-07-08 jrick if (err)
3542 24df9a28 2023-07-08 jrick return err;
3543 24df9a28 2023-07-08 jrick }
3544 24df9a28 2023-07-08 jrick
3545 24df9a28 2023-07-08 jrick committer_time = got_object_commit_get_committer_time(commit);
3546 24df9a28 2023-07-08 jrick if (gmtime_r(&committer_time, &tm) == NULL) {
3547 24df9a28 2023-07-08 jrick err = got_error_from_errno("gmtime_r");
3548 24df9a28 2023-07-08 jrick goto done;
3549 24df9a28 2023-07-08 jrick }
3550 24df9a28 2023-07-08 jrick if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm) == 0) {
3551 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_NO_SPACE);
3552 24df9a28 2023-07-08 jrick goto done;
3553 24df9a28 2023-07-08 jrick }
3554 24df9a28 2023-07-08 jrick
3555 24df9a28 2023-07-08 jrick err = got_object_commit_get_logmsg(&logmsg0, commit);
3556 24df9a28 2023-07-08 jrick if (err)
3557 24df9a28 2023-07-08 jrick goto done;
3558 24df9a28 2023-07-08 jrick
3559 24df9a28 2023-07-08 jrick s = logmsg0;
3560 24df9a28 2023-07-08 jrick while (isspace((unsigned char)s[0]))
3561 24df9a28 2023-07-08 jrick s++;
3562 24df9a28 2023-07-08 jrick
3563 24df9a28 2023-07-08 jrick nl = strchr(s, '\n');
3564 24df9a28 2023-07-08 jrick if (nl) {
3565 24df9a28 2023-07-08 jrick *nl = '\0';
3566 24df9a28 2023-07-08 jrick }
3567 24df9a28 2023-07-08 jrick
3568 24df9a28 2023-07-08 jrick if (ref_str)
3569 24df9a28 2023-07-08 jrick printf("%s%-7s %s\n", datebuf, ref_str, s);
3570 24df9a28 2023-07-08 jrick else
3571 24df9a28 2023-07-08 jrick printf("%s%.7s %s\n", datebuf, id_str, s);
3572 24df9a28 2023-07-08 jrick
3573 24df9a28 2023-07-08 jrick if (fflush(stdout) != 0 && err == NULL)
3574 24df9a28 2023-07-08 jrick err = got_error_from_errno("fflush");
3575 24df9a28 2023-07-08 jrick done:
3576 24df9a28 2023-07-08 jrick free(id_str);
3577 24df9a28 2023-07-08 jrick free(ref_str);
3578 24df9a28 2023-07-08 jrick free(logmsg0);
3579 24df9a28 2023-07-08 jrick return err;
3580 24df9a28 2023-07-08 jrick }
3581 24df9a28 2023-07-08 jrick
3582 24df9a28 2023-07-08 jrick static const struct got_error *
3583 24df9a28 2023-07-08 jrick print_diffstat(struct got_diffstat_cb_arg *dsa, const char *header)
3584 24df9a28 2023-07-08 jrick {
3585 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
3586 24df9a28 2023-07-08 jrick
3587 24df9a28 2023-07-08 jrick if (header != NULL)
3588 24df9a28 2023-07-08 jrick printf("%s\n", header);
3589 24df9a28 2023-07-08 jrick
3590 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, dsa->paths, entry) {
3591 24df9a28 2023-07-08 jrick struct got_diff_changed_path *cp = pe->data;
3592 24df9a28 2023-07-08 jrick int pad = dsa->max_path_len - pe->path_len + 1;
3593 24df9a28 2023-07-08 jrick
3594 24df9a28 2023-07-08 jrick printf(" %c %s%*c | %*d+ %*d-\n", cp->status, pe->path, pad,
3595 24df9a28 2023-07-08 jrick ' ', dsa->add_cols + 1, cp->add, dsa->rm_cols + 1, cp->rm);
3596 24df9a28 2023-07-08 jrick }
3597 24df9a28 2023-07-08 jrick printf("\n%d file%s changed, %d insertion%s(+), %d deletion%s(-)\n\n",
3598 24df9a28 2023-07-08 jrick dsa->nfiles, dsa->nfiles > 1 ? "s" : "", dsa->ins,
3599 24df9a28 2023-07-08 jrick dsa->ins != 1 ? "s" : "", dsa->del, dsa->del != 1 ? "s" : "");
3600 24df9a28 2023-07-08 jrick
3601 24df9a28 2023-07-08 jrick if (fflush(stdout) != 0)
3602 24df9a28 2023-07-08 jrick return got_error_from_errno("fflush");
3603 24df9a28 2023-07-08 jrick
3604 24df9a28 2023-07-08 jrick return NULL;
3605 24df9a28 2023-07-08 jrick }
3606 24df9a28 2023-07-08 jrick
3607 24df9a28 2023-07-08 jrick static const struct got_error *
3608 24df9a28 2023-07-08 jrick printfile(FILE *f)
3609 24df9a28 2023-07-08 jrick {
3610 24df9a28 2023-07-08 jrick char buf[8192];
3611 24df9a28 2023-07-08 jrick size_t r;
3612 24df9a28 2023-07-08 jrick
3613 24df9a28 2023-07-08 jrick if (fseeko(f, 0L, SEEK_SET) == -1)
3614 24df9a28 2023-07-08 jrick return got_error_from_errno("fseek");
3615 24df9a28 2023-07-08 jrick
3616 24df9a28 2023-07-08 jrick for (;;) {
3617 24df9a28 2023-07-08 jrick r = fread(buf, 1, sizeof(buf), f);
3618 24df9a28 2023-07-08 jrick if (r == 0) {
3619 24df9a28 2023-07-08 jrick if (ferror(f))
3620 24df9a28 2023-07-08 jrick return got_error_from_errno("fread");
3621 24df9a28 2023-07-08 jrick if (feof(f))
3622 24df9a28 2023-07-08 jrick break;
3623 24df9a28 2023-07-08 jrick }
3624 24df9a28 2023-07-08 jrick if (fwrite(buf, 1, r, stdout) != r)
3625 24df9a28 2023-07-08 jrick return got_ferror(stdout, GOT_ERR_IO);
3626 24df9a28 2023-07-08 jrick }
3627 24df9a28 2023-07-08 jrick
3628 24df9a28 2023-07-08 jrick return NULL;
3629 24df9a28 2023-07-08 jrick }
3630 24df9a28 2023-07-08 jrick
3631 24df9a28 2023-07-08 jrick static const struct got_error *
3632 24df9a28 2023-07-08 jrick print_commit(struct got_commit_object *commit, struct got_object_id *id,
3633 24df9a28 2023-07-08 jrick struct got_repository *repo, const char *path,
3634 24df9a28 2023-07-08 jrick struct got_pathlist_head *changed_paths,
3635 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg *diffstat, int show_patch, int diff_context,
3636 24df9a28 2023-07-08 jrick struct got_reflist_object_id_map *refs_idmap, const char *custom_refs_str,
3637 24df9a28 2023-07-08 jrick const char *prefix)
3638 24df9a28 2023-07-08 jrick {
3639 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
3640 24df9a28 2023-07-08 jrick FILE *f = NULL;
3641 24df9a28 2023-07-08 jrick char *id_str, *datestr, *logmsg0, *logmsg, *line;
3642 24df9a28 2023-07-08 jrick char datebuf[26];
3643 24df9a28 2023-07-08 jrick time_t committer_time;
3644 24df9a28 2023-07-08 jrick const char *author, *committer;
3645 24df9a28 2023-07-08 jrick char *refs_str = NULL;
3646 24df9a28 2023-07-08 jrick
3647 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
3648 24df9a28 2023-07-08 jrick if (err)
3649 24df9a28 2023-07-08 jrick return err;
3650 24df9a28 2023-07-08 jrick
3651 24df9a28 2023-07-08 jrick if (custom_refs_str == NULL) {
3652 24df9a28 2023-07-08 jrick struct got_reflist_head *refs;
3653 24df9a28 2023-07-08 jrick refs = got_reflist_object_id_map_lookup(refs_idmap, id);
3654 24df9a28 2023-07-08 jrick if (refs) {
3655 24df9a28 2023-07-08 jrick err = build_refs_str(&refs_str, refs, id, repo, 0);
3656 24df9a28 2023-07-08 jrick if (err)
3657 24df9a28 2023-07-08 jrick goto done;
3658 24df9a28 2023-07-08 jrick }
3659 24df9a28 2023-07-08 jrick }
3660 24df9a28 2023-07-08 jrick
3661 24df9a28 2023-07-08 jrick printf(GOT_COMMIT_SEP_STR);
3662 24df9a28 2023-07-08 jrick if (custom_refs_str)
3663 24df9a28 2023-07-08 jrick printf("%s %s (%s)\n", prefix ? prefix : "commit", id_str,
3664 24df9a28 2023-07-08 jrick custom_refs_str);
3665 24df9a28 2023-07-08 jrick else
3666 24df9a28 2023-07-08 jrick printf("%s %s%s%s%s\n", prefix ? prefix : "commit", id_str,
3667 24df9a28 2023-07-08 jrick refs_str ? " (" : "", refs_str ? refs_str : "",
3668 24df9a28 2023-07-08 jrick refs_str ? ")" : "");
3669 24df9a28 2023-07-08 jrick free(id_str);
3670 24df9a28 2023-07-08 jrick id_str = NULL;
3671 24df9a28 2023-07-08 jrick free(refs_str);
3672 24df9a28 2023-07-08 jrick refs_str = NULL;
3673 24df9a28 2023-07-08 jrick printf("from: %s\n", got_object_commit_get_author(commit));
3674 24df9a28 2023-07-08 jrick author = got_object_commit_get_author(commit);
3675 24df9a28 2023-07-08 jrick committer = got_object_commit_get_committer(commit);
3676 24df9a28 2023-07-08 jrick if (strcmp(author, committer) != 0)
3677 24df9a28 2023-07-08 jrick printf("via: %s\n", committer);
3678 24df9a28 2023-07-08 jrick committer_time = got_object_commit_get_committer_time(commit);
3679 24df9a28 2023-07-08 jrick datestr = get_datestr(&committer_time, datebuf);
3680 24df9a28 2023-07-08 jrick if (datestr)
3681 24df9a28 2023-07-08 jrick printf("date: %s UTC\n", datestr);
3682 24df9a28 2023-07-08 jrick if (got_object_commit_get_nparents(commit) > 1) {
3683 24df9a28 2023-07-08 jrick const struct got_object_id_queue *parent_ids;
3684 24df9a28 2023-07-08 jrick struct got_object_qid *qid;
3685 24df9a28 2023-07-08 jrick int n = 1;
3686 24df9a28 2023-07-08 jrick parent_ids = got_object_commit_get_parent_ids(commit);
3687 24df9a28 2023-07-08 jrick STAILQ_FOREACH(qid, parent_ids, entry) {
3688 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, &qid->id);
3689 24df9a28 2023-07-08 jrick if (err)
3690 24df9a28 2023-07-08 jrick goto done;
3691 24df9a28 2023-07-08 jrick printf("parent %d: %s\n", n++, id_str);
3692 24df9a28 2023-07-08 jrick free(id_str);
3693 24df9a28 2023-07-08 jrick id_str = NULL;
3694 24df9a28 2023-07-08 jrick }
3695 24df9a28 2023-07-08 jrick }
3696 24df9a28 2023-07-08 jrick
3697 24df9a28 2023-07-08 jrick err = got_object_commit_get_logmsg(&logmsg0, commit);
3698 24df9a28 2023-07-08 jrick if (err)
3699 24df9a28 2023-07-08 jrick goto done;
3700 24df9a28 2023-07-08 jrick
3701 24df9a28 2023-07-08 jrick logmsg = logmsg0;
3702 24df9a28 2023-07-08 jrick do {
3703 24df9a28 2023-07-08 jrick line = strsep(&logmsg, "\n");
3704 24df9a28 2023-07-08 jrick if (line)
3705 24df9a28 2023-07-08 jrick printf(" %s\n", line);
3706 24df9a28 2023-07-08 jrick } while (line);
3707 24df9a28 2023-07-08 jrick free(logmsg0);
3708 24df9a28 2023-07-08 jrick
3709 24df9a28 2023-07-08 jrick if (changed_paths && diffstat == NULL) {
3710 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
3711 24df9a28 2023-07-08 jrick
3712 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, changed_paths, entry) {
3713 24df9a28 2023-07-08 jrick struct got_diff_changed_path *cp = pe->data;
3714 24df9a28 2023-07-08 jrick
3715 24df9a28 2023-07-08 jrick printf(" %c %s\n", cp->status, pe->path);
3716 24df9a28 2023-07-08 jrick }
3717 24df9a28 2023-07-08 jrick printf("\n");
3718 24df9a28 2023-07-08 jrick }
3719 24df9a28 2023-07-08 jrick if (show_patch) {
3720 24df9a28 2023-07-08 jrick if (diffstat) {
3721 24df9a28 2023-07-08 jrick f = got_opentemp();
3722 24df9a28 2023-07-08 jrick if (f == NULL) {
3723 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentemp");
3724 24df9a28 2023-07-08 jrick goto done;
3725 24df9a28 2023-07-08 jrick }
3726 24df9a28 2023-07-08 jrick }
3727 24df9a28 2023-07-08 jrick
3728 24df9a28 2023-07-08 jrick err = print_patch(commit, id, path, diff_context, diffstat,
3729 24df9a28 2023-07-08 jrick repo, diffstat == NULL ? stdout : f);
3730 24df9a28 2023-07-08 jrick if (err)
3731 24df9a28 2023-07-08 jrick goto done;
3732 24df9a28 2023-07-08 jrick }
3733 24df9a28 2023-07-08 jrick if (diffstat) {
3734 24df9a28 2023-07-08 jrick err = print_diffstat(diffstat, NULL);
3735 24df9a28 2023-07-08 jrick if (err)
3736 24df9a28 2023-07-08 jrick goto done;
3737 24df9a28 2023-07-08 jrick if (show_patch) {
3738 24df9a28 2023-07-08 jrick err = printfile(f);
3739 24df9a28 2023-07-08 jrick if (err)
3740 24df9a28 2023-07-08 jrick goto done;
3741 24df9a28 2023-07-08 jrick }
3742 24df9a28 2023-07-08 jrick }
3743 24df9a28 2023-07-08 jrick if (show_patch)
3744 24df9a28 2023-07-08 jrick printf("\n");
3745 24df9a28 2023-07-08 jrick
3746 24df9a28 2023-07-08 jrick if (fflush(stdout) != 0 && err == NULL)
3747 24df9a28 2023-07-08 jrick err = got_error_from_errno("fflush");
3748 24df9a28 2023-07-08 jrick done:
3749 24df9a28 2023-07-08 jrick if (f && fclose(f) == EOF && err == NULL)
3750 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
3751 24df9a28 2023-07-08 jrick free(id_str);
3752 24df9a28 2023-07-08 jrick free(refs_str);
3753 24df9a28 2023-07-08 jrick return err;
3754 24df9a28 2023-07-08 jrick }
3755 24df9a28 2023-07-08 jrick
3756 24df9a28 2023-07-08 jrick static const struct got_error *
3757 24df9a28 2023-07-08 jrick print_commits(struct got_object_id *root_id, struct got_object_id *end_id,
3758 24df9a28 2023-07-08 jrick struct got_repository *repo, const char *path, int show_changed_paths,
3759 24df9a28 2023-07-08 jrick int show_diffstat, int show_patch, const char *search_pattern,
3760 24df9a28 2023-07-08 jrick int diff_context, int limit, int log_branches, int reverse_display_order,
3761 24df9a28 2023-07-08 jrick struct got_reflist_object_id_map *refs_idmap, int one_line,
3762 24df9a28 2023-07-08 jrick FILE *tmpfile)
3763 24df9a28 2023-07-08 jrick {
3764 24df9a28 2023-07-08 jrick const struct got_error *err;
3765 24df9a28 2023-07-08 jrick struct got_commit_graph *graph;
3766 24df9a28 2023-07-08 jrick regex_t regex;
3767 24df9a28 2023-07-08 jrick int have_match;
3768 24df9a28 2023-07-08 jrick struct got_object_id_queue reversed_commits;
3769 24df9a28 2023-07-08 jrick struct got_object_qid *qid;
3770 24df9a28 2023-07-08 jrick struct got_commit_object *commit;
3771 24df9a28 2023-07-08 jrick struct got_pathlist_head changed_paths;
3772 24df9a28 2023-07-08 jrick
3773 24df9a28 2023-07-08 jrick STAILQ_INIT(&reversed_commits);
3774 24df9a28 2023-07-08 jrick TAILQ_INIT(&changed_paths);
3775 24df9a28 2023-07-08 jrick
3776 24df9a28 2023-07-08 jrick if (search_pattern && regcomp(&regex, search_pattern,
3777 24df9a28 2023-07-08 jrick REG_EXTENDED | REG_NOSUB | REG_NEWLINE))
3778 24df9a28 2023-07-08 jrick return got_error_msg(GOT_ERR_REGEX, search_pattern);
3779 24df9a28 2023-07-08 jrick
3780 24df9a28 2023-07-08 jrick err = got_commit_graph_open(&graph, path, !log_branches);
3781 24df9a28 2023-07-08 jrick if (err)
3782 24df9a28 2023-07-08 jrick return err;
3783 24df9a28 2023-07-08 jrick err = got_commit_graph_iter_start(graph, root_id, repo,
3784 24df9a28 2023-07-08 jrick check_cancelled, NULL);
3785 24df9a28 2023-07-08 jrick if (err)
3786 24df9a28 2023-07-08 jrick goto done;
3787 24df9a28 2023-07-08 jrick for (;;) {
3788 24df9a28 2023-07-08 jrick struct got_object_id id;
3789 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg dsa = { 0, 0, 0, 0, 0, 0,
3790 24df9a28 2023-07-08 jrick &changed_paths, 0, 0, GOT_DIFF_ALGORITHM_PATIENCE };
3791 24df9a28 2023-07-08 jrick
3792 24df9a28 2023-07-08 jrick if (sigint_received || sigpipe_received)
3793 24df9a28 2023-07-08 jrick break;
3794 24df9a28 2023-07-08 jrick
3795 24df9a28 2023-07-08 jrick err = got_commit_graph_iter_next(&id, graph, repo,
3796 24df9a28 2023-07-08 jrick check_cancelled, NULL);
3797 24df9a28 2023-07-08 jrick if (err) {
3798 24df9a28 2023-07-08 jrick if (err->code == GOT_ERR_ITER_COMPLETED)
3799 24df9a28 2023-07-08 jrick err = NULL;
3800 24df9a28 2023-07-08 jrick break;
3801 24df9a28 2023-07-08 jrick }
3802 24df9a28 2023-07-08 jrick
3803 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, &id);
3804 24df9a28 2023-07-08 jrick if (err)
3805 24df9a28 2023-07-08 jrick break;
3806 24df9a28 2023-07-08 jrick
3807 24df9a28 2023-07-08 jrick if ((show_changed_paths || (show_diffstat && !show_patch))
3808 24df9a28 2023-07-08 jrick && !reverse_display_order) {
3809 24df9a28 2023-07-08 jrick err = get_changed_paths(&changed_paths, commit, repo,
3810 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL);
3811 24df9a28 2023-07-08 jrick if (err)
3812 24df9a28 2023-07-08 jrick break;
3813 24df9a28 2023-07-08 jrick }
3814 24df9a28 2023-07-08 jrick
3815 24df9a28 2023-07-08 jrick if (search_pattern) {
3816 24df9a28 2023-07-08 jrick err = match_commit(&have_match, &id, commit, &regex);
3817 24df9a28 2023-07-08 jrick if (err) {
3818 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
3819 24df9a28 2023-07-08 jrick break;
3820 24df9a28 2023-07-08 jrick }
3821 24df9a28 2023-07-08 jrick if (have_match == 0 && show_changed_paths)
3822 24df9a28 2023-07-08 jrick match_changed_paths(&have_match,
3823 24df9a28 2023-07-08 jrick &changed_paths, &regex);
3824 24df9a28 2023-07-08 jrick if (have_match == 0 && show_patch) {
3825 24df9a28 2023-07-08 jrick err = match_patch(&have_match, commit, &id,
3826 24df9a28 2023-07-08 jrick path, diff_context, repo, &regex, tmpfile);
3827 24df9a28 2023-07-08 jrick if (err)
3828 24df9a28 2023-07-08 jrick break;
3829 24df9a28 2023-07-08 jrick }
3830 24df9a28 2023-07-08 jrick if (have_match == 0) {
3831 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
3832 24df9a28 2023-07-08 jrick got_pathlist_free(&changed_paths,
3833 24df9a28 2023-07-08 jrick GOT_PATHLIST_FREE_ALL);
3834 24df9a28 2023-07-08 jrick continue;
3835 24df9a28 2023-07-08 jrick }
3836 24df9a28 2023-07-08 jrick }
3837 24df9a28 2023-07-08 jrick
3838 24df9a28 2023-07-08 jrick if (reverse_display_order) {
3839 24df9a28 2023-07-08 jrick err = got_object_qid_alloc(&qid, &id);
3840 24df9a28 2023-07-08 jrick if (err)
3841 24df9a28 2023-07-08 jrick break;
3842 24df9a28 2023-07-08 jrick STAILQ_INSERT_HEAD(&reversed_commits, qid, entry);
3843 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
3844 24df9a28 2023-07-08 jrick } else {
3845 24df9a28 2023-07-08 jrick if (one_line)
3846 24df9a28 2023-07-08 jrick err = print_commit_oneline(commit, &id,
3847 24df9a28 2023-07-08 jrick repo, refs_idmap);
3848 24df9a28 2023-07-08 jrick else
3849 24df9a28 2023-07-08 jrick err = print_commit(commit, &id, repo, path,
3850 24df9a28 2023-07-08 jrick (show_changed_paths || show_diffstat) ?
3851 24df9a28 2023-07-08 jrick &changed_paths : NULL,
3852 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL, show_patch,
3853 24df9a28 2023-07-08 jrick diff_context, refs_idmap, NULL, NULL);
3854 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
3855 24df9a28 2023-07-08 jrick if (err)
3856 24df9a28 2023-07-08 jrick break;
3857 24df9a28 2023-07-08 jrick }
3858 24df9a28 2023-07-08 jrick if ((limit && --limit == 0) ||
3859 24df9a28 2023-07-08 jrick (end_id && got_object_id_cmp(&id, end_id) == 0))
3860 24df9a28 2023-07-08 jrick break;
3861 24df9a28 2023-07-08 jrick
3862 24df9a28 2023-07-08 jrick got_pathlist_free(&changed_paths, GOT_PATHLIST_FREE_ALL);
3863 24df9a28 2023-07-08 jrick }
3864 24df9a28 2023-07-08 jrick if (reverse_display_order) {
3865 24df9a28 2023-07-08 jrick STAILQ_FOREACH(qid, &reversed_commits, entry) {
3866 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg dsa = { 0, 0, 0, 0, 0, 0,
3867 24df9a28 2023-07-08 jrick &changed_paths, 0, 0, GOT_DIFF_ALGORITHM_PATIENCE };
3868 24df9a28 2023-07-08 jrick
3869 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo,
3870 24df9a28 2023-07-08 jrick &qid->id);
3871 24df9a28 2023-07-08 jrick if (err)
3872 24df9a28 2023-07-08 jrick break;
3873 24df9a28 2023-07-08 jrick if (show_changed_paths ||
3874 24df9a28 2023-07-08 jrick (show_diffstat && !show_patch)) {
3875 24df9a28 2023-07-08 jrick err = get_changed_paths(&changed_paths, commit,
3876 24df9a28 2023-07-08 jrick repo, show_diffstat ? &dsa : NULL);
3877 24df9a28 2023-07-08 jrick if (err)
3878 24df9a28 2023-07-08 jrick break;
3879 24df9a28 2023-07-08 jrick }
3880 24df9a28 2023-07-08 jrick if (one_line)
3881 24df9a28 2023-07-08 jrick err = print_commit_oneline(commit, &qid->id,
3882 24df9a28 2023-07-08 jrick repo, refs_idmap);
3883 24df9a28 2023-07-08 jrick else
3884 24df9a28 2023-07-08 jrick err = print_commit(commit, &qid->id, repo, path,
3885 24df9a28 2023-07-08 jrick (show_changed_paths || show_diffstat) ?
3886 24df9a28 2023-07-08 jrick &changed_paths : NULL,
3887 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL, show_patch,
3888 24df9a28 2023-07-08 jrick diff_context, refs_idmap, NULL, NULL);
3889 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
3890 24df9a28 2023-07-08 jrick if (err)
3891 24df9a28 2023-07-08 jrick break;
3892 24df9a28 2023-07-08 jrick got_pathlist_free(&changed_paths, GOT_PATHLIST_FREE_ALL);
3893 24df9a28 2023-07-08 jrick }
3894 24df9a28 2023-07-08 jrick }
3895 24df9a28 2023-07-08 jrick done:
3896 24df9a28 2023-07-08 jrick while (!STAILQ_EMPTY(&reversed_commits)) {
3897 24df9a28 2023-07-08 jrick qid = STAILQ_FIRST(&reversed_commits);
3898 24df9a28 2023-07-08 jrick STAILQ_REMOVE_HEAD(&reversed_commits, entry);
3899 24df9a28 2023-07-08 jrick got_object_qid_free(qid);
3900 24df9a28 2023-07-08 jrick }
3901 24df9a28 2023-07-08 jrick got_pathlist_free(&changed_paths, GOT_PATHLIST_FREE_ALL);
3902 24df9a28 2023-07-08 jrick if (search_pattern)
3903 24df9a28 2023-07-08 jrick regfree(&regex);
3904 24df9a28 2023-07-08 jrick got_commit_graph_close(graph);
3905 24df9a28 2023-07-08 jrick return err;
3906 24df9a28 2023-07-08 jrick }
3907 24df9a28 2023-07-08 jrick
3908 24df9a28 2023-07-08 jrick __dead static void
3909 24df9a28 2023-07-08 jrick usage_log(void)
3910 24df9a28 2023-07-08 jrick {
3911 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s log [-bdPpRs] [-C number] [-c commit] "
3912 24df9a28 2023-07-08 jrick "[-l N] [-r repository-path] [-S search-pattern] [-x commit] "
3913 24df9a28 2023-07-08 jrick "[path]\n", getprogname());
3914 24df9a28 2023-07-08 jrick exit(1);
3915 24df9a28 2023-07-08 jrick }
3916 24df9a28 2023-07-08 jrick
3917 24df9a28 2023-07-08 jrick static int
3918 24df9a28 2023-07-08 jrick get_default_log_limit(void)
3919 24df9a28 2023-07-08 jrick {
3920 24df9a28 2023-07-08 jrick const char *got_default_log_limit;
3921 24df9a28 2023-07-08 jrick long long n;
3922 24df9a28 2023-07-08 jrick const char *errstr;
3923 24df9a28 2023-07-08 jrick
3924 24df9a28 2023-07-08 jrick got_default_log_limit = getenv("GOT_LOG_DEFAULT_LIMIT");
3925 24df9a28 2023-07-08 jrick if (got_default_log_limit == NULL)
3926 24df9a28 2023-07-08 jrick return 0;
3927 24df9a28 2023-07-08 jrick n = strtonum(got_default_log_limit, 0, INT_MAX, &errstr);
3928 24df9a28 2023-07-08 jrick if (errstr != NULL)
3929 24df9a28 2023-07-08 jrick return 0;
3930 24df9a28 2023-07-08 jrick return n;
3931 24df9a28 2023-07-08 jrick }
3932 24df9a28 2023-07-08 jrick
3933 24df9a28 2023-07-08 jrick static const struct got_error *
3934 24df9a28 2023-07-08 jrick cmd_log(int argc, char *argv[])
3935 24df9a28 2023-07-08 jrick {
3936 24df9a28 2023-07-08 jrick const struct got_error *error;
3937 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
3938 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
3939 24df9a28 2023-07-08 jrick struct got_object_id *start_id = NULL, *end_id = NULL;
3940 24df9a28 2023-07-08 jrick char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
3941 24df9a28 2023-07-08 jrick const char *start_commit = NULL, *end_commit = NULL;
3942 24df9a28 2023-07-08 jrick const char *search_pattern = NULL;
3943 24df9a28 2023-07-08 jrick int diff_context = -1, ch;
3944 24df9a28 2023-07-08 jrick int show_changed_paths = 0, show_patch = 0, limit = 0, log_branches = 0;
3945 24df9a28 2023-07-08 jrick int show_diffstat = 0, reverse_display_order = 0, one_line = 0;
3946 24df9a28 2023-07-08 jrick const char *errstr;
3947 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
3948 24df9a28 2023-07-08 jrick struct got_reflist_object_id_map *refs_idmap = NULL;
3949 24df9a28 2023-07-08 jrick FILE *tmpfile = NULL;
3950 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
3951 24df9a28 2023-07-08 jrick
3952 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
3953 24df9a28 2023-07-08 jrick
3954 24df9a28 2023-07-08 jrick #ifndef PROFILE
3955 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
3956 24df9a28 2023-07-08 jrick NULL)
3957 24df9a28 2023-07-08 jrick == -1)
3958 24df9a28 2023-07-08 jrick err(1, "pledge");
3959 24df9a28 2023-07-08 jrick #endif
3960 24df9a28 2023-07-08 jrick
3961 24df9a28 2023-07-08 jrick limit = get_default_log_limit();
3962 24df9a28 2023-07-08 jrick
3963 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "bC:c:dl:PpRr:S:sx:")) != -1) {
3964 24df9a28 2023-07-08 jrick switch (ch) {
3965 24df9a28 2023-07-08 jrick case 'b':
3966 24df9a28 2023-07-08 jrick log_branches = 1;
3967 24df9a28 2023-07-08 jrick break;
3968 24df9a28 2023-07-08 jrick case 'C':
3969 24df9a28 2023-07-08 jrick diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
3970 24df9a28 2023-07-08 jrick &errstr);
3971 24df9a28 2023-07-08 jrick if (errstr != NULL)
3972 24df9a28 2023-07-08 jrick errx(1, "number of context lines is %s: %s",
3973 24df9a28 2023-07-08 jrick errstr, optarg);
3974 24df9a28 2023-07-08 jrick break;
3975 24df9a28 2023-07-08 jrick case 'c':
3976 24df9a28 2023-07-08 jrick start_commit = optarg;
3977 24df9a28 2023-07-08 jrick break;
3978 24df9a28 2023-07-08 jrick case 'd':
3979 24df9a28 2023-07-08 jrick show_diffstat = 1;
3980 24df9a28 2023-07-08 jrick break;
3981 24df9a28 2023-07-08 jrick case 'l':
3982 24df9a28 2023-07-08 jrick limit = strtonum(optarg, 0, INT_MAX, &errstr);
3983 24df9a28 2023-07-08 jrick if (errstr != NULL)
3984 24df9a28 2023-07-08 jrick errx(1, "number of commits is %s: %s",
3985 24df9a28 2023-07-08 jrick errstr, optarg);
3986 24df9a28 2023-07-08 jrick break;
3987 24df9a28 2023-07-08 jrick case 'P':
3988 24df9a28 2023-07-08 jrick show_changed_paths = 1;
3989 24df9a28 2023-07-08 jrick break;
3990 24df9a28 2023-07-08 jrick case 'p':
3991 24df9a28 2023-07-08 jrick show_patch = 1;
3992 24df9a28 2023-07-08 jrick break;
3993 24df9a28 2023-07-08 jrick case 'R':
3994 24df9a28 2023-07-08 jrick reverse_display_order = 1;
3995 24df9a28 2023-07-08 jrick break;
3996 24df9a28 2023-07-08 jrick case 'r':
3997 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
3998 24df9a28 2023-07-08 jrick if (repo_path == NULL)
3999 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
4000 24df9a28 2023-07-08 jrick optarg);
4001 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
4002 24df9a28 2023-07-08 jrick break;
4003 24df9a28 2023-07-08 jrick case 'S':
4004 24df9a28 2023-07-08 jrick search_pattern = optarg;
4005 24df9a28 2023-07-08 jrick break;
4006 24df9a28 2023-07-08 jrick case 's':
4007 24df9a28 2023-07-08 jrick one_line = 1;
4008 24df9a28 2023-07-08 jrick break;
4009 24df9a28 2023-07-08 jrick case 'x':
4010 24df9a28 2023-07-08 jrick end_commit = optarg;
4011 24df9a28 2023-07-08 jrick break;
4012 24df9a28 2023-07-08 jrick default:
4013 24df9a28 2023-07-08 jrick usage_log();
4014 24df9a28 2023-07-08 jrick /* NOTREACHED */
4015 24df9a28 2023-07-08 jrick }
4016 24df9a28 2023-07-08 jrick }
4017 24df9a28 2023-07-08 jrick
4018 24df9a28 2023-07-08 jrick argc -= optind;
4019 24df9a28 2023-07-08 jrick argv += optind;
4020 24df9a28 2023-07-08 jrick
4021 24df9a28 2023-07-08 jrick if (diff_context == -1)
4022 24df9a28 2023-07-08 jrick diff_context = 3;
4023 24df9a28 2023-07-08 jrick else if (!show_patch)
4024 24df9a28 2023-07-08 jrick errx(1, "-C requires -p");
4025 24df9a28 2023-07-08 jrick
4026 24df9a28 2023-07-08 jrick if (one_line && (show_patch || show_changed_paths || show_diffstat))
4027 24df9a28 2023-07-08 jrick errx(1, "cannot use -s with -d, -p or -P");
4028 24df9a28 2023-07-08 jrick
4029 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
4030 24df9a28 2023-07-08 jrick if (cwd == NULL) {
4031 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
4032 24df9a28 2023-07-08 jrick goto done;
4033 24df9a28 2023-07-08 jrick }
4034 24df9a28 2023-07-08 jrick
4035 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
4036 24df9a28 2023-07-08 jrick if (error != NULL)
4037 24df9a28 2023-07-08 jrick goto done;
4038 24df9a28 2023-07-08 jrick
4039 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4040 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
4041 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
4042 24df9a28 2023-07-08 jrick goto done;
4043 24df9a28 2023-07-08 jrick error = NULL;
4044 24df9a28 2023-07-08 jrick }
4045 24df9a28 2023-07-08 jrick
4046 24df9a28 2023-07-08 jrick if (argc == 1) {
4047 24df9a28 2023-07-08 jrick if (worktree) {
4048 24df9a28 2023-07-08 jrick error = got_worktree_resolve_path(&path, worktree,
4049 24df9a28 2023-07-08 jrick argv[0]);
4050 24df9a28 2023-07-08 jrick if (error)
4051 24df9a28 2023-07-08 jrick goto done;
4052 24df9a28 2023-07-08 jrick } else {
4053 24df9a28 2023-07-08 jrick path = strdup(argv[0]);
4054 24df9a28 2023-07-08 jrick if (path == NULL) {
4055 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
4056 24df9a28 2023-07-08 jrick goto done;
4057 24df9a28 2023-07-08 jrick }
4058 24df9a28 2023-07-08 jrick }
4059 24df9a28 2023-07-08 jrick } else if (argc != 0)
4060 24df9a28 2023-07-08 jrick usage_log();
4061 24df9a28 2023-07-08 jrick
4062 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4063 24df9a28 2023-07-08 jrick repo_path = worktree ?
4064 24df9a28 2023-07-08 jrick strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
4065 24df9a28 2023-07-08 jrick }
4066 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4067 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
4068 24df9a28 2023-07-08 jrick goto done;
4069 24df9a28 2023-07-08 jrick }
4070 24df9a28 2023-07-08 jrick
4071 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4072 24df9a28 2023-07-08 jrick if (error != NULL)
4073 24df9a28 2023-07-08 jrick goto done;
4074 24df9a28 2023-07-08 jrick
4075 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1,
4076 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_root_path(worktree) : NULL);
4077 24df9a28 2023-07-08 jrick if (error)
4078 24df9a28 2023-07-08 jrick goto done;
4079 24df9a28 2023-07-08 jrick
4080 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
4081 24df9a28 2023-07-08 jrick if (error)
4082 24df9a28 2023-07-08 jrick goto done;
4083 24df9a28 2023-07-08 jrick
4084 24df9a28 2023-07-08 jrick error = got_reflist_object_id_map_create(&refs_idmap, &refs, repo);
4085 24df9a28 2023-07-08 jrick if (error)
4086 24df9a28 2023-07-08 jrick goto done;
4087 24df9a28 2023-07-08 jrick
4088 24df9a28 2023-07-08 jrick if (start_commit == NULL) {
4089 24df9a28 2023-07-08 jrick struct got_reference *head_ref;
4090 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
4091 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo,
4092 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_head_ref_name(worktree)
4093 24df9a28 2023-07-08 jrick : GOT_REF_HEAD, 0);
4094 24df9a28 2023-07-08 jrick if (error != NULL)
4095 24df9a28 2023-07-08 jrick goto done;
4096 24df9a28 2023-07-08 jrick error = got_ref_resolve(&start_id, repo, head_ref);
4097 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
4098 24df9a28 2023-07-08 jrick if (error != NULL)
4099 24df9a28 2023-07-08 jrick goto done;
4100 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo,
4101 24df9a28 2023-07-08 jrick start_id);
4102 24df9a28 2023-07-08 jrick if (error != NULL)
4103 24df9a28 2023-07-08 jrick goto done;
4104 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
4105 24df9a28 2023-07-08 jrick } else {
4106 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&start_id, NULL,
4107 24df9a28 2023-07-08 jrick start_commit, GOT_OBJ_TYPE_COMMIT, &refs, repo);
4108 24df9a28 2023-07-08 jrick if (error != NULL)
4109 24df9a28 2023-07-08 jrick goto done;
4110 24df9a28 2023-07-08 jrick }
4111 24df9a28 2023-07-08 jrick if (end_commit != NULL) {
4112 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&end_id, NULL,
4113 24df9a28 2023-07-08 jrick end_commit, GOT_OBJ_TYPE_COMMIT, &refs, repo);
4114 24df9a28 2023-07-08 jrick if (error != NULL)
4115 24df9a28 2023-07-08 jrick goto done;
4116 24df9a28 2023-07-08 jrick }
4117 24df9a28 2023-07-08 jrick
4118 24df9a28 2023-07-08 jrick if (worktree) {
4119 24df9a28 2023-07-08 jrick /*
4120 24df9a28 2023-07-08 jrick * If a path was specified on the command line it was resolved
4121 24df9a28 2023-07-08 jrick * to a path in the work tree above. Prepend the work tree's
4122 24df9a28 2023-07-08 jrick * path prefix to obtain the corresponding in-repository path.
4123 24df9a28 2023-07-08 jrick */
4124 24df9a28 2023-07-08 jrick if (path) {
4125 24df9a28 2023-07-08 jrick const char *prefix;
4126 24df9a28 2023-07-08 jrick prefix = got_worktree_get_path_prefix(worktree);
4127 24df9a28 2023-07-08 jrick if (asprintf(&in_repo_path, "%s%s%s", prefix,
4128 24df9a28 2023-07-08 jrick (path[0] != '\0') ? "/" : "", path) == -1) {
4129 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
4130 24df9a28 2023-07-08 jrick goto done;
4131 24df9a28 2023-07-08 jrick }
4132 24df9a28 2023-07-08 jrick }
4133 24df9a28 2023-07-08 jrick } else
4134 24df9a28 2023-07-08 jrick error = got_repo_map_path(&in_repo_path, repo,
4135 24df9a28 2023-07-08 jrick path ? path : "");
4136 24df9a28 2023-07-08 jrick if (error != NULL)
4137 24df9a28 2023-07-08 jrick goto done;
4138 24df9a28 2023-07-08 jrick if (in_repo_path) {
4139 24df9a28 2023-07-08 jrick free(path);
4140 24df9a28 2023-07-08 jrick path = in_repo_path;
4141 24df9a28 2023-07-08 jrick }
4142 24df9a28 2023-07-08 jrick
4143 24df9a28 2023-07-08 jrick if (worktree) {
4144 24df9a28 2023-07-08 jrick /* Release work tree lock. */
4145 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
4146 24df9a28 2023-07-08 jrick worktree = NULL;
4147 24df9a28 2023-07-08 jrick }
4148 24df9a28 2023-07-08 jrick
4149 24df9a28 2023-07-08 jrick if (search_pattern && show_patch) {
4150 24df9a28 2023-07-08 jrick tmpfile = got_opentemp();
4151 24df9a28 2023-07-08 jrick if (tmpfile == NULL) {
4152 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
4153 24df9a28 2023-07-08 jrick goto done;
4154 24df9a28 2023-07-08 jrick }
4155 24df9a28 2023-07-08 jrick }
4156 24df9a28 2023-07-08 jrick
4157 24df9a28 2023-07-08 jrick error = print_commits(start_id, end_id, repo, path ? path : "",
4158 24df9a28 2023-07-08 jrick show_changed_paths, show_diffstat, show_patch, search_pattern,
4159 24df9a28 2023-07-08 jrick diff_context, limit, log_branches, reverse_display_order,
4160 24df9a28 2023-07-08 jrick refs_idmap, one_line, tmpfile);
4161 24df9a28 2023-07-08 jrick done:
4162 24df9a28 2023-07-08 jrick free(path);
4163 24df9a28 2023-07-08 jrick free(repo_path);
4164 24df9a28 2023-07-08 jrick free(cwd);
4165 8a00d20d 2023-07-12 jrick free(start_id);
4166 8a00d20d 2023-07-12 jrick free(end_id);
4167 24df9a28 2023-07-08 jrick if (worktree)
4168 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
4169 24df9a28 2023-07-08 jrick if (repo) {
4170 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
4171 24df9a28 2023-07-08 jrick if (error == NULL)
4172 24df9a28 2023-07-08 jrick error = close_err;
4173 24df9a28 2023-07-08 jrick }
4174 24df9a28 2023-07-08 jrick if (pack_fds) {
4175 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
4176 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
4177 24df9a28 2023-07-08 jrick if (error == NULL)
4178 24df9a28 2023-07-08 jrick error = pack_err;
4179 24df9a28 2023-07-08 jrick }
4180 24df9a28 2023-07-08 jrick if (refs_idmap)
4181 24df9a28 2023-07-08 jrick got_reflist_object_id_map_free(refs_idmap);
4182 24df9a28 2023-07-08 jrick if (tmpfile && fclose(tmpfile) == EOF && error == NULL)
4183 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
4184 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
4185 24df9a28 2023-07-08 jrick return error;
4186 24df9a28 2023-07-08 jrick }
4187 24df9a28 2023-07-08 jrick
4188 24df9a28 2023-07-08 jrick __dead static void
4189 24df9a28 2023-07-08 jrick usage_diff(void)
4190 24df9a28 2023-07-08 jrick {
4191 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s diff [-adPsw] [-C number] [-c commit] "
4192 24df9a28 2023-07-08 jrick "[-r repository-path] [object1 object2 | path ...]\n",
4193 24df9a28 2023-07-08 jrick getprogname());
4194 24df9a28 2023-07-08 jrick exit(1);
4195 24df9a28 2023-07-08 jrick }
4196 24df9a28 2023-07-08 jrick
4197 24df9a28 2023-07-08 jrick struct print_diff_arg {
4198 24df9a28 2023-07-08 jrick struct got_repository *repo;
4199 24df9a28 2023-07-08 jrick struct got_worktree *worktree;
4200 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg *diffstat;
4201 24df9a28 2023-07-08 jrick int diff_context;
4202 24df9a28 2023-07-08 jrick const char *id_str;
4203 24df9a28 2023-07-08 jrick int header_shown;
4204 24df9a28 2023-07-08 jrick int diff_staged;
4205 24df9a28 2023-07-08 jrick enum got_diff_algorithm diff_algo;
4206 24df9a28 2023-07-08 jrick int ignore_whitespace;
4207 24df9a28 2023-07-08 jrick int force_text_diff;
4208 24df9a28 2023-07-08 jrick FILE *f1;
4209 24df9a28 2023-07-08 jrick FILE *f2;
4210 24df9a28 2023-07-08 jrick FILE *outfile;
4211 24df9a28 2023-07-08 jrick };
4212 24df9a28 2023-07-08 jrick
4213 24df9a28 2023-07-08 jrick /*
4214 24df9a28 2023-07-08 jrick * Create a file which contains the target path of a symlink so we can feed
4215 24df9a28 2023-07-08 jrick * it as content to the diff engine.
4216 24df9a28 2023-07-08 jrick */
4217 24df9a28 2023-07-08 jrick static const struct got_error *
4218 24df9a28 2023-07-08 jrick get_symlink_target_file(int *fd, int dirfd, const char *de_name,
4219 24df9a28 2023-07-08 jrick const char *abspath)
4220 24df9a28 2023-07-08 jrick {
4221 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
4222 24df9a28 2023-07-08 jrick char target_path[PATH_MAX];
4223 24df9a28 2023-07-08 jrick ssize_t target_len, outlen;
4224 24df9a28 2023-07-08 jrick
4225 24df9a28 2023-07-08 jrick *fd = -1;
4226 24df9a28 2023-07-08 jrick
4227 24df9a28 2023-07-08 jrick if (dirfd != -1) {
4228 24df9a28 2023-07-08 jrick target_len = readlinkat(dirfd, de_name, target_path, PATH_MAX);
4229 24df9a28 2023-07-08 jrick if (target_len == -1)
4230 24df9a28 2023-07-08 jrick return got_error_from_errno2("readlinkat", abspath);
4231 24df9a28 2023-07-08 jrick } else {
4232 24df9a28 2023-07-08 jrick target_len = readlink(abspath, target_path, PATH_MAX);
4233 24df9a28 2023-07-08 jrick if (target_len == -1)
4234 24df9a28 2023-07-08 jrick return got_error_from_errno2("readlink", abspath);
4235 24df9a28 2023-07-08 jrick }
4236 24df9a28 2023-07-08 jrick
4237 24df9a28 2023-07-08 jrick *fd = got_opentempfd();
4238 24df9a28 2023-07-08 jrick if (*fd == -1)
4239 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentempfd");
4240 24df9a28 2023-07-08 jrick
4241 24df9a28 2023-07-08 jrick outlen = write(*fd, target_path, target_len);
4242 24df9a28 2023-07-08 jrick if (outlen == -1) {
4243 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
4244 24df9a28 2023-07-08 jrick goto done;
4245 24df9a28 2023-07-08 jrick }
4246 24df9a28 2023-07-08 jrick
4247 24df9a28 2023-07-08 jrick if (lseek(*fd, 0, SEEK_SET) == -1) {
4248 24df9a28 2023-07-08 jrick err = got_error_from_errno2("lseek", abspath);
4249 24df9a28 2023-07-08 jrick goto done;
4250 24df9a28 2023-07-08 jrick }
4251 24df9a28 2023-07-08 jrick done:
4252 24df9a28 2023-07-08 jrick if (err) {
4253 24df9a28 2023-07-08 jrick close(*fd);
4254 24df9a28 2023-07-08 jrick *fd = -1;
4255 24df9a28 2023-07-08 jrick }
4256 24df9a28 2023-07-08 jrick return err;
4257 24df9a28 2023-07-08 jrick }
4258 24df9a28 2023-07-08 jrick
4259 24df9a28 2023-07-08 jrick static const struct got_error *
4260 24df9a28 2023-07-08 jrick print_diff(void *arg, unsigned char status, unsigned char staged_status,
4261 24df9a28 2023-07-08 jrick const char *path, struct got_object_id *blob_id,
4262 24df9a28 2023-07-08 jrick struct got_object_id *staged_blob_id, struct got_object_id *commit_id,
4263 24df9a28 2023-07-08 jrick int dirfd, const char *de_name)
4264 24df9a28 2023-07-08 jrick {
4265 24df9a28 2023-07-08 jrick struct print_diff_arg *a = arg;
4266 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
4267 24df9a28 2023-07-08 jrick struct got_blob_object *blob1 = NULL;
4268 24df9a28 2023-07-08 jrick int fd = -1, fd1 = -1, fd2 = -1;
4269 24df9a28 2023-07-08 jrick FILE *f2 = NULL;
4270 24df9a28 2023-07-08 jrick char *abspath = NULL, *label1 = NULL;
4271 24df9a28 2023-07-08 jrick struct stat sb;
4272 24df9a28 2023-07-08 jrick off_t size1 = 0;
4273 24df9a28 2023-07-08 jrick int f2_exists = 0;
4274 24df9a28 2023-07-08 jrick
4275 24df9a28 2023-07-08 jrick memset(&sb, 0, sizeof(sb));
4276 24df9a28 2023-07-08 jrick
4277 24df9a28 2023-07-08 jrick if (a->diff_staged) {
4278 24df9a28 2023-07-08 jrick if (staged_status != GOT_STATUS_MODIFY &&
4279 24df9a28 2023-07-08 jrick staged_status != GOT_STATUS_ADD &&
4280 24df9a28 2023-07-08 jrick staged_status != GOT_STATUS_DELETE)
4281 24df9a28 2023-07-08 jrick return NULL;
4282 24df9a28 2023-07-08 jrick } else {
4283 24df9a28 2023-07-08 jrick if (staged_status == GOT_STATUS_DELETE)
4284 24df9a28 2023-07-08 jrick return NULL;
4285 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_NONEXISTENT)
4286 24df9a28 2023-07-08 jrick return got_error_set_errno(ENOENT, path);
4287 24df9a28 2023-07-08 jrick if (status != GOT_STATUS_MODIFY &&
4288 24df9a28 2023-07-08 jrick status != GOT_STATUS_ADD &&
4289 24df9a28 2023-07-08 jrick status != GOT_STATUS_DELETE &&
4290 24df9a28 2023-07-08 jrick status != GOT_STATUS_CONFLICT)
4291 24df9a28 2023-07-08 jrick return NULL;
4292 24df9a28 2023-07-08 jrick }
4293 24df9a28 2023-07-08 jrick
4294 24df9a28 2023-07-08 jrick err = got_opentemp_truncate(a->f1);
4295 24df9a28 2023-07-08 jrick if (err)
4296 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentemp_truncate");
4297 24df9a28 2023-07-08 jrick err = got_opentemp_truncate(a->f2);
4298 24df9a28 2023-07-08 jrick if (err)
4299 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentemp_truncate");
4300 24df9a28 2023-07-08 jrick
4301 24df9a28 2023-07-08 jrick if (!a->header_shown) {
4302 24df9a28 2023-07-08 jrick if (fprintf(a->outfile, "diff %s%s\n",
4303 24df9a28 2023-07-08 jrick a->diff_staged ? "-s " : "",
4304 24df9a28 2023-07-08 jrick got_worktree_get_root_path(a->worktree)) < 0) {
4305 24df9a28 2023-07-08 jrick err = got_error_from_errno("fprintf");
4306 24df9a28 2023-07-08 jrick goto done;
4307 24df9a28 2023-07-08 jrick }
4308 24df9a28 2023-07-08 jrick if (fprintf(a->outfile, "commit - %s\n", a->id_str) < 0) {
4309 24df9a28 2023-07-08 jrick err = got_error_from_errno("fprintf");
4310 24df9a28 2023-07-08 jrick goto done;
4311 24df9a28 2023-07-08 jrick }
4312 24df9a28 2023-07-08 jrick if (fprintf(a->outfile, "path + %s%s\n",
4313 24df9a28 2023-07-08 jrick got_worktree_get_root_path(a->worktree),
4314 24df9a28 2023-07-08 jrick a->diff_staged ? " (staged changes)" : "") < 0) {
4315 24df9a28 2023-07-08 jrick err = got_error_from_errno("fprintf");
4316 24df9a28 2023-07-08 jrick goto done;
4317 24df9a28 2023-07-08 jrick }
4318 24df9a28 2023-07-08 jrick a->header_shown = 1;
4319 24df9a28 2023-07-08 jrick }
4320 24df9a28 2023-07-08 jrick
4321 24df9a28 2023-07-08 jrick if (a->diff_staged) {
4322 24df9a28 2023-07-08 jrick const char *label1 = NULL, *label2 = NULL;
4323 24df9a28 2023-07-08 jrick switch (staged_status) {
4324 24df9a28 2023-07-08 jrick case GOT_STATUS_MODIFY:
4325 24df9a28 2023-07-08 jrick label1 = path;
4326 24df9a28 2023-07-08 jrick label2 = path;
4327 24df9a28 2023-07-08 jrick break;
4328 24df9a28 2023-07-08 jrick case GOT_STATUS_ADD:
4329 24df9a28 2023-07-08 jrick label2 = path;
4330 24df9a28 2023-07-08 jrick break;
4331 24df9a28 2023-07-08 jrick case GOT_STATUS_DELETE:
4332 24df9a28 2023-07-08 jrick label1 = path;
4333 24df9a28 2023-07-08 jrick break;
4334 24df9a28 2023-07-08 jrick default:
4335 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_FILE_STATUS);
4336 24df9a28 2023-07-08 jrick }
4337 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
4338 24df9a28 2023-07-08 jrick if (fd1 == -1) {
4339 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
4340 24df9a28 2023-07-08 jrick goto done;
4341 24df9a28 2023-07-08 jrick }
4342 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
4343 24df9a28 2023-07-08 jrick if (fd2 == -1) {
4344 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
4345 24df9a28 2023-07-08 jrick goto done;
4346 24df9a28 2023-07-08 jrick }
4347 24df9a28 2023-07-08 jrick err = got_diff_objects_as_blobs(NULL, NULL, a->f1, a->f2,
4348 24df9a28 2023-07-08 jrick fd1, fd2, blob_id, staged_blob_id, label1, label2,
4349 24df9a28 2023-07-08 jrick a->diff_algo, a->diff_context, a->ignore_whitespace,
4350 24df9a28 2023-07-08 jrick a->force_text_diff, a->diffstat, a->repo, a->outfile);
4351 24df9a28 2023-07-08 jrick goto done;
4352 24df9a28 2023-07-08 jrick }
4353 24df9a28 2023-07-08 jrick
4354 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
4355 24df9a28 2023-07-08 jrick if (fd1 == -1) {
4356 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_opentempfd");
4357 24df9a28 2023-07-08 jrick goto done;
4358 24df9a28 2023-07-08 jrick }
4359 24df9a28 2023-07-08 jrick
4360 24df9a28 2023-07-08 jrick if (staged_status == GOT_STATUS_ADD ||
4361 24df9a28 2023-07-08 jrick staged_status == GOT_STATUS_MODIFY) {
4362 24df9a28 2023-07-08 jrick char *id_str;
4363 24df9a28 2023-07-08 jrick err = got_object_open_as_blob(&blob1, a->repo, staged_blob_id,
4364 24df9a28 2023-07-08 jrick 8192, fd1);
4365 24df9a28 2023-07-08 jrick if (err)
4366 24df9a28 2023-07-08 jrick goto done;
4367 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, staged_blob_id);
4368 24df9a28 2023-07-08 jrick if (err)
4369 24df9a28 2023-07-08 jrick goto done;
4370 24df9a28 2023-07-08 jrick if (asprintf(&label1, "%s (staged)", id_str) == -1) {
4371 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
4372 24df9a28 2023-07-08 jrick free(id_str);
4373 24df9a28 2023-07-08 jrick goto done;
4374 24df9a28 2023-07-08 jrick }
4375 24df9a28 2023-07-08 jrick free(id_str);
4376 24df9a28 2023-07-08 jrick } else if (status != GOT_STATUS_ADD) {
4377 24df9a28 2023-07-08 jrick err = got_object_open_as_blob(&blob1, a->repo, blob_id, 8192,
4378 24df9a28 2023-07-08 jrick fd1);
4379 24df9a28 2023-07-08 jrick if (err)
4380 24df9a28 2023-07-08 jrick goto done;
4381 24df9a28 2023-07-08 jrick }
4382 24df9a28 2023-07-08 jrick
4383 24df9a28 2023-07-08 jrick if (status != GOT_STATUS_DELETE) {
4384 24df9a28 2023-07-08 jrick if (asprintf(&abspath, "%s/%s",
4385 24df9a28 2023-07-08 jrick got_worktree_get_root_path(a->worktree), path) == -1) {
4386 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
4387 24df9a28 2023-07-08 jrick goto done;
4388 24df9a28 2023-07-08 jrick }
4389 24df9a28 2023-07-08 jrick
4390 24df9a28 2023-07-08 jrick if (dirfd != -1) {
4391 24df9a28 2023-07-08 jrick fd = openat(dirfd, de_name,
4392 24df9a28 2023-07-08 jrick O_RDONLY | O_NOFOLLOW | O_CLOEXEC);
4393 24df9a28 2023-07-08 jrick if (fd == -1) {
4394 24df9a28 2023-07-08 jrick if (!got_err_open_nofollow_on_symlink()) {
4395 24df9a28 2023-07-08 jrick err = got_error_from_errno2("openat",
4396 24df9a28 2023-07-08 jrick abspath);
4397 24df9a28 2023-07-08 jrick goto done;
4398 24df9a28 2023-07-08 jrick }
4399 24df9a28 2023-07-08 jrick err = get_symlink_target_file(&fd, dirfd,
4400 24df9a28 2023-07-08 jrick de_name, abspath);
4401 24df9a28 2023-07-08 jrick if (err)
4402 24df9a28 2023-07-08 jrick goto done;
4403 24df9a28 2023-07-08 jrick }
4404 24df9a28 2023-07-08 jrick } else {
4405 24df9a28 2023-07-08 jrick fd = open(abspath, O_RDONLY | O_NOFOLLOW | O_CLOEXEC);
4406 24df9a28 2023-07-08 jrick if (fd == -1) {
4407 24df9a28 2023-07-08 jrick if (!got_err_open_nofollow_on_symlink()) {
4408 24df9a28 2023-07-08 jrick err = got_error_from_errno2("open",
4409 24df9a28 2023-07-08 jrick abspath);
4410 24df9a28 2023-07-08 jrick goto done;
4411 24df9a28 2023-07-08 jrick }
4412 24df9a28 2023-07-08 jrick err = get_symlink_target_file(&fd, dirfd,
4413 24df9a28 2023-07-08 jrick de_name, abspath);
4414 24df9a28 2023-07-08 jrick if (err)
4415 24df9a28 2023-07-08 jrick goto done;
4416 24df9a28 2023-07-08 jrick }
4417 24df9a28 2023-07-08 jrick }
4418 24df9a28 2023-07-08 jrick if (fstatat(fd, abspath, &sb, AT_SYMLINK_NOFOLLOW) == -1) {
4419 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fstatat", abspath);
4420 24df9a28 2023-07-08 jrick goto done;
4421 24df9a28 2023-07-08 jrick }
4422 24df9a28 2023-07-08 jrick f2 = fdopen(fd, "r");
4423 24df9a28 2023-07-08 jrick if (f2 == NULL) {
4424 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fdopen", abspath);
4425 24df9a28 2023-07-08 jrick goto done;
4426 24df9a28 2023-07-08 jrick }
4427 24df9a28 2023-07-08 jrick fd = -1;
4428 24df9a28 2023-07-08 jrick f2_exists = 1;
4429 24df9a28 2023-07-08 jrick }
4430 24df9a28 2023-07-08 jrick
4431 24df9a28 2023-07-08 jrick if (blob1) {
4432 24df9a28 2023-07-08 jrick err = got_object_blob_dump_to_file(&size1, NULL, NULL,
4433 24df9a28 2023-07-08 jrick a->f1, blob1);
4434 24df9a28 2023-07-08 jrick if (err)
4435 24df9a28 2023-07-08 jrick goto done;
4436 24df9a28 2023-07-08 jrick }
4437 24df9a28 2023-07-08 jrick
4438 24df9a28 2023-07-08 jrick err = got_diff_blob_file(blob1, a->f1, size1, label1, f2 ? f2 : a->f2,
4439 24df9a28 2023-07-08 jrick f2_exists, &sb, path, GOT_DIFF_ALGORITHM_PATIENCE, a->diff_context,
4440 24df9a28 2023-07-08 jrick a->ignore_whitespace, a->force_text_diff, a->diffstat, a->outfile);
4441 24df9a28 2023-07-08 jrick done:
4442 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && err == NULL)
4443 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
4444 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && err == NULL)
4445 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
4446 24df9a28 2023-07-08 jrick if (blob1)
4447 24df9a28 2023-07-08 jrick got_object_blob_close(blob1);
4448 24df9a28 2023-07-08 jrick if (fd != -1 && close(fd) == -1 && err == NULL)
4449 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
4450 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && err == NULL)
4451 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
4452 24df9a28 2023-07-08 jrick free(abspath);
4453 24df9a28 2023-07-08 jrick return err;
4454 24df9a28 2023-07-08 jrick }
4455 24df9a28 2023-07-08 jrick
4456 24df9a28 2023-07-08 jrick static const struct got_error *
4457 24df9a28 2023-07-08 jrick cmd_diff(int argc, char *argv[])
4458 24df9a28 2023-07-08 jrick {
4459 24df9a28 2023-07-08 jrick const struct got_error *error;
4460 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
4461 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
4462 24df9a28 2023-07-08 jrick char *cwd = NULL, *repo_path = NULL;
4463 24df9a28 2023-07-08 jrick const char *commit_args[2] = { NULL, NULL };
4464 24df9a28 2023-07-08 jrick int ncommit_args = 0;
4465 24df9a28 2023-07-08 jrick struct got_object_id *ids[2] = { NULL, NULL };
4466 24df9a28 2023-07-08 jrick char *labels[2] = { NULL, NULL };
4467 24df9a28 2023-07-08 jrick int type1 = GOT_OBJ_TYPE_ANY, type2 = GOT_OBJ_TYPE_ANY;
4468 24df9a28 2023-07-08 jrick int diff_context = 3, diff_staged = 0, ignore_whitespace = 0, ch, i;
4469 24df9a28 2023-07-08 jrick int force_text_diff = 0, force_path = 0, rflag = 0, show_diffstat = 0;
4470 24df9a28 2023-07-08 jrick const char *errstr;
4471 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
4472 24df9a28 2023-07-08 jrick struct got_pathlist_head diffstat_paths, paths;
4473 24df9a28 2023-07-08 jrick FILE *f1 = NULL, *f2 = NULL, *outfile = NULL;
4474 24df9a28 2023-07-08 jrick int fd1 = -1, fd2 = -1;
4475 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
4476 24df9a28 2023-07-08 jrick struct got_diffstat_cb_arg dsa;
4477 24df9a28 2023-07-08 jrick
4478 24df9a28 2023-07-08 jrick memset(&dsa, 0, sizeof(dsa));
4479 24df9a28 2023-07-08 jrick
4480 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
4481 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
4482 24df9a28 2023-07-08 jrick TAILQ_INIT(&diffstat_paths);
4483 24df9a28 2023-07-08 jrick
4484 24df9a28 2023-07-08 jrick #ifndef PROFILE
4485 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
4486 24df9a28 2023-07-08 jrick NULL) == -1)
4487 24df9a28 2023-07-08 jrick err(1, "pledge");
4488 24df9a28 2023-07-08 jrick #endif
4489 24df9a28 2023-07-08 jrick
4490 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "aC:c:dPr:sw")) != -1) {
4491 24df9a28 2023-07-08 jrick switch (ch) {
4492 24df9a28 2023-07-08 jrick case 'a':
4493 24df9a28 2023-07-08 jrick force_text_diff = 1;
4494 24df9a28 2023-07-08 jrick break;
4495 24df9a28 2023-07-08 jrick case 'C':
4496 24df9a28 2023-07-08 jrick diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
4497 24df9a28 2023-07-08 jrick &errstr);
4498 24df9a28 2023-07-08 jrick if (errstr != NULL)
4499 24df9a28 2023-07-08 jrick errx(1, "number of context lines is %s: %s",
4500 24df9a28 2023-07-08 jrick errstr, optarg);
4501 24df9a28 2023-07-08 jrick break;
4502 24df9a28 2023-07-08 jrick case 'c':
4503 24df9a28 2023-07-08 jrick if (ncommit_args >= 2)
4504 24df9a28 2023-07-08 jrick errx(1, "too many -c options used");
4505 24df9a28 2023-07-08 jrick commit_args[ncommit_args++] = optarg;
4506 24df9a28 2023-07-08 jrick break;
4507 24df9a28 2023-07-08 jrick case 'd':
4508 24df9a28 2023-07-08 jrick show_diffstat = 1;
4509 24df9a28 2023-07-08 jrick break;
4510 24df9a28 2023-07-08 jrick case 'P':
4511 24df9a28 2023-07-08 jrick force_path = 1;
4512 24df9a28 2023-07-08 jrick break;
4513 24df9a28 2023-07-08 jrick case 'r':
4514 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
4515 24df9a28 2023-07-08 jrick if (repo_path == NULL)
4516 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
4517 24df9a28 2023-07-08 jrick optarg);
4518 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
4519 24df9a28 2023-07-08 jrick rflag = 1;
4520 24df9a28 2023-07-08 jrick break;
4521 24df9a28 2023-07-08 jrick case 's':
4522 24df9a28 2023-07-08 jrick diff_staged = 1;
4523 24df9a28 2023-07-08 jrick break;
4524 24df9a28 2023-07-08 jrick case 'w':
4525 24df9a28 2023-07-08 jrick ignore_whitespace = 1;
4526 24df9a28 2023-07-08 jrick break;
4527 24df9a28 2023-07-08 jrick default:
4528 24df9a28 2023-07-08 jrick usage_diff();
4529 24df9a28 2023-07-08 jrick /* NOTREACHED */
4530 24df9a28 2023-07-08 jrick }
4531 24df9a28 2023-07-08 jrick }
4532 24df9a28 2023-07-08 jrick
4533 24df9a28 2023-07-08 jrick argc -= optind;
4534 24df9a28 2023-07-08 jrick argv += optind;
4535 24df9a28 2023-07-08 jrick
4536 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
4537 24df9a28 2023-07-08 jrick if (cwd == NULL) {
4538 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
4539 24df9a28 2023-07-08 jrick goto done;
4540 24df9a28 2023-07-08 jrick }
4541 24df9a28 2023-07-08 jrick
4542 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
4543 24df9a28 2023-07-08 jrick if (error != NULL)
4544 24df9a28 2023-07-08 jrick goto done;
4545 24df9a28 2023-07-08 jrick
4546 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4547 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
4548 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
4549 24df9a28 2023-07-08 jrick goto done;
4550 24df9a28 2023-07-08 jrick else
4551 24df9a28 2023-07-08 jrick error = NULL;
4552 24df9a28 2023-07-08 jrick if (worktree) {
4553 24df9a28 2023-07-08 jrick repo_path =
4554 24df9a28 2023-07-08 jrick strdup(got_worktree_get_repo_path(worktree));
4555 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4556 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
4557 24df9a28 2023-07-08 jrick goto done;
4558 24df9a28 2023-07-08 jrick }
4559 24df9a28 2023-07-08 jrick } else {
4560 24df9a28 2023-07-08 jrick repo_path = strdup(cwd);
4561 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
4562 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
4563 24df9a28 2023-07-08 jrick goto done;
4564 24df9a28 2023-07-08 jrick }
4565 24df9a28 2023-07-08 jrick }
4566 24df9a28 2023-07-08 jrick }
4567 24df9a28 2023-07-08 jrick
4568 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4569 24df9a28 2023-07-08 jrick free(repo_path);
4570 24df9a28 2023-07-08 jrick if (error != NULL)
4571 24df9a28 2023-07-08 jrick goto done;
4572 24df9a28 2023-07-08 jrick
4573 24df9a28 2023-07-08 jrick if (show_diffstat) {
4574 24df9a28 2023-07-08 jrick dsa.paths = &diffstat_paths;
4575 24df9a28 2023-07-08 jrick dsa.force_text = force_text_diff;
4576 24df9a28 2023-07-08 jrick dsa.ignore_ws = ignore_whitespace;
4577 24df9a28 2023-07-08 jrick dsa.diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
4578 24df9a28 2023-07-08 jrick }
4579 24df9a28 2023-07-08 jrick
4580 24df9a28 2023-07-08 jrick if (rflag || worktree == NULL || ncommit_args > 0) {
4581 24df9a28 2023-07-08 jrick if (force_path) {
4582 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_NOT_IMPL,
4583 24df9a28 2023-07-08 jrick "-P option can only be used when diffing "
4584 24df9a28 2023-07-08 jrick "a work tree");
4585 24df9a28 2023-07-08 jrick goto done;
4586 24df9a28 2023-07-08 jrick }
4587 24df9a28 2023-07-08 jrick if (diff_staged) {
4588 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_NOT_IMPL,
4589 24df9a28 2023-07-08 jrick "-s option can only be used when diffing "
4590 24df9a28 2023-07-08 jrick "a work tree");
4591 24df9a28 2023-07-08 jrick goto done;
4592 24df9a28 2023-07-08 jrick }
4593 24df9a28 2023-07-08 jrick }
4594 24df9a28 2023-07-08 jrick
4595 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1,
4596 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_root_path(worktree) : NULL);
4597 24df9a28 2023-07-08 jrick if (error)
4598 24df9a28 2023-07-08 jrick goto done;
4599 24df9a28 2023-07-08 jrick
4600 24df9a28 2023-07-08 jrick if ((!force_path && argc == 2) || ncommit_args > 0) {
4601 24df9a28 2023-07-08 jrick int obj_type = (ncommit_args > 0 ?
4602 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT : GOT_OBJ_TYPE_ANY);
4603 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
4604 24df9a28 2023-07-08 jrick NULL);
4605 24df9a28 2023-07-08 jrick if (error)
4606 24df9a28 2023-07-08 jrick goto done;
4607 24df9a28 2023-07-08 jrick for (i = 0; i < (ncommit_args > 0 ? ncommit_args : argc); i++) {
4608 24df9a28 2023-07-08 jrick const char *arg;
4609 24df9a28 2023-07-08 jrick if (ncommit_args > 0)
4610 24df9a28 2023-07-08 jrick arg = commit_args[i];
4611 24df9a28 2023-07-08 jrick else
4612 24df9a28 2023-07-08 jrick arg = argv[i];
4613 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&ids[i], &labels[i],
4614 24df9a28 2023-07-08 jrick arg, obj_type, &refs, repo);
4615 24df9a28 2023-07-08 jrick if (error) {
4616 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_REF &&
4617 24df9a28 2023-07-08 jrick error->code != GOT_ERR_NO_OBJ)
4618 24df9a28 2023-07-08 jrick goto done;
4619 24df9a28 2023-07-08 jrick if (ncommit_args > 0)
4620 24df9a28 2023-07-08 jrick goto done;
4621 24df9a28 2023-07-08 jrick error = NULL;
4622 24df9a28 2023-07-08 jrick break;
4623 24df9a28 2023-07-08 jrick }
4624 24df9a28 2023-07-08 jrick }
4625 24df9a28 2023-07-08 jrick }
4626 24df9a28 2023-07-08 jrick
4627 24df9a28 2023-07-08 jrick f1 = got_opentemp();
4628 24df9a28 2023-07-08 jrick if (f1 == NULL) {
4629 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
4630 24df9a28 2023-07-08 jrick goto done;
4631 24df9a28 2023-07-08 jrick }
4632 24df9a28 2023-07-08 jrick
4633 24df9a28 2023-07-08 jrick f2 = got_opentemp();
4634 24df9a28 2023-07-08 jrick if (f2 == NULL) {
4635 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
4636 24df9a28 2023-07-08 jrick goto done;
4637 24df9a28 2023-07-08 jrick }
4638 24df9a28 2023-07-08 jrick
4639 24df9a28 2023-07-08 jrick outfile = got_opentemp();
4640 24df9a28 2023-07-08 jrick if (outfile == NULL) {
4641 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
4642 24df9a28 2023-07-08 jrick goto done;
4643 24df9a28 2023-07-08 jrick }
4644 24df9a28 2023-07-08 jrick
4645 24df9a28 2023-07-08 jrick if (ncommit_args == 0 && (ids[0] == NULL || ids[1] == NULL)) {
4646 24df9a28 2023-07-08 jrick struct print_diff_arg arg;
4647 24df9a28 2023-07-08 jrick char *id_str;
4648 24df9a28 2023-07-08 jrick
4649 24df9a28 2023-07-08 jrick if (worktree == NULL) {
4650 24df9a28 2023-07-08 jrick if (argc == 2 && ids[0] == NULL) {
4651 24df9a28 2023-07-08 jrick error = got_error_path(argv[0], GOT_ERR_NO_OBJ);
4652 24df9a28 2023-07-08 jrick goto done;
4653 24df9a28 2023-07-08 jrick } else if (argc == 2 && ids[1] == NULL) {
4654 24df9a28 2023-07-08 jrick error = got_error_path(argv[1], GOT_ERR_NO_OBJ);
4655 24df9a28 2023-07-08 jrick goto done;
4656 24df9a28 2023-07-08 jrick } else if (argc > 0) {
4657 24df9a28 2023-07-08 jrick error = got_error_fmt(GOT_ERR_NOT_WORKTREE,
4658 24df9a28 2023-07-08 jrick "%s", "specified paths cannot be resolved");
4659 24df9a28 2023-07-08 jrick goto done;
4660 24df9a28 2023-07-08 jrick } else {
4661 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_NOT_WORKTREE);
4662 24df9a28 2023-07-08 jrick goto done;
4663 24df9a28 2023-07-08 jrick }
4664 24df9a28 2023-07-08 jrick }
4665 24df9a28 2023-07-08 jrick
4666 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv,
4667 24df9a28 2023-07-08 jrick worktree);
4668 24df9a28 2023-07-08 jrick if (error)
4669 24df9a28 2023-07-08 jrick goto done;
4670 24df9a28 2023-07-08 jrick
4671 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str,
4672 24df9a28 2023-07-08 jrick got_worktree_get_base_commit_id(worktree));
4673 24df9a28 2023-07-08 jrick if (error)
4674 24df9a28 2023-07-08 jrick goto done;
4675 24df9a28 2023-07-08 jrick arg.repo = repo;
4676 24df9a28 2023-07-08 jrick arg.worktree = worktree;
4677 24df9a28 2023-07-08 jrick arg.diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
4678 24df9a28 2023-07-08 jrick arg.diff_context = diff_context;
4679 24df9a28 2023-07-08 jrick arg.id_str = id_str;
4680 24df9a28 2023-07-08 jrick arg.header_shown = 0;
4681 24df9a28 2023-07-08 jrick arg.diff_staged = diff_staged;
4682 24df9a28 2023-07-08 jrick arg.ignore_whitespace = ignore_whitespace;
4683 24df9a28 2023-07-08 jrick arg.force_text_diff = force_text_diff;
4684 24df9a28 2023-07-08 jrick arg.diffstat = show_diffstat ? &dsa : NULL;
4685 24df9a28 2023-07-08 jrick arg.f1 = f1;
4686 24df9a28 2023-07-08 jrick arg.f2 = f2;
4687 24df9a28 2023-07-08 jrick arg.outfile = outfile;
4688 24df9a28 2023-07-08 jrick
4689 24df9a28 2023-07-08 jrick error = got_worktree_status(worktree, &paths, repo, 0,
4690 24df9a28 2023-07-08 jrick print_diff, &arg, check_cancelled, NULL);
4691 24df9a28 2023-07-08 jrick free(id_str);
4692 24df9a28 2023-07-08 jrick if (error)
4693 24df9a28 2023-07-08 jrick goto done;
4694 24df9a28 2023-07-08 jrick
4695 24df9a28 2023-07-08 jrick if (show_diffstat && dsa.nfiles > 0) {
4696 24df9a28 2023-07-08 jrick char *header;
4697 24df9a28 2023-07-08 jrick
4698 24df9a28 2023-07-08 jrick if (asprintf(&header, "diffstat %s%s",
4699 24df9a28 2023-07-08 jrick diff_staged ? "-s " : "",
4700 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree)) == -1) {
4701 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
4702 24df9a28 2023-07-08 jrick goto done;
4703 24df9a28 2023-07-08 jrick }
4704 24df9a28 2023-07-08 jrick
4705 24df9a28 2023-07-08 jrick error = print_diffstat(&dsa, header);
4706 24df9a28 2023-07-08 jrick free(header);
4707 24df9a28 2023-07-08 jrick if (error)
4708 24df9a28 2023-07-08 jrick goto done;
4709 24df9a28 2023-07-08 jrick }
4710 24df9a28 2023-07-08 jrick
4711 24df9a28 2023-07-08 jrick error = printfile(outfile);
4712 24df9a28 2023-07-08 jrick goto done;
4713 24df9a28 2023-07-08 jrick }
4714 24df9a28 2023-07-08 jrick
4715 24df9a28 2023-07-08 jrick if (ncommit_args == 1) {
4716 24df9a28 2023-07-08 jrick struct got_commit_object *commit;
4717 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, ids[0]);
4718 24df9a28 2023-07-08 jrick if (error)
4719 24df9a28 2023-07-08 jrick goto done;
4720 24df9a28 2023-07-08 jrick
4721 24df9a28 2023-07-08 jrick labels[1] = labels[0];
4722 24df9a28 2023-07-08 jrick ids[1] = ids[0];
4723 24df9a28 2023-07-08 jrick if (got_object_commit_get_nparents(commit) > 0) {
4724 24df9a28 2023-07-08 jrick const struct got_object_id_queue *pids;
4725 24df9a28 2023-07-08 jrick struct got_object_qid *pid;
4726 24df9a28 2023-07-08 jrick pids = got_object_commit_get_parent_ids(commit);
4727 24df9a28 2023-07-08 jrick pid = STAILQ_FIRST(pids);
4728 24df9a28 2023-07-08 jrick ids[0] = got_object_id_dup(&pid->id);
4729 24df9a28 2023-07-08 jrick if (ids[0] == NULL) {
4730 24df9a28 2023-07-08 jrick error = got_error_from_errno(
4731 24df9a28 2023-07-08 jrick "got_object_id_dup");
4732 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
4733 24df9a28 2023-07-08 jrick goto done;
4734 24df9a28 2023-07-08 jrick }
4735 24df9a28 2023-07-08 jrick error = got_object_id_str(&labels[0], ids[0]);
4736 24df9a28 2023-07-08 jrick if (error) {
4737 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
4738 24df9a28 2023-07-08 jrick goto done;
4739 24df9a28 2023-07-08 jrick }
4740 24df9a28 2023-07-08 jrick } else {
4741 24df9a28 2023-07-08 jrick ids[0] = NULL;
4742 24df9a28 2023-07-08 jrick labels[0] = strdup("/dev/null");
4743 24df9a28 2023-07-08 jrick if (labels[0] == NULL) {
4744 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
4745 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
4746 24df9a28 2023-07-08 jrick goto done;
4747 24df9a28 2023-07-08 jrick }
4748 24df9a28 2023-07-08 jrick }
4749 24df9a28 2023-07-08 jrick
4750 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
4751 24df9a28 2023-07-08 jrick }
4752 24df9a28 2023-07-08 jrick
4753 24df9a28 2023-07-08 jrick if (ncommit_args == 0 && argc > 2) {
4754 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_BAD_PATH,
4755 24df9a28 2023-07-08 jrick "path arguments cannot be used when diffing two objects");
4756 24df9a28 2023-07-08 jrick goto done;
4757 24df9a28 2023-07-08 jrick }
4758 24df9a28 2023-07-08 jrick
4759 24df9a28 2023-07-08 jrick if (ids[0]) {
4760 24df9a28 2023-07-08 jrick error = got_object_get_type(&type1, repo, ids[0]);
4761 24df9a28 2023-07-08 jrick if (error)
4762 24df9a28 2023-07-08 jrick goto done;
4763 24df9a28 2023-07-08 jrick }
4764 24df9a28 2023-07-08 jrick
4765 24df9a28 2023-07-08 jrick error = got_object_get_type(&type2, repo, ids[1]);
4766 24df9a28 2023-07-08 jrick if (error)
4767 24df9a28 2023-07-08 jrick goto done;
4768 24df9a28 2023-07-08 jrick if (type1 != GOT_OBJ_TYPE_ANY && type1 != type2) {
4769 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_OBJ_TYPE);
4770 24df9a28 2023-07-08 jrick goto done;
4771 24df9a28 2023-07-08 jrick }
4772 24df9a28 2023-07-08 jrick if (type1 == GOT_OBJ_TYPE_BLOB && argc > 2) {
4773 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_OBJ_TYPE,
4774 24df9a28 2023-07-08 jrick "path arguments cannot be used when diffing blobs");
4775 24df9a28 2023-07-08 jrick goto done;
4776 24df9a28 2023-07-08 jrick }
4777 24df9a28 2023-07-08 jrick
4778 24df9a28 2023-07-08 jrick for (i = 0; ncommit_args > 0 && i < argc; i++) {
4779 24df9a28 2023-07-08 jrick char *in_repo_path;
4780 24df9a28 2023-07-08 jrick struct got_pathlist_entry *new;
4781 24df9a28 2023-07-08 jrick if (worktree) {
4782 24df9a28 2023-07-08 jrick const char *prefix;
4783 24df9a28 2023-07-08 jrick char *p;
4784 24df9a28 2023-07-08 jrick error = got_worktree_resolve_path(&p, worktree,
4785 24df9a28 2023-07-08 jrick argv[i]);
4786 24df9a28 2023-07-08 jrick if (error)
4787 24df9a28 2023-07-08 jrick goto done;
4788 24df9a28 2023-07-08 jrick prefix = got_worktree_get_path_prefix(worktree);
4789 24df9a28 2023-07-08 jrick while (prefix[0] == '/')
4790 24df9a28 2023-07-08 jrick prefix++;
4791 24df9a28 2023-07-08 jrick if (asprintf(&in_repo_path, "%s%s%s", prefix,
4792 24df9a28 2023-07-08 jrick (p[0] != '\0' && prefix[0] != '\0') ? "/" : "",
4793 24df9a28 2023-07-08 jrick p) == -1) {
4794 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
4795 24df9a28 2023-07-08 jrick free(p);
4796 24df9a28 2023-07-08 jrick goto done;
4797 24df9a28 2023-07-08 jrick }
4798 24df9a28 2023-07-08 jrick free(p);
4799 24df9a28 2023-07-08 jrick } else {
4800 24df9a28 2023-07-08 jrick char *mapped_path, *s;
4801 24df9a28 2023-07-08 jrick error = got_repo_map_path(&mapped_path, repo, argv[i]);
4802 24df9a28 2023-07-08 jrick if (error)
4803 24df9a28 2023-07-08 jrick goto done;
4804 24df9a28 2023-07-08 jrick s = mapped_path;
4805 24df9a28 2023-07-08 jrick while (s[0] == '/')
4806 24df9a28 2023-07-08 jrick s++;
4807 24df9a28 2023-07-08 jrick in_repo_path = strdup(s);
4808 24df9a28 2023-07-08 jrick if (in_repo_path == NULL) {
4809 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
4810 24df9a28 2023-07-08 jrick free(mapped_path);
4811 24df9a28 2023-07-08 jrick goto done;
4812 24df9a28 2023-07-08 jrick }
4813 24df9a28 2023-07-08 jrick free(mapped_path);
4814 24df9a28 2023-07-08 jrick
4815 24df9a28 2023-07-08 jrick }
4816 24df9a28 2023-07-08 jrick error = got_pathlist_insert(&new, &paths, in_repo_path, NULL);
4817 24df9a28 2023-07-08 jrick if (error || new == NULL /* duplicate */)
4818 24df9a28 2023-07-08 jrick free(in_repo_path);
4819 24df9a28 2023-07-08 jrick if (error)
4820 24df9a28 2023-07-08 jrick goto done;
4821 24df9a28 2023-07-08 jrick }
4822 24df9a28 2023-07-08 jrick
4823 24df9a28 2023-07-08 jrick if (worktree) {
4824 24df9a28 2023-07-08 jrick /* Release work tree lock. */
4825 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
4826 24df9a28 2023-07-08 jrick worktree = NULL;
4827 24df9a28 2023-07-08 jrick }
4828 24df9a28 2023-07-08 jrick
4829 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
4830 24df9a28 2023-07-08 jrick if (fd1 == -1) {
4831 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentempfd");
4832 24df9a28 2023-07-08 jrick goto done;
4833 24df9a28 2023-07-08 jrick }
4834 24df9a28 2023-07-08 jrick
4835 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
4836 24df9a28 2023-07-08 jrick if (fd2 == -1) {
4837 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentempfd");
4838 24df9a28 2023-07-08 jrick goto done;
4839 24df9a28 2023-07-08 jrick }
4840 24df9a28 2023-07-08 jrick
4841 24df9a28 2023-07-08 jrick switch (type1 == GOT_OBJ_TYPE_ANY ? type2 : type1) {
4842 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_BLOB:
4843 24df9a28 2023-07-08 jrick error = got_diff_objects_as_blobs(NULL, NULL, f1, f2,
4844 24df9a28 2023-07-08 jrick fd1, fd2, ids[0], ids[1], NULL, NULL,
4845 24df9a28 2023-07-08 jrick GOT_DIFF_ALGORITHM_PATIENCE, diff_context,
4846 24df9a28 2023-07-08 jrick ignore_whitespace, force_text_diff,
4847 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL, repo, outfile);
4848 24df9a28 2023-07-08 jrick break;
4849 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TREE:
4850 24df9a28 2023-07-08 jrick error = got_diff_objects_as_trees(NULL, NULL, f1, f2, fd1, fd2,
4851 24df9a28 2023-07-08 jrick ids[0], ids[1], &paths, "", "",
4852 24df9a28 2023-07-08 jrick GOT_DIFF_ALGORITHM_PATIENCE, diff_context,
4853 24df9a28 2023-07-08 jrick ignore_whitespace, force_text_diff,
4854 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL, repo, outfile);
4855 24df9a28 2023-07-08 jrick break;
4856 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_COMMIT:
4857 24df9a28 2023-07-08 jrick fprintf(outfile, "diff %s %s\n", labels[0], labels[1]);
4858 24df9a28 2023-07-08 jrick error = got_diff_objects_as_commits(NULL, NULL, f1, f2,
4859 24df9a28 2023-07-08 jrick fd1, fd2, ids[0], ids[1], &paths,
4860 24df9a28 2023-07-08 jrick GOT_DIFF_ALGORITHM_PATIENCE, diff_context,
4861 24df9a28 2023-07-08 jrick ignore_whitespace, force_text_diff,
4862 24df9a28 2023-07-08 jrick show_diffstat ? &dsa : NULL, repo, outfile);
4863 24df9a28 2023-07-08 jrick break;
4864 24df9a28 2023-07-08 jrick default:
4865 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_OBJ_TYPE);
4866 24df9a28 2023-07-08 jrick }
4867 24df9a28 2023-07-08 jrick if (error)
4868 24df9a28 2023-07-08 jrick goto done;
4869 24df9a28 2023-07-08 jrick
4870 24df9a28 2023-07-08 jrick if (show_diffstat && dsa.nfiles > 0) {
4871 24df9a28 2023-07-08 jrick char *header = NULL;
4872 24df9a28 2023-07-08 jrick
4873 24df9a28 2023-07-08 jrick if (asprintf(&header, "diffstat %s %s",
4874 24df9a28 2023-07-08 jrick labels[0], labels[1]) == -1) {
4875 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
4876 24df9a28 2023-07-08 jrick goto done;
4877 24df9a28 2023-07-08 jrick }
4878 24df9a28 2023-07-08 jrick
4879 24df9a28 2023-07-08 jrick error = print_diffstat(&dsa, header);
4880 24df9a28 2023-07-08 jrick free(header);
4881 24df9a28 2023-07-08 jrick if (error)
4882 24df9a28 2023-07-08 jrick goto done;
4883 24df9a28 2023-07-08 jrick }
4884 24df9a28 2023-07-08 jrick
4885 24df9a28 2023-07-08 jrick error = printfile(outfile);
4886 24df9a28 2023-07-08 jrick
4887 24df9a28 2023-07-08 jrick done:
4888 24df9a28 2023-07-08 jrick free(labels[0]);
4889 24df9a28 2023-07-08 jrick free(labels[1]);
4890 24df9a28 2023-07-08 jrick free(ids[0]);
4891 24df9a28 2023-07-08 jrick free(ids[1]);
4892 24df9a28 2023-07-08 jrick if (worktree)
4893 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
4894 24df9a28 2023-07-08 jrick if (repo) {
4895 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
4896 24df9a28 2023-07-08 jrick if (error == NULL)
4897 24df9a28 2023-07-08 jrick error = close_err;
4898 24df9a28 2023-07-08 jrick }
4899 24df9a28 2023-07-08 jrick if (pack_fds) {
4900 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
4901 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
4902 24df9a28 2023-07-08 jrick if (error == NULL)
4903 24df9a28 2023-07-08 jrick error = pack_err;
4904 24df9a28 2023-07-08 jrick }
4905 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
4906 24df9a28 2023-07-08 jrick got_pathlist_free(&diffstat_paths, GOT_PATHLIST_FREE_ALL);
4907 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
4908 24df9a28 2023-07-08 jrick if (outfile && fclose(outfile) == EOF && error == NULL)
4909 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
4910 24df9a28 2023-07-08 jrick if (f1 && fclose(f1) == EOF && error == NULL)
4911 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
4912 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && error == NULL)
4913 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
4914 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && error == NULL)
4915 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
4916 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && error == NULL)
4917 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
4918 24df9a28 2023-07-08 jrick return error;
4919 24df9a28 2023-07-08 jrick }
4920 24df9a28 2023-07-08 jrick
4921 24df9a28 2023-07-08 jrick __dead static void
4922 24df9a28 2023-07-08 jrick usage_blame(void)
4923 24df9a28 2023-07-08 jrick {
4924 24df9a28 2023-07-08 jrick fprintf(stderr,
4925 24df9a28 2023-07-08 jrick "usage: %s blame [-c commit] [-r repository-path] path\n",
4926 24df9a28 2023-07-08 jrick getprogname());
4927 24df9a28 2023-07-08 jrick exit(1);
4928 24df9a28 2023-07-08 jrick }
4929 24df9a28 2023-07-08 jrick
4930 24df9a28 2023-07-08 jrick struct blame_line {
4931 24df9a28 2023-07-08 jrick int annotated;
4932 24df9a28 2023-07-08 jrick char *id_str;
4933 24df9a28 2023-07-08 jrick char *committer;
4934 24df9a28 2023-07-08 jrick char datebuf[11]; /* YYYY-MM-DD + NUL */
4935 24df9a28 2023-07-08 jrick };
4936 24df9a28 2023-07-08 jrick
4937 24df9a28 2023-07-08 jrick struct blame_cb_args {
4938 24df9a28 2023-07-08 jrick struct blame_line *lines;
4939 24df9a28 2023-07-08 jrick int nlines;
4940 24df9a28 2023-07-08 jrick int nlines_prec;
4941 24df9a28 2023-07-08 jrick int lineno_cur;
4942 24df9a28 2023-07-08 jrick off_t *line_offsets;
4943 24df9a28 2023-07-08 jrick FILE *f;
4944 24df9a28 2023-07-08 jrick struct got_repository *repo;
4945 24df9a28 2023-07-08 jrick };
4946 24df9a28 2023-07-08 jrick
4947 24df9a28 2023-07-08 jrick static const struct got_error *
4948 24df9a28 2023-07-08 jrick blame_cb(void *arg, int nlines, int lineno,
4949 24df9a28 2023-07-08 jrick struct got_commit_object *commit, struct got_object_id *id)
4950 24df9a28 2023-07-08 jrick {
4951 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
4952 24df9a28 2023-07-08 jrick struct blame_cb_args *a = arg;
4953 24df9a28 2023-07-08 jrick struct blame_line *bline;
4954 24df9a28 2023-07-08 jrick char *line = NULL;
4955 24df9a28 2023-07-08 jrick size_t linesize = 0;
4956 24df9a28 2023-07-08 jrick off_t offset;
4957 24df9a28 2023-07-08 jrick struct tm tm;
4958 24df9a28 2023-07-08 jrick time_t committer_time;
4959 24df9a28 2023-07-08 jrick
4960 24df9a28 2023-07-08 jrick if (nlines != a->nlines ||
4961 24df9a28 2023-07-08 jrick (lineno != -1 && lineno < 1) || lineno > a->nlines)
4962 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_RANGE);
4963 24df9a28 2023-07-08 jrick
4964 24df9a28 2023-07-08 jrick if (sigint_received)
4965 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_ITER_COMPLETED);
4966 24df9a28 2023-07-08 jrick
4967 24df9a28 2023-07-08 jrick if (lineno == -1)
4968 24df9a28 2023-07-08 jrick return NULL; /* no change in this commit */
4969 24df9a28 2023-07-08 jrick
4970 24df9a28 2023-07-08 jrick /* Annotate this line. */
4971 24df9a28 2023-07-08 jrick bline = &a->lines[lineno - 1];
4972 24df9a28 2023-07-08 jrick if (bline->annotated)
4973 24df9a28 2023-07-08 jrick return NULL;
4974 24df9a28 2023-07-08 jrick err = got_object_id_str(&bline->id_str, id);
4975 24df9a28 2023-07-08 jrick if (err)
4976 24df9a28 2023-07-08 jrick return err;
4977 24df9a28 2023-07-08 jrick
4978 24df9a28 2023-07-08 jrick bline->committer = strdup(got_object_commit_get_committer(commit));
4979 24df9a28 2023-07-08 jrick if (bline->committer == NULL) {
4980 24df9a28 2023-07-08 jrick err = got_error_from_errno("strdup");
4981 24df9a28 2023-07-08 jrick goto done;
4982 24df9a28 2023-07-08 jrick }
4983 24df9a28 2023-07-08 jrick
4984 24df9a28 2023-07-08 jrick committer_time = got_object_commit_get_committer_time(commit);
4985 24df9a28 2023-07-08 jrick if (gmtime_r(&committer_time, &tm) == NULL)
4986 24df9a28 2023-07-08 jrick return got_error_from_errno("gmtime_r");
4987 24df9a28 2023-07-08 jrick if (strftime(bline->datebuf, sizeof(bline->datebuf), "%G-%m-%d",
4988 24df9a28 2023-07-08 jrick &tm) == 0) {
4989 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_NO_SPACE);
4990 24df9a28 2023-07-08 jrick goto done;
4991 24df9a28 2023-07-08 jrick }
4992 24df9a28 2023-07-08 jrick bline->annotated = 1;
4993 24df9a28 2023-07-08 jrick
4994 24df9a28 2023-07-08 jrick /* Print lines annotated so far. */
4995 24df9a28 2023-07-08 jrick bline = &a->lines[a->lineno_cur - 1];
4996 24df9a28 2023-07-08 jrick if (!bline->annotated)
4997 24df9a28 2023-07-08 jrick goto done;
4998 24df9a28 2023-07-08 jrick
4999 24df9a28 2023-07-08 jrick offset = a->line_offsets[a->lineno_cur - 1];
5000 24df9a28 2023-07-08 jrick if (fseeko(a->f, offset, SEEK_SET) == -1) {
5001 24df9a28 2023-07-08 jrick err = got_error_from_errno("fseeko");
5002 24df9a28 2023-07-08 jrick goto done;
5003 24df9a28 2023-07-08 jrick }
5004 24df9a28 2023-07-08 jrick
5005 24df9a28 2023-07-08 jrick while (a->lineno_cur <= a->nlines && bline->annotated) {
5006 24df9a28 2023-07-08 jrick char *smallerthan, *at, *nl, *committer;
5007 24df9a28 2023-07-08 jrick size_t len;
5008 24df9a28 2023-07-08 jrick
5009 24df9a28 2023-07-08 jrick if (getline(&line, &linesize, a->f) == -1) {
5010 24df9a28 2023-07-08 jrick if (ferror(a->f))
5011 24df9a28 2023-07-08 jrick err = got_error_from_errno("getline");
5012 24df9a28 2023-07-08 jrick break;
5013 24df9a28 2023-07-08 jrick }
5014 24df9a28 2023-07-08 jrick
5015 24df9a28 2023-07-08 jrick committer = bline->committer;
5016 24df9a28 2023-07-08 jrick smallerthan = strchr(committer, '<');
5017 24df9a28 2023-07-08 jrick if (smallerthan && smallerthan[1] != '\0')
5018 24df9a28 2023-07-08 jrick committer = smallerthan + 1;
5019 24df9a28 2023-07-08 jrick at = strchr(committer, '@');
5020 24df9a28 2023-07-08 jrick if (at)
5021 24df9a28 2023-07-08 jrick *at = '\0';
5022 24df9a28 2023-07-08 jrick len = strlen(committer);
5023 24df9a28 2023-07-08 jrick if (len >= 9)
5024 24df9a28 2023-07-08 jrick committer[8] = '\0';
5025 24df9a28 2023-07-08 jrick
5026 24df9a28 2023-07-08 jrick nl = strchr(line, '\n');
5027 24df9a28 2023-07-08 jrick if (nl)
5028 24df9a28 2023-07-08 jrick *nl = '\0';
5029 24df9a28 2023-07-08 jrick printf("%.*d) %.8s %s %-8s %s\n", a->nlines_prec, a->lineno_cur,
5030 24df9a28 2023-07-08 jrick bline->id_str, bline->datebuf, committer, line);
5031 24df9a28 2023-07-08 jrick
5032 24df9a28 2023-07-08 jrick a->lineno_cur++;
5033 24df9a28 2023-07-08 jrick bline = &a->lines[a->lineno_cur - 1];
5034 24df9a28 2023-07-08 jrick }
5035 24df9a28 2023-07-08 jrick done:
5036 24df9a28 2023-07-08 jrick free(line);
5037 24df9a28 2023-07-08 jrick return err;
5038 24df9a28 2023-07-08 jrick }
5039 24df9a28 2023-07-08 jrick
5040 24df9a28 2023-07-08 jrick static const struct got_error *
5041 24df9a28 2023-07-08 jrick cmd_blame(int argc, char *argv[])
5042 24df9a28 2023-07-08 jrick {
5043 24df9a28 2023-07-08 jrick const struct got_error *error;
5044 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
5045 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
5046 24df9a28 2023-07-08 jrick char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5047 24df9a28 2023-07-08 jrick char *link_target = NULL;
5048 24df9a28 2023-07-08 jrick struct got_object_id *obj_id = NULL;
5049 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
5050 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
5051 24df9a28 2023-07-08 jrick struct got_blob_object *blob = NULL;
5052 24df9a28 2023-07-08 jrick char *commit_id_str = NULL;
5053 24df9a28 2023-07-08 jrick struct blame_cb_args bca;
5054 24df9a28 2023-07-08 jrick int ch, obj_type, i, fd1 = -1, fd2 = -1, fd3 = -1;
5055 24df9a28 2023-07-08 jrick off_t filesize;
5056 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
5057 24df9a28 2023-07-08 jrick FILE *f1 = NULL, *f2 = NULL;
5058 24df9a28 2023-07-08 jrick
5059 24df9a28 2023-07-08 jrick fd1 = got_opentempfd();
5060 24df9a28 2023-07-08 jrick if (fd1 == -1)
5061 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentempfd");
5062 24df9a28 2023-07-08 jrick
5063 24df9a28 2023-07-08 jrick memset(&bca, 0, sizeof(bca));
5064 24df9a28 2023-07-08 jrick
5065 24df9a28 2023-07-08 jrick #ifndef PROFILE
5066 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
5067 24df9a28 2023-07-08 jrick NULL) == -1)
5068 24df9a28 2023-07-08 jrick err(1, "pledge");
5069 24df9a28 2023-07-08 jrick #endif
5070 24df9a28 2023-07-08 jrick
5071 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5072 24df9a28 2023-07-08 jrick switch (ch) {
5073 24df9a28 2023-07-08 jrick case 'c':
5074 24df9a28 2023-07-08 jrick commit_id_str = optarg;
5075 24df9a28 2023-07-08 jrick break;
5076 24df9a28 2023-07-08 jrick case 'r':
5077 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
5078 24df9a28 2023-07-08 jrick if (repo_path == NULL)
5079 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
5080 24df9a28 2023-07-08 jrick optarg);
5081 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
5082 24df9a28 2023-07-08 jrick break;
5083 24df9a28 2023-07-08 jrick default:
5084 24df9a28 2023-07-08 jrick usage_blame();
5085 24df9a28 2023-07-08 jrick /* NOTREACHED */
5086 24df9a28 2023-07-08 jrick }
5087 24df9a28 2023-07-08 jrick }
5088 24df9a28 2023-07-08 jrick
5089 24df9a28 2023-07-08 jrick argc -= optind;
5090 24df9a28 2023-07-08 jrick argv += optind;
5091 24df9a28 2023-07-08 jrick
5092 24df9a28 2023-07-08 jrick if (argc == 1)
5093 24df9a28 2023-07-08 jrick path = argv[0];
5094 24df9a28 2023-07-08 jrick else
5095 24df9a28 2023-07-08 jrick usage_blame();
5096 24df9a28 2023-07-08 jrick
5097 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
5098 24df9a28 2023-07-08 jrick if (cwd == NULL) {
5099 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
5100 24df9a28 2023-07-08 jrick goto done;
5101 24df9a28 2023-07-08 jrick }
5102 24df9a28 2023-07-08 jrick
5103 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
5104 24df9a28 2023-07-08 jrick if (error != NULL)
5105 24df9a28 2023-07-08 jrick goto done;
5106 24df9a28 2023-07-08 jrick
5107 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
5108 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
5109 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
5110 24df9a28 2023-07-08 jrick goto done;
5111 24df9a28 2023-07-08 jrick else
5112 24df9a28 2023-07-08 jrick error = NULL;
5113 24df9a28 2023-07-08 jrick if (worktree) {
5114 24df9a28 2023-07-08 jrick repo_path =
5115 24df9a28 2023-07-08 jrick strdup(got_worktree_get_repo_path(worktree));
5116 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
5117 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
5118 24df9a28 2023-07-08 jrick if (error)
5119 24df9a28 2023-07-08 jrick goto done;
5120 24df9a28 2023-07-08 jrick }
5121 24df9a28 2023-07-08 jrick } else {
5122 24df9a28 2023-07-08 jrick repo_path = strdup(cwd);
5123 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
5124 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
5125 24df9a28 2023-07-08 jrick goto done;
5126 24df9a28 2023-07-08 jrick }
5127 24df9a28 2023-07-08 jrick }
5128 24df9a28 2023-07-08 jrick }
5129 24df9a28 2023-07-08 jrick
5130 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
5131 24df9a28 2023-07-08 jrick if (error != NULL)
5132 24df9a28 2023-07-08 jrick goto done;
5133 24df9a28 2023-07-08 jrick
5134 24df9a28 2023-07-08 jrick if (worktree) {
5135 24df9a28 2023-07-08 jrick const char *prefix = got_worktree_get_path_prefix(worktree);
5136 24df9a28 2023-07-08 jrick char *p;
5137 24df9a28 2023-07-08 jrick
5138 24df9a28 2023-07-08 jrick error = got_worktree_resolve_path(&p, worktree, path);
5139 24df9a28 2023-07-08 jrick if (error)
5140 24df9a28 2023-07-08 jrick goto done;
5141 24df9a28 2023-07-08 jrick if (asprintf(&in_repo_path, "%s%s%s", prefix,
5142 24df9a28 2023-07-08 jrick (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
5143 24df9a28 2023-07-08 jrick p) == -1) {
5144 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
5145 24df9a28 2023-07-08 jrick free(p);
5146 24df9a28 2023-07-08 jrick goto done;
5147 24df9a28 2023-07-08 jrick }
5148 24df9a28 2023-07-08 jrick free(p);
5149 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5150 24df9a28 2023-07-08 jrick } else {
5151 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5152 24df9a28 2023-07-08 jrick if (error)
5153 24df9a28 2023-07-08 jrick goto done;
5154 24df9a28 2023-07-08 jrick error = got_repo_map_path(&in_repo_path, repo, path);
5155 24df9a28 2023-07-08 jrick }
5156 24df9a28 2023-07-08 jrick if (error)
5157 24df9a28 2023-07-08 jrick goto done;
5158 24df9a28 2023-07-08 jrick
5159 24df9a28 2023-07-08 jrick if (commit_id_str == NULL) {
5160 24df9a28 2023-07-08 jrick struct got_reference *head_ref;
5161 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo, worktree ?
5162 24df9a28 2023-07-08 jrick got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
5163 24df9a28 2023-07-08 jrick if (error != NULL)
5164 24df9a28 2023-07-08 jrick goto done;
5165 24df9a28 2023-07-08 jrick error = got_ref_resolve(&commit_id, repo, head_ref);
5166 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
5167 24df9a28 2023-07-08 jrick if (error != NULL)
5168 24df9a28 2023-07-08 jrick goto done;
5169 24df9a28 2023-07-08 jrick } else {
5170 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
5171 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
5172 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
5173 24df9a28 2023-07-08 jrick NULL);
5174 24df9a28 2023-07-08 jrick if (error)
5175 24df9a28 2023-07-08 jrick goto done;
5176 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
5177 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
5178 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
5179 24df9a28 2023-07-08 jrick if (error)
5180 24df9a28 2023-07-08 jrick goto done;
5181 24df9a28 2023-07-08 jrick }
5182 24df9a28 2023-07-08 jrick
5183 24df9a28 2023-07-08 jrick if (worktree) {
5184 24df9a28 2023-07-08 jrick /* Release work tree lock. */
5185 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
5186 24df9a28 2023-07-08 jrick worktree = NULL;
5187 24df9a28 2023-07-08 jrick }
5188 24df9a28 2023-07-08 jrick
5189 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, commit_id);
5190 24df9a28 2023-07-08 jrick if (error)
5191 24df9a28 2023-07-08 jrick goto done;
5192 24df9a28 2023-07-08 jrick
5193 24df9a28 2023-07-08 jrick error = got_object_resolve_symlinks(&link_target, in_repo_path,
5194 24df9a28 2023-07-08 jrick commit, repo);
5195 24df9a28 2023-07-08 jrick if (error)
5196 24df9a28 2023-07-08 jrick goto done;
5197 24df9a28 2023-07-08 jrick
5198 24df9a28 2023-07-08 jrick error = got_object_id_by_path(&obj_id, repo, commit,
5199 24df9a28 2023-07-08 jrick link_target ? link_target : in_repo_path);
5200 24df9a28 2023-07-08 jrick if (error)
5201 24df9a28 2023-07-08 jrick goto done;
5202 24df9a28 2023-07-08 jrick
5203 24df9a28 2023-07-08 jrick error = got_object_get_type(&obj_type, repo, obj_id);
5204 24df9a28 2023-07-08 jrick if (error)
5205 24df9a28 2023-07-08 jrick goto done;
5206 24df9a28 2023-07-08 jrick
5207 24df9a28 2023-07-08 jrick if (obj_type != GOT_OBJ_TYPE_BLOB) {
5208 24df9a28 2023-07-08 jrick error = got_error_path(link_target ? link_target : in_repo_path,
5209 24df9a28 2023-07-08 jrick GOT_ERR_OBJ_TYPE);
5210 24df9a28 2023-07-08 jrick goto done;
5211 24df9a28 2023-07-08 jrick }
5212 24df9a28 2023-07-08 jrick
5213 24df9a28 2023-07-08 jrick error = got_object_open_as_blob(&blob, repo, obj_id, 8192, fd1);
5214 24df9a28 2023-07-08 jrick if (error)
5215 24df9a28 2023-07-08 jrick goto done;
5216 24df9a28 2023-07-08 jrick bca.f = got_opentemp();
5217 24df9a28 2023-07-08 jrick if (bca.f == NULL) {
5218 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
5219 24df9a28 2023-07-08 jrick goto done;
5220 24df9a28 2023-07-08 jrick }
5221 24df9a28 2023-07-08 jrick error = got_object_blob_dump_to_file(&filesize, &bca.nlines,
5222 24df9a28 2023-07-08 jrick &bca.line_offsets, bca.f, blob);
5223 24df9a28 2023-07-08 jrick if (error || bca.nlines == 0)
5224 24df9a28 2023-07-08 jrick goto done;
5225 24df9a28 2023-07-08 jrick
5226 24df9a28 2023-07-08 jrick /* Don't include \n at EOF in the blame line count. */
5227 24df9a28 2023-07-08 jrick if (bca.line_offsets[bca.nlines - 1] == filesize)
5228 24df9a28 2023-07-08 jrick bca.nlines--;
5229 24df9a28 2023-07-08 jrick
5230 24df9a28 2023-07-08 jrick bca.lines = calloc(bca.nlines, sizeof(*bca.lines));
5231 24df9a28 2023-07-08 jrick if (bca.lines == NULL) {
5232 24df9a28 2023-07-08 jrick error = got_error_from_errno("calloc");
5233 24df9a28 2023-07-08 jrick goto done;
5234 24df9a28 2023-07-08 jrick }
5235 24df9a28 2023-07-08 jrick bca.lineno_cur = 1;
5236 24df9a28 2023-07-08 jrick bca.nlines_prec = 0;
5237 24df9a28 2023-07-08 jrick i = bca.nlines;
5238 24df9a28 2023-07-08 jrick while (i > 0) {
5239 24df9a28 2023-07-08 jrick i /= 10;
5240 24df9a28 2023-07-08 jrick bca.nlines_prec++;
5241 24df9a28 2023-07-08 jrick }
5242 24df9a28 2023-07-08 jrick bca.repo = repo;
5243 24df9a28 2023-07-08 jrick
5244 24df9a28 2023-07-08 jrick fd2 = got_opentempfd();
5245 24df9a28 2023-07-08 jrick if (fd2 == -1) {
5246 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentempfd");
5247 24df9a28 2023-07-08 jrick goto done;
5248 24df9a28 2023-07-08 jrick }
5249 24df9a28 2023-07-08 jrick fd3 = got_opentempfd();
5250 24df9a28 2023-07-08 jrick if (fd3 == -1) {
5251 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentempfd");
5252 24df9a28 2023-07-08 jrick goto done;
5253 24df9a28 2023-07-08 jrick }
5254 24df9a28 2023-07-08 jrick f1 = got_opentemp();
5255 24df9a28 2023-07-08 jrick if (f1 == NULL) {
5256 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
5257 24df9a28 2023-07-08 jrick goto done;
5258 24df9a28 2023-07-08 jrick }
5259 24df9a28 2023-07-08 jrick f2 = got_opentemp();
5260 24df9a28 2023-07-08 jrick if (f2 == NULL) {
5261 24df9a28 2023-07-08 jrick error = got_error_from_errno("got_opentemp");
5262 24df9a28 2023-07-08 jrick goto done;
5263 24df9a28 2023-07-08 jrick }
5264 24df9a28 2023-07-08 jrick error = got_blame(link_target ? link_target : in_repo_path, commit_id,
5265 24df9a28 2023-07-08 jrick repo, GOT_DIFF_ALGORITHM_PATIENCE, blame_cb, &bca,
5266 24df9a28 2023-07-08 jrick check_cancelled, NULL, fd2, fd3, f1, f2);
5267 24df9a28 2023-07-08 jrick done:
5268 24df9a28 2023-07-08 jrick free(in_repo_path);
5269 24df9a28 2023-07-08 jrick free(link_target);
5270 24df9a28 2023-07-08 jrick free(repo_path);
5271 24df9a28 2023-07-08 jrick free(cwd);
5272 24df9a28 2023-07-08 jrick free(commit_id);
5273 24df9a28 2023-07-08 jrick free(obj_id);
5274 24df9a28 2023-07-08 jrick if (commit)
5275 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
5276 24df9a28 2023-07-08 jrick
5277 24df9a28 2023-07-08 jrick if (fd1 != -1 && close(fd1) == -1 && error == NULL)
5278 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
5279 24df9a28 2023-07-08 jrick if (fd2 != -1 && close(fd2) == -1 && error == NULL)
5280 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
5281 24df9a28 2023-07-08 jrick if (fd3 != -1 && close(fd3) == -1 && error == NULL)
5282 24df9a28 2023-07-08 jrick error = got_error_from_errno("close");
5283 24df9a28 2023-07-08 jrick if (f1 && fclose(f1) == EOF && error == NULL)
5284 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
5285 24df9a28 2023-07-08 jrick if (f2 && fclose(f2) == EOF && error == NULL)
5286 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
5287 24df9a28 2023-07-08 jrick
5288 24df9a28 2023-07-08 jrick if (blob)
5289 24df9a28 2023-07-08 jrick got_object_blob_close(blob);
5290 24df9a28 2023-07-08 jrick if (worktree)
5291 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
5292 24df9a28 2023-07-08 jrick if (repo) {
5293 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
5294 24df9a28 2023-07-08 jrick if (error == NULL)
5295 24df9a28 2023-07-08 jrick error = close_err;
5296 24df9a28 2023-07-08 jrick }
5297 24df9a28 2023-07-08 jrick if (pack_fds) {
5298 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
5299 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
5300 24df9a28 2023-07-08 jrick if (error == NULL)
5301 24df9a28 2023-07-08 jrick error = pack_err;
5302 24df9a28 2023-07-08 jrick }
5303 24df9a28 2023-07-08 jrick if (bca.lines) {
5304 24df9a28 2023-07-08 jrick for (i = 0; i < bca.nlines; i++) {
5305 24df9a28 2023-07-08 jrick struct blame_line *bline = &bca.lines[i];
5306 24df9a28 2023-07-08 jrick free(bline->id_str);
5307 24df9a28 2023-07-08 jrick free(bline->committer);
5308 24df9a28 2023-07-08 jrick }
5309 24df9a28 2023-07-08 jrick free(bca.lines);
5310 24df9a28 2023-07-08 jrick }
5311 24df9a28 2023-07-08 jrick free(bca.line_offsets);
5312 24df9a28 2023-07-08 jrick if (bca.f && fclose(bca.f) == EOF && error == NULL)
5313 24df9a28 2023-07-08 jrick error = got_error_from_errno("fclose");
5314 24df9a28 2023-07-08 jrick return error;
5315 24df9a28 2023-07-08 jrick }
5316 24df9a28 2023-07-08 jrick
5317 24df9a28 2023-07-08 jrick __dead static void
5318 24df9a28 2023-07-08 jrick usage_tree(void)
5319 24df9a28 2023-07-08 jrick {
5320 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s tree [-iR] [-c commit] [-r repository-path] "
5321 24df9a28 2023-07-08 jrick "[path]\n", getprogname());
5322 24df9a28 2023-07-08 jrick exit(1);
5323 24df9a28 2023-07-08 jrick }
5324 24df9a28 2023-07-08 jrick
5325 24df9a28 2023-07-08 jrick static const struct got_error *
5326 24df9a28 2023-07-08 jrick print_entry(struct got_tree_entry *te, const char *id, const char *path,
5327 24df9a28 2023-07-08 jrick const char *root_path, struct got_repository *repo)
5328 24df9a28 2023-07-08 jrick {
5329 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
5330 24df9a28 2023-07-08 jrick int is_root_path = (strcmp(path, root_path) == 0);
5331 24df9a28 2023-07-08 jrick const char *modestr = "";
5332 24df9a28 2023-07-08 jrick mode_t mode = got_tree_entry_get_mode(te);
5333 24df9a28 2023-07-08 jrick char *link_target = NULL;
5334 24df9a28 2023-07-08 jrick
5335 24df9a28 2023-07-08 jrick path += strlen(root_path);
5336 24df9a28 2023-07-08 jrick while (path[0] == '/')
5337 24df9a28 2023-07-08 jrick path++;
5338 24df9a28 2023-07-08 jrick
5339 24df9a28 2023-07-08 jrick if (got_object_tree_entry_is_submodule(te))
5340 24df9a28 2023-07-08 jrick modestr = "$";
5341 24df9a28 2023-07-08 jrick else if (S_ISLNK(mode)) {
5342 24df9a28 2023-07-08 jrick int i;
5343 24df9a28 2023-07-08 jrick
5344 24df9a28 2023-07-08 jrick err = got_tree_entry_get_symlink_target(&link_target, te, repo);
5345 24df9a28 2023-07-08 jrick if (err)
5346 24df9a28 2023-07-08 jrick return err;
5347 24df9a28 2023-07-08 jrick for (i = 0; link_target[i] != '\0'; i++) {
5348 24df9a28 2023-07-08 jrick if (!isprint((unsigned char)link_target[i]))
5349 24df9a28 2023-07-08 jrick link_target[i] = '?';
5350 24df9a28 2023-07-08 jrick }
5351 24df9a28 2023-07-08 jrick
5352 24df9a28 2023-07-08 jrick modestr = "@";
5353 24df9a28 2023-07-08 jrick }
5354 24df9a28 2023-07-08 jrick else if (S_ISDIR(mode))
5355 24df9a28 2023-07-08 jrick modestr = "/";
5356 24df9a28 2023-07-08 jrick else if (mode & S_IXUSR)
5357 24df9a28 2023-07-08 jrick modestr = "*";
5358 24df9a28 2023-07-08 jrick
5359 24df9a28 2023-07-08 jrick printf("%s%s%s%s%s%s%s\n", id ? id : "", path,
5360 24df9a28 2023-07-08 jrick is_root_path ? "" : "/", got_tree_entry_get_name(te), modestr,
5361 24df9a28 2023-07-08 jrick link_target ? " -> ": "", link_target ? link_target : "");
5362 24df9a28 2023-07-08 jrick
5363 24df9a28 2023-07-08 jrick free(link_target);
5364 24df9a28 2023-07-08 jrick return NULL;
5365 24df9a28 2023-07-08 jrick }
5366 24df9a28 2023-07-08 jrick
5367 24df9a28 2023-07-08 jrick static const struct got_error *
5368 24df9a28 2023-07-08 jrick print_tree(const char *path, struct got_commit_object *commit,
5369 24df9a28 2023-07-08 jrick int show_ids, int recurse, const char *root_path,
5370 24df9a28 2023-07-08 jrick struct got_repository *repo)
5371 24df9a28 2023-07-08 jrick {
5372 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
5373 24df9a28 2023-07-08 jrick struct got_object_id *tree_id = NULL;
5374 24df9a28 2023-07-08 jrick struct got_tree_object *tree = NULL;
5375 24df9a28 2023-07-08 jrick int nentries, i;
5376 24df9a28 2023-07-08 jrick
5377 24df9a28 2023-07-08 jrick err = got_object_id_by_path(&tree_id, repo, commit, path);
5378 24df9a28 2023-07-08 jrick if (err)
5379 24df9a28 2023-07-08 jrick goto done;
5380 24df9a28 2023-07-08 jrick
5381 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree, repo, tree_id);
5382 24df9a28 2023-07-08 jrick if (err)
5383 24df9a28 2023-07-08 jrick goto done;
5384 24df9a28 2023-07-08 jrick nentries = got_object_tree_get_nentries(tree);
5385 24df9a28 2023-07-08 jrick for (i = 0; i < nentries; i++) {
5386 24df9a28 2023-07-08 jrick struct got_tree_entry *te;
5387 24df9a28 2023-07-08 jrick char *id = NULL;
5388 24df9a28 2023-07-08 jrick
5389 24df9a28 2023-07-08 jrick if (sigint_received || sigpipe_received)
5390 24df9a28 2023-07-08 jrick break;
5391 24df9a28 2023-07-08 jrick
5392 24df9a28 2023-07-08 jrick te = got_object_tree_get_entry(tree, i);
5393 24df9a28 2023-07-08 jrick if (show_ids) {
5394 24df9a28 2023-07-08 jrick char *id_str;
5395 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str,
5396 24df9a28 2023-07-08 jrick got_tree_entry_get_id(te));
5397 24df9a28 2023-07-08 jrick if (err)
5398 24df9a28 2023-07-08 jrick goto done;
5399 24df9a28 2023-07-08 jrick if (asprintf(&id, "%s ", id_str) == -1) {
5400 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
5401 24df9a28 2023-07-08 jrick free(id_str);
5402 24df9a28 2023-07-08 jrick goto done;
5403 24df9a28 2023-07-08 jrick }
5404 24df9a28 2023-07-08 jrick free(id_str);
5405 24df9a28 2023-07-08 jrick }
5406 24df9a28 2023-07-08 jrick err = print_entry(te, id, path, root_path, repo);
5407 24df9a28 2023-07-08 jrick free(id);
5408 24df9a28 2023-07-08 jrick if (err)
5409 24df9a28 2023-07-08 jrick goto done;
5410 24df9a28 2023-07-08 jrick
5411 24df9a28 2023-07-08 jrick if (recurse && S_ISDIR(got_tree_entry_get_mode(te))) {
5412 24df9a28 2023-07-08 jrick char *child_path;
5413 24df9a28 2023-07-08 jrick if (asprintf(&child_path, "%s%s%s", path,
5414 24df9a28 2023-07-08 jrick path[0] == '/' && path[1] == '\0' ? "" : "/",
5415 24df9a28 2023-07-08 jrick got_tree_entry_get_name(te)) == -1) {
5416 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
5417 24df9a28 2023-07-08 jrick goto done;
5418 24df9a28 2023-07-08 jrick }
5419 24df9a28 2023-07-08 jrick err = print_tree(child_path, commit, show_ids, 1,
5420 24df9a28 2023-07-08 jrick root_path, repo);
5421 24df9a28 2023-07-08 jrick free(child_path);
5422 24df9a28 2023-07-08 jrick if (err)
5423 24df9a28 2023-07-08 jrick goto done;
5424 24df9a28 2023-07-08 jrick }
5425 24df9a28 2023-07-08 jrick }
5426 24df9a28 2023-07-08 jrick done:
5427 24df9a28 2023-07-08 jrick if (tree)
5428 24df9a28 2023-07-08 jrick got_object_tree_close(tree);
5429 24df9a28 2023-07-08 jrick free(tree_id);
5430 24df9a28 2023-07-08 jrick return err;
5431 24df9a28 2023-07-08 jrick }
5432 24df9a28 2023-07-08 jrick
5433 24df9a28 2023-07-08 jrick static const struct got_error *
5434 24df9a28 2023-07-08 jrick cmd_tree(int argc, char *argv[])
5435 24df9a28 2023-07-08 jrick {
5436 24df9a28 2023-07-08 jrick const struct got_error *error;
5437 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
5438 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
5439 24df9a28 2023-07-08 jrick const char *path, *refname = NULL;
5440 24df9a28 2023-07-08 jrick char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5441 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
5442 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
5443 24df9a28 2023-07-08 jrick char *commit_id_str = NULL;
5444 24df9a28 2023-07-08 jrick int show_ids = 0, recurse = 0;
5445 24df9a28 2023-07-08 jrick int ch;
5446 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
5447 24df9a28 2023-07-08 jrick
5448 24df9a28 2023-07-08 jrick #ifndef PROFILE
5449 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
5450 24df9a28 2023-07-08 jrick NULL) == -1)
5451 24df9a28 2023-07-08 jrick err(1, "pledge");
5452 24df9a28 2023-07-08 jrick #endif
5453 24df9a28 2023-07-08 jrick
5454 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:iRr:")) != -1) {
5455 24df9a28 2023-07-08 jrick switch (ch) {
5456 24df9a28 2023-07-08 jrick case 'c':
5457 24df9a28 2023-07-08 jrick commit_id_str = optarg;
5458 24df9a28 2023-07-08 jrick break;
5459 24df9a28 2023-07-08 jrick case 'i':
5460 24df9a28 2023-07-08 jrick show_ids = 1;
5461 24df9a28 2023-07-08 jrick break;
5462 24df9a28 2023-07-08 jrick case 'R':
5463 24df9a28 2023-07-08 jrick recurse = 1;
5464 24df9a28 2023-07-08 jrick break;
5465 24df9a28 2023-07-08 jrick case 'r':
5466 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
5467 24df9a28 2023-07-08 jrick if (repo_path == NULL)
5468 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
5469 24df9a28 2023-07-08 jrick optarg);
5470 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
5471 24df9a28 2023-07-08 jrick break;
5472 24df9a28 2023-07-08 jrick default:
5473 24df9a28 2023-07-08 jrick usage_tree();
5474 24df9a28 2023-07-08 jrick /* NOTREACHED */
5475 24df9a28 2023-07-08 jrick }
5476 24df9a28 2023-07-08 jrick }
5477 24df9a28 2023-07-08 jrick
5478 24df9a28 2023-07-08 jrick argc -= optind;
5479 24df9a28 2023-07-08 jrick argv += optind;
5480 24df9a28 2023-07-08 jrick
5481 24df9a28 2023-07-08 jrick if (argc == 1)
5482 24df9a28 2023-07-08 jrick path = argv[0];
5483 24df9a28 2023-07-08 jrick else if (argc > 1)
5484 24df9a28 2023-07-08 jrick usage_tree();
5485 24df9a28 2023-07-08 jrick else
5486 24df9a28 2023-07-08 jrick path = NULL;
5487 24df9a28 2023-07-08 jrick
5488 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
5489 24df9a28 2023-07-08 jrick if (cwd == NULL) {
5490 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
5491 24df9a28 2023-07-08 jrick goto done;
5492 24df9a28 2023-07-08 jrick }
5493 24df9a28 2023-07-08 jrick
5494 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
5495 24df9a28 2023-07-08 jrick if (error != NULL)
5496 24df9a28 2023-07-08 jrick goto done;
5497 24df9a28 2023-07-08 jrick
5498 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
5499 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
5500 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
5501 24df9a28 2023-07-08 jrick goto done;
5502 24df9a28 2023-07-08 jrick else
5503 24df9a28 2023-07-08 jrick error = NULL;
5504 24df9a28 2023-07-08 jrick if (worktree) {
5505 24df9a28 2023-07-08 jrick repo_path =
5506 24df9a28 2023-07-08 jrick strdup(got_worktree_get_repo_path(worktree));
5507 24df9a28 2023-07-08 jrick if (repo_path == NULL)
5508 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
5509 24df9a28 2023-07-08 jrick if (error)
5510 24df9a28 2023-07-08 jrick goto done;
5511 24df9a28 2023-07-08 jrick } else {
5512 24df9a28 2023-07-08 jrick repo_path = strdup(cwd);
5513 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
5514 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
5515 24df9a28 2023-07-08 jrick goto done;
5516 24df9a28 2023-07-08 jrick }
5517 24df9a28 2023-07-08 jrick }
5518 24df9a28 2023-07-08 jrick }
5519 24df9a28 2023-07-08 jrick
5520 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
5521 24df9a28 2023-07-08 jrick if (error != NULL)
5522 24df9a28 2023-07-08 jrick goto done;
5523 24df9a28 2023-07-08 jrick
5524 24df9a28 2023-07-08 jrick if (worktree) {
5525 24df9a28 2023-07-08 jrick const char *prefix = got_worktree_get_path_prefix(worktree);
5526 24df9a28 2023-07-08 jrick char *p;
5527 24df9a28 2023-07-08 jrick
5528 24df9a28 2023-07-08 jrick if (path == NULL || got_path_is_root_dir(path))
5529 24df9a28 2023-07-08 jrick path = "";
5530 24df9a28 2023-07-08 jrick error = got_worktree_resolve_path(&p, worktree, path);
5531 24df9a28 2023-07-08 jrick if (error)
5532 24df9a28 2023-07-08 jrick goto done;
5533 24df9a28 2023-07-08 jrick if (asprintf(&in_repo_path, "%s%s%s", prefix,
5534 24df9a28 2023-07-08 jrick (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
5535 24df9a28 2023-07-08 jrick p) == -1) {
5536 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
5537 24df9a28 2023-07-08 jrick free(p);
5538 24df9a28 2023-07-08 jrick goto done;
5539 24df9a28 2023-07-08 jrick }
5540 24df9a28 2023-07-08 jrick free(p);
5541 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5542 24df9a28 2023-07-08 jrick if (error)
5543 24df9a28 2023-07-08 jrick goto done;
5544 24df9a28 2023-07-08 jrick } else {
5545 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
5546 24df9a28 2023-07-08 jrick if (error)
5547 24df9a28 2023-07-08 jrick goto done;
5548 24df9a28 2023-07-08 jrick if (path == NULL)
5549 24df9a28 2023-07-08 jrick path = "/";
5550 24df9a28 2023-07-08 jrick error = got_repo_map_path(&in_repo_path, repo, path);
5551 24df9a28 2023-07-08 jrick if (error != NULL)
5552 24df9a28 2023-07-08 jrick goto done;
5553 24df9a28 2023-07-08 jrick }
5554 24df9a28 2023-07-08 jrick
5555 24df9a28 2023-07-08 jrick if (commit_id_str == NULL) {
5556 24df9a28 2023-07-08 jrick struct got_reference *head_ref;
5557 24df9a28 2023-07-08 jrick if (worktree)
5558 24df9a28 2023-07-08 jrick refname = got_worktree_get_head_ref_name(worktree);
5559 24df9a28 2023-07-08 jrick else
5560 24df9a28 2023-07-08 jrick refname = GOT_REF_HEAD;
5561 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo, refname, 0);
5562 24df9a28 2023-07-08 jrick if (error != NULL)
5563 24df9a28 2023-07-08 jrick goto done;
5564 24df9a28 2023-07-08 jrick error = got_ref_resolve(&commit_id, repo, head_ref);
5565 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
5566 24df9a28 2023-07-08 jrick if (error != NULL)
5567 24df9a28 2023-07-08 jrick goto done;
5568 24df9a28 2023-07-08 jrick } else {
5569 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
5570 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
5571 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name,
5572 24df9a28 2023-07-08 jrick NULL);
5573 24df9a28 2023-07-08 jrick if (error)
5574 24df9a28 2023-07-08 jrick goto done;
5575 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
5576 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
5577 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
5578 24df9a28 2023-07-08 jrick if (error)
5579 24df9a28 2023-07-08 jrick goto done;
5580 24df9a28 2023-07-08 jrick }
5581 24df9a28 2023-07-08 jrick
5582 24df9a28 2023-07-08 jrick if (worktree) {
5583 24df9a28 2023-07-08 jrick /* Release work tree lock. */
5584 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
5585 24df9a28 2023-07-08 jrick worktree = NULL;
5586 24df9a28 2023-07-08 jrick }
5587 24df9a28 2023-07-08 jrick
5588 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, commit_id);
5589 24df9a28 2023-07-08 jrick if (error)
5590 24df9a28 2023-07-08 jrick goto done;
5591 24df9a28 2023-07-08 jrick
5592 24df9a28 2023-07-08 jrick error = print_tree(in_repo_path, commit, show_ids, recurse,
5593 24df9a28 2023-07-08 jrick in_repo_path, repo);
5594 24df9a28 2023-07-08 jrick done:
5595 24df9a28 2023-07-08 jrick free(in_repo_path);
5596 24df9a28 2023-07-08 jrick free(repo_path);
5597 24df9a28 2023-07-08 jrick free(cwd);
5598 24df9a28 2023-07-08 jrick free(commit_id);
5599 24df9a28 2023-07-08 jrick if (commit)
5600 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
5601 24df9a28 2023-07-08 jrick if (worktree)
5602 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
5603 24df9a28 2023-07-08 jrick if (repo) {
5604 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
5605 24df9a28 2023-07-08 jrick if (error == NULL)
5606 24df9a28 2023-07-08 jrick error = close_err;
5607 24df9a28 2023-07-08 jrick }
5608 24df9a28 2023-07-08 jrick if (pack_fds) {
5609 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
5610 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
5611 24df9a28 2023-07-08 jrick if (error == NULL)
5612 24df9a28 2023-07-08 jrick error = pack_err;
5613 24df9a28 2023-07-08 jrick }
5614 24df9a28 2023-07-08 jrick return error;
5615 24df9a28 2023-07-08 jrick }
5616 24df9a28 2023-07-08 jrick
5617 24df9a28 2023-07-08 jrick __dead static void
5618 24df9a28 2023-07-08 jrick usage_status(void)
5619 24df9a28 2023-07-08 jrick {
5620 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s status [-I] [-S status-codes] "
5621 24df9a28 2023-07-08 jrick "[-s status-codes] [path ...]\n", getprogname());
5622 24df9a28 2023-07-08 jrick exit(1);
5623 24df9a28 2023-07-08 jrick }
5624 24df9a28 2023-07-08 jrick
5625 24df9a28 2023-07-08 jrick struct got_status_arg {
5626 24df9a28 2023-07-08 jrick char *status_codes;
5627 24df9a28 2023-07-08 jrick int suppress;
5628 24df9a28 2023-07-08 jrick };
5629 24df9a28 2023-07-08 jrick
5630 24df9a28 2023-07-08 jrick static const struct got_error *
5631 24df9a28 2023-07-08 jrick print_status(void *arg, unsigned char status, unsigned char staged_status,
5632 24df9a28 2023-07-08 jrick const char *path, struct got_object_id *blob_id,
5633 24df9a28 2023-07-08 jrick struct got_object_id *staged_blob_id, struct got_object_id *commit_id,
5634 24df9a28 2023-07-08 jrick int dirfd, const char *de_name)
5635 24df9a28 2023-07-08 jrick {
5636 24df9a28 2023-07-08 jrick struct got_status_arg *st = arg;
5637 24df9a28 2023-07-08 jrick
5638 24df9a28 2023-07-08 jrick if (status == staged_status && (status == GOT_STATUS_DELETE))
5639 24df9a28 2023-07-08 jrick status = GOT_STATUS_NO_CHANGE;
5640 24df9a28 2023-07-08 jrick if (st != NULL && st->status_codes) {
5641 24df9a28 2023-07-08 jrick size_t ncodes = strlen(st->status_codes);
5642 24df9a28 2023-07-08 jrick int i, j = 0;
5643 24df9a28 2023-07-08 jrick
5644 24df9a28 2023-07-08 jrick for (i = 0; i < ncodes ; i++) {
5645 24df9a28 2023-07-08 jrick if (st->suppress) {
5646 24df9a28 2023-07-08 jrick if (status == st->status_codes[i] ||
5647 24df9a28 2023-07-08 jrick staged_status == st->status_codes[i]) {
5648 24df9a28 2023-07-08 jrick j++;
5649 24df9a28 2023-07-08 jrick continue;
5650 24df9a28 2023-07-08 jrick }
5651 24df9a28 2023-07-08 jrick } else {
5652 24df9a28 2023-07-08 jrick if (status == st->status_codes[i] ||
5653 24df9a28 2023-07-08 jrick staged_status == st->status_codes[i])
5654 24df9a28 2023-07-08 jrick break;
5655 24df9a28 2023-07-08 jrick }
5656 24df9a28 2023-07-08 jrick }
5657 24df9a28 2023-07-08 jrick
5658 24df9a28 2023-07-08 jrick if (st->suppress && j == 0)
5659 24df9a28 2023-07-08 jrick goto print;
5660 24df9a28 2023-07-08 jrick
5661 24df9a28 2023-07-08 jrick if (i == ncodes)
5662 24df9a28 2023-07-08 jrick return NULL;
5663 24df9a28 2023-07-08 jrick }
5664 24df9a28 2023-07-08 jrick print:
5665 24df9a28 2023-07-08 jrick printf("%c%c %s\n", status, staged_status, path);
5666 24df9a28 2023-07-08 jrick return NULL;
5667 24df9a28 2023-07-08 jrick }
5668 24df9a28 2023-07-08 jrick
5669 24df9a28 2023-07-08 jrick static const struct got_error *
5670 24df9a28 2023-07-08 jrick cmd_status(int argc, char *argv[])
5671 24df9a28 2023-07-08 jrick {
5672 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
5673 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
5674 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
5675 24df9a28 2023-07-08 jrick struct got_status_arg st;
5676 24df9a28 2023-07-08 jrick char *cwd = NULL;
5677 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
5678 24df9a28 2023-07-08 jrick int ch, i, no_ignores = 0;
5679 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
5680 24df9a28 2023-07-08 jrick
5681 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
5682 24df9a28 2023-07-08 jrick
5683 24df9a28 2023-07-08 jrick memset(&st, 0, sizeof(st));
5684 24df9a28 2023-07-08 jrick st.status_codes = NULL;
5685 24df9a28 2023-07-08 jrick st.suppress = 0;
5686 24df9a28 2023-07-08 jrick
5687 24df9a28 2023-07-08 jrick #ifndef PROFILE
5688 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
5689 24df9a28 2023-07-08 jrick NULL) == -1)
5690 24df9a28 2023-07-08 jrick err(1, "pledge");
5691 24df9a28 2023-07-08 jrick #endif
5692 24df9a28 2023-07-08 jrick
5693 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "IS:s:")) != -1) {
5694 24df9a28 2023-07-08 jrick switch (ch) {
5695 24df9a28 2023-07-08 jrick case 'I':
5696 24df9a28 2023-07-08 jrick no_ignores = 1;
5697 24df9a28 2023-07-08 jrick break;
5698 24df9a28 2023-07-08 jrick case 'S':
5699 24df9a28 2023-07-08 jrick if (st.status_codes != NULL && st.suppress == 0)
5700 24df9a28 2023-07-08 jrick option_conflict('S', 's');
5701 24df9a28 2023-07-08 jrick st.suppress = 1;
5702 24df9a28 2023-07-08 jrick /* fallthrough */
5703 24df9a28 2023-07-08 jrick case 's':
5704 24df9a28 2023-07-08 jrick for (i = 0; optarg[i] != '\0'; i++) {
5705 24df9a28 2023-07-08 jrick switch (optarg[i]) {
5706 24df9a28 2023-07-08 jrick case GOT_STATUS_MODIFY:
5707 24df9a28 2023-07-08 jrick case GOT_STATUS_ADD:
5708 24df9a28 2023-07-08 jrick case GOT_STATUS_DELETE:
5709 24df9a28 2023-07-08 jrick case GOT_STATUS_CONFLICT:
5710 24df9a28 2023-07-08 jrick case GOT_STATUS_MISSING:
5711 24df9a28 2023-07-08 jrick case GOT_STATUS_OBSTRUCTED:
5712 24df9a28 2023-07-08 jrick case GOT_STATUS_UNVERSIONED:
5713 24df9a28 2023-07-08 jrick case GOT_STATUS_MODE_CHANGE:
5714 24df9a28 2023-07-08 jrick case GOT_STATUS_NONEXISTENT:
5715 24df9a28 2023-07-08 jrick break;
5716 24df9a28 2023-07-08 jrick default:
5717 24df9a28 2023-07-08 jrick errx(1, "invalid status code '%c'",
5718 24df9a28 2023-07-08 jrick optarg[i]);
5719 24df9a28 2023-07-08 jrick }
5720 24df9a28 2023-07-08 jrick }
5721 24df9a28 2023-07-08 jrick if (ch == 's' && st.suppress)
5722 24df9a28 2023-07-08 jrick option_conflict('s', 'S');
5723 24df9a28 2023-07-08 jrick st.status_codes = optarg;
5724 24df9a28 2023-07-08 jrick break;
5725 24df9a28 2023-07-08 jrick default:
5726 24df9a28 2023-07-08 jrick usage_status();
5727 24df9a28 2023-07-08 jrick /* NOTREACHED */
5728 24df9a28 2023-07-08 jrick }
5729 24df9a28 2023-07-08 jrick }
5730 24df9a28 2023-07-08 jrick
5731 24df9a28 2023-07-08 jrick argc -= optind;
5732 24df9a28 2023-07-08 jrick argv += optind;
5733 24df9a28 2023-07-08 jrick
5734 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
5735 24df9a28 2023-07-08 jrick if (cwd == NULL) {
5736 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
5737 24df9a28 2023-07-08 jrick goto done;
5738 24df9a28 2023-07-08 jrick }
5739 24df9a28 2023-07-08 jrick
5740 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
5741 24df9a28 2023-07-08 jrick if (error != NULL)
5742 24df9a28 2023-07-08 jrick goto done;
5743 24df9a28 2023-07-08 jrick
5744 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
5745 24df9a28 2023-07-08 jrick if (error) {
5746 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
5747 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "status", cwd);
5748 24df9a28 2023-07-08 jrick goto done;
5749 24df9a28 2023-07-08 jrick }
5750 24df9a28 2023-07-08 jrick
5751 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
5752 24df9a28 2023-07-08 jrick NULL, pack_fds);
5753 24df9a28 2023-07-08 jrick if (error != NULL)
5754 24df9a28 2023-07-08 jrick goto done;
5755 24df9a28 2023-07-08 jrick
5756 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1,
5757 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
5758 24df9a28 2023-07-08 jrick if (error)
5759 24df9a28 2023-07-08 jrick goto done;
5760 24df9a28 2023-07-08 jrick
5761 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
5762 24df9a28 2023-07-08 jrick if (error)
5763 24df9a28 2023-07-08 jrick goto done;
5764 24df9a28 2023-07-08 jrick
5765 24df9a28 2023-07-08 jrick error = got_worktree_status(worktree, &paths, repo, no_ignores,
5766 24df9a28 2023-07-08 jrick print_status, &st, check_cancelled, NULL);
5767 24df9a28 2023-07-08 jrick done:
5768 24df9a28 2023-07-08 jrick if (pack_fds) {
5769 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
5770 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
5771 24df9a28 2023-07-08 jrick if (error == NULL)
5772 24df9a28 2023-07-08 jrick error = pack_err;
5773 24df9a28 2023-07-08 jrick }
5774 24df9a28 2023-07-08 jrick if (repo) {
5775 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
5776 24df9a28 2023-07-08 jrick if (error == NULL)
5777 24df9a28 2023-07-08 jrick error = close_err;
5778 24df9a28 2023-07-08 jrick }
5779 24df9a28 2023-07-08 jrick
5780 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
5781 24df9a28 2023-07-08 jrick free(cwd);
5782 24df9a28 2023-07-08 jrick return error;
5783 24df9a28 2023-07-08 jrick }
5784 24df9a28 2023-07-08 jrick
5785 24df9a28 2023-07-08 jrick __dead static void
5786 24df9a28 2023-07-08 jrick usage_tag(void)
5787 24df9a28 2023-07-08 jrick {
5788 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s tag [-lVv] [-c commit] [-m message] "
5789 24df9a28 2023-07-08 jrick "[-r repository-path] [-s signer-id] name\n", getprogname());
5790 24df9a28 2023-07-08 jrick exit(1);
5791 24df9a28 2023-07-08 jrick }
5792 24df9a28 2023-07-08 jrick
5793 24df9a28 2023-07-08 jrick #if 0
5794 24df9a28 2023-07-08 jrick static const struct got_error *
5795 24df9a28 2023-07-08 jrick sort_tags(struct got_reflist_head *sorted, struct got_reflist_head *tags)
5796 24df9a28 2023-07-08 jrick {
5797 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
5798 24df9a28 2023-07-08 jrick struct got_reflist_entry *re, *se, *new;
5799 24df9a28 2023-07-08 jrick struct got_object_id *re_id, *se_id;
5800 24df9a28 2023-07-08 jrick struct got_tag_object *re_tag, *se_tag;
5801 24df9a28 2023-07-08 jrick time_t re_time, se_time;
5802 24df9a28 2023-07-08 jrick
5803 24df9a28 2023-07-08 jrick STAILQ_FOREACH(re, tags, entry) {
5804 24df9a28 2023-07-08 jrick se = STAILQ_FIRST(sorted);
5805 24df9a28 2023-07-08 jrick if (se == NULL) {
5806 24df9a28 2023-07-08 jrick err = got_reflist_entry_dup(&new, re);
5807 24df9a28 2023-07-08 jrick if (err)
5808 24df9a28 2023-07-08 jrick return err;
5809 24df9a28 2023-07-08 jrick STAILQ_INSERT_HEAD(sorted, new, entry);
5810 24df9a28 2023-07-08 jrick continue;
5811 24df9a28 2023-07-08 jrick } else {
5812 24df9a28 2023-07-08 jrick err = got_ref_resolve(&re_id, repo, re->ref);
5813 24df9a28 2023-07-08 jrick if (err)
5814 24df9a28 2023-07-08 jrick break;
5815 24df9a28 2023-07-08 jrick err = got_object_open_as_tag(&re_tag, repo, re_id);
5816 24df9a28 2023-07-08 jrick free(re_id);
5817 24df9a28 2023-07-08 jrick if (err)
5818 24df9a28 2023-07-08 jrick break;
5819 24df9a28 2023-07-08 jrick re_time = got_object_tag_get_tagger_time(re_tag);
5820 24df9a28 2023-07-08 jrick got_object_tag_close(re_tag);
5821 24df9a28 2023-07-08 jrick }
5822 24df9a28 2023-07-08 jrick
5823 24df9a28 2023-07-08 jrick while (se) {
5824 24df9a28 2023-07-08 jrick err = got_ref_resolve(&se_id, repo, re->ref);
5825 24df9a28 2023-07-08 jrick if (err)
5826 24df9a28 2023-07-08 jrick break;
5827 24df9a28 2023-07-08 jrick err = got_object_open_as_tag(&se_tag, repo, se_id);
5828 24df9a28 2023-07-08 jrick free(se_id);
5829 24df9a28 2023-07-08 jrick if (err)
5830 24df9a28 2023-07-08 jrick break;
5831 24df9a28 2023-07-08 jrick se_time = got_object_tag_get_tagger_time(se_tag);
5832 24df9a28 2023-07-08 jrick got_object_tag_close(se_tag);
5833 24df9a28 2023-07-08 jrick
5834 24df9a28 2023-07-08 jrick if (se_time > re_time) {
5835 24df9a28 2023-07-08 jrick err = got_reflist_entry_dup(&new, re);
5836 24df9a28 2023-07-08 jrick if (err)
5837 24df9a28 2023-07-08 jrick return err;
5838 24df9a28 2023-07-08 jrick STAILQ_INSERT_AFTER(sorted, se, new, entry);
5839 24df9a28 2023-07-08 jrick break;
5840 24df9a28 2023-07-08 jrick }
5841 24df9a28 2023-07-08 jrick se = STAILQ_NEXT(se, entry);
5842 24df9a28 2023-07-08 jrick continue;
5843 24df9a28 2023-07-08 jrick }
5844 24df9a28 2023-07-08 jrick }
5845 24df9a28 2023-07-08 jrick done:
5846 24df9a28 2023-07-08 jrick return err;
5847 24df9a28 2023-07-08 jrick }
5848 24df9a28 2023-07-08 jrick #endif
5849 24df9a28 2023-07-08 jrick
5850 24df9a28 2023-07-08 jrick static const struct got_error *
5851 24df9a28 2023-07-08 jrick get_tag_refname(char **refname, const char *tag_name)
5852 24df9a28 2023-07-08 jrick {
5853 24df9a28 2023-07-08 jrick const struct got_error *err;
5854 24df9a28 2023-07-08 jrick
5855 24df9a28 2023-07-08 jrick if (strncmp("refs/tags/", tag_name, 10) == 0) {
5856 24df9a28 2023-07-08 jrick *refname = strdup(tag_name);
5857 24df9a28 2023-07-08 jrick if (*refname == NULL)
5858 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
5859 24df9a28 2023-07-08 jrick } else if (asprintf(refname, "refs/tags/%s", tag_name) == -1) {
5860 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
5861 24df9a28 2023-07-08 jrick *refname = NULL;
5862 24df9a28 2023-07-08 jrick return err;
5863 24df9a28 2023-07-08 jrick }
5864 24df9a28 2023-07-08 jrick
5865 24df9a28 2023-07-08 jrick return NULL;
5866 24df9a28 2023-07-08 jrick }
5867 24df9a28 2023-07-08 jrick
5868 24df9a28 2023-07-08 jrick static const struct got_error *
5869 24df9a28 2023-07-08 jrick list_tags(struct got_repository *repo, const char *tag_name, int verify_tags,
5870 24df9a28 2023-07-08 jrick const char *allowed_signers, const char *revoked_signers, int verbosity)
5871 24df9a28 2023-07-08 jrick {
5872 24df9a28 2023-07-08 jrick static const struct got_error *err = NULL;
5873 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
5874 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
5875 24df9a28 2023-07-08 jrick char *wanted_refname = NULL;
5876 24df9a28 2023-07-08 jrick int bad_sigs = 0;
5877 24df9a28 2023-07-08 jrick
5878 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
5879 24df9a28 2023-07-08 jrick
5880 24df9a28 2023-07-08 jrick err = got_ref_list(&refs, repo, "refs/tags", got_ref_cmp_tags, repo);
5881 24df9a28 2023-07-08 jrick if (err)
5882 24df9a28 2023-07-08 jrick return err;
5883 24df9a28 2023-07-08 jrick
5884 24df9a28 2023-07-08 jrick if (tag_name) {
5885 24df9a28 2023-07-08 jrick struct got_reference *ref;
5886 24df9a28 2023-07-08 jrick err = get_tag_refname(&wanted_refname, tag_name);
5887 24df9a28 2023-07-08 jrick if (err)
5888 24df9a28 2023-07-08 jrick goto done;
5889 24df9a28 2023-07-08 jrick /* Wanted tag reference should exist. */
5890 24df9a28 2023-07-08 jrick err = got_ref_open(&ref, repo, wanted_refname, 0);
5891 24df9a28 2023-07-08 jrick if (err)
5892 24df9a28 2023-07-08 jrick goto done;
5893 24df9a28 2023-07-08 jrick got_ref_close(ref);
5894 24df9a28 2023-07-08 jrick }
5895 24df9a28 2023-07-08 jrick
5896 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &refs, entry) {
5897 24df9a28 2023-07-08 jrick const char *refname;
5898 24df9a28 2023-07-08 jrick char *refstr, *tagmsg0, *tagmsg, *line, *id_str, *datestr;
5899 24df9a28 2023-07-08 jrick char datebuf[26];
5900 24df9a28 2023-07-08 jrick const char *tagger, *ssh_sig = NULL;
5901 24df9a28 2023-07-08 jrick char *sig_msg = NULL;
5902 24df9a28 2023-07-08 jrick time_t tagger_time;
5903 24df9a28 2023-07-08 jrick struct got_object_id *id;
5904 24df9a28 2023-07-08 jrick struct got_tag_object *tag;
5905 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
5906 24df9a28 2023-07-08 jrick
5907 24df9a28 2023-07-08 jrick refname = got_ref_get_name(re->ref);
5908 24df9a28 2023-07-08 jrick if (strncmp(refname, "refs/tags/", 10) != 0 ||
5909 24df9a28 2023-07-08 jrick (wanted_refname && strcmp(refname, wanted_refname) != 0))
5910 24df9a28 2023-07-08 jrick continue;
5911 24df9a28 2023-07-08 jrick refname += 10;
5912 24df9a28 2023-07-08 jrick refstr = got_ref_to_str(re->ref);
5913 24df9a28 2023-07-08 jrick if (refstr == NULL) {
5914 24df9a28 2023-07-08 jrick err = got_error_from_errno("got_ref_to_str");
5915 24df9a28 2023-07-08 jrick break;
5916 24df9a28 2023-07-08 jrick }
5917 24df9a28 2023-07-08 jrick
5918 24df9a28 2023-07-08 jrick err = got_ref_resolve(&id, repo, re->ref);
5919 24df9a28 2023-07-08 jrick if (err)
5920 24df9a28 2023-07-08 jrick break;
5921 24df9a28 2023-07-08 jrick err = got_object_open_as_tag(&tag, repo, id);
5922 24df9a28 2023-07-08 jrick if (err) {
5923 24df9a28 2023-07-08 jrick if (err->code != GOT_ERR_OBJ_TYPE) {
5924 24df9a28 2023-07-08 jrick free(id);
5925 24df9a28 2023-07-08 jrick break;
5926 24df9a28 2023-07-08 jrick }
5927 24df9a28 2023-07-08 jrick /* "lightweight" tag */
5928 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, id);
5929 24df9a28 2023-07-08 jrick if (err) {
5930 24df9a28 2023-07-08 jrick free(id);
5931 24df9a28 2023-07-08 jrick break;
5932 24df9a28 2023-07-08 jrick }
5933 24df9a28 2023-07-08 jrick tagger = got_object_commit_get_committer(commit);
5934 24df9a28 2023-07-08 jrick tagger_time =
5935 24df9a28 2023-07-08 jrick got_object_commit_get_committer_time(commit);
5936 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, id);
5937 24df9a28 2023-07-08 jrick free(id);
5938 24df9a28 2023-07-08 jrick if (err)
5939 24df9a28 2023-07-08 jrick break;
5940 24df9a28 2023-07-08 jrick } else {
5941 24df9a28 2023-07-08 jrick free(id);
5942 24df9a28 2023-07-08 jrick tagger = got_object_tag_get_tagger(tag);
5943 24df9a28 2023-07-08 jrick tagger_time = got_object_tag_get_tagger_time(tag);
5944 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str,
5945 24df9a28 2023-07-08 jrick got_object_tag_get_object_id(tag));
5946 24df9a28 2023-07-08 jrick if (err)
5947 24df9a28 2023-07-08 jrick break;
5948 24df9a28 2023-07-08 jrick }
5949 24df9a28 2023-07-08 jrick
5950 24df9a28 2023-07-08 jrick if (tag && verify_tags) {
5951 24df9a28 2023-07-08 jrick ssh_sig = got_sigs_get_tagmsg_ssh_signature(
5952 24df9a28 2023-07-08 jrick got_object_tag_get_message(tag));
5953 24df9a28 2023-07-08 jrick if (ssh_sig && allowed_signers == NULL) {
5954 24df9a28 2023-07-08 jrick err = got_error_msg(
5955 24df9a28 2023-07-08 jrick GOT_ERR_VERIFY_TAG_SIGNATURE,
5956 24df9a28 2023-07-08 jrick "SSH signature verification requires "
5957 24df9a28 2023-07-08 jrick "setting allowed_signers in "
5958 24df9a28 2023-07-08 jrick "got.conf(5)");
5959 24df9a28 2023-07-08 jrick break;
5960 24df9a28 2023-07-08 jrick }
5961 24df9a28 2023-07-08 jrick }
5962 24df9a28 2023-07-08 jrick
5963 24df9a28 2023-07-08 jrick printf("%stag %s %s\n", GOT_COMMIT_SEP_STR, refname, refstr);
5964 24df9a28 2023-07-08 jrick free(refstr);
5965 24df9a28 2023-07-08 jrick printf("from: %s\n", tagger);
5966 24df9a28 2023-07-08 jrick datestr = get_datestr(&tagger_time, datebuf);
5967 24df9a28 2023-07-08 jrick if (datestr)
5968 24df9a28 2023-07-08 jrick printf("date: %s UTC\n", datestr);
5969 24df9a28 2023-07-08 jrick if (commit)
5970 24df9a28 2023-07-08 jrick printf("object: %s %s\n", GOT_OBJ_LABEL_COMMIT, id_str);
5971 24df9a28 2023-07-08 jrick else {
5972 24df9a28 2023-07-08 jrick switch (got_object_tag_get_object_type(tag)) {
5973 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_BLOB:
5974 24df9a28 2023-07-08 jrick printf("object: %s %s\n", GOT_OBJ_LABEL_BLOB,
5975 24df9a28 2023-07-08 jrick id_str);
5976 24df9a28 2023-07-08 jrick break;
5977 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TREE:
5978 24df9a28 2023-07-08 jrick printf("object: %s %s\n", GOT_OBJ_LABEL_TREE,
5979 24df9a28 2023-07-08 jrick id_str);
5980 24df9a28 2023-07-08 jrick break;
5981 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_COMMIT:
5982 24df9a28 2023-07-08 jrick printf("object: %s %s\n", GOT_OBJ_LABEL_COMMIT,
5983 24df9a28 2023-07-08 jrick id_str);
5984 24df9a28 2023-07-08 jrick break;
5985 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TAG:
5986 24df9a28 2023-07-08 jrick printf("object: %s %s\n", GOT_OBJ_LABEL_TAG,
5987 24df9a28 2023-07-08 jrick id_str);
5988 24df9a28 2023-07-08 jrick break;
5989 24df9a28 2023-07-08 jrick default:
5990 24df9a28 2023-07-08 jrick break;
5991 24df9a28 2023-07-08 jrick }
5992 24df9a28 2023-07-08 jrick }
5993 24df9a28 2023-07-08 jrick free(id_str);
5994 24df9a28 2023-07-08 jrick
5995 24df9a28 2023-07-08 jrick if (ssh_sig) {
5996 24df9a28 2023-07-08 jrick err = got_sigs_verify_tag_ssh(&sig_msg, tag, ssh_sig,
5997 24df9a28 2023-07-08 jrick allowed_signers, revoked_signers, verbosity);
5998 24df9a28 2023-07-08 jrick if (err && err->code == GOT_ERR_BAD_TAG_SIGNATURE)
5999 24df9a28 2023-07-08 jrick bad_sigs = 1;
6000 24df9a28 2023-07-08 jrick else if (err)
6001 24df9a28 2023-07-08 jrick break;
6002 24df9a28 2023-07-08 jrick printf("signature: %s", sig_msg);
6003 24df9a28 2023-07-08 jrick free(sig_msg);
6004 24df9a28 2023-07-08 jrick sig_msg = NULL;
6005 24df9a28 2023-07-08 jrick }
6006 24df9a28 2023-07-08 jrick
6007 24df9a28 2023-07-08 jrick if (commit) {
6008 24df9a28 2023-07-08 jrick err = got_object_commit_get_logmsg(&tagmsg0, commit);
6009 24df9a28 2023-07-08 jrick if (err)
6010 24df9a28 2023-07-08 jrick break;
6011 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
6012 24df9a28 2023-07-08 jrick } else {
6013 24df9a28 2023-07-08 jrick tagmsg0 = strdup(got_object_tag_get_message(tag));
6014 24df9a28 2023-07-08 jrick got_object_tag_close(tag);
6015 24df9a28 2023-07-08 jrick if (tagmsg0 == NULL) {
6016 24df9a28 2023-07-08 jrick err = got_error_from_errno("strdup");
6017 24df9a28 2023-07-08 jrick break;
6018 24df9a28 2023-07-08 jrick }
6019 24df9a28 2023-07-08 jrick }
6020 24df9a28 2023-07-08 jrick
6021 24df9a28 2023-07-08 jrick tagmsg = tagmsg0;
6022 24df9a28 2023-07-08 jrick do {
6023 24df9a28 2023-07-08 jrick line = strsep(&tagmsg, "\n");
6024 24df9a28 2023-07-08 jrick if (line)
6025 24df9a28 2023-07-08 jrick printf(" %s\n", line);
6026 24df9a28 2023-07-08 jrick } while (line);
6027 24df9a28 2023-07-08 jrick free(tagmsg0);
6028 24df9a28 2023-07-08 jrick }
6029 24df9a28 2023-07-08 jrick done:
6030 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
6031 24df9a28 2023-07-08 jrick free(wanted_refname);
6032 24df9a28 2023-07-08 jrick
6033 24df9a28 2023-07-08 jrick if (err == NULL && bad_sigs)
6034 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_BAD_TAG_SIGNATURE);
6035 24df9a28 2023-07-08 jrick return err;
6036 24df9a28 2023-07-08 jrick }
6037 24df9a28 2023-07-08 jrick
6038 24df9a28 2023-07-08 jrick static const struct got_error *
6039 24df9a28 2023-07-08 jrick get_tag_message(char **tagmsg, char **tagmsg_path, const char *commit_id_str,
6040 24df9a28 2023-07-08 jrick const char *tag_name, const char *repo_path)
6041 24df9a28 2023-07-08 jrick {
6042 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
6043 24df9a28 2023-07-08 jrick char *template = NULL, *initial_content = NULL;
6044 24df9a28 2023-07-08 jrick char *editor = NULL;
6045 24df9a28 2023-07-08 jrick int initial_content_len;
6046 24df9a28 2023-07-08 jrick int fd = -1;
6047 24df9a28 2023-07-08 jrick
6048 24df9a28 2023-07-08 jrick if (asprintf(&template, GOT_TMPDIR_STR "/got-tagmsg") == -1) {
6049 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
6050 24df9a28 2023-07-08 jrick goto done;
6051 24df9a28 2023-07-08 jrick }
6052 24df9a28 2023-07-08 jrick
6053 24df9a28 2023-07-08 jrick initial_content_len = asprintf(&initial_content,
6054 24df9a28 2023-07-08 jrick "\n# tagging commit %s as %s\n",
6055 24df9a28 2023-07-08 jrick commit_id_str, tag_name);
6056 24df9a28 2023-07-08 jrick if (initial_content_len == -1) {
6057 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
6058 24df9a28 2023-07-08 jrick goto done;
6059 24df9a28 2023-07-08 jrick }
6060 24df9a28 2023-07-08 jrick
6061 24df9a28 2023-07-08 jrick err = got_opentemp_named_fd(tagmsg_path, &fd, template, "");
6062 24df9a28 2023-07-08 jrick if (err)
6063 24df9a28 2023-07-08 jrick goto done;
6064 24df9a28 2023-07-08 jrick
6065 24df9a28 2023-07-08 jrick if (write(fd, initial_content, initial_content_len) == -1) {
6066 24df9a28 2023-07-08 jrick err = got_error_from_errno2("write", *tagmsg_path);
6067 24df9a28 2023-07-08 jrick goto done;
6068 24df9a28 2023-07-08 jrick }
6069 24df9a28 2023-07-08 jrick if (close(fd) == -1) {
6070 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", *tagmsg_path);
6071 24df9a28 2023-07-08 jrick goto done;
6072 24df9a28 2023-07-08 jrick }
6073 24df9a28 2023-07-08 jrick fd = -1;
6074 24df9a28 2023-07-08 jrick
6075 24df9a28 2023-07-08 jrick err = get_editor(&editor);
6076 24df9a28 2023-07-08 jrick if (err)
6077 24df9a28 2023-07-08 jrick goto done;
6078 24df9a28 2023-07-08 jrick err = edit_logmsg(tagmsg, editor, *tagmsg_path, initial_content,
6079 24df9a28 2023-07-08 jrick initial_content_len, 1);
6080 24df9a28 2023-07-08 jrick done:
6081 24df9a28 2023-07-08 jrick free(initial_content);
6082 24df9a28 2023-07-08 jrick free(template);
6083 24df9a28 2023-07-08 jrick free(editor);
6084 24df9a28 2023-07-08 jrick
6085 24df9a28 2023-07-08 jrick if (fd != -1 && close(fd) == -1 && err == NULL)
6086 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", *tagmsg_path);
6087 24df9a28 2023-07-08 jrick
6088 24df9a28 2023-07-08 jrick if (err) {
6089 24df9a28 2023-07-08 jrick free(*tagmsg);
6090 24df9a28 2023-07-08 jrick *tagmsg = NULL;
6091 24df9a28 2023-07-08 jrick }
6092 24df9a28 2023-07-08 jrick return err;
6093 24df9a28 2023-07-08 jrick }
6094 24df9a28 2023-07-08 jrick
6095 24df9a28 2023-07-08 jrick static const struct got_error *
6096 24df9a28 2023-07-08 jrick add_tag(struct got_repository *repo, const char *tagger,
6097 24df9a28 2023-07-08 jrick const char *tag_name, const char *commit_arg, const char *tagmsg_arg,
6098 24df9a28 2023-07-08 jrick const char *signer_id, int verbosity)
6099 24df9a28 2023-07-08 jrick {
6100 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
6101 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL, *tag_id = NULL;
6102 24df9a28 2023-07-08 jrick char *label = NULL, *commit_id_str = NULL;
6103 24df9a28 2023-07-08 jrick struct got_reference *ref = NULL;
6104 24df9a28 2023-07-08 jrick char *refname = NULL, *tagmsg = NULL;
6105 24df9a28 2023-07-08 jrick char *tagmsg_path = NULL, *tag_id_str = NULL;
6106 24df9a28 2023-07-08 jrick int preserve_tagmsg = 0;
6107 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
6108 24df9a28 2023-07-08 jrick
6109 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
6110 24df9a28 2023-07-08 jrick
6111 24df9a28 2023-07-08 jrick /*
6112 24df9a28 2023-07-08 jrick * Don't let the user create a tag name with a leading '-'.
6113 24df9a28 2023-07-08 jrick * While technically a valid reference name, this case is usually
6114 24df9a28 2023-07-08 jrick * an unintended typo.
6115 24df9a28 2023-07-08 jrick */
6116 24df9a28 2023-07-08 jrick if (tag_name[0] == '-')
6117 24df9a28 2023-07-08 jrick return got_error_path(tag_name, GOT_ERR_REF_NAME_MINUS);
6118 24df9a28 2023-07-08 jrick
6119 24df9a28 2023-07-08 jrick err = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
6120 24df9a28 2023-07-08 jrick if (err)
6121 24df9a28 2023-07-08 jrick goto done;
6122 24df9a28 2023-07-08 jrick
6123 24df9a28 2023-07-08 jrick err = got_repo_match_object_id(&commit_id, &label, commit_arg,
6124 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, &refs, repo);
6125 24df9a28 2023-07-08 jrick if (err)
6126 24df9a28 2023-07-08 jrick goto done;
6127 24df9a28 2023-07-08 jrick
6128 24df9a28 2023-07-08 jrick err = got_object_id_str(&commit_id_str, commit_id);
6129 24df9a28 2023-07-08 jrick if (err)
6130 24df9a28 2023-07-08 jrick goto done;
6131 24df9a28 2023-07-08 jrick
6132 24df9a28 2023-07-08 jrick err = get_tag_refname(&refname, tag_name);
6133 24df9a28 2023-07-08 jrick if (err)
6134 24df9a28 2023-07-08 jrick goto done;
6135 24df9a28 2023-07-08 jrick if (strncmp("refs/tags/", tag_name, 10) == 0)
6136 24df9a28 2023-07-08 jrick tag_name += 10;
6137 24df9a28 2023-07-08 jrick
6138 24df9a28 2023-07-08 jrick err = got_ref_open(&ref, repo, refname, 0);
6139 24df9a28 2023-07-08 jrick if (err == NULL) {
6140 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_TAG_EXISTS);
6141 24df9a28 2023-07-08 jrick goto done;
6142 24df9a28 2023-07-08 jrick } else if (err->code != GOT_ERR_NOT_REF)
6143 24df9a28 2023-07-08 jrick goto done;
6144 24df9a28 2023-07-08 jrick
6145 24df9a28 2023-07-08 jrick if (tagmsg_arg == NULL) {
6146 24df9a28 2023-07-08 jrick err = get_tag_message(&tagmsg, &tagmsg_path, commit_id_str,
6147 24df9a28 2023-07-08 jrick tag_name, got_repo_get_path(repo));
6148 24df9a28 2023-07-08 jrick if (err) {
6149 24df9a28 2023-07-08 jrick if (err->code != GOT_ERR_COMMIT_MSG_EMPTY &&
6150 24df9a28 2023-07-08 jrick tagmsg_path != NULL)
6151 24df9a28 2023-07-08 jrick preserve_tagmsg = 1;
6152 24df9a28 2023-07-08 jrick goto done;
6153 24df9a28 2023-07-08 jrick }
6154 24df9a28 2023-07-08 jrick /* Editor is done; we can now apply unveil(2) */
6155 24df9a28 2023-07-08 jrick err = got_sigs_apply_unveil();
6156 24df9a28 2023-07-08 jrick if (err)
6157 24df9a28 2023-07-08 jrick goto done;
6158 24df9a28 2023-07-08 jrick err = apply_unveil(got_repo_get_path(repo), 0, NULL);
6159 24df9a28 2023-07-08 jrick if (err)
6160 24df9a28 2023-07-08 jrick goto done;
6161 24df9a28 2023-07-08 jrick }
6162 24df9a28 2023-07-08 jrick
6163 24df9a28 2023-07-08 jrick err = got_object_tag_create(&tag_id, tag_name, commit_id,
6164 24df9a28 2023-07-08 jrick tagger, time(NULL), tagmsg ? tagmsg : tagmsg_arg, signer_id, repo,
6165 24df9a28 2023-07-08 jrick verbosity);
6166 24df9a28 2023-07-08 jrick if (err) {
6167 24df9a28 2023-07-08 jrick if (tagmsg_path)
6168 24df9a28 2023-07-08 jrick preserve_tagmsg = 1;
6169 24df9a28 2023-07-08 jrick goto done;
6170 24df9a28 2023-07-08 jrick }
6171 24df9a28 2023-07-08 jrick
6172 24df9a28 2023-07-08 jrick err = got_ref_alloc(&ref, refname, tag_id);
6173 24df9a28 2023-07-08 jrick if (err) {
6174 24df9a28 2023-07-08 jrick if (tagmsg_path)
6175 24df9a28 2023-07-08 jrick preserve_tagmsg = 1;
6176 24df9a28 2023-07-08 jrick goto done;
6177 24df9a28 2023-07-08 jrick }
6178 24df9a28 2023-07-08 jrick
6179 24df9a28 2023-07-08 jrick err = got_ref_write(ref, repo);
6180 24df9a28 2023-07-08 jrick if (err) {
6181 24df9a28 2023-07-08 jrick if (tagmsg_path)
6182 24df9a28 2023-07-08 jrick preserve_tagmsg = 1;
6183 24df9a28 2023-07-08 jrick goto done;
6184 24df9a28 2023-07-08 jrick }
6185 24df9a28 2023-07-08 jrick
6186 24df9a28 2023-07-08 jrick err = got_object_id_str(&tag_id_str, tag_id);
6187 24df9a28 2023-07-08 jrick if (err) {
6188 24df9a28 2023-07-08 jrick if (tagmsg_path)
6189 24df9a28 2023-07-08 jrick preserve_tagmsg = 1;
6190 24df9a28 2023-07-08 jrick goto done;
6191 24df9a28 2023-07-08 jrick }
6192 24df9a28 2023-07-08 jrick printf("Created tag %s\n", tag_id_str);
6193 24df9a28 2023-07-08 jrick done:
6194 24df9a28 2023-07-08 jrick if (preserve_tagmsg) {
6195 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: tag message preserved in %s\n",
6196 24df9a28 2023-07-08 jrick getprogname(), tagmsg_path);
6197 24df9a28 2023-07-08 jrick } else if (tagmsg_path && unlink(tagmsg_path) == -1 && err == NULL)
6198 24df9a28 2023-07-08 jrick err = got_error_from_errno2("unlink", tagmsg_path);
6199 24df9a28 2023-07-08 jrick free(tag_id_str);
6200 24df9a28 2023-07-08 jrick if (ref)
6201 24df9a28 2023-07-08 jrick got_ref_close(ref);
6202 24df9a28 2023-07-08 jrick free(commit_id);
6203 24df9a28 2023-07-08 jrick free(commit_id_str);
6204 24df9a28 2023-07-08 jrick free(refname);
6205 24df9a28 2023-07-08 jrick free(tagmsg);
6206 24df9a28 2023-07-08 jrick free(tagmsg_path);
6207 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
6208 24df9a28 2023-07-08 jrick return err;
6209 24df9a28 2023-07-08 jrick }
6210 24df9a28 2023-07-08 jrick
6211 24df9a28 2023-07-08 jrick static const struct got_error *
6212 24df9a28 2023-07-08 jrick cmd_tag(int argc, char *argv[])
6213 24df9a28 2023-07-08 jrick {
6214 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
6215 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
6216 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
6217 24df9a28 2023-07-08 jrick char *cwd = NULL, *repo_path = NULL, *commit_id_str = NULL;
6218 24df9a28 2023-07-08 jrick char *gitconfig_path = NULL, *tagger = NULL;
6219 24df9a28 2023-07-08 jrick char *allowed_signers = NULL, *revoked_signers = NULL;
6220 24df9a28 2023-07-08 jrick const char *signer_id = NULL;
6221 24df9a28 2023-07-08 jrick const char *tag_name = NULL, *commit_id_arg = NULL, *tagmsg = NULL;
6222 24df9a28 2023-07-08 jrick int ch, do_list = 0, verify_tags = 0, verbosity = 0;
6223 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
6224 24df9a28 2023-07-08 jrick
6225 24df9a28 2023-07-08 jrick #ifndef PROFILE
6226 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
6227 24df9a28 2023-07-08 jrick "sendfd unveil", NULL) == -1)
6228 24df9a28 2023-07-08 jrick err(1, "pledge");
6229 24df9a28 2023-07-08 jrick #endif
6230 24df9a28 2023-07-08 jrick
6231 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:lm:r:s:Vv")) != -1) {
6232 24df9a28 2023-07-08 jrick switch (ch) {
6233 24df9a28 2023-07-08 jrick case 'c':
6234 24df9a28 2023-07-08 jrick commit_id_arg = optarg;
6235 24df9a28 2023-07-08 jrick break;
6236 24df9a28 2023-07-08 jrick case 'l':
6237 24df9a28 2023-07-08 jrick do_list = 1;
6238 24df9a28 2023-07-08 jrick break;
6239 24df9a28 2023-07-08 jrick case 'm':
6240 24df9a28 2023-07-08 jrick tagmsg = optarg;
6241 24df9a28 2023-07-08 jrick break;
6242 24df9a28 2023-07-08 jrick case 'r':
6243 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
6244 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
6245 24df9a28 2023-07-08 jrick error = got_error_from_errno2("realpath",
6246 24df9a28 2023-07-08 jrick optarg);
6247 24df9a28 2023-07-08 jrick goto done;
6248 24df9a28 2023-07-08 jrick }
6249 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
6250 24df9a28 2023-07-08 jrick break;
6251 24df9a28 2023-07-08 jrick case 's':
6252 24df9a28 2023-07-08 jrick signer_id = optarg;
6253 24df9a28 2023-07-08 jrick break;
6254 24df9a28 2023-07-08 jrick case 'V':
6255 24df9a28 2023-07-08 jrick verify_tags = 1;
6256 24df9a28 2023-07-08 jrick break;
6257 24df9a28 2023-07-08 jrick case 'v':
6258 24df9a28 2023-07-08 jrick if (verbosity < 0)
6259 24df9a28 2023-07-08 jrick verbosity = 0;
6260 24df9a28 2023-07-08 jrick else if (verbosity < 3)
6261 24df9a28 2023-07-08 jrick verbosity++;
6262 24df9a28 2023-07-08 jrick break;
6263 24df9a28 2023-07-08 jrick default:
6264 24df9a28 2023-07-08 jrick usage_tag();
6265 24df9a28 2023-07-08 jrick /* NOTREACHED */
6266 24df9a28 2023-07-08 jrick }
6267 24df9a28 2023-07-08 jrick }
6268 24df9a28 2023-07-08 jrick
6269 24df9a28 2023-07-08 jrick argc -= optind;
6270 24df9a28 2023-07-08 jrick argv += optind;
6271 24df9a28 2023-07-08 jrick
6272 24df9a28 2023-07-08 jrick if (do_list || verify_tags) {
6273 24df9a28 2023-07-08 jrick if (commit_id_arg != NULL)
6274 24df9a28 2023-07-08 jrick errx(1,
6275 24df9a28 2023-07-08 jrick "-c option can only be used when creating a tag");
6276 24df9a28 2023-07-08 jrick if (tagmsg) {
6277 24df9a28 2023-07-08 jrick if (do_list)
6278 24df9a28 2023-07-08 jrick option_conflict('l', 'm');
6279 24df9a28 2023-07-08 jrick else
6280 24df9a28 2023-07-08 jrick option_conflict('V', 'm');
6281 24df9a28 2023-07-08 jrick }
6282 24df9a28 2023-07-08 jrick if (signer_id) {
6283 24df9a28 2023-07-08 jrick if (do_list)
6284 24df9a28 2023-07-08 jrick option_conflict('l', 's');
6285 24df9a28 2023-07-08 jrick else
6286 24df9a28 2023-07-08 jrick option_conflict('V', 's');
6287 24df9a28 2023-07-08 jrick }
6288 24df9a28 2023-07-08 jrick if (argc > 1)
6289 24df9a28 2023-07-08 jrick usage_tag();
6290 24df9a28 2023-07-08 jrick } else if (argc != 1)
6291 24df9a28 2023-07-08 jrick usage_tag();
6292 24df9a28 2023-07-08 jrick
6293 24df9a28 2023-07-08 jrick if (argc == 1)
6294 24df9a28 2023-07-08 jrick tag_name = argv[0];
6295 24df9a28 2023-07-08 jrick
6296 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
6297 24df9a28 2023-07-08 jrick if (cwd == NULL) {
6298 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
6299 24df9a28 2023-07-08 jrick goto done;
6300 24df9a28 2023-07-08 jrick }
6301 24df9a28 2023-07-08 jrick
6302 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
6303 24df9a28 2023-07-08 jrick if (error != NULL)
6304 24df9a28 2023-07-08 jrick goto done;
6305 24df9a28 2023-07-08 jrick
6306 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
6307 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
6308 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
6309 24df9a28 2023-07-08 jrick goto done;
6310 24df9a28 2023-07-08 jrick else
6311 24df9a28 2023-07-08 jrick error = NULL;
6312 24df9a28 2023-07-08 jrick if (worktree) {
6313 24df9a28 2023-07-08 jrick repo_path =
6314 24df9a28 2023-07-08 jrick strdup(got_worktree_get_repo_path(worktree));
6315 24df9a28 2023-07-08 jrick if (repo_path == NULL)
6316 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
6317 24df9a28 2023-07-08 jrick if (error)
6318 24df9a28 2023-07-08 jrick goto done;
6319 24df9a28 2023-07-08 jrick } else {
6320 24df9a28 2023-07-08 jrick repo_path = strdup(cwd);
6321 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
6322 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
6323 24df9a28 2023-07-08 jrick goto done;
6324 24df9a28 2023-07-08 jrick }
6325 24df9a28 2023-07-08 jrick }
6326 24df9a28 2023-07-08 jrick }
6327 24df9a28 2023-07-08 jrick
6328 24df9a28 2023-07-08 jrick if (do_list || verify_tags) {
6329 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6330 24df9a28 2023-07-08 jrick if (error != NULL)
6331 24df9a28 2023-07-08 jrick goto done;
6332 24df9a28 2023-07-08 jrick error = get_allowed_signers(&allowed_signers, repo, worktree);
6333 24df9a28 2023-07-08 jrick if (error)
6334 24df9a28 2023-07-08 jrick goto done;
6335 24df9a28 2023-07-08 jrick error = get_revoked_signers(&revoked_signers, repo, worktree);
6336 24df9a28 2023-07-08 jrick if (error)
6337 24df9a28 2023-07-08 jrick goto done;
6338 24df9a28 2023-07-08 jrick if (worktree) {
6339 24df9a28 2023-07-08 jrick /* Release work tree lock. */
6340 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
6341 24df9a28 2023-07-08 jrick worktree = NULL;
6342 24df9a28 2023-07-08 jrick }
6343 24df9a28 2023-07-08 jrick
6344 24df9a28 2023-07-08 jrick /*
6345 24df9a28 2023-07-08 jrick * Remove "cpath" promise unless needed for signature tmpfile
6346 24df9a28 2023-07-08 jrick * creation.
6347 24df9a28 2023-07-08 jrick */
6348 24df9a28 2023-07-08 jrick if (verify_tags)
6349 24df9a28 2023-07-08 jrick got_sigs_apply_unveil();
6350 24df9a28 2023-07-08 jrick else {
6351 24df9a28 2023-07-08 jrick #ifndef PROFILE
6352 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath flock proc exec sendfd "
6353 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
6354 24df9a28 2023-07-08 jrick err(1, "pledge");
6355 24df9a28 2023-07-08 jrick #endif
6356 24df9a28 2023-07-08 jrick }
6357 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
6358 24df9a28 2023-07-08 jrick if (error)
6359 24df9a28 2023-07-08 jrick goto done;
6360 24df9a28 2023-07-08 jrick error = list_tags(repo, tag_name, verify_tags, allowed_signers,
6361 24df9a28 2023-07-08 jrick revoked_signers, verbosity);
6362 24df9a28 2023-07-08 jrick } else {
6363 24df9a28 2023-07-08 jrick error = get_gitconfig_path(&gitconfig_path);
6364 24df9a28 2023-07-08 jrick if (error)
6365 24df9a28 2023-07-08 jrick goto done;
6366 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, gitconfig_path,
6367 24df9a28 2023-07-08 jrick pack_fds);
6368 24df9a28 2023-07-08 jrick if (error != NULL)
6369 24df9a28 2023-07-08 jrick goto done;
6370 24df9a28 2023-07-08 jrick
6371 24df9a28 2023-07-08 jrick error = get_author(&tagger, repo, worktree);
6372 24df9a28 2023-07-08 jrick if (error)
6373 24df9a28 2023-07-08 jrick goto done;
6374 24df9a28 2023-07-08 jrick if (signer_id == NULL)
6375 24df9a28 2023-07-08 jrick signer_id = get_signer_id(repo, worktree);
6376 24df9a28 2023-07-08 jrick
6377 24df9a28 2023-07-08 jrick if (tagmsg) {
6378 24df9a28 2023-07-08 jrick if (signer_id) {
6379 24df9a28 2023-07-08 jrick error = got_sigs_apply_unveil();
6380 24df9a28 2023-07-08 jrick if (error)
6381 24df9a28 2023-07-08 jrick goto done;
6382 24df9a28 2023-07-08 jrick }
6383 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0, NULL);
6384 24df9a28 2023-07-08 jrick if (error)
6385 24df9a28 2023-07-08 jrick goto done;
6386 24df9a28 2023-07-08 jrick }
6387 24df9a28 2023-07-08 jrick
6388 24df9a28 2023-07-08 jrick if (commit_id_arg == NULL) {
6389 24df9a28 2023-07-08 jrick struct got_reference *head_ref;
6390 24df9a28 2023-07-08 jrick struct got_object_id *commit_id;
6391 24df9a28 2023-07-08 jrick error = got_ref_open(&head_ref, repo,
6392 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_head_ref_name(worktree)
6393 24df9a28 2023-07-08 jrick : GOT_REF_HEAD, 0);
6394 24df9a28 2023-07-08 jrick if (error)
6395 24df9a28 2023-07-08 jrick goto done;
6396 24df9a28 2023-07-08 jrick error = got_ref_resolve(&commit_id, repo, head_ref);
6397 24df9a28 2023-07-08 jrick got_ref_close(head_ref);
6398 24df9a28 2023-07-08 jrick if (error)
6399 24df9a28 2023-07-08 jrick goto done;
6400 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
6401 24df9a28 2023-07-08 jrick free(commit_id);
6402 24df9a28 2023-07-08 jrick if (error)
6403 24df9a28 2023-07-08 jrick goto done;
6404 24df9a28 2023-07-08 jrick }
6405 24df9a28 2023-07-08 jrick
6406 24df9a28 2023-07-08 jrick if (worktree) {
6407 24df9a28 2023-07-08 jrick /* Release work tree lock. */
6408 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
6409 24df9a28 2023-07-08 jrick worktree = NULL;
6410 24df9a28 2023-07-08 jrick }
6411 24df9a28 2023-07-08 jrick
6412 24df9a28 2023-07-08 jrick error = add_tag(repo, tagger, tag_name,
6413 24df9a28 2023-07-08 jrick commit_id_str ? commit_id_str : commit_id_arg, tagmsg,
6414 24df9a28 2023-07-08 jrick signer_id, verbosity);
6415 24df9a28 2023-07-08 jrick }
6416 24df9a28 2023-07-08 jrick done:
6417 24df9a28 2023-07-08 jrick if (repo) {
6418 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
6419 24df9a28 2023-07-08 jrick if (error == NULL)
6420 24df9a28 2023-07-08 jrick error = close_err;
6421 24df9a28 2023-07-08 jrick }
6422 24df9a28 2023-07-08 jrick if (worktree)
6423 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
6424 24df9a28 2023-07-08 jrick if (pack_fds) {
6425 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
6426 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
6427 24df9a28 2023-07-08 jrick if (error == NULL)
6428 24df9a28 2023-07-08 jrick error = pack_err;
6429 24df9a28 2023-07-08 jrick }
6430 24df9a28 2023-07-08 jrick free(cwd);
6431 24df9a28 2023-07-08 jrick free(repo_path);
6432 24df9a28 2023-07-08 jrick free(gitconfig_path);
6433 24df9a28 2023-07-08 jrick free(commit_id_str);
6434 24df9a28 2023-07-08 jrick free(tagger);
6435 24df9a28 2023-07-08 jrick free(allowed_signers);
6436 24df9a28 2023-07-08 jrick free(revoked_signers);
6437 24df9a28 2023-07-08 jrick return error;
6438 24df9a28 2023-07-08 jrick }
6439 24df9a28 2023-07-08 jrick
6440 24df9a28 2023-07-08 jrick __dead static void
6441 24df9a28 2023-07-08 jrick usage_add(void)
6442 24df9a28 2023-07-08 jrick {
6443 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s add [-IR] path ...\n", getprogname());
6444 24df9a28 2023-07-08 jrick exit(1);
6445 24df9a28 2023-07-08 jrick }
6446 24df9a28 2023-07-08 jrick
6447 24df9a28 2023-07-08 jrick static const struct got_error *
6448 24df9a28 2023-07-08 jrick add_progress(void *arg, unsigned char status, const char *path)
6449 24df9a28 2023-07-08 jrick {
6450 24df9a28 2023-07-08 jrick while (path[0] == '/')
6451 24df9a28 2023-07-08 jrick path++;
6452 24df9a28 2023-07-08 jrick printf("%c %s\n", status, path);
6453 24df9a28 2023-07-08 jrick return NULL;
6454 24df9a28 2023-07-08 jrick }
6455 24df9a28 2023-07-08 jrick
6456 24df9a28 2023-07-08 jrick static const struct got_error *
6457 24df9a28 2023-07-08 jrick cmd_add(int argc, char *argv[])
6458 24df9a28 2023-07-08 jrick {
6459 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
6460 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
6461 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
6462 24df9a28 2023-07-08 jrick char *cwd = NULL;
6463 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
6464 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
6465 24df9a28 2023-07-08 jrick int ch, can_recurse = 0, no_ignores = 0;
6466 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
6467 24df9a28 2023-07-08 jrick
6468 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
6469 24df9a28 2023-07-08 jrick
6470 24df9a28 2023-07-08 jrick #ifndef PROFILE
6471 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
6472 24df9a28 2023-07-08 jrick NULL) == -1)
6473 24df9a28 2023-07-08 jrick err(1, "pledge");
6474 24df9a28 2023-07-08 jrick #endif
6475 24df9a28 2023-07-08 jrick
6476 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "IR")) != -1) {
6477 24df9a28 2023-07-08 jrick switch (ch) {
6478 24df9a28 2023-07-08 jrick case 'I':
6479 24df9a28 2023-07-08 jrick no_ignores = 1;
6480 24df9a28 2023-07-08 jrick break;
6481 24df9a28 2023-07-08 jrick case 'R':
6482 24df9a28 2023-07-08 jrick can_recurse = 1;
6483 24df9a28 2023-07-08 jrick break;
6484 24df9a28 2023-07-08 jrick default:
6485 24df9a28 2023-07-08 jrick usage_add();
6486 24df9a28 2023-07-08 jrick /* NOTREACHED */
6487 24df9a28 2023-07-08 jrick }
6488 24df9a28 2023-07-08 jrick }
6489 24df9a28 2023-07-08 jrick
6490 24df9a28 2023-07-08 jrick argc -= optind;
6491 24df9a28 2023-07-08 jrick argv += optind;
6492 24df9a28 2023-07-08 jrick
6493 24df9a28 2023-07-08 jrick if (argc < 1)
6494 24df9a28 2023-07-08 jrick usage_add();
6495 24df9a28 2023-07-08 jrick
6496 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
6497 24df9a28 2023-07-08 jrick if (cwd == NULL) {
6498 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
6499 24df9a28 2023-07-08 jrick goto done;
6500 24df9a28 2023-07-08 jrick }
6501 24df9a28 2023-07-08 jrick
6502 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
6503 24df9a28 2023-07-08 jrick if (error != NULL)
6504 24df9a28 2023-07-08 jrick goto done;
6505 24df9a28 2023-07-08 jrick
6506 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
6507 24df9a28 2023-07-08 jrick if (error) {
6508 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
6509 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "add", cwd);
6510 24df9a28 2023-07-08 jrick goto done;
6511 24df9a28 2023-07-08 jrick }
6512 24df9a28 2023-07-08 jrick
6513 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
6514 24df9a28 2023-07-08 jrick NULL, pack_fds);
6515 24df9a28 2023-07-08 jrick if (error != NULL)
6516 24df9a28 2023-07-08 jrick goto done;
6517 24df9a28 2023-07-08 jrick
6518 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1,
6519 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
6520 24df9a28 2023-07-08 jrick if (error)
6521 24df9a28 2023-07-08 jrick goto done;
6522 24df9a28 2023-07-08 jrick
6523 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
6524 24df9a28 2023-07-08 jrick if (error)
6525 24df9a28 2023-07-08 jrick goto done;
6526 24df9a28 2023-07-08 jrick
6527 24df9a28 2023-07-08 jrick if (!can_recurse) {
6528 24df9a28 2023-07-08 jrick char *ondisk_path;
6529 24df9a28 2023-07-08 jrick struct stat sb;
6530 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &paths, entry) {
6531 24df9a28 2023-07-08 jrick if (asprintf(&ondisk_path, "%s/%s",
6532 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree),
6533 24df9a28 2023-07-08 jrick pe->path) == -1) {
6534 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
6535 24df9a28 2023-07-08 jrick goto done;
6536 24df9a28 2023-07-08 jrick }
6537 24df9a28 2023-07-08 jrick if (lstat(ondisk_path, &sb) == -1) {
6538 24df9a28 2023-07-08 jrick if (errno == ENOENT) {
6539 24df9a28 2023-07-08 jrick free(ondisk_path);
6540 24df9a28 2023-07-08 jrick continue;
6541 24df9a28 2023-07-08 jrick }
6542 24df9a28 2023-07-08 jrick error = got_error_from_errno2("lstat",
6543 24df9a28 2023-07-08 jrick ondisk_path);
6544 24df9a28 2023-07-08 jrick free(ondisk_path);
6545 24df9a28 2023-07-08 jrick goto done;
6546 24df9a28 2023-07-08 jrick }
6547 24df9a28 2023-07-08 jrick free(ondisk_path);
6548 24df9a28 2023-07-08 jrick if (S_ISDIR(sb.st_mode)) {
6549 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_BAD_PATH,
6550 24df9a28 2023-07-08 jrick "adding directories requires -R option");
6551 24df9a28 2023-07-08 jrick goto done;
6552 24df9a28 2023-07-08 jrick }
6553 24df9a28 2023-07-08 jrick }
6554 24df9a28 2023-07-08 jrick }
6555 24df9a28 2023-07-08 jrick
6556 24df9a28 2023-07-08 jrick error = got_worktree_schedule_add(worktree, &paths, add_progress,
6557 24df9a28 2023-07-08 jrick NULL, repo, no_ignores);
6558 24df9a28 2023-07-08 jrick done:
6559 24df9a28 2023-07-08 jrick if (repo) {
6560 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
6561 24df9a28 2023-07-08 jrick if (error == NULL)
6562 24df9a28 2023-07-08 jrick error = close_err;
6563 24df9a28 2023-07-08 jrick }
6564 24df9a28 2023-07-08 jrick if (worktree)
6565 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
6566 24df9a28 2023-07-08 jrick if (pack_fds) {
6567 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
6568 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
6569 24df9a28 2023-07-08 jrick if (error == NULL)
6570 24df9a28 2023-07-08 jrick error = pack_err;
6571 24df9a28 2023-07-08 jrick }
6572 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
6573 24df9a28 2023-07-08 jrick free(cwd);
6574 24df9a28 2023-07-08 jrick return error;
6575 24df9a28 2023-07-08 jrick }
6576 24df9a28 2023-07-08 jrick
6577 24df9a28 2023-07-08 jrick __dead static void
6578 24df9a28 2023-07-08 jrick usage_remove(void)
6579 24df9a28 2023-07-08 jrick {
6580 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s remove [-fkR] [-s status-codes] path ...\n",
6581 24df9a28 2023-07-08 jrick getprogname());
6582 24df9a28 2023-07-08 jrick exit(1);
6583 24df9a28 2023-07-08 jrick }
6584 24df9a28 2023-07-08 jrick
6585 24df9a28 2023-07-08 jrick static const struct got_error *
6586 24df9a28 2023-07-08 jrick print_remove_status(void *arg, unsigned char status,
6587 24df9a28 2023-07-08 jrick unsigned char staged_status, const char *path)
6588 24df9a28 2023-07-08 jrick {
6589 24df9a28 2023-07-08 jrick while (path[0] == '/')
6590 24df9a28 2023-07-08 jrick path++;
6591 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_NONEXISTENT)
6592 24df9a28 2023-07-08 jrick return NULL;
6593 24df9a28 2023-07-08 jrick if (status == staged_status && (status == GOT_STATUS_DELETE))
6594 24df9a28 2023-07-08 jrick status = GOT_STATUS_NO_CHANGE;
6595 24df9a28 2023-07-08 jrick printf("%c%c %s\n", status, staged_status, path);
6596 24df9a28 2023-07-08 jrick return NULL;
6597 24df9a28 2023-07-08 jrick }
6598 24df9a28 2023-07-08 jrick
6599 24df9a28 2023-07-08 jrick static const struct got_error *
6600 24df9a28 2023-07-08 jrick cmd_remove(int argc, char *argv[])
6601 24df9a28 2023-07-08 jrick {
6602 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
6603 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
6604 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
6605 24df9a28 2023-07-08 jrick const char *status_codes = NULL;
6606 24df9a28 2023-07-08 jrick char *cwd = NULL;
6607 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
6608 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
6609 24df9a28 2023-07-08 jrick int ch, delete_local_mods = 0, can_recurse = 0, keep_on_disk = 0, i;
6610 24df9a28 2023-07-08 jrick int ignore_missing_paths = 0;
6611 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
6612 24df9a28 2023-07-08 jrick
6613 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
6614 24df9a28 2023-07-08 jrick
6615 24df9a28 2023-07-08 jrick #ifndef PROFILE
6616 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
6617 24df9a28 2023-07-08 jrick NULL) == -1)
6618 24df9a28 2023-07-08 jrick err(1, "pledge");
6619 24df9a28 2023-07-08 jrick #endif
6620 24df9a28 2023-07-08 jrick
6621 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "fkRs:")) != -1) {
6622 24df9a28 2023-07-08 jrick switch (ch) {
6623 24df9a28 2023-07-08 jrick case 'f':
6624 24df9a28 2023-07-08 jrick delete_local_mods = 1;
6625 24df9a28 2023-07-08 jrick ignore_missing_paths = 1;
6626 24df9a28 2023-07-08 jrick break;
6627 24df9a28 2023-07-08 jrick case 'k':
6628 24df9a28 2023-07-08 jrick keep_on_disk = 1;
6629 24df9a28 2023-07-08 jrick break;
6630 24df9a28 2023-07-08 jrick case 'R':
6631 24df9a28 2023-07-08 jrick can_recurse = 1;
6632 24df9a28 2023-07-08 jrick break;
6633 24df9a28 2023-07-08 jrick case 's':
6634 24df9a28 2023-07-08 jrick for (i = 0; optarg[i] != '\0'; i++) {
6635 24df9a28 2023-07-08 jrick switch (optarg[i]) {
6636 24df9a28 2023-07-08 jrick case GOT_STATUS_MODIFY:
6637 24df9a28 2023-07-08 jrick delete_local_mods = 1;
6638 24df9a28 2023-07-08 jrick break;
6639 24df9a28 2023-07-08 jrick case GOT_STATUS_MISSING:
6640 24df9a28 2023-07-08 jrick ignore_missing_paths = 1;
6641 24df9a28 2023-07-08 jrick break;
6642 24df9a28 2023-07-08 jrick default:
6643 24df9a28 2023-07-08 jrick errx(1, "invalid status code '%c'",
6644 24df9a28 2023-07-08 jrick optarg[i]);
6645 24df9a28 2023-07-08 jrick }
6646 24df9a28 2023-07-08 jrick }
6647 24df9a28 2023-07-08 jrick status_codes = optarg;
6648 24df9a28 2023-07-08 jrick break;
6649 24df9a28 2023-07-08 jrick default:
6650 24df9a28 2023-07-08 jrick usage_remove();
6651 24df9a28 2023-07-08 jrick /* NOTREACHED */
6652 24df9a28 2023-07-08 jrick }
6653 24df9a28 2023-07-08 jrick }
6654 24df9a28 2023-07-08 jrick
6655 24df9a28 2023-07-08 jrick argc -= optind;
6656 24df9a28 2023-07-08 jrick argv += optind;
6657 24df9a28 2023-07-08 jrick
6658 24df9a28 2023-07-08 jrick if (argc < 1)
6659 24df9a28 2023-07-08 jrick usage_remove();
6660 24df9a28 2023-07-08 jrick
6661 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
6662 24df9a28 2023-07-08 jrick if (cwd == NULL) {
6663 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
6664 24df9a28 2023-07-08 jrick goto done;
6665 24df9a28 2023-07-08 jrick }
6666 24df9a28 2023-07-08 jrick
6667 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
6668 24df9a28 2023-07-08 jrick if (error != NULL)
6669 24df9a28 2023-07-08 jrick goto done;
6670 24df9a28 2023-07-08 jrick
6671 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
6672 24df9a28 2023-07-08 jrick if (error) {
6673 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
6674 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "remove", cwd);
6675 24df9a28 2023-07-08 jrick goto done;
6676 24df9a28 2023-07-08 jrick }
6677 24df9a28 2023-07-08 jrick
6678 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
6679 24df9a28 2023-07-08 jrick NULL, pack_fds);
6680 24df9a28 2023-07-08 jrick if (error)
6681 24df9a28 2023-07-08 jrick goto done;
6682 24df9a28 2023-07-08 jrick
6683 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1,
6684 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
6685 24df9a28 2023-07-08 jrick if (error)
6686 24df9a28 2023-07-08 jrick goto done;
6687 24df9a28 2023-07-08 jrick
6688 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
6689 24df9a28 2023-07-08 jrick if (error)
6690 24df9a28 2023-07-08 jrick goto done;
6691 24df9a28 2023-07-08 jrick
6692 24df9a28 2023-07-08 jrick if (!can_recurse) {
6693 24df9a28 2023-07-08 jrick char *ondisk_path;
6694 24df9a28 2023-07-08 jrick struct stat sb;
6695 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &paths, entry) {
6696 24df9a28 2023-07-08 jrick if (asprintf(&ondisk_path, "%s/%s",
6697 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree),
6698 24df9a28 2023-07-08 jrick pe->path) == -1) {
6699 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
6700 24df9a28 2023-07-08 jrick goto done;
6701 24df9a28 2023-07-08 jrick }
6702 24df9a28 2023-07-08 jrick if (lstat(ondisk_path, &sb) == -1) {
6703 24df9a28 2023-07-08 jrick if (errno == ENOENT) {
6704 24df9a28 2023-07-08 jrick free(ondisk_path);
6705 24df9a28 2023-07-08 jrick continue;
6706 24df9a28 2023-07-08 jrick }
6707 24df9a28 2023-07-08 jrick error = got_error_from_errno2("lstat",
6708 24df9a28 2023-07-08 jrick ondisk_path);
6709 24df9a28 2023-07-08 jrick free(ondisk_path);
6710 24df9a28 2023-07-08 jrick goto done;
6711 24df9a28 2023-07-08 jrick }
6712 24df9a28 2023-07-08 jrick free(ondisk_path);
6713 24df9a28 2023-07-08 jrick if (S_ISDIR(sb.st_mode)) {
6714 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_BAD_PATH,
6715 24df9a28 2023-07-08 jrick "removing directories requires -R option");
6716 24df9a28 2023-07-08 jrick goto done;
6717 24df9a28 2023-07-08 jrick }
6718 24df9a28 2023-07-08 jrick }
6719 24df9a28 2023-07-08 jrick }
6720 24df9a28 2023-07-08 jrick
6721 24df9a28 2023-07-08 jrick error = got_worktree_schedule_delete(worktree, &paths,
6722 24df9a28 2023-07-08 jrick delete_local_mods, status_codes, print_remove_status, NULL,
6723 24df9a28 2023-07-08 jrick repo, keep_on_disk, ignore_missing_paths);
6724 24df9a28 2023-07-08 jrick done:
6725 24df9a28 2023-07-08 jrick if (repo) {
6726 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
6727 24df9a28 2023-07-08 jrick if (error == NULL)
6728 24df9a28 2023-07-08 jrick error = close_err;
6729 24df9a28 2023-07-08 jrick }
6730 24df9a28 2023-07-08 jrick if (worktree)
6731 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
6732 24df9a28 2023-07-08 jrick if (pack_fds) {
6733 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
6734 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
6735 24df9a28 2023-07-08 jrick if (error == NULL)
6736 24df9a28 2023-07-08 jrick error = pack_err;
6737 24df9a28 2023-07-08 jrick }
6738 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
6739 24df9a28 2023-07-08 jrick free(cwd);
6740 24df9a28 2023-07-08 jrick return error;
6741 24df9a28 2023-07-08 jrick }
6742 24df9a28 2023-07-08 jrick
6743 24df9a28 2023-07-08 jrick __dead static void
6744 24df9a28 2023-07-08 jrick usage_patch(void)
6745 24df9a28 2023-07-08 jrick {
6746 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s patch [-nR] [-c commit] [-p strip-count] "
6747 24df9a28 2023-07-08 jrick "[patchfile]\n", getprogname());
6748 24df9a28 2023-07-08 jrick exit(1);
6749 24df9a28 2023-07-08 jrick }
6750 24df9a28 2023-07-08 jrick
6751 24df9a28 2023-07-08 jrick static const struct got_error *
6752 24df9a28 2023-07-08 jrick patch_from_stdin(int *patchfd)
6753 24df9a28 2023-07-08 jrick {
6754 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
6755 24df9a28 2023-07-08 jrick ssize_t r;
6756 24df9a28 2023-07-08 jrick char buf[BUFSIZ];
6757 24df9a28 2023-07-08 jrick sig_t sighup, sigint, sigquit;
6758 24df9a28 2023-07-08 jrick
6759 24df9a28 2023-07-08 jrick *patchfd = got_opentempfd();
6760 24df9a28 2023-07-08 jrick if (*patchfd == -1)
6761 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentempfd");
6762 24df9a28 2023-07-08 jrick
6763 24df9a28 2023-07-08 jrick sighup = signal(SIGHUP, SIG_DFL);
6764 24df9a28 2023-07-08 jrick sigint = signal(SIGINT, SIG_DFL);
6765 24df9a28 2023-07-08 jrick sigquit = signal(SIGQUIT, SIG_DFL);
6766 24df9a28 2023-07-08 jrick
6767 24df9a28 2023-07-08 jrick for (;;) {
6768 24df9a28 2023-07-08 jrick r = read(0, buf, sizeof(buf));
6769 24df9a28 2023-07-08 jrick if (r == -1) {
6770 24df9a28 2023-07-08 jrick err = got_error_from_errno("read");
6771 24df9a28 2023-07-08 jrick break;
6772 24df9a28 2023-07-08 jrick }
6773 24df9a28 2023-07-08 jrick if (r == 0)
6774 24df9a28 2023-07-08 jrick break;
6775 24df9a28 2023-07-08 jrick if (write(*patchfd, buf, r) == -1) {
6776 24df9a28 2023-07-08 jrick err = got_error_from_errno("write");
6777 24df9a28 2023-07-08 jrick break;
6778 24df9a28 2023-07-08 jrick }
6779 24df9a28 2023-07-08 jrick }
6780 24df9a28 2023-07-08 jrick
6781 24df9a28 2023-07-08 jrick signal(SIGHUP, sighup);
6782 24df9a28 2023-07-08 jrick signal(SIGINT, sigint);
6783 24df9a28 2023-07-08 jrick signal(SIGQUIT, sigquit);
6784 24df9a28 2023-07-08 jrick
6785 24df9a28 2023-07-08 jrick if (err == NULL && lseek(*patchfd, 0, SEEK_SET) == -1)
6786 24df9a28 2023-07-08 jrick err = got_error_from_errno("lseek");
6787 24df9a28 2023-07-08 jrick
6788 24df9a28 2023-07-08 jrick if (err != NULL) {
6789 24df9a28 2023-07-08 jrick close(*patchfd);
6790 24df9a28 2023-07-08 jrick *patchfd = -1;
6791 24df9a28 2023-07-08 jrick }
6792 24df9a28 2023-07-08 jrick
6793 24df9a28 2023-07-08 jrick return err;
6794 24df9a28 2023-07-08 jrick }
6795 24df9a28 2023-07-08 jrick
6796 24df9a28 2023-07-08 jrick struct got_patch_progress_arg {
6797 24df9a28 2023-07-08 jrick int did_something;
6798 24df9a28 2023-07-08 jrick int conflicts;
6799 24df9a28 2023-07-08 jrick int rejects;
6800 24df9a28 2023-07-08 jrick };
6801 24df9a28 2023-07-08 jrick
6802 24df9a28 2023-07-08 jrick static const struct got_error *
6803 24df9a28 2023-07-08 jrick patch_progress(void *arg, const char *old, const char *new,
6804 24df9a28 2023-07-08 jrick unsigned char status, const struct got_error *error, int old_from,
6805 24df9a28 2023-07-08 jrick int old_lines, int new_from, int new_lines, int offset,
6806 24df9a28 2023-07-08 jrick int ws_mangled, const struct got_error *hunk_err)
6807 24df9a28 2023-07-08 jrick {
6808 24df9a28 2023-07-08 jrick const char *path = new == NULL ? old : new;
6809 24df9a28 2023-07-08 jrick struct got_patch_progress_arg *a = arg;
6810 24df9a28 2023-07-08 jrick
6811 24df9a28 2023-07-08 jrick while (*path == '/')
6812 24df9a28 2023-07-08 jrick path++;
6813 24df9a28 2023-07-08 jrick
6814 24df9a28 2023-07-08 jrick if (status != GOT_STATUS_NO_CHANGE &&
6815 24df9a28 2023-07-08 jrick status != 0 /* per-hunk progress */) {
6816 24df9a28 2023-07-08 jrick printf("%c %s\n", status, path);
6817 24df9a28 2023-07-08 jrick a->did_something = 1;
6818 24df9a28 2023-07-08 jrick }
6819 24df9a28 2023-07-08 jrick
6820 24df9a28 2023-07-08 jrick if (hunk_err == NULL) {
6821 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_CANNOT_UPDATE)
6822 24df9a28 2023-07-08 jrick a->rejects++;
6823 24df9a28 2023-07-08 jrick else if (status == GOT_STATUS_CONFLICT)
6824 24df9a28 2023-07-08 jrick a->conflicts++;
6825 24df9a28 2023-07-08 jrick }
6826 24df9a28 2023-07-08 jrick
6827 24df9a28 2023-07-08 jrick if (error != NULL)
6828 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
6829 24df9a28 2023-07-08 jrick
6830 24df9a28 2023-07-08 jrick if (offset != 0 || hunk_err != NULL || ws_mangled) {
6831 24df9a28 2023-07-08 jrick printf("@@ -%d,%d +%d,%d @@ ", old_from,
6832 24df9a28 2023-07-08 jrick old_lines, new_from, new_lines);
6833 24df9a28 2023-07-08 jrick if (hunk_err != NULL)
6834 24df9a28 2023-07-08 jrick printf("%s\n", hunk_err->msg);
6835 24df9a28 2023-07-08 jrick else if (offset != 0)
6836 24df9a28 2023-07-08 jrick printf("applied with offset %d\n", offset);
6837 24df9a28 2023-07-08 jrick else
6838 24df9a28 2023-07-08 jrick printf("hunk contains mangled whitespace\n");
6839 24df9a28 2023-07-08 jrick }
6840 24df9a28 2023-07-08 jrick
6841 24df9a28 2023-07-08 jrick return NULL;
6842 24df9a28 2023-07-08 jrick }
6843 24df9a28 2023-07-08 jrick
6844 24df9a28 2023-07-08 jrick static void
6845 24df9a28 2023-07-08 jrick print_patch_progress_stats(struct got_patch_progress_arg *ppa)
6846 24df9a28 2023-07-08 jrick {
6847 24df9a28 2023-07-08 jrick if (!ppa->did_something)
6848 24df9a28 2023-07-08 jrick return;
6849 24df9a28 2023-07-08 jrick
6850 24df9a28 2023-07-08 jrick if (ppa->conflicts > 0)
6851 24df9a28 2023-07-08 jrick printf("Files with merge conflicts: %d\n", ppa->conflicts);
6852 24df9a28 2023-07-08 jrick
6853 24df9a28 2023-07-08 jrick if (ppa->rejects > 0) {
6854 24df9a28 2023-07-08 jrick printf("Files where patch failed to apply: %d\n",
6855 24df9a28 2023-07-08 jrick ppa->rejects);
6856 24df9a28 2023-07-08 jrick }
6857 24df9a28 2023-07-08 jrick }
6858 24df9a28 2023-07-08 jrick
6859 24df9a28 2023-07-08 jrick static const struct got_error *
6860 24df9a28 2023-07-08 jrick cmd_patch(int argc, char *argv[])
6861 24df9a28 2023-07-08 jrick {
6862 24df9a28 2023-07-08 jrick const struct got_error *error = NULL, *close_error = NULL;
6863 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
6864 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
6865 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
6866 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
6867 24df9a28 2023-07-08 jrick const char *commit_id_str = NULL;
6868 24df9a28 2023-07-08 jrick struct stat sb;
6869 24df9a28 2023-07-08 jrick const char *errstr;
6870 24df9a28 2023-07-08 jrick char *cwd = NULL;
6871 24df9a28 2023-07-08 jrick int ch, nop = 0, strip = -1, reverse = 0;
6872 24df9a28 2023-07-08 jrick int patchfd;
6873 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
6874 24df9a28 2023-07-08 jrick struct got_patch_progress_arg ppa;
6875 24df9a28 2023-07-08 jrick
6876 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
6877 24df9a28 2023-07-08 jrick
6878 24df9a28 2023-07-08 jrick #ifndef PROFILE
6879 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr proc exec sendfd flock "
6880 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
6881 24df9a28 2023-07-08 jrick err(1, "pledge");
6882 24df9a28 2023-07-08 jrick #endif
6883 24df9a28 2023-07-08 jrick
6884 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:np:R")) != -1) {
6885 24df9a28 2023-07-08 jrick switch (ch) {
6886 24df9a28 2023-07-08 jrick case 'c':
6887 24df9a28 2023-07-08 jrick commit_id_str = optarg;
6888 24df9a28 2023-07-08 jrick break;
6889 24df9a28 2023-07-08 jrick case 'n':
6890 24df9a28 2023-07-08 jrick nop = 1;
6891 24df9a28 2023-07-08 jrick break;
6892 24df9a28 2023-07-08 jrick case 'p':
6893 24df9a28 2023-07-08 jrick strip = strtonum(optarg, 0, INT_MAX, &errstr);
6894 24df9a28 2023-07-08 jrick if (errstr != NULL)
6895 24df9a28 2023-07-08 jrick errx(1, "pathname strip count is %s: %s",
6896 24df9a28 2023-07-08 jrick errstr, optarg);
6897 24df9a28 2023-07-08 jrick break;
6898 24df9a28 2023-07-08 jrick case 'R':
6899 24df9a28 2023-07-08 jrick reverse = 1;
6900 24df9a28 2023-07-08 jrick break;
6901 24df9a28 2023-07-08 jrick default:
6902 24df9a28 2023-07-08 jrick usage_patch();
6903 24df9a28 2023-07-08 jrick /* NOTREACHED */
6904 24df9a28 2023-07-08 jrick }
6905 24df9a28 2023-07-08 jrick }
6906 24df9a28 2023-07-08 jrick
6907 24df9a28 2023-07-08 jrick argc -= optind;
6908 24df9a28 2023-07-08 jrick argv += optind;
6909 24df9a28 2023-07-08 jrick
6910 24df9a28 2023-07-08 jrick if (argc == 0) {
6911 24df9a28 2023-07-08 jrick error = patch_from_stdin(&patchfd);
6912 24df9a28 2023-07-08 jrick if (error)
6913 24df9a28 2023-07-08 jrick return error;
6914 24df9a28 2023-07-08 jrick } else if (argc == 1) {
6915 24df9a28 2023-07-08 jrick patchfd = open(argv[0], O_RDONLY);
6916 24df9a28 2023-07-08 jrick if (patchfd == -1) {
6917 24df9a28 2023-07-08 jrick error = got_error_from_errno2("open", argv[0]);
6918 24df9a28 2023-07-08 jrick return error;
6919 24df9a28 2023-07-08 jrick }
6920 24df9a28 2023-07-08 jrick if (fstat(patchfd, &sb) == -1) {
6921 24df9a28 2023-07-08 jrick error = got_error_from_errno2("fstat", argv[0]);
6922 24df9a28 2023-07-08 jrick goto done;
6923 24df9a28 2023-07-08 jrick }
6924 24df9a28 2023-07-08 jrick if (!S_ISREG(sb.st_mode)) {
6925 24df9a28 2023-07-08 jrick error = got_error_path(argv[0], GOT_ERR_BAD_FILETYPE);
6926 24df9a28 2023-07-08 jrick goto done;
6927 24df9a28 2023-07-08 jrick }
6928 24df9a28 2023-07-08 jrick } else
6929 24df9a28 2023-07-08 jrick usage_patch();
6930 24df9a28 2023-07-08 jrick
6931 24df9a28 2023-07-08 jrick if ((cwd = getcwd(NULL, 0)) == NULL) {
6932 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
6933 24df9a28 2023-07-08 jrick goto done;
6934 24df9a28 2023-07-08 jrick }
6935 24df9a28 2023-07-08 jrick
6936 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
6937 24df9a28 2023-07-08 jrick if (error != NULL)
6938 24df9a28 2023-07-08 jrick goto done;
6939 24df9a28 2023-07-08 jrick
6940 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
6941 24df9a28 2023-07-08 jrick if (error != NULL)
6942 24df9a28 2023-07-08 jrick goto done;
6943 24df9a28 2023-07-08 jrick
6944 24df9a28 2023-07-08 jrick const char *repo_path = got_worktree_get_repo_path(worktree);
6945 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6946 24df9a28 2023-07-08 jrick if (error != NULL)
6947 24df9a28 2023-07-08 jrick goto done;
6948 24df9a28 2023-07-08 jrick
6949 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
6950 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
6951 24df9a28 2023-07-08 jrick if (error != NULL)
6952 24df9a28 2023-07-08 jrick goto done;
6953 24df9a28 2023-07-08 jrick
6954 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
6955 24df9a28 2023-07-08 jrick if (error)
6956 24df9a28 2023-07-08 jrick goto done;
6957 24df9a28 2023-07-08 jrick
6958 24df9a28 2023-07-08 jrick if (commit_id_str != NULL) {
6959 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
6960 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
6961 24df9a28 2023-07-08 jrick if (error)
6962 24df9a28 2023-07-08 jrick goto done;
6963 24df9a28 2023-07-08 jrick }
6964 24df9a28 2023-07-08 jrick
6965 24df9a28 2023-07-08 jrick memset(&ppa, 0, sizeof(ppa));
6966 24df9a28 2023-07-08 jrick error = got_patch(patchfd, worktree, repo, nop, strip, reverse,
6967 24df9a28 2023-07-08 jrick commit_id, patch_progress, &ppa, check_cancelled, NULL);
6968 24df9a28 2023-07-08 jrick print_patch_progress_stats(&ppa);
6969 24df9a28 2023-07-08 jrick done:
6970 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
6971 24df9a28 2023-07-08 jrick free(commit_id);
6972 24df9a28 2023-07-08 jrick if (repo) {
6973 24df9a28 2023-07-08 jrick close_error = got_repo_close(repo);
6974 24df9a28 2023-07-08 jrick if (error == NULL)
6975 24df9a28 2023-07-08 jrick error = close_error;
6976 24df9a28 2023-07-08 jrick }
6977 24df9a28 2023-07-08 jrick if (worktree != NULL) {
6978 24df9a28 2023-07-08 jrick close_error = got_worktree_close(worktree);
6979 24df9a28 2023-07-08 jrick if (error == NULL)
6980 24df9a28 2023-07-08 jrick error = close_error;
6981 24df9a28 2023-07-08 jrick }
6982 24df9a28 2023-07-08 jrick if (pack_fds) {
6983 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
6984 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
6985 24df9a28 2023-07-08 jrick if (error == NULL)
6986 24df9a28 2023-07-08 jrick error = pack_err;
6987 24df9a28 2023-07-08 jrick }
6988 24df9a28 2023-07-08 jrick free(cwd);
6989 24df9a28 2023-07-08 jrick return error;
6990 24df9a28 2023-07-08 jrick }
6991 24df9a28 2023-07-08 jrick
6992 24df9a28 2023-07-08 jrick __dead static void
6993 24df9a28 2023-07-08 jrick usage_revert(void)
6994 24df9a28 2023-07-08 jrick {
6995 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s revert [-pR] [-F response-script] path ...\n",
6996 24df9a28 2023-07-08 jrick getprogname());
6997 24df9a28 2023-07-08 jrick exit(1);
6998 24df9a28 2023-07-08 jrick }
6999 24df9a28 2023-07-08 jrick
7000 24df9a28 2023-07-08 jrick static const struct got_error *
7001 24df9a28 2023-07-08 jrick revert_progress(void *arg, unsigned char status, const char *path)
7002 24df9a28 2023-07-08 jrick {
7003 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_UNVERSIONED)
7004 24df9a28 2023-07-08 jrick return NULL;
7005 24df9a28 2023-07-08 jrick
7006 24df9a28 2023-07-08 jrick while (path[0] == '/')
7007 24df9a28 2023-07-08 jrick path++;
7008 24df9a28 2023-07-08 jrick printf("%c %s\n", status, path);
7009 24df9a28 2023-07-08 jrick return NULL;
7010 24df9a28 2023-07-08 jrick }
7011 24df9a28 2023-07-08 jrick
7012 24df9a28 2023-07-08 jrick struct choose_patch_arg {
7013 24df9a28 2023-07-08 jrick FILE *patch_script_file;
7014 24df9a28 2023-07-08 jrick const char *action;
7015 24df9a28 2023-07-08 jrick };
7016 24df9a28 2023-07-08 jrick
7017 24df9a28 2023-07-08 jrick static const struct got_error *
7018 24df9a28 2023-07-08 jrick show_change(unsigned char status, const char *path, FILE *patch_file, int n,
7019 24df9a28 2023-07-08 jrick int nchanges, const char *action)
7020 24df9a28 2023-07-08 jrick {
7021 24df9a28 2023-07-08 jrick const struct got_error *err;
7022 24df9a28 2023-07-08 jrick char *line = NULL;
7023 24df9a28 2023-07-08 jrick size_t linesize = 0;
7024 24df9a28 2023-07-08 jrick ssize_t linelen;
7025 24df9a28 2023-07-08 jrick
7026 24df9a28 2023-07-08 jrick switch (status) {
7027 24df9a28 2023-07-08 jrick case GOT_STATUS_ADD:
7028 24df9a28 2023-07-08 jrick printf("A %s\n%s this addition? [y/n] ", path, action);
7029 24df9a28 2023-07-08 jrick break;
7030 24df9a28 2023-07-08 jrick case GOT_STATUS_DELETE:
7031 24df9a28 2023-07-08 jrick printf("D %s\n%s this deletion? [y/n] ", path, action);
7032 24df9a28 2023-07-08 jrick break;
7033 24df9a28 2023-07-08 jrick case GOT_STATUS_MODIFY:
7034 24df9a28 2023-07-08 jrick if (fseek(patch_file, 0L, SEEK_SET) == -1)
7035 24df9a28 2023-07-08 jrick return got_error_from_errno("fseek");
7036 24df9a28 2023-07-08 jrick printf(GOT_COMMIT_SEP_STR);
7037 24df9a28 2023-07-08 jrick while ((linelen = getline(&line, &linesize, patch_file)) != -1)
7038 24df9a28 2023-07-08 jrick printf("%s", line);
7039 24df9a28 2023-07-08 jrick if (linelen == -1 && ferror(patch_file)) {
7040 24df9a28 2023-07-08 jrick err = got_error_from_errno("getline");
7041 24df9a28 2023-07-08 jrick free(line);
7042 24df9a28 2023-07-08 jrick return err;
7043 24df9a28 2023-07-08 jrick }
7044 24df9a28 2023-07-08 jrick free(line);
7045 24df9a28 2023-07-08 jrick printf(GOT_COMMIT_SEP_STR);
7046 24df9a28 2023-07-08 jrick printf("M %s (change %d of %d)\n%s this change? [y/n/q] ",
7047 24df9a28 2023-07-08 jrick path, n, nchanges, action);
7048 24df9a28 2023-07-08 jrick break;
7049 24df9a28 2023-07-08 jrick default:
7050 24df9a28 2023-07-08 jrick return got_error_path(path, GOT_ERR_FILE_STATUS);
7051 24df9a28 2023-07-08 jrick }
7052 24df9a28 2023-07-08 jrick
7053 24df9a28 2023-07-08 jrick fflush(stdout);
7054 24df9a28 2023-07-08 jrick return NULL;
7055 24df9a28 2023-07-08 jrick }
7056 24df9a28 2023-07-08 jrick
7057 24df9a28 2023-07-08 jrick static const struct got_error *
7058 24df9a28 2023-07-08 jrick choose_patch(int *choice, void *arg, unsigned char status, const char *path,
7059 24df9a28 2023-07-08 jrick FILE *patch_file, int n, int nchanges)
7060 24df9a28 2023-07-08 jrick {
7061 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
7062 24df9a28 2023-07-08 jrick char *line = NULL;
7063 24df9a28 2023-07-08 jrick size_t linesize = 0;
7064 24df9a28 2023-07-08 jrick ssize_t linelen;
7065 24df9a28 2023-07-08 jrick int resp = ' ';
7066 24df9a28 2023-07-08 jrick struct choose_patch_arg *a = arg;
7067 24df9a28 2023-07-08 jrick
7068 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_NONE;
7069 24df9a28 2023-07-08 jrick
7070 24df9a28 2023-07-08 jrick if (a->patch_script_file) {
7071 24df9a28 2023-07-08 jrick char *nl;
7072 24df9a28 2023-07-08 jrick err = show_change(status, path, patch_file, n, nchanges,
7073 24df9a28 2023-07-08 jrick a->action);
7074 24df9a28 2023-07-08 jrick if (err)
7075 24df9a28 2023-07-08 jrick return err;
7076 24df9a28 2023-07-08 jrick linelen = getline(&line, &linesize, a->patch_script_file);
7077 24df9a28 2023-07-08 jrick if (linelen == -1) {
7078 24df9a28 2023-07-08 jrick if (ferror(a->patch_script_file))
7079 24df9a28 2023-07-08 jrick return got_error_from_errno("getline");
7080 24df9a28 2023-07-08 jrick return NULL;
7081 24df9a28 2023-07-08 jrick }
7082 24df9a28 2023-07-08 jrick nl = strchr(line, '\n');
7083 24df9a28 2023-07-08 jrick if (nl)
7084 24df9a28 2023-07-08 jrick *nl = '\0';
7085 24df9a28 2023-07-08 jrick if (strcmp(line, "y") == 0) {
7086 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_YES;
7087 24df9a28 2023-07-08 jrick printf("y\n");
7088 24df9a28 2023-07-08 jrick } else if (strcmp(line, "n") == 0) {
7089 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_NO;
7090 24df9a28 2023-07-08 jrick printf("n\n");
7091 24df9a28 2023-07-08 jrick } else if (strcmp(line, "q") == 0 &&
7092 24df9a28 2023-07-08 jrick status == GOT_STATUS_MODIFY) {
7093 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_QUIT;
7094 24df9a28 2023-07-08 jrick printf("q\n");
7095 24df9a28 2023-07-08 jrick } else
7096 24df9a28 2023-07-08 jrick printf("invalid response '%s'\n", line);
7097 24df9a28 2023-07-08 jrick free(line);
7098 24df9a28 2023-07-08 jrick return NULL;
7099 24df9a28 2023-07-08 jrick }
7100 24df9a28 2023-07-08 jrick
7101 24df9a28 2023-07-08 jrick while (resp != 'y' && resp != 'n' && resp != 'q') {
7102 24df9a28 2023-07-08 jrick err = show_change(status, path, patch_file, n, nchanges,
7103 24df9a28 2023-07-08 jrick a->action);
7104 24df9a28 2023-07-08 jrick if (err)
7105 24df9a28 2023-07-08 jrick return err;
7106 24df9a28 2023-07-08 jrick resp = getchar();
7107 24df9a28 2023-07-08 jrick if (resp == '\n')
7108 24df9a28 2023-07-08 jrick resp = getchar();
7109 24df9a28 2023-07-08 jrick if (status == GOT_STATUS_MODIFY) {
7110 24df9a28 2023-07-08 jrick if (resp != 'y' && resp != 'n' && resp != 'q') {
7111 24df9a28 2023-07-08 jrick printf("invalid response '%c'\n", resp);
7112 24df9a28 2023-07-08 jrick resp = ' ';
7113 24df9a28 2023-07-08 jrick }
7114 24df9a28 2023-07-08 jrick } else if (resp != 'y' && resp != 'n') {
7115 24df9a28 2023-07-08 jrick printf("invalid response '%c'\n", resp);
7116 24df9a28 2023-07-08 jrick resp = ' ';
7117 24df9a28 2023-07-08 jrick }
7118 24df9a28 2023-07-08 jrick }
7119 24df9a28 2023-07-08 jrick
7120 24df9a28 2023-07-08 jrick if (resp == 'y')
7121 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_YES;
7122 24df9a28 2023-07-08 jrick else if (resp == 'n')
7123 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_NO;
7124 24df9a28 2023-07-08 jrick else if (resp == 'q' && status == GOT_STATUS_MODIFY)
7125 24df9a28 2023-07-08 jrick *choice = GOT_PATCH_CHOICE_QUIT;
7126 24df9a28 2023-07-08 jrick
7127 24df9a28 2023-07-08 jrick return NULL;
7128 24df9a28 2023-07-08 jrick }
7129 24df9a28 2023-07-08 jrick
7130 24df9a28 2023-07-08 jrick struct wt_commitable_path_arg {
7131 24df9a28 2023-07-08 jrick struct got_pathlist_head *commit_paths;
7132 24df9a28 2023-07-08 jrick int *has_changes;
7133 24df9a28 2023-07-08 jrick };
7134 24df9a28 2023-07-08 jrick
7135 24df9a28 2023-07-08 jrick /*
7136 24df9a28 2023-07-08 jrick * Shortcut work tree status callback to determine if the set of paths scanned
7137 24df9a28 2023-07-08 jrick * has at least one versioned path that is being modified and, if not NULL, is
7138 24df9a28 2023-07-08 jrick * in the arg->commit_paths list. Set arg and return GOT_ERR_FILE_MODIFIED as
7139 24df9a28 2023-07-08 jrick * soon as a path is passed with a status that satisfies this criteria.
7140 24df9a28 2023-07-08 jrick */
7141 24df9a28 2023-07-08 jrick static const struct got_error *
7142 24df9a28 2023-07-08 jrick worktree_has_commitable_path(void *arg, unsigned char status,
7143 24df9a28 2023-07-08 jrick unsigned char staged_status, const char *path,
7144 24df9a28 2023-07-08 jrick struct got_object_id *blob_id, struct got_object_id *staged_blob_id,
7145 24df9a28 2023-07-08 jrick struct got_object_id *commit_id, int dirfd, const char *de_name)
7146 24df9a28 2023-07-08 jrick {
7147 24df9a28 2023-07-08 jrick struct wt_commitable_path_arg *a = arg;
7148 24df9a28 2023-07-08 jrick
7149 24df9a28 2023-07-08 jrick if (status == staged_status && (status == GOT_STATUS_DELETE))
7150 24df9a28 2023-07-08 jrick status = GOT_STATUS_NO_CHANGE;
7151 24df9a28 2023-07-08 jrick
7152 24df9a28 2023-07-08 jrick if (!(status == GOT_STATUS_NO_CHANGE ||
7153 24df9a28 2023-07-08 jrick status == GOT_STATUS_UNVERSIONED) ||
7154 24df9a28 2023-07-08 jrick staged_status != GOT_STATUS_NO_CHANGE) {
7155 24df9a28 2023-07-08 jrick if (a->commit_paths != NULL) {
7156 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
7157 24df9a28 2023-07-08 jrick
7158 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, a->commit_paths, entry) {
7159 24df9a28 2023-07-08 jrick if (strncmp(path, pe->path,
7160 24df9a28 2023-07-08 jrick pe->path_len) == 0) {
7161 24df9a28 2023-07-08 jrick *a->has_changes = 1;
7162 24df9a28 2023-07-08 jrick break;
7163 24df9a28 2023-07-08 jrick }
7164 24df9a28 2023-07-08 jrick }
7165 24df9a28 2023-07-08 jrick } else
7166 24df9a28 2023-07-08 jrick *a->has_changes = 1;
7167 24df9a28 2023-07-08 jrick
7168 24df9a28 2023-07-08 jrick if (*a->has_changes)
7169 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_FILE_MODIFIED);
7170 24df9a28 2023-07-08 jrick }
7171 24df9a28 2023-07-08 jrick
7172 24df9a28 2023-07-08 jrick return NULL;
7173 24df9a28 2023-07-08 jrick }
7174 24df9a28 2023-07-08 jrick
7175 24df9a28 2023-07-08 jrick /*
7176 24df9a28 2023-07-08 jrick * Check that the changeset of the commit identified by id is
7177 24df9a28 2023-07-08 jrick * comprised of at least one modified path that is being committed.
7178 24df9a28 2023-07-08 jrick */
7179 24df9a28 2023-07-08 jrick static const struct got_error *
7180 24df9a28 2023-07-08 jrick commit_path_changed_in_worktree(struct wt_commitable_path_arg *wcpa,
7181 24df9a28 2023-07-08 jrick struct got_object_id *id, struct got_worktree *worktree,
7182 24df9a28 2023-07-08 jrick struct got_repository *repo)
7183 24df9a28 2023-07-08 jrick {
7184 24df9a28 2023-07-08 jrick const struct got_error *err;
7185 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
7186 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL, *pcommit = NULL;
7187 24df9a28 2023-07-08 jrick struct got_tree_object *tree = NULL, *ptree = NULL;
7188 24df9a28 2023-07-08 jrick struct got_object_qid *pid;
7189 24df9a28 2023-07-08 jrick
7190 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
7191 24df9a28 2023-07-08 jrick
7192 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, id);
7193 24df9a28 2023-07-08 jrick if (err)
7194 24df9a28 2023-07-08 jrick goto done;
7195 24df9a28 2023-07-08 jrick
7196 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree, repo,
7197 24df9a28 2023-07-08 jrick got_object_commit_get_tree_id(commit));
7198 24df9a28 2023-07-08 jrick if (err)
7199 24df9a28 2023-07-08 jrick goto done;
7200 24df9a28 2023-07-08 jrick
7201 24df9a28 2023-07-08 jrick pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
7202 24df9a28 2023-07-08 jrick if (pid != NULL) {
7203 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&pcommit, repo, &pid->id);
7204 24df9a28 2023-07-08 jrick if (err)
7205 24df9a28 2023-07-08 jrick goto done;
7206 24df9a28 2023-07-08 jrick
7207 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&ptree, repo,
7208 24df9a28 2023-07-08 jrick got_object_commit_get_tree_id(pcommit));
7209 24df9a28 2023-07-08 jrick if (err)
7210 24df9a28 2023-07-08 jrick goto done;
7211 24df9a28 2023-07-08 jrick }
7212 24df9a28 2023-07-08 jrick
7213 24df9a28 2023-07-08 jrick err = got_diff_tree(ptree, tree, NULL, NULL, -1, -1, "", "", repo,
7214 24df9a28 2023-07-08 jrick got_diff_tree_collect_changed_paths, &paths, 0);
7215 24df9a28 2023-07-08 jrick if (err)
7216 24df9a28 2023-07-08 jrick goto done;
7217 24df9a28 2023-07-08 jrick
7218 24df9a28 2023-07-08 jrick err = got_worktree_status(worktree, &paths, repo, 0,
7219 24df9a28 2023-07-08 jrick worktree_has_commitable_path, wcpa, check_cancelled, NULL);
7220 24df9a28 2023-07-08 jrick if (err && err->code == GOT_ERR_FILE_MODIFIED) {
7221 24df9a28 2023-07-08 jrick /*
7222 24df9a28 2023-07-08 jrick * At least one changed path in the referenced commit is
7223 24df9a28 2023-07-08 jrick * modified in the work tree, that's all we need to know!
7224 24df9a28 2023-07-08 jrick */
7225 24df9a28 2023-07-08 jrick err = NULL;
7226 24df9a28 2023-07-08 jrick }
7227 24df9a28 2023-07-08 jrick
7228 24df9a28 2023-07-08 jrick done:
7229 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_ALL);
7230 24df9a28 2023-07-08 jrick if (commit)
7231 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
7232 24df9a28 2023-07-08 jrick if (pcommit)
7233 24df9a28 2023-07-08 jrick got_object_commit_close(pcommit);
7234 24df9a28 2023-07-08 jrick if (tree)
7235 24df9a28 2023-07-08 jrick got_object_tree_close(tree);
7236 24df9a28 2023-07-08 jrick if (ptree)
7237 24df9a28 2023-07-08 jrick got_object_tree_close(ptree);
7238 24df9a28 2023-07-08 jrick return err;
7239 24df9a28 2023-07-08 jrick }
7240 24df9a28 2023-07-08 jrick
7241 24df9a28 2023-07-08 jrick /*
7242 24df9a28 2023-07-08 jrick * Remove any "logmsg" reference comprised entirely of paths that have
7243 24df9a28 2023-07-08 jrick * been reverted in this work tree. If any path in the logmsg ref changeset
7244 24df9a28 2023-07-08 jrick * remains in a changed state in the worktree, do not remove the reference.
7245 24df9a28 2023-07-08 jrick */
7246 24df9a28 2023-07-08 jrick static const struct got_error *
7247 24df9a28 2023-07-08 jrick rm_logmsg_ref(struct got_worktree *worktree, struct got_repository *repo)
7248 24df9a28 2023-07-08 jrick {
7249 24df9a28 2023-07-08 jrick const struct got_error *err;
7250 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
7251 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
7252 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
7253 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
7254 24df9a28 2023-07-08 jrick struct wt_commitable_path_arg wcpa;
7255 24df9a28 2023-07-08 jrick char *uuidstr = NULL;
7256 24df9a28 2023-07-08 jrick
7257 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
7258 24df9a28 2023-07-08 jrick
7259 24df9a28 2023-07-08 jrick err = got_worktree_get_uuid(&uuidstr, worktree);
7260 24df9a28 2023-07-08 jrick if (err)
7261 24df9a28 2023-07-08 jrick goto done;
7262 24df9a28 2023-07-08 jrick
7263 24df9a28 2023-07-08 jrick err = got_ref_list(&refs, repo, "refs/got/worktree",
7264 24df9a28 2023-07-08 jrick got_ref_cmp_by_name, repo);
7265 24df9a28 2023-07-08 jrick if (err)
7266 24df9a28 2023-07-08 jrick goto done;
7267 24df9a28 2023-07-08 jrick
7268 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &refs, entry) {
7269 24df9a28 2023-07-08 jrick const char *refname;
7270 24df9a28 2023-07-08 jrick int has_changes = 0;
7271 24df9a28 2023-07-08 jrick
7272 24df9a28 2023-07-08 jrick refname = got_ref_get_name(re->ref);
7273 24df9a28 2023-07-08 jrick
7274 24df9a28 2023-07-08 jrick if (!strncmp(refname, GOT_WORKTREE_CHERRYPICK_REF_PREFIX,
7275 24df9a28 2023-07-08 jrick GOT_WORKTREE_CHERRYPICK_REF_PREFIX_LEN))
7276 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_CHERRYPICK_REF_PREFIX_LEN + 1;
7277 24df9a28 2023-07-08 jrick else if (!strncmp(refname, GOT_WORKTREE_BACKOUT_REF_PREFIX,
7278 24df9a28 2023-07-08 jrick GOT_WORKTREE_BACKOUT_REF_PREFIX_LEN))
7279 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_BACKOUT_REF_PREFIX_LEN + 1;
7280 24df9a28 2023-07-08 jrick else
7281 24df9a28 2023-07-08 jrick continue;
7282 24df9a28 2023-07-08 jrick
7283 24df9a28 2023-07-08 jrick if (strncmp(refname, uuidstr, GOT_WORKTREE_UUID_STRLEN) == 0)
7284 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_UUID_STRLEN + 1; /* skip '-' */
7285 24df9a28 2023-07-08 jrick else
7286 24df9a28 2023-07-08 jrick continue;
7287 24df9a28 2023-07-08 jrick
7288 24df9a28 2023-07-08 jrick err = got_repo_match_object_id(&commit_id, NULL, refname,
7289 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, NULL, repo);
7290 24df9a28 2023-07-08 jrick if (err)
7291 24df9a28 2023-07-08 jrick goto done;
7292 24df9a28 2023-07-08 jrick
7293 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, commit_id);
7294 24df9a28 2023-07-08 jrick if (err)
7295 24df9a28 2023-07-08 jrick goto done;
7296 24df9a28 2023-07-08 jrick
7297 24df9a28 2023-07-08 jrick wcpa.commit_paths = NULL;
7298 24df9a28 2023-07-08 jrick wcpa.has_changes = &has_changes;
7299 24df9a28 2023-07-08 jrick
7300 24df9a28 2023-07-08 jrick err = commit_path_changed_in_worktree(&wcpa, commit_id,
7301 24df9a28 2023-07-08 jrick worktree, repo);
7302 24df9a28 2023-07-08 jrick if (err)
7303 24df9a28 2023-07-08 jrick goto done;
7304 24df9a28 2023-07-08 jrick
7305 24df9a28 2023-07-08 jrick if (!has_changes) {
7306 24df9a28 2023-07-08 jrick err = got_ref_delete(re->ref, repo);
7307 24df9a28 2023-07-08 jrick if (err)
7308 24df9a28 2023-07-08 jrick goto done;
7309 24df9a28 2023-07-08 jrick }
7310 24df9a28 2023-07-08 jrick
7311 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
7312 24df9a28 2023-07-08 jrick commit = NULL;
7313 24df9a28 2023-07-08 jrick free(commit_id);
7314 24df9a28 2023-07-08 jrick commit_id = NULL;
7315 24df9a28 2023-07-08 jrick }
7316 24df9a28 2023-07-08 jrick
7317 24df9a28 2023-07-08 jrick done:
7318 24df9a28 2023-07-08 jrick free(uuidstr);
7319 24df9a28 2023-07-08 jrick free(commit_id);
7320 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
7321 24df9a28 2023-07-08 jrick if (commit)
7322 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
7323 24df9a28 2023-07-08 jrick return err;
7324 24df9a28 2023-07-08 jrick }
7325 24df9a28 2023-07-08 jrick
7326 24df9a28 2023-07-08 jrick static const struct got_error *
7327 24df9a28 2023-07-08 jrick cmd_revert(int argc, char *argv[])
7328 24df9a28 2023-07-08 jrick {
7329 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
7330 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
7331 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
7332 24df9a28 2023-07-08 jrick char *cwd = NULL, *path = NULL;
7333 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
7334 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
7335 24df9a28 2023-07-08 jrick int ch, can_recurse = 0, pflag = 0;
7336 24df9a28 2023-07-08 jrick FILE *patch_script_file = NULL;
7337 24df9a28 2023-07-08 jrick const char *patch_script_path = NULL;
7338 24df9a28 2023-07-08 jrick struct choose_patch_arg cpa;
7339 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
7340 24df9a28 2023-07-08 jrick
7341 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
7342 24df9a28 2023-07-08 jrick
7343 24df9a28 2023-07-08 jrick #ifndef PROFILE
7344 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
7345 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
7346 24df9a28 2023-07-08 jrick err(1, "pledge");
7347 24df9a28 2023-07-08 jrick #endif
7348 24df9a28 2023-07-08 jrick
7349 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "F:pR")) != -1) {
7350 24df9a28 2023-07-08 jrick switch (ch) {
7351 24df9a28 2023-07-08 jrick case 'F':
7352 24df9a28 2023-07-08 jrick patch_script_path = optarg;
7353 24df9a28 2023-07-08 jrick break;
7354 24df9a28 2023-07-08 jrick case 'p':
7355 24df9a28 2023-07-08 jrick pflag = 1;
7356 24df9a28 2023-07-08 jrick break;
7357 24df9a28 2023-07-08 jrick case 'R':
7358 24df9a28 2023-07-08 jrick can_recurse = 1;
7359 24df9a28 2023-07-08 jrick break;
7360 24df9a28 2023-07-08 jrick default:
7361 24df9a28 2023-07-08 jrick usage_revert();
7362 24df9a28 2023-07-08 jrick /* NOTREACHED */
7363 24df9a28 2023-07-08 jrick }
7364 24df9a28 2023-07-08 jrick }
7365 24df9a28 2023-07-08 jrick
7366 24df9a28 2023-07-08 jrick argc -= optind;
7367 24df9a28 2023-07-08 jrick argv += optind;
7368 24df9a28 2023-07-08 jrick
7369 24df9a28 2023-07-08 jrick if (argc < 1)
7370 24df9a28 2023-07-08 jrick usage_revert();
7371 24df9a28 2023-07-08 jrick if (patch_script_path && !pflag)
7372 24df9a28 2023-07-08 jrick errx(1, "-F option can only be used together with -p option");
7373 24df9a28 2023-07-08 jrick
7374 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
7375 24df9a28 2023-07-08 jrick if (cwd == NULL) {
7376 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
7377 24df9a28 2023-07-08 jrick goto done;
7378 24df9a28 2023-07-08 jrick }
7379 24df9a28 2023-07-08 jrick
7380 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
7381 24df9a28 2023-07-08 jrick if (error != NULL)
7382 24df9a28 2023-07-08 jrick goto done;
7383 24df9a28 2023-07-08 jrick
7384 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
7385 24df9a28 2023-07-08 jrick if (error) {
7386 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
7387 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "revert", cwd);
7388 24df9a28 2023-07-08 jrick goto done;
7389 24df9a28 2023-07-08 jrick }
7390 24df9a28 2023-07-08 jrick
7391 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
7392 24df9a28 2023-07-08 jrick NULL, pack_fds);
7393 24df9a28 2023-07-08 jrick if (error != NULL)
7394 24df9a28 2023-07-08 jrick goto done;
7395 24df9a28 2023-07-08 jrick
7396 24df9a28 2023-07-08 jrick if (patch_script_path) {
7397 24df9a28 2023-07-08 jrick patch_script_file = fopen(patch_script_path, "re");
7398 24df9a28 2023-07-08 jrick if (patch_script_file == NULL) {
7399 24df9a28 2023-07-08 jrick error = got_error_from_errno2("fopen",
7400 24df9a28 2023-07-08 jrick patch_script_path);
7401 24df9a28 2023-07-08 jrick goto done;
7402 24df9a28 2023-07-08 jrick }
7403 24df9a28 2023-07-08 jrick }
7404 24df9a28 2023-07-08 jrick
7405 24df9a28 2023-07-08 jrick /*
7406 24df9a28 2023-07-08 jrick * XXX "c" perm needed on repo dir to delete merge references.
7407 24df9a28 2023-07-08 jrick */
7408 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
7409 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
7410 24df9a28 2023-07-08 jrick if (error)
7411 24df9a28 2023-07-08 jrick goto done;
7412 24df9a28 2023-07-08 jrick
7413 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
7414 24df9a28 2023-07-08 jrick if (error)
7415 24df9a28 2023-07-08 jrick goto done;
7416 24df9a28 2023-07-08 jrick
7417 24df9a28 2023-07-08 jrick if (!can_recurse) {
7418 24df9a28 2023-07-08 jrick char *ondisk_path;
7419 24df9a28 2023-07-08 jrick struct stat sb;
7420 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &paths, entry) {
7421 24df9a28 2023-07-08 jrick if (asprintf(&ondisk_path, "%s/%s",
7422 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree),
7423 24df9a28 2023-07-08 jrick pe->path) == -1) {
7424 24df9a28 2023-07-08 jrick error = got_error_from_errno("asprintf");
7425 24df9a28 2023-07-08 jrick goto done;
7426 24df9a28 2023-07-08 jrick }
7427 24df9a28 2023-07-08 jrick if (lstat(ondisk_path, &sb) == -1) {
7428 24df9a28 2023-07-08 jrick if (errno == ENOENT) {
7429 24df9a28 2023-07-08 jrick free(ondisk_path);
7430 24df9a28 2023-07-08 jrick continue;
7431 24df9a28 2023-07-08 jrick }
7432 24df9a28 2023-07-08 jrick error = got_error_from_errno2("lstat",
7433 24df9a28 2023-07-08 jrick ondisk_path);
7434 24df9a28 2023-07-08 jrick free(ondisk_path);
7435 24df9a28 2023-07-08 jrick goto done;
7436 24df9a28 2023-07-08 jrick }
7437 24df9a28 2023-07-08 jrick free(ondisk_path);
7438 24df9a28 2023-07-08 jrick if (S_ISDIR(sb.st_mode)) {
7439 24df9a28 2023-07-08 jrick error = got_error_msg(GOT_ERR_BAD_PATH,
7440 24df9a28 2023-07-08 jrick "reverting directories requires -R option");
7441 24df9a28 2023-07-08 jrick goto done;
7442 24df9a28 2023-07-08 jrick }
7443 24df9a28 2023-07-08 jrick }
7444 24df9a28 2023-07-08 jrick }
7445 24df9a28 2023-07-08 jrick
7446 24df9a28 2023-07-08 jrick cpa.patch_script_file = patch_script_file;
7447 24df9a28 2023-07-08 jrick cpa.action = "revert";
7448 24df9a28 2023-07-08 jrick error = got_worktree_revert(worktree, &paths, revert_progress, NULL,
7449 24df9a28 2023-07-08 jrick pflag ? choose_patch : NULL, &cpa, repo);
7450 24df9a28 2023-07-08 jrick
7451 24df9a28 2023-07-08 jrick error = rm_logmsg_ref(worktree, repo);
7452 24df9a28 2023-07-08 jrick done:
7453 24df9a28 2023-07-08 jrick if (patch_script_file && fclose(patch_script_file) == EOF &&
7454 24df9a28 2023-07-08 jrick error == NULL)
7455 24df9a28 2023-07-08 jrick error = got_error_from_errno2("fclose", patch_script_path);
7456 24df9a28 2023-07-08 jrick if (repo) {
7457 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
7458 24df9a28 2023-07-08 jrick if (error == NULL)
7459 24df9a28 2023-07-08 jrick error = close_err;
7460 24df9a28 2023-07-08 jrick }
7461 24df9a28 2023-07-08 jrick if (worktree)
7462 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
7463 24df9a28 2023-07-08 jrick if (pack_fds) {
7464 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
7465 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
7466 24df9a28 2023-07-08 jrick if (error == NULL)
7467 24df9a28 2023-07-08 jrick error = pack_err;
7468 24df9a28 2023-07-08 jrick }
7469 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
7470 24df9a28 2023-07-08 jrick free(path);
7471 24df9a28 2023-07-08 jrick free(cwd);
7472 24df9a28 2023-07-08 jrick return error;
7473 24df9a28 2023-07-08 jrick }
7474 24df9a28 2023-07-08 jrick
7475 24df9a28 2023-07-08 jrick __dead static void
7476 24df9a28 2023-07-08 jrick usage_commit(void)
7477 24df9a28 2023-07-08 jrick {
7478 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s commit [-CNnS] [-A author] [-F path] "
7479 24df9a28 2023-07-08 jrick "[-m message] [path ...]\n", getprogname());
7480 24df9a28 2023-07-08 jrick exit(1);
7481 24df9a28 2023-07-08 jrick }
7482 24df9a28 2023-07-08 jrick
7483 24df9a28 2023-07-08 jrick struct collect_commit_logmsg_arg {
7484 24df9a28 2023-07-08 jrick const char *cmdline_log;
7485 24df9a28 2023-07-08 jrick const char *prepared_log;
7486 24df9a28 2023-07-08 jrick const char *merged_log;
7487 24df9a28 2023-07-08 jrick int non_interactive;
7488 24df9a28 2023-07-08 jrick const char *editor;
7489 24df9a28 2023-07-08 jrick const char *worktree_path;
7490 24df9a28 2023-07-08 jrick const char *repo_path;
7491 24df9a28 2023-07-08 jrick const char *dial_proto;
7492 24df9a28 2023-07-08 jrick char *logmsg_path;
7493 24df9a28 2023-07-08 jrick };
7494 24df9a28 2023-07-08 jrick
7495 24df9a28 2023-07-08 jrick static const struct got_error *
7496 24df9a28 2023-07-08 jrick read_prepared_logmsg(char **logmsg, const char *path)
7497 24df9a28 2023-07-08 jrick {
7498 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
7499 24df9a28 2023-07-08 jrick FILE *f = NULL;
7500 24df9a28 2023-07-08 jrick struct stat sb;
7501 24df9a28 2023-07-08 jrick size_t r;
7502 24df9a28 2023-07-08 jrick
7503 24df9a28 2023-07-08 jrick *logmsg = NULL;
7504 24df9a28 2023-07-08 jrick memset(&sb, 0, sizeof(sb));
7505 24df9a28 2023-07-08 jrick
7506 24df9a28 2023-07-08 jrick f = fopen(path, "re");
7507 24df9a28 2023-07-08 jrick if (f == NULL)
7508 24df9a28 2023-07-08 jrick return got_error_from_errno2("fopen", path);
7509 24df9a28 2023-07-08 jrick
7510 24df9a28 2023-07-08 jrick if (fstat(fileno(f), &sb) == -1) {
7511 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fstat", path);
7512 24df9a28 2023-07-08 jrick goto done;
7513 24df9a28 2023-07-08 jrick }
7514 24df9a28 2023-07-08 jrick if (sb.st_size == 0) {
7515 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_COMMIT_MSG_EMPTY);
7516 24df9a28 2023-07-08 jrick goto done;
7517 24df9a28 2023-07-08 jrick }
7518 24df9a28 2023-07-08 jrick
7519 24df9a28 2023-07-08 jrick *logmsg = malloc(sb.st_size + 1);
7520 24df9a28 2023-07-08 jrick if (*logmsg == NULL) {
7521 24df9a28 2023-07-08 jrick err = got_error_from_errno("malloc");
7522 24df9a28 2023-07-08 jrick goto done;
7523 24df9a28 2023-07-08 jrick }
7524 24df9a28 2023-07-08 jrick
7525 24df9a28 2023-07-08 jrick r = fread(*logmsg, 1, sb.st_size, f);
7526 24df9a28 2023-07-08 jrick if (r != sb.st_size) {
7527 24df9a28 2023-07-08 jrick if (ferror(f))
7528 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fread", path);
7529 24df9a28 2023-07-08 jrick else
7530 24df9a28 2023-07-08 jrick err = got_error(GOT_ERR_IO);
7531 24df9a28 2023-07-08 jrick goto done;
7532 24df9a28 2023-07-08 jrick }
7533 24df9a28 2023-07-08 jrick (*logmsg)[sb.st_size] = '\0';
7534 24df9a28 2023-07-08 jrick done:
7535 24df9a28 2023-07-08 jrick if (fclose(f) == EOF && err == NULL)
7536 24df9a28 2023-07-08 jrick err = got_error_from_errno2("fclose", path);
7537 24df9a28 2023-07-08 jrick if (err) {
7538 24df9a28 2023-07-08 jrick free(*logmsg);
7539 24df9a28 2023-07-08 jrick *logmsg = NULL;
7540 24df9a28 2023-07-08 jrick }
7541 24df9a28 2023-07-08 jrick return err;
7542 24df9a28 2023-07-08 jrick }
7543 24df9a28 2023-07-08 jrick
7544 24df9a28 2023-07-08 jrick static const struct got_error *
7545 24df9a28 2023-07-08 jrick collect_commit_logmsg(struct got_pathlist_head *commitable_paths,
7546 24df9a28 2023-07-08 jrick const char *diff_path, char **logmsg, void *arg)
7547 24df9a28 2023-07-08 jrick {
7548 24df9a28 2023-07-08 jrick char *initial_content = NULL;
7549 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
7550 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
7551 24df9a28 2023-07-08 jrick char *template = NULL;
7552 24df9a28 2023-07-08 jrick char *prepared_msg = NULL, *merged_msg = NULL;
7553 24df9a28 2023-07-08 jrick struct collect_commit_logmsg_arg *a = arg;
7554 24df9a28 2023-07-08 jrick int initial_content_len;
7555 24df9a28 2023-07-08 jrick int fd = -1;
7556 24df9a28 2023-07-08 jrick size_t len;
7557 24df9a28 2023-07-08 jrick
7558 24df9a28 2023-07-08 jrick /* if a message was specified on the command line, just use it */
7559 24df9a28 2023-07-08 jrick if (a->cmdline_log != NULL && *a->cmdline_log != '\0') {
7560 24df9a28 2023-07-08 jrick len = strlen(a->cmdline_log) + 1;
7561 24df9a28 2023-07-08 jrick *logmsg = malloc(len + 1);
7562 24df9a28 2023-07-08 jrick if (*logmsg == NULL)
7563 24df9a28 2023-07-08 jrick return got_error_from_errno("malloc");
7564 24df9a28 2023-07-08 jrick strlcpy(*logmsg, a->cmdline_log, len);
7565 24df9a28 2023-07-08 jrick return NULL;
7566 24df9a28 2023-07-08 jrick } else if (a->prepared_log != NULL && a->non_interactive)
7567 24df9a28 2023-07-08 jrick return read_prepared_logmsg(logmsg, a->prepared_log);
7568 24df9a28 2023-07-08 jrick
7569 24df9a28 2023-07-08 jrick if (asprintf(&template, "%s/logmsg", a->worktree_path) == -1)
7570 24df9a28 2023-07-08 jrick return got_error_from_errno("asprintf");
7571 24df9a28 2023-07-08 jrick
7572 24df9a28 2023-07-08 jrick err = got_opentemp_named_fd(&a->logmsg_path, &fd, template, "");
7573 24df9a28 2023-07-08 jrick if (err)
7574 24df9a28 2023-07-08 jrick goto done;
7575 24df9a28 2023-07-08 jrick
7576 24df9a28 2023-07-08 jrick if (a->prepared_log) {
7577 24df9a28 2023-07-08 jrick err = read_prepared_logmsg(&prepared_msg, a->prepared_log);
7578 24df9a28 2023-07-08 jrick if (err)
7579 24df9a28 2023-07-08 jrick goto done;
7580 24df9a28 2023-07-08 jrick } else if (a->merged_log) {
7581 24df9a28 2023-07-08 jrick err = read_prepared_logmsg(&merged_msg, a->merged_log);
7582 24df9a28 2023-07-08 jrick if (err)
7583 24df9a28 2023-07-08 jrick goto done;
7584 24df9a28 2023-07-08 jrick }
7585 24df9a28 2023-07-08 jrick
7586 24df9a28 2023-07-08 jrick initial_content_len = asprintf(&initial_content,
7587 24df9a28 2023-07-08 jrick "%s%s\n# changes to be committed:\n",
7588 24df9a28 2023-07-08 jrick prepared_msg ? prepared_msg : "",
7589 24df9a28 2023-07-08 jrick merged_msg ? merged_msg : "");
7590 24df9a28 2023-07-08 jrick if (initial_content_len == -1) {
7591 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
7592 24df9a28 2023-07-08 jrick goto done;
7593 24df9a28 2023-07-08 jrick }
7594 24df9a28 2023-07-08 jrick
7595 24df9a28 2023-07-08 jrick if (write(fd, initial_content, initial_content_len) == -1) {
7596 24df9a28 2023-07-08 jrick err = got_error_from_errno2("write", a->logmsg_path);
7597 24df9a28 2023-07-08 jrick goto done;
7598 24df9a28 2023-07-08 jrick }
7599 24df9a28 2023-07-08 jrick
7600 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, commitable_paths, entry) {
7601 24df9a28 2023-07-08 jrick struct got_commitable *ct = pe->data;
7602 24df9a28 2023-07-08 jrick dprintf(fd, "# %c %s\n",
7603 24df9a28 2023-07-08 jrick got_commitable_get_status(ct),
7604 24df9a28 2023-07-08 jrick got_commitable_get_path(ct));
7605 24df9a28 2023-07-08 jrick }
7606 24df9a28 2023-07-08 jrick
7607 24df9a28 2023-07-08 jrick if (diff_path) {
7608 24df9a28 2023-07-08 jrick dprintf(fd, "# detailed changes can be viewed in %s\n",
7609 24df9a28 2023-07-08 jrick diff_path);
7610 24df9a28 2023-07-08 jrick }
7611 24df9a28 2023-07-08 jrick
7612 24df9a28 2023-07-08 jrick if (close(fd) == -1) {
7613 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", a->logmsg_path);
7614 24df9a28 2023-07-08 jrick goto done;
7615 24df9a28 2023-07-08 jrick }
7616 24df9a28 2023-07-08 jrick fd = -1;
7617 24df9a28 2023-07-08 jrick
7618 24df9a28 2023-07-08 jrick err = edit_logmsg(logmsg, a->editor, a->logmsg_path, initial_content,
7619 24df9a28 2023-07-08 jrick initial_content_len, a->prepared_log ? 0 : 1);
7620 24df9a28 2023-07-08 jrick done:
7621 24df9a28 2023-07-08 jrick free(initial_content);
7622 24df9a28 2023-07-08 jrick free(template);
7623 24df9a28 2023-07-08 jrick free(prepared_msg);
7624 24df9a28 2023-07-08 jrick free(merged_msg);
7625 24df9a28 2023-07-08 jrick
7626 24df9a28 2023-07-08 jrick if (fd != -1 && close(fd) == -1 && err == NULL)
7627 24df9a28 2023-07-08 jrick err = got_error_from_errno2("close", a->logmsg_path);
7628 24df9a28 2023-07-08 jrick
7629 24df9a28 2023-07-08 jrick /* Editor is done; we can now apply unveil(2) */
7630 24df9a28 2023-07-08 jrick if (err == NULL)
7631 24df9a28 2023-07-08 jrick err = got_dial_apply_unveil(a->dial_proto);
7632 24df9a28 2023-07-08 jrick if (err == NULL)
7633 24df9a28 2023-07-08 jrick err = apply_unveil(a->repo_path, 0, a->worktree_path);
7634 24df9a28 2023-07-08 jrick if (err) {
7635 24df9a28 2023-07-08 jrick free(*logmsg);
7636 24df9a28 2023-07-08 jrick *logmsg = NULL;
7637 24df9a28 2023-07-08 jrick }
7638 24df9a28 2023-07-08 jrick return err;
7639 24df9a28 2023-07-08 jrick }
7640 24df9a28 2023-07-08 jrick
7641 24df9a28 2023-07-08 jrick static const struct got_error *
7642 24df9a28 2023-07-08 jrick cat_logmsg(FILE *f, struct got_commit_object *commit, const char *idstr,
7643 24df9a28 2023-07-08 jrick const char *type, int has_content)
7644 24df9a28 2023-07-08 jrick {
7645 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
7646 24df9a28 2023-07-08 jrick char *logmsg = NULL;
7647 24df9a28 2023-07-08 jrick
7648 24df9a28 2023-07-08 jrick err = got_object_commit_get_logmsg(&logmsg, commit);
7649 24df9a28 2023-07-08 jrick if (err)
7650 24df9a28 2023-07-08 jrick return err;
7651 24df9a28 2023-07-08 jrick
7652 24df9a28 2023-07-08 jrick if (fprintf(f, "%s# log message of %s commit %s:%s",
7653 24df9a28 2023-07-08 jrick has_content ? "\n" : "", type, idstr, logmsg) < 0)
7654 24df9a28 2023-07-08 jrick err = got_ferror(f, GOT_ERR_IO);
7655 24df9a28 2023-07-08 jrick
7656 24df9a28 2023-07-08 jrick free(logmsg);
7657 24df9a28 2023-07-08 jrick return err;
7658 24df9a28 2023-07-08 jrick }
7659 24df9a28 2023-07-08 jrick
7660 24df9a28 2023-07-08 jrick /*
7661 24df9a28 2023-07-08 jrick * Lookup "logmsg" references of backed-out and cherrypicked commits
7662 24df9a28 2023-07-08 jrick * belonging to the current work tree. If found, and the worktree has
7663 24df9a28 2023-07-08 jrick * at least one modified file that was changed in the referenced commit,
7664 24df9a28 2023-07-08 jrick * add its log message to a new temporary file at *logmsg_path.
7665 24df9a28 2023-07-08 jrick * Add all refs found to matched_refs to be scheduled for removal on
7666 24df9a28 2023-07-08 jrick * successful commit.
7667 24df9a28 2023-07-08 jrick */
7668 24df9a28 2023-07-08 jrick static const struct got_error *
7669 24df9a28 2023-07-08 jrick lookup_logmsg_ref(char **logmsg_path, struct got_pathlist_head *paths,
7670 24df9a28 2023-07-08 jrick struct got_reflist_head *matched_refs, struct got_worktree *worktree,
7671 24df9a28 2023-07-08 jrick struct got_repository *repo)
7672 24df9a28 2023-07-08 jrick {
7673 24df9a28 2023-07-08 jrick const struct got_error *err;
7674 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
7675 24df9a28 2023-07-08 jrick struct got_object_id *id = NULL;
7676 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
7677 24df9a28 2023-07-08 jrick struct got_reflist_entry *re, *re_match;
7678 24df9a28 2023-07-08 jrick FILE *f = NULL;
7679 24df9a28 2023-07-08 jrick char *uuidstr = NULL;
7680 24df9a28 2023-07-08 jrick int added_logmsg = 0;
7681 24df9a28 2023-07-08 jrick
7682 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
7683 24df9a28 2023-07-08 jrick
7684 24df9a28 2023-07-08 jrick *logmsg_path = NULL;
7685 24df9a28 2023-07-08 jrick
7686 24df9a28 2023-07-08 jrick err = got_worktree_get_uuid(&uuidstr, worktree);
7687 24df9a28 2023-07-08 jrick if (err)
7688 24df9a28 2023-07-08 jrick goto done;
7689 24df9a28 2023-07-08 jrick
7690 24df9a28 2023-07-08 jrick err = got_ref_list(&refs, repo, "refs/got/worktree",
7691 24df9a28 2023-07-08 jrick got_ref_cmp_by_name, repo);
7692 24df9a28 2023-07-08 jrick if (err)
7693 24df9a28 2023-07-08 jrick goto done;
7694 24df9a28 2023-07-08 jrick
7695 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &refs, entry) {
7696 24df9a28 2023-07-08 jrick const char *refname, *type;
7697 24df9a28 2023-07-08 jrick struct wt_commitable_path_arg wcpa;
7698 24df9a28 2023-07-08 jrick int add_logmsg = 0;
7699 24df9a28 2023-07-08 jrick
7700 24df9a28 2023-07-08 jrick refname = got_ref_get_name(re->ref);
7701 24df9a28 2023-07-08 jrick
7702 24df9a28 2023-07-08 jrick if (strncmp(refname, GOT_WORKTREE_CHERRYPICK_REF_PREFIX,
7703 24df9a28 2023-07-08 jrick GOT_WORKTREE_CHERRYPICK_REF_PREFIX_LEN) == 0) {
7704 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_CHERRYPICK_REF_PREFIX_LEN + 1;
7705 24df9a28 2023-07-08 jrick type = "cherrypicked";
7706 24df9a28 2023-07-08 jrick } else if (strncmp(refname, GOT_WORKTREE_BACKOUT_REF_PREFIX,
7707 24df9a28 2023-07-08 jrick GOT_WORKTREE_BACKOUT_REF_PREFIX_LEN) == 0) {
7708 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_BACKOUT_REF_PREFIX_LEN + 1;
7709 24df9a28 2023-07-08 jrick type = "backed-out";
7710 24df9a28 2023-07-08 jrick } else
7711 24df9a28 2023-07-08 jrick continue;
7712 24df9a28 2023-07-08 jrick
7713 24df9a28 2023-07-08 jrick if (strncmp(refname, uuidstr, GOT_WORKTREE_UUID_STRLEN) == 0)
7714 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_UUID_STRLEN + 1; /* skip '-' */
7715 24df9a28 2023-07-08 jrick else
7716 24df9a28 2023-07-08 jrick continue;
7717 24df9a28 2023-07-08 jrick
7718 24df9a28 2023-07-08 jrick err = got_repo_match_object_id(&id, NULL, refname,
7719 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, NULL, repo);
7720 24df9a28 2023-07-08 jrick if (err)
7721 24df9a28 2023-07-08 jrick goto done;
7722 24df9a28 2023-07-08 jrick
7723 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, id);
7724 24df9a28 2023-07-08 jrick if (err)
7725 24df9a28 2023-07-08 jrick goto done;
7726 24df9a28 2023-07-08 jrick
7727 24df9a28 2023-07-08 jrick wcpa.commit_paths = paths;
7728 24df9a28 2023-07-08 jrick wcpa.has_changes = &add_logmsg;
7729 24df9a28 2023-07-08 jrick
7730 24df9a28 2023-07-08 jrick err = commit_path_changed_in_worktree(&wcpa, id,
7731 24df9a28 2023-07-08 jrick worktree, repo);
7732 24df9a28 2023-07-08 jrick if (err)
7733 24df9a28 2023-07-08 jrick goto done;
7734 24df9a28 2023-07-08 jrick
7735 24df9a28 2023-07-08 jrick if (add_logmsg) {
7736 24df9a28 2023-07-08 jrick if (f == NULL) {
7737 24df9a28 2023-07-08 jrick err = got_opentemp_named(logmsg_path, &f,
7738 24df9a28 2023-07-08 jrick "got-commit-logmsg", "");
7739 24df9a28 2023-07-08 jrick if (err)
7740 24df9a28 2023-07-08 jrick goto done;
7741 24df9a28 2023-07-08 jrick }
7742 24df9a28 2023-07-08 jrick err = cat_logmsg(f, commit, refname, type,
7743 24df9a28 2023-07-08 jrick added_logmsg);
7744 24df9a28 2023-07-08 jrick if (err)
7745 24df9a28 2023-07-08 jrick goto done;
7746 24df9a28 2023-07-08 jrick if (!added_logmsg)
7747 24df9a28 2023-07-08 jrick ++added_logmsg;
7748 24df9a28 2023-07-08 jrick
7749 24df9a28 2023-07-08 jrick err = got_reflist_entry_dup(&re_match, re);
7750 24df9a28 2023-07-08 jrick if (err)
7751 24df9a28 2023-07-08 jrick goto done;
7752 24df9a28 2023-07-08 jrick TAILQ_INSERT_HEAD(matched_refs, re_match, entry);
7753 24df9a28 2023-07-08 jrick }
7754 24df9a28 2023-07-08 jrick
7755 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
7756 24df9a28 2023-07-08 jrick commit = NULL;
7757 24df9a28 2023-07-08 jrick free(id);
7758 24df9a28 2023-07-08 jrick id = NULL;
7759 24df9a28 2023-07-08 jrick }
7760 24df9a28 2023-07-08 jrick
7761 24df9a28 2023-07-08 jrick done:
7762 24df9a28 2023-07-08 jrick free(id);
7763 24df9a28 2023-07-08 jrick free(uuidstr);
7764 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
7765 24df9a28 2023-07-08 jrick if (commit)
7766 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
7767 24df9a28 2023-07-08 jrick if (f && fclose(f) == EOF && err == NULL)
7768 24df9a28 2023-07-08 jrick err = got_error_from_errno("fclose");
7769 24df9a28 2023-07-08 jrick if (!added_logmsg) {
7770 24df9a28 2023-07-08 jrick if (*logmsg_path && unlink(*logmsg_path) != 0 && err == NULL)
7771 24df9a28 2023-07-08 jrick err = got_error_from_errno2("unlink", *logmsg_path);
7772 24df9a28 2023-07-08 jrick *logmsg_path = NULL;
7773 24df9a28 2023-07-08 jrick }
7774 24df9a28 2023-07-08 jrick return err;
7775 24df9a28 2023-07-08 jrick }
7776 24df9a28 2023-07-08 jrick
7777 24df9a28 2023-07-08 jrick static const struct got_error *
7778 24df9a28 2023-07-08 jrick cmd_commit(int argc, char *argv[])
7779 24df9a28 2023-07-08 jrick {
7780 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
7781 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
7782 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
7783 24df9a28 2023-07-08 jrick char *cwd = NULL, *id_str = NULL;
7784 24df9a28 2023-07-08 jrick struct got_object_id *id = NULL;
7785 24df9a28 2023-07-08 jrick const char *logmsg = NULL;
7786 24df9a28 2023-07-08 jrick char *prepared_logmsg = NULL, *merged_logmsg = NULL;
7787 24df9a28 2023-07-08 jrick struct collect_commit_logmsg_arg cl_arg;
7788 24df9a28 2023-07-08 jrick const char *author = NULL;
7789 24df9a28 2023-07-08 jrick char *gitconfig_path = NULL, *editor = NULL, *committer = NULL;
7790 24df9a28 2023-07-08 jrick int ch, rebase_in_progress, histedit_in_progress, preserve_logmsg = 0;
7791 24df9a28 2023-07-08 jrick int allow_bad_symlinks = 0, non_interactive = 0, merge_in_progress = 0;
7792 24df9a28 2023-07-08 jrick int show_diff = 1, commit_conflicts = 0;
7793 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
7794 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
7795 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
7796 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
7797 24df9a28 2023-07-08 jrick const struct got_gotconfig *repo_conf = NULL, *worktree_conf = NULL;
7798 24df9a28 2023-07-08 jrick const struct got_remote_repo *remotes, *remote = NULL;
7799 24df9a28 2023-07-08 jrick int nremotes;
7800 24df9a28 2023-07-08 jrick char *proto = NULL, *host = NULL, *port = NULL;
7801 24df9a28 2023-07-08 jrick char *repo_name = NULL, *server_path = NULL;
7802 24df9a28 2023-07-08 jrick const char *remote_name;
7803 24df9a28 2023-07-08 jrick int verbosity = 0;
7804 24df9a28 2023-07-08 jrick int i;
7805 24df9a28 2023-07-08 jrick
7806 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
7807 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
7808 24df9a28 2023-07-08 jrick cl_arg.logmsg_path = NULL;
7809 24df9a28 2023-07-08 jrick
7810 24df9a28 2023-07-08 jrick #ifndef PROFILE
7811 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
7812 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
7813 24df9a28 2023-07-08 jrick err(1, "pledge");
7814 24df9a28 2023-07-08 jrick #endif
7815 24df9a28 2023-07-08 jrick
7816 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "A:CF:m:NnS")) != -1) {
7817 24df9a28 2023-07-08 jrick switch (ch) {
7818 24df9a28 2023-07-08 jrick case 'A':
7819 24df9a28 2023-07-08 jrick author = optarg;
7820 24df9a28 2023-07-08 jrick error = valid_author(author);
7821 24df9a28 2023-07-08 jrick if (error)
7822 24df9a28 2023-07-08 jrick return error;
7823 24df9a28 2023-07-08 jrick break;
7824 24df9a28 2023-07-08 jrick case 'C':
7825 24df9a28 2023-07-08 jrick commit_conflicts = 1;
7826 24df9a28 2023-07-08 jrick break;
7827 24df9a28 2023-07-08 jrick case 'F':
7828 24df9a28 2023-07-08 jrick if (logmsg != NULL)
7829 24df9a28 2023-07-08 jrick option_conflict('F', 'm');
7830 24df9a28 2023-07-08 jrick prepared_logmsg = realpath(optarg, NULL);
7831 24df9a28 2023-07-08 jrick if (prepared_logmsg == NULL)
7832 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
7833 24df9a28 2023-07-08 jrick optarg);
7834 24df9a28 2023-07-08 jrick break;
7835 24df9a28 2023-07-08 jrick case 'm':
7836 24df9a28 2023-07-08 jrick if (prepared_logmsg)
7837 24df9a28 2023-07-08 jrick option_conflict('m', 'F');
7838 24df9a28 2023-07-08 jrick logmsg = optarg;
7839 24df9a28 2023-07-08 jrick break;
7840 24df9a28 2023-07-08 jrick case 'N':
7841 24df9a28 2023-07-08 jrick non_interactive = 1;
7842 24df9a28 2023-07-08 jrick break;
7843 24df9a28 2023-07-08 jrick case 'n':
7844 24df9a28 2023-07-08 jrick show_diff = 0;
7845 24df9a28 2023-07-08 jrick break;
7846 24df9a28 2023-07-08 jrick case 'S':
7847 24df9a28 2023-07-08 jrick allow_bad_symlinks = 1;
7848 24df9a28 2023-07-08 jrick break;
7849 24df9a28 2023-07-08 jrick default:
7850 24df9a28 2023-07-08 jrick usage_commit();
7851 24df9a28 2023-07-08 jrick /* NOTREACHED */
7852 24df9a28 2023-07-08 jrick }
7853 24df9a28 2023-07-08 jrick }
7854 24df9a28 2023-07-08 jrick
7855 24df9a28 2023-07-08 jrick argc -= optind;
7856 24df9a28 2023-07-08 jrick argv += optind;
7857 24df9a28 2023-07-08 jrick
7858 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
7859 24df9a28 2023-07-08 jrick if (cwd == NULL) {
7860 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
7861 24df9a28 2023-07-08 jrick goto done;
7862 24df9a28 2023-07-08 jrick }
7863 24df9a28 2023-07-08 jrick
7864 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
7865 24df9a28 2023-07-08 jrick if (error != NULL)
7866 24df9a28 2023-07-08 jrick goto done;
7867 24df9a28 2023-07-08 jrick
7868 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
7869 24df9a28 2023-07-08 jrick if (error) {
7870 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
7871 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "commit", cwd);
7872 24df9a28 2023-07-08 jrick goto done;
7873 24df9a28 2023-07-08 jrick }
7874 24df9a28 2023-07-08 jrick
7875 24df9a28 2023-07-08 jrick error = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
7876 24df9a28 2023-07-08 jrick if (error)
7877 24df9a28 2023-07-08 jrick goto done;
7878 24df9a28 2023-07-08 jrick if (rebase_in_progress) {
7879 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_REBASING);
7880 24df9a28 2023-07-08 jrick goto done;
7881 24df9a28 2023-07-08 jrick }
7882 24df9a28 2023-07-08 jrick
7883 24df9a28 2023-07-08 jrick error = got_worktree_histedit_in_progress(&histedit_in_progress,
7884 24df9a28 2023-07-08 jrick worktree);
7885 24df9a28 2023-07-08 jrick if (error)
7886 24df9a28 2023-07-08 jrick goto done;
7887 24df9a28 2023-07-08 jrick
7888 24df9a28 2023-07-08 jrick error = get_gitconfig_path(&gitconfig_path);
7889 24df9a28 2023-07-08 jrick if (error)
7890 24df9a28 2023-07-08 jrick goto done;
7891 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, got_worktree_get_repo_path(worktree),
7892 24df9a28 2023-07-08 jrick gitconfig_path, pack_fds);
7893 24df9a28 2023-07-08 jrick if (error != NULL)
7894 24df9a28 2023-07-08 jrick goto done;
7895 24df9a28 2023-07-08 jrick
7896 24df9a28 2023-07-08 jrick error = got_worktree_merge_in_progress(&merge_in_progress, worktree, repo);
7897 24df9a28 2023-07-08 jrick if (error)
7898 24df9a28 2023-07-08 jrick goto done;
7899 24df9a28 2023-07-08 jrick if (merge_in_progress) {
7900 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_MERGE_BUSY);
7901 24df9a28 2023-07-08 jrick goto done;
7902 24df9a28 2023-07-08 jrick }
7903 24df9a28 2023-07-08 jrick
7904 24df9a28 2023-07-08 jrick error = get_author(&committer, repo, worktree);
7905 24df9a28 2023-07-08 jrick if (error)
7906 24df9a28 2023-07-08 jrick goto done;
7907 24df9a28 2023-07-08 jrick
7908 24df9a28 2023-07-08 jrick if (author == NULL)
7909 24df9a28 2023-07-08 jrick author = committer;
7910 24df9a28 2023-07-08 jrick
7911 24df9a28 2023-07-08 jrick remote_name = GOT_SEND_DEFAULT_REMOTE_NAME;
7912 24df9a28 2023-07-08 jrick worktree_conf = got_worktree_get_gotconfig(worktree);
7913 24df9a28 2023-07-08 jrick if (worktree_conf) {
7914 24df9a28 2023-07-08 jrick got_gotconfig_get_remotes(&nremotes, &remotes,
7915 24df9a28 2023-07-08 jrick worktree_conf);
7916 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
7917 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
7918 24df9a28 2023-07-08 jrick remote = &remotes[i];
7919 24df9a28 2023-07-08 jrick break;
7920 24df9a28 2023-07-08 jrick }
7921 24df9a28 2023-07-08 jrick }
7922 24df9a28 2023-07-08 jrick }
7923 24df9a28 2023-07-08 jrick if (remote == NULL) {
7924 24df9a28 2023-07-08 jrick repo_conf = got_repo_get_gotconfig(repo);
7925 24df9a28 2023-07-08 jrick if (repo_conf) {
7926 24df9a28 2023-07-08 jrick got_gotconfig_get_remotes(&nremotes, &remotes,
7927 24df9a28 2023-07-08 jrick repo_conf);
7928 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
7929 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
7930 24df9a28 2023-07-08 jrick remote = &remotes[i];
7931 24df9a28 2023-07-08 jrick break;
7932 24df9a28 2023-07-08 jrick }
7933 24df9a28 2023-07-08 jrick }
7934 24df9a28 2023-07-08 jrick }
7935 24df9a28 2023-07-08 jrick }
7936 24df9a28 2023-07-08 jrick if (remote == NULL) {
7937 24df9a28 2023-07-08 jrick got_repo_get_gitconfig_remotes(&nremotes, &remotes, repo);
7938 24df9a28 2023-07-08 jrick for (i = 0; i < nremotes; i++) {
7939 24df9a28 2023-07-08 jrick if (strcmp(remotes[i].name, remote_name) == 0) {
7940 24df9a28 2023-07-08 jrick remote = &remotes[i];
7941 24df9a28 2023-07-08 jrick break;
7942 24df9a28 2023-07-08 jrick }
7943 24df9a28 2023-07-08 jrick }
7944 24df9a28 2023-07-08 jrick }
7945 24df9a28 2023-07-08 jrick if (remote == NULL) {
7946 24df9a28 2023-07-08 jrick error = got_error_path(remote_name, GOT_ERR_NO_REMOTE);
7947 24df9a28 2023-07-08 jrick goto done;
7948 24df9a28 2023-07-08 jrick }
7949 24df9a28 2023-07-08 jrick
7950 24df9a28 2023-07-08 jrick error = got_dial_parse_uri(&proto, &host, &port, &server_path,
7951 24df9a28 2023-07-08 jrick &repo_name, remote->fetch_url);
7952 24df9a28 2023-07-08 jrick if (error)
7953 24df9a28 2023-07-08 jrick goto done;
7954 24df9a28 2023-07-08 jrick
7955 24df9a28 2023-07-08 jrick if (strcmp(proto, "git") == 0) {
7956 24df9a28 2023-07-08 jrick #ifndef PROFILE
7957 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
7958 24df9a28 2023-07-08 jrick "sendfd dns inet unveil", NULL) == -1)
7959 24df9a28 2023-07-08 jrick err(1, "pledge");
7960 24df9a28 2023-07-08 jrick #endif
7961 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "git+ssh") == 0 ||
7962 24df9a28 2023-07-08 jrick strcmp(proto, "ssh") == 0) {
7963 24df9a28 2023-07-08 jrick #ifndef PROFILE
7964 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec "
7965 24df9a28 2023-07-08 jrick "sendfd unveil", NULL) == -1)
7966 24df9a28 2023-07-08 jrick err(1, "pledge");
7967 24df9a28 2023-07-08 jrick #endif
7968 24df9a28 2023-07-08 jrick } else if (strcmp(proto, "http") == 0 ||
7969 24df9a28 2023-07-08 jrick strcmp(proto, "git+http") == 0) {
7970 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_NOT_IMPL);
7971 24df9a28 2023-07-08 jrick goto done;
7972 24df9a28 2023-07-08 jrick } else {
7973 24df9a28 2023-07-08 jrick error = got_error_path(proto, GOT_ERR_BAD_PROTO);
7974 24df9a28 2023-07-08 jrick goto done;
7975 24df9a28 2023-07-08 jrick }
7976 24df9a28 2023-07-08 jrick
7977 24df9a28 2023-07-08 jrick
7978 24df9a28 2023-07-08 jrick /*if (verbosity >= 0) {
7979 24df9a28 2023-07-08 jrick printf("Connecting to \"%s\" %s://%s%s%s%s%s\n",
7980 24df9a28 2023-07-08 jrick remote->name, proto, host,
7981 24df9a28 2023-07-08 jrick port ? ":" : "", port ? port : "",
7982 24df9a28 2023-07-08 jrick *server_path == '/' ? "" : "/", server_path);
7983 24df9a28 2023-07-08 jrick }*/
7984 24df9a28 2023-07-08 jrick
7985 24df9a28 2023-07-08 jrick /*
7986 24df9a28 2023-07-08 jrick * unveil(2) traverses exec(2); if an editor is used we have
7987 24df9a28 2023-07-08 jrick * to apply unveil after the log message has been written during
7988 24df9a28 2023-07-08 jrick * the callback.
7989 24df9a28 2023-07-08 jrick */
7990 24df9a28 2023-07-08 jrick if (logmsg == NULL || strlen(logmsg) == 0)
7991 24df9a28 2023-07-08 jrick error = get_editor(&editor);
7992 24df9a28 2023-07-08 jrick else {
7993 24df9a28 2023-07-08 jrick error = got_dial_apply_unveil(proto);
7994 24df9a28 2023-07-08 jrick if (error)
7995 24df9a28 2023-07-08 jrick goto done;
7996 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
7997 24df9a28 2023-07-08 jrick got_worktree_get_root_path(worktree));
7998 24df9a28 2023-07-08 jrick }
7999 24df9a28 2023-07-08 jrick if (error)
8000 24df9a28 2023-07-08 jrick goto done;
8001 24df9a28 2023-07-08 jrick
8002 24df9a28 2023-07-08 jrick if (prepared_logmsg == NULL) {
8003 24df9a28 2023-07-08 jrick error = lookup_logmsg_ref(&merged_logmsg,
8004 24df9a28 2023-07-08 jrick argc > 0 ? &paths : NULL, &refs, worktree, repo);
8005 24df9a28 2023-07-08 jrick if (error)
8006 24df9a28 2023-07-08 jrick goto done;
8007 24df9a28 2023-07-08 jrick }
8008 24df9a28 2023-07-08 jrick
8009 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv, worktree);
8010 24df9a28 2023-07-08 jrick if (error)
8011 24df9a28 2023-07-08 jrick goto done;
8012 24df9a28 2023-07-08 jrick
8013 24df9a28 2023-07-08 jrick cl_arg.editor = editor;
8014 24df9a28 2023-07-08 jrick cl_arg.cmdline_log = logmsg;
8015 24df9a28 2023-07-08 jrick cl_arg.prepared_log = prepared_logmsg;
8016 24df9a28 2023-07-08 jrick cl_arg.merged_log = merged_logmsg;
8017 24df9a28 2023-07-08 jrick cl_arg.non_interactive = non_interactive;
8018 24df9a28 2023-07-08 jrick cl_arg.worktree_path = got_worktree_get_root_path(worktree);
8019 24df9a28 2023-07-08 jrick cl_arg.repo_path = got_repo_get_path(repo);
8020 24df9a28 2023-07-08 jrick cl_arg.dial_proto = proto;
8021 24df9a28 2023-07-08 jrick error = got_worktree_cvg_commit(&id, worktree, &paths, author,
8022 24df9a28 2023-07-08 jrick committer, allow_bad_symlinks, show_diff, commit_conflicts,
8023 24df9a28 2023-07-08 jrick collect_commit_logmsg, &cl_arg, print_status, NULL, proto, host,
8024 24df9a28 2023-07-08 jrick port, server_path, verbosity, remote, check_cancelled, repo);
8025 24df9a28 2023-07-08 jrick if (error) {
8026 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_COMMIT_MSG_EMPTY &&
8027 24df9a28 2023-07-08 jrick cl_arg.logmsg_path != NULL)
8028 24df9a28 2023-07-08 jrick preserve_logmsg = 1;
8029 24df9a28 2023-07-08 jrick goto done;
8030 24df9a28 2023-07-08 jrick }
8031 24df9a28 2023-07-08 jrick
8032 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str, id);
8033 24df9a28 2023-07-08 jrick if (error)
8034 24df9a28 2023-07-08 jrick goto done;
8035 24df9a28 2023-07-08 jrick printf("Created commit %s\n", id_str);
8036 24df9a28 2023-07-08 jrick
8037 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &refs, entry) {
8038 24df9a28 2023-07-08 jrick error = got_ref_delete(re->ref, repo);
8039 24df9a28 2023-07-08 jrick if (error)
8040 24df9a28 2023-07-08 jrick goto done;
8041 24df9a28 2023-07-08 jrick }
8042 24df9a28 2023-07-08 jrick
8043 24df9a28 2023-07-08 jrick done:
8044 24df9a28 2023-07-08 jrick if (preserve_logmsg) {
8045 24df9a28 2023-07-08 jrick fprintf(stderr, "%s: log message preserved in %s\n",
8046 24df9a28 2023-07-08 jrick getprogname(), cl_arg.logmsg_path);
8047 24df9a28 2023-07-08 jrick } else if (cl_arg.logmsg_path && unlink(cl_arg.logmsg_path) == -1 &&
8048 24df9a28 2023-07-08 jrick error == NULL)
8049 24df9a28 2023-07-08 jrick error = got_error_from_errno2("unlink", cl_arg.logmsg_path);
8050 24df9a28 2023-07-08 jrick free(cl_arg.logmsg_path);
8051 24df9a28 2023-07-08 jrick if (merged_logmsg && unlink(merged_logmsg) == -1 && error == NULL)
8052 24df9a28 2023-07-08 jrick error = got_error_from_errno2("unlink", merged_logmsg);
8053 24df9a28 2023-07-08 jrick free(merged_logmsg);
8054 24df9a28 2023-07-08 jrick if (repo) {
8055 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
8056 24df9a28 2023-07-08 jrick if (error == NULL)
8057 24df9a28 2023-07-08 jrick error = close_err;
8058 24df9a28 2023-07-08 jrick }
8059 24df9a28 2023-07-08 jrick if (worktree)
8060 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
8061 24df9a28 2023-07-08 jrick if (pack_fds) {
8062 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
8063 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
8064 24df9a28 2023-07-08 jrick if (error == NULL)
8065 24df9a28 2023-07-08 jrick error = pack_err;
8066 24df9a28 2023-07-08 jrick }
8067 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
8068 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
8069 24df9a28 2023-07-08 jrick free(cwd);
8070 24df9a28 2023-07-08 jrick free(id_str);
8071 24df9a28 2023-07-08 jrick free(gitconfig_path);
8072 24df9a28 2023-07-08 jrick free(editor);
8073 24df9a28 2023-07-08 jrick free(committer);
8074 24df9a28 2023-07-08 jrick free(prepared_logmsg);
8075 24df9a28 2023-07-08 jrick return error;
8076 24df9a28 2023-07-08 jrick }
8077 24df9a28 2023-07-08 jrick
8078 24df9a28 2023-07-08 jrick /*
8079 24df9a28 2023-07-08 jrick * Print and if delete is set delete all ref_prefix references.
8080 24df9a28 2023-07-08 jrick * If wanted_ref is not NULL, only print or delete this reference.
8081 24df9a28 2023-07-08 jrick */
8082 24df9a28 2023-07-08 jrick static const struct got_error *
8083 24df9a28 2023-07-08 jrick process_logmsg_refs(const char *ref_prefix, size_t prefix_len,
8084 24df9a28 2023-07-08 jrick const char *wanted_ref, int delete, struct got_worktree *worktree,
8085 24df9a28 2023-07-08 jrick struct got_repository *repo)
8086 24df9a28 2023-07-08 jrick {
8087 24df9a28 2023-07-08 jrick const struct got_error *err;
8088 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
8089 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
8090 24df9a28 2023-07-08 jrick struct got_reflist_entry *re;
8091 24df9a28 2023-07-08 jrick struct got_reflist_object_id_map *refs_idmap = NULL;
8092 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
8093 24df9a28 2023-07-08 jrick struct got_object_id *id = NULL;
8094 24df9a28 2023-07-08 jrick const char *header_prefix;
8095 24df9a28 2023-07-08 jrick char *uuidstr = NULL;
8096 24df9a28 2023-07-08 jrick int found = 0;
8097 24df9a28 2023-07-08 jrick
8098 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
8099 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
8100 24df9a28 2023-07-08 jrick
8101 24df9a28 2023-07-08 jrick err = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, repo);
8102 24df9a28 2023-07-08 jrick if (err)
8103 24df9a28 2023-07-08 jrick goto done;
8104 24df9a28 2023-07-08 jrick
8105 24df9a28 2023-07-08 jrick err = got_reflist_object_id_map_create(&refs_idmap, &refs, repo);
8106 24df9a28 2023-07-08 jrick if (err)
8107 24df9a28 2023-07-08 jrick goto done;
8108 24df9a28 2023-07-08 jrick
8109 24df9a28 2023-07-08 jrick if (worktree != NULL) {
8110 24df9a28 2023-07-08 jrick err = got_worktree_get_uuid(&uuidstr, worktree);
8111 24df9a28 2023-07-08 jrick if (err)
8112 24df9a28 2023-07-08 jrick goto done;
8113 24df9a28 2023-07-08 jrick }
8114 24df9a28 2023-07-08 jrick
8115 24df9a28 2023-07-08 jrick if (wanted_ref) {
8116 24df9a28 2023-07-08 jrick if (strncmp(wanted_ref, "refs/heads/", 11) == 0)
8117 24df9a28 2023-07-08 jrick wanted_ref += 11;
8118 24df9a28 2023-07-08 jrick }
8119 24df9a28 2023-07-08 jrick
8120 24df9a28 2023-07-08 jrick if (strcmp(ref_prefix, GOT_WORKTREE_BACKOUT_REF_PREFIX) == 0)
8121 24df9a28 2023-07-08 jrick header_prefix = "backout";
8122 24df9a28 2023-07-08 jrick else
8123 24df9a28 2023-07-08 jrick header_prefix = "cherrypick";
8124 24df9a28 2023-07-08 jrick
8125 24df9a28 2023-07-08 jrick TAILQ_FOREACH(re, &refs, entry) {
8126 24df9a28 2023-07-08 jrick const char *refname, *wt;
8127 24df9a28 2023-07-08 jrick
8128 24df9a28 2023-07-08 jrick refname = got_ref_get_name(re->ref);
8129 24df9a28 2023-07-08 jrick
8130 24df9a28 2023-07-08 jrick err = check_cancelled(NULL);
8131 24df9a28 2023-07-08 jrick if (err)
8132 24df9a28 2023-07-08 jrick goto done;
8133 24df9a28 2023-07-08 jrick
8134 24df9a28 2023-07-08 jrick if (strncmp(refname, ref_prefix, prefix_len) == 0)
8135 24df9a28 2023-07-08 jrick refname += prefix_len + 1; /* skip '-' delimiter */
8136 24df9a28 2023-07-08 jrick else
8137 24df9a28 2023-07-08 jrick continue;
8138 24df9a28 2023-07-08 jrick
8139 24df9a28 2023-07-08 jrick wt = refname;
8140 24df9a28 2023-07-08 jrick
8141 24df9a28 2023-07-08 jrick if (worktree == NULL || strncmp(refname, uuidstr,
8142 24df9a28 2023-07-08 jrick GOT_WORKTREE_UUID_STRLEN) == 0)
8143 24df9a28 2023-07-08 jrick refname += GOT_WORKTREE_UUID_STRLEN + 1; /* skip '-' */
8144 24df9a28 2023-07-08 jrick else
8145 24df9a28 2023-07-08 jrick continue;
8146 24df9a28 2023-07-08 jrick
8147 24df9a28 2023-07-08 jrick err = got_repo_match_object_id(&id, NULL, refname,
8148 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, NULL, repo);
8149 24df9a28 2023-07-08 jrick if (err)
8150 24df9a28 2023-07-08 jrick goto done;
8151 24df9a28 2023-07-08 jrick
8152 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, id);
8153 24df9a28 2023-07-08 jrick if (err)
8154 24df9a28 2023-07-08 jrick goto done;
8155 24df9a28 2023-07-08 jrick
8156 24df9a28 2023-07-08 jrick if (wanted_ref)
8157 24df9a28 2023-07-08 jrick found = strncmp(wanted_ref, refname,
8158 24df9a28 2023-07-08 jrick strlen(wanted_ref)) == 0;
8159 24df9a28 2023-07-08 jrick if (wanted_ref && !found) {
8160 24df9a28 2023-07-08 jrick struct got_reflist_head *ci_refs;
8161 24df9a28 2023-07-08 jrick
8162 24df9a28 2023-07-08 jrick ci_refs = got_reflist_object_id_map_lookup(refs_idmap,
8163 24df9a28 2023-07-08 jrick id);
8164 24df9a28 2023-07-08 jrick
8165 24df9a28 2023-07-08 jrick if (ci_refs) {
8166 24df9a28 2023-07-08 jrick char *refs_str = NULL;
8167 24df9a28 2023-07-08 jrick char const *r = NULL;
8168 24df9a28 2023-07-08 jrick
8169 24df9a28 2023-07-08 jrick err = build_refs_str(&refs_str, ci_refs, id,
8170 24df9a28 2023-07-08 jrick repo, 1);
8171 24df9a28 2023-07-08 jrick if (err)
8172 24df9a28 2023-07-08 jrick goto done;
8173 24df9a28 2023-07-08 jrick
8174 24df9a28 2023-07-08 jrick r = refs_str;
8175 24df9a28 2023-07-08 jrick while (r) {
8176 24df9a28 2023-07-08 jrick if (strncmp(r, wanted_ref,
8177 24df9a28 2023-07-08 jrick strlen(wanted_ref)) == 0) {
8178 24df9a28 2023-07-08 jrick found = 1;
8179 24df9a28 2023-07-08 jrick break;
8180 24df9a28 2023-07-08 jrick }
8181 24df9a28 2023-07-08 jrick r = strchr(r, ' ');
8182 24df9a28 2023-07-08 jrick if (r)
8183 24df9a28 2023-07-08 jrick ++r;
8184 24df9a28 2023-07-08 jrick }
8185 24df9a28 2023-07-08 jrick free(refs_str);
8186 24df9a28 2023-07-08 jrick }
8187 24df9a28 2023-07-08 jrick }
8188 24df9a28 2023-07-08 jrick
8189 24df9a28 2023-07-08 jrick if (wanted_ref == NULL || found) {
8190 24df9a28 2023-07-08 jrick if (delete) {
8191 24df9a28 2023-07-08 jrick err = got_ref_delete(re->ref, repo);
8192 24df9a28 2023-07-08 jrick if (err)
8193 24df9a28 2023-07-08 jrick goto done;
8194 24df9a28 2023-07-08 jrick printf("Deleted: ");
8195 24df9a28 2023-07-08 jrick err = print_commit_oneline(commit, id, repo,
8196 24df9a28 2023-07-08 jrick refs_idmap);
8197 24df9a28 2023-07-08 jrick } else {
8198 24df9a28 2023-07-08 jrick /*
8199 24df9a28 2023-07-08 jrick * Print paths modified by commit to help
8200 24df9a28 2023-07-08 jrick * associate commits with worktree changes.
8201 24df9a28 2023-07-08 jrick */
8202 24df9a28 2023-07-08 jrick err = get_changed_paths(&paths, commit,
8203 24df9a28 2023-07-08 jrick repo, NULL);
8204 24df9a28 2023-07-08 jrick if (err)
8205 24df9a28 2023-07-08 jrick goto done;
8206 24df9a28 2023-07-08 jrick
8207 24df9a28 2023-07-08 jrick err = print_commit(commit, id, repo, NULL,
8208 24df9a28 2023-07-08 jrick &paths, NULL, 0, 0, refs_idmap, NULL,
8209 24df9a28 2023-07-08 jrick header_prefix);
8210 24df9a28 2023-07-08 jrick got_pathlist_free(&paths,
8211 24df9a28 2023-07-08 jrick GOT_PATHLIST_FREE_ALL);
8212 24df9a28 2023-07-08 jrick
8213 24df9a28 2023-07-08 jrick if (worktree == NULL)
8214 24df9a28 2023-07-08 jrick printf("work tree: %.*s\n\n",
8215 24df9a28 2023-07-08 jrick GOT_WORKTREE_UUID_STRLEN, wt);
8216 24df9a28 2023-07-08 jrick }
8217 24df9a28 2023-07-08 jrick if (err || found)
8218 24df9a28 2023-07-08 jrick goto done;
8219 24df9a28 2023-07-08 jrick }
8220 24df9a28 2023-07-08 jrick
8221 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8222 24df9a28 2023-07-08 jrick commit = NULL;
8223 24df9a28 2023-07-08 jrick free(id);
8224 24df9a28 2023-07-08 jrick id = NULL;
8225 24df9a28 2023-07-08 jrick }
8226 24df9a28 2023-07-08 jrick
8227 24df9a28 2023-07-08 jrick if (wanted_ref != NULL && !found)
8228 24df9a28 2023-07-08 jrick err = got_error_fmt(GOT_ERR_NOT_REF, "%s", wanted_ref);
8229 24df9a28 2023-07-08 jrick
8230 24df9a28 2023-07-08 jrick done:
8231 24df9a28 2023-07-08 jrick free(id);
8232 24df9a28 2023-07-08 jrick free(uuidstr);
8233 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
8234 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_ALL);
8235 24df9a28 2023-07-08 jrick if (refs_idmap)
8236 24df9a28 2023-07-08 jrick got_reflist_object_id_map_free(refs_idmap);
8237 24df9a28 2023-07-08 jrick if (commit)
8238 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8239 24df9a28 2023-07-08 jrick return err;
8240 24df9a28 2023-07-08 jrick }
8241 24df9a28 2023-07-08 jrick
8242 24df9a28 2023-07-08 jrick /*
8243 24df9a28 2023-07-08 jrick * Create new temp "logmsg" ref of the backed-out or cherrypicked commit
8244 24df9a28 2023-07-08 jrick * identified by id for log messages to prepopulate the editor on commit.
8245 24df9a28 2023-07-08 jrick */
8246 24df9a28 2023-07-08 jrick static const struct got_error *
8247 24df9a28 2023-07-08 jrick logmsg_ref(struct got_object_id *id, const char *prefix,
8248 24df9a28 2023-07-08 jrick struct got_worktree *worktree, struct got_repository *repo)
8249 24df9a28 2023-07-08 jrick {
8250 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
8251 24df9a28 2023-07-08 jrick char *idstr, *ref = NULL, *refname = NULL;
8252 24df9a28 2023-07-08 jrick int histedit_in_progress;
8253 24df9a28 2023-07-08 jrick int rebase_in_progress, merge_in_progress;
8254 24df9a28 2023-07-08 jrick
8255 24df9a28 2023-07-08 jrick /*
8256 24df9a28 2023-07-08 jrick * Silenty refuse to create merge reference if any histedit, merge,
8257 24df9a28 2023-07-08 jrick * or rebase operation is in progress.
8258 24df9a28 2023-07-08 jrick */
8259 24df9a28 2023-07-08 jrick err = got_worktree_histedit_in_progress(&histedit_in_progress,
8260 24df9a28 2023-07-08 jrick worktree);
8261 24df9a28 2023-07-08 jrick if (err)
8262 24df9a28 2023-07-08 jrick return err;
8263 24df9a28 2023-07-08 jrick if (histedit_in_progress)
8264 24df9a28 2023-07-08 jrick return NULL;
8265 24df9a28 2023-07-08 jrick
8266 24df9a28 2023-07-08 jrick err = got_worktree_rebase_in_progress(&rebase_in_progress, worktree);
8267 24df9a28 2023-07-08 jrick if (err)
8268 24df9a28 2023-07-08 jrick return err;
8269 24df9a28 2023-07-08 jrick if (rebase_in_progress)
8270 24df9a28 2023-07-08 jrick return NULL;
8271 24df9a28 2023-07-08 jrick
8272 24df9a28 2023-07-08 jrick err = got_worktree_merge_in_progress(&merge_in_progress, worktree,
8273 24df9a28 2023-07-08 jrick repo);
8274 24df9a28 2023-07-08 jrick if (err)
8275 24df9a28 2023-07-08 jrick return err;
8276 24df9a28 2023-07-08 jrick if (merge_in_progress)
8277 24df9a28 2023-07-08 jrick return NULL;
8278 24df9a28 2023-07-08 jrick
8279 24df9a28 2023-07-08 jrick err = got_object_id_str(&idstr, id);
8280 24df9a28 2023-07-08 jrick if (err)
8281 24df9a28 2023-07-08 jrick return err;
8282 24df9a28 2023-07-08 jrick
8283 24df9a28 2023-07-08 jrick err = got_worktree_get_logmsg_ref_name(&refname, worktree, prefix);
8284 24df9a28 2023-07-08 jrick if (err)
8285 24df9a28 2023-07-08 jrick goto done;
8286 24df9a28 2023-07-08 jrick
8287 24df9a28 2023-07-08 jrick if (asprintf(&ref, "%s-%s", refname, idstr) == -1) {
8288 24df9a28 2023-07-08 jrick err = got_error_from_errno("asprintf");
8289 24df9a28 2023-07-08 jrick goto done;
8290 24df9a28 2023-07-08 jrick }
8291 24df9a28 2023-07-08 jrick
8292 24df9a28 2023-07-08 jrick err = create_ref(ref, got_worktree_get_base_commit_id(worktree),
8293 24df9a28 2023-07-08 jrick -1, repo);
8294 24df9a28 2023-07-08 jrick done:
8295 24df9a28 2023-07-08 jrick free(ref);
8296 24df9a28 2023-07-08 jrick free(idstr);
8297 24df9a28 2023-07-08 jrick free(refname);
8298 24df9a28 2023-07-08 jrick return err;
8299 24df9a28 2023-07-08 jrick }
8300 24df9a28 2023-07-08 jrick
8301 24df9a28 2023-07-08 jrick __dead static void
8302 24df9a28 2023-07-08 jrick usage_cherrypick(void)
8303 24df9a28 2023-07-08 jrick {
8304 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s cherrypick [-lX] [commit-id]\n",
8305 24df9a28 2023-07-08 jrick getprogname());
8306 24df9a28 2023-07-08 jrick exit(1);
8307 24df9a28 2023-07-08 jrick }
8308 24df9a28 2023-07-08 jrick
8309 24df9a28 2023-07-08 jrick static const struct got_error *
8310 24df9a28 2023-07-08 jrick cmd_cherrypick(int argc, char *argv[])
8311 24df9a28 2023-07-08 jrick {
8312 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
8313 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
8314 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
8315 24df9a28 2023-07-08 jrick char *cwd = NULL, *commit_id_str = NULL;
8316 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
8317 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
8318 24df9a28 2023-07-08 jrick struct got_object_qid *pid;
8319 24df9a28 2023-07-08 jrick int ch, list_refs = 0, remove_refs = 0;
8320 24df9a28 2023-07-08 jrick struct got_update_progress_arg upa;
8321 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
8322 24df9a28 2023-07-08 jrick
8323 24df9a28 2023-07-08 jrick #ifndef PROFILE
8324 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
8325 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
8326 24df9a28 2023-07-08 jrick err(1, "pledge");
8327 24df9a28 2023-07-08 jrick #endif
8328 24df9a28 2023-07-08 jrick
8329 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "lX")) != -1) {
8330 24df9a28 2023-07-08 jrick switch (ch) {
8331 24df9a28 2023-07-08 jrick case 'l':
8332 24df9a28 2023-07-08 jrick list_refs = 1;
8333 24df9a28 2023-07-08 jrick break;
8334 24df9a28 2023-07-08 jrick case 'X':
8335 24df9a28 2023-07-08 jrick remove_refs = 1;
8336 24df9a28 2023-07-08 jrick break;
8337 24df9a28 2023-07-08 jrick default:
8338 24df9a28 2023-07-08 jrick usage_cherrypick();
8339 24df9a28 2023-07-08 jrick /* NOTREACHED */
8340 24df9a28 2023-07-08 jrick }
8341 24df9a28 2023-07-08 jrick }
8342 24df9a28 2023-07-08 jrick
8343 24df9a28 2023-07-08 jrick argc -= optind;
8344 24df9a28 2023-07-08 jrick argv += optind;
8345 24df9a28 2023-07-08 jrick
8346 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8347 24df9a28 2023-07-08 jrick if (argc != 0 && argc != 1)
8348 24df9a28 2023-07-08 jrick usage_cherrypick();
8349 24df9a28 2023-07-08 jrick } else if (argc != 1)
8350 24df9a28 2023-07-08 jrick usage_cherrypick();
8351 24df9a28 2023-07-08 jrick if (list_refs && remove_refs)
8352 24df9a28 2023-07-08 jrick option_conflict('l', 'X');
8353 24df9a28 2023-07-08 jrick
8354 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
8355 24df9a28 2023-07-08 jrick if (cwd == NULL) {
8356 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
8357 24df9a28 2023-07-08 jrick goto done;
8358 24df9a28 2023-07-08 jrick }
8359 24df9a28 2023-07-08 jrick
8360 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
8361 24df9a28 2023-07-08 jrick if (error != NULL)
8362 24df9a28 2023-07-08 jrick goto done;
8363 24df9a28 2023-07-08 jrick
8364 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
8365 24df9a28 2023-07-08 jrick if (error) {
8366 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8367 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_WORKTREE)
8368 24df9a28 2023-07-08 jrick goto done;
8369 24df9a28 2023-07-08 jrick } else {
8370 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
8371 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error,
8372 24df9a28 2023-07-08 jrick "cherrypick", cwd);
8373 24df9a28 2023-07-08 jrick goto done;
8374 24df9a28 2023-07-08 jrick }
8375 24df9a28 2023-07-08 jrick }
8376 24df9a28 2023-07-08 jrick
8377 24df9a28 2023-07-08 jrick error = got_repo_open(&repo,
8378 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_repo_path(worktree) : cwd,
8379 24df9a28 2023-07-08 jrick NULL, pack_fds);
8380 24df9a28 2023-07-08 jrick if (error != NULL)
8381 24df9a28 2023-07-08 jrick goto done;
8382 24df9a28 2023-07-08 jrick
8383 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
8384 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_root_path(worktree) : NULL);
8385 24df9a28 2023-07-08 jrick if (error)
8386 24df9a28 2023-07-08 jrick goto done;
8387 24df9a28 2023-07-08 jrick
8388 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8389 24df9a28 2023-07-08 jrick error = process_logmsg_refs(GOT_WORKTREE_CHERRYPICK_REF_PREFIX,
8390 24df9a28 2023-07-08 jrick GOT_WORKTREE_CHERRYPICK_REF_PREFIX_LEN,
8391 24df9a28 2023-07-08 jrick argc == 1 ? argv[0] : NULL, remove_refs, worktree, repo);
8392 24df9a28 2023-07-08 jrick goto done;
8393 24df9a28 2023-07-08 jrick }
8394 24df9a28 2023-07-08 jrick
8395 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL, argv[0],
8396 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, NULL, repo);
8397 24df9a28 2023-07-08 jrick if (error)
8398 24df9a28 2023-07-08 jrick goto done;
8399 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
8400 24df9a28 2023-07-08 jrick if (error)
8401 24df9a28 2023-07-08 jrick goto done;
8402 24df9a28 2023-07-08 jrick
8403 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, commit_id);
8404 24df9a28 2023-07-08 jrick if (error)
8405 24df9a28 2023-07-08 jrick goto done;
8406 24df9a28 2023-07-08 jrick pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
8407 24df9a28 2023-07-08 jrick memset(&upa, 0, sizeof(upa));
8408 24df9a28 2023-07-08 jrick error = got_worktree_merge_files(worktree, pid ? &pid->id : NULL,
8409 24df9a28 2023-07-08 jrick commit_id, repo, update_progress, &upa, check_cancelled,
8410 24df9a28 2023-07-08 jrick NULL);
8411 24df9a28 2023-07-08 jrick if (error != NULL)
8412 24df9a28 2023-07-08 jrick goto done;
8413 24df9a28 2023-07-08 jrick
8414 24df9a28 2023-07-08 jrick if (upa.did_something) {
8415 24df9a28 2023-07-08 jrick error = logmsg_ref(commit_id,
8416 24df9a28 2023-07-08 jrick GOT_WORKTREE_CHERRYPICK_REF_PREFIX, worktree, repo);
8417 24df9a28 2023-07-08 jrick if (error)
8418 24df9a28 2023-07-08 jrick goto done;
8419 24df9a28 2023-07-08 jrick printf("Merged commit %s\n", commit_id_str);
8420 24df9a28 2023-07-08 jrick }
8421 24df9a28 2023-07-08 jrick print_merge_progress_stats(&upa);
8422 24df9a28 2023-07-08 jrick done:
8423 24df9a28 2023-07-08 jrick free(cwd);
8424 24df9a28 2023-07-08 jrick if (commit)
8425 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8426 24df9a28 2023-07-08 jrick free(commit_id_str);
8427 24df9a28 2023-07-08 jrick if (worktree)
8428 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
8429 24df9a28 2023-07-08 jrick if (repo) {
8430 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
8431 24df9a28 2023-07-08 jrick if (error == NULL)
8432 24df9a28 2023-07-08 jrick error = close_err;
8433 24df9a28 2023-07-08 jrick }
8434 24df9a28 2023-07-08 jrick if (pack_fds) {
8435 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
8436 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
8437 24df9a28 2023-07-08 jrick if (error == NULL)
8438 24df9a28 2023-07-08 jrick error = pack_err;
8439 24df9a28 2023-07-08 jrick }
8440 24df9a28 2023-07-08 jrick
8441 24df9a28 2023-07-08 jrick return error;
8442 24df9a28 2023-07-08 jrick }
8443 24df9a28 2023-07-08 jrick
8444 24df9a28 2023-07-08 jrick __dead static void
8445 24df9a28 2023-07-08 jrick usage_backout(void)
8446 24df9a28 2023-07-08 jrick {
8447 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s backout [-lX] [commit-id]\n", getprogname());
8448 24df9a28 2023-07-08 jrick exit(1);
8449 24df9a28 2023-07-08 jrick }
8450 24df9a28 2023-07-08 jrick
8451 24df9a28 2023-07-08 jrick static const struct got_error *
8452 24df9a28 2023-07-08 jrick cmd_backout(int argc, char *argv[])
8453 24df9a28 2023-07-08 jrick {
8454 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
8455 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
8456 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
8457 24df9a28 2023-07-08 jrick char *cwd = NULL, *commit_id_str = NULL;
8458 24df9a28 2023-07-08 jrick struct got_object_id *commit_id = NULL;
8459 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
8460 24df9a28 2023-07-08 jrick struct got_object_qid *pid;
8461 24df9a28 2023-07-08 jrick int ch, list_refs = 0, remove_refs = 0;
8462 24df9a28 2023-07-08 jrick struct got_update_progress_arg upa;
8463 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
8464 24df9a28 2023-07-08 jrick
8465 24df9a28 2023-07-08 jrick #ifndef PROFILE
8466 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
8467 24df9a28 2023-07-08 jrick "unveil", NULL) == -1)
8468 24df9a28 2023-07-08 jrick err(1, "pledge");
8469 24df9a28 2023-07-08 jrick #endif
8470 24df9a28 2023-07-08 jrick
8471 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "lX")) != -1) {
8472 24df9a28 2023-07-08 jrick switch (ch) {
8473 24df9a28 2023-07-08 jrick case 'l':
8474 24df9a28 2023-07-08 jrick list_refs = 1;
8475 24df9a28 2023-07-08 jrick break;
8476 24df9a28 2023-07-08 jrick case 'X':
8477 24df9a28 2023-07-08 jrick remove_refs = 1;
8478 24df9a28 2023-07-08 jrick break;
8479 24df9a28 2023-07-08 jrick default:
8480 24df9a28 2023-07-08 jrick usage_backout();
8481 24df9a28 2023-07-08 jrick /* NOTREACHED */
8482 24df9a28 2023-07-08 jrick }
8483 24df9a28 2023-07-08 jrick }
8484 24df9a28 2023-07-08 jrick
8485 24df9a28 2023-07-08 jrick argc -= optind;
8486 24df9a28 2023-07-08 jrick argv += optind;
8487 24df9a28 2023-07-08 jrick
8488 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8489 24df9a28 2023-07-08 jrick if (argc != 0 && argc != 1)
8490 24df9a28 2023-07-08 jrick usage_backout();
8491 24df9a28 2023-07-08 jrick } else if (argc != 1)
8492 24df9a28 2023-07-08 jrick usage_backout();
8493 24df9a28 2023-07-08 jrick if (list_refs && remove_refs)
8494 24df9a28 2023-07-08 jrick option_conflict('l', 'X');
8495 24df9a28 2023-07-08 jrick
8496 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
8497 24df9a28 2023-07-08 jrick if (cwd == NULL) {
8498 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
8499 24df9a28 2023-07-08 jrick goto done;
8500 24df9a28 2023-07-08 jrick }
8501 24df9a28 2023-07-08 jrick
8502 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
8503 24df9a28 2023-07-08 jrick if (error != NULL)
8504 24df9a28 2023-07-08 jrick goto done;
8505 24df9a28 2023-07-08 jrick
8506 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
8507 24df9a28 2023-07-08 jrick if (error) {
8508 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8509 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_NOT_WORKTREE)
8510 24df9a28 2023-07-08 jrick goto done;
8511 24df9a28 2023-07-08 jrick } else {
8512 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
8513 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error,
8514 24df9a28 2023-07-08 jrick "backout", cwd);
8515 24df9a28 2023-07-08 jrick goto done;
8516 24df9a28 2023-07-08 jrick }
8517 24df9a28 2023-07-08 jrick }
8518 24df9a28 2023-07-08 jrick
8519 24df9a28 2023-07-08 jrick error = got_repo_open(&repo,
8520 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_repo_path(worktree) : cwd,
8521 24df9a28 2023-07-08 jrick NULL, pack_fds);
8522 24df9a28 2023-07-08 jrick if (error != NULL)
8523 24df9a28 2023-07-08 jrick goto done;
8524 24df9a28 2023-07-08 jrick
8525 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 0,
8526 24df9a28 2023-07-08 jrick worktree ? got_worktree_get_root_path(worktree) : NULL);
8527 24df9a28 2023-07-08 jrick if (error)
8528 24df9a28 2023-07-08 jrick goto done;
8529 24df9a28 2023-07-08 jrick
8530 24df9a28 2023-07-08 jrick if (list_refs || remove_refs) {
8531 24df9a28 2023-07-08 jrick error = process_logmsg_refs(GOT_WORKTREE_BACKOUT_REF_PREFIX,
8532 24df9a28 2023-07-08 jrick GOT_WORKTREE_BACKOUT_REF_PREFIX_LEN,
8533 24df9a28 2023-07-08 jrick argc == 1 ? argv[0] : NULL, remove_refs, worktree, repo);
8534 24df9a28 2023-07-08 jrick goto done;
8535 24df9a28 2023-07-08 jrick }
8536 24df9a28 2023-07-08 jrick
8537 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL, argv[0],
8538 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_COMMIT, NULL, repo);
8539 24df9a28 2023-07-08 jrick if (error)
8540 24df9a28 2023-07-08 jrick goto done;
8541 24df9a28 2023-07-08 jrick error = got_object_id_str(&commit_id_str, commit_id);
8542 24df9a28 2023-07-08 jrick if (error)
8543 24df9a28 2023-07-08 jrick goto done;
8544 24df9a28 2023-07-08 jrick
8545 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, commit_id);
8546 24df9a28 2023-07-08 jrick if (error)
8547 24df9a28 2023-07-08 jrick goto done;
8548 24df9a28 2023-07-08 jrick pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
8549 24df9a28 2023-07-08 jrick if (pid == NULL) {
8550 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_ROOT_COMMIT);
8551 24df9a28 2023-07-08 jrick goto done;
8552 24df9a28 2023-07-08 jrick }
8553 24df9a28 2023-07-08 jrick
8554 24df9a28 2023-07-08 jrick memset(&upa, 0, sizeof(upa));
8555 24df9a28 2023-07-08 jrick error = got_worktree_merge_files(worktree, commit_id, &pid->id,
8556 24df9a28 2023-07-08 jrick repo, update_progress, &upa, check_cancelled, NULL);
8557 24df9a28 2023-07-08 jrick if (error != NULL)
8558 24df9a28 2023-07-08 jrick goto done;
8559 24df9a28 2023-07-08 jrick
8560 24df9a28 2023-07-08 jrick if (upa.did_something) {
8561 24df9a28 2023-07-08 jrick error = logmsg_ref(commit_id, GOT_WORKTREE_BACKOUT_REF_PREFIX,
8562 24df9a28 2023-07-08 jrick worktree, repo);
8563 24df9a28 2023-07-08 jrick if (error)
8564 24df9a28 2023-07-08 jrick goto done;
8565 24df9a28 2023-07-08 jrick printf("Backed out commit %s\n", commit_id_str);
8566 24df9a28 2023-07-08 jrick }
8567 24df9a28 2023-07-08 jrick print_merge_progress_stats(&upa);
8568 24df9a28 2023-07-08 jrick done:
8569 24df9a28 2023-07-08 jrick free(cwd);
8570 24df9a28 2023-07-08 jrick if (commit)
8571 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8572 24df9a28 2023-07-08 jrick free(commit_id_str);
8573 24df9a28 2023-07-08 jrick if (worktree)
8574 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
8575 24df9a28 2023-07-08 jrick if (repo) {
8576 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
8577 24df9a28 2023-07-08 jrick if (error == NULL)
8578 24df9a28 2023-07-08 jrick error = close_err;
8579 24df9a28 2023-07-08 jrick }
8580 24df9a28 2023-07-08 jrick if (pack_fds) {
8581 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
8582 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
8583 24df9a28 2023-07-08 jrick if (error == NULL)
8584 24df9a28 2023-07-08 jrick error = pack_err;
8585 24df9a28 2023-07-08 jrick }
8586 24df9a28 2023-07-08 jrick return error;
8587 24df9a28 2023-07-08 jrick }
8588 24df9a28 2023-07-08 jrick
8589 24df9a28 2023-07-08 jrick __dead static void
8590 24df9a28 2023-07-08 jrick usage_cat(void)
8591 24df9a28 2023-07-08 jrick {
8592 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s cat [-P] [-c commit] [-r repository-path] "
8593 24df9a28 2023-07-08 jrick "arg ...\n", getprogname());
8594 24df9a28 2023-07-08 jrick exit(1);
8595 24df9a28 2023-07-08 jrick }
8596 24df9a28 2023-07-08 jrick
8597 24df9a28 2023-07-08 jrick static const struct got_error *
8598 24df9a28 2023-07-08 jrick cat_blob(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
8599 24df9a28 2023-07-08 jrick {
8600 24df9a28 2023-07-08 jrick const struct got_error *err;
8601 24df9a28 2023-07-08 jrick struct got_blob_object *blob;
8602 24df9a28 2023-07-08 jrick int fd = -1;
8603 24df9a28 2023-07-08 jrick
8604 24df9a28 2023-07-08 jrick fd = got_opentempfd();
8605 24df9a28 2023-07-08 jrick if (fd == -1)
8606 24df9a28 2023-07-08 jrick return got_error_from_errno("got_opentempfd");
8607 24df9a28 2023-07-08 jrick
8608 24df9a28 2023-07-08 jrick err = got_object_open_as_blob(&blob, repo, id, 8192, fd);
8609 24df9a28 2023-07-08 jrick if (err)
8610 24df9a28 2023-07-08 jrick goto done;
8611 24df9a28 2023-07-08 jrick
8612 24df9a28 2023-07-08 jrick err = got_object_blob_dump_to_file(NULL, NULL, NULL, outfile, blob);
8613 24df9a28 2023-07-08 jrick done:
8614 24df9a28 2023-07-08 jrick if (fd != -1 && close(fd) == -1 && err == NULL)
8615 24df9a28 2023-07-08 jrick err = got_error_from_errno("close");
8616 24df9a28 2023-07-08 jrick if (blob)
8617 24df9a28 2023-07-08 jrick got_object_blob_close(blob);
8618 24df9a28 2023-07-08 jrick return err;
8619 24df9a28 2023-07-08 jrick }
8620 24df9a28 2023-07-08 jrick
8621 24df9a28 2023-07-08 jrick static const struct got_error *
8622 24df9a28 2023-07-08 jrick cat_tree(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
8623 24df9a28 2023-07-08 jrick {
8624 24df9a28 2023-07-08 jrick const struct got_error *err;
8625 24df9a28 2023-07-08 jrick struct got_tree_object *tree;
8626 24df9a28 2023-07-08 jrick int nentries, i;
8627 24df9a28 2023-07-08 jrick
8628 24df9a28 2023-07-08 jrick err = got_object_open_as_tree(&tree, repo, id);
8629 24df9a28 2023-07-08 jrick if (err)
8630 24df9a28 2023-07-08 jrick return err;
8631 24df9a28 2023-07-08 jrick
8632 24df9a28 2023-07-08 jrick nentries = got_object_tree_get_nentries(tree);
8633 24df9a28 2023-07-08 jrick for (i = 0; i < nentries; i++) {
8634 24df9a28 2023-07-08 jrick struct got_tree_entry *te;
8635 24df9a28 2023-07-08 jrick char *id_str;
8636 24df9a28 2023-07-08 jrick if (sigint_received || sigpipe_received)
8637 24df9a28 2023-07-08 jrick break;
8638 24df9a28 2023-07-08 jrick te = got_object_tree_get_entry(tree, i);
8639 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, got_tree_entry_get_id(te));
8640 24df9a28 2023-07-08 jrick if (err)
8641 24df9a28 2023-07-08 jrick break;
8642 24df9a28 2023-07-08 jrick fprintf(outfile, "%s %.7o %s\n", id_str,
8643 24df9a28 2023-07-08 jrick got_tree_entry_get_mode(te),
8644 24df9a28 2023-07-08 jrick got_tree_entry_get_name(te));
8645 24df9a28 2023-07-08 jrick free(id_str);
8646 24df9a28 2023-07-08 jrick }
8647 24df9a28 2023-07-08 jrick
8648 24df9a28 2023-07-08 jrick got_object_tree_close(tree);
8649 24df9a28 2023-07-08 jrick return err;
8650 24df9a28 2023-07-08 jrick }
8651 24df9a28 2023-07-08 jrick
8652 24df9a28 2023-07-08 jrick static const struct got_error *
8653 24df9a28 2023-07-08 jrick cat_commit(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
8654 24df9a28 2023-07-08 jrick {
8655 24df9a28 2023-07-08 jrick const struct got_error *err;
8656 24df9a28 2023-07-08 jrick struct got_commit_object *commit;
8657 24df9a28 2023-07-08 jrick const struct got_object_id_queue *parent_ids;
8658 24df9a28 2023-07-08 jrick struct got_object_qid *pid;
8659 24df9a28 2023-07-08 jrick char *id_str = NULL;
8660 24df9a28 2023-07-08 jrick const char *logmsg = NULL;
8661 24df9a28 2023-07-08 jrick char gmtoff[6];
8662 24df9a28 2023-07-08 jrick
8663 24df9a28 2023-07-08 jrick err = got_object_open_as_commit(&commit, repo, id);
8664 24df9a28 2023-07-08 jrick if (err)
8665 24df9a28 2023-07-08 jrick return err;
8666 24df9a28 2023-07-08 jrick
8667 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, got_object_commit_get_tree_id(commit));
8668 24df9a28 2023-07-08 jrick if (err)
8669 24df9a28 2023-07-08 jrick goto done;
8670 24df9a28 2023-07-08 jrick
8671 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_COMMIT_LABEL_TREE, id_str);
8672 24df9a28 2023-07-08 jrick parent_ids = got_object_commit_get_parent_ids(commit);
8673 24df9a28 2023-07-08 jrick fprintf(outfile, "numparents %d\n",
8674 24df9a28 2023-07-08 jrick got_object_commit_get_nparents(commit));
8675 24df9a28 2023-07-08 jrick STAILQ_FOREACH(pid, parent_ids, entry) {
8676 24df9a28 2023-07-08 jrick char *pid_str;
8677 24df9a28 2023-07-08 jrick err = got_object_id_str(&pid_str, &pid->id);
8678 24df9a28 2023-07-08 jrick if (err)
8679 24df9a28 2023-07-08 jrick goto done;
8680 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_COMMIT_LABEL_PARENT, pid_str);
8681 24df9a28 2023-07-08 jrick free(pid_str);
8682 24df9a28 2023-07-08 jrick }
8683 24df9a28 2023-07-08 jrick got_date_format_gmtoff(gmtoff, sizeof(gmtoff),
8684 24df9a28 2023-07-08 jrick got_object_commit_get_author_gmtoff(commit));
8685 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s %lld %s\n", GOT_COMMIT_LABEL_AUTHOR,
8686 24df9a28 2023-07-08 jrick got_object_commit_get_author(commit),
8687 24df9a28 2023-07-08 jrick (long long)got_object_commit_get_author_time(commit),
8688 24df9a28 2023-07-08 jrick gmtoff);
8689 24df9a28 2023-07-08 jrick
8690 24df9a28 2023-07-08 jrick got_date_format_gmtoff(gmtoff, sizeof(gmtoff),
8691 24df9a28 2023-07-08 jrick got_object_commit_get_committer_gmtoff(commit));
8692 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s %lld %s\n", GOT_COMMIT_LABEL_COMMITTER,
8693 24df9a28 2023-07-08 jrick got_object_commit_get_committer(commit),
8694 24df9a28 2023-07-08 jrick (long long)got_object_commit_get_committer_time(commit),
8695 24df9a28 2023-07-08 jrick gmtoff);
8696 24df9a28 2023-07-08 jrick
8697 24df9a28 2023-07-08 jrick logmsg = got_object_commit_get_logmsg_raw(commit);
8698 24df9a28 2023-07-08 jrick fprintf(outfile, "messagelen %zd\n", strlen(logmsg));
8699 24df9a28 2023-07-08 jrick fprintf(outfile, "%s", logmsg);
8700 24df9a28 2023-07-08 jrick done:
8701 24df9a28 2023-07-08 jrick free(id_str);
8702 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8703 24df9a28 2023-07-08 jrick return err;
8704 24df9a28 2023-07-08 jrick }
8705 24df9a28 2023-07-08 jrick
8706 24df9a28 2023-07-08 jrick static const struct got_error *
8707 24df9a28 2023-07-08 jrick cat_tag(struct got_object_id *id, struct got_repository *repo, FILE *outfile)
8708 24df9a28 2023-07-08 jrick {
8709 24df9a28 2023-07-08 jrick const struct got_error *err;
8710 24df9a28 2023-07-08 jrick struct got_tag_object *tag;
8711 24df9a28 2023-07-08 jrick char *id_str = NULL;
8712 24df9a28 2023-07-08 jrick const char *tagmsg = NULL;
8713 24df9a28 2023-07-08 jrick char gmtoff[6];
8714 24df9a28 2023-07-08 jrick
8715 24df9a28 2023-07-08 jrick err = got_object_open_as_tag(&tag, repo, id);
8716 24df9a28 2023-07-08 jrick if (err)
8717 24df9a28 2023-07-08 jrick return err;
8718 24df9a28 2023-07-08 jrick
8719 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, got_object_tag_get_object_id(tag));
8720 24df9a28 2023-07-08 jrick if (err)
8721 24df9a28 2023-07-08 jrick goto done;
8722 24df9a28 2023-07-08 jrick
8723 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_OBJECT, id_str);
8724 24df9a28 2023-07-08 jrick
8725 24df9a28 2023-07-08 jrick switch (got_object_tag_get_object_type(tag)) {
8726 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_BLOB:
8727 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
8728 24df9a28 2023-07-08 jrick GOT_OBJ_LABEL_BLOB);
8729 24df9a28 2023-07-08 jrick break;
8730 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TREE:
8731 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
8732 24df9a28 2023-07-08 jrick GOT_OBJ_LABEL_TREE);
8733 24df9a28 2023-07-08 jrick break;
8734 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_COMMIT:
8735 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
8736 24df9a28 2023-07-08 jrick GOT_OBJ_LABEL_COMMIT);
8737 24df9a28 2023-07-08 jrick break;
8738 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TAG:
8739 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TYPE,
8740 24df9a28 2023-07-08 jrick GOT_OBJ_LABEL_TAG);
8741 24df9a28 2023-07-08 jrick break;
8742 24df9a28 2023-07-08 jrick default:
8743 24df9a28 2023-07-08 jrick break;
8744 24df9a28 2023-07-08 jrick }
8745 24df9a28 2023-07-08 jrick
8746 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s\n", GOT_TAG_LABEL_TAG,
8747 24df9a28 2023-07-08 jrick got_object_tag_get_name(tag));
8748 24df9a28 2023-07-08 jrick
8749 24df9a28 2023-07-08 jrick got_date_format_gmtoff(gmtoff, sizeof(gmtoff),
8750 24df9a28 2023-07-08 jrick got_object_tag_get_tagger_gmtoff(tag));
8751 24df9a28 2023-07-08 jrick fprintf(outfile, "%s%s %lld %s\n", GOT_TAG_LABEL_TAGGER,
8752 24df9a28 2023-07-08 jrick got_object_tag_get_tagger(tag),
8753 24df9a28 2023-07-08 jrick (long long)got_object_tag_get_tagger_time(tag),
8754 24df9a28 2023-07-08 jrick gmtoff);
8755 24df9a28 2023-07-08 jrick
8756 24df9a28 2023-07-08 jrick tagmsg = got_object_tag_get_message(tag);
8757 24df9a28 2023-07-08 jrick fprintf(outfile, "messagelen %zd\n", strlen(tagmsg));
8758 24df9a28 2023-07-08 jrick fprintf(outfile, "%s", tagmsg);
8759 24df9a28 2023-07-08 jrick done:
8760 24df9a28 2023-07-08 jrick free(id_str);
8761 24df9a28 2023-07-08 jrick got_object_tag_close(tag);
8762 24df9a28 2023-07-08 jrick return err;
8763 24df9a28 2023-07-08 jrick }
8764 24df9a28 2023-07-08 jrick
8765 24df9a28 2023-07-08 jrick static const struct got_error *
8766 24df9a28 2023-07-08 jrick cmd_cat(int argc, char *argv[])
8767 24df9a28 2023-07-08 jrick {
8768 24df9a28 2023-07-08 jrick const struct got_error *error;
8769 24df9a28 2023-07-08 jrick struct got_repository *repo = NULL;
8770 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
8771 24df9a28 2023-07-08 jrick char *cwd = NULL, *repo_path = NULL, *label = NULL;
8772 24df9a28 2023-07-08 jrick const char *commit_id_str = NULL;
8773 24df9a28 2023-07-08 jrick struct got_object_id *id = NULL, *commit_id = NULL;
8774 24df9a28 2023-07-08 jrick struct got_commit_object *commit = NULL;
8775 24df9a28 2023-07-08 jrick int ch, obj_type, i, force_path = 0;
8776 24df9a28 2023-07-08 jrick struct got_reflist_head refs;
8777 24df9a28 2023-07-08 jrick int *pack_fds = NULL;
8778 24df9a28 2023-07-08 jrick
8779 24df9a28 2023-07-08 jrick TAILQ_INIT(&refs);
8780 24df9a28 2023-07-08 jrick
8781 24df9a28 2023-07-08 jrick #ifndef PROFILE
8782 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
8783 24df9a28 2023-07-08 jrick NULL) == -1)
8784 24df9a28 2023-07-08 jrick err(1, "pledge");
8785 24df9a28 2023-07-08 jrick #endif
8786 24df9a28 2023-07-08 jrick
8787 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "c:Pr:")) != -1) {
8788 24df9a28 2023-07-08 jrick switch (ch) {
8789 24df9a28 2023-07-08 jrick case 'c':
8790 24df9a28 2023-07-08 jrick commit_id_str = optarg;
8791 24df9a28 2023-07-08 jrick break;
8792 24df9a28 2023-07-08 jrick case 'P':
8793 24df9a28 2023-07-08 jrick force_path = 1;
8794 24df9a28 2023-07-08 jrick break;
8795 24df9a28 2023-07-08 jrick case 'r':
8796 24df9a28 2023-07-08 jrick repo_path = realpath(optarg, NULL);
8797 24df9a28 2023-07-08 jrick if (repo_path == NULL)
8798 24df9a28 2023-07-08 jrick return got_error_from_errno2("realpath",
8799 24df9a28 2023-07-08 jrick optarg);
8800 24df9a28 2023-07-08 jrick got_path_strip_trailing_slashes(repo_path);
8801 24df9a28 2023-07-08 jrick break;
8802 24df9a28 2023-07-08 jrick default:
8803 24df9a28 2023-07-08 jrick usage_cat();
8804 24df9a28 2023-07-08 jrick /* NOTREACHED */
8805 24df9a28 2023-07-08 jrick }
8806 24df9a28 2023-07-08 jrick }
8807 24df9a28 2023-07-08 jrick
8808 24df9a28 2023-07-08 jrick argc -= optind;
8809 24df9a28 2023-07-08 jrick argv += optind;
8810 24df9a28 2023-07-08 jrick
8811 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
8812 24df9a28 2023-07-08 jrick if (cwd == NULL) {
8813 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
8814 24df9a28 2023-07-08 jrick goto done;
8815 24df9a28 2023-07-08 jrick }
8816 24df9a28 2023-07-08 jrick
8817 24df9a28 2023-07-08 jrick error = got_repo_pack_fds_open(&pack_fds);
8818 24df9a28 2023-07-08 jrick if (error != NULL)
8819 24df9a28 2023-07-08 jrick goto done;
8820 24df9a28 2023-07-08 jrick
8821 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
8822 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
8823 24df9a28 2023-07-08 jrick if (error && error->code != GOT_ERR_NOT_WORKTREE)
8824 24df9a28 2023-07-08 jrick goto done;
8825 24df9a28 2023-07-08 jrick if (worktree) {
8826 24df9a28 2023-07-08 jrick repo_path = strdup(
8827 24df9a28 2023-07-08 jrick got_worktree_get_repo_path(worktree));
8828 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
8829 24df9a28 2023-07-08 jrick error = got_error_from_errno("strdup");
8830 24df9a28 2023-07-08 jrick goto done;
8831 24df9a28 2023-07-08 jrick }
8832 24df9a28 2023-07-08 jrick
8833 24df9a28 2023-07-08 jrick /* Release work tree lock. */
8834 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
8835 24df9a28 2023-07-08 jrick worktree = NULL;
8836 24df9a28 2023-07-08 jrick }
8837 24df9a28 2023-07-08 jrick }
8838 24df9a28 2023-07-08 jrick
8839 24df9a28 2023-07-08 jrick if (repo_path == NULL) {
8840 24df9a28 2023-07-08 jrick repo_path = strdup(cwd);
8841 24df9a28 2023-07-08 jrick if (repo_path == NULL)
8842 24df9a28 2023-07-08 jrick return got_error_from_errno("strdup");
8843 24df9a28 2023-07-08 jrick }
8844 24df9a28 2023-07-08 jrick
8845 24df9a28 2023-07-08 jrick error = got_repo_open(&repo, repo_path, NULL, pack_fds);
8846 24df9a28 2023-07-08 jrick free(repo_path);
8847 24df9a28 2023-07-08 jrick if (error != NULL)
8848 24df9a28 2023-07-08 jrick goto done;
8849 24df9a28 2023-07-08 jrick
8850 24df9a28 2023-07-08 jrick error = apply_unveil(got_repo_get_path(repo), 1, NULL);
8851 24df9a28 2023-07-08 jrick if (error)
8852 24df9a28 2023-07-08 jrick goto done;
8853 24df9a28 2023-07-08 jrick
8854 24df9a28 2023-07-08 jrick error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
8855 24df9a28 2023-07-08 jrick if (error)
8856 24df9a28 2023-07-08 jrick goto done;
8857 24df9a28 2023-07-08 jrick
8858 24df9a28 2023-07-08 jrick if (commit_id_str == NULL)
8859 24df9a28 2023-07-08 jrick commit_id_str = GOT_REF_HEAD;
8860 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&commit_id, NULL,
8861 24df9a28 2023-07-08 jrick commit_id_str, GOT_OBJ_TYPE_COMMIT, &refs, repo);
8862 24df9a28 2023-07-08 jrick if (error)
8863 24df9a28 2023-07-08 jrick goto done;
8864 24df9a28 2023-07-08 jrick
8865 24df9a28 2023-07-08 jrick error = got_object_open_as_commit(&commit, repo, commit_id);
8866 24df9a28 2023-07-08 jrick if (error)
8867 24df9a28 2023-07-08 jrick goto done;
8868 24df9a28 2023-07-08 jrick
8869 24df9a28 2023-07-08 jrick for (i = 0; i < argc; i++) {
8870 24df9a28 2023-07-08 jrick if (force_path) {
8871 24df9a28 2023-07-08 jrick error = got_object_id_by_path(&id, repo, commit,
8872 24df9a28 2023-07-08 jrick argv[i]);
8873 24df9a28 2023-07-08 jrick if (error)
8874 24df9a28 2023-07-08 jrick break;
8875 24df9a28 2023-07-08 jrick } else {
8876 24df9a28 2023-07-08 jrick error = got_repo_match_object_id(&id, &label, argv[i],
8877 24df9a28 2023-07-08 jrick GOT_OBJ_TYPE_ANY, NULL /* do not resolve tags */,
8878 24df9a28 2023-07-08 jrick repo);
8879 24df9a28 2023-07-08 jrick if (error) {
8880 24df9a28 2023-07-08 jrick if (error->code != GOT_ERR_BAD_OBJ_ID_STR &&
8881 24df9a28 2023-07-08 jrick error->code != GOT_ERR_NOT_REF)
8882 24df9a28 2023-07-08 jrick break;
8883 24df9a28 2023-07-08 jrick error = got_object_id_by_path(&id, repo,
8884 24df9a28 2023-07-08 jrick commit, argv[i]);
8885 24df9a28 2023-07-08 jrick if (error)
8886 24df9a28 2023-07-08 jrick break;
8887 24df9a28 2023-07-08 jrick }
8888 24df9a28 2023-07-08 jrick }
8889 24df9a28 2023-07-08 jrick
8890 24df9a28 2023-07-08 jrick error = got_object_get_type(&obj_type, repo, id);
8891 24df9a28 2023-07-08 jrick if (error)
8892 24df9a28 2023-07-08 jrick break;
8893 24df9a28 2023-07-08 jrick
8894 24df9a28 2023-07-08 jrick switch (obj_type) {
8895 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_BLOB:
8896 24df9a28 2023-07-08 jrick error = cat_blob(id, repo, stdout);
8897 24df9a28 2023-07-08 jrick break;
8898 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TREE:
8899 24df9a28 2023-07-08 jrick error = cat_tree(id, repo, stdout);
8900 24df9a28 2023-07-08 jrick break;
8901 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_COMMIT:
8902 24df9a28 2023-07-08 jrick error = cat_commit(id, repo, stdout);
8903 24df9a28 2023-07-08 jrick break;
8904 24df9a28 2023-07-08 jrick case GOT_OBJ_TYPE_TAG:
8905 24df9a28 2023-07-08 jrick error = cat_tag(id, repo, stdout);
8906 24df9a28 2023-07-08 jrick break;
8907 24df9a28 2023-07-08 jrick default:
8908 24df9a28 2023-07-08 jrick error = got_error(GOT_ERR_OBJ_TYPE);
8909 24df9a28 2023-07-08 jrick break;
8910 24df9a28 2023-07-08 jrick }
8911 24df9a28 2023-07-08 jrick if (error)
8912 24df9a28 2023-07-08 jrick break;
8913 24df9a28 2023-07-08 jrick free(label);
8914 24df9a28 2023-07-08 jrick label = NULL;
8915 24df9a28 2023-07-08 jrick free(id);
8916 24df9a28 2023-07-08 jrick id = NULL;
8917 24df9a28 2023-07-08 jrick }
8918 24df9a28 2023-07-08 jrick done:
8919 24df9a28 2023-07-08 jrick free(label);
8920 24df9a28 2023-07-08 jrick free(id);
8921 24df9a28 2023-07-08 jrick free(commit_id);
8922 24df9a28 2023-07-08 jrick if (commit)
8923 24df9a28 2023-07-08 jrick got_object_commit_close(commit);
8924 24df9a28 2023-07-08 jrick if (worktree)
8925 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
8926 24df9a28 2023-07-08 jrick if (repo) {
8927 24df9a28 2023-07-08 jrick const struct got_error *close_err = got_repo_close(repo);
8928 24df9a28 2023-07-08 jrick if (error == NULL)
8929 24df9a28 2023-07-08 jrick error = close_err;
8930 24df9a28 2023-07-08 jrick }
8931 24df9a28 2023-07-08 jrick if (pack_fds) {
8932 24df9a28 2023-07-08 jrick const struct got_error *pack_err =
8933 24df9a28 2023-07-08 jrick got_repo_pack_fds_close(pack_fds);
8934 24df9a28 2023-07-08 jrick if (error == NULL)
8935 24df9a28 2023-07-08 jrick error = pack_err;
8936 24df9a28 2023-07-08 jrick }
8937 24df9a28 2023-07-08 jrick
8938 24df9a28 2023-07-08 jrick got_ref_list_free(&refs);
8939 24df9a28 2023-07-08 jrick return error;
8940 24df9a28 2023-07-08 jrick }
8941 24df9a28 2023-07-08 jrick
8942 24df9a28 2023-07-08 jrick __dead static void
8943 24df9a28 2023-07-08 jrick usage_info(void)
8944 24df9a28 2023-07-08 jrick {
8945 24df9a28 2023-07-08 jrick fprintf(stderr, "usage: %s info [path ...]\n",
8946 24df9a28 2023-07-08 jrick getprogname());
8947 24df9a28 2023-07-08 jrick exit(1);
8948 24df9a28 2023-07-08 jrick }
8949 24df9a28 2023-07-08 jrick
8950 24df9a28 2023-07-08 jrick static const struct got_error *
8951 24df9a28 2023-07-08 jrick print_path_info(void *arg, const char *path, mode_t mode, time_t mtime,
8952 24df9a28 2023-07-08 jrick struct got_object_id *blob_id, struct got_object_id *staged_blob_id,
8953 24df9a28 2023-07-08 jrick struct got_object_id *commit_id)
8954 24df9a28 2023-07-08 jrick {
8955 24df9a28 2023-07-08 jrick const struct got_error *err = NULL;
8956 24df9a28 2023-07-08 jrick char *id_str = NULL;
8957 24df9a28 2023-07-08 jrick char datebuf[128];
8958 24df9a28 2023-07-08 jrick struct tm mytm, *tm;
8959 24df9a28 2023-07-08 jrick struct got_pathlist_head *paths = arg;
8960 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
8961 24df9a28 2023-07-08 jrick
8962 24df9a28 2023-07-08 jrick /*
8963 24df9a28 2023-07-08 jrick * Clear error indication from any of the path arguments which
8964 24df9a28 2023-07-08 jrick * would cause this file index entry to be displayed.
8965 24df9a28 2023-07-08 jrick */
8966 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, paths, entry) {
8967 24df9a28 2023-07-08 jrick if (got_path_cmp(path, pe->path, strlen(path),
8968 24df9a28 2023-07-08 jrick pe->path_len) == 0 ||
8969 24df9a28 2023-07-08 jrick got_path_is_child(path, pe->path, pe->path_len))
8970 24df9a28 2023-07-08 jrick pe->data = NULL; /* no error */
8971 24df9a28 2023-07-08 jrick }
8972 24df9a28 2023-07-08 jrick
8973 24df9a28 2023-07-08 jrick printf(GOT_COMMIT_SEP_STR);
8974 24df9a28 2023-07-08 jrick if (S_ISLNK(mode))
8975 24df9a28 2023-07-08 jrick printf("symlink: %s\n", path);
8976 24df9a28 2023-07-08 jrick else if (S_ISREG(mode)) {
8977 24df9a28 2023-07-08 jrick printf("file: %s\n", path);
8978 24df9a28 2023-07-08 jrick printf("mode: %o\n", mode & (S_IRWXU | S_IRWXG | S_IRWXO));
8979 24df9a28 2023-07-08 jrick } else if (S_ISDIR(mode))
8980 24df9a28 2023-07-08 jrick printf("directory: %s\n", path);
8981 24df9a28 2023-07-08 jrick else
8982 24df9a28 2023-07-08 jrick printf("something: %s\n", path);
8983 24df9a28 2023-07-08 jrick
8984 24df9a28 2023-07-08 jrick tm = localtime_r(&mtime, &mytm);
8985 24df9a28 2023-07-08 jrick if (tm == NULL)
8986 24df9a28 2023-07-08 jrick return NULL;
8987 24df9a28 2023-07-08 jrick if (strftime(datebuf, sizeof(datebuf), "%c %Z", tm) == 0)
8988 24df9a28 2023-07-08 jrick return got_error(GOT_ERR_NO_SPACE);
8989 24df9a28 2023-07-08 jrick printf("timestamp: %s\n", datebuf);
8990 24df9a28 2023-07-08 jrick
8991 24df9a28 2023-07-08 jrick if (blob_id) {
8992 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, blob_id);
8993 24df9a28 2023-07-08 jrick if (err)
8994 24df9a28 2023-07-08 jrick return err;
8995 24df9a28 2023-07-08 jrick printf("based on blob: %s\n", id_str);
8996 24df9a28 2023-07-08 jrick free(id_str);
8997 24df9a28 2023-07-08 jrick }
8998 24df9a28 2023-07-08 jrick
8999 24df9a28 2023-07-08 jrick if (staged_blob_id) {
9000 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, staged_blob_id);
9001 24df9a28 2023-07-08 jrick if (err)
9002 24df9a28 2023-07-08 jrick return err;
9003 24df9a28 2023-07-08 jrick printf("based on staged blob: %s\n", id_str);
9004 24df9a28 2023-07-08 jrick free(id_str);
9005 24df9a28 2023-07-08 jrick }
9006 24df9a28 2023-07-08 jrick
9007 24df9a28 2023-07-08 jrick if (commit_id) {
9008 24df9a28 2023-07-08 jrick err = got_object_id_str(&id_str, commit_id);
9009 24df9a28 2023-07-08 jrick if (err)
9010 24df9a28 2023-07-08 jrick return err;
9011 24df9a28 2023-07-08 jrick printf("based on commit: %s\n", id_str);
9012 24df9a28 2023-07-08 jrick free(id_str);
9013 24df9a28 2023-07-08 jrick }
9014 24df9a28 2023-07-08 jrick
9015 24df9a28 2023-07-08 jrick return NULL;
9016 24df9a28 2023-07-08 jrick }
9017 24df9a28 2023-07-08 jrick
9018 24df9a28 2023-07-08 jrick static const struct got_error *
9019 24df9a28 2023-07-08 jrick cmd_info(int argc, char *argv[])
9020 24df9a28 2023-07-08 jrick {
9021 24df9a28 2023-07-08 jrick const struct got_error *error = NULL;
9022 24df9a28 2023-07-08 jrick struct got_worktree *worktree = NULL;
9023 24df9a28 2023-07-08 jrick char *cwd = NULL, *id_str = NULL;
9024 24df9a28 2023-07-08 jrick struct got_pathlist_head paths;
9025 24df9a28 2023-07-08 jrick char *uuidstr = NULL;
9026 24df9a28 2023-07-08 jrick int ch, show_files = 0;
9027 24df9a28 2023-07-08 jrick
9028 24df9a28 2023-07-08 jrick TAILQ_INIT(&paths);
9029 24df9a28 2023-07-08 jrick
9030 24df9a28 2023-07-08 jrick #ifndef PROFILE
9031 24df9a28 2023-07-08 jrick if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
9032 24df9a28 2023-07-08 jrick NULL) == -1)
9033 24df9a28 2023-07-08 jrick err(1, "pledge");
9034 24df9a28 2023-07-08 jrick #endif
9035 24df9a28 2023-07-08 jrick
9036 24df9a28 2023-07-08 jrick while ((ch = getopt(argc, argv, "")) != -1) {
9037 24df9a28 2023-07-08 jrick switch (ch) {
9038 24df9a28 2023-07-08 jrick default:
9039 24df9a28 2023-07-08 jrick usage_info();
9040 24df9a28 2023-07-08 jrick /* NOTREACHED */
9041 24df9a28 2023-07-08 jrick }
9042 24df9a28 2023-07-08 jrick }
9043 24df9a28 2023-07-08 jrick
9044 24df9a28 2023-07-08 jrick argc -= optind;
9045 24df9a28 2023-07-08 jrick argv += optind;
9046 24df9a28 2023-07-08 jrick
9047 24df9a28 2023-07-08 jrick cwd = getcwd(NULL, 0);
9048 24df9a28 2023-07-08 jrick if (cwd == NULL) {
9049 24df9a28 2023-07-08 jrick error = got_error_from_errno("getcwd");
9050 24df9a28 2023-07-08 jrick goto done;
9051 24df9a28 2023-07-08 jrick }
9052 24df9a28 2023-07-08 jrick
9053 24df9a28 2023-07-08 jrick error = got_worktree_open(&worktree, cwd);
9054 24df9a28 2023-07-08 jrick if (error) {
9055 24df9a28 2023-07-08 jrick if (error->code == GOT_ERR_NOT_WORKTREE)
9056 24df9a28 2023-07-08 jrick error = wrap_not_worktree_error(error, "info", cwd);
9057 24df9a28 2023-07-08 jrick goto done;
9058 24df9a28 2023-07-08 jrick }
9059 24df9a28 2023-07-08 jrick
9060 24df9a28 2023-07-08 jrick #ifndef PROFILE
9061 24df9a28 2023-07-08 jrick /* Remove "wpath cpath proc exec sendfd" promises. */
9062 24df9a28 2023-07-08 jrick if (pledge("stdio rpath flock unveil", NULL) == -1)
9063 24df9a28 2023-07-08 jrick err(1, "pledge");
9064 24df9a28 2023-07-08 jrick #endif
9065 24df9a28 2023-07-08 jrick error = apply_unveil(NULL, 0, got_worktree_get_root_path(worktree));
9066 24df9a28 2023-07-08 jrick if (error)
9067 24df9a28 2023-07-08 jrick goto done;
9068 24df9a28 2023-07-08 jrick
9069 24df9a28 2023-07-08 jrick if (argc >= 1) {
9070 24df9a28 2023-07-08 jrick error = get_worktree_paths_from_argv(&paths, argc, argv,
9071 24df9a28 2023-07-08 jrick worktree);
9072 24df9a28 2023-07-08 jrick if (error)
9073 24df9a28 2023-07-08 jrick goto done;
9074 24df9a28 2023-07-08 jrick show_files = 1;
9075 24df9a28 2023-07-08 jrick }
9076 24df9a28 2023-07-08 jrick
9077 24df9a28 2023-07-08 jrick error = got_object_id_str(&id_str,
9078 24df9a28 2023-07-08 jrick got_worktree_get_base_commit_id(worktree));
9079 24df9a28 2023-07-08 jrick if (error)
9080 24df9a28 2023-07-08 jrick goto done;
9081 24df9a28 2023-07-08 jrick
9082 24df9a28 2023-07-08 jrick error = got_worktree_get_uuid(&uuidstr, worktree);
9083 24df9a28 2023-07-08 jrick if (error)
9084 24df9a28 2023-07-08 jrick goto done;
9085 24df9a28 2023-07-08 jrick
9086 24df9a28 2023-07-08 jrick printf("work tree: %s\n", got_worktree_get_root_path(worktree));
9087 24df9a28 2023-07-08 jrick printf("work tree base commit: %s\n", id_str);
9088 24df9a28 2023-07-08 jrick printf("work tree path prefix: %s\n",
9089 24df9a28 2023-07-08 jrick got_worktree_get_path_prefix(worktree));
9090 24df9a28 2023-07-08 jrick printf("work tree branch reference: %s\n",
9091 24df9a28 2023-07-08 jrick got_worktree_get_head_ref_name(worktree));
9092 24df9a28 2023-07-08 jrick printf("work tree UUID: %s\n", uuidstr);
9093 24df9a28 2023-07-08 jrick printf("repository: %s\n", got_worktree_get_repo_path(worktree));
9094 24df9a28 2023-07-08 jrick
9095 24df9a28 2023-07-08 jrick if (show_files) {
9096 24df9a28 2023-07-08 jrick struct got_pathlist_entry *pe;
9097 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &paths, entry) {
9098 24df9a28 2023-07-08 jrick if (pe->path_len == 0)
9099 24df9a28 2023-07-08 jrick continue;
9100 24df9a28 2023-07-08 jrick /*
9101 24df9a28 2023-07-08 jrick * Assume this path will fail. This will be corrected
9102 24df9a28 2023-07-08 jrick * in print_path_info() in case the path does suceeed.
9103 24df9a28 2023-07-08 jrick */
9104 24df9a28 2023-07-08 jrick pe->data = (void *)got_error(GOT_ERR_BAD_PATH);
9105 24df9a28 2023-07-08 jrick }
9106 24df9a28 2023-07-08 jrick error = got_worktree_path_info(worktree, &paths,
9107 24df9a28 2023-07-08 jrick print_path_info, &paths, check_cancelled, NULL);
9108 24df9a28 2023-07-08 jrick if (error)
9109 24df9a28 2023-07-08 jrick goto done;
9110 24df9a28 2023-07-08 jrick TAILQ_FOREACH(pe, &paths, entry) {
9111 24df9a28 2023-07-08 jrick if (pe->data != NULL) {
9112 24df9a28 2023-07-08 jrick const struct got_error *perr;
9113 24df9a28 2023-07-08 jrick
9114 24df9a28 2023-07-08 jrick perr = pe->data;
9115 24df9a28 2023-07-08 jrick error = got_error_fmt(perr->code, "%s",
9116 24df9a28 2023-07-08 jrick pe->path);
9117 24df9a28 2023-07-08 jrick break;
9118 24df9a28 2023-07-08 jrick }
9119 24df9a28 2023-07-08 jrick }
9120 24df9a28 2023-07-08 jrick }
9121 24df9a28 2023-07-08 jrick done:
9122 24df9a28 2023-07-08 jrick if (worktree)
9123 24df9a28 2023-07-08 jrick got_worktree_close(worktree);
9124 24df9a28 2023-07-08 jrick got_pathlist_free(&paths, GOT_PATHLIST_FREE_PATH);
9125 24df9a28 2023-07-08 jrick free(cwd);
9126 24df9a28 2023-07-08 jrick free(id_str);
9127 24df9a28 2023-07-08 jrick free(uuidstr);
9128 24df9a28 2023-07-08 jrick return error;
9129 24df9a28 2023-07-08 jrick }