123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- // Copyright 2014 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #ifndef BASE_BIG_ENDIAN_H_
- #define BASE_BIG_ENDIAN_H_
- #include <stddef.h>
- #include <stdint.h>
- #include <type_traits>
- #include "base/base_export.h"
- #include "base/strings/string_piece.h"
- namespace base {
- // Read an integer (signed or unsigned) from |buf| in Big Endian order.
- // Note: this loop is unrolled with -O1 and above.
- // NOTE(szym): glibc dns-canon.c use ntohs(*(uint16_t*)ptr) which is
- // potentially unaligned.
- // This would cause SIGBUS on ARMv5 or earlier and ARMv6-M.
- template<typename T>
- inline void ReadBigEndian(const char buf[], T* out) {
- static_assert(std::is_integral<T>::value, "T has to be an integral type.");
- // Make an unsigned version of the output type to make shift possible
- // without UB.
- typename std::make_unsigned<T>::type unsigned_result = uint8_t{buf[0]};
- for (size_t i = 1; i < sizeof(T); ++i) {
- unsigned_result <<= 8;
- // Must cast to uint8_t to avoid clobbering by sign extension.
- unsigned_result |= static_cast<uint8_t>(buf[i]);
- }
- *out = unsigned_result;
- }
- // Write an integer (signed or unsigned) |val| to |buf| in Big Endian order.
- // Note: this loop is unrolled with -O1 and above.
- template<typename T>
- inline void WriteBigEndian(char buf[], T val) {
- static_assert(std::is_integral<T>::value, "T has to be an integral type.");
- auto unsigned_val = static_cast<typename std::make_unsigned<T>::type>(val);
- for (size_t i = 0; i < sizeof(T); ++i) {
- buf[sizeof(T) - i - 1] = static_cast<char>(unsigned_val & 0xFF);
- unsigned_val >>= 8;
- }
- }
- // Specializations to make clang happy about the (dead code) shifts above.
- template <>
- inline void ReadBigEndian<uint8_t>(const char buf[], uint8_t* out) {
- *out = buf[0];
- }
- template <>
- inline void WriteBigEndian<uint8_t>(char buf[], uint8_t val) {
- buf[0] = static_cast<char>(val);
- }
- template <>
- inline void ReadBigEndian<int8_t>(const char buf[], int8_t* out) {
- *out = buf[0];
- }
- template <>
- inline void WriteBigEndian<int8_t>(char buf[], int8_t val) {
- buf[0] = static_cast<char>(val);
- }
- // Allows reading integers in network order (big endian) while iterating over
- // an underlying buffer. All the reading functions advance the internal pointer.
- class BASE_EXPORT BigEndianReader {
- public:
- BigEndianReader(const char* buf, size_t len);
- const char* ptr() const { return ptr_; }
- size_t remaining() const { return end_ - ptr_; }
- bool Skip(size_t len);
- bool ReadBytes(void* out, size_t len);
- // Creates a StringPiece in |out| that points to the underlying buffer.
- bool ReadPiece(base::StringPiece* out, size_t len);
- bool ReadU8(uint8_t* value);
- bool ReadU16(uint16_t* value);
- bool ReadU32(uint32_t* value);
- bool ReadU64(uint64_t* value);
- // Reads a length-prefixed region:
- // 1. reads a big-endian length L from the buffer;
- // 2. sets |*out| to a StringPiece over the next L many bytes
- // of the buffer (beyond the end of the bytes encoding the length); and
- // 3. skips the main reader past this L-byte substring.
- //
- // Fails if reading a U8 or U16 fails, or if the parsed length is greater
- // than the number of bytes remaining in the stream.
- //
- // On failure, leaves the stream at the same position
- // as before the call.
- bool ReadU8LengthPrefixed(base::StringPiece* out);
- bool ReadU16LengthPrefixed(base::StringPiece* out);
- private:
- // Hidden to promote type safety.
- template<typename T>
- bool Read(T* v);
- template <typename T>
- bool ReadLengthPrefixed(base::StringPiece* out);
- const char* ptr_;
- const char* end_;
- };
- // Allows writing integers in network order (big endian) while iterating over
- // an underlying buffer. All the writing functions advance the internal pointer.
- class BASE_EXPORT BigEndianWriter {
- public:
- BigEndianWriter(char* buf, size_t len);
- char* ptr() const { return ptr_; }
- size_t remaining() const { return end_ - ptr_; }
- bool Skip(size_t len);
- bool WriteBytes(const void* buf, size_t len);
- bool WriteU8(uint8_t value);
- bool WriteU16(uint16_t value);
- bool WriteU32(uint32_t value);
- bool WriteU64(uint64_t value);
- private:
- // Hidden to promote type safety.
- template<typename T>
- bool Write(T v);
- char* ptr_;
- char* end_;
- };
- } // namespace base
- #endif // BASE_BIG_ENDIAN_H_
|