Windows创建大文件

工作中遇到创建固定大小的大文件,参考[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

探讨了如何创建大文件

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值