Quelltext /~heha/hs/borg.zip/DLG_LDOP.CPP

/************************************************************************
*		   dlg_ldopt.cpp					*
* Contains the dialog routines for the load file dialogboxes		*
************************************************************************/

#include <windows.h>
#include <stdio.h>		//sscanf...


#include "resource.h"
#include "exeload.h"
#include "proctab.h"
#include "dasm.h"
#include "disasm.h"
#include "help.h"
#include "debug.h"
#include "data.h"
//#include "gname.h"
//#include "relocs.h"
//#include "xref.h"

/************************************************************************
* moreoptions								*
* - advanced loading options						*
************************************************************************/
BOOL CALLBACK moreoptions(HWND Wnd,UINT Msg,WPARAM wParam,LPARAM) {

 switch (Msg) {
  case WM_INITDIALOG: {
   CheckDlgButton(Wnd,advanced_pushbp,    options.codedetect&CD_PUSHBP);
   CheckDlgButton(Wnd,advanced_aggressive,options.codedetect&CD_AGGRESSIVE);
   CheckDlgButton(Wnd,advanced_enter,     options.codedetect&CD_ENTER);
   CheckDlgButton(Wnd,advanced_movbx,     options.codedetect&CD_MOVBX);
   CheckDlgButton(Wnd,advanced_moveax,    options.codedetect&CD_MOVEAX);
   CheckDlgButton(Wnd,advanced_eaxfromesp,options.codedetect&CD_EAXFROMESP);
  }return TRUE;
  case WM_COMMAND: switch (LOWORD(wParam)) {
   case IDOK: {
    options.codedetect=0;
    if (IsDlgButtonChecked(Wnd,advanced_pushbp))     options.codedetect|=CD_PUSHBP;
    if (IsDlgButtonChecked(Wnd,advanced_aggressive)) options.codedetect|=CD_AGGRESSIVE;
    if (IsDlgButtonChecked(Wnd,advanced_enter))      options.codedetect|=CD_ENTER;
    if (IsDlgButtonChecked(Wnd,advanced_movbx))      options.codedetect|=CD_MOVBX;
    if (IsDlgButtonChecked(Wnd,advanced_moveax))     options.codedetect|=CD_MOVEAX;
    if (IsDlgButtonChecked(Wnd,advanced_eaxfromesp)) options.codedetect|=CD_EAXFROMESP;
   }nobreak;
   case IDCANCEL: EndDialog(Wnd,wParam);
  }
 }
 return FALSE;
}

