protocol-caps.c

   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-ll.h"
   7 #include "hex.h"
   8 #include "object.h"
   9 #include "object-store-ll.h"
  10 #include "string-list.h"
  11 #include "strbuf.h"
  12
  13 struct requested_info {
  14         unsigned size : 1;
  15 };
  16
  17 /*
  18  * Parses oids from the given line and collects them in the given
  19  * oid_str_list. Returns 1 if parsing was successful and 0 otherwise.
  20  */
  21 static int parse_oid(const char *line, struct string_list *oid_str_list)
  22 {
  23         const char *arg;
  24
  25         if (!skip_prefix(line, "oid ", &arg))
  26                 return 0;
  27
  28         string_list_append(oid_str_list, arg);
  29
  30         return 1;
  31 }
  32
  33 /*
  34  * Validates and send requested info back to the client. Any errors detected
  35  * are returned as they are detected.
  36  */
  37 static void send_info(struct repository *r, struct packet_writer *writer,
  38                       struct string_list *oid_str_list,
  39                       struct requested_info *info)
  40 {
  41         struct string_list_item *item;
  42         struct strbuf send_buffer = STRBUF_INIT;
  43
  44         if (!oid_str_list->nr)
  45                 return;
  46
  47         if (info->size)
  48                 packet_writer_write(writer, "size");
  49
  50         for_each_string_list_item (item, oid_str_list) {
  51                 const char *oid_str = item->string;
  52                 struct object_id oid;
  53                 unsigned long object_size;
  54
  55                 if (get_oid_hex(oid_str, &oid) < 0) {
  56                         packet_writer_error(
  57                                 writer,
  58                                 "object-info: protocol error, expected to get oid, not '%s'",
  59                                 oid_str);
  60                         continue;
  61                 }
  62
  63                 strbuf_addstr(&send_buffer, oid_str);
  64
  65                 if (info->size) {
  66                         if (oid_object_info(r, &oid, &object_size) < 0) {
  67                                 strbuf_addstr(&send_buffer, " ");
  68                         } else {
  69                                 strbuf_addf(&send_buffer, " %lu", object_size);
  70                         }
  71                 }
  72
  73                 packet_writer_write(writer, "%s", send_buffer.buf);
  74                 strbuf_reset(&send_buffer);
  75         }
  76         strbuf_release(&send_buffer);
  77 }
  78
  79 int cap_object_info(struct repository *r, struct packet_reader *request)
  80 {
  81         struct requested_info info = { 0 };
  82         struct packet_writer writer;
  83         struct string_list oid_str_list = STRING_LIST_INIT_DUP;
  84
  85         packet_writer_init(&writer, 1);
  86
  87         while (packet_reader_read(request) == PACKET_READ_NORMAL) {
  88                 if (!strcmp("size", request->line)) {
  89                         info.size = 1;
  90                         continue;
  91                 }
  92
  93                 if (parse_oid(request->line, &oid_str_list))
  94                         continue;
  95
  96                 packet_writer_error(&writer,
  97                                     "object-info: unexpected line: '%s'",
  98                                     request->line);
  99         }
 100
 101         if (request->status != PACKET_READ_FLUSH) {
 102                 packet_writer_error(
 103                         &writer, "object-info: expected flush after arguments");
 104                 die(_("object-info: expected flush after arguments"));
 105         }
 106
 107         send_info(r, &writer, &oid_str_list, &info);
 108
 109         string_list_clear(&oid_str_list, 1);
 110
 111         packet_flush(1);
 112
 113         return 0;
 114 }