[thirdparty/git.git] / bundle.c

#include "cache.h"
#include "lockfile.h"
#include "bundle.h"
#include "object.h"
#include "commit.h"
#include "diff.h"
#include "revision.h"
#include "list-objects.h"
#include "run-command.h"
#include "refs.h"
#include "argv-array.h"

static const char bundle_signature[] = "# v2 git bundle\n";

static void add_to_ref_list(const unsigned char *sha1, const char *name,
		struct ref_list *list)
{
	ALLOC_GROW(list->list, list->nr + 1, list->alloc);
	hashcpy(list->list[list->nr].sha1, sha1);
	list->list[list->nr].name = xstrdup(name);
	list->nr++;
}

static int parse_bundle_header(int fd, struct bundle_header *header,
			       const char *report_path)
{
	struct strbuf buf = STRBUF_INIT;
	int status = 0;

	/* The bundle header begins with the signature */
	if (strbuf_getwholeline_fd(&buf, fd, '\n') ||
	    strcmp(buf.buf, bundle_signature)) {
		if (report_path)
			error(_("'%s' does not look like a v2 bundle file"),
			      report_path);
		status = -1;
		goto abort;
	}

	/* The bundle header ends with an empty line */
	while (!strbuf_getwholeline_fd(&buf, fd, '\n') &&
	       buf.len && buf.buf[0] != '\n') {
		unsigned char sha1[20];
		int is_prereq = 0;

		if (*buf.buf == '-') {
			is_prereq = 1;
			strbuf_remove(&buf, 0, 1);
		}
		strbuf_rtrim(&buf);

		/*
		 * Tip lines have object name, SP, and refname.
		 * Prerequisites have object name that is optionally
		 * followed by SP and subject line.
		 */
		if (get_sha1_hex(buf.buf, sha1) ||
		    (buf.len > 40 && !isspace(buf.buf[40])) ||
		    (!is_prereq && buf.len <= 40)) {
			if (report_path)
				error(_("unrecognized header: %s%s (%d)"),
				      (is_prereq ? "-" : ""), buf.buf, (int)buf.len);
			status = -1;
			break;
		} else {
			if (is_prereq)
				add_to_ref_list(sha1, "", &header->prerequisites);
			else
				add_to_ref_list(sha1, buf.buf + 41, &header->references);
		}
	}

 abort:
	if (status) {
		close(fd);
		fd = -1;
	}
	strbuf_release(&buf);
	return fd;
}

int read_bundle_header(const char *path, struct bundle_header *header)
{
	int fd = open(path, O_RDONLY);

	if (fd < 0)
		return error(_("could not open '%s'"), path);
	return parse_bundle_header(fd, header, path);
}

int is_bundle(const char *path, int quiet)
{
	struct bundle_header header;
	int fd = open(path, O_RDONLY);

	if (fd < 0)
		return 0;
	memset(&header, 0, sizeof(header));
	fd = parse_bundle_header(fd, &header, quiet ? NULL : path);
	if (fd >= 0)
		close(fd);
	return (fd >= 0);
}

static int list_refs(struct ref_list *r, int argc, const char **argv)
{
	int i;

	for (i = 0; i < r->nr; i++) {
		if (argc > 1) {
			int j;
			for (j = 1; j < argc; j++)
				if (!strcmp(r->list[i].name, argv[j]))
					break;
			if (j == argc)
				continue;
		}
		printf("%s %s\n", sha1_to_hex(r->list[i].sha1),
				r->list[i].name);
	}
	return 0;
}

/* Remember to update object flag allocation in object.h */
#define PREREQ_MARK (1u<<16)

