aebf78151e4935629296b56967bd3d9b8942713f
[hmcfgusb] / flash-hmcfgusb.c
1 /* (not working) flasher for HM-CFG-USB
2 *
3 * Copyright (c) 2013 Michael Gernoth <michael@gernoth.net>
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a copy
6 * of this software and associated documentation files (the "Software"), to
7 * deal in the Software without restriction, including without limitation the
8 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
9 * sell copies of the Software, and to permit persons to whom the Software is
10 * furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice shall be included in
13 * all copies or substantial portions of the Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
21 * IN THE SOFTWARE.
22 */
23
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <unistd.h>
27 #include <stdint.h>
28 #include <string.h>
29 #include <strings.h>
30 #include <poll.h>
31 #include <errno.h>
32 #include <sys/types.h>
33 #include <sys/stat.h>
34 #include <fcntl.h>
35 #include <sys/time.h>
36 #include <libusb-1.0/libusb.h>
37
38 #include "hexdump.h"
39 #include "hmcfgusb.h"
40
41 struct recv_data {
42 int ack;
43 };
44
45 static int parse_hmcfgusb(uint8_t *buf, int buf_len, void *data)
46 {
47 struct recv_data *rdata = data;
48
49 if (buf_len != 1)
50 return 1;
51
52 rdata->ack = buf[0];
53
54 return 1;
55 }
56
57 static uint8_t ascii_to_nibble(uint8_t a)
58 {
59 uint8_t c = 0x00;
60
61 if ((a >= '0') && (a <= '9')) {
62 c = a - '0';
63 } else if ((a >= 'A') && (a <= 'F')) {
64 c = (a - 'A') + 10;
65 } else if ((a >= 'a') && (a <= 'f')) {
66 c = (a - 'a') + 10;
67 }
68
69 return c;
70 }
71
72 int main(int argc, char **argv)
73 {
74 struct hmcfgusb_dev *dev;
75 struct recv_data rdata;
76 uint8_t out[4096];
77 uint8_t buf[4096];
78 uint8_t *outp;
79 int fd;
80 int pfd;
81 int r;
82 int i;
83 int cnt;
84 int pkt;
85 int debug = 0;
86
87 hmcfgusb_set_debug(debug);
88
89 memset(&rdata, 0, sizeof(rdata));
90
91 fd = open("hmusbif.enc", O_RDONLY);
92 if (fd < 0) {
93 perror("Can't open hmusbif.enc");
94 exit(EXIT_FAILURE);
95 }
96
97 dev = hmcfgusb_init(parse_hmcfgusb, &rdata);
98 if (!dev) {
99 fprintf(stderr, "Can't initialize HM-CFG-USB\n");
100 exit(EXIT_FAILURE);
101 }
102
103 if (!dev->bootloader) {
104 fprintf(stderr, "HM-CFG-USB not in bootloader mode, entering bootloader.\n");
105 hmcfgusb_enter_bootloader(dev);
106 fprintf(stderr, "\nWaiting for device to reappear...\n");
107
108 do {
109 sleep(1);
110 } while ((dev = hmcfgusb_init(parse_hmcfgusb, &rdata)) == NULL);
111
112 if (!dev->bootloader) {
113 fprintf(stderr, "Can't enter bootloader, giving up!\n");
114 exit(EXIT_FAILURE);
115 }
116 }
117
118 printf("HM-CFG-USB opened!\n");
119
120 cnt = 0;
121 pkt = 0;
122 do {
123 int len;
124
125 memset(buf, 0, sizeof(buf));
126 r = read(fd, buf, 4);
127 if (r < 0) {
128 perror("read");
129 exit(EXIT_FAILURE);
130 } else if (r == 0) {
131 break;
132 } else if (r != 4) {
133 printf("can't get length information!\n");
134 exit(EXIT_FAILURE);
135 }
136
137 len = (ascii_to_nibble(buf[0]) & 0xf)<< 4;
138 len |= ascii_to_nibble(buf[1]) & 0xf;
139 len <<= 8;
140 len |= (ascii_to_nibble(buf[2]) & 0xf)<< 4;
141 len |= ascii_to_nibble(buf[3]) & 0xf;
142
143 r = read(fd, buf, len * 2);
144 if (r < 0) {
145 perror("read");
146 exit(EXIT_FAILURE);
147 } else if (r < len * 2) {
148 printf("short read, aborting (%d < %d)\n", r, len * 2);
149 break;
150 }
151
152 memset(out, 0, sizeof(out));
153 outp = out;
154 *outp++ = (pkt >> 8) & 0xff;
155 *outp++ = pkt & 0xff;
156 *outp++ = (len >> 8) & 0xff;
157 *outp++ = len & 0xff;
158 for (i = 0; i < r; i+=2) {
159 *outp = (ascii_to_nibble(buf[i]) & 0xf)<< 4;
160 *outp |= ascii_to_nibble(buf[i+1]) & 0xf;
161 outp++;
162 }
163 cnt = outp - out;
164 printf("Flashing %d bytes...\n", cnt);
165 if (debug)
166 hexdump(out, cnt, "F> ");
167
168 rdata.ack = 0;
169 if (!hmcfgusb_send(dev, out, cnt, 0)) {
170 perror("hmcfgusb_send");
171 exit(EXIT_FAILURE);
172 }
173
174 printf("Waiting for ack...\n");
175 do {
176 errno = 0;
177 pfd = hmcfgusb_poll(dev, 1);
178 if ((pfd < 0) && errno) {
179 perror("hmcfgusb_poll");
180 exit(EXIT_FAILURE);
181 }
182 if (rdata.ack) {
183 break;
184 }
185 } while (pfd < 0);
186
187 if (rdata.ack == 2) {
188 printf("Firmware update successfull!\n");
189 break;
190 }
191 pkt++;
192 } while (r > 0);
193
194 hmcfgusb_close(dev);
195
196 return EXIT_SUCCESS;
197 }
Impressum, Datenschutz