librats 1.0.2 → 2.1.4

package/native-src/src/bt_network.h

@@ -1,417 +0,0 @@
-#pragma once
-
-/**
- * @file bt_network.h
- * @brief Async network layer for BitTorrent peer connections
- * 
- * Provides efficient multiplexed I/O for managing multiple peer connections
- * using platform-optimal I/O polling (epoll/kqueue/WSAPoll).
- * Buffers are owned by BtPeerConnection (single source of truth) -
- * this class only handles socket I/O.
- */
-
-#include "bt_types.h"
-#include "bt_peer_connection.h"
-#include "io_poller.h"
-#include "socket.h"
-
-#include <vector>
-#include <queue>
-#include <unordered_map>
-#include <memory>
-#include <mutex>
-#include <atomic>
-#include <thread>
-#include <functional>
-#include <chrono>
-
-namespace librats {
-
-//=============================================================================
-// Network Configuration
-//=============================================================================
-
-/**
- * @brief Configuration for the network manager
- */
-struct BtNetworkConfig {
-    uint16_t listen_port;           ///< Port for incoming connections (0 = random)
-    size_t max_connections;         ///< Maximum total connections
-    bool enable_incoming;           ///< Accept incoming connections
-    int connect_timeout_ms;         ///< Timeout for outgoing connections (ms)
-    int poll_timeout_ms;            ///< Timeout for I/O poll (ms)
-    size_t send_buffer_high_water;  ///< High water mark for send buffer
-    PeerID peer_id;                 ///< Our peer ID (for incoming connections)
-    
-    BtNetworkConfig()
-        : listen_port(6881)
-        , max_connections(200)
-        , enable_incoming(true)
-        , connect_timeout_ms(30000)
-        , poll_timeout_ms(15)
-        , send_buffer_high_water(1024 * 1024)  // 1MB
-        , peer_id{} {}
-};
-
-//=============================================================================
-// Connection State
-//=============================================================================
-
-/**
- * @brief State of a network connection
- */
-enum class NetConnectionState : uint8_t {
-    Connecting,     ///< TCP connect in progress
-    Handshaking,    ///< Waiting for BitTorrent handshake
-    Connected,      ///< Fully connected and operational
-    Closing         ///< Connection being closed
-};
-
-/**
- * @brief Information about a pending outgoing connection
- */
-struct PendingConnect {
-    std::string ip;
-    uint16_t port;
-    BtInfoHash info_hash;
-    PeerID peer_id;
-    uint32_t num_pieces;
-    socket_t socket;
-    std::chrono::steady_clock::time_point start_time;
-    
-    PendingConnect() : port(0), num_pieces(0), socket(INVALID_SOCKET_VALUE) {}
-};
-
-/**
- * @brief Per-socket connection context
- * 
- * Note: Buffers are now owned by BtPeerConnection (single source of truth).
- * This struct only holds socket state and connection reference.
- */
-struct SocketContext {
-    socket_t socket;
-    BtInfoHash info_hash;
-    std::shared_ptr<BtPeerConnection> connection;
-    NetConnectionState state;
-    bool incoming;
-    std::chrono::steady_clock::time_point connected_at;
-    std::chrono::steady_clock::time_point last_activity;
-    
-    SocketContext()
-        : socket(INVALID_SOCKET_VALUE)
-        , state(NetConnectionState::Connecting)
-        , incoming(false) {}
-};
-
-/**
- * @brief Per-torrent registration info
- */
-struct TorrentRegistration {
-    BtInfoHash info_hash;
-    PeerID peer_id;
-    uint32_t num_pieces;
-    
-    TorrentRegistration() : num_pieces(0) {}
-};
-
-//=============================================================================
-// Deferred Callback Events (to avoid deadlocks)
-//=============================================================================
-
-/**
- * @brief Event for connected callback (deferred)
- */
-struct ConnectedEvent {
-    BtInfoHash info_hash;
-    std::shared_ptr<BtPeerConnection> connection;
-    socket_t socket;
-    bool incoming;
-};
-
-/**
- * @brief Event for data callback (deferred)
- */
-struct DataEvent {
-    BtInfoHash info_hash;
-    std::shared_ptr<BtPeerConnection> connection;
-    socket_t socket;
-};
-
-/**
- * @brief Event for disconnected callback (deferred)
- */
-struct DisconnectedEvent {
-    BtInfoHash info_hash;
-    std::shared_ptr<BtPeerConnection> connection;
-};
-
-//=============================================================================
-// Network Manager
-//=============================================================================
-
-/**
- * @brief Manages all BitTorrent peer connections
- * 
- * Provides async I/O via platform-optimal multiplexing:
- * - Linux:   epoll   (O(1) per event)
- * - macOS:   kqueue  (O(1) per event)
- * - Windows: WSAPoll (no FD_SETSIZE limit)
- * 
- * Features:
- * - Listens for incoming connections
- * - Manages outgoing connection queue
- * - Handles non-blocking reads/writes
- * - TCP_NODELAY for low-latency messaging
- * - Invokes callbacks for events
- * 
- * Thread-safety: The manager runs its own I/O thread.
- * Callbacks are invoked from the I/O thread.
- * Public methods are thread-safe.
- */
-class BtNetworkManager {
-public:
-    //=========================================================================
-    // Types
-    //=========================================================================
-    
-    /// Callback when a peer connection is established (handshake complete)
-    using ConnectedCallback = std::function<void(
-        const BtInfoHash& info_hash,
-        std::shared_ptr<BtPeerConnection> connection,
-        socket_t socket,
-        bool incoming
-    )>;
-    
-    /// Callback when a peer disconnects
-    using DisconnectedCallback = std::function<void(
-        const BtInfoHash& info_hash,
-        BtPeerConnection* connection
-    )>;
-    
-    /// Callback when data is received and processed
-    using DataCallback = std::function<void(
-        const BtInfoHash& info_hash,
-        BtPeerConnection* connection,
-        socket_t socket
-    )>;
-    
-    //=========================================================================
-    // Construction
-    //=========================================================================
-    
-    /**
-     * @brief Create network manager with config
-     */
-    explicit BtNetworkManager(const BtNetworkConfig& config);
-    
-    /**
-     * @brief Destructor - stops the manager
-     */
-    ~BtNetworkManager();
-    
-    // Non-copyable
-    BtNetworkManager(const BtNetworkManager&) = delete;
-    BtNetworkManager& operator=(const BtNetworkManager&) = delete;
-    
-    //=========================================================================
-    // Lifecycle
-    //=========================================================================
-    
-    /**
-     * @brief Start the network manager
-     * 
-     * Creates listen socket and starts I/O thread.
-     * 
-     * @return true if started successfully
-     */
-    bool start();
-    
-    /**
-     * @brief Stop the network manager
-     * 
-     * Closes all connections and stops I/O thread.
-     */
-    void stop();
-    
-    /**
-     * @brief Check if manager is running
-     */
-    bool is_running() const { return running_; }
-    
-    /**
-     * @brief Get the actual listen port
-     */
-    uint16_t listen_port() const { return actual_listen_port_; }
-    
-    //=========================================================================
-    // Torrent Registration
-    //=========================================================================
-    
-    /**
-     * @brief Register a torrent to accept connections for
-     * 
-     * @param info_hash Torrent info hash
-     * @param peer_id Our peer ID
-     * @param num_pieces Number of pieces in torrent
-     */
-    void register_torrent(const BtInfoHash& info_hash, 
-                          const PeerID& peer_id,
-                          uint32_t num_pieces);
-    
-    /**
-     * @brief Unregister a torrent
-     * 
-     * Closes all connections for this torrent.
-     */
-    void unregister_torrent(const BtInfoHash& info_hash);
-    
-    //=========================================================================
-    // Connection Management
-    //=========================================================================
-    
-    /**
-     * @brief Initiate connection to a peer
-     * 
-     * @param ip Peer IP address
-     * @param port Peer port
-     * @param info_hash Torrent info hash
-     * @param peer_id Our peer ID
-     * @param num_pieces Number of pieces (0 for magnet links)
-     * @return true if connection was queued
-     */
-    bool connect_peer(const std::string& ip, uint16_t port,
-                      const BtInfoHash& info_hash,
-                      const PeerID& peer_id,
-                      uint32_t num_pieces);
-    
-    /**
-     * @brief Close a connection
-     * 
-     * @param socket Socket to close
-     */
-    void close_connection(socket_t socket);
-    
-    /**
-     * @brief Send data to a peer
-     * 
-     * Queues data in the send buffer; actual send is async.
-     * 
-     * @param socket Target socket
-     * @param data Data to send
-     * @return true if queued successfully
-     */
-    bool send_to_peer(socket_t socket, const std::vector<uint8_t>& data);
-    
-    /**
-     * @brief Get number of active connections
-     */
-    size_t connection_count() const;
-    
-    /**
-     * @brief Get number of pending outgoing connections
-     */
-    size_t pending_connect_count() const;
-    
-    //=========================================================================
-    // Callbacks
-    //=========================================================================
-    
-    void set_connected_callback(ConnectedCallback cb) { 
-        on_connected_ = std::move(cb); 
-    }
-    
-    void set_disconnected_callback(DisconnectedCallback cb) { 
-        on_disconnected_ = std::move(cb); 
-    }
-    
-    void set_data_callback(DataCallback cb) { 
-        on_data_ = std::move(cb); 
-    }
-    
-private:
-    //=========================================================================
-    // Internal Methods
-    //=========================================================================
-    
-    /// Main I/O loop (runs in separate thread)
-    void io_loop();
-    
-    /// Synchronize poller state with current connections (under mutex)
-    void sync_poller();
-    
-    /// Process the queue of pending connections
-    void process_pending_connects();
-    
-    /// Accept incoming connections
-    void accept_incoming();
-    
-    /// Handle readable socket (returns true if connection should be closed)
-    bool handle_readable(socket_t socket,
-                        std::vector<ConnectedEvent>& connected_events,
-                        std::vector<DataEvent>& data_events);
-    
-    /// Handle writable socket (drain send buffer, returns true if connection should be closed)
-    bool handle_writable(socket_t socket);
-    
-    /// Handle data received from a peer (returns true if connection should be closed)
-    bool handle_peer_data(SocketContext& ctx,
-                         std::vector<ConnectedEvent>& connected_events,
-                         std::vector<DataEvent>& data_events);
-    
-    /// Close connection with cleanup (collects disconnected events)
-    void close_connection_internal(socket_t socket,
-                                  std::vector<DisconnectedEvent>& disconnected_events);
-    
-    /// Flush send buffers for a socket (returns true if connection should be closed)
-    bool flush_send_buffer(SocketContext& ctx);
-    
-    /// Create socket for outgoing connection
-    socket_t create_connect_socket(const std::string& ip, uint16_t port);
-    
-    /// Handle info_hash discovered from incoming connection handshake
-    void on_incoming_info_hash(BtPeerConnection* conn, const BtInfoHash& info_hash);
-    
-    /// Set TCP_NODELAY on a socket (disable Nagle's algorithm)
-    static bool set_tcp_nodelay(socket_t sock);
-    
-    //=========================================================================
-    // Data Members
-    //=========================================================================
-    
-    BtNetworkConfig config_;
-    
-    std::atomic<bool> running_;
-    std::atomic<uint16_t> actual_listen_port_;
-    
-    socket_t listen_socket_;
-    
-    mutable std::mutex mutex_;
-    
-    /// Platform-optimal I/O poller (epoll/kqueue/WSAPoll)
-    std::unique_ptr<IOPoller> poller_;
-    
-    /// Tracks current poller event mask per socket (to avoid redundant modify calls)
-    std::unordered_map<socket_t, uint32_t> poller_state_;
-    
-    /// Active connections by socket
-    std::unordered_map<socket_t, SocketContext> connections_;
-    
-    /// Registered torrents by info hash
-    std::unordered_map<BtInfoHash, TorrentRegistration, InfoHashHash> torrents_;
-    
-    /// Queue of pending outgoing connections
-    std::queue<PendingConnect> pending_connects_;
-    
-    /// Pending connects that are in progress (by socket)
-    std::unordered_map<socket_t, PendingConnect> connecting_;
-    
-    /// I/O thread
-    std::thread io_thread_;
-    
-    /// Callbacks
-    ConnectedCallback on_connected_;
-    DisconnectedCallback on_disconnected_;
-    DataCallback on_data_;
-};
-
-} // namespace librats