submodule.c 48.1 KB
Newer Older
1
/*
Edward Thomson committed
2
 * Copyright (C) the libgit2 contributors. All rights reserved.
3 4 5 6 7 8 9
 *
 * This file is part of libgit2, distributed under the GNU GPL v2 with
 * a Linking Exception. For full terms see the included COPYING file.
 */

#include "common.h"
#include "git2/config.h"
10
#include "git2/sys/config.h"
11 12 13
#include "git2/types.h"
#include "git2/index.h"
#include "buffer.h"
14
#include "buf_text.h"
15 16 17 18 19
#include "vector.h"
#include "posix.h"
#include "config_file.h"
#include "config.h"
#include "repository.h"
Russell Belfer committed
20 21 22
#include "submodule.h"
#include "tree.h"
#include "iterator.h"
23 24
#include "path.h"
#include "index.h"
Russell Belfer committed
25 26

#define GIT_MODULES_FILE ".gitmodules"
27

28 29 30
static git_cvar_map _sm_update_map[] = {
	{GIT_CVAR_STRING, "checkout", GIT_SUBMODULE_UPDATE_CHECKOUT},
	{GIT_CVAR_STRING, "rebase", GIT_SUBMODULE_UPDATE_REBASE},
Russell Belfer committed
31 32
	{GIT_CVAR_STRING, "merge", GIT_SUBMODULE_UPDATE_MERGE},
	{GIT_CVAR_STRING, "none", GIT_SUBMODULE_UPDATE_NONE},
33 34
	{GIT_CVAR_FALSE, NULL, GIT_SUBMODULE_UPDATE_NONE},
	{GIT_CVAR_TRUE, NULL, GIT_SUBMODULE_UPDATE_CHECKOUT},
35 36
};

37
static git_cvar_map _sm_ignore_map[] = {
Russell Belfer committed
38
	{GIT_CVAR_STRING, "none", GIT_SUBMODULE_IGNORE_NONE},
39
	{GIT_CVAR_STRING, "untracked", GIT_SUBMODULE_IGNORE_UNTRACKED},
Russell Belfer committed
40 41
	{GIT_CVAR_STRING, "dirty", GIT_SUBMODULE_IGNORE_DIRTY},
	{GIT_CVAR_STRING, "all", GIT_SUBMODULE_IGNORE_ALL},
42 43
	{GIT_CVAR_FALSE, NULL, GIT_SUBMODULE_IGNORE_NONE},
	{GIT_CVAR_TRUE, NULL, GIT_SUBMODULE_IGNORE_ALL},
44 45
};

46 47 48 49 50 51
static git_cvar_map _sm_recurse_map[] = {
	{GIT_CVAR_STRING, "on-demand", GIT_SUBMODULE_RECURSE_ONDEMAND},
	{GIT_CVAR_FALSE, NULL, GIT_SUBMODULE_RECURSE_NO},
	{GIT_CVAR_TRUE, NULL, GIT_SUBMODULE_RECURSE_YES},
};

52 53 54 55 56 57 58 59 60 61
enum {
	CACHE_OK = 0,
	CACHE_REFRESH = 1,
	CACHE_FLUSH = 2
};
enum {
	GITMODULES_EXISTING = 0,
	GITMODULES_CREATE = 1,
};

62
static kh_inline khint_t str_hash_no_trailing_slash(const char *s)
63
{
64
	khint_t h;
65

66
	for (h = 0; *s; ++s)
67
		if (s[1] != '\0' || *s != '/')
68
			h = (h << 5) - h + *s;
69

70
	return h;
71 72
}

73
static kh_inline int str_equal_no_trailing_slash(const char *a, const char *b)
74
{
75 76
	size_t alen = a ? strlen(a) : 0;
	size_t blen = b ? strlen(b) : 0;
77

78
	if (alen > 0 && a[alen - 1] == '/')
79
		alen--;
80
	if (blen > 0 && b[blen - 1] == '/')
81 82
		blen--;

83
	return (alen == blen && strncmp(a, b, alen) == 0);
84 85
}

Russell Belfer committed
86 87 88 89
__KHASH_IMPL(
	str, static kh_inline, const char *, void *, 1,
	str_hash_no_trailing_slash, str_equal_no_trailing_slash);

90
static int submodule_cache_init(git_repository *repo, int refresh);
91 92
static void submodule_cache_free(git_submodule_cache *cache);

93
static git_config_backend *open_gitmodules(git_submodule_cache *, int gitmod);
94 95
static int get_url_base(git_buf *url, git_repository *repo);
static int lookup_head_remote_key(git_buf *remote_key, git_repository *repo);
96
static int submodule_get(git_submodule **, git_submodule_cache *, const char *, const char *);
97
static int submodule_load_from_config(const git_config_entry *, void *);
98
static int submodule_load_from_wd_lite(git_submodule *);
99 100
static void submodule_get_index_status(unsigned int *, git_submodule *);
static void submodule_get_wd_status(unsigned int *, git_submodule *, git_repository *, git_submodule_ignore_t);
Russell Belfer committed
101 102 103 104 105 106 107 108 109 110 111 112 113

static int submodule_cmp(const void *a, const void *b)
{
	return strcmp(((git_submodule *)a)->name, ((git_submodule *)b)->name);
}

static int submodule_config_key_trunc_puts(git_buf *key, const char *suffix)
{
	ssize_t idx = git_buf_rfind(key, '.');
	git_buf_truncate(key, (size_t)(idx + 1));
	return git_buf_puts(key, suffix);
}

114 115 116
/* lookup submodule or return ENOTFOUND if it doesn't exist */
static int submodule_lookup(
	git_submodule **out,
117
	git_submodule_cache *cache,
118 119 120 121 122 123 124
	const char *name,
	const char *alternate)
{
	khiter_t pos;

	/* lock cache */

125
	pos = git_strmap_lookup_index(cache->submodules, name);
126

127 128
	if (!git_strmap_valid_index(cache->submodules, pos) && alternate)
		pos = git_strmap_lookup_index(cache->submodules, alternate);
129

130
	if (!git_strmap_valid_index(cache->submodules, pos)) {
131 132 133 134 135
		/* unlock cache */
		return GIT_ENOTFOUND; /* don't set error - caller will cope */
	}

	if (out != NULL) {
136
		git_submodule *sm = git_strmap_value_at(cache->submodules, pos);
137 138 139 140 141 142 143 144 145
		GIT_REFCOUNT_INC(sm);
		*out = sm;
	}

	/* unlock cache */

	return 0;
}

146 147 148 149 150 151 152 153 154 155 156 157
/* clear a set of flags on all submodules */
static void submodule_cache_clear_flags(
	git_submodule_cache *cache, uint32_t mask)
{
	git_submodule *sm;
	uint32_t inverted_mask = ~mask;

	git_strmap_foreach_value(cache->submodules, sm, {
		sm->flags &= inverted_mask;
	});
}

Russell Belfer committed
158 159 160 161
/*
 * PUBLIC APIS
 */

162 163 164 165
bool git_submodule__is_submodule(git_repository *repo, const char *name)
{
	git_strmap *map;

166
	if (submodule_cache_init(repo, CACHE_OK) < 0) {
167 168 169 170
		giterr_clear();
		return false;
	}

171
	if (!repo->_submodules || !(map = repo->_submodules->submodules))
172 173 174 175 176
		return false;

	return git_strmap_valid_index(map, git_strmap_lookup_index(map, name));
}

177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
static void submodule_set_lookup_error(int error, const char *name)
{
	if (!error)
		return;

	giterr_set(GITERR_SUBMODULE, (error == GIT_ENOTFOUND) ?
		"No submodule named '%s'" :
		"Submodule '%s' has not been added yet", name);
}

int git_submodule__lookup(
	git_submodule **out, /* NULL if user only wants to test existence */
	git_repository *repo,
	const char *name)    /* trailing slash is allowed */
{
	int error;

	assert(repo && name);

196
	if ((error = submodule_cache_init(repo, CACHE_OK)) < 0)
197 198 199 200 201 202 203 204
		return error;

	if ((error = submodule_lookup(out, repo->_submodules, name, NULL)) < 0)
		submodule_set_lookup_error(error, name);

	return error;
}

Russell Belfer committed
205
int git_submodule_lookup(
206
	git_submodule **out, /* NULL if user only wants to test existence */
Russell Belfer committed
207
	git_repository *repo,
208
	const char *name)    /* trailing slash is allowed */
Russell Belfer committed
209 210 211 212 213
{
	int error;

	assert(repo && name);

214
	if ((error = submodule_cache_init(repo, CACHE_REFRESH)) < 0)
Russell Belfer committed
215 216
		return error;

217
	if ((error = submodule_lookup(out, repo->_submodules, name, NULL)) < 0) {
Russell Belfer committed
218 219 220 221 222

		/* check if a plausible submodule exists at path */
		if (git_repository_workdir(repo)) {
			git_buf path = GIT_BUF_INIT;

223 224
			if (git_buf_join3(&path,
					'/', git_repository_workdir(repo), name, DOT_GIT) < 0)
Russell Belfer committed
225 226
				return -1;

227
			if (git_path_exists(path.ptr))
Russell Belfer committed
228 229 230 231 232
				error = GIT_EEXISTS;

			git_buf_free(&path);
		}

233
		submodule_set_lookup_error(error, name);
Russell Belfer committed
234 235
	}

236
	return error;
Russell Belfer committed
237
}
238

239 240 241 242 243
static void submodule_free_dup(void *sm)
{
	git_submodule_free(sm);
}

