summaryrefslogtreecommitdiffstats
path: root/xpcom/glue/nsINIParser.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'xpcom/glue/nsINIParser.cpp')
-rw-r--r--xpcom/glue/nsINIParser.cpp331
1 files changed, 331 insertions, 0 deletions
diff --git a/xpcom/glue/nsINIParser.cpp b/xpcom/glue/nsINIParser.cpp
new file mode 100644
index 000000000..53eae7292
--- /dev/null
+++ b/xpcom/glue/nsINIParser.cpp
@@ -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;
+}