225 lines
		
	
	
		
			7.2 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			225 lines
		
	
	
		
			7.2 KiB
		
	
	
	
		
			C++
		
	
	
	
//
 | 
						|
// Copyright(c) 2015 Gabi Melman.
 | 
						|
// Distributed under the MIT License (http://opensource.org/licenses/MIT)
 | 
						|
//
 | 
						|
 | 
						|
#pragma once
 | 
						|
 | 
						|
#include <cctype>
 | 
						|
#include <spdlog/common.h>
 | 
						|
 | 
						|
#if defined(__has_include)
 | 
						|
    #if __has_include(<version>)
 | 
						|
        #include <version>
 | 
						|
    #endif
 | 
						|
#endif
 | 
						|
 | 
						|
#if __cpp_lib_span >= 202002L
 | 
						|
    #include <span>
 | 
						|
#endif
 | 
						|
 | 
						|
//
 | 
						|
// Support for logging binary data as hex
 | 
						|
// format flags, any combination of the following:
 | 
						|
// {:X} - print in uppercase.
 | 
						|
// {:s} - don't separate each byte with space.
 | 
						|
// {:p} - don't print the position on each line start.
 | 
						|
// {:n} - don't split the output to lines.
 | 
						|
// {:a} - show ASCII if :n is not set
 | 
						|
 | 
						|
//
 | 
						|
// Examples:
 | 
						|
//
 | 
						|
// std::vector<char> v(200, 0x0b);
 | 
						|
// logger->info("Some buffer {}", spdlog::to_hex(v));
 | 
						|
// char buf[128];
 | 
						|
// logger->info("Some buffer {:X}", spdlog::to_hex(std::begin(buf), std::end(buf)));
 | 
						|
// logger->info("Some buffer {:X}", spdlog::to_hex(std::begin(buf), std::end(buf), 16));
 | 
						|
 | 
						|
