X-Git-Url: https://gitweb.ps.run/matrix_esp_thesis/blobdiff_plain/d43e8671acc5709c192e159e0d91626f0677cdf1..50f9d03ea29a8ae2f82d5edb5404fdebf4316b13:/src/matrix.h

diff --git a/src/matrix.h b/src/matrix.h
index d61c59b..81b6170 100644
--- a/src/matrix.h
+++ b/src/matrix.h
@@ -2,49 +2,482 @@
 #define MATRIX__H
 
 #include <stdbool.h>
+#include <stdlib.h>
 #include <string.h>
+#include <time.h>
 
 #include <olm/olm.h>
 
 
 
+#define USER_ID_SIZE 64
+#define ROOM_ID_SIZE 128
+#define SERVER_SIZE 20
+#define ACCESS_TOKEN_SIZE 40
+#define DEVICE_ID_SIZE 20
+#define EXPIRE_MS_SIZE 20
+#define REFRESH_TOKEN_SIZE 20
+#define MAX_URL_LEN 1024
 
+#define OLM_IDENTITY_KEYS_JSON_SIZE 128
+#define DEVICE_KEY_SIZE 44
+#define SIGNING_KEY_SIZE 44
+#define ONETIME_KEY_SIZE 44
+#define MASTER_KEY_SIZE 44
 
-typedef struct FixedBuffer {
-    void * ptr;
-    int size;
-    int len;
-} FixedBuffer;
+#define KEY_SHARE_EVENT_LEN 1024
 
-FixedBuffer
-FixedBuf(const char * str);
+#define OLM_ACCOUNT_MEMORY_SIZE 7528
+#define OLM_ACCOUNT_RANDOM_SIZE (32+32)
 
+#define OLM_SESSION_MEMORY_SIZE 3352
+#define OLM_ENCRYPT_RANDOM_SIZE 32
+#define OLM_OUTBOUND_SESSION_RANDOM_SIZE (32*2)
 
+#define OLM_ONETIME_KEYS_RANDOM_SIZE (32*10)
+#define OLM_KEY_ID_SIZE 32
 
