mirror of
https://repo.dactyloidae.xyz/Dactyloidae/UXP.git
synced 2026-09-07 08:18:41 +09:00
import FIREFOX_52_6_0esr_RELEASE from mozilla-esr52 hg repo
This commit is contained in:
Roy Tam
commit
dcd9973243
150858 changed files with 23884658 additions and 0 deletions
331
xpcom/glue/nsINIParser.cpp
Normal file
331
xpcom/glue/nsINIParser.cpp
Normal file
|
|
@ -0,0 +1,331 @@
|
|||
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
||||
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
||||
/* 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/. */
|
||||
|
||||
// Moz headers (alphabetical)
|
||||
#include "nsCRTGlue.h"
|
||||
#include "nsError.h"
|
||||
#include "nsIFile.h"
|
||||
#include "nsINIParser.h"
|
||||
#include "mozilla/FileUtils.h" // AutoFILE
|
||||
|
||||
// System headers (alphabetical)
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#ifdef XP_WIN
|
||||
#include <windows.h>
|
||||
#endif
|
||||
|
||||
#if defined(XP_WIN)
|
||||
#define READ_BINARYMODE L"rb"
|
||||
#else
|
||||
#define READ_BINARYMODE "r"
|
||||
#endif
|
||||
|
||||
using namespace mozilla;
|
||||
|
||||
#ifdef XP_WIN
|
||||
inline FILE*
|
||||
TS_tfopen(const char* aPath, const wchar_t* aMode)
|
||||
{
|
||||
wchar_t wPath[MAX_PATH];
|
||||
MultiByteToWideChar(CP_UTF8, 0, aPath, -1, wPath, MAX_PATH);
|
||||
return _wfopen(wPath, aMode);
|
||||
}
|
||||
#else
|
||||
inline FILE*
|
||||
TS_tfopen(const char* aPath, const char* aMode)
|
||||
{
|
||||
return fopen(aPath, aMode);
|
||||
}
|
||||
#endif
|
||||
|
||||
// Stack based FILE wrapper to ensure that fclose is called, copied from
|
||||
// toolkit/mozapps/update/updater/readstrings.cpp
|
||||
|
||||
class AutoFILE
|
||||
{
|
||||
public:
|
||||
explicit AutoFILE(FILE* aFp = nullptr) : fp_(aFp) {}
|
||||
~AutoFILE()
|
||||
{
|
||||
if (fp_) {
|
||||
fclose(fp_);
|
||||
}
|
||||
}
|
||||
operator FILE*() { return fp_; }
|
||||
FILE** operator&() { return &fp_; }
|
||||
void operator=(FILE* aFp) { fp_ = aFp; }
|
||||
private:
|
||||
FILE* fp_;
|
||||
};
|
||||
|
||||
nsresult
|
||||
nsINIParser::Init(nsIFile* aFile)
|
||||
{
|
||||
/* open the file. Don't use OpenANSIFileDesc, because you mustn't
|
||||
pass FILE* across shared library boundaries, which may be using
|
||||
different CRTs */
|
||||
|
||||
AutoFILE fd;
|
||||
|
||||
#ifdef XP_WIN
|
||||
nsAutoString path;
|
||||
nsresult rv = aFile->GetPath(path);
|
||||
if (NS_WARN_IF(NS_FAILED(rv))) {
|
||||
return rv;
|
||||
}
|
||||
|
||||
fd = _wfopen(path.get(), READ_BINARYMODE);
|
||||
#else
|
||||
nsAutoCString path;
|
||||
aFile->GetNativePath(path);
|
||||
|
||||
fd = fopen(path.get(), READ_BINARYMODE);
|
||||
#endif
|
||||
|
||||
if (!fd) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
return InitFromFILE(fd);
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsINIParser::Init(const char* aPath)
|
||||
{
|
||||
/* open the file */
|
||||
AutoFILE fd(TS_tfopen(aPath, READ_BINARYMODE));
|
||||
if (!fd) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
return InitFromFILE(fd);
|
||||
}
|
||||
|
||||
static const char kNL[] = "\r\n";
|
||||
static const char kEquals[] = "=";
|
||||
static const char kWhitespace[] = " \t";
|
||||
static const char kRBracket[] = "]";
|
||||
|
||||
nsresult
|
||||
nsINIParser::InitFromFILE(FILE* aFd)
|
||||
{
|
||||
/* get file size */
|
||||
if (fseek(aFd, 0, SEEK_END) != 0) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
long flen = ftell(aFd);
|
||||
/* zero-sized file, or an error */
|
||||
if (flen <= 0) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
/* malloc an internal buf the size of the file */
|
||||
mFileContents = MakeUnique<char[]>(flen + 2);
|
||||
if (!mFileContents) {
|
||||
return NS_ERROR_OUT_OF_MEMORY;
|
||||
}
|
||||
|
||||
/* read the file in one swoop */
|
||||
if (fseek(aFd, 0, SEEK_SET) != 0) {
|
||||
return NS_BASE_STREAM_OSERROR;
|
||||
}
|
||||
|
||||
int rd = fread(mFileContents.get(), sizeof(char), flen, aFd);
|
||||
if (rd != flen) {
|
||||
return NS_BASE_STREAM_OSERROR;
|
||||
}
|
||||
|
||||
// We write a UTF16 null so that the file is easier to convert to UTF8
|
||||
mFileContents[flen] = mFileContents[flen + 1] = '\0';
|
||||
|
||||
char* buffer = &mFileContents[0];
|
||||
|
||||
if (flen >= 3 &&
|
||||
mFileContents[0] == '\xEF' &&
|
||||
mFileContents[1] == '\xBB' &&
|
||||
mFileContents[2] == '\xBF') {
|
||||
// Someone set us up the Utf-8 BOM
|
||||
// This case is easy, since we assume that BOM-less
|
||||
// files are Utf-8 anyway. Just skip the BOM and process as usual.
|
||||
buffer = &mFileContents[3];
|
||||
}
|
||||
|
||||
#ifdef XP_WIN
|
||||
if (flen >= 2 &&
|
||||
mFileContents[0] == '\xFF' &&
|
||||
mFileContents[1] == '\xFE') {
|
||||
// Someone set us up the Utf-16LE BOM
|
||||
buffer = &mFileContents[2];
|
||||
// Get the size required for our Utf8 buffer
|
||||
flen = WideCharToMultiByte(CP_UTF8,
|
||||
0,
|
||||
reinterpret_cast<LPWSTR>(buffer),
|
||||
-1,
|
||||
nullptr,
|
||||
0,
|
||||
nullptr,
|
||||
nullptr);
|
||||
if (flen == 0) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
UniquePtr<char[]> utf8Buffer(new char[flen]);
|
||||
if (WideCharToMultiByte(CP_UTF8, 0, reinterpret_cast<LPWSTR>(buffer), -1,
|
||||
utf8Buffer.get(), flen, nullptr, nullptr) == 0) {
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
mFileContents = Move(utf8Buffer);
|
||||
buffer = mFileContents.get();
|
||||
}
|
||||
#endif
|
||||
|
||||
char* currSection = nullptr;
|
||||
|
||||
// outer loop tokenizes into lines
|
||||
while (char* token = NS_strtok(kNL, &buffer)) {
|
||||
if (token[0] == '#' || token[0] == ';') { // it's a comment
|
||||
continue;
|
||||
}
|
||||
|
||||
token = (char*)NS_strspnp(kWhitespace, token);
|
||||
if (!*token) { // empty line
|
||||
continue;
|
||||
}
|
||||
|
||||
if (token[0] == '[') { // section header!
|
||||
++token;
|
||||
currSection = token;
|
||||
|
||||
char* rb = NS_strtok(kRBracket, &token);
|
||||
if (!rb || NS_strtok(kWhitespace, &token)) {
|
||||
// there's either an unclosed [Section or a [Section]Moretext!
|
||||
// we could frankly decide that this INI file is malformed right
|
||||
// here and stop, but we won't... keep going, looking for
|
||||
// a well-formed [section] to continue working with
|
||||
currSection = nullptr;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!currSection) {
|
||||
// If we haven't found a section header (or we found a malformed
|
||||
// section header), don't bother parsing this line.
|
||||
continue;
|
||||
}
|
||||
|
||||
char* key = token;
|
||||
char* e = NS_strtok(kEquals, &token);
|
||||
if (!e || !token) {
|
||||
continue;
|
||||
}
|
||||
|
||||
INIValue* v;
|
||||
if (!mSections.Get(currSection, &v)) {
|
||||
v = new INIValue(key, token);
|
||||
if (!v) {
|
||||
return NS_ERROR_OUT_OF_MEMORY;
|
||||
}
|
||||
|
||||
mSections.Put(currSection, v);
|
||||
continue;
|
||||
}
|
||||
|
||||
// Check whether this key has already been specified; overwrite
|
||||
// if so, or append if not.
|
||||
while (v) {
|
||||
if (!strcmp(key, v->key)) {
|
||||
v->value = token;
|
||||
break;
|
||||
}
|
||||
if (!v->next) {
|
||||
v->next = MakeUnique<INIValue>(key, token);
|
||||
if (!v->next) {
|
||||
return NS_ERROR_OUT_OF_MEMORY;
|
||||
}
|
||||
break;
|
||||
}
|
||||
v = v->next.get();
|
||||
}
|
||||
NS_ASSERTION(v, "v should never be null coming out of this loop");
|
||||
}
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsINIParser::GetString(const char* aSection, const char* aKey,
|
||||
nsACString& aResult)
|
||||
{
|
||||
INIValue* val;
|
||||
mSections.Get(aSection, &val);
|
||||
|
||||
while (val) {
|
||||
if (strcmp(val->key, aKey) == 0) {
|
||||
aResult.Assign(val->value);
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
val = val->next.get();
|
||||
}
|
||||
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsINIParser::GetString(const char* aSection, const char* aKey,
|
||||
char* aResult, uint32_t aResultLen)
|
||||
{
|
||||
INIValue* val;
|
||||
mSections.Get(aSection, &val);
|
||||
|
||||
while (val) {
|
||||
if (strcmp(val->key, aKey) == 0) {
|
||||
strncpy(aResult, val->value, aResultLen);
|
||||
aResult[aResultLen - 1] = '\0';
|
||||
if (strlen(val->value) >= aResultLen) {
|
||||
return NS_ERROR_LOSS_OF_SIGNIFICANT_DATA;
|
||||
}
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
val = val->next.get();
|
||||
}
|
||||
|
||||
return NS_ERROR_FAILURE;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsINIParser::GetSections(INISectionCallback aCB, void* aClosure)
|
||||
{
|
||||
for (auto iter = mSections.Iter(); !iter.Done(); iter.Next()) {
|
||||
if (!aCB(iter.Key(), aClosure)) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
return NS_OK;
|
||||
}
|
||||
|
||||
nsresult
|
||||
nsINIParser::GetStrings(const char* aSection,
|
||||
INIStringCallback aCB, void* aClosure)
|
||||
{
|
||||
INIValue* val;
|
||||
|
||||
for (mSections.Get(aSection, &val);
|
||||
val;
|
||||
val = val->next.get()) {
|
||||
|
||||
if (!aCB(val->key, val->value, aClosure)) {
|
||||
return NS_OK;
|
||||
}
|
||||
}
|
||||
|
||||
return NS_OK;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue