00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054 #include <stdio.h>
00055 #include <stdlib.h>
00056 #include <assert.h>
00057 #include "zlib.h"
00058
00059 #define local static
00060
00061
00062 local void quit(char *why)
00063 {
00064 fprintf(stderr, "fitblk abort: %s\n", why);
00065 exit(1);
00066 }
00067
00068 #define RAWLEN 4096
00069
00070
00071
00072
00073 local int partcompress(FILE *in, z_streamp def)
00074 {
00075 int ret, flush;
00076 unsigned char raw[RAWLEN];
00077
00078 flush = Z_NO_FLUSH;
00079 do {
00080 def->avail_in = fread(raw, 1, RAWLEN, in);
00081 if (ferror(in))
00082 return Z_ERRNO;
00083 def->next_in = raw;
00084 if (feof(in))
00085 flush = Z_FINISH;
00086 ret = deflate(def, flush);
00087 assert(ret != Z_STREAM_ERROR);
00088 } while (def->avail_out != 0 && flush == Z_NO_FLUSH);
00089 return ret;
00090 }
00091
00092
00093
00094
00095
00096 local int recompress(z_streamp inf, z_streamp def)
00097 {
00098 int ret, flush;
00099 unsigned char raw[RAWLEN];
00100
00101 flush = Z_NO_FLUSH;
00102 do {
00103
00104 inf->avail_out = RAWLEN;
00105 inf->next_out = raw;
00106 ret = inflate(inf, Z_NO_FLUSH);
00107 assert(ret != Z_STREAM_ERROR && ret != Z_DATA_ERROR &&
00108 ret != Z_NEED_DICT);
00109 if (ret == Z_MEM_ERROR)
00110 return ret;
00111
00112
00113 def->avail_in = RAWLEN - inf->avail_out;
00114 def->next_in = raw;
00115 if (inf->avail_out != 0)
00116 flush = Z_FINISH;
00117 ret = deflate(def, flush);
00118 assert(ret != Z_STREAM_ERROR);
00119 } while (ret != Z_STREAM_END && def->avail_out != 0);
00120 return ret;
00121 }
00122
00123 #define EXCESS 256
00124 #define MARGIN 8
00125
00126
00127 int main(int argc, char **argv)
00128 {
00129 int ret;
00130 unsigned size;
00131 unsigned have;
00132 unsigned char *blk;
00133 unsigned char *tmp;
00134 z_stream def, inf;
00135
00136
00137 if (argc != 2)
00138 quit("need one argument: size of output block");
00139 ret = strtol(argv[1], argv + 1, 10);
00140 if (argv[1][0] != 0)
00141 quit("argument must be a number");
00142 if (ret < 8)
00143 quit("need positive size of 8 or greater");
00144 size = (unsigned)ret;
00145
00146
00147 blk = malloc(size + EXCESS);
00148 def.zalloc = Z_NULL;
00149 def.zfree = Z_NULL;
00150 def.opaque = Z_NULL;
00151 ret = deflateInit(&def, Z_DEFAULT_COMPRESSION);
00152 if (ret != Z_OK || blk == NULL)
00153 quit("out of memory");
00154
00155
00156 def.avail_out = size + EXCESS;
00157 def.next_out = blk;
00158 ret = partcompress(stdin, &def);
00159 if (ret == Z_ERRNO)
00160 quit("error reading input");
00161
00162
00163 if (ret == Z_STREAM_END && def.avail_out >= EXCESS) {
00164
00165 have = size + EXCESS - def.avail_out;
00166 if (fwrite(blk, 1, have, stdout) != have || ferror(stdout))
00167 quit("error writing output");
00168
00169
00170 ret = deflateEnd(&def);
00171 assert(ret != Z_STREAM_ERROR);
00172 free(blk);
00173 fprintf(stderr,
00174 "%u bytes unused out of %u requested (all input)\n",
00175 size - have, size);
00176 return 0;
00177 }
00178
00179
00180 inf.zalloc = Z_NULL;
00181 inf.zfree = Z_NULL;
00182 inf.opaque = Z_NULL;
00183 inf.avail_in = 0;
00184 inf.next_in = Z_NULL;
00185 ret = inflateInit(&inf);
00186 tmp = malloc(size + EXCESS);
00187 if (ret != Z_OK || tmp == NULL)
00188 quit("out of memory");
00189 ret = deflateReset(&def);
00190 assert(ret != Z_STREAM_ERROR);
00191
00192
00193 inf.avail_in = size + EXCESS;
00194 inf.next_in = blk;
00195 def.avail_out = size + EXCESS;
00196 def.next_out = tmp;
00197 ret = recompress(&inf, &def);
00198 if (ret == Z_MEM_ERROR)
00199 quit("out of memory");
00200
00201
00202 ret = inflateReset(&inf);
00203 assert(ret != Z_STREAM_ERROR);
00204 ret = deflateReset(&def);
00205 assert(ret != Z_STREAM_ERROR);
00206
00207
00208 inf.avail_in = size - MARGIN;
00209 inf.next_in = tmp;
00210 def.avail_out = size;
00211 def.next_out = blk;
00212 ret = recompress(&inf, &def);
00213 if (ret == Z_MEM_ERROR)
00214 quit("out of memory");
00215 assert(ret == Z_STREAM_END);
00216
00217
00218 have = size - def.avail_out;
00219 if (fwrite(blk, 1, have, stdout) != have || ferror(stdout))
00220 quit("error writing output");
00221
00222
00223 free(tmp);
00224 ret = inflateEnd(&inf);
00225 assert(ret != Z_STREAM_ERROR);
00226 ret = deflateEnd(&def);
00227 assert(ret != Z_STREAM_ERROR);
00228 free(blk);
00229 fprintf(stderr,
00230 "%u bytes unused out of %u requested (%lu input)\n",
00231 size - have, size, def.total_in);
00232 return 0;
00233 }