#include #include #include "omemo/store.h" GHashTable * session_store_new(void) { return g_hash_table_new_full(g_str_hash, g_str_equal, free, NULL); } GHashTable * pre_key_store_new(void) { return g_hash_table_new_full(g_direct_hash, g_direct_equal, NULL, (GDestroyNotify)signal_buffer_free); } GHashTable * signed_pre_key_store_new(void) { return g_hash_table_new_full(g_direct_hash, g_direct_equal, NULL, (GDestroyNotify)signal_buffer_free); } void identity_key_store_new(identity_key_store_t *identity_key_store) { identity_key_store->identity_key_store = g_hash_table_new_full(g_str_hash, g_str_equal, free, (GDestroyNotify)signal_buffer_free); identity_key_store->private = NULL; identity_key_store->public = NULL; } int load_session(signal_buffer **record, const signal_protocol_address *address, void *user_data) { GHashTable *session_store = (GHashTable *)user_data; GHashTable *device_store = NULL; device_store = g_hash_table_lookup(session_store, address->name); if (!device_store) { *record = NULL; return 0; } signal_buffer *original = g_hash_table_lookup(device_store, GINT_TO_POINTER(address->device_id)); if (!original) { *record = NULL; return 0; } *record = signal_buffer_copy(original); return 1; } int get_sub_device_sessions(signal_int_list **sessions, const char *name, size_t name_len, void *user_data) { GHashTable *session_store = (GHashTable *)user_data; GHashTable *device_store = NULL; GHashTableIter iter; gpointer key, value; device_store = g_hash_table_lookup(session_store, name); if (!device_store) { return SG_SUCCESS; } *sessions = signal_int_list_alloc(); g_hash_table_iter_init(&iter, device_store); while (g_hash_table_iter_next(&iter, &key, &value)) { signal_int_list_push_back(*sessions, GPOINTER_TO_INT(key)); } return SG_SUCCESS; } int store_session(const signal_protocol_address *address, uint8_t *record, size_t record_len, void *user_data) { GHashTable *session_store = (GHashTable *)user_data; GHashTable *device_store = NULL; device_store = g_hash_table_lookup(session_store, (void *)address->name); if (!device_store) { device_store = g_hash_table_new_full(g_direct_hash, g_direct_equal, NULL, (GDestroyNotify)signal_buffer_free); g_hash_table_insert(session_store, strdup(address->name), device_store); } signal_buffer *buffer = signal_buffer_create(record, record_len); g_hash_table_insert(device_store, GINT_TO_POINTER(address->device_id), buffer); return SG_SUCCESS; } int contains_session(const signal_protocol_address *address, void *user_data) { signal_buffer *record; load_session(&record, address, user_data); return record != NULL; } int delete_session(const signal_protocol_address *address, void *user_data) { GHashTable *session_store = (GHashTable *)user_data; GHashTable *device_store = NULL; device_store = g_hash_table_lookup(session_store, address->name); if (!device_store) { return SG_SUCCESS; } return g_hash_table_remove(device_store, GINT_TO_POINTER(address->device_id)); } int delete_all_sessions(const char *name, size_t name_len, void *user_data) { GHashTable *session_store = (GHashTable *)user_data; GHashTable *device_store = NULL; device_store = g_hash_table_lookup(session_store, name); if (!device_store) { return SG_SUCCESS; } guint len = g_hash_table_size(device_store); g_hash_table_remove_all(device_store); return len; } int load_pre_key(signal_buffer **record, uint32_t pre_key_id, void *user_data) { GHashTable *pre_key_store = (GHashTable *)user_data; *record = g_hash_table_lookup(pre_key_store, GINT_TO_POINTER(pre_key_id)); return SG_SUCCESS; } int store_pre_key(uint32_t pre_key_id, uint8_t *record, size_t record_len, void *user_data) { GHashTable *pre_key_store = (GHashTable *)user_data; signal_buffer *buffer = signal_buffer_create(record, record_len); g_hash_table_insert(pre_key_store, GINT_TO_POINTER(pre_key_id), buffer); return SG_SUCCESS; } int contains_pre_key(uint32_t pre_key_id, void *user_data) { signal_buffer *record; load_pre_key(&record, pre_key_id, user_data); return record != NULL; } int remove_pre_key(uint32_t pre_key_id, void *user_data) { GHashTable *pre_key_store = (GHashTable *)user_data; return g_hash_table_remove(pre_key_store, GINT_TO_POINTER(pre_key_id)); } int load_signed_pre_key(signal_buffer **record, uint32_t signed_pre_key_id, void *user_data) { GHashTable *signed_pre_key_store = (GHashTable *)user_data; *record = g_hash_table_lookup(signed_pre_key_store, GINT_TO_POINTER(signed_pre_key_id)); return SG_SUCCESS; } int store_signed_pre_key(uint32_t signed_pre_key_id, uint8_t *record, size_t record_len, void *user_data) { GHashTable *signed_pre_key_store = (GHashTable *)user_data; signal_buffer *buffer = signal_buffer_create(record, record_len); g_hash_table_insert(signed_pre_key_store, GINT_TO_POINTER(signed_pre_key_id), buffer); return SG_SUCCESS; } int contains_signed_pre_key(uint32_t signed_pre_key_id, void *user_data) { signal_buffer *record; load_signed_pre_key(&record, signed_pre_key_id, user_data); return record != NULL; } int remove_signed_pre_key(uint32_t signed_pre_key_id, void *user_data) { GHashTable *signed_pre_key_store = (GHashTable *)user_data; return g_hash_table_remove(signed_pre_key_store, GINT_TO_POINTER(signed_pre_key_id)); } int get_identity_key_pair(signal_buffer **public_data, signal_buffer **private_data, void *user_data) { identity_key_store_t *identity_key_store = (identity_key_store_t *)user_data; *public_data = signal_buffer_copy(identity_key_store->public); *private_data = signal_buffer_copy(identity_key_store->private); return SG_SUCCESS; } int get_local_registration_id(void *user_data, uint32_t *registration_id) { identity_key_store_t *identity_key_store = (identity_key_store_t *)user_data; *registration_id = identity_key_store->registration_id; return SG_SUCCESS; } int save_identity(const signal_protocol_address *address, uint8_t *key_data, size_t key_len, void *user_data) { identity_key_store_t *identity_key_store = (identity_key_store_t *)user_data; char *node = g_strdup_printf("%s:%d", address->name, address->device_id); signal_buffer *buffer = signal_buffer_create(key_data, key_len); g_hash_table_insert(identity_key_store->identity_key_store, node, buffer); return SG_SUCCESS; } int is_trusted_identity(const signal_protocol_address *address, uint8_t *key_data, size_t key_len, void *user_data) { identity_key_store_t *identity_key_store = (identity_key_store_t *)user_data; char *node = g_strdup_printf("%s:%d", address->name, address->device_id); signal_buffer *buffer = signal_buffer_create(key_data, key_len); signal_buffer *original = g_hash_table_lookup(identity_key_store->identity_key_store, node); return original == NULL || signal_buffer_compare(buffer, original); } int store_sender_key(const signal_protocol_sender_key_name *sender_key_name, uint8_t *record, size_t record_len, uint8_t *user_record, size_t user_record_len, void *user_data) { return SG_SUCCESS; } int load_sender_key(signal_buffer **record, signal_buffer **user_record, const signal_protocol_sender_key_name *sender_key_name, void *user_data) { return SG_SUCCESS; }