xref: /openbmc/linux/arch/arm/boot/compressed/misc.c (revision aa0d3bb77e780054babcd289484cf4c15180111b)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * misc.c
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  * This is a collection of several routines from gzip-1.0.3
51da177e4SLinus Torvalds  * adapted for Linux.
61da177e4SLinus Torvalds  *
71da177e4SLinus Torvalds  * malloc by Hannu Savolainen 1993 and Matthias Urlichs 1994
81da177e4SLinus Torvalds  *
91da177e4SLinus Torvalds  * Modified for ARM Linux by Russell King
101da177e4SLinus Torvalds  *
111da177e4SLinus Torvalds  * Nicolas Pitre <nico@visuaide.com>  1999/04/14 :
121da177e4SLinus Torvalds  *  For this code to run directly from Flash, all constant variables must
131da177e4SLinus Torvalds  *  be marked with 'const' and all other variables initialized at run-time
141da177e4SLinus Torvalds  *  only.  This way all non constant variables will end up in the bss segment,
151da177e4SLinus Torvalds  *  which should point to addresses in RAM and cleared to 0 on start.
161da177e4SLinus Torvalds  *  This allows for a much quicker boot time.
171da177e4SLinus Torvalds  */
181da177e4SLinus Torvalds 
191da177e4SLinus Torvalds unsigned int __machine_arch_type;
201da177e4SLinus Torvalds 
21*aa0d3bb7SRusty Russell #include <linux/compiler.h>	/* for inline */
22*aa0d3bb7SRusty Russell #include <linux/types.h>	/* for size_t */
23*aa0d3bb7SRusty Russell #include <linux/stddef.h>	/* for NULL */
24*aa0d3bb7SRusty Russell #include <asm/string.h>
251da177e4SLinus Torvalds 
261da177e4SLinus Torvalds #ifdef STANDALONE_DEBUG
271da177e4SLinus Torvalds #define putstr printf
28a081568dSRussell King #else
29a081568dSRussell King 
30a081568dSRussell King static void putstr(const char *ptr);
31a081568dSRussell King 
32a081568dSRussell King #include <linux/compiler.h>
33a09e64fbSRussell King #include <mach/uncompress.h>
341da177e4SLinus Torvalds 
351da177e4SLinus Torvalds #ifdef CONFIG_DEBUG_ICEDCC
367d95ded9STony Lindgren 
377d95ded9STony Lindgren #ifdef CONFIG_CPU_V6
387d95ded9STony Lindgren 
397d95ded9STony Lindgren static void icedcc_putc(int ch)
407d95ded9STony Lindgren {
417d95ded9STony Lindgren 	int status, i = 0x4000000;
427d95ded9STony Lindgren 
437d95ded9STony Lindgren 	do {
447d95ded9STony Lindgren 		if (--i < 0)
457d95ded9STony Lindgren 			return;
467d95ded9STony Lindgren 
477d95ded9STony Lindgren 		asm volatile ("mrc p14, 0, %0, c0, c1, 0" : "=r" (status));
487d95ded9STony Lindgren 	} while (status & (1 << 29));
497d95ded9STony Lindgren 
507d95ded9STony Lindgren 	asm("mcr p14, 0, %0, c0, c5, 0" : : "r" (ch));
517d95ded9STony Lindgren }
52c633c3cfSJean-Christop PLAGNIOL-VILLARD #elif defined(CONFIG_CPU_XSCALE)
53c633c3cfSJean-Christop PLAGNIOL-VILLARD 
54c633c3cfSJean-Christop PLAGNIOL-VILLARD static void icedcc_putc(int ch)
55c633c3cfSJean-Christop PLAGNIOL-VILLARD {
56c633c3cfSJean-Christop PLAGNIOL-VILLARD 	int status, i = 0x4000000;
57c633c3cfSJean-Christop PLAGNIOL-VILLARD 
58c633c3cfSJean-Christop PLAGNIOL-VILLARD 	do {
59c633c3cfSJean-Christop PLAGNIOL-VILLARD 		if (--i < 0)
60c633c3cfSJean-Christop PLAGNIOL-VILLARD 			return;
61c633c3cfSJean-Christop PLAGNIOL-VILLARD 
62c633c3cfSJean-Christop PLAGNIOL-VILLARD 		asm volatile ("mrc p14, 0, %0, c14, c0, 0" : "=r" (status));
63c633c3cfSJean-Christop PLAGNIOL-VILLARD 	} while (status & (1 << 28));
64c633c3cfSJean-Christop PLAGNIOL-VILLARD 
65c633c3cfSJean-Christop PLAGNIOL-VILLARD 	asm("mcr p14, 0, %0, c8, c0, 0" : : "r" (ch));
66c633c3cfSJean-Christop PLAGNIOL-VILLARD }
677d95ded9STony Lindgren 
687d95ded9STony Lindgren #else
697d95ded9STony Lindgren 
70de4533a0SRussell King static void icedcc_putc(int ch)
71de4533a0SRussell King {
72de4533a0SRussell King 	int status, i = 0x4000000;
73de4533a0SRussell King 
74de4533a0SRussell King 	do {
75de4533a0SRussell King 		if (--i < 0)
76de4533a0SRussell King 			return;
77de4533a0SRussell King 
78b2556da5SUwe Zeisberger 		asm volatile ("mrc p14, 0, %0, c0, c0, 0" : "=r" (status));
79de4533a0SRussell King 	} while (status & 2);
80de4533a0SRussell King 
81b2556da5SUwe Zeisberger 	asm("mcr p14, 0, %0, c1, c0, 0" : : "r" (ch));
82de4533a0SRussell King }
83de4533a0SRussell King 
847d95ded9STony Lindgren #endif
857d95ded9STony Lindgren 
86a081568dSRussell King #define putc(ch)	icedcc_putc(ch)
87a081568dSRussell King #define flush()	do { } while (0)
88a081568dSRussell King #endif
891da177e4SLinus Torvalds 
90a081568dSRussell King static void putstr(const char *ptr)
911da177e4SLinus Torvalds {
92a081568dSRussell King 	char c;
93a081568dSRussell King 
94a081568dSRussell King 	while ((c = *ptr++) != '\0') {
95a081568dSRussell King 		if (c == '\n')
96a081568dSRussell King 			putc('\r');
97a081568dSRussell King 		putc(c);
981da177e4SLinus Torvalds 	}
99a081568dSRussell King 
100a081568dSRussell King 	flush();
1011da177e4SLinus Torvalds }
1021da177e4SLinus Torvalds 
1031da177e4SLinus Torvalds #endif
1041da177e4SLinus Torvalds 
1051da177e4SLinus Torvalds #define __ptr_t void *
1061da177e4SLinus Torvalds 
10759f0cb0fSRussell King #define memzero(s,n) __memzero(s,n)
10859f0cb0fSRussell King 
1091da177e4SLinus Torvalds /*
1101da177e4SLinus Torvalds  * Optimised C version of memzero for the ARM.
1111da177e4SLinus Torvalds  */
1121da177e4SLinus Torvalds void __memzero (__ptr_t s, size_t n)
1131da177e4SLinus Torvalds {
1141da177e4SLinus Torvalds 	union { void *vp; unsigned long *ulp; unsigned char *ucp; } u;
1151da177e4SLinus Torvalds 	int i;
1161da177e4SLinus Torvalds 
1171da177e4SLinus Torvalds 	u.vp = s;
1181da177e4SLinus Torvalds 
1191da177e4SLinus Torvalds 	for (i = n >> 5; i > 0; i--) {
1201da177e4SLinus Torvalds 		*u.ulp++ = 0;
1211da177e4SLinus Torvalds 		*u.ulp++ = 0;
1221da177e4SLinus Torvalds 		*u.ulp++ = 0;
1231da177e4SLinus Torvalds 		*u.ulp++ = 0;
1241da177e4SLinus Torvalds 		*u.ulp++ = 0;
1251da177e4SLinus Torvalds 		*u.ulp++ = 0;
1261da177e4SLinus Torvalds 		*u.ulp++ = 0;
1271da177e4SLinus Torvalds 		*u.ulp++ = 0;
1281da177e4SLinus Torvalds 	}
1291da177e4SLinus Torvalds 
1301da177e4SLinus Torvalds 	if (n & 1 << 4) {
1311da177e4SLinus Torvalds 		*u.ulp++ = 0;
1321da177e4SLinus Torvalds 		*u.ulp++ = 0;
1331da177e4SLinus Torvalds 		*u.ulp++ = 0;
1341da177e4SLinus Torvalds 		*u.ulp++ = 0;
1351da177e4SLinus Torvalds 	}
1361da177e4SLinus Torvalds 
1371da177e4SLinus Torvalds 	if (n & 1 << 3) {
1381da177e4SLinus Torvalds 		*u.ulp++ = 0;
1391da177e4SLinus Torvalds 		*u.ulp++ = 0;
1401da177e4SLinus Torvalds 	}
1411da177e4SLinus Torvalds 
1421da177e4SLinus Torvalds 	if (n & 1 << 2)
1431da177e4SLinus Torvalds 		*u.ulp++ = 0;
1441da177e4SLinus Torvalds 
1451da177e4SLinus Torvalds 	if (n & 1 << 1) {
1461da177e4SLinus Torvalds 		*u.ucp++ = 0;
1471da177e4SLinus Torvalds 		*u.ucp++ = 0;
1481da177e4SLinus Torvalds 	}
1491da177e4SLinus Torvalds 
1501da177e4SLinus Torvalds 	if (n & 1)
1511da177e4SLinus Torvalds 		*u.ucp++ = 0;
1521da177e4SLinus Torvalds }
1531da177e4SLinus Torvalds 
1541da177e4SLinus Torvalds static inline __ptr_t memcpy(__ptr_t __dest, __const __ptr_t __src,
1551da177e4SLinus Torvalds 			    size_t __n)
1561da177e4SLinus Torvalds {
1571da177e4SLinus Torvalds 	int i = 0;
1581da177e4SLinus Torvalds 	unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
1591da177e4SLinus Torvalds 
1601da177e4SLinus Torvalds 	for (i = __n >> 3; i > 0; i--) {
1611da177e4SLinus Torvalds 		*d++ = *s++;
1621da177e4SLinus Torvalds 		*d++ = *s++;
1631da177e4SLinus Torvalds 		*d++ = *s++;
1641da177e4SLinus Torvalds 		*d++ = *s++;
1651da177e4SLinus Torvalds 		*d++ = *s++;
1661da177e4SLinus Torvalds 		*d++ = *s++;
1671da177e4SLinus Torvalds 		*d++ = *s++;
1681da177e4SLinus Torvalds 		*d++ = *s++;
1691da177e4SLinus Torvalds 	}
1701da177e4SLinus Torvalds 
1711da177e4SLinus Torvalds 	if (__n & 1 << 2) {
1721da177e4SLinus Torvalds 		*d++ = *s++;
1731da177e4SLinus Torvalds 		*d++ = *s++;
1741da177e4SLinus Torvalds 		*d++ = *s++;
1751da177e4SLinus Torvalds 		*d++ = *s++;
1761da177e4SLinus Torvalds 	}
1771da177e4SLinus Torvalds 
1781da177e4SLinus Torvalds 	if (__n & 1 << 1) {
1791da177e4SLinus Torvalds 		*d++ = *s++;
1801da177e4SLinus Torvalds 		*d++ = *s++;
1811da177e4SLinus Torvalds 	}
1821da177e4SLinus Torvalds 
1831da177e4SLinus Torvalds 	if (__n & 1)
1841da177e4SLinus Torvalds 		*d++ = *s++;
1851da177e4SLinus Torvalds 
1861da177e4SLinus Torvalds 	return __dest;
1871da177e4SLinus Torvalds }
1881da177e4SLinus Torvalds 
1891da177e4SLinus Torvalds /*
1901da177e4SLinus Torvalds  * gzip delarations
1911da177e4SLinus Torvalds  */
1921da177e4SLinus Torvalds #define OF(args)  args
1931da177e4SLinus Torvalds #define STATIC static
1941da177e4SLinus Torvalds 
1951da177e4SLinus Torvalds typedef unsigned char  uch;
1961da177e4SLinus Torvalds typedef unsigned short ush;
1971da177e4SLinus Torvalds typedef unsigned long  ulg;
1981da177e4SLinus Torvalds 
1991da177e4SLinus Torvalds #define WSIZE 0x8000		/* Window size must be at least 32k, */
2001da177e4SLinus Torvalds 				/* and a power of two */
2011da177e4SLinus Torvalds 
2021da177e4SLinus Torvalds static uch *inbuf;		/* input buffer */
2031da177e4SLinus Torvalds static uch window[WSIZE];	/* Sliding window buffer */
2041da177e4SLinus Torvalds 
2051da177e4SLinus Torvalds static unsigned insize;		/* valid bytes in inbuf */
2061da177e4SLinus Torvalds static unsigned inptr;		/* index of next byte to be processed in inbuf */
2071da177e4SLinus Torvalds static unsigned outcnt;		/* bytes in output buffer */
2081da177e4SLinus Torvalds 
2091da177e4SLinus Torvalds /* gzip flag byte */
2101da177e4SLinus Torvalds #define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
2111da177e4SLinus Torvalds #define CONTINUATION 0x02 /* bit 1 set: continuation of multi-part gzip file */
2121da177e4SLinus Torvalds #define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
2131da177e4SLinus Torvalds #define ORIG_NAME    0x08 /* bit 3 set: original file name present */
2141da177e4SLinus Torvalds #define COMMENT      0x10 /* bit 4 set: file comment present */
2151da177e4SLinus Torvalds #define ENCRYPTED    0x20 /* bit 5 set: file is encrypted */
2161da177e4SLinus Torvalds #define RESERVED     0xC0 /* bit 6,7:   reserved */
2171da177e4SLinus Torvalds 
2181da177e4SLinus Torvalds #define get_byte()  (inptr < insize ? inbuf[inptr++] : fill_inbuf())
2191da177e4SLinus Torvalds 
2201da177e4SLinus Torvalds /* Diagnostic functions */
2211da177e4SLinus Torvalds #ifdef DEBUG
2221da177e4SLinus Torvalds #  define Assert(cond,msg) {if(!(cond)) error(msg);}
2231da177e4SLinus Torvalds #  define Trace(x) fprintf x
2241da177e4SLinus Torvalds #  define Tracev(x) {if (verbose) fprintf x ;}
2251da177e4SLinus Torvalds #  define Tracevv(x) {if (verbose>1) fprintf x ;}
2261da177e4SLinus Torvalds #  define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
2271da177e4SLinus Torvalds #  define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
2281da177e4SLinus Torvalds #else
2291da177e4SLinus Torvalds #  define Assert(cond,msg)
2301da177e4SLinus Torvalds #  define Trace(x)
2311da177e4SLinus Torvalds #  define Tracev(x)
2321da177e4SLinus Torvalds #  define Tracevv(x)
2331da177e4SLinus Torvalds #  define Tracec(c,x)
2341da177e4SLinus Torvalds #  define Tracecv(c,x)
2351da177e4SLinus Torvalds #endif
2361da177e4SLinus Torvalds 
2371da177e4SLinus Torvalds static int  fill_inbuf(void);
2381da177e4SLinus Torvalds static void flush_window(void);
2391da177e4SLinus Torvalds static void error(char *m);
2401da177e4SLinus Torvalds 
2411da177e4SLinus Torvalds extern char input_data[];
2421da177e4SLinus Torvalds extern char input_data_end[];
2431da177e4SLinus Torvalds 
2441da177e4SLinus Torvalds static uch *output_data;
2451da177e4SLinus Torvalds static ulg output_ptr;
2461da177e4SLinus Torvalds static ulg bytes_out;
2471da177e4SLinus Torvalds 
2481da177e4SLinus Torvalds static void error(char *m);
2491da177e4SLinus Torvalds 
2501da177e4SLinus Torvalds static void putstr(const char *);
2511da177e4SLinus Torvalds 
2521da177e4SLinus Torvalds extern int end;
2531da177e4SLinus Torvalds static ulg free_mem_ptr;
2542d6ffccaSThomas Petazzoni static ulg free_mem_end_ptr;
2551da177e4SLinus Torvalds 
2562d6ffccaSThomas Petazzoni #ifdef STANDALONE_DEBUG
2572d6ffccaSThomas Petazzoni #define NO_INFLATE_MALLOC
2582d6ffccaSThomas Petazzoni #endif
2592d6ffccaSThomas Petazzoni 
2602d6ffccaSThomas Petazzoni #define ARCH_HAS_DECOMP_WDOG
2611da177e4SLinus Torvalds 
2621da177e4SLinus Torvalds #include "../../../../lib/inflate.c"
2631da177e4SLinus Torvalds 
2641da177e4SLinus Torvalds /* ===========================================================================
2651da177e4SLinus Torvalds  * Fill the input buffer. This is called only when the buffer is empty
2661da177e4SLinus Torvalds  * and at least one byte is really needed.
2671da177e4SLinus Torvalds  */
2681da177e4SLinus Torvalds int fill_inbuf(void)
2691da177e4SLinus Torvalds {
2701da177e4SLinus Torvalds 	if (insize != 0)
2711da177e4SLinus Torvalds 		error("ran out of input data");
2721da177e4SLinus Torvalds 
2731da177e4SLinus Torvalds 	inbuf = input_data;
2741da177e4SLinus Torvalds 	insize = &input_data_end[0] - &input_data[0];
2751da177e4SLinus Torvalds 
2761da177e4SLinus Torvalds 	inptr = 1;
2771da177e4SLinus Torvalds 	return inbuf[0];
2781da177e4SLinus Torvalds }
2791da177e4SLinus Torvalds 
2801da177e4SLinus Torvalds /* ===========================================================================
2811da177e4SLinus Torvalds  * Write the output window window[0..outcnt-1] and update crc and bytes_out.
2821da177e4SLinus Torvalds  * (Used for the decompressed data only.)
2831da177e4SLinus Torvalds  */
2841da177e4SLinus Torvalds void flush_window(void)
2851da177e4SLinus Torvalds {
2861da177e4SLinus Torvalds 	ulg c = crc;
2871da177e4SLinus Torvalds 	unsigned n;
2881da177e4SLinus Torvalds 	uch *in, *out, ch;
2891da177e4SLinus Torvalds 
2901da177e4SLinus Torvalds 	in = window;
2911da177e4SLinus Torvalds 	out = &output_data[output_ptr];
2921da177e4SLinus Torvalds 	for (n = 0; n < outcnt; n++) {
2931da177e4SLinus Torvalds 		ch = *out++ = *in++;
2941da177e4SLinus Torvalds 		c = crc_32_tab[((int)c ^ ch) & 0xff] ^ (c >> 8);
2951da177e4SLinus Torvalds 	}
2961da177e4SLinus Torvalds 	crc = c;
2971da177e4SLinus Torvalds 	bytes_out += (ulg)outcnt;
2981da177e4SLinus Torvalds 	output_ptr += (ulg)outcnt;
2991da177e4SLinus Torvalds 	outcnt = 0;
3001da177e4SLinus Torvalds 	putstr(".");
3011da177e4SLinus Torvalds }
3021da177e4SLinus Torvalds 
303f8c905d3SBen Dooks #ifndef arch_error
304f8c905d3SBen Dooks #define arch_error(x)
305f8c905d3SBen Dooks #endif
306f8c905d3SBen Dooks 
3071da177e4SLinus Torvalds static void error(char *x)
3081da177e4SLinus Torvalds {
309f8c905d3SBen Dooks 	arch_error(x);
310f8c905d3SBen Dooks 
3111da177e4SLinus Torvalds 	putstr("\n\n");
3121da177e4SLinus Torvalds 	putstr(x);
3131da177e4SLinus Torvalds 	putstr("\n\n -- System halted");
3141da177e4SLinus Torvalds 
3151da177e4SLinus Torvalds 	while(1);	/* Halt */
3161da177e4SLinus Torvalds }
3171da177e4SLinus Torvalds 
3181da177e4SLinus Torvalds #ifndef STANDALONE_DEBUG
3191da177e4SLinus Torvalds 
3201da177e4SLinus Torvalds ulg
3211da177e4SLinus Torvalds decompress_kernel(ulg output_start, ulg free_mem_ptr_p, ulg free_mem_ptr_end_p,
3221da177e4SLinus Torvalds 		  int arch_id)
3231da177e4SLinus Torvalds {
3241da177e4SLinus Torvalds 	output_data		= (uch *)output_start;	/* Points to kernel start */
3251da177e4SLinus Torvalds 	free_mem_ptr		= free_mem_ptr_p;
3262d6ffccaSThomas Petazzoni 	free_mem_end_ptr	= free_mem_ptr_end_p;
3271da177e4SLinus Torvalds 	__machine_arch_type	= arch_id;
3281da177e4SLinus Torvalds 
3291da177e4SLinus Torvalds 	arch_decomp_setup();
3301da177e4SLinus Torvalds 
3311da177e4SLinus Torvalds 	makecrc();
3321da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3331da177e4SLinus Torvalds 	gunzip();
3341da177e4SLinus Torvalds 	putstr(" done, booting the kernel.\n");
3351da177e4SLinus Torvalds 	return output_ptr;
3361da177e4SLinus Torvalds }
3371da177e4SLinus Torvalds #else
3381da177e4SLinus Torvalds 
3391da177e4SLinus Torvalds char output_buffer[1500*1024];
3401da177e4SLinus Torvalds 
3411da177e4SLinus Torvalds int main()
3421da177e4SLinus Torvalds {
3431da177e4SLinus Torvalds 	output_data = output_buffer;
3441da177e4SLinus Torvalds 
3451da177e4SLinus Torvalds 	makecrc();
3461da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3471da177e4SLinus Torvalds 	gunzip();
3481da177e4SLinus Torvalds 	putstr("done.\n");
3491da177e4SLinus Torvalds 	return 0;
3501da177e4SLinus Torvalds }
3511da177e4SLinus Torvalds #endif
3521da177e4SLinus Torvalds 
353