374 lines
11 KiB
C++
374 lines
11 KiB
C++
#include <wininetp.h>
|
|
|
|
|
|
#define SZ_COOKIE_EXPORT_HEADER "# Internet Explorer cookie file, exported for Netscape browsers."
|
|
#define IE_LOCAL_PREFIX "~~local~~"
|
|
#define IE_COOKIE_PREFIX "Cookie:"
|
|
|
|
|
|
void FileTimeToDosTime( FILETIME ft, DWORD* pTime_t);
|
|
|
|
|
|
struct CACHE_ENTRY_INFO_BUFFER : public INTERNET_CACHE_ENTRY_INFO
|
|
{
|
|
BYTE _ab[MAX_CACHE_ENTRY_INFO_SIZE];
|
|
};
|
|
|
|
|
|
//****************************************************
|
|
//
|
|
// FileOutputStream - utility
|
|
//
|
|
|
|
class FileOutputStream
|
|
{
|
|
public:
|
|
FileOutputStream()
|
|
: m_hFile(INVALID_HANDLE_VALUE), m_fError(FALSE), m_dwLastError(0)
|
|
{
|
|
}
|
|
|
|
~FileOutputStream()
|
|
{
|
|
if( m_hFile != INVALID_HANDLE_VALUE)
|
|
CloseHandle( m_hFile);
|
|
}
|
|
|
|
BOOL Load( LPCTSTR szFilename, BOOL fAppend)
|
|
{
|
|
m_hFile = CreateFile( szFilename, GENERIC_WRITE | GENERIC_READ, 0, NULL,
|
|
fAppend ? OPEN_ALWAYS : CREATE_ALWAYS,
|
|
FILE_ATTRIBUTE_NORMAL, NULL);
|
|
|
|
if( m_hFile == INVALID_HANDLE_VALUE)
|
|
{
|
|
m_fError = TRUE;
|
|
m_dwLastError = GetLastError();
|
|
return FALSE;
|
|
}
|
|
|
|
if( fAppend
|
|
&& SetFilePointer( m_hFile, 0, NULL, FILE_END) == 0xFFFFFFFF)
|
|
{
|
|
m_fError = TRUE;
|
|
m_dwLastError = GetLastError();
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL DumpStr( LPCSTR szString, DWORD cbSize)
|
|
{
|
|
DWORD dwTemp;
|
|
|
|
if( m_fError == TRUE)
|
|
return FALSE;
|
|
|
|
if( WriteFile( m_hFile, szString, cbSize, &dwTemp, NULL) == TRUE)
|
|
{
|
|
return TRUE;
|
|
}
|
|
else
|
|
{
|
|
m_fError = TRUE;
|
|
m_dwLastError = GetLastError();
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
BOOL WriteBool( BOOL fBool)
|
|
{
|
|
if( fBool == TRUE)
|
|
return DumpStr( "TRUE", sizeof("TRUE") - 1);
|
|
else
|
|
return DumpStr( "FALSE", sizeof("FALSE") - 1);
|
|
}
|
|
|
|
BOOL WriteTab()
|
|
{
|
|
return DumpStr( "\t", sizeof('\t'));
|
|
}
|
|
|
|
BOOL WriteNSNewline()
|
|
{
|
|
return DumpStr( "\r\n\r\n", sizeof("\r\n\r\n") - 1);
|
|
}
|
|
|
|
BOOL IsError()
|
|
{
|
|
return m_fError;
|
|
}
|
|
|
|
private:
|
|
HANDLE m_hFile;
|
|
BOOL m_fError;
|
|
DWORD m_dwLastError;
|
|
};
|
|
|
|
|
|
//****************************************************
|
|
//
|
|
// ExportCookieFile
|
|
//
|
|
|
|
// per-line format of Netscape cookie file
|
|
//DOMAIN \t is_given_domain(TRUE|FALSE) \t PATH \t secure(TRUE|FALSE) \t time_t(EXPIRES) \t NAME \t VALUE
|
|
|
|
BOOLAPI ExportCookieFileW( IN LPCWSTR szFilename, IN BOOL fAppend)
|
|
{
|
|
MEMORYPACKET mpFilename;
|
|
ALLOC_MB(szFilename,0,mpFilename);
|
|
if (!mpFilename.psStr)
|
|
{
|
|
return FALSE;
|
|
}
|
|
UNICODE_TO_ANSI(szFilename,mpFilename);
|
|
|
|
return ExportCookieFileA( mpFilename.psStr, fAppend);
|
|
};
|
|
|
|
|
|
|
|
BOOLAPI ExportCookieFileA( IN LPCSTR szFilename, IN BOOL fAppend)
|
|
{
|
|
BOOL retVal = FALSE;
|
|
|
|
CHAR* cursor;
|
|
DWORD dwTemp;
|
|
|
|
FileOutputStream outFile;
|
|
|
|
CACHE_ENTRY_INFO_BUFFER *pcei = new CACHE_ENTRY_INFO_BUFFER;
|
|
|
|
if (pcei == NULL)
|
|
goto doneExportCookieFile;
|
|
|
|
DWORD cbCeiSize = sizeof(*pcei);
|
|
HANDLE hEnumeration = FindFirstUrlCacheEntry( IE_COOKIE_PREFIX, pcei, &cbCeiSize);
|
|
if( hEnumeration == NULL)
|
|
goto doneExportCookieFile;
|
|
|
|
if( outFile.Load( szFilename, fAppend) != TRUE)
|
|
goto doneExportCookieFile;
|
|
|
|
if( !fAppend)
|
|
{
|
|
outFile.DumpStr( SZ_COOKIE_EXPORT_HEADER, sizeof(SZ_COOKIE_EXPORT_HEADER) - 1);
|
|
outFile.WriteNSNewline();
|
|
}
|
|
|
|
if( outFile.IsError() == TRUE)
|
|
goto doneExportCookieFile;
|
|
|
|
// enumerate over the cookie cache entries.
|
|
//Each cache entry will have a file which contains multiple cookies,
|
|
//so inside this loop we will be enumerating over individual cookies.
|
|
do
|
|
{
|
|
CHAR* pFileBuf = NULL;
|
|
DWORD cbFileBufSize;
|
|
CHAR* pDomain;
|
|
|
|
//Generate the DOMAIN for all the cookies in this entry
|
|
// example Urls in pEntry:
|
|
// "Cookie:t-franks@amazon.com/"
|
|
// "Cookie:t-franks@~~local~~/c:\local\path\"
|
|
// The correspondingDomain entry in NS cookie.txt:
|
|
// "amazon.com"
|
|
// "" -> local paths are empty strings to NS
|
|
cursor = pcei->lpszSourceUrlName;
|
|
|
|
while( *cursor != '@' && *cursor != '\0')
|
|
cursor++;
|
|
|
|
if( *cursor != '@')
|
|
goto doneWithEntryOnExportCookie;
|
|
|
|
pDomain = ++cursor;
|
|
|
|
if( StrCmpN( pDomain, IE_LOCAL_PREFIX, sizeof(IE_LOCAL_PREFIX) - 1) != 0)
|
|
{
|
|
// if we have a non-local domain, clip off the path.
|
|
while( *cursor != '/' && *cursor != '\0')
|
|
cursor++;
|
|
|
|
if( *cursor != '/')
|
|
goto doneWithEntryOnExportCookie;
|
|
|
|
*cursor = '\0';
|
|
}
|
|
else
|
|
{
|
|
// Else local cookies have a blank domain in the Netscape cookie files.
|
|
pDomain[0] = '\0';
|
|
}
|
|
|
|
// load the attached file and enumerate through the contained cookies.
|
|
if( (pcei->lpszLocalFileName == NULL)
|
|
|| (ReadFileToBuffer( pcei->lpszLocalFileName, (BYTE**)&pFileBuf, &cbFileBufSize)
|
|
!= TRUE))
|
|
{
|
|
goto doneWithEntryOnExportCookie;
|
|
}
|
|
|
|
CHAR* pFileCursor;
|
|
pFileCursor = pFileBuf;
|
|
while( pFileCursor < pFileBuf + cbFileBufSize)
|
|
{
|
|
CHAR *pszName, *pszValue, *pszHash, *pszFlags,
|
|
*pszExpireTimeLow, *pszExpireTimeHigh,
|
|
*pszLastTimeHigh, *pszLastTimeLow,
|
|
*pszDelimiter;
|
|
FILETIME ftExpire;
|
|
DWORD dwFlags;
|
|
|
|
// Get the first token (cookie name).
|
|
pszName = StrTokEx(&pFileCursor, "\n");
|
|
if (!pszName) // Cookie name.
|
|
{
|
|
// Normal termination of the parse.
|
|
goto doneWithEntryOnExportCookie;
|
|
}
|
|
|
|
// Parse the rest of the cookie
|
|
if( *pFileCursor == '\n')
|
|
{
|
|
pszValue = pFileCursor;
|
|
*pszValue = '\0';
|
|
pFileCursor++;
|
|
}
|
|
else
|
|
pszValue = StrTokEx(&pFileCursor, "\n"); // Cookie value.
|
|
|
|
pszHash = StrTokEx(&pFileCursor, "\n"); // Combo of domain and path.
|
|
pszFlags = StrTokEx(&pFileCursor, "\n"); // Cookie flags.
|
|
pszExpireTimeLow = StrTokEx(&pFileCursor, "\n"); // Expire time.
|
|
pszExpireTimeHigh = StrTokEx(&pFileCursor, "\n");
|
|
pszLastTimeLow = StrTokEx(&pFileCursor, "\n"); // Last Modified time.
|
|
pszLastTimeHigh = StrTokEx(&pFileCursor, "\n");
|
|
pszDelimiter = StrTokEx(&pFileCursor, "\n"); // Delimiter should be "*"
|
|
|
|
// Abnormal termination of parse.
|
|
if (!pszDelimiter || pszDelimiter[0] != '*')
|
|
{
|
|
INET_ASSERT(FALSE);
|
|
goto doneWithEntryOnExportCookie;
|
|
}
|
|
|
|
// Get the expire time.
|
|
ftExpire.dwLowDateTime = atoi(pszExpireTimeLow);
|
|
ftExpire.dwHighDateTime = atoi(pszExpireTimeHigh);
|
|
|
|
// Get the flags
|
|
dwFlags = atoi(pszFlags);
|
|
|
|
// If this is a session cookie, its non-persistent
|
|
//and shouldn't be saved to file.
|
|
// (session cookies shouldn't be in the index file, anyhow)
|
|
INET_ASSERT( (dwFlags & COOKIE_SESSION) == 0 ? TRUE : FALSE);
|
|
if( (dwFlags & COOKIE_SESSION) != 0)
|
|
continue;
|
|
|
|
// process parsed contents.
|
|
CHAR* pszNSPath;
|
|
DWORD timetExpire;
|
|
|
|
// Example IE cookie file:
|
|
// pszHash = "amazon.com/main/"
|
|
// pszHash = "~~local~~/c:\local\path\"
|
|
// corresponding Netscape path format:
|
|
// pszNSPath = "/main/"
|
|
// pszNSPath = "/c|/local/path/"
|
|
cursor = pszHash;
|
|
|
|
// move to the first '/'
|
|
while( *cursor != '/' && *cursor != '\0')
|
|
cursor++;
|
|
if( *cursor != '/')
|
|
continue;
|
|
|
|
pszNSPath = cursor;
|
|
|
|
// If this is a path on the Hard Disk, turn the '\\'s to '/'s
|
|
//and ':'s to '|'s. (to conform with funky NS format)
|
|
if( StrCmpN( pszHash, IE_LOCAL_PREFIX, sizeof(IE_LOCAL_PREFIX) - 1) == 0)
|
|
{
|
|
while( *cursor != '\0')
|
|
{
|
|
if( *cursor == '\\')
|
|
*cursor = '/';
|
|
|
|
if( *cursor == ':')
|
|
*cursor = '|';
|
|
|
|
cursor++;
|
|
}
|
|
}
|
|
|
|
// convert to Netscape time format
|
|
FileTimeToDosTime( ftExpire, &timetExpire);
|
|
|
|
outFile.DumpStr( pDomain, lstrlen( pDomain));
|
|
outFile.WriteTab();
|
|
outFile.WriteBool( TRUE);
|
|
outFile.WriteTab();
|
|
outFile.DumpStr( pszNSPath, lstrlen( pszNSPath));
|
|
outFile.WriteTab();
|
|
outFile.WriteBool( (dwFlags & COOKIE_SECURE) != 0 ? TRUE : FALSE);
|
|
outFile.WriteTab();
|
|
|
|
CHAR expireBuffer[16];
|
|
wsprintf( expireBuffer, "%lu", timetExpire);
|
|
outFile.DumpStr( expireBuffer, lstrlen( expireBuffer));
|
|
outFile.WriteTab();
|
|
|
|
outFile.DumpStr( pszName, lstrlen( pszName));
|
|
outFile.WriteTab();
|
|
outFile.DumpStr( pszValue, lstrlen( pszValue));
|
|
outFile.WriteNSNewline();
|
|
}
|
|
|
|
doneWithEntryOnExportCookie:
|
|
if( pFileBuf != NULL)
|
|
delete [] pFileBuf;
|
|
|
|
if( outFile.IsError() == TRUE)
|
|
goto doneExportCookieFile;
|
|
}
|
|
while( FindNextUrlCacheEntry( hEnumeration, pcei, &(cbCeiSize = sizeof(*pcei))));
|
|
|
|
retVal = TRUE;
|
|
|
|
doneExportCookieFile:
|
|
|
|
if (pcei)
|
|
delete pcei;
|
|
|
|
if( hEnumeration != NULL)
|
|
FindCloseUrlCache( hEnumeration);
|
|
|
|
return retVal;
|
|
}
|
|
|
|
// reverse of part of transformation in InternetTimeFromTime_t
|
|
// A filetime is the numbers of 100 ns since Jan 1, 1601, while
|
|
//a dostime is the number of seconds since Jan 1, 1970.
|
|
void FileTimeToDosTime( FILETIME ft, DWORD* pTime_t)
|
|
{
|
|
//dwl1970Offset is the number of ns from 1601 to 1970
|
|
const DWORDLONG dwl1970Offset = 0x019dae9064bafa80;
|
|
|
|
DWORDLONG lVal;
|
|
|
|
lVal = ft.dwLowDateTime;
|
|
lVal |= Int64ShllMod32( Int64ShllMod32( ft.dwHighDateTime, 16), 16 );
|
|
|
|
*pTime_t = (DWORD)((lVal - dwl1970Offset) / 10000000);
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
|