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