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