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

octave: update deps
author Pascal Bellard <pascal.bellard@slitaz.org>
date Fri Aug 31 08:57:15 2012 +0200 (2012-08-31)
parents 20cd0d6b9f64
children 60c00c202374
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@12210 26
pascal@12210 27 #define ALIGN1
pascal@12215 28
pascal@12215 29 static uint8_t wbuffer[64]; /* always correctly aligned for uint64_t */
pascal@12215 30 static uint64_t total64; /* must be directly before hash[] */
pascal@12215 31 static uint32_t hash[8]; /* 4 elements for md5, 5 for sha1, 8 for sha256 */
pascal@12210 32
pascal@12210 33 /* Emit a string of hex representation of bytes */
pascal@12215 34 static char* bin2hex(char *p)
pascal@12210 35 {
pascal@12215 36 static const char bb_hexdigits_upcase[] ALIGN1 = "0123456789abcdef";
pascal@12215 37 int count = 16;
pascal@12215 38 const char *cp = (const char *) hash;
pascal@12210 39 while (count) {
pascal@12210 40 unsigned char c = *cp++;
pascal@12210 41 /* put lowercase hex digits */
pascal@12215 42 *p++ = bb_hexdigits_upcase[c >> 4];
pascal@12215 43 *p++ = bb_hexdigits_upcase[c & 0xf];
pascal@12210 44 count--;
pascal@12210 45 }
pascal@12210 46 return p;
pascal@12210 47 }
pascal@12210 48
pascal@12210 49 //#define rotl32(x,n) (((x) << (n)) | ((x) >> (32 - (n))))
pascal@12210 50 static uint32_t rotl32(uint32_t x, unsigned n)
pascal@12210 51 {
pascal@12210 52 return (x << n) | (x >> (32 - n));
pascal@12210 53 }
pascal@12210 54
pascal@12215 55 static void md5_process_block64(void);
pascal@12210 56
pascal@12210 57 /* Feed data through a temporary buffer.
pascal@12210 58 * The internal buffer remembers previous data until it has 64
pascal@12210 59 * bytes worth to pass on.
pascal@12210 60 */
pascal@12215 61 static void common64_hash(const void *buffer, size_t len)
pascal@12210 62 {
pascal@12215 63 unsigned bufpos = total64 & 63;
pascal@12210 64
pascal@12215 65 total64 += len;
pascal@12210 66
pascal@12210 67 while (1) {
pascal@12210 68 unsigned remaining = 64 - bufpos;
pascal@12210 69 if (remaining > len)
pascal@12210 70 remaining = len;
pascal@12210 71 /* Copy data into aligned buffer */
pascal@12215 72 memcpy(wbuffer + bufpos, buffer, remaining);
pascal@12210 73 len -= remaining;
pascal@12210 74 buffer = (const char *)buffer + remaining;
pascal@12210 75 bufpos += remaining;
pascal@12210 76 /* clever way to do "if (bufpos != 64) break; ... ; bufpos = 0;" */
pascal@12210 77 bufpos -= 64;
pascal@12210 78 if (bufpos != 0)
pascal@12210 79 break;
pascal@12210 80 /* Buffer is filled up, process it */
pascal@12215 81 md5_process_block64();
pascal@12210 82 /*bufpos = 0; - already is */
pascal@12210 83 }
pascal@12210 84 }
pascal@12210 85
pascal@12210 86 /* Process the remaining bytes in the buffer */
pascal@12215 87 static void common64_end(void)
pascal@12210 88 {
pascal@12215 89 unsigned bufpos = total64 & 63;
pascal@12210 90 /* Pad the buffer to the next 64-byte boundary with 0x80,0,0,0... */
pascal@12215 91 wbuffer[bufpos++] = 0x80;
pascal@12210 92
pascal@12210 93 /* This loop iterates either once or twice, no more, no less */
pascal@12210 94 while (1) {
pascal@12210 95 unsigned remaining = 64 - bufpos;
pascal@12215 96 memset(wbuffer + bufpos, 0, remaining);
pascal@12210 97 /* Do we have enough space for the length count? */
pascal@12210 98 if (remaining >= 8) {
pascal@12210 99 /* Store the 64-bit counter of bits in the buffer */
pascal@12215 100 uint64_t t = total64 << 3;
pascal@12210 101 /* wbuffer is suitably aligned for this */
pascal@12215 102 *(uint64_t *) (&wbuffer[64 - 8]) = t;
pascal@12210 103 }
pascal@12215 104 md5_process_block64();
pascal@12210 105 if (remaining >= 8)
pascal@12210 106 break;
pascal@12210 107 bufpos = 0;
pascal@12210 108 }
pascal@12210 109 }
pascal@12210 110
pascal@12210 111 /* These are the four functions used in the four steps of the MD5 algorithm
pascal@12210 112 * and defined in the RFC 1321. The first function is a little bit optimized
pascal@12210 113 * (as found in Colin Plumbs public domain implementation).
pascal@12210 114 * #define FF(b, c, d) ((b & c) | (~b & d))
pascal@12210 115 */
pascal@12210 116 #undef FF
pascal@12210 117 #undef FG
pascal@12210 118 #undef FH
pascal@12210 119 #undef FI
pascal@12210 120 #define FF(b, c, d) (d ^ (b & (c ^ d)))
pascal@12210 121 #define FG(b, c, d) FF(d, b, c)
pascal@12210 122 #define FH(b, c, d) (b ^ c ^ d)
pascal@12210 123 #define FI(b, c, d) (c ^ (b | ~d))
pascal@12210 124
pascal@12210 125 /* Hash a single block, 64 bytes long and 4-byte aligned */
pascal@12215 126 static void md5_process_block64(void)
pascal@12210 127 {
pascal@12210 128 /* Before we start, one word to the strange constants.
pascal@12210 129 They are defined in RFC 1321 as
pascal@12210 130 T[i] = (int)(4294967296.0 * fabs(sin(i))), i=1..64
pascal@12210 131 */
pascal@12210 132 static const uint32_t C_array[] = {
pascal@12210 133 /* round 1 */
pascal@12210 134 0xd76aa478, 0xe8c7b756, 0x242070db, 0xc1bdceee,
pascal@12210 135 0xf57c0faf, 0x4787c62a, 0xa8304613, 0xfd469501,
pascal@12210 136 0x698098d8, 0x8b44f7af, 0xffff5bb1, 0x895cd7be,
pascal@12210 137 0x6b901122, 0xfd987193, 0xa679438e, 0x49b40821,
pascal@12210 138 /* round 2 */
pascal@12210 139 0xf61e2562, 0xc040b340, 0x265e5a51, 0xe9b6c7aa,
pascal@12210 140 0xd62f105d, 0x02441453, 0xd8a1e681, 0xe7d3fbc8,
pascal@12210 141 0x21e1cde6, 0xc33707d6, 0xf4d50d87, 0x455a14ed,
pascal@12210 142 0xa9e3e905, 0xfcefa3f8, 0x676f02d9, 0x8d2a4c8a,
pascal@12210 143 /* round 3 */
pascal@12210 144 0xfffa3942, 0x8771f681, 0x6d9d6122, 0xfde5380c,
pascal@12210 145 0xa4beea44, 0x4bdecfa9, 0xf6bb4b60, 0xbebfbc70,
pascal@12210 146 0x289b7ec6, 0xeaa127fa, 0xd4ef3085, 0x4881d05,
pascal@12210 147 0xd9d4d039, 0xe6db99e5, 0x1fa27cf8, 0xc4ac5665,
pascal@12210 148 /* round 4 */
pascal@12210 149 0xf4292244, 0x432aff97, 0xab9423a7, 0xfc93a039,
pascal@12210 150 0x655b59c3, 0x8f0ccc92, 0xffeff47d, 0x85845dd1,
pascal@12210 151 0x6fa87e4f, 0xfe2ce6e0, 0xa3014314, 0x4e0811a1,
pascal@12210 152 0xf7537e82, 0xbd3af235, 0x2ad7d2bb, 0xeb86d391
pascal@12210 153 };
pascal@12210 154 static const char P_array[] ALIGN1 = {
pascal@12210 155 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, /* 1 */
pascal@12210 156 1, 6, 11, 0, 5, 10, 15, 4, 9, 14, 3, 8, 13, 2, 7, 12, /* 2 */
pascal@12210 157 5, 8, 11, 14, 1, 4, 7, 10, 13, 0, 3, 6, 9, 12, 15, 2, /* 3 */
pascal@12210 158 0, 7, 14, 5, 12, 3, 10, 1, 8, 15, 6, 13, 4, 11, 2, 9 /* 4 */
pascal@12210 159 };
pascal@12215 160 uint32_t *words = (void*) wbuffer;
pascal@12215 161 uint32_t A = hash[0];
pascal@12215 162 uint32_t B = hash[1];
pascal@12215 163 uint32_t C = hash[2];
pascal@12215 164 uint32_t D = hash[3];
pascal@12210 165
pascal@12210 166 static const char S_array[] ALIGN1 = {
pascal@12210 167 7, 12, 17, 22,
pascal@12210 168 5, 9, 14, 20,
pascal@12210 169 4, 11, 16, 23,
pascal@12210 170 6, 10, 15, 21
pascal@12210 171 };
pascal@12210 172 const uint32_t *pc;
pascal@12210 173 const char *pp;
pascal@12210 174 const char *ps;
pascal@12210 175 int i;
pascal@12210 176 uint32_t temp;
pascal@12210 177
pascal@12210 178
pascal@12210 179 pc = C_array;
pascal@12210 180 pp = P_array;
pascal@12210 181 ps = S_array - 4;
pascal@12210 182
pascal@12210 183 for (i = 0; i < 64; i++) {
pascal@12210 184 if ((i & 0x0f) == 0)
pascal@12210 185 ps += 4;
pascal@12210 186 temp = A;
pascal@12210 187 switch (i >> 4) {
pascal@12210 188 case 0:
pascal@12210 189 temp += FF(B, C, D);
pascal@12210 190 break;
pascal@12210 191 case 1:
pascal@12210 192 temp += FG(B, C, D);
pascal@12210 193 break;
pascal@12210 194 case 2:
pascal@12210 195 temp += FH(B, C, D);
pascal@12210 196 break;
pascal@12210 197 case 3:
pascal@12210 198 temp += FI(B, C, D);
pascal@12210 199 }
pascal@12210 200 temp += words[(int) (*pp++)] + *pc++;
pascal@12210 201 temp = rotl32(temp, ps[i & 3]);
pascal@12210 202 temp += B;
pascal@12210 203 A = D;
pascal@12210 204 D = C;
pascal@12210 205 C = B;
pascal@12210 206 B = temp;
pascal@12210 207 }
pascal@12210 208 /* Add checksum to the starting values */
pascal@12215 209 hash[0] += A;
pascal@12215 210 hash[1] += B;
pascal@12215 211 hash[2] += C;
pascal@12215 212 hash[3] += D;
pascal@12210 213
pascal@12210 214 }
pascal@12210 215 #undef FF
pascal@12210 216 #undef FG
pascal@12210 217 #undef FH
pascal@12210 218 #undef FI
pascal@12210 219
pascal@12210 220 /* Initialize structure containing state of computation.
pascal@12210 221 * (RFC 1321, 3.3: Step 3)
pascal@12210 222 */
pascal@12215 223 static void md5_begin(void)
pascal@12210 224 {
pascal@12215 225 hash[0] = 0x67452301;
pascal@12215 226 hash[1] = 0xefcdab89;
pascal@12215 227 hash[2] = 0x98badcfe;
pascal@12215 228 hash[3] = 0x10325476;
pascal@12215 229 total64 = 0;
pascal@12210 230 }
pascal@12210 231
pascal@12210 232 /* Used also for sha1 and sha256 */
pascal@12215 233 #define md5_hash common64_hash
pascal@12210 234
pascal@12210 235 /* Process the remaining bytes in the buffer and put result from CTX
pascal@12210 236 * in first 16 bytes following RESBUF. The result is always in little
pascal@12210 237 * endian byte order, so that a byte-wise output yields to the wanted
pascal@12210 238 * ASCII representation of the message digest.
pascal@12210 239 */
pascal@12215 240 #define md5_end common64_end
pascal@12210 241
pascal@12210 242 /*
pascal@12210 243 * Copyright (C) 2003 Glenn L. McGrath
pascal@12210 244 * Copyright (C) 2003-2004 Erik Andersen
pascal@12210 245 *
pascal@12210 246 * Licensed under GPLv2 or later, see file LICENSE in this source tree.
pascal@12210 247 */
pascal@12210 248
pascal@12215 249 static char *unrockridge(const char *name)
pascal@12212 250 {
pascal@12212 251 static char buffer[256];
pascal@12215 252 int i = 0, j = 8;
pascal@12215 253 while (*name && i < 255) {
pascal@12215 254 char c = *name++;
pascal@12215 255 //if (c == '\\') c = '/';
pascal@12215 256 if (c == '.') {
pascal@12215 257 for (j = i; --j >= 0 && buffer[j] != '/';)
pascal@12215 258 if (buffer[j] == '.') buffer[j] = '_';
pascal@12215 259 if (i - j > 9) i = j + 9;
pascal@12215 260 j = i + 4;
pascal@12212 261 }
pascal@12215 262 else if (c == '/') j = i + 9;
pascal@12212 263 else if (i >= j) continue;
pascal@12215 264 else if (c >= 'a' && c <= 'z') c += 'A' - 'a';
pascal@12215 265 else if ((c < 'A' || c > 'Z') && (c < '0' || c > '9')) c = '_';
pascal@12215 266 buffer[i++] = c;
pascal@12212 267 }
pascal@12215 268 buffer[i] = 0;
pascal@12212 269 return buffer;
pascal@12212 270 }
pascal@12212 271
pascal@12210 272 static uint8_t *hash_file(const char *filename)
pascal@12210 273 {
pascal@12210 274 int src_fd, count;
pascal@12215 275 uint8_t in_buf[4096];
pascal@12215 276 static uint8_t hash_value[16*2+1];
pascal@12210 277
pascal@12210 278 src_fd = open(filename, O_RDONLY);
pascal@12210 279 if (src_fd < 0) {
pascal@12215 280 src_fd = open(unrockridge(filename), O_RDONLY);
pascal@12212 281 }
pascal@12212 282 if (src_fd < 0) {
pascal@12210 283 return NULL;
pascal@12210 284 }
pascal@12210 285
pascal@12215 286 md5_begin();
pascal@12210 287 while ((count = read(src_fd, in_buf, 4096)) > 0) {
pascal@12215 288 md5_hash(in_buf, count);
pascal@12210 289 }
pascal@12210 290
pascal@12210 291 close(src_fd);
pascal@12215 292
pascal@12215 293 if (count)
pascal@12215 294 return NULL;
pascal@12215 295
pascal@12215 296 md5_end();
pascal@12215 297 bin2hex((char *)hash_value);
pascal@12210 298
pascal@12210 299 return hash_value;
pascal@12210 300 }
pascal@12210 301
pascal@13295 302 static int main_say(int argc, char **argv)
pascal@13295 303 {
pascal@13295 304 int i;
pascal@13295 305 for (i = 1; i < argc; i++) {
pascal@13295 306 printf("%s ",argv[i]);
pascal@13295 307 }
pascal@13295 308 sleep(5);
pascal@13295 309 }
pascal@13295 310
pascal@12273 311 static int main_md5sum(int argc, char **argv)
pascal@12210 312 {
pascal@12215 313 int files = 0, tested = 0, good = 0;
pascal@12215 314 static char clear_eol[] = " ";
pascal@12210 315
pascal@12210 316 (void) argc;
pascal@12210 317 /* -c implied */
pascal@12287 318 argv++;
pascal@12210 319 do {
pascal@12210 320 FILE *fp;
pascal@12215 321 char eol, *line, buffer[256];
pascal@12210 322 fp = fopen(*argv,"r");
pascal@12212 323 if (fp == NULL)
pascal@12215 324 fp = fopen(unrockridge(*argv),"r");
pascal@12210 325
pascal@12210 326 while ((line = fgets(buffer,256,fp)) != NULL) {
pascal@12210 327 uint8_t *hash_value;
pascal@12210 328 char *filename_ptr, *status;
pascal@12210 329 int len = strlen(line);
pascal@12210 330
pascal@12210 331 if (line[0] < '0')
pascal@12210 332 continue;
pascal@12210 333 if (line[len-1] == '\n')
pascal@12210 334 line[len-1] = 0;
pascal@12210 335 filename_ptr = strstr(line, " ");
pascal@12210 336 /* handle format for binary checksums */
pascal@12210 337 if (filename_ptr == NULL) {
pascal@12210 338 filename_ptr = strstr(line, " *");
pascal@12210 339 }
pascal@12210 340 if (filename_ptr == NULL) {
pascal@12210 341 continue;
pascal@12210 342 }
pascal@12210 343 *filename_ptr = '\0';
pascal@12210 344 *++filename_ptr = '/';
pascal@12212 345 if (filename_ptr[1] == '/')
pascal@12212 346 filename_ptr++;
pascal@12210 347
pascal@12215 348 files++;
pascal@12215 349 status = "NOT CHECKED";
pascal@12215 350 eol = '\n';
pascal@12211 351 hash_value = hash_file(filename_ptr);
pascal@12211 352 if (hash_value) {
pascal@12215 353 tested++;
pascal@12215 354 status = "BROKEN";
pascal@12215 355 if (!strcmp((char*)hash_value, line)) {
pascal@12215 356 good++;
pascal@12215 357 status = "OK";
pascal@12215 358 eol = ' ';
pascal@12210 359 }
pascal@12210 360 }
pascal@12215 361 printf("\r%s: %s%s%c", filename_ptr, status, clear_eol, eol);
pascal@12210 362 }
pascal@12210 363 fclose(fp);
pascal@12210 364 } while (*++argv);
pascal@12215 365 printf("\r%d files OK, %d broken, %d not checked.%s\n",
pascal@12215 366 good, tested - good, files - tested, clear_eol);
pascal@12287 367 sleep(5);
pascal@12273 368 return 0;
pascal@12215 369 }
pascal@12210 370
pascal@12273 371 /*
pascal@12273 372 * ifmem.c
pascal@12273 373 *
pascal@12273 374 * Run one command if the memory is large enought, and another if it isn't.
pascal@12273 375 *
pascal@12273 376 * Usage:
pascal@12273 377 *
pascal@12273 378 * label boot_kernel
pascal@12273 379 * kernel ifmem.c
pascal@12273 380 * append size_in_KB boot_large [size_in_KB boot_medium] boot_small
pascal@12273 381 *
pascal@12273 382 * label boot_large
pascal@12273 383 * kernel vmlinuz_large_memory
pascal@12273 384 * append ...
pascal@12273 385 *
pascal@12273 386 * label boot_small
pascal@12273 387 * kernel vmlinuz_small_memory
pascal@12273 388 * append ...
pascal@12273 389 */
pascal@12273 390
pascal@12273 391 #include <inttypes.h>
pascal@12273 392 #include <alloca.h>
pascal@12273 393 #include <syslinux/boot.h>
pascal@12273 394
pascal@12273 395 struct e820_data {
pascal@12273 396 uint64_t base;
pascal@12273 397 uint64_t len;
pascal@12273 398 uint32_t type;
pascal@12273 399 uint32_t extattr;
pascal@12273 400 } __attribute__((packed));
pascal@12273 401
pascal@12273 402 // Get memory size in Kb
pascal@12273 403 static unsigned long memory_size(void)
pascal@12273 404 {
pascal@12273 405 uint64_t bytes = 0;
pascal@12273 406 static com32sys_t ireg, oreg;
pascal@12273 407 static struct e820_data ed;
pascal@12273 408
pascal@12273 409 ireg.eax.w[0] = 0xe820;
pascal@12273 410 ireg.edx.l = 0x534d4150;
pascal@12273 411 ireg.ecx.l = sizeof(struct e820_data);
pascal@12273 412 ireg.edi.w[0] = OFFS(__com32.cs_bounce);
pascal@12273 413 ireg.es = SEG(__com32.cs_bounce);
pascal@12273 414
pascal@12273 415 ed.extattr = 1;
pascal@12273 416
pascal@12273 417 do {
pascal@12273 418 memcpy(__com32.cs_bounce, &ed, sizeof ed);
pascal@12273 419
pascal@12273 420 __intcall(0x15, &ireg, &oreg);
pascal@12273 421 if (oreg.eflags.l & EFLAGS_CF ||
pascal@12273 422 oreg.eax.l != 0x534d4150 ||
pascal@12273 423 oreg.ecx.l < 20)
pascal@12273 424 break;
pascal@12273 425
pascal@12273 426 memcpy(&ed, __com32.cs_bounce, sizeof ed);
pascal@12273 427
pascal@12273 428 if (ed.type == 1)
pascal@12273 429 bytes += ed.len;
pascal@12273 430
pascal@12273 431 ireg.ebx.l = oreg.ebx.l;
pascal@12273 432 } while (ireg.ebx.l);
pascal@12273 433
pascal@12273 434 if (!bytes) {
pascal@12273 435 memset(&ireg, 0, sizeof ireg);
pascal@12273 436 ireg.eax.w[0] = 0x8800;
pascal@12273 437 __intcall(0x15, &ireg, &oreg);
pascal@12273 438 return ireg.eax.w[0];
pascal@12273 439 }
pascal@12273 440 return bytes >> 10;
pascal@12273 441 }
pascal@12273 442
pascal@12349 443 static void usage(const char *msg)
pascal@12349 444 {
pascal@12349 445 fprintf(stderr,"\n%s\n.",msg);
pascal@12349 446 sleep(5);
pascal@12349 447 exit(1);
pascal@12349 448 }
pascal@12349 449
pascal@12273 450 static int main_ifmem(int argc, char *argv[])
pascal@12273 451 {
pascal@12273 452 int i;
pascal@12273 453 unsigned long ram_size;
pascal@12273 454
pascal@12273 455 if (argc < 4) {
pascal@12349 456 usage("Usage: ifmem.c32 size_KB boot_large_memory boot_small_memory");
pascal@12273 457 }
pascal@12273 458
pascal@12273 459 // find target according to ram size
pascal@12273 460 ram_size = memory_size();
pascal@12273 461 printf("Total memory found %luK.\n",ram_size);
pascal@12273 462 ram_size += (1 << 10); // add 1M to round boundaries...
pascal@12273 463
pascal@12273 464 i = 1;
pascal@12273 465 do {
pascal@12273 466 char *s = argv[i];
pascal@12273 467 char *p = s;
pascal@12273 468 unsigned long scale = 1;
pascal@12273 469
pascal@12273 470 while (*p >= '0' && *p <= '9') p++;
pascal@12273 471 switch (*p | 0x20) {
pascal@12273 472 case 'g': scale <<= 10;
pascal@12273 473 case 'm': scale <<= 10;
pascal@12273 474 default : *p = 0; break;
pascal@12273 475 }
pascal@12273 476 i++; // seek to label
pascal@12273 477 if (ram_size >= scale * strtoul(s, NULL, 0)) break;
pascal@12273 478 i++; // next size or default label
pascal@12273 479 } while (i + 1 < argc);
pascal@12273 480
pascal@12349 481 if (i != argc) syslinux_run_command(argv[i]);
pascal@12273 482 else syslinux_run_default();
pascal@12273 483 return -1;
pascal@12273 484 }
pascal@12273 485
pascal@12273 486 #include <syslinux/reboot.h>
pascal@12273 487
pascal@12273 488 static int main_reboot(int argc, char *argv[])
pascal@12273 489 {
pascal@12273 490 int warm = 0;
pascal@12273 491 int i;
pascal@12273 492
pascal@12273 493 for (i = 1; i < argc; i++) {
pascal@12273 494 if (!strcmp(argv[i], "-w") || !strcmp(argv[i], "--warm"))
pascal@12273 495 warm = 1;
pascal@12273 496 }
pascal@12273 497
pascal@12273 498 syslinux_reboot(warm);
pascal@12273 499 }
pascal@12273 500
pascal@12273 501 /* APM poweroff module.
pascal@12273 502 * based on poweroff.asm, Copyright 2009 Sebastian Herbszt
pascal@12273 503 */
pascal@12273 504
pascal@12273 505 static int main_poweroff(int argc, char *argv[])
pascal@12273 506 {
pascal@12273 507 static com32sys_t ireg, oreg;
pascal@12273 508 static char notsupported[] ="APM 1.1+ not supported";
pascal@12273 509 unsigned i;
pascal@12273 510 static struct {
pascal@12273 511 unsigned short ax;
pascal@12273 512 unsigned short bx;
pascal@12273 513 unsigned short cx;
pascal@12273 514 char *msg;
pascal@12273 515 } inst[] = {
pascal@12273 516 { 0x5300, // APM Installation Check (00h)
pascal@12273 517 0, // APM BIOS (0000h)
pascal@12273 518 0, "APM not present" },
pascal@12273 519 { 0x5301, // APM Real Mode Interface Connect (01h)
pascal@12273 520 0, // APM BIOS (0000h)
pascal@12273 521 0, "APM RM interface connect failed" },
pascal@12273 522 { 0x530E, // APM Driver Version (0Eh)
pascal@12273 523 0, // APM BIOS (0000h)
pascal@12273 524 0x0101, // APM Driver Version version 1.1
pascal@12273 525 notsupported },
pascal@12273 526 { 0x5307, // Set Power State (07h)
pascal@12273 527 1, // All devices power managed by the APM
pascal@12273 528 3, // Power state off
pascal@12273 529 "Power off failed" }
pascal@12273 530 };
pascal@12273 531
pascal@12273 532 (void) argc;
pascal@12273 533 (void) argv;
pascal@12273 534 for (i = 0; i < sizeof(inst)/sizeof(inst[0]); i++) {
pascal@12273 535 char *msg = inst[i].msg;
pascal@12273 536
pascal@12273 537 ireg.eax.w[0] = inst[i].ax;
pascal@12273 538 ireg.ebx.w[0] = inst[i].bx;
pascal@12273 539 ireg.ecx.w[0] = inst[i].cx;
pascal@12273 540 __intcall(0x15, &ireg, &oreg);
pascal@12273 541 if ((oreg.eflags.l & EFLAGS_CF) == 0) {
pascal@12273 542 switch (inst[i].ax) {
pascal@12273 543 case 0x5300 :
pascal@12273 544 if (oreg.ebx.w[0] != 0x504D /* 'PM' */) break;
pascal@12273 545 msg = "Power management disabled";
pascal@12273 546 if (oreg.ecx.w[0] & 8) break; // bit 3 APM BIOS Power Management disabled
pascal@12273 547 case 0x530E :
pascal@12273 548 msg = notsupported;
pascal@12273 549 if (oreg.eax.w[0] < 0x101) break;
pascal@12273 550 default : continue;
pascal@12273 551 }
pascal@12273 552 }
pascal@12273 553 printf("%s.\n", msg);
pascal@12273 554 return 1;
pascal@12273 555 }
pascal@12273 556 return 0;
pascal@12273 557 }
pascal@12273 558
pascal@12287 559 /*
pascal@12287 560 * Copyright 2009 Intel Corporation; author: H. Peter Anvin
pascal@12287 561 */
pascal@12287 562
pascal@12287 563 #include <syslinux/keyboard.h>
pascal@12287 564 #include <syslinux/loadfile.h>
pascal@12287 565 #include <syslinux/adv.h>
pascal@12287 566
pascal@12349 567 static void setlinuxarg(int slot, int argc, char *argv[])
pascal@12349 568 {
pascal@12349 569 for (; argc--; argv++)
pascal@12349 570 syslinux_setadv(slot++, strlen(*argv), *argv);
pascal@12349 571 }
pascal@12349 572
pascal@12287 573 static int main_kbdmap(int argc, char *argv[])
pascal@12287 574 {
pascal@12287 575 const struct syslinux_keyboard_map *const kmap = syslinux_keyboard_map();
pascal@12287 576 size_t map_size, size, i;
pascal@12287 577 char *kbdmap, *msg;
pascal@12287 578
pascal@12287 579 if (argc < 3)
pascal@12349 580 usage("Usage: kbdmap archive.cpio mapfile [cmdline]..");
pascal@12287 581
pascal@12293 582 // Save extra cmdline arguments
pascal@12349 583 setlinuxarg(1, argc - 3, argv + 3);
pascal@12293 584
pascal@13318 585 msg="Append to kernel parameters: ";
pascal@13318 586 for (i = 3; i < argc; i++, msg = " ")
pascal@13318 587 printf("%s%s",msg,argv[i]);
pascal@13318 588 printf("\n\n Hit RETURN to continue.\n");
pascal@13317 589
pascal@12287 590 msg = "Load error";
pascal@12287 591 if (kmap->version != 1 ||
pascal@12287 592 loadfile(argv[1], (void **) &kbdmap, &map_size) ||
pascal@12287 593 strncmp(kbdmap, "07070", 5))
pascal@12287 594 goto kbdmap_error;
pascal@12287 595
pascal@12287 596 // search for mapfile in cpio archive
pascal@12287 597 for (i = 0; i < map_size;) {
pascal@12287 598 int len, j;
pascal@12287 599 char *name;
pascal@12287 600
pascal@12287 601 for (j = size = 0; j < 8; j++) {
pascal@12287 602 char c = kbdmap[54 + i + j] - '0';
pascal@12287 603 if (c > 9) c += '0' + 10 - 'A';
pascal@12287 604 size <<= 4;
pascal@12287 605 size += c;
pascal@12287 606 }
pascal@12287 607 i += 110;
pascal@12287 608 name = kbdmap + i;
pascal@12287 609 len = 1 + strlen(name);
pascal@12287 610 i += len;
pascal@12287 611 i += ((-i)&3);
pascal@12287 612 if (!strcmp(name, argv[2])) {
pascal@12287 613 kbdmap += i;
pascal@12287 614 break;
pascal@12287 615 }
pascal@12287 616 i += size + ((-size)&3);
pascal@12287 617 }
pascal@12287 618
pascal@12287 619 msg = "Filename error";
pascal@12287 620 if (i >= map_size)
pascal@12287 621 goto kbdmap_error;
pascal@12287 622
pascal@12287 623 msg = "Format error";
pascal@12287 624 if (size != kmap->length)
pascal@12287 625 goto kbdmap_error;
pascal@12287 626
pascal@12287 627 memcpy(kmap->map, kbdmap, size);
pascal@12287 628
pascal@12287 629 return 0;
pascal@12287 630
pascal@12287 631 kbdmap_error:
pascal@12287 632 printf("%s.\n",msg);
pascal@12287 633 return 1;
pascal@12287 634 }
pascal@12287 635
pascal@12287 636 /* ----------------------------------------------------------------------- *
pascal@12287 637 *
pascal@12287 638 * Copyright 2007-2008 H. Peter Anvin - All Rights Reserved
pascal@12287 639 * Copyright 2009 Intel Corporation; author: H. Peter Anvin
pascal@12287 640 *
pascal@12287 641 * Permission is hereby granted, free of charge, to any person
pascal@12287 642 * obtaining a copy of this software and associated documentation
pascal@12287 643 * files (the "Software"), to deal in the Software without
pascal@12287 644 * restriction, including without limitation the rights to use,
pascal@12287 645 * copy, modify, merge, publish, distribute, sublicense, and/or
pascal@12287 646 * sell copies of the Software, and to permit persons to whom
pascal@12287 647 * the Software is furnished to do so, subject to the following
pascal@12287 648 * conditions:
pascal@12287 649 *
pascal@12287 650 * The above copyright notice and this permission notice shall
pascal@12287 651 * be included in all copies or substantial portions of the Software.
pascal@12287 652 *
pascal@12287 653 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
pascal@12287 654 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
pascal@12287 655 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
pascal@12287 656 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
pascal@12287 657 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
pascal@12287 658 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
pascal@12287 659 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
pascal@12287 660 * OTHER DEALINGS IN THE SOFTWARE.
pascal@12287 661 *
pascal@12287 662 * ----------------------------------------------------------------------- */
pascal@12287 663
pascal@12287 664 /*
pascal@12287 665 * linux.c
pascal@12287 666 *
pascal@12287 667 * Sample module to load Linux kernels. This module can also create
pascal@12287 668 * a file out of the DHCP return data if running under PXELINUX.
pascal@12287 669 *
pascal@12287 670 * If -dhcpinfo is specified, the DHCP info is written into the file
pascal@12287 671 * /dhcpinfo.dat in the initramfs.
pascal@12287 672 *
pascal@12287 673 * Usage: linux.c32 [-dhcpinfo] kernel arguments...
pascal@12287 674 */
pascal@12287 675
pascal@12287 676 #include <stdbool.h>
pascal@12287 677 #include <stdlib.h>
pascal@12287 678 #include <stdio.h>
pascal@12287 679 #include <string.h>
pascal@12287 680 #include <console.h>
pascal@12287 681 #include <syslinux/loadfile.h>
pascal@12287 682 #include <syslinux/linux.h>
pascal@12287 683 #include <syslinux/pxe.h>
pascal@12287 684
pascal@12287 685 const char *progname = "linux.c32";
pascal@12287 686
pascal@12287 687 /* Find the last instance of a particular command line argument
pascal@12287 688 (which should include the final =; do not use for boolean arguments) */
pascal@12287 689 static char *find_argument(char **argv, const char *argument)
pascal@12287 690 {
pascal@12287 691 int la = strlen(argument);
pascal@12287 692 char **arg;
pascal@12287 693 char *ptr = NULL;
pascal@12287 694
pascal@12287 695 for (arg = argv; *arg; arg++) {
pascal@12287 696 if (!memcmp(*arg, argument, la))
pascal@12287 697 ptr = *arg + la;
pascal@12287 698 }
pascal@12287 699
pascal@12287 700 return ptr;
pascal@12287 701 }
pascal@12287 702
pascal@12287 703 /* Search for a boolean argument; return its position, or 0 if not present */
pascal@12287 704 static int find_boolean(char **argv, const char *argument)
pascal@12287 705 {
pascal@12287 706 char **arg;
pascal@12287 707
pascal@12287 708 for (arg = argv; *arg; arg++) {
pascal@12287 709 if (!strcmp(*arg, argument))
pascal@12287 710 return (arg - argv) + 1;
pascal@12287 711 }
pascal@12287 712
pascal@12287 713 return 0;
pascal@12287 714 }
pascal@12287 715
pascal@12287 716 /* Stitch together the command line from a set of argv's */
pascal@12287 717 static char *make_cmdline(char **argv)
pascal@12287 718 {
pascal@12287 719 char **arg;
pascal@12287 720 size_t bytes, size;
pascal@12287 721 char *cmdline, *p;
pascal@12287 722 int i;
pascal@12287 723
pascal@12287 724 bytes = 1; /* Just in case we have a zero-entry cmdline */
pascal@12287 725 for (arg = argv; *arg; arg++) {
pascal@12287 726 bytes += strlen(*arg) + 1;
pascal@12287 727 }
pascal@12287 728 for (i = 0; i < 255; i++)
pascal@12287 729 if (syslinux_getadv(i, &size))
pascal@12287 730 bytes += ++size;
pascal@12287 731
pascal@12287 732 p = cmdline = malloc(bytes);
pascal@12287 733 if (!cmdline)
pascal@12287 734 return NULL;
pascal@12287 735
pascal@12287 736 for (arg = argv; *arg; arg++) {
pascal@12287 737 int len = strlen(*arg);
pascal@12287 738 memcpy(p, *arg, len);
pascal@12287 739 p[len] = ' ';
pascal@12287 740 p += len + 1;
pascal@12287 741 }
pascal@12287 742
pascal@12287 743 for (i = 0; i < 255; i++) {
pascal@12287 744 const void *q = syslinux_getadv(i, &size);
pascal@12287 745 if (q == NULL) continue;
pascal@12287 746 memcpy(p, q, size);
pascal@12287 747 p[size] = ' ';
pascal@12287 748 p += size + 1;
pascal@12287 749 }
pascal@12287 750
pascal@12287 751 if (p > cmdline)
pascal@12287 752 p--; /* Remove the last space */
pascal@12287 753 *p = '\0';
pascal@12287 754
pascal@12287 755 return cmdline;
pascal@12287 756 }
pascal@12287 757
pascal@12287 758 static int main_linux(int argc, char *argv[])
pascal@12287 759 {
pascal@12287 760 const char *kernel_name;
pascal@12287 761 struct initramfs *initramfs;
pascal@12287 762 char *cmdline;
pascal@12287 763 char *boot_image;
pascal@12287 764 void *kernel_data;
pascal@12287 765 size_t kernel_len;
pascal@12287 766 bool opt_dhcpinfo = false;
pascal@12287 767 bool opt_quiet = false;
pascal@12287 768 void *dhcpdata;
pascal@12287 769 size_t dhcplen;
pascal@12287 770 char **argp, *arg, *p;
pascal@12287 771
pascal@12287 772 (void)argc;
pascal@12287 773 argp = argv + 1;
pascal@12287 774
pascal@12287 775 while ((arg = *argp) && arg[0] == '-') {
pascal@12287 776 if (!strcmp("-dhcpinfo", arg)) {
pascal@12287 777 opt_dhcpinfo = true;
pascal@12287 778 } else {
pascal@12287 779 fprintf(stderr, "%s: unknown option: %s\n", progname, arg);
pascal@12287 780 return 1;
pascal@12287 781 }
pascal@12287 782 argp++;
pascal@12287 783 }
pascal@12287 784
pascal@12287 785 if (!arg) {
pascal@12287 786 fprintf(stderr, "%s: missing kernel name\n", progname);
pascal@12287 787 return 1;
pascal@12287 788 }
pascal@12287 789
pascal@12287 790 kernel_name = arg;
pascal@12287 791
pascal@12287 792 boot_image = malloc(strlen(kernel_name) + 12);
pascal@12287 793 if (!boot_image)
pascal@12287 794 goto bail;
pascal@12287 795 strcpy(boot_image, "BOOT_IMAGE=");
pascal@12287 796 strcpy(boot_image + 11, kernel_name);
pascal@12287 797 /* argp now points to the kernel name, and the command line follows.
pascal@12287 798 Overwrite the kernel name with the BOOT_IMAGE= argument, and thus
pascal@12287 799 we have the final argument. */
pascal@12287 800 *argp = boot_image;
pascal@12287 801
pascal@12287 802 if (find_boolean(argp, "quiet"))
pascal@12287 803 opt_quiet = true;
pascal@12287 804
pascal@12287 805 if (!opt_quiet)
pascal@12287 806 printf("Loading %s... ", kernel_name);
pascal@12287 807 if (loadfile(kernel_name, &kernel_data, &kernel_len)) {
pascal@12287 808 if (opt_quiet)
pascal@12287 809 printf("Loading %s ", kernel_name);
pascal@12287 810 printf("failed!\n");
pascal@12287 811 goto bail;
pascal@12287 812 }
pascal@12287 813 if (!opt_quiet)
pascal@12287 814 printf("ok\n");
pascal@12287 815
pascal@12287 816 cmdline = make_cmdline(argp);
pascal@12287 817 if (!cmdline)
pascal@12287 818 goto bail;
pascal@12287 819
pascal@12287 820 /* Initialize the initramfs chain */
pascal@12287 821 initramfs = initramfs_init();
pascal@12287 822 if (!initramfs)
pascal@12287 823 goto bail;
pascal@12287 824
pascal@12287 825 if ((arg = find_argument(argp, "initrd="))) {
pascal@12287 826 do {
pascal@12287 827 p = strchr(arg, ',');
pascal@12287 828 if (p)
pascal@12287 829 *p = '\0';
pascal@12287 830
pascal@12287 831 if (!opt_quiet)
pascal@12287 832 printf("Loading %s... ", arg);
pascal@12287 833 if (initramfs_load_archive(initramfs, arg)) {
pascal@12287 834 if (opt_quiet)
pascal@12287 835 printf("Loading %s ", kernel_name);
pascal@12287 836 printf("failed!\n");
pascal@12287 837 goto bail;
pascal@12287 838 }
pascal@12287 839 if (!opt_quiet)
pascal@12287 840 printf("ok\n");
pascal@12287 841
pascal@12287 842 if (p)
pascal@12287 843 *p++ = ',';
pascal@12287 844 } while ((arg = p));
pascal@12287 845 }
pascal@12287 846
pascal@12287 847 /* Append the DHCP info */
pascal@12287 848 if (opt_dhcpinfo &&
pascal@12287 849 !pxe_get_cached_info(PXENV_PACKET_TYPE_DHCP_ACK, &dhcpdata, &dhcplen)) {
pascal@12287 850 if (initramfs_add_file(initramfs, dhcpdata, dhcplen, dhcplen,
pascal@12287 851 "/dhcpinfo.dat", 0, 0755))
pascal@12287 852 goto bail;
pascal@12287 853 }
pascal@12287 854
pascal@12287 855 /* This should not return... */
pascal@12287 856 syslinux_boot_linux(kernel_data, kernel_len, initramfs, cmdline);
pascal@12287 857
pascal@12287 858 bail:
pascal@12287 859 fprintf(stderr, "Kernel load failure (insufficient memory?)\n");
pascal@12287 860 return 1;
pascal@12287 861 }
pascal@12287 862
pascal@12349 863 static int main_setarg(int argc, char *argv[])
pascal@12349 864 {
pascal@12349 865 if (argc < 3) {
pascal@12349 866 usage("Usage: setarg.c32 argnum [args]...");
pascal@12349 867 }
pascal@12349 868 setlinuxarg(atoi(argv[1]), argc - 2, argv + 2);
pascal@12349 869 return 0;
pascal@12349 870 }
pascal@12349 871
pascal@12349 872 static int main_ifarg(int argc, char *argv[])
pascal@12349 873 {
pascal@12349 874 int i;
pascal@12349 875 size_t size;
pascal@12349 876
pascal@12349 877 if (argc < 3) {
pascal@12349 878 usage("Usage: ifarg.c32 [argnum labelifset]... labelifnoneset");
pascal@12349 879 }
pascal@12349 880 for (i = 1; i < argc - 1; i += 2) {
pascal@13295 881 int n = atoi(argv[i]);
pascal@13295 882 if (n == -1) {
pascal@13295 883 for (n = 0; n < 255; n++) {
pascal@13316 884 if (syslinux_getadv(n, &size))
pascal@13295 885 goto found;
pascal@13295 886 }
pascal@13295 887 continue;
pascal@13295 888 }
pascal@13295 889 else if (! syslinux_getadv(n, &size)) continue;
pascal@13295 890 found:
pascal@13295 891 syslinux_run_command(argv[i+1]);
pascal@12349 892 }
pascal@12349 893 if (i != argc) syslinux_run_command(argv[i]);
pascal@12349 894 else syslinux_run_default();
pascal@12349 895 return 0;
pascal@12349 896 }
pascal@12349 897
pascal@12349 898 /* ----------------------------------------------------------------------- *
pascal@12349 899 *
pascal@12349 900 * Copyright 2007-2008 H. Peter Anvin - All Rights Reserved
pascal@12349 901 *
pascal@12349 902 * This program is free software; you can redistribute it and/or modify
pascal@12349 903 * it under the terms of the GNU General Public License as published by
pascal@12349 904 * the Free Software Foundation, Inc., 53 Temple Place Ste 330,
pascal@12349 905 * Boston MA 02111-1307, USA; either version 2 of the License, or
pascal@12349 906 * (at your option) any later version; incorporated herein by reference.
pascal@12349 907 *
pascal@12349 908 * ----------------------------------------------------------------------- */
pascal@12349 909
pascal@12349 910 static int main_listarg(int argc, char *argv[])
pascal@12349 911 {
pascal@12349 912 uint8_t *p, *ep;
pascal@12349 913 size_t s = syslinux_adv_size();
pascal@12349 914 char buf[256];
pascal@12349 915
pascal@12349 916 (void) argc;
pascal@12349 917 (void) argv;
pascal@12349 918 p = syslinux_adv_ptr();
pascal@12349 919
pascal@12349 920 printf("args size: %zd bytes at %p\n", s, p);
pascal@12349 921
pascal@12349 922 ep = p + s; /* Need at least opcode+len */
pascal@12349 923 while (p < ep - 1 && *p) {
pascal@12349 924 int t = *p++;
pascal@12349 925 int l = *p++;
pascal@12349 926
pascal@12349 927 if (p + l > ep)
pascal@12349 928 break;
pascal@12349 929
pascal@12349 930 memcpy(buf, p, l);
pascal@12349 931 buf[l] = '\0';
pascal@12349 932
pascal@12349 933 printf("arg %3d: \"%s\"\n", t, buf);
pascal@12349 934
pascal@12349 935 p += l;
pascal@12349 936 }
pascal@12349 937 sleep(5);
pascal@12349 938 return 0;
pascal@12349 939 }
pascal@12349 940
pascal@12273 941 int main(int argc, char *argv[])
pascal@12273 942 {
pascal@12273 943 unsigned i;
pascal@12273 944 static struct {
pascal@12273 945 char *name;
pascal@12273 946 int (*main)(int argc, char *argv[]);
pascal@12273 947 } bin[] = {
pascal@13295 948 { "say", main_say },
pascal@13295 949 { "md5sum", main_md5sum },
pascal@13295 950 { "ifmem", main_ifmem },
pascal@13295 951 { "reboot", main_reboot },
pascal@12287 952 { "poweroff", main_poweroff },
pascal@13295 953 { "kbdmap", main_kbdmap },
pascal@13295 954 { "linux", main_linux },
pascal@13295 955 { "setarg", main_setarg },
pascal@13295 956 { "ifarg", main_ifarg },
pascal@13295 957 { "listarg", main_listarg }
pascal@12273 958 };
pascal@12273 959
pascal@12273 960 openconsole(&dev_null_r, &dev_stdcon_w);
pascal@12273 961
pascal@12273 962 if (strstr(argv[0], "c32box")) { argc--; argv++; }
pascal@12273 963 for (i = 0; i < sizeof(bin)/sizeof(bin[0]); i++)
pascal@12273 964 if (strstr(argv[0], bin[i].name))
pascal@12273 965 return bin[i].main(argc, argv);
pascal@12287 966 printf("No %s in c32box modules\n", argv[0]);
pascal@12287 967 for (i = 0; i < sizeof(bin)/sizeof(bin[0]); i++)
pascal@12287 968 printf(" %s \n",bin[i].name);
pascal@12273 969 return 1;
pascal@12273 970 }