packet and cypher fixes

1 files changed, 9 insertions, 3 deletions

diff --git a/encryptedPacket.h b/encryptedPacket.h
index 7b70e17..83b831e 100644
--- a/encryptedPacket.h
+++ b/encryptedPacket.h
@@ -34,7 +34,7 @@
 #include "datatypes.h"
 #include "buffer.h"
 #include "authTag.h"
-
+class Cypher;
 class EncryptedPacket : public Buffer
 {
 public:
@@ -98,7 +98,7 @@ public:
    * Get the maximum payload size
    * @return maximum payload size
    */
-  u_int32_t getSize() const;
+  u_int32_t getMaxLength() const;
 
   /**
    * Set the real length of the payload
@@ -111,6 +111,8 @@ public:
   AuthTag getAuthTag() const;
   void setAuthTag(AuthTag& tag);
 
+	void setPayloadLength(u_int8_t payload_length);
+
                        
 //  bool hasHeader() const;
 //  Packet& withHeader(bool b);
@@ -131,9 +133,13 @@ private:
 
   struct HeaderStruct* header_;
   AuthTag* auth_tag_;
-  u_int32_t size_;
+  u_int32_t max_length_;
 
   static const u_int32_t AUTHTAG_SIZE = 10;     // 10byte
+protected:
+	friend class Cypher;
+	u_int8_t * payload_;
+	u_int32_t payload_length_;
 };
 
 #endif