Russell Belfer committed
244 245 246 247 248 249
int git_submodule_foreach(
	git_repository *repo,
	int (*callback)(git_submodule *sm, const char *name, void *payload),
	void *payload)
{
	int error;
250
	size_t i;
Russell Belfer committed
251
	git_submodule *sm;
252 253
	git_submodule_cache *cache;
	git_vector snapshot = GIT_VECTOR_INIT;
Russell Belfer committed
254 255 256

	assert(repo && callback);

257
	if ((error = submodule_cache_init(repo, CACHE_REFRESH)) < 0)
Russell Belfer committed
258 259
		return error;

260 261 262 263 264 265 266 267 268 269 270 271
	cache = repo->_submodules;

	if (git_mutex_lock(&cache->lock) < 0) {
		giterr_set(GITERR_OS, "Unable to acquire lock on submodule cache");
		return -1;
	}

	if (!(error = git_vector_init(
			&snapshot, kh_size(cache->submodules), submodule_cmp))) {

		git_strmap_foreach_value(cache->submodules, sm, {
			if ((error = git_vector_insert(&snapshot, sm)) < 0)
Russell Belfer committed
272
				break;
273 274 275 276 277
			GIT_REFCOUNT_INC(sm);
		});
	}

	git_mutex_unlock(&cache->lock);
Russell Belfer committed
278

279 280 281 282 283 284
	if (error < 0)
		goto done;

	git_vector_uniq(&snapshot, submodule_free_dup);

	git_vector_foreach(&snapshot, i, sm) {
285
		if ((error = callback(sm, sm->name, payload)) != 0) {
286
			giterr_set_after_callback(error);
Russell Belfer committed
287
			break;
288
		}
289
	}
Russell Belfer committed
290

291 292 293 294
done:
	git_vector_foreach(&snapshot, i, sm)
		git_submodule_free(sm);
	git_vector_free(&snapshot);
Russell Belfer committed
295 296 297 298

	return error;
}

299
void git_submodule_cache_free(git_repository *repo)
Russell Belfer committed
300
{
301
	git_submodule_cache *cache;
Russell Belfer committed
302 303 304

	assert(repo);

305 306
	if ((cache = git__swap(repo->_submodules, NULL)) != NULL)
		submodule_cache_free(cache);
Russell Belfer committed
307 308 309
}

int git_submodule_add_setup(
310
	git_submodule **out,
Russell Belfer committed
311 312 313 314 315 316
	git_repository *repo,
	const char *url,
	const char *path,
	int use_gitlink)
{
	int error = 0;
Ben Straub committed
317
	git_config_backend *mods = NULL;
318
	git_submodule *sm = NULL;
Russell Belfer committed
319
	git_buf name = GIT_BUF_INIT, real_url = GIT_BUF_INIT;
320
	git_repository_init_options initopt = GIT_REPOSITORY_INIT_OPTIONS_INIT;
Russell Belfer committed
321 322 323 324 325 326
	git_repository *subrepo = NULL;

	assert(repo && url && path);

	/* see if there is already an entry for this submodule */

327
	if (git_submodule_lookup(NULL, repo, path) < 0)
Russell Belfer committed
328 329 330
		giterr_clear();
	else {
		giterr_set(GITERR_SUBMODULE,
331
			"Attempt to add submodule '%s' that already exists", path);
Russell Belfer committed
332 333 334 335
		return GIT_EEXISTS;
	}

	/* resolve parameters */
336
	if ((error = git_submodule_resolve_url(&real_url, repo, url)) < 0)
Russell Belfer committed
337 338 339 340 341 342 343 344 345 346 347 348 349 350 351
		goto cleanup;

	/* validate and normalize path */

	if (git__prefixcmp(path, git_repository_workdir(repo)) == 0)
		path += strlen(git_repository_workdir(repo));

	if (git_path_root(path) >= 0) {
		giterr_set(GITERR_SUBMODULE, "Submodule path must be a relative path");
		error = -1;
		goto cleanup;
	}

	/* update .gitmodules */

352
	if (!(mods = open_gitmodules(repo->_submodules, GITMODULES_CREATE))) {
Russell Belfer committed
353
		giterr_set(GITERR_SUBMODULE,
354
			"Adding submodules to a bare repository is not supported");
Russell Belfer committed
355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391
		return -1;
	}

	if ((error = git_buf_printf(&name, "submodule.%s.path", path)) < 0 ||
		(error = git_config_file_set_string(mods, name.ptr, path)) < 0)
		goto cleanup;

	if ((error = submodule_config_key_trunc_puts(&name, "url")) < 0 ||
		(error = git_config_file_set_string(mods, name.ptr, real_url.ptr)) < 0)
		goto cleanup;

	git_buf_clear(&name);

	/* init submodule repository and add origin remote as needed */

	error = git_buf_joinpath(&name, git_repository_workdir(repo), path);
	if (error < 0)
		goto cleanup;

	/* New style: sub-repo goes in <repo-dir>/modules/<name>/ with a
	 * gitlink in the sub-repo workdir directory to that repository
	 *
	 * Old style: sub-repo goes directly into repo/<name>/.git/
	 */

	initopt.flags = GIT_REPOSITORY_INIT_MKPATH |
		GIT_REPOSITORY_INIT_NO_REINIT;
	initopt.origin_url = real_url.ptr;

	if (git_path_exists(name.ptr) &&
		git_path_contains(&name, DOT_GIT))
	{
		/* repo appears to already exist - reinit? */
	}
	else if (use_gitlink) {
		git_buf repodir = GIT_BUF_INIT;

392 393
		error = git_buf_join3(
			&repodir, '/', git_repository_path(repo), "modules", path);
Russell Belfer committed
394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
		if (error < 0)
			goto cleanup;

		initopt.workdir_path = name.ptr;
		initopt.flags |= GIT_REPOSITORY_INIT_NO_DOTGIT_DIR;

		error = git_repository_init_ext(&subrepo, repodir.ptr, &initopt);

		git_buf_free(&repodir);
	}
	else {
		error = git_repository_init_ext(&subrepo, name.ptr, &initopt);
	}
	if (error < 0)
		goto cleanup;

	/* add submodule to hash and "reload" it */

412 413 414 415 416 417
	if (git_mutex_lock(&repo->_submodules->lock) < 0) {
		giterr_set(GITERR_OS, "Unable to acquire lock on submodule cache");
		error = -1;
		goto cleanup;
	}

418
	if (!(error = submodule_get(&sm, repo->_submodules, path, NULL)) &&
419
		!(error = git_submodule_reload(sm, false)))
420
		error = git_submodule_init(sm, false);
Russell Belfer committed
421

422 423
	git_mutex_unlock(&repo->_submodules->lock);

Russell Belfer committed
424
cleanup:
425 426 427 428 429 430
	if (error && sm) {
		git_submodule_free(sm);
		sm = NULL;
	}
	if (out != NULL)
		*out = sm;
Russell Belfer committed
431

432
	git_config_file_free(mods);
Russell Belfer committed
433 434 435 436 437 438 439 440 441 442 443 444 445 446
	git_repository_free(subrepo);
	git_buf_free(&real_url);
	git_buf_free(&name);

	return error;
}

int git_submodule_add_finalize(git_submodule *sm)
{
	int error;
	git_index *index;

	assert(sm);

447
	if ((error = git_repository_index__weakptr(&index, sm->repo)) < 0 ||
448
		(error = git_index_add_bypath(index, GIT_MODULES_FILE)) < 0)
Russell Belfer committed
449 450
		return error;

451
	return git_submodule_add_to_index(sm, true);
Russell Belfer committed
452 453
}

454
int git_submodule_add_to_index(git_submodule *sm, int write_index)
Russell Belfer committed
455 456
{
	int error;
457
	git_repository *sm_repo = NULL;
Russell Belfer committed
458 459 460 461 462 463 464 465
	git_index *index;
	git_buf path = GIT_BUF_INIT;
	git_commit *head;
	git_index_entry entry;
	struct stat st;

	assert(sm);

466 467 468
	/* force reload of wd OID by git_submodule_open */
	sm->flags = sm->flags & ~GIT_SUBMODULE_STATUS__WD_OID_VALID;

469
	if ((error = git_repository_index__weakptr(&index, sm->repo)) < 0 ||
Russell Belfer committed
470
		(error = git_buf_joinpath(
471
			&path, git_repository_workdir(sm->repo), sm->path)) < 0 ||
Russell Belfer committed
472 473 474 475 476 477 478 479 480 481
		(error = git_submodule_open(&sm_repo, sm)) < 0)
		goto cleanup;

	/* read stat information for submodule working directory */
	if (p_stat(path.ptr, &st) < 0) {
		giterr_set(GITERR_SUBMODULE,
			"Cannot add submodule without working directory");
		error = -1;
		goto cleanup;
	}
482 483

	memset(&entry, 0, sizeof(entry));
484
	entry.path = sm->path;
485 486
	git_index_entry__init_from_stat(
		&entry, &st, !(git_index_caps(index) & GIT_INDEXCAP_NO_FILEMODE));
Russell Belfer committed
487 488 489 490 491 492 493 494

	/* calling git_submodule_open will have set sm->wd_oid if possible */
	if ((sm->flags & GIT_SUBMODULE_STATUS__WD_OID_VALID) == 0) {
		giterr_set(GITERR_SUBMODULE,
			"Cannot add submodule without HEAD to index");
		error = -1;
		goto cleanup;
	}
495
	git_oid_cpy(&entry.id, &sm->wd_oid);
Russell Belfer committed
496 497 498 499 500 501 502 503 504 505 506

	if ((error = git_commit_lookup(&head, sm_repo, &sm->wd_oid)) < 0)
		goto cleanup;

	entry.ctime.seconds = git_commit_time(head);
	entry.ctime.nanoseconds = 0;
	entry.mtime.seconds = git_commit_time(head);
	entry.mtime.nanoseconds = 0;

	git_commit_free(head);

507
	/* add it */
Edward Thomson committed
508
	error = git_index_add(index, &entry);
Russell Belfer committed
509

510 511 512 513 514 515 516 517
	/* write it, if requested */
	if (!error && write_index) {
		error = git_index_write(index);

		if (!error)
			git_oid_cpy(&sm->index_oid, &sm->wd_oid);
	}

Russell Belfer committed
518 519 520 521 522 523
cleanup:
	git_repository_free(sm_repo);
	git_buf_free(&path);
	return error;
}

