main.c

linux下网络下载软件的源码

		if((childpid == -1) && (errno != EINTR))
			break;
}


void DownloadLoop (URLp URL)
/*************************************
**	void DownloadLoop (URLp)
**
** Pre:  URL has been Parse()'d
** Post: Connects to server and downloads the file
*************************************/
{	char COMMAND[128];
	int ARCHIVES;

	while(URL->retry == 1 /* || (RESUMES_LEFT != 0 && !URL->done && URL->retry) */)
	{	URL->done = 0;
		ARCHIVES = 0;
		if (RESUMES_LEFT > 0)
			RESUMES_LEFT --;

		if (Connect(URL))
		{	FD_ZERO(&testfds);
			FD_SET(URL->sockfd, &testfds);
			if ((URL->protocol == FTP && ReadFTPListing(URL)) ||
				(URL->protocol == HTTP && ReadHTTPHeader(URL)))
			{	start = time((time_t *)0);
				end = time((time_t *)0);
				FD_ZERO(&testfds);
				FD_SET(URL->sockfd, &testfds);
				while (!URL->done)
					Download(URL);
			}
		}

		if (OUTPUT_LEVEL > 0)
			printf("\n");
		close (URL->sockfd);

		if (URL->retry == 1 || (RESUMES_LEFT != 0 && !URL->done))
		{	if (OUTPUT_LEVEL > 1)
				printf("Retrying in %d seconds...\n\n", WAIT_RETRY);
			sleep(WAIT_RETRY);
		} else if (TESTFILES)
		{	if (strstr(strrchr(URL->filename, '.'), "gz") || strstr(strrchr(URL->filename, '.'), "tgz"))
			{	strcpy(COMMAND, "gzip -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "tar"))
			{	strcpy(COMMAND, "tar -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "zip"))
			{	strcpy(COMMAND, "unzip -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "arj"))
			{	strcpy(COMMAND, "unarj t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "ace"))
			{	strcpy(COMMAND, "unace -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "rar"))
			{	strcpy(COMMAND, "unrar -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "bz2"))
			{	strcpy(COMMAND, "bzip2 -t ");
				ARCHIVES = 1;
			} else if (strstr(strrchr(URL->filename, '.'), "rpm"))
			{	strcpy(COMMAND, "rpm --nopgp --checksig ");
				ARCHIVES = 1;
			}

			if (ARCHIVES)
				if(system(strcat(COMMAND, URL->filename)) == 0)
				{	printf("**** %s is a valid archive!\n\n\n", URL->filename);
					if (CALL_ME)
						MailNotify(URL,1);
				} else
				{	printf("!!!! %s HAS ERRORS!!!\n\n\n", URL->filename);
					if (CALL_ME)
						MailNotify(URL,0);
				} else if (CALL_ME)
					MailNotify(URL,-1);
		}
	}
}


void MailNotifyFinish()
/*************************************
**    void MailNotify(URLp URL, int valid)
**
** Pre:  downloading finished
** Post: a mail is sent
**	Written by Eric Gaudet.
*************************************/
{	FILE *fp;
	char tmpstrmail[1024];
	NODE *ptr = list->next;

	if (OUTPUT_LEVEL > 6)
		printf("Sending mail\n");

	sprintf(tmpstrmail,"mail -s \"[GREED] Your files have been downloaded\" %s > /dev/null",
			CALL_ADDR);
	fp=popen(tmpstrmail,"w");

	fprintf(fp,"You're download is complete!\nFiles retirieved:\n");
	while (ptr != NULL)
	{	if (ptr->URL.filename == NULL)
			fprintf(fp,"* From the directory: %s\n", ptr->URL.name);
		else
			fprintf(fp,"> %s (%ld bytes)\n", ptr->URL.filename, ptr->URL.ltotal);
		ptr = ptr->next;
	}
	fprintf(fp,"\n\n");

	fprintf(fp,"\n%s  [Get and Resume Elite EDition] - By "
		"Anoakie Turner\r\n", CURRENT_VERSION);
	pclose(fp);
}

void MailNotify(URLp URL, int valid)
/*************************************
**    void MailNotify(URLp URL, int valid)
**
** Pre:  downloading finished
** Post: a mail is sent
**	Written by Eric Gaudet.
*************************************/
{	FILE *fp;
	char tmpstrmail[1024];

	if (OUTPUT_LEVEL > 6)
		printf("Sending mail\n");

	if (URL)
	{	sprintf(tmpstrmail,"mail -s \"[GREED] %s downloaded\" %s > /dev/null",
			URL->filename,CALL_ADDR);
		fp=popen(tmpstrmail,"w");

		switch (valid)
		{	case 1:
				fprintf(fp,"The following archive has been sucessfully downloaded and tested :\n");
				break;
			case 0:
				fprintf(fp,"The following archive has been downloaded, but appears to be invalid :\n");
				break;
			default:
				fprintf(fp,"The following file has been downloaded :\n");
				break;
		}
		fprintf(fp,"%s (%ld bytes)\n", URL->filename, URL->ltotal);
	}
	else
	{
		sprintf(tmpstrmail,"mail -s \"[GREED] running\" %s > /dev/null",
			CALL_ADDR);
		fp=popen(tmpstrmail,"w");
	}

	fprintf(fp,"\n%s  [Get and Resume Elite EDition] - By "
		"Anoakie Turner\r\n", CURRENT_VERSION);
	pclose(fp);
}


void SendBug()
/*************************************
*************************************/
{	FILE *fp;
	char *buffer;
	char mail[64000];
	char email[600];
	char tmpstrmail[1024];
	int result;
	buffer = mail;


	printf(
"*** IMPORTANT ***\nBefore you try to send mail, make SURE you have sendmail set up correcly!!!\n"
"If not, the mail will"
" never get to me!\nIf you don't think you have sendmail set up or if you're not really sure (or if you run RedHat or"
" Mandrake),\nthen just send it with a regular e-mail program!\nThis is only for people who know what they are"
" doing!\nA copy of this e-mail will be mailed to yourself for verification that the mail was sent...\nIf you don't"
" get it, then you don't have sendmail installed\ncorrectly and you'll need to send me the bug through a regular"
" e-mail program!\n***\n\n");
	printf("E-mail address: ");
	result = scanf("%s", email);
	printf("Please describe the bug as well as you can, how it happened, if it's reproducable, "
		"what OS you are running, what compiler you're using, and any other in"
		"formation that will help me find and remove this bug from future versions of greed."
		"\n\nPress CTRL-D on a blank line when finished.\n****\n\n");
	while(result != EOF)
	{	result = scanf("%s", buffer);
		buffer += strlen(buffer);
		buffer[0] = ' ';
		buffer ++;
		buffer[0] = '\0';
	}

	sprintf(tmpstrmail,"mail -s \"[GREED] BUG REPORT! [GREED]\" %s > /dev/null",
			"[e-mail]");
	fp=popen(tmpstrmail,"w");

	fprintf(fp,"E-mail from: %s\r\n\r\n", email);
	fprintf(fp,"Problem:\n%s", mail);
	fprintf(fp,"\n\n");

	fprintf(fp,"\n%s  [Get and Resume Elite EDition] - By "
		"Anoakie Turner\r\n", CURRENT_VERSION);
	printf("\n\nThank you for reporting this bug!  Without the help of the many GREED users, "
		"GREED wouldn't be as great as it is now!\nExiting...\n\n");
	pclose(fp);
}


void Switches (int i, char *argv[])
/*************************************
**	void Switches (int, char*)
**
** Post: Parses strings in argv and set global variables accordingly.
*************************************/
{	if (argv[i][0] == '-')
	{	switch (argv[i][1])
		{	case '*': /* For developpers only : DO NOT print in usage. */
				printf("*** Debug Info Mode On\n");
				DEBUG=1;
				break;
			case '0': 
				RESUME = 0;
				break;
			case 'N': 
			case 'n': 
				argv[i] = GGR;
				break;
			case 'M':
			case 'm':
				CALL_ME = 1;
				CALL_ADDR = argv[i] + 2;
				printf("%s will be notified by mail\n", CALL_ADDR);
				/* MailNotify(0,1); */
				break;
			case 'O':
			case 'o':
				if (argv[i][2] >= '0' || argv[i][2] <= '9')
					OUTPUT_LEVEL = argv[i][2] - '0';
				break;
			case 'R':
			case 'r':
				if (argv[i][2] == '\0')
					REFERRER = 0;
				else
					strcpy(REF, argv[i] + 2);
				break;
			case 'S':
			case 's':
				STDOUT = 1;
				break;
			case 'T':
			case 't':
				TESTFILES = 1;
				break;
			case 'V':
			case 'v':
				printf("%s  [Get and Resume Elite EDition] - By "
				 "Anoakie Turner\r\n", CURRENT_VERSION);
				exit(EXIT_SUCCESS);
				break;
			case 'F':
			case 'f':
				CHECKIN = 0;
				break;
			case 'G':
			case 'g':
				IMPORT_GR = 1;
				GR_FILE = argv[i] + 2;
				break;
			case 'I':
			case 'i':
				STDIN = 1;
				break;
			case 'L':
			case 'l':
				RECURSE = atoi(argv[i] + 2);
				break;
			case 'D':
			case 'd':
				if (argv[i][2] != '\0')
					FORK = atoi(argv[i] + 2);
				else
					FORK = 1;
				break;
			case 'C':
			case 'c':
				argv[i] += 2;
				printf("\n\n<-- %s\n", argv[i]);
				unescape_url(argv[i]);
				printf("--> %s!\n\n", argv[i]);
				break;
			case 'B':
			case 'b':
				BACKGROUND = 1;
				break;
			case 'Z':
			case 'z':
				SendBug();
				exit(0);
				break;
			case 'W':
			case 'w':
				NEWRC = 1;
				break;
			default:
				printf("\n\n\t\t*** Unknown switch [%s]! ***\n\n", argv[i]);
			case '?':
				printf("%s  [Get and Resume Elite EDition] - By "
				"Anoakie Turner\r\n", CURRENT_VERSION);
				printf("\n\nUSEAGE: greed URL(1) URL(2) ... URL(N)\n"
				"i.e.: greed %s ftp://some.where/out/there.tar.gz\n\n", GGR);
printf("Options:\n"
"-b\tSends greed into background downloading mode.\n"
"-c<url>\tchanges the %%##'s in a URL to characters.\n"
"-d\tTells greed to download 2 files at once.\n"
"-d<#>\tTells greed to download # files at once.\n"
"-f\tIgnores the greed.in (default download list) file if it exists.\n"
"-g<FL>\tImports a windows GetRight file (*.GRX).\n"
"\t\ti.e. greed -gMyDownloads.grx\n"
"-i\tReads URLs from standard input.\n"
"\t\ti.e. greed -I < download.lst\n"
"-l<#>\tSets the level of recursion for FTP transfers.\n"
"-m<adr>\tsend a mail to [adr] (by sendmail) when a download is complete\n"
"-n\tDownloads the newest version of GREED!\n"
"-o<#>\tSets the level of output (default 5, no output = 0).\n"
"-r\tSets the referrer OFF.\n"
"-r<URL>\tSets the referrer to URL.\n"
"-s\tPrints the file that is being downloaded to the screen instead of a file.\n"
"-t\tTests archives after download to see if they are valid.\n"
"-w\tCreates new .greedrc in ${HOME}/.greedrc.\n"
"-z\tSend me a bug!\n"
"-0\tRestarts file at 0 bytes\n"
"...\tUsage: greed URL...FILE, where URL is the url and FILE is the file retrieved.\n"
"\t\tNOTE: This only works if you have the correct programs to test with!.\n"
"\n\nALL OTHER OPTIONS (including proxy support) CAN BE FOUND IN ~/.greedrc\n");
				exit(0);
				break;
		}
	}
}