Mercurial > hg > xemacs-beta
diff src/unexcw.cc @ 245:51092a27c943 r20-5b21
Import from CVS: tag r20-5b21
author | cvs |
---|---|
date | Mon, 13 Aug 2007 10:17:54 +0200 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/unexcw.cc Mon Aug 13 10:17:54 2007 +0200 @@ -0,0 +1,606 @@ +/* unexec for GNU Emacs on Cygwin32. + Copyright (C) 1994, 1998 Free Software Foundation, Inc. + +This file is part of XEmacs. + +XEmacs 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, or (at your option) any +later version. + +XEmacs 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 XEmacs; see the file COPYING. If not, write to the Free +Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA +02111-1307, USA. + +*/ + +/* Adapted from unexnt.c Andy Piper (andyp@parallax.co.uk) 13-1-98 */ + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <fcntl.h> +#include <config.h> +#include <string.h> + +#define PERROR(arg) perror(arg);exit(-1) + +#ifndef HAVE_COFF_H +extern "C" void +unexec (char *, char *, void *, void *, void *) +{ + PERROR("cannot unexec() coff.h not installed"); +} + +extern "C" void run_time_remap (char *) +{} +#else + +#include <windows.h> +/* + * unfortunately we need this c++ to get at the internals of cygwin + */ +class pinfo; +class per_process +{ + public: + char *initial_sp; + + /* The offset of these 3 values can never change. */ + /* magic_biscuit is the size of this class and should never change. */ + int magic_biscuit; + int version_major; + int version_minor; + + struct _reent **impure_ptr_ptr; + char ***envptr; + + /* Used to point to the memory machine we should use - + usually points back into the dll, but can be overridden by the user. */ + void *(*malloc)(size_t); + void (*free)(void *); + void *(*realloc)(void *, size_t); + + int *fmode_ptr; + + int (*main)(int, char **, char **); + void (**ctors)(); + void (**dtors)(); + + /* For fork */ + void *data_start; + void *data_end; + void *bss_start; + void *bss_end; + + /* For future expansion of values set by the app. */ + void *public_reserved[4]; + + /* The rest are *internal* to cygwin.dll. + Those that are here because we want the child to inherit the value from + the parent (which happens when bss is copied) are marked as such. */ + + /* FIXME: Which of these can go elsewhere? */ + + /* FIXME: Delete, make `self' a global. */ + pinfo *self; /* pointer only valid in self process */ + + /* non-zero of ctors have been run. Inherited from parent. */ + int run_ctors_p; + + /* These will be non-zero if the above (malloc,free,realloc) have been + overridden. */ + /* FIXME: not currently used */ + int __imp_malloc; + int __imp_free; + int __imp_realloc; + + /* Heap management. Inherited from parent. */ + void *base; /* bottom of the heap */ + void *ptr; /* current index into heap */ + int size; /* current size of heap */ + + /* Mask of what to trace. Inherited from parent. + See sys/strace.h for details. The value of this is changeable from other + tasks via the `cygwin' utility so we want this in the shared data area + (and thus the process table since there's one of these per task). + However, we also want to turn on stracing as soon as possible and + therefore before we know which process table entry to use. So we put it + here, and have a pointer to it in the process table. */ + int strace_mask; + + /* Non-zero means the task was forked. The value is the pid. + Inherited from parent. */ + int forkee; + + void *hmodule; + + void* /*HANDLE*/ signal_arrived; + /* For future expansion, so apps won't have to be relinked if we + add an item. */ + void *internal_reserved[9]; + + /* struct file_queue *dq; !!! this may need to be nuked ? */ +}; + +extern per_process cygwin_statu; /* pointer into the application's static data */ + +#include <coff.h> + +#define ALLOC_UNIT 0xFFFF +#define ALLOC_MASK ~((unsigned long)(ALLOC_UNIT)) +#define ALIGN_ALLOC(addr) \ +((((unsigned long)addr) + ALLOC_UNIT) & ALLOC_MASK) +#define SIZEOF_PER_PROCESS (42 * 4) + +/* + * Heap related stuff. + */ +#define get_reserved_heap_size() (*heap_size) +#define get_committed_heap_size() \ +(int)((unsigned char*)(*heap_index)-(unsigned char*)(*heap_base)) +#define get_heap_start() (*heap_base) +#define get_heap_end() (*heap_index) + +extern "C" { +void** heap_base = &cygwin_statu.base; +void** heap_index = &cygwin_statu.ptr; +int* heap_size = &cygwin_statu.size; +int* heap_flag = &cygwin_statu.forkee; +void* per_process_data = &cygwin_statu; +/* To prevent zero-initialized variables from being placed into the bss + section, use non-zero values to represent an uninitialized state. */ +#define UNINIT_PTR ((void *) 0xF0A0F0A0) +#define UNINIT_LONG (0xF0A0F0A0L) + +void* local_heap_base=UNINIT_PTR; +void* local_heap_index=UNINIT_PTR; +unsigned long local_heap_size=UNINIT_LONG; + +/* Recreate the heap created during dumping. */ + +enum { + HEAP_UNINITIALIZED = 1, + HEAP_UNLOADED, + HEAP_LOADED +}; + +/* Basically, our "initialized" flag. */ +int heap_state = HEAP_UNINITIALIZED; + +/* So we can find our heap in the file to recreate it. */ +unsigned long heap_index_in_executable = UNINIT_LONG; + +static void get_section_info (int a_out, char* a_name); +static void copy_executable_and_dump_data_section (int a_out, int a_new); +static void dump_heap (int a_out, int a_new); +static void dup_file_area(int a_out, int a_new, long size); + +/* Cached info about the .data section in the executable. */ +void* data_start_va = UNINIT_PTR; +unsigned long data_size = UNINIT_LONG; + +/* Cached info about the .bss section in the executable. */ +void* bss_start = UNINIT_PTR; +unsigned long bss_size = UNINIT_LONG; +FILHDR f_hdr; +PEAOUTHDR f_ohdr; +SCNHDR f_data, f_bss, f_text, f_idata; +} +#define PERROR(arg) perror(arg);exit(-1) +#define CHECK_AOUT_POS(a) \ +if (lseek(a_out, 0, SEEK_CUR) != a) \ +{ \ + printf("we are at %lx, should be at %lx\n", \ + lseek(a_out, 0, SEEK_CUR), a); \ + exit(-1); \ +} + +/* Dump out .data and .bss sections into a new executable. */ +extern "C" void +unexec (char *out_name, char *in_name, void *start_data, void *, void *) +{ + /* ugly nt hack - should be in lisp */ + char new_name[MAX_PATH], a_name[MAX_PATH]; + char *ptr; + + /* Make sure that the input and output filenames have the + ".exe" extension...patch them up if they don't. */ + strcpy (a_name, in_name); + ptr = a_name + strlen (a_name) - 4; + if (strcmp (ptr, ".exe")) + strcat (a_name, ".exe"); + + strcpy (new_name, out_name); + ptr = new_name + strlen (new_name) - 4; + if (strcmp (ptr, ".exe")) + strcat (new_name, ".exe"); + /* save heap info in our data segment so that we can recreate after + dumping */ + + local_heap_base = *heap_base; + local_heap_size = *heap_size; + local_heap_index = *heap_index; + + /* We need to round off our heap to NT's allocation unit (64KB). */ + /* round_heap (get_allocation_unit ()); */ + + int a_new, a_out = -1; + + if (a_name && (a_out = open (a_name, O_RDONLY)) < 0) + { + PERROR (a_name); + } + if ((a_new = creat (new_name, 0666)) < 0) + { + PERROR (new_name); + } + + /* Get the interesting section info, like start and size of .bss... */ + get_section_info (a_out, a_name); + + /* Set the flag (before dumping). */ + heap_state = HEAP_UNLOADED; + + copy_executable_and_dump_data_section (a_out, a_new); + dump_heap (a_out, a_new); + + close(a_out); + close(a_new); +} + +/* Flip through the executable and cache the info necessary for dumping. */ +static void get_section_info (int a_out, char* a_name) +{ + if (read (a_out, &f_hdr, sizeof (f_hdr)) != sizeof (f_hdr)) + { + PERROR (a_name); + } + + if (f_hdr.e_magic != DOSMAGIC) + { + PERROR("unknown exe header"); + } + + /* Check the NT header signature ... */ + if (f_hdr.nt_signature != NT_SIGNATURE) + { + PERROR("invalid nt header"); + } + + /* Flip through the sections for .data and .bss ... */ + if (f_hdr.f_opthdr > 0) + { + if (read (a_out, &f_ohdr, AOUTSZ) != AOUTSZ) + { + PERROR (a_name); + } + } + /* Loop through .data & .bss section headers, copying them in */ + lseek (a_out, sizeof (f_hdr) + f_hdr.f_opthdr, 0); + + if (read (a_out, &f_text, sizeof (f_text)) != sizeof (f_text) + && + strcmp (f_text.s_name, ".text")) + { + PERROR ("no .text section"); + } + + /* The .bss section. */ + if (read (a_out, &f_bss, sizeof (f_bss)) != sizeof (f_bss) + && + strcmp (f_bss.s_name, ".bss")) + { + PERROR ("no .bss section"); + } + extern int my_ebss; + bss_start = (void *) ((char*)f_ohdr.ImageBase + f_bss.s_vaddr); + bss_size = (unsigned long)((char*)&my_ebss-(char*)bss_start); + + /* must keep bss data that we want to be blank as blank */ + printf("found bss - keeping %lx of %lx bytes\n", bss_size, f_ohdr.bsize); + + /* The .data section. */ + if (read (a_out, &f_data, sizeof (f_data)) != sizeof (f_data) + && + strcmp (f_data.s_name, ".data")) + { + PERROR ("no .data section"); + } + + /* From lastfile.c */ + extern char my_edata[]; + + /* The .data section. */ + data_start_va = (void *) ((char*)f_ohdr.ImageBase + f_data.s_vaddr); + + /* We want to only write Emacs data back to the executable, + not any of the library data (if library data is included, + then a dumped Emacs won't run on system versions other + than the one Emacs was dumped on). */ + data_size = my_edata - data_start_va; + + /* The .idata section. */ + if (read (a_out, &f_idata, sizeof (f_idata)) != sizeof (f_idata) + && + strcmp (f_idata.s_name, ".idata")) + { + PERROR ("no .idata section"); + } +} + +/* The dump routines. */ + +static void +copy_executable_and_dump_data_section (int a_out, int a_new) +{ + long size=0; + unsigned long new_data_size, new_bss_size, f_data_s_vaddr, + file_sz_change, f_data_s_scnptr, bss_padding; + int i; + SCNHDR section; + /* calculate new sizes f_ohdr.dsize is the total initalized data + size on disk which is f_data.s_size + f_idata.s_size. + f_ohdr.data_start is the base addres of all data and so should + not be changed. *.s_vaddr is the virtual address of the start + of the section normalzed from f_ohdr.ImageBase. *.s_paddr + appears to be the number of bytes in the section actually used + (whereas *.s_size is aligned). + + bsize is now 0 since subsumed into .data + dsize is dsize + (f_data.s_vaddr - f_bss.s_vaddr) + f_data.s_vaddr is f_bss.s_vaddr + f_data.s_size is new dsize maybe. + what about s_paddr & s_scnptr? */ + /* this is the amount the file increases in size */ + *heap_flag=1; // kludge to get mem to remap + new_bss_size=f_data.s_vaddr - f_bss.s_vaddr; + file_sz_change=new_bss_size; + new_data_size=f_ohdr.dsize + new_bss_size; + f_data_s_scnptr = f_data.s_scnptr; + f_data_s_vaddr = f_data.s_vaddr; + f_data.s_vaddr = f_bss.s_vaddr; + f_data.s_paddr += new_bss_size; + + if (f_data.s_size + f_idata.s_size != f_ohdr.dsize) + { + printf("section size doesn't tally with dsize %lx != %lx\n", + f_data.s_size + f_idata.s_size, f_ohdr.dsize); + } + f_data.s_size += new_bss_size; + lseek (a_new, 0, SEEK_SET); + /* write file header */ + f_hdr.f_symptr += file_sz_change; + f_hdr.f_nscns--; + printf("writing file header\n"); + if (write(a_new, &f_hdr, sizeof(f_hdr)) != sizeof(f_hdr)) + { + PERROR("failed to write file header"); + } + /* write optional header fixing dsize & bsize*/ + printf("writing optional header\n"); + printf("new data size is %lx, >= %lx\n", new_data_size, + f_ohdr.dsize + f_ohdr.bsize); + if (new_data_size < f_ohdr.dsize + f_ohdr.bsize ) + { + PERROR("new data size is < approx"); + } + f_ohdr.dsize=new_data_size; + f_ohdr.bsize=0; + if (write(a_new, &f_ohdr, sizeof(f_ohdr)) != sizeof(f_ohdr)) + { + PERROR("failed to write optional header"); + } + /* write text as is */ + printf("writing text header (unchanged)\n"); + + if (write(a_new, &f_text, sizeof(f_text)) != sizeof(f_text)) + { + PERROR("failed to write text header"); + } + + /* write new data header */ + printf("writing .data header\n"); + + if (write(a_new, &f_data, sizeof(f_data)) != sizeof(f_data)) + { + PERROR("failed to write data header"); + } + + printf("writing .idata header\n"); + f_idata.s_scnptr += file_sz_change; + if (f_idata.s_lnnoptr != 0) f_idata.s_lnnoptr += file_sz_change; + if (f_idata.s_relptr != 0) f_idata.s_relptr += file_sz_change; + if (write(a_new, &f_idata, sizeof(f_idata)) != sizeof(f_idata)) + { + PERROR("failed to write idata header"); + } + + /* copy other section headers adjusting the file offset */ + for (i=0; i<(f_hdr.f_nscns-3); i++) + { + if (read (a_out, §ion, sizeof (section)) != sizeof (section)) + { + PERROR ("no .data section"); + } + + section.s_scnptr += file_sz_change; + if (section.s_lnnoptr != 0) section.s_lnnoptr += file_sz_change; + if (section.s_relptr != 0) section.s_relptr += file_sz_change; + + if (write(a_new, §ion, sizeof(section)) != sizeof(section)) + { + PERROR("failed to write data header"); + } + } + + /* dump bss to maintain offsets */ + memset(&f_bss, 0, sizeof(f_bss)); + if (write(a_new, &f_bss, sizeof(f_bss)) != sizeof(f_bss)) + { + PERROR("failed to write bss header"); + } + + + size=lseek(a_new, 0, SEEK_CUR); + CHECK_AOUT_POS(size); + + /* copy eveything else until start of data */ + size = f_data_s_scnptr - lseek (a_out, 0, SEEK_CUR); + + printf ("copying executable up to data section ... %lx bytes\n", + size); + dup_file_area(a_out, a_new, size); + + CHECK_AOUT_POS(f_data_s_scnptr); + + /* dump bss + padding between sections */ + printf ("dumping .bss into executable... %lx bytes\n", bss_size); + if (write(a_new, bss_start, bss_size) != (int)bss_size) + { + PERROR("failed to write bss section"); + } + /* pad, needs to be zero */ + bss_padding = new_bss_size - bss_size; + printf ("padding .bss ... %lx bytes\n", bss_padding); + void* empty_space = malloc(bss_padding); + memset(empty_space, 0, bss_padding); + if (write(a_new, empty_space, bss_padding) != (int)bss_padding) + { + PERROR("failed to write bss section"); + } + free(empty_space); + + /* Get a pointer to the raw data in our address space. */ + printf ("dumping .data section... %lx bytes\n", data_size); + if (write(a_new, data_start_va, data_size) != (int)data_size) + { + PERROR("failed to write data section"); + } + + lseek(a_out, f_data_s_scnptr + data_size, SEEK_SET); + + size = (((unsigned long)per_process_data-f_ohdr.ImageBase)-f_data_s_vaddr) + - data_size; + /* write rest of .data to cygwin per process */ + printf ("copying from .data to cygwin per_process... %lx bytes\n", size); + dup_file_area(a_out, a_new, size); + + /* now write out the per process information */ + printf ("dumping to cygwin per_process... %x bytes at %p\n", + SIZEOF_PER_PROCESS, per_process_data); + + per_process newpp; + memset(&newpp, 0, SIZEOF_PER_PROCESS); + newpp.base = cygwin_statu.base; + newpp.ptr = cygwin_statu.ptr; + newpp.size = cygwin_statu.size; + + if (write(a_new, &newpp, SIZEOF_PER_PROCESS) + != (int)SIZEOF_PER_PROCESS) + { + PERROR("failed to write per_process info"); + } + free(empty_space); + + /* dump the rest */ + size = lseek(a_out, SIZEOF_PER_PROCESS, SEEK_CUR); + size = f_idata.s_scnptr - size; + dup_file_area(a_out, a_new, size); + + // lseek(a_out, f_idata.s_scnptr, SEEK_CUR); + CHECK_AOUT_POS(f_idata.s_scnptr); + /* now dump - idata don't need to do this cygwin ds is in .data! */ + printf ("dumping .idata section... %lx bytes\n", f_idata.s_size); + + dup_file_area(a_out,a_new,f_idata.s_size); + + /* write rest of file */ + printf ("writing rest of file\n"); + size = lseek(a_out, 0, SEEK_END); + size = size - (f_idata.s_scnptr + f_idata.s_size); /* length remaining in a_out */ + lseek(a_out, f_idata.s_scnptr + f_idata.s_size, SEEK_SET); + + dup_file_area(a_out, a_new, size); +} + +/* + * copy from aout to anew + */ +static void dup_file_area(int a_out, int a_new, long size) +{ + char page[BUFSIZ]; + long n; + for (; size > 0; size -= sizeof (page)) + { + n = size > sizeof (page) ? sizeof (page) : size; + if (read (a_out, page, n) != n || write (a_new, page, n) != n) + { + PERROR ("dump_out()"); + } + } +} + +static void dump_heap (int a_out, int a_new) +{ + void *heap_data; + unsigned long heap_size; + + printf ("Dumping heap into executable...\n"); + + heap_size = get_committed_heap_size (); + heap_data = get_heap_start (); + + printf ("heap start in process - %p \n", heap_data); + printf ("heap size in bytes %lx\n", heap_size); + + /* nt version rounds heap start - don't see why we should */ + heap_index_in_executable = lseek(a_new, 0, SEEK_CUR); + + if (write(a_new, heap_data, heap_size) != (int)heap_size) + { + PERROR("failed to write data section"); + } +} + +extern "C" void run_time_remap (char *a_name) +{ + int a_out=-1; + + if ((*heap_base)!=local_heap_base + ||(*heap_index)<local_heap_index + ||(*heap_size)<local_heap_size) + { + PERROR("heap parameters not in bss"); + } + + if (a_name && (a_out = open (a_name, O_RDONLY)) < 0) + { + PERROR (a_name); + } + + /* load the heap */ + lseek(a_out, heap_index_in_executable, SEEK_SET); + + if (read (a_out, get_heap_start(), + (int)((unsigned char*)(local_heap_index) + -(unsigned char*)(local_heap_base))) < 0) + { + PERROR (a_name); + } + close(a_out); + + /* switch to new heap */ + heap_state=HEAP_LOADED; + *heap_flag=0; + + close(a_out); +} + +#endif /* HAVE_COFF_H */