mirror of
https://github.com/sfall-team/int2ssl.git
synced 2026-07-27 16:52:42 -07:00
initial commit
This commit is contained in:
alexeevdv
+305
@@ -0,0 +1,305 @@
|
||||
// ProcTable.cpp : implementation file
|
||||
//
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "ProcTable.h"
|
||||
#include "Utility.h"
|
||||
#include "ObjectAttributes.h"
|
||||
|
||||
#include "Hacks/CString.h"
|
||||
#include <iostream>
|
||||
|
||||
// CProcDescriptor
|
||||
|
||||
CProcDescriptor::CProcDescriptor() :
|
||||
m_ulNameOffset(0),
|
||||
m_ulType(0),
|
||||
m_ulTime(0),
|
||||
m_ulExpressionOffset(0),
|
||||
m_ulBodyOffset(0),
|
||||
m_ulNumArgs(0)
|
||||
{
|
||||
}
|
||||
|
||||
CProcDescriptor::CProcDescriptor(const CProcDescriptor& Item) :
|
||||
m_ulNameOffset(Item.m_ulNameOffset),
|
||||
m_ulType(Item.m_ulType),
|
||||
m_ulTime(Item.m_ulTime),
|
||||
m_ulExpressionOffset(Item.m_ulExpressionOffset),
|
||||
m_ulBodyOffset(Item.m_ulBodyOffset),
|
||||
m_ulNumArgs(Item.m_ulNumArgs)
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
CProcDescriptor::~CProcDescriptor()
|
||||
{
|
||||
}
|
||||
|
||||
CProcDescriptor& CProcDescriptor::operator = (const CProcDescriptor& Item)
|
||||
{
|
||||
if (&Item != this) {
|
||||
m_ulNameOffset = Item.m_ulNameOffset;
|
||||
m_ulType = Item.m_ulType;
|
||||
m_ulTime = Item.m_ulTime;
|
||||
m_ulExpressionOffset = Item.m_ulExpressionOffset;
|
||||
m_ulBodyOffset = Item.m_ulBodyOffset;
|
||||
m_ulNumArgs = Item.m_ulNumArgs;
|
||||
}
|
||||
|
||||
return (*this);
|
||||
}
|
||||
|
||||
void CProcDescriptor::Serialize(CArchive& ar)
|
||||
{
|
||||
//if (ar.IsStoring()) {
|
||||
if (false)
|
||||
{
|
||||
// ASSERT(FALSE);
|
||||
}
|
||||
else {
|
||||
UINT uiTotalRead;
|
||||
|
||||
uiTotalRead = ReadMSBULong(ar, m_ulNameOffset);
|
||||
uiTotalRead += ReadMSBULong(ar, m_ulType);
|
||||
uiTotalRead += ReadMSBULong(ar, m_ulTime);
|
||||
uiTotalRead += ReadMSBULong(ar, m_ulExpressionOffset);
|
||||
uiTotalRead += ReadMSBULong(ar, m_ulBodyOffset);
|
||||
uiTotalRead += ReadMSBULong(ar, m_ulNumArgs);
|
||||
|
||||
// printf("=================\n");
|
||||
// printf("uiTotalRead = %u\n", uiTotalRead);
|
||||
// printf("m_ulNameOffset = 0x%08X\n", m_ulNameOffset);
|
||||
// printf("m_ulType = 0x%08X\n", m_ulType);
|
||||
// printf("m_ulTime = 0x%08X\n", m_ulTime);
|
||||
// printf("m_ulExpressionOffset = 0x%08X\n", m_ulExpressionOffset);
|
||||
// printf("m_ulBodyOffset = 0x%08X\n", m_ulBodyOffset);
|
||||
// printf("m_ulNumArgs = 0x%08X\n", m_ulNumArgs);
|
||||
|
||||
|
||||
if (uiTotalRead != (sizeof(ULONG) * 6)) {
|
||||
printf("Error: Unable read procedure descriptor\n");
|
||||
AfxThrowUserException();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void CProcDescriptor::Dump(CArchive& ar)
|
||||
{
|
||||
CString strOutLine;
|
||||
CString strType;
|
||||
|
||||
if (m_ulType != 0x00000000) {
|
||||
strType = "( ";
|
||||
|
||||
if (m_ulType & P_TIMED) {
|
||||
strType += "timed ";
|
||||
}
|
||||
|
||||
if (m_ulType & P_CONDITIONAL) {
|
||||
strType += "conditional ";
|
||||
}
|
||||
|
||||
if (m_ulType & P_IMPORT) {
|
||||
strType += "import ";
|
||||
}
|
||||
|
||||
if (m_ulType & P_EXPORT) {
|
||||
strType += "export ";
|
||||
}
|
||||
|
||||
if (m_ulType & P_CRITICAL) {
|
||||
strType += "critical ";
|
||||
}
|
||||
|
||||
strType += ")";
|
||||
}
|
||||
else {
|
||||
strType = "No special types";
|
||||
}
|
||||
|
||||
strOutLine.Format("Name offset: 0x%08X\n", m_ulNameOffset);
|
||||
ar.WriteString(strOutLine);
|
||||
strOutLine.Format("Type: 0x%08X // %s\n", m_ulType, strType.c_str());
|
||||
ar.WriteString(strOutLine);
|
||||
strOutLine.Format("Time: 0x%08X // %d\n", m_ulTime, m_ulTime);
|
||||
ar.WriteString(strOutLine);
|
||||
strOutLine.Format("Expression offset: 0x%08X\n", m_ulExpressionOffset);
|
||||
ar.WriteString(strOutLine);
|
||||
strOutLine.Format("Body offset: 0x%08X\n", m_ulBodyOffset);
|
||||
ar.WriteString(strOutLine);
|
||||
strOutLine.Format("Number of args: 0x%08X // %d\n", m_ulNumArgs, m_ulNumArgs);
|
||||
ar.WriteString(strOutLine);
|
||||
}
|
||||
|
||||
|
||||
// CProcTable
|
||||
|
||||
CProcTable::CProcTable()
|
||||
{
|
||||
}
|
||||
|
||||
CProcTable::~CProcTable()
|
||||
{
|
||||
}
|
||||
|
||||
struct ProcBodyOffset {
|
||||
ULONG m_ulProcIndex;
|
||||
ULONG m_ulBodyOffset;
|
||||
};
|
||||
|
||||
int compareProcBodyOffsets(const void* elem0, const void* elem1)
|
||||
{
|
||||
ProcBodyOffset offset0 = *((ProcBodyOffset*)elem0);
|
||||
ProcBodyOffset offset1 = *((ProcBodyOffset*)elem1);
|
||||
|
||||
if (offset0.m_ulBodyOffset < offset1.m_ulBodyOffset) {
|
||||
return -1;
|
||||
}
|
||||
else if (offset0.m_ulBodyOffset > offset1.m_ulBodyOffset) {
|
||||
return 1;
|
||||
}
|
||||
else {
|
||||
if (offset0.m_ulProcIndex < offset1.m_ulProcIndex) {
|
||||
return -1;
|
||||
}
|
||||
else if (offset0.m_ulProcIndex > offset1.m_ulProcIndex) {
|
||||
return 1;
|
||||
}
|
||||
else {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void CProcTable::Serialize(CArchive& ar)
|
||||
{
|
||||
//if (ar.IsStoring()) {
|
||||
if(false)
|
||||
{
|
||||
// ASSERT(FALSE);
|
||||
}
|
||||
else {
|
||||
m_Table.RemoveAll();
|
||||
m_ProcSize.RemoveAll();
|
||||
|
||||
ULONG ulSizeOfTable;
|
||||
ULONG ulRead;
|
||||
|
||||
ar.Flush();
|
||||
ULONG ulSizeOfFile = ULONG(ar.GetFile()->GetLength());
|
||||
|
||||
ulRead = ReadMSBULong(ar, ulSizeOfTable);
|
||||
|
||||
if (ulRead != sizeof(ulSizeOfTable)) {
|
||||
printf("Error: Unable read size of procedures table\n");
|
||||
AfxThrowUserException();
|
||||
}
|
||||
|
||||
m_Table.SetSize(ulSizeOfTable);
|
||||
m_ProcSize.SetSize(ulSizeOfTable);
|
||||
|
||||
ULONG ulIndexOfProcOffset = 0;
|
||||
ProcBodyOffset* pOffsets = new ProcBodyOffset[ulSizeOfTable + 1];
|
||||
|
||||
if (pOffsets == NULL) {
|
||||
// AfxThrowMemoryException();
|
||||
throw 1;
|
||||
}
|
||||
|
||||
for(ULONG i = 0; i < ulSizeOfTable; i++) {
|
||||
// printf("======== %u =========\n", i);
|
||||
m_Table.at(i).Serialize(ar);
|
||||
m_ProcSize.at(i) = 0; // Initialize size of procedure
|
||||
|
||||
if (!(m_Table.at(i).m_ulType & P_IMPORT)) {
|
||||
if ((m_Table.at(i).m_ulBodyOffset != 0) && (m_Table.at(i).m_ulBodyOffset != ulSizeOfFile)) {
|
||||
pOffsets[ulIndexOfProcOffset].m_ulProcIndex = i;
|
||||
pOffsets[ulIndexOfProcOffset].m_ulBodyOffset = m_Table.at(i).m_ulBodyOffset;
|
||||
ulIndexOfProcOffset++;
|
||||
}
|
||||
else {
|
||||
m_Table.at(i).m_ulType |= P_NOTIMPLEMENTED;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pOffsets[ulIndexOfProcOffset].m_ulProcIndex = ulIndexOfProcOffset + 1;
|
||||
pOffsets[ulIndexOfProcOffset].m_ulBodyOffset = ulSizeOfFile;
|
||||
|
||||
qsort(pOffsets, ulIndexOfProcOffset, sizeof(ProcBodyOffset), compareProcBodyOffsets);
|
||||
|
||||
// Sizes of procedures
|
||||
for(ULONG i = 0; i < ulIndexOfProcOffset; i++) {
|
||||
if (pOffsets[i + 1].m_ulBodyOffset >= pOffsets[i].m_ulBodyOffset) {
|
||||
m_ProcSize.at(pOffsets[i].m_ulProcIndex) = pOffsets[i + 1].m_ulBodyOffset - pOffsets[i].m_ulBodyOffset;
|
||||
}
|
||||
else {
|
||||
printf("Error: Procedures are not sorted in ascending address order\n");
|
||||
AfxThrowUserException();
|
||||
}
|
||||
}
|
||||
|
||||
delete [] pOffsets;
|
||||
|
||||
for(ULONG i = 0; i < ulSizeOfTable; i++) {
|
||||
if (m_Table.at(i).m_ulBodyOffset != 0) {
|
||||
m_ulOffsetOfProcSection = m_Table.at(i).m_ulBodyOffset;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// // For debugging only
|
||||
// printf("Total: %d\n", m_Table.GetSize());
|
||||
//
|
||||
// for(i = 0; i < ulSizeOfTable; i++) {
|
||||
// printf("Offset: 0x%08X, size: 0x%08X\n", m_Table.at(i).m_ulBodyOffset, m_ProcSize.at(i));
|
||||
// }
|
||||
|
||||
// printf("m_ulOffsetOfProcSection: 0x%08X\n", m_ulOffsetOfProcSection);
|
||||
}
|
||||
}
|
||||
|
||||
INT_PTR CProcTable::GetSize()
|
||||
{
|
||||
return m_Table.GetSize();
|
||||
}
|
||||
|
||||
ULONG CProcTable::GetSizeOfProc(INT_PTR nIndex)
|
||||
{
|
||||
if ((nIndex < 0) || (nIndex > m_ProcSize.GetUpperBound())) {
|
||||
printf("Warning: Invalid index of procedure (%d). Exception will be thrown\n", nIndex);
|
||||
AfxThrowUserException();
|
||||
}
|
||||
|
||||
return m_ProcSize.at(nIndex);
|
||||
}
|
||||
|
||||
ULONG CProcTable::GetOffsetOfProcSection()
|
||||
{
|
||||
return m_ulOffsetOfProcSection;
|
||||
}
|
||||
|
||||
|
||||
void CProcTable::Dump(CArchive& ar)
|
||||
{
|
||||
CString strOutLine;
|
||||
|
||||
for(INT_PTR i = 0; i < m_Table.GetSize(); i++) {
|
||||
strOutLine.Format("======== Procedure %d ========\n", i);
|
||||
ar.WriteString(strOutLine);
|
||||
m_Table.at(i).Dump(ar);
|
||||
ar.WriteString("\n");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
CProcDescriptor& CProcTable::operator [] (INT_PTR nIndex)
|
||||
{
|
||||
if ((nIndex < 0) || (nIndex > m_ProcSize.GetUpperBound())) {
|
||||
printf("Warning: Invalid index of procedure (%d). Exception will be thrown\n", nIndex);
|
||||
AfxThrowUserException();
|
||||
}
|
||||
|
||||
return m_Table.at(nIndex);
|
||||
}
|
||||
Reference in New Issue
Block a user