bool KFile::setFile(KData fullFilePath, int mode)
{
if (m_bIsOpen)
{
fclose (m_hFile);
m_bIsOpen = false;
}
#ifndef WIN32
replaceAll(fullFilePath);
#endif
if (fullFilePath.find("/") == -1) //斜杠……
{
m_kFilePath = "./";
m_kFileName = fullFilePath;
}
else
{
KData tStr = fullFilePath;
if (tStr[tStr.length() - 1] == '/')
{
return false;
}
else
{
int pos = tStr.findlast("/");
m_kFilePath = tStr.substr(0, pos + 1);
m_kFileName = tStr.substr(pos + 1, tStr.length() - pos - 1);
}
}
return open(mode);
}
bool KDirectory::createDir(const KData& dir)
{
int pos = 0;
while (true)
{
if ((pos = dir.find("/", pos)) == -1)
pos = dir.length();
KData dtDir = dir.substr(0, pos);
pos++;
if (!dtDir.isEmpty())
{
if (!isDirectoryExist(dtDir))
{
/*
if ( -1 == mkdir(dtDir.getData(),S_IRWXU|S_IRWXG|S_IRWXO) )
{
return false;
}
*/
if (MKDIR(dtDir.getData()) != 0)
{
return false;
}
}
}
if (pos >= (int) dir.length())
break;
}
return true;
}
void vCpLog( int pri, const char* file, int line, const char* fmt, va_list ap )
{
assert( pri >= 0 && pri <= LAST_PRIORITY );
char datebuf[ DATEBUF_SIZE ];
memset(datebuf, 0, DATEBUF_SIZE);
int datebufCharsRemaining;
struct timeval tv;
struct timezone tz;
int result = gettimeofday( &tv, &tz );
if( result == -1 )
{
datebuf [0] = '\0';
}
else
{
const time_t timeInSeconds = (time_t) tv.tv_sec;
strftime( datebuf, DATEBUF_SIZE, "%m-%d %H:%M:%S", localtime ( &timeInSeconds ) );
}
char msbuf[10];
sprintf( msbuf, ".%3.3d", (int)(tv.tv_usec/1000) );
datebufCharsRemaining = DATEBUF_SIZE - strlen( datebuf );
strncat( datebuf, msbuf, datebufCharsRemaining - 1 );
datebuf[DATEBUF_SIZE - 1] = '\0';
if( bConnServer )
{
char outBuff[2048];
memset(outBuff, 0, 2048);
KData ret = "[";
ret = ret + KData(datebuf) + KData("]") + KData(priNameShort[pri]) + KData(" ");
vsprintf( outBuff, fmt, ap );
ret += outBuff;
LogPack *pLogPack = (LogPack*)outBuff;
pLogPack->lLogMsgSize = htonl( ret.length() ) ;
strcpy( pLogPack->strLogMsg, ret.getDataBuf() );
spClientSock->transmit( (char*)pLogPack, sizeof( long )+ret.length() );
}
if( mainLogFunc == NULL )
{
fprintf( cpLogFd, "[%s] %s: ", datebuf, priNameShort[pri] );
vfprintf( cpLogFd, fmt, ap );
fprintf( cpLogFd, "\n" );
fflush( cpLogFd );
}
else
{
char outBuff[2048];
memset(outBuff, 0, 2048);
KData ret = "[";
ret = ret + KData(datebuf) + "]" + priNameShort[pri] + " ";
vsprintf( outBuff, fmt, ap );
ret += outBuff;
mainLogFunc( ret, pri );
}
}
int KFile::write(const KData& data)
{
const char* pKData = data.getDataBuf();
int len = data.length();
int iWrited = 0;
while (iWrited < len)
{
int iRet = write((unsigned char*) (pKData + iWrited), len - iWrited);
if (iRet <= 0)
break;
iWrited += iRet;
}
return iWrited;
}
void
KNetworkAddress::setHostName( const KData& theAddress )
{
const char* cstrAddress;
KData sAddress, sPort;
unsigned int length = theAddress.length();
unsigned int colonPos = length;
cstrAddress = theAddress.getData();
for ( int i = 0; *cstrAddress != '\0'; cstrAddress++, i++ )
{
if ( *cstrAddress == ':' )
{
colonPos = i;
break;
}
}
if ( colonPos != length )
{
sAddress = theAddress.substr( 0, colonPos );
sPort = theAddress.substr( colonPos+1, length-colonPos-1 );
colonPos = atoi( sPort.getData() );
if ( colonPos )
{
aPort = colonPos;
}
else
{
}
}
else // No ':' in input string;
{
sAddress = theAddress;
}
rawHostName = sAddress;
ipAddressSet = false;
}
int KHttp::getHttpFileLength(const KData & fullUrl)
{
KData server;
KData httpFile;
KData dtFullUrl = fullUrl;
if (isEqualNoCase(dtFullUrl.substr(0, 7), "http://"))
{
dtFullUrl = dtFullUrl.substr(7);
}
int pos = dtFullUrl.find("/");
if (pos == -1)
{
return -1;
}
else
{
server = dtFullUrl.substr(0, pos);
httpFile = dtFullUrl.substr(pos);
}
m_bChunked = false;
m_iWriteLen = 0;
m_iLength = 0;
m_clientSock.close();
m_kCnnect.close();
_respHeadMap.clear();
m_iNotifyPos = 0;
m_iNotifyGap = 0;
m_iContentLength = 0;
m_iStatusCode = 0;
bool bGet = (_paramMap.size() == 0);
KFile file;
KData httpRequest;
KData dtPost;
if (bGet)
{
httpRequest = "GET ";
}
else
{
httpRequest = "POST ";
map<KData, KData>::iterator iter;
for (iter = _paramMap.begin(); iter != _paramMap.end(); iter++)
{
if (iter != _paramMap.begin())
dtPost += "&";
dtPost += iter->first;
dtPost += "=";
dtPost += iter->second;
}
}
httpRequest += httpFile;
httpRequest += " HTTP/1.1";
httpRequest += CRLF;
httpRequest += "Host: ";
httpRequest += server;
httpRequest += CRLF;
httpRequest += "Accept: */*";
httpRequest += CRLF;
if (!m_dtUserAgent.isEmpty())
{
httpRequest += "User-Agent: ";
httpRequest += m_dtUserAgent;
httpRequest += CRLF;
}
httpRequest += "Pragma: no-cache";
httpRequest += CRLF;
httpRequest += "Cache-Control: no-cache";
httpRequest += CRLF;
httpRequest += "Connection: close";
httpRequest += CRLF;
if (!bGet)
{
httpRequest += "Content-Type: application/x-www-form-urlencoded";
httpRequest += CRLF;
httpRequest += "Content-Length: ";
httpRequest += KData((int) dtPost.length());
httpRequest += CRLF;
}
httpRequest += CRLF;
char buff[MTU] =
{ 0 };
m_clientSock.setServer(server, 80);
if (!m_clientSock.connect())
{
return 0;
}
m_kCnnect = m_clientSock.getConn();
m_kCnnect.setTimeout(m_timeout);
if (m_kCnnect.writeData(httpRequest) != (int) httpRequest.length())
return 0;
m_iWritedBytes += httpRequest.length();
if (!bGet)
{
if (m_kCnnect.writeData(dtPost) != (int) dtPost.length())
return 0;
m_iWritedBytes += dtPost.length();
}
int iRead;
m_iStatusCode = 0;
memset(buff, 0, MTU);
if ((iRead = m_kCnnect.readLine(buff, MTU)) <= 0)
{
m_iReadedBytes += iRead;
return 0;
}
KData dtKData;
KData dtLine(buff, iRead);
if (dtLine.match(SP, &dtKData, true) == NOT_FOUND)
{
return 0;
}
if (dtKData != "HTTP/1.1" && dtKData != "HTTP/1.0")
{
return 0;
}
if (dtLine.match(SP, &dtKData, true) == NOT_FOUND)
{
return 0;
}
m_iStatusCode = (int) dtKData;
while ((iRead = m_kCnnect.readLine(buff, MTU)) > 0)
{
m_iReadedBytes += iRead;
KData dtLine(buff, iRead);
KData dtBefVal;
if (FOUND == dtLine.match(":", &dtBefVal, true))
{
dtBefVal.removeSpaces();
dtLine.removeSpaces();
if (isEqualNoCase(dtBefVal, "Content-Length"))
{
return (int) dtLine;
}
}
}
return 0;
}
int KHttp::getHttpFile(const KData& kServer, const KData& kHttpFile,
const KData& kSaveFile, int nStartPos)
{
LOGD("Entry getHttpFile");
m_dtHttpServer = kServer;
m_dtHttpFile = kHttpFile;
m_bChunked = false;
m_iWriteLen = nStartPos;
m_iLength = 0;
m_clientSock.close();
m_kCnnect.close();
_respHeadMap.clear();
m_iNotifyPos = 0;
m_iNotifyGap = 0;
m_iContentLength = 0;
m_iStatusCode = 0;
bool bGet = (_paramMap.size() == 0);
KFile kFile;
KData kHttpRequest;
KData kDataPost;
if (bGet)
{
kHttpRequest = "GET ";
}
else
{
kHttpRequest = "POST ";
map<KData, KData>::iterator iter;
for (iter = _paramMap.begin(); iter != _paramMap.end(); iter++)
{
if (iter != _paramMap.begin())
kDataPost += "&";
kDataPost += iter->first;
kDataPost += "=";
kDataPost += iter->second;
}
}
kHttpRequest += kHttpFile;
kHttpRequest += " HTTP/1.1";
kHttpRequest += CRLF;
kHttpRequest += "Host: ";
kHttpRequest += kServer;
kHttpRequest += CRLF;
kHttpRequest += "Accept: */*";
kHttpRequest += CRLF;
if (!m_dtUserAgent.isEmpty())
{
kHttpRequest += "User-Agent: ";
kHttpRequest += m_dtUserAgent;
kHttpRequest += CRLF;
}
if (nStartPos > 0)
{
kHttpRequest += "RANGE: bytes=";
kHttpRequest += nStartPos;
kHttpRequest += "-";
kHttpRequest += CRLF;
}
kHttpRequest += "Pragma: no-cache";
kHttpRequest += CRLF;
kHttpRequest += "Cache-Control: no-cache";
kHttpRequest += CRLF;
kHttpRequest += "Connection: close";
kHttpRequest += CRLF;
if (!bGet)
{
kHttpRequest += "Content-Type: application/x-www-form-urlencoded";
kHttpRequest += CRLF;
kHttpRequest += "Content-Length: ";
kHttpRequest += KData((int) kDataPost.length());
kHttpRequest += CRLF;
}
kHttpRequest += CRLF;
char szBuffer[MTU] = {0};
if (m_dtHttpProxy.isEmpty())
{
LOGD("Connect to Server: %s", kServer.getData() );
m_clientSock.setServer(kServer, 80);
}
else
{
LOGD("Connect to Server: %s", m_dtHttpProxy.getData() );
m_clientSock.setServer(m_dtHttpProxy, 80);
}
m_clientSock.initSocket();
if (!m_clientSock.connect())
{
LOGERROR("m_clientSock.connect() failed");
return -1;
}
m_kCnnect = m_clientSock.getConn();
m_kCnnect.setTimeout(m_timeout);
if (m_kCnnect.writeData(kHttpRequest) != (int) kHttpRequest.length())
{
LOGERROR(" m_conn.writeData(httpRequest) != (int)httpRequest.length() failed");
return -1;
}
m_iWritedBytes += kHttpRequest.length();
if (!bGet)
{
if (m_kCnnect.writeData(kDataPost) != (int) kDataPost.length())
{
LOGERROR("m_conn.writeData(dtPost) != (int)dtPost.length() failed");
return -1;
}
m_iWritedBytes += kDataPost.length();
}
int iRead = 0;
m_iStatusCode = 0;
bool bRun = true;
memset(szBuffer, 0, MTU);
if ((iRead = m_kCnnect.readLine(szBuffer, MTU)) <= 0)
{
LOGERROR( "Read command line err: %d", iRead );
m_iReadedBytes += iRead;
return 0;
}
KData dtKData;
KData dtLine(szBuffer, iRead);
if (dtLine.match(SP, &dtKData, true) == NOT_FOUND)
{
LOGERROR("Read command line mactch space err" );
return 0;
}
if (dtKData != "HTTP/1.1" && dtKData != "HTTP/1.0")
{
LOGERROR( "GET HTTP HEAD ERR" );
return 0;
}
if (dtLine.match(SP, &dtKData, true) == NOT_FOUND)
{
LOGERROR("Read command line mactch space 2 err" );
return 0;
}
m_iStatusCode = (int)dtKData;
LOGD("Ready to while ( (iRead = m_conn.readLine(buff,MTU)) > 0 )");
while ((iRead = m_kCnnect.readLine(szBuffer, MTU)) > 0)
{
m_iReadedBytes += iRead;
KData dtLine(szBuffer, iRead);
KData dtBefVal;
if (FOUND == dtLine.match(":", &dtBefVal, true))
{
dtBefVal.removeSpaces();
dtLine.removeSpaces();
if (isEqualNoCase(dtBefVal, "Content-Length"))
{
m_iLength = (int) dtLine + nStartPos;
if (m_iNotifyPercent > 0 && m_iNotifyPercent <= 100)
{
m_iNotifyPos = m_iNotifyGap = m_iLength
/ (100 / m_iNotifyPercent);
}
}
else if (isEqualNoCase(dtBefVal, "Transfer-Encoding"))
{
if (isEqualNoCase(dtLine, "chunked"))
{
m_bChunked = true;
}
}
_respHeadMap[dtBefVal] = dtLine;
}
}
if (iRead < 0)
{
LOGERROR("read err" );
return -1;
}
if (m_iStatusCode != 200 && m_iStatusCode != 206)
{
if (m_iStatusCode == 302)
{
KData dtRedirectUrl = getRespFieldValue("Location");
LOGD("Ready to return getHttpFile( dtRedirectUrl, savefile, startpos );");
return getHttpFile(dtRedirectUrl, kSaveFile, nStartPos);
}
LOGERROR(" m_iStatusCode!=200 && m_iStatusCode!=206");
return 0;
}
FILE* pkFile = 0;
string strFile = kSaveFile.getData();
if (nStartPos <= 0)
{
pkFile = fopen(strFile.c_str(),"wb+");
}
else
{
pkFile = fopen(strFile.c_str(), "wb+");
fclose(pkFile);
pkFile = fopen(strFile.c_str(),"r+b");
if (0 == pkFile)
{
return -1;
}
fseek(pkFile, nStartPos, SEEK_SET);
}
if (m_bChunked)
{
LOGD("m_bChunked is true!");
unsigned char* pBuff = new unsigned char[MTU];
int iBuffLen = MTU;
while ((iRead = m_kCnnect.readLine(szBuffer, MTU)) > 0)
{
m_iReadedBytes += iRead;
if (iRead > 8)
{
return -1;
}
int nLength = KData(szBuffer, iRead).HexToInt();
if (nLength <= 0)
{
delete [] pBuff;
return m_iWriteLen;
}
if (nLength > iBuffLen)
{
delete[] pBuff;
iBuffLen = nLength;
pBuff = new unsigned char[iBuffLen];
}
int iReaded = 0;
memset(pBuff, 0, nLength);
m_kCnnect.readData(pBuff, nLength, iReaded);
m_iReadedBytes += iReaded;
kFile.write(pBuff, iReaded);
m_iWriteLen += iReaded;
if (m_iLength > 0)
{
if (m_iWriteLen >= m_iLength)
{
if (m_iNotifyGap > 0)
{
m_pNotifyCallback(m_pNotifyParam, 100, m_iWriteLen,
m_iLength);
}
break;
}
if (m_iNotifyGap > 0 && m_iWriteLen > m_iNotifyPos)
{
int nPercent = int((m_iWriteLen / (float) m_iLength) * 100);
m_iNotifyPos += m_iNotifyGap;
m_pNotifyCallback(m_pNotifyParam, nPercent, m_iWriteLen,
m_iLength);
}
}
if (iReaded != nLength)
{
delete[] pBuff;
return m_iWriteLen;
}
if (m_kCnnect.readLine(szBuffer, MTU) != 0)
{
return m_iWriteLen;
}
}
delete[] pBuff;
return m_iWriteLen;
}
else
{
string strFile = kSaveFile.getData();
while ((iRead = m_kCnnect.readn(szBuffer, MTU)) > 0 && bRun)
{
m_iReadedBytes += iRead;
unsigned int uiWriteSize = fwrite((unsigned char*) szBuffer,1,iRead,pkFile);
//kFile.write((unsigned char*) szBuffer, iRead);
m_iWriteLen += iRead;
if (m_iLength > 0)
{
if (m_iWriteLen >= m_iLength)
{
if (m_iNotifyGap > 0)
{
m_pNotifyCallback(m_pNotifyParam, 100, m_iWriteLen,
m_iLength);
}
break;
}
if (m_iNotifyGap > 0 && m_iWriteLen > m_iNotifyPos)
{
int nPercent = int((m_iWriteLen / (float) m_iLength) * 100);
m_iNotifyPos += m_iNotifyGap;
m_pNotifyCallback(m_pNotifyParam, nPercent, m_iWriteLen,
m_iLength);
}
}
while (m_bPause)
{
vsleep(1000);
}
}
//fclose(pkFile);
}
fclose(pkFile);
m_kCnnect.close();
kFile.closeFile();
LOGD("return m_iWriteLen;");
return m_iWriteLen;
}