commands.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <strings.h>
   4 #include <stdlib.h>
   5 #include <unistd.h>
   6
   7 #include <libubox/blobmsg.h>
   8
   9 #include "uqmi.h"
  10 #include "commands.h"
  11
  12 static struct blob_buf status;
  13
  14 static void no_cb(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg)
  15 {
  16 }
  17
  18 static void cmd_version_cb(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg)
  19 {
  20         struct qmi_ctl_get_version_info_response res;
  21         char name_buf[16];
  22         int i;
  23
  24         qmi_parse_ctl_get_version_info_response(msg, &res);
  25         for (i = 0; i < res.data.service_list_n; i++) {
  26                 sprintf(name_buf, "service_%d", res.data.service_list[i].service);
  27                 blobmsg_printf(&status, name_buf, "%d,%d",
  28                         res.data.service_list[i].major_version,
  29                         res.data.service_list[i].minor_version);
  30         }
  31 }
  32
  33 static enum qmi_cmd_result
  34 cmd_version_prepare(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg, char *arg)
  35 {
  36         qmi_set_ctl_get_version_info_request(msg);
  37         return QMI_CMD_REQUEST;
  38 }
  39
  40 #define cmd_get_client_id_cb no_cb
  41 static enum qmi_cmd_result
  42 cmd_get_client_id_prepare(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg, char *arg)
  43 {
  44         QmiService svc = qmi_service_get_by_name(arg);
  45
  46         if (svc < 0) {
  47                 fprintf(stderr, "Invalid service name '%s'\n", arg);
  48                 return QMI_CMD_EXIT;
  49         }
  50
  51         if (qmi_service_connect(qmi, svc, -1)) {
  52                 fprintf(stderr, "Failed to connect to service\n");
  53                 return QMI_CMD_EXIT;
  54         }
  55
  56         printf("%d\n", qmi_service_get_client_id(qmi, svc));
  57         return QMI_CMD_DONE;
  58 }
  59
  60 #define cmd_set_client_id_cb no_cb
  61 static enum qmi_cmd_result
  62 cmd_set_client_id_prepare(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg, char *arg)
  63 {
  64         QmiService svc;
  65         int id;
  66         char *s;
  67
  68         s = strchr(arg, ',');
  69         if (!s) {
  70                 fprintf(stderr, "Invalid argument\n");
  71                 return QMI_CMD_EXIT;
  72         }
  73         *s = 0;
  74         s++;
  75
  76         id = strtoul(s, &s, 0);
  77         if (s && *s) {
  78                 fprintf(stderr, "Invalid argument\n");
  79                 return QMI_CMD_EXIT;
  80         }
  81
  82         svc = qmi_service_get_by_name(arg);
  83         if (svc < 0) {
  84                 fprintf(stderr, "Invalid service name '%s'\n", arg);
  85                 return QMI_CMD_EXIT;
  86         }
  87
  88         if (qmi_service_connect(qmi, svc, id)) {
  89                 fprintf(stderr, "Failed to connect to service\n");
  90                 return QMI_CMD_EXIT;
  91         }
  92
  93         return QMI_CMD_DONE;
  94 }
  95
  96 #include "commands-wds.c"
  97 #include "commands-dms.c"
  98 #include "commands-nas.c"
  99 #include "commands-wms.c"
 100
 101 #define __uqmi_command(_name, _optname, _arg, _type) \
 102         [__UQMI_COMMAND_##_name] = { \
 103                 .name = #_optname, \
 104                 .type = _type, \
 105                 .prepare = cmd_##_name##_prepare, \
 106                 .cb = cmd_##_name##_cb, \
 107         }
 108
 109 const struct uqmi_cmd_handler uqmi_cmd_handler[__UQMI_COMMAND_LAST] = {
 110         __uqmi_commands
 111 };
 112 #undef __uqmi_command
 113
 114 static struct uqmi_cmd *cmds;
 115 static int n_cmds;
 116
 117 void uqmi_add_command(char *arg, int cmd)
 118 {
 119         int idx = n_cmds++;
 120
 121         cmds = realloc(cmds, n_cmds * sizeof(*cmds));
 122         cmds[idx].handler = &uqmi_cmd_handler[cmd];
 123         cmds[idx].arg = optarg;
 124 }
 125
 126 static void uqmi_print_result(struct blob_attr *data)
 127 {
 128         struct blob_attr *cur;
 129         int rem;
 130
 131         blob_for_each_attr(cur, data, rem) {
 132                 switch (blobmsg_type(cur)) {
 133                 case BLOBMSG_TYPE_STRING:
 134                         printf("%s=%s\n", blobmsg_name(cur), (char *) blobmsg_data(cur));
 135                         break;
 136                 case BLOBMSG_TYPE_INT32:
 137                         printf("%s=%d\n", blobmsg_name(cur), (int32_t) blobmsg_get_u32(cur));
 138                         break;
 139                 case BLOBMSG_TYPE_INT8:
 140                         printf("%s=%s\n", blobmsg_name(cur), blobmsg_get_u8(cur) ? "true" : "false");
 141                         break;
 142                 }
 143         }
 144 }
 145
 146 static bool __uqmi_run_commands(struct qmi_dev *qmi, bool option)
 147 {
 148         static char buf[2048];
 149         static struct qmi_request req;
 150         int i;
 151
 152         for (i = 0; i < n_cmds; i++) {
 153                 enum qmi_cmd_result res;
 154                 bool cmd_option = cmds[i].handler->type == CMD_TYPE_OPTION;
 155                 bool do_break = false;
 156
 157                 if (cmd_option != option)
 158                         continue;
 159
 160                 blob_buf_init(&status, 0);
 161                 if (cmds[i].handler->type > QMI_SERVICE_CTL &&
 162                     qmi_service_connect(qmi, cmds[i].handler->type, -1)) {
 163                         blobmsg_printf(&status, "error", "failed to connect to service");
 164                         res = QMI_CMD_EXIT;
 165                 } else {
 166                         res = cmds[i].handler->prepare(qmi, &req, (void *) buf, cmds[i].arg);
 167                 }
 168
 169                 if (res == QMI_CMD_REQUEST) {
 170                         qmi_request_start(qmi, &req, (void *) buf, cmds[i].handler->cb);
 171                         req.no_error_cb = true;
 172                         if (qmi_request_wait(qmi, &req))
 173                                 blobmsg_add_string(&status, "error", qmi_get_error_str(req.ret));
 174                 } else if (res == QMI_CMD_EXIT) {
 175                         do_break = true;
 176                 }
 177
 178                 uqmi_print_result(status.head);
 179                 if (do_break)
 180                         return false;
 181         }
 182         return true;
 183 }
 184
 185 void uqmi_run_commands(struct qmi_dev *qmi)
 186 {
 187         if (__uqmi_run_commands(qmi, true))
 188                 __uqmi_run_commands(qmi, false);
 189         free(cmds);
 190         cmds = NULL;
 191         n_cmds = 0;
 192 }