1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
| //===-- StringExtractor.h ---------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef utility_StringExtractor_h_
#define utility_StringExtractor_h_
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/StringRef.h"
#include <stddef.h>
#include <stdint.h>
#include <string>
class StringExtractor {
public:
enum { BigEndian = 0, LittleEndian = 1 };
// Constructors and Destructors
StringExtractor();
StringExtractor(llvm::StringRef packet_str);
StringExtractor(const char *packet_cstr);
virtual ~StringExtractor();
void Reset(llvm::StringRef str) {
m_packet = str;
m_index = 0;
}
// Returns true if the file position is still valid for the data contained in
// this string extractor object.
bool IsGood() const { return m_index != UINT64_MAX; }
uint64_t GetFilePos() const { return m_index; }
void SetFilePos(uint32_t idx) { m_index = idx; }
void Clear() {
m_packet.clear();
m_index = 0;
}
void SkipSpaces();
llvm::StringRef GetStringRef() const { return m_packet; }
bool Empty() { return m_packet.empty(); }
size_t GetBytesLeft() {
if (m_index < m_packet.size())
return m_packet.size() - m_index;
return 0;
}
char GetChar(char fail_value = '\0');
char PeekChar(char fail_value = '\0') {
const char *cstr = Peek();
if (cstr)
return cstr[0];
return fail_value;
}
int DecodeHexU8();
uint8_t GetHexU8(uint8_t fail_value = 0, bool set_eof_on_fail = true);
bool GetHexU8Ex(uint8_t &ch, bool set_eof_on_fail = true);
bool GetNameColonValue(llvm::StringRef &name, llvm::StringRef &value);
int32_t GetS32(int32_t fail_value, int base = 0);
uint32_t GetU32(uint32_t fail_value, int base = 0);
int64_t GetS64(int64_t fail_value, int base = 0);
uint64_t GetU64(uint64_t fail_value, int base = 0);
uint32_t GetHexMaxU32(bool little_endian, uint32_t fail_value);
uint64_t GetHexMaxU64(bool little_endian, uint64_t fail_value);
size_t GetHexBytes(llvm::MutableArrayRef<uint8_t> dest,
uint8_t fail_fill_value);
size_t GetHexBytesAvail(llvm::MutableArrayRef<uint8_t> dest);
size_t GetHexByteString(std::string &str);
size_t GetHexByteStringFixedLength(std::string &str, uint32_t nibble_length);
size_t GetHexByteStringTerminatedBy(std::string &str, char terminator);
bool ConsumeFront(const llvm::StringRef &str);
const char *Peek() {
if (m_index < m_packet.size())
return m_packet.c_str() + m_index;
return nullptr;
}
protected:
bool fail() {
m_index = UINT64_MAX;
return false;
}
/// The string in which to extract data.
std::string m_packet;
/// When extracting data from a packet, this index will march along as things
/// get extracted. If set to UINT64_MAX the end of the packet data was
/// reached when decoding information.
uint64_t m_index;
};
#endif // utility_StringExtractor_h_
|