/*
* This program was advertised on unix-wizards.  I have had such a large
* response I'm sending it out to the world.
*
* Here is the source.  It works fine under 4.1bsd, I see no fundamental
* reason why it shouldn't work on an 11. (Except possibly small format
* changes in exec header or user structure.)  No documentation yet.
* Usage is
*        undump new-a.out-file [old-a.out-file] [core-file]
* where old-a.out-file and core-file default to "a.out" and "core",
* respectively.  Probably should have an option to not require
* old-a.out-file if the core came from a 407 file.
*
* It doesn't preserve open files, and the program is re-entered at main
* when you run it.  It's used locally to dump a lisp and restart it.
*
* It requires a local subroutine called scanargs, somewhat similar to
* getopt (I think).  You should be able to easily get around this, though.
* =Spencer
*/


/*
*      Barry Shein ([email protected]) - Add SUN/3 changes, tried and
*      worked on TeX but not certain about OMAGIC files. Build it
*      on a SUN/3 with -Dsun3 on the command line. (1/26/86)
*
*
*      Believed to work on a Sun2 with O.S. 3.2 as well.  If it
*      doesn't, try the 4.2undump.c (P. A. MacKay. (2/23/86)
*/
/*
* undump.c - Convert a core file to an a.out.
*
* Author:      Spencer W. Thomas
*              Computer Science Dept.
*              University of Utah
* Date:        Wed Feb 17 1982
* Copyright (c) 1982 Spencer W. Thomas
*
* Usage:
* undump new-a.out [a.out] [core]
*/

#include <stdio.h>
#include <sys/param.h>
#include <sys/dir.h>

#ifdef sun3
#include <sys/core.h>
#include <machine/param.h>
#else
#include <sys/user.h>
#endif

#include <sys/stat.h>
#include <a.out.h>

#ifdef sun3
#define PSIZE       NBPG
#else
#define PSIZE       10240
#endif sun3

#define TEXITUDE

#ifdef sun3
struct core c ;
#else
struct user u;
#endif

struct exec hdr, ohdr;

main(argc, argv)
char **argv;
{
   char *new_name, *a_out_name = "a.out", *core_name = "core";
   FILE *new, *a_out, *core;

   if (scanargs(argc, argv, "undump new-a.out!s a.out%s core%s",
           &new_name, &a_out_name, &core_name)
                   != 1)
       exit(1);

   if ((a_out = fopen(a_out_name, "r")) == NULL)
   {
       perror(a_out_name);
       exit(1);
   }
   if ((core = fopen(core_name, "r")) == NULL)
   {
       perror(core_name);
       exit(1);
   }
   if ((new = fopen(new_name, "w")) == NULL)
   {
       perror(new_name);
       exit(1);
   }

#ifdef sun3
   read_c(core);
#else
   read_u(core);
#endif
   make_hdr(new, a_out);
   copy_text(new, a_out);
   copy_data(new, core);
   copy_sym(new, a_out);
   fclose(new);
   fclose(core);
   fclose(a_out);
   mark_x(new_name);
}

#ifdef sun3
read_c(core) FILE *core;
{
 if(fread(&c, sizeof c, 1, core) != 1) {
   perror("Couldn't read core header");
   exit(1);
 }
 if(c.c_magic != CORE_MAGIC) {
   fprintf(stderr,"Not a core file\n");
   exit(1);
 }
}

#else
/*
* read the u structure from the core file.
*/
read_u(core)
FILE *core;
{
   if ( fread(&u, sizeof u, 1, core) != 1 )
   {
       perror("Couldn't read user structure from core file");
       exit(1);
   }
}
#endif sun3

/*
* Make the header in the new a.out from the header in the old one
* modified by the new data size.
*/
make_hdr(new, a_out)
FILE *new, *a_out;
{
   if (fread(&hdr, sizeof hdr, 1, a_out) != 1)
   {
       perror("Couldn't read header from a.out file");
       exit(1);
   }
   ohdr = hdr;
   if N_BADMAG(hdr)
   {
       fprintf(stderr, "a.out file doesn't have legal magic number\n");
       exit(1);
   }
#ifdef sun3
   if (hdr.a_magic != c.c_aouthdr.a_magic ||
       hdr.a_text  != c.c_aouthdr.a_text  ||
       hdr.a_data  != c.c_aouthdr.a_data  ||
       hdr.a_entry != c.c_aouthdr.a_entry)
#else
   if (hdr.a_magic != u.u_exdata.ux_mag ||
       hdr.a_text != u.u_exdata.ux_tsize ||
       hdr.a_data != u.u_exdata.ux_dsize ||
       hdr.a_entry != u.u_exdata.ux_entloc)
#endif sun3
   {
       fprintf(stderr, "Core file didn't come from this a.out\n");
#ifndef TEXITUDE
       exit(1);
#endif
   }

   printf("Data segment size was %u", hdr.a_data);
#ifdef sun3
   hdr.a_data = c.c_dsize;
#else
   hdr.a_data = ctob(u.u_dsize);
#endif sun3
   hdr.a_bss = 0;                      /* all data is inited now! */
   printf(" now is %u\n", hdr.a_data);
   if (fwrite(&hdr, sizeof hdr, 1, new) != 1)
   {
       perror("Couldn't write header to new a.out file");
       exit(1);
   }
}

