2 * Copyright (c) 2011 Nathan Whitehorn
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 #include <sys/param.h>
35 static int fetch_files(int nfiles, char **urls);
40 char *diststring = strdup(getenv("DISTRIBUTIONS"));
42 int i, nfetched, ndists = 0;
43 for (i = 0; diststring[i] != 0; i++)
44 if (isspace(diststring[i]) && !isspace(diststring[i+1]))
46 ndists++; /* Last one */
48 urls = calloc(ndists, sizeof(const char *));
50 fprintf(stderr, "Out of memory!\n");
54 init_dialog(stdin, stdout);
55 dialog_vars.backtitle = __DECONST(char *, "FreeBSD Installer");
58 for (i = 0; i < ndists; i++) {
59 urls[i] = malloc(PATH_MAX);
60 sprintf(urls[i], "%s/%s", getenv("BSDINSTALL_DISTSITE"),
61 strsep(&diststring, " \t"));
64 if (chdir(getenv("BSDINSTALL_DISTDIR")) != 0) {
66 sprintf(error, "Could could change to directory %s: %s\n",
67 getenv("BSDINSTALL_DISTDIR"), strerror(errno));
68 dialog_msgbox("Error", error, 0, 0, TRUE);
73 nfetched = fetch_files(ndists, urls);
78 for (i = 0; i < ndists; i++)
82 return ((nfetched == ndists) ? 0 : 1);
86 fetch_files(int nfiles, char **urls)
89 FILE *fetch_out, *file_out;
90 struct url_stat ustat;
91 off_t total_bytes, current_bytes, fsize;
96 int i, progress, last_progress;
97 int nsuccess = 0; /* Number of files successfully downloaded */
101 /* Make the transfer list for dialog */
102 items = calloc(sizeof(char *), nfiles * 2);
104 fprintf(stderr, "Out of memory!\n");
108 for (i = 0; i < nfiles; i++) {
109 items[i*2] = strrchr(urls[i], '/');
110 if (items[i*2] != NULL)
113 items[i*2] = urls[i];
114 items[i*2 + 1] = "Pending";
117 dialog_msgbox("", "Connecting to server.\nPlease wait...", 0, 0, FALSE);
119 /* Try to stat all the files */
121 for (i = 0; i < nfiles; i++) {
122 if (fetchStatURL(urls[i], &ustat, "") == 0 && ustat.size > 0)
123 total_bytes += ustat.size;
127 for (i = 0; i < nfiles; i++) {
128 last_progress = progress;
129 if (total_bytes == 0)
130 progress = (i*100)/nfiles;
132 fetchLastErrCode = 0;
133 fetch_out = fetchXGetURL(urls[i], &ustat, "");
134 if (fetch_out == NULL) {
135 snprintf(errormsg, sizeof(errormsg),
136 "Error while fetching %s: %s\n", urls[i],
138 items[i*2 + 1] = "Failed";
139 dialog_msgbox("Fetch Error", errormsg, 0, 0,
144 items[i*2 + 1] = "In Progress";
146 file_out = fopen(items[i*2], "w+");
147 if (file_out == NULL) {
148 snprintf(errormsg, sizeof(errormsg),
149 "Error while fetching %s: %s\n",
150 urls[i], strerror(errno));
151 items[i*2 + 1] = "Failed";
152 dialog_msgbox("Fetch Error", errormsg, 0, 0,
158 while ((chunk = fread(block, 1, sizeof(block), fetch_out))
160 if (fwrite(block, 1, chunk, file_out) < chunk)
163 current_bytes += chunk;
166 if (total_bytes > 0) {
167 last_progress = progress;
168 progress = (current_bytes*100)/total_bytes;
171 if (ustat.size > 0) {
172 sprintf(status, "-%jd", (fsize*100)/ustat.size);
173 items[i*2 + 1] = status;
176 if (progress > last_progress)
177 dialog_mixedgauge("Fetching Distribution",
178 "Fetching distribution files...", 0, 0,
180 __DECONST(char **, items));
183 if (ustat.size > 0 && fsize < ustat.size) {
184 if (fetchLastErrCode == 0)
185 snprintf(errormsg, sizeof(errormsg),
186 "Error while fetching %s: %s\n",
187 urls[i], strerror(errno));
189 snprintf(errormsg, sizeof(errormsg),
190 "Error while fetching %s: %s\n",
191 urls[i], fetchLastErrString);
192 items[i*2 + 1] = "Failed";
193 dialog_msgbox("Fetch Error", errormsg, 0, 0,
196 items[i*2 + 1] = "Done";