Linux BT下载(7)-种子解析模块设计与实现1

种子解析模块设计与实现

解析文件主要在parse_metafile.h和parse_metafile.c中实现。

//parse_metafile.h 解析文件
#ifndef PARSE_METAFILE
#define PARSE_METAFILE

//保存种子文件中获取的tracker的URL
typedef struct _Announce_list
{
	char announce[128];
	struct _Announce_list *next;
} Announce_list;

//保存各个待下载的文件的路径长度
typedef struct _Files
{
	char path[256];
	long length;
	struct _Files *next;
} Files;

int read_metafile(char *metafile_name);			//读取种子文件
int find_keyword(char *keyword, long *position);//在种子文件中查找关键字
int read_announce_list();   			//获取各个tracker服务器的地址
int add_an_announce(char *url);		//向tracker列表添加一个url

int get_piece_length();	//获取每个piece的长度，一般为256KB
int get_pieces();		//获得各个piece的hash值

int is_multi_files();	//判断下载的是单个还是多个文件
int get_file_name();    //获取文件名，对于多文件获取的目录名
int get_file_length();	//获取待下载的文件总长度
int get_files_length_path();//获取文件的路径和长度，对多文件种子有效

int get_info_hash();	//由info关键词对应的值计算info_hash
int get_peer_id();  	//生成peer_id,每个peer都有一20字节的peer_id

void release_memory_in_parse_metafile();	//释放parse_metafile.c中动态分配的内存
int parse_metafile(char *metafile);	//调用本文件中定义的函数，完成解析种子文件

#endif

以下是parse_metafile.c文件的头部

//parse_metafile.c  解析文件
#include <stdio.h>
#include <ctype.h>
#include <malloc.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include "parse_metafile.h"
#include "shal.h"

char *metafile_content = NULL;	//保存种子的内容
long filesize;					//种子的长度

int piece_length = 0;		//每个piece的长度一般为256KB
char *pieces = NULL;		//保存每个pieces的hash值，每个为20B
int pieces_length = 0;		//缓冲区pieces的长度

int multi_file = 0;		//指明是单文件还是多文件
char *file_name = NULL;	//对于单文件，存放文件；对于多文件，存放目录名
long long file_length = 0;	//存放待下载的文件的总长度
Files *files_head = NULL; 	//对多文件种子有效，存放各个文件的路径和长度
unsigned char info_hash[20];	//保存info_hash的值，连接tracker和peer时使用
unsigned char peer_id[20];	//保存peer_id的值，连接peer时使用

Announce_list *announce_list_head = NULL;	//用于保存所有trackr的服务器的URL

具体函数如下:

int read_metafile(char *metafile_name)，此函数的说明在代码中不再另写。

/*
* int read_metafile(char *metafile_name)
*解析种子文件
*metafile_name是种子文件名
*处理成功返回0，否则返回-1
*将种子文件的内容读入到全局变量metafile_content所指的缓存区中

*说明 编译器预定义的宏__FILE__和__LINE__在程序中可以直接使用。
*__FILE__代表该宏所在源文件的文件名，在源文件parse_metafile.c中
*该宏的值等于"parse_metafile.c",宏__LINE__的值为__LINE__所在的行
*号
*
*
*/
int read_metafile(char *metafile_name)
{
	long i;
	
	//以二进制、只读的方式打开文件
	FILE *fp = fopen(metafile_name, "rb");
	if(fp == NULL)
	{
		printf("%s:%d can not open file\n", __FILE__, __LINE__);
		return -1;
	}
	
	//获得种子文件的长度，fieszie为全局变量,在parse_metafile_name
	fseek(fp , 0, SEEK_END);
	filesize = ftell(fp);
	if(fileszie == -1)
	{
		printf("%s:%d fseek failed\n", __FILE__,__LINE__);
		return -1;
	}
	metafile_content = (char *)malloc(filesize+1);
	if(metafile_content == NULL)
	{
		printf("%s:%d malloc failed\n", __FILE__,__LINE__);
		return -1;
	}
	
	//读取种子文件的内容到metafile_content缓冲区中
	fseek(fp, 0, SEEK_SET);
	for(i = 0; i < filesize; i++)
	{
		metafile_content[i] = fgetc(fp);
	}
	
	metafile_content[i] = '\0';
	fclose(fp);

	
#ifdef DEBUG
	printf("metafile size is: &ld\n", filesize);
#endif
	return 0;
}

int find_keyword(char *keyword,long *position)

