工作中遇到创建固定大小的大文件,参考[1]中的内容,重新整理了代码,如下:
#include <stdio.h>
#include <windows.h>
#include <tlhelp32.h>
#include <string.h>
#include <tchar.h>
#include <Psapi.h>
#include <io.h>
#include <direct.h>
#include <sys/stat.h>
#include <sys/types.h>
wchar_t * ANSIToUnicode( const char* str )
{
int textlen ;
wchar_t * result;
textlen = MultiByteToWideChar( CP_ACP, 0, str,-1, NULL,0 );
result = (wchar_t *)malloc((textlen+1)*sizeof(wchar_t));
memset(result,0,(textlen+1)*sizeof(wchar_t));
MultiByteToWideChar(CP_ACP, 0,str,-1,(LPWSTR)result,textlen );
return result;
}
int create_size_file(char *file, long long file_size)
{
HMODULE module;
HANDLE token = 0;
HANDLE file_handle;
typedef int (WINAPI *sfvd)(HANDLE hFile, LONGLONG ValidDataLength);
sfvd SetFileValidData;
TOKEN_PRIVILEGES tp = { 0 };
LPCTSTR filename = NULL;
LONG size_lo = 0;
LONG size_hi = 0;
module = GetModuleHandle(L"kernel32.dll");
if (module == NULL) {
fprintf(stderr, "get module[kernel32.dll] error[%d]\n", GetLastError());
return -1;
}
SetFileValidData = (sfvd)GetProcAddress(module, ("SetFileValidData"));
if (!SetFileValidData) {
fprintf(stderr, "GetProcAddress error[%d]\n", GetLastError());
return -1;
}
if (!OpenProcessToken(INVALID_HANDLE_VALUE, TOKEN_ALL_ACCESS, &token)) {
fprintf(stderr, "OpenProcessToken error[%d]\n", GetLastError());
return -1;
}
// SE_MANAGE_VOLUME_NAME = "SeManageVolumePrivilege"
if (!LookupPrivilegeValue(0, L"SeManageVolumePrivilege", &tp.Privileges->Luid)) {
fprintf(stderr, "LookupPrivilegeValue error[%d]\n", GetLastError());
CloseHandle(token);
return -1;
}
// 提升权限
tp.PrivilegeCount = 1;
tp.Privileges->Attributes = SE_PRIVILEGE_ENABLED;
if (!AdjustTokenPrivileges(token, 0, &tp, sizeof (tp), 0, 0)) {
fprintf(stderr, "AdjustTokenPrivileges error[%d]\n", GetLastError());
CloseHandle(token);
return -1;
}
filename = (LPCTSTR)ANSIToUnicode(file);
file_handle = CreateFile(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (file_handle == INVALID_HANDLE_VALUE) {
fprintf(stderr, "CreateFile error[%d]\n", GetLastError());
CloseHandle(token);
free((void *)filename);
return -1;
}
size_lo = (LONG)file_size;
size_hi = (LONG)(file_size >> 32);
if (SetFilePointer(file_handle, size_lo, &size_hi, FILE_BEGIN) == HFILE_ERROR) {
fprintf(stderr, "SetFilePointer error[%d]\n", GetLastError());
CloseHandle(file_handle);
CloseHandle(token);
free((void *)filename);
return -1;
}
if (SetEndOfFile(file_handle) == 0) {
fprintf(stderr, "SetEndOfFile error[%d]\n", GetLastError());
CloseHandle(file_handle);
CloseHandle(token);
free((void *)filename);
return -1;
}
// 不对磁盘进行清0操作
if (!SetFileValidData(file_handle, file_size)) {
fprintf(stderr, "SetFileValidData error[%d]\n", GetLastError());
CloseHandle(file_handle);
CloseHandle(token);
free((void *)filename);
return -1;
}
CloseHandle(file_handle);
CloseHandle(token);
free((void *)filename);
return 0;
}
int main(int argc, char* argv[])
{
FILE *fp = NULL;
char buf[] = "1234567890";
int size = 0;
long long file_size = (long long)2*1024*1024*1024;
char *filename = "D:\\a.txt";
if (argc >= 2) {
filename = argv[1];
}
if (create_size_file(filename, file_size) < 0) {
printf("create file error\n");
return -1;
}
printf("create_size_file ok\n");
fp = fopen(filename, "rb+");
if (fp == NULL) {
printf("fopen file error\n");
return -1;
}
if (_fseeki64 (fp, file_size - 10, SEEK_SET) < 0) {
printf("fseek error\n");
return -1;
}
printf("fseek ok\n");
size = fwrite(buf, 1, sizeof(buf)-1, fp);
if (size < 0) {
printf("fwrite error\n");
return -1;
}
printf("fwrite ok\n");
fclose(fp);
return 0;
}
参考资料:
[1] http://bbs.csdn.net/topics/391006799?list=lz
探讨了如何创建大文件