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