.. _program_listing_file__tmp_ws_src_warehouse_ros_sqlite_include_warehouse_ros_sqlite_utils.hpp: Program Listing for File utils.hpp ================================== |exhale_lsh| :ref:`Return to documentation for file ` (``/tmp/ws/src/warehouse_ros_sqlite/include/warehouse_ros_sqlite/utils.hpp``) .. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS .. code-block:: cpp // Copyright 2020 Bjarne von Horn // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions are met: // // * Redistributions of source code must retain the above copyright // notice, this list of conditions and the following disclaimer. // // * Redistributions in binary form must reproduce the above copyright // notice, this list of conditions and the following disclaimer in the // documentation and/or other materials provided with the distribution. // // * Neither the name of the copyright holder nor the names of its // contributors may be used to endorse or promote products derived from // this software without specific prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE // POSSIBILITY OF SUCH DAMAGE. // SPDX-License-Identifier: BSD-3-Clause #ifndef WAREHOUSE_ROS_SQLITE__UTILS_HPP_ #define WAREHOUSE_ROS_SQLITE__UTILS_HPP_ #include #include #include #include #include #include #include #include extern "C" { struct sqlite3_stmt; struct sqlite3; } namespace warehouse_ros_sqlite { struct WAREHOUSE_ROS_SQLITE_EXPORT Sqlite3StmtDeleter { void operator()(sqlite3_stmt * stmt) const; }; WAREHOUSE_ROS_SQLITE_EXPORT void sqlite3_delete(sqlite3 * db); using sqlite3_stmt_ptr = std::unique_ptr; using sqlite3_ptr = std::shared_ptr; namespace schema { namespace detail { template void check_do_escape(std::string & /*unused*/, char /*unused*/) { } template void check_do_escape(std::string & s, char c) { if (c == escaped_char) { s.push_back(escaped_char); } check_do_escape(s, c); } template std::string escape(const std::string & s) { std::string ans; ans.reserve(4 * sizeof...(escaped_chars) + s.size()); for (const auto c : s) { ans.push_back(c); check_do_escape(ans, c); } return ans; } } // namespace detail constexpr const char * DB_NAME = "main"; constexpr const char * METADATA_COLUMN_PREFIX = "M_"; constexpr const char * DATA_COLUMN_NAME = "Data"; constexpr const char * TABLE_NAME_PREFIX = "T_"; constexpr const char * M_D5_TABLE_NAME = "WarehouseIndex"; constexpr const char * M_D5_TABLE_INDEX_COLUMN = "MangledTableName"; constexpr const char * M_D5_TABLE_M_D5_COLUMN = "MessageMD5"; constexpr const char * M_D5_TABLE_DATATYPE_COLUMN = "MessageDataType"; constexpr const char * M_D5_TABLE_TABLE_COLUMN = "WarehouseCollectionName"; constexpr const char * M_D5_TABLE_DATABASE_COLUMN = "WarehouseDatabaseName"; const int DATA_COLUMN_INDEX = 0; const int VERSION = 10; using escaped_columnname = std::string; using escaped_tablename = std::string; inline std::string escape_identifier(const std::string & s) { return "\"" + detail::escape<'"'>(s) + "\""; } inline escaped_columnname escape_columnname_with_prefix(const std::string & c) { return escape_identifier(METADATA_COLUMN_PREFIX + c); } inline std::string escape_string_literal_without_quotes(const std::string & c) { return schema::detail::escape<'\''>(c); } inline std::string mangle_database_and_collection_name( const std::string & db_name, const std::string & collection_name) { return TABLE_NAME_PREFIX + detail::escape<'@'>(db_name) + "@" + detail::escape<'@'>( collection_name); } inline escaped_tablename escape_and_mangle_database_and_collection_name( const std::string & db_name, const std::string & collection_name) { return "\"" + detail::escape<'@', '"'>(TABLE_NAME_PREFIX + db_name) + "@" + detail::escape<'@', '"'>(collection_name) + "\""; } } // namespace schema struct WAREHOUSE_ROS_SQLITE_EXPORT NullValue { }; inline WAREHOUSE_ROS_SQLITE_EXPORT std::array parse_md5_hexstring( const std::string & md5) { std::array binary_md5; if (md5.size() == 16) { std::copy(md5.begin(), md5.end(), binary_md5.begin()); return binary_md5; } if (md5.size() != 32) { throw std::invalid_argument("md5.size() must equal 32"); } size_t md5_idx = 0; for (auto & c : binary_md5) { char * end; const auto substr = md5.substr(md5_idx, 2); const auto t = std::strtoul(substr.c_str(), &end, 16); if (substr.c_str() + 2 != end) { throw std::invalid_argument("md5 is not hex string"); } c = static_cast(t); md5_idx += 2; } return binary_md5; } inline WAREHOUSE_ROS_SQLITE_EXPORT std::string verify_md5_string(const std::string & md5) { static const char characters[] = "0123456789ABCDEF"; if (md5.size() == 32) { return md5; } else if (md5.size() != 16) { throw std::invalid_argument("not a valid md5 string"); } std::string ans(32, 0); auto it = ans.begin(); for (const auto val : md5) { *it++ = characters[static_cast(val) >> 4]; *it++ = characters[static_cast(val) & 0x0F]; } return ans; } } // namespace warehouse_ros_sqlite #endif // WAREHOUSE_ROS_SQLITE__UTILS_HPP_