#ifndef LLARP_BENCODE_H #define LLARP_BENCODE_H #include #include #include #include #include #ifdef __cplusplus extern "C" { #endif static bool INLINE bencode_write_bytestring(llarp_buffer_t* buff, const void* data, size_t sz) { if(!llarp_buffer_writef(buff, "%ld:", sz)) return false; return llarp_buffer_write(buff, data, sz); } static bool INLINE bencode_write_int(llarp_buffer_t* buff, int i) { return llarp_buffer_writef(buff, "i%de", i); } static bool INLINE bencode_write_uint16(llarp_buffer_t* buff, uint16_t i) { return llarp_buffer_writef(buff, "i%de", i); } static bool INLINE bencode_write_int64(llarp_buffer_t* buff, int64_t i) { return llarp_buffer_writef(buff, "i%lde", i); } static bool INLINE bencode_write_uint64(llarp_buffer_t* buff, uint64_t i) { return llarp_buffer_writef(buff, "i%lde", i); } static bool INLINE bencode_write_sizeint(llarp_buffer_t* buff, size_t i) { return llarp_buffer_writef(buff, "i%lde", i); } static bool INLINE bencode_start_list(llarp_buffer_t* buff) { return llarp_buffer_write(buff, "l", 1); } static bool INLINE bencode_start_dict(llarp_buffer_t* buff) { return llarp_buffer_write(buff, "d", 1); } static bool INLINE bencode_end(llarp_buffer_t* buff) { return llarp_buffer_write(buff, "e", 1); } static bool INLINE bencode_write_version_entry(llarp_buffer_t* buff) { return llarp_buffer_writef(buff, "1:vi%de", LLARP_PROTO_VERSION); } static bool INLINE bdecode_read_integer(struct llarp_buffer_t* buffer, uint64_t* result) { size_t len; if(*buffer->cur != 'i') return false; char numbuf[32]; buffer->cur++; len = llarp_buffer_read_until(buffer, 'e', (byte_t*)numbuf, sizeof(numbuf) - 1); if(!len) return false; buffer->cur++; numbuf[len] = 0; *result = atol(numbuf); return true; } static bool INLINE bdecode_read_string(llarp_buffer_t* buffer, llarp_buffer_t* result) { size_t len; int slen; char numbuf[10]; len = llarp_buffer_read_until(buffer, ':', (byte_t*)numbuf, sizeof(numbuf) - 1); if(!len) return false; numbuf[len] = 0; slen = atoi(numbuf); if(slen < 0) return false; buffer->cur++; len = llarp_buffer_size_left(buffer); if(len < slen) return false; result->base = buffer->cur; result->cur = buffer->cur; result->sz = slen; buffer->cur += slen; return true; } struct dict_reader { llarp_buffer_t* buffer; void* user; /** * called when we got a key string, return true to continue iteration * called with null key on done iterating */ bool (*on_key)(struct dict_reader*, llarp_buffer_t*); }; static bool INLINE bdecode_read_dict(llarp_buffer_t* buff, struct dict_reader* r) { llarp_buffer_t strbuf; r->buffer = buff; if(*r->buffer->cur != 'd') return false; r->buffer->cur++; while(llarp_buffer_size_left(r->buffer) && *r->buffer->cur != 'e') { if(bdecode_read_string(r->buffer, &strbuf)) { if(!r->on_key(r, &strbuf)) return false; } } if(*r->buffer->cur != 'e') return false; r->buffer->cur++; return r->on_key(r, 0); } struct list_reader { llarp_buffer_t* buffer; void* user; bool (*on_item)(struct list_reader*, bool); }; static bool INLINE bdecode_read_list(llarp_buffer_t* buff, struct list_reader* r) { r->buffer = buff; if(*r->buffer->cur != 'l') return false; r->buffer->cur++; while(llarp_buffer_size_left(r->buffer) && *r->buffer->cur != 'e') { if(!r->on_item(r, true)) return false; } if(*r->buffer->cur != 'e') return false; r->buffer->cur++; return r->on_item(r, false); } #ifdef __cplusplus } #endif #endif