/*
* Copy the text from the a.out to the new a.out
*/
copy_text(new, a_out)
FILE *new, *a_out;
{
   char page[PSIZE];
   int txtcnt = hdr.a_text;

#ifdef sun3
   fseek(new,(long) N_TXTOFF(hdr)+sizeof(hdr),0);
   fseek(a_out,(long) N_TXTOFF(hdr)+sizeof(hdr),0);
   txtcnt -= sizeof(hdr);
#else
   fseek(new, N_TXTOFF(hdr), 0);
   fseek(a_out, N_TXTOFF(hdr), 0);
#endif sun3
   if (hdr.a_magic == OMAGIC)
   {
       printf("a.out file is not shared text, getting text from core file\n");
       fseek(a_out, hdr.a_text, 1);    /* skip over text */
       return;
   }
   while (txtcnt >= PSIZE)
   {
       if (fread(page, PSIZE, 1, a_out) != 1)
       {
           perror("Read failure on a.out text");
           exit(1);
       }
       if (fwrite(page, PSIZE, 1, new) != 1)
       {
           perror("Write failure in text segment");
           exit(1);
       }
       txtcnt -= PSIZE;
   }
   if (txtcnt)
   {
       if (fread(page, txtcnt, 1, a_out) != 1)
       {
           perror("Read failure on a.out text");
           exit(1);
       }
       if (fwrite(page, txtcnt, 1, new) != 1)
       {
           perror("Write failure in text segment");
           exit(1);
       }
   }
}

/*
* copy the data from the core file to the new a.out
*/
copy_data(new, core)
FILE *new, *core;
{
   char page[PSIZE];
#ifdef sun3
   int datacnt = c.c_dsize;
#else
   int datacnt = ctob(u.u_dsize);
#endif sun3

   if (hdr.a_magic == OMAGIC)
#ifdef sun3
       datacnt += c.c_tsize;   /* BZS - not sure about this */
#else
       datacnt += u.u_tsize;
#endif

#ifndef sun3    /* for the sun3 the data should be right where we left off */
   fseek(core, ctob(UPAGES), 0);
#endif
   while (datacnt >= PSIZE)
   {
       if (fread(page, PSIZE, 1, core) != 1)
       {
           perror("Read failure on core data");
           exit(1);
       }
       if (fwrite(page, PSIZE, 1, new) != 1)
       {
           perror("Write failure in data segment");
           exit(1);
       }
       datacnt -= PSIZE;
   }
   if (datacnt)
   {
       if (fread(page, datacnt, 1, core) != 1)
       {
           perror("Read failure on core data");
           exit(1);
       }
       if (fwrite(page, datacnt, 1, new) != 1)
       {
           perror("Write failure in data segment");
           exit(1);
       }
   }
}

/*
* Copy the relocation information and symbol table from the a.out to the new
*/
copy_sym(new, a_out)
FILE *new, *a_out;
{
   char page[PSIZE];
   int n;

#ifdef sun3
   fseek(a_out, N_SYMOFF(ohdr), 0);
#else
   fseek(a_out, ohdr.a_data, 1);       /* skip over data segment */
#endif sun3
   while ((n = fread(page, 1, PSIZE, a_out)) > 0)
   {
       if (fwrite(page, 1, n, new) != n)
       {
           perror("Error writing symbol table to new a.out");
           fprintf(stderr, "new a.out should be ok otherwise\n");
           return;
       }
   }
   if (n < 0)
   {
       perror("Error reading symbol table from a.out");
       fprintf(stderr, "new a.out should be ok otherwise\n");
   }
}

/*
* After succesfully building the new a.out, mark it executable
*/
mark_x(name)
char *name;
{
   struct stat sbuf;
   int um;

   um = umask(777);
   umask(um);
   if (stat(name, &sbuf) == -1)
   {
       perror ("Can't stat new a.out");
       fprintf(stderr, "Setting protection to %o\n", 0777 & ~um);
       sbuf.st_mode = 0777;
   }
   sbuf.st_mode |= 0111 & ~um;
   if (chmod(name, sbuf.st_mode) == -1)
       perror("Couldn't change mode of new a.out to executable");

}