Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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
#include "FileOperations.h"
#include <stdio.h>
#include <stdlib.h>
#if defined(_WIN32) || defined(_WIN64)
#include "StringOperations.h"
#include <direct.h>
#include <io.h>
#include <windows.h>
#else
#include <sys/file.h>
#include <sys/time.h>
#include <unistd.h>
#endif
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/types.h>
// Make sure that all directories on path exist, excluding the final element of
// the path.
void ensurePath(std::string Path) {
size_t Pos = 0;
if (Path[0] == PATHSEP_CHAR) {
Pos++;
}
while ((Pos = Path.find(PATHSEP_CHAR, Pos)) != std::string::npos) {
std::string Portion = Path.substr(0, Pos);
if (!Portion.empty()) {
#if defined(_WIN32) || defined(_WIN64)
int Err = _mkdir(Portion.c_str());
#else
int Err = mkdir(Portion.c_str(), 0775);
#endif
if (Err == -1 && errno != EEXIST) {
perror("mkdir failed");
exit(1);
}
}
Pos++;
}
}
#if defined(_WIN32) || defined(_WIN64)
AutoLockFile::AutoLockFile(const std::string &SrcFile,
const std::string &DstFile) {
this->Filename = DstFile;
std::string Hash = hash(SrcFile);
std::string MutexName = std::string("Local\\searchfox-") + Hash;
std::wstring WideMutexName;
WideMutexName.assign(MutexName.begin(), MutexName.end());
Handle = CreateMutex(nullptr, false, WideMutexName.c_str());
if (Handle == NULL) {
return;
}
if (WaitForSingleObject(Handle, INFINITE) != WAIT_OBJECT_0) {
return;
}
}
AutoLockFile::~AutoLockFile() {
ReleaseMutex(Handle);
CloseHandle(Handle);
}
bool AutoLockFile::success() { return Handle != NULL; }
FILE *AutoLockFile::openTmp() {
int TmpDescriptor = _open((Filename + ".tmp").c_str(),
_O_WRONLY | _O_APPEND | _O_CREAT | _O_BINARY, 0666);
return _fdopen(TmpDescriptor, "ab");
}
bool AutoLockFile::moveTmp() {
if (_unlink(Filename.c_str()) == -1) {
if (errno != ENOENT) {
return false;
}
}
return rename((Filename + ".tmp").c_str(), Filename.c_str()) == 0;
}
std::string getAbsolutePath(const std::string &Filename) {
char Full[_MAX_PATH];
if (!_fullpath(Full, Filename.c_str(), _MAX_PATH)) {
return std::string("");
}
return std::string(Full);
}
#else
AutoLockFile::AutoLockFile(const std::string &SrcFile,
const std::string &DstFile) {
this->Filename = DstFile;
FileDescriptor = open(SrcFile.c_str(), O_RDONLY);
if (FileDescriptor == -1) {
return;
}
do {
int rv = flock(FileDescriptor, LOCK_EX);
if (rv == 0) {
break;
}
} while (true);
}
AutoLockFile::~AutoLockFile() { close(FileDescriptor); }
bool AutoLockFile::success() { return FileDescriptor != -1; }
FILE *AutoLockFile::openTmp() {
int TmpDescriptor =
open((Filename + ".tmp").c_str(), O_WRONLY | O_APPEND | O_CREAT, 0666);
return fdopen(TmpDescriptor, "ab");
}
bool AutoLockFile::moveTmp() {
if (unlink(Filename.c_str()) == -1) {
if (errno != ENOENT) {
return false;
}
}
return rename((Filename + ".tmp").c_str(), Filename.c_str()) == 0;
}
std::string getAbsolutePath(const std::string &Filename) {
char Full[4096];
if (!realpath(Filename.c_str(), Full)) {
return std::string("");
}
return std::string(Full);
}
#endif