git

1
#include "git-compat-util.h"
2
#include "protocol-caps.h"
3
#include "gettext.h"
4
#include "hex.h"
5
#include "pkt-line.h"
6
#include "hash.h"
7
#include "hex.h"
8
#include "object.h"
9
#include "object-store-ll.h"
10
#include "repository.h"
11
#include "string-list.h"
12
#include "strbuf.h"
13

14
struct requested_info {
15
	unsigned size : 1;
16
};
17

18
/*
19
 * Parses oids from the given line and collects them in the given
20
 * oid_str_list. Returns 1 if parsing was successful and 0 otherwise.
21
 */
22
static int parse_oid(const char *line, struct string_list *oid_str_list)
23
{
24
	const char *arg;
25

26
	if (!skip_prefix(line, "oid ", &arg))
27
		return 0;
28

29
	string_list_append(oid_str_list, arg);
30

31
	return 1;
32
}
33

34
/*
35
 * Validates and send requested info back to the client. Any errors detected
36
 * are returned as they are detected.
37
 */
38
static void send_info(struct repository *r, struct packet_writer *writer,
39
		      struct string_list *oid_str_list,
40
		      struct requested_info *info)
41
{
42
	struct string_list_item *item;
43
	struct strbuf send_buffer = STRBUF_INIT;
44

45
	if (!oid_str_list->nr)
46
		return;
47

48
	if (info->size)
49
		packet_writer_write(writer, "size");
50

51
	for_each_string_list_item (item, oid_str_list) {
52
		const char *oid_str = item->string;
53
		struct object_id oid;
54
		unsigned long object_size;
55

56
		if (get_oid_hex_algop(oid_str, &oid, r->hash_algo) < 0) {
57
			packet_writer_error(
58
				writer,
59
				"object-info: protocol error, expected to get oid, not '%s'",
60
				oid_str);
61
			continue;
62
		}
63

64
		strbuf_addstr(&send_buffer, oid_str);
65

66
		if (info->size) {
67
			if (oid_object_info(r, &oid, &object_size) < 0) {
68
				strbuf_addstr(&send_buffer, " ");
69
			} else {
70
				strbuf_addf(&send_buffer, " %lu", object_size);
71
			}
72
		}
73

74
		packet_writer_write(writer, "%s", send_buffer.buf);
75
		strbuf_reset(&send_buffer);
76
	}
77
	strbuf_release(&send_buffer);
78
}
79

80
int cap_object_info(struct repository *r, struct packet_reader *request)
81
{
82
	struct requested_info info = { 0 };
83
	struct packet_writer writer;
84
	struct string_list oid_str_list = STRING_LIST_INIT_DUP;
85

86
	packet_writer_init(&writer, 1);
87

88
	while (packet_reader_read(request) == PACKET_READ_NORMAL) {
89
		if (!strcmp("size", request->line)) {
90
			info.size = 1;
91
			continue;
92
		}
93

94
		if (parse_oid(request->line, &oid_str_list))
95
			continue;
96

97
		packet_writer_error(&writer,
98
				    "object-info: unexpected line: '%s'",
99
				    request->line);
100
	}
101

102
	if (request->status != PACKET_READ_FLUSH) {
103
		packet_writer_error(
104
			&writer, "object-info: expected flush after arguments");
105
		die(_("object-info: expected flush after arguments"));
106
	}
107

108
	send_info(r, &writer, &oid_str_list, &info);
109

110
	string_list_clear(&oid_str_list, 1);
111

112
	packet_flush(1);
113

114
	return 0;
115
}
116