判断文件是否PE文件,是否GUI程序

BOOL IsWindowsApp( CString strPathName )
 {
     if ( ! PathFileExists( strPathName ) )
         return FALSE;
 
     // 根据 PE 签名判断当前文件是否合法的 PE 文件
     HANDLE hFile = CreateFile( strPathName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
     if ( hFile == INVALID_HANDLE_VALUE ) {
         TRACE1( "Failed To Open File %s !\n", strPathName );
         return FALSE;
     }
 
     HANDLE hMMFile = CreateFileMapping( hFile, NULL, PAGE_READONLY, 0, 0, NULL );
     if ( hMMFile == INVALID_HANDLE_VALUE ) {
         CloseHandle( hFile );
         return FALSE;
     }
 
     LPVOID pvMem = MapViewOfFile( hMMFile, FILE_MAP_READ, 0, 0, 0 );
     if ( ! pvMem ) {
         CloseHandle( hMMFile );
         CloseHandle( hFile );
         return FALSE;
     }
 
     // 是否包含有 DOS 签名
     if ( *( USHORT* ) pvMem != IMAGE_DOS_SIGNATURE ) {
         UnmapViewOfFile( pvMem );
         CloseHandle( hMMFile );
         CloseHandle( hFile );
         return FALSE;
     }
 
     // 是否包含有 NT 签名
     if ( *( ( DWORD* ) ( ( PBYTE ) pvMem + ( ( PIMAGE_DOS_HEADER ) pvMem )->e_lfanew ) ) != IMAGE_NT_SIGNATURE ) {
         UnmapViewOfFile( pvMem );
         CloseHandle( hMMFile );
         CloseHandle( hFile );
         return FALSE;
     }
 
     LPVOID pvOptionalHeader =  ( PBYTE ) pvMem + ( ( PIMAGE_DOS_HEADER ) pvMem )->e_lfanew + sizeof( DWORD ) + sizeof( IMAGE_FILE_HEADER );
 
     IMAGE_OPTIONAL_HEADER ioh;
     CopyMemory( & ioh, pvOptionalHeader, sizeof( IMAGE_OPTIONAL_HEADER ) );
 
     if ( ioh.Subsystem == IMAGE_SUBSYSTEM_WINDOWS_GUI ) {
         UnmapViewOfFile( pvMem );
         CloseHandle( hMMFile );
         CloseHandle( hFile );   
 
         return TRUE;
     }
 
     return FALSE;
 }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值