cvs.zerfleddert.de Git - proxmark3-svn/blob

1 //-----------------------------------------------------------------------------

2 // This code is licensed to you under the terms of the GNU GPL, version 2 or,

3 // at your option, any later version. See the LICENSE.txt file for the text of

4 // the license.

5 //-----------------------------------------------------------------------------

6 // Compression tool for FPGA config files. Compress several *.bit files at

7 // compile time. Decompression is done at run time (see fpgaloader.c).

8 // This uses the zlib library tuned to this specific case. The small file sizes

9 // allow to use "insane" parameters for optimum compression ratio.

10 //-----------------------------------------------------------------------------

12 #include <stdio.h>

13 #include <stdlib.h>

14 #include <string.h>

15 #include <stdint.h>

16 #include <stdbool.h>

17 #include "zlib.h"

19 #define MAX(a,b) ((a)>(b)?(a):(b))

21 // zlib configuration

22 #define COMPRESS_LEVEL 9 // use best possible compression

23 #define COMPRESS_WINDOW_BITS 15 // default = max = 15 for a window of 2^15 = 32KBytes

24 #define COMPRESS_MEM_LEVEL 9 // determines the amount of memory allocated during compression. Default = 8.

25 /* COMPRESS_STRATEGY can be

26 Z_DEFAULT_STRATEGY (the default),

27 Z_FILTERED (more huffmann, less string matching),

28 Z_HUFFMAN_ONLY (huffman only, no string matching)

29 Z_RLE (distances limited to one)

30 Z_FIXED (prevents the use of dynamic Huffman codes)

31 */

32 #define COMPRESS_STRATEGY Z_DEFAULT_STRATEGY

33 // zlib tuning parameters:

34 #define COMPRESS_GOOD_LENGTH 258

35 #define COMPRESS_MAX_LAZY 258

36 #define COMPRESS_MAX_NICE_LENGTH 258

37 #define COMPRESS_MAX_CHAIN 8192

39 #define FPGA_INTERLEAVE_SIZE 288 // (the FPGA's internal config frame size is 288 bits. Interleaving with 288 bytes should give best compression)

40 #define FPGA_CONFIG_SIZE 42336 // our current fpga_[lh]f.bit files are 42175 bytes. Rounded up to next multiple of FPGA_INTERLEAVE_SIZE

 static void usage(void)

43 {

         fprintf(stderr, "Usage: fpga_compress <infile1> <infile2> ... <infile_n> <outfile>\n");

         fprintf(stderr, "          Combine n FPGA bitstream files and compress them into one.\n\n");

         fprintf(stderr, "       fpga_compress -d <infile> <outfile>");

         fprintf(stderr, "          Decompress <infile>. Write result to <outfile>");

48 }

 static voidpf fpga_deflate_malloc(voidpf opaque, uInt items, uInt size)

52 {

         return malloc(items*size);

54 }

 static void fpga_deflate_free(voidpf opaque, voidpf address)

58 {

59 return free(address);

60 }

 static bool all_feof(FILE *infile[], uint8_t num_infiles)

64 {

         for (uint16_t i = 0; i < num_infiles; i++) {

                 if (!feof(infile[i])) {

67 return false;

68 }

69 }

71 return true;

72 }

 int zlib_compress(FILE *infile[], uint8_t num_infiles, FILE *outfile)

76 {

77 uint8_t *fpga_config;

78 uint32_t i;

79 int ret;

80 uint8_t c;

81 z_stream compressed_fpga_stream;

         fpga_config = malloc(num_infiles * FPGA_CONFIG_SIZE);

85 // read the input files. Interleave them into fpga_config[]

86 i = 0;

87 do {

                 if (i >= num_infiles * FPGA_CONFIG_SIZE) {

                         fprintf(stderr, "Input files too big (total > %d bytes). These are probably not PM3 FPGA config files.\n", num_infiles*FPGA_CONFIG_SIZE);

                         for(uint16_t j = 0; j < num_infiles; j++) {

92 fclose(infile[j]);

93 }

94 free(fpga_config);

95 return(EXIT_FAILURE);

96 }

                 for(uint16_t j = 0; j < num_infiles; j++) {

                         for(uint16_t k = 0; k < FPGA_INTERLEAVE_SIZE; k++) {

                                 c = fgetc(infile[j]);

                                 if (!feof(infile[j])) {

102 fpga_config[i++] = c;

                                 } else if (num_infiles > 1) {

104 fpga_config[i++] = '\0';

105 }

106 }

107 }

108

         } while (!all_feof(infile, num_infiles));

110

111 // initialize zlib structures

112 compressed_fpga_stream.next_in = fpga_config;

113 compressed_fpga_stream.avail_in = i;

114 compressed_fpga_stream.zalloc = fpga_deflate_malloc;

115 compressed_fpga_stream.zfree = fpga_deflate_free;

116 compressed_fpga_stream.opaque = Z_NULL;

117 ret = deflateInit2(&compressed_fpga_stream,

118 COMPRESS_LEVEL,

119 Z_DEFLATED,

120 COMPRESS_WINDOW_BITS,

121 COMPRESS_MEM_LEVEL,

122 COMPRESS_STRATEGY);

123

124 // estimate the size of the compressed output

         unsigned int outsize_max = deflateBound(&compressed_fpga_stream, compressed_fpga_stream.avail_in);

         uint8_t *outbuf = malloc(outsize_max);

127 compressed_fpga_stream.next_out = outbuf;

128 compressed_fpga_stream.avail_out = outsize_max;

129

130 if (ret == Z_OK) {

131 ret = deflateTune(&compressed_fpga_stream,

132 COMPRESS_GOOD_LENGTH,

133 COMPRESS_MAX_LAZY,

134 COMPRESS_MAX_NICE_LENGTH,

135 COMPRESS_MAX_CHAIN);

136 }

137

138 if (ret == Z_OK) {

                 ret = deflate(&compressed_fpga_stream, Z_FINISH);

140 }

141

         fprintf(stderr, "compressed %d input bytes to %lu output bytes\n", i, compressed_fpga_stream.total_out);

143

144 if (ret != Z_STREAM_END) {

                 fprintf(stderr, "Error in deflate(): %d %s\n", ret, compressed_fpga_stream.msg);

146 free(outbuf);

147 deflateEnd(&compressed_fpga_stream);

                 for(uint16_t j = 0; j < num_infiles; j++) {

149 fclose(infile[j]);

150 }

151 fclose(outfile);

152 free(infile);

153 free(fpga_config);

154 return(EXIT_FAILURE);

155 }

156

         for (i = 0; i < compressed_fpga_stream.total_out; i++) {

                 fputc(outbuf[i], outfile);

159 }

160

161 free(outbuf);

162 deflateEnd(&compressed_fpga_stream);

         for(uint16_t j = 0; j < num_infiles; j++) {

164 fclose(infile[j]);

165 }

166 fclose(outfile);

167 free(infile);

168 free(fpga_config);

169

170 return(EXIT_SUCCESS);

171

172 }

