能力值：
( LV2，RANK：10 )
6 楼
晕，我写错了，实在是不好意思。
如果可以的话，你可以把变量定义到函数外面。我当时是在Class里面多定义了几个成员变量，用它们存的。
在函数里面定义变量会破坏堆栈，因为这样编译出来，会变成 mov dword ptr [ebp-xx]，eax 的形势，而这里刚进来的时候没有 push ebp 和 mov ebp, esp。
这个是我以前用 AheadLib 改的一个例子，拿send() 和 recv() 举个例子吧，调用原函数前后的操作。红色是我自己加的，其他的是自动生成就有的。这个我以前测试过，没问题。
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// AheadLib 命名空间
namespace AheadLib
{
HMODULE m_hModule = NULL; // 原始模块句柄
DWORD m_dwReturn[500] = {0}; // 原始函数返回地址
[COLOR="Red"]LPVOID m_ptrBuffer = 0;
DWORD m_size = 0;
HANDLE m_hFile = NULL;
DWORD m_byteCount = 0;[/COLOR]
// 加载原始模块
inline BOOL WINAPI Load()
{
TCHAR tzPath[MAX_PATH];
TCHAR tzTemp[MAX_PATH * 2];
GetSystemDirectory(tzPath, MAX_PATH);
lstrcat(tzPath, TEXT("\\ws2_32"));
m_hModule = LoadLibrary(tzPath);
if (m_hModule == NULL)
{
wsprintf(tzTemp, TEXT("无法加载 %s，程序无法正常运行。"), tzPath);
MessageBox(NULL, tzTemp, TEXT("AheadLib"), MB_ICONSTOP);
}
[COLOR="red"]m_hFile = CreateFile("MyLog.log", GENERIC_WRITE | GENERIC_READ, FILE_SHARE_WRITE | FILE_SHARE_READ, NULL, OPEN_ALWAYS, 0, 0);
if (m_hFile == INVALID_HANDLE_VALUE)
MessageBox(NULL, "Can not open log file!", "ws2_32.dll", MB_ICONSTOP);
else
{
SetFilePointer(m_hFile, 0, 0, FILE_END);
WriteFile(m_hFile, TEXT("========== START ===========\r\n"), 30, &m_byteCount, 0);
FlushFileBuffers(m_hFile);
}[/COLOR]
return (m_hModule != NULL);
}
// 释放原始模块
inline VOID WINAPI Free()
{
if (m_hModule)
{
FreeLibrary(m_hModule);
}
[COLOR="red"]if (m_hFile)
{
WriteFile(m_hFile, "=========== END ============\r\n", 30, &m_byteCount, 0);
WriteFile(m_hFile, m_ptrBuffer, m_size, &m_byteCount, 0);
FlushFileBuffers(m_hFile);
CloseHandle(m_hFile);
}[/COLOR]
}
// 获取原始函数地址
FARPROC WINAPI GetAddress(PCSTR pszProcName)
{
FARPROC fpAddress;
CHAR szProcName[16];
TCHAR tzTemp[MAX_PATH];
fpAddress = GetProcAddress(m_hModule, pszProcName);
if (fpAddress == NULL)
{
if (HIWORD(pszProcName) == 0)
{
wsprintf(szProcName, "%d", pszProcName);
pszProcName = szProcName;
}
wsprintf(tzTemp, TEXT("无法找到函数 %hs，程序无法正常运行。"), pszProcName);
MessageBox(NULL, tzTemp, TEXT("AheadLib"), MB_ICONSTOP);
ExitProcess(-2);
}
return fpAddress;
}
}
using namespace AheadLib;
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 导出函数
ALCDECL AheadLib_send(void)
{
// 保存返回地址
__asm POP m_dwReturn[18 * TYPE long];
[COLOR="red"]__asm
{
push eax
mov eax, [esp+8]
mov m_ptrBuffer, eax
mov eax, [esp+12]
mov m_size, eax
}
if (m_hFile != INVALID_HANDLE_VALUE)
{
WriteFile(m_hFile, TEXT("---------- send() ----------\r\n"), 30, &m_byteCount, 0);
WriteFile(m_hFile, m_ptrBuffer, m_size, &m_byteCount, 0);
FlushFileBuffers(m_hFile);
}
__asm pop eax[/COLOR]
// 调用原始函数
GetAddress("send")();
// 转跳到返回地址
__asm JMP m_dwReturn[18 * TYPE long];
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 导出函数
ALCDECL AheadLib_recv(void)
{
// 保存返回地址
__asm POP m_dwReturn[15 * TYPE long];
[COLOR="red"]__asm
{
push eax;
mov eax, [esp+8]
mov m_ptrBuffer, eax
mov eax, [esp+12]
mov m_size, eax
pop eax
}[/COLOR]
// 调用原始函数
GetAddress("recv")();
[COLOR="red"]__asm push eax
if (m_hFile != INVALID_HANDLE_VALUE)
{
WriteFile(m_hFile, "---------- recv() ----------\r\n", 30, &m_byteCount, 0);
WriteFile(m_hFile, m_ptrBuffer, m_size, &m_byteCount, 0);
FlushFileBuffers(m_hFile);
}
__asm pop eax[/COLOR]
// 转跳到返回地址
__asm JMP m_dwReturn[15 * TYPE long];
}
能力值：
( LV2，RANK：10 )
8 楼
// 导出函数
#pragma comment(linker, "/EXPORT:accept=_AheadLib_accept,@1")
#pragma comment(linker, "/EXPORT:bind=_AheadLib_bind,@2")
#pragma comment(linker, "/EXPORT:closesocket=_AheadLib_closesocket,@3")
#pragma comment(linker, "/EXPORT:connect=_AheadLib_connect,@4")
#pragma comment(linker, "/EXPORT:getpeername=_AheadLib_getpeername,@5")
#pragma comment(linker, "/EXPORT:getsockname=_AheadLib_getsockname,@6")
#pragma comment(linker, "/EXPORT:getsockopt=_AheadLib_getsockopt,@7")
#pragma comment(linker, "/EXPORT:htonl=_AheadLib_htonl,@8")
#pragma comment(linker, "/EXPORT:htons=_AheadLib_htons,@9")
#pragma comment(linker, "/EXPORT:ioctlsocket=_AheadLib_ioctlsocket,@10")
#pragma comment(linker, "/EXPORT:inet_addr=_AheadLib_inet_addr,@11")
#pragma comment(linker, "/EXPORT:inet_ntoa=_AheadLib_inet_ntoa,@12")
#pragma comment(linker, "/EXPORT:listen=_AheadLib_listen,@13")
#pragma comment(linker, "/EXPORT:ntohl=_AheadLib_ntohl,@14")
#pragma comment(linker, "/EXPORT:ntohs=_AheadLib_ntohs,@15")
#pragma comment(linker, "/EXPORT:recv=_AheadLib_recv,@16")
#pragma comment(linker, "/EXPORT:recvfrom=_AheadLib_recvfrom,@17")
#pragma comment(linker, "/EXPORT:select=_AheadLib_select,@18")
#pragma comment(linker, "/EXPORT:send=_AheadLib_send,@19")
#pragma comment(linker, "/EXPORT:sendto=_AheadLib_sendto,@20")
#pragma comment(linker, "/EXPORT:setsockopt=_AheadLib_setsockopt,@21")
#pragma comment(linker, "/EXPORT:shutdown=_AheadLib_shutdown,@22")
...
程序开始声明了
DWORD m_dwReturn[500] = {0}; // 原始函数返回地址
用来存放返回地址，按照导出列表 recv() 排16，send()排19，因为数组是0开始的，所以各减了1，就成了15和18。
我当时做这个的时候只是记录发送和接受的数据，所以只修改了send(), sendto(), recv(), recvfrom(), WSASend(), WSASendTo(), WSARecv(), WSARecvFrom()，内容基本都差不多。我也是照着上面那个地址里的例子做的，他讲得挺详细的。其他的dll我就没做过了。
能力值：
( LV2，RANK：10 )
10 楼
因为不太懂汇编,所以我把接收发送数据的函数单独搞出来同样是调用相同写包的函数,在send函数中调用包写文件函数的时候容易出错,比如测试时候导致IE经常出错. icersg
希望能有人继续，不知道connect()当如何记录，这个代码生成的工具好在做补丁，而做监听记录就麻烦了点