524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541
const char *git_submodule_ignore_to_str(git_submodule_ignore_t ignore)
{
	int i;
	for (i = 0; i < (int)ARRAY_SIZE(_sm_ignore_map); ++i)
		if (_sm_ignore_map[i].map_value == ignore)
			return _sm_ignore_map[i].str_match;
	return NULL;
}

const char *git_submodule_update_to_str(git_submodule_update_t update)
{
	int i;
	for (i = 0; i < (int)ARRAY_SIZE(_sm_update_map); ++i)
		if (_sm_update_map[i].map_value == update)
			return _sm_update_map[i].str_match;
	return NULL;
}

542 543 544 545 546 547 548 549 550
const char *git_submodule_recurse_to_str(git_submodule_recurse_t recurse)
{
	int i;
	for (i = 0; i < (int)ARRAY_SIZE(_sm_recurse_map); ++i)
		if (_sm_recurse_map[i].map_value == recurse)
			return _sm_recurse_map[i].str_match;
	return NULL;
}

Russell Belfer committed
551 552 553
int git_submodule_save(git_submodule *submodule)
{
	int error = 0;
Ben Straub committed
554
	git_config_backend *mods;
Russell Belfer committed
555
	git_buf key = GIT_BUF_INIT;
556
	const char *val;
Russell Belfer committed
557 558 559

	assert(submodule);

560
	mods = open_gitmodules(submodule->repo->_submodules, GITMODULES_CREATE);
Russell Belfer committed
561 562
	if (!mods) {
		giterr_set(GITERR_SUBMODULE,
563
			"Adding submodules to a bare repository is not supported");
Russell Belfer committed
564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579
		return -1;
	}

	if ((error = git_buf_printf(&key, "submodule.%s.", submodule->name)) < 0)
		goto cleanup;

	/* save values for path, url, update, ignore, fetchRecurseSubmodules */

	if ((error = submodule_config_key_trunc_puts(&key, "path")) < 0 ||
		(error = git_config_file_set_string(mods, key.ptr, submodule->path)) < 0)
		goto cleanup;

	if ((error = submodule_config_key_trunc_puts(&key, "url")) < 0 ||
		(error = git_config_file_set_string(mods, key.ptr, submodule->url)) < 0)
		goto cleanup;

580 581 582 583
	if ((error = submodule_config_key_trunc_puts(&key, "branch")) < 0 ||
		(error = git_config_file_set_string(mods, key.ptr, submodule->branch)) < 0)
		goto cleanup;

Russell Belfer committed
584
	if (!(error = submodule_config_key_trunc_puts(&key, "update")) &&
585
		(val = git_submodule_update_to_str(submodule->update)) != NULL)
Russell Belfer committed
586 587 588 589 590
		error = git_config_file_set_string(mods, key.ptr, val);
	if (error < 0)
		goto cleanup;

	if (!(error = submodule_config_key_trunc_puts(&key, "ignore")) &&
591
		(val = git_submodule_ignore_to_str(submodule->ignore)) != NULL)
Russell Belfer committed
592 593 594 595
		error = git_config_file_set_string(mods, key.ptr, val);
	if (error < 0)
		goto cleanup;

596 597 598 599
	if (!(error = submodule_config_key_trunc_puts(&key, "fetchRecurseSubmodules")) &&
		(val = git_submodule_recurse_to_str(submodule->fetch_recurse)) != NULL)
		error = git_config_file_set_string(mods, key.ptr, val);
	if (error < 0)
Russell Belfer committed
600 601 602 603 604 605
		goto cleanup;

	/* update internal defaults */

	submodule->ignore_default = submodule->ignore;
	submodule->update_default = submodule->update;
606
	submodule->fetch_recurse_default = submodule->fetch_recurse;
Russell Belfer committed
607 608 609
	submodule->flags |= GIT_SUBMODULE_STATUS_IN_CONFIG;

cleanup:
610
	git_config_file_free(mods);
Russell Belfer committed
611 612 613 614 615 616 617 618
	git_buf_free(&key);

	return error;
}

git_repository *git_submodule_owner(git_submodule *submodule)
{
	assert(submodule);
619
	return submodule->repo;
Russell Belfer committed
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
}

const char *git_submodule_name(git_submodule *submodule)
{
	assert(submodule);
	return submodule->name;
}

const char *git_submodule_path(git_submodule *submodule)
{
	assert(submodule);
	return submodule->path;
}

const char *git_submodule_url(git_submodule *submodule)
{
	assert(submodule);
	return submodule->url;
}

640 641 642 643
int git_submodule_resolve_url(git_buf *out, git_repository *repo, const char *url)
{
	int error = 0;

644 645 646
	assert(out && repo && url);

	git_buf_sanitize(out);
Carlos Martín Nieto committed
647

648
	if (git_path_is_relative(url)) {
649
		if (!(error = get_url_base(out, repo)))
650 651 652 653 654 655 656 657 658 659 660
			error = git_path_apply_relative(out, url);
	} else if (strchr(url, ':') != NULL || url[0] == '/') {
		error = git_buf_sets(out, url);
	} else {
		giterr_set(GITERR_SUBMODULE, "Invalid format for submodule URL");
		error = -1;
	}

	return error;
}

661 662 663 664 665 666
const char *git_submodule_branch(git_submodule *submodule)
{
	assert(submodule);
	return submodule->branch;
}

Russell Belfer committed
667 668 669 670 671 672 673 674 675 676 677 678
int git_submodule_set_url(git_submodule *submodule, const char *url)
{
	assert(submodule && url);

	git__free(submodule->url);

	submodule->url = git__strdup(url);
	GITERR_CHECK_ALLOC(submodule->url);

	return 0;
}

679
const git_oid *git_submodule_index_id(git_submodule *submodule)
Russell Belfer committed
680 681 682 683 684 685 686 687 688
{
	assert(submodule);

	if (submodule->flags & GIT_SUBMODULE_STATUS__INDEX_OID_VALID)
		return &submodule->index_oid;
	else
		return NULL;
}

689
const git_oid *git_submodule_head_id(git_submodule *submodule)
Russell Belfer committed
690 691 692 693 694 695 696 697 698
{
	assert(submodule);

	if (submodule->flags & GIT_SUBMODULE_STATUS__HEAD_OID_VALID)
		return &submodule->head_oid;
	else
		return NULL;
}

699
const git_oid *git_submodule_wd_id(git_submodule *submodule)
Russell Belfer committed
700 701 702
{
	assert(submodule);

703
	/* load unless we think we have a valid oid */
Russell Belfer committed
704 705 706 707
	if (!(submodule->flags & GIT_SUBMODULE_STATUS__WD_OID_VALID)) {
		git_repository *subrepo;

		/* calling submodule open grabs the HEAD OID if possible */
708
		if (!git_submodule_open_bare(&subrepo, submodule))
Russell Belfer committed
709
			git_repository_free(subrepo);
710 711
		else
			giterr_clear();
Russell Belfer committed
712 713 714 715 716 717 718 719 720 721 722
	}

	if (submodule->flags & GIT_SUBMODULE_STATUS__WD_OID_VALID)
		return &submodule->wd_oid;
	else
		return NULL;
}

git_submodule_ignore_t git_submodule_ignore(git_submodule *submodule)
{
	assert(submodule);
723 724
	return (submodule->ignore < GIT_SUBMODULE_IGNORE_NONE) ?
		GIT_SUBMODULE_IGNORE_NONE : submodule->ignore;
Russell Belfer committed
725 726 727 728 729 730 731 732 733
}

git_submodule_ignore_t git_submodule_set_ignore(
	git_submodule *submodule, git_submodule_ignore_t ignore)
{
	git_submodule_ignore_t old;

	assert(submodule);

734
	if (ignore == GIT_SUBMODULE_IGNORE_RESET)
Russell Belfer committed
735 736 737 738 739 740 741 742 743 744
		ignore = submodule->ignore_default;

	old = submodule->ignore;
	submodule->ignore = ignore;
	return old;
}

git_submodule_update_t git_submodule_update(git_submodule *submodule)
{
	assert(submodule);
745 746
	return (submodule->update < GIT_SUBMODULE_UPDATE_CHECKOUT) ?
		GIT_SUBMODULE_UPDATE_CHECKOUT : submodule->update;
Russell Belfer committed
747 748 749 750 751 752 753 754 755
}

git_submodule_update_t git_submodule_set_update(
	git_submodule *submodule, git_submodule_update_t update)
{
	git_submodule_update_t old;

	assert(submodule);

756
	if (update == GIT_SUBMODULE_UPDATE_RESET)
Russell Belfer committed
757 758 759 760 761 762 763
		update = submodule->update_default;

	old = submodule->update;
	submodule->update = update;
	return old;
}

764
git_submodule_recurse_t git_submodule_fetch_recurse_submodules(
765 766 767 768 769 770
	git_submodule *submodule)
{
	assert(submodule);
	return submodule->fetch_recurse;
}

