lldatapacker.h
上传用户:king477883
上传日期:2021-03-01
资源大小:9553k
文件大小:14k
源码类别:

游戏引擎

开发平台:

C++ Builder

  1. /** 
  2.  * @file lldatapacker.h
  3.  * @brief Data packer declaration for tightly storing binary data.
  4.  *
  5.  * $LicenseInfo:firstyear=2002&license=viewergpl$
  6.  * 
  7.  * Copyright (c) 2002-2010, Linden Research, Inc.
  8.  * 
  9.  * Second Life Viewer Source Code
  10.  * The source code in this file ("Source Code") is provided by Linden Lab
  11.  * to you under the terms of the GNU General Public License, version 2.0
  12.  * ("GPL"), unless you have obtained a separate licensing agreement
  13.  * ("Other License"), formally executed by you and Linden Lab.  Terms of
  14.  * the GPL can be found in doc/GPL-license.txt in this distribution, or
  15.  * online at http://secondlifegrid.net/programs/open_source/licensing/gplv2
  16.  * 
  17.  * There are special exceptions to the terms and conditions of the GPL as
  18.  * it is applied to this Source Code. View the full text of the exception
  19.  * in the file doc/FLOSS-exception.txt in this software distribution, or
  20.  * online at
  21.  * http://secondlifegrid.net/programs/open_source/licensing/flossexception
  22.  * 
  23.  * By copying, modifying or distributing this software, you acknowledge
  24.  * that you have read and understood your obligations described above,
  25.  * and agree to abide by those obligations.
  26.  * 
  27.  * ALL LINDEN LAB SOURCE CODE IS PROVIDED "AS IS." LINDEN LAB MAKES NO
  28.  * WARRANTIES, EXPRESS, IMPLIED OR OTHERWISE, REGARDING ITS ACCURACY,
  29.  * COMPLETENESS OR PERFORMANCE.
  30.  * $/LicenseInfo$
  31.  */
  32. #ifndef LL_LLDATAPACKER_H 
  33. #define LL_LLDATAPACKER_H
  34. class LLColor4;
  35. class LLColor4U;
  36. class LLVector2;
  37. class LLVector3;
  38. class LLVector4;
  39. class LLUUID;
  40. class LLDataPacker
  41. {
  42. public:
  43. virtual ~LLDataPacker() {}
  44. // Not required to override, but error to call?
  45. virtual void reset();
  46. virtual void dumpBufferToLog();
  47. virtual BOOL hasNext() const = 0;
  48. virtual BOOL packString(const std::string& value, const char *name) = 0;
  49. virtual BOOL unpackString(std::string& value, const char *name) = 0;
  50. virtual BOOL packBinaryData(const U8 *value, S32 size, const char *name) = 0;
  51. virtual BOOL unpackBinaryData(U8 *value, S32 &size, const char *name) = 0;
  52. // Constant size binary data packing
  53. virtual BOOL packBinaryDataFixed(const U8 *value, S32 size, const char *name) = 0;
  54. virtual BOOL unpackBinaryDataFixed(U8 *value, S32 size, const char *name) = 0;
  55. virtual BOOL packU8(const U8 value, const char *name) = 0;
  56. virtual BOOL unpackU8(U8 &value, const char *name) = 0;
  57. virtual BOOL packU16(const U16 value, const char *name) = 0;
  58. virtual BOOL unpackU16(U16 &value, const char *name) = 0;
  59. virtual BOOL packU32(const U32 value, const char *name) = 0;
  60. virtual BOOL unpackU32(U32 &value, const char *name) = 0;
  61. virtual BOOL packS32(const S32 value, const char *name) = 0;
  62. virtual BOOL unpackS32(S32 &value, const char *name) = 0;
  63. virtual BOOL packF32(const F32 value, const char *name) = 0;
  64. virtual BOOL unpackF32(F32 &value, const char *name) = 0;
  65. // Packs a float into an integer, using the given size
  66. // and picks the right U* data type to pack into.
  67. BOOL packFixed(const F32 value, const char *name,
  68. const BOOL is_signed, const U32 int_bits, const U32 frac_bits);
  69. BOOL unpackFixed(F32 &value, const char *name,
  70. const BOOL is_signed, const U32 int_bits, const U32 frac_bits);
  71. virtual BOOL packColor4(const LLColor4 &value, const char *name) = 0;
  72. virtual BOOL unpackColor4(LLColor4 &value, const char *name) = 0;
  73. virtual BOOL packColor4U(const LLColor4U &value, const char *name) = 0;
  74. virtual BOOL unpackColor4U(LLColor4U &value, const char *name) = 0;
  75. virtual BOOL packVector2(const LLVector2 &value, const char *name) = 0;
  76. virtual BOOL unpackVector2(LLVector2 &value, const char *name) = 0;
  77. virtual BOOL packVector3(const LLVector3 &value, const char *name) = 0;
  78. virtual BOOL unpackVector3(LLVector3 &value, const char *name) = 0;
  79. virtual BOOL packVector4(const LLVector4 &value, const char *name) = 0;
  80. virtual BOOL unpackVector4(LLVector4 &value, const char *name) = 0;
  81. virtual BOOL packUUID(const LLUUID &value, const char *name) = 0;
  82. virtual BOOL unpackUUID(LLUUID &value, const char *name) = 0;
  83. U32 getPassFlags() const { return mPassFlags; }
  84. void setPassFlags(U32 flags) { mPassFlags = flags; }
  85. protected:
  86. LLDataPacker();
  87. protected:
  88. U32 mPassFlags;
  89. BOOL mWriteEnabled; // disable this to do things like determine filesize without actually copying data
  90. };
  91. class LLDataPackerBinaryBuffer : public LLDataPacker
  92. {
  93. public:
  94. LLDataPackerBinaryBuffer(U8 *bufferp, S32 size)
  95. : LLDataPacker(),
  96. mBufferp(bufferp),
  97. mCurBufferp(bufferp),
  98. mBufferSize(size)
  99. {
  100. mWriteEnabled = TRUE;
  101. }
  102. LLDataPackerBinaryBuffer()
  103. : LLDataPacker(),
  104. mBufferp(NULL),
  105. mCurBufferp(NULL),
  106. mBufferSize(0)
  107. {
  108. }
  109. /*virtual*/ BOOL packString(const std::string& value, const char *name);
  110. /*virtual*/ BOOL unpackString(std::string& value, const char *name);
  111. /*virtual*/ BOOL packBinaryData(const U8 *value, S32 size, const char *name);
  112. /*virtual*/ BOOL unpackBinaryData(U8 *value, S32 &size, const char *name);
  113. // Constant size binary data packing
  114. /*virtual*/ BOOL packBinaryDataFixed(const U8 *value, S32 size, const char *name);
  115. /*virtual*/ BOOL unpackBinaryDataFixed(U8 *value, S32 size, const char *name);
  116. /*virtual*/ BOOL packU8(const U8 value, const char *name);
  117. /*virtual*/ BOOL unpackU8(U8 &value, const char *name);
  118. /*virtual*/ BOOL packU16(const U16 value, const char *name);
  119. /*virtual*/ BOOL unpackU16(U16 &value, const char *name);
  120. /*virtual*/ BOOL packU32(const U32 value, const char *name);
  121. /*virtual*/ BOOL unpackU32(U32 &value, const char *name);
  122. /*virtual*/ BOOL packS32(const S32 value, const char *name);
  123. /*virtual*/ BOOL unpackS32(S32 &value, const char *name);
  124. /*virtual*/ BOOL packF32(const F32 value, const char *name);
  125. /*virtual*/ BOOL unpackF32(F32 &value, const char *name);
  126. /*virtual*/ BOOL packColor4(const LLColor4 &value, const char *name);
  127. /*virtual*/ BOOL unpackColor4(LLColor4 &value, const char *name);
  128. /*virtual*/ BOOL packColor4U(const LLColor4U &value, const char *name);
  129. /*virtual*/ BOOL unpackColor4U(LLColor4U &value, const char *name);
  130. /*virtual*/ BOOL packVector2(const LLVector2 &value, const char *name);
  131. /*virtual*/ BOOL unpackVector2(LLVector2 &value, const char *name);
  132. /*virtual*/ BOOL packVector3(const LLVector3 &value, const char *name);
  133. /*virtual*/ BOOL unpackVector3(LLVector3 &value, const char *name);
  134. /*virtual*/ BOOL packVector4(const LLVector4 &value, const char *name);
  135. /*virtual*/ BOOL unpackVector4(LLVector4 &value, const char *name);
  136. /*virtual*/ BOOL packUUID(const LLUUID &value, const char *name);
  137. /*virtual*/ BOOL unpackUUID(LLUUID &value, const char *name);
  138. S32 getCurrentSize() const { return (S32)(mCurBufferp - mBufferp); }
  139. S32 getBufferSize() const { return mBufferSize; }
  140. void reset() { mCurBufferp = mBufferp; mWriteEnabled = (mCurBufferp != NULL); }
  141. void freeBuffer() { delete [] mBufferp; mBufferp = mCurBufferp = NULL; mBufferSize = 0; mWriteEnabled = FALSE; }
  142. void assignBuffer(U8 *bufferp, S32 size)
  143. {
  144. mBufferp = bufferp;
  145. mCurBufferp = bufferp;
  146. mBufferSize = size;
  147. mWriteEnabled = TRUE;
  148. }
  149. const LLDataPackerBinaryBuffer& operator=(const LLDataPackerBinaryBuffer &a);
  150. /*virtual*/ BOOL hasNext() const { return getCurrentSize() < getBufferSize(); }
  151. /*virtual*/ void dumpBufferToLog();
  152. protected:
  153. inline BOOL verifyLength(const S32 data_size, const char *name);
  154. U8 *mBufferp;
  155. U8 *mCurBufferp;
  156. S32 mBufferSize;
  157. };
  158. inline BOOL LLDataPackerBinaryBuffer::verifyLength(const S32 data_size, const char *name)
  159. {
  160. if (mWriteEnabled && (mCurBufferp - mBufferp) > mBufferSize - data_size)
  161. {
  162. llwarns << "Buffer overflow in BinaryBuffer length verify, field name " << name << "!" << llendl;
  163. llwarns << "Current pos: " << (int)(mCurBufferp - mBufferp) << " Buffer size: " << mBufferSize << " Data size: " << data_size << llendl;
  164. return FALSE;
  165. }
  166. return TRUE;
  167. }
  168. class LLDataPackerAsciiBuffer : public LLDataPacker
  169. {
  170. public:
  171. LLDataPackerAsciiBuffer(char* bufferp, S32 size)
  172. {
  173. mBufferp = bufferp;
  174. mCurBufferp = bufferp;
  175. mBufferSize = size;
  176. mPassFlags = 0;
  177. mIncludeNames = FALSE;
  178. mWriteEnabled = TRUE;
  179. }
  180. LLDataPackerAsciiBuffer()
  181. {
  182. mBufferp = NULL;
  183. mCurBufferp = NULL;
  184. mBufferSize = 0;
  185. mPassFlags = 0;
  186. mIncludeNames = FALSE;
  187. mWriteEnabled = FALSE;
  188. }
  189. /*virtual*/ BOOL packString(const std::string& value, const char *name);
  190. /*virtual*/ BOOL unpackString(std::string& value, const char *name);
  191. /*virtual*/ BOOL packBinaryData(const U8 *value, S32 size, const char *name);
  192. /*virtual*/ BOOL unpackBinaryData(U8 *value, S32 &size, const char *name);
  193. // Constant size binary data packing
  194. /*virtual*/ BOOL packBinaryDataFixed(const U8 *value, S32 size, const char *name);
  195. /*virtual*/ BOOL unpackBinaryDataFixed(U8 *value, S32 size, const char *name);
  196. /*virtual*/ BOOL packU8(const U8 value, const char *name);
  197. /*virtual*/ BOOL unpackU8(U8 &value, const char *name);
  198. /*virtual*/ BOOL packU16(const U16 value, const char *name);
  199. /*virtual*/ BOOL unpackU16(U16 &value, const char *name);
  200. /*virtual*/ BOOL packU32(const U32 value, const char *name);
  201. /*virtual*/ BOOL unpackU32(U32 &value, const char *name);
  202. /*virtual*/ BOOL packS32(const S32 value, const char *name);
  203. /*virtual*/ BOOL unpackS32(S32 &value, const char *name);
  204. /*virtual*/ BOOL packF32(const F32 value, const char *name);
  205. /*virtual*/ BOOL unpackF32(F32 &value, const char *name);
  206. /*virtual*/ BOOL packColor4(const LLColor4 &value, const char *name);
  207. /*virtual*/ BOOL unpackColor4(LLColor4 &value, const char *name);
  208. /*virtual*/ BOOL packColor4U(const LLColor4U &value, const char *name);
  209. /*virtual*/ BOOL unpackColor4U(LLColor4U &value, const char *name);
  210. /*virtual*/ BOOL packVector2(const LLVector2 &value, const char *name);
  211. /*virtual*/ BOOL unpackVector2(LLVector2 &value, const char *name);
  212. /*virtual*/ BOOL packVector3(const LLVector3 &value, const char *name);
  213. /*virtual*/ BOOL unpackVector3(LLVector3 &value, const char *name);
  214. /*virtual*/ BOOL packVector4(const LLVector4 &value, const char *name);
  215. /*virtual*/ BOOL unpackVector4(LLVector4 &value, const char *name);
  216. /*virtual*/ BOOL packUUID(const LLUUID &value, const char *name);
  217. /*virtual*/ BOOL unpackUUID(LLUUID &value, const char *name);
  218. void setIncludeNames(BOOL b) { mIncludeNames = b; }
  219. // Include the trailing NULL so it's always a valid string
  220. S32 getCurrentSize() const { return (S32)(mCurBufferp - mBufferp) + 1; }
  221. S32 getBufferSize() const { return mBufferSize; }
  222. /*virtual*/ void reset() { mCurBufferp = mBufferp; mWriteEnabled = (mCurBufferp != NULL); }
  223. /*virtual*/ BOOL hasNext() const { return getCurrentSize() < getBufferSize(); }
  224. inline void freeBuffer();
  225. inline void assignBuffer(char* bufferp, S32 size);
  226. void dump();
  227. protected:
  228. void writeIndentedName(const char *name);
  229. BOOL getValueStr(const char *name, char *out_value, const S32 value_len);
  230. protected:
  231. inline BOOL verifyLength(const S32 data_size, const char *name);
  232. char *mBufferp;
  233. char *mCurBufferp;
  234. S32 mBufferSize;
  235. BOOL mIncludeNames; // useful for debugging, print the name of each field
  236. };
  237. inline void LLDataPackerAsciiBuffer::freeBuffer()
  238. {
  239. delete [] mBufferp; 
  240. mBufferp = mCurBufferp = NULL; 
  241. mBufferSize = 0;
  242. mWriteEnabled = FALSE;
  243. }
  244. inline void LLDataPackerAsciiBuffer::assignBuffer(char* bufferp, S32 size)
  245. {
  246. mBufferp = bufferp;
  247. mCurBufferp = bufferp;
  248. mBufferSize = size;
  249. mWriteEnabled = TRUE;
  250. }
  251. inline BOOL LLDataPackerAsciiBuffer::verifyLength(const S32 data_size, const char *name)
  252. {
  253. if (mWriteEnabled && (mCurBufferp - mBufferp) > mBufferSize - data_size)
  254. {
  255. llwarns << "Buffer overflow in AsciiBuffer length verify, field name " << name << "!" << llendl;
  256. llwarns << "Current pos: " << (int)(mCurBufferp - mBufferp) << " Buffer size: " << mBufferSize << " Data size: " << data_size << llendl;
  257. return FALSE;
  258. }
  259. return TRUE;
  260. }
  261. class LLDataPackerAsciiFile : public LLDataPacker
  262. {
  263. public:
  264. LLDataPackerAsciiFile(LLFILE *fp, const S32 indent = 2)
  265. :  LLDataPacker(),
  266. mIndent(indent),
  267. mFP(fp),
  268. mOutputStream(NULL),
  269. mInputStream(NULL)
  270. {
  271. }
  272. LLDataPackerAsciiFile(std::ostream& output_stream, const S32 indent = 2)
  273. :  LLDataPacker(),
  274. mIndent(indent),
  275. mFP(NULL),
  276. mOutputStream(&output_stream),
  277. mInputStream(NULL)
  278. {
  279. mWriteEnabled = TRUE;
  280. }
  281. LLDataPackerAsciiFile(std::istream& input_stream, const S32 indent = 2)
  282. :  LLDataPacker(),
  283. mIndent(indent),
  284. mFP(NULL),
  285. mOutputStream(NULL),
  286. mInputStream(&input_stream)
  287. {
  288. }
  289. /*virtual*/ BOOL packString(const std::string& value, const char *name);
  290. /*virtual*/ BOOL unpackString(std::string& value, const char *name);
  291. /*virtual*/ BOOL packBinaryData(const U8 *value, S32 size, const char *name);
  292. /*virtual*/ BOOL unpackBinaryData(U8 *value, S32 &size, const char *name);
  293. /*virtual*/ BOOL packBinaryDataFixed(const U8 *value, S32 size, const char *name);
  294. /*virtual*/ BOOL unpackBinaryDataFixed(U8 *value, S32 size, const char *name);
  295. /*virtual*/ BOOL packU8(const U8 value, const char *name);
  296. /*virtual*/ BOOL unpackU8(U8 &value, const char *name);
  297. /*virtual*/ BOOL packU16(const U16 value, const char *name);
  298. /*virtual*/ BOOL unpackU16(U16 &value, const char *name);
  299. /*virtual*/ BOOL packU32(const U32 value, const char *name);
  300. /*virtual*/ BOOL unpackU32(U32 &value, const char *name);
  301. /*virtual*/ BOOL packS32(const S32 value, const char *name);
  302. /*virtual*/ BOOL unpackS32(S32 &value, const char *name);
  303. /*virtual*/ BOOL packF32(const F32 value, const char *name);
  304. /*virtual*/ BOOL unpackF32(F32 &value, const char *name);
  305. /*virtual*/ BOOL packColor4(const LLColor4 &value, const char *name);
  306. /*virtual*/ BOOL unpackColor4(LLColor4 &value, const char *name);
  307. /*virtual*/ BOOL packColor4U(const LLColor4U &value, const char *name);
  308. /*virtual*/ BOOL unpackColor4U(LLColor4U &value, const char *name);
  309. /*virtual*/ BOOL packVector2(const LLVector2 &value, const char *name);
  310. /*virtual*/ BOOL unpackVector2(LLVector2 &value, const char *name);
  311. /*virtual*/ BOOL packVector3(const LLVector3 &value, const char *name);
  312. /*virtual*/ BOOL unpackVector3(LLVector3 &value, const char *name);
  313. /*virtual*/ BOOL packVector4(const LLVector4 &value, const char *name);
  314. /*virtual*/ BOOL unpackVector4(LLVector4 &value, const char *name);
  315. /*virtual*/ BOOL packUUID(const LLUUID &value, const char *name);
  316. /*virtual*/ BOOL unpackUUID(LLUUID &value, const char *name);
  317. protected:
  318. void writeIndentedName(const char *name);
  319. BOOL getValueStr(const char *name, char *out_value, const S32 value_len);
  320. /*virtual*/ BOOL hasNext() const { return true; }
  321. protected:
  322. S32 mIndent;
  323. LLFILE *mFP;
  324. std::ostream* mOutputStream;
  325. std::istream* mInputStream;
  326. };
  327. #endif // LL_LLDATAPACKER