freshlybakedca.ke
Git fork
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 "odb.h"
10#include "repository.h"
11#include "string-list.h"
12#include "strbuf.h"
13
14struct 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 */
22static 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 */
38static 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 (odb_read_object_info(r->objects, &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
80int 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}