]> git.ozlabs.org Git - yaboot.git/blobdiff - second/yaboot.c
Remove devel debugging code
[yaboot.git] / second / yaboot.c
index fd5dee3c5399e818f9469ec5540aa79e2af0ce53..3fbb103ab4d942134be70c0855ceda1b58f7cb24 100644 (file)
@@ -1,25 +1,25 @@
 /*
- *  Yaboot - secondary boot loader for Linux on PowerPC. 
+ *  Yaboot - secondary boot loader for Linux on PowerPC.
  *
  *  Copyright (C) 2001, 2002 Ethan Benson
  *
  *  Copyright (C) 1999, 2000, 2001 Benjamin Herrenschmidt
- *  
+ *
  *  IBM CHRP support
- *  
+ *
  *  Copyright (C) 2001 Peter Bergner
  *
  *  portions based on poof
- *  
+ *
  *  Copyright (C) 1999 Marius Vollmer
- *  
+ *
  *  portions based on quik
- *  
+ *
  *  Copyright (C) 1996 Paul Mackerras.
- *  
+ *
  *  Because this program is derived from the corresponding file in the
  *  silo-0.64 distribution, it is also
- *  
+ *
  *  Copyright (C) 1996 Pete A. Zaitcev
  *                1996 Maurizio Plaza
  *                1996 David S. Miller
@@ -111,7 +111,7 @@ static void     setup_display(void);
 /* Locals & globals */
 
 int useconf = 0;
-char bootdevice[1024];
+char bootdevice[BOOTDEVSZ];
 char *password = NULL;
 struct boot_fspec_t boot;
 int _machine = _MACH_Pmac;
@@ -180,7 +180,7 @@ yaboot_start (unsigned long r3, unsigned long r4, unsigned long r5)
 
      /* OF seems to do it, but I'm not very confident */
      memset(&__bss_start, 0, &_end - &__bss_start);
-       
+
      /* Check for quik first stage bootloader (but I don't think we are
       * compatible with it anyway, I'll look into backporting to older OF
       * versions later
@@ -192,7 +192,7 @@ yaboot_start (unsigned long r3, unsigned long r4, unsigned long r5)
 
      /* Initialize OF interface */
      prom_init ((prom_entry) r5);
-       
+
      /* Allocate some memory for malloc'ator */
      malloc_base = prom_claim((void *)MALLOCADDR, MALLOCSIZE, 0);
      if (malloc_base == (void *)-1) {
@@ -203,7 +203,7 @@ yaboot_start (unsigned long r3, unsigned long r4, unsigned long r5)
      malloc_init(malloc_base, MALLOCSIZE);
      DEBUG_F("Malloc buffer allocated at %p (%d bytes)\n",
             malloc_base, MALLOCSIZE);
-               
+
      /* A few useless DEBUG_F's */
      DEBUG_F("reloc_offset :  %ld         (should be 0)\n", reloc_offset());
      DEBUG_F("test_bss     :  %d         (should be 0)\n", test_bss);
@@ -226,7 +226,7 @@ yaboot_start (unsigned long r3, unsigned long r4, unsigned long r5)
                    _machine = _MACH_chrp;
          }
      }
-       
+
      DEBUG_F("Running on _machine = %d\n", _machine);
      DEBUG_SLEEP;
 
@@ -241,9 +241,9 @@ yaboot_start (unsigned long r3, unsigned long r4, unsigned long r5)
 
      /* Return to OF */
      prom_exit();
-       
+
      return result;
-       
+
 }
 
 #ifdef CONFIG_COLOR_TEXT
@@ -260,13 +260,13 @@ check_color_text_ui(char *color)
          i++;
      }
      return -1;
