fast-import: fix unalinged allocation and access
[git/git.git] / config.c
CommitLineData
10bea152
JS
1/*
2 * GIT - The information manager from hell
3 *
4 * Copyright (C) Linus Torvalds, 2005
5 * Copyright (C) Johannes Schindelin, 2005
6 *
7 */
17712991 8#include "cache.h"
7f0e39fa 9#include "exec_cmd.h"
17712991
LT
10
11#define MAXNAME (256)
12
13static FILE *config_file;
0dccc7dc 14static const char *config_file_name;
17712991 15static int config_linenr;
960ccca6
DH
16static int zlib_compression_seen;
17
17712991
LT
18static int get_next_char(void)
19{
20 int c;
21 FILE *f;
22
23 c = '\n';
24 if ((f = config_file) != NULL) {
25 c = fgetc(f);
db2c075d
JH
26 if (c == '\r') {
27 /* DOS like systems */
28 c = fgetc(f);
29 if (c != '\n') {
30 ungetc(c, f);
31 c = '\r';
32 }
33 }
17712991
LT
34 if (c == '\n')
35 config_linenr++;
36 if (c == EOF) {
37 config_file = NULL;
38 c = '\n';
39 }
40 }
41 return c;
42}
43
44static char *parse_value(void)
45{
46 static char value[1024];
47 int quote = 0, comment = 0, len = 0, space = 0;
48
49 for (;;) {
50 int c = get_next_char();
51 if (len >= sizeof(value))
52 return NULL;
53 if (c == '\n') {
54 if (quote)
55 return NULL;
56 value[len] = 0;
57 return value;
58 }
59 if (comment)
60 continue;
61 if (isspace(c) && !quote) {
62 space = 1;
63 continue;
64 }
7ebdba61
JS
65 if (!quote) {
66 if (c == ';' || c == '#') {
67 comment = 1;
68 continue;
69 }
70 }
17712991
LT
71 if (space) {
72 if (len)
73 value[len++] = ' ';
74 space = 0;
75 }
76 if (c == '\\') {
77 c = get_next_char();
78 switch (c) {
79 case '\n':
80 continue;
81 case 't':
82 c = '\t';
83 break;
84 case 'b':
85 c = '\b';
86 break;
87 case 'n':
88 c = '\n';
89 break;
5cbb401d
LT
90 /* Some characters escape as themselves */
91 case '\\': case '"':
92 break;
93 /* Reject unknown escape sequences */
94 default:
95 return NULL;
17712991
LT
96 }
97 value[len++] = c;
98 continue;
99 }
100 if (c == '"') {
101 quote = 1-quote;
102 continue;
103 }
17712991
LT
104 value[len++] = c;
105 }
106}
107
38c5afa8
LT
108static inline int iskeychar(int c)
109{
110 return isalnum(c) || c == '-';
111}
112
17712991
LT
113static int get_value(config_fn_t fn, char *name, unsigned int len)
114{
115 int c;
116 char *value;
117
118 /* Get the full name */
119 for (;;) {
120 c = get_next_char();
121 if (c == EOF)
122 break;
38c5afa8 123 if (!iskeychar(c))
17712991
LT
124 break;
125 name[len++] = tolower(c);
126 if (len >= MAXNAME)
127 return -1;
128 }
129 name[len] = 0;
130 while (c == ' ' || c == '\t')
131 c = get_next_char();
132
133 value = NULL;
134 if (c != '\n') {
135 if (c != '=')
136 return -1;
137 value = parse_value();
138 if (!value)
139 return -1;
140 }
141 return fn(name, value);
142}
143
d14f7764
LT
144static int get_extended_base_var(char *name, int baselen, int c)
145{
146 do {
147 if (c == '\n')
148 return -1;
149 c = get_next_char();
150 } while (isspace(c));
151
152 /* We require the format to be '[base "extension"]' */
153 if (c != '"')
154 return -1;
155 name[baselen++] = '.';
156
157 for (;;) {
158 int c = get_next_char();
159 if (c == '\n')
160 return -1;
161 if (c == '"')
162 break;
163 if (c == '\\') {
164 c = get_next_char();
165 if (c == '\n')
166 return -1;
167 }
168 name[baselen++] = c;
169 if (baselen > MAXNAME / 2)
170 return -1;
171 }
172
173 /* Final ']' */
174 if (get_next_char() != ']')
175 return -1;
176 return baselen;
177}
178
17712991
LT
179static int get_base_var(char *name)
180{
181 int baselen = 0;
182
183 for (;;) {
184 int c = get_next_char();
185 if (c == EOF)
186 return -1;
187 if (c == ']')
188 return baselen;
d14f7764
LT
189 if (isspace(c))
190 return get_extended_base_var(name, baselen, c);
38c5afa8 191 if (!iskeychar(c) && c != '.')
17712991
LT
192 return -1;
193 if (baselen > MAXNAME / 2)
194 return -1;
195 name[baselen++] = tolower(c);
196 }
197}
198
199static int git_parse_file(config_fn_t fn)
200{
201 int comment = 0;
202 int baselen = 0;
203 static char var[MAXNAME];
204
205 for (;;) {
206 int c = get_next_char();
207 if (c == '\n') {
208 /* EOF? */
209 if (!config_file)
210 return 0;
211 comment = 0;
212 continue;
213 }
214 if (comment || isspace(c))
215 continue;
216 if (c == '#' || c == ';') {
217 comment = 1;
218 continue;
219 }
220 if (c == '[') {
221 baselen = get_base_var(var);
222 if (baselen <= 0)
223 break;
224 var[baselen++] = '.';
225 var[baselen] = 0;
226 continue;
227 }
228 if (!isalpha(c))
229 break;
128af9d1 230 var[baselen] = tolower(c);
17712991
LT
231 if (get_value(fn, var, baselen+1) < 0)
232 break;
233 }
4f629539 234 die("bad config file line %d in %s", config_linenr, config_file_name);
17712991
LT
235}
236
0b87b6e0
BD
237static unsigned long get_unit_factor(const char *end)
238{
239 if (!*end)
240 return 1;
241 else if (!strcasecmp(end, "k"))
242 return 1024;
243 else if (!strcasecmp(end, "m"))
244 return 1024 * 1024;
245 else if (!strcasecmp(end, "g"))
246 return 1024 * 1024 * 1024;
247 die("unknown unit: '%s'", end);
248}
249
250int git_parse_long(const char *value, long *ret)
251{
252 if (value && *value) {
253 char *end;
254 long val = strtol(value, &end, 0);
255 *ret = val * get_unit_factor(end);
256 return 1;
257 }
258 return 0;
259}
260
261int git_parse_ulong(const char *value, unsigned long *ret)
17712991
LT
262{
263 if (value && *value) {
264 char *end;
0b87b6e0
BD
265 unsigned long val = strtoul(value, &end, 0);
266 *ret = val * get_unit_factor(end);
267 return 1;
17712991 268 }
0b87b6e0
BD
269 return 0;
270}
271
272int git_config_int(const char *name, const char *value)
273{
274 long ret;
275 if (!git_parse_long(value, &ret))
276 die("bad config value for '%s' in %s", name, config_file_name);
277 return ret;
278}
279
280unsigned long git_config_ulong(const char *name, const char *value)
281{
282 unsigned long ret;
283 if (!git_parse_ulong(value, &ret))
284 die("bad config value for '%s' in %s", name, config_file_name);
285 return ret;
17712991
LT
286}
287
288int git_config_bool(const char *name, const char *value)
289{
290 if (!value)
291 return 1;
292 if (!*value)
293 return 0;
624314fd 294 if (!strcasecmp(value, "true") || !strcasecmp(value, "yes"))
17712991 295 return 1;
624314fd 296 if (!strcasecmp(value, "false") || !strcasecmp(value, "no"))
17712991
LT
297 return 0;
298 return git_config_int(name, value) != 0;
299}
300
301int git_default_config(const char *var, const char *value)
302{
303 /* This needs a better name */
304 if (!strcmp(var, "core.filemode")) {
305 trust_executable_bit = git_config_bool(var, value);
306 return 0;
307 }
308
9378c161
JH
309 if (!strcmp(var, "core.quotepath")) {
310 quote_path_fully = git_config_bool(var, value);
311 return 0;
312 }
313
78a8d641
JS
314 if (!strcmp(var, "core.symlinks")) {
315 has_symlinks = git_config_bool(var, value);
316 return 0;
317 }
318
7d1864ce
JH
319 if (!strcmp(var, "core.bare")) {
320 is_bare_repository_cfg = git_config_bool(var, value);
321 return 0;
322 }
323
5f73076c
JH
324 if (!strcmp(var, "core.ignorestat")) {
325 assume_unchanged = git_config_bool(var, value);
326 return 0;
327 }
328
e388c738
JH
329 if (!strcmp(var, "core.prefersymlinkrefs")) {
330 prefer_symlink_refs = git_config_bool(var, value);
f8348be3
JS
331 return 0;
332 }
333
6de08ae6
SP
334 if (!strcmp(var, "core.logallrefupdates")) {
335 log_all_ref_updates = git_config_bool(var, value);
336 return 0;
337 }
338
2f8acdb3
JH
339 if (!strcmp(var, "core.warnambiguousrefs")) {
340 warn_ambiguous_refs = git_config_bool(var, value);
341 return 0;
342 }
343
960ccca6 344 if (!strcmp(var, "core.loosecompression")) {
12f6c308
JBH
345 int level = git_config_int(var, value);
346 if (level == -1)
347 level = Z_DEFAULT_COMPRESSION;
348 else if (level < 0 || level > Z_BEST_COMPRESSION)
349 die("bad zlib compression level %d", level);
350 zlib_compression_level = level;
960ccca6
DH
351 zlib_compression_seen = 1;
352 return 0;
353 }
354
355 if (!strcmp(var, "core.compression")) {
356 int level = git_config_int(var, value);
357 if (level == -1)
358 level = Z_DEFAULT_COMPRESSION;
359 else if (level < 0 || level > Z_BEST_COMPRESSION)
360 die("bad zlib compression level %d", level);
361 core_compression_level = level;
362 core_compression_seen = 1;
363 if (!zlib_compression_seen)
364 zlib_compression_level = level;
12f6c308
JBH
365 return 0;
366 }
367
60bb8b14 368 if (!strcmp(var, "core.packedgitwindowsize")) {
5faaf246 369 int pgsz_x2 = getpagesize() * 2;
60bb8b14 370 packed_git_window_size = git_config_int(var, value);
5faaf246
JH
371
372 /* This value must be multiple of (pagesize * 2) */
373 packed_git_window_size /= pgsz_x2;
374 if (packed_git_window_size < 1)
375 packed_git_window_size = 1;
376 packed_git_window_size *= pgsz_x2;
60bb8b14
SP
377 return 0;
378 }
379
77ccc5bb
SP
380 if (!strcmp(var, "core.packedgitlimit")) {
381 packed_git_limit = git_config_int(var, value);
382 return 0;
383 }
384
18bdec11
SP
385 if (!strcmp(var, "core.deltabasecachelimit")) {
386 delta_base_cache_limit = git_config_int(var, value);
387 return 0;
388 }
389
6c510bee 390 if (!strcmp(var, "core.autocrlf")) {
d7f46334
LT
391 if (value && !strcasecmp(value, "input")) {
392 auto_crlf = -1;
393 return 0;
394 }
6c510bee
LT
395 auto_crlf = git_config_bool(var, value);
396 return 0;
397 }
398
e1b10391 399 if (!strcmp(var, "user.name")) {
817151e6 400 strlcpy(git_default_name, value, sizeof(git_default_name));
e1b10391
LT
401 return 0;
402 }
403
404 if (!strcmp(var, "user.email")) {
817151e6 405 strlcpy(git_default_email, value, sizeof(git_default_email));
e1b10391
LT
406 return 0;
407 }
408
4e72dcec 409 if (!strcmp(var, "i18n.commitencoding")) {
dbb2b41a 410 git_commit_encoding = xstrdup(value);
4e72dcec
JH
411 return 0;
412 }
413
d2c11a38 414 if (!strcmp(var, "i18n.logoutputencoding")) {
dbb2b41a 415 git_log_output_encoding = xstrdup(value);
d2c11a38
JH
416 return 0;
417 }
418
419
a159ca0c 420 if (!strcmp(var, "pager.color") || !strcmp(var, "color.pager")) {
aa086eb8
ML
421 pager_use_color = git_config_bool(var,value);
422 return 0;
423 }
424
54adf370
BG
425 if (!strcmp(var, "core.pager")) {
426 pager_program = xstrdup(value);
427 return 0;
428 }
429
4d87b9c5
JS
430 if (!strcmp(var, "core.editor")) {
431 editor_program = xstrdup(value);
432 return 0;
433 }
434
039bc64e
JH
435 if (!strcmp(var, "core.excludesfile")) {
436 if (!value)
437 die("core.excludesfile without value");
438 excludes_file = xstrdup(value);
439 return 0;
440 }
441
a9cc857a 442 if (!strcmp(var, "core.whitespace")) {
cf1b7869 443 whitespace_rule_cfg = parse_whitespace_rule(value);
a9cc857a
JH
444 return 0;
445 }
446
1ab661dd 447 /* Add other config variables here and to Documentation/config.txt. */
17712991
LT
448 return 0;
449}
450
4f629539 451int git_config_from_file(config_fn_t fn, const char *filename)
17712991
LT
452{
453 int ret;
4f629539 454 FILE *f = fopen(filename, "r");
17712991
LT
455
456 ret = -1;
457 if (f) {
458 config_file = f;
4f629539 459 config_file_name = filename;
17712991
LT
460 config_linenr = 1;
461 ret = git_parse_file(fn);
462 fclose(f);
4f629539 463 config_file_name = NULL;
17712991
LT
464 }
465 return ret;
466}
10bea152 467
506b17b1
JS
468const char *git_etc_gitconfig(void)
469{
7f0e39fa
JS
470 static const char *system_wide;
471 if (!system_wide) {
472 system_wide = ETC_GITCONFIG;
473 if (!is_absolute_path(system_wide)) {
474 /* interpret path relative to exec-dir */
475 const char *exec_path = git_exec_path();
476 system_wide = prefix_path(exec_path, strlen(exec_path),
477 system_wide);
478 }
479 }
480 return system_wide;
506b17b1
JS
481}
482
4f629539
JH
483int git_config(config_fn_t fn)
484{
5f1a63e0
JS
485 int ret = 0;
486 char *repo_config = NULL;
487 const char *home = NULL, *filename;
488
489 /* $GIT_CONFIG makes git read _only_ the given config file,
490 * $GIT_CONFIG_LOCAL will make it process it in addition to the
491 * global config file, the same way it would the per-repository
492 * config file otherwise. */
d4ebc36c 493 filename = getenv(CONFIG_ENVIRONMENT);
5f1a63e0 494 if (!filename) {
506b17b1
JS
495 if (!access(git_etc_gitconfig(), R_OK))
496 ret += git_config_from_file(fn, git_etc_gitconfig());
5f1a63e0 497 home = getenv("HOME");
d4ebc36c 498 filename = getenv(CONFIG_LOCAL_ENVIRONMENT);
5f1a63e0 499 if (!filename)
9befac47 500 filename = repo_config = xstrdup(git_path("config"));
7f29f7a9 501 }
5f1a63e0
JS
502
503 if (home) {
9befac47 504 char *user_config = xstrdup(mkpath("%s/.gitconfig", home));
e33d0611 505 if (!access(user_config, R_OK))
5f1a63e0
JS
506 ret = git_config_from_file(fn, user_config);
507 free(user_config);
508 }
509
510 ret += git_config_from_file(fn, filename);
4cac42b1 511 free(repo_config);
5f1a63e0 512 return ret;
4f629539
JH
513}
514
10bea152
JS
515/*
516 * Find all the stuff for git_config_set() below.
517 */
4ddba79d
JS
518
519#define MAX_MATCHES 512
520
10bea152
JS
521static struct {
522 int baselen;
523 char* key;
f98d863d 524 int do_not_match;
10bea152 525 regex_t* value_regex;
4ddba79d 526 int multi_replace;
dc49cd76 527 size_t offset[MAX_MATCHES];
10bea152
JS
528 enum { START, SECTION_SEEN, SECTION_END_SEEN, KEY_SEEN } state;
529 int seen;
530} store;
531
f98d863d
JS
532static int matches(const char* key, const char* value)
533{
534 return !strcmp(key, store.key) &&
535 (store.value_regex == NULL ||
536 (store.do_not_match ^
537 !regexec(store.value_regex, value, 0, NULL, 0)));
538}
539
10bea152
JS
540static int store_aux(const char* key, const char* value)
541{
ae9ee41d
JH
542 const char *ep;
543 size_t section_len;
544
10bea152
JS
545 switch (store.state) {
546 case KEY_SEEN:
f98d863d 547 if (matches(key, value)) {
4ddba79d 548 if (store.seen == 1 && store.multi_replace == 0) {
10bea152
JS
549 fprintf(stderr,
550 "Warning: %s has multiple values\n",
551 key);
4ddba79d
JS
552 } else if (store.seen >= MAX_MATCHES) {
553 fprintf(stderr, "Too many matches\n");
554 return 1;
10bea152 555 }
4ddba79d
JS
556
557 store.offset[store.seen] = ftell(config_file);
10bea152
JS
558 store.seen++;
559 }
560 break;
561 case SECTION_SEEN:
ae9ee41d
JH
562 /*
563 * What we are looking for is in store.key (both
564 * section and var), and its section part is baselen
565 * long. We found key (again, both section and var).
566 * We would want to know if this key is in the same
567 * section as what we are looking for. We already
568 * know we are in the same section as what should
569 * hold store.key.
570 */
571 ep = strrchr(key, '.');
572 section_len = ep - key;
573
574 if ((section_len != store.baselen) ||
575 memcmp(key, store.key, section_len+1)) {
10bea152
JS
576 store.state = SECTION_END_SEEN;
577 break;
ae9ee41d
JH
578 }
579
580 /*
581 * Do not increment matches: this is no match, but we
582 * just made sure we are in the desired section.
583 */
584 store.offset[store.seen] = ftell(config_file);
10bea152
JS
585 /* fallthru */
586 case SECTION_END_SEEN:
587 case START:
f98d863d 588 if (matches(key, value)) {
4ddba79d 589 store.offset[store.seen] = ftell(config_file);
10bea152
JS
590 store.state = KEY_SEEN;
591 store.seen++;
d14f7764
LT
592 } else {
593 if (strrchr(key, '.') - key == store.baselen &&
bdf0ef08 594 !strncmp(key, store.key, store.baselen)) {
93ddef3e 595 store.state = SECTION_SEEN;
bdf0ef08 596 store.offset[store.seen] = ftell(config_file);
d14f7764 597 }
bdf0ef08 598 }
10bea152
JS
599 }
600 return 0;
601}
602
b79d18c9 603static int write_error(void)
480c9e52
AW
604{
605 fprintf(stderr, "Failed to write new configuration file\n");
606
607 /* Same error code as "failed to rename". */
608 return 4;
609}
610
611static int store_write_section(int fd, const char* key)
10bea152 612{
d14f7764
LT
613 const char *dot = strchr(key, '.');
614 int len1 = store.baselen, len2 = -1;
615
616 dot = strchr(key, '.');
617 if (dot) {
618 int dotlen = dot - key;
619 if (dotlen < len1) {
620 len2 = len1 - dotlen - 1;
621 len1 = dotlen;
622 }
623 }
624
480c9e52
AW
625 if (write_in_full(fd, "[", 1) != 1 ||
626 write_in_full(fd, key, len1) != len1)
627 return 0;
d14f7764 628 if (len2 >= 0) {
480c9e52
AW
629 if (write_in_full(fd, " \"", 2) != 2)
630 return 0;
d14f7764
LT
631 while (--len2 >= 0) {
632 unsigned char c = *++dot;
633 if (c == '"')
480c9e52
AW
634 if (write_in_full(fd, "\\", 1) != 1)
635 return 0;
636 if (write_in_full(fd, &c, 1) != 1)
637 return 0;
d14f7764 638 }
480c9e52
AW
639 if (write_in_full(fd, "\"", 1) != 1)
640 return 0;
d14f7764 641 }
480c9e52
AW
642 if (write_in_full(fd, "]\n", 2) != 2)
643 return 0;
644
645 return 1;
10bea152
JS
646}
647
480c9e52 648static int store_write_pair(int fd, const char* key, const char* value)
10bea152
JS
649{
650 int i;
480c9e52 651 int length = strlen(key+store.baselen+1);
cdd4fb15
BG
652 int quote = 0;
653
6281f394
JM
654 /*
655 * Check to see if the value needs to be surrounded with a dq pair.
656 * Note that problematic characters are always backslash-quoted; this
657 * check is about not losing leading or trailing SP and strings that
658 * follow beginning-of-comment characters (i.e. ';' and '#') by the
659 * configuration parser.
660 */
cdd4fb15
BG
661 if (value[0] == ' ')
662 quote = 1;
663 for (i = 0; value[i]; i++)
664 if (value[i] == ';' || value[i] == '#')
665 quote = 1;
6281f394 666 if (i && value[i-1] == ' ')
cdd4fb15 667 quote = 1;
10bea152 668
480c9e52
AW
669 if (write_in_full(fd, "\t", 1) != 1 ||
670 write_in_full(fd, key+store.baselen+1, length) != length ||
671 write_in_full(fd, " = ", 3) != 3)
672 return 0;
cdd4fb15
BG
673 if (quote && write_in_full(fd, "\"", 1) != 1)
674 return 0;
10bea152
JS
675 for (i = 0; value[i]; i++)
676 switch (value[i]) {
480c9e52
AW
677 case '\n':
678 if (write_in_full(fd, "\\n", 2) != 2)
679 return 0;
680 break;
681 case '\t':
682 if (write_in_full(fd, "\\t", 2) != 2)
683 return 0;
684 break;
685 case '"':
686 case '\\':
687 if (write_in_full(fd, "\\", 1) != 1)
688 return 0;
689 default:
690 if (write_in_full(fd, value+i, 1) != 1)
691 return 0;
692 break;
693 }
cdd4fb15
BG
694 if (quote && write_in_full(fd, "\"", 1) != 1)
695 return 0;
480c9e52
AW
696 if (write_in_full(fd, "\n", 1) != 1)
697 return 0;
698 return 1;
10bea152
JS
699}
700
dc49cd76
SP
701static ssize_t find_beginning_of_line(const char* contents, size_t size,
702 size_t offset_, int* found_bracket)
4ddba79d 703{
dc49cd76
SP
704 size_t equal_offset = size, bracket_offset = size;
705 ssize_t offset;
4ddba79d 706
a6080a0a 707 for (offset = offset_-2; offset > 0
4ddba79d
JS
708 && contents[offset] != '\n'; offset--)
709 switch (contents[offset]) {
710 case '=': equal_offset = offset; break;
711 case ']': bracket_offset = offset; break;
712 }
713 if (bracket_offset < equal_offset) {
714 *found_bracket = 1;
715 offset = bracket_offset+1;
716 } else
717 offset++;
718
719 return offset;
720}
721
10bea152
JS
722int git_config_set(const char* key, const char* value)
723{
4ddba79d 724 return git_config_set_multivar(key, value, NULL, 0);
10bea152
JS
725}
726
727/*
728 * If value==NULL, unset in (remove from) config,
729 * if value_regex!=NULL, disregard key/value pairs where value does not match.
4ddba79d
JS
730 * if multi_replace==0, nothing, or only one matching key/value is replaced,
731 * else all matching key/values (regardless how many) are removed,
732 * before the new pair is written.
10bea152
JS
733 *
734 * Returns 0 on success.
735 *
736 * This function does this:
737 *
738 * - it locks the config file by creating ".git/config.lock"
739 *
740 * - it then parses the config using store_aux() as validator to find
741 * the position on the key/value pair to replace. If it is to be unset,
742 * it must be found exactly once.
743 *
744 * - the config file is mmap()ed and the part before the match (if any) is
745 * written to the lock file, then the changed part and the rest.
746 *
747 * - the config file is removed and the lock file rename()d to it.
748 *
749 */
750int git_config_set_multivar(const char* key, const char* value,
4ddba79d 751 const char* value_regex, int multi_replace)
10bea152 752{
d14f7764 753 int i, dot;
f8ba655e 754 int fd = -1, in_fd;
dafc88b1 755 int ret;
9c3796fc 756 char* config_filename;
6cbf973c 757 struct lock_file *lock = NULL;
b17e659d 758 const char* last_dot = strrchr(key, '.');
4ddba79d 759
d4ebc36c 760 config_filename = getenv(CONFIG_ENVIRONMENT);
9c3796fc 761 if (!config_filename) {
d4ebc36c 762 config_filename = getenv(CONFIG_LOCAL_ENVIRONMENT);
9c3796fc
JS
763 if (!config_filename)
764 config_filename = git_path("config");
765 }
9befac47 766 config_filename = xstrdup(config_filename);
9c3796fc 767
10bea152
JS
768 /*
769 * Since "key" actually contains the section name and the real
770 * key name separated by a dot, we have to know where the dot is.
771 */
b17e659d 772
dafc88b1 773 if (last_dot == NULL) {
10bea152 774 fprintf(stderr, "key does not contain a section: %s\n", key);
dafc88b1
SH
775 ret = 2;
776 goto out_free;
10bea152 777 }
b17e659d
JS
778 store.baselen = last_dot - key;
779
780 store.multi_replace = multi_replace;
10bea152
JS
781
782 /*
783 * Validate the key and while at it, lower case it for matching.
784 */
2d7320d0 785 store.key = xmalloc(strlen(key) + 1);
d14f7764
LT
786 dot = 0;
787 for (i = 0; key[i]; i++) {
788 unsigned char c = key[i];
789 if (c == '.')
790 dot = 1;
791 /* Leave the extended basename untouched.. */
792 if (!dot || i > store.baselen) {
38c5afa8 793 if (!iskeychar(c) || (i == store.baselen+1 && !isalpha(c))) {
d14f7764
LT
794 fprintf(stderr, "invalid key: %s\n", key);
795 free(store.key);
796 ret = 1;
797 goto out_free;
798 }
799 c = tolower(c);
6f71686e
JS
800 } else if (c == '\n') {
801 fprintf(stderr, "invalid key (newline): %s\n", key);
802 free(store.key);
803 ret = 1;
804 goto out_free;
d14f7764
LT
805 }
806 store.key[i] = c;
807 }
3dd94e3b 808 store.key[i] = 0;
10bea152
JS
809
810 /*
6cbf973c 811 * The lock serves a purpose in addition to locking: the new
10bea152
JS
812 * contents of .git/config will be written into it.
813 */
6cbf973c
BS
814 lock = xcalloc(sizeof(struct lock_file), 1);
815 fd = hold_lock_file_for_update(lock, config_filename, 0);
816 if (fd < 0) {
10bea152
JS
817 fprintf(stderr, "could not lock config file\n");
818 free(store.key);
dafc88b1
SH
819 ret = -1;
820 goto out_free;
10bea152
JS
821 }
822
823 /*
824 * If .git/config does not exist yet, write a minimal version.
825 */
88fb958b
AR
826 in_fd = open(config_filename, O_RDONLY);
827 if ( in_fd < 0 ) {
10bea152
JS
828 free(store.key);
829
88fb958b
AR
830 if ( ENOENT != errno ) {
831 error("opening %s: %s", config_filename,
832 strerror(errno));
dafc88b1
SH
833 ret = 3; /* same as "invalid config file" */
834 goto out_free;
88fb958b 835 }
10bea152
JS
836 /* if nothing to unset, error out */
837 if (value == NULL) {
dafc88b1
SH
838 ret = 5;
839 goto out_free;
10bea152
JS
840 }
841
842 store.key = (char*)key;
480c9e52 843 if (!store_write_section(fd, key) ||
93c1e079
JH
844 !store_write_pair(fd, key, value))
845 goto write_err_out;
846 } else {
88fb958b 847 struct stat st;
10bea152 848 char* contents;
dc49cd76
SP
849 size_t contents_sz, copy_begin, copy_end;
850 int i, new_line = 0;
10bea152
JS
851
852 if (value_regex == NULL)
853 store.value_regex = NULL;
854 else {
f98d863d
JS
855 if (value_regex[0] == '!') {
856 store.do_not_match = 1;
857 value_regex++;
858 } else
859 store.do_not_match = 0;
860
2d7320d0 861 store.value_regex = (regex_t*)xmalloc(sizeof(regex_t));
10bea152
JS
862 if (regcomp(store.value_regex, value_regex,
863 REG_EXTENDED)) {
7246ed43 864 fprintf(stderr, "Invalid pattern: %s\n",
10bea152
JS
865 value_regex);
866 free(store.value_regex);
dafc88b1
SH
867 ret = 6;
868 goto out_free;
10bea152
JS
869 }
870 }
871
4ddba79d 872 store.offset[0] = 0;
10bea152
JS
873 store.state = START;
874 store.seen = 0;
875
876 /*
877 * After this, store.offset will contain the *end* offset
878 * of the last match, or remain at 0 if no match was found.
879 * As a side effect, we make sure to transform only a valid
880 * existing config file.
881 */
9c3796fc 882 if (git_config_from_file(store_aux, config_filename)) {
10bea152
JS
883 fprintf(stderr, "invalid config file\n");
884 free(store.key);
885 if (store.value_regex != NULL) {
886 regfree(store.value_regex);
887 free(store.value_regex);
888 }
dafc88b1
SH
889 ret = 3;
890 goto out_free;
10bea152
JS
891 }
892
893 free(store.key);
894 if (store.value_regex != NULL) {
895 regfree(store.value_regex);
896 free(store.value_regex);
897 }
898
4ddba79d
JS
899 /* if nothing to unset, or too many matches, error out */
900 if ((store.seen == 0 && value == NULL) ||
901 (store.seen > 1 && multi_replace == 0)) {
dafc88b1
SH
902 ret = 5;
903 goto out_free;
10bea152
JS
904 }
905
88fb958b 906 fstat(in_fd, &st);
dc49cd76
SP
907 contents_sz = xsize_t(st.st_size);
908 contents = xmmap(NULL, contents_sz, PROT_READ,
10bea152
JS
909 MAP_PRIVATE, in_fd, 0);
910 close(in_fd);
911
4ddba79d
JS
912 if (store.seen == 0)
913 store.seen = 1;
914
915 for (i = 0, copy_begin = 0; i < store.seen; i++) {
916 if (store.offset[i] == 0) {
dc49cd76 917 store.offset[i] = copy_end = contents_sz;
4ddba79d
JS
918 } else if (store.state != KEY_SEEN) {
919 copy_end = store.offset[i];
10bea152 920 } else
4ddba79d 921 copy_end = find_beginning_of_line(
dc49cd76 922 contents, contents_sz,
4ddba79d
JS
923 store.offset[i]-2, &new_line);
924
925 /* write the first part of the config */
926 if (copy_end > copy_begin) {
93c1e079
JH
927 if (write_in_full(fd, contents + copy_begin,
928 copy_end - copy_begin) <
929 copy_end - copy_begin)
930 goto write_err_out;
931 if (new_line &&
932 write_in_full(fd, "\n", 1) != 1)
933 goto write_err_out;
4ddba79d
JS
934 }
935 copy_begin = store.offset[i];
10bea152
JS
936 }
937
10bea152
JS
938 /* write the pair (value == NULL means unset) */
939 if (value != NULL) {
93c1e079
JH
940 if (store.state == START) {
941 if (!store_write_section(fd, key))
942 goto write_err_out;
480c9e52 943 }
93c1e079
JH
944 if (!store_write_pair(fd, key, value))
945 goto write_err_out;
10bea152
JS
946 }
947
948 /* write the rest of the config */
dc49cd76 949 if (copy_begin < contents_sz)
93c1e079 950 if (write_in_full(fd, contents + copy_begin,
dc49cd76
SP
951 contents_sz - copy_begin) <
952 contents_sz - copy_begin)
93c1e079 953 goto write_err_out;
10bea152 954
dc49cd76 955 munmap(contents, contents_sz);
10bea152
JS
956 }
957
6cbf973c
BS
958 if (close(fd) || commit_lock_file(lock) < 0) {
959 fprintf(stderr, "Cannot commit config file!\n");
dafc88b1
SH
960 ret = 4;
961 goto out_free;
10bea152
JS
962 }
963
6cbf973c
BS
964 /* fd is closed, so don't try to close it below. */
965 fd = -1;
966 /*
967 * lock is committed, so don't try to roll it back below.
968 * NOTE: Since lockfile.c keeps a linked list of all created
969 * lock_file structures, it isn't safe to free(lock). It's
970 * better to just leave it hanging around.
971 */
972 lock = NULL;
dafc88b1
SH
973 ret = 0;
974
975out_free:
f8ba655e
JH
976 if (0 <= fd)
977 close(fd);
6cbf973c
BS
978 if (lock)
979 rollback_lock_file(lock);
4cac42b1 980 free(config_filename);
dafc88b1 981 return ret;
93c1e079
JH
982
983write_err_out:
984 ret = write_error();
985 goto out_free;
986
10bea152
JS
987}
988
118f8b24
PB
989static int section_name_match (const char *buf, const char *name)
990{
991 int i = 0, j = 0, dot = 0;
992 for (; buf[i] && buf[i] != ']'; i++) {
993 if (!dot && isspace(buf[i])) {
994 dot = 1;
995 if (name[j++] != '.')
996 break;
997 for (i++; isspace(buf[i]); i++)
998 ; /* do nothing */
999 if (buf[i] != '"')
1000 break;
1001 continue;
1002 }
1003 if (buf[i] == '\\' && dot)
1004 i++;
1005 else if (buf[i] == '"' && dot) {
1006 for (i++; isspace(buf[i]); i++)
1007 ; /* do_nothing */
1008 break;
1009 }
1010 if (buf[i] != name[j++])
1011 break;
1012 }
1013 return (buf[i] == ']' && name[j] == 0);
1014}
1015
1016/* if new_name == NULL, the section is removed instead */
0667fcfb
JS
1017int git_config_rename_section(const char *old_name, const char *new_name)
1018{
118f8b24 1019 int ret = 0, remove = 0;
fc1905bb 1020 char *config_filename;
0667fcfb
JS
1021 struct lock_file *lock = xcalloc(sizeof(struct lock_file), 1);
1022 int out_fd;
1023 char buf[1024];
1024
d4ebc36c 1025 config_filename = getenv(CONFIG_ENVIRONMENT);
0667fcfb 1026 if (!config_filename) {
d4ebc36c 1027 config_filename = getenv(CONFIG_LOCAL_ENVIRONMENT);
0667fcfb
JS
1028 if (!config_filename)
1029 config_filename = git_path("config");
1030 }
1031 config_filename = xstrdup(config_filename);
1032 out_fd = hold_lock_file_for_update(lock, config_filename, 0);
fc1905bb
JH
1033 if (out_fd < 0) {
1034 ret = error("Could not lock config file!");
1035 goto out;
1036 }
0667fcfb 1037
fc1905bb 1038 if (!(config_file = fopen(config_filename, "rb"))) {
01ebb9dc
GB
1039 /* no config file means nothing to rename, no error */
1040 goto unlock_and_out;
fc1905bb 1041 }
0667fcfb
JS
1042
1043 while (fgets(buf, sizeof(buf), config_file)) {
1044 int i;
480c9e52 1045 int length;
0667fcfb
JS
1046 for (i = 0; buf[i] && isspace(buf[i]); i++)
1047 ; /* do nothing */
1048 if (buf[i] == '[') {
1049 /* it's a section */
118f8b24
PB
1050 if (section_name_match (&buf[i+1], old_name)) {
1051 ret++;
1052 if (new_name == NULL) {
1053 remove = 1;
0667fcfb
JS
1054 continue;
1055 }
0667fcfb 1056 store.baselen = strlen(new_name);
480c9e52
AW
1057 if (!store_write_section(out_fd, new_name)) {
1058 ret = write_error();
1059 goto out;
1060 }
0667fcfb
JS
1061 continue;
1062 }
118f8b24 1063 remove = 0;
0667fcfb 1064 }
118f8b24
PB
1065 if (remove)
1066 continue;
480c9e52
AW
1067 length = strlen(buf);
1068 if (write_in_full(out_fd, buf, length) != length) {
1069 ret = write_error();
1070 goto out;
1071 }
0667fcfb 1072 }
fc1905bb 1073 fclose(config_file);
01ebb9dc 1074 unlock_and_out:
0667fcfb 1075 if (close(out_fd) || commit_lock_file(lock) < 0)
480c9e52 1076 ret = error("Cannot commit config file!");
fc1905bb
JH
1077 out:
1078 free(config_filename);
0667fcfb
JS
1079 return ret;
1080}