main.c

   1 /*
   2  * uqmi -- tiny QMI support implementation
   3  *
   4  * Copyright (C) 2014-2015 Felix Fietkau <nbd@openwrt.org>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  19  * Boston, MA 02110-1301 USA.
  20  */
  21
  22 #include <libubox/uloop.h>
  23 #include <libubox/utils.h>
  24 #include <stdio.h>
  25 #include <stdlib.h>
  26 #include <fcntl.h>
  27 #include <unistd.h>
  28 #include <errno.h>
  29 #include <getopt.h>
  30 #include <signal.h>
  31
  32 #include "uqmi.h"
  33 #include "commands.h"
  34
  35 static const char *device;
  36
  37 #define CMD_OPT(_arg) (-2 - _arg)
  38
  39 #define __uqmi_command(_name, _optname, _arg, _option) { #_optname, _arg##_argument, NULL, CMD_OPT(__UQMI_COMMAND_##_name) }
  40 static const struct option uqmi_getopt[] = {
  41         __uqmi_commands,
  42         { "single", no_argument, NULL, 's' },
  43         { "device", required_argument, NULL, 'd' },
  44         { "keep-client-id", required_argument, NULL, 'k' },
  45         { "release-client-id", required_argument, NULL, 'r' },
  46         { NULL, 0, NULL, 0 }
  47 };
  48 #undef __uqmi_command
  49
  50 static int usage(const char *progname)
  51 {
  52         fprintf(stderr, "Usage: %s <options|actions>\n"
  53                 "Options:\n"
  54                 "  --single, -s:                     Print output as a single line (for scripts)\n"
  55                 "  --device=NAME, -d NAME:           Set device name to NAME (required)\n"
  56                 "  --keep-client-id <name>:          Keep Client ID for service <name>\n"
  57                 "  --release-client-id <name>:       Release Client ID after exiting\n"
  58                 "\n"
  59                 "Services:                           dms, nas, pds, wds, wms\n"
  60                 "\n"
  61                 "Actions:\n"
  62                 "  --get-versions:                   Get service versions\n"
  63                 "  --set-client-id <name>,<id>:      Set Client ID for service <name> to <id>\n"
  64                 "                                    (implies --keep-client-id)\n"
  65                 "  --get-client-id <name>:           Connect and get Client ID for service <name>\n"
  66                 "                                    (implies --keep-client-id)\n"
  67                 wds_helptext
  68                 dms_helptext
  69                 nas_helptext
  70                 wms_helptext
  71                 wda_helptext
  72                 "\n", progname);
  73         return 1;
  74 }
  75
  76 static void keep_client_id(struct qmi_dev *qmi, const char *optarg)
  77 {
  78         QmiService svc = qmi_service_get_by_name(optarg);
  79         if (svc < 0) {
  80                 fprintf(stderr, "Invalid service %s\n", optarg);
  81                 exit(1);
  82         }
  83         qmi_service_get_client_id(qmi, svc);
  84 }
  85
  86 static void release_client_id(struct qmi_dev *qmi, const char *optarg)
  87 {
  88         QmiService svc = qmi_service_get_by_name(optarg);
  89         if (svc < 0) {
  90                 fprintf(stderr, "Invalid service %s\n", optarg);
  91                 exit(1);
  92         }
  93         qmi_service_release_client_id(qmi, svc);
  94 }
  95
  96 static void handle_exit_signal(int signal)
  97 {
  98         cancel_all_requests = true;
  99         uloop_end();
 100 }
 101
 102 int main(int argc, char **argv)
 103 {
 104         static struct qmi_dev dev;
 105         int ch, ret;
 106
 107         uloop_init();
 108         signal(SIGINT, handle_exit_signal);
 109         signal(SIGTERM, handle_exit_signal);
 110
 111         while ((ch = getopt_long(argc, argv, "d:k:s", uqmi_getopt, NULL)) != -1) {
 112                 int cmd_opt = CMD_OPT(ch);
 113
 114                 if (ch < 0 && cmd_opt >= 0 && cmd_opt < __UQMI_COMMAND_LAST) {
 115                         uqmi_add_command(optarg, cmd_opt);
 116                         continue;
 117                 }
 118
 119                 switch(ch) {
 120                 case 'r':
 121                         release_client_id(&dev, optarg);
 122                         break;
 123                 case 'k':
 124                         keep_client_id(&dev, optarg);
 125                         break;
 126                 case 'd':
 127                         device = optarg;
 128                         break;
 129                 case 's':
 130                         single_line = true;
 131                         break;
 132                 default:
 133                         return usage(argv[0]);
 134                 }
 135         }
 136
 137         if (!device) {
 138                 fprintf(stderr, "No device given\n");
 139                 return usage(argv[0]);
 140         }
 141
 142         if (qmi_device_open(&dev, device)) {
 143                 fprintf(stderr, "Failed to open device\n");
 144                 return 2;
 145         }
 146
 147         ret = uqmi_run_commands(&dev) ? 0 : -1;
 148
 149         qmi_device_close(&dev);
 150
 151         return ret;
 152 }