mirror of
https://github.com/roytam1/palemoon27.git
synced 2026-06-02 10:49:56 +00:00
Pale Moon
122 lines
3.0 KiB
C++
122 lines
3.0 KiB
C++
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#if defined(XP_WIN)
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
|
|
#include "updatelogging.h"
|
|
|
|
UpdateLog::UpdateLog() : logFP(nullptr)
|
|
{
|
|
}
|
|
|
|
void UpdateLog::Init(NS_tchar* sourcePath,
|
|
const NS_tchar* fileName,
|
|
const NS_tchar* alternateFileName,
|
|
bool append)
|
|
{
|
|
if (logFP)
|
|
return;
|
|
|
|
#ifdef XP_WIN
|
|
GetTempFileNameW(sourcePath, L"log", 0, mTmpFilePath);
|
|
int dstFilePathLen = NS_tsnprintf(mDstFilePath,
|
|
sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
|
|
NS_T("%s/%s"), sourcePath, append ? alternateFileName : fileName);
|
|
|
|
// If the destination path was over the length limit,
|
|
// disable logging by skipping opening the file and setting logFP.
|
|
if ((dstFilePathLen > 0) &&
|
|
(dstFilePathLen <
|
|
static_cast<int>(sizeof(mDstFilePath)/sizeof(mDstFilePath[0])))) {
|
|
if (append) {
|
|
CopyFileW(mDstFilePath, mTmpFilePath, FALSE);
|
|
}
|
|
logFP = NS_tfopen(mTmpFilePath, append ? NS_T("a") : NS_T("w"));
|
|
|
|
// Delete this file now so it is possible to tell from the unelevated
|
|
// updater process if the elevated updater process has written the log.
|
|
DeleteFileW(mDstFilePath);
|
|
}
|
|
#else
|
|
int dstFilePathLen = 0;
|
|
if (alternateFileName && NS_taccess(mDstFilePath, F_OK)) {
|
|
dstFilePathLen = NS_tsnprintf(mDstFilePath,
|
|
sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
|
|
NS_T("%s/%s"), sourcePath, alternateFileName);
|
|
} else {
|
|
dstFilePathLen = NS_tsnprintf(mDstFilePath,
|
|
sizeof(mDstFilePath)/sizeof(mDstFilePath[0]),
|
|
NS_T("%s/%s"), sourcePath, fileName);
|
|
}
|
|
|
|
// If the destination path was over the length limit,
|
|
// disable logging by skipping opening the file and setting logFP.
|
|
if ((dstFilePathLen > 0) &&
|
|
(dstFilePathLen <
|
|
static_cast<int>(sizeof(mDstFilePath)/sizeof(mDstFilePath[0])))) {
|
|
logFP = NS_tfopen(mDstFilePath, append ? NS_T("a") : NS_T("w"));
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void UpdateLog::Finish()
|
|
{
|
|
if (!logFP)
|
|
return;
|
|
|
|
fclose(logFP);
|
|
logFP = nullptr;
|
|
|
|
#ifdef XP_WIN
|
|
// When the log file already exists then the elevated updater has already
|
|
// written the log file and the temp file for the log should be discarded.
|
|
if (!NS_taccess(mDstFilePath, F_OK)) {
|
|
DeleteFileW(mTmpFilePath);
|
|
} else {
|
|
MoveFileW(mTmpFilePath, mDstFilePath);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void UpdateLog::Flush()
|
|
{
|
|
if (!logFP)
|
|
return;
|
|
|
|
fflush(logFP);
|
|
}
|
|
|
|
void UpdateLog::Printf(const char *fmt, ... )
|
|
{
|
|
if (!logFP)
|
|
return;
|
|
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
vfprintf(logFP, fmt, ap);
|
|
fprintf(logFP, "\n");
|
|
va_end(ap);
|
|
}
|
|
|
|
void UpdateLog::WarnPrintf(const char *fmt, ... )
|
|
{
|
|
if (!logFP)
|
|
return;
|
|
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
fprintf(logFP, "*** Warning: ");
|
|
vfprintf(logFP, fmt, ap);
|
|
fprintf(logFP, "***\n");
|
|
va_end(ap);
|
|
}
|