Rizin
unix-like reverse engineering framework and cli tools
Macros | Functions
mztools.c File Reference
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "zlib.h"
#include "unzip.h"

Go to the source code of this file.

Macros

#define READ_8(adr)   ((unsigned char)*(adr))
 
#define READ_16(adr)   ( READ_8(adr) | (READ_8(adr+1) << 8) )
 
#define READ_32(adr)   ( READ_16(adr) | (READ_16((adr)+2) << 16) )
 
#define WRITE_8(buff, n)
 
#define WRITE_16(buff, n)
 
#define WRITE_32(buff, n)
 

Functions

int ZEXPORT unzRepair (char *file, const char *fileOut, const char *fileOutTmp, uLong *nRecovered, uLong *bytesRecovered) const
 

Macro Definition Documentation

◆ READ_16

#define READ_16 (   adr)    ( READ_8(adr) | (READ_8(adr+1) << 8) )

Definition at line 15 of file mztools.c.

◆ READ_32

#define READ_32 (   adr)    ( READ_16(adr) | (READ_16((adr)+2) << 16) )

Definition at line 16 of file mztools.c.

◆ READ_8

#define READ_8 (   adr)    ((unsigned char)*(adr))

Definition at line 14 of file mztools.c.

◆ WRITE_16

#define WRITE_16 (   buff,
  n 
)
Value:
do { \
WRITE_8((unsigned char*)(buff), n); \
WRITE_8(((unsigned char*)(buff)) + 1, (n) >> 8); \
} while(0)
n
int n
Definition: mipsasm.c:19

Definition at line 21 of file mztools.c.

◆ WRITE_32

#define WRITE_32 (   buff,
  n 
)
Value:
do { \
WRITE_16((unsigned char*)(buff), (n) & 0xffff); \
WRITE_16((unsigned char*)(buff) + 2, (n) >> 16); \
} while(0)

Definition at line 25 of file mztools.c.

◆ WRITE_8

#define WRITE_8 (   buff,
  n 
)
Value:
do { \
*((unsigned char*)(buff)) = (unsigned char) ((n) & 0xff); \
} while(0)

Definition at line 18 of file mztools.c.

Function Documentation

◆ unzRepair()

int ZEXPORT unzRepair ( char*  file,
const char*  fileOut,
const char*  fileOutTmp,
uLong*  nRecovered,
uLong*  bytesRecovered 
) const

Definition at line 30 of file mztools.c.

