Prepare 1.3.7-pre1
[yaboot.git] / util / addnote.c
index 74c8efc8c6801710b8af645ffb444ceb68e644e6..90434c8c2a3a63259e115ee8bc82bb5f2e88d038 100644 (file)
@@ -1,18 +1,28 @@
 /*
- * Program to hack in a PT_NOTE program header entry in an ELF file.
- * This is needed for OF on RS/6000s to load an image correctly.
- * Note that OF needs a program header entry for the note, not an
- * ELF section.
+ *  addnote.c - Program to hack in a PT_NOTE program header entry in an ELF file.
+ *             This is needed for OF on RS/6000s to load an image correctly.
+ *             Note that OF needs a program header entry for the note, not an
+ *             ELF section.
  *
- * Copyright 2000 Paul Mackerras.
+ *  Copyright 2000 Paul Mackerras.
  *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version
- * 2 of the License, or (at your option) any later version.
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
  *
- * Usage: addnote zImage
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  */
+
+/* Usage: addnote zImage */
+
 #include <stdio.h>
 #include <fcntl.h>
 #include <unistd.h>
@@ -22,12 +32,12 @@ char arch[] = "PowerPC";
 
 #define N_DESCR        6
 unsigned int descr[N_DESCR] = {
-       0xffffffff,             /* real-mode = true */
-       0x00c00000,             /* real-base, i.e. where we expect OF to be */
-       0xffffffff,             /* real-size */
-       0xffffffff,             /* virt-base */
-       0xffffffff,             /* virt-size */
-       0x4000,                 /* load-base */
+     0xffffffff,               /* real-mode = true */
+     0x00c00000,               /* real-base, i.e. where we expect OF to be */
+     0xffffffff,               /* real-size */
+     0xffffffff,               /* virt-base */
+     0xffffffff,               /* virt-size */
+     0x4000,                   /* load-base */
 };
 
 unsigned char buf[512];