771
git_submodule_recurse_t git_submodule_set_fetch_recurse_submodules(
772
	git_submodule *submodule,
773
	git_submodule_recurse_t fetch_recurse_submodules)
774
{
775
	git_submodule_recurse_t old;
776 777 778

	assert(submodule);

779 780 781
	if (fetch_recurse_submodules == GIT_SUBMODULE_RECURSE_RESET)
		fetch_recurse_submodules = submodule->fetch_recurse_default;

782
	old = submodule->fetch_recurse;
783
	submodule->fetch_recurse = fetch_recurse_submodules;
784 785 786
	return old;
}

787
int git_submodule_init(git_submodule *sm, int overwrite)
Russell Belfer committed
788 789
{
	int error;
790
	const char *val;
791 792
	git_buf key = GIT_BUF_INIT;
	git_config *cfg = NULL;
Russell Belfer committed
793

794
	if (!sm->url) {
Russell Belfer committed
795
		giterr_set(GITERR_SUBMODULE,
796
			"No URL configured for submodule '%s'", sm->name);
Russell Belfer committed
797 798 799
		return -1;
	}

800
	if ((error = git_repository_config(&cfg, sm->repo)) < 0)
Russell Belfer committed
801 802
		return error;

803 804 805 806 807 808 809
	/* write "submodule.NAME.url" */

	if ((error = git_buf_printf(&key, "submodule.%s.url", sm->name)) < 0 ||
		(error = git_config__update_entry(
			cfg, key.ptr, sm->url, overwrite != 0, false)) < 0)
		goto cleanup;

Russell Belfer committed
810 811
	/* write "submodule.NAME.update" if not default */

812 813 814 815 816 817 818 819 820 821 822 823 824
	val = (sm->update == GIT_SUBMODULE_UPDATE_CHECKOUT) ?
		NULL : git_submodule_update_to_str(sm->update);

	if ((error = git_buf_printf(&key, "submodule.%s.update", sm->name)) < 0 ||
		(error = git_config__update_entry(
			cfg, key.ptr, val, overwrite != 0, false)) < 0)
		goto cleanup;

	/* success */

cleanup:
	git_config_free(cfg);
	git_buf_free(&key);
Russell Belfer committed
825 826 827 828

	return error;
}

829
int git_submodule_sync(git_submodule *sm)
Russell Belfer committed
830
{
831 832 833
	int error = 0;
	git_config *cfg = NULL;
	git_buf key = GIT_BUF_INIT;
834
	git_repository *smrepo = NULL;
835 836

	if (!sm->url) {
Russell Belfer committed
837
		giterr_set(GITERR_SUBMODULE,
838
			"No URL configured for submodule '%s'", sm->name);
Russell Belfer committed
839 840 841 842 843
		return -1;
	}

	/* copy URL over to config only if it already exists */

844 845 846 847 848 849
	if (!(error = git_repository_config__weakptr(&cfg, sm->repo)) &&
		!(error = git_buf_printf(&key, "submodule.%s.url", sm->name)))
		error = git_config__update_entry(cfg, key.ptr, sm->url, true, true);

	/* if submodule exists in the working directory, update remote url */

850 851 852 853
	if (!error &&
		(sm->flags & GIT_SUBMODULE_STATUS_IN_WD) != 0 &&
		!(error = git_submodule_open(&smrepo, sm)))
	{
854 855 856 857 858
		git_buf remote_name = GIT_BUF_INIT;

		if ((error = git_repository_config__weakptr(&cfg, smrepo)) < 0)
			/* return error from reading submodule config */;
		else if ((error = lookup_head_remote_key(&remote_name, smrepo)) < 0) {
859
			giterr_clear();
860 861 862 863 864
			error = git_buf_sets(&key, "branch.origin.remote");
		} else {
			error = git_buf_join3(
				&key, '.', "branch", remote_name.ptr, "remote");
			git_buf_free(&remote_name);
865 866
		}

867 868
		if (!error)
			error = git_config__update_entry(cfg, key.ptr, sm->url, true, false);
869 870 871 872 873 874 875

		git_repository_free(smrepo);
	}

	git_buf_free(&key);

	return error;
Russell Belfer committed
876 877
}

878 879
static int git_submodule__open(
	git_repository **subrepo, git_submodule *sm, bool bare)
Russell Belfer committed
880 881 882
{
	int error;
	git_buf path = GIT_BUF_INIT;
883 884
	unsigned int flags = GIT_REPOSITORY_OPEN_NO_SEARCH;
	const char *wd;
Russell Belfer committed
885

886
	assert(sm && subrepo);
Russell Belfer committed
887

888 889 890
	if (git_repository__ensure_not_bare(
			sm->repo, "open submodule repository") < 0)
		return GIT_EBAREREPO;
Russell Belfer committed
891

892
	wd = git_repository_workdir(sm->repo);
Russell Belfer committed
893

894 895
	if (git_buf_joinpath(&path, wd, sm->path) < 0 ||
		git_buf_joinpath(&path, path.ptr, DOT_GIT) < 0)
Russell Belfer committed
896 897
		return -1;

898 899 900 901
	sm->flags = sm->flags &
		~(GIT_SUBMODULE_STATUS_IN_WD |
		  GIT_SUBMODULE_STATUS__WD_OID_VALID |
		  GIT_SUBMODULE_STATUS__WD_SCANNED);
Russell Belfer committed
902

903 904
	if (bare)
		flags |= GIT_REPOSITORY_OPEN_BARE;
Russell Belfer committed
905

906
	error = git_repository_open_ext(subrepo, path.ptr, flags, wd);
907

908 909 910 911
	/* if we opened the submodule successfully, grab HEAD OID, etc. */
	if (!error) {
		sm->flags |= GIT_SUBMODULE_STATUS_IN_WD |
			GIT_SUBMODULE_STATUS__WD_SCANNED;
912

913 914 915 916 917 918 919 920 921
		if (!git_reference_name_to_id(&sm->wd_oid, *subrepo, GIT_HEAD_FILE))
			sm->flags |= GIT_SUBMODULE_STATUS__WD_OID_VALID;
		else
			giterr_clear();
	} else if (git_path_exists(path.ptr)) {
		sm->flags |= GIT_SUBMODULE_STATUS__WD_SCANNED |
			GIT_SUBMODULE_STATUS_IN_WD;
	} else {
		git_buf_rtruncate_at_char(&path, '/'); /* remove "/.git" */
922

923 924 925
		if (git_path_isdir(path.ptr))
			sm->flags |= GIT_SUBMODULE_STATUS__WD_SCANNED;
	}
926

927
	git_buf_free(&path);
928

929 930
	return error;
}
931

932 933 934 935
int git_submodule_open_bare(git_repository **subrepo, git_submodule *sm)
{
	return git_submodule__open(subrepo, sm, true);
}
Russell Belfer committed
936

937 938 939
int git_submodule_open(git_repository **subrepo, git_submodule *sm)
{
	return git_submodule__open(subrepo, sm, false);
Russell Belfer committed
940 941
}

942
int git_submodule_reload_all(git_repository *repo, int force)
Russell Belfer committed
943
{
944
	return submodule_cache_init(repo, force ? CACHE_FLUSH : CACHE_REFRESH);
Russell Belfer committed
945 946
}

947 948
static void submodule_update_from_index_entry(
	git_submodule *sm, const git_index_entry *ie)
Russell Belfer committed
949
{
950
	bool already_found = (sm->flags & GIT_SUBMODULE_STATUS_IN_INDEX) != 0;
Russell Belfer committed
951

952 953 954 955 956 957 958
	if (!S_ISGITLINK(ie->mode)) {
		if (!already_found)
			sm->flags |= GIT_SUBMODULE_STATUS__INDEX_NOT_SUBMODULE;
	} else {
		if (already_found)
			sm->flags |= GIT_SUBMODULE_STATUS__INDEX_MULTIPLE_ENTRIES;
		else
959
			git_oid_cpy(&sm->index_oid, &ie->id);
Russell Belfer committed
960

961 962 963 964 965 966 967 968 969
		sm->flags |= GIT_SUBMODULE_STATUS_IN_INDEX |
			GIT_SUBMODULE_STATUS__INDEX_OID_VALID;
	}
}

static int submodule_update_index(git_submodule *sm)
{
	git_index *index;
	const git_index_entry *ie;
Russell Belfer committed
970

971
	if (git_repository_index__weakptr(&index, sm->repo) < 0)
Russell Belfer committed
972 973
		return -1;

974
	sm->flags = sm->flags &
975 976 977
		~(GIT_SUBMODULE_STATUS_IN_INDEX |
		  GIT_SUBMODULE_STATUS__INDEX_OID_VALID);

978 979
	if (!(ie = git_index_get_bypath(index, sm->path, 0)))
		return 0;
Russell Belfer committed
980

981 982 983 984 985 986 987 988 989 990 991 992 993 994 995
	submodule_update_from_index_entry(sm, ie);

	return 0;
}

static void submodule_update_from_head_data(
	git_submodule *sm, mode_t mode, const git_oid *id)
{
	if (!S_ISGITLINK(mode))
		sm->flags |= GIT_SUBMODULE_STATUS__HEAD_NOT_SUBMODULE;
	else {
		git_oid_cpy(&sm->head_oid, id);

		sm->flags |= GIT_SUBMODULE_STATUS_IN_HEAD |
			GIT_SUBMODULE_STATUS__HEAD_OID_VALID;
Russell Belfer committed
996
	}
997
}
Russell Belfer committed
998

