diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/bencoding.c | 54 |
1 files changed, 43 insertions, 11 deletions
diff --git a/utils/bencoding.c b/utils/bencoding.c index a60e1b3..ae80546 100644 --- a/utils/bencoding.c +++ b/utils/bencoding.c @@ -7,29 +7,61 @@ #include <bencoding.c> #define S0(x) (x ? x : "") int main (int argc, char ** argv) { - if (argc != 1+1) - error_at_line(1, 0, __FILE__, __LINE__, "%s encode < json || %s decode < bencoding", S0(argv[0]), S0(argv[0])); + if (argc < 1+1) + error_at_line(1, 0, __FILE__, __LINE__, "%s encode < json || %s decode < bencoding || %s path path/to/obj < bencoding || %s foreach < bencoding", S0(argv[0]), S0(argv[0]), S0(argv[0]), S0(argv[0])); + if (argv[1][0] == 'p' && argc != 1+2) + error_at_line(1, 0, __FILE__, __LINE__, "set path!"); int size = 2048; int len = 0; char * in = malloc(size); while (!feof(stdin) && !ferror(stdin)) { if (!in) error_at_line(2, 0, __FILE__, __LINE__, "heap alloc failed"); - len += fread(in, 1, size-len-1, stdin); + len += fread(in+len, 1, size-len-1, stdin); if ((size - len) < 1024) in = realloc(in, size *= 2); } if (argv[1][0] == 'e') error_at_line(3, 0, __FILE__, __LINE__, "N/I"); struct bencoding * bencoding = bdecode(in, size, terminate); - len = b2json_length(bencoding); - char out[len+1]; - char * end = b2json(out, bencoding); - *end = '\0'; - puts(out); - if (end - out != len) - error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len); - fprintf(stderr, "len: %d\n", len); + if (argv[1][0] == 'd') { + len = b2json_length(bencoding); + char out[len+1]; + char * end = b2json(out, bencoding); + *end = '\0'; + puts(out); + if (end - out != len) + error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len); + fprintf(stderr, "len: %d\n", len); + } + if (argv[1][0] == 'p') { + len = b2json_length(bpath(bencoding, argv[2])); + char out[len+1]; + char * end = b2json(out, bpath(bencoding, argv[2])); + *end = '\0'; + puts(out); + if (end - out != len) + error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len); + fprintf(stderr, "len: %d\n", len); + } + if (argv[1][0] == 'f') { + bforeach (value, bencoding) { + len = b2json_length(value->key); + char out[len+1]; + char * end = b2json(out, value->key); + *end = '\0'; + if (end - out != len) + error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out, len); + printf("key(%d): %s\n", len, out); + len = b2json_length(value); + char out2[len+1]; + end = b2json(out2, value); + *end = '\0'; + if (end - out2 != len) + error_at_line(4, 0, __FILE__, __LINE__, "b2json wrote %ld instead of %d bytes.", end-out2, len); + printf("value(%d): %s\n", len, out2); + } + } free_bencoding(bencoding); free(in); return 0; |