peer_connection.h

Go to the documentation of this file.

#ifndef __PEER_CONNECTION_H
#define __PEER_CONNECTION_H
 
#include <stdint.h>
#include <pthread.h>
 
// Maximum number of peers to track
#define MAX_NODES 100
 
// Bitcoin mainnet magic bytes:
#define BITCOIN_MAINNET_MAGIC 0xD9B4BEF9
 
// Default Bitcoin mainnet port:
#define BITCOIN_MAINNET_PORT 8333
 
#define htole16(x) ((uint16_t)((((x) & 0xFF) << 8) | (((x) >> 8) & 0xFF)))
#define htole32(x) ((uint32_t)((((x) & 0xFF) << 24) | (((x) >> 8) & 0xFF00) | (((x) >> 16) & 0xFF) | (((x) >> 24) & 0xFF000000)))
#define htole64(x) ((uint64_t)((((x) & 0xFF) << 56) | (((x) >> 8) & 0xFF00) | (((x) >> 16) & 0xFF0000) | (((x) >> 24) & 0xFF000000) | (((x) >> 32) & 0xFF00000000) | (((x) >> 40) & 0xFF0000000000) | (((x) >> 48) & 0xFF000000000000) | (((x) >> 56) & 0xFF00000000000000)))
 
#define MAX_LOCATOR_COUNT 10
#define GENESIS_BLOCK_HASH "0000000000000000000000000000000000000000000000000000000000000000"
#define HEADERS_FILE "headers.dat"
#define MAX_HEADERS_COUNT 2000
 
// Structure for Bitcoin P2P message header (24 bytes).
// For reference: https://en.bitcoin.it/wiki/Protocol_documentation#Message_structure
#pragma pack(push, 1)
typedef struct
{
    unsigned int magic; // Magic value indicating message origin network
    char command[12]; // ASCII command (null-padded)
    unsigned int length; // Payload size (little-endian)
    unsigned char checksum[4]; // First 4 bytes of double SHA-256 of the payload
} bitcoin_msg_header;
#pragma pack(pop)
 
typedef struct
{
    char ip_address[64];
    uint16_t port;
    int socket_fd;
    pthread_t thread;
    int is_connected;
    int operation_in_progress;
    uint64_t compact_blocks;
    uint64_t fee_rate;
} Node;
 
// global array of nodes
extern Node nodes[MAX_NODES];
 
void list_connected_nodes();
 
int get_idx(const char* ip_address);
 
void send_getaddr_and_wait(int idx);
 
int connect_to_peer(const char* ip_addr);
 
void disconnect(int node_id);
 
 
unsigned char* load_blocks_from_file(const char* filename, size_t* payload_len);
 
void send_getheaders_and_wait(int idx);
 
void send_headers(int idx, const unsigned char* start_hash, const unsigned char* stop_hash);
 
void send_getblocks_and_wait(int idx);
 
void send_getdata_and_wait(int idx, const unsigned char* hashes, size_t hash_count);
 
void send_inv_and_wait(int idx, const unsigned char* inv_data, size_t inv_count);
 
void send_tx(int idx, const unsigned char* tx_data, size_t tx_size);
 
#endif // __PEER_CONNECTION_H