xref: /openbmc/linux/arch/arm/boot/compressed/misc.c (revision c633c3cfcf41efe720020553aa3d4782fa0b9bd5)
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 
211da177e4SLinus Torvalds #include <linux/string.h>
221da177e4SLinus Torvalds 
231da177e4SLinus Torvalds #ifdef STANDALONE_DEBUG
241da177e4SLinus Torvalds #define putstr printf
25a081568dSRussell King #else
26a081568dSRussell King 
27a081568dSRussell King static void putstr(const char *ptr);
28a081568dSRussell King 
29a081568dSRussell King #include <linux/compiler.h>
30a09e64fbSRussell King #include <mach/uncompress.h>
311da177e4SLinus Torvalds 
321da177e4SLinus Torvalds #ifdef CONFIG_DEBUG_ICEDCC
337d95ded9STony Lindgren 
347d95ded9STony Lindgren #ifdef CONFIG_CPU_V6
357d95ded9STony Lindgren 
367d95ded9STony Lindgren static void icedcc_putc(int ch)
377d95ded9STony Lindgren {
387d95ded9STony Lindgren 	int status, i = 0x4000000;
397d95ded9STony Lindgren 
407d95ded9STony Lindgren 	do {
417d95ded9STony Lindgren 		if (--i < 0)
427d95ded9STony Lindgren 			return;
437d95ded9STony Lindgren 
447d95ded9STony Lindgren 		asm volatile ("mrc p14, 0, %0, c0, c1, 0" : "=r" (status));
457d95ded9STony Lindgren 	} while (status & (1 << 29));
467d95ded9STony Lindgren 
477d95ded9STony Lindgren 	asm("mcr p14, 0, %0, c0, c5, 0" : : "r" (ch));
487d95ded9STony Lindgren }
49*c633c3cfSJean-Christop PLAGNIOL-VILLARD #elif defined(CONFIG_CPU_XSCALE)
50*c633c3cfSJean-Christop PLAGNIOL-VILLARD 
51*c633c3cfSJean-Christop PLAGNIOL-VILLARD static void icedcc_putc(int ch)
52*c633c3cfSJean-Christop PLAGNIOL-VILLARD {
53*c633c3cfSJean-Christop PLAGNIOL-VILLARD 	int status, i = 0x4000000;
54*c633c3cfSJean-Christop PLAGNIOL-VILLARD 
55*c633c3cfSJean-Christop PLAGNIOL-VILLARD 	do {
56*c633c3cfSJean-Christop PLAGNIOL-VILLARD 		if (--i < 0)
57*c633c3cfSJean-Christop PLAGNIOL-VILLARD 			return;
58*c633c3cfSJean-Christop PLAGNIOL-VILLARD 
59*c633c3cfSJean-Christop PLAGNIOL-VILLARD 		asm volatile ("mrc p14, 0, %0, c14, c0, 0" : "=r" (status));
60*c633c3cfSJean-Christop PLAGNIOL-VILLARD 	} while (status & (1 << 28));
61*c633c3cfSJean-Christop PLAGNIOL-VILLARD 
62*c633c3cfSJean-Christop PLAGNIOL-VILLARD 	asm("mcr p14, 0, %0, c8, c0, 0" : : "r" (ch));
63*c633c3cfSJean-Christop PLAGNIOL-VILLARD }
647d95ded9STony Lindgren 
657d95ded9STony Lindgren #else
667d95ded9STony Lindgren 
67de4533a0SRussell King static void icedcc_putc(int ch)
68de4533a0SRussell King {
69de4533a0SRussell King 	int status, i = 0x4000000;
70de4533a0SRussell King 
71de4533a0SRussell King 	do {
72de4533a0SRussell King 		if (--i < 0)
73de4533a0SRussell King 			return;
74de4533a0SRussell King 
75b2556da5SUwe Zeisberger 		asm volatile ("mrc p14, 0, %0, c0, c0, 0" : "=r" (status));
76de4533a0SRussell King 	} while (status & 2);
77de4533a0SRussell King 
78b2556da5SUwe Zeisberger 	asm("mcr p14, 0, %0, c1, c0, 0" : : "r" (ch));
79de4533a0SRussell King }
80de4533a0SRussell King 
817d95ded9STony Lindgren #endif
827d95ded9STony Lindgren 
83a081568dSRussell King #define putc(ch)	icedcc_putc(ch)
84a081568dSRussell King #define flush()	do { } while (0)
85a081568dSRussell King #endif
861da177e4SLinus Torvalds 
87a081568dSRussell King static void putstr(const char *ptr)
881da177e4SLinus Torvalds {
89a081568dSRussell King 	char c;
90a081568dSRussell King 
91a081568dSRussell King 	while ((c = *ptr++) != '\0') {
92a081568dSRussell King 		if (c == '\n')
93a081568dSRussell King 			putc('\r');
94a081568dSRussell King 		putc(c);
951da177e4SLinus Torvalds 	}
96a081568dSRussell King 
97a081568dSRussell King 	flush();
981da177e4SLinus Torvalds }
991da177e4SLinus Torvalds 
1001da177e4SLinus Torvalds #endif
1011da177e4SLinus Torvalds 
1021da177e4SLinus Torvalds #define __ptr_t void *
1031da177e4SLinus Torvalds 
10459f0cb0fSRussell King #define memzero(s,n) __memzero(s,n)
10559f0cb0fSRussell King 
1061da177e4SLinus Torvalds /*
1071da177e4SLinus Torvalds  * Optimised C version of memzero for the ARM.
1081da177e4SLinus Torvalds  */
1091da177e4SLinus Torvalds void __memzero (__ptr_t s, size_t n)
1101da177e4SLinus Torvalds {
1111da177e4SLinus Torvalds 	union { void *vp; unsigned long *ulp; unsigned char *ucp; } u;
1121da177e4SLinus Torvalds 	int i;
1131da177e4SLinus Torvalds 
1141da177e4SLinus Torvalds 	u.vp = s;
1151da177e4SLinus Torvalds 
1161da177e4SLinus Torvalds 	for (i = n >> 5; i > 0; i--) {
1171da177e4SLinus Torvalds 		*u.ulp++ = 0;
1181da177e4SLinus Torvalds 		*u.ulp++ = 0;
1191da177e4SLinus Torvalds 		*u.ulp++ = 0;
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 	}
1261da177e4SLinus Torvalds 
1271da177e4SLinus Torvalds 	if (n & 1 << 4) {
1281da177e4SLinus Torvalds 		*u.ulp++ = 0;
1291da177e4SLinus Torvalds 		*u.ulp++ = 0;
1301da177e4SLinus Torvalds 		*u.ulp++ = 0;
1311da177e4SLinus Torvalds 		*u.ulp++ = 0;
1321da177e4SLinus Torvalds 	}
1331da177e4SLinus Torvalds 
1341da177e4SLinus Torvalds 	if (n & 1 << 3) {
1351da177e4SLinus Torvalds 		*u.ulp++ = 0;
1361da177e4SLinus Torvalds 		*u.ulp++ = 0;
1371da177e4SLinus Torvalds 	}
1381da177e4SLinus Torvalds 
1391da177e4SLinus Torvalds 	if (n & 1 << 2)
1401da177e4SLinus Torvalds 		*u.ulp++ = 0;
1411da177e4SLinus Torvalds 
1421da177e4SLinus Torvalds 	if (n & 1 << 1) {
1431da177e4SLinus Torvalds 		*u.ucp++ = 0;
1441da177e4SLinus Torvalds 		*u.ucp++ = 0;
1451da177e4SLinus Torvalds 	}
1461da177e4SLinus Torvalds 
1471da177e4SLinus Torvalds 	if (n & 1)
1481da177e4SLinus Torvalds 		*u.ucp++ = 0;
1491da177e4SLinus Torvalds }
1501da177e4SLinus Torvalds 
1511da177e4SLinus Torvalds static inline __ptr_t memcpy(__ptr_t __dest, __const __ptr_t __src,
1521da177e4SLinus Torvalds 			    size_t __n)
1531da177e4SLinus Torvalds {
1541da177e4SLinus Torvalds 	int i = 0;
1551da177e4SLinus Torvalds 	unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
1561da177e4SLinus Torvalds 
1571da177e4SLinus Torvalds 	for (i = __n >> 3; i > 0; i--) {
1581da177e4SLinus Torvalds 		*d++ = *s++;
1591da177e4SLinus Torvalds 		*d++ = *s++;
1601da177e4SLinus Torvalds 		*d++ = *s++;
1611da177e4SLinus Torvalds 		*d++ = *s++;
1621da177e4SLinus Torvalds 		*d++ = *s++;
1631da177e4SLinus Torvalds 		*d++ = *s++;
1641da177e4SLinus Torvalds 		*d++ = *s++;
1651da177e4SLinus Torvalds 		*d++ = *s++;
1661da177e4SLinus Torvalds 	}
1671da177e4SLinus Torvalds 
1681da177e4SLinus Torvalds 	if (__n & 1 << 2) {
1691da177e4SLinus Torvalds 		*d++ = *s++;
1701da177e4SLinus Torvalds 		*d++ = *s++;
1711da177e4SLinus Torvalds 		*d++ = *s++;
1721da177e4SLinus Torvalds 		*d++ = *s++;
1731da177e4SLinus Torvalds 	}
1741da177e4SLinus Torvalds 
1751da177e4SLinus Torvalds 	if (__n & 1 << 1) {
1761da177e4SLinus Torvalds 		*d++ = *s++;
1771da177e4SLinus Torvalds 		*d++ = *s++;
1781da177e4SLinus Torvalds 	}
1791da177e4SLinus Torvalds 
1801da177e4SLinus Torvalds 	if (__n & 1)
1811da177e4SLinus Torvalds 		*d++ = *s++;
1821da177e4SLinus Torvalds 
1831da177e4SLinus Torvalds 	return __dest;
1841da177e4SLinus Torvalds }
1851da177e4SLinus Torvalds 
1861da177e4SLinus Torvalds /*
1871da177e4SLinus Torvalds  * gzip delarations
1881da177e4SLinus Torvalds  */
1891da177e4SLinus Torvalds #define OF(args)  args
1901da177e4SLinus Torvalds #define STATIC static
1911da177e4SLinus Torvalds 
1921da177e4SLinus Torvalds typedef unsigned char  uch;
1931da177e4SLinus Torvalds typedef unsigned short ush;
1941da177e4SLinus Torvalds typedef unsigned long  ulg;
1951da177e4SLinus Torvalds 
1961da177e4SLinus Torvalds #define WSIZE 0x8000		/* Window size must be at least 32k, */
1971da177e4SLinus Torvalds 				/* and a power of two */
1981da177e4SLinus Torvalds 
1991da177e4SLinus Torvalds static uch *inbuf;		/* input buffer */
2001da177e4SLinus Torvalds static uch window[WSIZE];	/* Sliding window buffer */
2011da177e4SLinus Torvalds 
2021da177e4SLinus Torvalds static unsigned insize;		/* valid bytes in inbuf */
2031da177e4SLinus Torvalds static unsigned inptr;		/* index of next byte to be processed in inbuf */
2041da177e4SLinus Torvalds static unsigned outcnt;		/* bytes in output buffer */
2051da177e4SLinus Torvalds 
2061da177e4SLinus Torvalds /* gzip flag byte */
2071da177e4SLinus Torvalds #define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
2081da177e4SLinus Torvalds #define CONTINUATION 0x02 /* bit 1 set: continuation of multi-part gzip file */
2091da177e4SLinus Torvalds #define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
2101da177e4SLinus Torvalds #define ORIG_NAME    0x08 /* bit 3 set: original file name present */
2111da177e4SLinus Torvalds #define COMMENT      0x10 /* bit 4 set: file comment present */
2121da177e4SLinus Torvalds #define ENCRYPTED    0x20 /* bit 5 set: file is encrypted */
2131da177e4SLinus Torvalds #define RESERVED     0xC0 /* bit 6,7:   reserved */
2141da177e4SLinus Torvalds 
2151da177e4SLinus Torvalds #define get_byte()  (inptr < insize ? inbuf[inptr++] : fill_inbuf())
2161da177e4SLinus Torvalds 
2171da177e4SLinus Torvalds /* Diagnostic functions */
2181da177e4SLinus Torvalds #ifdef DEBUG
2191da177e4SLinus Torvalds #  define Assert(cond,msg) {if(!(cond)) error(msg);}
2201da177e4SLinus Torvalds #  define Trace(x) fprintf x
2211da177e4SLinus Torvalds #  define Tracev(x) {if (verbose) fprintf x ;}
2221da177e4SLinus Torvalds #  define Tracevv(x) {if (verbose>1) fprintf x ;}
2231da177e4SLinus Torvalds #  define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
2241da177e4SLinus Torvalds #  define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
2251da177e4SLinus Torvalds #else
2261da177e4SLinus Torvalds #  define Assert(cond,msg)
2271da177e4SLinus Torvalds #  define Trace(x)
2281da177e4SLinus Torvalds #  define Tracev(x)
2291da177e4SLinus Torvalds #  define Tracevv(x)
2301da177e4SLinus Torvalds #  define Tracec(c,x)
2311da177e4SLinus Torvalds #  define Tracecv(c,x)
2321da177e4SLinus Torvalds #endif
2331da177e4SLinus Torvalds 
2341da177e4SLinus Torvalds static int  fill_inbuf(void);
2351da177e4SLinus Torvalds static void flush_window(void);
2361da177e4SLinus Torvalds static void error(char *m);
2371da177e4SLinus Torvalds 
2381da177e4SLinus Torvalds extern char input_data[];
2391da177e4SLinus Torvalds extern char input_data_end[];
2401da177e4SLinus Torvalds 
2411da177e4SLinus Torvalds static uch *output_data;
2421da177e4SLinus Torvalds static ulg output_ptr;
2431da177e4SLinus Torvalds static ulg bytes_out;
2441da177e4SLinus Torvalds 
2451da177e4SLinus Torvalds static void error(char *m);
2461da177e4SLinus Torvalds 
2471da177e4SLinus Torvalds static void putstr(const char *);
2481da177e4SLinus Torvalds 
2491da177e4SLinus Torvalds extern int end;
2501da177e4SLinus Torvalds static ulg free_mem_ptr;
2512d6ffccaSThomas Petazzoni static ulg free_mem_end_ptr;
2521da177e4SLinus Torvalds 
2532d6ffccaSThomas Petazzoni #ifdef STANDALONE_DEBUG
2542d6ffccaSThomas Petazzoni #define NO_INFLATE_MALLOC
2552d6ffccaSThomas Petazzoni #endif
2562d6ffccaSThomas Petazzoni 
2572d6ffccaSThomas Petazzoni #define ARCH_HAS_DECOMP_WDOG
2581da177e4SLinus Torvalds 
2591da177e4SLinus Torvalds #include "../../../../lib/inflate.c"
2601da177e4SLinus Torvalds 
2611da177e4SLinus Torvalds /* ===========================================================================
2621da177e4SLinus Torvalds  * Fill the input buffer. This is called only when the buffer is empty
2631da177e4SLinus Torvalds  * and at least one byte is really needed.
2641da177e4SLinus Torvalds  */
2651da177e4SLinus Torvalds int fill_inbuf(void)
2661da177e4SLinus Torvalds {
2671da177e4SLinus Torvalds 	if (insize != 0)
2681da177e4SLinus Torvalds 		error("ran out of input data");
2691da177e4SLinus Torvalds 
2701da177e4SLinus Torvalds 	inbuf = input_data;
2711da177e4SLinus Torvalds 	insize = &input_data_end[0] - &input_data[0];
2721da177e4SLinus Torvalds 
2731da177e4SLinus Torvalds 	inptr = 1;
2741da177e4SLinus Torvalds 	return inbuf[0];
2751da177e4SLinus Torvalds }
2761da177e4SLinus Torvalds 
2771da177e4SLinus Torvalds /* ===========================================================================
2781da177e4SLinus Torvalds  * Write the output window window[0..outcnt-1] and update crc and bytes_out.
2791da177e4SLinus Torvalds  * (Used for the decompressed data only.)
2801da177e4SLinus Torvalds  */
2811da177e4SLinus Torvalds void flush_window(void)
2821da177e4SLinus Torvalds {
2831da177e4SLinus Torvalds 	ulg c = crc;
2841da177e4SLinus Torvalds 	unsigned n;
2851da177e4SLinus Torvalds 	uch *in, *out, ch;
2861da177e4SLinus Torvalds 
2871da177e4SLinus Torvalds 	in = window;
2881da177e4SLinus Torvalds 	out = &output_data[output_ptr];
2891da177e4SLinus Torvalds 	for (n = 0; n < outcnt; n++) {
2901da177e4SLinus Torvalds 		ch = *out++ = *in++;
2911da177e4SLinus Torvalds 		c = crc_32_tab[((int)c ^ ch) & 0xff] ^ (c >> 8);
2921da177e4SLinus Torvalds 	}
2931da177e4SLinus Torvalds 	crc = c;
2941da177e4SLinus Torvalds 	bytes_out += (ulg)outcnt;
2951da177e4SLinus Torvalds 	output_ptr += (ulg)outcnt;
2961da177e4SLinus Torvalds 	outcnt = 0;
2971da177e4SLinus Torvalds 	putstr(".");
2981da177e4SLinus Torvalds }
2991da177e4SLinus Torvalds 
300f8c905d3SBen Dooks #ifndef arch_error
301f8c905d3SBen Dooks #define arch_error(x)
302f8c905d3SBen Dooks #endif
303f8c905d3SBen Dooks 
3041da177e4SLinus Torvalds static void error(char *x)
3051da177e4SLinus Torvalds {
306f8c905d3SBen Dooks 	arch_error(x);
307f8c905d3SBen Dooks 
3081da177e4SLinus Torvalds 	putstr("\n\n");
3091da177e4SLinus Torvalds 	putstr(x);
3101da177e4SLinus Torvalds 	putstr("\n\n -- System halted");
3111da177e4SLinus Torvalds 
3121da177e4SLinus Torvalds 	while(1);	/* Halt */
3131da177e4SLinus Torvalds }
3141da177e4SLinus Torvalds 
3151da177e4SLinus Torvalds #ifndef STANDALONE_DEBUG
3161da177e4SLinus Torvalds 
3171da177e4SLinus Torvalds ulg
3181da177e4SLinus Torvalds decompress_kernel(ulg output_start, ulg free_mem_ptr_p, ulg free_mem_ptr_end_p,
3191da177e4SLinus Torvalds 		  int arch_id)
3201da177e4SLinus Torvalds {
3211da177e4SLinus Torvalds 	output_data		= (uch *)output_start;	/* Points to kernel start */
3221da177e4SLinus Torvalds 	free_mem_ptr		= free_mem_ptr_p;
3232d6ffccaSThomas Petazzoni 	free_mem_end_ptr	= free_mem_ptr_end_p;
3241da177e4SLinus Torvalds 	__machine_arch_type	= arch_id;
3251da177e4SLinus Torvalds 
3261da177e4SLinus Torvalds 	arch_decomp_setup();
3271da177e4SLinus Torvalds 
3281da177e4SLinus Torvalds 	makecrc();
3291da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3301da177e4SLinus Torvalds 	gunzip();
3311da177e4SLinus Torvalds 	putstr(" done, booting the kernel.\n");
3321da177e4SLinus Torvalds 	return output_ptr;
3331da177e4SLinus Torvalds }
3341da177e4SLinus Torvalds #else
3351da177e4SLinus Torvalds 
3361da177e4SLinus Torvalds char output_buffer[1500*1024];
3371da177e4SLinus Torvalds 
3381da177e4SLinus Torvalds int main()
3391da177e4SLinus Torvalds {
3401da177e4SLinus Torvalds 	output_data = output_buffer;
3411da177e4SLinus Torvalds 
3421da177e4SLinus Torvalds 	makecrc();
3431da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3441da177e4SLinus Torvalds 	gunzip();
3451da177e4SLinus Torvalds 	putstr("done.\n");
3461da177e4SLinus Torvalds 	return 0;
3471da177e4SLinus Torvalds }
3481da177e4SLinus Torvalds #endif
3491da177e4SLinus Torvalds 
350