/************************************************************************
* checktypebox								*
* - after a file has been chosen to load and before the file is loaded	*
*   this is displayed for the user to set options for analysis, file	*
*   type, etc.								*
************************************************************************/
//dialog proc for verifying type and
// initial options
BOOL CALLBACK checktypebox(HWND Wnd,UINT Msg,WPARAM wParam,LPARAM lParam) {
 static int exetype;
 int i;
 TCHAR *db;	// Text for DefaultButton

 switch (Msg) {
  case WM_INITDIALOG: {
   exetype=floader.getexetype();
   i=0;
   options.loadaddr.segm=0x1000;
   options.loadaddr.offs=0;
   CheckDlgButton(Wnd,IDC_DEFAULTBUTTON,true);
   switch (exetype) {
    case NE_EXE:
     db=T("NE Executable");
     options.processor=PROC_80486;
     options.mode16=true;
    break;
    case COM_EXE:
     db=T("COM File");
     options.processor=PROC_80386;
     options.mode16=true;
     options.loadaddr.offs=0x100;
    break;
    case SYS_EXE:
     db=T("SYS File");
     options.processor=PROC_80386;
     options.mode16=true;
     options.loadaddr.offs=0x00;
    break;
    case PE_EXE:
     db=T("PE Executable");
     options.processor=PROC_PENTIUM;
     options.mode16=false;
    break;
    case OS2_EXE:
     db=T("OS2 Executable");
     options.processor=PROC_PENTIUM;
     options.mode16=false;
    break;
    case LE_EXE:
     db=T("LE Executable");
     options.processor=PROC_80486;
     options.mode16=false;
    break;
    case MZ_EXE:
     db=T("COM File");
     CheckRadioButton(Wnd,IDC_DEFAULTBUTTON,IDC_BINBUTTON,IDC_DOSBUTTON);
     options.processor=PROC_80386;
     options.mode16=true;
     break;
    case BIN_EXE:
    default:
     db=T("COM File");
     CheckRadioButton(Wnd,IDC_DEFAULTBUTTON,IDC_BINBUTTON,IDC_BINBUTTON);
     options.processor=PROC_8086;
     options.mode16=true;
    break;
   }
   SetDlgItemText(Wnd,IDC_DEFAULTBUTTON,db);

   options.mode32=!options.mode16;
   CheckDlgButton(Wnd,load_debug,options.loaddebug);
   CheckDlgButton(Wnd,demangle_names,options.demangle);
   CheckDlgButton(Wnd,IDC_16DASM,options.mode16);
   CheckDlgButton(Wnd,IDC_32DASM,options.mode32);
   CheckDlgButton(Wnd,IDC_LOADDATA,options.loaddata);
   CheckDlgButton(Wnd,IDC_LOADRESOURCES,options.loadresources);
   HWND w=GetDlgItem(Wnd,IDC_LISTBOX1);
   while (procnames[i].num) {
    SendMessage(w,LB_ADDSTRING,0,(LPARAM)(LPCTSTR)procnames[i].name);
    if(options.processor==procnames[i].num)
    SendMessage(w,LB_SETCURSEL,i,0);
    i++;
   }
   TCHAR s[20];
   wsprintf(s,"%x",options.loadaddr.segm);
   SetDlgItemText(Wnd,IDC_SEGEDIT,s);
   wsprintf(s,"%lx",options.loadaddr.offs);
   SetDlgItemText(Wnd,IDC_OFFSEDIT,s);
  }return TRUE;

  case WM_COMMAND: switch (LOWORD(wParam)) {
   case IDC_SEGEDIT: if (HIWORD(wParam)==EN_CHANGE) {
    TCHAR s[20];
    GetWindowText((HWND)lParam,s,elemof(s));
    sscanf(s,"%x",&options.loadaddr.segm);
   }break;
   case IDC_OFFSEDIT: if (HIWORD(wParam)==EN_CHANGE) {
    TCHAR s[20];
    GetWindowText((HWND)lParam,s,elemof(s));
    sscanf(s,"%lx",&options.loadaddr.offs);
   }break;
   case IDC_HELPBUTTON1: {
    DialogBox(hInst,MAKEINTRESOURCE(HELPDIALOG_1),Wnd,helpbox1);
   }break;
   case more_options: {
    DialogBox(hInst,MAKEINTRESOURCE(Advanced_Options),Wnd,moreoptions);
   }break;
   case load_debug: {
    options.loaddebug=!options.loaddebug;
    CheckDlgButton(Wnd,load_debug,options.loaddebug);
   }break;
   case demangle_names: {
    options.demangle=!options.demangle;
    CheckDlgButton(Wnd,demangle_names,options.demangle);
   }break;
   case IDC_16DASM: {
    options.mode16=!options.mode16;
    CheckDlgButton(Wnd,IDC_16DASM,options.mode16);
   }break;
   case IDC_32DASM: {
    options.mode32=!options.mode32;
    CheckDlgButton(Wnd,IDC_32DASM,options.mode32);
   }break;
   case IDC_LOADDATA: {
    options.loaddata=!options.loaddata;
    CheckDlgButton(Wnd,IDC_LOADDATA,options.loaddata);
   }break;
   case IDC_LOADRESOURCES: {
    options.loadresources=!options.loadresources;
    CheckDlgButton(Wnd,IDC_LOADRESOURCES,options.loadresources);
   }break;
   case IDOK: {
    if(!IsDlgButtonChecked(Wnd,IDC_DEFAULTBUTTON)) {
     if(IsDlgButtonChecked(Wnd,IDC_DOSBUTTON)) floader.setexetype(MZ_EXE);
     else floader.setexetype(BIN_EXE);
    }else if (exetype==BIN_EXE || exetype==MZ_EXE) floader.setexetype(COM_EXE);
    options.processor=procnames[SendDlgItemMessage(Wnd,IDC_LISTBOX1,LB_GETCURSEL,0,0)].num;
   }nobreak;
   case IDCANCEL: EndDialog(Wnd,wParam);
  }
 }
 return FALSE;
}

