1
0
mirror of https://github.com/bol-van/zapret.git synced 2025-05-02 03:42:58 +03:00

tpws : better gzip checking. fixing missing fclose

This commit is contained in:
bolvan 2019-05-10 08:47:23 +03:00
parent f8dd9f3d23
commit 5dd7834314
10 changed files with 26 additions and 26 deletions
binaries
aarch64
armhf
mips32r1-lsb
mips32r1-msb
mips64r2-msb
ppc
x86
x86_64
tpws

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

@ -77,5 +77,7 @@ zerr:
bool is_gzip(FILE* F) bool is_gzip(FILE* F)
{ {
unsigned char magic[2]; unsigned char magic[2];
return !fseek(F,0,SEEK_SET) && fread(magic, 1, 2, F)==2 && magic[0]==0x1F && magic[1]==0x8B; bool b = !fseek(F,0,SEEK_SET) && fread(magic, 1, 2, F)==2 && magic[0]==0x1F && magic[1]==0x8B;
fseek(F,0,SEEK_SET);
return b;
} }

@ -41,11 +41,14 @@ bool LoadHostList(strpool **hostlist, char *filename)
fprintf(stderr, "Could not open %s\n", filename); fprintf(stderr, "Could not open %s\n", filename);
return false; return false;
} }
if ((r=z_readfile(F,&zbuf,&zsize))==Z_OK)
{
printf("zlib compression detected. uncompressed size : %zu\n", zsize); if (is_gzip(F))
{
r = z_readfile(F,&zbuf,&zsize);
fclose(F); fclose(F);
if (r==Z_OK)
{
printf("zlib compression detected. uncompressed size : %zu\n", zsize);
p = zbuf; p = zbuf;
e = zbuf + zsize; e = zbuf + zsize;
@ -61,20 +64,14 @@ bool LoadHostList(strpool **hostlist, char *filename)
} }
free(zbuf); free(zbuf);
} }
else if (r!=Z_DATA_ERROR) else
{ {
fprintf(stderr, "zlib decompression failed : result %d\n",r); fprintf(stderr, "zlib decompression failed : result %d\n",r);
return false; return false;
} }
}
else else
{ {
if (is_gzip(F))
{
fprintf(stderr, "hostlist is gzip but is broken : %s\n",filename);
return false;
}
fseek(F,0,SEEK_SET);
printf("loading plain text list\n",r); printf("loading plain text list\n",r);
while (fgets(s, 256, F)) while (fgets(s, 256, F))
@ -90,6 +87,7 @@ bool LoadHostList(strpool **hostlist, char *filename)
} }
fclose(F); fclose(F);
} }
printf("Loaded %d hosts from %s\n", ct, filename); printf("Loaded %d hosts from %s\n", ct, filename);
return true; return true;
} }