C.前面看了一大段是不是有点云里雾里的啊?是有点罗嗦,但是俗话说:“万事总是开头难”OK,现在总算可以写主文件了。
下面就是ant.c
#include "ant.h"
#include "signal.h"
//0表示不用,1表示用代理
int use_proxy;
//代理的地址
char proxy_server[100];
//端口
u_short proxy_port;
//蚂蚁数目
int ant_num;
//下载量
int bulk;
//下载进程
int progress;
char cfgfile[256];
char logfile[256];
struct Ant* ants[10];
struct Mission ms;
//响应异常退出,将url、ant数据结构中的变量值,存放到log。
void justpause()
{
int i;
FILE* log;
log=fopen(logfile,"w");
fprintf(log,"%s ",ms.url);
for(i=0;iamount!=0)
fprintf(log, "%d:%d ",ants[i]->position,ants[i]->amount);
fclose(log);
exit(1);
}
//总算挨到主了,呵呵。。。。。
int main(int argc,char** argv)
{
int n;
int maxfd;
char savefile[256];
FILE* goods;
FILE* log;
fd_set rset,aset;
int finished;
int length;
char status[80];
char c;
struct sigaction sa;
//初始化环境参数
init_env( );
//对命令行传进来的参数做判断
while((c=getopt(argc,argv,"n:u:s:p:l:h"))!=EOF){
switch (c){
//取得蚂蚁数目
case 'n':
ant_num=atoi(optarg);
if(ant_num > MAXANTNUM)
ant_num = MAXANTNUM;
break;
//取得目标的url地址。
case 'u':
strcpy(ms.url, optarg);
break;
//取得代理的地址
case 's':
strcpy(proxy_server, optarg);
use_proxy=1;
break;
//取得代理的端口
case 'p':
use_proxy=1;
proxy_port=(u_short)atoi(optarg);
break;
//取得日志文件名
case 'l':
strcpy(logfile,optarg);
break;
case 'h':
printf("usage: ant -n antnum -u url -s proxyserver -p
prxoy_port -l logfile ");
return 1;
}
}
//从日志文件中读取url
if(logfile[0] != 0){
log = fopen(logfile, "r+");
strcpy(ms.url,get_url_from_log(log));
}
//将值赋代理的相关变量
if(use_proxy){
strcpy(ms.host,proxy_server);
ms.port=proxy_port;
}else{
strcpy(ms.host, extract_from_url(ms.url, SITENAME));
ms.port = atoi(extract_from_url(ms.url, PORTNAME));
}
//如果有下载日志,从日志中取相关信息
if(logfile[0] != 0){
printf("The file you havent download,continue.......");
n=0;
strcpy(ms.url,get_url_from_log(log));
while (1) {
ants[n] = (struct Ant*)malloc(sizeof(struct Ant));
memset(ants[n], 0, sizeof(struct Ant));
if(!(assign_mission(log, ants[n])))
break;
n++;
if(n > MAXANTNUM){
printf("You logfile must be wrong! ");
exit(1);
}
}
fclose(log);
free(ants[n]);
ant_num=n;
}
else{
//取得下载文件的大小,为每只蚂蚁分配下载位置。
bulk=get_size_of_url(&ms);
if(bulk==0){
printf("Cannot get the size of this object! ");
ant_num=1;
ants[0] = (struct Ant*)malloc(sizeof(struct Ant));
ants[0]->position=0;
ants[0]->amount=1073741823;
}
else if(bulk==-1){
printf("Cannot connect to the site! ");
exit(1);
}
else{
for(n=0; nposition = (bulk/ant_num)*n;
if(n!=(ant_num-1))
ants[n]->amount = bulk/ant_num;
else
ants[n]->amount = bulk-(bulk/ant_num)*(ant_num-1);
}
}
}
/