1 /*
2  * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 /*
18  * State information for a tracked file in a work tree.
19  * When written to disk, multi-byte fields are written in big-endian.
20  * Some fields are based on results from stat(2). These are only used in
21  * order to detect modifications made to on-disk files, they are never
22  * applied back to the filesystem.
23  */
24 struct got_fileindex_entry {
25 	RB_ENTRY(got_fileindex_entry) entry;
26 	uint64_t ctime_sec;
27 	uint64_t ctime_nsec;
28 	uint64_t mtime_sec;
29 	uint64_t mtime_nsec;
30 	uint32_t uid;
31 	uint32_t gid;
32 	/*
33 	 * On-disk size is truncated to the lower 32 bits.
34 	 * The value is only used to check for modifications anyway.
35 	 */
36 	uint32_t size;
37 
38 	uint16_t mode;
39 #define GOT_FILEIDX_MODE_FILE_TYPE	0x000f
40 #define GOT_FILEIDX_MODE_REGULAR_FILE	1
41 #define GOT_FILEIDX_MODE_SYMLINK	2
42 #define GOT_FILEIDX_MODE_PERMS		0xff10
43 #define GOT_FILEIDX_MODE_PERMS_SHIFT	4
44 
45 	/* SHA1 of corresponding blob in repository. */
46 	uint8_t blob_sha1[SHA1_DIGEST_LENGTH];
47 
48 	/* SHA1 of corresponding base commit in repository. */
49 	uint8_t commit_sha1[SHA1_DIGEST_LENGTH];
50 
51 	uint32_t flags;
52 
53 	/*
54 	 * UNIX-style path, relative to work tree root.
55 	 * Variable length, and NUL-padded to a multiple of 8 on disk.
56 	 */
57 	char *path;
58 
59 	/*
60 	 * (since GOT_FILE_INDEX_VERSION 2)
61 	 * SHA1 of staged blob in repository if stage equals either
62 	 * GOT_FILEIDX_STAGE_MODIFY or GOT_FILEIDX_STAGE_ADD.
63 	 * Otherwise, this field is not written to disk.
64 	 */
65 	uint8_t staged_blob_sha1[SHA1_DIGEST_LENGTH];
66 };
67 
68 /* Modifications explicitly staged for commit. */
69 #define GOT_FILEIDX_STAGE_NONE		0
70 #define GOT_FILEIDX_STAGE_MODIFY	1
71 #define GOT_FILEIDX_STAGE_ADD		2
72 #define GOT_FILEIDX_STAGE_DELETE	3
73 
74 struct got_fileindex;
75 
76 RB_HEAD(got_fileindex_tree, got_fileindex_entry);
77 
78 size_t got_fileindex_entry_path_len(const struct got_fileindex_entry *);
79 
80 static inline int
81 got_fileindex_cmp(const struct got_fileindex_entry *e1,
82     const struct got_fileindex_entry *e2)
83 {
84 	return got_path_cmp(e1->path, e2->path,
85 	    got_fileindex_entry_path_len(e1),
86 	    got_fileindex_entry_path_len(e2));
87 }
88 
89 RB_PROTOTYPE(got_fileindex_tree, got_fileindex_entry, entry, got_fileindex_cmp);
90 
91 /* On-disk file index header structure. */
92 struct got_fileindex_hdr {
93 	uint32_t signature;	/* big-endian */
94 #define GOT_FILE_INDEX_SIGNATURE	0x676f7449 /* 'g', 'o', 't', 'I' */
95 	uint32_t version;	/* big-endian */
96 #define GOT_FILE_INDEX_VERSION	2
97 	uint32_t nentries;	/* big-endian */
98 	/* list of concatenated fileindex entries */
99 	uint8_t sha1[SHA1_DIGEST_LENGTH]; /* checksum of above on-disk data */
100 };
101 
102 uint16_t got_fileindex_perms_from_st(struct stat *);
103 mode_t got_fileindex_perms_to_st(struct got_fileindex_entry *);
104 
105 const struct got_error *got_fileindex_entry_update(struct got_fileindex_entry *,
106     const char *, uint8_t *, uint8_t *, int);
107 const struct got_error *got_fileindex_entry_alloc(struct got_fileindex_entry **,
108     const char *, const char *, uint8_t *, uint8_t *);
109 void got_fileindex_entry_free(struct got_fileindex_entry *);
110 
111 struct got_fileindex *got_fileindex_alloc(void);
112 void got_fileindex_free(struct got_fileindex *);
113 const struct got_error *got_fileindex_write(struct got_fileindex *, FILE *);
114 const struct got_error *got_fileindex_entry_add(struct got_fileindex *,
115     struct got_fileindex_entry *);
116 void got_fileindex_entry_remove(struct got_fileindex *,
117     struct got_fileindex_entry *);
118 struct got_fileindex_entry *got_fileindex_entry_get(struct got_fileindex *,
119     const char *, size_t);
120 const struct got_error *got_fileindex_read(struct got_fileindex *, FILE *);
121 typedef const struct got_error *(*got_fileindex_cb)(void *,
122     struct got_fileindex_entry *);
123 const struct got_error *got_fileindex_for_each_entry_safe(
124     struct got_fileindex *, got_fileindex_cb cb, void *);
125 
126 typedef const struct got_error *(*got_fileindex_diff_tree_old_new_cb)(void *,
127     struct got_fileindex_entry *, struct got_tree_entry *, const char *);
128 typedef const struct got_error *(*got_fileindex_diff_tree_old_cb)(void *,
129     struct got_fileindex_entry *, const char *);
130 typedef const struct got_error *(*got_fileindex_diff_tree_new_cb)(void *,
131     struct got_tree_entry *, const char *);
132 struct got_fileindex_diff_tree_cb {
133 	got_fileindex_diff_tree_old_new_cb diff_old_new;
134 	got_fileindex_diff_tree_old_cb diff_old;
135 	got_fileindex_diff_tree_new_cb diff_new;
136 };
137 const struct got_error *got_fileindex_diff_tree(struct got_fileindex *,
138     struct got_tree_object *, const char *, const char *,
139     struct got_repository *, struct got_fileindex_diff_tree_cb *, void *);
140 
141 typedef const struct got_error *(*got_fileindex_diff_dir_old_new_cb)(void *,
142     struct got_fileindex_entry *, struct dirent *, const char *, int);
143 typedef const struct got_error *(*got_fileindex_diff_dir_old_cb)(void *,
144     struct got_fileindex_entry *, const char *);
145 typedef const struct got_error *(*got_fileindex_diff_dir_new_cb)(void *,
146     struct dirent *, const char *, int);
147 struct got_fileindex_diff_dir_cb {
148 	got_fileindex_diff_dir_old_new_cb diff_old_new;
149 	got_fileindex_diff_dir_old_cb diff_old;
150 	got_fileindex_diff_dir_new_cb diff_new;
151 };
152 const struct got_error *got_fileindex_diff_dir(struct got_fileindex *, int,
153     const char *, const char *, struct got_repository *,
154     struct got_fileindex_diff_dir_cb *, void *);
155 
156 int got_fileindex_entry_has_blob(struct got_fileindex_entry *);
157 int got_fileindex_entry_has_commit(struct got_fileindex_entry *);
158 int got_fileindex_entry_has_file_on_disk(struct got_fileindex_entry *);
159 uint32_t got_fileindex_entry_stage_get(const struct got_fileindex_entry *);
160 void got_fileindex_entry_stage_set(struct got_fileindex_entry *ie, uint32_t);
161 
162 void got_fileindex_entry_mark_deleted_from_disk(struct got_fileindex_entry *);