1//===-- StringExtractor.h ---------------------------------------*- C++ -*-===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#ifndef utility_StringExtractor_h_
11#define utility_StringExtractor_h_
12
13// C Includes
14// C++ Includes
15#include <string>
16#include <stdint.h>
17
18// Other libraries and framework includes
19// Project includes
20
21class StringExtractor
22{
23public:
24
25    enum {
26        BigEndian = 0,
27        LittleEndian = 1
28    };
29    //------------------------------------------------------------------
30    // Constructors and Destructors
31    //------------------------------------------------------------------
32    StringExtractor();
33    StringExtractor(const char *packet_cstr);
34    StringExtractor(const StringExtractor& rhs);
35    virtual ~StringExtractor();
36
37    //------------------------------------------------------------------
38    // Operators
39    //------------------------------------------------------------------
40    const StringExtractor&
41    operator=(const StringExtractor& rhs);
42
43    // Returns true if the file position is still valid for the data
44    // contained in this string extractor object.
45    bool
46    IsGood() const
47    {
48        return m_index != UINT64_MAX;
49    }
50
51    uint64_t
52    GetFilePos () const
53    {
54        return m_index;
55    }
56
57    void
58    SetFilePos (uint32_t idx)
59    {
60        m_index = idx;
61    }
62
63    void
64    Clear ()
65    {
66        m_packet.clear();
67        m_index = 0;
68    }
69
70    std::string &
71    GetStringRef ()
72    {
73        return m_packet;
74    }
75
76    const std::string &
77    GetStringRef () const
78    {
79        return m_packet;
80    }
81
82    bool
83    Empty()
84    {
85        return m_packet.empty();
86    }
87
88    size_t
89    GetBytesLeft ()
90    {
91        if (m_index < m_packet.size())
92            return m_packet.size() - m_index;
93        return 0;
94    }
95    char
96    GetChar (char fail_value = '\0');
97
98    uint8_t
99    GetHexU8 (uint8_t fail_value = 0, bool set_eof_on_fail = true);
100
101    bool
102    GetNameColonValue (std::string &name, std::string &value);
103
104    uint32_t
105    GetU32 (uint32_t fail_value, int base = 0);
106
107    uint32_t
108    GetHexMaxU32 (bool little_endian, uint32_t fail_value);
109
110    uint64_t
111    GetHexMaxU64 (bool little_endian, uint64_t fail_value);
112
113    size_t
114    GetHexBytes (void *dst, size_t dst_len, uint8_t fail_fill_value);
115
116    uint64_t
117    GetHexWithFixedSize (uint32_t byte_size, bool little_endian, uint64_t fail_value);
118
119    size_t
120    GetHexByteString (std::string &str);
121
122    const char *
123    Peek ()
124    {
125        if (m_index < m_packet.size())
126            return m_packet.c_str() + m_index;
127        return NULL;
128    }
129
130protected:
131    //------------------------------------------------------------------
132    // For StringExtractor only
133    //------------------------------------------------------------------
134    std::string m_packet;   // The string in which to extract data.
135    uint64_t m_index;       // When extracting data from a packet, this index
136                            // will march along as things get extracted. If set
137                            // to UINT64_MAX the end of the packet data was
138                            // reached when decoding information
139};
140
141#endif  // utility_StringExtractor_h_
142