173

174

 int zlib_decompress(FILE *infile, FILE *outfile)

176 {

177 #define DECOMPRESS_BUF_SIZE 1024

178 uint8_t outbuf[DECOMPRESS_BUF_SIZE];

179 uint8_t inbuf[DECOMPRESS_BUF_SIZE];

180 int ret;

181

182 z_stream compressed_fpga_stream;

183

184 // initialize zlib structures

185 compressed_fpga_stream.next_in = inbuf;

186 compressed_fpga_stream.avail_in = 0;

187 compressed_fpga_stream.next_out = outbuf;

188 compressed_fpga_stream.avail_out = DECOMPRESS_BUF_SIZE;

189 compressed_fpga_stream.zalloc = fpga_deflate_malloc;

190 compressed_fpga_stream.zfree = fpga_deflate_free;

191 compressed_fpga_stream.opaque = Z_NULL;

192

         ret = inflateInit2(&compressed_fpga_stream, 0);

194

195 do {

                 if (compressed_fpga_stream.avail_in == 0) {

197 compressed_fpga_stream.next_in = inbuf;

198 uint16_t i = 0;

199 do {

                                 int c = fgetc(infile);

                                 if (!feof(infile)) {

                                         inbuf[i++] = c & 0xFF;

203 compressed_fpga_stream.avail_in++;

204 } else {

205 break;

206 }

                         } while (i < DECOMPRESS_BUF_SIZE);

208 }

209

                 ret = inflate(&compressed_fpga_stream, Z_SYNC_FLUSH);

211

                 if (ret != Z_OK && ret != Z_STREAM_END) {

213 break;

214 }

215

                 if (compressed_fpga_stream.avail_out == 0) {

                         for (uint16_t i = 0; i < DECOMPRESS_BUF_SIZE; i++) {

                                 fputc(outbuf[i], outfile);

219 }

220 compressed_fpga_stream.avail_out = DECOMPRESS_BUF_SIZE;

221 compressed_fpga_stream.next_out = outbuf;

222 }

         } while (ret == Z_OK);

224

         if (ret == Z_STREAM_END) {  // reached end of input

226 uint16_t i = 0;

                 while (compressed_fpga_stream.avail_out < DECOMPRESS_BUF_SIZE) {

                         fputc(outbuf[i++], outfile);

229 compressed_fpga_stream.avail_out++;

230 }

231 fclose(outfile);

232 fclose(infile);

233 return(EXIT_SUCCESS);

234 } else {

                 fprintf(stderr, "Error. Inflate() returned error %d, %s", ret, compressed_fpga_stream.msg);

236 fclose(outfile);

237 fclose(infile);

238 return(EXIT_FAILURE);

239 }

240

241 }

242

243

 int main(int argc, char **argv)

245 {

246 FILE **infiles;

247 FILE *outfile;

248

         if (argc == 1 || argc == 2) {

250 usage();

251 return(EXIT_FAILURE);

252 }

253

         if (!strcmp(argv[1], "-d")) {                   // Decompress

                 infiles = calloc(1, sizeof(FILE*));

                 if (argc != 4) {

257 usage();

258 return(EXIT_FAILURE);

259 }

                 infiles[0] = fopen(argv[2], "rb");

                 if (infiles[0] == NULL) {

                         fprintf(stderr, "Error. Cannot open input file %s", argv[2]);

263 return(EXIT_FAILURE);

264 }

                 outfile = fopen(argv[3], "wb");

266 if (outfile == NULL) {

                         fprintf(stderr, "Error. Cannot open output file %s", argv[3]);

268 return(EXIT_FAILURE);

269 }

                 return zlib_decompress(infiles[0], outfile);

271

272 } else { // Compress

273

                 infiles = calloc(argc-2, sizeof(FILE*));

                 for (uint16_t i = 0; i < argc-2; i++) { 

                         infiles[i] = fopen(argv[i+1], "rb");

                         if (infiles[i] == NULL) {

                                 fprintf(stderr, "Error. Cannot open input file %s", argv[i+1]);

279 return(EXIT_FAILURE);

280 }

281 }

                 outfile = fopen(argv[argc-1], "wb");

283 if (outfile == NULL) {

                         fprintf(stderr, "Error. Cannot open output file %s", argv[argc-1]);

285 return(EXIT_FAILURE);

286 }

                 return zlib_compress(infiles, argc-2, outfile);

288 }

289 }