src/unexalpha.c

   1 /* Unexec for DEC alpha.  schoepf@sc.ZIB-Berlin.DE (Rainer Schoepf).
   2
   3    Copyright (C) 1994 Free Software Foundation, Inc.
   4
   5 This file is part of GNU Emacs.
   6
   7 GNU Emacs is free software; you can redistribute it and/or modify
   8 it under the terms of the GNU General Public License as published by
   9 the Free Software Foundation; either version 2, or (at your option)
  10 any later version.
  11
  12 GNU Emacs is distributed in the hope that it will be useful,
  13 but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 GNU General Public License for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with GNU Emacs; see the file COPYING.  If not, write to
  19 the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20 Boston, MA 02111-1307, USA.  */
  21
  22 /* Synched up with: FSF 19.31. */
  23
  24 \f
  25 #include <config.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <unistd.h>
  29 #include <sys/types.h>
  30 #include <sys/file.h>
  31 #include <sys/stat.h>
  32 #include <sys/mman.h>
  33 #include <stdio.h>
  34 #include <errno.h>
  35 #include <varargs.h>
  36 #include <filehdr.h>
  37 #include <aouthdr.h>
  38 #include <scnhdr.h>
  39 #include <syms.h>
  40
  41 static void fatal_unexec (char *, char *);
  42 static void mark_x (char *);
  43
  44 #define READ(_fd, _buffer, _size, _error_message, _error_arg) \
  45         errno = EEOF; \
  46         if (read (_fd, _buffer, _size) != _size) \
  47           fatal_unexec (_error_message, _error_arg);
  48
  49 #define WRITE(_fd, _buffer, _size, _error_message, _error_arg) \
  50         if (write (_fd, _buffer, _size) != _size) \
  51           fatal_unexec (_error_message, _error_arg);
  52
  53 #define SEEK(_fd, _position, _error_message, _error_arg) \
  54         errno = EEOF; \
  55         if (lseek (_fd, _position, L_SET) != _position) \
  56           fatal_unexec (_error_message, _error_arg);
  57
  58 #define EEOF -1
  59
  60 static struct scnhdr *text_section;
  61 static struct scnhdr *init_section;
  62 static struct scnhdr *finit_section;
  63 static struct scnhdr *rdata_section;
  64 static struct scnhdr *rconst_section;
  65 static struct scnhdr *data_section;
  66 static struct scnhdr *pdata_section;
  67 static struct scnhdr *xdata_section;
  68 static struct scnhdr *got_section;
  69 static struct scnhdr *lit8_section;
  70 static struct scnhdr *lit4_section;
  71 static struct scnhdr *sdata_section;
  72 static struct scnhdr *sbss_section;
  73 static struct scnhdr *bss_section;
  74
  75 static unsigned long Brk;
  76
  77 struct headers {
  78     struct filehdr fhdr;
  79     struct aouthdr aout;
  80     struct scnhdr section[_MIPS_NSCNS_MAX];
  81 };
  82
  83
  84 /* Define name of label for entry point for the dumped executable.  */
  85
  86 #ifndef DEFAULT_ENTRY_ADDRESS
  87 #define DEFAULT_ENTRY_ADDRESS __start
  88 #endif
  89 EXTERN_C int DEFAULT_ENTRY_ADDRESS (void);
  90
  91 \f
  92 int
  93 unexec (char *new_name, char *a_name,
  94         unsigned long data_start,
  95         unsigned long bss_start,
  96         unsigned long entry_address)
  97 {
  98   int new, old;
  99   char * oldptr;
 100   struct headers ohdr, nhdr;
 101   struct stat stat;
 102   long pagesize, brk;
 103   long newsyms, symrel;
 104   int i;
 105   long vaddr, scnptr;
 106 #define BUFSIZE 8192
 107   char buffer[BUFSIZE];
 108
 109   if ((old = open (a_name, O_RDONLY)) < 0)
 110     fatal_unexec ("opening %s", a_name);
 111
 112   new = creat (new_name, 0666);
 113   if (new < 0) fatal_unexec ("creating %s", new_name);
 114
 115   if ((fstat (old, &stat) == -1))
 116     fatal_unexec ("fstat %s", a_name);
 117
 118   oldptr = (char *)mmap (0, stat.st_size, PROT_READ, MAP_FILE|MAP_SHARED, old, 0);
 119
 120   if (oldptr == (char *)-1)
 121     fatal_unexec ("mmap %s", a_name);
 122
 123   close (old);
 124
 125   /* This is a copy of the a.out header of the original executable */
 126
 127   ohdr = (*(struct headers *)oldptr);
 128
 129   /* This is where we build the new header from the in-memory copy */
 130
 131   nhdr = *((struct headers *)TEXT_START);
 132
 133   /* First do some consistency checks */
 134
 135   if (nhdr.fhdr.f_magic != ALPHAMAGIC
 136       && nhdr.fhdr.f_magic != ALPHAUMAGIC)
 137     {
 138       fprintf (stderr, "unexec: input file magic number is %x, not %x or %x.\n",
 139                nhdr.fhdr.f_magic, ALPHAMAGIC, ALPHAUMAGIC);
 140       exit (1);
 141     }
 142
 143   if (nhdr.fhdr.f_opthdr != sizeof (nhdr.aout))
 144     {
 145       fprintf (stderr, "unexec: input a.out header is %d bytes, not %ld.\n",
 146                nhdr.fhdr.f_opthdr, (long) (sizeof (nhdr.aout)));
 147       exit (1);
 148     }
 149   if (nhdr.aout.magic != ZMAGIC)
 150     {
 151       fprintf (stderr, "unexec: input file a.out magic number is %o, not %o.\n",
 152                nhdr.aout.magic, ZMAGIC);
 153       exit (1);
 154     }
 155
 156
 157   /* Now check the existence of certain header section and grab
 158      their addresses. */
 159
 160 #define CHECK_SCNHDR(ptr, name, flags)                                  \
 161   ptr = NULL;                                                           \
 162   for (i = 0; i < nhdr.fhdr.f_nscns && !ptr; i++)                       \
 163     if (strcmp (nhdr.section[i].s_name, name) == 0)                     \
 164       {                                                                 \
 165         if (nhdr.section[i].s_flags != flags)                           \
 166           fprintf (stderr, "unexec: %x flags (%x expected) in %s section.\n", \
 167                    nhdr.section[i].s_flags, flags, name);               \
 168         ptr = nhdr.section + i;                                         \
 169       }                                                                 \
 170
 171   CHECK_SCNHDR (text_section,  _TEXT,  STYP_TEXT);
 172   CHECK_SCNHDR (init_section,  _INIT,  STYP_INIT);
 173 #ifdef _FINI
 174   CHECK_SCNHDR (finit_section, _FINI,  STYP_FINI);
 175 #endif /* _FINI */
 176   CHECK_SCNHDR (rdata_section, _RDATA, STYP_RDATA);
 177 #ifdef _RCONST
 178   CHECK_SCNHDR (rconst_section, _RCONST, STYP_RCONST);
 179 #endif
 180 #ifdef _PDATA
 181   CHECK_SCNHDR (pdata_section, _PDATA, STYP_PDATA);
 182 #endif /* _PDATA */
 183 #ifdef _GOT
 184   CHECK_SCNHDR (got_section,   _GOT,   STYP_GOT);
 185 #endif /* _GOT */
 186   CHECK_SCNHDR (data_section,  _DATA,  STYP_DATA);
 187 #ifdef _XDATA
 188   CHECK_SCNHDR (xdata_section, _XDATA, STYP_XDATA);
 189 #endif /* _XDATA */
 190 #ifdef _LIT8
 191   CHECK_SCNHDR (lit8_section,  _LIT8,  STYP_LIT8);
 192   CHECK_SCNHDR (lit4_section,  _LIT4,  STYP_LIT4);
 193 #endif /* _LIT8 */
 194   CHECK_SCNHDR (sdata_section, _SDATA, STYP_SDATA);
 195   CHECK_SCNHDR (sbss_section,  _SBSS,  STYP_SBSS);
 196   CHECK_SCNHDR (bss_section,   _BSS,   STYP_BSS);
 197
 198
 199   pagesize = getpagesize ();
 200   brk = (((long) (sbrk (0))) + pagesize - 1) & (-pagesize);
 201
 202   /* Remember the current break */
 203
 204   Brk = brk;
 205
 206   nhdr.aout.dsize = brk - DATA_START;
 207   nhdr.aout.bsize = 0;
 208   if (entry_address == 0)
 209     {
 210       nhdr.aout.entry = (unsigned long)DEFAULT_ENTRY_ADDRESS;
 211     }
 212   else
 213     nhdr.aout.entry = entry_address;
 214
 215   nhdr.aout.bss_start = nhdr.aout.data_start + nhdr.aout.dsize;
 216
 217   if (rdata_section != NULL)
 218     {
 219       rdata_section->s_size = data_start - DATA_START;
 220
 221       /* Adjust start and virtual addresses of rdata_section, too.  */
 222       rdata_section->s_vaddr = DATA_START;
 223       rdata_section->s_paddr = DATA_START;
 224       rdata_section->s_scnptr = text_section->s_scnptr + nhdr.aout.tsize;
 225     }
 226
 227   data_section->s_vaddr = data_start;
 228   data_section->s_paddr = data_start;
 229   data_section->s_size = brk - data_start;
 230
 231   if (rdata_section != NULL)
 232     {
 233       data_section->s_scnptr = rdata_section->s_scnptr + rdata_section->s_size;
 234     }
 235
 236   vaddr = data_section->s_vaddr + data_section->s_size;
 237   scnptr = data_section->s_scnptr + data_section->s_size;
 238   if (lit8_section != NULL)
 239     {
 240       lit8_section->s_vaddr = vaddr;
 241       lit8_section->s_paddr = vaddr;
 242       lit8_section->s_size = 0;
 243       lit8_section->s_scnptr = scnptr;
 244     }
 245   if (lit4_section != NULL)
 246     {
 247       lit4_section->s_vaddr = vaddr;
 248       lit4_section->s_paddr = vaddr;
 249       lit4_section->s_size = 0;
 250       lit4_section->s_scnptr = scnptr;
 251     }
 252   if (sdata_section != NULL)
 253     {
 254       sdata_section->s_vaddr = vaddr;
 255       sdata_section->s_paddr = vaddr;
 256       sdata_section->s_size = 0;
 257       sdata_section->s_scnptr = scnptr;
 258     }
 259 #ifdef _XDATA
 260   if (xdata_section != NULL)
 261     {
 262       xdata_section->s_vaddr = vaddr;
 263       xdata_section->s_paddr = vaddr;
 264       xdata_section->s_size = 0;
 265       xdata_section->s_scnptr = scnptr;
 266     }
 267 #endif
 268 #ifdef _GOT
 269   if (got_section != NULL)
 270     {
 271       got_section->s_vaddr = vaddr;
 272       got_section->s_paddr = vaddr;
 273       got_section->s_size = 0;
 274       got_section->s_scnptr = scnptr;
 275     }
 276 #endif /*_GOT */
 277   if (sbss_section != NULL)
 278     {
 279       sbss_section->s_vaddr = vaddr;
 280       sbss_section->s_paddr = vaddr;
 281       sbss_section->s_size = 0;
 282       sbss_section->s_scnptr = scnptr;
 283     }
 284   if (bss_section != NULL)
 285     {
 286       bss_section->s_vaddr = vaddr;
 287       bss_section->s_paddr = vaddr;
 288       bss_section->s_size = 0;
 289       bss_section->s_scnptr = scnptr;
 290     }
 291
 292   WRITE (new, (char *)TEXT_START, nhdr.aout.tsize,
 293          "writing text section to %s", new_name);
 294   WRITE (new, (char *)DATA_START, nhdr.aout.dsize,
 295          "writing data section to %s", new_name);
 296
 297
 298   /*
 299    * Construct new symbol table header
 300    */
 301
 302   memcpy (buffer, oldptr + nhdr.fhdr.f_symptr, cbHDRR);
 303
 304 #define symhdr ((pHDRR)buffer)
 305   newsyms = nhdr.aout.tsize + nhdr.aout.dsize;
 306   symrel = newsyms - nhdr.fhdr.f_symptr;
 307   nhdr.fhdr.f_symptr = newsyms;
 308   symhdr->cbLineOffset += symrel;
 309   symhdr->cbDnOffset += symrel;
 310   symhdr->cbPdOffset += symrel;
 311   symhdr->cbSymOffset += symrel;
 312   symhdr->cbOptOffset += symrel;
 313   symhdr->cbAuxOffset += symrel;
 314   symhdr->cbSsOffset += symrel;
 315   symhdr->cbSsExtOffset += symrel;
 316   symhdr->cbFdOffset += symrel;
 317   symhdr->cbRfdOffset += symrel;
 318   symhdr->cbExtOffset += symrel;
 319
 320   WRITE (new, buffer, cbHDRR, "writing symbol table header of %s", new_name);
 321
 322   /*
 323    * Copy the symbol table and line numbers
 324    */
 325   WRITE (new, oldptr + ohdr.fhdr.f_symptr + cbHDRR,
 326          stat.st_size - ohdr.fhdr.f_symptr - cbHDRR,
 327          "writing symbol table of %s", new_name);
 328
 329 #if 0
 330
 331 /* Not needed for now */
 332
 333   update_dynamic_symbols (oldptr, new_name, new, newsyms,
 334                           ((pHDRR) (oldptr + ohdr.fhdr.f_symptr))->issExtMax,
 335                           ((pHDRR) (oldptr + ohdr.fhdr.f_symptr))->cbExtOffset,
 336                           ((pHDRR) (oldptr + ohdr.fhdr.f_symptr))->cbSsExtOffset);
 337
 338 #endif
 339
 340 #undef symhdr
 341
 342   SEEK (new, 0, "seeking to start of header in %s", new_name);
 343   WRITE (new, &nhdr, sizeof (nhdr),
 344          "writing header of %s", new_name);
 345
 346   close (old);
 347   close (new);
 348   mark_x (new_name);
 349   return 0;
 350 }
 351
 352
 353 #if 0
 354
 355 /* Not needed for now */
 356
 357 /* The following function updates the values of some symbols
 358    that are used by the dynamic loader:
 359
 360    _edata
 361    _end
 362
 363 */
 364
 365 int
 366 update_dynamic_symbols (
 367      char *old,                 /* Pointer to old executable */
 368      char *new_name,            /* Name of new executable */
 369      int new,                   /* File descriptor for new executable */
 370      long newsyms,              /* Offset of Symbol table in new executable */
 371      int nsyms,                 /* Number of symbol table entries */
 372      long symoff,               /* Offset of External Symbols in old file */
 373      long stroff)               /* Offset of string table in old file */
 374 {
 375   long i;
 376   int found = 0;
 377   EXTR n_end, n_edata;
 378
 379   /* We go through the symbol table entries until we have found the two
 380      symbols. */
 381
 382   /* cbEXTR is the size of an external symbol table entry */
 383
 384   for (i = 0; i < nsyms && found < 2; i += cbEXTR)
 385     {
 386       REGISTER pEXTR x = (pEXTR) (old + symoff + i);
 387       char *s;
 388
 389       s = old + stroff + x->asym.iss; /* name of the symbol */
 390
 391       if (!strcmp(s,"_edata"))
 392         {
 393           found++;
 394           memcpy (&n_edata, x, cbEXTR);
 395           n_edata.asym.value = Brk;
 396           SEEK (new, newsyms + cbHDRR + i,
 397                 "seeking to symbol _edata in %s", new_name);
 398           WRITE (new, &n_edata, cbEXTR,
 399                  "writing symbol table entry for _edata into %s", new_name);
 400         }
 401       else if (!strcmp(s,"_end"))
 402         {
 403           found++;
 404           memcpy (&n_end, x, cbEXTR);
 405           n_end.asym.value = Brk;
 406           SEEK (new, newsyms + cbHDRR + i,
 407                 "seeking to symbol _end in %s", new_name);
 408           WRITE (new, &n_end, cbEXTR,
 409                  "writing symbol table entry for _end into %s", new_name);
 410         }
 411     }
 412
 413 }
 414
 415 #endif
 416
 417 \f
 418 /*
 419  * mark_x
 420  *
 421  * After successfully building the new a.out, mark it executable
 422  */
 423
 424 static void
 425 mark_x (char *name)
 426 {
 427   struct stat sbuf;
 428   int um = umask (777);
 429   umask (um);
 430   if (stat (name, &sbuf) < 0)
 431     fatal_unexec ("getting protection on %s", name);
 432   sbuf.st_mode |= 0111 & ~um;
 433   if (chmod (name, sbuf.st_mode) < 0)
 434     fatal_unexec ("setting protection on %s", name);
 435 }
 436
 437 static void
 438 fatal_unexec (char *s, char *arg)
 439 {
 440   if (errno == EEOF)
 441     fputs ("unexec: unexpected end of file, ", stderr);
 442   else
 443     fprintf (stderr, "unexec: %s, ", strerror (errno));
 444   fprintf (stderr, s, arg);
 445   fputs (".\n", stderr);
 446   exit (1);
 447 }