metainfo.h

浏览该文件的文档.

#ifndef METAINFO_H
#define METAINFO_H

#include <stdio.h>
#include <time.h>
#include <inttypes.h>

#define HASH_SIZE 20

struct BNode;

struct Tracker
{
    char method[10];        
    char host[128];         
    char port[10];          
    char request[128];      
    int sfd;                
    int timerfd;            
};

#define SUB_NA 0

#define SUB_DOWNLOAD 1

#define SUB_FINISH 2

#define WAIT_THRESHOLD 10.0

struct PieceInfo
{
    unsigned char  hash[HASH_SIZE]; 
    int            nr_owners;       
    int            is_downloaded;   

    unsigned char *substate;        
    time_t        *subtimer;        
};

struct WaitPeer
{
    int fd;               
    union {
        uint32_t addr;    
        uint8_t ip[4];    
    };
    uint16_t port;        
    int direction;        
};

struct MetaInfo
{
    size_t file_size;                   
    FILE *file;                         
    unsigned char info_hash[HASH_SIZE]; 

    uint32_t piece_size;                
    size_t nr_pieces;                   
    size_t bitfield_size;               
    uint32_t sub_size;                  
    size_t sub_count;                   
    struct PieceInfo *pieces;           

    unsigned short port;                
    int listen_fd;                      
    int timerfd;                        
    int nr_peers;                       
    struct Peer **peers;                
    int nr_wait_peers;                  
    struct WaitPeer *wait_peers;        
    size_t nr_trackers;                 
    struct Tracker *trackers;           
};

void free_metainfo(struct MetaInfo **pmi);

void extract_trackers(struct MetaInfo *mi, const struct BNode *ast);

void metainfo_load_file(struct MetaInfo *mi, const struct BNode *ast);

void extract_pieces(struct MetaInfo *mi, const struct BNode *ast);

void add_peer(struct MetaInfo *mi, struct Peer *p);

void del_peer_by_fd(struct MetaInfo *mi, int fd);

struct Peer *get_peer_by_fd(struct MetaInfo *mi, int fd);

struct Peer * get_peer_by_addr(struct MetaInfo *mi, uint32_t addr, uint16_t port);

int check_substate(struct MetaInfo *mi, int index);

struct Tracker *get_tracker_by_fd(struct MetaInfo *mi, int sfd);

struct Tracker *get_tracker_by_timer(struct MetaInfo *mi, int timerfd);

void add_wait_peer(struct MetaInfo *mi, int fd, uint32_t addr, uint16_t port);

int get_wait_peer_index_by_fd(struct MetaInfo *mi, int fd);

int find_wait_peer_fd_by_addr(struct MetaInfo *mi, uint32_t addr, uint16_t port);

void rm_wait_peer(struct MetaInfo *mi, int index);

#endif  // METAINFO_H