X-Git-Url: https://gitweb.ps.run/matrix_esp_thesis/blobdiff_plain/8d8ae609f0201ec4640738ff49b768e899695423..b231872efcb97e52856f5972efd161c6e9b03cd6:/src/matrix.c diff --git a/src/matrix.c b/src/matrix.c index 06409b8..40c2e5b 100644 --- a/src/matrix.c +++ b/src/matrix.c @@ -11,9 +11,11 @@ #define ENCRYPTED_REQUEST_SIZE (1024*5) #define ENCRYPTED_EVENT_SIZE (1024*10) -#define ROOMEVENT_REQUEST_SIZE 256 -#define ROOMEVENT_RESPONSE_SIZE 1024 -#define ROOMEVENT_URL "/_matrix/client/v3/rooms/%s/send/%s/%d" +#define ROOM_SEND_REQUEST_SIZE 256 +#define ROOM_SEND_RESPONSE_SIZE 1024 +#define ROOM_SEND_URL "/_matrix/client/v3/rooms/%s/send/%s/%d" + +#define ROOMKEY_REQUEST_SIZE (1024*4) #define TODEVICE_EVENT_SIZE (1024*5) #define TODEVICE_URL "/_matrix/client/v3/sendToDevice/%s/%d" @@ -31,9 +33,14 @@ #define KEYS_CLAIM_REQUEST_SIZE 1024 #define KEYS_CLAIM_RESPONSE_SIZE 1024 -#define JSON_QUERY_SIZE 128 +#define SYNC_TIMEOUT 5000 +#define JSON_QUERY_SIZE 128 +#define JSON_MAX_INDICES 100 +#define JSON_MAX_ENTRY_SIZE 1024 +#define MAX(a,b) ((a) > (b) ? (a) : (b)) +#define MIN(a,b) ((a) < (b) ? (a) : (b)) void Randomize( @@ -76,6 +83,69 @@ JsonEscape( return true; } +bool +JsonCanonicalize( + const char * sIn, int sInLen, + char * sOut, int sOutCap) +{ + snprintf(sOut, sOutCap, "{}"); + + int koff, klen, voff, vlen, vtype, off; + + struct Key { + const char * ptr; + int len; + }; + + struct Key keys[JSON_MAX_INDICES]; + int numKeys = 0; + + for (off = 0; (off = mjson_next(sIn, sInLen, off, &koff, &klen, &voff, &vlen, &vtype)) != 0; ) { + keys[numKeys].ptr = sIn + koff; + keys[numKeys].len = klen; + numKeys++; + } + + for (int i = 0; i < numKeys; i++) { + for (int j = i; j < numKeys; j++) { + if ( + strncmp( + keys[i].ptr, + keys[j].ptr, + MIN(keys[i].len, keys[j].len) + ) > 0 + ) { + struct Key k = keys[i]; + keys[i] = keys[j]; + keys[j] = k; + } + } + } + + for (int i = 0; i < numKeys; i++) { + char jp[JSON_QUERY_SIZE]; + snprintf(jp, JSON_QUERY_SIZE, "$.%.*s", keys[i].len-2, keys[i].ptr+1); + + const char * valPtr; + int valLen; + mjson_find(sIn, sInLen, jp, &valPtr, &valLen); + + static char newEntry[JSON_MAX_ENTRY_SIZE]; + snprintf(newEntry, JSON_MAX_ENTRY_SIZE, "{%.*s:%.*s}", keys[i].len, keys[i].ptr, valLen, valPtr); + + char * buffer = strdup(sOut); + + struct mjson_fixedbuf fb = { sOut, sOutCap, 0 }; + mjson_merge(buffer, strlen(buffer), newEntry, strlen(newEntry), mjson_print_fixed_buf, &fb); + + free(buffer); + } + + // TODO: recursively sort entries + + return true; +} + bool JsonSign( MatrixClient * client, const char * sIn, int sInLen, @@ -89,8 +159,8 @@ bool JsonSign( int signatureLen = res; - char thisSigningKey[DEVICE_KEY_SIZE]; - MatrixOlmAccountGetSigningKey(&client->olmAccount, thisSigningKey, DEVICE_KEY_SIZE); + static char thisSigningKey[SIGNING_KEY_SIZE]; + MatrixOlmAccountGetSigningKey(&client->olmAccount, thisSigningKey, SIGNING_KEY_SIZE); static char signatureJson[JSON_SIGNATURE_SIZE]; int signatureJsonLen = @@ -103,7 +173,8 @@ bool JsonSign( "}" "}", client->userId, - "1", + //"1", + client->deviceId, signatureLen, signature); struct mjson_fixedbuf result = { sOut, sOutCap, 0 }; @@ -181,7 +252,35 @@ MatrixOlmAccountGetSigningKey( return true; } -// TODO:in/outbound sessions +bool +MatrixOlmSessionFrom( + MatrixOlmSession * session, + OlmAccount * olmAccount, + const char * deviceId, + const char * deviceKey, + const char * encrypted) +{ + memset(session, 0, sizeof(MatrixOlmSession)); + + session->deviceId = deviceId; + + session->session = + olm_session(session->memory); + + char * encryptedCopy = strdup(encrypted); + + size_t res = + olm_create_inbound_session_from(session->session, olmAccount, + deviceKey, strlen(deviceKey), + encryptedCopy, strlen(encryptedCopy)); + + if (res == olm_error()) { + printf("error olm:%s\n", olm_session_last_error(session->session)); + } + + return res != olm_error(); +} + bool MatrixOlmSessionTo( MatrixOlmSession * session, @@ -211,7 +310,7 @@ MatrixOlmSessionTo( printf("error olm:%s\n", olm_session_last_error(session->session)); } - return session->session != NULL; + return res != olm_error(); } bool @@ -257,6 +356,85 @@ MatrixOlmSessionEncrypt( return res != olm_error(); } +bool +MatrixOlmSessionDecrypt( + MatrixOlmSession * session, + size_t messageType, + char * encrypted, + char * outBuffer, int outBufferCap) +{ + static uint8_t random[OLM_ENCRYPT_RANDOM_SIZE]; + Randomize(random, OLM_ENCRYPT_RANDOM_SIZE); + + size_t res = + olm_decrypt(session->session, + messageType, + encrypted, strlen(encrypted), + outBuffer, outBufferCap); + + if (res != olm_error() && res < outBufferCap) + outBuffer[res] = '\0'; + + return res != olm_error(); +} + +bool +MatrixMegolmInSessionInit( + MatrixMegolmInSession * session, + const char * roomId, + const char * sessionId, + const char * sessionKey, int sessionKeyLen) +{ + memset(session, 0, sizeof(MatrixMegolmInSession)); + + strncpy(session->roomId, roomId, sizeof(session->roomId)); + strncpy(session->id, sessionId, sizeof(session->id)); + strncpy(session->key, sessionKey, sizeof(session->key)); + + session->session = + olm_inbound_group_session(session->memory); + + size_t res = + olm_init_inbound_group_session( + // olm_import_inbound_group_session( + session->session, + (const uint8_t *)sessionKey, sessionKeyLen); + if (res == olm_error()) { + printf("Error initializing Megolm session: %s\n", olm_inbound_group_session_last_error(session->session)); + } + + return res != olm_error(); +} + +bool +MatrixMegolmInSessionDecrypt( + MatrixMegolmInSession * session, + const char * encrypted, int encryptedLen, + char * outDecrypted, int outDecryptedCap) +{ + // uint8_t buffer[1024]; + // memcpy(buffer, encrypted, encryptedLen); + + uint32_t megolmInMessageIndex; + + size_t res = + olm_group_decrypt(session->session, + (uint8_t *)encrypted, encryptedLen, + (uint8_t *)outDecrypted, outDecryptedCap, + &megolmInMessageIndex); + + printf("message index: %d\n", megolmInMessageIndex); + + if (res == olm_error()) { + printf("error decrypting megolm message: %s\n", olm_inbound_group_session_last_error(session->session)); + } + else { + printf("decrypted len: %d\n", res); + } + + return true; +} + // https://matrix.org/docs/guides/end-to-end-encryption-implementation-guide#starting-a-megolm-session bool MatrixMegolmOutSessionInit( @@ -510,7 +688,7 @@ MatrixClientUploadOnetimeKeys( static char requestBuffer[KEYS_UPLOAD_REQUEST_SIZE]; mjson_snprintf(requestBuffer, KEYS_UPLOAD_REQUEST_SIZE, - "{\"one_time_keys\":{"); + "{"); static char onetimeKeysBuffer[1024]; olm_account_one_time_keys(client->olmAccount.account, @@ -524,14 +702,15 @@ MatrixClientUploadOnetimeKeys( while ((off = mjson_next(keys, keysLen, off, &koff, &klen, &voff, &vlen, &vtype)) != 0) { static char keyJson[JSON_ONETIME_KEY_SIZE]; - snprintf(keyJson, JSON_ONETIME_KEY_SIZE, - "{\"key\":\"%.*s\"}", - vlen-2, keys + voff+1); + int keyJsonLen = + snprintf(keyJson, JSON_ONETIME_KEY_SIZE, + "{\"key\":\"%.*s\"}", + vlen-2, keys + voff+1); static char keyJsonSigned[JSON_ONETIME_KEY_SIGNED_SIZE]; JsonSign(client, - keyJson, JSON_ONETIME_KEY_SIZE, + keyJson, keyJsonLen, keyJsonSigned, JSON_ONETIME_KEY_SIGNED_SIZE); mjson_snprintf(requestBuffer+strlen(requestBuffer), KEYS_UPLOAD_REQUEST_SIZE-strlen(requestBuffer), @@ -540,13 +719,28 @@ MatrixClientUploadOnetimeKeys( keyJsonSigned); } - mjson_snprintf(requestBuffer+strlen(requestBuffer)-1, KEYS_UPLOAD_REQUEST_SIZE-strlen(requestBuffer), - "}}"); + if (requestBuffer[strlen(requestBuffer)-1] == ',') + requestBuffer[strlen(requestBuffer)-1] = '\0'; + + mjson_snprintf(requestBuffer+strlen(requestBuffer), KEYS_UPLOAD_REQUEST_SIZE-strlen(requestBuffer), + "}"); + + // static char onetimeKeysSignedBuffer[KEYS_UPLOAD_REQUEST_SIGNED_SIZE]; + // JsonSign(client, + // requestBuffer, strlen(requestBuffer), + // onetimeKeysSignedBuffer, KEYS_UPLOAD_REQUEST_SIZE); + + // static char finalEvent[KEYS_UPLOAD_REQUEST_SIGNED_SIZE]; + // snprintf(finalEvent, KEYS_UPLOAD_REQUEST_SIGNED_SIZE, + // "{\"one_time_keys\":%s}", onetimeKeysSignedBuffer); + static char finalEvent[KEYS_UPLOAD_REQUEST_SIGNED_SIZE]; + snprintf(finalEvent, KEYS_UPLOAD_REQUEST_SIGNED_SIZE, + "{\"one_time_keys\":%s}", requestBuffer); static char responseBuffer[KEYS_UPLOAD_RESPONSE_SIZE]; MatrixHttpPost(client, KEYS_UPLOAD_URL, - requestBuffer, + finalEvent, responseBuffer, KEYS_UPLOAD_RESPONSE_SIZE, true); @@ -565,31 +759,35 @@ MatrixClientUploadDeviceKey( static char deviceKeysBuffer[KEYS_UPLOAD_REQUEST_SIZE]; - mjson_snprintf(deviceKeysBuffer, KEYS_UPLOAD_REQUEST_SIZE, - "{\"device_keys\":{" - "\"algorithms\":[\"m.olm.v1.curve25519-aes-sha2\",\"m.megolm.v1.aes-sha2\"]," - "\"device_id\":\"%s\"," - "\"keys\":{" - "\"curve25519:%s\":\"%s\"," - "\"ed25519:%s\":\"%s\"" - "}," - "\"user_id\":\"%s\"" - "}}", - client->deviceId, - client->deviceId, thisDeviceKey, - client->deviceId, thisSigningKey, - client->userId); + int deviceKeysBufferLen = + mjson_snprintf(deviceKeysBuffer, KEYS_UPLOAD_REQUEST_SIZE, + "{" + "\"algorithms\":[\"m.olm.v1.curve25519-aes-sha2\",\"m.megolm.v1.aes-sha2\"]," + "\"device_id\":\"%s\"," + "\"keys\":{" + "\"curve25519:%s\":\"%s\"," + "\"ed25519:%s\":\"%s\"" + "}," + "\"user_id\":\"%s\"" + "}", + client->deviceId, + client->deviceId, thisDeviceKey, + client->deviceId, thisSigningKey, + client->userId); static char deviceKeysSignedBuffer[KEYS_UPLOAD_REQUEST_SIGNED_SIZE]; JsonSign(client, - deviceKeysBuffer, KEYS_UPLOAD_REQUEST_SIZE, + deviceKeysBuffer, deviceKeysBufferLen, deviceKeysSignedBuffer, KEYS_UPLOAD_REQUEST_SIZE); - + + static char finalEvent[KEYS_UPLOAD_REQUEST_SIGNED_SIZE]; + snprintf(finalEvent, KEYS_UPLOAD_REQUEST_SIGNED_SIZE, + "{\"device_keys\":%s}", deviceKeysSignedBuffer); static char responseBuffer[KEYS_UPLOAD_RESPONSE_SIZE]; MatrixHttpPost(client, KEYS_UPLOAD_URL, - deviceKeysSignedBuffer, + finalEvent, responseBuffer, KEYS_UPLOAD_RESPONSE_SIZE, true); @@ -715,14 +913,14 @@ MatrixClientSendEvent( { static char requestUrl[MAX_URL_LEN]; sprintf(requestUrl, - ROOMEVENT_URL, roomId, msgType, (int)time(NULL)); + ROOM_SEND_URL, roomId, msgType, (int)time(NULL)); - static char responseBuffer[ROOMEVENT_RESPONSE_SIZE]; + static char responseBuffer[ROOM_SEND_RESPONSE_SIZE]; bool result = MatrixHttpPut(client, requestUrl, msgBody, - responseBuffer, ROOMEVENT_RESPONSE_SIZE, + responseBuffer, ROOM_SEND_RESPONSE_SIZE, true); return result; @@ -738,7 +936,7 @@ MatrixClientSendEventEncrypted( const char * msgBody) { // event json - static char requestBuffer[ROOMEVENT_REQUEST_SIZE]; + static char requestBuffer[ROOM_SEND_REQUEST_SIZE]; sprintf(requestBuffer, "{" "\"type\":\"%s\"," @@ -751,7 +949,8 @@ MatrixClientSendEventEncrypted( // get megolm session MatrixMegolmOutSession * outSession; - MatrixClientGetMegolmOutSession(client, roomId, &outSession); + if (! MatrixClientGetMegolmOutSession(client, roomId, &outSession)) + MatrixClientNewMegolmOutSession(client, roomId, &outSession); // encrypt static char encryptedBuffer[ENCRYPTED_REQUEST_SIZE]; @@ -793,15 +992,60 @@ MatrixClientSendEventEncrypted( bool MatrixClientSync( MatrixClient * client, - char * outSyncBuffer, int outSyncCap) + char * outSyncBuffer, int outSyncCap, + const char * nextBatch) { + // filter={\"event_fields\":[\"to_device\"]} + static char url[MAX_URL_LEN]; + snprintf(url, MAX_URL_LEN, + "/_matrix/client/v3/sync?timeout=%d%s", + SYNC_TIMEOUT, + strlen(nextBatch) > 0 ? "&since=" : ""); + + int index = strlen(url); + + for (size_t i = 0; i < strlen(nextBatch); i++) { + char c = nextBatch[i]; + + if (c == '~') { + url[index++] = '%'; + url[index++] = '7'; + url[index++] = 'E'; + } + else { + url[index++] = c; + } + } + url[index] = '\0'; + return MatrixHttpGet(client, - "/_matrix/client/v3/sync", + url, outSyncBuffer, outSyncCap, true); } +// https://spec.matrix.org/v1.7/client-server-api/#get_matrixclientv3roomsroomideventeventid +bool +MatrixClientGetRoomEvent( + MatrixClient * client, + const char * roomId, + const char * eventId, + char * outEvent, int outEventCap) +{ + static char url[MAX_URL_LEN]; + snprintf(url, MAX_URL_LEN, + "/_matrix/client/v3/rooms/%s/event/%s", + roomId, + eventId); + + return + MatrixHttpGet(client, + url, + outEvent, outEventCap, + true); +} + bool MatrixClientShareMegolmOutSession( MatrixClient * client, @@ -823,16 +1067,6 @@ MatrixClientShareMegolmOutSession( session->key ); - // // get olm session - // MatrixOlmSession * olmSession; - // MatrixClientGetOlmSession(client, userId, deviceId, &olmSession); - - // // encrypt - // char encryptedBuffer[KEY_SHARE_EVENT_LEN]; - // MatrixOlmSessionEncrypt(olmSession, - // eventBuffer, - // encryptedBuffer, KEY_SHARE_EVENT_LEN); - // send MatrixClientSendToDeviceEncrypted(client, userId, @@ -874,23 +1108,6 @@ MatrixClientShareMegolmOutSessionTest( return true; } -// bool -// MatrixClientSetMegolmOutSession( -// MatrixClient * client, -// const char * roomId, -// MatrixMegolmOutSession session) -// { -// if (client->numMegolmOutSessions < 10) -// { -// session.roomId = roomId; -// client->megolmOutSessions[client->numMegolmOutSessions] = session; -// client->numMegolmOutSessions++; - -// return true; -// } -// return false; -// } - bool MatrixClientGetMegolmOutSession( MatrixClient * client, @@ -906,8 +1123,27 @@ MatrixClientGetMegolmOutSession( } } - if (MatrixClientInitMegolmOutSession(client, roomId)) { - *outSession = &client->megolmOutSessions[client->numMegolmOutSessions-1]; + return false; +} + +bool +MatrixClientNewMegolmOutSession( + MatrixClient * client, + const char * roomId, + MatrixMegolmOutSession ** outSession) +{ + if (client->numMegolmOutSessions < NUM_MEGOLM_SESSIONS) + { + MatrixMegolmOutSession * result = + &client->megolmOutSessions[client->numMegolmOutSessions]; + + MatrixMegolmOutSessionInit(result, + roomId); + + *outSession = result; + + client->numMegolmOutSessions++; + return true; } @@ -915,25 +1151,137 @@ MatrixClientGetMegolmOutSession( } bool -MatrixClientInitMegolmOutSession( +MatrixClientGetMegolmInSession( MatrixClient * client, - const char * roomId) + const char * roomId, int roomIdLen, + const char * sessionId, int sessionIdLen, + MatrixMegolmInSession ** outSession) { - if (client->numMegolmOutSessions < NUM_MEGOLM_SESSIONS) + for (int i = 0; i < client->numMegolmInSessions; i++) { - MatrixMegolmOutSessionInit( - &client->megolmOutSessions[client->numMegolmOutSessions], - roomId); + if (strncmp(client->megolmInSessions[i].roomId, roomId, roomIdLen) == 0 && + strncmp(client->megolmInSessions[i].id, sessionId, sessionIdLen) == 0) + { + *outSession = &client->megolmInSessions[i]; + return true; + } + } + + return false; +} + +bool +MatrixClientNewMegolmInSession( + MatrixClient * client, + const char * roomId, + const char * sessionId, + const char * sessionKey, + MatrixMegolmInSession ** outSession) +{ + if (client->numMegolmInSessions < NUM_MEGOLM_SESSIONS) + { + MatrixMegolmInSession * result = + &client->megolmInSessions[client->numMegolmInSessions]; - client->numMegolmOutSessions++; + MatrixMegolmInSessionInit(result, + roomId, + sessionId, + sessionKey, strlen(sessionKey)); + + *outSession = result; + + client->numMegolmInSessions++; + + return true; + } + + return false; +} + +bool +MatrixClientRequestMegolmInSession( + MatrixClient * client, + const char * roomId, + const char * sessionId, + const char * senderKey, + const char * userId, + const char * deviceId) +{ + // TODO: cancel requests + MatrixClientSendDummy(client, userId, deviceId); + + static char event[ROOMKEY_REQUEST_SIZE]; + snprintf(event, ROOMKEY_REQUEST_SIZE, + "{" + "\"action\":\"request\"," + "\"body\":{" + "\"algorithm\":\"m.megolm.v1.aes-sha2\"," + "\"room_id\":\"%s\"," + "\"sender_key\":\"%s\"," + "\"session_id\":\"%s\"" + "}," + "\"request_id\":\"%d\"," + "\"requesting_device_id\":\"%s\"" + "}", + roomId, + senderKey, + sessionId, + time(NULL), + client->deviceId); + + + MatrixClientSendToDevice(client, + userId, + deviceId, + event, + "m.room_key_request"); + + return true; +} + +bool +MatrixClientGetOlmSessionIn( + MatrixClient * client, + const char * userId, + const char * deviceId, + const char * encrypted, + MatrixOlmSession ** outSession) +{ + for (int i = 0; i < client->numOlmSessions; i++) + { + if (strcmp(client->olmSessions[i].deviceId, deviceId) == 0) + { + *outSession = &client->olmSessions[i]; + return true; + } + } + + if (client->numOlmSessions < NUM_OLM_SESSIONS) + { + static char deviceKey[DEVICE_KEY_SIZE]; + MatrixClientRequestDeviceKey(client, + deviceId, + deviceKey, DEVICE_KEY_SIZE); + + MatrixOlmSessionFrom( + &client->olmSessions[client->numOlmSessions], + client->olmAccount.account, + deviceId, + deviceKey, + encrypted); + + *outSession = &client->olmSessions[client->numOlmSessions]; + + client->numOlmSessions++; return true; } + return false; } bool -MatrixClientGetOlmSession( +MatrixClientGetOlmSessionOut( MatrixClient * client, const char * userId, const char * deviceId, @@ -1004,14 +1352,16 @@ MatrixClientSendToDevice( deviceId, message); - static char responseBuffer[ROOMEVENT_RESPONSE_SIZE]; + static char responseBuffer[ROOM_SEND_RESPONSE_SIZE]; bool result = MatrixHttpPut(client, requestUrl, eventBuffer, - responseBuffer, ROOMEVENT_RESPONSE_SIZE, + responseBuffer, ROOM_SEND_RESPONSE_SIZE, true); + printf("%s\n", responseBuffer); + return result; } @@ -1025,7 +1375,7 @@ MatrixClientSendToDeviceEncrypted( { // get olm session MatrixOlmSession * olmSession; - MatrixClientGetOlmSession(client, userId, deviceId, &olmSession); + MatrixClientGetOlmSessionOut(client, userId, deviceId, &olmSession); // create event json char targetDeviceKey[DEVICE_KEY_SIZE]; @@ -1056,8 +1406,6 @@ MatrixClientSendToDeviceEncrypted( userId, // recipient user id targetSigningKey, // recipient device key thisSigningKey); - - printf("%s\n", eventBuffer); // encrypt static char encryptedBuffer[ENCRYPTED_REQUEST_SIZE]; @@ -1097,6 +1445,20 @@ MatrixClientSendToDeviceEncrypted( "m.room.encrypted"); } +bool +MatrixClientSendDummy( + MatrixClient * client, + const char * userId, + const char * deviceId) +{ + return MatrixClientSendToDeviceEncrypted( + client, + userId, + deviceId, + "{}", + "m.dummy"); +} + bool MatrixClientFindDevice( MatrixClient * client,