cmd/load.c

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