Extend StandardEject sequence to include LUN 1
[project/usbmode.git] / switch.c
1 #include <unistd.h>
2 #include "switch.h"
3
4 enum {
5 DATA_MODE,
6 DATA_MSG,
7 DATA_INTERFACE,
8 DATA_MSG_EP,
9 DATA_RES_EP,
10 DATA_RESPONSE,
11 DATA_RELEASE_DELAY,
12 DATA_CONFIG,
13 DATA_ALT,
14 DATA_DEV_CLASS,
15 __DATA_MAX
16 };
17
18 static void detach_driver(struct usbdev_data *data)
19 {
20 libusb_detach_kernel_driver(data->devh, data->interface);
21 }
22
23 struct msg_entry {
24 char *data;
25 int len;
26 };
27
28 static int send_msg(struct usbdev_data *data, struct msg_entry *msg)
29 {
30 int transferred;
31
32 return libusb_bulk_transfer(data->devh, data->msg_endpoint,
33 (void *) msg->data, msg->len,
34 &transferred, 3000);
35 }
36
37 static int read_response(struct usbdev_data *data, int len)
38 {
39 unsigned char *buf;
40 int ret, transferred;
41
42 if (len < 13)
43 len = 13;
44 buf = alloca(len);
45 ret = libusb_bulk_transfer(data->devh, data->response_endpoint,
46 buf, len, &transferred, 3000);
47 libusb_bulk_transfer(data->devh, data->response_endpoint,
48 buf, 13, &transferred, 100);
49 return ret;
50 }
51
52 static void send_messages(struct usbdev_data *data, struct msg_entry *msg, int n_msg)
53 {
54 int i, len;
55
56 libusb_claim_interface(data->devh, data->interface);
57 libusb_clear_halt(data->devh, data->msg_endpoint);
58
59 for (i = 0; i < n_msg; i++) {
60 if (send_msg(data, &msg[i])) {
61 fprintf(stderr, "Failed to send switch message\n");
62 continue;
63 }
64
65 if (!data->need_response)
66 continue;
67
68 if (!memcmp(msg[i].data, "\x55\x53\x42\x43", 4))
69 len = 13;
70 else
71 len = msg[i].len;
72
73 if (read_response(data, len))
74 return;
75 }
76
77 libusb_clear_halt(data->devh, data->msg_endpoint);
78 libusb_clear_halt(data->devh, data->response_endpoint);
79
80 usleep(200000);
81
82 if (data->release_delay)
83 usleep(data->release_delay * 1000);
84
85 libusb_release_interface(data->devh, data->interface);
86 return;
87 }
88
89 static void send_config_messages(struct usbdev_data *data, struct blob_attr *attr)
90 {
91 struct blob_attr *cur;
92 int rem, n_msg = 0;
93 struct msg_entry *msg;
94
95 blobmsg_for_each_attr(cur, attr, rem)
96 n_msg++;
97
98 msg = alloca(n_msg * sizeof(*msg));
99 n_msg = 0;
100 blobmsg_for_each_attr(cur, attr, rem) {
101 int msg_nr;
102
103 if (blobmsg_type(cur) != BLOBMSG_TYPE_INT32) {
104 fprintf(stderr, "Invalid data in message list\n");
105 return;
106 }
107
108 msg_nr = blobmsg_get_u32(cur);
109 if (msg_nr >= n_messages) {
110 fprintf(stderr, "Message index out of range!\n");
111 return;
112 }
113
114 msg[n_msg].data = messages[msg_nr];
115 msg[n_msg++].len = message_len[msg_nr];
116 }
117
118 send_messages(data, msg, n_msg);
119 }
120
121 static void handle_generic(struct usbdev_data *data, struct blob_attr **tb)
122 {
123 detach_driver(data);
124 send_config_messages(data, tb[DATA_MSG]);
125 }
126
127 static void send_control_packet(struct usbdev_data *data, uint8_t type, uint8_t req,
128 uint16_t val, uint16_t idx, int len)
129 {
130 unsigned char *buffer = alloca(len ? len : 1);
131
132 libusb_control_transfer(data->devh, type, req, val, idx, buffer, len, 1000);
133 }
134
135 static void handle_huawei(struct usbdev_data *data, struct blob_attr **tb)
136 {
137 int type = LIBUSB_REQUEST_TYPE_STANDARD | LIBUSB_RECIPIENT_DEVICE;
138 send_control_packet(data, type, LIBUSB_REQUEST_SET_FEATURE, 1, 0, 0);
139 }
140
141 static void handle_huaweinew(struct usbdev_data *data, struct blob_attr **tb)
142 {
143 static struct msg_entry msgs[] = {
144 {
145 "\x55\x53\x42\x43\x12\x34\x56\x78\x00\x00\x00\x00\x00\x00\x00\x11"
146 "\x06\x20\x00\x00\x01\x01\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
147 }
148 };
149
150 detach_driver(data);
151 data->need_response = false;
152 send_messages(data, msgs, ARRAY_SIZE(msgs));
153 }
154
155 static void handle_standardeject(struct usbdev_data *data, struct blob_attr **tb)
156 {
157 static struct msg_entry msgs[] = {
158 {
159 "\x55\x53\x42\x43\x12\x34\x56\x78\x00\x00\x00\x00\x00\x00\x06\x1e"
160 "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
161 }, {
162 "\x55\x53\x42\x43\x12\x34\x56\x79\x00\x00\x00\x00\x00\x00\x06\x1b"
163 "\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
164 }, {
165 "\x55\x53\x42\x43\x12\x34\x56\x78\x00\x00\x00\x00\x00\x01\x06\x1e"
166 "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
167 }, {
168 "\x55\x53\x42\x43\x12\x34\x56\x79\x00\x00\x00\x00\x00\x01\x06\x1b"
169 "\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
170 }
171 };
172
173 detach_driver(data);
174 data->need_response = true;
175 send_messages(data, msgs, ARRAY_SIZE(msgs));
176 }
177
178 static void handle_sierra(struct usbdev_data *data, struct blob_attr **tb)
179 {
180 int type = LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE;
181 send_control_packet(data, type, LIBUSB_REQUEST_SET_INTERFACE, 1, 0, 0);
182 }
183
184 static void handle_sony(struct usbdev_data *data, struct blob_attr **tb)
185 {
186 int type = LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_ENDPOINT_IN;
187 int i;
188
189 detach_driver(data);
190 send_control_packet(data, type, 0x11, 2, 0, 3);
191
192 libusb_close(data->devh);
193 sleep(5);
194
195 for (i = 0; i < 25; i++) {
196 data->devh = libusb_open_device_with_vid_pid(usb,
197 data->desc.idVendor, data->desc.idProduct);
198 if (data->devh)
199 break;
200 }
201
202 send_control_packet(data, type, 0x11, 2, 0, 3);
203 }
204
205 static void handle_qisda(struct usbdev_data *data, struct blob_attr **tb)
206 {
207 static unsigned char buffer[] = "\x05\x8c\x04\x08\xa0\xee\x20\x00\x5c\x01\x04\x08\x98\xcd\xea\xbf";
208 int type = LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE;
209
210 libusb_control_transfer(data->devh, type, 0x04, 0, 0, buffer, 16, 1000);
211 }
212
213 static void handle_gct(struct usbdev_data *data, struct blob_attr **tb)
214 {
215 int type = LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE | LIBUSB_ENDPOINT_IN;
216
217 detach_driver(data);
218
219 if (libusb_claim_interface(data->devh, data->interface))
220 return;
221
222 send_control_packet(data, type, 0xa0, 0, data->interface, 1);
223 send_control_packet(data, type, 0xfe, 0, data->interface, 1);
224
225 libusb_release_interface(data->devh, data->interface);
226 }
227
228 static void handle_kobil(struct usbdev_data *data, struct blob_attr **tb)
229 {
230 int type = LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE | LIBUSB_ENDPOINT_IN;
231
232 detach_driver(data);
233 send_control_packet(data, type, 0x88, 0, 0, 8);
234 }
235
236 static void handle_sequans(struct usbdev_data *data, struct blob_attr **tb)
237 {
238 int type = LIBUSB_REQUEST_TYPE_VENDOR | LIBUSB_RECIPIENT_DEVICE;
239 send_control_packet(data, type, LIBUSB_REQUEST_SET_INTERFACE, 2, 0, 0);
240 }
241
242 static void mobile_action_interrupt_msg(struct usbdev_data *data, void *msg, int n_in)
243 {
244 unsigned char *buf = alloca(8);
245 int ep_out = 0x02, ep_in = 0x81;
246 int transferred;
247 int i;
248
249 if (msg)
250 libusb_interrupt_transfer(data->devh, ep_out, msg, 8, &transferred, 1000);
251 for (i = 0; i < n_in; i++)
252 libusb_interrupt_transfer(data->devh, ep_in, buf, 8, &transferred, 1000);
253 }
254
255 static void handle_mobile_action(struct usbdev_data *data, struct blob_attr **tb)
256 {
257 int type = LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE;
258 char *msg[] = {
259 "\xb0\x04\x00\x00\x02\x90\x26\x86",
260 "\x37\x01\xfe\xdb\xc1\x33\x1f\x83",
261 "\x37\x0e\xb5\x9d\x3b\x8a\x91\x51",
262 "\x34\x87\xba\x0d\xfc\x8a\x91\x51",
263 "\x37\x01\xfe\xdb\xc1\x33\x1f\x83",
264 "\x37\x0e\xb5\x9d\x3b\x8a\x91\x51",
265 "\x34\x87\xba\x0d\xfc\x8a\x91\x51",
266 "\x33\x04\xfe\x00\xf4\x6c\x1f\xf0",
267 "\x32\x07\xfe\xf0\x29\xb9\x3a\xf0"
268 };
269 int i;
270
271 for (i = 0; i < 2; i++)
272 libusb_control_transfer(data->devh, type, 0x09, 0x0300, 0, (void *) msg[0], 8, 1000);
273 mobile_action_interrupt_msg(data, NULL, 2);
274 mobile_action_interrupt_msg(data, msg[1], 1);
275 mobile_action_interrupt_msg(data, msg[2], 1);
276 mobile_action_interrupt_msg(data, msg[3], 63);
277 mobile_action_interrupt_msg(data, msg[4], 1);
278 mobile_action_interrupt_msg(data, msg[5], 1);
279 mobile_action_interrupt_msg(data, msg[6], 73);
280 mobile_action_interrupt_msg(data, msg[7], 1);
281 mobile_action_interrupt_msg(data, msg[8], 1);
282 }
283
284 static void handle_cisco(struct usbdev_data *data, struct blob_attr **tb)
285 {
286 static struct msg_entry msgs[] = {
287 {
288 "\x55\x53\x42\x43\xf8\x3b\xcd\x81\x00\x02\x00\x00\x80\x00\x0a\xfd"
289 "\x00\x00\x00\x03\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
290 }, {
291 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x02\x00\x00\x80\x00\x0a\xfd"
292 "\x00\x00\x00\x07\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
293 }, {
294 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x00\x00\x00\x00\x00\x0a\xfd"
295 "\x00\x01\x00\x07\x10\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
296 }, {
297 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x02\x00\x00\x80\x00\x0a\xfd"
298 "\x00\x02\x00\x23\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
299 }, {
300 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x00\x00\x00\x00\x00\x0a\xfd"
301 "\x00\x03\x00\x23\x82\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
302 }, {
303 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x02\x00\x00\x80\x00\x0a\xfd"
304 "\x00\x02\x00\x26\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
305 }, {
306 "\x55\x53\x42\x43\x98\x43\x00\x82\x00\x00\x00\x00\x00\x00\x0a\xfd"
307 "\x00\x03\x00\x26\xc8\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
308 }, {
309 "\x55\x53\x42\x43\xd8\x4c\x04\x82\x00\x02\x00\x00\x80\x00\x0a\xfd"
310 "\x00\x00\x10\x73\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
311 }, {
312 "\x55\x53\x42\x43\xd8\x4c\x04\x82\x00\x02\x00\x00\x80\x00\x0a\xfd"
313 "\x00\x02\x00\x24\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00", 31
314 }, {
315 "\x55\x53\x42\x43\xd8\x4c\x04\x82\x00\x00\x00\x00\x00\x00\x0a\xfd"
316 "\x00\x03\x00\x24\x13\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
317 }, {
318 "\x55\x53\x42\x43\xd8\x4c\x04\x82\x00\x00\x00\x00\x00\x00\x0a\xfd"
319 "\x00\x01\x10\x73\x24\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 31
320 }
321
322 };
323
324 detach_driver(data);
325 data->need_response = true;
326 send_messages(data, msgs, ARRAY_SIZE(msgs));
327 }
328
329 static void handle_mbim(struct usbdev_data *data, struct blob_attr **tb)
330 {
331 int j;
332
333 if (data->desc.bNumConfigurations < 2)
334 return;
335
336 for (j = 0; j < data->desc.bNumConfigurations; j++) {
337 struct libusb_config_descriptor *config;
338 int i;
339
340 libusb_get_config_descriptor(data->dev, j, &config);
341
342 for (i = 0; i < config->bNumInterfaces; i++) {
343 if (config->interface[i].altsetting[0].bInterfaceClass == 2) {
344 if (config->interface[i].altsetting[0].bInterfaceSubClass == 0x0e) {
345 struct libusb_config_descriptor *active;
346 int count = 5;
347
348 libusb_get_active_config_descriptor(data->dev, &active);
349 if (active->bConfigurationValue == config->bConfigurationValue)
350 return;
351 while ((libusb_set_configuration(data->devh, config->bConfigurationValue) < 0) && --count)
352 libusb_detach_kernel_driver(data->devh, active->interface[0].altsetting[0].bInterfaceNumber);
353
354 libusb_free_config_descriptor(config);
355 return;
356 }
357 }
358 }
359
360 libusb_free_config_descriptor(config);
361 }
362 }
363
364 static void set_alt_setting(struct usbdev_data *data, int setting)
365 {
366 if (libusb_claim_interface(data->devh, data->interface))
367 return;
368
369 libusb_set_interface_alt_setting(data->devh, data->interface, setting);
370 libusb_release_interface(data->devh, data->interface);
371 }
372
373 enum {
374 MODE_GENERIC,
375 MODE_HUAWEI,
376 MODE_HUAWEINEW,
377 MODE_SIERRA,
378 MODE_STDEJECT,
379 MODE_SONY,
380 MODE_QISDA,
381 MODE_GCT,
382 MODE_KOBIL,
383 MODE_SEQUANS,
384 MODE_MOBILE_ACTION,
385 MODE_CISCO,
386 MODE_MBIM,
387 __MODE_MAX
388 };
389
390 static const struct {
391 const char *name;
392 void (*cb)(struct usbdev_data *data, struct blob_attr **tb);
393 } modeswitch_cb[__MODE_MAX] = {
394 [MODE_GENERIC] = { "Generic", handle_generic },
395 [MODE_STDEJECT] = { "StandardEject", handle_standardeject },
396 [MODE_HUAWEI] = { "Huawei", handle_huawei },
397 [MODE_HUAWEINEW] = { "HuaweiNew", handle_huaweinew },
398 [MODE_SIERRA] = { "Sierra", handle_sierra },
399 [MODE_SONY] = { "Sony", handle_sony },
400 [MODE_QISDA] = { "Qisda", handle_qisda },
401 [MODE_GCT] = { "GCT", handle_gct },
402 [MODE_KOBIL] = { "Kobil", handle_kobil },
403 [MODE_SEQUANS] = { "Sequans", handle_sequans },
404 [MODE_MOBILE_ACTION] = { "MobileAction", handle_mobile_action },
405 [MODE_CISCO] = { "Cisco", handle_cisco },
406 [MODE_MBIM] = { "MBIM", handle_mbim },
407 };
408
409 void handle_switch(struct usbdev_data *data)
410 {
411 static const struct blobmsg_policy data_policy[__DATA_MAX] = {
412 [DATA_MODE] = { .name = "mode", .type = BLOBMSG_TYPE_STRING },
413 [DATA_MSG] = { .name = "msg", .type = BLOBMSG_TYPE_ARRAY },
414 [DATA_INTERFACE] = { .name = "interface", .type = BLOBMSG_TYPE_INT32 },
415 [DATA_MSG_EP] = { .name = "msg_endpoint", .type = BLOBMSG_TYPE_INT32 },
416 [DATA_RES_EP] = { .name = "response_endpoint", .type = BLOBMSG_TYPE_INT32 },
417 [DATA_RESPONSE] = { .name = "response", .type = BLOBMSG_TYPE_BOOL },
418 [DATA_CONFIG] = { .name = "config", .type = BLOBMSG_TYPE_INT32 },
419 [DATA_ALT] = { .name = "alt", .type = BLOBMSG_TYPE_INT32 },
420 [DATA_DEV_CLASS] = { .name = "t_class", .type = BLOBMSG_TYPE_INT32 },
421 };
422 struct blob_attr *tb[__DATA_MAX];
423 int mode = MODE_GENERIC;
424 int t_class = 0;
425
426 blobmsg_parse(data_policy, __DATA_MAX, tb, blobmsg_data(data->info), blobmsg_data_len(data->info));
427
428 if (tb[DATA_DEV_CLASS])
429 t_class = blobmsg_get_u32(tb[DATA_DEV_CLASS]);
430
431 if (tb[DATA_INTERFACE])
432 data->interface = blobmsg_get_u32(tb[DATA_INTERFACE]);
433
434 if (tb[DATA_MSG_EP])
435 data->msg_endpoint = blobmsg_get_u32(tb[DATA_MSG_EP]);
436
437 if (tb[DATA_RES_EP])
438 data->response_endpoint = blobmsg_get_u32(tb[DATA_RES_EP]);
439
440 if (tb[DATA_RELEASE_DELAY])
441 data->release_delay = blobmsg_get_u32(tb[DATA_RELEASE_DELAY]);
442
443 if (tb[DATA_RESPONSE])
444 data->need_response = blobmsg_get_bool(tb[DATA_RESPONSE]);
445
446 if (t_class > 0 && data->dev_class != t_class)
447 return;
448
449 if (tb[DATA_MODE]) {
450 const char *modestr;
451 int i;
452
453 modestr = blobmsg_data(tb[DATA_MODE]);
454 for (i = 0; i < __MODE_MAX; i++) {
455 if (strcmp(modeswitch_cb[i].name, modestr) != 0)
456 continue;
457
458 mode = i;
459 break;
460 }
461 }
462
463 modeswitch_cb[mode].cb(data, tb);
464
465 if (tb[DATA_CONFIG]) {
466 int config, config_new;
467
468 config_new = blobmsg_get_u32(tb[DATA_CONFIG]);
469 if (libusb_get_configuration(data->devh, &config) ||
470 config != config_new)
471 libusb_set_configuration(data->devh, config_new);
472 }
473
474 if (tb[DATA_ALT]) {
475 int new = blobmsg_get_u32(tb[DATA_ALT]);
476 set_alt_setting(data, new);
477 }
478 }