]> git.sur5r.net Git - u-boot/blob - common/xyzModem.c
riscv: checkpatch: Fix Macro argument reuse
[u-boot] / common / xyzModem.c
1 /*
2  *==========================================================================
3  *
4  *      xyzModem.c
5  *
6  *      RedBoot stream handler for xyzModem protocol
7  *
8  *==========================================================================
9  * SPDX-License-Identifier:     eCos-2.0
10  *==========================================================================
11  *#####DESCRIPTIONBEGIN####
12  *
13  * Author(s):    gthomas
14  * Contributors: gthomas, tsmith, Yoshinori Sato
15  * Date:         2000-07-14
16  * Purpose:
17  * Description:
18  *
19  * This code is part of RedBoot (tm).
20  *
21  *####DESCRIPTIONEND####
22  *
23  *==========================================================================
24  */
25 #include <common.h>
26 #include <xyzModem.h>
27 #include <stdarg.h>
28 #include <crc.h>
29
30 /* Assumption - run xyzModem protocol over the console port */
31
32 /* Values magic to the protocol */
33 #define SOH 0x01
34 #define STX 0x02
35 #define EOT 0x04
36 #define ACK 0x06
37 #define BSP 0x08
38 #define NAK 0x15
39 #define CAN 0x18
40 #define EOF 0x1A                /* ^Z for DOS officionados */
41
42 /* Data & state local to the protocol */
43 static struct
44 {
45   int *__chan;
46   unsigned char pkt[1024], *bufp;
47   unsigned char blk, cblk, crc1, crc2;
48   unsigned char next_blk;       /* Expected block */
49   int len, mode, total_retries;
50   int total_SOH, total_STX, total_CAN;
51   bool crc_mode, at_eof, tx_ack;
52   unsigned long file_length, read_length;
53 } xyz;
54
55 #define xyzModem_CHAR_TIMEOUT            2000   /* 2 seconds */
56 #define xyzModem_MAX_RETRIES             20
57 #define xyzModem_MAX_RETRIES_WITH_CRC    10
58 #define xyzModem_CAN_COUNT                3     /* Wait for 3 CAN before quitting */
59
60
61 typedef int cyg_int32;
62 static int
63 CYGACC_COMM_IF_GETC_TIMEOUT (char chan, char *c)
64 {
65
66   ulong now = get_timer(0);
67   while (!tstc ())
68     {
69       if (get_timer(now) > xyzModem_CHAR_TIMEOUT)
70         break;
71     }
72   if (tstc ())
73     {
74       *c = getc ();
75       return 1;
76     }
77   return 0;
78 }
79
80 static void
81 CYGACC_COMM_IF_PUTC (char x, char y)
82 {
83   putc (y);
84 }
85
86 /* Validate a hex character */
87 __inline__ static bool
88 _is_hex (char c)
89 {
90   return (((c >= '0') && (c <= '9')) ||
91           ((c >= 'A') && (c <= 'F')) || ((c >= 'a') && (c <= 'f')));
92 }
93
94 /* Convert a single hex nibble */
95 __inline__ static int
96 _from_hex (char c)
97 {
98   int ret = 0;
99
100   if ((c >= '0') && (c <= '9'))
101     {
102       ret = (c - '0');
103     }
104   else if ((c >= 'a') && (c <= 'f'))
105     {
106       ret = (c - 'a' + 0x0a);
107     }
108   else if ((c >= 'A') && (c <= 'F'))
109     {
110       ret = (c - 'A' + 0x0A);
111     }
112   return ret;
113 }
114
115 /* Convert a character to lower case */
116 __inline__ static char
117 _tolower (char c)
118 {
119   if ((c >= 'A') && (c <= 'Z'))
120     {
121       c = (c - 'A') + 'a';
122     }
123   return c;
124 }
125
126 /* Parse (scan) a number */
127 static bool
128 parse_num (char *s, unsigned long *val, char **es, char *delim)
129 {
130   bool first = true;
131   int radix = 10;
132   char c;
133   unsigned long result = 0;
134   int digit;
135
136   while (*s == ' ')
137     s++;
138   while (*s)
139     {
140       if (first && (s[0] == '0') && (_tolower (s[1]) == 'x'))
141         {
142           radix = 16;
143           s += 2;
144         }
145       first = false;
146       c = *s++;
147       if (_is_hex (c) && ((digit = _from_hex (c)) < radix))
148         {
149           /* Valid digit */
150           result = (result * radix) + digit;
151         }
152       else
153         {
154           if (delim != (char *) 0)
155             {
156               /* See if this character is one of the delimiters */
157               char *dp = delim;
158               while (*dp && (c != *dp))
159                 dp++;
160               if (*dp)
161                 break;          /* Found a good delimiter */
162             }
163           return false;         /* Malformatted number */
164         }
165     }
166   *val = result;
167   if (es != (char **) 0)
168     {
169       *es = s;
170     }
171   return true;
172 }
173
174
175 #ifdef DEBUG
176 /*
177  * Note: this debug setup works by storing the strings in a fixed buffer
178  */
179 static char zm_debug_buf[8192];
180 static char *zm_out = zm_debug_buf;
181 static char *zm_out_start = zm_debug_buf;
182
183 static int
184 zm_dprintf (char *fmt, ...)
185 {
186   int len;
187   va_list args;
188
189   va_start (args, fmt);
190   len = diag_vsprintf (zm_out, fmt, args);
191   zm_out += len;
192   return len;
193 }
194
195 static void
196 zm_flush (void)
197 {
198   zm_out = zm_out_start;
199 }
200
201 static void
202 zm_dump_buf (void *buf, int len)
203 {
204
205 }
206
207 static unsigned char zm_buf[2048];
208 static unsigned char *zm_bp;
209
210 static void
211 zm_new (void)
212 {
213   zm_bp = zm_buf;
214 }
215
216 static void
217 zm_save (unsigned char c)
218 {
219   *zm_bp++ = c;
220 }
221
222 static void
223 zm_dump (int line)
224 {
225   zm_dprintf ("Packet at line: %d\n", line);
226   zm_dump_buf (zm_buf, zm_bp - zm_buf);
227 }
228
229 #define ZM_DEBUG(x) x
230 #else
231 #define ZM_DEBUG(x)
232 #endif
233
234 /* Wait for the line to go idle */
235 static void
236 xyzModem_flush (void)
237 {
238   int res;
239   char c;
240   while (true)
241     {
242       res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, &c);
243       if (!res)
244         return;
245     }
246 }
247
248 static int
249 xyzModem_get_hdr (void)
250 {
251   char c;
252   int res;
253   bool hdr_found = false;
254   int i, can_total, hdr_chars;
255   unsigned short cksum;
256
257   ZM_DEBUG (zm_new ());
258   /* Find the start of a header */
259   can_total = 0;
260   hdr_chars = 0;
261
262   if (xyz.tx_ack)
263     {
264       CYGACC_COMM_IF_PUTC (*xyz.__chan, ACK);
265       xyz.tx_ack = false;
266     }
267   while (!hdr_found)
268     {
269       res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, &c);
270       ZM_DEBUG (zm_save (c));
271       if (res)
272         {
273           hdr_chars++;
274           switch (c)
275             {
276             case SOH:
277               xyz.total_SOH++;
278             case STX:
279               if (c == STX)
280                 xyz.total_STX++;
281               hdr_found = true;
282               break;
283             case CAN:
284               xyz.total_CAN++;
285               ZM_DEBUG (zm_dump (__LINE__));
286               if (++can_total == xyzModem_CAN_COUNT)
287                 {
288                   return xyzModem_cancel;
289                 }
290               else
291                 {
292                   /* Wait for multiple CAN to avoid early quits */
293                   break;
294                 }
295             case EOT:
296               /* EOT only supported if no noise */
297               if (hdr_chars == 1)
298                 {
299                   CYGACC_COMM_IF_PUTC (*xyz.__chan, ACK);
300                   ZM_DEBUG (zm_dprintf ("ACK on EOT #%d\n", __LINE__));
301                   ZM_DEBUG (zm_dump (__LINE__));
302                   return xyzModem_eof;
303                 }
304             default:
305               /* Ignore, waiting for start of header */
306               ;
307             }
308         }
309       else
310         {
311           /* Data stream timed out */
312           xyzModem_flush ();    /* Toss any current input */
313           ZM_DEBUG (zm_dump (__LINE__));
314           CYGACC_CALL_IF_DELAY_US ((cyg_int32) 250000);
315           return xyzModem_timeout;
316         }
317     }
318
319   /* Header found, now read the data */
320   res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, (char *) &xyz.blk);
321   ZM_DEBUG (zm_save (xyz.blk));
322   if (!res)
323     {
324       ZM_DEBUG (zm_dump (__LINE__));
325       return xyzModem_timeout;
326     }
327   res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, (char *) &xyz.cblk);
328   ZM_DEBUG (zm_save (xyz.cblk));
329   if (!res)
330     {
331       ZM_DEBUG (zm_dump (__LINE__));
332       return xyzModem_timeout;
333     }
334   xyz.len = (c == SOH) ? 128 : 1024;
335   xyz.bufp = xyz.pkt;
336   for (i = 0; i < xyz.len; i++)
337     {
338       res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, &c);
339       ZM_DEBUG (zm_save (c));
340       if (res)
341         {
342           xyz.pkt[i] = c;
343         }
344       else
345         {
346           ZM_DEBUG (zm_dump (__LINE__));
347           return xyzModem_timeout;
348         }
349     }
350   res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, (char *) &xyz.crc1);
351   ZM_DEBUG (zm_save (xyz.crc1));
352   if (!res)
353     {
354       ZM_DEBUG (zm_dump (__LINE__));
355       return xyzModem_timeout;
356     }
357   if (xyz.crc_mode)
358     {
359       res = CYGACC_COMM_IF_GETC_TIMEOUT (*xyz.__chan, (char *) &xyz.crc2);
360       ZM_DEBUG (zm_save (xyz.crc2));
361       if (!res)
362         {
363           ZM_DEBUG (zm_dump (__LINE__));
364           return xyzModem_timeout;
365         }
366     }
367   ZM_DEBUG (zm_dump (__LINE__));
368   /* Validate the message */
369   if ((xyz.blk ^ xyz.cblk) != (unsigned char) 0xFF)
370     {
371       ZM_DEBUG (zm_dprintf
372                 ("Framing error - blk: %x/%x/%x\n", xyz.blk, xyz.cblk,
373                  (xyz.blk ^ xyz.cblk)));
374       ZM_DEBUG (zm_dump_buf (xyz.pkt, xyz.len));
375       xyzModem_flush ();
376       return xyzModem_frame;
377     }
378   /* Verify checksum/CRC */
379   if (xyz.crc_mode)
380     {
381       cksum = crc16_ccitt(0, xyz.pkt, xyz.len);
382       if (cksum != ((xyz.crc1 << 8) | xyz.crc2))
383         {
384           ZM_DEBUG (zm_dprintf ("CRC error - recvd: %02x%02x, computed: %x\n",
385                                 xyz.crc1, xyz.crc2, cksum & 0xFFFF));
386           return xyzModem_cksum;
387         }
388     }
389   else
390     {
391       cksum = 0;
392       for (i = 0; i < xyz.len; i++)
393         {
394           cksum += xyz.pkt[i];
395         }
396       if (xyz.crc1 != (cksum & 0xFF))
397         {
398           ZM_DEBUG (zm_dprintf
399                     ("Checksum error - recvd: %x, computed: %x\n", xyz.crc1,
400                      cksum & 0xFF));
401           return xyzModem_cksum;
402         }
403     }
404   /* If we get here, the message passes [structural] muster */
405   return 0;
406 }
407
408 int
409 xyzModem_stream_open (connection_info_t * info, int *err)
410 {
411   int stat = 0;
412   int retries = xyzModem_MAX_RETRIES;
413   int crc_retries = xyzModem_MAX_RETRIES_WITH_CRC;
414
415 /*    ZM_DEBUG(zm_out = zm_out_start); */
416 #ifdef xyzModem_zmodem
417   if (info->mode == xyzModem_zmodem)
418     {
419       *err = xyzModem_noZmodem;
420       return -1;
421     }
422 #endif
423
424 /* TODO: CHECK ! */
425   int dummy = 0;
426   xyz.__chan = &dummy;
427   xyz.len = 0;
428   xyz.crc_mode = true;
429   xyz.at_eof = false;
430   xyz.tx_ack = false;
431   xyz.mode = info->mode;
432   xyz.total_retries = 0;
433   xyz.total_SOH = 0;
434   xyz.total_STX = 0;
435   xyz.total_CAN = 0;
436   xyz.read_length = 0;
437   xyz.file_length = 0;
438
439   CYGACC_COMM_IF_PUTC (*xyz.__chan, (xyz.crc_mode ? 'C' : NAK));
440
441   if (xyz.mode == xyzModem_xmodem)
442     {
443       /* X-modem doesn't have an information header - exit here */
444       xyz.next_blk = 1;
445       return 0;
446     }
447
448   while (retries-- > 0)
449     {
450       stat = xyzModem_get_hdr ();
451       if (stat == 0)
452         {
453           /* Y-modem file information header */
454           if (xyz.blk == 0)
455             {
456               /* skip filename */
457               while (*xyz.bufp++);
458               /* get the length */
459               parse_num ((char *) xyz.bufp, &xyz.file_length, NULL, " ");
460               /* The rest of the file name data block quietly discarded */
461               xyz.tx_ack = true;
462             }
463           xyz.next_blk = 1;
464           xyz.len = 0;
465           return 0;
466         }
467       else if (stat == xyzModem_timeout)
468         {
469           if (--crc_retries <= 0)
470             xyz.crc_mode = false;
471           CYGACC_CALL_IF_DELAY_US (5 * 100000); /* Extra delay for startup */
472           CYGACC_COMM_IF_PUTC (*xyz.__chan, (xyz.crc_mode ? 'C' : NAK));
473           xyz.total_retries++;
474           ZM_DEBUG (zm_dprintf ("NAK (%d)\n", __LINE__));
475         }
476       if (stat == xyzModem_cancel)
477         {
478           break;
479         }
480     }
481   *err = stat;
482   ZM_DEBUG (zm_flush ());
483   return -1;
484 }
485
486 int
487 xyzModem_stream_read (char *buf, int size, int *err)
488 {
489   int stat, total, len;
490   int retries;
491
492   total = 0;
493   stat = xyzModem_cancel;
494   /* Try and get 'size' bytes into the buffer */
495   while (!xyz.at_eof && (size > 0))
496     {
497       if (xyz.len == 0)
498         {
499           retries = xyzModem_MAX_RETRIES;
500           while (retries-- > 0)
501             {
502               stat = xyzModem_get_hdr ();
503               if (stat == 0)
504                 {
505                   if (xyz.blk == xyz.next_blk)
506                     {
507                       xyz.tx_ack = true;
508                       ZM_DEBUG (zm_dprintf
509                                 ("ACK block %d (%d)\n", xyz.blk, __LINE__));
510                       xyz.next_blk = (xyz.next_blk + 1) & 0xFF;
511
512                       if (xyz.mode == xyzModem_xmodem || xyz.file_length == 0)
513                         {
514                           /* Data blocks can be padded with ^Z (EOF) characters */
515                           /* This code tries to detect and remove them */
516                           if ((xyz.bufp[xyz.len - 1] == EOF) &&
517                               (xyz.bufp[xyz.len - 2] == EOF) &&
518                               (xyz.bufp[xyz.len - 3] == EOF))
519                             {
520                               while (xyz.len
521                                      && (xyz.bufp[xyz.len - 1] == EOF))
522                                 {
523                                   xyz.len--;
524                                 }
525                             }
526                         }
527
528                       /*
529                        * See if accumulated length exceeds that of the file.
530                        * If so, reduce size (i.e., cut out pad bytes)
531                        * Only do this for Y-modem (and Z-modem should it ever
532                        * be supported since it can fall back to Y-modem mode).
533                        */
534                       if (xyz.mode != xyzModem_xmodem && 0 != xyz.file_length)
535                         {
536                           xyz.read_length += xyz.len;
537                           if (xyz.read_length > xyz.file_length)
538                             {
539                               xyz.len -= (xyz.read_length - xyz.file_length);
540                             }
541                         }
542                       break;
543                     }
544                   else if (xyz.blk == ((xyz.next_blk - 1) & 0xFF))
545                     {
546                       /* Just re-ACK this so sender will get on with it */
547                       CYGACC_COMM_IF_PUTC (*xyz.__chan, ACK);
548                       continue; /* Need new header */
549                     }
550                   else
551                     {
552                       stat = xyzModem_sequence;
553                     }
554                 }
555               if (stat == xyzModem_cancel)
556                 {
557                   break;
558                 }
559               if (stat == xyzModem_eof)
560                 {
561                   CYGACC_COMM_IF_PUTC (*xyz.__chan, ACK);
562                   ZM_DEBUG (zm_dprintf ("ACK (%d)\n", __LINE__));
563                   if (xyz.mode == xyzModem_ymodem)
564                     {
565                       CYGACC_COMM_IF_PUTC (*xyz.__chan,
566                                            (xyz.crc_mode ? 'C' : NAK));
567                       xyz.total_retries++;
568                       ZM_DEBUG (zm_dprintf ("Reading Final Header\n"));
569                       stat = xyzModem_get_hdr ();
570                       CYGACC_COMM_IF_PUTC (*xyz.__chan, ACK);
571                       ZM_DEBUG (zm_dprintf ("FINAL ACK (%d)\n", __LINE__));
572                     }
573                   xyz.at_eof = true;
574                   break;
575                 }
576               CYGACC_COMM_IF_PUTC (*xyz.__chan, (xyz.crc_mode ? 'C' : NAK));
577               xyz.total_retries++;
578               ZM_DEBUG (zm_dprintf ("NAK (%d)\n", __LINE__));
579             }
580           if (stat < 0)
581             {
582               *err = stat;
583               xyz.len = -1;
584               return total;
585             }
586         }
587       /* Don't "read" data from the EOF protocol package */
588       if (!xyz.at_eof)
589         {
590           len = xyz.len;
591           if (size < len)
592             len = size;
593           memcpy (buf, xyz.bufp, len);
594           size -= len;
595           buf += len;
596           total += len;
597           xyz.len -= len;
598           xyz.bufp += len;
599         }
600     }
601   return total;
602 }
603
604 void
605 xyzModem_stream_close (int *err)
606 {
607   diag_printf
608     ("xyzModem - %s mode, %d(SOH)/%d(STX)/%d(CAN) packets, %d retries\n",
609      xyz.crc_mode ? "CRC" : "Cksum", xyz.total_SOH, xyz.total_STX,
610      xyz.total_CAN, xyz.total_retries);
611   ZM_DEBUG (zm_flush ());
612 }
613
614 /* Need to be able to clean out the input buffer, so have to take the */
615 /* getc */
616 void
617 xyzModem_stream_terminate (bool abort, int (*getc) (void))
618 {
619   int c;
620
621   if (abort)
622     {
623       ZM_DEBUG (zm_dprintf ("!!!! TRANSFER ABORT !!!!\n"));
624       switch (xyz.mode)
625         {
626         case xyzModem_xmodem:
627         case xyzModem_ymodem:
628           /* The X/YMODEM Spec seems to suggest that multiple CAN followed by an equal */
629           /* number of Backspaces is a friendly way to get the other end to abort. */
630           CYGACC_COMM_IF_PUTC (*xyz.__chan, CAN);
631           CYGACC_COMM_IF_PUTC (*xyz.__chan, CAN);
632           CYGACC_COMM_IF_PUTC (*xyz.__chan, CAN);
633           CYGACC_COMM_IF_PUTC (*xyz.__chan, CAN);
634           CYGACC_COMM_IF_PUTC (*xyz.__chan, BSP);
635           CYGACC_COMM_IF_PUTC (*xyz.__chan, BSP);
636           CYGACC_COMM_IF_PUTC (*xyz.__chan, BSP);
637           CYGACC_COMM_IF_PUTC (*xyz.__chan, BSP);
638           /* Now consume the rest of what's waiting on the line. */
639           ZM_DEBUG (zm_dprintf ("Flushing serial line.\n"));
640           xyzModem_flush ();
641           xyz.at_eof = true;
642           break;
643 #ifdef xyzModem_zmodem
644         case xyzModem_zmodem:
645           /* Might support it some day I suppose. */
646 #endif
647           break;
648         }
649     }
650   else
651     {
652       ZM_DEBUG (zm_dprintf ("Engaging cleanup mode...\n"));
653       /*
654        * Consume any trailing crap left in the inbuffer from
655        * previous received blocks. Since very few files are an exact multiple
656        * of the transfer block size, there will almost always be some gunk here.
657        * If we don't eat it now, RedBoot will think the user typed it.
658        */
659       ZM_DEBUG (zm_dprintf ("Trailing gunk:\n"));
660       while ((c = (*getc) ()) > -1)
661         ;
662       ZM_DEBUG (zm_dprintf ("\n"));
663       /*
664        * Make a small delay to give terminal programs like minicom
665        * time to get control again after their file transfer program
666        * exits.
667        */
668       CYGACC_CALL_IF_DELAY_US ((cyg_int32) 250000);
669     }
670 }
671
672 char *
673 xyzModem_error (int err)
674 {
675   switch (err)
676     {
677     case xyzModem_access:
678       return "Can't access file";
679       break;
680     case xyzModem_noZmodem:
681       return "Sorry, zModem not available yet";
682       break;
683     case xyzModem_timeout:
684       return "Timed out";
685       break;
686     case xyzModem_eof:
687       return "End of file";
688       break;
689     case xyzModem_cancel:
690       return "Cancelled";
691       break;
692     case xyzModem_frame:
693       return "Invalid framing";
694       break;
695     case xyzModem_cksum:
696       return "CRC/checksum error";
697       break;
698     case xyzModem_sequence:
699       return "Block sequence error";
700       break;
701     default:
702       return "Unknown error";
703       break;
704     }
705 }
706
707 /*
708  * RedBoot interface
709  */