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