netinit.c 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703
  1. /* vim: ts=4 sw=4 noexpandtab
  2. * This file is part of ToaruOS and is released under the terms
  3. * of the NCSA / University of Illinois License - see LICENSE.md
  4. * Copyright (C) 2015-2017 K. Lange
  5. *
  6. * netinit
  7. *
  8. * Download, decompress, and mount a root filesystem from the
  9. * network and run the `/bin/init` contained therein.
  10. *
  11. */
  12. #include <stdio.h>
  13. #include <string.h>
  14. #include <stdlib.h>
  15. #include <signal.h>
  16. #include <unistd.h>
  17. #include <fcntl.h>
  18. #include <errno.h>
  19. #include <time.h>
  20. #include <syscall.h>
  21. #include <sys/wait.h>
  22. #include <sys/ioctl.h>
  23. #include <sys/utsname.h>
  24. #include <sys/time.h>
  25. #include <getopt.h>
  26. #include <syscall.h>
  27. #define DEFAULT_URL "http://10.0.2.1:8080/netboot.img"
  28. #include <pthread.h>
  29. #include <kernel/video.h>
  30. #include "../lib/list.c"
  31. #include "../lib/hashmap.c"
  32. #include "terminal-font.h"
  33. extern int mount(char* src,char* tgt,char* typ,unsigned long,void*);
  34. #define SIZE 512
  35. struct http_req {
  36. char domain[SIZE];
  37. char path[SIZE];
  38. int port;
  39. int ssl;
  40. };
  41. struct {
  42. int show_headers;
  43. const char * output_file;
  44. const char * cookie;
  45. FILE * out;
  46. } fetch_options = {0};
  47. #define TRACE(msg,...) do { \
  48. char tmp[512]; \
  49. sprintf(tmp, msg, ##__VA_ARGS__); \
  50. fprintf(stderr, "%s", tmp); \
  51. fflush(stderr); \
  52. print_string(tmp); \
  53. } while(0)
  54. static int has_video = 0;
  55. static int width, height, depth;
  56. static char * framebuffer;
  57. static struct timeval start;
  58. static int framebuffer_fd;
  59. #define char_height 20
  60. #define char_width 9
  61. #define BG_COLOR 0xFF050505
  62. #define FG_COLOR 0xFFCCCCCC
  63. #define EX_COLOR 0xFF999999
  64. static void set_point(int x, int y, uint32_t value) {
  65. uint32_t * disp = (uint32_t *)framebuffer;
  66. uint32_t * cell = &disp[y * width + x];
  67. *cell = value;
  68. }
  69. static void write_char(int x, int y, int val, uint32_t color) {
  70. if (val > 128) {
  71. val = 4;
  72. }
  73. #ifdef number_font
  74. uint8_t * c = number_font[val];
  75. for (uint8_t i = 0; i < char_height; ++i) {
  76. for (uint8_t j = 0; j < char_width; ++j) {
  77. if (c[i] & (1 << (8-j))) {
  78. set_point(x+j,y+i,color);
  79. } else {
  80. set_point(x+j,y+i,BG_COLOR);
  81. }
  82. }
  83. }
  84. #else
  85. uint16_t * c = large_font[val];
  86. for (uint8_t i = 0; i < char_height; ++i) {
  87. for (uint8_t j = 0; j < char_width; ++j) {
  88. if (c[i] & (1 << (15-j))) {
  89. set_point(x+j,y+i,color);
  90. } else {
  91. set_point(x+j,y+i,BG_COLOR);
  92. }
  93. }
  94. }
  95. #endif
  96. }
  97. #define BUF_SIZE 255
  98. static void read_http_line(char * buf, FILE * f) {
  99. memset(buf, 0x00, BUF_SIZE);
  100. fgets(buf, BUF_SIZE, f);
  101. char * _r = strchr(buf, '\r');
  102. if (_r) {
  103. *_r = '\0';
  104. }
  105. if (!_r) {
  106. _r = strchr(buf, '\n'); /* that's not right, but, whatever */
  107. if (_r) {
  108. *_r = '\0';
  109. }
  110. }
  111. }
  112. static unsigned short * textmemptr = (unsigned short *)0xB8000;
  113. static void placech(unsigned char c, int x, int y, int attr) {
  114. unsigned short *where;
  115. unsigned att = attr << 8;
  116. where = textmemptr + (y * 80 + x);
  117. *where = c | att;
  118. }
  119. #define LEFT_PAD 40
  120. static int x = LEFT_PAD;
  121. static int y = 0;
  122. static int vx = 0;
  123. static int vy = 0;
  124. static void print_string(char * msg) {
  125. if (!has_video) {
  126. while (*msg) {
  127. placech(' ',vx,vy,0);
  128. switch (*msg) {
  129. case '\n':
  130. vx = 0;
  131. vy += 1;
  132. if (vy == 25) {
  133. /* scroll */
  134. memcpy(textmemptr,textmemptr + 80,sizeof(unsigned short) * 80 * 24);
  135. memset(textmemptr + 80 * 24, 0, sizeof(unsigned short) * 80);
  136. vy = 24;
  137. }
  138. break;
  139. case '\033':
  140. msg++;
  141. if (*msg == '[') {
  142. msg++;
  143. if (*msg == 'G') {
  144. vx = 0;
  145. }
  146. if (*msg == 'K') {
  147. int last_x = vx;
  148. while (vx < 80) {
  149. placech(' ',vx,vy,0);
  150. vx++;
  151. }
  152. vx = last_x;
  153. }
  154. }
  155. break;
  156. default:
  157. placech(*msg,vx,vy,0xF);
  158. vx++;
  159. break;
  160. }
  161. placech('_',vx,vy,0xF);
  162. msg++;
  163. }
  164. } else {
  165. while (*msg) {
  166. write_char(x,y,' ',BG_COLOR);
  167. switch (*msg) {
  168. case '\n':
  169. x = LEFT_PAD;
  170. y += char_height;
  171. if (y > height - 30) {
  172. y = 0;
  173. }
  174. break;
  175. case '\033':
  176. msg++;
  177. if (*msg == '[') {
  178. msg++;
  179. if (*msg == 'G') {
  180. x = LEFT_PAD;
  181. }
  182. if (*msg == 'K') {
  183. int last_x = x;
  184. while (x < width) {
  185. write_char(x,y,' ',FG_COLOR);
  186. x += char_width;
  187. }
  188. x = last_x;
  189. }
  190. }
  191. break;
  192. default:
  193. write_char(x,y,*msg,FG_COLOR);
  194. x += char_width;
  195. break;
  196. }
  197. write_char(x,y,'_',EX_COLOR);
  198. msg++;
  199. }
  200. }
  201. }
  202. void parse_url(char * d, struct http_req * r) {
  203. if (strstr(d, "http://") == d) {
  204. d += strlen("http://");
  205. char * s = strstr(d, "/");
  206. if (!s) {
  207. strcpy(r->domain, d);
  208. strcpy(r->path, "");
  209. } else {
  210. *s = 0;
  211. s++;
  212. strcpy(r->domain, d);
  213. strcpy(r->path, s);
  214. }
  215. if (strstr(r->domain,":")) {
  216. char * port = strstr(r->domain,":");
  217. *port = '\0';
  218. port++;
  219. r->port = atoi(port);
  220. } else {
  221. r->port = 80;
  222. }
  223. r->ssl = 0;
  224. } else if (strstr(d, "https://") == d) {
  225. d += strlen("https://");
  226. char * s = strstr(d, "/");
  227. if (!s) {
  228. strcpy(r->domain, d);
  229. strcpy(r->path, "");
  230. } else {
  231. *s = 0;
  232. s++;
  233. strcpy(r->domain, d);
  234. strcpy(r->path, s);
  235. }
  236. if (strstr(r->domain,":")) {
  237. char * port = strstr(r->domain,":");
  238. *port = '\0';
  239. port++;
  240. r->port = atoi(port);
  241. } else {
  242. r->port = 443;
  243. }
  244. r->ssl = 1;
  245. } else {
  246. TRACE("sorry, can't parse %s\n", d);
  247. exit(1);
  248. }
  249. }
  250. static void bad_response(void) {
  251. TRACE("Bad response.\n");
  252. exit(1);
  253. }
  254. static char * img = "/tmp/netboot.img";
  255. static void update_video(int sig) {
  256. (void)sig;
  257. ioctl(framebuffer_fd, IO_VID_WIDTH, &width);
  258. ioctl(framebuffer_fd, IO_VID_HEIGHT, &height);
  259. ioctl(framebuffer_fd, IO_VID_DEPTH, &depth);
  260. ioctl(framebuffer_fd, IO_VID_ADDR, &framebuffer);
  261. ioctl(framebuffer_fd, IO_VID_SIGNAL, NULL);
  262. /* Clear the screen */
  263. for (int y = 0; y < height; ++y) {
  264. for (int x = 0; x < width; ++x) {
  265. set_point(x,y,BG_COLOR);
  266. }
  267. }
  268. x = LEFT_PAD;
  269. y = 0;
  270. if (sig) {
  271. TRACE("(video display changed to %d x %d)\n", width, height);
  272. }
  273. }
  274. static volatile int watchdog_success = 0;
  275. static void network_error(int is_thread) {
  276. TRACE("\n\n");
  277. TRACE("ERROR: Network does not seem to be available, or unable to reach host.\n");
  278. TRACE(" Please check your VM configuration.\n");
  279. if (is_thread) {
  280. pthread_exit(0);
  281. } else {
  282. exit(1);
  283. }
  284. }
  285. static void * watchdog_func(void * garbage) {
  286. (void)garbage;
  287. int i = 0;
  288. while (i < 5) {
  289. usleep(1000000);
  290. if (watchdog_success) {
  291. pthread_exit(0);
  292. }
  293. i++;
  294. }
  295. network_error(1);
  296. return NULL;
  297. }
  298. #define BAR_WIDTH 20
  299. #define bar_perc "||||||||||||||||||||"
  300. #define bar_spac " "
  301. static void draw_progress(size_t content_length, size_t size) {
  302. struct timeval now;
  303. gettimeofday(&now, NULL);
  304. TRACE("\033[G%6dkB",(int)size/1024);
  305. if (content_length) {
  306. int percent = (size * BAR_WIDTH) / (content_length);
  307. TRACE(" / %6dkB [%.*s%.*s]", (int)content_length/1024, percent,bar_perc,BAR_WIDTH-percent,bar_spac);
  308. }
  309. double timediff = (double)(now.tv_sec - start.tv_sec) + (double)(now.tv_usec - start.tv_usec)/1000000.0;
  310. if (timediff > 0.0) {
  311. double rate = (double)(size) / timediff;
  312. double s = rate/(1024.0) * 8.0;
  313. if (s > 1024.0) {
  314. TRACE(" %.2f mbps", s/1024.0);
  315. } else {
  316. TRACE(" %.2f kbps", s);
  317. }
  318. if (content_length) {
  319. if (rate > 0.0) {
  320. double remaining = (double)(content_length - size) / rate;
  321. TRACE(" (%.2f sec remaining)", remaining);
  322. }
  323. }
  324. }
  325. TRACE("\033[K");
  326. }
  327. static unsigned int crctab[256] = {
  328. 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419, 0x706af48f,
  329. 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
  330. 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91, 0x1db71064, 0x6ab020f2,
  331. 0xf3b97148, 0x84be41de, 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
  332. 0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
  333. 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
  334. 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b, 0x35b5a8fa, 0x42b2986c,
  335. 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
  336. 0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423,
  337. 0xcfba9599, 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
  338. 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190, 0x01db7106,
  339. 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
  340. 0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d,
  341. 0x91646c97, 0xe6635c01, 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
  342. 0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
  343. 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
  344. 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2, 0x4adfa541, 0x3dd895d7,
  345. 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
  346. 0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa,
  347. 0xbe0b1010, 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
  348. 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17, 0x2eb40d81,
  349. 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
  350. 0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683, 0xe3630b12, 0x94643b84,
  351. 0x0d6d6a3e, 0x7a6a5aa8, 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
  352. 0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
  353. 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
  354. 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5, 0xd6d6a3e8, 0xa1d1937e,
  355. 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
  356. 0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55,
  357. 0x316e8eef, 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
  358. 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe, 0xb2bd0b28,
  359. 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
  360. 0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a, 0x9c0906a9, 0xeb0e363f,
  361. 0x72076785, 0x05005713, 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
  362. 0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
  363. 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
  364. 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c, 0x8f659eff, 0xf862ae69,
  365. 0x616bffd3, 0x166ccf45, 0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
  366. 0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc,
  367. 0x40df0b66, 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
  368. 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605, 0xcdd70693,
  369. 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
  370. 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d
  371. };
  372. /* This is taken from the kernel/sys/version.c */
  373. #if (defined(__GNUC__) || defined(__GNUG__)) && !(defined(__clang__) || defined(__INTEL_COMPILER))
  374. # define COMPILER_VERSION "gcc " __VERSION__
  375. #elif (defined(__clang__))
  376. # define COMPILER_VERSION "clang " __clang_version__
  377. #else
  378. # define COMPILER_VERSION "unknown-compiler how-did-you-do-that"
  379. #endif
  380. int main(int argc, char * argv[]) {
  381. int _stdin = open("/dev/null", O_RDONLY);
  382. int _stdout = open("/dev/ttyS0", O_WRONLY);
  383. int _stderr = open("/dev/ttyS0", O_WRONLY);
  384. if (_stdout < 0) {
  385. _stdout = open("/dev/null", O_WRONLY);
  386. _stderr = open("/dev/null", O_WRONLY);
  387. }
  388. (void)_stdin;
  389. (void)_stdout;
  390. (void)_stderr;
  391. mount("x", "/tmp", "tmpfs", 0, NULL);
  392. int tmpfd = open("/proc/framebuffer", O_RDONLY);
  393. if (tmpfd < 0) {
  394. has_video = 0;
  395. memset(textmemptr, 0, sizeof(unsigned short) * 80 * 25);
  396. } else {
  397. has_video = 1;
  398. framebuffer_fd = open("/dev/fb0", O_RDONLY);
  399. update_video(0);
  400. signal(SIGWINEVENT, update_video);
  401. }
  402. TRACE("\n\nToaruOS Netinit Host\n\n");
  403. TRACE("ToaruOS is free software under the NCSA / University of Illinois license.\n");
  404. TRACE(" https://toaruos.org/ https://git.toaruos.org/klange/toaruos\n\n");
  405. struct utsname u;
  406. uname(&u);
  407. TRACE("%s %s %s %s\n", u.sysname, u.nodename, u.release, u.version);
  408. {
  409. char kernel_version[512] = {0};
  410. int fd = open("/proc/compiler", O_RDONLY);
  411. read(fd, kernel_version, 512);
  412. if (kernel_version[strlen(kernel_version)-1] == '\n') {
  413. kernel_version[strlen(kernel_version)-1] = '\0';
  414. }
  415. TRACE(" Kernel was built with: %s\n", kernel_version);
  416. }
  417. TRACE(" Netinit binary was built with: %s\n", COMPILER_VERSION);
  418. TRACE("\n");
  419. if (has_video) {
  420. TRACE("Display is %dx%d (%d bpp), framebuffer at 0x%x\n", width, height, depth, (unsigned int)framebuffer);
  421. }
  422. TRACE("\n");
  423. TRACE("Sleeping for a moment to let network initialize...\n");
  424. sleep(2);
  425. #define LINE_LEN 100
  426. char line[LINE_LEN];
  427. FILE * f = fopen("/proc/netif", "r");
  428. while (fgets(line, LINE_LEN, f) != NULL) {
  429. if (strstr(line, "ip:") == line) {
  430. char * value = strchr(line,'\t')+1;
  431. *strchr(value,'\n') = '\0';
  432. TRACE(" IP address: %s\n", value);
  433. } else if (strstr(line, "device:") == line) {
  434. char * value = strchr(line,'\t')+1;
  435. *strchr(value,'\n') = '\0';
  436. TRACE(" Network Driver: %s\n", value);
  437. } else if (strstr(line, "mac:") == line) {
  438. char * value = strchr(line,'\t')+1;
  439. *strchr(value,'\n') = '\0';
  440. TRACE(" MAC address: %s\n", value);
  441. } else if (strstr(line, "dns:") == line) {
  442. char * value = strchr(line,'\t')+1;
  443. *strchr(value,'\n') = '\0';
  444. TRACE(" DNS server: %s\n", value);
  445. } else if (strstr(line, "gateway:") == line) {
  446. char * value = strchr(line,'\t')+1;
  447. *strchr(value,'\n') = '\0';
  448. TRACE(" Gateway: %s\n", value);
  449. } else if (strstr(line,"no network") == line){
  450. network_error(0);
  451. }
  452. memset(line, 0, LINE_LEN);
  453. }
  454. fclose(f);
  455. struct http_req my_req;
  456. /* TODO: Extract URL from kcmdline */
  457. if (argc < 2) {
  458. parse_url(DEFAULT_URL, &my_req);
  459. } else {
  460. parse_url(argv[1], &my_req);
  461. }
  462. char file[100];
  463. sprintf(file, "/dev/net/%s:%d", my_req.domain, my_req.port);
  464. TRACE("Fetching from %s... ", my_req.domain);
  465. fetch_options.out = fopen(img,"w+");
  466. pthread_t watchdog;
  467. pthread_create(&watchdog, NULL, watchdog_func, NULL);
  468. f = fopen(file,"r+");
  469. if (!f) {
  470. network_error(0);
  471. }
  472. #if 0
  473. char * vbuf = malloc(10240);
  474. setvbuf(f,vbuf,_IOLBF,10240);
  475. #endif
  476. watchdog_success = 1;
  477. TRACE("Connection established.\n");
  478. fprintf(f,
  479. "GET /%s HTTP/1.0\r\n"
  480. "User-Agent: curl/7.35.0\r\n"
  481. "Host: %s\r\n"
  482. "Accept: */*\r\n"
  483. "\r\n", my_req.path, my_req.domain);
  484. gettimeofday(&start, NULL);
  485. hashmap_t * headers = hashmap_create(10);
  486. /* Parse response */
  487. {
  488. char buf[BUF_SIZE];
  489. read_http_line(buf, f);
  490. TRACE("[%s]\n", buf);
  491. char * elements[3];
  492. elements[0] = buf;
  493. elements[1] = strchr(elements[0], ' ');
  494. if (!elements[1]) bad_response();
  495. *elements[1] = '\0';
  496. elements[1]++;
  497. elements[2] = strchr(elements[1], ' ');
  498. if (!elements[2]) bad_response();
  499. *elements[2] = '\0';
  500. elements[2]++;
  501. if (strcmp(elements[1], "200")) {
  502. TRACE("Bad response code: %s\n", elements[1]);
  503. return 1;
  504. }
  505. }
  506. while (1) {
  507. char buf[BUF_SIZE];
  508. read_http_line(buf, f);
  509. if (!*buf) {
  510. TRACE("(done with headers)\n");
  511. break;
  512. }
  513. /* Split */
  514. char * name = buf;
  515. char * value = strstr(buf, ": ");
  516. if (!value) bad_response();
  517. *value = '\0';
  518. value += 2;
  519. hashmap_set(headers, name, strdup(value));
  520. }
  521. #if 1
  522. TRACE("Dumping headers.\n");
  523. list_t * hash_keys = hashmap_keys(headers);
  524. foreach(_key, hash_keys) {
  525. char * key = (char *)_key->value;
  526. TRACE("[%s] = %s\n", key, (char*)hashmap_get(headers, key));
  527. }
  528. list_free(hash_keys);
  529. free(hash_keys);
  530. #endif
  531. /* determine how many bytes we should read now */
  532. if (!hashmap_has(headers, "Content-Length")) {
  533. TRACE("Don't know how much to read.\n");
  534. return 1;
  535. }
  536. int bytes_to_read = atoi(hashmap_get(headers, "Content-Length"));
  537. size_t bytes_total = (size_t)bytes_to_read;
  538. size_t bytes_read = 0;
  539. #define RBUF_SIZE 10240
  540. char * buf = malloc(RBUF_SIZE);
  541. uint32_t crc32 = 0xffffffff;
  542. while (bytes_to_read > 0) {
  543. size_t r = fread(buf, 1, bytes_to_read < RBUF_SIZE ? bytes_to_read : RBUF_SIZE, f);
  544. fwrite(buf, 1, r, fetch_options.out);
  545. for (size_t i = 0; i < r; ++i) {
  546. int ind = (crc32 ^ buf[i]) & 0xFF;
  547. crc32 = (crc32 >> 8) ^ (crctab[ind]);
  548. }
  549. bytes_to_read -= r;
  550. bytes_read += r;
  551. draw_progress(bytes_total, bytes_read);
  552. }
  553. crc32 ^= 0xffffffff;
  554. free(buf);
  555. TRACE("\nDone: 0x%x\n", (unsigned int)crc32);
  556. fflush(fetch_options.out);
  557. fclose(fetch_options.out);
  558. #if 0
  559. FILE * xtmp = fopen(img, "r");
  560. crc32 = 0xffffffff;
  561. int tab = 0;
  562. size_t bytesread = 0;
  563. while (!feof(xtmp)) {
  564. uint8_t buf[1024];
  565. size_t r = fread(buf, 1, 1024, xtmp);
  566. for (size_t i = 0; i < r; ++i) {
  567. if (tab & 0x01) {
  568. TRACE("%2x ", (unsigned char)buf[i]);
  569. } else {
  570. TRACE("%2x", (unsigned char)buf[i]);
  571. }
  572. tab++;
  573. if (tab == 32){
  574. tab = 0;
  575. TRACE("\n");
  576. }
  577. int ind = (crc32 ^ buf[i]) & 0xFF;
  578. crc32 = (crc32 >> 8) ^ (crctab[ind]);
  579. }
  580. bytesread += r;
  581. }
  582. crc32 ^= 0xffffffff;
  583. TRACE("\nDisk crc32: 0x%x (%d)\n", (unsigned int)crc32, bytesread);
  584. #endif
  585. TRACE("Mounting filesystem... ");
  586. int err = 0;
  587. if ((err = mount(img, "/", "ext2", 0, NULL))) {
  588. TRACE("Mount error: %d; errno=%d\n", err, errno);
  589. return 0;
  590. } else {
  591. TRACE("Done.\n");
  592. }
  593. FILE * tmp = fopen("/bin/init","r");
  594. if (!tmp) {
  595. TRACE("/bin/init missing?\n");
  596. } else {
  597. TRACE("/bin/init exists, filesystem successfully mounted\n");
  598. fclose(tmp);
  599. }
  600. TRACE("Executing init...\n");
  601. char * const _argv[] = {
  602. "/bin/init",
  603. NULL,
  604. };
  605. execve("/bin/init",_argv,NULL);
  606. TRACE("ERROR: If you are seeing this, there was a problem\n");
  607. TRACE(" executing the init binary from the downloaded\n");
  608. TRACE(" filesystem. This may indicate a corrupted\n");
  609. TRACE(" download. Please try again.\n");
  610. return 0;
  611. }