xref: /freebsd/usr.sbin/bsdinstall/distfetch/distfetch.c (revision d2ce15bd43b3a1dcce08eecbff8d5d359946d972)
1 /*-
2  * Copyright (c) 2011 Nathan Whitehorn
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
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.
13  *
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
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #include <sys/param.h>
30 #include <stdio.h>
31 #include <errno.h>
32 #include <fetch.h>
33 #include <dialog.h>
34 
35 static int fetch_files(int nfiles, char **urls);
36 
37 int
38 main(void)
39 {
40 	char *diststring;
41 	char **urls;
42 	int i, nfetched, ndists = 0;
43 
44 	if (getenv("DISTRIBUTIONS") == NULL) {
45 		fprintf(stderr, "DISTRIBUTIONS variable is not set\n");
46 		return (1);
47 	}
48 
49 	diststring = strdup(getenv("DISTRIBUTIONS"));
50 	for (i = 0; diststring[i] != 0; i++)
51 		if (isspace(diststring[i]) && !isspace(diststring[i+1]))
52 			ndists++;
53 	ndists++; /* Last one */
54 
55 	urls = calloc(ndists, sizeof(const char *));
56 	if (urls == NULL) {
57 		fprintf(stderr, "Out of memory!\n");
58 		free(diststring);
59 		return (1);
60 	}
61 
62 	init_dialog(stdin, stdout);
63 	dialog_vars.backtitle = __DECONST(char *, "FreeBSD Installer");
64 	dlg_put_backtitle();
65 
66 	for (i = 0; i < ndists; i++) {
67 		urls[i] = malloc(PATH_MAX);
68 		sprintf(urls[i], "%s/%s", getenv("BSDINSTALL_DISTSITE"),
69 		    strsep(&diststring, " \t"));
70 	}
71 
72 	if (chdir(getenv("BSDINSTALL_DISTDIR")) != 0) {
73 		char error[512];
74 		sprintf(error, "Could could change to directory %s: %s\n",
75 		    getenv("BSDINSTALL_DISTDIR"), strerror(errno));
76 		dialog_msgbox("Error", error, 0, 0, TRUE);
77 		end_dialog();
78 		return (1);
79 	}
80 
81 	nfetched = fetch_files(ndists, urls);
82 
83 	end_dialog();
84 
85 	free(diststring);
86 	for (i = 0; i < ndists; i++)
87 		free(urls[i]);
88 	free(urls);
89 
90 	return ((nfetched == ndists) ? 0 : 1);
91 }
92 
93 static int
94 fetch_files(int nfiles, char **urls)
95 {
96 	const char **items;
97 	FILE *fetch_out, *file_out;
98 	struct url_stat ustat;
99 	off_t total_bytes, current_bytes, fsize;
100 	char status[8];
101 	char errormsg[512];
102 	uint8_t block[4096];
103 	size_t chunk;
104 	int i, progress, last_progress;
105 	int nsuccess = 0; /* Number of files successfully downloaded */
106 
107 	progress = 0;
108 
109 	/* Make the transfer list for dialog */
110 	items = calloc(sizeof(char *), nfiles * 2);
111 	if (items == NULL) {
112 		fprintf(stderr, "Out of memory!\n");
113 		return (-1);
114 	}
115 
116 	for (i = 0; i < nfiles; i++) {
117 		items[i*2] = strrchr(urls[i], '/');
118 		if (items[i*2] != NULL)
119 			items[i*2]++;
120 		else
121 			items[i*2] = urls[i];
122 		items[i*2 + 1] = "Pending";
123 	}
124 
125 	dialog_msgbox("", "Connecting to server.\nPlease wait...", 0, 0, FALSE);
126 
127 	/* Try to stat all the files */
128 	total_bytes = 0;
129 	for (i = 0; i < nfiles; i++) {
130 		if (fetchStatURL(urls[i], &ustat, "") == 0 && ustat.size > 0)
131 			total_bytes += ustat.size;
132 	}
133 
134 	current_bytes = 0;
135 	for (i = 0; i < nfiles; i++) {
136 		last_progress = progress;
137 		if (total_bytes == 0)
138 			progress = (i*100)/nfiles;
139 
140 		fetchLastErrCode = 0;
141 		fetch_out = fetchXGetURL(urls[i], &ustat, "");
142 		if (fetch_out == NULL) {
143 			snprintf(errormsg, sizeof(errormsg),
144 			    "Error while fetching %s: %s\n", urls[i],
145 			    fetchLastErrString);
146 			items[i*2 + 1] = "Failed";
147 			dialog_msgbox("Fetch Error", errormsg, 0, 0,
148 			    TRUE);
149 			continue;
150 		}
151 
152 		items[i*2 + 1] = "In Progress";
153 		fsize = 0;
154 		file_out = fopen(items[i*2], "w+");
155 		if (file_out == NULL) {
156 			snprintf(errormsg, sizeof(errormsg),
157 			    "Error while fetching %s: %s\n",
158 			    urls[i], strerror(errno));
159 			items[i*2 + 1] = "Failed";
160 			dialog_msgbox("Fetch Error", errormsg, 0, 0,
161 			    TRUE);
162 			fclose(fetch_out);
163 			continue;
164 		}
165 
166 		while ((chunk = fread(block, 1, sizeof(block), fetch_out))
167 		    > 0) {
168 			if (fwrite(block, 1, chunk, file_out) < chunk)
169 				break;
170 
171 			current_bytes += chunk;
172 			fsize += chunk;
173 
174 			if (total_bytes > 0) {
175 				last_progress = progress;
176 				progress = (current_bytes*100)/total_bytes;
177 			}
178 
179 			if (ustat.size > 0) {
180 				sprintf(status, "-%jd", (fsize*100)/ustat.size);
181 				items[i*2 + 1] = status;
182 			}
183 
184 			if (progress > last_progress)
185 				dialog_mixedgauge("Fetching Distribution",
186 				    "Fetching distribution files...", 0, 0,
187 				    progress, nfiles,
188 				    __DECONST(char **, items));
189 		}
190 
191 		if (ustat.size > 0 && fsize < ustat.size) {
192 			if (fetchLastErrCode == 0)
193 				snprintf(errormsg, sizeof(errormsg),
194 				    "Error while fetching %s: %s\n",
195 				    urls[i], strerror(errno));
196 			else
197 				snprintf(errormsg, sizeof(errormsg),
198 				    "Error while fetching %s: %s\n",
199 				    urls[i], fetchLastErrString);
200 			items[i*2 + 1] = "Failed";
201 			dialog_msgbox("Fetch Error", errormsg, 0, 0,
202 				    TRUE);
203 		} else {
204 			items[i*2 + 1] = "Done";
205 			nsuccess++;
206 		}
207 
208 		fclose(fetch_out);
209 		fclose(file_out);
210 	}
211 
212 	free(items);
213 	return (nsuccess);
214 }
215 
216