999 1000 1001
static int submodule_update_head(git_submodule *submodule)
{
	git_tree *head = NULL;
nulltoken committed
1002
	git_tree_entry *te = NULL;
Russell Belfer committed
1003

1004 1005 1006
	submodule->flags = submodule->flags &
		~(GIT_SUBMODULE_STATUS_IN_HEAD |
		  GIT_SUBMODULE_STATUS__HEAD_OID_VALID);
Russell Belfer committed
1007

1008 1009 1010 1011 1012 1013
	/* if we can't look up file in current head, then done */
	if (git_repository_head_tree(&head, submodule->repo) < 0 ||
		git_tree_entry_bypath(&te, head, submodule->path) < 0)
		giterr_clear();
	else
		submodule_update_from_head_data(submodule, te->attr, &te->oid);
Russell Belfer committed
1014

nulltoken committed
1015
	git_tree_entry_free(te);
1016 1017 1018
	git_tree_free(head);
	return 0;
}
1019

1020

1021
int git_submodule_reload(git_submodule *sm, int force)
1022 1023 1024
{
	int error = 0;
	git_config_backend *mods;
1025
	git_submodule_cache *cache;
Russell Belfer committed
1026

1027 1028
	GIT_UNUSED(force);

1029 1030 1031
	assert(sm);

	cache = sm->repo->_submodules;
Russell Belfer committed
1032

1033
	/* refresh index data */
1034
	if ((error = submodule_update_index(sm)) < 0)
1035
		return error;
1036 1037

	/* refresh HEAD tree data */
1038
	if ((error = submodule_update_head(sm)) < 0)
1039
		return error;
Russell Belfer committed
1040

1041
	/* done if bare */
1042
	if (git_repository_is_bare(sm->repo))
1043 1044
		return error;

Russell Belfer committed
1045
	/* refresh config data */
1046
	mods = open_gitmodules(cache, GITMODULES_EXISTING);
1047
	if (mods != NULL) {
Russell Belfer committed
1048 1049 1050
		git_buf path = GIT_BUF_INIT;

		git_buf_sets(&path, "submodule\\.");
1051
		git_buf_text_puts_escape_regex(&path, sm->name);
Russell Belfer committed
1052 1053 1054 1055
		git_buf_puts(&path, ".*");

		if (git_buf_oom(&path))
			error = -1;
1056
		else
Russell Belfer committed
1057
			error = git_config_file_foreach_match(
1058
				mods, path.ptr, submodule_load_from_config, cache);
Russell Belfer committed
1059 1060

		git_buf_free(&path);
1061
		git_config_file_free(mods);
Russell Belfer committed
1062

1063 1064 1065
		if (error < 0)
			return error;
	}
1066 1067

	/* refresh wd data */
1068 1069 1070
	sm->flags &=
		~(GIT_SUBMODULE_STATUS_IN_WD | GIT_SUBMODULE_STATUS__WD_OID_VALID |
		  GIT_SUBMODULE_STATUS__WD_FLAGS);
1071

1072
	return submodule_load_from_wd_lite(sm);
Russell Belfer committed
1073 1074
}

1075 1076
static void submodule_copy_oid_maybe(
	git_oid *tgt, const git_oid *src, bool valid)
Russell Belfer committed
1077
{
1078 1079 1080 1081 1082 1083 1084
	if (tgt) {
		if (valid)
			memcpy(tgt, src, sizeof(*tgt));
		else
			memset(tgt, 0, sizeof(*tgt));
	}
}
1085

1086 1087 1088 1089 1090 1091 1092 1093 1094 1095
int git_submodule__status(
	unsigned int *out_status,
	git_oid *out_head_id,
	git_oid *out_index_id,
	git_oid *out_wd_id,
	git_submodule *sm,
	git_submodule_ignore_t ign)
{
	unsigned int status;
	git_repository *smrepo = NULL;
Russell Belfer committed
1096

1097
	if (ign < GIT_SUBMODULE_IGNORE_NONE)
1098
		ign = sm->ignore;
1099

1100 1101 1102 1103
	/* only return location info if ignore == all */
	if (ign == GIT_SUBMODULE_IGNORE_ALL) {
		*out_status = (sm->flags & GIT_SUBMODULE_STATUS__IN_FLAGS);
		return 0;
1104
	}
Russell Belfer committed
1105

1106 1107 1108 1109 1110 1111 1112
	/* refresh the index OID */
	if (submodule_update_index(sm) < 0)
		return -1;

	/* refresh the HEAD OID */
	if (submodule_update_head(sm) < 0)
		return -1;
Russell Belfer committed
1113

1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143
	/* for ignore == dirty, don't scan the working directory */
	if (ign == GIT_SUBMODULE_IGNORE_DIRTY) {
		/* git_submodule_open_bare will load WD OID data */
		if (git_submodule_open_bare(&smrepo, sm) < 0)
			giterr_clear();
		else
			git_repository_free(smrepo);
		smrepo = NULL;
	} else if (git_submodule_open(&smrepo, sm) < 0) {
		giterr_clear();
		smrepo = NULL;
	}

	status = GIT_SUBMODULE_STATUS__CLEAR_INTERNAL(sm->flags);

	submodule_get_index_status(&status, sm);
	submodule_get_wd_status(&status, sm, smrepo, ign);

	git_repository_free(smrepo);

	*out_status = status;

	submodule_copy_oid_maybe(out_head_id, &sm->head_oid,
		(sm->flags & GIT_SUBMODULE_STATUS__HEAD_OID_VALID) != 0);
	submodule_copy_oid_maybe(out_index_id, &sm->index_oid,
		(sm->flags & GIT_SUBMODULE_STATUS__INDEX_OID_VALID) != 0);
	submodule_copy_oid_maybe(out_wd_id, &sm->wd_oid,
		(sm->flags & GIT_SUBMODULE_STATUS__WD_OID_VALID) != 0);

	return 0;
Russell Belfer committed
1144 1145
}

1146
int git_submodule_status(unsigned int *status, git_submodule *sm)
1147
{
1148
	assert(status && sm);
1149

1150
	return git_submodule__status(status, NULL, NULL, NULL, sm, 0);
1151
}
1152

1153 1154 1155 1156 1157 1158
int git_submodule_location(unsigned int *location, git_submodule *sm)
{
	assert(location && sm);

	return git_submodule__status(
		location, NULL, NULL, NULL, sm, GIT_SUBMODULE_IGNORE_ALL);
1159 1160 1161
}


Russell Belfer committed
1162 1163 1164 1165
/*
 * INTERNAL FUNCTIONS
 */

1166 1167
static int submodule_alloc(
	git_submodule **out, git_submodule_cache *cache, const char *name)
1168
{
1169
	size_t namelen;
1170
	git_submodule *sm;
1171

1172
	if (!name || !(namelen = strlen(name))) {
1173
		giterr_set(GITERR_SUBMODULE, "Invalid submodule name");
1174
		return -1;
1175 1176
	}

1177
	sm = git__calloc(1, sizeof(git_submodule));
1178
	GITERR_CHECK_ALLOC(sm);
1179

1180 1181 1182
	sm->name = sm->path = git__strdup(name);
	if (!sm->name) {
		git__free(sm);
1183
		return -1;
1184
	}
Russell Belfer committed
1185

1186
	GIT_REFCOUNT_INC(sm);
1187 1188
	sm->ignore = sm->ignore_default = GIT_SUBMODULE_IGNORE_NONE;
	sm->update = sm->update_default = GIT_SUBMODULE_UPDATE_CHECKOUT;
1189
	sm->fetch_recurse = sm->fetch_recurse_default = GIT_SUBMODULE_RECURSE_NO;
1190
	sm->repo   = cache->repo;
1191
	sm->branch = NULL;
1192

1193 1194
	*out = sm;
	return 0;
Russell Belfer committed
1195 1196
}

1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
static void submodule_cache_remove_item(
	git_submodule_cache *cache,
	git_submodule *item,
	bool free_after_remove)
{
	git_strmap *map;
	const char *name, *alt;

	if (!cache || !(map = cache->submodules) || !item)
		return;

	name = item->name;
	alt  = (item->path != item->name) ? item->path : NULL;

	for (; name; name = alt, alt = NULL) {
		khiter_t pos = git_strmap_lookup_index(map, name);
		git_submodule *found;

		if (!git_strmap_valid_index(map, pos))
			continue;

		found = git_strmap_value_at(map, pos);

		if (found != item)
			continue;

		git_strmap_set_value_at(map, pos, NULL);
		git_strmap_delete_at(map, pos);

		if (free_after_remove)
			git_submodule_free(found);
	}
}

1231
static void submodule_release(git_submodule *sm)
Russell Belfer committed
1232 1233 1234 1235
{
	if (!sm)
		return;

1236 1237
	if (sm->repo) {
		git_submodule_cache *cache = sm->repo->_submodules;
1238
		sm->repo = NULL;
1239
		submodule_cache_remove_item(cache, sm, false);
1240 1241
	}

1242 1243 1244 1245
	if (sm->path != sm->name)
		git__free(sm->path);
	git__free(sm->name);
	git__free(sm->url);
1246
	git__free(sm->branch);
1247 1248 1249
	git__memzero(sm, sizeof(*sm));
	git__free(sm);
}
Russell Belfer committed
1250

1251 1252 1253 1254 1255
void git_submodule_free(git_submodule *sm)
{
	if (!sm)
		return;
	GIT_REFCOUNT_DEC(sm, submodule_release);
Russell Belfer committed
1256 1257
}

1258
static int submodule_get(
1259
	git_submodule **out,
1260
	git_submodule_cache *cache,
1261 1262
	const char *name,
	const char *alternate)
