使用文件映射获取打开的文件名

#include "stdafx.h"
#include <Windows.h>
#include <stdio.h>
#include <tchar.h>
#include <cstring>
#include <psapi.h>
#include <psapi.h>
#include <strsafe.h>
#pragma comment(lib, "psapi.lib")

LPCTSTR GetFileNameFromHandle(HANDLE hFile)
{
static TCHAR pszFilename[MAX_PATH+1];
memset(pszFilename, 0, sizeof(pszFilename));

HANDLE hFileMap = NULL;
void* pMapMem = NULL;

do
{
// Get the file size.
DWORD dwFileSizeHi = 0;
DWORD dwFileSizeLo = GetFileSize(hFile, &dwFileSizeHi);

if( dwFileSizeLo == 0 && dwFileSizeHi == 0 )
{
_tprintf( _T("Cannot map a file with a length of zero.\n") );
break;
}

// Create a file mapping object.
DWORD dwMapSize = min(((ULONGLONG)dwFileSizeHi + (ULONGLONG)dwFileSizeLo), 64);
hFileMap = CreateFileMapping(hFile,
NULL,
PAGE_READONLY,
0,
dwMapSize,
NULL);
if (hFileMap == NULL)
{
_tprintf(_T("CreateFileMapping failed with : %ld\n"), GetLastError());
break;
}

// Create a file mapping to get the file name.
pMapMem = MapViewOfFile(hFileMap, FILE_MAP_READ, 0, 0, 1);
if (pMapMem == NULL)
{
_tprintf( _T("MapViewOfFile failed with : %ld\n"), GetLastError());
break;
}

if (!GetMappedFileName (GetCurrentProcess(), pMapMem, pszFilename, MAX_PATH))
{
_tprintf( _T("GetCurrentProcess failed with : %ld\n"), GetLastError());
break;
}

// Translate path with device name to drive letters.
const int BUFSIZE = 512;

TCHAR szTemp[BUFSIZE + 1];
szTemp[0] = '\0';

if (!GetLogicalDriveStrings(BUFSIZE, szTemp))
{
_tprintf( _T("GetLogicalDriveStrings failed with : %ld\n"), GetLastError());
break;
}

TCHAR szName[MAX_PATH];
TCHAR szDrive[3] = TEXT(" :");
BOOL bFound = FALSE;
TCHAR* p = szTemp;

do
{
// Copy the drive letter to the template string
*szDrive = *p;

// Look up each device name
if (QueryDosDevice(szDrive, szName, BUFSIZE))
{
UINT uNameLen = _tcslen(szName);

if (uNameLen < MAX_PATH)
{
bFound = _tcsnicmp(pszFilename, szName, uNameLen) == 0;

if (bFound)
{
// Reconstruct pszFilename using szTempFile
// Replace device path with DOS path
TCHAR szTempFile[MAX_PATH];
StringCchPrintf(szTempFile,
MAX_PATH,
TEXT("%s%s"),
szDrive,
pszFilename+uNameLen);
StringCchCopyN(pszFilename, MAX_PATH+1, szTempFile, _tcslen(szTempFile));

_tprintf(TEXT("File name is %s\n"), pszFilename);

break;
}
}
}

// Go to the next NULL character.
while (*p++);
} while (!bFound && *p); // end of string

}while(0);

if(pMapMem)
{
UnmapViewOfFile(pMapMem);
pMapMem = NULL;
}

if(hFileMap)
{
CloseHandle(hFileMap);
hFileMap = NULL;
}

return(pszFilename);
}

//搜索C:\\*.txt的文件做测试
int Test()
{
HANDLE hFile;
HANDLE hFind;
WIN32_FIND_DATA wfd = {0};
TCHAR szPath[MAX_PATH] = { _T("c:\\") };
TCHAR szFile[MAX_PATH] = {0};
_tcscpy_s(szFile, szPath);
_tcscat_s(szFile, _T("*.txt"));
hFind = FindFirstFile(szFile, &wfd);
if (hFind == INVALID_HANDLE_VALUE)
{
_tprintf( _T("can not find a file") );
return 1;
}

do
{
_tprintf( _T("find %s at current dir\n"), wfd.cFileName);

_tcscpy_s(szFile, szPath);
_tcscat_s(szFile, wfd.cFileName);

hFile = CreateFile(szFile,
GENERIC_READ,
0,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
_tprintf( _T("create file error, %d"), GetLastError());
}
else
{
GetFileNameFromHandle(hFile);
CloseHandle(hFile);
}
}while(FindNextFile(hFind, &wfd));

FindClose(hFind);

system("pause");
return 0;
}