主函数的设计和实现
主函数main调用前面所提到的模块的函数,主要功能是对下载和上传。
main.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <malloc.h>
#include "data.h"
#include "tracker.h"
#include "bitfield.h"
#include "torrent.h"
#include "parse_metafile.h"
#include "signal_hander.h"
#include "policy.h"
#include "log.h"
// #define DEBUG
int main(int argc, char *argv[])
{
int ret;
if(argc != 2) {
printf("usage:%s metafile\n",argv[0]);
exit(-1);
}
// 设置信号处理函数
ret = set_signal_hander();
if(ret != 0) { printf("%s:%d error\n",__FILE__,__LINE__); return -1; }
// 解析种子文件
ret = parse_metafile(argv[1]);
if(ret != 0) { printf("%s:%d error\n",__FILE__,__LINE__); return -1; }
// 初始化非阻塞peer
init_unchoke_peers();
// 创建用于保存下载数据的文件
ret = create_files();
if(ret != 0) { printf("%s:%d error\n",__FILE__,__LINE__); return -1; }
// 创建位图
ret = create_bitfield();
if(ret != 0) { printf("%s:%d error\n",__FILE__,__LINE__); return -1; }
// 创建缓冲区
ret = create_btcache();
if(ret != 0) { printf("%s:%d error\n",__FILE__,__LINE__); return -1; }
// 负责与所有Peer收发数据、交换消息
download_upload_with_peers();
// 做一些清理工作,主要是释放动态分配的内存
do_clear_work();
return 0;
}
为了编译的方便编写了Makefile文件。
Makefile
CC=gcc
CFLAGS= -Iinclude -Wall -g -DDEBUG
LDFLAGS=-L./lib -Wl,-rpath=./lib -Wl,-rpath=/usr/local/lib
ttorrent: main.o parse_metafile.o tracker.o bitfield.o sha1.o message.o peer.o data.o policy.o torrent.o bterror.o log.o signal_hander.o
$(CC) -o $@ $(LDFLAGS) $^ -ldl
clean:
rm -rf *.o ttorrent
编译和运行
$ make
$ ./ttorrent test.torrent
(1)上述内容均来自以前书籍(《Linux C编程实战》),仅是为了保存记忆以文字形式记录于此。
(2)谢 谢!