git.oblomov.eu Git - wine/blob - tools/winedump/dump.c

   1 /*
   2  *      File dumping utility
   3  *
   4  *      Copyright 2001,2005 Eric Pouech
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #include <stdlib.h>
  25 #include <stdarg.h>
  26 #include <stdio.h>
  27 #ifdef HAVE_UNISTD_H
  28 # include <unistd.h>
  29 #endif
  30 #include <time.h>
  31 #ifdef HAVE_SYS_TYPES_H
  32 # include <sys/types.h>
  33 #endif
  34 #ifdef HAVE_SYS_STAT_H
  35 # include <sys/stat.h>
  36 #endif
  37 #ifdef HAVE_SYS_MMAN_H
  38 #include <sys/mman.h>
  39 #endif
  40 #include <fcntl.h>
  41
  42 #define NONAMELESSUNION
  43 #define NONAMELESSSTRUCT
  44 #include "windef.h"
  45 #include "winbase.h"
  46 #include "winedump.h"
  47 #include "pe.h"
  48
  49 static void*                    dump_base;
  50 static unsigned long            dump_total_len;
  51
  52 void dump_data( const unsigned char *ptr, unsigned int size, const char *prefix )
  53 {
  54     unsigned int i, j;
  55
  56     printf( "%s%08x: ", prefix, 0 );
  57     if (!ptr)
  58     {
  59         printf("NULL\n");
  60         return;
  61     }
  62     for (i = 0; i < size; i++)
  63     {
  64         printf( "%02x%c", ptr[i], (i % 16 == 7) ? '-' : ' ' );
  65         if ((i % 16) == 15)
  66         {
  67             printf( " " );
  68             for (j = 0; j < 16; j++)
  69                 printf( "%c", isprint(ptr[i-15+j]) ? ptr[i-15+j] : '.' );
  70             if (i < size-1) printf( "\n%s%08x: ", prefix, i + 1 );
  71         }
  72     }
  73     if (i % 16)
  74     {
  75         printf( "%*s ", 3 * (16-(i%16)), "" );
  76         for (j = 0; j < i % 16; j++)
  77             printf( "%c", isprint(ptr[i-(i%16)+j]) ? ptr[i-(i%16)+j] : '.' );
  78     }
  79     printf( "\n" );
  80 }
  81
  82 const char*     get_time_str(unsigned long _t)
  83 {
  84     time_t      t = (time_t)_t;
  85     const char *str = ctime(&t);
  86     size_t len;
  87     static char buf[128];
  88
  89     if (!str) /* not valid time */
  90     {
  91         strcpy(buf, "not valid time");
  92         return buf;
  93     }
  94
  95     len = strlen(str);
  96     /* FIXME: I don't get the same values from MS' pedump running under Wine...
  97      * I wonder if Wine isn't broken wrt to GMT settings...
  98      */
  99     if (len && str[len-1] == '\n') len--;
 100     if (len >= sizeof(buf)) len = sizeof(buf) - 1;
 101     memcpy( buf, str, len );
 102     buf[len] = 0;
 103     return buf;
 104 }
 105
 106 unsigned int strlenW( const WCHAR *str )
 107 {
 108     const WCHAR *s = str;
 109     while (*s) s++;
 110     return s - str;
 111 }
 112
 113 void dump_unicode_str( const WCHAR *str, int len )
 114 {
 115     if (len == -1) len = strlenW( str );
 116     printf( "L\"");
 117     while (len-- > 0 && *str)
 118     {
 119         WCHAR c = *str++;
 120         switch (c)
 121         {
 122         case '\n': printf( "\\n" ); break;
 123         case '\r': printf( "\\r" ); break;
 124         case '\t': printf( "\\t" ); break;
 125         case '"':  printf( "\\\"" ); break;
 126         case '\\': printf( "\\\\" ); break;
 127         default:
 128             if (c >= ' ' && c <= 126) putchar(c);
 129             else printf( "\\u%04x",c);
 130         }
 131     }
 132     printf( "\"" );
 133 }
 134
 135 void*   PRD(unsigned long prd, unsigned long len)
 136 {
 137     return (prd + len > dump_total_len) ? NULL : (char*)dump_base + prd;
 138 }
 139
 140 unsigned long Offset(void* ptr)
 141 {
 142     if (ptr < dump_base) {printf("<<<<<ptr below\n");return 0;}
 143     if ((char *)ptr >= (char*)dump_base + dump_total_len) {printf("<<<<<ptr above\n");return 0;}
 144     return (char*)ptr - (char*)dump_base;
 145 }
 146
 147 static  void    do_dump( enum FileSig sig, void* pmt )
 148 {
 149     if (sig == SIG_NE)
 150     {
 151         ne_dump( dump_base, dump_total_len );
 152         return;
 153     }
 154
 155     if (sig == SIG_LE)
 156     {
 157         le_dump( dump_base, dump_total_len );
 158         return;
 159     }
 160
 161     pe_dump(pmt);
 162 }
 163
 164 static  enum FileSig    check_headers(void** pmt)
 165 {
 166     WORD*               pw;
 167     DWORD*              pdw;
 168     IMAGE_DOS_HEADER*   dh;
 169     enum FileSig        sig;
 170
 171     pw = PRD(0, sizeof(WORD));
 172     if (!pw) {printf("Can't get main signature, aborting\n"); return 0;}
 173
 174     *pmt = NULL;
 175     switch (*pw)
 176     {
 177     case IMAGE_DOS_SIGNATURE:
 178         sig = SIG_DOS;
 179         dh = PRD(0, sizeof(IMAGE_DOS_HEADER));
 180         if (dh && dh->e_lfanew >= sizeof(*dh)) /* reasonable DOS header ? */
 181         {
 182             /* the signature is the first DWORD */
 183             pdw = PRD(dh->e_lfanew, sizeof(DWORD));
 184             if (pdw)
 185             {
 186                 if (*pdw == IMAGE_NT_SIGNATURE)
 187                 {
 188                     *pmt = PRD(dh->e_lfanew, sizeof(DWORD)+sizeof(IMAGE_FILE_HEADER));
 189                     sig = SIG_PE;
 190                 }
 191                 else if (*(WORD *)pdw == IMAGE_OS2_SIGNATURE)
 192                 {
 193                     sig = SIG_NE;
 194                 }
 195                 else if (*(WORD *)pdw == IMAGE_VXD_SIGNATURE)
 196                 {
 197                     sig = SIG_LE;
 198                 }
 199                 else
 200                 {
 201                     printf("No PE Signature found\n");
 202                 }
 203             }
 204             else
 205             {
 206                 printf("Can't get the extented signature, aborting\n");
 207             }
 208         }
 209         break;
 210     case 0x4944: /* "DI" */
 211         sig = SIG_DBG;
 212         break;
 213     case 0x444D: /* "MD" */
 214         pdw = PRD(0, sizeof(DWORD));
 215         if (pdw && *pdw == 0x504D444D) /* "MDMP" */
 216             sig = SIG_MDMP;
 217         else
 218             sig = SIG_UNKNOWN;
 219         break;
 220     default:
 221         printf("No known main signature (%.2s/%x), aborting\n", (char*)pw, *pw);
 222         sig = SIG_UNKNOWN;
 223     }
 224
 225     return sig;
 226 }
 227
 228 int dump_analysis(const char* name, void (*fn)(enum FileSig, void*), enum FileSig wanted_sig)
 229 {
 230     int                 fd;
 231     enum FileSig        effective_sig;
 232     int                 ret = 1;
 233     struct stat         s;
 234     void*               pmt;
 235
 236     setbuf(stdout, NULL);
 237
 238     fd = open(name, O_RDONLY | O_BINARY);
 239     if (fd == -1) fatal("Can't open file");
 240
 241     if (fstat(fd, &s) < 0) fatal("Can't get size");
 242     dump_total_len = s.st_size;
 243
 244 #ifdef HAVE_MMAP
 245     if ((dump_base = mmap(NULL, dump_total_len, PROT_READ, MAP_PRIVATE, fd, 0)) == (void *)-1)
 246 #endif
 247     {
 248         if (!(dump_base = malloc( dump_total_len ))) fatal( "Out of memory" );
 249         if ((unsigned long)read( fd, dump_base, dump_total_len ) != dump_total_len) fatal( "Cannot read file" );
 250     }
 251
 252     effective_sig = check_headers(&pmt);
 253
 254     if (effective_sig == SIG_UNKNOWN)
 255     {
 256         printf("Can't get a recognized file signature, aborting\n");
 257         ret = 0;
 258     }
 259     else if (wanted_sig == SIG_UNKNOWN || wanted_sig == effective_sig)
 260     {
 261         switch (effective_sig)
 262         {
 263         case SIG_UNKNOWN: /* shouldn't happen... */
 264             ret = 0; break;
 265         case SIG_PE:
 266         case SIG_NE:
 267         case SIG_LE:
 268             printf("Contents of \"%s\": %ld bytes\n\n", name, dump_total_len);
 269             (*fn)(effective_sig, pmt);
 270             break;
 271         case SIG_DBG:
 272             dump_separate_dbg();
 273             break;
 274         case SIG_DOS:
 275             ret = 0; break;
 276         case SIG_MDMP:
 277             mdmp_dump();
 278             break;
 279         }
 280     }
 281     else
 282     {
 283         printf("Can't get a suitable file signature, aborting\n");
 284         ret = 0;
 285     }
 286
 287     if (ret) printf("Done dumping %s\n", name);
 288 #ifdef HAVE_MMAP
 289     if (munmap(dump_base, dump_total_len) == -1)
 290 #endif
 291     {
 292         free( dump_base );
 293     }
 294     close(fd);
 295
 296     return ret;
 297 }
 298
 299 void    dump_file(const char* name)
 300 {
 301     dump_analysis(name, do_dump, SIG_UNKNOWN);
 302 }