1 files changed, 84 insertions, 1 deletions
diff --git a/src/net/manaserv/messagein.cpp b/src/net/manaserv/messagein.cpp
index 4994de4c..8065d313 100644
--- a/src/net/manaserv/messagein.cpp
+++ b/src/net/manaserv/messagein.cpp
@@ -27,15 +27,41 @@
 namespace ManaServ {
 
 MessageIn::MessageIn(const char *data, unsigned int length):
-        Net::MessageIn(data, length)
+    mData(data),
+    mLength(length),
+    mDebugMode(false),
+    mPos(0)
 {
     // Read the message ID
     mId = readInt16();
+
+    // Read and clear the debug flag
+    mDebugMode = mId & ManaServ::XXMSG_DEBUG_FLAG;
+    mId &= ~ManaServ::XXMSG_DEBUG_FLAG;
+}
+
+uint8_t MessageIn::readInt8()
+{
+    uint8_t value = 0;
+
+    if (!readValueType(ManaServ::Int8))
+        return value;
+
+    if (mPos < mLength)
+    {
+        value = mData[mPos];
+    }
+    mPos++;
+    return value;
 }
 
 uint16_t MessageIn::readInt16()
 {
     uint16_t value = 0;
+
+    if (!readValueType(ManaServ::Int16))
+        return value;
+
     if (mPos + 2 <= mLength)
     {
         uint16_t t;
@@ -49,6 +75,10 @@ uint16_t MessageIn::readInt16()
 uint32_t MessageIn::readInt32()
 {
     uint32_t value = 0;
+
+    if (!readValueType(ManaServ::Int32))
+        return value;
+
     if (mPos + 4 <= mLength)
     {
         uint32_t t;
@@ -59,4 +89,57 @@ uint32_t MessageIn::readInt32()
     return value;
 }
 
+std::string MessageIn::readString(int length)
+{
+    if (!readValueType(ManaServ::String))
+        return std::string();
+
+    if (mDebugMode)
+    {
+        int fixedLength = (int16_t) readInt16();
+        if (fixedLength != length)
+        {
+            // String does not have the expected length
+            mPos = mLength + 1;
+            return std::string();
+        }
+    }
+
+    // Get string length
+    if (length < 0)
+    {
+        length = readInt16();
+    }
+
+    // Make sure the string isn't erroneous
+    if (length < 0 || mPos + length > mLength)
+    {
+        mPos = mLength + 1;
+        return std::string();
+    }
+
+    // Read the string
+    const char *stringBeg = mData + mPos;
+    const char *stringEnd = (const char *)memchr(stringBeg, '\0', length);
+    std::string readString(stringBeg,
+                           stringEnd ? stringEnd - stringBeg : length);
+    mPos += length;
+
+    return readString;
+}
+
+bool MessageIn::readValueType(ManaServ::ValueType type)
+{
+    if (!mDebugMode) // Verification not possible
+        return true;
+
+    if (mPos >= mLength)
+        return false;
+
+    uint8_t t = mData[mPos];
+    ++mPos;
+
+    return t == type;
 }
+
+} // ManaServ