| 1 | // Copyright (C) 2018 Intel Corporation. |
| 2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only |
| 3 | |
| 4 | #ifndef QCBORSTREAMREADER_H |
| 5 | #define QCBORSTREAMREADER_H |
| 6 | |
| 7 | #include <QtCore/qbytearray.h> |
| 8 | #include <QtCore/qcborcommon.h> |
| 9 | #include <QtCore/qfloat16.h> |
| 10 | #include <QtCore/qscopedpointer.h> |
| 11 | #include <QtCore/qstring.h> |
| 12 | #include <QtCore/qstringview.h> |
| 13 | |
| 14 | QT_REQUIRE_CONFIG(cborstreamreader); |
| 15 | |
| 16 | /* X11 headers use these values too, but as defines */ |
| 17 | #if defined(False) && defined(True) |
| 18 | # undef True |
| 19 | # undef False |
| 20 | #endif |
| 21 | |
| 22 | QT_BEGIN_NAMESPACE |
| 23 | |
| 24 | class QIODevice; |
| 25 | |
| 26 | class QCborStreamReaderPrivate; |
| 27 | class Q_CORE_EXPORT QCborStreamReader |
| 28 | { |
| 29 | Q_GADGET |
| 30 | public: |
| 31 | enum Type : quint8 { |
| 32 | UnsignedInteger = 0x00, |
| 33 | NegativeInteger = 0x20, |
| 34 | ByteString = 0x40, |
| 35 | ByteArray = ByteString, |
| 36 | TextString = 0x60, |
| 37 | String = TextString, |
| 38 | Array = 0x80, |
| 39 | Map = 0xa0, |
| 40 | Tag = 0xc0, |
| 41 | SimpleType = 0xe0, |
| 42 | HalfFloat = 0xf9, |
| 43 | Float16 = HalfFloat, |
| 44 | Float = 0xfa, |
| 45 | Double = 0xfb, |
| 46 | |
| 47 | Invalid = 0xff |
| 48 | }; |
| 49 | Q_ENUM(Type) |
| 50 | |
| 51 | enum StringResultCode { |
| 52 | EndOfString = 0, |
| 53 | Ok = 1, |
| 54 | Error = -1 |
| 55 | }; |
| 56 | template <typename Container> struct StringResult { |
| 57 | Container data; |
| 58 | StringResultCode status = Error; |
| 59 | }; |
| 60 | Q_ENUM(StringResultCode) |
| 61 | |
| 62 | QCborStreamReader(); |
| 63 | QCborStreamReader(const char *data, qsizetype len); |
| 64 | QCborStreamReader(const quint8 *data, qsizetype len); |
| 65 | explicit QCborStreamReader(const QByteArray &data); |
| 66 | explicit QCborStreamReader(QIODevice *device); |
| 67 | ~QCborStreamReader(); |
| 68 | Q_DISABLE_COPY(QCborStreamReader) |
| 69 | |
| 70 | void setDevice(QIODevice *device); |
| 71 | QIODevice *device() const; |
| 72 | void addData(const QByteArray &data); |
| 73 | void addData(const char *data, qsizetype len); |
| 74 | void addData(const quint8 *data, qsizetype len) |
| 75 | { addData(data: reinterpret_cast<const char *>(data), len); } |
| 76 | void reparse(); |
| 77 | void clear(); |
| 78 | void reset(); |
| 79 | |
| 80 | #if QT_CORE_REMOVED_SINCE(6, 7) |
| 81 | QCborError lastError(); |
| 82 | #endif |
| 83 | QCborError lastError() const; |
| 84 | |
| 85 | qint64 currentOffset() const; |
| 86 | |
| 87 | bool isValid() const { return !isInvalid(); } |
| 88 | |
| 89 | int containerDepth() const; |
| 90 | QCborStreamReader::Type parentContainerType() const; |
| 91 | bool hasNext() const noexcept Q_DECL_PURE_FUNCTION; |
| 92 | bool next(int maxRecursion = 10000); |
| 93 | |
| 94 | Type type() const { return QCborStreamReader::Type(type_); } |
| 95 | bool isUnsignedInteger() const { return type() == UnsignedInteger; } |
| 96 | bool isNegativeInteger() const { return type() == NegativeInteger; } |
| 97 | bool isInteger() const { return quint8(type()) <= quint8(NegativeInteger); } |
| 98 | bool isByteArray() const { return type() == ByteArray; } |
| 99 | bool isString() const { return type() == String; } |
| 100 | bool isArray() const { return type() == Array; } |
| 101 | bool isMap() const { return type() == Map; } |
| 102 | bool isTag() const { return type() == Tag; } |
| 103 | bool isSimpleType() const { return type() == SimpleType; } |
| 104 | bool isFloat16() const { return type() == Float16; } |
| 105 | bool isFloat() const { return type() == Float; } |
| 106 | bool isDouble() const { return type() == Double; } |
| 107 | bool isInvalid() const { return type() == Invalid; } |
| 108 | |
| 109 | bool isSimpleType(QCborSimpleType st) const { return isSimpleType() && toSimpleType() == st; } |
| 110 | bool isFalse() const { return isSimpleType(st: QCborSimpleType::False); } |
| 111 | bool isTrue() const { return isSimpleType(st: QCborSimpleType::True); } |
| 112 | bool isBool() const { return isFalse() || isTrue(); } |
| 113 | bool isNull() const { return isSimpleType(st: QCborSimpleType::Null); } |
| 114 | bool isUndefined() const { return isSimpleType(st: QCborSimpleType::Undefined); } |
| 115 | |
| 116 | bool isLengthKnown() const noexcept Q_DECL_PURE_FUNCTION; |
| 117 | quint64 length() const; |
| 118 | |
| 119 | bool isContainer() const { return isMap() || isArray(); } |
| 120 | bool enterContainer() { Q_ASSERT(isContainer()); return _enterContainer_helper(); } |
| 121 | bool leaveContainer(); |
| 122 | |
| 123 | bool readAndAppendToString(QString &dst) |
| 124 | { Q_ASSERT(isString()); return _readAndAppendToString_helper(dst); } |
| 125 | bool readAndAppendToUtf8String(QByteArray &dst) |
| 126 | { Q_ASSERT(isString()); return _readAndAppendToUtf8String_helper(dst); } |
| 127 | bool readAndAppendToByteArray(QByteArray &dst) |
| 128 | { Q_ASSERT(isByteArray()); return _readAndAppendToByteArray_helper(dst); } |
| 129 | StringResult<QString> readString() { Q_ASSERT(isString()); return _readString_helper(); } |
| 130 | StringResult<QByteArray> readUtf8String() { Q_ASSERT(isString()); return _readUtf8String_helper(); } |
| 131 | StringResult<QByteArray> readByteArray(){ Q_ASSERT(isByteArray()); return _readByteArray_helper(); } |
| 132 | qsizetype currentStringChunkSize() const{ Q_ASSERT(isString() || isByteArray()); return _currentStringChunkSize(); } |
| 133 | StringResult<qsizetype> readStringChunk(char *ptr, qsizetype maxlen); |
| 134 | |
| 135 | bool toBool() const { Q_ASSERT(isBool()); return value64 - int(QCborSimpleType::False); } |
| 136 | QCborTag toTag() const { Q_ASSERT(isTag()); return QCborTag(value64); } |
| 137 | quint64 toUnsignedInteger() const { Q_ASSERT(isUnsignedInteger()); return value64; } |
| 138 | QCborNegativeInteger toNegativeInteger() const { Q_ASSERT(isNegativeInteger()); return QCborNegativeInteger(value64 + 1); } |
| 139 | QCborSimpleType toSimpleType() const{ Q_ASSERT(isSimpleType()); return QCborSimpleType(value64); } |
| 140 | qfloat16 toFloat16() const { Q_ASSERT(isFloat16()); return _toFloatingPoint<qfloat16>(); } |
| 141 | float toFloat() const { Q_ASSERT(isFloat()); return _toFloatingPoint<float>(); } |
| 142 | double toDouble() const { Q_ASSERT(isDouble()); return _toFloatingPoint<double>(); } |
| 143 | |
| 144 | qint64 toInteger() const |
| 145 | { |
| 146 | Q_ASSERT(isInteger()); |
| 147 | qint64 v = qint64(value64); |
| 148 | if (isNegativeInteger()) |
| 149 | return -v - 1; |
| 150 | return v; |
| 151 | } |
| 152 | QString readAllString() |
| 153 | { |
| 154 | QString dst; |
| 155 | if (!readAndAppendToString(dst)) |
| 156 | dst = QString{}; |
| 157 | return dst; |
| 158 | } |
| 159 | QByteArray readAllUtf8String() |
| 160 | { |
| 161 | QByteArray dst; |
| 162 | if (!readAndAppendToUtf8String(dst)) |
| 163 | dst = QByteArray{}; |
| 164 | return dst; |
| 165 | } |
| 166 | QByteArray readAllByteArray() |
| 167 | { |
| 168 | QByteArray dst; |
| 169 | if (!readAndAppendToByteArray(dst)) |
| 170 | dst = QByteArray{}; |
| 171 | return dst; |
| 172 | } |
| 173 | |
| 174 | private: |
| 175 | void preparse(); |
| 176 | bool _enterContainer_helper(); |
| 177 | StringResult<QString> _readString_helper(); |
| 178 | StringResult<QByteArray> _readUtf8String_helper(); |
| 179 | StringResult<QByteArray> _readByteArray_helper(); |
| 180 | qsizetype _currentStringChunkSize() const; |
| 181 | bool _readAndAppendToString_helper(QString &); |
| 182 | bool _readAndAppendToUtf8String_helper(QByteArray &); |
| 183 | bool _readAndAppendToByteArray_helper(QByteArray &); |
| 184 | |
| 185 | template <typename FP> FP _toFloatingPoint() const noexcept |
| 186 | { |
| 187 | using UIntFP = typename QIntegerForSizeof<FP>::Unsigned; |
| 188 | UIntFP u = UIntFP(value64); |
| 189 | FP f; |
| 190 | memcpy(static_cast<void *>(&f), &u, sizeof(f)); |
| 191 | return f; |
| 192 | } |
| 193 | |
| 194 | friend QCborStreamReaderPrivate; |
| 195 | friend class QCborContainerPrivate; |
| 196 | quint64 value64; |
| 197 | QScopedPointer<QCborStreamReaderPrivate> d; |
| 198 | quint8 type_; |
| 199 | quint8 reserved[3] = {}; |
| 200 | }; |
| 201 | |
| 202 | QT_END_NAMESPACE |
| 203 | |
| 204 | #if defined(QT_X11_DEFINES_FOUND) |
| 205 | # define True 1 |
| 206 | # define False 0 |
| 207 | #endif |
| 208 | |
| 209 | #endif // QCBORSTREAMREADER_H |
| 210 | |