以文本方式查看主题

-  曙海教育集团论坛  (http://sun4.cn/bbs/index.asp)
--  Linux应用开发  (http://sun4.cn/bbs/list.asp?boardid=32)
----  linux下的应用开发起步--简化蚂蚁制作(三)  (http://sun4.cn/bbs/dispbbs.asp?boardid=32&id=1660)

--  作者:wangxinxin
--  发布时间:2010-11-23 10:29:54
--  linux下的应用开发起步--简化蚂蚁制作(三)

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);

  }

  }

  }

  /