测试专用，跨平台批量生成指定大小及个数的文件(非稀疏文件)

最近有点不爽，心里不痛快，一个东东搞了很久了还没搞好，耐心都快弄没了,火大。不提这个了，写了个批量生成目录文件和指定大小的

文件的东西用于测试分享下。顺便看了下稀疏文件相关的东东，这个在另一篇了。

效果大致如下 ：

功能没得什么说的，直接附上代码: 

/*************************************************
Author: xiongchuanliang
Description: 批量生成指定大小，指定个数的文件，及子目录
g++ -o test custsize.cpp 
**************************************************/

#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#if defined(WIN32)
	#include <Windows.h>
	#include <Winioctl.h>
	#include <direct.h> 
	#include <io.h>
#else
	#include <sys/stat.h>
	#include <sys/types.h>
	#include <unistd.h> 
	#include <errno.h>
	#include <fcntl.h>
#endif

//生成的文件的文件名后缀
#define FILE_SUFFIX ".log"

// 生成不存存的子目录
int createdir(char * sdirname)
{
	int ret = 0 ;
	if(sdirname == NULL) return 1;
#if defined(WIN32)
	if( _access(sdirname,NULL)!=0 )
	{
		if(_mkdir(sdirname) != 0 )
		{
			printf("ERR mkdir 错误码:%d\n",GetLastError());
			return -1;
		}
	}
#else
	if( access(sdirname,F_OK)!=0 )
	{
		if( mkdir(sdirname,0755) == -1 )
		{
			//perror("ERR mkdir error ");			
			printf("ERR mkdir 错误码:%d\n",errno);
			return -1;
		} 
	}
#endif
	return 0;
}

//生成指定大小的文件
int testfile(char * sdirname,	 //当前目录
			 const int prefix,   //文件前缀
			 const int filenum , //当前目录需生成文件个数
			 const long setsize, //文件大小
			 const long fcount,  //总文件个数
			 long &cfcount)      //已生成数
{
	char scfilename[500] = {0};
	if(sdirname == NULL) return 1;

	for(int k=0;k<filenum;k++)
	{
		
#if defined(WIN32)
       sprintf(scfilename,"fsutil file createnew %s/%d_%d%s %d ",
			sdirname,prefix,k + 1,FILE_SUFFIX,setsize);
		
#else
		//创建指定大小非稀疏文件
		//fallocate -o 0 -l 1024m /u01/a.log
        sprintf(scfilename,"fallocate -o 0 -l %dk %s/%d_%d%s ",
        	setsize,sdirname,prefix,k + 1,FILE_SUFFIX);
#endif
        system(scfilename);
		cfcount++;
		printf("\n进度[ %d / %d ] \n", cfcount,fcount);
		memset(scfilename,0,sizeof(char)*500);
	}
	return 0;
}

//生成子目录,及测试文件
int testdir(char * sdirname,	//当前目录
			const int dirlevel ,//目录层次
			const int dirnum,	//子目录数
			const int filenum , //当前目录需生成文件个数
			const long setsize, //文件大小
			const long fcount)  //已生成数
{
	char sclevel[500] = {0};
	char schilddir[500] = {0};
	long cfcount = 0;
	if(sdirname == NULL) return 1;
	if( createdir(sdirname) != 0 ) return 1;
	for(int i=0;i< dirlevel;i++)
	{
		sprintf(sclevel,"%s/%d",sdirname,i + 1);
		if( createdir(sclevel) != 0 ) return 1;
		if( testfile(sclevel,i+1,filenum,setsize,fcount,cfcount) != 0 )return 1;
		for(int j=0;j<dirnum;j++)
		{
			sprintf(schilddir,"%s/%d",sclevel,j + 1);
			if( createdir(schilddir) != 0 ) return 1;
			if( testfile(schilddir,j+1,filenum,setsize,fcount,cfcount) )return 1;
			memset(schilddir,0,sizeof(char)*500);
		}
		memset(sclevel,0,sizeof(char)*500);
	}
	return 1;
}

