⭐ 欢迎来到虫虫下载站! | 📦 资源下载 📁 资源专辑 ℹ️ 关于我们
⭐ 虫虫下载站

📄 tools.cpp

📁 在Linux下使用GCC编制的FTP服务器
💻 CPP
字号:
// Tools.cpp: implementation of the Tools:: class.
//
//////////////////////////////////////////////////////////////////////
#include "Tools.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

char *Tools::TrimCopy(char *pszDst,const char *pszSrc,char chTag)
{
	strcpy(pszDst,pszSrc);
	return Trim(pszDst,chTag);
}

char *Tools::Trim(char *pszBuf,char chTag)
{
	char *pszRet=pszBuf;
	while(*pszBuf==chTag)
		pszBuf++;
	if(pszRet<pszBuf)
		strcpy(pszRet,pszBuf);
	pszBuf=pszRet+strlen(pszRet)-1;
	while(pszBuf>pszRet && *pszBuf==chTag)
		*pszBuf--=0;
	return pszRet;
}

char *Tools::StrCopyByTag(char *pszDst,
						 int iDstSize,
						 const char *pszSrc,
						 const char *pszStartTag,
						 const char *pszEndTag)
{
    char *pHead,*pTail;
    *pszDst=0;
    if((pHead=strstr(pszSrc,pszStartTag))==NULL)
        return NULL;
    pHead+=strlen(pszStartTag);
    if((pTail=strstr(pHead,pszEndTag))==NULL)
        return NULL;
    if(pTail-pHead>iDstSize-1)
        return NULL;
    strncpy(pszDst,pHead,pTail-pHead);
	pszDst[pTail-pHead]=0;
	return pszDst;
}

bool Tools::IStartWith(const void *pSrc,const char *szSub){
    char *szSrc=(char *)pSrc;
    while(*szSrc && *szSub)
        if(toupper(*szSrc++)!=toupper(*szSub++))
            return false;
		return !*szSub;
}

bool Tools::CreateDirs(const char *pszDir,int iMode)
{
    char szCurDir[MAX_PATH];
    const char *pszPreDir=pszDir;
	//
    while((pszPreDir=strstr(pszPreDir,"/"))!=NULL)
    {
        strncpy(szCurDir,pszDir,pszPreDir-pszDir);
        szCurDir[pszPreDir-pszDir]='\0';
		if(!IsDir(szCurDir) && mkdir(szCurDir,iMode)!=0){
			return false;
		}
		pszPreDir++;
    }
	if(!IsDir(pszDir) && mkdir(pszDir,iMode)!=0)
		return false;
    return true;    
}

char *Tools::UrlRelToAbs(char *pszDstUrl,
						 int iDstSize,
						 const char *pszSrcUrl,
						 const char *pszRelUrl)
{
	assert(*pszSrcUrl=='/');
	const char *pszSrcTail;
	int iLen;
    //szRelUrl是绝对路径的情况
    if(*pszRelUrl=='/')
    {
		if(strstr(pszRelUrl,".."))//不允许这类情况的出现
			return NULL;
		if((int)strlen(pszRelUrl)>iDstSize-1)
			return NULL;//目标缓冲区存不下
		strcpy(pszDstUrl,pszRelUrl);
    }else{//szRelUrl是相对路径的情况
		pszSrcTail=pszSrcUrl+strlen(pszSrcUrl);
		//先处理"../../"的情况
		while(memcmp(pszRelUrl,"..",2)==0){
			while(--pszSrcTail>pszSrcUrl){
				if(*pszSrcTail=='/')
					break;
			}
			if(pszSrcTail<pszSrcUrl){
				return NULL;//到根目录以上了
			}
			pszRelUrl+=2;
			if(*pszRelUrl=='/')
				pszRelUrl++;
		}
		if(strstr(pszRelUrl,".."))
			return NULL;//不允许"../a/../b"之类的情况出现
		if(pszSrcTail-pszSrcUrl+(int)strlen(pszRelUrl)>iDstSize-2)
			return NULL;//目标缓冲区存不下
		memcpy(pszDstUrl,pszSrcUrl,pszSrcTail-pszSrcUrl);
		memcpy(pszDstUrl+(pszSrcTail-pszSrcUrl),"/\0",2);
		strcat(pszDstUrl,pszRelUrl);
	}
	iLen=strlen(pszDstUrl);
	if(iLen>1 && pszDstUrl[iLen-1]=='/')
		pszDstUrl[iLen-1]='\0';
    return pszDstUrl;
}

char *Tools::ReplaceAll(const char *szSrc,
						const char *szSubOld,
						const char *szSubNew)
{
	unsigned nFindCount=0;
	const char *szSrcPos=szSrc;
	char *szDst,*szDstPos,*szFind;
	unsigned nSrcLen=strlen(szSrc);
	unsigned nSubOldLen=strlen(szSubOld);
	unsigned nSubNewLen=strlen(szSubNew);
	
	//{{第一遍扫描,获取目标缓冲区大小
	while((szFind=strstr(szSrcPos,szSubOld))!=NULL)
	{
		nFindCount++;
		szSrcPos=szFind+nSubOldLen;
	}
	//}}第一遍扫描
	
	if(nFindCount<1)
		return NULL;
	szDst=new char[nSrcLen+(nSubNewLen-nSubOldLen)*nFindCount+1];
	szSrcPos=szSrc;
	szDstPos=szDst;
	
	//{{第二遍扫描,逐块拷贝
	while((szFind=strstr(szSrcPos,szSubOld))!=NULL)
	{
		memcpy(szDstPos,szSrcPos,szFind-szSrcPos);
		szDstPos+=szFind-szSrcPos;
		memcpy(szDstPos,szSubNew,nSubNewLen);
		szDstPos+=nSubNewLen;
		szSrcPos=szFind+nSubOldLen;
	}
	//}}第二遍扫描
	
	memcpy(szDstPos,szSrcPos,strlen(szSrcPos));
	szDstPos[strlen(szSrcPos)]=0;
	return szDst;
}

char *Tools::UrlEncode(char *szUrl,unsigned nUrlSize)
{
	char *szTemp;
	char cUnsafes[]={' '};		//现只处理空格情况'类似于UrlEscape()
	char szUnsafe[2],szSafe[4];
	for(int i=0;i<int(sizeof(cUnsafes));i++)
	{
		sprintf(szUnsafe,"%c",cUnsafes[i]);
		sprintf(szSafe,"%%%02x",cUnsafes[i]);
		szTemp=ReplaceAll(szUrl,szUnsafe,szSafe);
		if(szTemp)
		{
			unsigned nLen=min(strlen(szTemp),nUrlSize-1);
			strncpy(szUrl,szTemp,nLen);
			szUrl[nLen]=0;
			delete []szTemp;
		}
	}
	return szUrl;
}


