Linux BT下载(17)-主函数的设计和实现

主函数的设计和实现

主函数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）谢 谢！