| #include "pkt-line.h" |
| |
| static void pack_line(const char *line) |
| { |
| if (!strcmp(line, "0000") || !strcmp(line, "0000\n")) |
| packet_flush(1); |
| else if (!strcmp(line, "0001") || !strcmp(line, "0001\n")) |
| packet_delim(1); |
| else |
| packet_write_fmt(1, "%s", line); |
| } |
| |
| static void pack(int argc, const char **argv) |
| { |
| if (argc) { /* read from argv */ |
| int i; |
| for (i = 0; i < argc; i++) |
| pack_line(argv[i]); |
| } else { /* read from stdin */ |
| char line[LARGE_PACKET_MAX]; |
| while (fgets(line, sizeof(line), stdin)) { |
| pack_line(line); |
| } |
| } |
| } |
| |
| static void unpack(void) |
| { |
| struct packet_reader reader; |
| packet_reader_init(&reader, 0, NULL, 0, |
| PACKET_READ_GENTLE_ON_EOF | |
| PACKET_READ_CHOMP_NEWLINE); |
| |
| while (packet_reader_read(&reader) != PACKET_READ_EOF) { |
| switch (reader.status) { |
| case PACKET_READ_EOF: |
| break; |
| case PACKET_READ_NORMAL: |
| printf("%s\n", reader.line); |
| break; |
| case PACKET_READ_FLUSH: |
| printf("0000\n"); |
| break; |
| case PACKET_READ_DELIM: |
| printf("0001\n"); |
| break; |
| } |
| } |
| } |
| |
| int cmd_main(int argc, const char **argv) |
| { |
| if (argc < 2) |
| die("too few arguments"); |
| |
| if (!strcmp(argv[1], "pack")) |
| pack(argc - 2, argv + 2); |
| else if (!strcmp(argv[1], "unpack")) |
| unpack(); |
| else |
| die("invalid argument '%s'", argv[1]); |
| |
| return 0; |
| } |