Blame src/tree-cache.c

Packit Service 20376f
/*
Packit Service 20376f
 * Copyright (C) the libgit2 contributors. All rights reserved.
Packit Service 20376f
 *
Packit Service 20376f
 * This file is part of libgit2, distributed under the GNU GPL v2 with
Packit Service 20376f
 * a Linking Exception. For full terms see the included COPYING file.
Packit Service 20376f
 */
Packit Service 20376f
Packit Service 20376f
#include "tree-cache.h"
Packit Service 20376f
#include "pool.h"
Packit Service 20376f
#include "tree.h"
Packit Service 20376f
Packit Service 20376f
static git_tree_cache *find_child(
Packit Service 20376f
	const git_tree_cache *tree, const char *path, const char *end)
Packit Service 20376f
{
Packit Service 20376f
	size_t i, dirlen = end ? (size_t)(end - path) : strlen(path);
Packit Service 20376f
Packit Service 20376f
	for (i = 0; i < tree->children_count; ++i) {
Packit Service 20376f
		git_tree_cache *child = tree->children[i];
Packit Service 20376f
Packit Service 20376f
		if (child->namelen == dirlen && !memcmp(path, child->name, dirlen))
Packit Service 20376f
			return child;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	return NULL;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
void git_tree_cache_invalidate_path(git_tree_cache *tree, const char *path)
Packit Service 20376f
{
Packit Service 20376f
	const char *ptr = path, *end;
Packit Service 20376f
Packit Service 20376f
	if (tree == NULL)
Packit Service 20376f
		return;
Packit Service 20376f
Packit Service 20376f
	tree->entry_count = -1;
Packit Service 20376f
Packit Service 20376f
	while (ptr != NULL) {
Packit Service 20376f
		end = strchr(ptr, '/');
Packit Service 20376f
Packit Service 20376f
		if (end == NULL) /* End of path */
Packit Service 20376f
			break;
Packit Service 20376f
Packit Service 20376f
		tree = find_child(tree, ptr, end);
Packit Service 20376f
		if (tree == NULL) /* We don't have that tree */
Packit Service 20376f
			return;
Packit Service 20376f
Packit Service 20376f
		tree->entry_count = -1;
Packit Service 20376f
		ptr = end + 1;
Packit Service 20376f
	}
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
const git_tree_cache *git_tree_cache_get(const git_tree_cache *tree, const char *path)
Packit Service 20376f
{
Packit Service 20376f
	const char *ptr = path, *end;
Packit Service 20376f
Packit Service 20376f
	if (tree == NULL) {
Packit Service 20376f
		return NULL;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	while (1) {
Packit Service 20376f
		end = strchr(ptr, '/');
Packit Service 20376f
Packit Service 20376f
		tree = find_child(tree, ptr, end);
Packit Service 20376f
		if (tree == NULL) /* Can't find it */
Packit Service 20376f
			return NULL;
Packit Service 20376f
Packit Service 20376f
		if (end == NULL || *end + 1 == '\0')
Packit Service 20376f
			return tree;
Packit Service 20376f
Packit Service 20376f
		ptr = end + 1;
Packit Service 20376f
	}
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
static int read_tree_internal(git_tree_cache **out,
Packit Service 20376f
			      const char **buffer_in, const char *buffer_end,
Packit Service 20376f
			      git_pool *pool)
Packit Service 20376f
{
Packit Service 20376f
	git_tree_cache *tree = NULL;
Packit Service 20376f
	const char *name_start, *buffer;
Packit Service 20376f
	int count;
Packit Service 20376f
Packit Service 20376f
	buffer = name_start = *buffer_in;
Packit Service 20376f
Packit Service 20376f
	if ((buffer = memchr(buffer, '\0', buffer_end - buffer)) == NULL)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	if (++buffer >= buffer_end)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	if (git_tree_cache_new(&tree, name_start, pool) < 0)
Packit Service 20376f
		return -1;
Packit Service 20376f
Packit Service 20376f
	/* Blank-terminated ASCII decimal number of entries in this tree */
Packit Service 20376f
	if (git__strntol32(&count, buffer, buffer_end - buffer, &buffer, 10) < 0)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	tree->entry_count = count;
Packit Service 20376f
Packit Service 20376f
	if (*buffer != ' ' || ++buffer >= buffer_end)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	 /* Number of children of the tree, newline-terminated */
Packit Service 20376f
	if (git__strntol32(&count, buffer, buffer_end - buffer, &buffer, 10) < 0 || count < 0)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	tree->children_count = count;
Packit Service 20376f
Packit Service 20376f
	if (*buffer != '\n' || ++buffer > buffer_end)
Packit Service 20376f
		goto corrupted;
Packit Service 20376f
Packit Service 20376f
	/* The SHA1 is only there if it's not invalidated */
Packit Service 20376f
	if (tree->entry_count >= 0) {
Packit Service 20376f
		/* 160-bit SHA-1 for this tree and it's children */
Packit Service 20376f
		if (buffer + GIT_OID_RAWSZ > buffer_end)
Packit Service 20376f
			goto corrupted;
Packit Service 20376f
Packit Service 20376f
		git_oid_fromraw(&tree->oid, (const unsigned char *)buffer);
Packit Service 20376f
		buffer += GIT_OID_RAWSZ;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	/* Parse children: */
Packit Service 20376f
	if (tree->children_count > 0) {
Packit Service 20376f
		unsigned int i;
Packit Service 20376f
Packit Service 20376f
		tree->children = git_pool_malloc(pool, tree->children_count * sizeof(git_tree_cache *));
Packit Service 20376f
		GITERR_CHECK_ALLOC(tree->children);
Packit Service 20376f
Packit Service 20376f
		memset(tree->children, 0x0, tree->children_count * sizeof(git_tree_cache *));
Packit Service 20376f
Packit Service 20376f
		for (i = 0; i < tree->children_count; ++i) {
Packit Service 20376f
			if (read_tree_internal(&tree->children[i], &buffer, buffer_end, pool) < 0)
Packit Service 20376f
				goto corrupted;
Packit Service 20376f
		}
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	*buffer_in = buffer;
Packit Service 20376f
	*out = tree;
Packit Service 20376f
	return 0;
Packit Service 20376f
Packit Service 20376f
 corrupted:
Packit Service 20376f
	giterr_set(GITERR_INDEX, "corrupted TREE extension in index");
Packit Service 20376f
	return -1;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
int git_tree_cache_read(git_tree_cache **tree, const char *buffer, size_t buffer_size, git_pool *pool)
Packit Service 20376f
{
Packit Service 20376f
	const char *buffer_end = buffer + buffer_size;
Packit Service 20376f
Packit Service 20376f
	if (read_tree_internal(tree, &buffer, buffer_end, pool) < 0)
Packit Service 20376f
		return -1;
Packit Service 20376f
Packit Service 20376f
	if (buffer < buffer_end) {
Packit Service 20376f
		giterr_set(GITERR_INDEX, "corrupted TREE extension in index (unexpected trailing data)");
Packit Service 20376f
		return -1;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	return 0;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
static int read_tree_recursive(git_tree_cache *cache, const git_tree *tree, git_pool *pool)
Packit Service 20376f
{
Packit Service 20376f
	git_repository *repo;
Packit Service 20376f
	size_t i, j, nentries, ntrees;
Packit Service 20376f
	int error;
Packit Service 20376f
Packit Service 20376f
	repo = git_tree_owner(tree);
Packit Service 20376f
Packit Service 20376f
	git_oid_cpy(&cache->oid, git_tree_id(tree));
Packit Service 20376f
	nentries = git_tree_entrycount(tree);
Packit Service 20376f
Packit Service 20376f
	/*
Packit Service 20376f
	 * We make sure we know how many trees we need to allocate for
Packit Service 20376f
	 * so we don't have to realloc and change the pointers for the
Packit Service 20376f
	 * parents.
Packit Service 20376f
	 */
Packit Service 20376f
	ntrees = 0;
Packit Service 20376f
	for (i = 0; i < nentries; i++) {
Packit Service 20376f
		const git_tree_entry *entry;
Packit Service 20376f
Packit Service 20376f
		entry = git_tree_entry_byindex(tree, i);
Packit Service 20376f
		if (git_tree_entry_filemode(entry) == GIT_FILEMODE_TREE)
Packit Service 20376f
			ntrees++;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	cache->children_count = ntrees;
Packit Service 20376f
	cache->children = git_pool_mallocz(pool, ntrees * sizeof(git_tree_cache *));
Packit Service 20376f
	GITERR_CHECK_ALLOC(cache->children);
Packit Service 20376f
Packit Service 20376f
	j = 0;
Packit Service 20376f
	for (i = 0; i < nentries; i++) {
Packit Service 20376f
		const git_tree_entry *entry;
Packit Service 20376f
		git_tree *subtree;
Packit Service 20376f
Packit Service 20376f
		entry = git_tree_entry_byindex(tree, i);
Packit Service 20376f
		if (git_tree_entry_filemode(entry) != GIT_FILEMODE_TREE) {
Packit Service 20376f
			cache->entry_count++;
Packit Service 20376f
			continue;
Packit Service 20376f
		}
Packit Service 20376f
Packit Service 20376f
		if ((error = git_tree_cache_new(&cache->children[j], git_tree_entry_name(entry), pool)) < 0)
Packit Service 20376f
			return error;
Packit Service 20376f
Packit Service 20376f
		if ((error = git_tree_lookup(&subtree, repo, git_tree_entry_id(entry))) < 0)
Packit Service 20376f
			return error;
Packit Service 20376f
Packit Service 20376f
		error = read_tree_recursive(cache->children[j], subtree, pool);
Packit Service 20376f
		git_tree_free(subtree);
Packit Service 20376f
		cache->entry_count += cache->children[j]->entry_count;
Packit Service 20376f
		j++;
Packit Service 20376f
Packit Service 20376f
		if (error < 0)
Packit Service 20376f
			return error;
Packit Service 20376f
	}
Packit Service 20376f
Packit Service 20376f
	return 0;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
int git_tree_cache_read_tree(git_tree_cache **out, const git_tree *tree, git_pool *pool)
Packit Service 20376f
{
Packit Service 20376f
	int error;
Packit Service 20376f
	git_tree_cache *cache;
Packit Service 20376f
Packit Service 20376f
	if ((error = git_tree_cache_new(&cache, "", pool)) < 0)
Packit Service 20376f
		return error;
Packit Service 20376f
Packit Service 20376f
	if ((error = read_tree_recursive(cache, tree, pool)) < 0)
Packit Service 20376f
		return error;
Packit Service 20376f
Packit Service 20376f
	*out = cache;
Packit Service 20376f
	return 0;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
int git_tree_cache_new(git_tree_cache **out, const char *name, git_pool *pool)
Packit Service 20376f
{
Packit Service 20376f
	size_t name_len;
Packit Service 20376f
	git_tree_cache *tree;
Packit Service 20376f
Packit Service 20376f
	name_len = strlen(name);
Packit Service 20376f
	tree = git_pool_malloc(pool, sizeof(git_tree_cache) + name_len + 1);
Packit Service 20376f
	GITERR_CHECK_ALLOC(tree);
Packit Service 20376f
Packit Service 20376f
	memset(tree, 0x0, sizeof(git_tree_cache));
Packit Service 20376f
	/* NUL-terminated tree name */
Packit Service 20376f
	tree->namelen = name_len;
Packit Service 20376f
	memcpy(tree->name, name, name_len);
Packit Service 20376f
	tree->name[name_len] = '\0';
Packit Service 20376f
Packit Service 20376f
	*out = tree;
Packit Service 20376f
	return 0;
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
static void write_tree(git_buf *out, git_tree_cache *tree)
Packit Service 20376f
{
Packit Service 20376f
	size_t i;
Packit Service 20376f
Packit Service 20376f
	git_buf_printf(out, "%s%c%"PRIdZ" %"PRIuZ"\n", tree->name, 0, tree->entry_count, tree->children_count);
Packit Service 20376f
Packit Service 20376f
	if (tree->entry_count != -1)
Packit Service 20376f
		git_buf_put(out, (const char *) &tree->oid, GIT_OID_RAWSZ);
Packit Service 20376f
Packit Service 20376f
	for (i = 0; i < tree->children_count; i++)
Packit Service 20376f
		write_tree(out, tree->children[i]);
Packit Service 20376f
}
Packit Service 20376f
Packit Service 20376f
int git_tree_cache_write(git_buf *out, git_tree_cache *tree)
Packit Service 20376f
{
Packit Service 20376f
	write_tree(out, tree);
Packit Service 20376f
Packit Service 20376f
	return git_buf_oom(out) ? -1 : 0;
Packit Service 20376f
}