int verify_bundle(struct bundle_header *header, int verbose)
{
	/*
	 * Do fast check, then if any prereqs are missing then go line by line
	 * to be verbose about the errors
	 */
	struct ref_list *p = &header->prerequisites;
	struct rev_info revs;
	const char *argv[] = {NULL, "--all", NULL};
	struct object_array refs;
	struct commit *commit;
	int i, ret = 0, req_nr;
	const char *message = _("Repository lacks these prerequisite commits:");

	init_revisions(&revs, NULL);
	for (i = 0; i < p->nr; i++) {
		struct ref_list_entry *e = p->list + i;
		struct object *o = parse_object(e->sha1);
		if (o) {
			o->flags |= PREREQ_MARK;
			add_pending_object(&revs, o, e->name);
			continue;
		}
		if (++ret == 1)
			error("%s", message);
		error("%s %s", sha1_to_hex(e->sha1), e->name);
	}
	if (revs.pending.nr != p->nr)
		return ret;
	req_nr = revs.pending.nr;
	setup_revisions(2, argv, &revs, NULL);

	refs = revs.pending;
	revs.leak_pending = 1;

	if (prepare_revision_walk(&revs))
		die(_("revision walk setup failed"));

	i = req_nr;
	while (i && (commit = get_revision(&revs)))
		if (commit->object.flags & PREREQ_MARK)
			i--;

	for (i = 0; i < req_nr; i++)
		if (!(refs.objects[i].item->flags & SHOWN)) {
			if (++ret == 1)
				error("%s", message);
			error("%s %s", sha1_to_hex(refs.objects[i].item->sha1),
				refs.objects[i].name);
		}

	clear_commit_marks_for_object_array(&refs, ALL_REV_FLAGS);
	free(refs.objects);

	if (verbose) {
		struct ref_list *r;

		r = &header->references;
		printf_ln(Q_("The bundle contains this ref:",
			     "The bundle contains these %d refs:",
			     r->nr),
			  r->nr);
		list_refs(r, 0, NULL);
		r = &header->prerequisites;
		if (!r->nr) {
			printf_ln(_("The bundle records a complete history."));
		} else {
			printf_ln(Q_("The bundle requires this ref:",
				     "The bundle requires these %d refs:",
				     r->nr),
				  r->nr);
			list_refs(r, 0, NULL);
		}
	}
	return ret;
}

int list_bundle_refs(struct bundle_header *header, int argc, const char **argv)
{
	return list_refs(&header->references, argc, argv);
}

static int is_tag_in_date_range(struct object *tag, struct rev_info *revs)
{
	unsigned long size;
	enum object_type type;
	char *buf = NULL, *line, *lineend;
	unsigned long date;
	int result = 1;

	if (revs->max_age == -1 && revs->min_age == -1)
		goto out;

	buf = read_sha1_file(tag->sha1, &type, &size);
	if (!buf)
		goto out;
	line = memmem(buf, size, "\ntagger ", 8);
	if (!line++)
		goto out;
	lineend = memchr(line, '\n', buf + size - line);
	line = memchr(line, '>', lineend ? lineend - line : buf + size - line);
	if (!line++)
		goto out;
	date = strtoul(line, NULL, 10);
	result = (revs->max_age == -1 || revs->max_age < date) &&
		(revs->min_age == -1 || revs->min_age > date);
out:
	free(buf);
	return result;
}

static int write_pack_data(int bundle_fd, struct lock_file *lock, struct rev_info *revs)
{
	struct child_process pack_objects = CHILD_PROCESS_INIT;
	int i;

	argv_array_pushl(&pack_objects.args,
			 "pack-objects", "--all-progress-implied",
			 "--stdout", "--thin", "--delta-base-offset",
			 NULL);
	pack_objects.in = -1;
	pack_objects.out = bundle_fd;
	pack_objects.git_cmd = 1;
	if (start_command(&pack_objects))
		return error(_("Could not spawn pack-objects"));

	/*
	 * start_command closed bundle_fd if it was > 1
	 * so set the lock fd to -1 so commit_lock_file()
	 * won't fail trying to close it.
	 */
	lock->fd = -1;

	for (i = 0; i < revs->pending.nr; i++) {
		struct object *object = revs->pending.objects[i].item;
		if (object->flags & UNINTERESTING)
			write_or_die(pack_objects.in, "^", 1);
		write_or_die(pack_objects.in, sha1_to_hex(object->sha1), 40);
		write_or_die(pack_objects.in, "\n", 1);
	}
	close(pack_objects.in);
	if (finish_command(&pack_objects))
		return error(_("pack-objects died"));
	return 0;
}

