dwmlscdoc.cpp

一个反汇编器源码 一个反汇编器源码

// DWmlscDoc.cpp : implementation of the CDWmlscDoc class
//

#include "stdafx.h"
#include "DWmlsc.h"

#include "DWmlscDoc.h"
#include "decompiler.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CDWmlscDoc

IMPLEMENT_DYNCREATE(CDWmlscDoc, CDocument)

BEGIN_MESSAGE_MAP(CDWmlscDoc, CDocument)
	//{{AFX_MSG_MAP(CDWmlscDoc)
		// NOTE - the ClassWizard will add and remove mapping macros here.
		//    DO NOT EDIT what you see in these blocks of generated code!
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDWmlscDoc construction/destruction

CDWmlscDoc::CDWmlscDoc()
{
	// TODO: add one-time construction code here
	m_binCodeBuf = NULL;
	m_codeLen = 0;
}

CDWmlscDoc::~CDWmlscDoc()
{
}

BOOL CDWmlscDoc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;

	((CEditView*)m_viewList.GetHead())->SetWindowText(NULL);

	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)

	return TRUE;
}


/////////////////////////////////////////////////////////////////////////////
// CDWmlscDoc serialization

void CDWmlscDoc::Serialize(CArchive& ar)
{
	if(ar.IsStoring() == FALSE)
	{
		m_codeLen = ar.GetFile()->GetLength();
		m_binCodeBuf = new BYTE[m_codeLen];
		ar.Read(m_binCodeBuf,m_codeLen);

		m_result_code.RemoveAll();

		if(DeCompile(m_binCodeBuf,m_codeLen,m_result_code) == false)
		{
			MessageBox(NULL,"Sorry,Decompile faild!","Error",MB_OK|MB_ICONSTOP);
			return;
		}

		SetModifiedFlag(TRUE);
	}
	else
	{
		if(m_binCodeBuf)
		{
			ar.Write(m_binCodeBuf,m_codeLen);
			CString file_name = ar.GetFile()->GetFileName();
			file_name.MakeLower();
			if(file_name.Find(".wmlsc") < 0)
			{
				MessageBox(NULL,"Invalidate ext name:" + file_name,"Invalidate pathname",MB_OK|MB_ICONSTOP);
				return;
			}
			else
			{
				file_name.Replace(".wmlsc",".wmls");
				CFile file;
				if(file.Open(file_name,CFile::modeWrite,NULL) == TRUE)
				{
					if(MessageBox(NULL,"Overwrite " + file_name + "?","Warning",MB_YESNO|MB_ICONINFORMATION) != IDYES)
					{
						file.Close();
						return;
					}
				}
				else
				{
					if(file.Open(file_name,CFile::modeCreate|CFile::modeWrite,NULL) == FALSE)
					{
						MessageBox(NULL,"Create File " + file_name + "Error!","Error",MB_OK|MB_ICONSTOP);
						return;
					}
				}
				POSITION pos = m_result_code.GetHeadPosition();
				while(pos != NULL)
				{
					CString line = m_result_code.GetNext(pos);
					file.Write(line.GetBuffer(0),line.GetLength());
					file.Write("\r\n",2);
				}

				file.Close();
			}
		}
	}
}

/////////////////////////////////////////////////////////////////////////////
// CDWmlscDoc diagnostics

#ifdef _DEBUG
void CDWmlscDoc::AssertValid() const
{
	CDocument::AssertValid();
}

void CDWmlscDoc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CDWmlscDoc commands

void CDWmlscDoc::DeleteContents() 
{
	delete [] m_binCodeBuf;	
	CDocument::DeleteContents();
	m_binCodeBuf = NULL;
	m_codeLen = 0;
	for(int i=0;i<g_FuncTable.GetSize();i++)
	{
		Function func = g_FuncTable.ElementAt(i);
		delete [] func.CodeArray;
	}
	g_FuncTable.RemoveAll();
	g_ConstTable.RemoveAll();
}