make uqmi_add_error return QMI_CMD_EXIT
[project/uqmi.git] / 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                 "                                    (implies --keep-client-id)\n"
37                 "  --release-client-id <name>:       Release Client ID after exiting\n"
38                 "\n"
39                 "Services:                           dms, nas, pds, wds, wms\n"
40                 "\n"
41                 "Actions:\n"
42                 "  --get-versions:                   Get service versions\n"
43                 "  --set-client-id <name>,<id>:      Set Client ID for service <name> to <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 }