diff options
Diffstat (limited to 'src/encryptedPacket.h')
| -rw-r--r-- | src/encryptedPacket.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/encryptedPacket.h b/src/encryptedPacket.h index c924ad9..8bf38f5 100644 --- a/src/encryptedPacket.h +++ b/src/encryptedPacket.h @@ -48,6 +48,7 @@ #include "datatypes.h" #include "buffer.h" +#include "packetSource.h" class Cipher; class EncryptedPacket : public Buffer @@ -143,6 +144,9 @@ public: void removeAuthTag(); uint8_t* getAuthTag(); uint32_t getAuthTagLength(); + + void setEndpoint(PacketSourceEndpoint & ); + PacketSourceEndpoint getEndpoint() const; private: EncryptedPacket(); @@ -166,6 +170,7 @@ private: uint8_t* payload_; uint8_t* auth_tag_; uint32_t auth_tag_length_; + PacketSourceEndpoint endpoint_; }; #endif |