int Tools::Base64Encode(char* pDst,
						const unsigned char* pSrc,
						int nSrcLen,
						int nMaxLineLen)
{
	const char EnBase64Tab[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
    unsigned char c1, c2, c3;    // 输入缓冲区读出3个字节
    int nDstLen = 0;             // 输出的字符计数
    int nLineLen = 0;            // 输出的行长度计数
    int nDiv = nSrcLen / 3;      // 输入数据长度除以3得到的倍数
    int nMod = nSrcLen % 3;      // 输入数据长度除以3得到的余数
	
    // 每次取3个字节,编码成4个字符
    for (int i = 0; i < nDiv; i ++)
    {
        // 取3个字节
        c1 = *pSrc++;
        c2 = *pSrc++;
        c3 = *pSrc++;
		
        // 编码成4个字符
        *pDst++ = EnBase64Tab[c1 >> 2];
        *pDst++ = EnBase64Tab[((c1 << 4) | (c2 >> 4)) & 0x3f];
        *pDst++ = EnBase64Tab[((c2 << 2) | (c3 >> 6)) & 0x3f];
        *pDst++ = EnBase64Tab[c3 & 0x3f];
        nLineLen += 4;
        nDstLen += 4;
		
        // 输出换行?
        if (nLineLen > nMaxLineLen - 4)
        {
            *pDst++ = '\r';
            *pDst++ = '\n';
            nLineLen = 0;
            nDstLen += 2;
        }
    }
	
    // 编码余下的字节
    if (nMod == 1)
    {
        c1 = *pSrc++;
        *pDst++ = EnBase64Tab[(c1 & 0xfc) >> 2];
        *pDst++ = EnBase64Tab[((c1 & 0x03) << 4)];
        *pDst++ = '=';
        *pDst++ = '=';
        nLineLen += 4;
        nDstLen += 4;
    }
    else if (nMod == 2)
    {
        c1 = *pSrc++;
        c2 = *pSrc++;
        *pDst++ = EnBase64Tab[(c1 & 0xfc) >> 2];
        *pDst++ = EnBase64Tab[((c1 & 0x03) << 4) | ((c2 & 0xf0) >> 4)];
        *pDst++ = EnBase64Tab[((c2 & 0x0f) << 2)];
        *pDst++ = '=';
        nDstLen += 4;
    }
	
    // 输出加个结束符
    *pDst = '\0';
	
    return nDstLen;
}


int Tools::Base64Decode(unsigned char* pDst,
						const char* pSrc,  
						int nSrcLen)
{
	//Base64解码方法中,最简单的也是查表法:将64个可打印字符的值作为索引,查表得到的值(范围为0-63)依次连起来,拼凑成字节形式输出,就得到解码结果。 
	const char DeBase64Tab[] =
	{
		0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
			0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
			62,        // '+'
			0, 0, 0,
			63,        // '/'
			52, 53, 54, 55, 56, 57, 58, 59, 60, 61,        // '0'-'9'
			0, 0, 0, 0, 0, 0, 0,
			0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,
			13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,        // 'A'-'Z'
			0, 0, 0, 0, 0, 0,
			26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38,
			39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51,        // 'a'-'z'
	};
    int nDstLen=0;          // 输出的字符计数
    int nValue;             // 解码用到的长整数
    int i=0;
	
    // 取4个字符,解码到一个长整数,再经过移位得到3个字节
    while (i < nSrcLen)
    {
        if (*pSrc != '\r' && *pSrc!='\n')
        {
            nValue = DeBase64Tab[*pSrc++] << 18;
            nValue += DeBase64Tab[*pSrc++] << 12;
            *pDst++ = (nValue & 0x00ff0000) >> 16;
            nDstLen++;
			
            if (*pSrc != '=')
            {
                nValue += DeBase64Tab[*pSrc++] << 6;
                *pDst++ = (nValue & 0x0000ff00) >> 8;
                nDstLen++;
				
                if (*pSrc != '=')
                {
                    nValue += DeBase64Tab[*pSrc++];
                    *pDst++ =nValue & 0x000000ff;
                    nDstLen++;
                }
            }
			
            i += 4;
        }
        else        // 回车换行,跳过
        {
            pSrc++;
            i++;
        }
	}
	
    // 输出加个结束符
    *pDst = '\0';
	
    return nDstLen;
}

bool Tools::IsFile(const char *pszFile)
{
	struct stat fInfo;
	if(IsDir(pszFile))
		return false;
	return(stat(pszFile,&fInfo) != -1 && errno != ENOENT );
}

bool Tools::IsDir(const char *pszDir)
{
	struct stat fInfo;
	return(stat(pszDir,&fInfo) != -1 && S_ISDIR(fInfo.st_mode));
}

int Tools::GetFileSize(const char *pszFile)
{
	struct stat fInfo;
	if(stat(pszFile,&fInfo) == -1)
		return -1;
	return fInfo.st_size;
}

void Tools::Char2HexStr(char *pszHex,char cSrc)
{
	const char szHex[]="0123456789ABCDEF";
	pszHex[0]=szHex[(unsigned char)cSrc>>4];
	pszHex[1]=szHex[cSrc & 0x0F];
	pszHex[2]=0;
}

bool Tools::Log(const char *pszFormat,...)
{
	char szNow[30];	
	time_t tNow;
	FILE *pStream;
	tm *ptNow;
	va_list lstArg;
	//
	pStream=fopen("./log.txt","ab");
	if(pStream==NULL)
		return false;
	time(&tNow);					//将当前的时间值赋值给tNow
	ptNow=localtime(&tNow);
	strftime(szNow,					//传出参数,接收时间字符串
			 sizeof(szNow),			//传出参数的大小(字节)
			 "%Y-%m-%d %H:%M:%S ",	//时间转换的格式
			 ptNow);				//时间数据源
	fputs(szNow,pStream);
	va_start(lstArg,pszFormat);
	vfprintf(pStream,pszFormat,lstArg);
	va_end(lstArg);
	fputs("\n",pStream);
	fclose(pStream);
	return true;
}

bool Tools::LogFile(const char *pszFile,
					const char *pszFormat,...)
{
	char szNow[30];	
	time_t tNow;
	tm *ptNow;
	FILE *pStream;
	va_list lstArg;
	//
	pStream=fopen(pszFile,"ab");
	if(pStream==NULL)
		return false;
	time(&tNow);					//将当前的时间值赋值给tNow
	ptNow=localtime(&tNow);
	strftime(szNow,					//传出参数,接收时间字符串
			 sizeof(szNow),			//传出参数的大小(字节)
			 "%Y-%m-%d %H:%M:%S ",	//时间转换的格式
			 ptNow);				//时间数据源
	fputs(szNow,pStream);
	va_start(lstArg,pszFormat);
	vfprintf(pStream,pszFormat,lstArg);
	va_end(lstArg);
	fputs("\n",pStream);
	fclose(pStream);
	return true;
}

bool Tools::LogMem(void *pAddr,int iSize,const char *pszInfo)
{
	char szBuf[32];
	FILE *pStream=fopen("./log.txt","ab");
	time_t tNow;
	tm *ptNow;
	int iRead=0;
	//
	if(pStream==NULL)
		return false;
	time(&tNow);					//将当前的时间值赋值给tNow
	ptNow=localtime(&tNow);
	strftime(szBuf,					//传出参数,接收时间字符串
			 sizeof(szBuf),			//传出参数的大小(字节)
			 "%Y-%m-%d %H:%M:%S ",	//时间转换的格式
			 ptNow);				//时间数据源
	fprintf(pStream,"%s %s addr:%08x:\n",szBuf,pszInfo,(int)pAddr);
	while(iRead<iSize){
		Char2HexStr(szBuf,*((char *)pAddr)++);
		fputs(szBuf,pStream);
		if(++iRead%16 == 0){
			fputs("\n",pStream);
		}else{
			fputs(" ",pStream);
		}
	}
	if(iRead%16)
		fputs("\n",pStream);
	fclose(pStream);
	return true;
}

int Tools::ReadLine(char *pszBuf,
					int iSize,
					FILE *pStream,
					bool bKeepLineSymbol)
{
	char ch=0;
	int iRead=0;
	while(iSize>1 && (ch=fgetc(pStream))!=EOF){
		iRead++;
		if(ch=='\r')
			continue;
		if(ch=='\n'){
			if(bKeepLineSymbol){
				*pszBuf++=ch;
			}
			break;		
		}
		*pszBuf++=ch;
		iSize--;
	}
	*pszBuf=0;
	//Read last character on this line;
	if(ch!='\n' && ch!=EOF){
		while((ch=fgetc(pStream))!=EOF){
			if(ch=='\n'){
				break;
			}	
		}
	}
	return iRead;
}

char *Tools::ReadConfig(const char *pszName,const char *pszValDef,
					    char *pszVal,int iSize,const char *pszFile)
{
	char szLine[1024];//只支持每行字符个数小于1024的配置文件
	char szName[128];//只支持属性名小于128的情况
	char *pszTag;
	FILE *pStream=fopen(pszFile,"rb");
	int iRet=0;
	//
	if(pStream==NULL){
		strcpy(pszVal,pszValDef);
		return pszVal;
	}
	*pszVal=0;
	while(ReadLine(szLine,sizeof(szLine),pStream)>0){
		if((pszTag=strstr(szLine,"="))==NULL)
			continue;
		*pszTag++=0;
		if(pszTag-szLine>(int)sizeof(szName))
			continue;
		TrimCopy(szName,szLine);
		if(*szName=='#')//是注释,不予理会,继续搜寻
			continue;
		if(strcmp(szName,pszName)!=0)
			continue;
		if((iRet=strlen(pszTag))>iSize-1){
			strncpy(pszVal,pszTag,iSize-1);
			pszVal[iSize-1]=0;
		}else{
			strcpy(pszVal,pszTag);
		}
		Trim(pszVal,' ');
		Trim(pszVal,'"');
		break;
	}
	fclose(pStream);
	if(*pszVal==0)
		strcpy(pszVal,pszValDef);
	return pszVal;
}

bool Tools::WriteConfig(const char *pszName,
					    const char *pszVal,
						const char *pszFile)
{
	char szLine[1024];//只支持每行字符个数小于1024的配置文件
	char szName[128];//只支持属性名小于128的情况
	FILE *pStream;
	int iPos,iRead=0;
	bool bFound=false;
	//不存在的话先创建
	if(!IsFile(pszFile)){
		if((pStream=fopen(pszFile,"w"))==NULL){
			return false;
		}
		fclose(pStream);
	}
	if((pStream=fopen(pszFile,"r+b"))==NULL)
		return false;
	while((iRead=ReadLine(szLine,sizeof(szLine),pStream))>0){
		if(!StrCopyByTag(szName,sizeof(szName),szLine,"","="))
			continue;
		if(*(Trim(szName))=='#')//是注释,不予理会,继续搜寻
			continue;
		if(strcmp(szName,pszName)==0){
			bFound=true;
			break;
		}
	}
	if(bFound){	//更新配置
		iPos=ftell(pStream);
		iPos-=iRead;
		char *pszBuf=new char[GetFileSize(pszFile)-iPos+strlen(pszName)+strlen(pszVal)+2];
		sprintf(pszBuf,"%s=%s\n",pszName,pszVal);
		while(ReadLine(szLine,sizeof(szLine),pStream,true)>0){
			strcat(pszBuf,szLine);
		}
		fseek(pStream,iPos,SEEK_SET);
		fwrite(pszBuf,1,strlen(pszBuf),pStream);
		delete pszBuf;
	}else{		//添加配置
		fprintf(pStream,"%s=%s\n",pszName,pszVal);
	}
	fclose(pStream);
	return true;
}

void Tools::MemDump(const void *pVoid, int iSize,const char *pszInfo)
{
	char *pch,szChar[3],szBuf[67]={0};
	int iRead=0;
	//
	if(pszInfo)
		printf("%s:\n",pszInfo);
	pch=(char *)pVoid;
	memset(szBuf,' ',66);
	while(iRead<iSize){
		Char2HexStr(szChar,*pch);
		memcpy(szBuf+3*(iRead%16),szChar,2);
		if(isprint(*pch))
			szBuf[50+iRead%16]=*pch;
		else
			szBuf[50+iRead%16]='.';
		if(++iRead%16 == 0){
			puts(szBuf);
			memset(szBuf,' ',66);
		}
		pch++;
	}
	if(iRead%16)
		puts(szBuf);
}

void Tools::MemDump(const string &sVal,const char *pszInfo)
{
	MemDump(sVal.c_str(),sVal.length(),pszInfo);
}

bool Tools::IsEmpty(const char *pszVal)
{
	while(*pszVal)
		if(*pszVal++!=' ')
			return true;
	return false;
}

⌨️ 快捷键说明

复制代码 Ctrl + C
搜索代码 Ctrl + F
全屏模式 F11
切换主题 Ctrl + Shift + D
显示快捷键 ?
增大字号 Ctrl + =
减小字号 Ctrl + -