static int compute_and_write_prerequisites(int bundle_fd,
					   struct rev_info *revs,
					   int argc, const char **argv)
{
	struct child_process rls = CHILD_PROCESS_INIT;
	struct strbuf buf = STRBUF_INIT;
	FILE *rls_fout;
	int i;

	argv_array_pushl(&rls.args,
			 "rev-list", "--boundary", "--pretty=oneline",
			 NULL);
	for (i = 1; i < argc; i++)
		argv_array_push(&rls.args, argv[i]);
	rls.out = -1;
	rls.git_cmd = 1;
	if (start_command(&rls))
		return -1;
	rls_fout = xfdopen(rls.out, "r");
	while (strbuf_getwholeline(&buf, rls_fout, '\n') != EOF) {
		unsigned char sha1[20];
		if (buf.len > 0 && buf.buf[0] == '-') {
			write_or_die(bundle_fd, buf.buf, buf.len);
			if (!get_sha1_hex(buf.buf + 1, sha1)) {
				struct object *object = parse_object_or_die(sha1, buf.buf);
				object->flags |= UNINTERESTING;
				add_pending_object(revs, object, buf.buf);
			}
		} else if (!get_sha1_hex(buf.buf, sha1)) {
			struct object *object = parse_object_or_die(sha1, buf.buf);
			object->flags |= SHOWN;
		}
	}
	strbuf_release(&buf);
	fclose(rls_fout);
	if (finish_command(&rls))
		return error(_("rev-list died"));
	return 0;
}

/*
 * Write out bundle refs based on the tips already
 * parsed into revs.pending. As a side effect, may
 * manipulate revs.pending to include additional
 * necessary objects (like tags).
 *
 * Returns the number of refs written, or negative
 * on error.
 */
static int write_bundle_refs(int bundle_fd, struct rev_info *revs)
{
	int i;
	int ref_count = 0;

	for (i = 0; i < revs->pending.nr; i++) {
		struct object_array_entry *e = revs->pending.objects + i;
		unsigned char sha1[20];
		char *ref;
		const char *display_ref;
		int flag;

		if (e->item->flags & UNINTERESTING)
			continue;
		if (dwim_ref(e->name, strlen(e->name), sha1, &ref) != 1)
			continue;
		if (read_ref_full(e->name, RESOLVE_REF_READING, sha1, &flag))
			flag = 0;
		display_ref = (flag & REF_ISSYMREF) ? e->name : ref;

		if (e->item->type == OBJ_TAG &&
				!is_tag_in_date_range(e->item, revs)) {
			e->item->flags |= UNINTERESTING;
			continue;
		}

		/*
		 * Make sure the refs we wrote out is correct; --max-count and
		 * other limiting options could have prevented all the tips
		 * from getting output.
		 *
		 * Non commit objects such as tags and blobs do not have
		 * this issue as they are not affected by those extra
		 * constraints.
		 */
		if (!(e->item->flags & SHOWN) && e->item->type == OBJ_COMMIT) {
			warning(_("ref '%s' is excluded by the rev-list options"),
				e->name);
			free(ref);
			continue;
		}
		/*
		 * If you run "git bundle create bndl v1.0..v2.0", the
		 * name of the positive ref is "v2.0" but that is the
		 * commit that is referenced by the tag, and not the tag
		 * itself.
		 */
		if (hashcmp(sha1, e->item->sha1)) {
			/*
			 * Is this the positive end of a range expressed
			 * in terms of a tag (e.g. v2.0 from the range
			 * "v1.0..v2.0")?
			 */
			struct commit *one = lookup_commit_reference(sha1);
			struct object *obj;

			if (e->item == &(one->object)) {
				/*
				 * Need to include e->name as an
				 * independent ref to the pack-objects
				 * input, so that the tag is included
				 * in the output; otherwise we would
				 * end up triggering "empty bundle"
				 * error.
				 */
				obj = parse_object_or_die(sha1, e->name);
				obj->flags |= SHOWN;
				add_pending_object(revs, obj, e->name);
			}
			free(ref);
			continue;
		}

		ref_count++;
		write_or_die(bundle_fd, sha1_to_hex(e->item->sha1), 40);
		write_or_die(bundle_fd, " ", 1);
		write_or_die(bundle_fd, display_ref, strlen(display_ref));
		write_or_die(bundle_fd, "\n", 1);
		free(ref);
	}

	/* end header */
	write_or_die(bundle_fd, "\n", 1);
	return ref_count;
}