Russell Belfer committed
1263
{
1264
	int error = 0;
Russell Belfer committed
1265 1266 1267
	khiter_t pos;
	git_submodule *sm;

1268
	pos = git_strmap_lookup_index(cache->submodules, name);
Russell Belfer committed
1269

1270 1271
	if (!git_strmap_valid_index(cache->submodules, pos) && alternate)
		pos = git_strmap_lookup_index(cache->submodules, alternate);
Russell Belfer committed
1272

1273 1274 1275
	if (!git_strmap_valid_index(cache->submodules, pos)) {
		if ((error = submodule_alloc(&sm, cache, name)) < 0)
			return error;
Russell Belfer committed
1276

1277 1278 1279
		/* insert value at name - if another thread beats us to it, then use
		 * their record and release our own.
		 */
1280
		pos = kh_put(str, cache->submodules, sm->name, &error);
1281

1282 1283 1284
		if (error < 0)
			goto done;
		else if (error == 0) {
1285
			git_submodule_free(sm);
1286
			sm = git_strmap_value_at(cache->submodules, pos);
1287
		} else {
1288
			error = 0;
1289
			git_strmap_set_value_at(cache->submodules, pos, sm);
1290 1291
		}
	} else {
1292
		sm = git_strmap_value_at(cache->submodules, pos);
Russell Belfer committed
1293 1294
	}

1295 1296 1297 1298 1299 1300 1301
done:
	if (error < 0)
		git_submodule_free(sm);
	else if (out) {
		GIT_REFCOUNT_INC(sm);
		*out = sm;
	}
Russell Belfer committed
1302

1303
	return error;
1304 1305
}

1306 1307 1308 1309 1310
static int submodule_config_error(const char *property, const char *value)
{
	giterr_set(GITERR_INVALID,
		"Invalid value for submodule '%s' property: '%s'", property, value);
	return -1;
1311 1312
}

1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340
int git_submodule_parse_ignore(git_submodule_ignore_t *out, const char *value)
{
	int val;

	if (git_config_lookup_map_value(
			&val, _sm_ignore_map, ARRAY_SIZE(_sm_ignore_map), value) < 0) {
		*out = GIT_SUBMODULE_IGNORE_NONE;
		return submodule_config_error("ignore", value);
	}

	*out = (git_submodule_ignore_t)val;
	return 0;
}

int git_submodule_parse_update(git_submodule_update_t *out, const char *value)
{
	int val;

	if (git_config_lookup_map_value(
			&val, _sm_update_map, ARRAY_SIZE(_sm_update_map), value) < 0) {
		*out = GIT_SUBMODULE_UPDATE_CHECKOUT;
		return submodule_config_error("update", value);
	}

	*out = (git_submodule_update_t)val;
	return 0;
}

1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354
int git_submodule_parse_recurse(git_submodule_recurse_t *out, const char *value)
{
	int val;

	if (git_config_lookup_map_value(
			&val, _sm_recurse_map, ARRAY_SIZE(_sm_recurse_map), value) < 0) {
		*out = GIT_SUBMODULE_RECURSE_YES;
		return submodule_config_error("recurse", value);
	}

	*out = (git_submodule_recurse_t)val;
	return 0;
}

Russell Belfer committed
1355
static int submodule_load_from_config(
1356
	const git_config_entry *entry, void *payload)
1357
{
1358
	git_submodule_cache *cache = payload;
1359
	const char *namestart, *property;
1360
	const char *key = entry->name, *value = entry->value, *path;
1361
	char *alternate = NULL, *replaced = NULL;
1362
	git_buf name = GIT_BUF_INIT;
1363
	git_submodule *sm = NULL;
1364
	int error = 0;
1365 1366 1367 1368 1369 1370

	if (git__prefixcmp(key, "submodule.") != 0)
		return 0;

	namestart = key + strlen("submodule.");
	property  = strrchr(namestart, '.');
1371 1372

	if (!property || (property == namestart))
1373
		return 0;
1374

1375
	property++;
1376
	path = !strcasecmp(property, "path") ? value : NULL;
1377

1378
	if ((error = git_buf_set(&name, namestart, property - namestart - 1)) < 0 ||
1379
		(error = submodule_get(&sm, cache, name.ptr, path)) < 0)
1380
		goto done;
1381

Russell Belfer committed
1382 1383
	sm->flags |= GIT_SUBMODULE_STATUS_IN_CONFIG;

1384 1385 1386
	/* Only from config might we get differing names & paths.  If so, then
	 * update the submodule and insert under the alternative key.
	 */
1387

1388 1389 1390
	/* TODO: if case insensitive filesystem, then the following strcmps
	 * should be strcasecmp
	 */
1391

1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
	if (strcmp(sm->name, name.ptr) != 0) { /* name changed */
		if (!strcmp(sm->path, name.ptr)) { /* already set as path */
			replaced = sm->name;
			sm->name = sm->path;
		} else {
			if (sm->name != sm->path)
				replaced = sm->name;
			alternate = sm->name = git_buf_detach(&name);
		}
	}
	else if (path && strcmp(path, sm->path) != 0) { /* path changed */
		if (!strcmp(sm->name, value)) { /* already set as name */
			replaced = sm->path;
			sm->path = sm->name;
		} else {
			if (sm->path != sm->name)
				replaced = sm->path;
			if ((alternate = git__strdup(value)) == NULL) {
				error = -1;
				goto done;
			}
			sm->path = alternate;
1414
		}
1415
	}
1416

1417 1418 1419 1420 1421 1422 1423 1424
	/* Deregister under name being replaced */
	if (replaced) {
		git_strmap_delete(cache->submodules, replaced);
		git_submodule_free(sm);
		git__free(replaced);
	}

	/* Insert under alternate key */
1425 1426
	if (alternate) {
		void *old_sm = NULL;
1427
		git_strmap_insert2(cache->submodules, alternate, sm, old_sm, error);
1428

1429 1430
		if (error < 0)
			goto done;
1431 1432 1433 1434
		if (error > 0)
			error = 0;

		GIT_REFCOUNT_INC(sm); /* increase refcount for new key */
1435 1436 1437

		/* if we replaced an old module under this key, release the old one */
		if (old_sm && ((git_submodule *)old_sm) != sm) {
1438
			git_submodule_free(old_sm);
1439 1440
			/* TODO: log warning about multiple submodules with same path */
		}
1441 1442
	}

Russell Belfer committed
1443 1444 1445 1446
	/* TODO: Look up path in index and if it is present but not a GITLINK
	 * then this should be deleted (at least to match git's behavior)
	 */

1447
	if (path)
1448
		goto done;
1449 1450

	/* copy other properties into submodule entry */
Russell Belfer committed
1451
	if (strcasecmp(property, "url") == 0) {
1452 1453 1454
		git__free(sm->url);
		sm->url = NULL;

1455 1456 1457 1458
		if (value != NULL && (sm->url = git__strdup(value)) == NULL) {
			error = -1;
			goto done;
		}
1459
	}
1460 1461 1462 1463 1464 1465 1466 1467 1468
	else if (strcasecmp(property, "branch") == 0) {
		git__free(sm->branch);
		sm->branch = NULL;

		if (value != NULL && (sm->branch = git__strdup(value)) == NULL) {
			error = -1;
			goto done;
		}
	}
Russell Belfer committed
1469
	else if (strcasecmp(property, "update") == 0) {
1470 1471
		if ((error = git_submodule_parse_update(&sm->update, value)) < 0)
			goto done;
1472
		sm->update_default = sm->update;
1473
	}
Russell Belfer committed
1474
	else if (strcasecmp(property, "fetchRecurseSubmodules") == 0) {
1475 1476
		if ((error = git_submodule_parse_recurse(&sm->fetch_recurse, value)) < 0)
			goto done;
1477
		sm->fetch_recurse_default = sm->fetch_recurse;
1478
	}
Russell Belfer committed
1479
	else if (strcasecmp(property, "ignore") == 0) {
1480 1481
		if ((error = git_submodule_parse_ignore(&sm->ignore, value)) < 0)
			goto done;
1482
		sm->ignore_default = sm->ignore;
1483 1484 1485
	}
	/* ignore other unknown submodule properties */

1486
done:
1487
	git_submodule_free(sm); /* offset refcount inc from submodule_get() */
1488
	git_buf_free(&name);
1489
	return error;
1490 1491
}

1492
static int submodule_load_from_wd_lite(git_submodule *sm)
Russell Belfer committed
1493 1494 1495
{
	git_buf path = GIT_BUF_INIT;

1496
	if (git_buf_joinpath(&path, git_repository_workdir(sm->repo), sm->path) < 0)
1497
		return -1;
Russell Belfer committed
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508

	if (git_path_isdir(path.ptr))
		sm->flags |= GIT_SUBMODULE_STATUS__WD_SCANNED;

	if (git_path_contains(&path, DOT_GIT))
		sm->flags |= GIT_SUBMODULE_STATUS_IN_WD;

	git_buf_free(&path);
	return 0;
}

1509 1510
static int submodule_cache_refresh_from_index(
	git_submodule_cache *cache, git_index *idx)
