client/subauth.c

   1 /*
   2  *      Sub-authentication Client
   3  *
   4  *      (c) 2017 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #include <ucw/lib.h>
   8 #include <ucw/opt.h>
   9 #include <ucw-json/json.h>
  10
  11 #include <errno.h>
  12 #include <fcntl.h>
  13 #include <sys/socket.h>
  14 #include <sys/un.h>
  15 #include <unistd.h>
  16
  17 #include "autoconf.h"
  18
  19 /*** Arguments ***/
  20
  21 static char *socket_path = INSTALL_RUN_DIR "/subauthd.socket";
  22
  23 static char *arg_login;
  24 static char *arg_zone;
  25 static char *arg_ident;
  26 static int debug;
  27
  28 /*** Communication with the daemon ***/
  29
  30 static struct json_context *js;
  31 static struct json_node *rq;
  32 static struct json_node *rp;
  33
  34 static void set_string(struct json_node *n, const char *key, const char *val)
  35 {
  36   json_object_set(n, key, json_new_string(js, val));
  37 }
  38
  39 static struct json_node *get_child(struct json_node *obj, const char *key, bool mandatory, uint type)
  40 {
  41   struct json_node *n = json_object_get(obj, key);
  42   if (!n)
  43     {
  44       if (mandatory)
  45         die("Malformed reply: missing %s", key);
  46     }
  47   else if (n->type != type)
  48     die("Malformed reply: %s has wrong type", key);
  49   return n;
  50 }
  51
  52 static void op_new(const char *op)
  53 {
  54   js = json_new();
  55   rq = json_new_object(js);
  56   set_string(rq, "cmd", op);
  57 }
  58
  59 static void op_debug(const char *msg, struct json_node *n)
  60 {
  61   if (!debug)
  62     return;
  63
  64   struct fastbuf *out = bfdopen_shared(1, 4096);
  65   bprintf(out, ">>> %s\n", msg);
  66   js->format_options = JSON_FORMAT_INDENT;
  67   json_write(js, out, n);
  68   js->format_options = 0;
  69   bclose(out);
  70 }
  71
  72 static void op_run(void)
  73 {
  74   int sk = socket(PF_UNIX, SOCK_SEQPACKET, 0);
  75   if (sk < 0)
  76     die("socket(PF_UNIX, SOCK_SEQPACKET): %m");
  77
  78   struct sockaddr_un sun;
  79   sun.sun_family = AF_UNIX;
  80   if (strlen(socket_path) >= sizeof(sun.sun_path))
  81     die("Socket path too long");
  82   strcpy(sun.sun_path, socket_path);
  83
  84   if (connect(sk, (struct sockaddr *) &sun, sizeof(sun)) < 0)
  85     die("Cannot connect to %s: %m", socket_path);
  86
  87   op_debug("Request", rq);
  88
  89   struct fastbuf *rq_fb = fbgrow_create(4096);
  90   json_write(js, rq_fb, rq);
  91   byte *rq_buf;
  92   uint rq_len = fbgrow_get_buf(rq_fb, &rq_buf);
  93   if (send(sk, rq_buf, rq_len, 0) < 0)
  94     die("Cannot send request: %m");
  95   bclose(rq_fb);
  96
  97   byte rp_buf[4096];            // FIXME
  98   int rp_len = recv(sk, rp_buf, sizeof(rp_buf), 0);
  99   if (rp_len < 0)
 100     die("Cannot receive reply: %m");
 101
 102   struct fastbuf rp_fb;
 103   fbbuf_init_read(&rp_fb, rp_buf, rp_len, 0);
 104   rp = json_parse(js, &rp_fb);
 105   op_debug("Reply", rp);
 106
 107   close(sk);
 108
 109   if (rp->type != JSON_OBJECT)
 110     die("Malformed reply: Top-level node is not an object");
 111   struct json_node *err = get_child(rp, "error", 1, JSON_STRING);
 112   if (strcmp(err->string, ""))
 113     {
 114       fprintf(stderr, "Error: %s\n", err->string);
 115       exit(1);
 116     }
 117 }
 118
 119 /*** Commands ***/
 120
 121 static void cmd_set_passwd(void)
 122 {
 123 }
 124
 125 static void cmd_list(void)
 126 {
 127   op_new("list");
 128   op_run();
 129 }
 130
 131 static void cmd_raw(void)
 132 {
 133   op_new("");
 134
 135   struct fastbuf *in = bfdopen_shared(0, 4096);
 136   rq = json_parse(js, in);
 137   bclose(in);
 138
 139   op_run();
 140 }
 141
 142 enum command {
 143   CMD_SET_PASSWD,
 144   CMD_LIST,
 145   CMD_RAW,
 146   CMD_MAX
 147 };
 148
 149 void (* const command_handlers[CMD_MAX])(void) = {
 150   [CMD_SET_PASSWD] = cmd_set_passwd,
 151   [CMD_LIST] = cmd_list,
 152   [CMD_RAW] = cmd_raw,
 153 };
 154
 155 static int command = -1;
 156
 157 /*** Main ***/
 158
 159 static const struct opt_section options = {
 160   OPT_ITEMS {
 161     OPT_HELP("A client to the sub-authentication daemon."),
 162     OPT_HELP("Usage: subauth [general options] <command> [options]"),
 163     OPT_HELP(""),
 164     OPT_HELP("General options:"),
 165     OPT_STRING('s', "socket", socket_path, OPT_REQUIRED_VALUE, "path\tPath to daemon control socket"),
 166     OPT_BOOL(0, "debug", debug, 0, "\tDump raw communication with the daemon"),
 167     OPT_HELP_OPTION,
 168     OPT_HELP(""),
 169     OPT_HELP("Commands:"),
 170     OPT_SWITCH(0, "passwd", command, CMD_SET_PASSWD, OPT_SINGLE, "\tSet password"),
 171     OPT_SWITCH(0, "list", command, CMD_LIST, OPT_SINGLE, "\tList tokens for all zones"),
 172     OPT_SWITCH(0, "raw", command, CMD_RAW, OPT_SINGLE, "\tSend raw JSON command to the daemon"),
 173     OPT_HELP(""),
 174     OPT_HELP("Command options:"),
 175     OPT_STRING('u', "user", arg_login, OPT_REQUIRED_VALUE, "login\tUser to act on (default: whoever calls it)"),
 176     OPT_STRING('z', "zone", arg_zone, OPT_REQUIRED_VALUE, "zone\tAuthentication zone"),
 177     OPT_STRING('i', "ident", arg_ident, OPT_REQUIRED_VALUE, "id\tToken ID"),
 178     OPT_END
 179   }
 180 };
 181
 182 int main(int argc UNUSED, char **argv)
 183 {
 184   opt_parse(&options, argv+1);
 185
 186   if (command < 0)
 187     opt_failure("No command given");
 188   if (!command_handlers[command])
 189     opt_failure("Command not implemented");
 190   command_handlers[command]();
 191
 192   return 0;
 193 }