1 /*
2 * Copyright (c) 2015, 2019, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 #include "IniFile.h"
27 #include "Helpers.h"
28
29 #include <string>
30
31
32 IniFile::IniFile() : ISectionalPropertyContainer() {
33 }
34
35 IniFile::~IniFile() {
36 for (OrderedMap<TString, IniSectionData*>::iterator iterator =
37 FMap.begin(); iterator != FMap.end(); iterator++) {
38 JPPair<TString, IniSectionData*> *item = *iterator;
39 delete item->second;
40 }
41 }
42
43 bool IniFile::LoadFromFile(const TString FileName) {
44 bool result = false;
45 Platform& platform = Platform::GetInstance();
46
47 std::list<TString> contents = platform.LoadFromFile(FileName);
48
49 if (contents.empty() == false) {
50 bool found = false;
51
52 // Determine the if file is an INI file or property file.
53 // Assign FDefaultSection if it is
54 // an INI file. Otherwise FDefaultSection is NULL.
55 for (std::list<TString>::const_iterator iterator = contents.begin();
56 iterator != contents.end(); iterator++) {
57 TString line = *iterator;
58
59 if (line[0] == ';') {
60 // Semicolon is a comment so ignore the line.
61 continue;
62 }
63 else {
64 if (line[0] == '[') {
65 found = true;
66 }
67
68 break;
69 }
70 }
71
72 if (found == true) {
73 TString sectionName;
74
75 for (std::list<TString>::const_iterator iterator = contents.begin();
76 iterator != contents.end(); iterator++) {
77 TString line = *iterator;
78
79 if (line[0] == ';') {
80 // Semicolon is a comment so ignore the line.
81 continue;
82 }
83 else if (line[0] == '[' && line[line.length() - 1] == ']') {
84 sectionName = line.substr(1, line.size() - 2);
85 }
86 else if (sectionName.empty() == false) {
87 TString name;
88 TString value;
89
90 if (Helpers::SplitOptionIntoNameValue(
91 line, name, value) == true) {
92 Append(sectionName, name, value);
93 }
94 }
95 }
96
97 result = true;
98 }
99 }
100
101 return result;
102 }
103
104 bool IniFile::SaveToFile(const TString FileName, bool ownerOnly) {
105 bool result = false;
106
107 std::list<TString> contents;
108 std::vector<TString> keys = FMap.GetKeys();
109
110 for (unsigned int index = 0; index < keys.size(); index++) {
111 TString name = keys[index];
112 IniSectionData *section;
113
114 if (FMap.GetValue(name, section) == true && section != NULL) {
115 contents.push_back(_T("[") + name + _T("]"));
116 std::list<TString> lines = section->GetLines();
117 contents.insert(contents.end(), lines.begin(), lines.end());
118 contents.push_back(_T(""));
119 }
120 }
121
122 Platform& platform = Platform::GetInstance();
123 platform.SaveToFile(FileName, contents, ownerOnly);
124 result = true;
125 return result;
126 }
127
128 void IniFile::Append(const TString SectionName,
129 const TString Key, TString Value) {
130 if (FMap.ContainsKey(SectionName) == true) {
131 IniSectionData* section;
132
133 if (FMap.GetValue(SectionName, section) == true && section != NULL) {
134 section->SetValue(Key, Value);
135 }
136 }
137 else {
138 IniSectionData *section = new IniSectionData();
139 section->SetValue(Key, Value);
140 FMap.Append(SectionName, section);
141 }
142 }
143
144 void IniFile::AppendSection(const TString SectionName,
145 OrderedMap<TString, TString> Values) {
146 if (FMap.ContainsKey(SectionName) == true) {
147 IniSectionData* section;
148
149 if (FMap.GetValue(SectionName, section) == true && section != NULL) {
150 section->Append(Values);
151 }
152 }
153 else {
154 IniSectionData *section = new IniSectionData(Values);
155 FMap.Append(SectionName, section);
156 }
157 }
158
159 bool IniFile::GetValue(const TString SectionName,
160 const TString Key, TString& Value) {
161 bool result = false;
162 IniSectionData* section;
163
164 if (FMap.GetValue(SectionName, section) == true && section != NULL) {
165 result = section->GetValue(Key, Value);
166 }
167
168 return result;
169 }
170
171 bool IniFile::SetValue(const TString SectionName,
172 const TString Key, TString Value) {
173 bool result = false;
174 IniSectionData* section;
175
176 if (FMap.GetValue(SectionName, section) && section != NULL) {
177 result = section->SetValue(Key, Value);
178 }
179 else {
180 Append(SectionName, Key, Value);
181 }
182
183
184 return result;
185 }
186
187 bool IniFile::GetSection(const TString SectionName,
188 OrderedMap<TString, TString> &Data) {
189 bool result = false;
190
191 if (FMap.ContainsKey(SectionName) == true) {
192 IniSectionData* section = NULL;
193
194 if (FMap.GetValue(SectionName, section) == true && section != NULL) {
195 OrderedMap<TString, TString> data = section->GetData();
196 Data.Append(data);
197 result = true;
198 }
199 }
200
201 return result;
202 }
203
204 bool IniFile::ContainsSection(const TString SectionName) {
205 return FMap.ContainsKey(SectionName);
206 }
207
208 //----------------------------------------------------------------------------
209
210 IniSectionData::IniSectionData() {
211 FMap.SetAllowDuplicates(true);
212 }
213
214 IniSectionData::IniSectionData(OrderedMap<TString, TString> Values) {
215 FMap = Values;
216 }
217
218 std::vector<TString> IniSectionData::GetKeys() {
219 return FMap.GetKeys();
220 }
221
222 std::list<TString> IniSectionData::GetLines() {
223 std::list<TString> result;
224 std::vector<TString> keys = FMap.GetKeys();
225
226 for (unsigned int index = 0; index < keys.size(); index++) {
227 TString name = keys[index];
228 TString value;
229
230 if (FMap.GetValue(name, value) == true) {
231 name = Helpers::ReplaceString(name, _T("="), _T("\\="));
232 value = Helpers::ReplaceString(value, _T("="), _T("\\="));
233
234 TString line = name + _T('=') + value;
235 result.push_back(line);
236 }
237 }
238
239 return result;
240 }
241
242 OrderedMap<TString, TString> IniSectionData::GetData() {
243 OrderedMap<TString, TString> result = FMap;
244 return result;
245 }
246
247 bool IniSectionData::GetValue(const TString Key, TString& Value) {
248 return FMap.GetValue(Key, Value);
249 }
250
251 bool IniSectionData::SetValue(const TString Key, TString Value) {
252 return FMap.SetValue(Key, Value);
253 }
254
255 void IniSectionData::Append(OrderedMap<TString, TString> Values) {
256 FMap.Append(Values);
257 }
258
259 size_t IniSectionData::GetCount() {
260 return FMap.Count();
261 }