//输入条件
int test()
{
	char type;
	char sdirname[500] = {0};
	long block,filesize,setsize;
	int  dirnum,dirlevel,filenum,fcount;

	printf("------------------------批量生成工具------------------------\n\n");
	printf("待创建主目录:");
	scanf("%s",sdirname);
	printf("\n子目录层数:");
	scanf("%d",&dirlevel);
	printf("\n每层子目录数:");
	scanf("%d",&dirnum);
	printf("\n每层子目录文件数:");
	scanf("%d",&filenum);
	printf("\n每个文件大小(单位:k,m,g):");
	scanf("%d%c",&filesize,&type);
	//printf("\n是否为稀疏文件(sparse-files) 请选y或n:");
	//scanf("%d",&filetype);
	switch(type)
	{
	case 'K':
	case 'k':
		block = 1024;
		break;
	case 'M':
	case 'm':
		block = 1024 * 1024;
		setsize = filesize * block ;
		break;
	case 'G':
	case 'g':
		block = 1024 * 1024 * 1024;
		break;
	default:
		printf("文件大小设置有误. 应为1024k这种方式输入。\n");
		return -1;
		break;
	}
	setsize = filesize * block ;
	fcount = dirnum * dirlevel * filenum + dirlevel * filenum;
	printf("--------------------------------\n");
	printf("文件总数: %d \n",fcount);
	printf("总大小  : %d %c \n",fcount * filesize,type);
	printf("--------------------------------\n\n");
	testdir(sdirname,dirlevel,dirnum,filenum,setsize, fcount );
	printf("\n全部生成完毕.\n");
	return 0;
}


int main(void)
{
	test();
#if defined(WIN32)
	system("pause");
#endif
	return 0;
}

创建的是非稀疏文件,UNIX/Linux下没找到合适的方法，找了三四种全是稀疏文件文件的，就不管了。反正目前测试了Windows下也够用了。

 对于Windows下创建指定文件，用得 fsutil 命令，要是不想弄这个，用代码实现，也可做到。 附个例子:

/*************************************************
Author: xiongchuanliang
Description: Windows下生成指定大小的文件，非稀疏文件
**************************************************/

#include <stdlib.h>
#include <stdio.h>

#include <Windows.h>
#include <Winioctl.h>
#include <direct.h> // _mkdir
#include <io.h>

void PrintError(DWORD dwErr) {
	char szMsg[256];
	DWORD dwFlags = FORMAT_MESSAGE_IGNORE_INSERTS |
		FORMAT_MESSAGE_MAX_WIDTH_MASK |
		FORMAT_MESSAGE_FROM_SYSTEM;
	if (!::FormatMessageA(dwFlags, NULL, dwErr, 0, szMsg, sizeof(szMsg), NULL)) strcpy(szMsg, "Unknown error.");
	printf(szMsg);
	printf("\n");
}

int testfile()
{
	HANDLE hFile ;
	DWORD dwWriteaddr = 0;
	DWORD dwRet =0,dwFPos = 0;
	BOOL bRet = FALSE;
	hFile = CreateFileA("C:\\xcl\\demo3.txt",GENERIC_READ|GENERIC_WRITE,
		FILE_SHARE_READ|FILE_SHARE_WRITE,
		NULL,
		CREATE_ALWAYS,
		0,
		NULL);
	if(hFile == INVALID_HANDLE_VALUE)
	{
		printf("创建文件失败!\n");
		PrintError(GetLastError());
		return 1;
	}

	//生成指定大小的文件
	// 0x100000 = 1048576 = 1 MB
	dwFPos = SetFilePointer(hFile,0x100000,NULL,FILE_BEGIN);
	if (dwFPos == INVALID_SET_FILE_POINTER)
	{
		printf("SetFilePointer发生错误.\n");
		PrintError(GetLastError());
		return 1;
	}

	bRet = WriteFile(hFile,"XCL\0",4,&dwWriteaddr,NULL);
	if (!bRet)
	{
		printf("WriteFile发生错误.\n");
		PrintError(GetLastError());
		return 1;
	}

	//除了在文件最末尾会有"XCL"这三个字符外,其它部份将会被填上0,实实在在的占用系统空间.
	dwFPos = SetEndOfFile(hFile);
	if (dwFPos == INVALID_SET_FILE_POINTER)
	{
		printf("SetEndOfFile发生错误.\n");
		PrintError(GetLastError());
		return 1;
	}

	CloseHandle(hFile);
	return 0;
}

int main(void)
{
	testfile();

	system("pause");

	return 0;
}

用上面的方法就够了。 如果要稀疏文件生成方法的，可以看我后一篇：稀疏文件(sparse-files) 生成汇总

附注:  fsutil 命令查看盘信息的写法。

C:\Documents and Settings\Administrator>fsutil fsinfo volumeinfo c:\

卷名 : WINXP

卷序列号 : 0x88f16579

组件长度最大值 : 255

文件系统名 : FAT32

保留文件名的大小写

支持文件名中的 Unicode

C:\Documents and Settings\Administrator>fsutil fsinfo volumeinfo d:\

卷名 : Install

卷序列号 : 0xcf722f5

组件长度最大值 : 255

文件系统名 : NTFS

支持区分大小写的文件名

保留文件名的大小写

支持文件名中的 Unicode

保留并加强 ACL

支持基于文件的压缩

支持磁盘配额

支持稀疏文件

支持重分析点

支持对象标识符

支持加密文件系统

支持带有名称的数据流

MAIL:[email protected]

BLOG: http://blog.csdn.net/xcl168