xyz-modem: Close stream after processing/sending terminate sequence
[platform/kernel/u-boot.git] / cmd / load.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000-2004
4  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5  */
6
7 /*
8  * Serial up- and download support
9  */
10 #include <common.h>
11 #include <command.h>
12 #include <console.h>
13 #include <cpu_func.h>
14 #include <efi_loader.h>
15 #include <env.h>
16 #include <exports.h>
17 #include <flash.h>
18 #include <image.h>
19 #include <mapmem.h>
20 #include <net.h>
21 #include <s_record.h>
22 #include <serial.h>
23 #include <xyzModem.h>
24 #include <asm/cache.h>
25 #include <asm/global_data.h>
26 #include <linux/delay.h>
27
28 DECLARE_GLOBAL_DATA_PTR;
29
30 #if defined(CONFIG_CMD_LOADB)
31 static ulong load_serial_ymodem(ulong offset, int mode);
32 #endif
33
34 #if defined(CONFIG_CMD_LOADS)
35 static ulong load_serial(long offset);
36 static int read_record(char *buf, ulong len);
37 # if defined(CONFIG_CMD_SAVES)
38 static int save_serial(ulong offset, ulong size);
39 static int write_record(char *buf);
40 #endif
41
42 static int do_echo = 1;
43 #endif
44
45 /* -------------------------------------------------------------------- */
46
47 #if defined(CONFIG_CMD_LOADS)
48 static int do_load_serial(struct cmd_tbl *cmdtp, int flag, int argc,
49                           char *const argv[])
50 {
51         long offset = 0;
52         ulong addr;
53         int i;
54         char *env_echo;
55         int rcode = 0;
56 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
57         int load_baudrate, current_baudrate;
58
59         load_baudrate = current_baudrate = gd->baudrate;
60 #endif
61
62         env_echo = env_get("loads_echo");
63         if (env_echo && *env_echo == '1')
64                 do_echo = 1;
65         else
66                 do_echo = 0;
67
68 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
69         if (argc >= 2) {
70                 offset = simple_strtol(argv[1], NULL, 16);
71         }
72         if (argc == 3) {
73                 load_baudrate = (int)dectoul(argv[2], NULL);
74
75                 /* default to current baudrate */
76                 if (load_baudrate == 0)
77                         load_baudrate = current_baudrate;
78         }
79         if (load_baudrate != current_baudrate) {
80                 printf("## Switch baudrate to %d bps and press ENTER ...\n",
81                         load_baudrate);
82                 udelay(50000);
83                 gd->baudrate = load_baudrate;
84                 serial_setbrg();
85                 udelay(50000);
86                 for (;;) {
87                         if (getchar() == '\r')
88                                 break;
89                 }
90         }
91 #else   /* ! CONFIG_SYS_LOADS_BAUD_CHANGE */
92         if (argc == 2) {
93                 offset = simple_strtol(argv[1], NULL, 16);
94         }
95 #endif  /* CONFIG_SYS_LOADS_BAUD_CHANGE */
96
97         printf("## Ready for S-Record download ...\n");
98
99         addr = load_serial(offset);
100
101         /*
102          * Gather any trailing characters (for instance, the ^D which
103          * is sent by 'cu' after sending a file), and give the
104          * box some time (100 * 1 ms)
105          */
106         for (i=0; i<100; ++i) {
107                 if (tstc()) {
108                         getchar();
109                 }
110                 udelay(1000);
111         }
112
113         if (addr == ~0) {
114                 printf("## S-Record download aborted\n");
115                 rcode = 1;
116         } else {
117                 printf("## Start Addr      = 0x%08lX\n", addr);
118                 image_load_addr = addr;
119         }
120
121 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
122         if (load_baudrate != current_baudrate) {
123                 printf("## Switch baudrate to %d bps and press ESC ...\n",
124                         current_baudrate);
125                 udelay(50000);
126                 gd->baudrate = current_baudrate;
127                 serial_setbrg();
128                 udelay(50000);
129                 for (;;) {
130                         if (getchar() == 0x1B) /* ESC */
131                                 break;
132                 }
133         }
134 #endif
135         return rcode;
136 }
137
138 static ulong load_serial(long offset)
139 {
140         char    record[SREC_MAXRECLEN + 1];     /* buffer for one S-Record      */
141         char    binbuf[SREC_MAXBINLEN];         /* buffer for binary data       */
142         int     binlen;                         /* no. of data bytes in S-Rec.  */
143         int     type;                           /* return code for record type  */
144         ulong   addr;                           /* load address from S-Record   */
145         ulong   size;                           /* number of bytes transferred  */
146         ulong   store_addr;
147         ulong   start_addr = ~0;
148         ulong   end_addr   =  0;
149         int     line_count =  0;
150
151         while (read_record(record, SREC_MAXRECLEN + 1) >= 0) {
152                 type = srec_decode(record, &binlen, &addr, binbuf);
153
154                 if (type < 0) {
155                         return (~0);            /* Invalid S-Record             */
156                 }
157
158                 switch (type) {
159                 case SREC_DATA2:
160                 case SREC_DATA3:
161                 case SREC_DATA4:
162                     store_addr = addr + offset;
163 #ifdef CONFIG_MTD_NOR_FLASH
164                     if (addr2info(store_addr)) {
165                         int rc;
166
167                         rc = flash_write((char *)binbuf,store_addr,binlen);
168                         if (rc != 0) {
169                                 flash_perror(rc);
170                                 return (~0);
171                         }
172                     } else
173 #endif
174                     {
175                         memcpy((char *)(store_addr), binbuf, binlen);
176                     }
177                     if ((store_addr) < start_addr)
178                         start_addr = store_addr;
179                     if ((store_addr + binlen - 1) > end_addr)
180                         end_addr = store_addr + binlen - 1;
181                     break;
182                 case SREC_END2:
183                 case SREC_END3:
184                 case SREC_END4:
185                     udelay(10000);
186                     size = end_addr - start_addr + 1;
187                     printf("\n"
188                             "## First Load Addr = 0x%08lX\n"
189                             "## Last  Load Addr = 0x%08lX\n"
190                             "## Total Size      = 0x%08lX = %ld Bytes\n",
191                             start_addr, end_addr, size, size
192                     );
193                     flush_cache(start_addr, size);
194                     env_set_hex("filesize", size);
195                     return (addr);
196                 case SREC_START:
197                     break;
198                 default:
199                     break;
200                 }
201                 if (!do_echo) { /* print a '.' every 100 lines */
202                         if ((++line_count % 100) == 0)
203                                 putc('.');
204                 }
205         }
206
207         return (~0);                    /* Download aborted             */
208 }
209
210 static int read_record(char *buf, ulong len)
211 {
212         char *p;
213         char c;
214
215         --len;  /* always leave room for terminating '\0' byte */
216
217         for (p=buf; p < buf+len; ++p) {
218                 c = getchar();          /* read character               */
219                 if (do_echo)
220                         putc(c);        /* ... and echo it              */
221
222                 switch (c) {
223                 case '\r':
224                 case '\n':
225                         *p = '\0';
226                         return (p - buf);
227                 case '\0':
228                 case 0x03:                      /* ^C - Control C               */
229                         return (-1);
230                 default:
231                         *p = c;
232                 }
233
234                 /* Check for the console hangup (if any different from serial) */
235                 if (gd->jt->getc != getchar) {
236                         if (ctrlc())
237                                 return (-1);
238                 }
239         }
240
241         /* line too long - truncate */
242         *p = '\0';
243         return (p - buf);
244 }
245
246 #if defined(CONFIG_CMD_SAVES)
247
248 int do_save_serial(struct cmd_tbl *cmdtp, int flag, int argc,
249                    char *const argv[])
250 {
251         ulong offset = 0;
252         ulong size   = 0;
253 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
254         int save_baudrate, current_baudrate;
255
256         save_baudrate = current_baudrate = gd->baudrate;
257 #endif
258
259         if (argc >= 2) {
260                 offset = hextoul(argv[1], NULL);
261         }
262 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
263         if (argc >= 3) {
264                 size = hextoul(argv[2], NULL);
265         }
266         if (argc == 4) {
267                 save_baudrate = (int)dectoul(argv[3], NULL);
268
269                 /* default to current baudrate */
270                 if (save_baudrate == 0)
271                         save_baudrate = current_baudrate;
272         }
273         if (save_baudrate != current_baudrate) {
274                 printf("## Switch baudrate to %d bps and press ENTER ...\n",
275                         save_baudrate);
276                 udelay(50000);
277                 gd->baudrate = save_baudrate;
278                 serial_setbrg();
279                 udelay(50000);
280                 for (;;) {
281                         if (getchar() == '\r')
282                                 break;
283                 }
284         }
285 #else   /* ! CONFIG_SYS_LOADS_BAUD_CHANGE */
286         if (argc == 3) {
287                 size = hextoul(argv[2], NULL);
288         }
289 #endif  /* CONFIG_SYS_LOADS_BAUD_CHANGE */
290
291         printf("## Ready for S-Record upload, press ENTER to proceed ...\n");
292         for (;;) {
293                 if (getchar() == '\r')
294                         break;
295         }
296         if (save_serial(offset, size)) {
297                 printf("## S-Record upload aborted\n");
298         } else {
299                 printf("## S-Record upload complete\n");
300         }
301 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
302         if (save_baudrate != current_baudrate) {
303                 printf("## Switch baudrate to %d bps and press ESC ...\n",
304                         (int)current_baudrate);
305                 udelay(50000);
306                 gd->baudrate = current_baudrate;
307                 serial_setbrg();
308                 udelay(50000);
309                 for (;;) {
310                         if (getchar() == 0x1B) /* ESC */
311                                 break;
312                 }
313         }
314 #endif
315         return 0;
316 }
317
318 #define SREC3_START                             "S0030000FC\n"
319 #define SREC3_FORMAT                    "S3%02X%08lX%s%02X\n"
320 #define SREC3_END                               "S70500000000FA\n"
321 #define SREC_BYTES_PER_RECORD   16
322
323 static int save_serial(ulong address, ulong count)
324 {
325         int i, c, reclen, checksum, length;
326         char *hex = "0123456789ABCDEF";
327         char    record[2*SREC_BYTES_PER_RECORD+16];     /* buffer for one S-Record      */
328         char    data[2*SREC_BYTES_PER_RECORD+1];        /* buffer for hex data  */
329
330         reclen = 0;
331         checksum  = 0;
332
333         if(write_record(SREC3_START))                   /* write the header */
334                 return (-1);
335         do {
336                 if(count) {                                             /* collect hex data in the buffer  */
337                         c = *(volatile uchar*)(address + reclen);       /* get one byte    */
338                         checksum += c;                                                  /* accumulate checksum */
339                         data[2*reclen]   = hex[(c>>4)&0x0f];
340                         data[2*reclen+1] = hex[c & 0x0f];
341                         data[2*reclen+2] = '\0';
342                         ++reclen;
343                         --count;
344                 }
345                 if(reclen == SREC_BYTES_PER_RECORD || count == 0) {
346                         /* enough data collected for one record: dump it */
347                         if(reclen) {    /* build & write a data record: */
348                                 /* address + data + checksum */
349                                 length = 4 + reclen + 1;
350
351                                 /* accumulate length bytes into checksum */
352                                 for(i = 0; i < 2; i++)
353                                         checksum += (length >> (8*i)) & 0xff;
354
355                                 /* accumulate address bytes into checksum: */
356                                 for(i = 0; i < 4; i++)
357                                         checksum += (address >> (8*i)) & 0xff;
358
359                                 /* make proper checksum byte: */
360                                 checksum = ~checksum & 0xff;
361
362                                 /* output one record: */
363                                 sprintf(record, SREC3_FORMAT, length, address, data, checksum);
364                                 if(write_record(record))
365                                         return (-1);
366                         }
367                         address  += reclen;  /* increment address */
368                         checksum  = 0;
369                         reclen    = 0;
370                 }
371         }
372         while(count);
373         if(write_record(SREC3_END))     /* write the final record */
374                 return (-1);
375         return(0);
376 }
377
378 static int write_record(char *buf)
379 {
380         char c;
381
382         while((c = *buf++))
383                 putc(c);
384
385         /* Check for the console hangup (if any different from serial) */
386
387         if (ctrlc()) {
388             return (-1);
389         }
390         return (0);
391 }
392 # endif
393
394 #endif
395
396
397 #if defined(CONFIG_CMD_LOADB)
398 /*
399  * loadb command (load binary) included
400  */
401 #define XON_CHAR        17
402 #define XOFF_CHAR       19
403 #define START_CHAR      0x01
404 #define ETX_CHAR        0x03
405 #define END_CHAR        0x0D
406 #define SPACE           0x20
407 #define K_ESCAPE        0x23
408 #define SEND_TYPE       'S'
409 #define DATA_TYPE       'D'
410 #define ACK_TYPE        'Y'
411 #define NACK_TYPE       'N'
412 #define BREAK_TYPE      'B'
413 #define tochar(x) ((char) (((x) + SPACE) & 0xff))
414 #define untochar(x) ((int) (((x) - SPACE) & 0xff))
415
416 static void set_kerm_bin_mode(unsigned long *);
417 static int k_recv(void);
418 static ulong load_serial_bin(ulong offset);
419
420
421 static char his_eol;        /* character he needs at end of packet */
422 static int  his_pad_count;  /* number of pad chars he needs */
423 static char his_pad_char;   /* pad chars he needs */
424 static char his_quote;      /* quote chars he'll use */
425
426 static int do_load_serial_bin(struct cmd_tbl *cmdtp, int flag, int argc,
427                               char *const argv[])
428 {
429         ulong offset = 0;
430         ulong addr;
431         int load_baudrate, current_baudrate;
432         int rcode = 0;
433         char *s;
434
435         /* pre-set offset from CONFIG_SYS_LOAD_ADDR */
436         offset = CONFIG_SYS_LOAD_ADDR;
437
438         /* pre-set offset from $loadaddr */
439         s = env_get("loadaddr");
440         if (s)
441                 offset = hextoul(s, NULL);
442
443         load_baudrate = current_baudrate = gd->baudrate;
444
445         if (argc >= 2) {
446                 offset = hextoul(argv[1], NULL);
447         }
448         if (argc == 3) {
449                 load_baudrate = (int)dectoul(argv[2], NULL);
450
451                 /* default to current baudrate */
452                 if (load_baudrate == 0)
453                         load_baudrate = current_baudrate;
454         }
455
456         if (load_baudrate != current_baudrate) {
457                 printf("## Switch baudrate to %d bps and press ENTER ...\n",
458                         load_baudrate);
459                 udelay(50000);
460                 gd->baudrate = load_baudrate;
461                 serial_setbrg();
462                 udelay(50000);
463                 for (;;) {
464                         if (getchar() == '\r')
465                                 break;
466                 }
467         }
468
469         if (strcmp(argv[0],"loady")==0) {
470                 printf("## Ready for binary (ymodem) download "
471                         "to 0x%08lX at %d bps...\n",
472                         offset,
473                         load_baudrate);
474
475                 addr = load_serial_ymodem(offset, xyzModem_ymodem);
476
477         } else if (strcmp(argv[0],"loadx")==0) {
478                 printf("## Ready for binary (xmodem) download "
479                         "to 0x%08lX at %d bps...\n",
480                         offset,
481                         load_baudrate);
482
483                 addr = load_serial_ymodem(offset, xyzModem_xmodem);
484
485         } else {
486
487                 printf("## Ready for binary (kermit) download "
488                         "to 0x%08lX at %d bps...\n",
489                         offset,
490                         load_baudrate);
491                 addr = load_serial_bin(offset);
492
493                 if (addr == ~0) {
494                         image_load_addr = 0;
495                         printf("## Binary (kermit) download aborted\n");
496                         rcode = 1;
497                 } else {
498                         printf("## Start Addr      = 0x%08lX\n", addr);
499                         image_load_addr = addr;
500                 }
501         }
502         if (load_baudrate != current_baudrate) {
503                 printf("## Switch baudrate to %d bps and press ESC ...\n",
504                         current_baudrate);
505                 udelay(50000);
506                 gd->baudrate = current_baudrate;
507                 serial_setbrg();
508                 udelay(50000);
509                 for (;;) {
510                         if (getchar() == 0x1B) /* ESC */
511                                 break;
512                 }
513         }
514
515         return rcode;
516 }
517
518
519 static ulong load_serial_bin(ulong offset)
520 {
521         int size, i;
522
523         set_kerm_bin_mode((ulong *) offset);
524         size = k_recv();
525
526         /*
527          * Gather any trailing characters (for instance, the ^D which
528          * is sent by 'cu' after sending a file), and give the
529          * box some time (100 * 1 ms)
530          */
531         for (i=0; i<100; ++i) {
532                 if (tstc()) {
533                         getchar();
534                 }
535                 udelay(1000);
536         }
537
538         if (size == 0)
539                 return ~0; /* Download aborted */
540
541         flush_cache(offset, size);
542
543         printf("## Total Size      = 0x%08x = %d Bytes\n", size, size);
544         env_set_hex("filesize", size);
545
546         return offset;
547 }
548
549 static void send_pad(void)
550 {
551         int count = his_pad_count;
552
553         while (count-- > 0)
554                 putc(his_pad_char);
555 }
556
557 /* converts escaped kermit char to binary char */
558 static char ktrans(char in)
559 {
560         if ((in & 0x60) == 0x40) {
561                 return (char) (in & ~0x40);
562         } else if ((in & 0x7f) == 0x3f) {
563                 return (char) (in | 0x40);
564         } else
565                 return in;
566 }
567
568 static int chk1(char *buffer)
569 {
570         int total = 0;
571
572         while (*buffer) {
573                 total += *buffer++;
574         }
575         return (int) ((total + ((total >> 6) & 0x03)) & 0x3f);
576 }
577
578 static void s1_sendpacket(char *packet)
579 {
580         send_pad();
581         while (*packet) {
582                 putc(*packet++);
583         }
584 }
585
586 static char a_b[24];
587 static void send_ack(int n)
588 {
589         a_b[0] = START_CHAR;
590         a_b[1] = tochar(3);
591         a_b[2] = tochar(n);
592         a_b[3] = ACK_TYPE;
593         a_b[4] = '\0';
594         a_b[4] = tochar(chk1(&a_b[1]));
595         a_b[5] = his_eol;
596         a_b[6] = '\0';
597         s1_sendpacket(a_b);
598 }
599
600 static void send_nack(int n)
601 {
602         a_b[0] = START_CHAR;
603         a_b[1] = tochar(3);
604         a_b[2] = tochar(n);
605         a_b[3] = NACK_TYPE;
606         a_b[4] = '\0';
607         a_b[4] = tochar(chk1(&a_b[1]));
608         a_b[5] = his_eol;
609         a_b[6] = '\0';
610         s1_sendpacket(a_b);
611 }
612
613
614 static void (*os_data_init)(void);
615 static void (*os_data_char)(char new_char);
616 static int os_data_state, os_data_state_saved;
617 static char *os_data_addr, *os_data_addr_saved;
618 static char *bin_start_address;
619
620 static void bin_data_init(void)
621 {
622         os_data_state = 0;
623         os_data_addr = bin_start_address;
624 }
625
626 static void os_data_save(void)
627 {
628         os_data_state_saved = os_data_state;
629         os_data_addr_saved = os_data_addr;
630 }
631
632 static void os_data_restore(void)
633 {
634         os_data_state = os_data_state_saved;
635         os_data_addr = os_data_addr_saved;
636 }
637
638 static void bin_data_char(char new_char)
639 {
640         switch (os_data_state) {
641         case 0:                                 /* data */
642                 *os_data_addr++ = new_char;
643                 break;
644         }
645 }
646
647 static void set_kerm_bin_mode(unsigned long *addr)
648 {
649         bin_start_address = (char *) addr;
650         os_data_init = bin_data_init;
651         os_data_char = bin_data_char;
652 }
653
654
655 /* k_data_* simply handles the kermit escape translations */
656 static int k_data_escape, k_data_escape_saved;
657 static void k_data_init(void)
658 {
659         k_data_escape = 0;
660         os_data_init();
661 }
662
663 static void k_data_save(void)
664 {
665         k_data_escape_saved = k_data_escape;
666         os_data_save();
667 }
668
669 static void k_data_restore(void)
670 {
671         k_data_escape = k_data_escape_saved;
672         os_data_restore();
673 }
674
675 static void k_data_char(char new_char)
676 {
677         if (k_data_escape) {
678                 /* last char was escape - translate this character */
679                 os_data_char(ktrans(new_char));
680                 k_data_escape = 0;
681         } else {
682                 if (new_char == his_quote) {
683                         /* this char is escape - remember */
684                         k_data_escape = 1;
685                 } else {
686                         /* otherwise send this char as-is */
687                         os_data_char(new_char);
688                 }
689         }
690 }
691
692 #define SEND_DATA_SIZE  20
693 static char send_parms[SEND_DATA_SIZE];
694 static char *send_ptr;
695
696 /* handle_send_packet interprits the protocol info and builds and
697    sends an appropriate ack for what we can do */
698 static void handle_send_packet(int n)
699 {
700         int length = 3;
701         int bytes;
702
703         /* initialize some protocol parameters */
704         his_eol = END_CHAR;             /* default end of line character */
705         his_pad_count = 0;
706         his_pad_char = '\0';
707         his_quote = K_ESCAPE;
708
709         /* ignore last character if it filled the buffer */
710         if (send_ptr == &send_parms[SEND_DATA_SIZE - 1])
711                 --send_ptr;
712         bytes = send_ptr - send_parms;  /* how many bytes we'll process */
713         do {
714                 if (bytes-- <= 0)
715                         break;
716                 /* handle MAXL - max length */
717                 /* ignore what he says - most I'll take (here) is 94 */
718                 a_b[++length] = tochar(94);
719                 if (bytes-- <= 0)
720                         break;
721                 /* handle TIME - time you should wait for my packets */
722                 /* ignore what he says - don't wait for my ack longer than 1 second */
723                 a_b[++length] = tochar(1);
724                 if (bytes-- <= 0)
725                         break;
726                 /* handle NPAD - number of pad chars I need */
727                 /* remember what he says - I need none */
728                 his_pad_count = untochar(send_parms[2]);
729                 a_b[++length] = tochar(0);
730                 if (bytes-- <= 0)
731                         break;
732                 /* handle PADC - pad chars I need */
733                 /* remember what he says - I need none */
734                 his_pad_char = ktrans(send_parms[3]);
735                 a_b[++length] = 0x40;   /* He should ignore this */
736                 if (bytes-- <= 0)
737                         break;
738                 /* handle EOL - end of line he needs */
739                 /* remember what he says - I need CR */
740                 his_eol = untochar(send_parms[4]);
741                 a_b[++length] = tochar(END_CHAR);
742                 if (bytes-- <= 0)
743                         break;
744                 /* handle QCTL - quote control char he'll use */
745                 /* remember what he says - I'll use '#' */
746                 his_quote = send_parms[5];
747                 a_b[++length] = '#';
748                 if (bytes-- <= 0)
749                         break;
750                 /* handle QBIN - 8-th bit prefixing */
751                 /* ignore what he says - I refuse */
752                 a_b[++length] = 'N';
753                 if (bytes-- <= 0)
754                         break;
755                 /* handle CHKT - the clock check type */
756                 /* ignore what he says - I do type 1 (for now) */
757                 a_b[++length] = '1';
758                 if (bytes-- <= 0)
759                         break;
760                 /* handle REPT - the repeat prefix */
761                 /* ignore what he says - I refuse (for now) */
762                 a_b[++length] = 'N';
763                 if (bytes-- <= 0)
764                         break;
765                 /* handle CAPAS - the capabilities mask */
766                 /* ignore what he says - I only do long packets - I don't do windows */
767                 a_b[++length] = tochar(2);      /* only long packets */
768                 a_b[++length] = tochar(0);      /* no windows */
769                 a_b[++length] = tochar(94);     /* large packet msb */
770                 a_b[++length] = tochar(94);     /* large packet lsb */
771         } while (0);
772
773         a_b[0] = START_CHAR;
774         a_b[1] = tochar(length);
775         a_b[2] = tochar(n);
776         a_b[3] = ACK_TYPE;
777         a_b[++length] = '\0';
778         a_b[length] = tochar(chk1(&a_b[1]));
779         a_b[++length] = his_eol;
780         a_b[++length] = '\0';
781         s1_sendpacket(a_b);
782 }
783
784 /* k_recv receives a OS Open image file over kermit line */
785 static int k_recv(void)
786 {
787         char new_char;
788         char k_state, k_state_saved;
789         int sum;
790         int done;
791         int length;
792         int n, last_n;
793         int len_lo, len_hi;
794
795         /* initialize some protocol parameters */
796         his_eol = END_CHAR;             /* default end of line character */
797         his_pad_count = 0;
798         his_pad_char = '\0';
799         his_quote = K_ESCAPE;
800
801         /* initialize the k_recv and k_data state machine */
802         done = 0;
803         k_state = 0;
804         k_data_init();
805         k_state_saved = k_state;
806         k_data_save();
807         n = 0;                          /* just to get rid of a warning */
808         last_n = -1;
809
810         /* expect this "type" sequence (but don't check):
811            S: send initiate
812            F: file header
813            D: data (multiple)
814            Z: end of file
815            B: break transmission
816          */
817
818         /* enter main loop */
819         while (!done) {
820                 /* set the send packet pointer to begining of send packet parms */
821                 send_ptr = send_parms;
822
823                 /* With each packet, start summing the bytes starting with the length.
824                    Save the current sequence number.
825                    Note the type of the packet.
826                    If a character less than SPACE (0x20) is received - error.
827                  */
828
829 #if 0
830                 /* OLD CODE, Prior to checking sequence numbers */
831                 /* first have all state machines save current states */
832                 k_state_saved = k_state;
833                 k_data_save ();
834 #endif
835
836                 /* get a packet */
837                 /* wait for the starting character or ^C */
838                 for (;;) {
839                         switch (getchar()) {
840                         case START_CHAR:        /* start packet */
841                                 goto START;
842                         case ETX_CHAR:          /* ^C waiting for packet */
843                                 return (0);
844                         default:
845                                 ;
846                         }
847                 }
848 START:
849                 /* get length of packet */
850                 sum = 0;
851                 new_char = getchar();
852                 if ((new_char & 0xE0) == 0)
853                         goto packet_error;
854                 sum += new_char & 0xff;
855                 length = untochar(new_char);
856                 /* get sequence number */
857                 new_char = getchar();
858                 if ((new_char & 0xE0) == 0)
859                         goto packet_error;
860                 sum += new_char & 0xff;
861                 n = untochar(new_char);
862                 --length;
863
864                 /* NEW CODE - check sequence numbers for retried packets */
865                 /* Note - this new code assumes that the sequence number is correctly
866                  * received.  Handling an invalid sequence number adds another layer
867                  * of complexity that may not be needed - yet!  At this time, I'm hoping
868                  * that I don't need to buffer the incoming data packets and can write
869                  * the data into memory in real time.
870                  */
871                 if (n == last_n) {
872                         /* same sequence number, restore the previous state */
873                         k_state = k_state_saved;
874                         k_data_restore();
875                 } else {
876                         /* new sequence number, checkpoint the download */
877                         last_n = n;
878                         k_state_saved = k_state;
879                         k_data_save();
880                 }
881                 /* END NEW CODE */
882
883                 /* get packet type */
884                 new_char = getchar();
885                 if ((new_char & 0xE0) == 0)
886                         goto packet_error;
887                 sum += new_char & 0xff;
888                 k_state = new_char;
889                 --length;
890                 /* check for extended length */
891                 if (length == -2) {
892                         /* (length byte was 0, decremented twice) */
893                         /* get the two length bytes */
894                         new_char = getchar();
895                         if ((new_char & 0xE0) == 0)
896                                 goto packet_error;
897                         sum += new_char & 0xff;
898                         len_hi = untochar(new_char);
899                         new_char = getchar();
900                         if ((new_char & 0xE0) == 0)
901                                 goto packet_error;
902                         sum += new_char & 0xff;
903                         len_lo = untochar(new_char);
904                         length = len_hi * 95 + len_lo;
905                         /* check header checksum */
906                         new_char = getchar();
907                         if ((new_char & 0xE0) == 0)
908                                 goto packet_error;
909                         if (new_char != tochar((sum + ((sum >> 6) & 0x03)) & 0x3f))
910                                 goto packet_error;
911                         sum += new_char & 0xff;
912 /* --length; */ /* new length includes only data and block check to come */
913                 }
914                 /* bring in rest of packet */
915                 while (length > 1) {
916                         new_char = getchar();
917                         if ((new_char & 0xE0) == 0)
918                                 goto packet_error;
919                         sum += new_char & 0xff;
920                         --length;
921                         if (k_state == DATA_TYPE) {
922                                 /* pass on the data if this is a data packet */
923                                 k_data_char (new_char);
924                         } else if (k_state == SEND_TYPE) {
925                                 /* save send pack in buffer as is */
926                                 *send_ptr++ = new_char;
927                                 /* if too much data, back off the pointer */
928                                 if (send_ptr >= &send_parms[SEND_DATA_SIZE])
929                                         --send_ptr;
930                         }
931                 }
932                 /* get and validate checksum character */
933                 new_char = getchar();
934                 if ((new_char & 0xE0) == 0)
935                         goto packet_error;
936                 if (new_char != tochar((sum + ((sum >> 6) & 0x03)) & 0x3f))
937                         goto packet_error;
938                 /* get END_CHAR */
939                 new_char = getchar();
940                 if (new_char != END_CHAR) {
941                   packet_error:
942                         /* restore state machines */
943                         k_state = k_state_saved;
944                         k_data_restore();
945                         /* send a negative acknowledge packet in */
946                         send_nack(n);
947                 } else if (k_state == SEND_TYPE) {
948                         /* crack the protocol parms, build an appropriate ack packet */
949                         handle_send_packet(n);
950                 } else {
951                         /* send simple acknowledge packet in */
952                         send_ack(n);
953                         /* quit if end of transmission */
954                         if (k_state == BREAK_TYPE)
955                                 done = 1;
956                 }
957         }
958         return ((ulong) os_data_addr - (ulong) bin_start_address);
959 }
960
961 static int getcxmodem(void) {
962         if (tstc())
963                 return (getchar());
964         return -1;
965 }
966 static ulong load_serial_ymodem(ulong offset, int mode)
967 {
968         int size;
969         int err;
970         int res;
971         connection_info_t info;
972         char ymodemBuf[1024];
973         ulong store_addr = ~0;
974         ulong addr = 0;
975
976         size = 0;
977         info.mode = mode;
978         res = xyzModem_stream_open(&info, &err);
979         if (!res) {
980
981                 while ((res =
982                         xyzModem_stream_read(ymodemBuf, 1024, &err)) > 0) {
983                         store_addr = addr + offset;
984                         size += res;
985                         addr += res;
986 #ifdef CONFIG_MTD_NOR_FLASH
987                         if (addr2info(store_addr)) {
988                                 int rc;
989
990                                 rc = flash_write((char *) ymodemBuf,
991                                                   store_addr, res);
992                                 if (rc != 0) {
993                                         flash_perror(rc);
994                                         return (~0);
995                                 }
996                         } else
997 #endif
998                         {
999                                 memcpy((char *)(store_addr), ymodemBuf,
1000                                         res);
1001                         }
1002
1003                 }
1004                 if (IS_ENABLED(CONFIG_CMD_BOOTEFI))
1005                         efi_set_bootdev("Uart", "", "",
1006                                         map_sysmem(offset, 0), size);
1007
1008         } else {
1009                 printf("%s\n", xyzModem_error(err));
1010         }
1011
1012         xyzModem_stream_terminate(false, &getcxmodem);
1013         xyzModem_stream_close(&err);
1014
1015
1016         flush_cache(offset, ALIGN(size, ARCH_DMA_MINALIGN));
1017
1018         printf("## Total Size      = 0x%08x = %d Bytes\n", size, size);
1019         env_set_hex("filesize", size);
1020
1021         return offset;
1022 }
1023
1024 #endif
1025
1026 /* -------------------------------------------------------------------- */
1027
1028 #if defined(CONFIG_CMD_LOADS)
1029
1030 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
1031 U_BOOT_CMD(
1032         loads, 3, 0,    do_load_serial,
1033         "load S-Record file over serial line",
1034         "[ off ] [ baud ]\n"
1035         "    - load S-Record file over serial line"
1036         " with offset 'off' and baudrate 'baud'"
1037 );
1038
1039 #else   /* ! CONFIG_SYS_LOADS_BAUD_CHANGE */
1040 U_BOOT_CMD(
1041         loads, 2, 0,    do_load_serial,
1042         "load S-Record file over serial line",
1043         "[ off ]\n"
1044         "    - load S-Record file over serial line with offset 'off'"
1045 );
1046 #endif  /* CONFIG_SYS_LOADS_BAUD_CHANGE */
1047
1048 /*
1049  * SAVES always requires LOADS support, but not vice versa
1050  */
1051
1052
1053 #if defined(CONFIG_CMD_SAVES)
1054 #ifdef  CONFIG_SYS_LOADS_BAUD_CHANGE
1055 U_BOOT_CMD(
1056         saves, 4, 0,    do_save_serial,
1057         "save S-Record file over serial line",
1058         "[ off ] [size] [ baud ]\n"
1059         "    - save S-Record file over serial line"
1060         " with offset 'off', size 'size' and baudrate 'baud'"
1061 );
1062 #else   /* ! CONFIG_SYS_LOADS_BAUD_CHANGE */
1063 U_BOOT_CMD(
1064         saves, 3, 0,    do_save_serial,
1065         "save S-Record file over serial line",
1066         "[ off ] [size]\n"
1067         "    - save S-Record file over serial line with offset 'off' and size 'size'"
1068 );
1069 #endif  /* CONFIG_SYS_LOADS_BAUD_CHANGE */
1070 #endif  /* CONFIG_CMD_SAVES */
1071 #endif  /* CONFIG_CMD_LOADS */
1072
1073
1074 #if defined(CONFIG_CMD_LOADB)
1075 U_BOOT_CMD(
1076         loadb, 3, 0,    do_load_serial_bin,
1077         "load binary file over serial line (kermit mode)",
1078         "[ addr [ baud ] ]\n"
1079         "    - load binary file over serial line"
1080         " at address 'addr' with baudrate 'baud'"
1081 );
1082
1083 U_BOOT_CMD(
1084         loadx, 3, 0,    do_load_serial_bin,
1085         "load binary file over serial line (xmodem mode)",
1086         "[ addr [ baud ] ]\n"
1087         "    - load binary file over serial line"
1088         " at address 'addr' with baudrate 'baud'"
1089 );
1090
1091 U_BOOT_CMD(
1092         loady, 3, 0,    do_load_serial_bin,
1093         "load binary file over serial line (ymodem mode)",
1094         "[ addr [ baud ] ]\n"
1095         "    - load binary file over serial line"
1096         " at address 'addr' with baudrate 'baud'"
1097 );
1098
1099 #endif  /* CONFIG_CMD_LOADB */