1 #define USE_THE_REPOSITORY_VARIABLE
3 #include "git-compat-util.h"
4 #include "tmp-objdir.h"
6 #include "chdir-notify.h"
8 #include "environment.h"
9 #include "object-file.h"
11 #include "string-list.h"
15 #include "object-store-ll.h"
16 #include "repository.h"
21 struct object_directory
*prev_odb
;
26 * Allow only one tmp_objdir at a time in a running process, which simplifies
27 * our atexit cleanup routines. It's doubtful callers will ever need
28 * more than one, and we can expand later if so. You can have many such
29 * tmp_objdirs simultaneously in many processes, of course.
31 static struct tmp_objdir
*the_tmp_objdir
;
33 static void tmp_objdir_free(struct tmp_objdir
*t
)
35 strbuf_release(&t
->path
);
36 strvec_clear(&t
->env
);
40 int tmp_objdir_destroy(struct tmp_objdir
*t
)
47 if (t
== the_tmp_objdir
)
48 the_tmp_objdir
= NULL
;
51 restore_primary_odb(t
->prev_odb
, t
->path
.buf
);
53 err
= remove_dir_recursively(&t
->path
, 0);
60 static void remove_tmp_objdir(void)
62 tmp_objdir_destroy(the_tmp_objdir
);
65 void tmp_objdir_discard_objects(struct tmp_objdir
*t
)
67 remove_dir_recursively(&t
->path
, REMOVE_DIR_KEEP_TOPLEVEL
);
71 * These env_* functions are for setting up the child environment; the
72 * "replace" variant overrides the value of any existing variable with that
73 * "key". The "append" variant puts our new value at the end of a list,
74 * separated by PATH_SEP (which is what separate values in
75 * GIT_ALTERNATE_OBJECT_DIRECTORIES).
77 static void env_append(struct strvec
*env
, const char *key
, const char *val
)
79 struct strbuf quoted
= STRBUF_INIT
;
83 * Avoid quoting if it's not necessary, for maximum compatibility
84 * with older parsers which don't understand the quoting.
86 if (*val
== '"' || strchr(val
, PATH_SEP
)) {
87 strbuf_addch("ed
, '"');
88 quote_c_style(val
, "ed
, NULL
, 1);
89 strbuf_addch("ed
, '"');
95 strvec_pushf(env
, "%s=%s", key
, val
);
97 strvec_pushf(env
, "%s=%s%c%s", key
, old
, PATH_SEP
, val
);
99 strbuf_release("ed
);
102 static void env_replace(struct strvec
*env
, const char *key
, const char *val
)
104 strvec_pushf(env
, "%s=%s", key
, val
);
107 static int setup_tmp_objdir(const char *root
)
112 path
= xstrfmt("%s/pack", root
);
113 ret
= mkdir(path
, 0777);
119 struct tmp_objdir
*tmp_objdir_create(const char *prefix
)
121 static int installed_handlers
;
122 struct tmp_objdir
*t
;
125 BUG("only one tmp_objdir can be used at a time");
127 t
= xcalloc(1, sizeof(*t
));
128 strbuf_init(&t
->path
, 0);
129 strvec_init(&t
->env
);
132 * Use a string starting with tmp_ so that the builtin/prune.c code
133 * can recognize any stale objdirs left behind by a crash and delete
136 strbuf_addf(&t
->path
, "%s/tmp_objdir-%s-XXXXXX",
137 repo_get_object_directory(the_repository
), prefix
);
139 if (!mkdtemp(t
->path
.buf
)) {
140 /* free, not destroy, as we never touched the filesystem */
146 if (!installed_handlers
) {
147 atexit(remove_tmp_objdir
);
148 installed_handlers
++;
151 if (setup_tmp_objdir(t
->path
.buf
)) {
152 tmp_objdir_destroy(t
);
156 env_append(&t
->env
, ALTERNATE_DB_ENVIRONMENT
,
157 absolute_path(repo_get_object_directory(the_repository
)));
158 env_replace(&t
->env
, DB_ENVIRONMENT
, absolute_path(t
->path
.buf
));
159 env_replace(&t
->env
, GIT_QUARANTINE_ENVIRONMENT
,
160 absolute_path(t
->path
.buf
));
166 * Make sure we copy packfiles and their associated metafiles in the correct
167 * order. All of these ends_with checks are slightly expensive to do in
168 * the midst of a sorting routine, but in practice it shouldn't matter.
169 * We will have a relatively small number of packfiles to order, and loose
170 * objects exit early in the first line.
172 static int pack_copy_priority(const char *name
)
174 if (!starts_with(name
, "pack"))
176 if (ends_with(name
, ".keep"))
178 if (ends_with(name
, ".pack"))
180 if (ends_with(name
, ".rev"))
182 if (ends_with(name
, ".idx"))
187 static int pack_copy_cmp(const char *a
, const char *b
)
189 return pack_copy_priority(a
) - pack_copy_priority(b
);
192 static int read_dir_paths(struct string_list
*out
, const char *path
)
201 while ((de
= readdir(dh
)))
202 if (de
->d_name
[0] != '.')
203 string_list_append(out
, de
->d_name
);
209 static int migrate_paths(struct strbuf
*src
, struct strbuf
*dst
);
211 static int migrate_one(struct strbuf
*src
, struct strbuf
*dst
)
215 if (stat(src
->buf
, &st
) < 0)
217 if (S_ISDIR(st
.st_mode
)) {
218 if (!mkdir(dst
->buf
, 0777)) {
219 if (adjust_shared_perm(dst
->buf
))
221 } else if (errno
!= EEXIST
)
223 return migrate_paths(src
, dst
);
225 return finalize_object_file(src
->buf
, dst
->buf
);
228 static int migrate_paths(struct strbuf
*src
, struct strbuf
*dst
)
230 size_t src_len
= src
->len
, dst_len
= dst
->len
;
231 struct string_list paths
= STRING_LIST_INIT_DUP
;
235 if (read_dir_paths(&paths
, src
->buf
) < 0)
237 paths
.cmp
= pack_copy_cmp
;
238 string_list_sort(&paths
);
240 for (i
= 0; i
< paths
.nr
; i
++) {
241 const char *name
= paths
.items
[i
].string
;
243 strbuf_addf(src
, "/%s", name
);
244 strbuf_addf(dst
, "/%s", name
);
246 ret
|= migrate_one(src
, dst
);
248 strbuf_setlen(src
, src_len
);
249 strbuf_setlen(dst
, dst_len
);
252 string_list_clear(&paths
, 0);
256 int tmp_objdir_migrate(struct tmp_objdir
*t
)
258 struct strbuf src
= STRBUF_INIT
, dst
= STRBUF_INIT
;
265 if (the_repository
->objects
->odb
->will_destroy
)
266 BUG("migrating an ODB that was marked for destruction");
267 restore_primary_odb(t
->prev_odb
, t
->path
.buf
);
271 strbuf_addbuf(&src
, &t
->path
);
272 strbuf_addstr(&dst
, repo_get_object_directory(the_repository
));
274 ret
= migrate_paths(&src
, &dst
);
276 strbuf_release(&src
);
277 strbuf_release(&dst
);
279 tmp_objdir_destroy(t
);
283 const char **tmp_objdir_env(const struct tmp_objdir
*t
)
290 void tmp_objdir_add_as_alternate(const struct tmp_objdir
*t
)
292 add_to_alternates_memory(t
->path
.buf
);
295 void tmp_objdir_replace_primary_odb(struct tmp_objdir
*t
, int will_destroy
)
298 BUG("the primary object database is already replaced");
299 t
->prev_odb
= set_temporary_primary_odb(t
->path
.buf
, will_destroy
);
300 t
->will_destroy
= will_destroy
;
303 struct tmp_objdir
*tmp_objdir_unapply_primary_odb(void)
305 if (!the_tmp_objdir
|| !the_tmp_objdir
->prev_odb
)
308 restore_primary_odb(the_tmp_objdir
->prev_odb
, the_tmp_objdir
->path
.buf
);
309 the_tmp_objdir
->prev_odb
= NULL
;
310 return the_tmp_objdir
;
313 void tmp_objdir_reapply_primary_odb(struct tmp_objdir
*t
, const char *old_cwd
,
318 path
= reparent_relative_path(old_cwd
, new_cwd
, t
->path
.buf
);
319 strbuf_reset(&t
->path
);
320 strbuf_addstr(&t
->path
, path
);
322 tmp_objdir_replace_primary_odb(t
, t
->will_destroy
);