/*
int find_keyword(char *keyword, long *position)
功能:从种子文件中查找某个关键字
参数:keyword为要查找的关键字，position用于返回关键字第一个字符所在的下标.
返回:成功找到关键字返回1，未找到返回0，执行失败返回-1
*/
int find_keyword(char *keyword, long *position)
{
	long i;
	
	*position = -1;
	if(keyword == NULL)
		return 0;
	
	for(i = 0; i < filesize - strlen(keyword); i++)
	{
		if( memcmp(&metafile_content[i], keyword, strlen(keyword))==0)
		{
			*position = i;
			return 1;
		}
	}
	return 0;
}

函数说明:此函数用于查找某个关键字（B编码的关键字）。例如：“8:announce”和“13:announce-list”之后是Tracker服务器的地址(URL)，找到该关键字后，可以获取Tracker的地址。

int read_announce_list()

/*
功能:获得Tracker地址，并将获得的地址保存到全局变量announce_list_head指向的List中
*/
int read_announce_list()
{
	Announce_list *node = NULL;
	Announce_list *p = NULL;
	int len = 0;
	long i;
	
	if( find_keyword("13:announce-list", &i) == 0)
	{//无"13:announce-list"关键字
		if( find_keyword("8:announce", &i) == 1)
		{
			//跳过"8:announce"
			i = i + strlen("8:announce");	//i未加前是字符串的起始位置
			
			//检测字符串后边的字符是否是阿拉伯数字
			//获取URL的长度
			while(isdigit(metafile_content[i]))
			{
				len = len * 10 + (metafile_content[i] - '0');
				i++;
			}
			i++; //跳过':'
			
			//申请堆空间保存Tracker的URL
			node = (Announce_list *)malloc(sizeof(Announce_list));
			strcpy(node->announce, &metafile_content[i], len);
			node->announce[len] = '\0';
			node->next = NULL;
			announce_list_head = node;
		}
	}
	else
	{/*如果有13:announce-list关键字不用处理8:announce关键词
	 **使用备用的URL(备用URL中包含关键字"8:announce"包含的URL)
	 **关键字"13:announce-list"之后的第一个字符为列表的起始字符'l'
	 **该列表中含有两个元素，这两个元素的类型也都是列表
	 **
	 */
		i = i + strlen("13:announce-list");
		i++;	//跳过'1'
		while(metafile_content[i] != 'e')
		{
			i++;   //跳过'l'
			/*
			*检查输入的参数是否为阿拉伯数字
			*获取URL的长度
			*/
			
			while(isdigit(metafile_content[i]))
			{
				len = len * 10 + (metafile_content[i] - '0');
				i++;
			}
			if(metafile_content[i] == ':')
				i++;	//跳过':'
			else
				return -1;
			//只处理以http开头的tracker地址，不处理以udp开头的地址
			if(memcmp(&metafile_content[i], "http", 4) == 0)
			{
				node = (Announce_list *)malloc(sizeof(Announce_list));
				strcpy(node->announce, &metafile_content[i],len);
				node->announce[len] = '\0';
				node->next = NULL;
				
				if(announce_list_head == NULL)
				{
					announce_list_head = node;
				}
				else
				{
					p = announce_list_head;
					while(p->next != NULL)
						p = p->next;	//使p指针指向最后个节点
					p->next = node; 	//node成为tracker列表的最后一个节点
				}
			}
			
			i = i + len;
			len = 0;
			i++;	//跳过'e'
			if(i >= filesize)
				return -1;
		}	//while循环结束
	}
#ifdef DEBUG
	p = announce_list_head;
	while(p != NULL)
	{
		printf("%s\n", p->announce);
		p = p->next;
	}
#endif
	return 0;
}

某种子文件的开头部分如下，可以帮助理解我的注释：

d8:announce32:http://tk.greedland.net/announce13:announce-listl132:http://tk.greedland.net/announcee113:http://th2.greedland.net/announceee...

第一个字符'd'是B编码中字典的起始符，接着是关键字“8:announce”,该关键字是长度为8的字符串，其对应的值为长度为32的字符串"32:http://tk.greedland,net/announce"是一个Tracker服务器地址(URL)，接着是关键字"13：annunce-list",该关键字对应的是一个列表，应为关键字"13:announce-list"之后是第一个字符的起始字符"l"，该列表中还有这两个元素的类型也都是列表。

如果有关键字"13:announce-list"哪吗就不用处理关键字"8:announce"的原因在于，前者对应的值 必定包含对应的后者的值。

...