1 files changed, 78 insertions, 0 deletions
diff --git a/include/axolotl/session.hh b/include/axolotl/session.hh
new file mode 100644
index 0000000..c69699d
--- /dev/null
+++ b/include/axolotl/session.hh
@@ -0,0 +1,78 @@
+#ifndef AXOLOTL_SESSION_HH_
+#define AXOLOTL_SESSION_HH_
+
+#include "axolotl/ratchet.hh"
+
+namespace axolotl {
+
+struct RemoteKey {
+    std::uint32_t id;
+    Curve25519PublicKey key;
+};
+
+struct RemoteKeys {
+};
+
+
+enum struct MessageType {
+    PRE_KEY_MESSAGE = 0,
+    MESSAGE = 1,
+};
+
+
+struct Session {
+    bool received_message;
+    RemoteKey alice_identity_key;
+    RemoteKey alice_base_key;
+    RemoteKey bob_identity_key;
+    RemoteKey bob_one_time_key;
+    Ratchet ratchet;
+
+    void initialise_outbound_session_random_length();
+
+    void initialise_outbound_session(
+        Account const & local_account,
+        RemoteKey const & identity_key,
+        RemoteKey const & one_time_key,
+        std::uint8_t const * random, std::size_t random_length
+    );
+
+    void initialise_inbound_session(
+        Account & local_account,
+        std::uint8_t const * one_time_key_message, std::size_t message_length
+    );
+
+    void matches_inbound_session(
+        std::uint8_t const * one_time_key_message, std::size_t message_length
+    );
+
+    MessageType encrypt_message_type();
+
+    std::size_t encrypt_message_length(
+        std::size_t plaintext_length
+    );
+
+    std::size_t encrypt_random_length();
+
+    std::size_t encrypt(
+        std::uint8_t const * plaintext, std::size_t plaintext_length,
+        std::uint8_t const * random, std::size_t random_length,
+        std::uint8_t * message, std::size_t message_length
+    );
+
+    std::size_t decrypt_max_plaintext_length(
+        MessageType message_type,
+        std::uint8_t const * message, std::size_t message_length
+    );
+
+    std::size_t decrypt(
+        MessageType message_type,
+        std::uint8_t const * message, std::size_t message_length,
+        std::uint8_t * plaintext, std::size_t max_plaintext_length
+    );
+};
+
+
+} // namespace axolotl
+
+#endif /* AXOLOTL_SESSION_HH_ */