int create_bundle(struct bundle_header *header, const char *path,
		  int argc, const char **argv)
{
	static struct lock_file lock;
	int bundle_fd = -1;
	int bundle_to_stdout;
	int ref_count = 0;
	struct rev_info revs;

	bundle_to_stdout = !strcmp(path, "-");
	if (bundle_to_stdout)
		bundle_fd = 1;
	else
		bundle_fd = hold_lock_file_for_update(&lock, path,
						      LOCK_DIE_ON_ERROR);

	/* write signature */
	write_or_die(bundle_fd, bundle_signature, strlen(bundle_signature));

	/* init revs to list objects for pack-objects later */
	save_commit_buffer = 0;
	init_revisions(&revs, NULL);

	/* write prerequisites */
	if (compute_and_write_prerequisites(bundle_fd, &revs, argc, argv))
		return -1;

	argc = setup_revisions(argc, argv, &revs, NULL);

	if (argc > 1)
		return error(_("unrecognized argument: %s"), argv[1]);

	object_array_remove_duplicates(&revs.pending);

	ref_count = write_bundle_refs(bundle_fd, &revs);
	if (!ref_count)
		die(_("Refusing to create empty bundle."));
	else if (ref_count < 0)
		return -1;

	/* write pack */
	if (write_pack_data(bundle_fd, &lock, &revs))
		return -1;

	if (!bundle_to_stdout) {
		if (commit_lock_file(&lock))
			die_errno(_("cannot create '%s'"), path);
	}
	return 0;
}

int unbundle(struct bundle_header *header, int bundle_fd, int flags)
{
	const char *argv_index_pack[] = {"index-pack",
					 "--fix-thin", "--stdin", NULL, NULL};
	struct child_process ip = CHILD_PROCESS_INIT;

	if (flags & BUNDLE_VERBOSE)
		argv_index_pack[3] = "-v";

	if (verify_bundle(header, 0))
		return -1;
	ip.argv = argv_index_pack;
	ip.in = bundle_fd;
	ip.no_stdout = 1;
	ip.git_cmd = 1;
	if (run_command(&ip))
		return error(_("index-pack died"));
	return 0;
}
Commit	Line	Data
30415d50	1	#include "cache.h"
697cc8ef	2	#include "lockfile.h"
30415d50 JS	3	#include "bundle.h"
	4	#include "object.h"
	5	#include "commit.h"
	6	#include "diff.h"
	7	#include "revision.h"
	8	#include "list-objects.h"
	9	#include "run-command.h"
fa303836	10	#include "refs.h"
1e2371ea	11	#include "argv-array.h"
30415d50 JS	12
	13	static const char bundle_signature[] = "# v2 git bundle\n";
	14
	15	static void add_to_ref_list(const unsigned char sha1, const char name,
	16	struct ref_list *list)
	17	{
5cbbe13a	18	ALLOC_GROW(list->list, list->nr + 1, list->alloc);
50546b15	19	hashcpy(list->list[list->nr].sha1, sha1);
30415d50 JS	20	list->list[list->nr].name = xstrdup(name);
	21	list->nr++;
	22	}
	23
2727b71f JH	24	static int parse_bundle_header(int fd, struct bundle_header *header,
2727b71f JH	25	const char *report_path)
f3fa1838	26	{
e9ee84cf	27	struct strbuf buf = STRBUF_INIT;
e9ee84cf	28	int status = 0;
30415d50	29
e9ee84cf	30	/* The bundle header begins with the signature */
5e8617f5	31	if (strbuf_getwholeline_fd(&buf, fd, '\n') \|\|
e9ee84cf	32	strcmp(buf.buf, bundle_signature)) {
2727b71f	33	if (report_path)
8a1e7eac	34	error(_("'%s' does not look like a v2 bundle file"),
2727b71f	35	report_path);
e9ee84cf JH	36	status = -1;
e9ee84cf JH	37	goto abort;
30415d50	38	}
e9ee84cf JH	39
e9ee84cf JH	40	/* The bundle header ends with an empty line */
5e8617f5	41	while (!strbuf_getwholeline_fd(&buf, fd, '\n') &&
e9ee84cf	42	buf.len && buf.buf[0] != '\n') {
30415d50	43	unsigned char sha1[20];
e9ee84cf JH	44	int is_prereq = 0;
	45
	46	if (*buf.buf == '-') {
	47	is_prereq = 1;
	48	strbuf_remove(&buf, 0, 1);
30415d50	49	}
e9ee84cf	50	strbuf_rtrim(&buf);
30415d50	51
e9ee84cf JH	52	/*
	53	* Tip lines have object name, SP, and refname.
	54	* Prerequisites have object name that is optionally
	55	* followed by SP and subject line.
	56	*/
	57	if (get_sha1_hex(buf.buf, sha1) \|\|
5446e33f	58	(buf.len > 40 && !isspace(buf.buf[40])) \|\|
e9ee84cf	59	(!is_prereq && buf.len <= 40)) {
2727b71f	60	if (report_path)
8a1e7eac	61	error(_("unrecognized header: %s%s (%d)"),
2727b71f	62	(is_prereq ? "-" : ""), buf.buf, (int)buf.len);
e9ee84cf JH	63	status = -1;
	64	break;
	65	} else {
	66	if (is_prereq)
	67	add_to_ref_list(sha1, "", &header->prerequisites);
	68	else
	69	add_to_ref_list(sha1, buf.buf + 41, &header->references);
30415d50	70	}
30415d50	71	}
e9ee84cf JH	72
	73	abort:
	74	if (status) {
	75	close(fd);
	76	fd = -1;
30415d50	77	}
e9ee84cf	78	strbuf_release(&buf);
30415d50 JS	79	return fd;
	80	}
	81
