summaryrefslogtreecommitdiff
path: root/src/ilasm/binstr.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/ilasm/binstr.h')
-rw-r--r--src/ilasm/binstr.h77
1 files changed, 77 insertions, 0 deletions
diff --git a/src/ilasm/binstr.h b/src/ilasm/binstr.h
new file mode 100644
index 0000000000..275b810a5c
--- /dev/null
+++ b/src/ilasm/binstr.h
@@ -0,0 +1,77 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+/**************************************************************************/
+/* a binary string (blob) class */
+
+#ifndef BINSTR_H
+#define BINSTR_H
+
+#include <string.h> // for memmove, memcpy ...
+
+#ifdef _PREFAST_
+#pragma warning(push)
+#pragma warning(disable:22008) // "Suppress PREfast warnings about integer overflow"
+#endif
+
+class BinStr {
+public:
+ BinStr() { len = 0L; max = 8L; ptr_ = buff; }
+ BinStr(BYTE* pb, DWORD cb) { len = cb; max = cb+8; ptr_ = pb; }
+ ~BinStr() { if (ptr_ != buff) delete [] ptr_; }
+
+ void insertInt8(int val) { if (len >= max) Realloc(); memmove(ptr_+1, ptr_, len); *ptr_ = val; len++; }
+ void insertInt32(int val) { if (len + 4 > max) Realloc(); memmove(ptr_+4, ptr_, len); SET_UNALIGNED_32(&ptr_[0], val); len+=4; }
+ void appendInt8(int val) { if (len >= max) Realloc(); ptr_[len++] = val; }
+ void appendInt16(int val) { if (len + 2 > max) Realloc(); SET_UNALIGNED_16(&ptr_[len], val); len += 2; }
+ void appendInt32(int val) { if (len + 4 > max) Realloc(); SET_UNALIGNED_32(&ptr_[len], val); len += 4; }
+ void appendInt64(__int64 *pval) { if (len + 8 > max) Realloc(8); SET_UNALIGNED_64(&ptr_[len],(*pval)); len += 8; }
+ unsigned __int8* getBuff(unsigned size) {
+ if (len + size > max) Realloc(size);
+ _ASSERTE(len + size <= max);
+ unsigned __int8* ret = &ptr_[len];
+ len += size;
+ return(ret);
+ }
+ void append(BinStr* str) {
+ memcpy(getBuff(str->length()), str->ptr(), str->length());
+ }
+
+ void appendFrom(BinStr* str, unsigned ix) {
+ _ASSERTE(str->length() >= ix);
+ if (str->length() >= ix)
+ {
+ memcpy(getBuff(str->length()-ix), str->ptr()+ix, str->length()-ix);
+ }
+ }
+
+ void remove(unsigned size) { _ASSERTE(len >= size); len -= size; }
+
+ unsigned __int8* ptr() { return(ptr_); }
+ unsigned length() { return(len); }
+
+private:
+ void Realloc(unsigned atLeast = 4) {
+ max = max * 2;
+ if (max < atLeast + len)
+ max = atLeast + len;
+ _ASSERTE(max >= len + atLeast);
+ unsigned __int8* newPtr = new unsigned __int8[max];
+ memcpy(newPtr, ptr_, len);
+ if (ptr_ != buff) delete [] ptr_;
+ ptr_ = newPtr;
+ }
+
+private:
+ unsigned len;
+ unsigned max;
+ unsigned __int8 *ptr_;
+ unsigned __int8 buff[8];
+};
+BinStr* BinStrToUnicode(BinStr* pSource, bool Swap = false);
+#ifdef _PREFAST_
+#pragma warning(pop)
+#endif
+
+#endif
+