Blame view

bootloader/u-boot_2015_04/lib/gunzip.c 2.09 KB
6b13f685e   김민수   BSP 최초 추가
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
  /*
   * (C) Copyright 2000-2006
   * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include <common.h>
  #include <watchdog.h>
  #include <command.h>
  #include <image.h>
  #include <malloc.h>
  #include <u-boot/zlib.h>
  
  #define	ZALLOC_ALIGNMENT	16
  #define HEAD_CRC		2
  #define EXTRA_FIELD		4
  #define ORIG_NAME		8
  #define COMMENT			0x10
  #define RESERVED		0xe0
  #define DEFLATED		8
  
  void *gzalloc(void *x, unsigned items, unsigned size)
  {
  	void *p;
  
  	size *= items;
  	size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1);
  
  	p = malloc (size);
  
  	return (p);
  }
  
  void gzfree(void *x, void *addr, unsigned nb)
  {
  	free (addr);
  }
  
  int gunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp)
  {
  	int i, flags;
  
  	/* skip header */
  	i = 10;
  	flags = src[3];
  	if (src[2] != DEFLATED || (flags & RESERVED) != 0) {
  		puts ("Error: Bad gzipped data
  ");
  		return (-1);
  	}
  	if ((flags & EXTRA_FIELD) != 0)
  		i = 12 + src[10] + (src[11] << 8);
  	if ((flags & ORIG_NAME) != 0)
  		while (src[i++] != 0)
  			;
  	if ((flags & COMMENT) != 0)
  		while (src[i++] != 0)
  			;
  	if ((flags & HEAD_CRC) != 0)
  		i += 2;
  	if (i >= *lenp) {
  		puts ("Error: gunzip out of data in header
  ");
  		return (-1);
  	}
  
  	return zunzip(dst, dstlen, src, lenp, 1, i);
  }
  
  /*
   * Uncompress blocks compressed with zlib without headers
   */
  int zunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp,
  						int stoponerr, int offset)
  {
  	z_stream s;
  	int err = 0;
  	int r;
  
  	s.zalloc = gzalloc;
  	s.zfree = gzfree;
  
  	r = inflateInit2(&s, -MAX_WBITS);
  	if (r != Z_OK) {
  		printf ("Error: inflateInit2() returned %d
  ", r);
  		return -1;
  	}
  	s.next_in = src + offset;
  	s.avail_in = *lenp - offset;
  	s.next_out = dst;
  	s.avail_out = dstlen;
  	do {
  		r = inflate(&s, Z_FINISH);
  		if (stoponerr == 1 && r != Z_STREAM_END &&
  		    (s.avail_out == 0 || r != Z_BUF_ERROR)) {
  			printf("Error: inflate() returned %d
  ", r);
  			err = -1;
  			break;
  		}
  		s.avail_in = *lenp - offset - (int)(s.next_out - (unsigned char*)dst);
  	} while (r == Z_BUF_ERROR);
  	*lenp = s.next_out - (unsigned char *) dst;
  	inflateEnd(&s);
  
  	return err;
  }