aljaz/lightning
1
0
Fork 0
mirror of https://github.com/aljazceru/lightning.git synced 2025-12-19 15:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

commando: Save blacklist runes to datastore

Browse Source
This commit is contained in:
ShahanaFarooqui
2023-03-31 11:54:37 +10:30
committed by Rusty Russell
parent a4ed3ae72e
commit 7ad04a9949
1 changed files with 49 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
plugins/commando.c
View File

@@ -1,5 +1,6 @@
#include "config.h"
#include <ccan/array_size/array_size.h>
#include <ccan/cast/cast.h>
#include <ccan/crypto/siphash24/siphash24.h>
#include <ccan/htable/htable_type.h>
#include <ccan/json_escape/json_escape.h>
@@ -963,6 +964,30 @@ static struct command_result *save_rune(struct command *cmd,
forward_error, rune);
}
static void towire_blacklist(u8 **pptr, const struct blacklist *b)
{
for (size_t i = 0; i < tal_count(b); i++) {
towire_u64(pptr, b[i].start);
towire_u64(pptr, b[i].end);
}
}
static struct blacklist *fromwire_blacklist(const tal_t *ctx,
const u8 **cursor,
size_t *max)
{
struct blacklist *blist = tal_arr(ctx, struct blacklist, 0);
while (*max > 0) {
struct blacklist b;
b.start = fromwire_u64(cursor, max);
b.end = fromwire_u64(cursor, max);
tal_arr_expand(&blist, b);
}
if (!*cursor) {
return tal_free(blist);
}
return blist;
}
static struct command_result *json_commando_rune(struct command *cmd,
const char *buffer,
@@ -1092,11 +1117,20 @@ static struct command_result *list_blacklist(struct command *cmd)
return command_finished(cmd, js);
}
static struct command_result *blacklist_save_done(struct command *cmd,
const char *buf,
const jsmntok_t *result,
void *unused)
{
return list_blacklist(cmd);
}
static struct command_result *json_commando_blacklist(struct command *cmd,
const char *buffer,
const jsmntok_t *params)
{
u64 *start, *end;
u8 *bwire;
struct blacklist *entry, *newblacklist;
if (!param(cmd, buffer, params,
@@ -1145,7 +1179,9 @@ static struct command_result *json_commando_blacklist(struct command *cmd,
}
tal_free(blacklist);
blacklist = newblacklist;
return list_blacklist(cmd);
bwire = tal_arr(tmpctx, u8, 0);
towire_blacklist(&bwire, blacklist);
return jsonrpc_set_datastore_binary(cmd->plugin, cmd, "commando/blacklist", bwire, "create-or-replace", blacklist_save_done, NULL, NULL);
}
static struct command_result *json_commando_listrunes(struct command *cmd,
@@ -1186,7 +1222,19 @@ static const char *init(struct plugin *p,
{
struct secret rune_secret;
const char *err;
u8 *bwire;
if (rpc_scan_datastore_hex(tmpctx, p, "commando/blacklist",
JSON_SCAN_TAL(tmpctx, json_tok_bin_from_hex,
&bwire)) == NULL) {
size_t max = tal_bytelen(bwire);
blacklist = fromwire_blacklist(p, cast_const2(const u8 **,
&bwire),
&max);
if (blacklist == NULL) {
plugin_err(p, "Invalid commando/blacklist");
}
}
outgoing_commands = tal_arr(p, struct commando *, 0);
incoming_commands = tal_arr(p, struct commando *, 0);
usage_table = tal(p, struct usage_table);