1
0
mirror of https://git.tartarus.org/simon/putty.git synced 2025-01-09 09:27:59 +00:00
putty-source/testzlib.c
Simon Tatham 5d718ef64b Whitespace rationalisation of entire code base.
The number of people has been steadily increasing who read our source
code with an editor that thinks tab stops are 4 spaces apart, as
opposed to the traditional tty-derived 8 that the PuTTY code expects.

So I've been wondering for ages about just fixing it, and switching to
a spaces-only policy throughout the code. And I recently found out
about 'git blame -w', which should make this change not too disruptive
for the purposes of source-control archaeology; so perhaps now is the
time.

While I'm at it, I've also taken the opportunity to remove all the
trailing spaces from source lines (on the basis that git dislikes
them, and is the only thing that seems to have a strong opinion one
way or the other).
    
Apologies to anyone downstream of this code who has complicated patch
sets to rebase past this change. I don't intend it to be needed again.
2019-09-08 20:29:21 +01:00

110 lines
2.9 KiB
C

/*
* Main program to compile sshzlib.c into a zlib decoding tool.
*
* This is potentially a handy tool in its own right for picking apart
* Zip files or PDFs or PNGs, because it accepts the bare Deflate
* format and the zlib wrapper format, unlike 'zcat' which accepts
* only the gzip wrapper format.
*
* It's also useful as a means for a fuzzer to get reasonably direct
* access to PuTTY's zlib decompressor.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "defs.h"
#include "ssh.h"
void out_of_memory(void)
{
fprintf(stderr, "Out of memory!\n");
exit(1);
}
int main(int argc, char **argv)
{
unsigned char buf[16], *outbuf;
int ret, outlen;
ssh_decompressor *handle;
int noheader = false, opts = true;
char *filename = NULL;
FILE *fp;
while (--argc) {
char *p = *++argv;
if (p[0] == '-' && opts) {
if (!strcmp(p, "-d")) {
noheader = true;
} else if (!strcmp(p, "--")) {
opts = false; /* next thing is filename */
} else if (!strcmp(p, "--help")) {
printf("usage: testzlib decode zlib (RFC1950) data"
" from standard input\n");
printf(" testzlib -d decode Deflate (RFC1951) data"
" from standard input\n");
printf(" testzlib --help display this text\n");
return 0;
} else {
fprintf(stderr, "unknown command line option '%s'\n", p);
return 1;
}
} else if (!filename) {
filename = p;
} else {
fprintf(stderr, "can only handle one filename\n");
return 1;
}
}
handle = ssh_decompressor_new(&ssh_zlib);
if (noheader) {
/*
* Provide missing zlib header if -d was specified.
*/
static const unsigned char ersatz_zlib_header[] = { 0x78, 0x9C };
ssh_decompressor_decompress(
handle, ersatz_zlib_header, sizeof(ersatz_zlib_header),
&outbuf, &outlen);
assert(outlen == 0);
}
if (filename)
fp = fopen(filename, "rb");
else
fp = stdin;
if (!fp) {
assert(filename);
fprintf(stderr, "unable to open '%s'\n", filename);
return 1;
}
while (1) {
ret = fread(buf, 1, sizeof(buf), fp);
if (ret <= 0)
break;
ssh_decompressor_decompress(handle, buf, ret, &outbuf, &outlen);
if (outbuf) {
if (outlen)
fwrite(outbuf, 1, outlen, stdout);
sfree(outbuf);
} else {
fprintf(stderr, "decoding error\n");
fclose(fp);
return 1;
}
}
ssh_decompressor_free(handle);
if (filename)
fclose(fp);
return 0;
}