/*

* Helper to submit via JSON-RPC and get back response.

*/

#include "json.h"

#include "pettycoin_dir.h"

#include <ccan/err/err.h>

#include <ccan/opt/opt.h>

#include <ccan/read_write_all/read_write_all.h>

#include <ccan/str/str.h>

#include <ccan/tal/str/str.h>

#include <stdio.h>

#include <sys/socket.h>

#include <sys/types.h>

#include <sys/un.h>

#include <unistd.h>

#define NO_ERROR 0

#define ERROR_FROM_PETTYCOIN 1

#define ERROR_TALKING_TO_PETTYCOIN 2

#define ERROR_USAGE 3

/* Tal wrappers for opt. */

static void *opt_allocfn(size_t size)

{

return tal_alloc_(NULL, size, false, TAL_LABEL("opt_allocfn", ""));

}

static void *tal_reallocfn(void *ptr, size_t size)

{

if (!ptr)

return opt_allocfn(size);

tal_resize_(&ptr, 1, size, false);

return ptr;

}

static void tal_freefn(void *ptr)

{

tal_free(ptr);

}

/* Simple test code to create a gateway transaction */

int main(int argc, char *argv[])

{

int fd, i, off;

const char *method;

char *cmd, *resp, *idstr, *rpc_filename;

char *result_end;

struct sockaddr_un addr;

jsmntok_t *toks;

const jsmntok_t *result, *error, *id;

char *pettycoin_dir;

const tal_t *ctx = tal(NULL, char);

size_t num_opens, num_closes;

bool valid;

err_set_progname(argv[0]);

opt_set_alloc(opt_allocfn, tal_reallocfn, tal_freefn);

pettycoin_dir_register_opts(ctx, &pettycoin_dir, &rpc_filename);

opt_register_noarg("--help|-h", opt_usage_and_exit,

"<command> [<params>...]", "Show this message");

opt_register_noarg("--version|-V", opt_version_and_exit, VERSION,

"Display version and exit");

opt_early_parse(argc, argv, opt_log_stderr_exit);

opt_parse(&argc, argv, opt_log_stderr_exit);

method = argv[1];

if (!method)

errx(ERROR_USAGE, "Need at least one argument\n%s",

opt_usage(argv[0], NULL));

if (chdir(pettycoin_dir) != 0)

err(ERROR_TALKING_TO_PETTYCOIN, "Moving into '%s'",

pettycoin_dir);

fd = socket(AF_UNIX, SOCK_STREAM, 0);

if (strlen(rpc_filename) + 1 > sizeof(addr.sun_path))

errx(ERROR_USAGE, "rpc filename '%s' too long", rpc_filename);

strcpy(addr.sun_path, rpc_filename);

addr.sun_family = AF_UNIX;

if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) != 0)

err(ERROR_TALKING_TO_PETTYCOIN,

"Connecting to '%s'", rpc_filename);

idstr = tal_fmt(ctx, "pettycoin_query-%i", getpid());

cmd = tal_fmt(ctx,

"{ \"method\" : \"%s\", \"id\" : \"%s\", \"params\" : [ ",

method, idstr);

for (i = 2; i < argc; i++) {

/* Numbers are left unquoted, and quoted things left alone. */

if (strspn(argv[i], "0123456789") == strlen(argv[i])

|| argv[i][0] == '"')

tal_append_fmt(&cmd, "%s", argv[i]);

else

tal_append_fmt(&cmd, "\"%s\"", argv[i]);

if (i != argc - 1)

tal_append_fmt(&cmd, ", ");

}

tal_append_fmt(&cmd, "] }");

if (!write_all(fd, cmd, strlen(cmd)))

err(ERROR_TALKING_TO_PETTYCOIN, "Writing command");

resp = tal_arr(cmd, char, 100);

off = 0;

num_opens = num_closes = 0;

while ((i = read(fd, resp + off, tal_count(resp) - 1 - off)) > 0) {

resp[off + i] = '\0';

num_opens += strcount(resp + off, "{");

num_closes += strcount(resp + off, "}");

off += i;

if (off == tal_count(resp) - 1)

tal_resize(&resp, tal_count(resp) * 2);

/* parsing huge outputs is slow: do quick check first. */

if (num_opens == num_closes && strstr(resp, "\"result\""))

break;

}

if (i < 0)

err(ERROR_TALKING_TO_PETTYCOIN, "reading response");

/* Parsing huge results is too slow, so hack fastpath common case */

result_end = tal_fmt(ctx, ", \"error\" : null, \"id\" : \"%s\" }\n",

idstr);

if (strstarts(resp, "{ \"result\" : ") && strends(resp, result_end)) {

/* Result is OK, so dump it */

resp += strlen("{ \"result\" : ");

printf("%.*s\n", (int)(strlen(resp) - strlen(result_end)), resp);

tal_free(ctx);

return 0;

}

toks = json_parse_input(resp, off, &valid);

if (!toks || !valid)

errx(ERROR_TALKING_TO_PETTYCOIN,

"Malformed response '%s'", resp);

result = json_get_member(resp, toks, "result");

if (!result)

errx(ERROR_TALKING_TO_PETTYCOIN,

"Missing 'result' in response '%s'", resp);

error = json_get_member(resp, toks, "error");

if (!error)

errx(ERROR_TALKING_TO_PETTYCOIN,

"Missing 'error' in response '%s'", resp);

id = json_get_member(resp, toks, "id");

if (!id)

errx(ERROR_TALKING_TO_PETTYCOIN,

"Missing 'id' in response '%s'", resp);

if (!json_tok_streq(resp, id, idstr))

errx(ERROR_TALKING_TO_PETTYCOIN,

"Incorrect 'id' in response: %.*s",

json_tok_len(id), json_tok_contents(resp, id));

if (json_tok_is_null(resp, error)) {

printf("%.*s\n",

json_tok_len(result),

json_tok_contents(resp, result));

tal_free(ctx);

return 0;

}

printf("%.*s\n",

json_tok_len(error), json_tok_contents(resp, error));

tal_free(ctx);

return 1;

}