mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-19 08:15:31 +00:00
86cb4f4e4d
While looking at this code I found a couple of places where errors could get ignored or silently discarded and result in corrupt data. This checks for the errors and fails harder. Depends on D10353 Differential Revision: https://phabricator.services.mozilla.com/D10354 --HG-- extra : moz-landing-system : lando
133 lines
3.2 KiB
C++
133 lines
3.2 KiB
C++
/* -*- 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
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "FileOperations.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
#include <direct.h>
|
|
#include <io.h>
|
|
#include <windows.h>
|
|
#include "StringOperations.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 &Filename) {
|
|
std::string Hash = hash(Filename);
|
|
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;
|
|
}
|
|
|
|
FileDescriptor = _open(Filename.c_str(), _O_RDWR | _O_CREAT | _O_BINARY, 0666);
|
|
}
|
|
|
|
AutoLockFile::~AutoLockFile() {
|
|
_close(FileDescriptor);
|
|
|
|
ReleaseMutex(Handle);
|
|
CloseHandle(Handle);
|
|
}
|
|
|
|
bool AutoLockFile::success() {
|
|
return Handle != NULL && FileDescriptor != -1;
|
|
}
|
|
|
|
FILE *AutoLockFile::openFile(const char *Mode) {
|
|
_lseek(FileDescriptor, 0, SEEK_SET);
|
|
return _fdopen(_dup(FileDescriptor), Mode);
|
|
}
|
|
|
|
bool AutoLockFile::truncateFile(size_t Length) {
|
|
return _chsize(FileDescriptor, Length) == 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 &Filename) {
|
|
FileDescriptor = open(Filename.c_str(), O_RDWR | O_CREAT, 0666);
|
|
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::openFile(const char *Mode) {
|
|
lseek(FileDescriptor, 0, SEEK_SET);
|
|
return fdopen(dup(FileDescriptor), Mode);
|
|
}
|
|
|
|
bool AutoLockFile::truncateFile(size_t Length) {
|
|
return ftruncate(FileDescriptor, Length) == 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
|