2727b71f JH	82	int read_bundle_header(const char path, struct bundle_header header)
	83	{
	84	int fd = open(path, O_RDONLY);
	85
30415d50	86	if (fd < 0)
8a1e7eac	87	return error(_("could not open '%s'"), path);
2727b71f JH	88	return parse_bundle_header(fd, header, path);
	89	}
	90
	91	int is_bundle(const char *path, int quiet)
	92	{
	93	struct bundle_header header;
	94	int fd = open(path, O_RDONLY);
	95
	96	if (fd < 0)
	97	return 0;
	98	memset(&header, 0, sizeof(header));
	99	fd = parse_bundle_header(fd, &header, quiet ? NULL : path);
	100	if (fd >= 0)
	101	close(fd);
	102	return (fd >= 0);
30415d50 JS	103	}
	104
	105	static int list_refs(struct ref_list r, int argc, const char *argv)
	106	{
	107	int i;
	108
	109	for (i = 0; i < r->nr; i++) {
	110	if (argc > 1) {
	111	int j;
	112	for (j = 1; j < argc; j++)
	113	if (!strcmp(r->list[i].name, argv[j]))
	114	break;
	115	if (j == argc)
	116	continue;
	117	}
	118	printf("%s %s\n", sha1_to_hex(r->list[i].sha1),
	119	r->list[i].name);
	120	}
	121	return 0;
	122	}
	123
208acbfb	124	/* Remember to update object flag allocation in object.h */
30415d50 JS	125	#define PREREQ_MARK (1u<<16)
	126
	127	int verify_bundle(struct bundle_header *header, int verbose)
	128	{
	129	/*
	130	* Do fast check, then if any prereqs are missing then go line by line
	131	* to be verbose about the errors
	132	*/
	133	struct ref_list *p = &header->prerequisites;
	134	struct rev_info revs;
a80aad7b	135	const char *argv[] = {NULL, "--all", NULL};
30415d50 JS	136	struct object_array refs;
	137	struct commit *commit;
	138	int i, ret = 0, req_nr;
8a1e7eac	139	const char *message = _("Repository lacks these prerequisite commits:");
30415d50 JS	140
	141	init_revisions(&revs, NULL);
	142	for (i = 0; i < p->nr; i++) {
	143	struct ref_list_entry *e = p->list + i;
	144	struct object *o = parse_object(e->sha1);
	145	if (o) {
	146	o->flags \|= PREREQ_MARK;
	147	add_pending_object(&revs, o, e->name);
	148	continue;
	149	}
	150	if (++ret == 1)
9db56f71	151	error("%s", message);
30415d50 JS	152	error("%s %s", sha1_to_hex(e->sha1), e->name);
	153	}
	154	if (revs.pending.nr != p->nr)
	155	return ret;
	156	req_nr = revs.pending.nr;
	157	setup_revisions(2, argv, &revs, NULL);
	158
5be78599 RS	159	refs = revs.pending;
5be78599 RS	160	revs.leak_pending = 1;
30415d50	161
3d51e1b5	162	if (prepare_revision_walk(&revs))
8a1e7eac	163	die(_("revision walk setup failed"));
30415d50 JS	164
	165	i = req_nr;
	166	while (i && (commit = get_revision(&revs)))
	167	if (commit->object.flags & PREREQ_MARK)
	168	i--;
	169
	170	for (i = 0; i < req_nr; i++)
	171	if (!(refs.objects[i].item->flags & SHOWN)) {
	172	if (++ret == 1)
9db56f71	173	error("%s", message);
30415d50 JS	174	error("%s %s", sha1_to_hex(refs.objects[i].item->sha1),
	175	refs.objects[i].name);
	176	}
	177
86a0a408	178	clear_commit_marks_for_object_array(&refs, ALL_REV_FLAGS);
5be78599	179	free(refs.objects);
30415d50 JS	180
	181	if (verbose) {
	182	struct ref_list *r;
	183
	184	r = &header->references;
a02ffe0e LF	185	printf_ln(Q_("The bundle contains this ref:",
a02ffe0e LF	186	"The bundle contains these %d refs:",
8a1e7eac NTND	187	r->nr),
8a1e7eac NTND	188	r->nr);
30415d50	189	list_refs(r, 0, NULL);
71ba6b10	190	r = &header->prerequisites;
8c3710fd JH	191	if (!r->nr) {
	192	printf_ln(_("The bundle records a complete history."));
	193	} else {
a02ffe0e LF	194	printf_ln(Q_("The bundle requires this ref:",
a02ffe0e LF	195	"The bundle requires these %d refs:",
8c3710fd JH	196	r->nr),
	197	r->nr);
	198	list_refs(r, 0, NULL);
	199	}
30415d50 JS	200	}
	201	return ret;
	202	}
	203
	204	int list_bundle_refs(struct bundle_header header, int argc, const char *argv)
	205	{
	206	return list_refs(&header->references, argc, argv);
	207	}
	208
