tildefriends/src/ssb.rpc.c

666 lines
25 KiB
C
Raw Normal View History

#include "ssb.rpc.h"
#include "ssb.h"
#include "trace.h"
#include <base64c.h>
#include <sodium/crypto_hash_sha256.h>
#include <sodium/crypto_sign.h>
#include <sqlite3.h>
#include <uv.h>
#include <string.h>
typedef struct _tf_ssb_blob_wants_t tf_ssb_blob_wants_t;
typedef struct _tf_ssb_blob_wants_t
{
tf_ssb_rpc_t* rpc;
tf_ssb_connection_t* connection;
int32_t request_number;
int64_t rowid;
tf_ssb_blob_wants_t* next;
int open_count;
} tf_ssb_blob_wants_t;
typedef struct _tf_ssb_rpc_t
{
tf_ssb_blob_wants_t* wants;
uv_async_t wants_async;
} tf_ssb_rpc_t;
const char** tf_ssb_get_following_deep(tf_ssb_t* ssb, const char** ids, int depth);
static void _tf_ssb_rpc_blob_has(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
JSContext* context = tf_ssb_connection_get_context(connection);
sqlite3* db = tf_ssb_connection_get_db(connection);
sqlite3_stmt* statement;
JSValue args_array = JS_GetPropertyStr(context, args, "args");
JSValue blob_id_value = JS_GetPropertyUint32(context, args_array, 0);
const char* blob_id = JS_ToCString(context, blob_id_value);
bool have = false;
if (sqlite3_prepare(db, "SELECT 1 FROM blobs WHERE id = $1", -1, &statement, NULL) == SQLITE_OK) {
have =
sqlite3_bind_text(statement, 1, blob_id, -1, NULL) == SQLITE_OK &&
sqlite3_step(statement) == SQLITE_ROW;
sqlite3_finalize(statement);
}
JS_FreeCString(context, blob_id);
JS_FreeValue(context, blob_id_value);
JS_FreeValue(context, args_array);
uint8_t send_flags =
(flags & k_ssb_rpc_flag_stream) |
k_ssb_rpc_flag_json |
k_ssb_rpc_flag_end_error;
const char* result = have ? "true" : "false";
tf_ssb_connection_rpc_send(connection, send_flags, -request_number, (const uint8_t*)result, strlen(result), NULL, NULL);
}
static void _tf_ssb_rpc_blob_get(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
tf_ssb_t* ssb = tf_ssb_connection_get_ssb(connection);
JSContext* context = tf_ssb_connection_get_context(connection);
JSValue blob_ids = JS_GetPropertyStr(context, args, "args");
JSValue length_value = JS_GetPropertyStr(context, blob_ids, "length");
int32_t length = 0;
JS_ToInt32(context, &length, length_value);
JS_FreeValue(context, length_value);
for (int i = 0; i < length; i++) {
JSValue blob_id_value = JS_GetPropertyUint32(context, blob_ids, i);
const char* blob_id = JS_ToCString(context, blob_id_value);
uint8_t* blob = NULL;
size_t blob_size = 0;
if (tf_ssb_blob_get(ssb, blob_id, &blob, &blob_size)) {
static const size_t k_block_size = 64 * 1024;
for (size_t offset = 0; offset < blob_size; offset += k_block_size) {
size_t block_size = offset + k_block_size < blob_size ? k_block_size : (blob_size - offset);
tf_ssb_connection_rpc_send(connection, k_ssb_rpc_flag_binary | k_ssb_rpc_flag_stream, -request_number, blob, block_size, NULL, NULL);
}
free(blob);
}
JS_FreeCString(context, blob_id);
JS_FreeValue(context, blob_id_value);
}
JS_FreeValue(context, blob_ids);
}
typedef struct _tf_ssb_connection_blobs_get_t
{
char blob_id[BLOB_ID_LEN];
size_t n;
size_t size;
crypto_hash_sha256_state hash;
uint8_t data[];
} tf_ssb_blobs_get_t;
static void _tf_ssb_connection_on_rpc_blobs_get_response(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
tf_ssb_t* ssb = tf_ssb_connection_get_ssb(connection);
tf_ssb_blobs_get_t* get = user_data;
if (get) {
if (get->n + size <= get->size) {
memcpy(get->data + get->n, message, size);
crypto_hash_sha256_update(&get->hash, message, size);
get->n += size;
}
printf("received %zd / %zd for %s\n", get->n, get->size, get->blob_id);
if (get->n == get->size) {
uint8_t hash[crypto_hash_sha256_BYTES];
crypto_hash_sha256_final(&get->hash, hash);
char hash64[256];
base64c_encode(hash, sizeof(hash), (uint8_t*)hash64, sizeof(hash64));
char id[512];
snprintf(id, sizeof(id), "&%s.sha256", hash64);
if (strcmp(id, get->blob_id) == 0) {
if (tf_ssb_blob_store(ssb, get->data, get->size, id, sizeof(id))) {
printf("stored blob %s\n", get->blob_id);
} else {
printf("failed to store %s\n", get->blob_id);
}
} else {
printf("blob does not match id %s vs. %s\n", id, get->blob_id);
}
tf_ssb_connection_remove_request(connection, -request_number);
free(get);
}
}
}
void tf_ssb_rpc_send_blobs_get(tf_ssb_connection_t* connection, const char* blob_id, size_t size)
{
JSContext* context = tf_ssb_connection_get_context(connection);
JSValue message = JS_NewObject(context);
JS_SetPropertyStr(context, message, "type", JS_NewString(context, "source"));
JSValue nameval = JS_NewArray(context);
JS_SetPropertyUint32(context, nameval, 0, JS_NewString(context, "blobs"));
JS_SetPropertyUint32(context, nameval, 1, JS_NewString(context, "get"));
JS_SetPropertyStr(context, message, "name", nameval);
JSValue argsval = JS_NewArray(context);
JS_SetPropertyUint32(context, argsval, 0, JS_NewString(context, blob_id));
JS_SetPropertyStr(context, message, "args", argsval);
JSValue str = JS_JSONStringify(context, message, JS_NULL, JS_NULL);
size_t len;
const char* cstr = JS_ToCStringLen(context, &len, str);
tf_ssb_blobs_get_t* get = malloc(sizeof(tf_ssb_blobs_get_t) + size);
*get = (tf_ssb_blobs_get_t) { .size = size };
snprintf(get->blob_id, sizeof(get->blob_id), "%s", blob_id);
crypto_hash_sha256_init(&get->hash);
tf_ssb_connection_rpc_send(connection, k_ssb_rpc_flag_json | k_ssb_rpc_flag_stream, tf_ssb_connection_next_request_number(connection), (const uint8_t*)cstr, len, _tf_ssb_connection_on_rpc_blobs_get_response, get);
JS_FreeCString(context, cstr);
JS_FreeValue(context, str);
JS_FreeValue(context, message);
}
static void _tf_ssb_connection_on_rpc_blobs_createWants_response(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
tf_ssb_t* ssb = tf_ssb_connection_get_ssb(connection);
JSContext* context = tf_ssb_connection_get_context(connection);
JSPropertyEnum* ptab;
uint32_t plen;
JS_GetOwnPropertyNames(context, &ptab, &plen, args, JS_GPN_STRING_MASK);
for (uint32_t i = 0; i < plen; ++i) {
JSPropertyDescriptor desc;
JSValue key_value = JS_NULL;
if (JS_GetOwnProperty(context, &desc, args, ptab[i].atom) == 1) {
key_value = desc.value;
}
int32_t size;
if (JS_ToInt32(context, &size, key_value) == 0) {
JSValue key = JS_AtomToString(context, ptab[i].atom);
const char* blob_id = JS_ToCString(context, key);
if (size >= 0 &&
size < k_ssb_blob_bytes_max) {
tf_ssb_rpc_send_blobs_get(connection, blob_id, size);
} else if (size < 0) {
size_t blob_size = 0;
if (tf_ssb_blob_get(ssb, blob_id, NULL, &blob_size)) {
JSValue size_response = JS_NewObject(context);
JS_SetPropertyStr(context, size_response, blob_id, JS_NewInt64(context, blob_size));
JSValue jsonval = JS_JSONStringify(context, size_response, JS_NULL, JS_NULL);
size_t len;
const char* json = JS_ToCStringLen(context, &len, jsonval);
tf_ssb_rpc_t* rpc = tf_ssb_get_rpc(ssb);
if (rpc->wants) {
tf_ssb_connection_rpc_send(connection, k_ssb_rpc_flag_json | k_ssb_rpc_flag_stream, -rpc->wants->request_number, (uint8_t*)json, len, NULL, NULL);
}
JS_FreeCString(context, json);
JS_FreeValue(context, jsonval);
JS_FreeValue(context, size_response);
}
}
JS_FreeCString(context, blob_id);
JS_FreeValue(context, key);
}
JS_FreeValue(context, key_value);
}
for (uint32_t i = 0; i < plen; ++i) {
JS_FreeAtom(context, ptab[i].atom);
}
js_free(context, ptab);
}
void tf_ssb_rpc_send_blobs_createWants(tf_ssb_connection_t* connection)
{
const char* k_createWants = "{\"name\": [\"blobs\", \"createWants\"], \"type\": \"source\", \"args\": []}";
tf_ssb_connection_rpc_send(connection, k_ssb_rpc_flag_json | k_ssb_rpc_flag_stream, tf_ssb_connection_next_request_number(connection), (const uint8_t*)k_createWants, strlen(k_createWants), _tf_ssb_connection_on_rpc_blobs_createWants_response, NULL);
}
static void _tf_ssb_rpc_remove_wants(tf_ssb_rpc_t* rpc, tf_ssb_connection_t* connection)
{
for (tf_ssb_blob_wants_t** it = &rpc->wants; *it; it = &(*it)->next) {
if ((*it)->connection == connection) {
tf_ssb_blob_wants_t* wants = *it;
*it = wants->next;
free(wants);
return;
}
}
}
static void _tf_ssb_blob_wants_update(tf_ssb_blob_wants_t* wants)
{
static const int k_messages_per_query = 16;
sqlite3* db = tf_ssb_connection_get_db(wants->connection);
sqlite3_stmt* statement;
if (wants->rowid <= 0) {
if (sqlite3_prepare(db, "SELECT MAX(messages.rowid) FROM messages", -1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_step(statement) == SQLITE_ROW) {
wants->rowid = sqlite3_column_int64(statement, 0);
}
sqlite3_finalize(statement);
}
}
if (sqlite3_prepare(db,
"SELECT DISTINCT json.value FROM messages, json_tree(messages.content) AS json "
"LEFT OUTER JOIN blobs ON json.value = blobs.id "
"WHERE "
" messages.rowid <= ?1 AND messages.rowid > ?2 AND "
" json.value LIKE '&%%.sha256' AND "
" length(json.value) = ?3 AND "
" blobs.content IS NULL",
-1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_bind_int64(statement, 1, wants->rowid) == SQLITE_OK &&
sqlite3_bind_int64(statement, 2, wants->rowid - k_messages_per_query) == SQLITE_OK &&
sqlite3_bind_int64(statement, 3, BLOB_ID_LEN - 1) == SQLITE_OK)
{
while (sqlite3_step(statement) == SQLITE_ROW) {
JSContext* context = tf_ssb_connection_get_context(wants->connection);
JSValue want = JS_NewObject(context);
JS_SetPropertyStr(context, want, (const char*)sqlite3_column_text(statement, 0), JS_NewInt32(context, -1));
JSValue jsonval = JS_JSONStringify(context, want, JS_NULL, JS_NULL);
size_t len;
const char* json = JS_ToCStringLen(context, &len, jsonval);
++wants->open_count;
tf_ssb_connection_rpc_send(wants->connection, k_ssb_rpc_flag_json | k_ssb_rpc_flag_stream, -wants->request_number, (const uint8_t*)json, len, NULL, NULL);
JS_FreeCString(context, json);
JS_FreeValue(context, jsonval);
JS_FreeValue(context, want);
}
wants->rowid -= k_messages_per_query;
} else {
printf("bind failed: %s\n", sqlite3_errmsg(db));
}
sqlite3_finalize(statement);
} else {
printf("prepare failed: %s\n", sqlite3_errmsg(db));
}
if (wants->rowid <= 0) {
tf_ssb_connection_rpc_send(wants->connection, k_ssb_rpc_flag_json | k_ssb_rpc_flag_stream, -wants->request_number, (const uint8_t*)"{}", 2, NULL, NULL);
} else {
uv_async_send(&wants->rpc->wants_async);
}
}
static void _tf_ssb_rpc_blobs_createWants(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
tf_ssb_rpc_t* rpc = user_data;
tf_ssb_blob_wants_t* wants = malloc(sizeof(tf_ssb_blob_wants_t));
*wants = (tf_ssb_blob_wants_t) {
.rpc = rpc,
.connection = connection,
.request_number = request_number,
.rowid = -1,
.next = rpc->wants,
};
rpc->wants = wants;
_tf_ssb_blob_wants_update(wants);
}
static void _tf_ssb_rpc_createHistoryStream(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue args, const uint8_t* message, size_t size, void* user_data)
{
JSContext* context = tf_ssb_connection_get_context(connection);
sqlite3* db = tf_ssb_connection_get_db(connection);
JSValue streamArgs = JS_GetPropertyStr(context, args, "args");
JSValue obj = JS_GetPropertyUint32(context, streamArgs, 0);
JSValue idval = JS_GetPropertyStr(context, obj, "id");
const char* author = JS_ToCString(context, idval);
int64_t seq = 0;
sqlite3_stmt* statement;
JSValue seqval = JS_GetPropertyStr(context, obj, "seq");
JS_ToInt64(context, &seq, seqval);
JS_FreeValue(context, seqval);
const char* query = "SELECT previous, sequence, timestamp, hash, content, signature FROM messages WHERE author = $1 AND sequence >= $2 ORDER BY sequence";
if (sqlite3_prepare(db, query, -1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_bind_text(statement, 1, author, -1, NULL) == SQLITE_OK &&
sqlite3_bind_int64(statement, 2, seq) == SQLITE_OK) {
while (sqlite3_step(statement) == SQLITE_ROW) {
JSValue message = JS_NewObject(context);
const char* previous = (const char*)sqlite3_column_text(statement, 0);
JS_SetPropertyStr(context, message, "previous", previous ? JS_NewString(context, previous) : JS_NULL);
JS_SetPropertyStr(context, message, "author", JS_NewString(context, author));
JS_SetPropertyStr(context, message, "sequence", JS_NewInt64(context, sqlite3_column_int64(statement, 1)));
JS_SetPropertyStr(context, message, "timestamp", JS_NewInt64(context, sqlite3_column_int64(statement, 2)));
JS_SetPropertyStr(context, message, "hash", JS_NewString(context, (const char*)sqlite3_column_text(statement, 3)));
const char* contentstr = (const char*)sqlite3_column_text(statement, 4);
JSValue content = JS_ParseJSON(context, contentstr, strlen(contentstr), NULL);
JS_SetPropertyStr(context, message, "content", content);
JS_SetPropertyStr(context, message, "signature", JS_NewString(context, (const char*)sqlite3_column_text(statement, 5)));
JSValue jsonval = JS_JSONStringify(context, message, JS_NULL, JS_NULL);
size_t len;
const char* json = JS_ToCStringLen(context, &len, jsonval);
if (tf_ssb_verify_and_strip_signature(context, message, NULL, 0)) {
tf_ssb_connection_rpc_send(connection, flags, -request_number, (const uint8_t*)json, len, NULL, NULL);
} else {
printf("message signature is invalid\n");
}
JS_FreeCString(context, json);
JS_FreeValue(context, jsonval);
JS_FreeValue(context, message);
}
}
sqlite3_finalize(statement);
} else {
printf("prepare failed: %s\n", sqlite3_errmsg(db));
}
tf_ssb_connection_rpc_send(connection, flags | k_ssb_rpc_flag_end_error, -request_number, (const uint8_t*)"true", 4, NULL, NULL);
JS_FreeValue(context, obj);
JS_FreeCString(context, author);
JS_FreeValue(context, idval);
JS_FreeValue(context, streamArgs);
}
static void _tf_ssb_connection_on_rpc_createHistoryStream_response(tf_ssb_connection_t* connection, uint8_t flags, int32_t request_number, JSValue val, const uint8_t* message, size_t size, void* user_data)
{
tf_ssb_rpc_t* rpc = user_data;
JSContext* context = tf_ssb_connection_get_context(connection);
char signature[crypto_sign_BYTES + 128];
char id[crypto_hash_sha256_BYTES * 2 + 1];
tf_ssb_calculate_message_id(context, val, id, sizeof(id));
if (tf_ssb_verify_and_strip_signature(context, val, signature, sizeof(signature))) {
tf_ssb_store_message(tf_ssb_connection_get_ssb(connection), context, id, val, signature);
} else {
printf("failed to verify message\n");
}
uv_async_send(&rpc->wants_async);
}
void tf_ssb_rpc_send_createHistoryStream(tf_ssb_connection_t* connection, const char* id)
{
JSContext* context = tf_ssb_connection_get_context(connection);
JSValue message = JS_NewObject(context);
JS_SetPropertyStr(context, message, "type", JS_NewString(context, "source"));
JSValue nameval = JS_NewArray(context);
JS_SetPropertyUint32(context, nameval, 0, JS_NewString(context, "createHistoryStream"));
JS_SetPropertyStr(context, message, "name", nameval);
JSValue obj = JS_NewObject(context);
JS_SetPropertyStr(context, obj, "id", JS_NewString(context, id));
JS_SetPropertyStr(context, obj, "keys", JS_FALSE);
int64_t sequence = 0;
tf_ssb_t* ssb = tf_ssb_connection_get_ssb(connection);
if (tf_ssb_get_latest_message_by_author(ssb, id, &sequence, NULL, 0)) {
JS_SetPropertyStr(context, obj, "seq", JS_NewInt64(context, sequence));
}
JSValue argsval = JS_NewArray(context);
JS_SetPropertyUint32(context, argsval, 0, obj);
JS_SetPropertyStr(context, message, "args", argsval);
JSValue str = JS_JSONStringify(context, message, JS_NULL, JS_NULL);
size_t len;
const char* cstr = JS_ToCStringLen(context, &len, str);
tf_ssb_rpc_t* rpc = tf_ssb_get_rpc(ssb);
tf_ssb_connection_rpc_send(connection, 0xa, tf_ssb_connection_next_request_number(connection), (const uint8_t*)cstr, len, _tf_ssb_connection_on_rpc_createHistoryStream_response, rpc);
JS_FreeCString(context, cstr);
JS_FreeValue(context, str);
JS_FreeValue(context, message);
}
static void _tf_ssb_rpc_on_connections_changed(tf_ssb_t* ssb, tf_ssb_change_t change, tf_ssb_connection_t* connection, void* user_data)
{
if (change != k_tf_ssb_change_connect) {
return;
}
char id[k_id_base64_len];
if (tf_ssb_connection_get_id(connection, id, sizeof(id))) {
const char** ids = tf_ssb_get_following_deep(ssb, (const char*[]) { id, NULL }, 2);
for (int i = 0; ids && ids[i]; i++) {
tf_ssb_rpc_send_createHistoryStream(connection, ids[i]);
}
free(ids);
}
}
static void _tf_ssb_add_id(const char*** results, int* results_count, int* results_capacity, const char* id)
{
for (int i = 0; i < *results_count; i++) {
if (strcmp((*results)[i], id) == 0) {
return;
}
}
if (*results_count + 1 > *results_capacity) {
int old_capacity = *results_capacity;
*results_capacity = (*results_capacity + 1) * 2;
*results = realloc(*results, sizeof(const char*) * (*results_capacity));
memset(*results + *results_count, 0, sizeof(const char*) * (*results_capacity - old_capacity));
}
(*results)[(*results_count)++] = strdup(id);
}
const char** tf_ssb_get_following(tf_ssb_t* ssb, const char* id)
{
sqlite3* db = tf_ssb_get_db(ssb);
JSContext* context = tf_ssb_get_context(ssb);
const int k_version = 0;
int64_t rowid = 0;
const char** results = NULL;
int results_count = 0;
int results_capacity = 0;
sqlite3_stmt* statement;
if (sqlite3_prepare(db, "SELECT value FROM properties WHERE id = ? AND key = (? || ':following')", -1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_bind_text(statement, 1, "core", -1, NULL) == SQLITE_OK &&
sqlite3_bind_text(statement, 2, id, -1, NULL) == SQLITE_OK &&
sqlite3_step(statement) == SQLITE_ROW) {
JSValue cache = JS_ParseJSON(context, (const char*)sqlite3_column_text(statement, 0), sqlite3_column_bytes(statement, 0), NULL);
JSValue version_value = JS_GetPropertyStr(context, cache, "version");
int32_t version = 0;
JS_ToInt32(context, &version, version_value);
JS_FreeValue(context, version_value);
if (version == k_version) {
JSValue rowid_value = JS_GetPropertyStr(context, cache, "rowid");
JS_ToInt64(context, &rowid, rowid_value);
JS_FreeValue(context, rowid_value);
JSValue ids = JS_GetPropertyStr(context, cache, "following");
JSValue length_value = JS_GetPropertyStr(context, ids, "length");
int32_t length = 0;
JS_ToInt32(context, &length, length_value);
for (int i = 0; i < length; i++) {
JSValue id = JS_GetPropertyUint32(context, ids, i);
const char* id_string = JS_ToCString(context, id);
_tf_ssb_add_id(&results, &results_count, &results_capacity, id_string);
JS_FreeCString(context, id_string);
JS_FreeValue(context, id);
}
JS_FreeValue(context, length_value);
JS_FreeValue(context, ids);
}
JS_FreeValue(context, cache);
}
sqlite3_finalize(statement);
}
int64_t loaded_rowid = rowid;
JSValue cache = JS_UNDEFINED;
JS_FreeValue(context, cache);
if (sqlite3_prepare(db, "SELECT "
"json_extract(content, '$.contact'), "
"json_extract(content, '$.following'), "
"rowid "
"FROM messages "
"WHERE author = ? AND "
"rowid > ? AND "
"json_extract(content, '$.type') = 'contact' UNION "
"SELECT NULL, NULL, MAX(rowid) FROM messages "
"ORDER BY rowid", -1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_bind_text(statement, 1, id, -1, NULL) == SQLITE_OK &&
sqlite3_bind_int64(statement, 2, rowid) == SQLITE_OK) {
while (sqlite3_step(statement) == SQLITE_ROW) {
const char* contact = (const char*)sqlite3_column_text(statement, 0);
if (sqlite3_column_type(statement, 0) != SQLITE_NULL) {
if (sqlite3_column_int(statement, 1)) {
_tf_ssb_add_id(&results, &results_count, &results_capacity, contact);
} else {
for (int i = 0; i < results_count; i++) {
if (strcmp(results[i], contact) == 0) {
free((void*)results[i]);
results[i] = results[--results_count];
}
}
}
}
rowid = sqlite3_column_int64(statement, 2);
}
}
sqlite3_finalize(statement);
}
if (rowid != loaded_rowid) {
JSValue cache = JS_NewObject(context);
JS_SetPropertyStr(context, cache, "version", JS_NewInt32(context, k_version));
JS_SetPropertyStr(context, cache, "rowid", JS_NewInt64(context, rowid));
JSValue ids = JS_NewArray(context);
for (int i = 0; i < results_count; i++) {
JS_SetPropertyUint32(context, ids, i, JS_NewString(context, results[i]));
}
JS_SetPropertyStr(context, cache, "following", ids);
JSValue json_value = JS_JSONStringify(context, cache, JS_NULL, JS_NULL);
const char* json = JS_ToCString(context, json_value);
JS_FreeValue(context, json_value);
JS_FreeValue(context, cache);
if (sqlite3_prepare(db, "INSERT OR REPLACE INTO properties (id, key, value) VALUES (?, ? || ':following', ?)", -1, &statement, NULL) == SQLITE_OK) {
if (sqlite3_bind_text(statement, 1, "core", -1, NULL) == SQLITE_OK &&
sqlite3_bind_text(statement, 2, id, -1, NULL) == SQLITE_OK &&
sqlite3_bind_text(statement, 3, json, -1, NULL) == SQLITE_OK) {
sqlite3_step(statement);
}
sqlite3_finalize(statement);
}
JS_FreeCString(context, json);
}
size_t size = (results_count + 1) * sizeof(const char*);
for (int i = 0; i < results_count; i++) {
size += strlen(results[i]) + 1;
}
char** result = malloc(size);
char* p = (char*)result + (results_count + 1) * sizeof(const char*);
for (int i = 0; i < results_count; i++) {
result[i] = p;
size_t length = strlen(results[i]) + 1;
memcpy(p, results[i], length);
free((void*)results[i]);
p += length;
}
result[results_count] = NULL;
free((void*)results);
return (const char**)result;
}
const char** tf_ssb_get_following_deep(tf_ssb_t* ssb, const char** ids, int depth)
{
const char** results = NULL;
int results_count = 0;
int results_capacity = 0;
for (int i = 0; ids[i]; i++) {
_tf_ssb_add_id(&results, &results_count, &results_capacity, ids[i]);
}
for (int i = 0; ids[i]; i++) {
const char** following = tf_ssb_get_following(ssb, ids[i]);
for (int j = 0; following[j]; j++) {
_tf_ssb_add_id(&results, &results_count, &results_capacity, following[j]);
}
free(following);
}
size_t size = (results_count + 1) * sizeof(const char*);
for (int i = 0; i < results_count; i++) {
size += strlen(results[i]) + 1;
}
char** result = malloc(size);
char* p = (char*)result + (results_count + 1) * sizeof(const char*);
for (int i = 0; i < results_count; i++) {
result[i] = p;
size_t length = strlen(results[i]) + 1;
memcpy(p, results[i], length);
free((void*)results[i]);
p += length;
}
result[results_count] = NULL;
free((void*)results);
if (depth > 1) {
const char** r = tf_ssb_get_following_deep(ssb, (const char**)result, depth - 1);
free(result);
result = (char**)r;
}
return (const char**)result;
}
static void _tf_ssb_rpc_connections_changed_callback(tf_ssb_t* ssb, tf_ssb_change_t change, tf_ssb_connection_t* connection, void* user_data)
{
tf_ssb_rpc_t* rpc = user_data;
if (change == k_tf_ssb_change_connect) {
tf_ssb_rpc_send_blobs_createWants(connection);
} else if (change == k_tf_ssb_change_remove) {
_tf_ssb_rpc_remove_wants(rpc, connection);
}
}
static void _tf_ssb_rpc_wants_async(uv_async_t* async)
{
tf_ssb_rpc_t* rpc = async->data;
tf_ssb_blob_wants_t* it = rpc->wants;
if (it)
{
rpc->wants = it->next;
it->next = NULL;
if (rpc->wants) {
for (tf_ssb_blob_wants_t* tail = rpc->wants; tail; tail = tail->next) {
if (!tail->next) {
tail->next = it;
break;
}
}
} else {
rpc->wants = it;
}
_tf_ssb_blob_wants_update(it);
}
}
tf_ssb_rpc_t* tf_ssb_rpc_create(tf_ssb_t* ssb)
{
tf_ssb_rpc_t* rpc = malloc(sizeof(tf_ssb_rpc_t));
*rpc = (tf_ssb_rpc_t) {
.wants_async.data = rpc,
};
tf_ssb_add_connections_changed_callback(ssb, _tf_ssb_rpc_connections_changed_callback, rpc);
tf_ssb_add_connections_changed_callback(ssb, _tf_ssb_rpc_on_connections_changed, NULL);
tf_ssb_register_rpc(ssb, (const char*[]) { "blobs", "has", NULL }, _tf_ssb_rpc_blob_has, NULL);
tf_ssb_register_rpc(ssb, (const char*[]) { "blobs", "get", NULL }, _tf_ssb_rpc_blob_get, NULL);
tf_ssb_register_rpc(ssb, (const char*[]) { "blobs", "createWants", NULL }, _tf_ssb_rpc_blobs_createWants, rpc);
tf_ssb_register_rpc(ssb, (const char*[]) { "createHistoryStream", NULL }, _tf_ssb_rpc_createHistoryStream, NULL);
uv_async_init(tf_ssb_get_loop(ssb), &rpc->wants_async, _tf_ssb_rpc_wants_async);
uv_unref((uv_handle_t*)&rpc->wants_async);
return rpc;
}
static void _tf_ssb_rpc_handle_closed(uv_handle_t* handle)
{
free(handle->data);
}
void tf_ssb_rpc_destroy(tf_ssb_rpc_t* rpc)
{
uv_close((uv_handle_t*)&rpc->wants_async, _tf_ssb_rpc_handle_closed);
}