rlezip2.cpp

压缩文件的一个解法,我们老师自己写也来的,相当强!

/*
	文件打包和解包程序 by 郭 炜
	本程序用法：
	rlezip  -e dirname filename  将文件夹 dirname 中的所有内容打包到 filename 文件里去 
								  dirname 可以是相对路径,也可以是绝对路径

    rlezip -x filename           将文件 filename里的所有内容解出来,放到当前目录下

	本程序没有进行压缩
	解压函数 DeCompressDir 省略

    结果文件 结构如下：
	每一个文件和文件夹，在结果文件中都有 一个 tagFileInfo结构与之对应。如果一个 tagFileInfo
	代表一个文件，那么在该 tagFileInfo后就紧跟着该文件的内容.

	//关键函数： FindFirstFile, FindNextFile  , _chdir, _mkdir

*/


#include <windows.h>
#include <stdio.h>
#include <string.h>
#include <direct.h>
#include <io.h>
#include <fstream.h> 


int g_nCurDirLayer = 0;  //当前正在处理的文件夹所处的层次
const int ATTR_FILE = 0;
const int ATTR_DIR =  1;
const int BUF_LEN = 1024;
ofstream g_ofData;
struct tagFileInfo {
	char szFileName[256];  //文件名或文件夹名字
	int nLayer;			   //文件或文件夹所在的层次,
	                       // 根文件夹所处层次为0 ,
	                       // 如果一个文件夹处于n层，
	                       // 那么该文件夹下的文件就都是 n+1层的
	int nLength;		   //文件压缩后的长度
	int nAttr;			   //为 ATTR_FILE 代表文件，
	                       //为 	ATTR_FILE 代表文件夹 
};
void CompressFile( char * szFile)
{
	tagFileInfo FileInfo;
	ifstream ifSrc;
	 
	

	memset( & FileInfo,0,sizeof(FileInfo));
	strcpy( FileInfo.szFileName,szFile);
	FileInfo.nLayer = g_nCurDirLayer;
	FileInfo.nAttr = ATTR_FILE;
	
	
	//因为本例子不压缩,所以  FileInfo.nLength 的值就是源文件长度,
	//可以事先得到.否则,就要在压缩完整个文件后,再回填了
	ifSrc.open(szFile,ios::in | ios::binary ); //以二进制只读方式打开文件
	ifSrc.seekg(0,ios::end); //定位读指针到文件末尾
	FileInfo.nLength = ifSrc.tellg(); //获取读指针位置，即文件长度
	ifSrc.seekg(0,ios::beg);  //将文件读指针设置到文件开头，以便以后读取

	
	
	g_ofData.write( (const char * ) & FileInfo,sizeof( FileInfo));
	char * pBuf = new char[BUF_LEN];
	int nReadedLen ;
	do {
		ifSrc.read( pBuf,BUF_LEN);
		nReadedLen = ifSrc.gcount(); //获取上次读操作读取的字节数
		g_ofData.write ( pBuf,nReadedLen);
	} while( nReadedLen > 0 );  //如果文件都读完，无法再读了，那么nReadedLen 就不会是正数
		
	ifSrc.close();
	delete [] pBuf;
}
void CompressDir( char * szDir )  {
	tagFileInfo FileInfo;
	int i,nLen;
	memset( & FileInfo,0,sizeof(FileInfo));
	nLen = strlen( szDir);
	for( i = nLen; szDir[i]!= '\\' && i > 0 ;i --);
	if( szDir[i] == '\\')
		i ++;
	strcpy( FileInfo.szFileName,szDir + i);
	FileInfo.nLayer = g_nCurDirLayer;
	FileInfo.nAttr = ATTR_DIR;
	
	g_ofData.write( (const char * ) & FileInfo,sizeof(FileInfo));
	_chdir( szDir );    //当前路径设为该文件夹
	WIN32_FIND_DATA stFindData;		
	HANDLE hFindClient;

	hFindClient = FindFirstFile( "*.*" , &stFindData );    
	if ( hFindClient == INVALID_HANDLE_VALUE ) {
		return ;
	}
	do {
		if( ! ( strcmp( stFindData.cFileName, "..") == 0 || 
			strcmp( stFindData.cFileName, ".") == 0 ) )  {  // . 和 .. 代表自己和父目录，要略过，不予处理
			if((FILE_ATTRIBUTE_DIRECTORY &  
				stFindData.dwFileAttributes) == FILE_ATTRIBUTE_DIRECTORY )  {
				g_nCurDirLayer ++; //stFindData.cFileName 文件夹的层次比当前文件夹即 szDir 多1 
				CompressDir( stFindData.cFileName);  
				g_nCurDirLayer --; //处理完stFindData.cFileName 文件夹后，层次值要恢复到 szDir 的层次值
			}
			else{ //是文件
				CompressFile( stFindData.cFileName);				
			}
		}	
	}while( FindNextFile( hFindClient,& stFindData));

	//关闭查找句柄
	FindClose( hFindClient);
	
	_chdir( ".." );   //返回上层文件夹（即父文件夹)
}
void DeCompressDir( char * szZippedFile)
{
	ifstream ifZip;
	tagFileInfo FileInfo;
	ifZip.open( szZippedFile,ios::in | ios::binary ); //　以二进制只读方式打开文件
	g_nCurDirLayer = 0;
	while(ifZip.read( (char *) & FileInfo,sizeof( FileInfo)) ) {
		
		//ifZip.read( (char *) & FileInfo,sizeof( FileInfo));
		//int nReadedLen = ifZip.gcount ();
		//if( nReadedLen <= 0 ) 
		//	break;

		//printf( "%s,%d\n",FileInfo.szFileName ,FileInfo.nLayer );
		if( FileInfo.nAttr == ATTR_DIR) {
			if ( g_nCurDirLayer == FileInfo.nLayer ) {   
				//正在处理的文件夹层次和 FileInfo.nLayer相等，
				 //说明 FileInfo 代表的文件夹是正在处理的文件夹
				 //的子文件夹
				_mkdir( FileInfo.szFileName );  //创建文件夹
				g_nCurDirLayer ++;
				_chdir( FileInfo.szFileName);  //改变当前文件夹，即进入刚刚创建的文件夹
			}
			else if( g_nCurDirLayer > FileInfo.nLayer ) {   //正在处理的文件夹层次比 FileInfo.nLayer 大，那么就要回退几层
				for( int i = 0;i < g_nCurDirLayer - FileInfo.nLayer ; i ++ ) {
					_chdir ( ".."); //返回上层文件夹（即父文件夹)
				}
				g_nCurDirLayer = FileInfo.nLayer ;
				_mkdir( FileInfo.szFileName );
				g_nCurDirLayer ++;
				_chdir( FileInfo.szFileName);
			}
		}
		else {
			if( g_nCurDirLayer > FileInfo.nLayer + 1 ) {  //如果一个文件夹处于n层，那么该文件夹下的文件就都是n+1层的
				for( int i = 0;i < g_nCurDirLayer - ( FileInfo.nLayer + 1 ); i ++ ) {
					_chdir ( ".."); //返回上层文件夹（即父文件夹)
				}
				g_nCurDirLayer = FileInfo.nLayer + 1;
			}
			ofstream ofDest;
			ofDest.open( FileInfo.szFileName ,ios::out | ios::binary ); //以二进制写方式打开文件
			char * pBuf = new char[BUF_LEN];
			int nTimes = FileInfo.nLength / BUF_LEN;
			int nRemain = FileInfo.nLength % BUF_LEN;
			int i;
			for( i = 0;i < nTimes; i ++) {
				ifZip.read( pBuf,BUF_LEN);
				ofDest.write( pBuf,BUF_LEN);
			}
			ifZip.read( pBuf,nRemain);
			ofDest.write( pBuf,nRemain);
			delete [] pBuf;
			ofDest.close();
		}
	}
	ifZip.close();
}
int main( int argc ,char * argv[])
{
	if( argc == 1 )
		return 0;
	if( stricmp( argv[1],"-e") == 0 ) {
		g_ofData.open(argv[3],ios::out | ios::binary); //以二进制写方式打开文件
		CompressDir( argv[2]);	
		g_ofData.close();
	}
	else if( stricmp( argv[1],"-x") ==0 )
		DeCompressDir( argv[2]);
	return 0;
}