/************************************************************************
* load									*
* - checks file header info, identifies the possible types of files,	*
*   gets the users file loading options and calls the specific exe	*
*   format loading routines.						*
************************************************************************/
bool loadexefile(LPCTSTR fname) {
 char mzhead[2],exthead[2];
  dword pe_offset;
  dword num;
  dword fsize;
  if(floader.efile!=INVALID_HANDLE_VALUE)
    return false;

  // just grab the file size first
  floader.efile=CreateFile(fname,GENERIC_READ,1,NULL,OPEN_EXISTING,0,NULL);
  fsize=GetFileSize(floader.efile,NULL);
  CloseHandle(floader.efile);
  if(!fsize)
  { MessageBox(MainWnd,"File appears to be of zero length ???",
      "Borg Message",MB_OK);
    return false;
  }

  floader.efile=CreateFile(fname,GENERIC_READ|GENERIC_WRITE,1,NULL,OPEN_EXISTING,0,NULL);
  if(floader.efile==INVALID_HANDLE_VALUE)
  { floader.efile=CreateFile(fname,GENERIC_READ,1,NULL,OPEN_EXISTING,0,NULL);
    if(floader.efile==INVALID_HANDLE_VALUE)
      return false;
    options.readonly=true;
    MessageBox(MainWnd,"Couldn't obtain write permission to file\n"
      "File opened readonly - will not be able to apply any patches",
      "Borg Message",MB_OK);
  }
  if(GetFileType(floader.efile)!=FILE_TYPE_DISK)
    return false;
  floader.exetype=BIN_EXE;
  if(ReadFile(floader.efile,mzhead,2,&num,NULL))
  { if((num==2)&&(((mzhead[0]=='M')&&(mzhead[1]=='Z'))||
       ((mzhead[0]=='Z')&&(mzhead[1]=='M'))))
    { SetFilePointer(floader.efile,0x3c,NULL,FILE_BEGIN);
      if(ReadFile(floader.efile,&pe_offset,4,&num,NULL))
	SetFilePointer(floader.efile,pe_offset,NULL,FILE_BEGIN);
      if(ReadFile(floader.efile,exthead,2,&num,NULL))
      { if(((short int *)exthead)[0]==0x4550)
	  floader.exetype=PE_EXE;
	else if(((short int *)exthead)[0]==0x454e)
	  floader.exetype=NE_EXE;
	else if(((short int *)exthead)[0]==0x454c)
	  floader.exetype=LE_EXE;
	else if(((short int *)exthead)[0]==0x584c)
	  floader.exetype=OS2_EXE;
	else
	  floader.exetype=MZ_EXE;
      }
    }
    else
    { if(lstrlen(fname)>3)
      { if(!lstrcmpi(fname+lstrlen(fname)-3,"com"))
	{ SetFilePointer(floader.efile,0,NULL,FILE_BEGIN);
	  floader.exetype=COM_EXE;
	}
	else if(!lstrcmpi(fname+lstrlen(fname)-3,"sys"))
	{ SetFilePointer(floader.efile,0,NULL,FILE_BEGIN);
	  floader.exetype=SYS_EXE;
	}
      }
    }
  }
  floader.fbuff=new byte[fsize];
  SetFilePointer(floader.efile,0x00,NULL,FILE_BEGIN);
  ReadFile(floader.efile,floader.fbuff,fsize,&num,NULL);
  if (DialogBox(hInst,MAKEINTRESOURCE(D_checktype),MainWnd,checktypebox)
    !=IDOK) {
   CloseHandle(floader.efile);
   return false;
  }
  if (!options.loadaddr.segm) {
   options.loadaddr.segm=0x1000;
   MessageBox(MainWnd,"Sorry - Can't use a zero segment base.\n"
     "Segment Base has been set to 0x1000","Borg Message",MB_OK);
  }

  dsm->dissettable();
  switch(floader.exetype)
  { case BIN_EXE:
      floader.readbinfile(fsize);
      break;
    case PE_EXE:
      floader.readpefile(pe_offset);
      break;
    case MZ_EXE:
      floader.readmzfile(fsize);
      break;
    case OS2_EXE:
      floader.reados2file();
      CloseHandle(floader.efile);
      floader.efile=INVALID_HANDLE_VALUE;
      floader.exetype=0;
      return false; // at the moment;
    case COM_EXE:
      floader.readcomfile(fsize);
      break;
    case SYS_EXE:
      floader.readsysfile(fsize);
      break;
    case LE_EXE:
      floader.readlefile();
      CloseHandle(floader.efile);
      floader.efile=INVALID_HANDLE_VALUE;
      floader.exetype=0;
      return false; // at the moment;
    case NE_EXE:
      floader.readnefile(pe_offset);
      break;
    default:
      CloseHandle(floader.efile);
      floader.efile=INVALID_HANDLE_VALUE;
      floader.exetype=0;
      return false;
  }
  return true;
}

/************************************************************************
* newfile								*
* - handles selecting a new file and its messages, using the standard	*
*   routine GetOpenFileName						*
* - starts up the secondary thread when the file is loaded		*
************************************************************************/
bool newfile(void) {
 UINT FilterIndex;
 if (getfiletoload(current_exe_name, elemof(current_exe_name),IDS_FILTERSTRING,FilterIndex)) {
  changemenus(true);	// Fenster zwecks Ausgabe erzeugen
  if (loadexefile(current_exe_name)) {
   TCHAR s[128];
   StatusMessage("File Opened");
   wsprintf(s,T("%s: %s"),winname,current_exe_name);
   SetWindowText(MainWnd,s);
//   InThread=true;
//   ThreadHandle=CreateThread(NULL,0,(LPTHREAD_START_ROUTINE)Thread,0,0,&ThreadId);
  }else{
   changemenus(false);
   MessageBox(MainWnd,"File open failed?",program_name,MB_OK|MB_ICONEXCLAMATION);
  }
 }
 return 0;
}

Vorgefundene Kodierung: UTF-80