1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#ifndef XML_INFO_READER_H
#define XML_INFO_READER_H
#include <QFile>
#include <QIODevice>
#include <QString>
#include <QStringEncoder>
#include <QTextStream>
#include <QXmlStreamReader>
#include <uibase/log.h>
#include <uibase/utility.h>
// This is from installer_fomod, but should probably not be duplicated here.
struct FomodInfoReader : QObject
{
Q_OBJECT
public:
struct XmlParseError : MOBase::Exception
{
XmlParseError(const QString& message) : MOBase::Exception(message) {}
};
static QByteArray skipXmlHeader(QIODevice& file)
{
static const unsigned char UTF16LE_BOM[] = {0xFF, 0xFE};
static const unsigned char UTF16BE_BOM[] = {0xFE, 0xFF};
static const unsigned char UTF8_BOM[] = {0xEF, 0xBB, 0xBF};
static const unsigned char UTF16LE[] = {0x3C, 0x00, 0x3F, 0x00};
static const unsigned char UTF16BE[] = {0x00, 0x3C, 0x00, 0x3F};
static const unsigned char UTF8[] = {0x3C, 0x3F, 0x78, 0x6D};
file.seek(0);
QByteArray rawBytes = file.read(4);
QTextStream stream(&file);
int bom = 0;
if (rawBytes.startsWith((const char*)UTF16LE_BOM)) {
stream.setEncoding(QStringEncoder::Encoding::Utf16LE);
bom = 2;
} else if (rawBytes.startsWith((const char*)UTF16BE_BOM)) {
stream.setEncoding(QStringEncoder::Encoding::Utf16BE);
bom = 2;
} else if (rawBytes.startsWith((const char*)UTF8_BOM)) {
stream.setEncoding(QStringEncoder::Encoding::Utf8);
bom = 3;
} else if (rawBytes.startsWith(QByteArray((const char*)UTF16LE, 4))) {
stream.setEncoding(QStringEncoder::Encoding::Utf16LE);
} else if (rawBytes.startsWith(QByteArray((const char*)UTF16BE, 4))) {
stream.setEncoding(QStringEncoder::Encoding::Utf16BE);
} else if (rawBytes.startsWith(QByteArray((const char*)UTF8, 4))) {
stream.setEncoding(QStringEncoder::Encoding::Utf8);
} // otherwise maybe the textstream knows the encoding?
stream.seek(bom);
QString header = stream.readLine();
if (!header.startsWith("<?")) {
// it was all for nothing, there is no header here...
stream.seek(bom);
}
// this seems to be necessary due to buffering in QTextStream
file.seek(stream.pos());
return file.readAll();
}
template <class Fn>
static auto readXml(QFile& file, Fn&& fn)
{
// List of encodings to try:
static const std::vector<QStringConverter::Encoding> encodings{
QStringConverter::Utf16, QStringConverter::Utf8, QStringConverter::Latin1};
std::string errorMessage;
try {
QXmlStreamReader reader(&file);
return fn(reader);
} catch (const XmlParseError& e) {
MOBase::log::warn(
"The {} in this file is incorrectly encoded ({}). Applying heuristics...",
file.fileName(), e.what());
}
// nmm's xml parser is less strict than the one from qt and allows files with
// wrong encoding in the header. Being strict here would be bad user experience
// this works around bad headers.
QByteArray headerlessData = skipXmlHeader(file);
// try parsing the file with several encodings to support broken files
for (auto encoding : encodings) {
MOBase::log::debug("Trying encoding {} for {}... ",
QStringConverter::nameForEncoding(encoding), file.fileName());
try {
QStringEncoder encoder(encoding);
QXmlStreamReader reader(
encoder.encode(QString("<?xml version=\"1.0\" encoding=\"%1\" ?>")
.arg(encoder.name())) +
headerlessData);
MOBase::log::debug("Interpreting {} as {}.", file.fileName(), encoder.name());
return fn(reader);
} catch (const XmlParseError& e) {
MOBase::log::debug("Not {}: {}.", QStringConverter::nameForEncoding(encoding),
e.what());
}
}
throw XmlParseError(
tr("Failed to parse %1. See console for details.").arg(file.fileName()));
}
static std::tuple<QString, int, QString> parseInfo(QXmlStreamReader& reader)
{
std::tuple<QString, int, QString> info{"", -1, ""};
while (!reader.atEnd()) {
switch (reader.readNext()) {
case QXmlStreamReader::StartElement: {
if (reader.name().toString() == "Name") {
std::get<0>(info) = reader.readElementText();
} else if (reader.name().toString() == "Author") {
} else if (reader.name().toString() == "Version") {
std::get<2>(info) = reader.readElementText();
} else if (reader.name().toString() == "Id") {
std::get<1>(info) = reader.readElementText().toInt();
} else if (reader.name().toString() == "Website") {
}
} break;
default: {
} break;
}
}
if (reader.hasError()) {
throw XmlParseError(
QString("%1 in line %2").arg(reader.errorString()).arg(reader.lineNumber()));
}
return info;
}
};
#endif
|