c9a42c4a JS	209	static int is_tag_in_date_range(struct object tag, struct rev_info revs)
	210	{
	211	unsigned long size;
	212	enum object_type type;
64045940	213	char buf = NULL, line, *lineend;
c9a42c4a	214	unsigned long date;
64045940	215	int result = 1;
c9a42c4a JS	216
c9a42c4a JS	217	if (revs->max_age == -1 && revs->min_age == -1)
64045940	218	goto out;
c9a42c4a JS	219
	220	buf = read_sha1_file(tag->sha1, &type, &size);
	221	if (!buf)
64045940	222	goto out;
c9a42c4a JS	223	line = memmem(buf, size, "\ntagger ", 8);
c9a42c4a JS	224	if (!line++)
64045940	225	goto out;
2c8544ab LF	226	lineend = memchr(line, '\n', buf + size - line);
2c8544ab LF	227	line = memchr(line, '>', lineend ? lineend - line : buf + size - line);
c9a42c4a	228	if (!line++)
64045940	229	goto out;
c9a42c4a	230	date = strtoul(line, NULL, 10);
64045940	231	result = (revs->max_age == -1 \|\| revs->max_age < date) &&
c9a42c4a	232	(revs->min_age == -1 \|\| revs->min_age > date);
64045940 RS	233	out:
	234	free(buf);
	235	return result;
c9a42c4a JS	236	}
c9a42c4a JS	237
5e626b91 JH	238	static int write_pack_data(int bundle_fd, struct lock_file lock, struct rev_info revs)
	239	{
	240	struct child_process pack_objects = CHILD_PROCESS_INIT;
	241	int i;
	242
	243	argv_array_pushl(&pack_objects.args,
	244	"pack-objects", "--all-progress-implied",
	245	"--stdout", "--thin", "--delta-base-offset",
	246	NULL);
	247	pack_objects.in = -1;
	248	pack_objects.out = bundle_fd;
	249	pack_objects.git_cmd = 1;
	250	if (start_command(&pack_objects))
	251	return error(_("Could not spawn pack-objects"));
	252
	253	/*
	254	* start_command closed bundle_fd if it was > 1
	255	* so set the lock fd to -1 so commit_lock_file()
	256	* won't fail trying to close it.
	257	*/
	258	lock->fd = -1;
	259
	260	for (i = 0; i < revs->pending.nr; i++) {
	261	struct object *object = revs->pending.objects[i].item;
	262	if (object->flags & UNINTERESTING)
	263	write_or_die(pack_objects.in, "^", 1);
	264	write_or_die(pack_objects.in, sha1_to_hex(object->sha1), 40);
	265	write_or_die(pack_objects.in, "\n", 1);
	266	}
	267	close(pack_objects.in);
	268	if (finish_command(&pack_objects))
	269	return error(_("pack-objects died"));
	270	return 0;
	271	}
	272
e8eb2512 JH	273	static int compute_and_write_prerequisites(int bundle_fd,
	274	struct rev_info *revs,
	275	int argc, const char **argv)
