]> CyberLeo.Net >> Repos - SourceForge/eyefi-config.git/blob - eyefi-config.c
Added dependencies on eyefi-config.h
[SourceForge/eyefi-config.git] / eyefi-config.c
1 /*
2  * eyefi-config.c
3  *
4  * Copyright (C) 2008 Dave Hansen <dave@sr71.net>
5  *
6  * This software may be redistributed and/or modified under the terms of
7  * the GNU General Public License ("GPL") version 2 as published by the
8  * Free Software Foundation.
9  */
10
11 #include "eyefi-config.h"
12
13 int eyefi_debug_level = 1;
14
15 int eyefi_printf(const char *fmt, ...)
16 {
17         va_list args;
18         int r;
19
20         va_start(args, fmt);
21         r = vprintf(fmt, args);
22         va_end(args);
23
24         return r;
25 }
26  
27 static char *eyefi_file_name(enum eyefi_file file)
28 {
29         switch (file) {
30         case REQC: return "reqc";
31         case REQM: return "reqm";
32         case RSPC: return "rspc";
33         case RSPM: return "rspm";
34         }
35
36         return NULL;
37 }
38
39 char *eyefi_file_on(enum eyefi_file file, char *mnt)
40 {
41         char *filename = eyefi_file_name(file);
42         char *full = malloc(PATHNAME_MAX);
43         
44         if (!full)
45                 return NULL;
46
47         sprintf(&full[0], "%s/EyeFi/%s", mnt, filename);
48         debug_printf(4, "eyefile nr: %d on '%s' is: '%s'\n", file, mnt, &full[0]);
49         return full;
50 }
51
52 /*
53  * This lets us get away with a static allocation
54  * for the buffer.  We make it size*2 so that we're
55  * guaranteed to be able to get a "size" buffer
56  * aligned inside of the larger one.
57  */
58 static char unaligned_buf[EYEFI_BUF_SIZE*2];
59 static void *eyefi_buf;
60
61 void *eyefi_response(void)
62 {
63         return eyefi_buf;
64 }
65
66 void dumpbuf(const char *buffer, int bytesToWrite)
67 {
68     int i;
69     static char linebuf[500];
70
71     for (i=0; i < bytesToWrite; i += 16) {
72         char *tmpbuf = &linebuf[0];
73         unsigned long sum = 0;
74         int j;
75 #define lprintf(args...)        do {            \
76         tmpbuf += sprintf(tmpbuf, ## args);\
77 } while (0)
78
79         lprintf("[%03d]: ", i);
80         for (j=0; j < 16; j++) {
81                 u8 c = ((unsigned char *)buffer)[i+j];
82                 lprintf("%02x ", (unsigned int)c);
83                 sum += c;
84         }
85         lprintf(" |");
86         for (j=0; j < 16; j++) {
87                 u8 c = ((unsigned char *)buffer)[i+j];
88                 if (c >= 'a' && c <= 'z')
89                         lprintf("%c", c);
90                 else if (c >= 'A' && c <= 'Z')
91                         lprintf("%c", c);
92                 else if (c >= '0' && c <= '9')
93                         lprintf("%c", c);
94                 else if (c >= 0x20 && c <= 127)
95                         lprintf("%c", c);
96                 else
97                         lprintf(".");
98         }
99         lprintf("|\n");
100         if (sum == 0)
101                 continue;
102         printf("%s", linebuf);
103         //if (i > 200)
104         //      break;
105     }
106 }
107
108 void read_from(enum eyefi_file);
109 void write_to(enum eyefi_file, void *, int);
110
111 struct card_seq_num eyefi_seq;
112 struct card_seq_num read_seq_from(enum eyefi_file file)
113 {
114         struct card_seq_num *ret;
115         read_from(file);
116         ret = eyefi_buf;
117         return *ret;
118 }
119
120 /*
121  * For O_DIRECT writes to files, we need
122  * to be 512 byte aligned on Linux, I think.
123  * So, just align this to something big
124  * and be done with it.  FIXME :)
125  *
126  * This probably isn't necessary on chdk,
127  * since I don't think it buffers I/O at
128  * all.
129  */
130 void align_buf(void)
131 {
132         unsigned long addr = (unsigned long)&unaligned_buf[EYEFI_BUF_SIZE];
133         addr &= ~(EYEFI_BUF_SIZE-1);
134         eyefi_buf = (void *)addr;
135         debug_printf(4, "buf: %p\n", eyefi_buf);
136         debug_printf(4, "unaligned: %p\n", &unaligned_buf[0]);
137 }
138
139 /*
140  * The real manager does this so we might
141  * as well, too.
142  */
143 void zero_card_files(void)
144 {
145         char zbuf[EYEFI_BUF_SIZE];
146
147         memset(&zbuf[0], 0, EYEFI_BUF_SIZE);
148         //write_to(REQM, zbuf, EYEFI_BUF_SIZE);
149         write_to(REQC, zbuf, EYEFI_BUF_SIZE);
150         write_to(RSPM, zbuf, EYEFI_BUF_SIZE);
151         write_to(RSPC, zbuf, EYEFI_BUF_SIZE);
152
153         read_from(REQM);
154         read_from(REQC);
155         read_from(RSPM);
156         read_from(RSPC);
157 }
158
159 void init_card()
160 {
161         char *mnt;
162         if (eyefi_buf != NULL)
163                 return;
164
165         debug_printf(2, "Initializing card...\n");
166         mnt = locate_eyefi_mount();
167         if (mnt == NULL)
168                 return;
169
170         align_buf();
171         zero_card_files();
172         eyefi_seq = read_seq_from(RSPC);
173         if (eyefi_seq.seq == 0)
174                 eyefi_seq.seq = 0x1234;
175         eyefi_seq.seq++;
176         debug_printf(2, "Done initializing card...\n");
177         debug_printf(2, "seq was: %04lx\n", eyefi_seq.seq);
178 }
179
180 static char *eyefi_file(enum eyefi_file file)
181 {
182         init_card();
183         return eyefi_file_on(file, locate_eyefi_mount());
184 }
185
186 void read_from(enum eyefi_file __file)
187 {
188         int ret;
189         int fd;
190         char *file = eyefi_file(__file);
191         
192         init_card();
193
194 retry:
195         fd = open(file, O_RDONLY);
196         if (fd < 0)
197                 open_error(file, fd);
198         fd_flush(fd);
199         ret = read(fd, eyefi_buf, EYEFI_BUF_SIZE);
200         if (eyefi_debug_level > 3)
201                 dumpbuf(eyefi_buf, 128);
202         if (ret < 0) {
203                 close(fd);
204                 perror("bad read, retrying...");
205                 goto retry;
206                 exit(1);
207         }
208         debug_printf(4, "read '%s': bytes: %d\n", file, ret);
209         /*
210          * There was a time when I was carefully recording how each response
211          * looked, and I counted the zeros in each response.  I don't care
212          * any more.
213         u8 c;
214         int zeros = 0;
215         int i;
216         for (i=0; i < EYEFI_BUF_SIZE; i++) {
217                 c = ((char *)eyefi_buf)[i];
218                 if (c == '\0') {
219                         zeros++;
220                         continue;
221                 }
222         }
223         */
224         free(file);
225         close(fd);
226 }
227
228 int fake_write = 0;
229 void write_to(enum eyefi_file __file, void *stuff, int len)
230 {
231         int ret;
232         int fd;
233         char *file;
234
235         if (fake_write)
236                 return;
237
238         init_card();
239         file = eyefi_file(__file);
240         if (len == -1)
241                 len = strlen(stuff);
242
243         if (eyefi_debug_level > 3) {
244                 debug_printf(3, "%s('%s', ..., %d)\n", __func__, file, len);
245                 dumpbuf(stuff, len);
246         }
247         memset(eyefi_buf, 0, EYEFI_BUF_SIZE);
248         memcpy(eyefi_buf, stuff, len);
249         fd = open(file, O_RDWR|O_CREAT, 0600);
250         if (fd < 0 )
251                 open_error(file, fd);
252         if (eyefi_debug_level > 3)
253                 dumpbuf(eyefi_buf, 128);
254         ret = write(fd, eyefi_buf, EYEFI_BUF_SIZE);
255         if (ret < 0)
256                 open_error(file, ret);
257         ret = fd_flush(fd);
258         if (ret < 0)
259                 open_error(file, ret);
260         close(fd);
261         debug_printf(3, "wrote %d bytes to '%s' (string was %d bytes)\n", ret, file, len);
262         if (ret < 0) {
263                 fprintf(stderr, "error writing to '%s': ", file);
264                 perror("");
265                 exit(ret);
266         }
267         free(file);
268 }       
269
270 #define write_struct(file, s) write_to((file), s, sizeof(*(s)))
271
272 void inc_seq(void)
273 {
274         /*
275          * Oddly enough, the sequence number appears
276          * to be of normal endianness.
277          */
278         //u32 tmpseq = be32_to_u32(seq.seq);
279         //seq.seq = u32_to_be32(tmpseq+1);
280         eyefi_seq.seq++;
281         write_struct(REQC, &eyefi_seq);
282 }
283
284 u32 eyefi_current_seq(void)
285 {
286         return eyefi_seq.seq;
287 }
288
289 int wait_for_response(void)
290 {
291         int good_rsp = 0;
292         u32 rsp = 0;
293         int i;
294         debug_printf(3, "waiting for response...\n");
295         inc_seq();
296         for (i = 0; i < 50; i++) {
297                 struct card_seq_num cardseq = read_seq_from(RSPC);
298                 debug_printf(3, "read rsp code: %lx, looking for: %lx raw: %lx\n", rsp, eyefi_current_seq(),
299                                 cardseq.seq);
300                 rsp = cardseq.seq;
301                 if (rsp == eyefi_current_seq()) {
302                         good_rsp = 1;
303                         break;
304                 }
305                 if (eyefi_debug_level > 4) {
306                         read_from(REQM);
307                         debug_printf(1, "command issued was: '%c'\n", ((char *)eyefi_buf)[0]);
308                 }
309                 usleep(300000);
310         }
311         if (!good_rsp) {
312                 debug_printf(1, "never saw card seq response\n");
313                 return -1;
314         }
315         debug_printf(3, "got good seq (%ld), reading RSPM...\n", rsp);
316         read_from(RSPM);
317         debug_printf(3, "done reading RSPM\n");
318         return 0;
319 }
320
321 char *net_test_states[] = {
322        "not scanning",
323        "locating network",
324        "verifying network key",
325        "waiting for DHCP",
326        "testing connection to Eye-Fi server",
327        "success",
328 };
329
330 char *net_test_state_name(u8 state)
331 {
332         int size = ARRAY_SIZE(net_test_states);
333         if (state >= size)
334                 return "unknown";
335         return net_test_states[state];
336 }
337
338 const char *net_types[] = {
339         "none",
340         "WEP",
341         "WPA",
342         "unknown1",
343         "WPA2",
344 };
345 const char net_type_unknown[] = "unknown";
346
347 const char *net_type_name(u8 type)
348 {
349         int size = ARRAY_SIZE(net_types);
350         debug_printf(3, "%s(%d): '%s' size: %d\n", __func__, type, net_types[type], size);
351         if (type >= size)
352                 return net_type_unknown;
353         return net_types[type];
354 }
355
356 static char lower(char c)
357 {
358         if ((c >= 'A') && (c <= 'Z'))
359                 c += ('a' - 'A');
360         return c;
361 }
362
363 int atoh(char c)
364 {
365         char lc = lower(c);
366         if ((c >= '0') && (c <= '9'))
367                 return c - '0';
368         else if ((lc >= 'a') && (lc <= 'z'))
369                 return (lc - 'a') + 10;
370         debug_printf(5, "non-hex character: '%c'/'%c'\n", c, lc);
371         return -1;
372 }
373
374 /*
375  * Take a string like "0ab1" and make it
376  * a series of bytes: { 0x0a, 0xb1 }
377  *
378  * @len is the strlen() of the ascii
379  *
380  * Destroys the original string.
381  */
382 char *convert_ascii_to_hex(char *ascii, int len)
383 {
384         int i;
385         if (len%2) {
386                 fprintf(stderr, "%s() must be even number of bytes: %d\n",
387                 __func__, len);
388                 exit(2);
389         }
390         for (i=0; i < len; i+=2) {
391                 int high = atoh(ascii[i]);
392                 int low  = atoh(ascii[i+1]);
393                 u8 byte = (high<<4 | low);
394                 if (high < 0 || low < 0) {
395                         fprintf(stderr, "unable to parse hex string: '%s'\n", ascii);
396                         return NULL;
397                 }
398                 debug_printf(6, "high: %02x low: %02x, both: %02x\n", high, low, byte);
399                 ascii[i/2] = byte;
400         }
401         for (i=len/2; i < len; i++)
402                 ascii[i] = '\0';
403         return &ascii[0];
404 }
405
406 int make_network_key(struct network_key *key, char *essid, char *pass)
407 {
408         char tmp[WPA_KEY_BYTES+WEP_KEY_BYTES];
409         int pass_len = strlen(pass);
410         char *hex_pass;
411         memset(key, 0, sizeof(*key));
412
413         strcpy(&tmp[0], pass);
414         eyefi_printf(" interpreting passphrase as ");
415         switch (pass_len) {
416                 case WPA_KEY_BYTES*2:
417                         eyefi_printf("hex WPA");
418                         hex_pass = convert_ascii_to_hex(tmp, pass_len);
419                         if (!hex_pass)
420                                 return -EINVAL;
421                         key->len = pass_len/2;
422                         memcpy(&key->wpa.key[0], hex_pass, key->len);
423                         break;
424                 case WEP_KEY_BYTES*2:
425                         eyefi_printf("hex WEP");
426                         hex_pass = convert_ascii_to_hex(tmp, strlen(pass));
427                         if (!hex_pass)
428                                 return -EINVAL;
429                         key->len = pass_len/2;
430                         memcpy(&key->wep.key[0], hex_pass, key->len);
431                         break;
432                 default:
433                         eyefi_printf("ASCII WPA");
434                         pbkdf2_sha1(pass, essid, strlen(essid), 4096,
435                                     &key->wpa.key[0], WPA_KEY_BYTES);
436                         key->len = WPA_KEY_BYTES;
437                         break;
438         }
439         eyefi_printf(" key (%d bytes)\n", key->len);
440         assert(key->len != 0);
441         return 0;
442 }
443
444 int card_info_cmd(enum card_info_subcommand cmd)
445 {
446         struct card_info_req cir;
447         cir.o = 'o';
448         cir.subcommand = cmd;
449
450         write_struct(REQM, &cir);
451         return wait_for_response();
452 }
453
454 u32 fetch_log_length(void)
455 {
456         debug_printf(3, "%s()\n", __func__);
457         card_info_cmd(LOG_LEN);
458         struct card_info_log_len *loglen = eyefi_buf;
459         return be32_to_u32(loglen->val);
460 }
461
462 struct card_firmware_info *fetch_card_firmware_info(void)
463 {
464         debug_printf(2, "%s()\n", __func__);
465         card_info_cmd(FIRMWARE_INFO);
466         return (struct card_firmware_info *)eyefi_buf;
467         return NULL;
468 }
469
470 struct testbuf {
471         char cmd;
472         u8 l1;
473         char name[100];
474 };
475
476 struct z {
477         char zeros[100];
478 } z;
479
480
481 char fwbuf[1<<20];
482 char zbuf[1<<20];
483 void testit0(void)
484 {
485         char c;
486         struct testbuf tb;
487         int i;
488         int fdin;
489         int fdout;
490         
491         card_info_cmd(3);
492         printf("o3 result:\n");
493         dumpbuf(eyefi_buf, 64);
494
495         memset(&zbuf[0], 0, EYEFI_BUF_SIZE);
496         zbuf[0] = 'o';
497         zbuf[1] = 2;
498
499         write_to(REQM, &zbuf[0], 16384);
500         printf("o2 written\n");
501         printf("seq: %x\n", (int)eyefi_seq.seq);
502         inc_seq();
503
504         for (i=0; i < 4; i++) {
505                 read_from(RSPC);
506                 printf("RSPC %d:\n", i);
507                 dumpbuf(eyefi_buf, 64);
508                 usleep(20000);
509         }
510
511         printf("RSPM1:\n");
512         read_from(RSPM);
513         dumpbuf(eyefi_buf, 64);
514
515         memset(&zbuf[0], 0, EYEFI_BUF_SIZE);
516         write_to(RSPM, zbuf, EYEFI_BUF_SIZE);
517         write_to(REQM, zbuf, EYEFI_BUF_SIZE);
518
519         fdin = open("/home/dave/projects/eyefi/EYEFIFWU.BIN.2.0001", O_RDONLY);
520         perror("fdin");
521         fdout = open("/media/EYE-FI/EYEFIFWU.BIN", O_WRONLY|O_CREAT);
522         perror("fdout");
523         if (fdin <= 0 || fdout <= 0)
524                 exit(1);
525         fd_flush(fdin);
526         i = read(fdin, &fwbuf[0], 524288);
527         perror("read");
528         if (i != 524288)
529                 exit(2);
530         i = write(fdout, &fwbuf[0], 524288);
531         fd_flush(fdout);
532         perror("write");
533         if (i != 524288)
534                 exit(3);
535
536         printf("RSPM2:\n");
537         read_from(RSPM);
538         dumpbuf(eyefi_buf, 64);
539
540         reboot_card();
541         printf("after reboot:\n");
542         dumpbuf(eyefi_buf, 64);
543
544         printf("cic3:\n");
545         card_info_cmd(3);
546         dumpbuf(eyefi_buf, 64);
547
548         printf("cic2:\n");
549         card_info_cmd(2);
550         dumpbuf(eyefi_buf, 64);
551
552         memset(&zbuf[0], 0, EYEFI_BUF_SIZE);
553         write_to(RSPM, zbuf, EYEFI_BUF_SIZE);
554         write_to(REQM, zbuf, EYEFI_BUF_SIZE);
555         
556         printf("cic2v2:\n");
557         card_info_cmd(2);
558         dumpbuf(eyefi_buf, 64);
559
560         exit(0);
561         strcpy(tb.name, "www.sr71.net/");
562         tb.l1 = strlen(tb.name);
563         for (i = 0; i < 10; i++) {
564                 tb.cmd = 'O';
565                 tb.l1 = i;
566                 write_struct(RSPM, &z);
567                 write_struct(REQM, &tb);
568                 wait_for_response();
569                 printf("buffer after O %d:\n", i);
570                 dumpbuf(eyefi_buf, 64);
571                 printf("----------------\n");
572                 write_struct(REQM, &tb);
573                 card_info_cmd(i);
574                 printf("card info(%d):\n", i);
575                 dumpbuf(eyefi_buf, 64);
576                 printf("-----------\n");
577         }
578         return;
579
580         strcpy(tb.name, "/public/eyefi/servname");
581         strcpy(tb.name, "/config/networks.xml");
582         //tb.len = strlen(tb.name);
583         tb.l1 = 0;
584         for (c = 'O'; c <= 'O'; c++) {
585                 tb.cmd = c;
586                 write_struct(REQM, &tb);
587                 wait_for_response();
588                 printf("dumping buffer:\n");
589                 dumpbuf(eyefi_buf, 64);
590                 printf("buffer dump done\n");
591         }
592 }
593
594 struct card_info_rsp_key *fetch_card_key(void)
595 {
596         struct card_info_rsp_key *key;
597
598         debug_printf(2, "%s()\n", __func__);
599         card_info_cmd(CARD_KEY);
600         key = eyefi_buf;
601         return key;
602 }
603
604 int issue_noarg_command(u8 cmd)
605 {
606         struct noarg_request req;
607         debug_printf(4, "%s() cmd: %d\n", __func__, cmd);
608         req.req = cmd;
609         write_struct(REQM, &req);
610         return wait_for_response();
611 }
612
613 struct scanned_net_list *scan_nets(void)
614 {
615         struct scanned_net_list *scanned;
616
617         debug_printf(2, "%s()\n", __func__);
618         issue_noarg_command('g');
619         scanned = eyefi_response();
620         return scanned;
621 }
622
623 struct configured_net_list *fetch_configured_nets(void)
624 {
625         struct configured_net_list *configured;
626
627         debug_printf(2, "%s()\n", __func__);
628         issue_noarg_command('l');
629         configured = eyefi_buf;
630         return configured;
631 }
632
633 void reboot_card(void)
634 {
635         debug_printf(2, "%s()\n", __func__);
636         issue_noarg_command('b');
637 }
638
639 int network_action(char cmd, char *essid, char *ascii_password)
640 {
641         struct net_request nr;
642         memset(&nr, 0, sizeof(nr));
643
644         nr.req = cmd;
645         strcpy(&nr.essid[0], essid);
646         nr.essid_len = strlen(essid);
647
648         if (ascii_password) {
649                 int ret = make_network_key(&nr.key, essid, ascii_password);
650                 if (ret)
651                         return ret;
652         }
653         write_struct(REQM, &nr);
654         return wait_for_response();
655 }
656
657 void add_network(char *essid, char *ascii_password)
658 {
659         debug_printf(2, "%s('%s', '%s')\n", __func__, essid, ascii_password);
660         network_action('a', essid, ascii_password);
661 }
662
663 void remove_network(char *essid)
664 {
665         debug_printf(2, "%s()\n", __func__);
666         network_action('d', essid, NULL);
667 }
668
669 int get_log_at_offset(u32 offset)
670 {
671         struct fetch_log_cmd cmd;
672         cmd.m = 'm';
673         cmd.offset = u32_to_be32(offset);
674
675         debug_printf(2, "getting log at offset: %08lx\n", offset);
676         write_struct(REQM, &cmd);
677         return wait_for_response();
678 }
679
680 void add_log_piece(u8 *log, int log_len, u8 *piece, int piece_pos, int piece_size)
681 {
682         debug_printf(2, "%s(%p, %d, %p, %d, %d)\n", __func__, log, log_len, piece, piece_pos, piece_size);
683         if (piece_pos + piece_size > log_len) {
684                 int overflow_by = (piece_pos + piece_size) - log_len;
685                 int piece_overrun_pos = piece_size - overflow_by;
686                 piece_size -= overflow_by;
687                 memcpy(&log[0], &piece[piece_overrun_pos], overflow_by);
688                 debug_printf(2, "writing %d bytes to logbuf[0] from piece[%d]\n",
689                                 overflow_by, piece_overrun_pos);
690         }
691         debug_printf(2, "writing %d bytes to logbuf[%d]\n", piece_size, piece_pos);
692         memcpy(&log[piece_pos], piece, piece_size);
693 }
694
695 int get_log_into(u8 *resbuf)
696 {
697         int total_bytes = 0;
698         int ret;
699         int i;
700         u32 log_start;
701         u32 log_end;
702         u32 __log_size = fetch_log_length();
703         int log_pieces = __log_size/EYEFI_BUF_SIZE;
704
705         if (__log_size <= 0)
706                 return __log_size;
707
708         /* There are 8 bytes of header in the first log entry
709          * to specify where the log starts and ends */
710         u32 log_size = __log_size - 8;
711
712         for (i = 0; i < log_pieces; i++) {
713                 debug_printf(1, "fetching EyeFi card log part %d/%d...",
714                                 i+1, log_pieces);
715                 fflush(NULL);
716                 ret = get_log_at_offset(EYEFI_BUF_SIZE*i);
717                 debug_printf(1, "done\n");
718                 u8 *log_piece;
719                 u32 log_piece_size;
720                 if (i == 0) {
721                         struct first_log_response *log = eyefi_buf;
722                         log_end = log_size - be32_to_u32(log->log_end);
723                         log_start = log_size - be32_to_u32(log->log_start);
724                         debug_printf(2, "log end:   0x%04lx\n", log_end);
725                         debug_printf(2, "log start: 0x%04lx\n", log_start);
726                         log_piece = &log->data[0];
727                         log_piece_size = ARRAY_SIZE(log->data);
728                 } else {
729                         struct rest_log_response *log = eyefi_buf;
730                         log_piece = &log->data[0];
731                         log_piece_size = ARRAY_SIZE(log->data);
732                 }
733                 add_log_piece(resbuf, log_size, log_piece, log_start, log_piece_size);
734                 total_bytes += log_piece_size;
735                 log_start += log_piece_size;
736                 log_start = log_start % log_size;
737         }
738         return total_bytes;
739 }
740