Merge pull request #10 from milosz275/blockchain

Blockchain

Author: himokkk

.gitignore

@@ -49,6 +49,8 @@ CMakeCache.txt
 *.log
 logs/
 logs
+history/
+history
 
 # VSCode
 *.vc.*
@@ -58,3 +60,9 @@ logs
 
 # BitLab history
 .*history.txt
+
+# Testing
+*.test
+
+# Blockchain
+*.dat

.vscode/settings.json

@@ -11,6 +11,7 @@
         "getdata",
         "getheaders",
         "getip",
+        "htole",
         "ipver",
         "libreadline",
         "lineptr",
@@ -45,7 +46,8 @@
         "bit": "c",
         "compare": "c",
         "limits": "c",
-        "new": "c"
+        "new": "c",
+        "errno.h": "c"
     },
     "C_Cpp.formatting": "vcFormat",
     "C_Cpp.autoAddFileAssociations": true,
@@ -57,4 +59,4 @@
         "editor.formatOnType": true,
         "editor.formatOnSave": true
     }
-}
+}

Doxyfile

@@ -937,6 +937,7 @@ INPUT_ENCODING         = UTF-8
 # *.vhdl, *.ucf, *.qsf and *.ice.
 
 FILE_PATTERNS          = *.c \
+                         *.cpp \
                          *.h \
                          *.md
 

README.md

