Windows C++ 获取文件夹大小（通过FindNextFile实现）

2019独角兽企业重金招聘Python工程师标准>>>

windows环境下，获取文件夹大小的方法，看着挺恶心的，但不失为一种方法。后面再看看能不能遇到好方法。

#include <windows.h>
#include <tchar.h> 
#include <stdio.h>
#include <strsafe.h>
#pragma comment(lib, "User32.lib")

void DisplayErrorBox(LPTSTR lpszFunction);

LARGE_INTEGER GetDirectorySize(TCHAR* szDir);

int _tmain(int argc, TCHAR *argv[])
{
	LARGE_INTEGER diretorySize;
	diretorySize.QuadPart = 0;
	TCHAR szDir[MAX_PATH];
	size_t length_of_arg;
	DWORD dwError = 0;
	
	if (argc != 2)
	{
		_tprintf(TEXT("\nUsage: %s <directory name>\n"), argv[0]);
		return (-1);
	}

	StringCchLength(argv[1], MAX_PATH, &length_of_arg);

	if (length_of_arg > (MAX_PATH - 3))
	{
		_tprintf(TEXT("\nDirectory path is too long.\n"));
		return (-1);
	}
	_tprintf(TEXT("\nTarget directory is %s\n\n"), argv[1]);

	StringCchCopy(szDir, MAX_PATH, argv[1]);	
	diretorySize = GetDirectorySize(szDir);

	_tprintf(TEXT("directory Size   %lld bytes\n"), diretorySize.QuadPart);

	return dwError;
}

void DisplayErrorBox(LPTSTR lpszFunction)
{
	LPVOID lpMsgBuf;
	LPVOID lpDisplayBuf;
	DWORD dw = GetLastError();

	FormatMessage(
		FORMAT_MESSAGE_ALLOCATE_BUFFER |
		FORMAT_MESSAGE_FROM_SYSTEM |
		FORMAT_MESSAGE_IGNORE_INSERTS,
		NULL,
		dw,
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
		(LPTSTR)&lpMsgBuf,
		0, NULL);

	// Display the error message and clean up

	lpDisplayBuf = (LPVOID)LocalAlloc(LMEM_ZEROINIT,
		(lstrlen((LPCTSTR)lpMsgBuf) + lstrlen((LPCTSTR)lpszFunction) + 40) * sizeof(TCHAR));
	StringCchPrintf((LPTSTR)lpDisplayBuf,
		LocalSize(lpDisplayBuf) / sizeof(TCHAR),
		TEXT("%s failed with error %d: %s"),
		lpszFunction, dw, lpMsgBuf);
	MessageBox(NULL, (LPCTSTR)lpDisplayBuf, TEXT("Error"), MB_OK);

	LocalFree(lpMsgBuf);
	LocalFree(lpDisplayBuf);
}

LARGE_INTEGER GetDirectorySize(TCHAR* szDir)
{
	WIN32_FIND_DATA ffd;
	LARGE_INTEGER diretorySize;
	diretorySize.QuadPart = 0;

	HANDLE hFind = INVALID_HANDLE_VALUE;
	DWORD dwError = 0;

	TCHAR szDirAll[MAX_PATH];
	StringCchCopy(szDirAll, MAX_PATH, szDir);
	StringCchCat(szDirAll, MAX_PATH, TEXT("\\*"));

	hFind = FindFirstFile(szDirAll, &ffd);
	if (INVALID_HANDLE_VALUE == hFind)
	{
		DisplayErrorBox(TEXT("FindFirstFile"));
		return diretorySize;
	}

	do
	{
		if (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			if (strcmp((char*)ffd.cFileName, ".")==0 || strcmp((char*)ffd.cFileName, "..")==0)
			{
				//过滤当前目录和上一层目录
				continue;
			}			

			TCHAR szSubDir[MAX_PATH];
			StringCchCopy(szSubDir, MAX_PATH, szDir);
			StringCchCat(szSubDir, MAX_PATH, TEXT("\\"));
			StringCchCat(szSubDir, MAX_PATH, ffd.cFileName);

			LARGE_INTEGER subDirSize = GetDirectorySize(szSubDir);
			diretorySize.QuadPart += subDirSize.QuadPart;
		}
		else
		{
			diretorySize.QuadPart += ffd.nFileSizeLow + ffd.nFileSizeHigh;
		}
	} while (FindNextFile(hFind, &ffd) != 0);

	dwError = GetLastError();
	if (dwError != ERROR_NO_MORE_FILES)
	{
		DisplayErrorBox(TEXT("FindFirstFile"));
	}

	FindClose(hFind);

	return diretorySize;
}

 

转载于:https://my.oschina.net/zidanzzg/blog/812834