-#define ACCESS_TOKEN_LEN 20 // TODO: fix
+#define OLM_SIGNATURE_SIZE 128
+
+#define MEGOLM_OUTBOUND_SESSION_MEMORY_SIZE 232
+#define MEGOLM_SESSION_ID_SIZE 44
+#define MEGOLM_SESSION_KEY_SIZE 306
+#define MEGOLM_INIT_RANDOM_SIZE (4*32 + 32)
+
+#define JSON_ONETIME_KEY_SIZE 128
+#define JSON_ONETIME_KEY_SIGNED_SIZE 256
+#define JSON_SIGNATURE_SIZE 256
+
+#define NUM_MEGOLM_SESSIONS 2
+#define NUM_OLM_SESSIONS 2
+#define NUM_DEVICES 10
+
+// HTTP
+
+typedef struct MatrixHttpConnection MatrixHttpConnection;
+
+bool
+MatrixHttpInit(
+    MatrixHttpConnection ** hc,
+    const char * host);
+
+bool
+MatrixHttpDeinit(
+    MatrixHttpConnection ** hc);
+    
+bool
+MatrixHttpSetAccessToken(
+    MatrixHttpConnection * hc,
+    const char * accessToken);
+
+bool
+MatrixHttpGet(
+    MatrixHttpConnection * hc,
+    const char * url,
+    char * outResponseBuffer, int outResponseCap,
+    bool authenticated);
+
+bool
+MatrixHttpPost(
+    MatrixHttpConnection * hc,
+    const char * url,
+    const char * requestBuffer,
+    char * outResponseBuffer, int outResponseCap,
+    bool authenticated);
+
+bool
+MatrixHttpPut(
+    MatrixHttpConnection * hc,
+    const char * url,
+    const char * requestBuffer,
+    char * outResponseBuffer, int outResponseCap,
+    bool authenticated);
+
+
+
+// Matrix Device
+
+typedef struct MatrixDevice {
+    char deviceId[DEVICE_ID_SIZE];
+    char deviceKey[DEVICE_KEY_SIZE];
+    char signingKey[SIGNING_KEY_SIZE];
+} MatrixDevice;
+
+
+// Matrix Olm Account
+
+typedef struct MatrixOlmAccount {
+    OlmAccount * account;
+    char memory[OLM_ACCOUNT_MEMORY_SIZE];
+} MatrixOlmAccount;
+
+bool
+MatrixOlmAccountInit(
+    MatrixOlmAccount * account);
+
+bool
+MatrixOlmAccountUnpickle(
+    MatrixOlmAccount * account,
+    void * pickled, int pickledLen,
+    const void * key, int keyLen);
+
+bool
+MatrixOlmAccountGetDeviceKey(
+    MatrixOlmAccount * account,
+    char * key, int keyCap);
+    
+bool
+MatrixOlmAccountGetSigningKey(
+    MatrixOlmAccount * account,
+    char * key, int keyCap);
+
+
+// Matrix Olm Session
+
+typedef struct MatrixOlmSession {
+    const char * deviceId; // TODO: char[]
+
+    int type;
+    OlmSession * session;
+    char memory[OLM_SESSION_MEMORY_SIZE];
+} MatrixOlmSession;
+
+bool
+MatrixOlmSessionUnpickle(
+    MatrixOlmSession * session,
+    const char * deviceId,
+    void * pickled, int pickledLen,
+    const void * key, int keyLen);
+
+bool
+MatrixOlmSessionFrom(
+    MatrixOlmSession * session,
+    OlmAccount * olmAccount,
+    const char * deviceId,
+    const char * deviceKey,
+    const char * encrypted);
+
+bool
+MatrixOlmSessionTo(
+    MatrixOlmSession * session,
+    OlmAccount * olmAccount,
+    const char * deviceId,
+    const char * deviceKey,
+    const char * deviceOnetimeKey);
+
+bool
+MatrixOlmSessionEncrypt(
+    MatrixOlmSession * session,
+    const char * plaintext,
+    char * outBuffer, int outBufferCap);
+
+bool
+MatrixOlmSessionDecrypt(
+    MatrixOlmSession * session,
+    size_t messageType,
+    char * encrypted,
+    char * outBuffer, int outBufferCap);
+
+
+// Matrix Megolm Session
+
+typedef struct MatrixMegolmInSession {
+    char roomId[ROOM_ID_SIZE];
+    char id[MEGOLM_SESSION_ID_SIZE];
+    char key[MEGOLM_SESSION_KEY_SIZE];
+
+    OlmInboundGroupSession * session;
+    char memory[MEGOLM_OUTBOUND_SESSION_MEMORY_SIZE];
+
+} MatrixMegolmInSession;
+
+bool
+MatrixMegolmInSessionInit(
+    MatrixMegolmInSession * session,
+    const char * roomId,
+    const char * sessionId,
+    const char * sessionKey, int sessionKeyLen);
+
+bool
+MatrixMegolmInSessionDecrypt(
+    MatrixMegolmInSession * session,
+    const char * encrypted, int encryptedLen,
+    char * outDecrypted, int outDecryptedCap);
+
+typedef struct MatrixMegolmOutSession {
+    char roomId[ROOM_ID_SIZE];
+    char id[MEGOLM_SESSION_ID_SIZE];
+    char key[MEGOLM_SESSION_KEY_SIZE];
+
+    OlmOutboundGroupSession * session;
+    char memory[MEGOLM_OUTBOUND_SESSION_MEMORY_SIZE];
+} MatrixMegolmOutSession;
+
+bool
+MatrixMegolmOutSessionInit(
+    MatrixMegolmOutSession * session,
+    const char * roomId);
+
+bool
+MatrixMegolmOutSessionEncrypt(
+    MatrixMegolmOutSession * session,
+    const char * plaintext,
+    char * outBuffer, int outBufferCap);
+
+
+// Matrix Client
 
 typedef struct MatrixClient {
-    OlmAccount * olmAcc;
-    char accessToken[ACCESS_TOKEN_LEN];
+    MatrixOlmAccount olmAccount;
+
+    MatrixMegolmInSession megolmInSessions[NUM_MEGOLM_SESSIONS];
+    int numMegolmInSessions;
+    MatrixMegolmOutSession megolmOutSessions[NUM_MEGOLM_SESSIONS];
+    int numMegolmOutSessions;
+    MatrixOlmSession olmSessions[NUM_OLM_SESSIONS];
+    int numOlmSessions;
+    
+    MatrixDevice devices[NUM_DEVICES];
+    int numDevices;
+    
+    // char deviceKey[DEVICE_KEY_SIZE];
+    // char signingKey[DEVICE_KEY_SIZE];
+
+    char userId[USER_ID_SIZE];
+    char accessToken[ACCESS_TOKEN_SIZE];
+    char deviceId[DEVICE_ID_SIZE];
+    char expireMs[EXPIRE_MS_SIZE];
+    char refreshToken[REFRESH_TOKEN_SIZE];
+    char masterKey[MASTER_KEY_SIZE];
+
+    bool verified;
+
+    MatrixHttpConnection * hc;
 } MatrixClient;
 
 bool
 MatrixClientInit(
+    MatrixClient * client);
+
+bool
+MatrixClientSetAccessToken(
+    MatrixClient * client,
+    const char * accessToken);
+
+bool
+MatrixClientSetDeviceId(
+    MatrixClient * client,
+    const char * deviceId);
+
+bool
+MatrixClientSetUserId(
+    MatrixClient * client,
+    const char * userId);
+
+bool
+MatrixClientGenerateOnetimeKeys(
     MatrixClient * client,
-    FixedBuffer server
-);
+    int numberOfKeys);
+
+bool
+MatrixClientUploadOnetimeKeys(
+    MatrixClient * client);
+
+bool
+MatrixClientUploadDeviceKeys(
+    MatrixClient * client);
+
+bool
+MatrixClientClaimOnetimeKey(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    char * outOnetimeKey, int outOnetimeKeyCap);
 
 bool
 MatrixClientLoginPassword(
     MatrixClient * client,
-    FixedBuffer username,
-    FixedBuffer password
-);
+    const char * username,
+    const char * password,
+    const char * displayName);
+    
+bool
+MatrixClientSendEvent(
+    MatrixClient * client,
+    const char * roomId,
+    const char * msgType,
+    const char * msgBody);
+    
+bool
+MatrixClientSendEventEncrypted(
+    MatrixClient * client,
+    const char * roomId,
+    const char * msgType,
+    const char * msgBody);
+
+void
+HandleEvent(
+    MatrixClient * client,
+    const char * event, int eventLen);
+
+void
+HandleRoomEvent(
+    MatrixClient * client,
+    const char * room, int roomLen,
+    const char * event, int eventLen);
+
+bool
+MatrixClientSync(
+    MatrixClient * client,
+    char * outSyncBuffer, int outSyncCap,
+    char * nextBatch, int nextBatchCap);
+
+bool
+MatrixClientGetRoomEvent(
+    MatrixClient * client,
+    const char * roomId,
+    const char * eventId,
+    char * outEvent, int outEventCap);
+
+bool
+MatrixClientShareMegolmOutSession(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    MatrixMegolmOutSession * session);
 
 bool