@@ -33,7 +33,6 @@ network's inner workings using the Bitcoin protocol.
 - [Usage](#usage)
 - [Config directory](#config-directory)
 - [Features](#features)
-- [Acknowledgements](#acknowledgements)
 - [License](#license)
 
 ## Installation
@@ -109,12 +108,11 @@ the following files:
 - `blocks.dat`: Block list file
 - `txs.dat`: Transaction list file -->
 
-Please feel free to link logs to your working directory:
+Please feel free to link logs and history to your current working directory:
 
 ```bash
-mkdir -p logs
-ln -s ~/.bitlab/logs/bitlab.log logs/bitlab.log
-ln -s ~/.bitlab/history/cli_history.txt logs/cli_history.txt
+ln -s ~/.bitlab/logs ./logs
+ln -s ~/.bitlab/history ./history
 ```
 
 > [!NOTE]
@@ -157,8 +155,7 @@ Stay connected with active peers and monitor peer availability:
 - **Network Alerts:**
         - Send `alert` messages to notify peers of important network events.
         - `alert` has been deprecated because of security risks so it won't be implemented
-        in this
-        app. [documentation thread](https://bitcoin.org/en/alert/2016-11-01-alert-retirement#reasons-for-retirement)
+        in this app. [documentation thread](https://bitcoin.org/en/alert/2016-11-01-alert-retirement#reasons-for-retirement)
 
 ### 4. Error Handling & Diagnostics
 
@@ -191,10 +188,6 @@ Efficiently share and request blocks and transactions with peers:
         - `block` message: Send or advertise a specific block.
         - `headers` message: Share up to 2,000 block headers for faster synchronization.
 
-## Acknowledgements
-
-- [Thread-safe logging](https://github.com/milosz275/secure-chat/blob/main/common/include/log.h)
-
 ## License
 
 This project is licensed under the MIT License - see

bitlab/include/cli.h

@@ -113,7 +113,7 @@ int cli_info(char** args);
 /**
  * Discovers Bitcoin peers.
  *
- * @param args The arguments passed to the function should be empty.
+ * @param args The arguments passed to the function should match those specified in 'help peerdiscovery'.
  * @return The exit code.
  */
 int cli_peer_discovery(char** args);
@@ -129,7 +129,7 @@ int cli_clear(char** args);
 /**
  * Prints CLI command help.
  *
- * @param args The arguments passed to this function should be empty.
+ * @param args The arguments passed to this function can be empty to get full program help or argument should specify one comand to get help for.
  * @return The exit code.
  */
 int cli_help(char** args);
@@ -177,6 +177,25 @@ int cli_getaddr(char** args);
  */
 int cli_disconnect(char** args);
 
+/**
+ * @brief Sends a 'getheaders' message to specified peer.
+ * 
+ * @param args The index of the peer.
+ * @return The exit code.
+ */
+int cli_getheaders(char** args);
+
+/**
+ * @brief Sends a 'getblocks' message to specified peer.
+ *
+ * @param args The index of the peer.
+ * @return The exit code.
+ */
+int cli_getblocks(char** args);
+
+int cli_inv(char** args);
+
+
 //// LINE HANDLING FUNCTIONS ////
 
 /**

bitlab/include/peer_connection.h

@@ -1,6 +1,66 @@
 #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)
+
+/**
+ * @brief The structure to store information about a connected peer.
+ *
+ * @param ip_address The IP address of the peer.
+ * @param port The port of the peer.
+ * @param socket_fd The socket file descriptor for communication.
+ * @param thread The thread assigned to this connection.
+ * @param is_connected The connection status.
+ * @param operation_in_progress The operation status.
+ * @param compact_blocks Does peer want to use compact blocks.
+ * @param fee_rate Min fee rate in sat/kB of transaction that peer allows.
+ */
+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];
+
 /**
  * @brief Lists all connected nodes and their details.
  *
@@ -9,6 +69,14 @@
  */
 void list_connected_nodes();
 
+/**
+ * @brief Get the index of the node with the given IP address.
+ *
+ * @param ip_address The IP address of the node.
+ * @return The index of the node in the global array.
+ */
+int get_idx(const char* ip_address);
+
 /**
  * @brief Sends a 'getaddr' message to the peer and waits for a response.
  *
@@ -39,4 +107,68 @@ 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);
+
+/**
+ * @brief Sends a 'getheaders' message to the peer and waits for a response.
+ *
+ * This function sends a 'getheaders' message to the peer identified by the given index
+ * and waits for a response. It is used to request a list of known peers from the connected peer.
+ *
+ * @param idx The index of the peer in the nodes array.
+ */
+void send_getheaders_and_wait(int idx);
+
+/**
+ * @brief Sends a 'headers' message to the peer.
+ *
+ * This function sends a 'headers' message to the peer identified by the given index.
+ * It retrieves the block headers from the local storage starting from the specified
+ * start hash up to the stop hash or the maximum number of headers allowed.
+ *
+ * @param idx The index of the peer in the nodes array.
+ * @param start_hash The hash of the first block header to send.
+ * @param stop_hash The hash of the last block header to send.
+ */
+void send_headers(int idx, const unsigned char* start_hash, const unsigned char* stop_hash);
+
+/**
+ * @brief Sends a 'getblocks' message to the peer and waits for a response.
+ *
+ * This function sends a 'getblocks' message to the peer identified by the given index
+ * and waits for a response. It is used to request a list of blocks from the connected peer.
+ * The response is processed and the blocks are saved to a file.
+ *
+ * @param idx The index of the peer in the nodes array.
+ */
+void send_getblocks_and_wait(int idx);
+
+/**
+ * @brief Sends a 'getdata' message to the peer and waits for a response.
+ *
+ * This function sends a 'getdata' message to the peer identified by the given index
+ * and waits for a response. It is used to request specific blocks or transactions
+ * from the connected peer based on the provided hashes. The response is saved to a file
+ * and logged to the Bitlab logs.
+ *
+ * @param idx The index of the peer in the nodes array.
+ * @param hashes An array of hashes representing the blocks or transactions to request.
+ * @param hash_count The number of hashes in the array.
+ */
+void send_getdata_and_wait(int idx, const unsigned char* hashes, size_t hash_count);
+
+/**
+ * @brief Sends an 'inv' message to the peer and waits for a response.
+ *
+ * This function sends an 'inv' message to the peer identified by the given index
+ * and waits for a response. It is used to advertise the knowledge of one or more objects
+ * (blocks or transactions). The inventory data is provided as input to the function.
+ *
+ * @param idx The index of the peer in the nodes array.
+ * @param inv_data An array of inventory vectors (type + hash).
+ * @param inv_count The number of inventory vectors in the array.
+ */
+void send_inv_and_wait(int idx, const unsigned char* inv_data, size_t inv_count);
+
 #endif // __PEER_CONNECTION_H