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
5 //-----------------------------------------------------------------------------
6 // Flasher frontend tool
7 //-----------------------------------------------------------------------------
16 #define MAX(a,b) ((a)>(b)?(a):(b))
19 #define COMPRESS_LEVEL 9 // use best possible compression
20 #define COMPRESS_WINDOW_BITS 15 // default = 15 for a window of 2^15 = 32KBytes
21 #define COMPRESS_MEM_LEVEL 9 // determines the amount of memory allocated during compression. Default = 8. Must be < 9
22 /* COMPRESS_STRATEGY can be
23 Z_DEFAULT_STRATEGY (the default),
24 Z_FILTERED (more huffmann, less string matching),
25 Z_HUFFMAN_ONLY (huffman only, no string matching)
26 Z_RLE (distances limited to one)
27 Z_FIXED (prevents the use of dynamic Huffman codes)
29 #define COMPRESS_STRATEGY Z_DEFAULT_STRATEGY
30 // zlib tuning parameters:
31 #define COMPRESS_GOOD_LENGTH 258
32 #define COMPRESS_MAX_LAZY 258
33 #define COMPRESS_MAX_NICE_LENGTH 258
34 #define COMPRESS_MAX_CHAIN 8192
36 #define FPGA_INTERLEAVE_SIZE 288 // (the FPGA's internal config frame size is 288 bits. Interleaving with 288 bytes should give best compression)
37 #define FPGA_CONFIG_SIZE 42336 // our current fpga_[lh]f.bit files are 42175 bytes. Rounded up to next multiple of FPGA_INTERLEAVE_SIZE
39 static void usage(char *argv0
)
41 fprintf(stderr
, "Usage: %s <infile1> <infile2> ... <infile_n> <outfile>\n\n", argv0
);
42 fprintf(stderr
, "Combines n FPGA bitstream files and compresses them into one.\n\n");
46 static voidpf
fpga_deflate_malloc(voidpf opaque
, uInt items
, uInt size
)
48 fprintf(stderr
, "zlib requested %d bytes\n", items
*size
);
49 return malloc(items
*size
);
53 static void fpga_deflate_free(voidpf opaque
, voidpf address
)
55 fprintf(stderr
, "zlib frees memory\n");
60 static bool all_feof(FILE *infile
[], uint8_t num_infiles
)
62 for (uint16_t i
= 0; i
< num_infiles
; i
++) {
63 if (!feof(infile
[i
])) {
72 int zlib_compress(FILE *infile
[], uint8_t num_infiles
, FILE *outfile
)
78 z_stream compressed_fpga_stream
;
80 fpga_config
= malloc(num_infiles
* FPGA_CONFIG_SIZE
);
82 // read the input files interleaving into fpga_config[]
85 for(uint16_t j
= 0; j
< num_infiles
; j
++) {
86 for(uint16_t k
= 0; k
< FPGA_INTERLEAVE_SIZE
; k
++) {
88 if (!feof(infile
[j
])) {
90 } else if (num_infiles
> 1) {
91 fpga_config
[i
++] = '\0';
96 if (i
> num_infiles
* FPGA_CONFIG_SIZE
) {
97 fprintf(stderr
, "Input files too big (total of %ld > %d bytes). These are probably not PM3 FPGA config files.", i
, num_infiles
*FPGA_CONFIG_SIZE
);
98 for(uint16_t j
= 0; j
< num_infiles
; j
++) {
103 } while (!all_feof(infile
, num_infiles
));
105 fprintf(stderr
, "Read a total of %ld bytes from %d files\n", i
, num_infiles
);
107 // initialize zlib structures
108 compressed_fpga_stream
.next_in
= fpga_config
;
109 compressed_fpga_stream
.avail_in
= i
;
110 compressed_fpga_stream
.zalloc
= fpga_deflate_malloc
;
111 compressed_fpga_stream
.zfree
= fpga_deflate_free
;
113 ret
= deflateInit2(&compressed_fpga_stream
,
116 COMPRESS_WINDOW_BITS
,
120 // estimate the size of the compressed output
121 unsigned int outsize_max
= deflateBound(&compressed_fpga_stream
, compressed_fpga_stream
.avail_in
);
122 fprintf(stderr
, "Allocating %ld bytes for output file (estimated upper bound)\n", outsize_max
);
123 uint8_t *outbuf
= malloc(outsize_max
);
124 compressed_fpga_stream
.next_out
= outbuf
;
125 compressed_fpga_stream
.avail_out
= outsize_max
;
129 ret
= deflateTune(&compressed_fpga_stream
,
130 COMPRESS_GOOD_LENGTH
,
132 COMPRESS_MAX_NICE_LENGTH
,
137 ret
= deflate(&compressed_fpga_stream
, Z_FINISH
);
140 fprintf(stderr
, "produced %d bytes of output\n", compressed_fpga_stream
.total_out
);
142 if (ret
!= Z_STREAM_END
) {
143 fprintf(stderr
, "Error in deflate(): %d %s\n", ret
, compressed_fpga_stream
.msg
);
145 deflateEnd(&compressed_fpga_stream
);
146 for(uint16_t j
= 0; j
< num_infiles
; j
++) {
155 for (i
= 0; i
< compressed_fpga_stream
.total_out
; i
++) {
156 fputc(outbuf
[i
], outfile
);
160 deflateEnd(&compressed_fpga_stream
);
161 for(uint16_t j
= 0; j
< num_infiles
; j
++) {
174 int main(int argc
, char **argv
)
179 if (argc
== 1 || argc
== 2) {
184 infiles
= calloc(argc
-2, sizeof(FILE*));
186 for (uint16_t i
= 0; i
< argc
-2; i
++) {
187 infiles
[i
] = fopen(argv
[i
+1], "rb");
188 if (infiles
[i
] == NULL
) {
189 fprintf(stderr
, "Error. Cannot open input file %s", argv
[i
+1]);
194 outfile
= fopen(argv
[argc
-1], "wb");
195 if (outfile
== NULL
) {
196 fprintf(stderr
, "Error. Cannot open output file %s", argv
[argc
-1]);
200 return zlib_compress(infiles
, argc
-2, outfile
);