t/helper/test-pkt-line.c

   1 #include "git-compat-util.h"
   2 #include "test-tool.h"
   3 #include "pkt-line.h"
   4 #include "sideband.h"
   5 #include "write-or-die.h"
   6
   7 static void pack_line(const char *line)
   8 {
   9         if (!strcmp(line, "0000") || !strcmp(line, "0000\n"))
  10                 packet_flush(1);
  11         else if (!strcmp(line, "0001") || !strcmp(line, "0001\n"))
  12                 packet_delim(1);
  13         else
  14                 packet_write_fmt(1, "%s", line);
  15 }
  16
  17 static void pack(int argc, const char **argv)
  18 {
  19         if (argc) { /* read from argv */
  20                 int i;
  21                 for (i = 0; i < argc; i++)
  22                         pack_line(argv[i]);
  23         } else { /* read from stdin */
  24                 char line[LARGE_PACKET_MAX];
  25                 while (fgets(line, sizeof(line), stdin)) {
  26                         pack_line(line);
  27                 }
  28         }
  29 }
  30
  31 static void pack_raw_stdin(void)
  32 {
  33         struct strbuf sb = STRBUF_INIT;
  34
  35         if (strbuf_read(&sb, 0, 0) < 0)
  36                 die_errno("failed to read from stdin");
  37         packet_write(1, sb.buf, sb.len);
  38         strbuf_release(&sb);
  39 }
  40
  41 static void unpack(void)
  42 {
  43         struct packet_reader reader;
  44         packet_reader_init(&reader, 0, NULL, 0,
  45                            PACKET_READ_GENTLE_ON_EOF |
  46                            PACKET_READ_CHOMP_NEWLINE);
  47
  48         while (packet_reader_read(&reader) != PACKET_READ_EOF) {
  49                 switch (reader.status) {
  50                 case PACKET_READ_EOF:
  51                         break;
  52                 case PACKET_READ_NORMAL:
  53                         printf("%s\n", reader.line);
  54                         break;
  55                 case PACKET_READ_FLUSH:
  56                         printf("0000\n");
  57                         break;
  58                 case PACKET_READ_DELIM:
  59                         printf("0001\n");
  60                         break;
  61                 case PACKET_READ_RESPONSE_END:
  62                         printf("0002\n");
  63                         break;
  64                 }
  65         }
  66 }
  67
  68 static void unpack_sideband(void)
  69 {
  70         struct packet_reader reader;
  71         packet_reader_init(&reader, 0, NULL, 0,
  72                            PACKET_READ_GENTLE_ON_EOF |
  73                            PACKET_READ_CHOMP_NEWLINE);
  74
  75         while (packet_reader_read(&reader) != PACKET_READ_EOF) {
  76                 int band;
  77                 int fd;
  78
  79                 switch (reader.status) {
  80                 case PACKET_READ_EOF:
  81                         break;
  82                 case PACKET_READ_NORMAL:
  83                         band = reader.line[0] & 0xff;
  84                         if (band < 1 || band > 2)
  85                                 continue; /* skip non-sideband packets */
  86                         fd = band;
  87
  88                         write_or_die(fd, reader.line + 1, reader.pktlen - 1);
  89                         break;
  90                 case PACKET_READ_FLUSH:
  91                         return;
  92                 case PACKET_READ_DELIM:
  93                 case PACKET_READ_RESPONSE_END:
  94                         break;
  95                 }
  96         }
  97 }
  98
  99 static int send_split_sideband(void)
 100 {
 101         const char *part1 = "Hello,";
 102         const char *primary = "\001primary: regular output\n";
 103         const char *part2 = " world!\n";
 104
 105         send_sideband(1, 2, part1, strlen(part1), LARGE_PACKET_MAX);
 106         packet_write(1, primary, strlen(primary));
 107         send_sideband(1, 2, part2, strlen(part2), LARGE_PACKET_MAX);
 108         packet_response_end(1);
 109
 110         return 0;
 111 }
 112
 113 static int receive_sideband(void)
 114 {
 115         return recv_sideband("sideband", 0, 1);
 116 }
 117
 118 int cmd__pkt_line(int argc, const char **argv)
 119 {
 120         if (argc < 2)
 121                 die("too few arguments");
 122
 123         if (!strcmp(argv[1], "pack"))
 124                 pack(argc - 2, argv + 2);
 125         else if (!strcmp(argv[1], "pack-raw-stdin"))
 126                 pack_raw_stdin();
 127         else if (!strcmp(argv[1], "unpack"))
 128                 unpack();
 129         else if (!strcmp(argv[1], "unpack-sideband"))
 130                 unpack_sideband();
 131         else if (!strcmp(argv[1], "send-split-sideband"))
 132                 send_split_sideband();
 133         else if (!strcmp(argv[1], "receive-sideband"))
 134                 receive_sideband();
 135         else
 136                 die("invalid argument '%s'", argv[1]);
 137
 138         return 0;
 139 }