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