36 {
37  int err = Z_OK;
38  FILE* fpZip = fopen(file, "rb");
39  FILE* fpOut = fopen(fileOut, "wb");
40  FILE* fpOutCD = fopen(fileOutTmp, "wb");
41  if (fpZip != NULL && fpOut != NULL) {
42  int entries = 0;
43  uLong totalBytes = 0;
44  char header[30];
45  char filename[1024];
46  char extra[1024];
47  int offset = 0;
48  int offsetCD = 0;
49  while ( fread(header, 1, 30, fpZip) == 30 ) {
50  int currentOffset = offset;
51 
52  /* File entry */
53  if (READ_32(header) == 0x04034b50) {
54  unsigned int version = READ_16(header + 4);
55  unsigned int gpflag = READ_16(header + 6);
56  unsigned int method = READ_16(header + 8);
57  unsigned int filetime = READ_16(header + 10);
58  unsigned int filedate = READ_16(header + 12);
59  unsigned int crc = READ_32(header + 14); /* crc */
60  unsigned int cpsize = READ_32(header + 18); /* compressed size */
61  unsigned int uncpsize = READ_32(header + 22); /* uncompressed sz */
62  unsigned int fnsize = READ_16(header + 26); /* file name length */
63  unsigned int extsize = READ_16(header + 28); /* extra field length */
64  filename[0] = extra[0] = '\0';
65 
66  /* Header */
67  if (fwrite(header, 1, 30, fpOut) == 30) {
68  offset += 30;
69  } else {
70  err = Z_ERRNO;
71  break;
72  }
73 
74  /* Filename */
75  if (fnsize > 0) {
76  if (fnsize < sizeof(filename)) {
77  if (fread(filename, 1, fnsize, fpZip) == fnsize) {
78  if (fwrite(filename, 1, fnsize, fpOut) == fnsize) {
79  offset += fnsize;
80  } else {
81  err = Z_ERRNO;
82  break;
83  }
84  } else {
85  err = Z_ERRNO;
86  break;
87  }
88  } else {
89  err = Z_ERRNO;
90  break;
91  }
92  } else {
93  err = Z_STREAM_ERROR;
94  break;
95  }
96 
97  /* Extra field */
98  if (extsize > 0) {
99  if (extsize < sizeof(extra)) {
100  if (fread(extra, 1, extsize, fpZip) == extsize) {
101  if (fwrite(extra, 1, extsize, fpOut) == extsize) {
102  offset += extsize;
103  } else {
104  err = Z_ERRNO;
105  break;
106  }
107  } else {
108  err = Z_ERRNO;
109  break;
110  }
111  } else {
112  err = Z_ERRNO;
113  break;
114  }
115  }
116 
117  /* Data */
118  {
119  int dataSize = cpsize;
120  if (dataSize == 0) {
121  dataSize = uncpsize;
122  }
123  if (dataSize > 0) {
124  char* data = malloc(dataSize);
125  if (data != NULL) {
126  if ((int)fread(data, 1, dataSize, fpZip) == dataSize) {
127  if ((int)fwrite(data, 1, dataSize, fpOut) == dataSize) {
128  offset += dataSize;
129  totalBytes += dataSize;
130  } else {
131  err = Z_ERRNO;
132  }
133  } else {
134  err = Z_ERRNO;
135  }
136  free(data);
137  if (err != Z_OK) {
138  break;
139  }
140  } else {
141  err = Z_MEM_ERROR;
142  break;
143  }
144  }
145  }
146 
147  /* Central directory entry */
148  {
149  char header[46];
150  char* comment = "";
151  int comsize = (int) strlen(comment);
152  WRITE_32(header, 0x02014b50);
153  WRITE_16(header + 4, version);
154  WRITE_16(header + 6, version);
155  WRITE_16(header + 8, gpflag);
156  WRITE_16(header + 10, method);
157  WRITE_16(header + 12, filetime);
158  WRITE_16(header + 14, filedate);
159  WRITE_32(header + 16, crc);
160  WRITE_32(header + 20, cpsize);
161  WRITE_32(header + 24, uncpsize);
162  WRITE_16(header + 28, fnsize);
163  WRITE_16(header + 30, extsize);
164  WRITE_16(header + 32, comsize);
165  WRITE_16(header + 34, 0); /* disk # */
166  WRITE_16(header + 36, 0); /* int attrb */
167  WRITE_32(header + 38, 0); /* ext attrb */
168  WRITE_32(header + 42, currentOffset);
169  /* Header */
170  if (fwrite(header, 1, 46, fpOutCD) == 46) {
171  offsetCD += 46;
172 
173  /* Filename */
174  if (fnsize > 0) {
175  if (fwrite(filename, 1, fnsize, fpOutCD) == fnsize) {
176  offsetCD += fnsize;
177  } else {
178  err = Z_ERRNO;
179  break;
180  }
181  } else {
182  err = Z_STREAM_ERROR;
183  break;
184  }
185 
186  /* Extra field */
187  if (extsize > 0) {
188  if (fwrite(extra, 1, extsize, fpOutCD) == extsize) {
189  offsetCD += extsize;
190  } else {
191  err = Z_ERRNO;
192  break;
193  }
194  }
195 
196  /* Comment field */
197  if (comsize > 0) {
198  if ((int)fwrite(comment, 1, comsize, fpOutCD) == comsize) {
199  offsetCD += comsize;
200  } else {
201  err = Z_ERRNO;
202  break;
203  }
204  }
205 
206 
207  } else {
208  err = Z_ERRNO;
209  break;
210  }
211  }
212 
213  /* Success */
214  entries++;
215 
216  } else {
217  break;
218  }
219  }
220 
221  /* Final central directory */
222  {
223  int entriesZip = entries;
224  char header[22];
225  char* comment = ""; // "ZIP File recovered by zlib/minizip/mztools";
226  int comsize = (int) strlen(comment);
227  if (entriesZip > 0xffff) {
228  entriesZip = 0xffff;
229  }
230  WRITE_32(header, 0x06054b50);
231  WRITE_16(header + 4, 0); /* disk # */
232  WRITE_16(header + 6, 0); /* disk # */
233  WRITE_16(header + 8, entriesZip); /* hack */
234  WRITE_16(header + 10, entriesZip); /* hack */
235  WRITE_32(header + 12, offsetCD); /* size of CD */
236  WRITE_32(header + 16, offset); /* offset to CD */
237  WRITE_16(header + 20, comsize); /* comment */
238 
239  /* Header */
240  if (fwrite(header, 1, 22, fpOutCD) == 22) {
241 
242  /* Comment field */
243  if (comsize > 0) {
244  if ((int)fwrite(comment, 1, comsize, fpOutCD) != comsize) {
245  err = Z_ERRNO;
246  }
247  }
248 
249  } else {
250  err = Z_ERRNO;
251  }
252  }
253 
254  /* Final merge (file + central directory) */
255  fclose(fpOutCD);
256  if (err == Z_OK) {
257  fpOutCD = fopen(fileOutTmp, "rb");
258  if (fpOutCD != NULL) {
259  int nRead;
260  char buffer[8192];
261  while ( (nRead = (int)fread(buffer, 1, sizeof(buffer), fpOutCD)) > 0) {
262  if ((int)fwrite(buffer, 1, nRead, fpOut) != nRead) {
263  err = Z_ERRNO;
264  break;
265  }
266  }
267  fclose(fpOutCD);
268  }
269  }
270 
271  /* Close */
272  fclose(fpZip);
273  fclose(fpOut);
274 
275  /* Wipe temporary file */
276  (void)remove(fileOutTmp);
277 
278  /* Number of recovered entries */
279  if (err == Z_OK) {
280  if (nRecovered != NULL) {
281  *nRecovered = entries;
282  }
283  if (bytesRecovered != NULL) {
284  *bytesRecovered = totalBytes;
285  }
286  }
287  } else {
288  err = Z_STREAM_ERROR;
289  }
290  return err;
291 }
err
static bool err
Definition: armass.c:435
entries
RzList * entries(RzBinFile *bf)
Definition: bin_ne.c:98
NULL
#define NULL
Definition: cris-opc.c:27
free
RZ_API void Ht_() free(HtName_(Ht) *ht)
Definition: ht_inc.c:130
filename
const char * filename
Definition: ioapi.h:137
offset
voidpf uLong offset
Definition: ioapi.h:144
malloc
void * malloc(size_t size)
Definition: malloc.c:123
header
#define header(is_bt, len_min, ret_op)
Definition: lz_encoder_mf.c:191
filetime
uLong filetime(char *f, tm_zip *tmzip, uLong *dt) const
Definition: minizip.c:141
READ_32
#define READ_32(adr)
Definition: mztools.c:16
READ_16
#define READ_16(adr)
Definition: mztools.c:15
WRITE_16
#define WRITE_16(buff, n)
Definition: mztools.c:21
WRITE_32
#define WRITE_32(buff, n)
Definition: mztools.c:25
benchmark.FILE
string FILE
Definition: benchmark.py:21
int
static int
Definition: sfsocketcall.h:114
buffer
Definition: buffer.h:15
file
Definition: gzappend.c:170
uLong
unsigned long uLong
Definition: zconf.h:394
Z_ERRNO
#define Z_ERRNO
Definition: zlib.h:180
Z_OK
#define Z_OK
Definition: zlib.h:177
Z_STREAM_ERROR
#define Z_STREAM_ERROR
Definition: zlib.h:181
Z_MEM_ERROR
#define Z_MEM_ERROR
Definition: zlib.h:183

References entries(), err, benchmark::FILE, filetime(), free(), header, int, malloc(), NULL, READ_16, READ_32, WRITE_16, WRITE_32, Z_ERRNO, Z_MEM_ERROR, Z_OK, and Z_STREAM_ERROR.