973d12b2a7
terminate called after throwing an instance of 'Poco::JSON::JSONException' ... #11 0x7f70540e006a in Poco::JSON::ParserImpl::parseImpl(std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char>> const&) (/usr/lib64/libPocoJSON.so.92+0x4206a) (BuildId: 149b3c1772e35fcbe5692d435f1b06820707bad3) #12 0x556a70a9511b in Poco::JSON::Parser::parse(std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char>> const&) /usr/include/Poco/JSON/Parser.h:202:9 #13 0x556a70a9511b in JsonUtil::parseJSON(std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char>> const&, Poco::SharedPtr<Poco::JSON::Object, Poco::ReferenceCounter, Poco::ReleasePolicy<Poco::JSON::Object>>&) /home/vmiklos/git/collaboraonline/online-fuzz/./common/JsonUtil.hpp:37:50 #14 0x556a70a7edde in AdminSocketHandler::handleMessage(std::vector<char, std::allocator<char>> const&) /home/vmiklos/git/collaboraonline/online-fuzz/wsd/Admin.cpp:325:13 Signed-off-by: Miklos Vajna <vmiklos@collabora.com> Change-Id: Iebe0f5d689032a8b3c1e5d38c5ee80d344d1cbed
249 lines
7.1 KiB
C++
249 lines
7.1 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
/*
|
|
* Copyright the Collabora Online contributors.
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* 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/.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "Util.hpp"
|
|
#include <Log.hpp>
|
|
|
|
#include <cassert>
|
|
#include <cstddef>
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <Poco/JSON/Object.h>
|
|
#include <Poco/JSON/Parser.h>
|
|
namespace JsonUtil
|
|
{
|
|
|
|
// Parse the json string and fill the Poco::JSON object
|
|
// Returns true if parsing successful otherwise false
|
|
inline bool parseJSON(const std::string& json, Poco::JSON::Object::Ptr& object)
|
|
{
|
|
const std::size_t index = json.find_first_of('{');
|
|
if (index != std::string::npos)
|
|
{
|
|
const std::string stringJSON = json.substr(index);
|
|
Poco::JSON::Parser parser;
|
|
try
|
|
{
|
|
const Poco::Dynamic::Var result = parser.parse(stringJSON);
|
|
object = result.extract<Poco::JSON::Object::Ptr>();
|
|
return true;
|
|
}
|
|
catch (const Poco::JSON::JSONException& exception)
|
|
{
|
|
LOG_WRN("parseJSON: failed to parse '" << stringJSON << "': '" << exception.what() << "'");
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
inline
|
|
int getLevenshteinDist(const std::string& string1, const std::string& string2)
|
|
{
|
|
int matrix[string1.size() + 1][string2.size() + 1];
|
|
std::memset(matrix, 0, sizeof(matrix[0][0]) * (string1.size() + 1) * (string2.size() + 1));
|
|
|
|
for (std::size_t i = 0; i < string1.size() + 1; i++)
|
|
{
|
|
for (std::size_t j = 0; j < string2.size() + 1; j++)
|
|
{
|
|
if (i == 0)
|
|
{
|
|
matrix[i][j] = j;
|
|
}
|
|
else if (j == 0)
|
|
{
|
|
matrix[i][j] = i;
|
|
}
|
|
else if (string1[i - 1] == string2[j - 1])
|
|
{
|
|
matrix[i][j] = matrix[i - 1][j - 1];
|
|
}
|
|
else
|
|
{
|
|
matrix[i][j] = 1 + std::min(std::min(matrix[i][j - 1], matrix[i - 1][j]),
|
|
matrix[i - 1][j - 1]);
|
|
}
|
|
}
|
|
}
|
|
|
|
return matrix[string1.size()][string2.size()];
|
|
}
|
|
|
|
/// Converts the given @valueVar to the type T, if possible.
|
|
/// @key is used for logging only.
|
|
template <typename T> T getJSONValue(const std::string& key, const Poco::Dynamic::Var valueVar)
|
|
{
|
|
try
|
|
{
|
|
return valueVar.convert<T>();
|
|
}
|
|
catch (const Poco::Exception& exc)
|
|
{
|
|
LOG_ERR("getJSONValue for ["
|
|
<< key << "]: " << exc.displayText()
|
|
<< (exc.nested() ? " (" + exc.nested()->displayText() + ')' : ""));
|
|
}
|
|
|
|
return T();
|
|
}
|
|
|
|
/// Gets value for @key directly from the given JSON in @object.
|
|
template <typename T> T getJSONValue(const Poco::JSON::Object::Ptr& object, const std::string& key)
|
|
{
|
|
return getJSONValue<T>(key, object->get(key));
|
|
}
|
|
|
|
/// Function that searches `object` for `key` and warns if there are minor mis-spellings involved.
|
|
/// Upon successful search, fills `value` with value found in object.
|
|
/// Removes the entry from the JSON object if @bRemove == true.
|
|
template <typename T>
|
|
bool findJSONValue(const Poco::JSON::Object::Ptr& object, const std::string& key, T& value)
|
|
{
|
|
// Try exact match first.
|
|
const Poco::Dynamic::Var var = object->get(key);
|
|
if (!var.isEmpty())
|
|
{
|
|
value = getJSONValue<T>(key, var);
|
|
|
|
LOG_TRC("Found JSON property [" << key << "] => [" << value << ']');
|
|
return true;
|
|
}
|
|
|
|
std::vector<std::string> propertyNames;
|
|
object->getNames(propertyNames);
|
|
|
|
// Check each property name against given key
|
|
// and warn for mis-spells with tolerance of 2.
|
|
const std::string keyLower = Util::toLower(key);
|
|
for (const std::string& userInput : propertyNames)
|
|
{
|
|
if (key != userInput)
|
|
{
|
|
const std::string userInputLower = Util::toLower(userInput);
|
|
|
|
// Mis-spelling tolerance.
|
|
const int levDist = getLevenshteinDist(keyLower, userInputLower);
|
|
if (levDist > 2)
|
|
continue; // Not even close, keep searching.
|
|
|
|
// We found something with some differences--warn and return.
|
|
LOG_ERR("Incorrect JSON property [" << userInput << "]. Did you mean [" << key <<
|
|
"] ? (Levenshtein distance: " << levDist << ')');
|
|
|
|
// Fail without exact match.
|
|
return false;
|
|
}
|
|
|
|
value = getJSONValue<T>(object, userInput);
|
|
|
|
LOG_TRC("Found JSON property [" << userInput << "] => [" << value << ']');
|
|
return true;
|
|
}
|
|
|
|
LOG_INF("Missing JSON property [" << key << "] will default to [" << value << ']');
|
|
return false;
|
|
}
|
|
|
|
static char getEscapementChar(char ch)
|
|
{
|
|
switch (ch)
|
|
{
|
|
case '\b':
|
|
return 'b';
|
|
case '\t':
|
|
return 't';
|
|
case '\n':
|
|
return 'n';
|
|
case '\f':
|
|
return 'f';
|
|
case '\r':
|
|
return 'r';
|
|
default:
|
|
return ch;
|
|
}
|
|
}
|
|
|
|
static void writeEscapedSequence(uint32_t ch, std::vector<char>& buf)
|
|
{
|
|
switch (ch)
|
|
{
|
|
case '\b':
|
|
case '\t':
|
|
case '\n':
|
|
case '\f':
|
|
case '\r':
|
|
case '"':
|
|
case '/':
|
|
case '\\':
|
|
buf.push_back('\\');
|
|
buf.push_back(getEscapementChar(ch));
|
|
break;
|
|
// Special processing of U+2028 and U+2029, which are valid JSON, but invalid JavaScript
|
|
// Write them in escaped '\u2028' or '\u2029' form
|
|
case 0x2028:
|
|
case 0x2029:
|
|
buf.push_back('\\');
|
|
buf.push_back('u');
|
|
buf.push_back('2');
|
|
buf.push_back('0');
|
|
buf.push_back('2');
|
|
buf.push_back(ch == 0x2028 ? '8' : '9');
|
|
break;
|
|
default:
|
|
assert(!"Unexpected character passed to writeEscapedSequence");
|
|
}
|
|
}
|
|
|
|
inline std::string escapeJSONValue(std::string val)
|
|
{
|
|
std::vector<char> buf;
|
|
buf.reserve(val.size() + 10); // some small initial extra space for escaping
|
|
for (size_t i = 0; i < val.size(); ++i)
|
|
{
|
|
const char ch = val[i];
|
|
switch(ch)
|
|
{
|
|
case '\b':
|
|
case '\t':
|
|
case '\n':
|
|
case '\f':
|
|
case '\r':
|
|
case '"':
|
|
case '/':
|
|
case '\\':
|
|
writeEscapedSequence(ch, buf);
|
|
break;
|
|
case '\xE2': // Special processing of U+2028 and U+2029
|
|
if (i + 2 < val.size() && val[i + 1] == '\x80'
|
|
&& (val[i + 2] == '\xA8' || val[i + 2] == '\xA9'))
|
|
{
|
|
writeEscapedSequence(val[i + 2] == '\xA8' ? 0x2028 : 0x2029, buf);
|
|
i += 2;
|
|
break;
|
|
}
|
|
// Fallthrough...
|
|
default:
|
|
buf.push_back(ch);
|
|
break;
|
|
}
|
|
}
|
|
return std::string(buf.data(), buf.size());
|
|
}
|
|
|
|
} // end namespace JsonUtil
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|
|
|