diff options
Diffstat (limited to 'common.c')
| -rw-r--r-- | common.c | 448 |
1 files changed, 0 insertions, 448 deletions
diff --git a/common.c b/common.c deleted file mode 100644 index 27f2734..0000000 --- a/common.c +++ /dev/null @@ -1,448 +0,0 @@ -#pragma once - -#include <stdbool.h> -#include <stddef.h> -#include <stdint.h> -#include <stdio.h> -#include <string.h> -#include <time.h> -// -#include "omen.c" - -#define likely(A) __builtin_expect(A, 1) -#define unlikely(A) __builtin_expect(A, 0) - -typedef uint8_t byte; - -typedef byte ipv4[ 4 ]; -typedef byte ipv6[ 16 ]; - -typedef byte hash[ 32 ]; -typedef byte public_key[ 32 ]; -typedef byte private_key[ 32 ]; -typedef byte signature[ 64 ]; - -const uint16_t default_port = 8040; - -enum client_input { - // ask server for its information (responds with audit_response) - audit_server = 0x9a, - - // ask server for a random set of neighbors - // (input uint8_t neighbor_count, responds with neighbors_info) - get_neighbors = 0x20, - - // monitor incoming packets - // (input struct firehose_request, responds with firehose_reply) - firehose = 0xf1, - - // give a packet to the relay - // (input struct data_packet, responds with push_packet_result) - push_packet = 0x88, - - // get a packet from the server - // (input hash, responds with get_packet_result) - get_packet = 0x87, - - // register a new neighbor relay to this server - // (input struct relay_neighbor, responds with register_neighbor_result) - register_neighbor = 0x73, -}; - -enum server_reply { - // pretty self-explanatory. - error = 0x1, - - // type `struct relay_info` - audit_response = 0xaf, - // type `struct relay_neighbor[ neighbor_count ]` - neighbors_info = 0x43, - // type `uint64_t new_offset, struct data_packet[ count ]` - firehose_reply = 0x6c, - // type `bool result_code` - push_packet_result = 0x4a, - // type `struct data_packet` - get_packet_result = 0x42, - // type `bool result_code` - register_neighbor_result = 0x6e, -}; - -typedef uint8_t get_neighbors_request; - -struct firehose_request { - uint8_t count; - uint64_t offset; -}; - -typedef uint64_t firehose_reply_new_offset; - -struct relay_neighbor { - bool active; // set to 0 to indicate empty quick connect slot - - bool has_ipv4; // whether the neighbor supports ipv4 connection - bool has_ipv6; // whether the neighbor supports ipv6 connection - ipv4 v4_addr; // ipv4 address of the neighbor (if supported) - ipv6 v6_addr; // ipv6 address of the neighbor (if supported) -}; - -struct relay_info { - // -- version info -- - byte major_version; - byte minor_version; - // -- motd -- - byte motd[ 128 ]; - // -- capabilities -- - struct { - uint16_t retention_hours; // how many hours packets will be retained for - bool can_store; // whether or not the relay is accepting new packets - byte leading_zeros; // relay challenge level - bool accepts_plain; // whether or not the relay accepts unencrypted packets - } capabilities; - // -- statistics -- - struct { - uint16_t neighbors; // current neighboring relay count for this relay - uint64_t held_packets; // how many packets the relay is holding - struct relay_neighbor quick_connect[ 4 ]; // four best neighbors for quick connection - } statistics; -}; - -// do not change -#define MAX_RECIPIENTS 32 - -struct data_packet { - // nonce + hash for PoW - uint32_t nonce; - hash hash; - - // content inside to be nonce + hashed - struct { - bool encrypted; // whether the contents is encrypted or plaintext - uint64_t created_time_ms; // unix timestamp for creation time in milliseconds - public_key author; // message author - public_key author_signing_pub; // message signing pubkey - uint8_t recipient_count; // number of recipients. leave as 0 for public. - public_key recipients[ 32 ]; // up to 32 recipients for packet - byte data[ 65536 ]; // packet data - signature signature; // data signature - } contents; -}; - -const size_t CONTENTS_SIZE = 1 + 8 + 32 + 32 + 1 + (32 * MAX_RECIPIENTS) + 65536 + 64; -const size_t PACKET_SIZE = 4 + 32 + (CONTENTS_SIZE); - -struct relay_neighbor s2relay_neighbor(size_t len, byte *stream, bool *success) { - if (len != 1 + 1 + 1 + 4 + 16) { - *success = false; - return (struct relay_neighbor) { 0 }; - } - - struct relay_neighbor out; - out.active = stream[ 0 ] != 0; - out.has_ipv4 = stream[ 1 ] != 0; - out.has_ipv6 = stream[ 2 ] != 0; - - memcpy(out.v4_addr, stream + 3, 4); - memcpy(out.v6_addr, stream + 7, 16); - - *success = true; - return out; -} - -void relay_neighbor2s(struct relay_neighbor data, byte *stream) { - stream[ 0 ] = data.active ? 1 : 0; - stream[ 1 ] = data.has_ipv4 ? 1 : 0; - stream[ 2 ] = data.has_ipv6 ? 1 : 0; - - memcpy(stream + 3, data.v4_addr, 4); - memcpy(stream + 7, data.v6_addr, 16); -} - -struct relay_info s2relay_info(size_t len, byte *stream, bool *success) { - if (len != 1 + 1 + 128 + 2 + 1 + 1 + 1 + 2 + 8 + (4 * (1 + 1 + 1 + 4 + 16))) { - *success = false; - return (struct relay_info) { 0 }; - } - - struct relay_info out; - size_t off = 0; - - out.major_version = stream[ off++ ]; - out.minor_version = stream[ off++ ]; - - memcpy(out.motd, stream + off, 128); - off += 128; - - out.capabilities.retention_hours = stream[ off ] + (stream[ off + 1 ] << 8); - off += 2; - out.capabilities.can_store = stream[ off++ ] != 0; - out.capabilities.leading_zeros = stream[ off++ ]; - out.capabilities.accepts_plain = stream[ off++ ] != 0; - - out.statistics.neighbors = stream[ off ] + (stream[ off + 1 ] << 8); - off += 2; - out.statistics.held_packets - = ((uint64_t) stream[ off ]) | ((uint64_t) stream[ off + 1 ] << 8) - | ((uint64_t) stream[ off + 2 ] << 16) | ((uint64_t) stream[ off + 3 ] << 24) - | ((uint64_t) stream[ off + 4 ] << 32) | ((uint64_t) stream[ off + 5 ] << 40) - | ((uint64_t) stream[ off + 6 ] << 48) | ((uint64_t) stream[ off + 7 ] << 56); - off += 8; - - for (int i = 0; i < 4; i++) { - bool ok; - out.statistics.quick_connect[ i ] = s2relay_neighbor(1 + 1 + 1 + 4 + 16, stream + off, &ok); - if (!ok) { - *success = false; - return (struct relay_info) { 0 }; - } - off += 1 + 1 + 1 + 4 + 16; - } - - *success = true; - return out; -} - -void relay_info2s(struct relay_info data, byte *stream) { - size_t off = 0; - - stream[ off++ ] = data.major_version; - stream[ off++ ] = data.minor_version; - - memcpy(stream + off, data.motd, 128); - off += 128; - - stream[ off++ ] = data.capabilities.retention_hours & 0xFF; - stream[ off++ ] = (data.capabilities.retention_hours >> 8) & 0xFF; - stream[ off++ ] = data.capabilities.can_store ? 1 : 0; - stream[ off++ ] = data.capabilities.leading_zeros; - stream[ off++ ] = data.capabilities.accepts_plain ? 1 : 0; - - stream[ off++ ] = data.statistics.neighbors & 0xFF; - stream[ off++ ] = (data.statistics.neighbors >> 8) & 0xFF; - - stream[ off++ ] = data.statistics.held_packets & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 8) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 16) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 24) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 32) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 40) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 48) & 0xFF; - stream[ off++ ] = (data.statistics.held_packets >> 56) & 0xFF; - - for (int i = 0; i < 4; i++) { - relay_neighbor2s(data.statistics.quick_connect[ i ], stream + off); - off += 1 + 1 + 1 + 4 + 16; - } -} - -struct data_packet s2data_packet(size_t len, byte *stream, bool *success) { - *success = false; - struct data_packet out = { 0 }; - if (len != PACKET_SIZE) { - printf("s2data_packet: length mismatch. expected %zu, got %zu\n", PACKET_SIZE, len); - return out; - } - - size_t off = 0; - - out.nonce = stream[ off ] | (stream[ off + 1 ] << 8) | (stream[ off + 2 ] << 16) - | (stream[ off + 3 ] << 24); - off += 4; - - memcpy(out.hash, stream + off, 32); - off += 32; - - out.contents.encrypted = stream[ off++ ] != 0; - - out.contents.created_time_ms - = ((uint64_t) stream[ off ]) | ((uint64_t) stream[ off + 1 ] << 8) - | ((uint64_t) stream[ off + 2 ] << 16) | ((uint64_t) stream[ off + 3 ] << 24) - | ((uint64_t) stream[ off + 4 ] << 32) | ((uint64_t) stream[ off + 5 ] << 40) - | ((uint64_t) stream[ off + 6 ] << 48) | ((uint64_t) stream[ off + 7 ] << 56); - off += 8; - - memcpy(out.contents.author, stream + off, 32); - off += 32; - - memcpy(out.contents.author_signing_pub, stream + off, 32); - off += 32; - - out.contents.recipient_count = stream[ off++ ]; - memcpy(out.contents.recipients, stream + off, 32 * MAX_RECIPIENTS); - off += 32 * MAX_RECIPIENTS; - - memcpy(out.contents.data, stream + off, 65536); - off += 65536; - - memcpy(out.contents.signature, stream + off, 64); - off += 64; - - *success = true; - return out; -} - -bool data_packet2s(struct data_packet data, byte *stream) { - if (data.contents.recipient_count > MAX_RECIPIENTS) { return false; } - - size_t off = 0; - - stream[ off++ ] = data.nonce & 0xFF; - stream[ off++ ] = (data.nonce >> 8) & 0xFF; - stream[ off++ ] = (data.nonce >> 16) & 0xFF; - stream[ off++ ] = (data.nonce >> 24) & 0xFF; - - memcpy(stream + off, data.hash, 32); - off += 32; - - stream[ off++ ] = data.contents.encrypted ? 1 : 0; - - for (int i = 0; i < 8; i++) stream[ off++ ] = (data.contents.created_time_ms >> (i * 8)) & 0xFF; - - memcpy(stream + off, data.contents.author, 32); - off += 32; - - memcpy(stream + off, data.contents.author_signing_pub, 32); - off += 32; - - stream[ off++ ] = data.contents.recipient_count; - memcpy(stream + off, data.contents.recipients, 32 * MAX_RECIPIENTS); - off += 32 * MAX_RECIPIENTS; - - memcpy(stream + off, data.contents.data, 65536); - off += 65536; - - memcpy(stream + off, data.contents.signature, 64); - off += 64; - - if (off != PACKET_SIZE) { - printf("data_packet2s: size mismatch wrote %zu expected %zu\n", off, PACKET_SIZE); - return false; - } - - return true; -} - -get_neighbors_request s2get_neighbors_request(size_t len, byte *stream, bool *success) { - if (len != 1) { - *success = false; - return 0; - } - *success = true; - return stream[ 0 ]; -} - -void get_neighbors_request2s(get_neighbors_request data, byte *stream) { stream[ 0 ] = data; } - -struct firehose_request s2firehose_request(size_t len, byte *stream, bool *success) { - if (len != 1 + 8) { - *success = false; - return (struct firehose_request) { 0 }; - } - struct firehose_request out; - out.count = stream[ 0 ]; - out.offset = ((uint64_t) stream[ 1 ]) | ((uint64_t) stream[ 2 ] << 8) - | ((uint64_t) stream[ 3 ] << 16) | ((uint64_t) stream[ 4 ] << 24) - | ((uint64_t) stream[ 5 ] << 32) | ((uint64_t) stream[ 6 ] << 40) - | ((uint64_t) stream[ 7 ] << 48) | ((uint64_t) stream[ 8 ] << 56); - *success = true; - return out; -} - -void firehose_request2s(struct firehose_request data, byte *stream) { - stream[ 0 ] = data.count; - stream[ 1 ] = data.offset & 0xFF; - stream[ 2 ] = (data.offset >> 8) & 0xFF; - stream[ 3 ] = (data.offset >> 16) & 0xFF; - stream[ 4 ] = (data.offset >> 24) & 0xFF; - stream[ 5 ] = (data.offset >> 32) & 0xFF; - stream[ 6 ] = (data.offset >> 40) & 0xFF; - stream[ 7 ] = (data.offset >> 48) & 0xFF; - stream[ 8 ] = (data.offset >> 56) & 0xFF; -} - -void s2hash(size_t len, byte *stream, bool *success, hash out) { - if (len != 32) { - *success = false; - return; - } - memcpy(out, stream, 32); - *success = true; -} - -void hash2s(hash data, byte *stream) { memcpy(stream, data, 32); } - -bool check_leading_zeros(const hash _hash, byte required_zeros) { - if (required_zeros == 0) return true; - - int full_bytes = required_zeros / 8; - int remaining_bits = required_zeros % 8; - - for (int i = 0; i < full_bytes; i++) { - if (_hash[ i ] != 0) return false; - } - - if (remaining_bits > 0) { - byte mask = (0xFF << (8 - remaining_bits)); - if ((_hash[ full_bytes ] & mask) != 0) return false; - } - - return true; -} - -uint64_t get_current_time_ms() { - struct timespec ts; - clock_gettime(CLOCK_REALTIME, &ts); - return (uint64_t) ts.tv_sec * 1000 + (uint64_t) ts.tv_nsec / 1000000; -} - -#define RELAY_NEIGHBOR_WIRE_LEN (1 + 1 + 1 + 4 + 16) - -static inline uint64_t read_u64_le(const byte *b) { - uint64_t v = 0; - for (int i = 0; i < 8; ++i) v |= (uint64_t) b[ i ] << (8 * i); - return v; -} - -static inline void write_u64_le(byte *b, uint64_t v) { - for (int i = 0; i < 8; ++i) b[ i ] = (v >> (8 * i)) & 0xFF; -} - -static inline uint32_t read_u32_le(const byte *b) { - uint32_t v = 0; - for (int i = 0; i < 4; ++i) v |= (uint32_t) b[ i ] << (8 * i); - return v; -} - -static inline void write_u32_le(byte *b, uint32_t v) { - for (int i = 0; i < 4; ++i) b[ i ] = (v >> (8 * i)) & 0xFF; -} - -static inline uint16_t read_u16_le(const byte *b) { - return (uint16_t) b[ 0 ] | ((uint16_t) b[ 1 ] << 8); -} - -static inline void write_u16_le(byte *b, uint16_t v) { - b[ 0 ] = v & 0xFF; - b[ 1 ] = (v >> 8) & 0xFF; -} - -struct relay_info run_server_audit(const char* ip, int port) { - byte data[] = { audit_server, 0, 0, 0, 0, 0, 0, 0 }; - byte* response = NULL; - size_t resp_len = 0; - bool success; - - int ret = study(ip, port, data, sizeof(data), &response, &resp_len); - if (ret != 0) { - printf("error: study returned %i\n", ret); - if (response) free(response); - return (struct relay_info) { 0 }; - } - - struct relay_info info = s2relay_info(resp_len - 1, response + 1, &success); - if (!success) { printf("error: failed to deserialize relay_info\n"); } - - free(response); - return info; -} - -pthread_mutex_t packet_storage_mutex = PTHREAD_MUTEX_INITIALIZER;
\ No newline at end of file |