Skip to content

Commit

Permalink
lightningd/plugin.c: Make plugin-exclusive loop support multiple plug…
Browse files Browse the repository at this point in the history
…ins.
  • Loading branch information
ZmnSCPxj committed Nov 23, 2020
1 parent 66b2bb9 commit ee0c13e
Show file tree
Hide file tree
Showing 3 changed files with 32 additions and 13 deletions.
20 changes: 14 additions & 6 deletions lightningd/plugin.c
Original file line number Diff line number Diff line change
Expand Up @@ -1715,20 +1715,28 @@ void plugin_request_send(struct plugin *plugin,
req->stream = NULL;
}

void *plugin_exclusive_loop(struct plugin *plugin)
void *plugins_exclusive_loop(struct plugin **plugins)
{
void *ret;
size_t i;
bool last = false;
assert(tal_count(plugins) != 0);

io_conn_out_exclusive(plugin->stdin_conn, true);
io_conn_exclusive(plugin->stdout_conn, true);
for (i = 0; i < tal_count(plugins); ++i) {
io_conn_out_exclusive(plugins[i]->stdin_conn, true);
io_conn_exclusive(plugins[i]->stdout_conn, true);
}

/* We don't service timers here, either! */
ret = io_loop(NULL, NULL);

io_conn_out_exclusive(plugin->stdin_conn, false);
if (io_conn_exclusive(plugin->stdout_conn, false))
for (i = 0; i < tal_count(plugins); ++i) {
io_conn_out_exclusive(plugins[i]->stdin_conn, false);
last = io_conn_exclusive(plugins[i]->stdout_conn, false);
}
if (last)
fatal("Still io_exclusive after removing plugin %s?",
plugin->cmd);
plugins[tal_count(plugins) - 1]->cmd);

return ret;
}
Expand Down
10 changes: 7 additions & 3 deletions lightningd/plugin.h
Original file line number Diff line number Diff line change
Expand Up @@ -304,12 +304,16 @@ void json_add_opt_disable_plugins(struct json_stream *response,
const struct plugins *plugins);

/**
* Used by db hooks which can't have any other I/O while talking to plugin.
* Used by db hooks which can't have any other I/O while talking to
* hooked plugins.
*
* Returns output of io_loop() (ie. whatever gets passed to io_break()
* @param plugins - a `tal`-allocated array of plugins that are the
* only ones we talk to.
*
* @return output of io_loop() (ie. whatever gets passed to io_break()
* to end exclusive loop).
*/
void *plugin_exclusive_loop(struct plugin *plugin);
void *plugins_exclusive_loop(struct plugin **plugins);

/**
* Add a directory to the plugin path to automatically load plugins.
Expand Down
15 changes: 11 additions & 4 deletions lightningd/plugin_hook.c
Original file line number Diff line number Diff line change
Expand Up @@ -330,12 +330,18 @@ void plugin_hook_db_sync(struct db *db)
struct jsonrpc_request *req;
struct plugin_hook_request *ph_req;
void *ret;
struct plugin *plugin;
struct plugin **plugins;
size_t i;

const char **changes = db_changes(db);
if (tal_count(hook->hooks) == 0)
return;

plugins = notleak(tal_arr(NULL, struct plugin *,
tal_count(hook->hooks)));
for (i = 0; i < tal_count(hook->hooks); ++i)
plugins[i] = hook->hooks[i]->plugin;

ph_req = notleak(tal(hook->hooks, struct plugin_hook_request));
/* FIXME: do IO logging for this! */
req = jsonrpc_request_start(NULL, hook->name, NULL, NULL,
Expand All @@ -344,7 +350,7 @@ void plugin_hook_db_sync(struct db *db)

ph_req->hook = hook;
ph_req->db = db;
plugin = ph_req->plugin = hook->hooks[0]->plugin;
ph_req->plugin = hook->hooks[0]->plugin;

json_add_num(req->stream, "data_version", db_data_version_get(db));

Expand All @@ -359,12 +365,13 @@ void plugin_hook_db_sync(struct db *db)
/* We can be called on way out of an io_loop, which is already breaking.
* That will make this immediately return; save the break value and call
* again, then hand it onwards. */
ret = plugin_exclusive_loop(plugin);
ret = plugins_exclusive_loop(plugins);
if (ret != ph_req) {
void *ret2 = plugin_exclusive_loop(plugin);
void *ret2 = plugins_exclusive_loop(plugins);
assert(ret2 == ph_req);
io_break(ret);
}
tal_free(plugins);
}

static void add_deps(const char ***arr,
Expand Down

0 comments on commit ee0c13e

Please sign in to comment.