mirror of
https://github.com/boostorg/mysql.git
synced 2025-05-12 14:11:41 +00:00
47 lines
1.5 KiB
C++
47 lines
1.5 KiB
C++
//
|
|
// Copyright (c) 2019-2025 Ruben Perez Hidalgo (rubenperez038 at gmail dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#include <boost/mysql/character_set.hpp>
|
|
#include <boost/mysql/escape_string.hpp>
|
|
#include <boost/mysql/string_view.hpp>
|
|
|
|
#include <string>
|
|
|
|
using namespace boost::mysql;
|
|
|
|
static bool bit_set(uint8_t value, size_t bit) { return value & (0x01 << bit); }
|
|
|
|
static bool call_escape_string(const uint8_t* data, size_t size) noexcept
|
|
{
|
|
// We need at least 1 byte (for options)
|
|
if (size < 1u)
|
|
return false;
|
|
const uint8_t* end = data + size;
|
|
|
|
// Options
|
|
bool backslash_slashes = bit_set(data[0], 0);
|
|
auto quot_ctx = bit_set(data[0], 1)
|
|
? (bit_set(data[0], 2) ? quoting_context::double_quote : quoting_context::backtick)
|
|
: quoting_context::single_quote;
|
|
++data;
|
|
|
|
// String to escape
|
|
string_view input(reinterpret_cast<const char*>(data), reinterpret_cast<const char*>(end));
|
|
|
|
// Perform the escaping
|
|
std::string escaped;
|
|
error_code ec = escape_string(input, {utf8mb4_charset, backslash_slashes}, quot_ctx, escaped);
|
|
return ec.failed() || escaped.empty();
|
|
}
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
|
|
{
|
|
// Note: this code should never throw exceptions, for any kind of input
|
|
call_escape_string(data, size);
|
|
return 0;
|
|
}
|