wok annotate syslinux/stuff/extra/md5sum.c @ rev 21820

syslinux/kbd: check kbd malloc pointer
author Pascal Bellard <pascal.bellard@slitaz.org>
date Sun Aug 25 11:41:26 2019 +0200 (2019-08-25)
parents 1fdfb6902d41
children 54edd52e6caa
rev   line source
pascal@12210 1 /*
pascal@12215 2 * Based on busybox code.
pascal@12210 3 *
pascal@12215 4 * Compute MD5 checksum of strings according to the
pascal@12215 5 * definition of MD5 in RFC 1321 from April 1992.
pascal@12210 6 *
pascal@12215 7 * Written by Ulrich Drepper <drepper@gnu.ai.mit.edu>, 1995.
pascal@12215 8 *
pascal@12215 9 * Copyright (C) 1995-1999 Free Software Foundation, Inc.
pascal@12215 10 * Copyright (C) 2001 Manuel Novoa III
pascal@12215 11 * Copyright (C) 2003 Glenn L. McGrath
pascal@12215 12 * Copyright (C) 2003 Erik Andersen
pascal@12210 13 * Copyright (C) 2010 Denys Vlasenko
pascal@12215 14 * Copyright (C) 2012 Pascal Bellard
pascal@12210 15 *
pascal@12215 16 * Licensed under GPLv2 or later
pascal@12210 17 */
pascal@12210 18
pascal@12210 19 #include <stdio.h>
pascal@12210 20 #include <stdlib.h>
pascal@12210 21 #include <string.h>
pascal@12210 22 #include <unistd.h>
pascal@12210 23 #include <fcntl.h>
pascal@12210 24 #include <console.h>
pascal@12210 25 #include <com32.h>
pascal@18674 26 #include <syslinux/config.h>
pascal@18674 27 #include <syslinux/disk.h>
pascal@12210 28
pascal@12210 29 #define ALIGN1
pascal@12215 30
pascal@12215 31 static uint8_t wbuffer[64]; /* always correctly aligned for uint64_t */
pascal@12215 32 static uint64_t total64; /* must be directly before hash[] */
pascal@12215 33 static uint32_t hash[8]; /* 4 elements for md5, 5 for sha1, 8 for sha256 */
pascal@12210 34
pascal@12210 35 /* Emit a string of hex representation of bytes */
pascal@12215 36 static char* bin2hex(char *p)
pascal@12210 37 {
pascal@12215 38 static const char bb_hexdigits_upcase[] ALIGN1 = "0123456789abcdef";
pascal@12215 39 int count = 16;
pascal@12215 40 const char *cp = (const char *) hash;
pascal@12210 41 while (count) {
pascal@12210 42 unsigned char c = *cp++;
pascal@12210 43 /* put lowercase hex digits */
pascal@12215 44 *p++ = bb_hexdigits_upcase[c >> 4];
pascal@12215 45 *p++ = bb_hexdigits_upcase[c & 0xf];
pascal@12210 46 count--;
pascal@12210 47 }
pascal@12210 48 return p;
pascal@12210 49 }
pascal@12210 50
pascal@12210 51 //#define rotl32(x,n) (((x) << (n)) | ((x) >> (32 - (n))))
pascal@12210 52 static uint32_t rotl32(uint32_t x, unsigned n)
pascal@12210 53 {
pascal@12210 54 return (x << n) | (x >> (32 - n));
pascal@12210 55 }
pascal@12210 56
pascal@12215 57 static void md5_process_block64(void);
pascal@12210 58
pascal@12210 59 /* Feed data through a temporary buffer.
pascal@12210 60 * The internal buffer remembers previous data until it has 64
pascal@12210 61 * bytes worth to pass on.
pascal@12210 62 */
pascal@12215 63 static void common64_hash(const void *buffer, size_t len)
pascal@12210 64 {
pascal@12215 65 unsigned bufpos = total64 & 63;
pascal@12210 66
pascal@12215 67 total64 += len;
pascal@12210 68
pascal@12210 69 while (1) {
pascal@12210 70 unsigned remaining = 64 - bufpos;
pascal@12210 71 if (remaining > len)
pascal@12210 72 remaining = len;
pascal@12210 73 /* Copy data into aligned buffer */
pascal@12215 74 memcpy(wbuffer + bufpos, buffer, remaining);
pascal@12210 75 len -= remaining;
pascal@12210 76 buffer = (const char *)buffer + remaining;
pascal@12210 77 bufpos += remaining;
pascal@12210 78 /* clever way to do "if (bufpos != 64) break; ... ; bufpos = 0;" */
pascal@12210 79 bufpos -= 64;
pascal@12210 80 if (bufpos != 0)
pascal@12210 81 break;
pascal@12210 82 /* Buffer is filled up, process it */
pascal@12215 83 md5_process_block64();
pascal@12210 84 /*bufpos = 0; - already is */
pascal@12210 85 }
pascal@12210 86 }
pascal@12210 87
pascal@12210 88 /* Process the remaining bytes in the buffer */
pascal@12215 89 static void common64_end(void)
pascal@12210 90 {
pascal@12215 91 unsigned bufpos = total64 & 63;
pascal@12210 92 /* Pad the buffer to the next 64-byte boundary with 0x80,0,0,0... */
pascal@12215 93 wbuffer[bufpos++] = 0x80;
pascal@12210 94
pascal@12210 95 /* This loop iterates either once or twice, no more, no less */
pascal@12210 96 while (1) {
pascal@12210 97 unsigned remaining = 64 - bufpos;
pascal@12215 98 memset(wbuffer + bufpos, 0, remaining);
pascal@12210 99 /* Do we have enough space for the length count? */
pascal@12210 100 if (remaining >= 8) {
pascal@12210 101 /* Store the 64-bit counter of bits in the buffer */
pascal@12215 102 uint64_t t = total64 << 3;
pascal@12210 103 /* wbuffer is suitably aligned for this */
pascal@12215 104 *(uint64_t *) (&wbuffer[64 - 8]) = t;
pascal@12210 105 }
pascal@12215 106 md5_process_block64();
pascal@12210 107 if (remaining >= 8)
pascal@12210 108 break;
pascal@12210 109 bufpos = 0;
pascal@12210 110 }
pascal@12210 111 }
pascal@12210 112
pascal@12210 113 /* These are the four functions used in the four steps of the MD5 algorithm
pascal@12210 114 * and defined in the RFC 1321. The first function is a little bit optimized
pascal@12210 115 * (as found in Colin Plumbs public domain implementation).
pascal@12210 116 * #define FF(b, c, d) ((b & c) | (~b & d))
pascal@12210 117 */
pascal@12210 118 #undef FF
pascal@12210 119 #undef FG
pascal@12210 120 #undef FH
pascal@12210 121 #undef FI
pascal@12210 122 #define FF(b, c, d) (d ^ (b & (c ^ d)))
pascal@12210 123 #define FG(b, c, d) FF(d, b, c)
pascal@12210 124 #define FH(b, c, d) (b ^ c ^ d)
pascal@12210 125 #define FI(b, c, d) (c ^ (b | ~d))
pascal@12210 126
pascal@12210 127 /* Hash a single block, 64 bytes long and 4-byte aligned */
pascal@12215 128 static void md5_process_block64(void)
pascal@12210 129 {
pascal@12210 130 /* Before we start, one word to the strange constants.
pascal@12210 131 They are defined in RFC 1321 as
pascal@12210 132 T[i] = (int)(4294967296.0 * fabs(sin(i))), i=1..64
pascal@12210 133 */
pascal@12210 134 static const uint32_t C_array[] = {
pascal@12210 135 /* round 1 */
pascal@12210 136 0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee,
pascal@12210 137 0xf57c0faf, 0x4787c62a, 0xa8304613, 0xfd469501,
pascal@12210 138 0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be,
pascal@12210 139 0x6b901122, 0xfd987193, 0xa679438e, 0x49b40821,
pascal@12210 140 /* round 2 */
pascal@12210 141 0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa,
pascal@12210 142 0xd62f105d, 0x02441453, 0xd8a1e681, 0xe7d3fbc8,
pascal@12210 143 0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed,
pascal@12210 144 0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a,
pascal@12210 145 /* round 3 */
pascal@12210 146 0xfffa3942, 0x8771f681, 0x6d9d6122, 0xfde5380c,
pascal@12210 147 0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70,
pascal@12210 148 0x289b7ec6, 0xeaa127fa, 0xd4ef3085, 0x4881d05,
pascal@12210 149 0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665,
pascal@12210 150 /* round 4 */
pascal@12210 151 0xf4292244, 0x432aff97, 0xab9423a7, 0xfc93a039,
pascal@12210 152 0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1,
pascal@12210 153 0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1,
pascal@12210 154 0xf7537e82, 0xbd3af235, 0x2ad7d2bb, 0xeb86d391
pascal@12210 155 };
pascal@12210 156 static const char P_array[] ALIGN1 = {
pascal@12210 157 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, /* 1 */
pascal@12210 158 1, 6, 11, 0, 5, 10, 15, 4, 9, 14, 3, 8, 13, 2, 7, 12, /* 2 */
pascal@12210 159 5, 8, 11, 14, 1, 4, 7, 10, 13, 0, 3, 6, 9, 12, 15, 2, /* 3 */
pascal@12210 160 0, 7, 14, 5, 12, 3, 10, 1, 8, 15, 6, 13, 4, 11, 2, 9 /* 4 */
pascal@12210 161 };
pascal@12215 162 uint32_t *words = (void*) wbuffer;
pascal@12215 163 uint32_t A = hash[0];
pascal@12215 164 uint32_t B = hash[1];
pascal@12215 165 uint32_t C = hash[2];
pascal@12215 166 uint32_t D = hash[3];
pascal@12210 167
pascal@12210 168 static const char S_array[] ALIGN1 = {
pascal@12210 169 7, 12, 17, 22,
pascal@12210 170 5, 9, 14, 20,
pascal@12210 171 4, 11, 16, 23,
pascal@12210 172 6, 10, 15, 21
pascal@12210 173 };
pascal@12210 174 const uint32_t *pc;
pascal@12210 175 const char *pp;
pascal@12210 176 const char *ps;
pascal@12210 177 int i;
pascal@12210 178 uint32_t temp;
pascal@12210 179
pascal@12210 180
pascal@12210 181 pc = C_array;
pascal@12210 182 pp = P_array;
pascal@12210 183 ps = S_array - 4;
pascal@12210 184
pascal@12210 185 for (i = 0; i < 64; i++) {
pascal@12210 186 if ((i & 0x0f) == 0)
pascal@12210 187 ps += 4;
pascal@12210 188 temp = A;
pascal@12210 189 switch (i >> 4) {
pascal@12210 190 case 0:
pascal@12210 191 temp += FF(B, C, D);
pascal@12210 192 break;
pascal@12210 193 case 1:
pascal@12210 194 temp += FG(B, C, D);
pascal@12210 195 break;
pascal@12210 196 case 2:
pascal@12210 197 temp += FH(B, C, D);
pascal@12210 198 break;
pascal@12210 199 case 3:
pascal@12210 200 temp += FI(B, C, D);
pascal@12210 201 }
pascal@12210 202 temp += words[(int) (*pp++)] + *pc++;
pascal@12210 203 temp = rotl32(temp, ps[i & 3]);
pascal@12210 204 temp += B;
pascal@12210 205 A = D;
pascal@12210 206 D = C;
pascal@12210 207 C = B;
pascal@12210 208 B = temp;
pascal@12210 209 }
pascal@12210 210 /* Add checksum to the starting values */
pascal@12215 211 hash[0] += A;
pascal@12215 212 hash[1] += B;
pascal@12215 213 hash[2] += C;
pascal@12215 214 hash[3] += D;
pascal@12210 215
pascal@12210 216 }
pascal@12210 217 #undef FF
pascal@12210 218 #undef FG
pascal@12210 219 #undef FH
pascal@12210 220 #undef FI
pascal@12210 221
pascal@12210 222 /* Initialize structure containing state of computation.
pascal@12210 223 * (RFC 1321, 3.3: Step 3)
pascal@12210 224 */
pascal@12215 225 static void md5_begin(void)
pascal@12210 226 {
pascal@12215 227 hash[0] = 0x67452301;
pascal@12215 228 hash[1] = 0xefcdab89;
pascal@12215 229 hash[2] = 0x98badcfe;
pascal@12215 230 hash[3] = 0x10325476;
pascal@12215 231 total64 = 0;
pascal@12210 232 }
pascal@12210 233
pascal@12210 234 /* Used also for sha1 and sha256 */
pascal@12215 235 #define md5_hash common64_hash
pascal@12210 236
pascal@12210 237 /* Process the remaining bytes in the buffer and put result from CTX
pascal@12210 238 * in first 16 bytes following RESBUF. The result is always in little
pascal@12210 239 * endian byte order, so that a byte-wise output yields to the wanted
pascal@12210 240 * ASCII representation of the message digest.
pascal@12210 241 */
pascal@12215 242 #define md5_end common64_end
pascal@12210 243
pascal@12210 244 /*
pascal@12210 245 * Copyright (C) 2003 Glenn L. McGrath
pascal@12210 246 * Copyright (C) 2003-2004 Erik Andersen
pascal@12210 247 *
pascal@12210 248 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
pascal@12210 249 */
pascal@12210 250
pascal@20062 251 #ifdef EXTRA
pascal@20062 252 #define WITH_UNROCKRIDGE
pascal@20062 253 #endif
pascal@20062 254 #ifdef WITH_UNROCKRIDGE
pascal@12215 255 static char *unrockridge(const char *name)
pascal@12212 256 {
pascal@12212 257 static char buffer[256];
pascal@12215 258 int i = 0, j = 8;
pascal@12215 259 while (*name && i < 255) {
pascal@12215 260 char c = *name++;
pascal@12215 261 //if (c == '\\') c = '/';
pascal@12215 262 if (c == '.') {
pascal@12215 263 for (j = i; --j >= 0 && buffer[j] != '/';)
pascal@12215 264 if (buffer[j] == '.') buffer[j] = '_';
pascal@12215 265 if (i - j > 9) i = j + 9;
pascal@12215 266 j = i + 4;
pascal@12212 267 }
pascal@12215 268 else if (c == '/') j = i + 9;
pascal@12212 269 else if (i >= j) continue;
pascal@12215 270 else if (c >= 'a' && c <= 'z') c += 'A' - 'a';
pascal@12215 271 else if ((c < 'A' || c > 'Z') && (c < '0' || c > '9')) c = '_';
pascal@12215 272 buffer[i++] = c;
pascal@12212 273 }
pascal@12215 274 buffer[i] = 0;
pascal@12212 275 return buffer;
pascal@12212 276 }
pascal@20062 277 #endif
pascal@12212 278
pascal@12210 279 static uint8_t *hash_file(const char *filename)
pascal@12210 280 {
pascal@12210 281 int src_fd, count;
pascal@12215 282 uint8_t in_buf[4096];
pascal@12215 283 static uint8_t hash_value[16*2+1];
pascal@12210 284
pascal@12210 285 src_fd = open(filename, O_RDONLY);
pascal@20062 286 #ifdef WITH_UNROCKRIDGE
pascal@12210 287 if (src_fd < 0) {
pascal@12215 288 src_fd = open(unrockridge(filename), O_RDONLY);
pascal@12212 289 }
pascal@20062 290 #endif
pascal@20062 291
pascal@12212 292 if (src_fd < 0) {
pascal@12210 293 return NULL;
pascal@12210 294 }
pascal@12210 295
pascal@12215 296 md5_begin();
pascal@12210 297 while ((count = read(src_fd, in_buf, 4096)) > 0) {
pascal@12215 298 md5_hash(in_buf, count);
pascal@12210 299 }
pascal@12210 300
pascal@12210 301 close(src_fd);
pascal@12215 302
pascal@12215 303 if (count)
pascal@12215 304 return NULL;
pascal@12215 305
pascal@12215 306 md5_end();
pascal@12215 307 bin2hex((char *)hash_value);
pascal@12210 308
pascal@12210 309 return hash_value;
pascal@12210 310 }
pascal@12210 311
pascal@20062 312 #ifdef EXTRA
pascal@13295 313 static int main_say(int argc, char **argv)
pascal@13295 314 {
pascal@13295 315 int i;
pascal@13295 316 for (i = 1; i < argc; i++) {
pascal@13295 317 printf("%s ",argv[i]);
pascal@13295 318 }
pascal@13295 319 sleep(5);
pascal@13551 320 return 0;
pascal@13295 321 }
pascal@20062 322 #endif
pascal@13295 323
pascal@12273 324 static int main_md5sum(int argc, char **argv)
pascal@12210 325 {
pascal@12215 326 int files = 0, tested = 0, good = 0;
pascal@12215 327 static char clear_eol[] = " ";
pascal@12210 328
pascal@12210 329 (void) argc;
pascal@12210 330 /* -c implied */
pascal@12287 331 argv++;
pascal@12210 332 do {
pascal@12210 333 FILE *fp;
pascal@13361 334 char eol, *line, buffer[4096];
pascal@12210 335 fp = fopen(*argv,"r");
pascal@20062 336 #ifdef WITH_UNROCKRIDGE
pascal@12212 337 if (fp == NULL)
pascal@12215 338 fp = fopen(unrockridge(*argv),"r");
pascal@20062 339 #endif
pascal@12210 340
pascal@13361 341 while ((line = fgets(buffer,sizeof(buffer),fp)) != NULL) {
pascal@12210 342 uint8_t *hash_value;
pascal@12210 343 char *filename_ptr, *status;
pascal@12210 344 int len = strlen(line);
pascal@12210 345
pascal@12210 346 if (line[0] < '0')
pascal@12210 347 continue;
pascal@12210 348 if (line[len-1] == '\n')
pascal@12210 349 line[len-1] = 0;
pascal@12210 350 filename_ptr = strstr(line, " ");
pascal@12210 351 /* handle format for binary checksums */
pascal@12210 352 if (filename_ptr == NULL) {
pascal@12210 353 filename_ptr = strstr(line, " *");
pascal@12210 354 }
pascal@12210 355 if (filename_ptr == NULL) {
pascal@12210 356 continue;
pascal@12210 357 }
pascal@12210 358 *filename_ptr = '\0';
pascal@12210 359 *++filename_ptr = '/';
pascal@12212 360 if (filename_ptr[1] == '/')
pascal@12212 361 filename_ptr++;
pascal@12210 362
pascal@12215 363 files++;
pascal@12215 364 status = "NOT CHECKED";
pascal@12215 365 eol = '\n';
pascal@12211 366 hash_value = hash_file(filename_ptr);
pascal@12211 367 if (hash_value) {
pascal@12215 368 tested++;
pascal@12215 369 status = "BROKEN";
pascal@12215 370 if (!strcmp((char*)hash_value, line)) {
pascal@12215 371 good++;
pascal@12215 372 status = "OK";
pascal@12215 373 eol = ' ';
pascal@12210 374 }
pascal@12210 375 }
pascal@12215 376 printf("\r%s: %s%s%c", filename_ptr, status, clear_eol, eol);
pascal@12210 377 }
pascal@12210 378 fclose(fp);
pascal@12210 379 } while (*++argv);
pascal@12215 380 printf("\r%d files OK, %d broken, %d not checked.%s\n",
pascal@12215 381 good, tested - good, files - tested, clear_eol);
pascal@12287 382 sleep(5);
pascal@12273 383 return 0;
pascal@12215 384 }
pascal@12210 385
pascal@12273 386 /*
pascal@12273 387 * ifmem.c
pascal@12273 388 *
pascal@12273 389 * Run one command if the memory is large enought, and another if it isn't.
pascal@12273 390 *
pascal@12273 391 * Usage:
pascal@12273 392 *
pascal@12273 393 * label boot_kernel
pascal@12273 394 * kernel ifmem.c
pascal@12273 395 * append size_in_KB boot_large [size_in_KB boot_medium] boot_small
pascal@12273 396 *
pascal@12273 397 * label boot_large
pascal@12273 398 * kernel vmlinuz_large_memory
pascal@12273 399 * append ...
pascal@12273 400 *
pascal@12273 401 * label boot_small
pascal@12273 402 * kernel vmlinuz_small_memory
pascal@12273 403 * append ...
pascal@12273 404 */
pascal@12273 405
pascal@12273 406 #include <inttypes.h>
pascal@12273 407 #include <alloca.h>
pascal@12273 408 #include <syslinux/boot.h>
pascal@12273 409
pascal@12273 410 struct e820_data {
pascal@12273 411 uint64_t base;
pascal@12273 412 uint64_t len;
pascal@12273 413 uint32_t type;
pascal@12273 414 uint32_t extattr;
pascal@12273 415 } __attribute__((packed));
pascal@12273 416
pascal@12273 417 // Get memory size in Kb
pascal@12273 418 static unsigned long memory_size(void)
pascal@12273 419 {
pascal@12273 420 uint64_t bytes = 0;
pascal@12273 421 static com32sys_t ireg, oreg;
pascal@12273 422 static struct e820_data ed;
pascal@12273 423
pascal@12273 424 ireg.eax.w[0] = 0xe820;
pascal@12273 425 ireg.edx.l = 0x534d4150;
pascal@12273 426 ireg.ecx.l = sizeof(struct e820_data);
pascal@12273 427 ireg.edi.w[0] = OFFS(__com32.cs_bounce);
pascal@12273 428 ireg.es = SEG(__com32.cs_bounce);
pascal@12273 429
pascal@12273 430 ed.extattr = 1;
pascal@12273 431
pascal@12273 432 do {
pascal@12273 433 memcpy(__com32.cs_bounce, &ed, sizeof ed);
pascal@12273 434
pascal@12273 435 __intcall(0x15, &ireg, &oreg);
pascal@12273 436 if (oreg.eflags.l & EFLAGS_CF ||
pascal@12273 437 oreg.eax.l != 0x534d4150 ||
pascal@12273 438 oreg.ecx.l < 20)
pascal@12273 439 break;
pascal@12273 440
pascal@12273 441 memcpy(&ed, __com32.cs_bounce, sizeof ed);
pascal@12273 442
pascal@12273 443 if (ed.type == 1)
pascal@12273 444 bytes += ed.len;
pascal@12273 445
pascal@12273 446 ireg.ebx.l = oreg.ebx.l;
pascal@12273 447 } while (ireg.ebx.l);
pascal@12273 448
pascal@12273 449 if (!bytes) {
pascal@12273 450 memset(&ireg, 0, sizeof ireg);
pascal@12273 451 ireg.eax.w[0] = 0x8800;
pascal@12273 452 __intcall(0x15, &ireg, &oreg);
pascal@12273 453 return ireg.eax.w[0];
pascal@12273 454 }
pascal@12273 455 return bytes >> 10;
pascal@12273 456 }
pascal@12273 457
pascal@12349 458 static void usage(const char *msg)
pascal@12349 459 {
pascal@12349 460 fprintf(stderr,"\n%s\n.",msg);
pascal@12349 461 sleep(5);
pascal@12349 462 exit(1);
pascal@12349 463 }
pascal@12349 464
pascal@12273 465 static int main_ifmem(int argc, char *argv[])
pascal@12273 466 {
pascal@12273 467 int i;
pascal@12273 468 unsigned long ram_size;
pascal@12273 469
pascal@12273 470 if (argc < 4) {
pascal@12349 471 usage("Usage: ifmem.c32 size_KB boot_large_memory boot_small_memory");
pascal@12273 472 }
pascal@12273 473
pascal@12273 474 // find target according to ram size
pascal@12273 475 ram_size = memory_size();
pascal@12273 476 printf("Total memory found %luK.\n",ram_size);
pascal@12273 477 ram_size += (1 << 10); // add 1M to round boundaries...
pascal@12273 478
pascal@12273 479 i = 1;
pascal@12273 480 do {
pascal@12273 481 char *s = argv[i];
pascal@12273 482 char *p = s;
pascal@12273 483 unsigned long scale = 1;
pascal@12273 484
pascal@12273 485 while (*p >= '0' && *p <= '9') p++;
pascal@12273 486 switch (*p | 0x20) {
pascal@12273 487 case 'g': scale <<= 10;
pascal@12273 488 case 'm': scale <<= 10;
pascal@12273 489 default : *p = 0; break;
pascal@12273 490 }
pascal@12273 491 i++; // seek to label
pascal@12273 492 if (ram_size >= scale * strtoul(s, NULL, 0)) break;
pascal@12273 493 i++; // next size or default label
pascal@12273 494 } while (i + 1 < argc);
pascal@12273 495
pascal@12349 496 if (i != argc) syslinux_run_command(argv[i]);
pascal@12273 497 else syslinux_run_default();
pascal@12273 498 return -1;
pascal@12273 499 }
pascal@12273 500
pascal@12273 501 #include <syslinux/reboot.h>
pascal@12273 502
pascal@12273 503 static int main_reboot(int argc, char *argv[])
pascal@12273 504 {
pascal@12273 505 int warm = 0;
pascal@12273 506 int i;
pascal@12273 507
pascal@12273 508 for (i = 1; i < argc; i++) {
pascal@20073 509 if (strstr(argv[i], "-w"))
pascal@12273 510 warm = 1;
pascal@12273 511 }
pascal@12273 512
pascal@12273 513 syslinux_reboot(warm);
pascal@12273 514 }
pascal@12273 515
pascal@12273 516 /* APM poweroff module.
pascal@12273 517 * based on poweroff.asm, Copyright 2009 Sebastian Herbszt
pascal@12273 518 */
pascal@12273 519
pascal@12273 520 static int main_poweroff(int argc, char *argv[])
pascal@12273 521 {
pascal@12273 522 static com32sys_t ireg, oreg;
pascal@12273 523 static char notsupported[] ="APM 1.1+ not supported";
pascal@12273 524 unsigned i;
pascal@12273 525 static struct {
pascal@12273 526 unsigned short ax;
pascal@12273 527 unsigned short bx;
pascal@12273 528 unsigned short cx;
pascal@12273 529 char *msg;
pascal@12273 530 } inst[] = {
pascal@12273 531 { 0x5300, // APM Installation Check (00h)
pascal@12273 532 0, // APM BIOS (0000h)
pascal@12273 533 0, "APM not present" },
pascal@12273 534 { 0x5301, // APM Real Mode Interface Connect (01h)
pascal@12273 535 0, // APM BIOS (0000h)
pascal@12273 536 0, "APM RM interface connect failed" },
pascal@12273 537 { 0x530E, // APM Driver Version (0Eh)
pascal@12273 538 0, // APM BIOS (0000h)
pascal@12273 539 0x0101, // APM Driver Version version 1.1
pascal@12273 540 notsupported },
pascal@12273 541 { 0x5307, // Set Power State (07h)
pascal@12273 542 1, // All devices power managed by the APM
pascal@12273 543 3, // Power state off
pascal@12273 544 "Power off failed" }
pascal@12273 545 };
pascal@12273 546
pascal@12273 547 (void) argc;
pascal@12273 548 (void) argv;
pascal@12273 549 for (i = 0; i < sizeof(inst)/sizeof(inst[0]); i++) {
pascal@12273 550 char *msg = inst[i].msg;
pascal@12273 551
pascal@12273 552 ireg.eax.w[0] = inst[i].ax;
pascal@12273 553 ireg.ebx.w[0] = inst[i].bx;
pascal@12273 554 ireg.ecx.w[0] = inst[i].cx;
pascal@12273 555 __intcall(0x15, &ireg, &oreg);
pascal@12273 556 if ((oreg.eflags.l & EFLAGS_CF) == 0) {
pascal@12273 557 switch (inst[i].ax) {
pascal@12273 558 case 0x5300 :
pascal@12273 559 if (oreg.ebx.w[0] != 0x504D /* 'PM' */) break;
pascal@12273 560 msg = "Power management disabled";
pascal@12273 561 if (oreg.ecx.w[0] & 8) break; // bit 3 APM BIOS Power Management disabled
pascal@12273 562 case 0x530E :
pascal@12273 563 msg = notsupported;
pascal@12273 564 if (oreg.eax.w[0] < 0x101) break;
pascal@12273 565 default : continue;
pascal@12273 566 }
pascal@12273 567 }
pascal@12273 568 printf("%s.\n", msg);
pascal@12273 569 return 1;
pascal@12273 570 }
pascal@12273 571 return 0;
pascal@12273 572 }
pascal@12273 573
pascal@12287 574 /*
pascal@12287 575 * Copyright 2009 Intel Corporation; author: H. Peter Anvin
pascal@12287 576 */
pascal@12287 577
pascal@12287 578 #include <syslinux/keyboard.h>
pascal@12287 579 #include <syslinux/loadfile.h>
pascal@12287 580 #include <syslinux/adv.h>
pascal@12287 581
pascal@12349 582 static void setlinuxarg(int slot, int argc, char *argv[])
pascal@12349 583 {
pascal@12349 584 for (; argc--; argv++)
pascal@12349 585 syslinux_setadv(slot++, strlen(*argv), *argv);
pascal@12349 586 }
pascal@12349 587
pascal@18758 588 #include "../../core/unlzma.c"
pascal@12287 589 static int main_kbdmap(int argc, char *argv[])
pascal@12287 590 {
pascal@12287 591 const struct syslinux_keyboard_map *const kmap = syslinux_keyboard_map();
pascal@12287 592 size_t map_size, size, i;
pascal@20073 593 char *kbdmap, *msg, *kbdfile, *kbdname;
pascal@20073 594 int skip = 3;
pascal@12287 595
pascal@12287 596 if (argc < 3)
pascal@20073 597 usage("Usage: kbdmap [archive.cpio] [mapfile] [cmdline]..");
pascal@20073 598
pascal@20073 599 if (loadfile(kbdfile = argv[1], (void **) &kbdmap, &map_size)) {
pascal@20073 600 kbdfile = "kbd";
pascal@20073 601 skip--;
pascal@20073 602 }
pascal@20073 603 else {
pascal@20073 604 free(kbdmap);
pascal@20073 605 }
pascal@20073 606
pascal@20073 607 if (strchr(kbdname = argv[skip-1],'=')) {
pascal@20073 608 for (i = --skip; argv[i]; i++) {
pascal@20073 609 if (!strncmp(argv[i], "kmap=", 5)) {
pascal@20073 610 kbdname = argv[i] + 5;
pascal@20073 611 break;
pascal@20073 612 }
pascal@20073 613 }
pascal@20073 614 }
pascal@12287 615
pascal@12293 616 // Save extra cmdline arguments
pascal@20073 617 setlinuxarg(1, argc - skip, argv + skip);
pascal@12293 618
pascal@13318 619 msg="Append to kernel parameters: ";
pascal@20073 620 for (i = skip; i < (size_t) argc; i++, msg = " ")
pascal@13318 621 printf("%s%s",msg,argv[i]);
pascal@13318 622 printf("\n\n Hit RETURN to continue.\n");
pascal@13317 623
pascal@12287 624 msg = "Load error";
pascal@12287 625 if (kmap->version != 1 ||
pascal@20073 626 loadfile(kbdfile, (void **) &kbdmap, &map_size))
pascal@18758 627 goto kbdmap_error;
pascal@18758 628 if (* (short *) kbdmap == 0x005D) {
pascal@21820 629 void *heap = malloc(2*(1846 + (768 << (3 + 0))) + 16);
pascal@18758 630 void *p = malloc(map_size = * (long *) (kbdmap + 5));
pascal@18758 631
pascal@21820 632 if (p) {
pascal@21820 633 unlzma(kbdmap, p, heap);
pascal@21820 634 free(heap);
pascal@21820 635 free(kbdmap);
pascal@21820 636 kbdmap = p;
pascal@21820 637 }
pascal@18758 638 }
pascal@18758 639 if (strncmp(kbdmap, "07070", 5))
pascal@12287 640 goto kbdmap_error;
pascal@12287 641
pascal@12287 642 // search for mapfile in cpio archive
pascal@12287 643 for (i = 0; i < map_size;) {
pascal@12287 644 int len, j;
pascal@12287 645 char *name;
pascal@12287 646
pascal@12287 647 for (j = size = 0; j < 8; j++) {
pascal@12287 648 char c = kbdmap[54 + i + j] - '0';
pascal@12287 649 if (c > 9) c += '0' + 10 - 'A';
pascal@12287 650 size <<= 4;
pascal@12287 651 size += c;
pascal@12287 652 }
pascal@12287 653 i += 110;
pascal@12287 654 name = kbdmap + i;
pascal@12287 655 len = 1 + strlen(name);
pascal@12287 656 i += len;
pascal@12287 657 i += ((-i)&3);
pascal@20073 658 if (!strcmp(name, kbdname)) {
pascal@12287 659 kbdmap += i;
pascal@12287 660 break;
pascal@12287 661 }
pascal@12287 662 i += size + ((-size)&3);
pascal@12287 663 }
pascal@12287 664
pascal@12287 665 msg = "Filename error";
pascal@12287 666 if (i >= map_size)
pascal@12287 667 goto kbdmap_error;
pascal@12287 668
pascal@12287 669 msg = "Format error";
pascal@12287 670 if (size != kmap->length)
pascal@12287 671 goto kbdmap_error;
pascal@12287 672
pascal@12287 673 memcpy(kmap->map, kbdmap, size);
pascal@12287 674
pascal@12287 675 return 0;
pascal@12287 676
pascal@12287 677 kbdmap_error:
pascal@12287 678 printf("%s.\n",msg);
pascal@12287 679 return 1;
pascal@12287 680 }
pascal@12287 681
pascal@12287 682 /* ----------------------------------------------------------------------- *
pascal@12287 683 *
pascal@12287 684 * Copyright 2007-2008 H. Peter Anvin - All Rights Reserved
pascal@13551 685 * Copyright 2009-2012 Intel Corporation; author: H. Peter Anvin
pascal@12287 686 *
pascal@12287 687 * Permission is hereby granted, free of charge, to any person
pascal@12287 688 * obtaining a copy of this software and associated documentation
pascal@12287 689 * files (the "Software"), to deal in the Software without
pascal@12287 690 * restriction, including without limitation the rights to use,
pascal@12287 691 * copy, modify, merge, publish, distribute, sublicense, and/or
pascal@12287 692 * sell copies of the Software, and to permit persons to whom
pascal@12287 693 * the Software is furnished to do so, subject to the following
pascal@12287 694 * conditions:
pascal@12287 695 *
pascal@12287 696 * The above copyright notice and this permission notice shall
pascal@12287 697 * be included in all copies or substantial portions of the Software.
pascal@12287 698 *
pascal@12287 699 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
pascal@12287 700 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
pascal@12287 701 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
pascal@12287 702 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
pascal@12287 703 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
pascal@12287 704 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
pascal@12287 705 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
pascal@12287 706 * OTHER DEALINGS IN THE SOFTWARE.
pascal@12287 707 *
pascal@12287 708 * ----------------------------------------------------------------------- */
pascal@12287 709
pascal@12287 710 /*
pascal@12287 711 * linux.c
pascal@12287 712 *
pascal@12287 713 * Sample module to load Linux kernels. This module can also create
pascal@12287 714 * a file out of the DHCP return data if running under PXELINUX.
pascal@12287 715 *
pascal@12287 716 * If -dhcpinfo is specified, the DHCP info is written into the file
pascal@12287 717 * /dhcpinfo.dat in the initramfs.
pascal@12287 718 *
pascal@12287 719 * Usage: linux.c32 [-dhcpinfo] kernel arguments...
pascal@12287 720 */
pascal@12287 721
pascal@13551 722 #include <errno.h>
pascal@12287 723 #include <stdbool.h>
pascal@12287 724 #include <stdlib.h>
pascal@12287 725 #include <stdio.h>
pascal@12287 726 #include <string.h>
pascal@12287 727 #include <console.h>
pascal@17057 728 #include <cpuid.h>
pascal@12287 729 #include <syslinux/loadfile.h>
pascal@12287 730 #include <syslinux/linux.h>
pascal@12287 731 #include <syslinux/pxe.h>
pascal@12287 732
pascal@18804 733 const char *progname = "c32box.c32";
pascal@12287 734
pascal@12287 735 /* Find the last instance of a particular command line argument
pascal@12287 736 (which should include the final =; do not use for boolean arguments) */
pascal@12287 737 static char *find_argument(char **argv, const char *argument)
pascal@12287 738 {
pascal@12287 739 int la = strlen(argument);
pascal@12287 740 char **arg;
pascal@12287 741 char *ptr = NULL;
pascal@12287 742
pascal@12287 743 for (arg = argv; *arg; arg++) {
pascal@12287 744 if (!memcmp(*arg, argument, la))
pascal@12287 745 ptr = *arg + la;
pascal@12287 746 }
pascal@12287 747
pascal@12287 748 return ptr;
pascal@12287 749 }
pascal@12287 750
pascal@12287 751 /* Search for a boolean argument; return its position, or 0 if not present */
pascal@20073 752 #if 1
pascal@20073 753 #define find_boolean(a,b) (find_argument(a,b) != NULL)
pascal@20073 754 #else
pascal@12287 755 static int find_boolean(char **argv, const char *argument)
pascal@12287 756 {
pascal@12287 757 char **arg;
pascal@12287 758
pascal@12287 759 for (arg = argv; *arg; arg++) {
pascal@12287 760 if (!strcmp(*arg, argument))
pascal@12287 761 return (arg - argv) + 1;
pascal@12287 762 }
pascal@12287 763
pascal@12287 764 return 0;
pascal@12287 765 }
pascal@20073 766 #endif
pascal@12287 767
pascal@18674 768 static int got_config;
pascal@18674 769 static char *custom_cmdline = "";
pascal@18674 770 static int custom_initrdlen;
pascal@20062 771 static int custom_initrdbase;
pascal@18674 772 static char *custom_buffer;
pascal@18674 773 static struct disk_info diskinfo;
pascal@18674 774
pascal@18674 775 static int has_custom_config(void)
pascal@18674 776 {
pascal@18674 777 const union syslinux_derivative_info *sdi;
pascal@18674 778
pascal@18674 779 if (got_config)
pascal@18674 780 goto done;
pascal@20073 781 got_config = -1;
pascal@18674 782 sdi = syslinux_derivative_info();
pascal@18674 783 if (sdi->c.filesystem != SYSLINUX_FS_ISOLINUX)
pascal@18674 784 goto fail;
pascal@18674 785 disk_get_params(sdi->iso.drive_number, &diskinfo);
pascal@18674 786 custom_buffer = disk_read_sectors(&diskinfo, 32768 / diskinfo.bps, 1);
pascal@20073 787 got_config = (*(unsigned long *) (custom_buffer + 80))
pascal@20062 788 * 2048 / diskinfo.bps;
pascal@20062 789 free(custom_buffer);
pascal@20062 790 custom_buffer = disk_read_sectors(&diskinfo, got_config, 1);
pascal@20062 791 if (!memcmp(custom_buffer,"#!boot ",7)) {
pascal@20062 792 char *p = custom_buffer+7+32+1;
pascal@18674 793
pascal@20062 794 if (!memcmp(p,"append=",7)) {
pascal@20062 795 custom_cmdline = p + 7;
pascal@20062 796 p = strchr(p,'\n');
pascal@20062 797 *p++ = 0;
pascal@20062 798 }
pascal@20062 799 if (!memcmp(p,"initrd:",7)) {
pascal@20062 800 custom_initrdlen = strtoul(p + 7, &custom_initrdbase, 10);
pascal@20062 801 custom_initrdbase += (got_config << 11) + 1 - (int) custom_buffer;
pascal@20062 802 }
pascal@20062 803 }
pascal@18674 804 fail:
pascal@18674 805 done:
pascal@18674 806 return got_config > 0;
pascal@18674 807 }
pascal@18674 808
pascal@18674 809 static int loadcustominitrd(void **data)
pascal@18674 810 {
pascal@18674 811 int n, len;
pascal@18674 812 char *p;
pascal@18674 813
pascal@20073 814 p = *data = malloc(custom_initrdlen);
pascal@18674 815 if (!p) return 0;
pascal@20062 816 for (len = custom_initrdlen; len != 0; len -= n, p += n) {
pascal@20062 817 free(custom_buffer);
pascal@20062 818 custom_buffer = disk_read_sectors(&diskinfo, custom_initrdbase >> 11,
pascal@20062 819 2048 / diskinfo.bps);
pascal@20062 820 n = 2048 - (custom_initrdbase & 2047);
pascal@18674 821 if (n > len)
pascal@18674 822 n = len;
pascal@20062 823 memcpy(p, custom_buffer + (custom_initrdbase & 2047), n);
pascal@20062 824 custom_initrdbase += n;
pascal@18674 825 }
pascal@18674 826 return 1;
pascal@18674 827 }
pascal@18674 828
pascal@12287 829 /* Stitch together the command line from a set of argv's */
pascal@12287 830 static char *make_cmdline(char **argv)
pascal@12287 831 {
pascal@12287 832 char **arg;
pascal@13552 833 size_t bytes, size;
pascal@12287 834 char *cmdline, *p;
pascal@13552 835 int i;
pascal@12287 836
pascal@12287 837 bytes = 1; /* Just in case we have a zero-entry cmdline */
pascal@12287 838 for (arg = argv; *arg; arg++) {
pascal@12287 839 bytes += strlen(*arg) + 1;
pascal@12287 840 }
pascal@13552 841 for (i = 0; i < 255; i++)
pascal@13552 842 if (syslinux_getadv(i, &size))
pascal@13552 843 bytes += ++size;
pascal@18674 844 if (has_custom_config())
pascal@18674 845 bytes += strlen(custom_cmdline) + 1;
pascal@13552 846
pascal@12287 847 p = cmdline = malloc(bytes);
pascal@12287 848 if (!cmdline)
pascal@12287 849 return NULL;
pascal@12287 850
pascal@12287 851 for (arg = argv; *arg; arg++) {
pascal@20073 852 size = strlen(*arg);
pascal@20073 853 memcpy(p, *arg, size);
pascal@20073 854 p[size] = ' ';
pascal@20073 855 p += size + 1;
pascal@12287 856 }
pascal@12287 857
pascal@13552 858 for (i = 0; i < 255; i++) {
pascal@13552 859 const void *q = syslinux_getadv(i, &size);
pascal@13552 860 if (q == NULL) continue;
pascal@13552 861 memcpy(p, q, size);
pascal@13552 862 p[size] = ' ';
pascal@13552 863 p += size + 1;
pascal@13552 864 }
pascal@13552 865
pascal@12287 866 if (p > cmdline)
pascal@12287 867 p--; /* Remove the last space */
pascal@12287 868 *p = '\0';
pascal@18674 869
pascal@18674 870 if (has_custom_config()) {
pascal@18674 871 *p++ = ' ';
pascal@19223 872 strcpy(p, custom_cmdline);
pascal@18674 873 }
pascal@12287 874
pascal@12287 875 return cmdline;
pascal@12287 876 }
pascal@12287 877
pascal@20062 878 #define cpu_has_cpuid() cpu_has_eflag(X86_EFLAGS_ID)
pascal@17057 879
pascal@17057 880 static bool __constfunc cpu_has_level(uint32_t level)
pascal@17057 881 {
pascal@17057 882 uint32_t group;
pascal@17057 883 uint32_t limit;
pascal@17057 884
pascal@17057 885 if (!cpu_has_cpuid())
pascal@17057 886 return false;
pascal@17057 887
pascal@17057 888 group = level & 0xffff0000;
pascal@17057 889 limit = cpuid_eax(group);
pascal@17057 890
pascal@17057 891 if ((limit & 0xffff0000) != group)
pascal@17057 892 return false;
pascal@17057 893
pascal@17057 894 if (level > limit)
pascal@17057 895 return false;
pascal@17057 896
pascal@17057 897 return true;
pascal@17057 898 }
pascal@17057 899
pascal@17057 900 /* This only supports feature groups 0 and 1, corresponding to the
pascal@17057 901 Intel and AMD EDX bit vectors. We can add more later if need be. */
pascal@17057 902 static bool __constfunc cpu_has_feature(int x)
pascal@17057 903 {
pascal@17057 904 uint32_t level = ((x & 1) << 31) | 1;
pascal@17057 905
pascal@17057 906 return cpu_has_level(level) && ((cpuid_edx(level) >> (x & 31) & 1));
pascal@17057 907 }
pascal@17057 908
pascal@18675 909 static const char *extfilename(const char *filename, char *ext, int feature)
pascal@17057 910 {
pascal@17060 911 #define NEWFILENAMESZ 256
pascal@17057 912 static char newfilename[NEWFILENAMESZ+1];
pascal@18675 913 const char *found = filename;
pascal@17060 914 char *new = newfilename;
pascal@17060 915 int fd;
pascal@17057 916
pascal@17057 917 if (strlen(filename) + strlen(ext) <= NEWFILENAMESZ) {
pascal@17058 918 strcpy(newfilename, filename);
pascal@17057 919 if (cpu_has_feature(feature)) {
pascal@17057 920 strcat(newfilename, ext);
pascal@17060 921 fd = open(new, O_RDONLY);
pascal@20062 922 #ifdef WITH_UNROCKRIDGE
pascal@17060 923 if (fd < 0)
pascal@17060 924 fd = open(new = unrockridge(new), O_RDONLY);
pascal@20062 925 #endif
pascal@17060 926 if (fd >= 0) {
pascal@17060 927 found = new;
pascal@17060 928 close(fd);
pascal@17060 929 }
pascal@17057 930 }
pascal@17057 931 }
pascal@17057 932 return found;
pascal@17057 933 }
pascal@17057 934
pascal@18675 935 static const char *bestextfilename(const char *filename)
pascal@17057 936 {
pascal@18675 937 const char *found;
pascal@17057 938
pascal@17057 939 //found = extfilename(filename, "fpu", X86_FEATURE_FPU);
pascal@17057 940 //found = extfilename(filename, "686", X86_FEATURE_CMOV);
pascal@17057 941 //found = extfilename(filename, "pae", X86_FEATURE_PAE);
pascal@17057 942 found = extfilename(filename, "64", X86_FEATURE_LM);
pascal@17057 943 //found = extfilename(filename, "guest", X86_FEATURE_HYPERVISOR);
pascal@17057 944 return found;
pascal@17057 945 }
pascal@17057 946
pascal@13551 947 static int setup_data_file(struct setup_data *setup_data,
pascal@13551 948 uint32_t type, const char *filename,
pascal@13551 949 bool opt_quiet)
pascal@13551 950 {
pascal@13551 951 if (!opt_quiet)
pascal@13551 952 printf("Loading %s... ", filename);
pascal@13551 953
pascal@13551 954 if (setup_data_load(setup_data, type, filename)) {
pascal@13551 955 if (opt_quiet)
pascal@13551 956 printf("Loading %s ", filename);
pascal@13551 957 printf("failed\n");
pascal@13551 958 return -1;
pascal@13551 959 }
pascal@13551 960
pascal@13551 961 if (!opt_quiet)
pascal@13551 962 printf("ok\n");
pascal@13551 963
pascal@13551 964 return 0;
pascal@13551 965 }
pascal@13551 966
pascal@12287 967 static int main_linux(int argc, char *argv[])
pascal@12287 968 {
pascal@12287 969 const char *kernel_name;
pascal@17057 970 const char *initrd_name;
pascal@12287 971 struct initramfs *initramfs;
pascal@13551 972 struct setup_data *setup_data;
pascal@20073 973 char *errmsg;
pascal@12287 974 char *cmdline;
pascal@12287 975 char *boot_image;
pascal@12287 976 void *kernel_data;
pascal@12287 977 size_t kernel_len;
pascal@12287 978 bool opt_dhcpinfo = false;
pascal@12287 979 bool opt_quiet = false;
pascal@12287 980 void *dhcpdata;
pascal@12287 981 size_t dhcplen;
pascal@13551 982 char **argp, **argl, *arg, *p;
pascal@13551 983
pascal@13551 984 openconsole(&dev_null_r, &dev_stdcon_w);
pascal@12287 985
pascal@12287 986 (void)argc;
pascal@12287 987
pascal@20073 988 for (argp = argv + 1; (arg = *argp) && arg[0] == '-'; argp++) {
pascal@12287 989 if (!strcmp("-dhcpinfo", arg)) {
pascal@12287 990 opt_dhcpinfo = true;
pascal@12287 991 } else {
pascal@20073 992 errmsg = "%s: unknown option: %s\n";
pascal@20073 993 goto unknown_option;
pascal@12287 994 }
pascal@12287 995 }
pascal@12287 996
pascal@12287 997 if (!arg) {
pascal@20073 998 errmsg = "%s: missing kernel name\n";
pascal@20073 999 unknown_option:
pascal@20073 1000 fprintf(stderr, errmsg, progname, arg);
pascal@12287 1001 return 1;
pascal@12287 1002 }
pascal@12287 1003
pascal@17061 1004 kernel_name = arg;
pascal@12287 1005
pascal@13551 1006 errno = 0;
pascal@12287 1007 boot_image = malloc(strlen(kernel_name) + 12);
pascal@13551 1008 if (!boot_image) {
pascal@20073 1009 errmsg = "Error allocating BOOT_IMAGE string: ";
pascal@20073 1010 goto bailmsg;
pascal@13551 1011 }
pascal@12287 1012 strcpy(boot_image, "BOOT_IMAGE=");
pascal@12287 1013 strcpy(boot_image + 11, kernel_name);
pascal@12287 1014 /* argp now points to the kernel name, and the command line follows.
pascal@12287 1015 Overwrite the kernel name with the BOOT_IMAGE= argument, and thus
pascal@12287 1016 we have the final argument. */
pascal@12287 1017 *argp = boot_image;
pascal@12287 1018
pascal@12287 1019 if (find_boolean(argp, "quiet"))
pascal@12287 1020 opt_quiet = true;
pascal@12287 1021
pascal@12287 1022 if (!opt_quiet)
pascal@12287 1023 printf("Loading %s... ", kernel_name);
pascal@13551 1024 errno = 0;
pascal@17061 1025 if (loadfile(bestextfilename(kernel_name), &kernel_data, &kernel_len)) {
pascal@12287 1026 if (opt_quiet)
pascal@12287 1027 printf("Loading %s ", kernel_name);
pascal@13551 1028 printf("failed: ");
pascal@12287 1029 goto bail;
pascal@12287 1030 }
pascal@12287 1031 if (!opt_quiet)
pascal@12287 1032 printf("ok\n");
pascal@12287 1033
pascal@13551 1034 errno = 0;
pascal@12287 1035 cmdline = make_cmdline(argp);
pascal@13551 1036 if (!cmdline) {
pascal@20073 1037 errmsg = "make_cmdline() failed: ";
pascal@20073 1038 goto bailmsg;
pascal@13551 1039 }
pascal@12287 1040
pascal@12287 1041 /* Initialize the initramfs chain */
pascal@13551 1042 errno = 0;
pascal@12287 1043 initramfs = initramfs_init();
pascal@13551 1044 if (!initramfs) {
pascal@20073 1045 errmsg = "initramfs_init() failed: ";
pascal@20073 1046 goto bailmsg;
pascal@13551 1047 }
pascal@12287 1048
pascal@12287 1049 if ((arg = find_argument(argp, "initrd="))) {
pascal@20073 1050 while (1) {
pascal@12287 1051 p = strchr(arg, ',');
pascal@12287 1052 if (p)
pascal@12287 1053 *p = '\0';
pascal@12287 1054
pascal@17061 1055 initrd_name = arg;
pascal@12287 1056 if (!opt_quiet)
pascal@17057 1057 printf("Loading %s... ", initrd_name);
pascal@13551 1058 errno = 0;
pascal@17061 1059 if (initramfs_load_archive(initramfs, bestextfilename(initrd_name))) {
pascal@12287 1060 if (opt_quiet)
pascal@17057 1061 printf("Loading %s ", initrd_name);
pascal@13551 1062 printf("failed: ");
pascal@12287 1063 goto bail;
pascal@12287 1064 }
pascal@12287 1065 if (!opt_quiet)
pascal@12287 1066 printf("ok\n");
pascal@12287 1067
pascal@20073 1068 if (!p)
pascal@20073 1069 break;
pascal@20073 1070
pascal@20073 1071 *p++ = ',';
pascal@20073 1072 arg = p;
pascal@20073 1073 }
pascal@12287 1074 }
pascal@12287 1075
pascal@12287 1076 /* Append the DHCP info */
pascal@12287 1077 if (opt_dhcpinfo &&
pascal@12287 1078 !pxe_get_cached_info(PXENV_PACKET_TYPE_DHCP_ACK, &dhcpdata, &dhcplen)) {
pascal@13551 1079 errno = 0;
pascal@12287 1080 if (initramfs_add_file(initramfs, dhcpdata, dhcplen, dhcplen,
pascal@13551 1081 "/dhcpinfo.dat", 0, 0755)) {
pascal@20073 1082 errmsg = "Unable to add DHCP info: ";
pascal@20073 1083 goto bailmsg;
pascal@13551 1084 }
pascal@13551 1085 }
pascal@13551 1086
pascal@18674 1087 if (has_custom_config() && custom_initrdlen) {
pascal@18674 1088 void *data;
pascal@18674 1089
pascal@18674 1090 if (!opt_quiet)
pascal@18674 1091 printf("Loading custom initrd... ");
pascal@18674 1092 if (loadcustominitrd(&data))
pascal@18674 1093 initramfs_add_data(initramfs, data, custom_initrdlen, custom_initrdlen, 4);
pascal@18674 1094 }
pascal@18674 1095
pascal@13551 1096 /* Handle dtb and eventually other setup data */
pascal@13551 1097 setup_data = setup_data_init();
pascal@13551 1098 if (!setup_data)
pascal@13551 1099 goto bail;
pascal@13551 1100
pascal@13551 1101 for (argl = argv; (arg = *argl); argl++) {
pascal@13551 1102 if (!memcmp(arg, "dtb=", 4)) {
pascal@13551 1103 if (setup_data_file(setup_data, SETUP_DTB, arg+4, opt_quiet))
pascal@13551 1104 goto bail;
pascal@13551 1105 } else if (!memcmp(arg, "blob.", 5)) {
pascal@13551 1106 uint32_t type;
pascal@13551 1107 char *ep;
pascal@13551 1108
pascal@13551 1109 type = strtoul(arg + 5, &ep, 10);
pascal@20073 1110 if (!type || ep[0] != '=' || !ep[1])
pascal@13551 1111 continue;
pascal@13551 1112
pascal@13551 1113 if (setup_data_file(setup_data, type, ep+1, opt_quiet))
pascal@13551 1114 goto bail;
pascal@13551 1115 }
pascal@12287 1116 }
pascal@12287 1117
pascal@12287 1118 /* This should not return... */
pascal@13551 1119 errno = 0;
pascal@13551 1120 syslinux_boot_linux(kernel_data, kernel_len, initramfs,
pascal@13551 1121 setup_data, cmdline);
pascal@20073 1122 errmsg = "syslinux_boot_linux() failed: ";
pascal@12287 1123
pascal@20073 1124 bailmsg:
pascal@20073 1125 fprintf(stderr, errmsg);
pascal@12287 1126 bail:
pascal@20073 1127 errmsg = "Error %d";
pascal@13551 1128 switch(errno) {
pascal@13551 1129 case ENOENT:
pascal@20073 1130 errmsg = "File not found";
pascal@13551 1131 break;
pascal@13551 1132 case ENOMEM:
pascal@20073 1133 errmsg = "Out of memory";
pascal@13551 1134 }
pascal@20073 1135 fprintf(stderr, errmsg, errno);
pascal@20073 1136 fprintf(stderr, "\n%luM RAM and %s bit cpu found.\n%s: Boot aborted!\n",
pascal@18855 1137 memory_size() >> 10,
pascal@20073 1138 cpu_has_feature(X86_FEATURE_LM) ? "64": "32",
pascal@20073 1139 progname);
pascal@12287 1140 return 1;
pascal@12287 1141 }
pascal@12287 1142
pascal@20062 1143 #ifdef EXTRA
pascal@12349 1144 static int main_setarg(int argc, char *argv[])
pascal@12349 1145 {
pascal@12349 1146 if (argc < 3) {
pascal@12349 1147 usage("Usage: setarg.c32 argnum [args]...");
pascal@12349 1148 }
pascal@12349 1149 setlinuxarg(atoi(argv[1]), argc - 2, argv + 2);
pascal@12349 1150 return 0;
pascal@12349 1151 }
pascal@12349 1152
pascal@12349 1153 static int main_ifarg(int argc, char *argv[])
pascal@12349 1154 {
pascal@12349 1155 int i;
pascal@12349 1156 size_t size;
pascal@12349 1157
pascal@12349 1158 if (argc < 3) {
pascal@12349 1159 usage("Usage: ifarg.c32 [argnum labelifset]... labelifnoneset");
pascal@12349 1160 }
pascal@12349 1161 for (i = 1; i < argc - 1; i += 2) {
pascal@13295 1162 int n = atoi(argv[i]);
pascal@13295 1163 if (n == -1) {
pascal@13295 1164 for (n = 0; n < 255; n++) {
pascal@13316 1165 if (syslinux_getadv(n, &size))
pascal@13295 1166 goto found;
pascal@13295 1167 }
pascal@13295 1168 continue;
pascal@13295 1169 }
pascal@13295 1170 else if (! syslinux_getadv(n, &size)) continue;
pascal@13295 1171 found:
pascal@13295 1172 syslinux_run_command(argv[i+1]);
pascal@12349 1173 }
pascal@12349 1174 if (i != argc) syslinux_run_command(argv[i]);
pascal@12349 1175 else syslinux_run_default();
pascal@12349 1176 return 0;
pascal@12349 1177 }
pascal@12349 1178
pascal@12349 1179 /* ----------------------------------------------------------------------- *
pascal@12349 1180 *
pascal@12349 1181 * Copyright 2007-2008 H. Peter Anvin - All Rights Reserved
pascal@12349 1182 *
pascal@12349 1183 * This program is free software; you can redistribute it and/or modify
pascal@12349 1184 * it under the terms of the GNU General Public License as published by
pascal@12349 1185 * the Free Software Foundation, Inc., 53 Temple Place Ste 330,
pascal@12349 1186 * Boston MA 02111-1307, USA; either version 2 of the License, or
pascal@12349 1187 * (at your option) any later version; incorporated herein by reference.
pascal@12349 1188 *
pascal@12349 1189 * ----------------------------------------------------------------------- */
pascal@12349 1190
pascal@12349 1191 static int main_listarg(int argc, char *argv[])
pascal@12349 1192 {
pascal@12349 1193 uint8_t *p, *ep;
pascal@12349 1194 size_t s = syslinux_adv_size();
pascal@12349 1195 char buf[256];
pascal@12349 1196
pascal@12349 1197 (void) argc;
pascal@12349 1198 (void) argv;
pascal@12349 1199 p = syslinux_adv_ptr();
pascal@12349 1200
pascal@12349 1201 printf("args size: %zd bytes at %p\n", s, p);
pascal@12349 1202
pascal@12349 1203 ep = p + s; /* Need at least opcode+len */
pascal@12349 1204 while (p < ep - 1 && *p) {
pascal@12349 1205 int t = *p++;
pascal@12349 1206 int l = *p++;
pascal@12349 1207
pascal@12349 1208 if (p + l > ep)
pascal@12349 1209 break;
pascal@12349 1210
pascal@12349 1211 memcpy(buf, p, l);
pascal@12349 1212 buf[l] = '\0';
pascal@12349 1213
pascal@12349 1214 printf("arg %3d: \"%s\"\n", t, buf);
pascal@12349 1215
pascal@12349 1216 p += l;
pascal@12349 1217 }
pascal@12349 1218 sleep(5);
pascal@12349 1219 return 0;
pascal@12349 1220 }
pascal@20062 1221 #endif
pascal@12349 1222
pascal@12273 1223 int main(int argc, char *argv[])
pascal@12273 1224 {
pascal@12273 1225 unsigned i;
pascal@12273 1226 static struct {
pascal@12273 1227 char *name;
pascal@12273 1228 int (*main)(int argc, char *argv[]);
pascal@12273 1229 } bin[] = {
pascal@20062 1230 #ifdef EXTRA
pascal@13295 1231 { "say", main_say },
pascal@13295 1232 { "setarg", main_setarg },
pascal@13295 1233 { "ifarg", main_ifarg },
pascal@20062 1234 { "listarg", main_listarg },
pascal@20062 1235 #endif
pascal@20062 1236 { "kbdmap", main_kbdmap },
pascal@20073 1237 { "kbd", main_kbdmap },
pascal@20062 1238 { "ifmem", main_ifmem },
pascal@20062 1239 { "linux", main_linux },
pascal@20062 1240 { "md5sum", main_md5sum },
pascal@20062 1241 { "reboot", main_reboot },
pascal@20062 1242 { "poweroff", main_poweroff }
pascal@12273 1243 };
pascal@12273 1244
pascal@12273 1245 openconsole(&dev_null_r, &dev_stdcon_w);
pascal@12273 1246
pascal@12273 1247 if (strstr(argv[0], "c32box")) { argc--; argv++; }
pascal@12273 1248 for (i = 0; i < sizeof(bin)/sizeof(bin[0]); i++)
pascal@12273 1249 if (strstr(argv[0], bin[i].name))
pascal@12273 1250 return bin[i].main(argc, argv);
pascal@12287 1251 printf("No %s in c32box modules\n", argv[0]);
pascal@12287 1252 for (i = 0; i < sizeof(bin)/sizeof(bin[0]); i++)
pascal@12287 1253 printf(" %s \n",bin[i].name);
pascal@12273 1254 return 1;
pascal@12273 1255 }