@@ -65,99 +75,106 @@ unsigned char elf_magic[4] = { 0x7f, 'E', 'L', 'F' };
 int
 main(int ac, char **av)
 {
-       int fd, n, i;
-       int ph, ps, np;
-       int nnote, ns;
-
-       if (ac != 2) {
-               fprintf(stderr, "Usage: %s elf-file\n", av[0]);
-               exit(1);
-       }
-       fd = open(av[1], O_RDWR);
-       if (fd < 0) {
-               perror(av[1]);
-               exit(1);
-       }
-
-       nnote = strlen(arch) + 1 + (N_DESCR + 3) * 4;
-
-       n = read(fd, buf, sizeof(buf));
-       if (n < 0) {
-               perror("read");
-               exit(1);
-       }
-
-       if (n < E_HSIZE || memcmp(&buf[E_IDENT+EI_MAGIC], elf_magic, 4) != 0)
-               goto notelf;
-
-       if (buf[E_IDENT+EI_CLASS] != ELFCLASS32
-           || buf[E_IDENT+EI_DATA] != ELFDATA2MSB) {
-               fprintf(stderr, "%s is not a big-endian 32-bit ELF image\n",
-                       av[1]);
-               exit(1);
-       }
-
-       ph = GET_32BE(E_PHOFF);
-       ps = GET_16BE(E_PHENTSIZE);
-       np = GET_16BE(E_PHNUM);
-       if (ph < E_HSIZE || ps < PH_HSIZE || np < 1)
-               goto notelf;
-       if (ph + (np + 1) * ps + nnote > n)
-               goto nospace;
-
-       for (i = 0; i < np; ++i) {
-               if (GET_32BE(ph + PH_TYPE) == PT_NOTE) {
-                       fprintf(stderr, "%s already has a note entry\n",
-                               av[1]);
-                       exit(0);
-               }
-               ph += ps;
-       }
-
-       /* XXX check that the area we want to use is all zeroes */
-       for (i = 0; i < ps + nnote; ++i)
-               if (buf[ph + i] != 0)
-                       goto nospace;
-
-       /* fill in the program header entry */
-       ns = ph + ps;
-       PUT_32BE(ph + PH_TYPE, PT_NOTE);
-       PUT_32BE(ph + PH_OFFSET, ns);
-       PUT_32BE(ph + PH_FILESZ, nnote);
-
-       /* fill in the note area we point to */
-       /* XXX we should probably make this a proper section */
-       PUT_32BE(ns, strlen(arch) + 1);
-       PUT_32BE(ns + 4, N_DESCR * 4);
-       PUT_32BE(ns + 8, 0x1275);
-       strcpy(&buf[ns + 12], arch);
-       ns += 12 + strlen(arch) + 1;
-       for (i = 0; i < N_DESCR; ++i)
-               PUT_32BE(ns + i * 4, descr[i]);
-
-       /* Update the number of program headers */
-       PUT_16BE(E_PHNUM, np + 1);
-
-       /* write back */
-       lseek(fd, (long) 0, SEEK_SET);
-       i = write(fd, buf, n);
-       if (i < 0) {
-               perror("write");
-               exit(1);
-       }
-       if (i < n) {
-               fprintf(stderr, "%s: write truncated\n", av[1]);
-               exit(1);
-       }
-
-       exit(0);
-
- notelf:
-       fprintf(stderr, "%s does not appear to be an ELF file\n", av[1]);
-       exit(1);
-
- nospace:
-       fprintf(stderr, "sorry, I can't find space in %s to put the note\n",
-               av[1]);
-       exit(1);
+     int fd, n, i;
+     int ph, ps, np;
+     int nnote, ns;
+
+     if (ac != 2) {
+         fprintf(stderr, "Usage: %s elf-file\n", av[0]);
+         exit(1);
+     }
+     fd = open(av[1], O_RDWR);
+     if (fd < 0) {
+         perror(av[1]);
+         exit(1);
+     }
+
+     nnote = strlen(arch) + 1 + (N_DESCR + 3) * 4;
+
+     n = read(fd, buf, sizeof(buf));
+     if (n < 0) {
+         perror("read");
+         exit(1);
+     }
+
+     if (n < E_HSIZE || memcmp(&buf[E_IDENT+EI_MAGIC], elf_magic, 4) != 0)
+         goto notelf;
+
+     if (buf[E_IDENT+EI_CLASS] != ELFCLASS32
+        || buf[E_IDENT+EI_DATA] != ELFDATA2MSB) {
+         fprintf(stderr, "%s is not a big-endian 32-bit ELF image\n",
+                 av[1]);
+         exit(1);
+     }
+
+     ph = GET_32BE(E_PHOFF);
+     ps = GET_16BE(E_PHENTSIZE);
+     np = GET_16BE(E_PHNUM);
+     if (ph < E_HSIZE || ps < PH_HSIZE || np < 1)
+         goto notelf;
+     if (ph + (np + 1) * ps + nnote > n)
+         goto nospace;
+
+     for (i = 0; i < np; ++i) {
+         if (GET_32BE(ph + PH_TYPE) == PT_NOTE) {
+              fprintf(stderr, "%s already has a note entry\n",
+                      av[1]);
+              exit(0);
+         }
+         ph += ps;
+     }
+
+     /* XXX check that the area we want to use is all zeroes */
+     for (i = 0; i < ps + nnote; ++i)
+         if (buf[ph + i] != 0)
+              goto nospace;
+
+     /* fill in the program header entry */
+     ns = ph + ps;
+     PUT_32BE(ph + PH_TYPE, PT_NOTE);
+     PUT_32BE(ph + PH_OFFSET, ns);
+     PUT_32BE(ph + PH_FILESZ, nnote);
+
+     /* fill in the note area we point to */
+     /* XXX we should probably make this a proper section */
+     PUT_32BE(ns, strlen(arch) + 1);
+     PUT_32BE(ns + 4, N_DESCR * 4);
+     PUT_32BE(ns + 8, 0x1275);
+     strcpy(&buf[ns + 12], arch);
+     ns += 12 + strlen(arch) + 1;
+     for (i = 0; i < N_DESCR; ++i)
+         PUT_32BE(ns + i * 4, descr[i]);
+
+     /* Update the number of program headers */
+     PUT_16BE(E_PHNUM, np + 1);
+
+     /* write back */
+     lseek(fd, (long) 0, SEEK_SET);
+     i = write(fd, buf, n);
+     if (i < 0) {
+         perror("write");
+         exit(1);
+     }
+     if (i < n) {
+         fprintf(stderr, "%s: write truncated\n", av[1]);
+         exit(1);
+     }
+
+     exit(0);
+
+notelf:
+     fprintf(stderr, "%s does not appear to be an ELF file\n", av[1]);
+     exit(1);
+
+nospace:
+     fprintf(stderr, "sorry, I can't find space in %s to put the note\n",
+            av[1]);
+     exit(1);
 }
+
+/* 
+ * Local variables:
+ * c-file-style: "k&r"
+ * c-basic-offset: 5
+ * End:
+ */