30415d50	276	{
d3180279	277	struct child_process rls = CHILD_PROCESS_INIT;
e8eb2512	278	struct strbuf buf = STRBUF_INIT;
30415d50	279	FILE *rls_fout;
e8eb2512	280	int i;
30415d50	281
92859f3a	282	argv_array_pushl(&rls.args,
1e2371ea JH	283	"rev-list", "--boundary", "--pretty=oneline",
	284	NULL);
	285	for (i = 1; i < argc; i++)
92859f3a	286	argv_array_push(&rls.args, argv[i]);
30415d50 JS	287	rls.out = -1;
	288	rls.git_cmd = 1;
	289	if (start_command(&rls))
	290	return -1;
41698375	291	rls_fout = xfdopen(rls.out, "r");
bc2fed49	292	while (strbuf_getwholeline(&buf, rls_fout, '\n') != EOF) {
30415d50	293	unsigned char sha1[20];
bc2fed49 TR	294	if (buf.len > 0 && buf.buf[0] == '-') {
	295	write_or_die(bundle_fd, buf.buf, buf.len);
	296	if (!get_sha1_hex(buf.buf + 1, sha1)) {
75a95490	297	struct object *object = parse_object_or_die(sha1, buf.buf);
30415d50	298	object->flags \|= UNINTERESTING;
e8eb2512	299	add_pending_object(revs, object, buf.buf);
30415d50	300	}
bc2fed49	301	} else if (!get_sha1_hex(buf.buf, sha1)) {
75a95490	302	struct object *object = parse_object_or_die(sha1, buf.buf);
30415d50 JS	303	object->flags \|= SHOWN;
	304	}
	305	}
bc2fed49	306	strbuf_release(&buf);
30415d50 JS	307	fclose(rls_fout);
30415d50 JS	308	if (finish_command(&rls))
8a1e7eac	309	return error(_("rev-list died"));
e8eb2512 JH	310	return 0;
	311	}
	312
d9362ef9 JK	313	/*
	314	* Write out bundle refs based on the tips already
	315	* parsed into revs.pending. As a side effect, may
	316	* manipulate revs.pending to include additional
	317	* necessary objects (like tags).
	318	*
	319	* Returns the number of refs written, or negative
	320	* on error.
	321	*/
	322	static int write_bundle_refs(int bundle_fd, struct rev_info *revs)
