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         { "mbim",  no_argument, NULL, 'm' },
  47         { NULL, 0, NULL, 0 }
  48 };
  49 #undef __uqmi_command
  50
  51 static int usage(const char *progname)
  52 {
  53         fprintf(stderr, "Usage: %s <options|actions>\n"
  54                 "Options:\n"
  55                 "  --single, -s:                     Print output as a single line (for scripts)\n"
  56                 "  --device=NAME, -d NAME:           Set device name to NAME (required)\n"
  57                 "  --keep-client-id <name>:          Keep Client ID for service <name>\n"
  58                 "  --release-client-id <name>:       Release Client ID after exiting\n"
  59                 "  --mbim, -m                        NAME is an MBIM device with EXT_QMUX support\n"
  60                 "\n"
  61                 "Services:                           dms, nas, pds, wds, wms\n"
  62                 "\n"
  63                 "Actions:\n"
  64                 "  --get-versions:                   Get service versions\n"
  65                 "  --set-client-id <name>,<id>:      Set Client ID for service <name> to <id>\n"
  66                 "                                    (implies --keep-client-id)\n"
  67                 "  --get-client-id <name>:           Connect and get Client ID for service <name>\n"
  68                 "                                    (implies --keep-client-id)\n"
  69                 wds_helptext
  70                 dms_helptext
  71                 nas_helptext
  72                 wms_helptext
  73                 wda_helptext
  74                 "\n", progname);
  75         return 1;
  76 }
  77
  78 static void keep_client_id(struct qmi_dev *qmi, const char *optarg)
  79 {
  80         QmiService svc = qmi_service_get_by_name(optarg);
  81         if (svc < 0) {
  82                 fprintf(stderr, "Invalid service %s\n", optarg);
  83                 exit(1);
  84         }
  85         qmi_service_get_client_id(qmi, svc);
  86 }
  87
  88 static void release_client_id(struct qmi_dev *qmi, const char *optarg)
  89 {
  90         QmiService svc = qmi_service_get_by_name(optarg);
  91         if (svc < 0) {
  92                 fprintf(stderr, "Invalid service %s\n", optarg);
  93                 exit(1);
  94         }
  95         qmi_service_release_client_id(qmi, svc);
  96 }
  97
  98 static void handle_exit_signal(int signal)
  99 {
 100         cancel_all_requests = true;
 101         uloop_end();
 102 }
 103
 104 int main(int argc, char **argv)
 105 {
 106         static struct qmi_dev dev;
 107         int ch, ret;
 108
 109         uloop_init();
 110         signal(SIGINT, handle_exit_signal);
 111         signal(SIGTERM, handle_exit_signal);
 112
 113         while ((ch = getopt_long(argc, argv, "d:k:sm", uqmi_getopt, NULL)) != -1) {
 114                 int cmd_opt = CMD_OPT(ch);
 115
 116                 if (ch < 0 && cmd_opt >= 0 && cmd_opt < __UQMI_COMMAND_LAST) {
 117                         uqmi_add_command(optarg, cmd_opt);
 118                         continue;
 119                 }
 120
 121                 switch(ch) {
 122                 case 'r':
 123                         release_client_id(&dev, optarg);
 124                         break;
 125                 case 'k':
 126                         keep_client_id(&dev, optarg);
 127                         break;
 128                 case 'd':
 129                         device = optarg;
 130                         break;
 131                 case 's':
 132                         single_line = true;
 133                         break;
 134                 case 'm':
 135                         dev.is_mbim = true;
 136                         break;
 137                 default:
 138                         return usage(argv[0]);
 139                 }
 140         }
 141
 142         if (!device) {
 143                 fprintf(stderr, "No device given\n");
 144                 return usage(argv[0]);
 145         }
 146
 147         if (qmi_device_open(&dev, device)) {
 148                 fprintf(stderr, "Failed to open device\n");
 149                 return 2;
 150         }
 151
 152         ret = uqmi_run_commands(&dev) ? 0 : -1;
 153
 154         qmi_device_close(&dev);
 155
 156         return ret;
 157 }