opkg_download.c

   1 /* vi: set noexpandtab sw=4 sts=4: */
   2 /* opkg_download.c - the itsy package management system
   3
   4    Carl D. Worth
   5
   6    Copyright (C) 2001 University of Southern California
   7    Copyright (C) 2008 OpenMoko Inc
   8
   9    This program is free software; you can redistribute it and/or
  10    modify it under the terms of the GNU General Public License as
  11    published by the Free Software Foundation; either version 2, or (at
  12    your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful, but
  15    WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17    General Public License for more details.
  18 */
  19
  20 #include <curl/curl.h>
  21
  22 #include "opkg.h"
  23 #include "opkg_download.h"
  24 #include "opkg_message.h"
  25
  26 #include "sprintf_alloc.h"
  27 #include "xsystem.h"
  28 #include "file_util.h"
  29 #include "str_util.h"
  30
  31 #ifdef OPKG_LIB
  32 #include "libopkg.h"
  33 opkg_download_progress_callback opkg_cb_download_progress = NULL;
  34 #endif
  35
  36 int
  37 curl_progress_func (char* url,
  38                     double t, /* dltotal */
  39                     double d, /* dlnow */
  40                     double ultotal,
  41                     double ulnow)
  42 {
  43     int i;
  44     int p = d*100/t;
  45
  46 #ifdef OPKG_LIB
  47     if (opkg_cb_download_progress)
  48     {
  49         opkg_cb_download_progress (p, url);
  50         return 0;
  51     }
  52 #endif
  53
  54     printf ("\r%3d%% |", p);
  55     for (i = 1; i < 73; i++)
  56     {
  57         if (i <= p * 0.73)
  58             printf ("=");
  59         else
  60             printf ("-");
  61     }
  62     printf ("|");
  63     fflush(stdout);
  64     return 0;
  65 }
  66
  67 int opkg_download(opkg_conf_t *conf, const char *src, const char *dest_file_name)
  68 {
  69     int err = 0;
  70
  71     char *src_basec = strdup(src);
  72     char *src_base = basename(src_basec);
  73     char *tmp_file_location;
  74     char *cmd;
  75
  76     opkg_message(conf,OPKG_NOTICE,"Downloading %s\n", src);
  77
  78     fflush(stdout);
  79
  80     if (str_starts_with(src, "file:")) {
  81         int ret;
  82         const char *file_src = src + 5;
  83         opkg_message(conf,OPKG_INFO,"Copying %s to %s...", file_src, dest_file_name);
  84         ret = file_copy(src + 5, dest_file_name);
  85         opkg_message(conf,OPKG_INFO,"Done\n");
  86         return ret;
  87     }
  88
  89     sprintf_alloc(&tmp_file_location, "%s/%s", conf->tmp_dir, src_base);
  90     err = unlink(tmp_file_location);
  91     if (err && errno != ENOENT) {
  92         opkg_message(conf,OPKG_ERROR, "%s: ERROR: failed to unlink %s: %s\n",
  93                 __FUNCTION__, tmp_file_location, strerror(errno));
  94         free(tmp_file_location);
  95         return errno;
  96     }
  97
  98     if (conf->http_proxy) {
  99         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: http_proxy = %s\n", conf->http_proxy);
 100         setenv("http_proxy", conf->http_proxy, 1);
 101     }
 102     if (conf->ftp_proxy) {
 103         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: ftp_proxy = %s\n", conf->ftp_proxy);
 104         setenv("ftp_proxy", conf->ftp_proxy, 1);
 105     }
 106     if (conf->no_proxy) {
 107         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: no_proxy = %s\n", conf->no_proxy);
 108         setenv("no_proxy", conf->no_proxy, 1);
 109     }
 110
 111     /* XXX: BUG rewrite to use execvp or else busybox's internal wget -Jamey 7/23/2002 */
 112 #if 0
 113     sprintf_alloc(&cmd, "wget --passive-ftp %s %s%s %s%s %s -P %s %s",
 114                   (conf->http_proxy || conf->ftp_proxy) ? "--proxy=on" : "",
 115                   conf->proxy_user ? "--proxy-user=" : "",
 116                   conf->proxy_user ? conf->proxy_user : "",
 117                   conf->proxy_passwd ? "--proxy-passwd=" : "",
 118                   conf->proxy_passwd ? conf->proxy_passwd : "",
 119                   conf->verbose_wget ? "" : "-q",
 120                   conf->tmp_dir,
 121                   src);
 122     err = xsystem(cmd);
 123     if (err) {
 124         if (err != -1) {
 125             opkg_message(conf,OPKG_ERROR, "%s: ERROR: Command failed with return value %d: `%s'\n",
 126                     __FUNCTION__, err, cmd);
 127         }
 128         unlink(tmp_file_location);
 129         free(tmp_file_location);
 130         free(src_basec);
 131         free(cmd);
 132         return EINVAL;
 133     }
 134     free(cmd);
 135 #endif
 136     CURL *curl;
 137     CURLcode res;
 138     FILE * file = fopen (tmp_file_location, "w");
 139
 140     curl = curl_easy_init ();
 141     if (curl)
 142     {
 143         curl_easy_setopt (curl, CURLOPT_URL, src);
 144         curl_easy_setopt (curl, CURLOPT_WRITEDATA, file);
 145         curl_easy_setopt (curl, CURLOPT_NOPROGRESS, 0);
 146         curl_easy_setopt (curl, CURLOPT_PROGRESSDATA, src);
 147         curl_easy_setopt (curl, CURLOPT_PROGRESSFUNCTION, curl_progress_func);
 148         if (conf->http_proxy || conf->ftp_proxy)
 149         {
 150             char *userpwd;
 151             sprintf_alloc (&userpwd, "%s:%s", conf->proxy_user,
 152                     conf->proxy_passwd);
 153             curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, userpwd);
 154             free (userpwd);
 155         }
 156         res = curl_easy_perform (curl);
 157         curl_easy_cleanup (curl);
 158         fclose (file);
 159
 160     }
 161     else
 162         return -1;
 163
 164     printf ("\n");
 165
 166     err = file_move(tmp_file_location, dest_file_name);
 167
 168     free(tmp_file_location);
 169     free(src_basec);
 170
 171     if (err) {
 172         return err;
 173     }
 174
 175     return 0;
 176 }
 177
 178 int opkg_download_pkg(opkg_conf_t *conf, pkg_t *pkg, const char *dir)
 179 {
 180     int err;
 181     char *url;
 182
 183     if (pkg->src == NULL) {
 184         opkg_message(conf,OPKG_ERROR, "ERROR: Package %s (parent %s) is not available from any configured src.\n",
 185                 pkg->name, pkg->parent->name);
 186         return -1;
 187     }
 188
 189     sprintf_alloc(&url, "%s/%s", pkg->src->value, pkg->filename);
 190
 191     /* XXX: BUG: The pkg->filename might be something like
 192        "../../foo.ipk". While this is correct, and exactly what we
 193        want to use to construct url above, here we actually need to
 194        use just the filename part, without any directory. */
 195     sprintf_alloc(&pkg->local_filename, "%s/%s", dir, pkg->filename);
 196
 197     err = opkg_download(conf, url, pkg->local_filename);
 198     free(url);
 199
 200     return err;
 201 }
 202
 203 /*
 204  * Downloads file from url, installs in package database, return package name.
 205  */
 206 int opkg_prepare_url_for_install(opkg_conf_t *conf, const char *url, char **namep)
 207 {
 208      int err = 0;
 209      pkg_t *pkg;
 210      pkg = pkg_new();
 211      if (pkg == NULL)
 212           return ENOMEM;
 213
 214      if (str_starts_with(url, "http://")
 215          || str_starts_with(url, "ftp://")) {
 216           char *tmp_file;
 217           char *file_basec = strdup(url);
 218           char *file_base = basename(file_basec);
 219
 220           sprintf_alloc(&tmp_file, "%s/%s", conf->tmp_dir, file_base);
 221           err = opkg_download(conf, url, tmp_file);
 222           if (err)
 223                return err;
 224
 225           err = pkg_init_from_file(pkg, tmp_file);
 226           if (err)
 227                return err;
 228           pkg->local_filename = strdup(tmp_file);
 229
 230           free(tmp_file);
 231           free(file_basec);
 232
 233      } else if (strcmp(&url[strlen(url) - 4], OPKG_PKG_EXTENSION) == 0
 234                 || strcmp(&url[strlen(url) - 4], DPKG_PKG_EXTENSION) == 0) {
 235
 236           err = pkg_init_from_file(pkg, url);
 237           if (err)
 238                return err;
 239           pkg->local_filename = strdup(url);
 240           opkg_message(conf, OPKG_DEBUG2, "Package %s provided by hand \(%s\).\n", pkg->name,pkg->local_filename);
 241           pkg->provided_by_hand = 1;
 242
 243      } else {
 244        pkg_deinit(pkg);
 245        free(pkg);
 246        return 0;
 247      }
 248
 249      if (!pkg->architecture) {
 250           opkg_message(conf, OPKG_ERROR, "Package %s has no Architecture defined.\n", pkg->name);
 251           return -EINVAL;
 252      }
 253
 254      pkg->dest = conf->default_dest;
 255      pkg->state_want = SW_INSTALL;
 256      pkg->state_flag |= SF_PREFER;
 257      pkg = hash_insert_pkg(&conf->pkg_hash, pkg, 1,conf);
 258      if ( pkg == NULL ){
 259         fprintf(stderr, "%s : This should never happen. Report this Bug in bugzilla please \n ",__FUNCTION__);
 260         return 0;
 261      }
 262      if (namep) {
 263           *namep = strdup(pkg->name);
 264      }
 265      return 0;
 266 }