blob: d23f9319fa7b9422078042029f88e57a76b3f391 [file] [log] [blame]
//===-- StdStringExtractor.cpp ----------------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "StdStringExtractor.h"
// C Includes
#include <stdlib.h>
// C++ Includes
// Other libraries and framework includes
// Project includes
static inline int xdigit_to_sint(char ch) {
if (ch >= 'a' && ch <= 'f')
return 10 + ch - 'a';
if (ch >= 'A' && ch <= 'F')
return 10 + ch - 'A';
if (ch >= '0' && ch <= '9')
return ch - '0';
return -1;
}
//----------------------------------------------------------------------
// StdStringExtractor constructor
//----------------------------------------------------------------------
StdStringExtractor::StdStringExtractor() : m_packet(), m_index(0) {}
StdStringExtractor::StdStringExtractor(const char *packet_cstr)
: m_packet(), m_index(0) {
if (packet_cstr)
m_packet.assign(packet_cstr);
}
//----------------------------------------------------------------------
// StdStringExtractor copy constructor
//----------------------------------------------------------------------
StdStringExtractor::StdStringExtractor(const StdStringExtractor &rhs)
: m_packet(rhs.m_packet), m_index(rhs.m_index) {}
//----------------------------------------------------------------------
// StdStringExtractor assignment operator
//----------------------------------------------------------------------
const StdStringExtractor &StdStringExtractor::
operator=(const StdStringExtractor &rhs) {
if (this != &rhs) {
m_packet = rhs.m_packet;
m_index = rhs.m_index;
}
return *this;
}
//----------------------------------------------------------------------
// Destructor
//----------------------------------------------------------------------
StdStringExtractor::~StdStringExtractor() {}
char StdStringExtractor::GetChar(char fail_value) {
if (m_index < m_packet.size()) {
char ch = m_packet[m_index];
++m_index;
return ch;
}
m_index = UINT64_MAX;
return fail_value;
}
//----------------------------------------------------------------------
// If a pair of valid hex digits exist at the head of the
// StdStringExtractor they are decoded into an unsigned byte and returned
// by this function
//
// If there is not a pair of valid hex digits at the head of the
// StdStringExtractor, it is left unchanged and -1 is returned
//----------------------------------------------------------------------
int StdStringExtractor::DecodeHexU8() {
SkipSpaces();
if (GetBytesLeft() < 2) {
return -1;
}
const int hi_nibble = xdigit_to_sint(m_packet[m_index]);
const int lo_nibble = xdigit_to_sint(m_packet[m_index + 1]);
if (hi_nibble == -1 || lo_nibble == -1) {
return -1;
}
m_index += 2;
return (uint8_t)((hi_nibble << 4) + lo_nibble);
}
//----------------------------------------------------------------------
// Extract an unsigned character from two hex ASCII chars in the packet
// string, or return fail_value on failure
//----------------------------------------------------------------------
uint8_t StdStringExtractor::GetHexU8(uint8_t fail_value, bool set_eof_on_fail) {
// On success, fail_value will be overwritten with the next
// character in the stream
GetHexU8Ex(fail_value, set_eof_on_fail);
return fail_value;
}
bool StdStringExtractor::GetHexU8Ex(uint8_t &ch, bool set_eof_on_fail) {
int byte = DecodeHexU8();
if (byte == -1) {
if (set_eof_on_fail || m_index >= m_packet.size())
m_index = UINT64_MAX;
// ch should not be changed in case of failure
return false;
}
ch = (uint8_t)byte;
return true;
}
uint32_t StdStringExtractor::GetU32(uint32_t fail_value, int base) {
if (m_index < m_packet.size()) {
char *end = nullptr;
const char *start = m_packet.c_str();
const char *cstr = start + m_index;
uint32_t result = static_cast<uint32_t>(::strtoul(cstr, &end, base));
if (end && end != cstr) {
m_index = end - start;
return result;
}
}
return fail_value;
}
int32_t StdStringExtractor::GetS32(int32_t fail_value, int base) {
if (m_index < m_packet.size()) {
char *end = nullptr;
const char *start = m_packet.c_str();
const char *cstr = start + m_index;
int32_t result = static_cast<int32_t>(::strtol(cstr, &end, base));
if (end && end != cstr) {
m_index = end - start;
return result;
}
}
return fail_value;
}
uint64_t StdStringExtractor::GetU64(uint64_t fail_value, int base) {
if (m_index < m_packet.size()) {
char *end = nullptr;
const char *start = m_packet.c_str();
const char *cstr = start + m_index;
uint64_t result = ::strtoull(cstr, &end, base);
if (end && end != cstr) {
m_index = end - start;
return result;
}
}
return fail_value;
}
int64_t StdStringExtractor::GetS64(int64_t fail_value, int base) {
if (m_index < m_packet.size()) {
char *end = nullptr;
const char *start = m_packet.c_str();
const char *cstr = start + m_index;
int64_t result = ::strtoll(cstr, &end, base);
if (end && end != cstr) {
m_index = end - start;
return result;
}
}
return fail_value;
}
uint32_t StdStringExtractor::GetHexMaxU32(bool little_endian,
uint32_t fail_value) {
uint32_t result = 0;
uint32_t nibble_count = 0;
SkipSpaces();
if (little_endian) {
uint32_t shift_amount = 0;
while (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
// Make sure we don't exceed the size of a uint32_t...
if (nibble_count >= (sizeof(uint32_t) * 2)) {
m_index = UINT64_MAX;
return fail_value;
}
uint8_t nibble_lo;
uint8_t nibble_hi = xdigit_to_sint(m_packet[m_index]);
++m_index;
if (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
nibble_lo = xdigit_to_sint(m_packet[m_index]);
++m_index;
result |= ((uint32_t)nibble_hi << (shift_amount + 4));
result |= ((uint32_t)nibble_lo << shift_amount);
nibble_count += 2;
shift_amount += 8;
} else {
result |= ((uint32_t)nibble_hi << shift_amount);
nibble_count += 1;
shift_amount += 4;
}
}
} else {
while (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
// Make sure we don't exceed the size of a uint32_t...
if (nibble_count >= (sizeof(uint32_t) * 2)) {
m_index = UINT64_MAX;
return fail_value;
}
uint8_t nibble = xdigit_to_sint(m_packet[m_index]);
// Big Endian
result <<= 4;
result |= nibble;
++m_index;
++nibble_count;
}
}
return result;
}
uint64_t StdStringExtractor::GetHexMaxU64(bool little_endian,
uint64_t fail_value) {
uint64_t result = 0;
uint32_t nibble_count = 0;
SkipSpaces();
if (little_endian) {
uint32_t shift_amount = 0;
while (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
// Make sure we don't exceed the size of a uint64_t...
if (nibble_count >= (sizeof(uint64_t) * 2)) {
m_index = UINT64_MAX;
return fail_value;
}
uint8_t nibble_lo;
uint8_t nibble_hi = xdigit_to_sint(m_packet[m_index]);
++m_index;
if (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
nibble_lo = xdigit_to_sint(m_packet[m_index]);
++m_index;
result |= ((uint64_t)nibble_hi << (shift_amount + 4));
result |= ((uint64_t)nibble_lo << shift_amount);
nibble_count += 2;
shift_amount += 8;
} else {
result |= ((uint64_t)nibble_hi << shift_amount);
nibble_count += 1;
shift_amount += 4;
}
}
} else {
while (m_index < m_packet.size() && ::isxdigit(m_packet[m_index])) {
// Make sure we don't exceed the size of a uint64_t...
if (nibble_count >= (sizeof(uint64_t) * 2)) {
m_index = UINT64_MAX;
return fail_value;
}
uint8_t nibble = xdigit_to_sint(m_packet[m_index]);
// Big Endian
result <<= 4;
result |= nibble;
++m_index;
++nibble_count;
}
}
return result;
}
size_t StdStringExtractor::GetHexBytes(void *dst_void, size_t dst_len,
uint8_t fail_fill_value) {
uint8_t *dst = (uint8_t *)dst_void;
size_t bytes_extracted = 0;
while (bytes_extracted < dst_len && GetBytesLeft()) {
dst[bytes_extracted] = GetHexU8(fail_fill_value);
if (IsGood())
++bytes_extracted;
else
break;
}
for (size_t i = bytes_extracted; i < dst_len; ++i)
dst[i] = fail_fill_value;
return bytes_extracted;
}
//----------------------------------------------------------------------
// Decodes all valid hex encoded bytes at the head of the
// StdStringExtractor, limited by dst_len.
//
// Returns the number of bytes successfully decoded
//----------------------------------------------------------------------
size_t StdStringExtractor::GetHexBytesAvail(void *dst_void, size_t dst_len) {
uint8_t *dst = (uint8_t *)dst_void;
size_t bytes_extracted = 0;
while (bytes_extracted < dst_len) {
int decode = DecodeHexU8();
if (decode == -1) {
break;
}
dst[bytes_extracted++] = (uint8_t)decode;
}
return bytes_extracted;
}
// Consume ASCII hex nibble character pairs until we have decoded byte_size
// bytes of data.
uint64_t StdStringExtractor::GetHexWithFixedSize(uint32_t byte_size,
bool little_endian,
uint64_t fail_value) {
if (byte_size <= 8 && GetBytesLeft() >= byte_size * 2) {
uint64_t result = 0;
uint32_t i;
if (little_endian) {
// Little Endian
uint32_t shift_amount;
for (i = 0, shift_amount = 0; i < byte_size && IsGood();
++i, shift_amount += 8) {
result |= ((uint64_t)GetHexU8() << shift_amount);
}
} else {
// Big Endian
for (i = 0; i < byte_size && IsGood(); ++i) {
result <<= 8;
result |= GetHexU8();
}
}
}
m_index = UINT64_MAX;
return fail_value;
}
size_t StdStringExtractor::GetHexByteString(std::string &str) {
str.clear();
str.reserve(GetBytesLeft() / 2);
char ch;
while ((ch = GetHexU8()) != '\0')
str.append(1, ch);
return str.size();
}
size_t StdStringExtractor::GetHexByteStringFixedLength(std::string &str,
uint32_t nibble_length) {
str.clear();
uint32_t nibble_count = 0;
for (const char *pch = Peek();
(nibble_count < nibble_length) && (pch != nullptr);
str.append(1, GetHexU8(0, false)), pch = Peek(), nibble_count += 2) {
}
return str.size();
}
size_t StdStringExtractor::GetHexByteStringTerminatedBy(std::string &str,
char terminator) {
str.clear();
char ch;
while ((ch = GetHexU8(0, false)) != '\0')
str.append(1, ch);
if (Peek() && *Peek() == terminator)
return str.size();
str.clear();
return str.size();
}
bool StdStringExtractor::GetNameColonValue(std::string &name,
std::string &value) {
// Read something in the form of NNNN:VVVV; where NNNN is any character
// that is not a colon, followed by a ':' character, then a value (one or
// more ';' chars), followed by a ';'
if (m_index < m_packet.size()) {
const size_t colon_idx = m_packet.find(':', m_index);
if (colon_idx != std::string::npos) {
const size_t semicolon_idx = m_packet.find(';', colon_idx);
if (semicolon_idx != std::string::npos) {
name.assign(m_packet, m_index, colon_idx - m_index);
value.assign(m_packet, colon_idx + 1, semicolon_idx - (colon_idx + 1));
m_index = semicolon_idx + 1;
return true;
}
}
}
m_index = UINT64_MAX;
return false;
}
void StdStringExtractor::SkipSpaces() {
const size_t n = m_packet.size();
while (m_index < n && isspace(m_packet[m_index]))
++m_index;
}