git / git

Cannot retrieve contributors at this time

507 lines (399 sloc) 11.7 KB

	#include "cache.h"
	#include "attr.h"
	#include "object.h"
	#include "blob.h"
	#include "commit.h"
	#include "tag.h"
	#include "tree.h"
	#include "delta.h"
	#include "pack.h"
	#include "tree-walk.h"
	#include "diff.h"
	#include "revision.h"
	#include "list-objects.h"
	#include "progress.h"
	#include "refs.h"
	#include "khash.h"
	#include "pack-bitmap.h"
	#include "pack-objects.h"
	#include "delta-islands.h"
	#include "oid-array.h"
	#include "config.h"

	KHASH_INIT(str, const char , void , 1, kh_str_hash_func, kh_str_hash_equal)

	static kh_oid_map_t *island_marks;
	static unsigned island_counter;
	static unsigned island_counter_core;

	static kh_str_t *remote_islands;

	struct remote_island {
	uint64_t hash;
	struct oid_array oids;
	};

	struct island_bitmap {
	uint32_t refcount;
	uint32_t bits[FLEX_ARRAY];
	};

	static uint32_t island_bitmap_size;

	/*
	* Allocate a new bitmap; if "old" is not NULL, the new bitmap will be a copy
	* of "old". Otherwise, the new bitmap is empty.
	*/
	static struct island_bitmap island_bitmap_new(const struct island_bitmap old)
	{
	size_t size = sizeof(struct island_bitmap) + (island_bitmap_size * 4);
	struct island_bitmap *b = xcalloc(1, size);

	if (old)
	memcpy(b, old, size);

	b->refcount = 1;
	return b;
	}

	static void island_bitmap_or(struct island_bitmap a, const struct island_bitmap b)
	{
	uint32_t i;

	for (i = 0; i < island_bitmap_size; ++i)
	a->bits[i] \|= b->bits[i];
	}

	static int island_bitmap_is_subset(struct island_bitmap *self,
	struct island_bitmap *super)
	{
	uint32_t i;

	if (self == super)
	return 1;

	for (i = 0; i < island_bitmap_size; ++i) {
	if ((self->bits[i] & super->bits[i]) != self->bits[i])
	return 0;
	}

	return 1;
	}

	#define ISLAND_BITMAP_BLOCK(x) (x / 32)
	#define ISLAND_BITMAP_MASK(x) (1 << (x % 32))

	static void island_bitmap_set(struct island_bitmap *self, uint32_t i)
	{
	self->bits[ISLAND_BITMAP_BLOCK(i)] \|= ISLAND_BITMAP_MASK(i);
	}

	static int island_bitmap_get(struct island_bitmap *self, uint32_t i)
	{
	return (self->bits[ISLAND_BITMAP_BLOCK(i)] & ISLAND_BITMAP_MASK(i)) != 0;
	}

	int in_same_island(const struct object_id trg_oid, const struct object_id src_oid)
	{
	khiter_t trg_pos, src_pos;

	/* If we aren't using islands, assume everything goes together. */
	if (!island_marks)
	return 1;

	/*
	* If we don't have a bitmap for the target, we can delta it
	* against anything -- it's not an important object
	*/
	trg_pos = kh_get_oid_map(island_marks, *trg_oid);
	if (trg_pos >= kh_end(island_marks))
	return 1;

	/*
	* if the source (our delta base) doesn't have a bitmap,
	* we don't want to base any deltas on it!
	*/
	src_pos = kh_get_oid_map(island_marks, *src_oid);
	if (src_pos >= kh_end(island_marks))
	return 0;

	return island_bitmap_is_subset(kh_value(island_marks, trg_pos),
	kh_value(island_marks, src_pos));
	}

	int island_delta_cmp(const struct object_id a, const struct object_id b)
	{
	khiter_t a_pos, b_pos;
	struct island_bitmap a_bitmap = NULL, b_bitmap = NULL;

	if (!island_marks)
	return 0;

	a_pos = kh_get_oid_map(island_marks, *a);
	if (a_pos < kh_end(island_marks))
	a_bitmap = kh_value(island_marks, a_pos);

	b_pos = kh_get_oid_map(island_marks, *b);
	if (b_pos < kh_end(island_marks))
	b_bitmap = kh_value(island_marks, b_pos);

	if (a_bitmap) {
	if (!b_bitmap \|\| !island_bitmap_is_subset(a_bitmap, b_bitmap))
	return -1;
	}
	if (b_bitmap) {
	if (!a_bitmap \|\| !island_bitmap_is_subset(b_bitmap, a_bitmap))
	return 1;
	}

	return 0;
	}

	static struct island_bitmap create_or_get_island_marks(struct object obj)
	{
	khiter_t pos;
	int hash_ret;

	pos = kh_put_oid_map(island_marks, obj->oid, &hash_ret);
	if (hash_ret)
	kh_value(island_marks, pos) = island_bitmap_new(NULL);

	return kh_value(island_marks, pos);
	}

	static void set_island_marks(struct object obj, struct island_bitmap marks)
	{
	struct island_bitmap *b;
	khiter_t pos;
	int hash_ret;

	pos = kh_put_oid_map(island_marks, obj->oid, &hash_ret);
	if (hash_ret) {
	/*
	* We don't have one yet; make a copy-on-write of the
	* parent.
	*/
	marks->refcount++;
	kh_value(island_marks, pos) = marks;
	return;
	}

	/*
	* We do have it. Make sure we split any copy-on-write before
	* updating.
	*/
	b = kh_value(island_marks, pos);
	if (b->refcount > 1) {
	b->refcount--;
	b = kh_value(island_marks, pos) = island_bitmap_new(b);
	}
	island_bitmap_or(b, marks);
	}

	static void mark_remote_island_1(struct repository *r,
	struct remote_island *rl,
	int is_core_island)
	{
	uint32_t i;

	for (i = 0; i < rl->oids.nr; ++i) {
	struct island_bitmap *marks;
	struct object *obj = parse_object(r, &rl->oids.oid[i]);

	if (!obj)
	continue;

	marks = create_or_get_island_marks(obj);
	island_bitmap_set(marks, island_counter);

	if (is_core_island && obj->type == OBJ_COMMIT)
	obj->flags \|= NEEDS_BITMAP;

	/* If it was a tag, also make sure we hit the underlying object. */
	while (obj && obj->type == OBJ_TAG) {
	obj = ((struct tag *)obj)->tagged;
	if (obj) {
	parse_object(r, &obj->oid);
	marks = create_or_get_island_marks(obj);
	island_bitmap_set(marks, island_counter);
	}
	}
	}

	if (is_core_island)
	island_counter_core = island_counter;

	island_counter++;
	}

	struct tree_islands_todo {
	struct object_entry *entry;
	unsigned int depth;
	};

	static int tree_depth_compare(const void a, const void b)
	{
	const struct tree_islands_todo *todo_a = a;
	const struct tree_islands_todo *todo_b = b;

	return todo_a->depth - todo_b->depth;
	}

	void resolve_tree_islands(struct repository *r,
	int progress,
	struct packing_data *to_pack)
	{
	struct progress *progress_state = NULL;
	struct tree_islands_todo *todo;
	int nr = 0;
	int i;

	if (!island_marks)
	return;

	/*
	* We process only trees, as commits and tags have already been handled
	* (and passed their marks on to root trees, as well. We must make sure
	* to process them in descending tree-depth order so that marks
	* propagate down the tree properly, even if a sub-tree is found in
	* multiple parent trees.
	*/
	ALLOC_ARRAY(todo, to_pack->nr_objects);
	for (i = 0; i < to_pack->nr_objects; i++) {
	if (oe_type(&to_pack->objects[i]) == OBJ_TREE) {
	todo[nr].entry = &to_pack->objects[i];
	todo[nr].depth = oe_tree_depth(to_pack, &to_pack->objects[i]);
	nr++;
	}
	}
	QSORT(todo, nr, tree_depth_compare);

	if (progress)
	progress_state = start_progress(_("Propagating island marks"), nr);

	for (i = 0; i < nr; i++) {
	struct object_entry *ent = todo[i].entry;
	struct island_bitmap *root_marks;
	struct tree *tree;
	struct tree_desc desc;
	struct name_entry entry;
	khiter_t pos;

	pos = kh_get_oid_map(island_marks, ent->idx.oid);
	if (pos >= kh_end(island_marks))
	continue;

	root_marks = kh_value(island_marks, pos);

	tree = lookup_tree(r, &ent->idx.oid);
	if (!tree \|\| parse_tree(tree) < 0)
	die(_("bad tree object %s"), oid_to_hex(&ent->idx.oid));

	init_tree_desc(&desc, tree->buffer, tree->size);
	while (tree_entry(&desc, &entry)) {
	struct object *obj;

	if (S_ISGITLINK(entry.mode))
	continue;

	obj = lookup_object(r, &entry.oid);
	if (!obj)
	continue;

	set_island_marks(obj, root_marks);
	}

	free_tree_buffer(tree);

	display_progress(progress_state, i+1);
	}

	stop_progress(&progress_state);
	free(todo);
	}

	static regex_t *island_regexes;
	static unsigned int island_regexes_alloc, island_regexes_nr;
	static const char *core_island_name;

	static int island_config_callback(const char k, const char v, void *cb)
	{
	if (!strcmp(k, "pack.island")) {
	struct strbuf re = STRBUF_INIT;

	if (!v)
	return config_error_nonbool(k);

	ALLOC_GROW(island_regexes, island_regexes_nr + 1, island_regexes_alloc);

	if (*v != '^')
	strbuf_addch(&re, '^');
	strbuf_addstr(&re, v);

	if (regcomp(&island_regexes[island_regexes_nr], re.buf, REG_EXTENDED))
	die(_("failed to load island regex for '%s': %s"), k, re.buf);

	strbuf_release(&re);
	island_regexes_nr++;
	return 0;
	}

	if (!strcmp(k, "pack.islandcore"))
	return git_config_string(&core_island_name, k, v);

	return 0;
	}

	static void add_ref_to_island(const char island_name, const struct object_id oid)
	{
	uint64_t sha_core;
	struct remote_island *rl = NULL;

	int hash_ret;
	khiter_t pos = kh_put_str(remote_islands, island_name, &hash_ret);

	if (hash_ret) {
	kh_key(remote_islands, pos) = xstrdup(island_name);
	kh_value(remote_islands, pos) = xcalloc(1, sizeof(struct remote_island));
	}

	rl = kh_value(remote_islands, pos);
	oid_array_append(&rl->oids, oid);

	memcpy(&sha_core, oid->hash, sizeof(uint64_t));
	rl->hash += sha_core;
	}

	static int find_island_for_ref(const char refname, const struct object_id oid,
	int flags, void *data)
	{
	/*
	* We should advertise 'ARRAY_SIZE(matches) - 2' as the max,
	* so we can diagnose below a config with more capture groups
	* than we support.
	*/
	regmatch_t matches[16];
	int i, m;
	struct strbuf island_name = STRBUF_INIT;

	/* walk backwards to get last-one-wins ordering */
	for (i = island_regexes_nr - 1; i >= 0; i--) {
	if (!regexec(&island_regexes[i], refname,
	ARRAY_SIZE(matches), matches, 0))
	break;
	}

	if (i < 0)
	return 0;

	if (matches[ARRAY_SIZE(matches) - 1].rm_so != -1)
	warning(_("island regex from config has "
	"too many capture groups (max=%d)"),
	(int)ARRAY_SIZE(matches) - 2);

	for (m = 1; m < ARRAY_SIZE(matches); m++) {
	regmatch_t *match = &matches[m];

	if (match->rm_so == -1)
	continue;

	if (island_name.len)
	strbuf_addch(&island_name, '-');

	strbuf_add(&island_name, refname + match->rm_so, match->rm_eo - match->rm_so);
	}

	add_ref_to_island(island_name.buf, oid);
	strbuf_release(&island_name);
	return 0;
	}

	static struct remote_island *get_core_island(void)
	{
	if (core_island_name) {
	khiter_t pos = kh_get_str(remote_islands, core_island_name);
	if (pos < kh_end(remote_islands))
	return kh_value(remote_islands, pos);
	}

	return NULL;
	}

	static void deduplicate_islands(struct repository *r)
	{
	struct remote_island island, core = NULL, **list;
	unsigned int island_count, dst, src, ref, i = 0;

	island_count = kh_size(remote_islands);
	ALLOC_ARRAY(list, island_count);

	kh_foreach_value(remote_islands, island, {
	list[i++] = island;
	});

	for (ref = 0; ref + 1 < island_count; ref++) {
	for (src = ref + 1, dst = src; src < island_count; src++) {
	if (list[ref]->hash == list[src]->hash)
	continue;

	if (src != dst)
	list[dst] = list[src];

	dst++;
	}
	island_count = dst;
	}

	island_bitmap_size = (island_count / 32) + 1;
	core = get_core_island();

	for (i = 0; i < island_count; ++i) {
	mark_remote_island_1(r, list[i], core && list[i]->hash == core->hash);
	}

	free(list);
	}

	void load_delta_islands(struct repository *r, int progress)
	{
	island_marks = kh_init_oid_map();
	remote_islands = kh_init_str();

	git_config(island_config_callback, NULL);
	for_each_ref(find_island_for_ref, NULL);
	deduplicate_islands(r);

	if (progress)
	fprintf(stderr, _("Marked %d islands, done.\n"), island_counter);
	}

	void propagate_island_marks(struct commit *commit)
	{
	khiter_t pos = kh_get_oid_map(island_marks, commit->object.oid);

	if (pos < kh_end(island_marks)) {
	struct commit_list *p;
	struct island_bitmap *root_marks = kh_value(island_marks, pos);

	parse_commit(commit);
	set_island_marks(&get_commit_tree(commit)->object, root_marks);
	for (p = commit->parents; p; p = p->next)
	set_island_marks(&p->item->object, root_marks);
	}
	}

	int compute_pack_layers(struct packing_data *to_pack)
	{
	uint32_t i;

	if (!core_island_name \|\| !island_marks)
	return 1;

	for (i = 0; i < to_pack->nr_objects; ++i) {
	struct object_entry *entry = &to_pack->objects[i];
	khiter_t pos = kh_get_oid_map(island_marks, entry->idx.oid);

	oe_set_layer(to_pack, entry, 1);

	if (pos < kh_end(island_marks)) {
	struct island_bitmap *bitmap = kh_value(island_marks, pos);

	if (island_bitmap_get(bitmap, island_counter_core))
	oe_set_layer(to_pack, entry, 0);
	}
	}

	return 2;
	}

git / git

git/delta-islands.c