1511 1512
{
	int error;
Russell Belfer committed
1513 1514
	git_iterator *i;
	const git_index_entry *entry;
1515

1516
	if ((error = git_iterator_for_index(&i, idx, 0, NULL, NULL)) < 0)
Russell Belfer committed
1517
		return error;
1518

1519
	while (!(error = git_iterator_advance(&entry, i))) {
1520
		khiter_t pos = git_strmap_lookup_index(cache->submodules, entry->path);
1521
		git_submodule *sm;
1522

1523 1524
		if (git_strmap_valid_index(cache->submodules, pos)) {
			sm = git_strmap_value_at(cache->submodules, pos);
1525

1526 1527 1528 1529 1530
			if (S_ISGITLINK(entry->mode))
				submodule_update_from_index_entry(sm, entry);
			else
				sm->flags |= GIT_SUBMODULE_STATUS__INDEX_NOT_SUBMODULE;
		} else if (S_ISGITLINK(entry->mode)) {
1531
			if (!submodule_get(&sm, cache, entry->path, NULL)) {
1532
				submodule_update_from_index_entry(sm, entry);
1533 1534
				git_submodule_free(sm);
			}
1535
		}
1536 1537
	}

1538 1539 1540
	if (error == GIT_ITEROVER)
		error = 0;

Russell Belfer committed
1541 1542 1543 1544 1545
	git_iterator_free(i);

	return error;
}

1546 1547
static int submodule_cache_refresh_from_head(
	git_submodule_cache *cache, git_tree *head)
Russell Belfer committed
1548 1549 1550 1551 1552
{
	int error;
	git_iterator *i;
	const git_index_entry *entry;

1553
	if ((error = git_iterator_for_tree(&i, head, 0, NULL, NULL)) < 0)
Russell Belfer committed
1554 1555
		return error;

1556
	while (!(error = git_iterator_advance(&entry, i))) {
1557
		khiter_t pos = git_strmap_lookup_index(cache->submodules, entry->path);
1558
		git_submodule *sm;
Russell Belfer committed
1559

1560 1561
		if (git_strmap_valid_index(cache->submodules, pos)) {
			sm = git_strmap_value_at(cache->submodules, pos);
1562

1563
			if (S_ISGITLINK(entry->mode))
1564
				submodule_update_from_head_data(sm, entry->mode, &entry->id);
1565 1566 1567
			else
				sm->flags |= GIT_SUBMODULE_STATUS__HEAD_NOT_SUBMODULE;
		} else if (S_ISGITLINK(entry->mode)) {
1568
			if (!submodule_get(&sm, cache, entry->path, NULL)) {
1569
				submodule_update_from_head_data(
1570
					sm, entry->mode, &entry->id);
1571 1572
				git_submodule_free(sm);
			}
1573
		}
Russell Belfer committed
1574 1575
	}

1576 1577 1578
	if (error == GIT_ITEROVER)
		error = 0;

Russell Belfer committed
1579 1580 1581 1582 1583
	git_iterator_free(i);

	return error;
}

Ben Straub committed
1584
static git_config_backend *open_gitmodules(
1585
	git_submodule_cache *cache,
1586
	int okay_to_create)
Russell Belfer committed
1587
{
1588
	const char *workdir = git_repository_workdir(cache->repo);
Russell Belfer committed
1589
	git_buf path = GIT_BUF_INIT;
Ben Straub committed
1590
	git_config_backend *mods = NULL;
Russell Belfer committed
1591 1592 1593 1594 1595 1596 1597 1598

	if (workdir != NULL) {
		if (git_buf_joinpath(&path, workdir, GIT_MODULES_FILE) != 0)
			return NULL;

		if (okay_to_create || git_path_isfile(path.ptr)) {
			/* git_config_file__ondisk should only fail if OOM */
			if (git_config_file__ondisk(&mods, path.ptr) < 0)
1599
				mods = NULL;
Russell Belfer committed
1600
			/* open should only fail here if the file is malformed */
1601
			else if (git_config_file_open(mods, GIT_CONFIG_LEVEL_LOCAL) < 0) {
Russell Belfer committed
1602 1603 1604
				git_config_file_free(mods);
				mods = NULL;
			}
1605 1606 1607
		}
	}

1608 1609
	git_buf_free(&path);

Russell Belfer committed
1610 1611 1612
	return mods;
}

1613
static void submodule_cache_free(git_submodule_cache *cache)
Russell Belfer committed
1614
{
1615
	git_submodule *sm;
Russell Belfer committed
1616

1617 1618
	if (!cache)
		return;
Russell Belfer committed
1619

1620 1621 1622 1623 1624
	git_strmap_foreach_value(cache->submodules, sm, {
		sm->repo = NULL; /* disconnect from repo */
		git_submodule_free(sm);
	});
	git_strmap_free(cache->submodules);
Russell Belfer committed
1625

1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
	git_buf_free(&cache->gitmodules_path);
	git_mutex_free(&cache->lock);
	git__free(cache);
}

static int submodule_cache_alloc(
	git_submodule_cache **out, git_repository *repo)
{
	git_submodule_cache *cache = git__calloc(1, sizeof(git_submodule_cache));
	GITERR_CHECK_ALLOC(cache);

	if (git_mutex_init(&cache->lock) < 0) {
		giterr_set(GITERR_OS, "Unable to initialize submodule cache lock");
		git__free(cache);
		return -1;
	}

1643
	if (git_strmap_alloc(&cache->submodules) < 0) {
1644 1645
		submodule_cache_free(cache);
		return -1;
1646 1647
	}

1648 1649 1650 1651 1652 1653 1654
	cache->repo = repo;
	git_buf_init(&cache->gitmodules_path, 0);

	*out = cache;
	return 0;
}

1655
static int submodule_cache_refresh(git_submodule_cache *cache, int refresh)
1656
{
1657
	int error = 0, update_index, update_head, update_gitmod;
1658 1659
	git_index *idx = NULL;
	git_tree *head = NULL;
1660
	const char *wd = NULL;
1661
	git_buf path = GIT_BUF_INIT;
1662 1663 1664
	git_submodule *sm;
	git_config_backend *mods = NULL;
	uint32_t mask;
1665

1666
	if (!cache || !cache->repo || !refresh)
1667 1668
		return 0;

1669 1670 1671 1672
	if (git_mutex_lock(&cache->lock) < 0) {
		giterr_set(GITERR_OS, "Unable to acquire lock on submodule cache");
		return -1;
	}
1673

1674
	/* get sources that we will need to check */
1675

1676 1677 1678 1679
	if (git_repository_index(&idx, cache->repo) < 0)
		giterr_clear();
	if (git_repository_head_tree(&head, cache->repo) < 0)
		giterr_clear();
Russell Belfer committed
1680

1681 1682 1683
	wd = git_repository_workdir(cache->repo);
	if (wd && (error = git_buf_joinpath(&path, wd, GIT_MODULES_FILE)) < 0)
		goto cleanup;
1684

1685
	/* check for invalidation */
1686

1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697
	if (refresh == CACHE_FLUSH)
		update_index = update_head = update_gitmod = true;
	else {
		update_index =
			!idx || git_index__changed_relative_to(idx, &cache->index_stamp);
		update_head =
			!head || !git_oid_equal(&cache->head_id, git_tree_id(head));

		update_gitmod = (wd != NULL) ?
			git_futils_filestamp_check(&cache->gitmodules_stamp, path.ptr) :
			(cache->gitmodules_stamp.mtime != 0);
1698
	}
Russell Belfer committed
1699

1700
	/* clear submodule flags that are to be refreshed */
Russell Belfer committed
1701

1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717
	mask = 0;
	if (!idx || update_index)
		mask |= GIT_SUBMODULE_STATUS_IN_INDEX |
			GIT_SUBMODULE_STATUS__INDEX_FLAGS |
			GIT_SUBMODULE_STATUS__INDEX_OID_VALID |
			GIT_SUBMODULE_STATUS__INDEX_MULTIPLE_ENTRIES;
	if (!head || update_head)
		mask |= GIT_SUBMODULE_STATUS_IN_HEAD |
			GIT_SUBMODULE_STATUS__HEAD_OID_VALID;
	if (update_gitmod)
		mask |= GIT_SUBMODULE_STATUS_IN_CONFIG;
	if (mask != 0)
		mask |= GIT_SUBMODULE_STATUS_IN_WD |
			GIT_SUBMODULE_STATUS__WD_SCANNED |
			GIT_SUBMODULE_STATUS__WD_FLAGS |
			GIT_SUBMODULE_STATUS__WD_OID_VALID;
1718 1719
	else
		goto cleanup; /* nothing to do */
1720

1721 1722 1723 1724 1725 1726 1727
	submodule_cache_clear_flags(cache, mask);

	/* add back submodule information from index */

	if (idx && update_index) {
		if ((error = submodule_cache_refresh_from_index(cache, idx)) < 0)
			goto cleanup;
1728

1729 1730
		git_futils_filestamp_set(
			&cache->index_stamp, git_index__filestamp(idx));
1731
	}
Russell Belfer committed
1732

1733
	/* add submodule information from HEAD */
Russell Belfer committed
1734

1735 1736 1737
	if (head && update_head) {
		if ((error = submodule_cache_refresh_from_head(cache, head)) < 0)
			goto cleanup;
1738

1739 1740
		git_oid_cpy(&cache->head_id, git_tree_id(head));
	}
Russell Belfer committed
1741

1742
	/* add submodule information from .gitmodules */
1743

1744 1745 1746
	if (wd && update_gitmod > 0) {
		if ((mods = open_gitmodules(cache, false)) != NULL &&
			(error = git_config_file_foreach(
1747 1748 1749 1750
				mods, submodule_load_from_config, cache)) < 0)
			goto cleanup;
	}

1751
	/* shallow scan submodules in work tree as needed */
1752

1753
	if (wd && mask != 0) {
1754
		git_strmap_foreach_value(cache->submodules, sm, {
1755 1756 1757
			submodule_load_from_wd_lite(sm);
		});
	}
1758

1759 1760 1761
	/* remove submodules that no longer exist */

	git_strmap_foreach_value(cache->submodules, sm, {
1762 1763 1764 1765 1766 1767
		/* purge unless in HEAD, index, or .gitmodules; no sm for wd only */
		if (sm != NULL &&
			!(sm->flags &
			 (GIT_SUBMODULE_STATUS_IN_HEAD |
			  GIT_SUBMODULE_STATUS_IN_INDEX |
			  GIT_SUBMODULE_STATUS_IN_CONFIG)))
1768 1769 1770
			submodule_cache_remove_item(cache, sm, true);
	});

1771
cleanup:
1772
	git_config_file_free(mods);
Russell Belfer committed
1773

1774 1775
	/* TODO: if we got an error, mark submodule config as invalid? */

1776 1777 1778 1779 1780 1781
	git_mutex_unlock(&cache->lock);

	git_index_free(idx);
	git_tree_free(head);
	git_buf_free(&path);

1782 1783 1784
	return error;
}

