ext/CrashServer/CrashHandler/SendRpt/IniFile.cpp

   1 #include "StdAfx.h"
   2 #include "IniFile.h"
   3 #include "utils.h"
   4
   5 IniFile::IniFile(const wchar_t* path)
   6 {
   7     CHandle file(CreateFile(path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, 0));
   8     if (file == INVALID_HANDLE_VALUE)
   9         throw std::runtime_error("ini file not found");
  10
  11     DWORD size = GetFileSize(file, NULL);
  12     std::vector<BYTE> data(size);
  13
  14     if (!ReadFile(file, data.data(), (DWORD)data.size(), &size, NULL))
  15         throw std::runtime_error("failed to read ini file");
  16
  17     Parse(data.data(), size);
  18 }
  19
  20 IniFile::IniFile(HMODULE hImage, DWORD resId)
  21 {
  22     auto data = ExtractDataFromResource(hImage, resId);
  23     Parse(data.first, data.second);
  24 }
  25
  26 const wchar_t* GetEol(const wchar_t* p, const wchar_t* end)
  27 {
  28     while (p < end && *p != L'\n')
  29         ++p;
  30     return p;
  31 }
  32
  33 void IniFile::Parse(LPCVOID data, size_t size)
  34 {
  35     LPCWSTR begin = (LPCWSTR)data;
  36     LPCWSTR end = begin + size / sizeof(*begin);
  37
  38     if (*begin == 0xFEFF)
  39         ++begin;
  40
  41     Section* currentSection = nullptr;
  42     CString* varToAppend = nullptr;
  43
  44     while (1)
  45     {
  46         const wchar_t* eol = GetEol(begin, end);
  47         if (begin >= eol)
  48             break;
  49
  50         CStringW line(begin, int(eol - begin));
  51         begin = eol + 1;
  52
  53         line.Trim(L" \t\r\n");
  54
  55         if (varToAppend)
  56         {
  57             bool appendNext = line.Right(1) == L"\\";
  58             if (appendNext)
  59                 line.Delete(line.GetLength() - 1);
  60             *varToAppend += line;
  61             if (!appendNext)
  62                 varToAppend = nullptr;
  63             continue;
  64         }
  65
  66         if (line.IsEmpty() || line[0] == L';')
  67             continue;
  68
  69         if (line[0] == L'[' && line[line.GetLength() - 1] == L']')
  70         {
  71             currentSection = &m_sections[line.Trim(L"[]")];
  72             continue;
  73         }
  74
  75         if (!currentSection)
  76             continue;
  77
  78         int equalPos = line.Find(L'=');
  79         if (equalPos == -1 || equalPos == 0)
  80             continue;
  81         CStringW var = line.Left(equalPos);
  82         CStringW text = line.Mid(equalPos + 1);
  83         if (text.Right(1) == L"\\")
  84         {
  85             text.Delete(text.GetLength() - 1);
  86             varToAppend = &(*currentSection)[var];
  87         }
  88
  89         (*currentSection)[var] = text;
  90     }
  91 }
  92
  93 CStringW IniFile::GetString(const wchar_t* section, const wchar_t* variable)
  94 {
  95     auto sec = m_sections.find(section);
  96     if (sec != m_sections.end())
  97     {
  98         auto var = sec->second.find(variable);
  99         if (var != sec->second.end())
 100             return var->second;
 101     }
 102     return CStringW();
 103 }