File: ZipFile.cpp

package info (click to toggle)
numptyphysics 0.2+svn157-0.3
  • links: PTS
  • area: main
  • in suites: buster, jessie, jessie-kfreebsd, sid, stretch
  • size: 2,596 kB
  • ctags: 2,896
  • sloc: cpp: 18,280; sh: 810; makefile: 205
file content (185 lines) | stat: -rw-r--r-- 4,467 bytes parent folder | download | duplicates (3)
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
/*
 * This file is part of NumptyPhysics
 * Copyright (C) 2008 Tim Edmonds
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 */

#include "ZipFile.h"
#include <string>
#include <cstring>
#include <cstdio>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <unistd.h>
#include <zlib.h>

//zip file structs and uncompress_int below lifted from navit project (GPL).
struct zip_lfh {
	int ziplocsig;
	short zipver;
	short zipgenfld;
	short zipmthd;
	short ziptime;
	short zipdate;
	int zipcrc;
	unsigned int zipsize;
	unsigned int zipuncmp;
	unsigned short zipfnln;
	unsigned short zipxtraln;
	char zipname[0];
} __attribute__ ((packed));

struct zip_cd {
	int zipcensig;
	char zipcver;
	char zipcos;
	char zipcvxt;
	char zipcexos;
	short zipcflg;
	short zipcmthd;
	short ziptim;
	short zipdat;
	int zipccrc;
	unsigned int zipcsiz;
	unsigned int zipcunc;
	unsigned short zipcfnl;
	unsigned short zipcxtl;
	unsigned short zipccml;
	unsigned short zipdsk;
	unsigned short zipint;
	unsigned int zipext;
	unsigned int zipofst;
	char zipcfn[0];	
} __attribute__ ((packed));

struct zip_eoc {
	int zipesig;
	unsigned short zipedsk;
	unsigned short zipecen;
	unsigned short zipenum;
	unsigned short zipecenn;
	unsigned int zipecsz;
	unsigned int zipeofst;
	short zipecoml;
	char zipecom[0];
} __attribute__ ((packed));

int uncompress_int(unsigned char *dest, int *destLen, 
		   const unsigned char *source, int sourceLen)
{
  z_stream stream;
  int err;
  
  stream.next_in = (Bytef*)source;
  stream.avail_in = (uInt)sourceLen;
  stream.next_out = dest;
  stream.avail_out = (uInt)*destLen;
  
  stream.zalloc = (alloc_func)0;
  stream.zfree = (free_func)0;
  
  err = inflateInit2(&stream, -MAX_WBITS);
  if (err != Z_OK) return err;
    
  err = inflate(&stream, Z_FINISH);
  if (err != Z_STREAM_END) {
    inflateEnd(&stream);
    if (err == Z_NEED_DICT || (err == Z_BUF_ERROR && stream.avail_in == 0))
      return Z_DATA_ERROR;
    return err;
    }
  *destLen = stream.total_out;
  
  err = inflateEnd(&stream);
  return err;
}



ZipFile::ZipFile(const std::string& fn)
{
  m_temp = NULL;
  m_fd=open(fn.c_str(), O_RDONLY);
  struct stat stat;
  if (fstat(m_fd, &stat)==0 && S_ISREG(stat.st_mode)) {
    m_dataLen = stat.st_size;
    // TODO - win32
    m_data = (unsigned char*)mmap(NULL,m_dataLen,PROT_READ,MAP_PRIVATE, m_fd, 0);
    if ( !m_data ) throw "mmap failed";
    if ( *(int*)&m_data[0] != 0x04034b50 ) throw "bad zip magic";
    m_eoc = (zip_eoc*)&m_data[m_dataLen-sizeof(zip_eoc)];
    m_firstcd = (zip_cd*)&m_data[m_eoc->zipeofst];
    if ( m_eoc && m_firstcd ) {
      m_entries = m_eoc->zipenum;
    } else {
      m_entries = 0;
    }
  } else {
    throw "invalid zip file";
  }
}


ZipFile::~ZipFile()
{
  delete[] m_temp;
  if ( m_data ) munmap( m_data, m_dataLen );
  if ( m_fd ) close( m_fd );
}


std::string ZipFile::entryName( int n )
{
  if ( n < 0 || n >= m_entries ) return std::string();
  zip_cd* cd = m_firstcd;
  for ( int count=0; cd < (zip_cd*)m_eoc && count < n; count++ ) {
    cd = (zip_cd*)(((char*)cd) + sizeof(zip_cd) + cd->zipcfnl + cd->zipcxtl + cd->zipccml);
  }    
  zip_lfh* lfh = (zip_lfh*)&m_data[cd->zipofst];

  if ( lfh ) {
    return std::string(lfh->zipname,lfh->zipfnln);
  }
  return std::string();
}

unsigned char* ZipFile::extract( int n, int *l )
{
  if ( n < 0 || n >= m_entries ) return NULL;
  zip_cd* cd = m_firstcd;
  for ( int count=0; cd < (zip_cd*)m_eoc && count < n; count++ ) {
    cd = (zip_cd*)(((char*)cd) + sizeof(zip_cd) + cd->zipcfnl + cd->zipcxtl + cd->zipccml);
  }    
  zip_lfh* lfh = (zip_lfh*)&m_data[cd->zipofst];

  if ( lfh ) {
    *l = lfh->zipuncmp;
    unsigned char* zdat = (unsigned char*)lfh + sizeof(*lfh) + lfh->zipfnln + lfh->zipxtraln;
    switch (lfh->zipmthd) {
    case 0: 
      return zdat;
    case 8:
      delete[] m_temp;
      m_temp = new unsigned char[*l];
      if ( uncompress_int(m_temp, l, zdat, lfh->zipsize) == Z_OK) {
	return m_temp;
      }
    }
  }
  return NULL;
}