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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
//===-- StringExtractor.h ---------------------------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef utility_StringExtractor_h_
#define utility_StringExtractor_h_
// Other libraries and framework includes
// Project includes
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/StringRef.h"
#include <stddef.h> // for size_t
#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);
StringExtractor(const StringExtractor &rhs);
virtual ~StringExtractor();
//------------------------------------------------------------------
// Operators
//------------------------------------------------------------------
const StringExtractor &operator=(const StringExtractor &rhs);
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();
std::string &GetStringRef() { return m_packet; }
const std::string &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);
uint64_t GetHexWithFixedSize(uint32_t byte_size, bool little_endian,
uint64_t fail_value);
size_t GetHexByteString(std::string &str);
size_t GetHexByteStringFixedLength(std::string &str, uint32_t nibble_length);
size_t GetHexByteStringTerminatedBy(std::string &str, char terminator);
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;
}
//------------------------------------------------------------------
// For StringExtractor only
//------------------------------------------------------------------
std::string m_packet; // The string in which to extract data.
uint64_t m_index; // 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
};
#endif // utility_StringExtractor_h_
|