-MatrixClientGetAccessToken(
+MatrixClientShareMegolmOutSessionTest(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    MatrixMegolmOutSession * session);
+
+bool
+MatrixClientGetMegolmOutSession(
+    MatrixClient * client,
+    const char * roomId,
+    MatrixMegolmOutSession ** outSession);
+
+bool
+MatrixClientNewMegolmOutSession(
+    MatrixClient * client,
+    const char * roomId,
+    MatrixMegolmOutSession ** outSession);
+
+bool
+MatrixClientGetMegolmInSession(
+    MatrixClient * client,
+    const char * roomId, int roomIdLen,
+    const char * sessionId, int sessionIdLen,
+    MatrixMegolmInSession ** outSession);
+
+bool
+MatrixClientNewMegolmInSession(
+    MatrixClient * client,
+    const char * roomId,
+    const char * sessionId,
+    const char * sessionKey,
+    MatrixMegolmInSession ** outSession);
+    
+bool
+MatrixClientRequestMegolmInSession(
+    MatrixClient * client,
+    const char * roomId,
+    const char * sessionId,
+    const char * senderKey,
+    const char * userId,
+    const char * deviceId); // TODO: remove deviceId (query all devices)
+
+bool
+MatrixClientGetOlmSession(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    MatrixOlmSession ** outSession);
+
+bool
+MatrixClientNewOlmSessionIn(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    const char * encrypted,
+    MatrixOlmSession ** outSession);
+    
+bool
+MatrixClientNewOlmSessionOut(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    MatrixOlmSession ** outSession);
+
+bool
+MatrixClientSendToDevice(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    const char * message,
+    const char * msgType);
+
+bool
+MatrixClientSendToDeviceEncrypted(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId,
+    const char * message,
+    const char * msgType);
+
+bool
+MatrixClientSendDummy(
+    MatrixClient * client,
+    const char * userId,
+    const char * deviceId);
+
+bool
+MatrixClientRequestDeviceKey(
+    MatrixClient * client,
+    const char * deviceId,
+    char * outDeviceKey, int outDeviceKeyCap);
+    
+bool
+MatrixClientRequestSigningKey(
+    MatrixClient * client,
+    const char * deviceId,
+    char * outSigningKey, int outSigningKeyCap);
+
+bool
+MatrixClientRequestMasterKey(
+    MatrixClient * client,
+    char * outMasterKey, int outMasterKeyCap);
+
+bool
+MatrixClientRequestDeviceKeys(
+    MatrixClient * client);
+
+bool
+MatrixClientDeleteDevice(
+    MatrixClient * client);
+
+
+// util
+
+void
+Randomize(uint8_t * random, int randomLen);
+
+bool
+JsonEscape(
+    const char * sIn, int sInLen,
+    char * sOut, int sOutCap);
+    
+bool
+JsonCanonicalize(
+    const char * sIn, int sInLen,
+    char * sOut, int sOutCap);
+    
+bool
+JsonSign(
     MatrixClient * client,
-    FixedBuffer * outBuffer
-);
+    const char * sIn, int sInLen,
+    char * sOut, int sOutCap);
 
 #endif