Streamer.h 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. /*
  2. * Copyright (c) 2020, Hüseyin ASLITÜRK <asliturk@hotmail.com>
  3. * Copyright (c) 2020, the SerenityOS developers.
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #pragma once
  8. #include <AK/Array.h>
  9. #include <AK/Endian.h>
  10. #include <AK/Types.h>
  11. #include <string.h>
  12. namespace Gfx {
  13. class Streamer {
  14. public:
  15. constexpr Streamer(u8 const* data, size_t size)
  16. : m_data_ptr(data)
  17. , m_size_remaining(size)
  18. {
  19. }
  20. template<typename T>
  21. constexpr bool read(T& value)
  22. {
  23. Array<u8, sizeof(T)> network_buffer {};
  24. auto network_value = new (network_buffer.data()) AK::NetworkOrdered<T> {};
  25. auto res = read_bytes(network_buffer.data(), sizeof(T));
  26. value = T(*network_value);
  27. return res;
  28. }
  29. constexpr bool read_bytes(u8* buffer, size_t count)
  30. {
  31. if (m_size_remaining < count) {
  32. return false;
  33. }
  34. memcpy(buffer, m_data_ptr, count);
  35. m_data_ptr += count;
  36. m_size_remaining -= count;
  37. return true;
  38. }
  39. constexpr bool at_end() const { return !m_size_remaining; }
  40. constexpr void step_back()
  41. {
  42. m_data_ptr -= 1;
  43. m_size_remaining += 1;
  44. }
  45. private:
  46. u8 const* m_data_ptr { nullptr };
  47. size_t m_size_remaining { 0 };
  48. };
  49. }