xref: /openbmc/linux/arch/arm/boot/compressed/misc.c (revision 2d6ffcca623a9a16df6cdfbe8250b7a5904a5f5e)
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>
30a081568dSRussell King #include <asm/arch/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 }
497d95ded9STony Lindgren 
507d95ded9STony Lindgren #else
517d95ded9STony Lindgren 
52de4533a0SRussell King static void icedcc_putc(int ch)
53de4533a0SRussell King {
54de4533a0SRussell King 	int status, i = 0x4000000;
55de4533a0SRussell King 
56de4533a0SRussell King 	do {
57de4533a0SRussell King 		if (--i < 0)
58de4533a0SRussell King 			return;
59de4533a0SRussell King 
60b2556da5SUwe Zeisberger 		asm volatile ("mrc p14, 0, %0, c0, c0, 0" : "=r" (status));
61de4533a0SRussell King 	} while (status & 2);
62de4533a0SRussell King 
63b2556da5SUwe Zeisberger 	asm("mcr p14, 0, %0, c1, c0, 0" : : "r" (ch));
64de4533a0SRussell King }
65de4533a0SRussell King 
667d95ded9STony Lindgren #endif
677d95ded9STony Lindgren 
68a081568dSRussell King #define putc(ch)	icedcc_putc(ch)
69a081568dSRussell King #define flush()	do { } while (0)
70a081568dSRussell King #endif
711da177e4SLinus Torvalds 
72a081568dSRussell King static void putstr(const char *ptr)
731da177e4SLinus Torvalds {
74a081568dSRussell King 	char c;
75a081568dSRussell King 
76a081568dSRussell King 	while ((c = *ptr++) != '\0') {
77a081568dSRussell King 		if (c == '\n')
78a081568dSRussell King 			putc('\r');
79a081568dSRussell King 		putc(c);
801da177e4SLinus Torvalds 	}
81a081568dSRussell King 
82a081568dSRussell King 	flush();
831da177e4SLinus Torvalds }
841da177e4SLinus Torvalds 
851da177e4SLinus Torvalds #endif
861da177e4SLinus Torvalds 
871da177e4SLinus Torvalds #define __ptr_t void *
881da177e4SLinus Torvalds 
891da177e4SLinus Torvalds /*
901da177e4SLinus Torvalds  * Optimised C version of memzero for the ARM.
911da177e4SLinus Torvalds  */
921da177e4SLinus Torvalds void __memzero (__ptr_t s, size_t n)
931da177e4SLinus Torvalds {
941da177e4SLinus Torvalds 	union { void *vp; unsigned long *ulp; unsigned char *ucp; } u;
951da177e4SLinus Torvalds 	int i;
961da177e4SLinus Torvalds 
971da177e4SLinus Torvalds 	u.vp = s;
981da177e4SLinus Torvalds 
991da177e4SLinus Torvalds 	for (i = n >> 5; i > 0; i--) {
1001da177e4SLinus Torvalds 		*u.ulp++ = 0;
1011da177e4SLinus Torvalds 		*u.ulp++ = 0;
1021da177e4SLinus Torvalds 		*u.ulp++ = 0;
1031da177e4SLinus Torvalds 		*u.ulp++ = 0;
1041da177e4SLinus Torvalds 		*u.ulp++ = 0;
1051da177e4SLinus Torvalds 		*u.ulp++ = 0;
1061da177e4SLinus Torvalds 		*u.ulp++ = 0;
1071da177e4SLinus Torvalds 		*u.ulp++ = 0;
1081da177e4SLinus Torvalds 	}
1091da177e4SLinus Torvalds 
1101da177e4SLinus Torvalds 	if (n & 1 << 4) {
1111da177e4SLinus Torvalds 		*u.ulp++ = 0;
1121da177e4SLinus Torvalds 		*u.ulp++ = 0;
1131da177e4SLinus Torvalds 		*u.ulp++ = 0;
1141da177e4SLinus Torvalds 		*u.ulp++ = 0;
1151da177e4SLinus Torvalds 	}
1161da177e4SLinus Torvalds 
1171da177e4SLinus Torvalds 	if (n & 1 << 3) {
1181da177e4SLinus Torvalds 		*u.ulp++ = 0;
1191da177e4SLinus Torvalds 		*u.ulp++ = 0;
1201da177e4SLinus Torvalds 	}
1211da177e4SLinus Torvalds 
1221da177e4SLinus Torvalds 	if (n & 1 << 2)
1231da177e4SLinus Torvalds 		*u.ulp++ = 0;
1241da177e4SLinus Torvalds 
1251da177e4SLinus Torvalds 	if (n & 1 << 1) {
1261da177e4SLinus Torvalds 		*u.ucp++ = 0;
1271da177e4SLinus Torvalds 		*u.ucp++ = 0;
1281da177e4SLinus Torvalds 	}
1291da177e4SLinus Torvalds 
1301da177e4SLinus Torvalds 	if (n & 1)
1311da177e4SLinus Torvalds 		*u.ucp++ = 0;
1321da177e4SLinus Torvalds }
1331da177e4SLinus Torvalds 
1341da177e4SLinus Torvalds static inline __ptr_t memcpy(__ptr_t __dest, __const __ptr_t __src,
1351da177e4SLinus Torvalds 			    size_t __n)
1361da177e4SLinus Torvalds {
1371da177e4SLinus Torvalds 	int i = 0;
1381da177e4SLinus Torvalds 	unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
1391da177e4SLinus Torvalds 
1401da177e4SLinus Torvalds 	for (i = __n >> 3; i > 0; i--) {
1411da177e4SLinus Torvalds 		*d++ = *s++;
1421da177e4SLinus Torvalds 		*d++ = *s++;
1431da177e4SLinus Torvalds 		*d++ = *s++;
1441da177e4SLinus Torvalds 		*d++ = *s++;
1451da177e4SLinus Torvalds 		*d++ = *s++;
1461da177e4SLinus Torvalds 		*d++ = *s++;
1471da177e4SLinus Torvalds 		*d++ = *s++;
1481da177e4SLinus Torvalds 		*d++ = *s++;
1491da177e4SLinus Torvalds 	}
1501da177e4SLinus Torvalds 
1511da177e4SLinus Torvalds 	if (__n & 1 << 2) {
1521da177e4SLinus Torvalds 		*d++ = *s++;
1531da177e4SLinus Torvalds 		*d++ = *s++;
1541da177e4SLinus Torvalds 		*d++ = *s++;
1551da177e4SLinus Torvalds 		*d++ = *s++;
1561da177e4SLinus Torvalds 	}
1571da177e4SLinus Torvalds 
1581da177e4SLinus Torvalds 	if (__n & 1 << 1) {
1591da177e4SLinus Torvalds 		*d++ = *s++;
1601da177e4SLinus Torvalds 		*d++ = *s++;
1611da177e4SLinus Torvalds 	}
1621da177e4SLinus Torvalds 
1631da177e4SLinus Torvalds 	if (__n & 1)
1641da177e4SLinus Torvalds 		*d++ = *s++;
1651da177e4SLinus Torvalds 
1661da177e4SLinus Torvalds 	return __dest;
1671da177e4SLinus Torvalds }
1681da177e4SLinus Torvalds 
1691da177e4SLinus Torvalds /*
1701da177e4SLinus Torvalds  * gzip delarations
1711da177e4SLinus Torvalds  */
1721da177e4SLinus Torvalds #define OF(args)  args
1731da177e4SLinus Torvalds #define STATIC static
1741da177e4SLinus Torvalds 
1751da177e4SLinus Torvalds typedef unsigned char  uch;
1761da177e4SLinus Torvalds typedef unsigned short ush;
1771da177e4SLinus Torvalds typedef unsigned long  ulg;
1781da177e4SLinus Torvalds 
1791da177e4SLinus Torvalds #define WSIZE 0x8000		/* Window size must be at least 32k, */
1801da177e4SLinus Torvalds 				/* and a power of two */
1811da177e4SLinus Torvalds 
1821da177e4SLinus Torvalds static uch *inbuf;		/* input buffer */
1831da177e4SLinus Torvalds static uch window[WSIZE];	/* Sliding window buffer */
1841da177e4SLinus Torvalds 
1851da177e4SLinus Torvalds static unsigned insize;		/* valid bytes in inbuf */
1861da177e4SLinus Torvalds static unsigned inptr;		/* index of next byte to be processed in inbuf */
1871da177e4SLinus Torvalds static unsigned outcnt;		/* bytes in output buffer */
1881da177e4SLinus Torvalds 
1891da177e4SLinus Torvalds /* gzip flag byte */
1901da177e4SLinus Torvalds #define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
1911da177e4SLinus Torvalds #define CONTINUATION 0x02 /* bit 1 set: continuation of multi-part gzip file */
1921da177e4SLinus Torvalds #define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
1931da177e4SLinus Torvalds #define ORIG_NAME    0x08 /* bit 3 set: original file name present */
1941da177e4SLinus Torvalds #define COMMENT      0x10 /* bit 4 set: file comment present */
1951da177e4SLinus Torvalds #define ENCRYPTED    0x20 /* bit 5 set: file is encrypted */
1961da177e4SLinus Torvalds #define RESERVED     0xC0 /* bit 6,7:   reserved */
1971da177e4SLinus Torvalds 
1981da177e4SLinus Torvalds #define get_byte()  (inptr < insize ? inbuf[inptr++] : fill_inbuf())
1991da177e4SLinus Torvalds 
2001da177e4SLinus Torvalds /* Diagnostic functions */
2011da177e4SLinus Torvalds #ifdef DEBUG
2021da177e4SLinus Torvalds #  define Assert(cond,msg) {if(!(cond)) error(msg);}
2031da177e4SLinus Torvalds #  define Trace(x) fprintf x
2041da177e4SLinus Torvalds #  define Tracev(x) {if (verbose) fprintf x ;}
2051da177e4SLinus Torvalds #  define Tracevv(x) {if (verbose>1) fprintf x ;}
2061da177e4SLinus Torvalds #  define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
2071da177e4SLinus Torvalds #  define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
2081da177e4SLinus Torvalds #else
2091da177e4SLinus Torvalds #  define Assert(cond,msg)
2101da177e4SLinus Torvalds #  define Trace(x)
2111da177e4SLinus Torvalds #  define Tracev(x)
2121da177e4SLinus Torvalds #  define Tracevv(x)
2131da177e4SLinus Torvalds #  define Tracec(c,x)
2141da177e4SLinus Torvalds #  define Tracecv(c,x)
2151da177e4SLinus Torvalds #endif
2161da177e4SLinus Torvalds 
2171da177e4SLinus Torvalds static int  fill_inbuf(void);
2181da177e4SLinus Torvalds static void flush_window(void);
2191da177e4SLinus Torvalds static void error(char *m);
2201da177e4SLinus Torvalds 
2211da177e4SLinus Torvalds extern char input_data[];
2221da177e4SLinus Torvalds extern char input_data_end[];
2231da177e4SLinus Torvalds 
2241da177e4SLinus Torvalds static uch *output_data;
2251da177e4SLinus Torvalds static ulg output_ptr;
2261da177e4SLinus Torvalds static ulg bytes_out;
2271da177e4SLinus Torvalds 
2281da177e4SLinus Torvalds static void error(char *m);
2291da177e4SLinus Torvalds 
2301da177e4SLinus Torvalds static void putstr(const char *);
2311da177e4SLinus Torvalds 
2321da177e4SLinus Torvalds extern int end;
2331da177e4SLinus Torvalds static ulg free_mem_ptr;
234*2d6ffccaSThomas Petazzoni static ulg free_mem_end_ptr;
2351da177e4SLinus Torvalds 
236*2d6ffccaSThomas Petazzoni #ifdef STANDALONE_DEBUG
237*2d6ffccaSThomas Petazzoni #define NO_INFLATE_MALLOC
238*2d6ffccaSThomas Petazzoni #endif
239*2d6ffccaSThomas Petazzoni 
240*2d6ffccaSThomas Petazzoni #define ARCH_HAS_DECOMP_WDOG
2411da177e4SLinus Torvalds 
2421da177e4SLinus Torvalds #include "../../../../lib/inflate.c"
2431da177e4SLinus Torvalds 
2441da177e4SLinus Torvalds /* ===========================================================================
2451da177e4SLinus Torvalds  * Fill the input buffer. This is called only when the buffer is empty
2461da177e4SLinus Torvalds  * and at least one byte is really needed.
2471da177e4SLinus Torvalds  */
2481da177e4SLinus Torvalds int fill_inbuf(void)
2491da177e4SLinus Torvalds {
2501da177e4SLinus Torvalds 	if (insize != 0)
2511da177e4SLinus Torvalds 		error("ran out of input data");
2521da177e4SLinus Torvalds 
2531da177e4SLinus Torvalds 	inbuf = input_data;
2541da177e4SLinus Torvalds 	insize = &input_data_end[0] - &input_data[0];
2551da177e4SLinus Torvalds 
2561da177e4SLinus Torvalds 	inptr = 1;
2571da177e4SLinus Torvalds 	return inbuf[0];
2581da177e4SLinus Torvalds }
2591da177e4SLinus Torvalds 
2601da177e4SLinus Torvalds /* ===========================================================================
2611da177e4SLinus Torvalds  * Write the output window window[0..outcnt-1] and update crc and bytes_out.
2621da177e4SLinus Torvalds  * (Used for the decompressed data only.)
2631da177e4SLinus Torvalds  */
2641da177e4SLinus Torvalds void flush_window(void)
2651da177e4SLinus Torvalds {
2661da177e4SLinus Torvalds 	ulg c = crc;
2671da177e4SLinus Torvalds 	unsigned n;
2681da177e4SLinus Torvalds 	uch *in, *out, ch;
2691da177e4SLinus Torvalds 
2701da177e4SLinus Torvalds 	in = window;
2711da177e4SLinus Torvalds 	out = &output_data[output_ptr];
2721da177e4SLinus Torvalds 	for (n = 0; n < outcnt; n++) {
2731da177e4SLinus Torvalds 		ch = *out++ = *in++;
2741da177e4SLinus Torvalds 		c = crc_32_tab[((int)c ^ ch) & 0xff] ^ (c >> 8);
2751da177e4SLinus Torvalds 	}
2761da177e4SLinus Torvalds 	crc = c;
2771da177e4SLinus Torvalds 	bytes_out += (ulg)outcnt;
2781da177e4SLinus Torvalds 	output_ptr += (ulg)outcnt;
2791da177e4SLinus Torvalds 	outcnt = 0;
2801da177e4SLinus Torvalds 	putstr(".");
2811da177e4SLinus Torvalds }
2821da177e4SLinus Torvalds 
283f8c905d3SBen Dooks #ifndef arch_error
284f8c905d3SBen Dooks #define arch_error(x)
285f8c905d3SBen Dooks #endif
286f8c905d3SBen Dooks 
2871da177e4SLinus Torvalds static void error(char *x)
2881da177e4SLinus Torvalds {
289f8c905d3SBen Dooks 	arch_error(x);
290f8c905d3SBen Dooks 
2911da177e4SLinus Torvalds 	putstr("\n\n");
2921da177e4SLinus Torvalds 	putstr(x);
2931da177e4SLinus Torvalds 	putstr("\n\n -- System halted");
2941da177e4SLinus Torvalds 
2951da177e4SLinus Torvalds 	while(1);	/* Halt */
2961da177e4SLinus Torvalds }
2971da177e4SLinus Torvalds 
2981da177e4SLinus Torvalds #ifndef STANDALONE_DEBUG
2991da177e4SLinus Torvalds 
3001da177e4SLinus Torvalds ulg
3011da177e4SLinus Torvalds decompress_kernel(ulg output_start, ulg free_mem_ptr_p, ulg free_mem_ptr_end_p,
3021da177e4SLinus Torvalds 		  int arch_id)
3031da177e4SLinus Torvalds {
3041da177e4SLinus Torvalds 	output_data		= (uch *)output_start;	/* Points to kernel start */
3051da177e4SLinus Torvalds 	free_mem_ptr		= free_mem_ptr_p;
306*2d6ffccaSThomas Petazzoni 	free_mem_end_ptr	= free_mem_ptr_end_p;
3071da177e4SLinus Torvalds 	__machine_arch_type	= arch_id;
3081da177e4SLinus Torvalds 
3091da177e4SLinus Torvalds 	arch_decomp_setup();
3101da177e4SLinus Torvalds 
3111da177e4SLinus Torvalds 	makecrc();
3121da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3131da177e4SLinus Torvalds 	gunzip();
3141da177e4SLinus Torvalds 	putstr(" done, booting the kernel.\n");
3151da177e4SLinus Torvalds 	return output_ptr;
3161da177e4SLinus Torvalds }
3171da177e4SLinus Torvalds #else
3181da177e4SLinus Torvalds 
3191da177e4SLinus Torvalds char output_buffer[1500*1024];
3201da177e4SLinus Torvalds 
3211da177e4SLinus Torvalds int main()
3221da177e4SLinus Torvalds {
3231da177e4SLinus Torvalds 	output_data = output_buffer;
3241da177e4SLinus Torvalds 
3251da177e4SLinus Torvalds 	makecrc();
3261da177e4SLinus Torvalds 	putstr("Uncompressing Linux...");
3271da177e4SLinus Torvalds 	gunzip();
3281da177e4SLinus Torvalds 	putstr("done.\n");
3291da177e4SLinus Torvalds 	return 0;
3301da177e4SLinus Torvalds }
3311da177e4SLinus Torvalds #endif
3321da177e4SLinus Torvalds 
333