a73406d2d5c61a9dcabe5db8830c60b6976d8fb7
[oweals/opkg-lede.git] / 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
32 int
33 curl_progress_func (void* data,
34                          double t, /* dltotal */
35                          double d, /* dlnow */
36                          double ultotal,
37                          double ulnow)
38 {
39     int i;
40     int p = d*100/t;
41     printf ("\r%3d%% |", p);
42     for (i = 1; i < 73; i++)
43     {
44         if (i <= p * 0.73)
45             printf ("=");
46         else
47             printf ("-");
48     }
49     printf ("|");
50     fflush(stdout);
51     return 0;
52 }
53
54 int opkg_download(opkg_conf_t *conf, const char *src, const char *dest_file_name)
55 {
56     int err = 0;
57
58     char *src_basec = strdup(src);
59     char *src_base = basename(src_basec);
60     char *tmp_file_location;
61     char *cmd;
62
63     opkg_message(conf,OPKG_NOTICE,"Downloading %s\n", src);
64         
65     fflush(stdout);
66     
67     if (str_starts_with(src, "file:")) {
68         int ret;
69         const char *file_src = src + 5;
70         opkg_message(conf,OPKG_INFO,"Copying %s to %s...", file_src, dest_file_name);
71         ret = file_copy(src + 5, dest_file_name);
72         opkg_message(conf,OPKG_INFO,"Done\n");
73         return ret;
74     }
75
76     sprintf_alloc(&tmp_file_location, "%s/%s", conf->tmp_dir, src_base);
77     err = unlink(tmp_file_location);
78     if (err && errno != ENOENT) {
79         opkg_message(conf,OPKG_ERROR, "%s: ERROR: failed to unlink %s: %s\n",
80                 __FUNCTION__, tmp_file_location, strerror(errno));
81         free(tmp_file_location);
82         return errno;
83     }
84
85     if (conf->http_proxy) {
86         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: http_proxy = %s\n", conf->http_proxy);
87         setenv("http_proxy", conf->http_proxy, 1);
88     }
89     if (conf->ftp_proxy) {
90         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: ftp_proxy = %s\n", conf->ftp_proxy);
91         setenv("ftp_proxy", conf->ftp_proxy, 1);
92     }
93     if (conf->no_proxy) {
94         opkg_message(conf,OPKG_DEBUG,"Setting environment variable: no_proxy = %s\n", conf->no_proxy);
95         setenv("no_proxy", conf->no_proxy, 1);
96     }
97
98     /* XXX: BUG rewrite to use execvp or else busybox's internal wget -Jamey 7/23/2002 */ 
99 #if 0
100     sprintf_alloc(&cmd, "wget --passive-ftp %s %s%s %s%s %s -P %s %s",
101                   (conf->http_proxy || conf->ftp_proxy) ? "--proxy=on" : "",
102                   conf->proxy_user ? "--proxy-user=" : "",
103                   conf->proxy_user ? conf->proxy_user : "",
104                   conf->proxy_passwd ? "--proxy-passwd=" : "",
105                   conf->proxy_passwd ? conf->proxy_passwd : "",
106                   conf->verbose_wget ? "" : "-q",
107                   conf->tmp_dir,
108                   src);
109     err = xsystem(cmd);
110     if (err) {
111         if (err != -1) {
112             opkg_message(conf,OPKG_ERROR, "%s: ERROR: Command failed with return value %d: `%s'\n",
113                     __FUNCTION__, err, cmd);
114         } 
115         unlink(tmp_file_location);
116         free(tmp_file_location);
117         free(src_basec);
118         free(cmd);
119         return EINVAL;
120     }
121     free(cmd);
122 #endif
123     CURL *curl;
124     CURLcode res;
125     FILE * file = fopen (tmp_file_location, "w");
126
127     curl = curl_easy_init ();
128     if (curl)
129     {
130         curl_easy_setopt (curl, CURLOPT_URL, src);
131         curl_easy_setopt (curl, CURLOPT_WRITEDATA, file);
132         curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
133         curl_easy_setopt (curl, CURLOPT_PROGRESSFUNCTION, curl_progress_func);
134         res = curl_easy_perform (curl);
135         curl_easy_cleanup (curl);
136         fclose (file);
137
138     }
139     else
140         return -1;
141
142     printf ("\n");
143
144     err = file_move(tmp_file_location, dest_file_name);
145
146     free(tmp_file_location);
147     free(src_basec);
148
149     if (err) {
150         return err;
151     }
152
153     return 0;
154 }
155
156 int opkg_download_pkg(opkg_conf_t *conf, pkg_t *pkg, const char *dir)
157 {
158     int err;
159     char *url;
160
161     if (pkg->src == NULL) {
162         opkg_message(conf,OPKG_ERROR, "ERROR: Package %s (parent %s) is not available from any configured src.\n",
163                 pkg->name, pkg->parent->name);
164         return -1;
165     }
166
167     sprintf_alloc(&url, "%s/%s", pkg->src->value, pkg->filename);
168
169     /* XXX: BUG: The pkg->filename might be something like
170        "../../foo.ipk". While this is correct, and exactly what we
171        want to use to construct url above, here we actually need to
172        use just the filename part, without any directory. */
173     sprintf_alloc(&pkg->local_filename, "%s/%s", dir, pkg->filename);
174
175     err = opkg_download(conf, url, pkg->local_filename);
176     free(url);
177
178     return err;
179 }
180
181 /*
182  * Downloads file from url, installs in package database, return package name. 
183  */
184 int opkg_prepare_url_for_install(opkg_conf_t *conf, const char *url, char **namep)
185 {
186      int err = 0;
187      pkg_t *pkg;
188      pkg = pkg_new();
189      if (pkg == NULL)
190           return ENOMEM;
191
192      if (str_starts_with(url, "http://")
193          || str_starts_with(url, "ftp://")) {
194           char *tmp_file;
195           char *file_basec = strdup(url);
196           char *file_base = basename(file_basec);
197
198           sprintf_alloc(&tmp_file, "%s/%s", conf->tmp_dir, file_base);
199           err = opkg_download(conf, url, tmp_file);
200           if (err)
201                return err;
202
203           err = pkg_init_from_file(pkg, tmp_file);
204           if (err)
205                return err;
206           pkg->local_filename = strdup(tmp_file);
207
208           free(tmp_file);
209           free(file_basec);
210
211      } else if (strcmp(&url[strlen(url) - 4], OPKG_PKG_EXTENSION) == 0
212                 || strcmp(&url[strlen(url) - 4], DPKG_PKG_EXTENSION) == 0) {
213
214           err = pkg_init_from_file(pkg, url);
215           if (err)
216                return err;
217           pkg->local_filename = strdup(url);
218           opkg_message(conf, OPKG_DEBUG2, "Package %s provided by hand \(%s\).\n", pkg->name,pkg->local_filename);
219           pkg->provided_by_hand = 1;
220
221      } else {
222        pkg_deinit(pkg);
223        free(pkg);
224        return 0;
225      }
226
227      if (!pkg->architecture) {
228           opkg_message(conf, OPKG_ERROR, "Package %s has no Architecture defined.\n", pkg->name);
229           return -EINVAL;
230      }
231
232      pkg->dest = conf->default_dest;
233      pkg->state_want = SW_INSTALL;
234      pkg->state_flag |= SF_PREFER;
235      pkg = hash_insert_pkg(&conf->pkg_hash, pkg, 1,conf);  
236      if ( pkg == NULL ){
237         fprintf(stderr, "%s : This should never happen. Report this Bug in bugzilla please \n ",__FUNCTION__);
238         return 0;
239      }
240      if (namep) {
241           *namep = strdup(pkg->name);
242      }
243      return 0;
244 }