-}      
+}
 #endif /* CONFIG_COLOR_TEXT */
 
 
 void print_message_file(char *filename)
 {
-     char *msg = NULL; 
+     char *msg = NULL;
      char *p, *endp;
      char *defdev = boot.dev;
      int defpart = boot.part;
@@ -284,7 +284,7 @@ void print_message_file(char *filename)
          if (p) {
               n = simple_strtol(p, &endp, 10);
               if (endp != p && *endp == 0)
-                   defpart = n;               
+                   defpart = n;
          }
 
      strncpy(msgpath, filename, sizeof(msgpath));
@@ -324,13 +324,13 @@ done:
  * config file. Handle the "\\" (blessed system folder)
  */
 static int
-load_config_file(char *device, char* path, int partition)
+load_config_file(struct boot_fspec_t *orig_fspec)
 {
      char *conf_file = NULL, *p;
      struct boot_file_t file;
      int sz, opened = 0, result = 0;
      char conf_path[512];
-     struct boot_fspec_t fspec;
+     struct boot_fspec_t fspec = *orig_fspec;
 
      /* Allocate a buffer for the config file */
      conf_file = malloc(CONFIG_FILE_MAX);
@@ -342,16 +342,15 @@ load_config_file(char *device, char* path, int partition)
      /* Build the path to the file */
      if (_machine == _MACH_chrp)
          strcpy(conf_path, "/etc/");
-     else if (path && *path)
-         strcpy(conf_path, path);
      else
          conf_path[0] = 0;
+     if (fspec.file && *fspec.file)
+         strcat(conf_path, fspec.file);
+     else
      strcat(conf_path, CONFIG_FILE_NAME);
 
      /* Open it */
-     fspec.dev = device;
      fspec.file = conf_path;
-     fspec.part = partition;
      result = open_file(&fspec, &file);
      if (result != FILE_ERR_OK) {
          prom_printf("%s:%d,", fspec.dev, fspec.part);
@@ -388,7 +387,7 @@ load_config_file(char *device, char* path, int partition)
          prom_interpret(p);
 
      password = cfg_get_strg(0, "password");
-       
+
 #ifdef CONFIG_COLOR_TEXT
      p = cfg_get_strg(0, "fgcolor");
      if (p) {
@@ -407,19 +406,19 @@ load_config_file(char *device, char* path, int partition)
      }
      if (bgcolor >= 0) {
          char temp[64];
-         sprintf(temp, "%x to background-color", bgcolor); 
-         prom_interpret(temp); 
+         sprintf(temp, "%x to background-color", bgcolor);
+         prom_interpret(temp);
 #if !DEBUG
          prom_printf("\xc");
 #endif /* !DEBUG */
      }
      if (fgcolor >= 0) {
          char temp[64];
-         sprintf(temp, "%x to foreground-color", fgcolor); 
-         prom_interpret(temp); 
+         sprintf(temp, "%x to foreground-color", fgcolor);
+         prom_interpret(temp);
      }
 #endif /* CONFIG_COLOR_TEXT */
-   
+
      p = cfg_get_strg(0, "init-message");
      if (p)
          prom_printf("%s\n", p);
@@ -429,18 +428,143 @@ load_config_file(char *device, char* path, int partition)
          print_message_file(p);
 
      result = 1;
-    
+
 bail:
 
      if (opened)
          file.fs->close(&file);
-    
+
      if (conf_file)
          free(conf_file);
-       
+
      return result;
 }
 
+/*
+ * "bootp-response" is the property name which is specified in
+ * the recommended practice doc for obp-tftp. However, pmac
+ * provides a "dhcp-response" property and chrp provides a
+ * "bootpreply-packet" property.  The latter appears to begin the
+ * bootp packet at offset 0x2a in the property for some reason.
+ */
+
+struct bootp_property_offset {
+     char *name; /* property name */
+     int offset; /* offset into property where bootp packet occurs */
+};
+static const struct bootp_property_offset bootp_response_properties[] = {
+     { .name = "bootp-response", .offset = 0 },
+     { .name = "dhcp-response", .offset = 0 },
+     { .name = "bootpreply-packet", .offset = 0x2a },
+};
+
+struct bootp_packet {
+     u8 op, htype, hlen, hops;
+     u32 xid;
+     u16 secs, flags;
+     u32 ciaddr, yiaddr, siaddr, giaddr;
+     unsigned char chaddr[16];
+     unsigned char sname[64];
+     unsigned char file[128];
+     /* vendor options go here if we need them */
+};
+
+/*
+ * Search for config file by MAC address, then by IP address.
+ * Basically copying pxelinux's algorithm.
+ * http://syslinux.zytor.com/pxe.php#config
+ */
+static int load_my_config_file(struct boot_fspec_t *orig_fspec)
+{
+     void *bootp_response = NULL;
+     char *propname;
+     struct bootp_packet *packet;
+     int i = 0, size, offset = 0, rc = 0;
+     prom_handle chosen;
+     struct boot_fspec_t fspec = *orig_fspec;
+#define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
+
+     chosen = prom_finddevice("/chosen");
+     if (chosen < 0) {
+          prom_printf("chosen=%d\n", chosen);
+         return 0;
+     }
+
+     for (i = 0; i < ARRAY_SIZE(bootp_response_properties); i++) {
+         propname = bootp_response_properties[i].name;
+         size = prom_getproplen(chosen, propname);
+         if (size <= 0)
+              continue;
+
+         DEBUG_F("using /chosen/%s\n", propname);
+         offset = bootp_response_properties[i].offset;
+         break;
+     }
+
+     if (size <= 0)
+         goto out;
+
+     if (sizeof(*packet) > size - offset) {
+         prom_printf("Malformed %s property?\n", propname);
+         goto out;
+     }
+
+     bootp_response = malloc(size);
+     if (!bootp_response)
+         goto out;
+
+     if (prom_getprop(chosen, propname, bootp_response, size) < 0)
+         goto out;
+
+     packet = bootp_response + offset;
+
+     /*
+      * First, try to match on mac address with the hardware type
+      * prepended.
+      */
+
+     /* 3 chars per byte in chaddr + 2 chars for htype + \0 */
+     fspec.file = malloc(packet->hlen * 3 + 2 + 1);
+     if (!fspec.file)
+         goto out;
+
+     sprintf(fspec.file, "%02x", packet->htype);
+
+     for (i = 0; i < packet->hlen; i++) {
+         char tmp[4];
+         sprintf(tmp, "-%02x", packet->chaddr[i]);
+         strcat(fspec.file, tmp);
+     }
+
+     rc = load_config_file(&fspec);
+     if (rc)
+         goto out;
+
+
+     /*
+      * Now try to match on IP.
+      */
+     free(fspec.file);
+     fspec.file = malloc(9);
+     sprintf(fspec.file, "%08X", packet->yiaddr);
+
+     while (strlen(fspec.file)) {
+         rc = load_config_file(&fspec);
+         if (rc)
+              goto out;
+         /* Chop one digit off the end, try again */
+         fspec.file[strlen(fspec.file) - 1] = '\0';
+     }
+
+ out:
+     if (rc) /* modify original only on success */
+         orig_fspec->file = fspec.file;
+     else
+         free(fspec.file);
+     free(bootp_response);
+     return rc;
+}
+
 void maintabfunc (void)
 {
      if (useconf) {
@@ -559,7 +683,7 @@ void check_password(char *str)
          if (!strncmp (password, "$1$", 3)) {
               if (!check_md5_password(passwdbuff, password))
                    return;
-         } 
+         }
          else if (!strcmp (password, passwdbuff))
               return;
 #else /* !MD5 */
@@ -602,7 +726,7 @@ int get_params(struct boot_param_t* params)
      params->rd.part = -1;
      params->sysmap.part = -1;
      defpart = boot.part;
-    
+
      cmdinit();
 
      if (first) {
@@ -747,7 +871,7 @@ int get_params(struct boot_param_t* params)
               check_password ("Restricted command.");
               return 1;
          }
-         return 1; 
+         return 1;
      }
 
      if (imagename[0] == '$') {
@@ -819,7 +943,7 @@ yaboot_text_ui(void)
      loadinfo_t          loadinfo;
      void                *initrd_more,*initrd_want;
      unsigned long       initrd_read;
-    
+
      loadinfo.load_loc = 0;
 
      for (;;) {
@@ -827,12 +951,12 @@ yaboot_text_ui(void)
          initrd_base = 0;
          sysmap_base = 0;
          sysmap_size = 0;
-       
+
          if (get_params(&params))
               return;
          if (!params.kernel.file)
               continue;
-       
+
          prom_printf("Please wait, loading kernel...\n");
 
          memset(&file, 0, sizeof(file));
@@ -906,7 +1030,7 @@ yaboot_text_ui(void)
                    free(params.sysmap.file);
                    params.sysmap.file=loc;
               }
-            
+
               result = open_file(&params.sysmap, &file);
               if (result != FILE_ERR_OK) {
                    prom_printf("%s:%d,", params.sysmap.dev, params.sysmap.part);
@@ -1007,7 +1131,7 @@ yaboot_text_ui(void)
          DEBUG_F(" done\n");
 
          if (flat_vmlinux) {
-              /* 
+              /*
                * Fill new boot infos (only if booting a vmlinux).
                *
                * The birec is low on memory, probably inside the malloc pool,
@@ -1027,12 +1151,12 @@ yaboot_text_ui(void)
               birec->tag = BI_FIRST;
               birec->size = sizeof(struct bi_record);
               birec = (struct bi_record *)((ulong)birec + birec->size);
-       
+
               birec->tag = BI_BOOTLOADER_ID;
               sprintf( (char *)birec->data, "yaboot");
               birec->size = sizeof(struct bi_record) + strlen("yaboot") + 1;
               birec = (struct bi_record *)((ulong)birec + birec->size);
-       
+
               birec->tag = BI_MACHTYPE;
               birec->data[0] = _machine;
               birec->size = sizeof(struct bi_record) + sizeof(ulong);
@@ -1176,7 +1300,7 @@ load_elf32(struct boot_file_t *file, loadinfo_t *loadinfo)
      if (loadinfo->base == (void *)-1) {
          prom_printf("Claim error, can't allocate kernel memory\n");
          goto bail;
-     } 
+     }
 
      DEBUG_F("After ELF parsing, load base: %p, mem_sz: 0x%08lx\n",
             loadinfo->base, loadinfo->memsize);
@@ -1205,7 +1329,7 @@ load_elf32(struct boot_file_t *file, loadinfo_t *loadinfo)
      }
 
      free(ph);
-    
+
      /* Return success at loading the Elf32 kernel */
      return 1;
 
@@ -1320,7 +1444,7 @@ load_elf64(struct boot_file_t *file, loadinfo_t *loadinfo)
      if (loadinfo->base == (void *)-1) {
          prom_printf("Claim error, can't allocate kernel memory\n");
          goto bail;
-     } 
+     }
 
      DEBUG_F("After ELF parsing, load base: %p, mem_sz: 0x%08lx\n",
             loadinfo->base, loadinfo->memsize);
@@ -1349,7 +1473,7 @@ load_elf64(struct boot_file_t *file, loadinfo_t *loadinfo)
      }
 
      free(ph);
-    
+
      /* Return success at loading the Elf64 kernel */
      return 1;
 
@@ -1433,7 +1557,7 @@ setup_display(void)
               DEBUG_F("Open screen result: %p\n", scrn);
          }
      }
-       
+
      if (scrn == PROM_INVALID_HANDLE) {
          prom_printf("No screen device found !/n");
          return;
@@ -1442,7 +1566,7 @@ setup_display(void)
          prom_set_color(scrn, i, default_colors[i*3],
                         default_colors[i*3+1], default_colors[i*3+2]);
      }
-     prom_printf("\x1b[1;37m\x1b[2;40m");      
+     prom_printf("\x1b[1;37m\x1b[2;40m");
 #ifdef COLOR_TEST
      for (i=0;i<16; i++) {
          prom_printf("\x1b[%d;%dm\x1b[1;47m%s \x1b[2;40m %s\n",
@@ -1456,7 +1580,7 @@ setup_display(void)
                      ansi_color_table[i].name,
                      ansi_color_table[i].name);
      }
-     prom_printf("\x1b[1;37m\x1b[2;40m");      
+     prom_printf("\x1b[1;37m\x1b[2;40m");
 #endif /* COLOR_TEST */
 
 #if !DEBUG
@@ -1473,11 +1597,11 @@ yaboot_main(void)
 
      if (_machine == _MACH_Pmac)
          setup_display();
-       
-     prom_get_chosen("bootpath", bootdevice, sizeof(bootdevice));
+
+     prom_get_chosen("bootpath", bootdevice, BOOTDEVSZ);
      DEBUG_F("/chosen/bootpath = %s\n", bootdevice);
      if (bootdevice[0] == 0) {
-         prom_get_options("boot-device", bootdevice, sizeof(bootdevice));
+         prom_get_options("boot-device", bootdevice, BOOTDEVSZ);
          DEBUG_F("boot-device = %s\n", bootdevice);
      }
      if (bootdevice[0] == 0) {
@@ -1514,7 +1638,17 @@ yaboot_main(void)
      DEBUG_F("After pmac path kludgeup: dev=%s, part=%d, file=%s\n",
             boot.dev, boot.part, boot.file);
 
-     useconf = load_config_file(boot.dev, boot.file, boot.part);
+     /*
+      * If we're doing a netboot, first look for one which matches our
+      * MAC address.
+      */
+     if (prom_get_devtype(boot.dev) == FILE_DEVICE_NET) {
+          prom_printf("Try to netboot\n");
+         useconf = load_my_config_file(&boot);
+     }
+
+     if (!useconf)
+         useconf = load_config_file(&boot);
 
      prom_printf("Welcome to yaboot version " VERSION "\n");
      prom_printf("Enter \"help\" to get some basic usage information\n");
@@ -1544,7 +1678,7 @@ yaboot_main(void)
      return 0;
 }
 
-/* 
+/*
  * Local variables:
  * c-file-style: "k&r"
  * c-basic-offset: 5