namespace spdlog {
 | 
						|
namespace details {
 | 
						|
 | 
						|
template <typename It>
 | 
						|
class dump_info {
 | 
						|
public:
 | 
						|
    dump_info(It range_begin, It range_end, size_t size_per_line)
 | 
						|
        : begin_(range_begin),
 | 
						|
          end_(range_end),
 | 
						|
          size_per_line_(size_per_line) {}
 | 
						|
 | 
						|
    // do not use begin() and end() to avoid collision with fmt/ranges
 | 
						|
    It get_begin() const { return begin_; }
 | 
						|
    It get_end() const { return end_; }
 | 
						|
    size_t size_per_line() const { return size_per_line_; }
 | 
						|
 | 
						|
private:
 | 
						|
    It begin_, end_;
 | 
						|
    size_t size_per_line_;
 | 
						|
};
 | 
						|
}  // namespace details
 | 
						|
 | 
						|
// create a dump_info that wraps the given container
 | 
						|
template <typename Container>
 | 
						|
inline details::dump_info<typename Container::const_iterator> to_hex(const Container &container,
 | 
						|
                                                                     size_t size_per_line = 32) {
 | 
						|
    static_assert(sizeof(typename Container::value_type) == 1,
 | 
						|
                  "sizeof(Container::value_type) != 1");
 | 
						|
    using Iter = typename Container::const_iterator;
 | 
						|
    return details::dump_info<Iter>(std::begin(container), std::end(container), size_per_line);
 | 
						|
}
 | 
						|
 | 
						|
#if __cpp_lib_span >= 202002L
 | 
						|
 | 
						|
template <typename Value, size_t Extent>
 | 
						|
inline details::dump_info<typename std::span<Value, Extent>::iterator> to_hex(
 | 
						|
    const std::span<Value, Extent> &container, size_t size_per_line = 32) {
 | 
						|
    using Container = std::span<Value, Extent>;
 | 
						|
    static_assert(sizeof(typename Container::value_type) == 1,
 | 
						|
                  "sizeof(Container::value_type) != 1");
 | 
						|
    using Iter = typename Container::iterator;
 | 
						|
    return details::dump_info<Iter>(std::begin(container), std::end(container), size_per_line);
 | 
						|
}
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
// create dump_info from ranges
 | 
						|
template <typename It>
 | 
						|
inline details::dump_info<It> to_hex(const It range_begin,
 | 
						|
                                     const It range_end,
 | 
						|
                                     size_t size_per_line = 32) {
 | 
						|
    return details::dump_info<It>(range_begin, range_end, size_per_line);
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace spdlog
 | 
						|
 | 
						|
namespace
 | 
						|
#ifdef SPDLOG_USE_STD_FORMAT
 | 
						|
    std
 | 
						|
#else
 | 
						|
    fmt
 | 
						|
#endif
 | 
						|
{
 | 
						|
 | 
						|
template <typename T>
 | 
						|
struct formatter<spdlog::details::dump_info<T>, char> {
 | 
						|
    const char delimiter = ' ';
 | 
						|
    bool put_newlines = true;
 | 
						|
    bool put_delimiters = true;
 | 
						|
    bool use_uppercase = false;
 | 
						|
    bool put_positions = true;  // position on start of each line
 | 
						|
    bool show_ascii = false;
 | 
						|
 | 
						|
    // parse the format string flags
 | 
						|
    template <typename ParseContext>
 | 
						|
    SPDLOG_CONSTEXPR_FUNC auto parse(ParseContext &ctx) -> decltype(ctx.begin()) {
 | 
						|
        auto it = ctx.begin();
 | 
						|
        while (it != ctx.end() && *it != '}') {
 | 
						|
            switch (*it) {
 | 
						|
                case 'X':
 | 
						|
                    use_uppercase = true;
 | 
						|
                    break;
 | 
						|
                case 's':
 | 
						|
                    put_delimiters = false;
 | 
						|
                    break;
 | 
						|
                case 'p':
 | 
						|
                    put_positions = false;
 | 
						|
                    break;
 | 
						|
                case 'n':
 | 
						|
                    put_newlines = false;
 | 
						|
                    show_ascii = false;
 | 
						|
                    break;
 | 
						|
                case 'a':
 | 
						|
                    if (put_newlines) {
 | 
						|
                        show_ascii = true;
 | 
						|
                    }
 | 
						|
                    break;
 | 
						|
            }
 | 
						|
 | 
						|
            ++it;
 | 
						|
        }
 | 
						|
        return it;
 | 
						|
    }
 | 
						|
 | 
						|
    // format the given bytes range as hex
 | 
						|
    template <typename FormatContext, typename Container>
 | 
						|
    auto format(const spdlog::details::dump_info<Container> &the_range, FormatContext &ctx) const
 | 
						|
        -> decltype(ctx.out()) {
 | 
						|
        SPDLOG_CONSTEXPR const char *hex_upper = "0123456789ABCDEF";
 | 
						|
        SPDLOG_CONSTEXPR const char *hex_lower = "0123456789abcdef";
 | 
						|
        const char *hex_chars = use_uppercase ? hex_upper : hex_lower;
 | 
						|
 | 
						|
#if !defined(SPDLOG_USE_STD_FORMAT) && FMT_VERSION < 60000
 | 
						|
        auto inserter = ctx.begin();
 | 
						|
#else
 | 
						|
        auto inserter = ctx.out();
 | 
						|
#endif
 | 
						|
 | 
						|
        int size_per_line = static_cast<int>(the_range.size_per_line());
 | 
						|
        auto start_of_line = the_range.get_begin();
 | 
						|
        for (auto i = the_range.get_begin(); i != the_range.get_end(); i++) {
 | 
						|
            auto ch = static_cast<unsigned char>(*i);
 | 
						|
 | 
						|
            if (put_newlines &&
 | 
						|
                (i == the_range.get_begin() || i - start_of_line >= size_per_line)) {
 | 
						|
                if (show_ascii && i != the_range.get_begin()) {
 | 
						|
                    *inserter++ = delimiter;
 | 
						|
                    *inserter++ = delimiter;
 | 
						|
                    for (auto j = start_of_line; j < i; j++) {
 | 
						|
                        auto pc = static_cast<unsigned char>(*j);
 | 
						|
                        *inserter++ = std::isprint(pc) ? static_cast<char>(*j) : '.';
 | 
						|
                    }
 | 
						|
                }
 | 
						|
 | 
						|
                put_newline(inserter, static_cast<size_t>(i - the_range.get_begin()));
 | 
						|
 | 
						|
                // put first byte without delimiter in front of it
 | 
						|
                *inserter++ = hex_chars[(ch >> 4) & 0x0f];
 | 
						|
                *inserter++ = hex_chars[ch & 0x0f];
 | 
						|
                start_of_line = i;
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
 | 
						|
            if (put_delimiters && i != the_range.get_begin()) {
 | 
						|
                *inserter++ = delimiter;
 | 
						|
            }
 | 
						|
 | 
						|
            *inserter++ = hex_chars[(ch >> 4) & 0x0f];
 | 
						|
            *inserter++ = hex_chars[ch & 0x0f];
 | 
						|
        }
 | 
						|
        if (show_ascii)  // add ascii to last line
 | 
						|
        {
 | 
						|
            if (the_range.get_end() - the_range.get_begin() > size_per_line) {
 | 
						|
                auto blank_num = size_per_line - (the_range.get_end() - start_of_line);
 | 
						|
                while (blank_num-- > 0) {
 | 
						|
                    *inserter++ = delimiter;
 | 
						|
                    *inserter++ = delimiter;
 | 
						|
                    if (put_delimiters) {
 | 
						|
                        *inserter++ = delimiter;
 | 
						|
                    }
 | 
						|
                }
 | 
						|
            }
 | 
						|
            *inserter++ = delimiter;
 | 
						|
            *inserter++ = delimiter;
 | 
						|
            for (auto j = start_of_line; j != the_range.get_end(); j++) {
 | 
						|
                auto pc = static_cast<unsigned char>(*j);
 | 
						|
                *inserter++ = std::isprint(pc) ? static_cast<char>(*j) : '.';
 | 
						|
            }
 | 
						|
        }
 | 
						|
        return inserter;
 | 
						|
    }
 | 
						|
 | 
						|
    // put newline(and position header)
 | 
						|
    template <typename It>
 | 
						|
    void put_newline(It inserter, std::size_t pos) const {
 | 
						|
#ifdef _WIN32
 | 
						|
        *inserter++ = '\r';
 | 
						|
#endif
 | 
						|
        *inserter++ = '\n';
 | 
						|
 | 
						|
        if (put_positions) {
 | 
						|
            spdlog::fmt_lib::format_to(inserter, SPDLOG_FMT_STRING("{:04X}: "), pos);
 | 
						|
        }
 | 
						|
    }
 | 
						|
};
 | 
						|
}  // namespace std
 |