peer.h

浏览该文件的文档.

#ifndef PEER_H
#define PEER_H

#include "metainfo.h"

#define PSTR_DEFAULT "BitTorrent protocol"

#define PSTRLEN_DEFAULT (sizeof(PSTR_DEFAULT) - 1)

#pragma pack(1)

typedef struct {
    uint8_t hs_pstrlen;               
    char    hs_pstr[19];              
    uint8_t hs_reserved[8];           
    uint8_t hs_info_hash[HASH_SIZE];  
    char    hs_peer_id[HASH_SIZE];    
} PeerHandShake;

enum {
    BT_CHOKE,
    BT_UNCHOKE,
    BT_INTERESTED,
    BT_NOT_INTERESTED,
    BT_HAVE,
    BT_BITFIELD,
    BT_REQUEST,
    BT_PIECE,
    BT_CANCEL,
};

extern const char *bt_types[];

struct PeerMsg {
    uint32_t len;                  
    uint8_t id;                    
    union {
        struct {
            uint32_t piece_index;  
        } have;                    

        uint8_t bitfield[0];       

        struct {
            uint32_t index;        
            uint32_t begin;        
            uint32_t length;       
        } request;                 

        struct {
            uint32_t index;        
            uint32_t begin;        
            uint8_t block[0];      
        } piece;                   

        struct {
            uint32_t index;        
            uint32_t begin;        
            uint32_t length;       
        } cancel;                  
    };
};
#pragma pack()

struct Peer
{
    int fd;                   
    char ip[16];              
    unsigned short port;      
    uint32_t addr;            
    unsigned char *bitfield;  
    int is_choked;            
    int is_interested;        
    int get_choked;           
    int get_interested;       
    int *requested_pieces;    
    int *requested_subpieces; 
    int requesting_index;     
    int requesting_begin;     
    int contribution;         
    unsigned wanted;          
    struct PeerMsg *msg;      
    time_t start_time;        
    double speed;             
};

struct PeerMsg *peer_get_packet(struct Peer *peer);

struct Peer *peer_new(int fd, size_t nr_pieces);

void peer_free(struct Peer **peer);

void print_bit(unsigned char *bytes, size_t bit_len);

void peer_set_bit(struct Peer *peer, unsigned bit_offset);

unsigned char peer_get_bit(struct Peer *peer, unsigned bit_offset);

void peer_send_msg(struct Peer *peer, struct PeerMsg *msg);

#endif  // PEER_H