BinaryReader.cpp 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. //
  2. // BinaryReader.cpp
  3. //
  4. // $Id: //poco/1.4/Foundation/src/BinaryReader.cpp#1 $
  5. //
  6. // Library: Foundation
  7. // Package: Streams
  8. // Module: BinaryReaderWriter
  9. //
  10. // Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
  11. // and Contributors.
  12. //
  13. // SPDX-License-Identifier: BSL-1.0
  14. //
  15. #include "Poco/BinaryReader.h"
  16. #include "Poco/ByteOrder.h"
  17. #include "Poco/TextEncoding.h"
  18. #include "Poco/TextConverter.h"
  19. #include <algorithm>
  20. namespace Poco {
  21. BinaryReader::BinaryReader(std::istream& istr, StreamByteOrder byteOrder):
  22. _istr(istr),
  23. _pTextConverter(0)
  24. {
  25. #if defined(POCO_ARCH_BIG_ENDIAN)
  26. _flipBytes = (byteOrder == LITTLE_ENDIAN_BYTE_ORDER);
  27. #else
  28. _flipBytes = (byteOrder == BIG_ENDIAN_BYTE_ORDER);
  29. #endif
  30. }
  31. BinaryReader::BinaryReader(std::istream& istr, TextEncoding& encoding, StreamByteOrder byteOrder):
  32. _istr(istr),
  33. _pTextConverter(new TextConverter(encoding, Poco::TextEncoding::global()))
  34. {
  35. #if defined(POCO_ARCH_BIG_ENDIAN)
  36. _flipBytes = (byteOrder == LITTLE_ENDIAN_BYTE_ORDER);
  37. #else
  38. _flipBytes = (byteOrder == BIG_ENDIAN_BYTE_ORDER);
  39. #endif
  40. }
  41. BinaryReader::~BinaryReader()
  42. {
  43. delete _pTextConverter;
  44. }
  45. BinaryReader& BinaryReader::operator >> (bool& value)
  46. {
  47. _istr.read((char*) &value, sizeof(value));
  48. return *this;
  49. }
  50. BinaryReader& BinaryReader::operator >> (char& value)
  51. {
  52. _istr.read((char*) &value, sizeof(value));
  53. return *this;
  54. }
  55. BinaryReader& BinaryReader::operator >> (unsigned char& value)
  56. {
  57. _istr.read((char*) &value, sizeof(value));
  58. return *this;
  59. }
  60. BinaryReader& BinaryReader::operator >> (signed char& value)
  61. {
  62. _istr.read((char*) &value, sizeof(value));
  63. return *this;
  64. }
  65. BinaryReader& BinaryReader::operator >> (short& value)
  66. {
  67. _istr.read((char*) &value, sizeof(value));
  68. if (_flipBytes) value = ByteOrder::flipBytes(value);
  69. return *this;
  70. }
  71. BinaryReader& BinaryReader::operator >> (unsigned short& value)
  72. {
  73. _istr.read((char*) &value, sizeof(value));
  74. if (_flipBytes) value = ByteOrder::flipBytes(value);
  75. return *this;
  76. }
  77. BinaryReader& BinaryReader::operator >> (int& value)
  78. {
  79. _istr.read((char*) &value, sizeof(value));
  80. if (_flipBytes) value = ByteOrder::flipBytes(value);
  81. return *this;
  82. }
  83. BinaryReader& BinaryReader::operator >> (unsigned int& value)
  84. {
  85. _istr.read((char*) &value, sizeof(value));
  86. if (_flipBytes) value = ByteOrder::flipBytes(value);
  87. return *this;
  88. }
  89. BinaryReader& BinaryReader::operator >> (long& value)
  90. {
  91. _istr.read((char*) &value, sizeof(value));
  92. #if defined(POCO_LONG_IS_64_BIT)
  93. if (_flipBytes) value = ByteOrder::flipBytes((Int64) value);
  94. #else
  95. if (_flipBytes) value = ByteOrder::flipBytes((Int32) value);
  96. #endif
  97. return *this;
  98. }
  99. BinaryReader& BinaryReader::operator >> (unsigned long& value)
  100. {
  101. _istr.read((char*) &value, sizeof(value));
  102. #if defined(POCO_LONG_IS_64_BIT)
  103. if (_flipBytes) value = ByteOrder::flipBytes((UInt64) value);
  104. #else
  105. if (_flipBytes) value = ByteOrder::flipBytes((UInt32) value);
  106. #endif
  107. return *this;
  108. }
  109. BinaryReader& BinaryReader::operator >> (float& value)
  110. {
  111. if (_flipBytes)
  112. {
  113. char* ptr = (char*) &value;
  114. ptr += sizeof(value);
  115. for (unsigned i = 0; i < sizeof(value); ++i)
  116. _istr.read(--ptr, 1);
  117. }
  118. else
  119. {
  120. _istr.read((char*) &value, sizeof(value));
  121. }
  122. return *this;
  123. }
  124. BinaryReader& BinaryReader::operator >> (double& value)
  125. {
  126. if (_flipBytes)
  127. {
  128. char* ptr = (char*) &value;
  129. ptr += sizeof(value);
  130. for (unsigned i = 0; i < sizeof(value); ++i)
  131. _istr.read(--ptr, 1);
  132. }
  133. else
  134. {
  135. _istr.read((char*) &value, sizeof(value));
  136. }
  137. return *this;
  138. }
  139. #if defined(POCO_HAVE_INT64) && !defined(POCO_LONG_IS_64_BIT)
  140. BinaryReader& BinaryReader::operator >> (Int64& value)
  141. {
  142. _istr.read((char*) &value, sizeof(value));
  143. if (_flipBytes) value = ByteOrder::flipBytes(value);
  144. return *this;
  145. }
  146. BinaryReader& BinaryReader::operator >> (UInt64& value)
  147. {
  148. _istr.read((char*) &value, sizeof(value));
  149. if (_flipBytes) value = ByteOrder::flipBytes(value);
  150. return *this;
  151. }
  152. #endif
  153. BinaryReader& BinaryReader::operator >> (std::string& value)
  154. {
  155. UInt32 size = 0;
  156. read7BitEncoded(size);
  157. value.clear();
  158. if (!_istr.good()) return *this;
  159. value.reserve(size);
  160. while (size--)
  161. {
  162. char c;
  163. if (!_istr.read(&c, 1).good()) break;
  164. value += c;
  165. }
  166. if (_pTextConverter)
  167. {
  168. std::string converted;
  169. _pTextConverter->convert(value, converted);
  170. std::swap(value, converted);
  171. }
  172. return *this;
  173. }
  174. void BinaryReader::read7BitEncoded(UInt32& value)
  175. {
  176. char c;
  177. value = 0;
  178. int s = 0;
  179. do
  180. {
  181. c = 0;
  182. _istr.read(&c, 1);
  183. UInt32 x = (c & 0x7F);
  184. x <<= s;
  185. value += x;
  186. s += 7;
  187. }
  188. while (c & 0x80);
  189. }
  190. #if defined(POCO_HAVE_INT64)
  191. void BinaryReader::read7BitEncoded(UInt64& value)
  192. {
  193. char c;
  194. value = 0;
  195. int s = 0;
  196. do
  197. {
  198. c = 0;
  199. _istr.read(&c, 1);
  200. UInt64 x = (c & 0x7F);
  201. x <<= s;
  202. value += x;
  203. s += 7;
  204. }
  205. while (c & 0x80);
  206. }
  207. #endif
  208. void BinaryReader::readRaw(std::streamsize length, std::string& value)
  209. {
  210. value.clear();
  211. value.reserve(static_cast<std::string::size_type>(length));
  212. while (length--)
  213. {
  214. char c;
  215. if (!_istr.read(&c, 1).good()) break;
  216. value += c;
  217. }
  218. }
  219. void BinaryReader::readRaw(char* buffer, std::streamsize length)
  220. {
  221. _istr.read(buffer, length);
  222. }
  223. void BinaryReader::readBOM()
  224. {
  225. UInt16 bom;
  226. _istr.read((char*) &bom, sizeof(bom));
  227. _flipBytes = bom != 0xFEFF;
  228. }
  229. } // namespace Poco