mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-07-02 05:29:35 +00:00
148 lines
3.4 KiB
C++
148 lines
3.4 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)
|
|
{
|
|
if (logFP) {
|
|
return;
|
|
}
|
|
|
|
int 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])))) {
|
|
#ifdef XP_WIN
|
|
GetTempFileNameW(sourcePath, L"log", 0, mTmpFilePath);
|
|
logFP = NS_tfopen(mTmpFilePath, 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);
|
|
#elif XP_MACOSX
|
|
logFP = NS_tfopen(mDstFilePath, NS_T("w"));
|
|
#else
|
|
// On platforms that have an updates directory in the installation directory
|
|
// (e.g. platforms other than Windows and Mac) the update log is written to
|
|
// a temporary file and then to the update log file. This is needed since
|
|
// the installation directory is moved during a replace request. This can be
|
|
// removed when the platform's updates directory is located outside of the
|
|
// installation directory.
|
|
logFP = tmpfile();
|
|
#endif
|
|
}
|
|
}
|
|
|
|
void UpdateLog::Finish()
|
|
{
|
|
if (!logFP) {
|
|
return;
|
|
}
|
|
|
|
#if !defined(XP_WIN) && !defined(XP_MACOSX)
|
|
const int blockSize = 1024;
|
|
char buffer[blockSize];
|
|
fflush(logFP);
|
|
rewind(logFP);
|
|
|
|
FILE *updateLogFP = NS_tfopen(mDstFilePath, NS_T("wb+"));
|
|
while (!feof(logFP)) {
|
|
size_t read = fread(buffer, 1, blockSize, logFP);
|
|
if (ferror(logFP)) {
|
|
fclose(logFP);
|
|
logFP = nullptr;
|
|
fclose(updateLogFP);
|
|
updateLogFP = nullptr;
|
|
return;
|
|
}
|
|
|
|
size_t written = 0;
|
|
|
|
while (written < read) {
|
|
size_t chunkWritten = fwrite(buffer, 1, read - written, updateLogFP);
|
|
if (chunkWritten <= 0) {
|
|
fclose(logFP);
|
|
logFP = nullptr;
|
|
fclose(updateLogFP);
|
|
updateLogFP = nullptr;
|
|
return;
|
|
}
|
|
|
|
written += chunkWritten;
|
|
}
|
|
}
|
|
fclose(updateLogFP);
|
|
updateLogFP = nullptr;
|
|
#endif
|
|
|
|
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);
|
|
}
|