Clean up work-tree handling
[git/git.git] / builtin-init-db.c
1 /*
2 * GIT - The information manager from hell
3 *
4 * Copyright (C) Linus Torvalds, 2005
5 */
6 #include "cache.h"
7 #include "builtin.h"
8
9 #ifndef DEFAULT_GIT_TEMPLATE_DIR
10 #define DEFAULT_GIT_TEMPLATE_DIR "/usr/share/git-core/templates"
11 #endif
12
13 #ifdef NO_TRUSTABLE_FILEMODE
14 #define TEST_FILEMODE 0
15 #else
16 #define TEST_FILEMODE 1
17 #endif
18
19 static void safe_create_dir(const char *dir, int share)
20 {
21 if (mkdir(dir, 0777) < 0) {
22 if (errno != EEXIST) {
23 perror(dir);
24 exit(1);
25 }
26 }
27 else if (share && adjust_shared_perm(dir))
28 die("Could not make %s writable by group\n", dir);
29 }
30
31 static int copy_file(const char *dst, const char *src, int mode)
32 {
33 int fdi, fdo, status;
34
35 mode = (mode & 0111) ? 0777 : 0666;
36 if ((fdi = open(src, O_RDONLY)) < 0)
37 return fdi;
38 if ((fdo = open(dst, O_WRONLY | O_CREAT | O_EXCL, mode)) < 0) {
39 close(fdi);
40 return fdo;
41 }
42 status = copy_fd(fdi, fdo);
43 if (close(fdo) != 0)
44 return error("%s: write error: %s", dst, strerror(errno));
45
46 if (!status && adjust_shared_perm(dst))
47 return -1;
48
49 return status;
50 }
51
52 static void copy_templates_1(char *path, int baselen,
53 char *template, int template_baselen,
54 DIR *dir)
55 {
56 struct dirent *de;
57
58 /* Note: if ".git/hooks" file exists in the repository being
59 * re-initialized, /etc/core-git/templates/hooks/update would
60 * cause git-init to fail here. I think this is sane but
61 * it means that the set of templates we ship by default, along
62 * with the way the namespace under .git/ is organized, should
63 * be really carefully chosen.
64 */
65 safe_create_dir(path, 1);
66 while ((de = readdir(dir)) != NULL) {
67 struct stat st_git, st_template;
68 int namelen;
69 int exists = 0;
70
71 if (de->d_name[0] == '.')
72 continue;
73 namelen = strlen(de->d_name);
74 if ((PATH_MAX <= baselen + namelen) ||
75 (PATH_MAX <= template_baselen + namelen))
76 die("insanely long template name %s", de->d_name);
77 memcpy(path + baselen, de->d_name, namelen+1);
78 memcpy(template + template_baselen, de->d_name, namelen+1);
79 if (lstat(path, &st_git)) {
80 if (errno != ENOENT)
81 die("cannot stat %s", path);
82 }
83 else
84 exists = 1;
85
86 if (lstat(template, &st_template))
87 die("cannot stat template %s", template);
88
89 if (S_ISDIR(st_template.st_mode)) {
90 DIR *subdir = opendir(template);
91 int baselen_sub = baselen + namelen;
92 int template_baselen_sub = template_baselen + namelen;
93 if (!subdir)
94 die("cannot opendir %s", template);
95 path[baselen_sub++] =
96 template[template_baselen_sub++] = '/';
97 path[baselen_sub] =
98 template[template_baselen_sub] = 0;
99 copy_templates_1(path, baselen_sub,
100 template, template_baselen_sub,
101 subdir);
102 closedir(subdir);
103 }
104 else if (exists)
105 continue;
106 else if (S_ISLNK(st_template.st_mode)) {
107 char lnk[256];
108 int len;
109 len = readlink(template, lnk, sizeof(lnk));
110 if (len < 0)
111 die("cannot readlink %s", template);
112 if (sizeof(lnk) <= len)
113 die("insanely long symlink %s", template);
114 lnk[len] = 0;
115 if (symlink(lnk, path))
116 die("cannot symlink %s %s", lnk, path);
117 }
118 else if (S_ISREG(st_template.st_mode)) {
119 if (copy_file(path, template, st_template.st_mode))
120 die("cannot copy %s to %s", template, path);
121 }
122 else
123 error("ignoring template %s", template);
124 }
125 }
126
127 static void copy_templates(const char *git_dir, int len, const char *template_dir)
128 {
129 char path[PATH_MAX];
130 char template_path[PATH_MAX];
131 int template_len;
132 DIR *dir;
133
134 if (!template_dir) {
135 template_dir = getenv(TEMPLATE_DIR_ENVIRONMENT);
136 if (!template_dir)
137 template_dir = DEFAULT_GIT_TEMPLATE_DIR;
138 }
139 strcpy(template_path, template_dir);
140 template_len = strlen(template_path);
141 if (template_path[template_len-1] != '/') {
142 template_path[template_len++] = '/';
143 template_path[template_len] = 0;
144 }
145 dir = opendir(template_path);
146 if (!dir) {
147 fprintf(stderr, "warning: templates not found %s\n",
148 template_dir);
149 return;
150 }
151
152 /* Make sure that template is from the correct vintage */
153 strcpy(template_path + template_len, "config");
154 repository_format_version = 0;
155 git_config_from_file(check_repository_format_version,
156 template_path);
157 template_path[template_len] = 0;
158
159 if (repository_format_version &&
160 repository_format_version != GIT_REPO_VERSION) {
161 fprintf(stderr, "warning: not copying templates of "
162 "a wrong format version %d from '%s'\n",
163 repository_format_version,
164 template_dir);
165 closedir(dir);
166 return;
167 }
168
169 memcpy(path, git_dir, len);
170 path[len] = 0;
171 copy_templates_1(path, len,
172 template_path, template_len,
173 dir);
174 closedir(dir);
175 }
176
177 static int create_default_files(const char *git_dir, const char *template_path)
178 {
179 unsigned len = strlen(git_dir);
180 static char path[PATH_MAX];
181 unsigned char sha1[20];
182 struct stat st1;
183 char repo_version_string[10];
184 int reinit;
185 int filemode;
186
187 if (len > sizeof(path)-50)
188 die("insane git directory %s", git_dir);
189 memcpy(path, git_dir, len);
190
191 if (len && path[len-1] != '/')
192 path[len++] = '/';
193
194 /*
195 * Create .git/refs/{heads,tags}
196 */
197 strcpy(path + len, "refs");
198 safe_create_dir(path, 1);
199 strcpy(path + len, "refs/heads");
200 safe_create_dir(path, 1);
201 strcpy(path + len, "refs/tags");
202 safe_create_dir(path, 1);
203
204 /* First copy the templates -- we might have the default
205 * config file there, in which case we would want to read
206 * from it after installing.
207 */
208 path[len] = 0;
209 copy_templates(path, len, template_path);
210
211 git_config(git_default_config);
212
213 /*
214 * We would have created the above under user's umask -- under
215 * shared-repository settings, we would need to fix them up.
216 */
217 if (shared_repository) {
218 path[len] = 0;
219 adjust_shared_perm(path);
220 strcpy(path + len, "refs");
221 adjust_shared_perm(path);
222 strcpy(path + len, "refs/heads");
223 adjust_shared_perm(path);
224 strcpy(path + len, "refs/tags");
225 adjust_shared_perm(path);
226 }
227
228 /*
229 * Create the default symlink from ".git/HEAD" to the "master"
230 * branch, if it does not exist yet.
231 */
232 strcpy(path + len, "HEAD");
233 reinit = !read_ref("HEAD", sha1);
234 if (!reinit) {
235 if (create_symref("HEAD", "refs/heads/master", NULL) < 0)
236 exit(1);
237 }
238
239 /* This forces creation of new config file */
240 sprintf(repo_version_string, "%d", GIT_REPO_VERSION);
241 git_config_set("core.repositoryformatversion", repo_version_string);
242
243 path[len] = 0;
244 strcpy(path + len, "config");
245
246 /* Check filemode trustability */
247 filemode = TEST_FILEMODE;
248 if (TEST_FILEMODE && !lstat(path, &st1)) {
249 struct stat st2;
250 filemode = (!chmod(path, st1.st_mode ^ S_IXUSR) &&
251 !lstat(path, &st2) &&
252 st1.st_mode != st2.st_mode);
253 }
254 git_config_set("core.filemode", filemode ? "true" : "false");
255
256 if (is_bare_repository())
257 git_config_set("core.bare", "true");
258 else {
259 const char *work_tree = get_git_work_tree();
260 git_config_set("core.bare", "false");
261 /* allow template config file to override the default */
262 if (log_all_ref_updates == -1)
263 git_config_set("core.logallrefupdates", "true");
264 if (work_tree != git_work_tree_cfg)
265 git_config_set("core.worktree", work_tree);
266 }
267 return reinit;
268 }
269
270 static const char init_db_usage[] =
271 "git-init [-q | --quiet] [--template=<template-directory>] [--shared]";
272
273 /*
274 * If you want to, you can share the DB area with any number of branches.
275 * That has advantages: you can save space by sharing all the SHA1 objects.
276 * On the other hand, it might just make lookup slower and messier. You
277 * be the judge. The default case is to have one DB per managed directory.
278 */
279 int cmd_init_db(int argc, const char **argv, const char *prefix)
280 {
281 const char *git_dir;
282 const char *sha1_dir;
283 const char *template_dir = NULL;
284 char *path;
285 int len, i, reinit;
286 int quiet = 0;
287
288 for (i = 1; i < argc; i++, argv++) {
289 const char *arg = argv[1];
290 if (!prefixcmp(arg, "--template="))
291 template_dir = arg+11;
292 else if (!strcmp(arg, "--shared"))
293 shared_repository = PERM_GROUP;
294 else if (!prefixcmp(arg, "--shared="))
295 shared_repository = git_config_perm("arg", arg+9);
296 else if (!strcmp(arg, "-q") || !strcmp(arg, "--quiet"))
297 quiet = 1;
298 else
299 usage(init_db_usage);
300 }
301
302 git_work_tree_cfg = xcalloc(PATH_MAX, 1);
303 if (!getcwd(git_work_tree_cfg, PATH_MAX))
304 die ("Cannot access current working directory.");
305 if (access(get_git_work_tree(), X_OK))
306 die ("Cannot access work tree '%s'", get_git_work_tree());
307
308 /*
309 * Set up the default .git directory contents
310 */
311 git_dir = getenv(GIT_DIR_ENVIRONMENT);
312 if (!git_dir)
313 git_dir = DEFAULT_GIT_DIR_ENVIRONMENT;
314 safe_create_dir(git_dir, 0);
315
316 /* Check to see if the repository version is right.
317 * Note that a newly created repository does not have
318 * config file, so this will not fail. What we are catching
319 * is an attempt to reinitialize new repository with an old tool.
320 */
321 check_repository_format();
322
323 reinit = create_default_files(git_dir, template_dir);
324
325 /*
326 * And set up the object store.
327 */
328 sha1_dir = get_object_directory();
329 len = strlen(sha1_dir);
330 path = xmalloc(len + 40);
331 memcpy(path, sha1_dir, len);
332
333 safe_create_dir(sha1_dir, 1);
334 strcpy(path+len, "/pack");
335 safe_create_dir(path, 1);
336 strcpy(path+len, "/info");
337 safe_create_dir(path, 1);
338
339 if (shared_repository) {
340 char buf[10];
341 /* We do not spell "group" and such, so that
342 * the configuration can be read by older version
343 * of git.
344 */
345 sprintf(buf, "%d", shared_repository);
346 git_config_set("core.sharedrepository", buf);
347 git_config_set("receive.denyNonFastforwards", "true");
348 }
349
350 if (!quiet)
351 printf("%s%s Git repository in %s/\n",
352 reinit ? "Reinitialized existing" : "Initialized empty",
353 shared_repository ? " shared" : "",
354 git_dir);
355
356 return 0;
357 }