CZipString ZipPlatform::GetTmpFileName(LPCTSTR lpszPath, ZIP_SIZE_TYPE uSizeNeeded)
{
TCHAR empty[] = _T("");
CZipString tempPath;
bool bCheckTemp = true;
if (lpszPath)
{
tempPath = lpszPath;
bCheckTemp = uSizeNeeded > 0 && GetDeviceFreeSpace(tempPath) < uSizeNeeded;
}
if (bCheckTemp)
{
DWORD size = GetTempPath(0, empty);
if (size == 0)
return (CZipString)empty;
GetTempPath(size, tempPath.GetBuffer(size));
tempPath.ReleaseBuffer();
if (GetDeviceFreeSpace(tempPath) < uSizeNeeded)
{
if (!GetCurrentDirectory(tempPath) || (uSizeNeeded > 0 && GetDeviceFreeSpace(tempPath) < uSizeNeeded))
return (CZipString)empty;
}
}
CZipString tempName;
if (!GetTempFileName(tempPath, _T("ZAR"), 0, tempName.GetBuffer(_MAX_PATH)))
return (CZipString)empty;
tempName.ReleaseBuffer();
return tempName;
}
int ZipPlatform::MultiByteToWide(const char* szIn, int iInSize, CZipString& szOut, UINT uCodePage)
{
szOut.Empty();
if (iInSize < 0)
{
iInSize = (int)std::string(szIn).length();
}
if (iInSize == 0)
{
return 0;
}
// iLen doesn't include terminating character
DWORD dwFlags = uCodePage <= CP_OEMCP ? MB_PRECOMPOSED : 0;
int iLen = MultiByteToWideChar(uCodePage, dwFlags, szIn, iInSize, NULL, 0);
if (iLen > 0)
{
iLen = MultiByteToWideChar(uCodePage, dwFlags, szIn, iInSize, szOut.GetBuffer(iLen), iLen);
szOut.ReleaseBuffer(iLen);
ASSERT(iLen > 0);
if (iLen == 0)
return -1;
#ifdef _ZIP_UNICODE_NORMALIZE
// if there is a problem with compilation here, you may need uncomment the block defining WINVER = 0x0600 at the bottom of _features.h file
if (IsNormalizedString(NormalizationC, szOut, iLen + 1) == TRUE)
return iLen;
int iNewLen = NormalizeString(NormalizationC, szOut, iLen, NULL, 0);
if (iNewLen <= 0)
{
return iLen;
}
CZipString szNormalized;
iNewLen = NormalizeString(NormalizationC, szOut, iLen, szNormalized.GetBuffer(iNewLen), iNewLen);
if (iNewLen <= 0)
{
szNormalized.ReleaseBuffer(0);
return iLen;
}
szNormalized.ReleaseBuffer(iNewLen);
szOut = szNormalized;
return iNewLen;
#else
return iLen;
#endif
}
else
{
return -1;
}
}
BOOL CZipException::GetErrorMessage(LPTSTR lpszError, UINT nMaxError,
UINT* )
{
if (!lpszError || !nMaxError)
return FALSE;
CZipString sz = GetErrorDescription();
if (sz.IsEmpty())
return FALSE;
UINT iLen = sz.GetLength();
if (nMaxError - 1 < iLen)
iLen = nMaxError - 1;
LPTSTR lpsz = sz.GetBuffer(iLen);
#if _MSC_VER >= 1400
#ifdef _UNICODE
wcsncpy_s(lpszError, nMaxError, lpsz, iLen);
#else
strncpy_s(lpszError, nMaxError, lpsz, iLen);
#endif
#else
#ifdef _UNICODE
wcsncpy(lpszError, lpsz, iLen);
#else
strncpy(lpszError, lpsz, iLen);
#endif
#endif
lpszError[iLen] = _T('\0');
return TRUE;
}
int ZipPlatform::SingleToWide(const CZipAutoBuffer &szSingle, CZipString& szWide, bool bUseAnsi)
{
int singleLen = szSingle.GetSize();
// iLen doesn't include terminating character
UINT uCodePage;
DWORD dwFlags;
if (bUseAnsi)
{
uCodePage = CP_ACP;
dwFlags = MB_PRECOMPOSED;
}
else
{
uCodePage = CP_UTF8;
dwFlags = 0;
}
int iLen = MultiByteToWideChar(uCodePage, dwFlags, szSingle.GetBuffer(), singleLen, NULL, 0);
if (iLen > 0)
{
iLen = MultiByteToWideChar(uCodePage, dwFlags, szSingle.GetBuffer(), singleLen,
szWide.GetBuffer(iLen) , iLen);
szWide.ReleaseBuffer(iLen);
ASSERT(iLen != 0);
}
else
{
szWide.Empty();
iLen --; // return -1
}
return iLen;
}
void ZipCompatibility::ConvertBufferToString(CZipString& szString, const CZipAutoBuffer& buffer, UINT uCodePage)
{
#ifdef _UNICODE
ZipPlatform::MultiByteToWide(buffer, szString, uCodePage);
#else
// iLen does not include the NULL character
int iLen;
if (uCodePage == CP_OEMCP)
{
CZipAutoBuffer buf;
buf = buffer;
ZipPlatform::AnsiOem(buf, false);
iLen = buf.GetSize();
memcpy(szString.GetBuffer(iLen), buf.GetBuffer(), iLen);
}
else
{
iLen = buffer.GetSize();
memcpy(szString.GetBuffer(iLen), buffer.GetBuffer(), iLen);
}
szString.ReleaseBuffer(iLen);
#endif
}
int ZipPlatform::SingleToWide(const CZipAutoBuffer &szSingle, CZipString& szWide)
{
int singleLen = szSingle.GetSize();
// iLen doesn't include terminating character
int iLen = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, szSingle.GetBuffer(), singleLen, NULL, 0);
if (iLen > 0)
{
iLen = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, szSingle.GetBuffer(), singleLen,
szWide.GetBuffer(iLen) , iLen);
szWide.ReleaseBuffer(iLen);
ASSERT(iLen != 0);
}
else
{
szWide.Empty();
iLen --; // return -1
}
return iLen;
}
CZipString ZipPlatform::GetTmpFileName(LPCTSTR lpszPath, ZIP_U32_U64 uSizeNeeded)
{
TCHAR empty[] = _T(""), prefix [] = _T("zar");
TCHAR* buf = NULL;
CZipString tempPath = lpszPath;
if (tempPath.IsEmpty())
tempPath = "/tmp";
if (ZipPlatform::GetDeviceFreeSpace(tempPath) < uSizeNeeded)
return empty;
CZipPathComponent::AppendSeparator(tempPath);
tempPath += prefix;
tempPath += _T("XXXXXX");
int handle = mkstemp(tempPath.GetBuffer(tempPath.GetLength()));
tempPath.ReleaseBuffer();
if (handle != -1)
{
close(handle); // we just create the file and open it later
return tempPath;
}
else
return empty;
}
ZIPINLINE bool ZipPlatform::RemoveFile(LPCTSTR lpszFileName, bool bThrow, int iMode)
{
if ((iMode & ZipPlatform::dfmRemoveReadOnly) != 0)
{
DWORD attr;
if (ZipPlatform::GetFileAttr(lpszFileName, attr)
&& (ZipCompatibility::GetAsInternalAttributes(attr, ZipPlatform::GetSystemID()) & ZipCompatibility::attROnly) != 0)
{
ZipPlatform::SetFileAttr(lpszFileName, ZipPlatform::GetDefaultAttributes());
}
}
#ifdef SHFileOperation
if ((iMode & ZipPlatform::dfmRecycleBin) == 0)
{
#endif
if (::DeleteFile((LPTSTR)(LPCTSTR)CZipPathComponent::AddPrefix(lpszFileName, false)))
return true;
#ifdef SHFileOperation
}
else
{
CZipString file = lpszFileName;
#if defined _UNICODE && _MSC_VER >= 1400
if (file.GetLength() >= MAX_PATH)
{
// cannot prefix for SHFileOperation, use short path
CZipString temp = CZipPathComponent::AddPrefix(lpszFileName, false);
DWORD size = ::GetShortPathName(temp, NULL, 0);
if (size != 0)
{
size = ::GetShortPathName(temp, file.GetBuffer(size), size);
file.ReleaseBuffer();
}
if (size == 0)
{
if (bThrow)
CZipException::Throw(CZipException::notRemoved, lpszFileName);
return false;
}
// GetShortPathName keeps the prefix - remove it
int prefixLength = CZipPathComponent::IsPrefixed(file);
if (prefixLength > 0)
{
file = file.Mid(prefixLength);
}
}
#endif
int length = file.GetLength();
CZipAutoBuffer buffer((length + 2) * sizeof(TCHAR), true); // double NULL is required
memcpy(buffer, (LPCTSTR)file, length * sizeof(TCHAR));
SHFILEOPSTRUCT shFileOp = {0};
shFileOp.wFunc = FO_DELETE;
shFileOp.fFlags = FOF_ALLOWUNDO | FOF_NOCONFIRMATION | FOF_SILENT | FOF_NOERRORUI;
shFileOp.pFrom = (LPCTSTR)(char*)buffer;
if (SHFileOperation(&shFileOp) == 0 && !shFileOp.fAnyOperationsAborted)
return true;
}
#endif
if (bThrow)
CZipException::Throw(CZipException::notRemoved, lpszFileName);
return false;
}