main.c

   1 #include <libubox/uloop.h>
   2 #include <libubox/utils.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <fcntl.h>
   6 #include <unistd.h>
   7 #include <errno.h>
   8 #include <getopt.h>
   9 #include <signal.h>
  10
  11 #include "uqmi.h"
  12 #include "commands.h"
  13
  14 static const char *device;
  15
  16 #define CMD_OPT(_arg) (-2 - _arg)
  17
  18 #define __uqmi_command(_name, _optname, _arg, _option) { #_optname, _arg##_argument, NULL, CMD_OPT(__UQMI_COMMAND_##_name) }
  19 static const struct option uqmi_getopt[] = {
  20         __uqmi_commands,
  21         { "single", no_argument, NULL, 's' },
  22         { "device", required_argument, NULL, 'd' },
  23         { "keep-client-id", required_argument, NULL, 'k' },
  24         { "release-client-id", required_argument, NULL, 'r' },
  25         { NULL, 0, NULL, 0 }
  26 };
  27 #undef __uqmi_command
  28
  29 static int usage(const char *progname)
  30 {
  31         fprintf(stderr, "Usage: %s <options|actions>\n"
  32                 "Options:\n"
  33                 "  --single, -s:                     Print output as a single line (for scripts)\n"
  34                 "  --device=NAME, -d NAME:           Set device name to NAME (required)\n"
  35                 "  --keep-client-id <name>:          Keep Client ID for service <name>\n"
  36                 "  --release-client-id <name>:       Release Client ID after exiting\n"
  37                 "\n"
  38                 "Services:                           dms, nas, pds, wds, wms\n"
  39                 "\n"
  40                 "Actions:\n"
  41                 "  --get-versions:                   Get service versions\n"
  42                 "  --set-client-id <name>,<id>:      Set Client ID for service <name> to <id>\n"
  43                 "                                    (implies --keep-client-id)\n"
  44                 "  --get-client-id <name>:           Connect and get Client ID for service <name>\n"
  45                 "                                    (implies --keep-client-id)\n"
  46                 wds_helptext
  47                 dms_helptext
  48                 nas_helptext
  49                 wms_helptext
  50                 "\n", progname);
  51         return 1;
  52 }
  53
  54 static void keep_client_id(struct qmi_dev *qmi, const char *optarg)
  55 {
  56         QmiService svc = qmi_service_get_by_name(optarg);
  57         if (svc < 0) {
  58                 fprintf(stderr, "Invalid service %s\n", optarg);
  59                 exit(1);
  60         }
  61         qmi_service_get_client_id(qmi, svc);
  62 }
  63
  64 static void release_client_id(struct qmi_dev *qmi, const char *optarg)
  65 {
  66         QmiService svc = qmi_service_get_by_name(optarg);
  67         if (svc < 0) {
  68                 fprintf(stderr, "Invalid service %s\n", optarg);
  69                 exit(1);
  70         }
  71         qmi_service_release_client_id(qmi, svc);
  72 }
  73
  74 static void handle_exit_signal(int signal)
  75 {
  76         cancel_all_requests = true;
  77         uloop_end();
  78 }
  79
  80 int main(int argc, char **argv)
  81 {
  82         static struct qmi_dev dev;
  83         int ch, ret;
  84
  85         uloop_init();
  86         signal(SIGINT, handle_exit_signal);
  87         signal(SIGTERM, handle_exit_signal);
  88
  89         while ((ch = getopt_long(argc, argv, "d:k:s", uqmi_getopt, NULL)) != -1) {
  90                 int cmd_opt = CMD_OPT(ch);
  91
  92                 if (ch < 0 && cmd_opt >= 0 && cmd_opt < __UQMI_COMMAND_LAST) {
  93                         uqmi_add_command(optarg, cmd_opt);
  94                         continue;
  95                 }
  96
  97                 switch(ch) {
  98                 case 'r':
  99                         release_client_id(&dev, optarg);
 100                         break;
 101                 case 'k':
 102                         keep_client_id(&dev, optarg);
 103                         break;
 104                 case 'd':
 105                         device = optarg;
 106                         break;
 107                 case 's':
 108                         single_line = true;
 109                         break;
 110                 default:
 111                         return usage(argv[0]);
 112                 }
 113         }
 114
 115         if (!device) {
 116                 fprintf(stderr, "No device given\n");
 117                 return usage(argv[0]);
 118         }
 119
 120         if (qmi_device_open(&dev, device)) {
 121                 fprintf(stderr, "Failed to open device\n");
 122                 return 2;
 123         }
 124
 125         ret = uqmi_run_commands(&dev) ? 0 : -1;
 126
 127         qmi_device_close(&dev);
 128
 129         return ret;
 130 }