prozilla/libprozilla/src/main.c

/******************************************************************************
   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$ */


#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);
    }
  else
    {
      /* Make sure home dir is never NULL */
      libprozrtinfo.home_dir = kstrdup(".");
    }

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