e8eb2512	323	{
d9362ef9 JK	324	int i;
d9362ef9 JK	325	int ref_count = 0;
b2a6d1c6	326
d9362ef9 JK	327	for (i = 0; i < revs->pending.nr; i++) {
d9362ef9 JK	328	struct object_array_entry *e = revs->pending.objects + i;
30415d50 JS	329	unsigned char sha1[20];
30415d50 JS	330	char *ref;
fa303836 JH	331	const char *display_ref;
fa303836 JH	332	int flag;
30415d50 JS	333
	334	if (e->item->flags & UNINTERESTING)
	335	continue;
	336	if (dwim_ref(e->name, strlen(e->name), sha1, &ref) != 1)
	337	continue;
7695d118	338	if (read_ref_full(e->name, RESOLVE_REF_READING, sha1, &flag))
fa303836 JH	339	flag = 0;
	340	display_ref = (flag & REF_ISSYMREF) ? e->name : ref;
	341
c9a42c4a	342	if (e->item->type == OBJ_TAG &&
d9362ef9	343	!is_tag_in_date_range(e->item, revs)) {
c9a42c4a JS	344	e->item->flags \|= UNINTERESTING;
	345	continue;
	346	}
	347
30415d50 JS	348	/*
	349	* Make sure the refs we wrote out is correct; --max-count and
	350	* other limiting options could have prevented all the tips
	351	* from getting output.
	352	*
	353	* Non commit objects such as tags and blobs do not have
	354	* this issue as they are not affected by those extra
	355	* constraints.
	356	*/
	357	if (!(e->item->flags & SHOWN) && e->item->type == OBJ_COMMIT) {
8a1e7eac	358	warning(_("ref '%s' is excluded by the rev-list options"),
30415d50 JS	359	e->name);
	360	free(ref);
	361	continue;
	362	}
	363	/*
	364	* If you run "git bundle create bndl v1.0..v2.0", the
	365	* name of the positive ref is "v2.0" but that is the
	366	* commit that is referenced by the tag, and not the tag
	367	* itself.
	368	*/
	369	if (hashcmp(sha1, e->item->sha1)) {
	370	/*
	371	* Is this the positive end of a range expressed
	372	* in terms of a tag (e.g. v2.0 from the range
	373	* "v1.0..v2.0")?
	374	*/
	375	struct commit *one = lookup_commit_reference(sha1);
	376	struct object *obj;
	377
	378	if (e->item == &(one->object)) {
	379	/*
	380	* Need to include e->name as an
	381	* independent ref to the pack-objects
	382	* input, so that the tag is included
	383	* in the output; otherwise we would
	384	* end up triggering "empty bundle"
	385	* error.
	386	*/
75a95490	387	obj = parse_object_or_die(sha1, e->name);
30415d50	388	obj->flags \|= SHOWN;
d9362ef9	389	add_pending_object(revs, obj, e->name);
30415d50 JS	390	}
	391	free(ref);
	392	continue;
	393	}
	394
	395	ref_count++;
	396	write_or_die(bundle_fd, sha1_to_hex(e->item->sha1), 40);
	397	write_or_die(bundle_fd, " ", 1);
fa303836	398	write_or_die(bundle_fd, display_ref, strlen(display_ref));
30415d50 JS	399	write_or_die(bundle_fd, "\n", 1);
	400	free(ref);
	401	}
30415d50 JS	402
	403	/* end header */
	404	write_or_die(bundle_fd, "\n", 1);
d9362ef9 JK	405	return ref_count;
	406	}
	407
	408	int create_bundle(struct bundle_header header, const char path,
	409	int argc, const char **argv)
	410	{
	411	static struct lock_file lock;
	412	int bundle_fd = -1;
	413	int bundle_to_stdout;
	414	int ref_count = 0;
	415	struct rev_info revs;
	416
	417	bundle_to_stdout = !strcmp(path, "-");
	418	if (bundle_to_stdout)
	419	bundle_fd = 1;
	420	else
	421	bundle_fd = hold_lock_file_for_update(&lock, path,
	422	LOCK_DIE_ON_ERROR);
	423
	424	/* write signature */
	425	write_or_die(bundle_fd, bundle_signature, strlen(bundle_signature));
	426
	427	/* init revs to list objects for pack-objects later */
	428	save_commit_buffer = 0;
	429	init_revisions(&revs, NULL);
	430
	431	/* write prerequisites */
	432	if (compute_and_write_prerequisites(bundle_fd, &revs, argc, argv))
	433	return -1;
	434
	435	argc = setup_revisions(argc, argv, &revs, NULL);
	436
	437	if (argc > 1)
	438	return error(_("unrecognized argument: %s"), argv[1]);
	439
	440	object_array_remove_duplicates(&revs.pending);
	441
	442	ref_count = write_bundle_refs(bundle_fd, &revs);
	443	if (!ref_count)
	444	die(_("Refusing to create empty bundle."));
	445	else if (ref_count < 0)
	446	return -1;
30415d50 JS	447
30415d50 JS	448	/* write pack */
5e626b91 JH	449	if (write_pack_data(bundle_fd, &lock, &revs))
5e626b91 JH	450	return -1;
4ed7cd3a	451
0f5cdf65 CH	452	if (!bundle_to_stdout) {
0f5cdf65 CH	453	if (commit_lock_file(&lock))
8a1e7eac	454	die_errno(_("cannot create '%s'"), path);
0f5cdf65	455	}
c20181e3	456	return 0;
30415d50 JS	457	}
30415d50 JS	458
be042aff	459	int unbundle(struct bundle_header *header, int bundle_fd, int flags)
30415d50 JS	460	{
30415d50 JS	461	const char *argv_index_pack[] = {"index-pack",
be042aff	462	"--fix-thin", "--stdin", NULL, NULL};
d3180279	463	struct child_process ip = CHILD_PROCESS_INIT;
30415d50	464
be042aff JH	465	if (flags & BUNDLE_VERBOSE)
	466	argv_index_pack[3] = "-v";
	467
30415d50 JS	468	if (verify_bundle(header, 0))
30415d50 JS	469	return -1;
30415d50 JS	470	ip.argv = argv_index_pack;
	471	ip.in = bundle_fd;
	472	ip.no_stdout = 1;
	473	ip.git_cmd = 1;
	474	if (run_command(&ip))
8a1e7eac	475	return error(_("index-pack died"));
30415d50 JS	476	return 0;
30415d50 JS	477	}