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