main.c

prozgui是一款Linxu下著名的下载工具

/******************************************************************************
 libprozilla - a download accelerator library
 Copyright (C) 2001 Kalum Somaratna

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
******************************************************************************/

/* The main file. */

/* $Id: main.c,v 1.40 2001/09/27 22:52:05 kalum Exp $ */


#include "common.h"
#include "prozilla.h"
#include "misc.h"
#include "getopt.h"
#include "debug.h"



static struct option long_opts[] = {
  /* { name    has_arg     *flag  val } */
  {"no-netrc", no_argument, NULL, 'n'},
  {"use-port", no_argument, NULL, 129},
  {"libdebug", no_argument, NULL, 130},
  {NULL, 0, NULL, 0}
};

libprozinfo libprozrtinfo;


/******************************************************************************
 Initialize the library.
******************************************************************************/
int proz_init(int argc, char **argv)
{
  int c;

  /* Gettext stuff */
  setlocale(LC_ALL, "");
  bindtextdomain(PACKAGE, LOCALEDIR);
  textdomain(PACKAGE);

  memset(&libprozrtinfo, 0, sizeof(libprozrtinfo));
  libprozrtinfo.argc = argc;
  libprozrtinfo.argv = argv;
  libprozrtinfo.debug_mode = TRUE;
  libprozrtinfo.ftp_use_pasv = TRUE;
  libprozrtinfo.ftp_default_user = kstrdup(DEFAULT_FTP_USER);
  libprozrtinfo.ftp_default_passwd = kstrdup(DEFAULT_FTP_PASSWD);
  libprozrtinfo.use_netrc = TRUE;
  libprozrtinfo.ftp_proxy = 0;
  libprozrtinfo.http_proxy = 0;
  /* 3 minutes should be enough for the default timeout */
  libprozrtinfo.conn_timeout.tv_sec = 120;
  libprozrtinfo.conn_timeout.tv_usec = 0;
  /* 15 secs for the default retry_delay */
  libprozrtinfo.conn_retry_delay.tv_sec = 15;
  libprozrtinfo.conn_retry_delay.tv_usec = 0;
  /*Default: try infinitely */
  libprozrtinfo.max_attempts = 0;
  /*No no-cache directive is given */
  libprozrtinfo.http_no_cache = FALSE;
  /*default is unlimited (0) bandwith */
  libprozrtinfo.max_bps_per_dl = 0;


  /* Parse the options and set the relevant ones for the library. */
  while ((c = getopt_long(argc, argv, "", long_opts, NULL)) != EOF)
  {
    switch (c)
    {
    case 129:
      libprozrtinfo.debug_mode = TRUE;
      break;
    default:
      continue;
    }
  }

  /* TODO Get home directory and read .netrc. */

  libprozrtinfo.home_dir = home_dir();

  if (libprozrtinfo.home_dir != NULL)
  {
    char *netrc_file = kmalloc(strlen(libprozrtinfo.home_dir)
			       + strlen(".netrc") + 2);

    sprintf(netrc_file, "%s/%s", libprozrtinfo.home_dir, ".netrc");

    libprozrtinfo.netrc_list = parse_netrc(netrc_file);
  }

  libprozrtinfo.dl_dir = kstrdup(".");
  libprozrtinfo.output_dir = kstrdup(".");
  libprozrtinfo.log_dir = kstrdup(".");
  debug_init();
  return 1;
}

/******************************************************************************
 ...
******************************************************************************/
void proz_shutdown(void)
{
  kfree(libprozrtinfo.http_proxy);
  kfree(libprozrtinfo.ftp_proxy);
  kfree(libprozrtinfo.dl_dir);
  kfree(libprozrtinfo.output_dir);
  kfree(libprozrtinfo.log_dir);
}

/******************************************************************************
 Abort with an error message.
******************************************************************************/
void proz_die(const char *format, ...)
{
  va_list args;
  char message[MAX_MSG_SIZE + 1];

  va_start(args, format);
  vsnprintf(message, MAX_MSG_SIZE, format, args);
  va_end(args);

  printf("%s\n", message);
  proz_shutdown();

  exit(EXIT_FAILURE);
}

/******************************************************************************
 ...
******************************************************************************/
void proz_set_http_proxy(proxy_info * proxy)
{
  if (libprozrtinfo.http_proxy)
    kfree(libprozrtinfo.http_proxy);
  libprozrtinfo.http_proxy = kmalloc(sizeof(proxy_info));
  memcpy(libprozrtinfo.http_proxy, proxy, sizeof(proxy_info));
}

/******************************************************************************
 ...
******************************************************************************/
void proz_set_ftp_proxy(proxy_info * proxy)
{
  if (libprozrtinfo.ftp_proxy)
    kfree(libprozrtinfo.ftp_proxy);
  libprozrtinfo.ftp_proxy = kmalloc(sizeof(proxy_info));
  memcpy(libprozrtinfo.ftp_proxy, proxy, sizeof(proxy_info));
}

void proz_use_http_proxy(boolean use)
{
  if (libprozrtinfo.http_proxy)
    libprozrtinfo.http_proxy->use_proxy = use;
}

void proz_use_ftp_proxy(boolean use)
{
  if (libprozrtinfo.ftp_proxy)
    libprozrtinfo.ftp_proxy->use_proxy = use;
}

void proz_set_connection_timeout(struct timeval *timeout)
{
  libprozrtinfo.conn_timeout.tv_sec = timeout->tv_sec;
  libprozrtinfo.conn_timeout.tv_usec = timeout->tv_usec;
}


void proz_set_connection_retry_delay(struct timeval *delay)
{
  libprozrtinfo.conn_retry_delay.tv_sec = delay->tv_sec;
  libprozrtinfo.conn_retry_delay.tv_usec = delay->tv_usec;
}


void proz_set_download_dir(char *dir)
{
  assert(dir != NULL);

  if (libprozrtinfo.dl_dir)
    kfree(libprozrtinfo.dl_dir);

  libprozrtinfo.dl_dir = kstrdup(dir);

}

void proz_set_logfile_dir(char *dir)
{
  assert(dir != NULL);
  if (libprozrtinfo.log_dir)
    kfree(libprozrtinfo.log_dir);

  libprozrtinfo.log_dir = kstrdup(dir);
}

void proz_set_output_dir(char *dir)
{
  assert(dir != NULL);

  if (libprozrtinfo.output_dir)
    kfree(libprozrtinfo.output_dir);

  libprozrtinfo.output_dir = kstrdup(dir);
}


char *proz_get_libprozilla_version()
{

  return strdup(VERSION);
}