2 5d56da81 2019-01-13 stsp * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
4 5261c201 2018-04-01 stsp * Permission to use, copy, modify, and distribute this software for any
5 5261c201 2018-04-01 stsp * purpose with or without fee is hereby granted, provided that the above
6 5261c201 2018-04-01 stsp * copyright notice and this permission notice appear in all copies.
8 5261c201 2018-04-01 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 5261c201 2018-04-01 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 5261c201 2018-04-01 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 5261c201 2018-04-01 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 5261c201 2018-04-01 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 5261c201 2018-04-01 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 5261c201 2018-04-01 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 5261c201 2018-04-01 stsp #include <sys/types.h>
18 5261c201 2018-04-01 stsp #include <sys/queue.h>
19 9e672c74 2019-03-11 stsp #include <sys/stat.h>
21 0fd469ce 2019-03-11 stsp #include <errno.h>
22 f77a24b0 2019-03-11 stsp #include <ctype.h>
23 a04f49d2 2019-02-04 stsp #include <dirent.h>
24 5261c201 2018-04-01 stsp #include <sha1.h>
25 5261c201 2018-04-01 stsp #include <stdio.h>
26 5261c201 2018-04-01 stsp #include <stdlib.h>
27 5261c201 2018-04-01 stsp #include <string.h>
28 5261c201 2018-04-01 stsp #include <util.h>
29 5261c201 2018-04-01 stsp #include <zlib.h>
30 788c352e 2018-06-16 stsp #include <time.h>
32 5261c201 2018-04-01 stsp #include "got_error.h"
33 5261c201 2018-04-01 stsp #include "got_object.h"
34 5261c201 2018-04-01 stsp #include "got_repository.h"
35 5261c201 2018-04-01 stsp #include "got_reference.h"
36 9e672c74 2019-03-11 stsp #include "got_opentemp.h"
38 5261c201 2018-04-01 stsp #include "got_lib_sha1.h"
39 5261c201 2018-04-01 stsp #include "got_lib_path.h"
40 5261c201 2018-04-01 stsp #include "got_lib_delta.h"
41 63581804 2018-07-09 stsp #include "got_lib_inflate.h"
42 5261c201 2018-04-01 stsp #include "got_lib_object.h"
43 f77a24b0 2019-03-11 stsp #include "got_lib_lockfile.h"
45 fb79db15 2019-02-01 stsp #ifndef nitems
46 fb79db15 2019-02-01 stsp #define nitems(_a) (sizeof(_a) / sizeof((_a)[0]))
49 d1f2edc9 2018-06-13 stsp #define GOT_REF_HEADS "heads"
50 d1f2edc9 2018-06-13 stsp #define GOT_REF_TAGS "tags"
51 d1f2edc9 2018-06-13 stsp #define GOT_REF_REMOTES "remotes"
53 5261c201 2018-04-01 stsp /* A symbolic reference. */
54 5261c201 2018-04-01 stsp struct got_symref {
59 5261c201 2018-04-01 stsp /* A non-symbolic reference (there is no better designation). */
60 5261c201 2018-04-01 stsp struct got_ref {
62 5261c201 2018-04-01 stsp u_int8_t sha1[SHA1_DIGEST_LENGTH];
65 5261c201 2018-04-01 stsp /* A reference which points to an arbitrary object. */
66 5261c201 2018-04-01 stsp struct got_reference {
67 5261c201 2018-04-01 stsp unsigned int flags;
68 5261c201 2018-04-01 stsp #define GOT_REF_IS_SYMBOLIC 0x01
71 5261c201 2018-04-01 stsp struct got_ref ref;
72 5261c201 2018-04-01 stsp struct got_symref symref;
76 5261c201 2018-04-01 stsp static const struct got_error *
77 5261c201 2018-04-01 stsp parse_symref(struct got_reference **ref, const char *name, const char *line)
79 5261c201 2018-04-01 stsp struct got_symref *symref;
80 5261c201 2018-04-01 stsp char *symref_name;
81 5261c201 2018-04-01 stsp char *symref_ref;
83 5261c201 2018-04-01 stsp if (line[0] == '\0')
84 30c0868d 2019-02-03 stsp return got_error(GOT_ERR_BAD_REF_DATA);
86 5261c201 2018-04-01 stsp symref_name = strdup(name);
87 5261c201 2018-04-01 stsp if (symref_name == NULL)
88 5261c201 2018-04-01 stsp return got_error_from_errno();
89 5261c201 2018-04-01 stsp symref_ref = strdup(line);
90 5261c201 2018-04-01 stsp if (symref_ref == NULL) {
91 5261c201 2018-04-01 stsp const struct got_error *err = got_error_from_errno();
92 5261c201 2018-04-01 stsp free(symref_name);
96 5261c201 2018-04-01 stsp *ref = calloc(1, sizeof(**ref));
97 5261c201 2018-04-01 stsp if (*ref == NULL)
98 5261c201 2018-04-01 stsp return got_error_from_errno();
99 5261c201 2018-04-01 stsp (*ref)->flags |= GOT_REF_IS_SYMBOLIC;
100 5261c201 2018-04-01 stsp symref = &((*ref)->ref.symref);
101 5261c201 2018-04-01 stsp symref->name = symref_name;
102 5261c201 2018-04-01 stsp symref->ref = symref_ref;
103 5261c201 2018-04-01 stsp return NULL;
106 5261c201 2018-04-01 stsp static const struct got_error *
107 5261c201 2018-04-01 stsp parse_ref_line(struct got_reference **ref, const char *name, const char *line)
109 5892cdd6 2019-03-10 stsp struct got_object_id id;
111 5261c201 2018-04-01 stsp if (strncmp(line, "ref: ", 5) == 0) {
113 5261c201 2018-04-01 stsp return parse_symref(ref, name, line);
116 5892cdd6 2019-03-10 stsp if (!got_parse_sha1_digest(id.sha1, line))
117 30c0868d 2019-02-03 stsp return got_error(GOT_ERR_BAD_REF_DATA);
119 5892cdd6 2019-03-10 stsp return got_ref_alloc(ref, name, &id);
122 5261c201 2018-04-01 stsp static const struct got_error *
123 5261c201 2018-04-01 stsp parse_ref_file(struct got_reference **ref, const char *name,
124 5261c201 2018-04-01 stsp const char *abspath)
126 5261c201 2018-04-01 stsp const struct got_error *err = NULL;
127 5261c201 2018-04-01 stsp FILE *f = fopen(abspath, "rb");
128 5261c201 2018-04-01 stsp char *line;
129 5261c201 2018-04-01 stsp size_t len;
130 5261c201 2018-04-01 stsp const char delim[3] = {'\0', '\0', '\0'};
132 5261c201 2018-04-01 stsp if (f == NULL)
133 1e37702e 2019-02-04 stsp return NULL;
135 5261c201 2018-04-01 stsp line = fparseln(f, &len, NULL, delim, 0);
136 5261c201 2018-04-01 stsp if (line == NULL) {
137 30c0868d 2019-02-03 stsp err = got_error(GOT_ERR_BAD_REF_DATA);
141 5261c201 2018-04-01 stsp err = parse_ref_line(ref, name, line);
143 5261c201 2018-04-01 stsp free(line);
144 fb43ecf1 2019-02-11 stsp if (fclose(f) != 0 && err == NULL)
145 fb43ecf1 2019-02-11 stsp err = got_error_from_errno();
146 5261c201 2018-04-01 stsp return err;
150 c5f754cc 2019-02-01 stsp is_well_known_ref(const char *refname)
152 c5f754cc 2019-02-01 stsp return (strcmp(refname, GOT_REF_HEAD) == 0 ||
153 5261c201 2018-04-01 stsp strcmp(refname, GOT_REF_ORIG_HEAD) == 0 ||
154 5261c201 2018-04-01 stsp strcmp(refname, GOT_REF_MERGE_HEAD) == 0 ||
155 c5f754cc 2019-02-01 stsp strcmp(refname, GOT_REF_FETCH_HEAD) == 0);
158 c5f754cc 2019-02-01 stsp static char *
159 c5f754cc 2019-02-01 stsp get_refs_dir_path(struct got_repository *repo, const char *refname)
161 c5f754cc 2019-02-01 stsp if (is_well_known_ref(refname) || strncmp(refname, "refs/", 5) == 0)
162 6e9da951 2019-01-06 stsp return strdup(got_repo_get_path_git_dir(repo));
164 5261c201 2018-04-01 stsp return got_repo_get_path_refs(repo);
168 f77a24b0 2019-03-11 stsp is_valid_ref_name(const char *name)
170 f77a24b0 2019-03-11 stsp const char *s, *slash, *seg;
171 f77a24b0 2019-03-11 stsp const char forbidden[] = { ' ', '~', '^', ':', '?', '*', '[' , '\\' };
172 f77a24b0 2019-03-11 stsp const char *forbidden_seq[] = { "//", "..", "@{" };
173 f77a24b0 2019-03-11 stsp const char *lfs = GOT_LOCKFILE_SUFFIX;
174 f77a24b0 2019-03-11 stsp const size_t lfs_len = sizeof(GOT_LOCKFILE_SUFFIX) - 1;
177 f77a24b0 2019-03-11 stsp if (name[0] == '@' && name[1] == '\0')
180 f77a24b0 2019-03-11 stsp slash = strchr(name, '/');
181 f77a24b0 2019-03-11 stsp if (slash == NULL)
186 f77a24b0 2019-03-11 stsp if (seg[0] == '\0' || seg[0] == '.' || seg[0] == '/')
188 f77a24b0 2019-03-11 stsp while (*s) {
189 f77a24b0 2019-03-11 stsp for (i = 0; i < nitems(forbidden); i++) {
190 f77a24b0 2019-03-11 stsp if (*s == forbidden[i])
193 f77a24b0 2019-03-11 stsp for (i = 0; i < nitems(forbidden_seq); i++) {
194 f77a24b0 2019-03-11 stsp if (s[0] == forbidden_seq[i][0] &&
195 f77a24b0 2019-03-11 stsp s[1] == forbidden_seq[i][1])
198 f77a24b0 2019-03-11 stsp if (iscntrl((unsigned char)s[0]))
200 f77a24b0 2019-03-11 stsp if (s[0] == '.' && s[1] == '\0')
202 f77a24b0 2019-03-11 stsp if (*s == '/') {
203 f77a24b0 2019-03-11 stsp const char *nextseg = s + 1;
204 f77a24b0 2019-03-11 stsp if (nextseg[0] == '\0' || nextseg[0] == '.' ||
205 f77a24b0 2019-03-11 stsp nextseg[0] == '/')
207 f77a24b0 2019-03-11 stsp if (seg <= s - lfs_len &&
208 f77a24b0 2019-03-11 stsp strncmp(s - lfs_len, lfs, lfs_len) == 0)
210 f77a24b0 2019-03-11 stsp seg = nextseg;
215 f77a24b0 2019-03-11 stsp if (seg <= s - lfs_len &&
216 f77a24b0 2019-03-11 stsp strncmp(s - lfs_len, lfs, lfs_len) == 0)
222 5892cdd6 2019-03-10 stsp const struct got_error *
223 5892cdd6 2019-03-10 stsp got_ref_alloc(struct got_reference **ref, const char *name,
224 5892cdd6 2019-03-10 stsp struct got_object_id *id)
226 5892cdd6 2019-03-10 stsp const struct got_error *err = NULL;
228 f77a24b0 2019-03-11 stsp if (!is_valid_ref_name(name))
229 f77a24b0 2019-03-11 stsp return got_error(GOT_ERR_BAD_REF_NAME);
231 5892cdd6 2019-03-10 stsp *ref = calloc(1, sizeof(**ref));
232 5892cdd6 2019-03-10 stsp if (*ref == NULL)
233 5892cdd6 2019-03-10 stsp return got_error_from_errno();
235 5892cdd6 2019-03-10 stsp memcpy(&(*ref)->ref.ref.sha1, id->sha1, SHA1_DIGEST_LENGTH);
236 5892cdd6 2019-03-10 stsp (*ref)->ref.ref.name = strdup(name);
237 5892cdd6 2019-03-10 stsp if ((*ref)->ref.ref.name == NULL) {
238 5892cdd6 2019-03-10 stsp err = got_error_from_errno();
239 5892cdd6 2019-03-10 stsp free(*ref);
240 5892cdd6 2019-03-10 stsp *ref = NULL;
242 5892cdd6 2019-03-10 stsp return err;
245 d1f2edc9 2018-06-13 stsp static const struct got_error *
246 fb79db15 2019-02-01 stsp parse_packed_ref_line(struct got_reference **ref, const char *abs_refname,
247 fb79db15 2019-02-01 stsp const char *line)
249 5892cdd6 2019-03-10 stsp struct got_object_id id;
250 5892cdd6 2019-03-10 stsp const char *name;
252 fb79db15 2019-02-01 stsp *ref = NULL;
254 532920c8 2019-02-01 stsp if (line[0] == '#' || line[0] == '^')
255 fb79db15 2019-02-01 stsp return NULL;
257 5892cdd6 2019-03-10 stsp if (!got_parse_sha1_digest(id.sha1, line))
258 30c0868d 2019-02-03 stsp return got_error(GOT_ERR_BAD_REF_DATA);
260 199a4027 2019-02-02 stsp if (abs_refname) {
261 199a4027 2019-02-02 stsp if (strcmp(line + SHA1_DIGEST_STRING_LENGTH, abs_refname) != 0)
262 199a4027 2019-02-02 stsp return NULL;
263 5892cdd6 2019-03-10 stsp name = abs_refname;
265 5892cdd6 2019-03-10 stsp name = line + SHA1_DIGEST_STRING_LENGTH;
267 5892cdd6 2019-03-10 stsp return got_ref_alloc(ref, name, &id);
270 fb79db15 2019-02-01 stsp static const struct got_error *
271 0dec1cc0 2019-02-01 stsp open_packed_ref(struct got_reference **ref, FILE *f, const char **subdirs,
272 0dec1cc0 2019-02-01 stsp int nsubdirs, const char *refname)
274 fb79db15 2019-02-01 stsp const struct got_error *err = NULL;
275 fb79db15 2019-02-01 stsp char *abs_refname;
276 fb79db15 2019-02-01 stsp char *line;
277 fb79db15 2019-02-01 stsp size_t len;
278 fb79db15 2019-02-01 stsp const char delim[3] = {'\0', '\0', '\0'};
279 0dec1cc0 2019-02-01 stsp int i, ref_is_absolute = (strncmp(refname, "refs/", 5) == 0);
281 1e37702e 2019-02-04 stsp *ref = NULL;
283 0dec1cc0 2019-02-01 stsp if (ref_is_absolute)
284 0dec1cc0 2019-02-01 stsp abs_refname = (char *)refname;
286 fb79db15 2019-02-01 stsp line = fparseln(f, &len, NULL, delim, 0);
287 1e37702e 2019-02-04 stsp if (line == NULL)
289 0dec1cc0 2019-02-01 stsp for (i = 0; i < nsubdirs; i++) {
290 0dec1cc0 2019-02-01 stsp if (!ref_is_absolute &&
291 0dec1cc0 2019-02-01 stsp asprintf(&abs_refname, "refs/%s/%s", subdirs[i],
292 0dec1cc0 2019-02-01 stsp refname) == -1)
293 0dec1cc0 2019-02-01 stsp return got_error_from_errno();
294 0dec1cc0 2019-02-01 stsp err = parse_packed_ref_line(ref, abs_refname, line);
295 0dec1cc0 2019-02-01 stsp if (!ref_is_absolute)
296 0dec1cc0 2019-02-01 stsp free(abs_refname);
297 532920c8 2019-02-01 stsp if (err || *ref != NULL)
300 fb79db15 2019-02-01 stsp free(line);
303 fb79db15 2019-02-01 stsp } while (*ref == NULL);
305 fb79db15 2019-02-01 stsp return err;
308 fb79db15 2019-02-01 stsp static const struct got_error *
309 d1f2edc9 2018-06-13 stsp open_ref(struct got_reference **ref, const char *path_refs, const char *subdir,
310 a04f49d2 2019-02-04 stsp const char *name)
312 d1f2edc9 2018-06-13 stsp const struct got_error *err = NULL;
313 a04f49d2 2019-02-04 stsp char *path = NULL;
314 a04f49d2 2019-02-04 stsp char *normpath = NULL;
315 a04f49d2 2019-02-04 stsp char *absname = NULL;
316 a04f49d2 2019-02-04 stsp int ref_is_absolute = (strncmp(name, "refs/", 5) == 0);
317 a04f49d2 2019-02-04 stsp int ref_is_well_known = is_well_known_ref(name);
319 1e37702e 2019-02-04 stsp *ref = NULL;
321 a04f49d2 2019-02-04 stsp if (ref_is_absolute || ref_is_well_known) {
322 a04f49d2 2019-02-04 stsp if (asprintf(&path, "%s/%s", path_refs, name) == -1)
323 a04f49d2 2019-02-04 stsp return got_error_from_errno();
324 a04f49d2 2019-02-04 stsp absname = (char *)name;
326 58908ed0 2019-03-11 stsp if (asprintf(&path, "%s/%s%s%s", path_refs, subdir,
327 58908ed0 2019-03-11 stsp subdir[0] ? "/" : "", name) == -1)
328 a04f49d2 2019-02-04 stsp return got_error_from_errno();
330 58908ed0 2019-03-11 stsp if (asprintf(&absname, "refs/%s%s%s",
331 58908ed0 2019-03-11 stsp subdir, subdir[0] ? "/" : "", name) == -1) {
332 a04f49d2 2019-02-04 stsp err = got_error_from_errno();
337 a04f49d2 2019-02-04 stsp normpath = got_path_normalize(path);
338 d1f2edc9 2018-06-13 stsp if (normpath == NULL) {
339 30c0868d 2019-02-03 stsp err = got_error_from_errno();
343 a04f49d2 2019-02-04 stsp err = parse_ref_file(ref, absname, normpath);
345 a04f49d2 2019-02-04 stsp if (!ref_is_absolute && !ref_is_well_known)
346 a04f49d2 2019-02-04 stsp free(absname);
347 a04f49d2 2019-02-04 stsp free(path);
348 d1f2edc9 2018-06-13 stsp free(normpath);
349 d1f2edc9 2018-06-13 stsp return err;
352 5261c201 2018-04-01 stsp const struct got_error *
353 5261c201 2018-04-01 stsp got_ref_open(struct got_reference **ref, struct got_repository *repo,
354 5261c201 2018-04-01 stsp const char *refname)
356 5261c201 2018-04-01 stsp const struct got_error *err = NULL;
357 c5f754cc 2019-02-01 stsp char *path_refs = NULL;
358 fb79db15 2019-02-01 stsp const char *subdirs[] = {
359 fb79db15 2019-02-01 stsp GOT_REF_HEADS, GOT_REF_TAGS, GOT_REF_REMOTES
361 c5f754cc 2019-02-01 stsp int i, well_known = is_well_known_ref(refname);
363 1e37702e 2019-02-04 stsp *ref = NULL;
365 e135804e 2019-02-10 stsp path_refs = get_refs_dir_path(repo, refname);
366 e135804e 2019-02-10 stsp if (path_refs == NULL) {
367 e135804e 2019-02-10 stsp err = got_error_from_errno();
371 c5f754cc 2019-02-01 stsp if (!well_known) {
372 c5f754cc 2019-02-01 stsp char *packed_refs_path;
375 e135804e 2019-02-10 stsp /* Search on-disk refs before packed refs! */
376 e135804e 2019-02-10 stsp for (i = 0; i < nitems(subdirs); i++) {
377 e135804e 2019-02-10 stsp err = open_ref(ref, path_refs, subdirs[i], refname);
378 e135804e 2019-02-10 stsp if (err || *ref)
382 c5f754cc 2019-02-01 stsp packed_refs_path = got_repo_get_path_packed_refs(repo);
383 e135804e 2019-02-10 stsp if (packed_refs_path == NULL) {
384 e135804e 2019-02-10 stsp err = got_error_from_errno();
388 c5f754cc 2019-02-01 stsp f = fopen(packed_refs_path, "rb");
389 c5f754cc 2019-02-01 stsp free(packed_refs_path);
390 c5f754cc 2019-02-01 stsp if (f != NULL) {
391 0dec1cc0 2019-02-01 stsp err = open_packed_ref(ref, f, subdirs, nitems(subdirs),
393 fb43ecf1 2019-02-11 stsp if (fclose(f) != 0 && err == NULL)
394 fb43ecf1 2019-02-11 stsp err = got_error_from_errno();
395 1e37702e 2019-02-04 stsp if (err || *ref)
400 fb79db15 2019-02-01 stsp err = open_ref(ref, path_refs, "", refname);
404 1e37702e 2019-02-04 stsp if (*ref == NULL)
405 1e37702e 2019-02-04 stsp err = got_error_not_ref(refname);
406 5261c201 2018-04-01 stsp free(path_refs);
407 5261c201 2018-04-01 stsp return err;
411 5261c201 2018-04-01 stsp got_ref_close(struct got_reference *ref)
413 5261c201 2018-04-01 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC)
414 5261c201 2018-04-01 stsp free(ref->ref.symref.name);
416 5261c201 2018-04-01 stsp free(ref->ref.ref.name);
420 5261c201 2018-04-01 stsp struct got_reference *
421 5261c201 2018-04-01 stsp got_ref_dup(struct got_reference *ref)
423 5261c201 2018-04-01 stsp struct got_reference *ret;
425 5261c201 2018-04-01 stsp ret = calloc(1, sizeof(*ret));
426 5261c201 2018-04-01 stsp if (ret == NULL)
427 5261c201 2018-04-01 stsp return NULL;
429 5261c201 2018-04-01 stsp ret->flags = ref->flags;
430 5261c201 2018-04-01 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC) {
431 5261c201 2018-04-01 stsp ret->ref.symref.name = strdup(ref->ref.symref.name);
432 5261c201 2018-04-01 stsp if (ret->ref.symref.name == NULL) {
434 5261c201 2018-04-01 stsp return NULL;
436 5261c201 2018-04-01 stsp ret->ref.symref.ref = strdup(ref->ref.symref.ref);
437 5261c201 2018-04-01 stsp if (ret->ref.symref.ref == NULL) {
438 5261c201 2018-04-01 stsp free(ret->ref.symref.name);
440 5261c201 2018-04-01 stsp return NULL;
443 5261c201 2018-04-01 stsp ref->ref.ref.name = strdup(ref->ref.ref.name);
444 5261c201 2018-04-01 stsp if (ref->ref.ref.name == NULL) {
446 5261c201 2018-04-01 stsp return NULL;
448 5261c201 2018-04-01 stsp memcpy(ret->ref.ref.sha1, ref->ref.ref.sha1,
449 5261c201 2018-04-01 stsp SHA1_DIGEST_LENGTH);
452 5261c201 2018-04-01 stsp return ret;
455 5261c201 2018-04-01 stsp static const struct got_error *
456 5261c201 2018-04-01 stsp resolve_symbolic_ref(struct got_reference **resolved,
457 5261c201 2018-04-01 stsp struct got_repository *repo, struct got_reference *ref)
459 5261c201 2018-04-01 stsp struct got_reference *nextref;
460 5261c201 2018-04-01 stsp const struct got_error *err;
462 5261c201 2018-04-01 stsp err = got_ref_open(&nextref, repo, ref->ref.symref.ref);
464 5261c201 2018-04-01 stsp return err;
466 5261c201 2018-04-01 stsp if (nextref->flags & GOT_REF_IS_SYMBOLIC)
467 5261c201 2018-04-01 stsp err = resolve_symbolic_ref(resolved, repo, nextref);
469 5261c201 2018-04-01 stsp *resolved = got_ref_dup(nextref);
471 5261c201 2018-04-01 stsp got_ref_close(nextref);
472 5261c201 2018-04-01 stsp return err;
475 5261c201 2018-04-01 stsp const struct got_error *
476 5261c201 2018-04-01 stsp got_ref_resolve(struct got_object_id **id, struct got_repository *repo,
477 5261c201 2018-04-01 stsp struct got_reference *ref)
479 5261c201 2018-04-01 stsp const struct got_error *err;
481 5261c201 2018-04-01 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC) {
482 5261c201 2018-04-01 stsp struct got_reference *resolved = NULL;
483 5261c201 2018-04-01 stsp err = resolve_symbolic_ref(&resolved, repo, ref);
484 5261c201 2018-04-01 stsp if (err == NULL)
485 5261c201 2018-04-01 stsp err = got_ref_resolve(id, repo, resolved);
486 5261c201 2018-04-01 stsp free(resolved);
487 5261c201 2018-04-01 stsp return err;
490 5261c201 2018-04-01 stsp *id = calloc(1, sizeof(**id));
491 5261c201 2018-04-01 stsp if (*id == NULL)
492 5261c201 2018-04-01 stsp return got_error_from_errno();
493 5261c201 2018-04-01 stsp memcpy((*id)->sha1, ref->ref.ref.sha1, SHA1_DIGEST_LENGTH);
494 5261c201 2018-04-01 stsp return NULL;
498 5261c201 2018-04-01 stsp got_ref_to_str(struct got_reference *ref)
502 271d2a38 2018-12-25 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC)
503 271d2a38 2018-12-25 stsp return strdup(ref->ref.symref.ref);
505 271d2a38 2018-12-25 stsp str = malloc(SHA1_DIGEST_STRING_LENGTH);
506 271d2a38 2018-12-25 stsp if (str == NULL)
507 271d2a38 2018-12-25 stsp return NULL;
509 271d2a38 2018-12-25 stsp if (got_sha1_digest_to_str(ref->ref.ref.sha1, str,
510 271d2a38 2018-12-25 stsp SHA1_DIGEST_STRING_LENGTH) == NULL) {
512 271d2a38 2018-12-25 stsp return NULL;
515 5261c201 2018-04-01 stsp return str;
518 0bd18d37 2019-02-01 stsp const char *
519 0bd18d37 2019-02-01 stsp got_ref_get_name(struct got_reference *ref)
521 0bd18d37 2019-02-01 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC)
522 0bd18d37 2019-02-01 stsp return ref->ref.symref.name;
524 0bd18d37 2019-02-01 stsp return ref->ref.ref.name;
527 199a4027 2019-02-02 stsp static const struct got_error *
528 29b5c214 2019-02-04 stsp insert_ref(struct got_reflist_head *refs, struct got_reference *ref,
529 199a4027 2019-02-02 stsp struct got_repository *repo)
531 199a4027 2019-02-02 stsp const struct got_error *err;
532 199a4027 2019-02-02 stsp struct got_object_id *id;
533 7a3c76f5 2019-02-05 stsp struct got_reflist_entry *new, *re, *prev;
536 7a3c76f5 2019-02-05 stsp err = got_ref_resolve(&id, repo, ref);
538 7a3c76f5 2019-02-05 stsp return err;
540 7a3c76f5 2019-02-05 stsp new = malloc(sizeof(*re));
541 7a3c76f5 2019-02-05 stsp if (new == NULL) {
543 7a3c76f5 2019-02-05 stsp return got_error_from_errno();
545 7a3c76f5 2019-02-05 stsp new->ref = ref;
546 7a3c76f5 2019-02-05 stsp new->id = id;
549 29b5c214 2019-02-04 stsp * We must de-duplicate entries on insert because packed-refs may
550 29b5c214 2019-02-04 stsp * contain redundant entries. On-disk refs take precedence.
551 29b5c214 2019-02-04 stsp * This code assumes that on-disk revs are read before packed-refs.
552 e397b6db 2019-02-04 stsp * We're iterating the list anyway, so insert elements sorted by name.
554 7a3c76f5 2019-02-05 stsp re = SIMPLEQ_FIRST(refs);
555 7a3c76f5 2019-02-05 stsp while (re) {
556 7a3c76f5 2019-02-05 stsp cmp = got_path_cmp(got_ref_get_name(re->ref),
557 7a3c76f5 2019-02-05 stsp got_ref_get_name(ref));
558 e397b6db 2019-02-04 stsp if (cmp == 0) {
559 7a3c76f5 2019-02-05 stsp free(ref); /* duplicate */
560 7a3c76f5 2019-02-05 stsp return NULL;
561 7a3c76f5 2019-02-05 stsp } else if (cmp > 0) {
563 7a3c76f5 2019-02-05 stsp SIMPLEQ_INSERT_AFTER(refs, prev, new, entry);
565 7a3c76f5 2019-02-05 stsp SIMPLEQ_INSERT_HEAD(refs, new, entry);
566 7a3c76f5 2019-02-05 stsp return NULL;
569 7a3c76f5 2019-02-05 stsp re = SIMPLEQ_NEXT(re, entry);
573 7a3c76f5 2019-02-05 stsp SIMPLEQ_INSERT_TAIL(refs, new, entry);
574 199a4027 2019-02-02 stsp return NULL;
577 a04f49d2 2019-02-04 stsp static const struct got_error *
578 29b5c214 2019-02-04 stsp gather_on_disk_refs(struct got_reflist_head *refs, const char *path_refs,
579 a04f49d2 2019-02-04 stsp const char *subdir, struct got_repository *repo)
581 a04f49d2 2019-02-04 stsp const struct got_error *err = NULL;
582 a04f49d2 2019-02-04 stsp DIR *d = NULL;
583 a04f49d2 2019-02-04 stsp char *path_subdir;
585 a04f49d2 2019-02-04 stsp if (asprintf(&path_subdir, "%s/%s", path_refs, subdir) == -1)
586 a04f49d2 2019-02-04 stsp return got_error_from_errno();
588 a04f49d2 2019-02-04 stsp d = opendir(path_subdir);
589 a04f49d2 2019-02-04 stsp if (d == NULL)
592 a04f49d2 2019-02-04 stsp while (1) {
593 a04f49d2 2019-02-04 stsp struct dirent *dent;
594 a04f49d2 2019-02-04 stsp struct got_reference *ref;
595 a04f49d2 2019-02-04 stsp char *child;
597 a04f49d2 2019-02-04 stsp dent = readdir(d);
598 a04f49d2 2019-02-04 stsp if (dent == NULL)
601 a04f49d2 2019-02-04 stsp if (strcmp(dent->d_name, ".") == 0 ||
602 a04f49d2 2019-02-04 stsp strcmp(dent->d_name, "..") == 0)
605 a04f49d2 2019-02-04 stsp switch (dent->d_type) {
606 a04f49d2 2019-02-04 stsp case DT_REG:
607 a04f49d2 2019-02-04 stsp err = open_ref(&ref, path_refs, subdir, dent->d_name);
611 29b5c214 2019-02-04 stsp err = insert_ref(refs, ref, repo);
616 a04f49d2 2019-02-04 stsp case DT_DIR:
617 a04f49d2 2019-02-04 stsp if (asprintf(&child, "%s%s%s", subdir,
618 a04f49d2 2019-02-04 stsp subdir[0] == '\0' ? "" : "/", dent->d_name) == -1) {
619 a04f49d2 2019-02-04 stsp err = got_error_from_errno();
622 29b5c214 2019-02-04 stsp err = gather_on_disk_refs(refs, path_refs, child, repo);
623 a04f49d2 2019-02-04 stsp free(child);
631 a04f49d2 2019-02-04 stsp closedir(d);
632 a04f49d2 2019-02-04 stsp free(path_subdir);
633 a04f49d2 2019-02-04 stsp return err;
636 199a4027 2019-02-02 stsp const struct got_error *
637 199a4027 2019-02-02 stsp got_ref_list(struct got_reflist_head *refs, struct got_repository *repo)
639 199a4027 2019-02-02 stsp const struct got_error *err;
640 a04f49d2 2019-02-04 stsp char *packed_refs_path, *path_refs = NULL;
641 29b5c214 2019-02-04 stsp FILE *f = NULL;
642 199a4027 2019-02-02 stsp struct got_reference *ref;
644 29b5c214 2019-02-04 stsp /* HEAD ref should always exist. */
645 29b5c214 2019-02-04 stsp path_refs = get_refs_dir_path(repo, GOT_REF_HEAD);
646 29b5c214 2019-02-04 stsp if (path_refs == NULL) {
647 29b5c214 2019-02-04 stsp err = got_error_from_errno();
650 29b5c214 2019-02-04 stsp err = open_ref(&ref, path_refs, "", GOT_REF_HEAD);
653 29b5c214 2019-02-04 stsp err = insert_ref(refs, ref, repo);
657 29b5c214 2019-02-04 stsp /* Gather on-disk refs before parsing packed-refs. */
658 29b5c214 2019-02-04 stsp free(path_refs);
659 29b5c214 2019-02-04 stsp path_refs = get_refs_dir_path(repo, "");
660 29b5c214 2019-02-04 stsp if (path_refs == NULL) {
661 29b5c214 2019-02-04 stsp err = got_error_from_errno();
664 29b5c214 2019-02-04 stsp err = gather_on_disk_refs(refs, path_refs, "", repo);
669 29b5c214 2019-02-04 stsp * The packed-refs file may contain redundant entries, in which
670 29b5c214 2019-02-04 stsp * case on-disk refs take precedence.
672 199a4027 2019-02-02 stsp packed_refs_path = got_repo_get_path_packed_refs(repo);
673 29b5c214 2019-02-04 stsp if (packed_refs_path == NULL) {
674 29b5c214 2019-02-04 stsp err = got_error_from_errno();
678 199a4027 2019-02-02 stsp f = fopen(packed_refs_path, "r");
679 199a4027 2019-02-02 stsp free(packed_refs_path);
681 199a4027 2019-02-02 stsp char *line;
682 199a4027 2019-02-02 stsp size_t len;
683 199a4027 2019-02-02 stsp const char delim[3] = {'\0', '\0', '\0'};
684 199a4027 2019-02-02 stsp while (1) {
685 199a4027 2019-02-02 stsp line = fparseln(f, &len, NULL, delim, 0);
686 199a4027 2019-02-02 stsp if (line == NULL)
688 199a4027 2019-02-02 stsp err = parse_packed_ref_line(&ref, NULL, line);
692 29b5c214 2019-02-04 stsp err = insert_ref(refs, ref, repo);
699 a04f49d2 2019-02-04 stsp free(path_refs);
700 fb43ecf1 2019-02-11 stsp if (f && fclose(f) != 0 && err == NULL)
701 fb43ecf1 2019-02-11 stsp err = got_error_from_errno();
702 199a4027 2019-02-02 stsp return err;
705 9e672c74 2019-03-11 stsp const struct got_error *
706 9e672c74 2019-03-11 stsp got_ref_write(struct got_reference *ref, struct got_repository *repo)
708 9e672c74 2019-03-11 stsp const struct got_error *err = NULL, *unlock_err = NULL;
709 9e672c74 2019-03-11 stsp const char *name = got_ref_get_name(ref);
710 9e672c74 2019-03-11 stsp char *path_refs = NULL, *path = NULL, *tmppath = NULL;
711 9e672c74 2019-03-11 stsp struct got_lockfile *lf = NULL;
712 9e672c74 2019-03-11 stsp FILE *f = NULL;
714 9e672c74 2019-03-11 stsp struct stat sb;
716 9e672c74 2019-03-11 stsp path_refs = get_refs_dir_path(repo, name);
717 9e672c74 2019-03-11 stsp if (path_refs == NULL) {
718 9e672c74 2019-03-11 stsp err = got_error_from_errno();
722 9e672c74 2019-03-11 stsp if (asprintf(&path, "%s/%s", path_refs, name) == -1) {
723 9e672c74 2019-03-11 stsp err = got_error_from_errno();
727 9e672c74 2019-03-11 stsp err = got_opentemp_named(&tmppath, &f, path);
728 9e672c74 2019-03-11 stsp if (f == NULL) {
729 9e672c74 2019-03-11 stsp err = got_error_from_errno();
733 9e672c74 2019-03-11 stsp if (ref->flags & GOT_REF_IS_SYMBOLIC) {
734 9e672c74 2019-03-11 stsp n = fprintf(f, "ref: %s\n", ref->ref.symref.ref);
735 9e672c74 2019-03-11 stsp if (n != strlen(ref->ref.symref.ref) + 6) {
736 9e672c74 2019-03-11 stsp err = got_ferror(f, GOT_ERR_IO);
740 9e672c74 2019-03-11 stsp char hex[SHA1_DIGEST_STRING_LENGTH];
741 9e672c74 2019-03-11 stsp if (got_sha1_digest_to_str(ref->ref.ref.sha1, hex,
742 9e672c74 2019-03-11 stsp sizeof(hex)) == NULL) {
743 9e672c74 2019-03-11 stsp err = got_error(GOT_ERR_BAD_REF_DATA);
746 9e672c74 2019-03-11 stsp n = fprintf(f, "%s\n", hex);
747 8fa2f096 2019-03-11 stsp if (n != sizeof(hex)) {
748 9e672c74 2019-03-11 stsp err = got_ferror(f, GOT_ERR_IO);
753 9e672c74 2019-03-11 stsp err = got_lockfile_lock(&lf, path);
757 9e672c74 2019-03-11 stsp /* XXX: check if old content matches our expectations? */
759 0fd469ce 2019-03-11 stsp if (stat(path, &sb) != 0 && errno != ENOENT) {
760 9e672c74 2019-03-11 stsp err = got_error_from_errno();
764 9e672c74 2019-03-11 stsp if (rename(tmppath, path) != 0) {
765 9e672c74 2019-03-11 stsp err = got_error_from_errno();
768 9e672c74 2019-03-11 stsp free(tmppath);
769 9e672c74 2019-03-11 stsp tmppath = NULL;
771 9e672c74 2019-03-11 stsp if (chmod(path, sb.st_mode) != 0) {
772 9e672c74 2019-03-11 stsp err = got_error_from_errno();
777 9e672c74 2019-03-11 stsp unlock_err = got_lockfile_unlock(lf);
779 9e672c74 2019-03-11 stsp if (fclose(f) != 0 && err == NULL)
780 9e672c74 2019-03-11 stsp err = got_error_from_errno();
782 9e672c74 2019-03-11 stsp free(path_refs);
783 9e672c74 2019-03-11 stsp free(path);
784 9e672c74 2019-03-11 stsp if (tmppath) {
785 9e672c74 2019-03-11 stsp if (unlink(tmppath) != 0 && err == NULL)
786 9e672c74 2019-03-11 stsp err = got_error_from_errno();
787 9e672c74 2019-03-11 stsp free(tmppath);
789 9e672c74 2019-03-11 stsp return err ? err : unlock_err;