今天在书上看到了一个关于加壳的程序,前面有一个判定是否是PE文件的函数。于是拿来试试。结果出现了好多问题。最头痛的就是CreateFile返回的是INVALID_HANDLE_VALUE的问题。
最开始我以为是自己的文件地址出了问题。于是换了一个地址又一个地址。问题依然存在。在网上查找资料。有一篇可能是正解,当时我完全忽略了。我怎么会知道是参数类型的问题。
原函数代码:(部分)
char m_cFileName[256]={0};
BOOL IsPeFile(char *pFileName)
{
char *pTempPath = NULL;
if(pFileName == NULL)
{
pTempPath = m_cFileName;
}
else
{
pTempPath = pFileName;
}
//创建指定文件句柄
HANDLE hFile = CreateFile(pTempPath,GENERIC_READ,FILE_SHARE_HEAD,NULL,
OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
if(!hFile || hFile == INVALID_HANDLE_VALUE)
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL,SUBLANG_DEFAULT),
(LPTSTR) &lpMsgBuf,
0,
NULL);
MessageBox(NULL,(LPCTSTR)lpMsgBuf,"Error",MB_OK | MB_ICONINFORMATION);
LocalFree(lpMsgBuf);
return FALSE;
}
.........................................................
int main()
{
char *filePath="D:\\各种工具\\PEditor\\PEditor.exe";
BOOL t;
t = IsPeFile(filePath);
if(t)
{ printf("OK\n");
return 1;
}
printf("no\n");
return 0;
}
---------------------------------------------------------------------------------------------------------
上面的代码就会出现我的说的那个问题。我改正后的代码如下:
TCHAR m_cFileName[256]={0};
BOOL IsPeFile(TCHAR *pFileName)
{
TCHAR *pTempPath = NULL;
if(pFileName == NULL)
{
pTempPath = m_cFileName;
}
else
{
pTempPath = pFileName;
}
//创建指定文件句柄
HANDLE hFile = CreateFile(pTempPath,GENERIC_READ,FILE_SHARE_HEAD,NULL,
OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
if(!hFile || hFile == INVALID_HANDLE_VALUE)
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL,SUBLANG_DEFAULT),
(LPTSTR) &lpMsgBuf,
0,
NULL);
MessageBox(NULL,(LPCTSTR)lpMsgBuf,"Error",MB_OK | MB_ICONINFORMATION);
LocalFree(lpMsgBuf);
return FALSE;
}
.........................................................
int main()
{
TCHAR *filePath=L"D:\\各种工具\\PEditor\\PEditor.exe";
BOOL t;
t = IsPeFile(filePath);
if(t)
{ printf("OK\n");
return 1;
}
printf("no\n");
return 0;
}
这样就是把char换成了TCHAR。然后就成功了。
关于TCHAR与char的关系区别,百度很多,不介绍了。
做一个记录。