sanitize pincode properly
[project/umbim.git] / cli.c
1 /*
2  * umbim
3  * Copyright (C) 2014 John Crispin <blogic@openwrt.org>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2
7  * as published by the Free Software Foundation
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  */
14
15 #define __STDC_FORMAT_MACROS
16 #include <inttypes.h>
17
18 #include <sys/types.h>
19 #include <sys/stat.h>
20
21 #include <alloca.h>
22 #include <fcntl.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27 #include <ctype.h>
28
29 #include <libubox/utils.h>
30 #include <libubox/uloop.h>
31
32 #include "mbim.h"
33
34 #include "data/mbim-service-basic-connect.h"
35
36 int return_code = -1;
37 int verbose;
38
39 struct mbim_handler *current_handler;
40 static uint8_t uuid_context_type_internet[16] = { 0x7E, 0x5E, 0x2A, 0x7E, 0x4E, 0x6F, 0x72, 0x72, 0x73, 0x6B, 0x65, 0x6E, 0x7E, 0x5E, 0x2A, 0x7E };
41 static int _argc;
42 static char **_argv;
43
44 static int
45 mbim_device_caps_response(void *buffer, int len)
46 {
47         struct mbim_basic_connect_device_caps_r *caps = (struct mbim_basic_connect_device_caps_r *) buffer;
48         char *deviceid, *firmwareinfo, *hardwareinfo;
49
50         if (len < sizeof(struct mbim_basic_connect_device_caps_r)) {
51                 fprintf(stderr, "message not long enough\n");
52                 return -1;
53         }
54
55         deviceid = mbim_get_string(&caps->deviceid, buffer);
56         firmwareinfo = mbim_get_string(&caps->firmwareinfo, buffer);
57         hardwareinfo = mbim_get_string(&caps->hardwareinfo, buffer);
58
59         printf("  devicetype: %04X - %s\n", le32toh(caps->devicetype),
60                 mbim_enum_string(mbim_device_type_values, le32toh(caps->devicetype)));
61         printf("  cellularclass: %04X\n", le32toh(caps->cellularclass));
62         printf("  voiceclass: %04X - %s\n", le32toh(caps->voiceclass),
63                 mbim_enum_string(mbim_voice_class_values, le32toh(caps->voiceclass)));
64         printf("  simclass: %04X\n", le32toh(caps->simclass));
65         printf("  dataclass: %04X\n", le32toh(caps->dataclass));
66         printf("  smscaps: %04X\n", le32toh(caps->smscaps));
67         printf("  controlcaps: %04X\n", le32toh(caps->controlcaps));
68         printf("  maxsessions: %04X\n", le32toh(caps->maxsessions));
69         printf("  deviceid: %s\n", deviceid);
70         printf("  firmwareinfo: %s\n", firmwareinfo);
71         printf("  hardwareinfo: %s\n", hardwareinfo);
72
73         return 0;
74 }
75
76 static int
77 mbim_pin_state_response(void *buffer, int len)
78 {
79         struct mbim_basic_connect_pin_r *pin = (struct mbim_basic_connect_pin_r *) buffer;
80
81         if (len < sizeof(struct mbim_basic_connect_pin_r)) {
82                 fprintf(stderr, "message not long enough\n");
83                 return -1;
84         }
85
86         if (le32toh(pin->pinstate) != MBIM_PIN_STATE_UNLOCKED) {
87                 fprintf(stderr, "required pin: %d - %s\n",
88                         le32toh(pin->pintype), mbim_enum_string(mbim_pin_type_values, le32toh(pin->pintype)));
89                 fprintf(stderr, "remaining attempts: %d\n", le32toh(pin->remainingattempts));
90                 return le32toh(pin->pintype);
91         }
92
93         fprintf(stderr, "Pin Unlocked\n");
94
95         return 0;
96 }
97
98 static int
99 mbim_registration_response(void *buffer, int len)
100 {
101         struct mbim_basic_connect_register_state_r *state = (struct mbim_basic_connect_register_state_r *) buffer;
102         char *provider_id, *provider_name, *roamingtext;
103
104         if (len < sizeof(struct mbim_basic_connect_register_state_r)) {
105                 fprintf(stderr, "message not long enough\n");
106                 return -1;
107         }
108
109         provider_id = mbim_get_string(&state->providerid, buffer);
110         provider_name = mbim_get_string(&state->providername, buffer);
111         roamingtext = mbim_get_string(&state->roamingtext, buffer);
112
113         printf("  nwerror: %04X - %s\n", le32toh(state->nwerror),
114                 mbim_enum_string(mbim_nw_error_values, le32toh(state->nwerror)));
115         printf("  registerstate: %04X - %s\n", le32toh(state->registerstate),
116                 mbim_enum_string(mbim_register_state_values, le32toh(state->registerstate)));
117         printf("  registermode: %04X - %s\n", le32toh(state->registermode),
118                 mbim_enum_string(mbim_register_mode_values, le32toh(state->registermode)));
119         printf("  availabledataclasses: %04X - %s\n", le32toh(state->availabledataclasses),
120                 mbim_enum_string(mbim_data_class_values, le32toh(state->availabledataclasses)));
121         printf("  currentcellularclass: %04X - %s\n", le32toh(state->currentcellularclass),
122                 mbim_enum_string(mbim_cellular_class_values, le32toh(state->currentcellularclass)));
123         printf("  provider_id: %s\n", provider_id);
124         printf("  provider_name: %s\n", provider_name);
125         printf("  roamingtext: %s\n", roamingtext);
126
127         if (le32toh(state->registerstate) == MBIM_REGISTER_STATE_HOME)
128                 return 0;
129
130         return le32toh(state->registerstate);
131 }
132
133 static int
134 mbim_subscriber_response(void *buffer, int len)
135 {
136         struct mbim_basic_connect_subscriber_ready_status_r *state = (struct mbim_basic_connect_subscriber_ready_status_r *) buffer;
137         char *subscriberid, *simiccid;
138         int nr;
139
140         if (len < sizeof(struct mbim_basic_connect_subscriber_ready_status_r)) {
141                 fprintf(stderr, "message not long enough\n");
142                 return -1;
143         }
144
145         subscriberid = mbim_get_string(&state->subscriberid, buffer);
146         simiccid = mbim_get_string(&state->simiccid, buffer);
147
148         printf("  readystate: %04X - %s\n", le32toh(state->readystate),
149                 mbim_enum_string(mbim_subscriber_ready_state_values, le32toh(state->readystate)));
150         printf("  simiccid: %s\n", simiccid);
151         printf("  subscriberid: %s\n", subscriberid);
152         if (le32toh(state->readyinfo) & MBIM_READY_INFO_FLAG_PROTECT_UNIQUE_ID)
153                 printf("  dont display subscriberID: 1\n");
154         for (nr = 0; nr < le32toh(state->telephonenumberscount); nr++) {
155                 struct mbim_string *str = buffer + le32toh(state->telephonenumbers) + (nr * sizeof(struct mbim_string));
156                 char *number = mbim_get_string(str, buffer);
157                 printf("  number: %s\n", number);
158         }
159
160         if (MBIM_SUBSCRIBER_READY_STATE_INITIALIZED == le32toh(state->readystate))
161                 return 0;
162
163         return le32toh(state->readystate);
164 }
165
166 static int
167 mbim_attach_response(void *buffer, int len)
168 {
169         struct mbim_basic_connect_packet_service_r *ps = (struct mbim_basic_connect_packet_service_r *) buffer;
170
171         if (len < sizeof(struct mbim_basic_connect_packet_service_r)) {
172                 fprintf(stderr, "message not long enough\n");
173                 return -1;
174         }
175
176         printf("  nwerror: %04X - %s\n", le32toh(ps->nwerror),
177                 mbim_enum_string(mbim_nw_error_values, le32toh(ps->nwerror)));
178         printf("  packetservicestate: %04X - %s\n", le32toh(ps->packetservicestate),
179                 mbim_enum_string(mbim_packet_service_state_values, le32toh(ps->packetservicestate)));
180         printf("  uplinkspeed: %"PRIu64"\n", le64toh(ps->uplinkspeed));
181         printf("  downlinkspeed: %"PRIu64"\n", le64toh(ps->downlinkspeed));
182
183         if (MBIM_PACKET_SERVICE_STATE_ATTACHED == le32toh(ps->packetservicestate))
184                 return 0;
185
186         return le32toh(ps->packetservicestate);
187 }
188
189 static int
190 mbim_connect_response(void *buffer, int len)
191 {
192         struct mbim_basic_connect_connect_r *c = (struct mbim_basic_connect_connect_r *) buffer;
193
194         if (len < sizeof(struct mbim_basic_connect_connect_r)) {
195                 fprintf(stderr, "message not long enough\n");
196                 return -1;
197         }
198
199         printf("  sessionid: %d\n", le32toh(c->sessionid));
200         printf("  activationstate: %04X - %s\n", le32toh(c->activationstate),
201                 mbim_enum_string(mbim_activation_state_values, le32toh(c->activationstate)));
202         printf("  voicecallstate: %04X - %s\n", le32toh(c->voicecallstate),
203                 mbim_enum_string(mbim_voice_call_state_values, le32toh(c->voicecallstate)));
204         printf("  nwerror: %04X - %s\n", le32toh(c->nwerror),
205                 mbim_enum_string(mbim_nw_error_values, le32toh(c->nwerror)));
206         printf("  iptype: %04X - %s\n", le32toh(c->iptype),
207                 mbim_enum_string(mbim_context_ip_type_values, le32toh(c->iptype)));
208
209         if (MBIM_ACTIVATION_STATE_ACTIVATED == le32toh(c->activationstate))
210                 return 0;
211
212         return le32toh(c->activationstate);
213 }
214
215 static int
216 mbim_config_response(void *buffer, int len)
217 {
218         struct mbim_basic_connect_ip_configuration_r *ip = (struct mbim_basic_connect_ip_configuration_r *) buffer;
219         char ipv4[16];
220         int i;
221
222         if (len < sizeof(struct mbim_basic_connect_ip_configuration_r)) {
223                 fprintf(stderr, "message not long enough\n");
224                 return -1;
225         }
226
227         if (le32toh(ip->ipv4configurationavailable) & MBIM_IP_CONFIGURATION_AVAILABLE_FLAG_ADDRESS)
228                 for (i = 0; i < le32toh(ip->ipv4addresscount); i++) {
229                         mbim_get_ipv4(buffer, ipv4, ip->ipv4address + (i * 4));
230                         printf("  ipv4address: %s\n", ipv4);
231                 }
232         if (le32toh(ip->ipv4configurationavailable) & MBIM_IP_CONFIGURATION_AVAILABLE_FLAG_DNS) {
233                 mbim_get_ipv4(buffer, ipv4, ip->ipv4gateway);
234                 printf("  ipv4gateway: %s\n", ipv4);
235         }
236         if (le32toh(ip->ipv4configurationavailable) & MBIM_IP_CONFIGURATION_AVAILABLE_FLAG_MTU)
237                 printf("  ipv4mtu: %d\n", le32toh(ip->ipv4mtu));
238         if (le32toh(ip->ipv4configurationavailable) & MBIM_IP_CONFIGURATION_AVAILABLE_FLAG_DNS)
239                 for (i = 0; i < le32toh(ip->ipv4dnsservercount); i++) {
240                         mbim_get_ipv4(buffer, ipv4, ip->ipv4dnsserver + (i * 4));
241                         printf("  ipv4dnsserver: %s\n", ipv4);
242                 }
243
244         printf("  ipv6configurationavailable: %04X\n", le32toh(ip->ipv6configurationavailable));
245
246         return 0;
247 }
248
249 static int
250 mbim_device_caps_request(void)
251 {
252         mbim_setup_command_msg(basic_connect, MBIM_MESSAGE_COMMAND_TYPE_QUERY, MBIM_CMD_BASIC_CONNECT_DEVICE_CAPS, 0);
253
254         return mbim_send_command_msg();
255 }
256
257 static int
258 mbim_pin_state_request(void)
259 {
260         mbim_setup_command_msg(basic_connect, MBIM_MESSAGE_COMMAND_TYPE_QUERY, MBIM_CMD_BASIC_CONNECT_PIN, 0);
261
262         return mbim_send_command_msg();
263 }
264
265 static int
266 mbim_registration_request(void)
267 {
268         mbim_setup_command_msg(basic_connect, MBIM_MESSAGE_COMMAND_TYPE_QUERY, MBIM_CMD_BASIC_CONNECT_REGISTER_STATE, 0);
269
270         return mbim_send_command_msg();
271 }
272
273 static int
274 mbim_subscriber_request(void)
275 {
276         mbim_setup_command_msg(basic_connect, MBIM_MESSAGE_COMMAND_TYPE_QUERY, MBIM_CMD_BASIC_CONNECT_SUBSCRIBER_READY_STATUS, 0);
277
278         return mbim_send_command_msg();
279 }
280
281 static int
282 _mbim_attach_request(int action)
283 {
284         struct mbim_basic_connect_packet_service_s *ps =
285                 (struct mbim_basic_connect_packet_service_s *) mbim_setup_command_msg(basic_connect,
286                         MBIM_MESSAGE_COMMAND_TYPE_SET, MBIM_CMD_BASIC_CONNECT_PACKET_SERVICE,
287                         sizeof(struct mbim_basic_connect_packet_service_s));
288
289         ps->packetserviceaction = htole32(action);
290
291         return mbim_send_command_msg();
292 }
293
294 static int
295 mbim_attach_request(void)
296 {
297         return _mbim_attach_request(MBIM_PACKET_SERVICE_ACTION_ATTACH);
298 }
299
300 static int
301 mbim_detach_request(void)
302 {
303         return _mbim_attach_request(MBIM_PACKET_SERVICE_ACTION_DETACH);
304 }
305
306 static int
307 mbim_connect_request(void)
308 {
309         struct mbim_basic_connect_connect_s *c =
310                 (struct mbim_basic_connect_connect_s *) mbim_setup_command_msg(basic_connect,
311                         MBIM_MESSAGE_COMMAND_TYPE_SET, MBIM_CMD_BASIC_CONNECT_CONNECT,
312                         sizeof(struct mbim_basic_connect_connect_s));
313
314         c->activationcommand = htole32(MBIM_ACTIVATION_COMMAND_ACTIVATE);
315         c->iptype = htole32(MBIM_CONTEXT_IP_TYPE_DEFAULT);
316         memcpy(c->contexttype, uuid_context_type_internet, 16);
317         if (_argc > 0)
318                 mbim_encode_string(&c->accessstring, *_argv);
319
320         return mbim_send_command_msg();
321 }
322
323 static int
324 mbim_disconnect_request(void)
325 {
326         struct mbim_basic_connect_connect_s *c =
327                 (struct mbim_basic_connect_connect_s *) mbim_setup_command_msg(basic_connect,
328                         MBIM_MESSAGE_COMMAND_TYPE_SET, MBIM_CMD_BASIC_CONNECT_CONNECT,
329                         sizeof(struct mbim_basic_connect_connect_s));
330
331         c->activationcommand = htole32(MBIM_ACTIVATION_COMMAND_DEACTIVATE);
332         memcpy(c->contexttype, uuid_context_type_internet, 16);
333
334         no_close = 0;
335
336         return mbim_send_command_msg();
337 }
338
339 static char*
340 mbim_pin_sanitize(char *pin)
341 {
342         char *p;
343
344         while (*pin && !isdigit(*pin))
345                 pin++;
346         p = pin;
347         if (!*p)
348                 return NULL;
349         while (*pin && isdigit(*pin))
350                 pin++;
351         if (*pin)
352                 *pin = '\0';
353
354         return p;
355 }
356
357 static int
358 mbim_pin_unlock_request(void)
359 {
360         struct mbim_basic_connect_pin_s *p =
361                 (struct mbim_basic_connect_pin_s *) mbim_setup_command_msg(basic_connect,
362                         MBIM_MESSAGE_COMMAND_TYPE_SET, MBIM_CMD_BASIC_CONNECT_PIN,
363                         sizeof(struct mbim_basic_connect_pin_s));
364         char *pin = mbim_pin_sanitize(_argv[0]);
365
366         if (!pin || !strlen(pin)) {
367                 fprintf(stderr, "failed to sanitize the pincode\n");
368                 return -1;
369         }
370
371         p->pintype = htole32(MBIM_PIN_TYPE_PIN1);
372         p->pinoperation = htole32(MBIM_PIN_OPERATION_ENTER);
373         mbim_encode_string(&p->pin, _argv[0]);
374
375         return mbim_send_command_msg();
376 }
377
378 static int
379 mbim_config_request(void)
380 {
381         mbim_setup_command_msg(basic_connect,
382                 MBIM_MESSAGE_COMMAND_TYPE_QUERY, MBIM_CMD_BASIC_CONNECT_IP_CONFIGURATION,
383                 sizeof(struct mbim_basic_connect_ip_configuration_q));
384
385         return mbim_send_command_msg();
386 }
387
388 static struct mbim_handler handlers[] = {
389         { "caps", 0, mbim_device_caps_request, mbim_device_caps_response },
390         { "pinstate", 0, mbim_pin_state_request, mbim_pin_state_response },
391         { "unlock", 1, mbim_pin_unlock_request, mbim_pin_state_response },
392         { "registration", 0, mbim_registration_request, mbim_registration_response },
393         { "subscriber", 0, mbim_subscriber_request, mbim_subscriber_response },
394         { "attach", 0, mbim_attach_request, mbim_attach_response },
395         { "detach", 0, mbim_detach_request, mbim_attach_response },
396         { "connect", 0, mbim_connect_request, mbim_connect_response },
397         { "disconnect", 0, mbim_disconnect_request, mbim_connect_response },
398         { "config", 0, mbim_config_request, mbim_config_response },
399 };
400
401 static int
402 usage(void)
403 {
404         fprintf(stderr, "Usage: mbim <caps|pinstate|unlock|connect|disconnect> [options]\n"
405                 "Options:\n"
406                 "    -d <device>        the device (/dev/cdc-wdmX)\n"
407                 "    -t <transaction>   the transaction id\n"
408                 "    -n                 no close\n\n"
409                 "    -v                 verbose\n\n");
410         return 1;
411 }
412
413 int
414 main(int argc, char **argv)
415 {
416         char *cmd, *device = NULL;
417         int no_open = 0, ch, i;
418
419         while ((ch = getopt(argc, argv, "nvd:t:")) != -1) {
420                 switch (ch) {
421                 case 'v':
422                         verbose = 1;
423                         break;
424                 case 'n':
425                         no_close = 1;
426                         break;
427                 case 'd':
428                         device = optarg;
429                         break;
430                 case 't':
431                         no_open = 1;
432                         transaction_id = atoi(optarg);
433                         break;
434                 default:
435                         return usage();
436                 }
437         }
438
439         if (!device || optind == argc)
440                 return usage();
441
442         cmd = argv[optind];
443         optind++;
444
445         _argc = argc - optind;
446         _argv = &argv[optind];
447
448         for (i = 0; i < ARRAY_SIZE(handlers); i++)
449                 if (!strcmp(cmd, handlers[i].name))
450                         current_handler = &handlers[i];
451
452         if (!current_handler || (optind + current_handler->argc > argc))
453                 return usage();
454
455         uloop_init();
456
457         mbim_open(device);
458         if (!no_open)
459                 mbim_send_open_msg();
460         else if (current_handler->request() < 0)
461                 return -1;
462
463         uloop_run();
464         uloop_done();
465
466         return return_code;
467 }