1785
static int submodule_cache_init(git_repository *repo, int cache_refresh)
1786 1787 1788 1789 1790 1791
{
	int error = 0;
	git_submodule_cache *cache = NULL;

	/* if submodules already exist, just refresh as requested */
	if (repo->_submodules)
1792
		return submodule_cache_refresh(repo->_submodules, cache_refresh);
1793 1794 1795

	/* otherwise create a new cache, load it, and atomically swap it in */
	if (!(error = submodule_cache_alloc(&cache, repo)) &&
1796
		!(error = submodule_cache_refresh(cache, CACHE_FLUSH)))
1797 1798 1799 1800 1801
		cache = git__compare_and_swap(&repo->_submodules, NULL, cache);

	/* might have raced with another thread to set cache, so free if needed */
	if (cache)
		submodule_cache_free(cache);
Russell Belfer committed
1802

1803 1804 1805
	return error;
}

Russell Belfer committed
1806 1807
/* Lookup name of remote of the local tracking branch HEAD points to */
static int lookup_head_remote_key(git_buf *remote_name, git_repository *repo)
1808
{
Russell Belfer committed
1809
	int error;
Russell Belfer committed
1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824
	git_reference *head = NULL;
	git_buf upstream_name = GIT_BUF_INIT;

	/* lookup and dereference HEAD */
	if ((error = git_repository_head(&head, repo)) < 0)
		return error;

	/* lookup remote tracking branch of HEAD */
	if (!(error = git_branch_upstream_name(
			&upstream_name, repo, git_reference_name(head))))
	{
		/* lookup remote of remote tracking branch */
		error = git_branch_remote_name(remote_name, repo, upstream_name.ptr);

		git_buf_free(&upstream_name);
1825
	}
Russell Belfer committed
1826

Russell Belfer committed
1827
	git_reference_free(head);
1828

1829 1830
	return error;
}
1831

Russell Belfer committed
1832 1833
/* Lookup the remote of the local tracking branch HEAD points to */
static int lookup_head_remote(git_remote **remote, git_repository *repo)
1834 1835
{
	int error;
Russell Belfer committed
1836
	git_buf remote_name = GIT_BUF_INIT;
1837

Russell Belfer committed
1838 1839 1840
	/* lookup remote of remote tracking branch name */
	if (!(error = lookup_head_remote_key(&remote_name, repo)))
		error = git_remote_load(remote, repo, remote_name.ptr);
1841

Russell Belfer committed
1842
	git_buf_free(&remote_name);
Russell Belfer committed
1843

1844 1845
	return error;
}
Russell Belfer committed
1846

Russell Belfer committed
1847 1848
/* Lookup remote, either from HEAD or fall back on origin */
static int lookup_default_remote(git_remote **remote, git_repository *repo)
1849
{
Russell Belfer committed
1850
	int error = lookup_head_remote(remote, repo);
Russell Belfer committed
1851

Russell Belfer committed
1852 1853 1854
	/* if that failed, use 'origin' instead */
	if (error == GIT_ENOTFOUND)
		error = git_remote_load(remote, repo, "origin");
Russell Belfer committed
1855

Russell Belfer committed
1856 1857 1858 1859 1860
	if (error == GIT_ENOTFOUND)
		giterr_set(
			GITERR_SUBMODULE,
			"Cannot get default remote for submodule - no local tracking "
			"branch for HEAD and origin does not exist");
Russell Belfer committed
1861 1862

	return error;
1863 1864
}

Russell Belfer committed
1865
static int get_url_base(git_buf *url, git_repository *repo)
1866 1867
{
	int error;
Russell Belfer committed
1868
	git_remote *remote = NULL;
1869

Russell Belfer committed
1870 1871 1872 1873 1874 1875 1876 1877 1878
	if (!(error = lookup_default_remote(&remote, repo))) {
		error = git_buf_sets(url, git_remote_url(remote));
		git_remote_free(remote);
	}
	else if (error == GIT_ENOTFOUND) {
		/* if repository does not have a default remote, use workdir instead */
		giterr_clear();
		error = git_buf_sets(url, git_repository_workdir(repo));
	}
1879 1880 1881 1882

	return error;
}

1883
static void submodule_get_index_status(unsigned int *status, git_submodule *sm)
Russell Belfer committed
1884
{
1885 1886
	const git_oid *head_oid  = git_submodule_head_id(sm);
	const git_oid *index_oid = git_submodule_index_id(sm);
Russell Belfer committed
1887

1888 1889
	*status = *status & ~GIT_SUBMODULE_STATUS__INDEX_FLAGS;

1890 1891 1892 1893 1894 1895 1896 1897
	if (!head_oid) {
		if (index_oid)
			*status |= GIT_SUBMODULE_STATUS_INDEX_ADDED;
	}
	else if (!index_oid)
		*status |= GIT_SUBMODULE_STATUS_INDEX_DELETED;
	else if (!git_oid_equal(head_oid, index_oid))
		*status |= GIT_SUBMODULE_STATUS_INDEX_MODIFIED;
Russell Belfer committed
1898 1899
}

1900 1901 1902 1903 1904
static void submodule_get_wd_status(
	unsigned int *status,
	git_submodule *sm,
	git_repository *sm_repo,
	git_submodule_ignore_t ign)
1905
{
1906 1907 1908 1909
	const git_oid *index_oid = git_submodule_index_id(sm);
	const git_oid *wd_oid =
		(sm->flags & GIT_SUBMODULE_STATUS__WD_OID_VALID) ? &sm->wd_oid : NULL;
	git_tree *sm_head = NULL;
1910
	git_index *index = NULL;
1911
	git_diff_options opt = GIT_DIFF_OPTIONS_INIT;
1912
	git_diff *diff;
1913

1914
	*status = *status & ~GIT_SUBMODULE_STATUS__WD_FLAGS;
Russell Belfer committed
1915

1916 1917 1918
	if (!index_oid) {
		if (wd_oid)
			*status |= GIT_SUBMODULE_STATUS_WD_ADDED;
Russell Belfer committed
1919
	}
1920 1921 1922 1923 1924 1925
	else if (!wd_oid) {
		if ((sm->flags & GIT_SUBMODULE_STATUS__WD_SCANNED) != 0 &&
			(sm->flags & GIT_SUBMODULE_STATUS_IN_WD) == 0)
			*status |= GIT_SUBMODULE_STATUS_WD_UNINITIALIZED;
		else
			*status |= GIT_SUBMODULE_STATUS_WD_DELETED;
Russell Belfer committed
1926
	}
1927 1928
	else if (!git_oid_equal(index_oid, wd_oid))
		*status |= GIT_SUBMODULE_STATUS_WD_MODIFIED;
Russell Belfer committed
1929

1930 1931 1932
	/* if we have no repo, then we're done */
	if (!sm_repo)
		return;
1933

1934 1935 1936 1937
	/* the diffs below could be optimized with an early termination
	 * option to the git_diff functions, but for now this is sufficient
	 * (and certainly no worse that what core git does).
	 */
1938

1939 1940
	if (ign == GIT_SUBMODULE_IGNORE_NONE)
		opt.flags |= GIT_DIFF_INCLUDE_UNTRACKED;
Russell Belfer committed
1941

1942 1943
	(void)git_repository_index__weakptr(&index, sm_repo);

1944
	/* if we don't have an unborn head, check diff with index */
1945 1946 1947 1948
	if (git_repository_head_tree(&sm_head, sm_repo) < 0)
		giterr_clear();
	else {
		/* perform head to index diff on submodule */
1949
		if (git_diff_tree_to_index(&diff, sm_repo, sm_head, index, &opt) < 0)
1950 1951
			giterr_clear();
		else {
1952
			if (git_diff_num_deltas(diff) > 0)
1953
				*status |= GIT_SUBMODULE_STATUS_WD_INDEX_MODIFIED;
1954
			git_diff_free(diff);
1955 1956
			diff = NULL;
		}
Russell Belfer committed
1957

1958
		git_tree_free(sm_head);
1959
	}
1960

1961
	/* perform index-to-workdir diff on submodule */
1962
	if (git_diff_index_to_workdir(&diff, sm_repo, index, &opt) < 0)
1963 1964 1965 1966
		giterr_clear();
	else {
		size_t untracked =
			git_diff_num_deltas_of_type(diff, GIT_DELTA_UNTRACKED);
1967

1968 1969
		if (untracked > 0)
			*status |= GIT_SUBMODULE_STATUS_WD_UNTRACKED;
1970

1971 1972
		if (git_diff_num_deltas(diff) != untracked)
			*status |= GIT_SUBMODULE_STATUS_WD_WD_MODIFIED;
Russell Belfer committed
1973

1974
		git_diff_free(diff);
1975
		diff = NULL;
1976
	}
1977
}