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_BAD_SYMLINK	3
43 #define GOT_FILEIDX_MODE_PERMS		0xfff0
44 #define GOT_FILEIDX_MODE_PERMS_SHIFT	4
45 
46 	/* SHA1 of corresponding blob in repository. */
47 	uint8_t blob_sha1[SHA1_DIGEST_LENGTH];
48 
49 	/* SHA1 of corresponding base commit in repository. */
50 	uint8_t commit_sha1[SHA1_DIGEST_LENGTH];
51 
52 	uint32_t flags;
53 
54 	/*
55 	 * UNIX-style path, relative to work tree root.
56 	 * Variable length, and NUL-padded to a multiple of 8 on disk.
57 	 */
58 	char *path;
59 
60 	/*
61 	 * (since GOT_FILE_INDEX_VERSION 2)
62 	 * SHA1 of staged blob in repository if stage equals either
63 	 * GOT_FILEIDX_STAGE_MODIFY or GOT_FILEIDX_STAGE_ADD.
64 	 * Otherwise, this field is not written to disk.
65 	 */
66 	uint8_t staged_blob_sha1[SHA1_DIGEST_LENGTH];
67 };
68 
69 /* Modifications explicitly staged for commit. */
70 #define GOT_FILEIDX_STAGE_NONE		0
71 #define GOT_FILEIDX_STAGE_MODIFY	1
72 #define GOT_FILEIDX_STAGE_ADD		2
73 #define GOT_FILEIDX_STAGE_DELETE	3
74 
75 struct got_fileindex;
76 
77 RB_HEAD(got_fileindex_tree, got_fileindex_entry);
78 
79 size_t got_fileindex_entry_path_len(const struct got_fileindex_entry *);
80 
81 static inline int
82 got_fileindex_cmp(const struct got_fileindex_entry *e1,
83     const struct got_fileindex_entry *e2)
84 {
85 	return got_path_cmp(e1->path, e2->path,
86 	    got_fileindex_entry_path_len(e1),
87 	    got_fileindex_entry_path_len(e2));
88 }
89 
90 RB_PROTOTYPE(got_fileindex_tree, got_fileindex_entry, entry, got_fileindex_cmp);
91 
92 /* On-disk file index header structure. */
93 struct got_fileindex_hdr {
94 	uint32_t signature;	/* big-endian */
95 #define GOT_FILE_INDEX_SIGNATURE	0x676f7449 /* 'g', 'o', 't', 'I' */
96 	uint32_t version;	/* big-endian */
97 #define GOT_FILE_INDEX_VERSION	2
98 	uint32_t nentries;	/* big-endian */
99 	/* list of concatenated fileindex entries */
100 	uint8_t sha1[SHA1_DIGEST_LENGTH]; /* checksum of above on-disk data */
101 };
102 
103 uint16_t got_fileindex_perms_from_st(struct stat *);
104 mode_t got_fileindex_perms_to_st(struct got_fileindex_entry *);
105 
106 const struct got_error *got_fileindex_entry_update(struct got_fileindex_entry *,
107     const char *, uint8_t *, uint8_t *, int);
108 const struct got_error *got_fileindex_entry_alloc(struct got_fileindex_entry **,
109     const char *);
110 void got_fileindex_entry_free(struct got_fileindex_entry *);
111 
112 struct got_fileindex *got_fileindex_alloc(void);
113 void got_fileindex_free(struct got_fileindex *);
114 const struct got_error *got_fileindex_write(struct got_fileindex *, FILE *);
115 const struct got_error *got_fileindex_entry_add(struct got_fileindex *,
116     struct got_fileindex_entry *);
117 void got_fileindex_entry_remove(struct got_fileindex *,
118     struct got_fileindex_entry *);
119 struct got_fileindex_entry *got_fileindex_entry_get(struct got_fileindex *,
120     const char *, size_t);
121 const struct got_error *got_fileindex_read(struct got_fileindex *, FILE *);
122 typedef const struct got_error *(*got_fileindex_cb)(void *,
123     struct got_fileindex_entry *);
124 const struct got_error *got_fileindex_for_each_entry_safe(
125     struct got_fileindex *, got_fileindex_cb cb, void *);
126 
127 typedef const struct got_error *(*got_fileindex_diff_tree_old_new_cb)(void *,
128     struct got_fileindex_entry *, struct got_tree_entry *, const char *);
129 typedef const struct got_error *(*got_fileindex_diff_tree_old_cb)(void *,
130     struct got_fileindex_entry *, const char *);
131 typedef const struct got_error *(*got_fileindex_diff_tree_new_cb)(void *,
132     struct got_tree_entry *, const char *);
133 struct got_fileindex_diff_tree_cb {
134 	got_fileindex_diff_tree_old_new_cb diff_old_new;
135 	got_fileindex_diff_tree_old_cb diff_old;
136 	got_fileindex_diff_tree_new_cb diff_new;
137 };
138 const struct got_error *got_fileindex_diff_tree(struct got_fileindex *,
139     struct got_tree_object *, const char *, const char *,
140     struct got_repository *, struct got_fileindex_diff_tree_cb *, void *);
141 
142 typedef const struct got_error *(*got_fileindex_diff_dir_old_new_cb)(void *,
143     struct got_fileindex_entry *, struct dirent *, const char *, int);
144 typedef const struct got_error *(*got_fileindex_diff_dir_old_cb)(void *,
145     struct got_fileindex_entry *, const char *);
146 typedef const struct got_error *(*got_fileindex_diff_dir_new_cb)(void *,
147     struct dirent *, const char *, int);
148 typedef const struct got_error *(*got_fileindex_diff_dir_traverse)(void *,
149     const char *, int);
150 struct got_fileindex_diff_dir_cb {
151 	got_fileindex_diff_dir_old_new_cb diff_old_new;
152 	got_fileindex_diff_dir_old_cb diff_old;
153 	got_fileindex_diff_dir_new_cb diff_new;
154 	got_fileindex_diff_dir_traverse diff_traverse;
155 };
156 const struct got_error *got_fileindex_diff_dir(struct got_fileindex *, int,
157     const char *, const char *, struct got_repository *,
158     struct got_fileindex_diff_dir_cb *, void *);
159 
160 int got_fileindex_entry_has_blob(struct got_fileindex_entry *);
161 int got_fileindex_entry_has_commit(struct got_fileindex_entry *);
162 int got_fileindex_entry_has_file_on_disk(struct got_fileindex_entry *);
163 uint32_t got_fileindex_entry_stage_get(const struct got_fileindex_entry *);
164 void got_fileindex_entry_stage_set(struct got_fileindex_entry *ie, uint32_t);
165 int got_fileindex_entry_filetype_get(struct got_fileindex_entry *);
166 const struct got_error *got_fileindex_entry_filetype_set(
167     struct got_fileindex_entry *, int);
168 
169 void got_fileindex_entry_mark_deleted_from_disk(struct got_fileindex_entry *);