[tor-commits] [stegotorus/master] crc32.{cc, h} and zpack.{cc, h} are not really steg code, pull them up one.

zwol at torproject.org zwol at torproject.org
Fri Jul 20 23:17:07 UTC 2012


commit dfc8211422231ad3065de3d32c9002277615d423
Author: Zack Weinberg <zackw at cmu.edu>
Date:   Mon Apr 23 15:44:35 2012 -0700

    crc32.{cc,h} and zpack.{cc,h} are not really steg code, pull them up one.
---
 Makefile.am       |    6 +-
 src/crc32.cc      |   83 ++++++++++++++
 src/crc32.h       |    6 +
 src/steg/crc32.cc |   83 --------------
 src/steg/crc32.h  |    6 -
 src/steg/zpack.cc |  311 -----------------------------------------------------
 src/steg/zpack.h  |   12 --
 src/zpack.cc      |  311 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/zpack.h       |   12 ++
 9 files changed, 415 insertions(+), 415 deletions(-)

diff --git a/Makefile.am b/Makefile.am
index 86ed489..7d72a96 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -19,7 +19,6 @@ PROTOCOLS = \
 STEGANOGRAPHERS = \
 	src/steg/b64cookies.cc \
 	src/steg/cookies.cc \
-	src/steg/crc32.cc \
 	src/steg/embed.cc \
 	src/steg/http.cc \
 	src/steg/jsSteg.cc \
@@ -27,11 +26,11 @@ STEGANOGRAPHERS = \
 	src/steg/nosteg_rr.cc \
 	src/steg/payloads.cc \
 	src/steg/pdfSteg.cc \
-	src/steg/swfSteg.cc \
-	src/steg/zpack.cc
+	src/steg/swfSteg.cc
 
 libstegotorus_a_SOURCES = \
 	src/base64.cc \
+	src/crc32.cc \
 	src/connections.cc \
 	src/crypt.cc \
 	src/network.cc \
@@ -40,6 +39,7 @@ libstegotorus_a_SOURCES = \
 	src/socks.cc \
 	src/steg.cc \
 	src/util.cc \
+	src/zpack.cc \
 	$(PROTOCOLS) $(STEGANOGRAPHERS)
 
 nodist_libstegotorus_a_SOURCES = protolist.cc steglist.cc
diff --git a/src/crc32.cc b/src/crc32.cc
new file mode 100644
index 0000000..1b18544
--- /dev/null
+++ b/src/crc32.cc
@@ -0,0 +1,83 @@
+#include "util.h"
+#include "crc32.h"
+
+#define CRC32C(c,d) (c=(c>>8)^crc_c[(c^(d))&0xFF])
+
+static const unsigned int crc_c[256] = {
+	0x00000000, 0x77073096, 0xee0e612c, 0x990951ba,
+	0x076dc419, 0x706af48f, 0xe963a535, 0x9e6495a3,
+	0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
+	0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91,
+	0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
+	0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
+	0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec,
+	0x14015c4f, 0x63066cd9, 0xfa0f3d63, 0x8d080df5,
+	0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
+	0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
+	0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940,
+	0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
+	0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116,
+	0x21b4f4b5, 0x56b3c423, 0xcfba9599, 0xb8bda50f,
+	0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
+	0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d,
+	0x76dc4190, 0x01db7106, 0x98d220bc, 0xefd5102a,
+	0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
+	0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818,
+	0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
+	0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
+	0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457,
+	0x65b0d9c6, 0x12b7e950, 0x8bbeb8ea, 0xfcb9887c,
+	0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
+	0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
+	0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb,
+	0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
+	0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9,
+	0x5005713c, 0x270241aa, 0xbe0b1010, 0xc90c2086,
+	0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
+	0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4,
+	0x59b33d17, 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad,
+	0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
+	0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683,
+	0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
+	0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
+	0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe,
+	0xf762575d, 0x806567cb, 0x196c3671, 0x6e6b06e7,
+	0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
+	0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
+	0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252,
+	0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
+	0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60,
+	0xdf60efc3, 0xa867df55, 0x316e8eef, 0x4669be79,
+	0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
+	0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f,
+	0xc5ba3bbe, 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04,
+	0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
+	0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a,
+	0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
+	0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
+	0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21,
+	0x86d3d2d4, 0xf1d4e242, 0x68ddb3f8, 0x1fda836e,
+	0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
+	0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
+	0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45,
+	0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
+	0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db,
+	0xaed16a4a, 0xd9d65adc, 0x40df0b66, 0x37d83bf0,
+	0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
+	0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6,
+	0xbad03605, 0xcdd70693, 0x54de5729, 0x23d967bf,
+	0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
+	0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d,
+};
+
+unsigned int
+generate_crc32c(const char *buffer, size_t length)
+{
+  size_t i;
+  unsigned int crc32 = ~0L;
+
+  for (i = 0; i < length; i++){
+      CRC32C(crc32, (unsigned char)buffer[i]);
+  }
+  return ~crc32;
+}
diff --git a/src/crc32.h b/src/crc32.h
new file mode 100644
index 0000000..0a1ee03
--- /dev/null
+++ b/src/crc32.h
@@ -0,0 +1,6 @@
+#ifndef _CRC32_H
+#define _CRC32_H
+
+unsigned int generate_crc32c(const char *string, size_t length);
+
+#endif
diff --git a/src/steg/crc32.cc b/src/steg/crc32.cc
deleted file mode 100644
index 1b18544..0000000
--- a/src/steg/crc32.cc
+++ /dev/null
@@ -1,83 +0,0 @@
-#include "util.h"
-#include "crc32.h"
-
-#define CRC32C(c,d) (c=(c>>8)^crc_c[(c^(d))&0xFF])
-
-static const unsigned int crc_c[256] = {
-	0x00000000, 0x77073096, 0xee0e612c, 0x990951ba,
-	0x076dc419, 0x706af48f, 0xe963a535, 0x9e6495a3,
-	0x0edb8832, 0x79dcb8a4, 0xe0d5e91e, 0x97d2d988,
-	0x09b64c2b, 0x7eb17cbd, 0xe7b82d07, 0x90bf1d91,
-	0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
-	0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7,
-	0x136c9856, 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec,
-	0x14015c4f, 0x63066cd9, 0xfa0f3d63, 0x8d080df5,
-	0x3b6e20c8, 0x4c69105e, 0xd56041e4, 0xa2677172,
-	0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
-	0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940,
-	0x32d86ce3, 0x45df5c75, 0xdcd60dcf, 0xabd13d59,
-	0x26d930ac, 0x51de003a, 0xc8d75180, 0xbfd06116,
-	0x21b4f4b5, 0x56b3c423, 0xcfba9599, 0xb8bda50f,
-	0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
-	0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d,
-	0x76dc4190, 0x01db7106, 0x98d220bc, 0xefd5102a,
-	0x71b18589, 0x06b6b51f, 0x9fbfe4a5, 0xe8b8d433,
-	0x7807c9a2, 0x0f00f934, 0x9609a88e, 0xe10e9818,
-	0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
-	0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e,
-	0x6c0695ed, 0x1b01a57b, 0x8208f4c1, 0xf50fc457,
-	0x65b0d9c6, 0x12b7e950, 0x8bbeb8ea, 0xfcb9887c,
-	0x62dd1ddf, 0x15da2d49, 0x8cd37cf3, 0xfbd44c65,
-	0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
-	0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb,
-	0x4369e96a, 0x346ed9fc, 0xad678846, 0xda60b8d0,
-	0x44042d73, 0x33031de5, 0xaa0a4c5f, 0xdd0d7cc9,
-	0x5005713c, 0x270241aa, 0xbe0b1010, 0xc90c2086,
-	0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
-	0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4,
-	0x59b33d17, 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad,
-	0xedb88320, 0x9abfb3b6, 0x03b6e20c, 0x74b1d29a,
-	0xead54739, 0x9dd277af, 0x04db2615, 0x73dc1683,
-	0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
-	0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1,
-	0xf00f9344, 0x8708a3d2, 0x1e01f268, 0x6906c2fe,
-	0xf762575d, 0x806567cb, 0x196c3671, 0x6e6b06e7,
-	0xfed41b76, 0x89d32be0, 0x10da7a5a, 0x67dd4acc,
-	0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
-	0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252,
-	0xd1bb67f1, 0xa6bc5767, 0x3fb506dd, 0x48b2364b,
-	0xd80d2bda, 0xaf0a1b4c, 0x36034af6, 0x41047a60,
-	0xdf60efc3, 0xa867df55, 0x316e8eef, 0x4669be79,
-	0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
-	0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f,
-	0xc5ba3bbe, 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04,
-	0xc2d7ffa7, 0xb5d0cf31, 0x2cd99e8b, 0x5bdeae1d,
-	0x9b64c2b0, 0xec63f226, 0x756aa39c, 0x026d930a,
-	0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
-	0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38,
-	0x92d28e9b, 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21,
-	0x86d3d2d4, 0xf1d4e242, 0x68ddb3f8, 0x1fda836e,
-	0x81be16cd, 0xf6b9265b, 0x6fb077e1, 0x18b74777,
-	0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
-	0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45,
-	0xa00ae278, 0xd70dd2ee, 0x4e048354, 0x3903b3c2,
-	0xa7672661, 0xd06016f7, 0x4969474d, 0x3e6e77db,
-	0xaed16a4a, 0xd9d65adc, 0x40df0b66, 0x37d83bf0,
-	0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
-	0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6,
-	0xbad03605, 0xcdd70693, 0x54de5729, 0x23d967bf,
-	0xb3667a2e, 0xc4614ab8, 0x5d681b02, 0x2a6f2b94,
-	0xb40bbe37, 0xc30c8ea1, 0x5a05df1b, 0x2d02ef8d,
-};
-
-unsigned int
-generate_crc32c(const char *buffer, size_t length)
-{
-  size_t i;
-  unsigned int crc32 = ~0L;
-
-  for (i = 0; i < length; i++){
-      CRC32C(crc32, (unsigned char)buffer[i]);
-  }
-  return ~crc32;
-}
diff --git a/src/steg/crc32.h b/src/steg/crc32.h
deleted file mode 100644
index 0a1ee03..0000000
--- a/src/steg/crc32.h
+++ /dev/null
@@ -1,6 +0,0 @@
-#ifndef _CRC32_H
-#define _CRC32_H
-
-unsigned int generate_crc32c(const char *string, size_t length);
-
-#endif
diff --git a/src/steg/zpack.cc b/src/steg/zpack.cc
deleted file mode 100644
index c3c650f..0000000
--- a/src/steg/zpack.cc
+++ /dev/null
@@ -1,311 +0,0 @@
-#include "util.h"
-#include "zpack.h"
-#include "zlib.h"
-#include "crc32.h"
-
-#define CHUNK 16384
-
-/* Compress from file source to file dest until EOF on source.
-   def() returns Z_OK on success, Z_MEM_ERROR if memory could not be
-   allocated for processing, Z_STREAM_ERROR if an invalid compression
-   level is supplied, Z_VERSION_ERROR if the version of zlib.h and the
-   version of the library linked do not match, or Z_ERRNO if there is
-   an error reading or writing the files. */
-
-ssize_t
-def(const char *source, size_t slen, char *dest, size_t dlen, int level)
-{
-  int ret, flush;
-  size_t have;
-  z_stream strm;
-  unsigned char in[CHUNK];
-  unsigned char out[CHUNK];
-  size_t dlen_orig = dlen;
-
-  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
-    return -1;
-
-  /* allocate deflate state */
-  strm.zalloc = Z_NULL;
-  strm.zfree = Z_NULL;
-  strm.opaque = Z_NULL;
-  ret = deflateInit(&strm, level);
-  if (ret != Z_OK)
-    return ret;
-
-  /* compress until end of file */
-  do {
-    if (slen > CHUNK)
-      strm.avail_in = CHUNK;
-    else
-      strm.avail_in = slen;
-
-    memcpy (in, source, strm.avail_in);
-    slen = slen - strm.avail_in;
-    source = source + strm.avail_in;
-
-    flush = (slen == 0) ? Z_FINISH : Z_NO_FLUSH;
-    strm.next_in = in;
-
-    /* run deflate() on input until output buffer not full, finish
-       compression if all of source has been read in */
-    do {
-      strm.avail_out = CHUNK;
-      strm.next_out = out;
-      ret = deflate(&strm, flush);    /* no bad return value */
-      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
-      have = CHUNK - strm.avail_out;
-
-      if (dlen < have) {
-        log_warn("dest buf too small - have %lu, need %lu",
-                 (unsigned long)dlen, (unsigned long)have);
-        return Z_ERRNO;
-      }
-
-      memcpy(dest, out, have);
-      dest += have;
-      dlen = dlen - have;
-    } while (strm.avail_out == 0);
-    log_assert(strm.avail_in == 0);     /* all input will be used */
-
-    /* done when last data in file processed */
-  } while (flush != Z_FINISH);
-  log_assert(ret == Z_STREAM_END);        /* stream will be complete */
-
-  /* clean up and return */
-  deflateEnd(&strm);
-  return (dlen_orig - dlen);
-}
-
-/* Decompress from file source to file dest until stream ends or EOF.
-   inf() returns Z_OK on success, Z_MEM_ERROR if memory could not be
-   allocated for processing, Z_DATA_ERROR if the deflate data is
-   invalid or incomplete, Z_VERSION_ERROR if the version of zlib.h and
-   the version of the library linked do not match, or Z_ERRNO if there
-   is an error reading or writing the files. */
-
-ssize_t
-inf(const char *source, size_t slen, char *dest, size_t dlen)
-{
-  int ret;
-  size_t have;
-  z_stream strm;
-  unsigned char in[CHUNK];
-  unsigned char out[CHUNK];
-  size_t dlen_orig = dlen;
-
-  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
-    return -1;
-
-  /* allocate inflate state */
-  strm.zalloc = Z_NULL;
-  strm.zfree = Z_NULL;
-  strm.opaque = Z_NULL;
-  strm.avail_in = 0;
-  strm.next_in = Z_NULL;
-  ret = inflateInit(&strm);
-  if (ret != Z_OK)
-    return ret;
-
-  /* decompress until deflate stream ends or end of file */
-  do {
-    if (slen == 0)
-      break;
-
-    if (slen > CHUNK)
-      strm.avail_in = CHUNK;
-    else
-      strm.avail_in = slen;
-
-    memcpy(in, source, strm.avail_in);
-    slen = slen - strm.avail_in;
-    source = source + strm.avail_in;
-    strm.next_in = in;
-
-    /* run inflate() on input until output buffer not full */
-    do {
-      strm.avail_out = CHUNK;
-      strm.next_out = out;
-      ret = inflate(&strm, Z_NO_FLUSH);
-      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
-      switch (ret) {
-      case Z_NEED_DICT:
-      case Z_DATA_ERROR:
-      case Z_MEM_ERROR:
-        inflateEnd(&strm);
-        return ret;
-      }
-      have = CHUNK - strm.avail_out;
-
-      if (dlen < have) {
-        log_warn("dest buf too small - have %lu, need %lu",
-                 (unsigned long)dlen, (unsigned long)have);
-        return Z_ERRNO;
-      }
-
-      memcpy(dest, out, have);
-      dest += have;
-      dlen = dlen - have;
-
-    } while (strm.avail_out == 0);
-
-    /* done when inflate() says it's done */
-  } while (ret != Z_STREAM_END);
-
-  /* clean up and return */
-  inflateEnd(&strm);
-
-  if (ret == Z_STREAM_END)
-    return dlen_orig - dlen;
-  return Z_DATA_ERROR;
-}
-
-/* assumes that we know there is exactly 10 bytes of gzip header */
-
-ssize_t
-gzInflate(const char *source, size_t slen, char *dest, size_t dlen)
-{
-  int ret;
-  size_t have;
-  z_stream strm;
-  unsigned char in[CHUNK];
-  unsigned char out[CHUNK];
-  size_t dlen_orig = dlen;
-
-  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
-    return -1;
-
-  /* allocate inflate state */
-  strm.zalloc = Z_NULL;
-  strm.zfree = Z_NULL;
-  strm.opaque = Z_NULL;
-  strm.avail_in = 0;
-  strm.next_in = Z_NULL;
-
-  ret = inflateInit2(&strm, -MAX_WBITS);
-  if (ret != Z_OK)
-    return ret;
-
-  source = source + 10;
-  slen -= 10;
-
-  /* decompress until deflate stream ends or end of file */
-  do {
-    if (slen == 0)
-      break;
-
-    if (slen > CHUNK)
-      strm.avail_in = CHUNK;
-    else
-      strm.avail_in = slen;
-
-    memcpy(in, source, strm.avail_in);
-    slen = slen - strm.avail_in;
-    source = source + strm.avail_in;
-    strm.next_in = in;
-
-    /* run inflate() on input until output buffer not full */
-    do {
-      strm.avail_out = CHUNK;
-      strm.next_out = out;
-      ret = inflate(&strm, Z_NO_FLUSH);
-      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
-      switch (ret) {
-      case Z_NEED_DICT:
-        ret = Z_DATA_ERROR;     /* and fall through */
-      case Z_DATA_ERROR:
-      case Z_MEM_ERROR:
-        inflateEnd(&strm);
-        return ret;
-      }
-      have = CHUNK - strm.avail_out;
-
-      if (dlen < have) {
-        log_warn("dest buf too small - have %lu, need %lu",
-                 (unsigned long)dlen, (unsigned long)have);
-        return Z_ERRNO;
-      }
-
-      memcpy(dest, out, have);
-      dest += have;
-      dlen = dlen - have;
-
-    } while (strm.avail_out == 0);
-
-    /* done when inflate() says it's done */
-  } while (ret != Z_STREAM_END);
-
-  /* clean up and return */
-  inflateEnd(&strm);
-
-  if (ret == Z_STREAM_END)
-    return dlen_orig - dlen;
-  return Z_DATA_ERROR;
-}
-
-ssize_t
-gzDeflate(const char *source, size_t slen, char *dest, size_t dlen,
-          time_t mtime)
-{
-  unsigned char *c;
-  unsigned long crc;
-  z_stream z;
-
-  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
-    return -1;
-
-  z.zalloc = Z_NULL;
-  z.zfree = Z_NULL;
-  z.opaque = Z_NULL;
-
-  if (Z_OK != deflateInit2(&z,
-                           Z_DEFAULT_COMPRESSION,
-                           Z_DEFLATED,
-                           -MAX_WBITS,  /* supress zlib-header */
-                           8,
-                           Z_DEFAULT_STRATEGY))
-    return -1;
-
-  z.next_in = (Bytef *)source;
-  z.avail_in = slen;
-  z.total_in = 0;
-
-  /* write gzip header */
-
-  c = (unsigned char *)dest;
-  c[0] = 0x1f;
-  c[1] = 0x8b;
-  c[2] = Z_DEFLATED;
-  c[3] = 0; /* options */
-  c[4] = (mtime >>  0) & 0xff;
-  c[5] = (mtime >>  8) & 0xff;
-  c[6] = (mtime >> 16) & 0xff;
-  c[7] = (mtime >> 24) & 0xff;
-  c[8] = 0x00; /* extra flags */
-  c[9] = 0x03; /* UNIX */
-
-  z.next_out = c + 10;
-  z.avail_out = dlen - 10 - 8;
-  z.total_out = 0;
-
-  if (deflate(&z, Z_FINISH) != Z_STREAM_END) {
-    deflateEnd(&z);
-    return -1;
-  }
-
-  crc = generate_crc32c(source, slen);
-
-  c = (unsigned char *)dest + 10 + z.total_out;
-  c[0] = (crc >>  0) & 0xff;
-  c[1] = (crc >>  8) & 0xff;
-  c[2] = (crc >> 16) & 0xff;
-  c[3] = (crc >> 24) & 0xff;
-  c[4] = (z.total_in >>  0) & 0xff;
-  c[5] = (z.total_in >>  8) & 0xff;
-  c[6] = (z.total_in >> 16) & 0xff;
-  c[7] = (z.total_in >> 24) & 0xff;
-
-  if (deflateEnd(&z) != Z_OK)
-    return -1;
-  return 10 + z.total_out + 8;
-}
diff --git a/src/steg/zpack.h b/src/steg/zpack.h
deleted file mode 100644
index 5d3f82f..0000000
--- a/src/steg/zpack.h
+++ /dev/null
@@ -1,12 +0,0 @@
-#ifndef _ZPACK_H
-#define _ZPACK_H
-
-ssize_t def(const char *source, size_t slen, char *dest, size_t dlen,
-            int level);
-ssize_t inf(const char *source, size_t slen, char *dest, size_t dlen);
-
-ssize_t gzInflate(const char *source, size_t slen, char *dest, size_t dlen);
-ssize_t gzDeflate(const char *source, size_t slen, char *dest, size_t dlen,
-                  time_t mtime);
-
-#endif
diff --git a/src/zpack.cc b/src/zpack.cc
new file mode 100644
index 0000000..c3c650f
--- /dev/null
+++ b/src/zpack.cc
@@ -0,0 +1,311 @@
+#include "util.h"
+#include "zpack.h"
+#include "zlib.h"
+#include "crc32.h"
+
+#define CHUNK 16384
+
+/* Compress from file source to file dest until EOF on source.
+   def() returns Z_OK on success, Z_MEM_ERROR if memory could not be
+   allocated for processing, Z_STREAM_ERROR if an invalid compression
+   level is supplied, Z_VERSION_ERROR if the version of zlib.h and the
+   version of the library linked do not match, or Z_ERRNO if there is
+   an error reading or writing the files. */
+
+ssize_t
+def(const char *source, size_t slen, char *dest, size_t dlen, int level)
+{
+  int ret, flush;
+  size_t have;
+  z_stream strm;
+  unsigned char in[CHUNK];
+  unsigned char out[CHUNK];
+  size_t dlen_orig = dlen;
+
+  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
+    return -1;
+
+  /* allocate deflate state */
+  strm.zalloc = Z_NULL;
+  strm.zfree = Z_NULL;
+  strm.opaque = Z_NULL;
+  ret = deflateInit(&strm, level);
+  if (ret != Z_OK)
+    return ret;
+
+  /* compress until end of file */
+  do {
+    if (slen > CHUNK)
+      strm.avail_in = CHUNK;
+    else
+      strm.avail_in = slen;
+
+    memcpy (in, source, strm.avail_in);
+    slen = slen - strm.avail_in;
+    source = source + strm.avail_in;
+
+    flush = (slen == 0) ? Z_FINISH : Z_NO_FLUSH;
+    strm.next_in = in;
+
+    /* run deflate() on input until output buffer not full, finish
+       compression if all of source has been read in */
+    do {
+      strm.avail_out = CHUNK;
+      strm.next_out = out;
+      ret = deflate(&strm, flush);    /* no bad return value */
+      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
+      have = CHUNK - strm.avail_out;
+
+      if (dlen < have) {
+        log_warn("dest buf too small - have %lu, need %lu",
+                 (unsigned long)dlen, (unsigned long)have);
+        return Z_ERRNO;
+      }
+
+      memcpy(dest, out, have);
+      dest += have;
+      dlen = dlen - have;
+    } while (strm.avail_out == 0);
+    log_assert(strm.avail_in == 0);     /* all input will be used */
+
+    /* done when last data in file processed */
+  } while (flush != Z_FINISH);
+  log_assert(ret == Z_STREAM_END);        /* stream will be complete */
+
+  /* clean up and return */
+  deflateEnd(&strm);
+  return (dlen_orig - dlen);
+}
+
+/* Decompress from file source to file dest until stream ends or EOF.
+   inf() returns Z_OK on success, Z_MEM_ERROR if memory could not be
+   allocated for processing, Z_DATA_ERROR if the deflate data is
+   invalid or incomplete, Z_VERSION_ERROR if the version of zlib.h and
+   the version of the library linked do not match, or Z_ERRNO if there
+   is an error reading or writing the files. */
+
+ssize_t
+inf(const char *source, size_t slen, char *dest, size_t dlen)
+{
+  int ret;
+  size_t have;
+  z_stream strm;
+  unsigned char in[CHUNK];
+  unsigned char out[CHUNK];
+  size_t dlen_orig = dlen;
+
+  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
+    return -1;
+
+  /* allocate inflate state */
+  strm.zalloc = Z_NULL;
+  strm.zfree = Z_NULL;
+  strm.opaque = Z_NULL;
+  strm.avail_in = 0;
+  strm.next_in = Z_NULL;
+  ret = inflateInit(&strm);
+  if (ret != Z_OK)
+    return ret;
+
+  /* decompress until deflate stream ends or end of file */
+  do {
+    if (slen == 0)
+      break;
+
+    if (slen > CHUNK)
+      strm.avail_in = CHUNK;
+    else
+      strm.avail_in = slen;
+
+    memcpy(in, source, strm.avail_in);
+    slen = slen - strm.avail_in;
+    source = source + strm.avail_in;
+    strm.next_in = in;
+
+    /* run inflate() on input until output buffer not full */
+    do {
+      strm.avail_out = CHUNK;
+      strm.next_out = out;
+      ret = inflate(&strm, Z_NO_FLUSH);
+      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
+      switch (ret) {
+      case Z_NEED_DICT:
+      case Z_DATA_ERROR:
+      case Z_MEM_ERROR:
+        inflateEnd(&strm);
+        return ret;
+      }
+      have = CHUNK - strm.avail_out;
+
+      if (dlen < have) {
+        log_warn("dest buf too small - have %lu, need %lu",
+                 (unsigned long)dlen, (unsigned long)have);
+        return Z_ERRNO;
+      }
+
+      memcpy(dest, out, have);
+      dest += have;
+      dlen = dlen - have;
+
+    } while (strm.avail_out == 0);
+
+    /* done when inflate() says it's done */
+  } while (ret != Z_STREAM_END);
+
+  /* clean up and return */
+  inflateEnd(&strm);
+
+  if (ret == Z_STREAM_END)
+    return dlen_orig - dlen;
+  return Z_DATA_ERROR;
+}
+
+/* assumes that we know there is exactly 10 bytes of gzip header */
+
+ssize_t
+gzInflate(const char *source, size_t slen, char *dest, size_t dlen)
+{
+  int ret;
+  size_t have;
+  z_stream strm;
+  unsigned char in[CHUNK];
+  unsigned char out[CHUNK];
+  size_t dlen_orig = dlen;
+
+  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
+    return -1;
+
+  /* allocate inflate state */
+  strm.zalloc = Z_NULL;
+  strm.zfree = Z_NULL;
+  strm.opaque = Z_NULL;
+  strm.avail_in = 0;
+  strm.next_in = Z_NULL;
+
+  ret = inflateInit2(&strm, -MAX_WBITS);
+  if (ret != Z_OK)
+    return ret;
+
+  source = source + 10;
+  slen -= 10;
+
+  /* decompress until deflate stream ends or end of file */
+  do {
+    if (slen == 0)
+      break;
+
+    if (slen > CHUNK)
+      strm.avail_in = CHUNK;
+    else
+      strm.avail_in = slen;
+
+    memcpy(in, source, strm.avail_in);
+    slen = slen - strm.avail_in;
+    source = source + strm.avail_in;
+    strm.next_in = in;
+
+    /* run inflate() on input until output buffer not full */
+    do {
+      strm.avail_out = CHUNK;
+      strm.next_out = out;
+      ret = inflate(&strm, Z_NO_FLUSH);
+      log_assert(ret != Z_STREAM_ERROR);  /* state not clobbered */
+      switch (ret) {
+      case Z_NEED_DICT:
+        ret = Z_DATA_ERROR;     /* and fall through */
+      case Z_DATA_ERROR:
+      case Z_MEM_ERROR:
+        inflateEnd(&strm);
+        return ret;
+      }
+      have = CHUNK - strm.avail_out;
+
+      if (dlen < have) {
+        log_warn("dest buf too small - have %lu, need %lu",
+                 (unsigned long)dlen, (unsigned long)have);
+        return Z_ERRNO;
+      }
+
+      memcpy(dest, out, have);
+      dest += have;
+      dlen = dlen - have;
+
+    } while (strm.avail_out == 0);
+
+    /* done when inflate() says it's done */
+  } while (ret != Z_STREAM_END);
+
+  /* clean up and return */
+  inflateEnd(&strm);
+
+  if (ret == Z_STREAM_END)
+    return dlen_orig - dlen;
+  return Z_DATA_ERROR;
+}
+
+ssize_t
+gzDeflate(const char *source, size_t slen, char *dest, size_t dlen,
+          time_t mtime)
+{
+  unsigned char *c;
+  unsigned long crc;
+  z_stream z;
+
+  if (slen > SIZE_T_CEILING || dlen > SIZE_T_CEILING)
+    return -1;
+
+  z.zalloc = Z_NULL;
+  z.zfree = Z_NULL;
+  z.opaque = Z_NULL;
+
+  if (Z_OK != deflateInit2(&z,
+                           Z_DEFAULT_COMPRESSION,
+                           Z_DEFLATED,
+                           -MAX_WBITS,  /* supress zlib-header */
+                           8,
+                           Z_DEFAULT_STRATEGY))
+    return -1;
+
+  z.next_in = (Bytef *)source;
+  z.avail_in = slen;
+  z.total_in = 0;
+
+  /* write gzip header */
+
+  c = (unsigned char *)dest;
+  c[0] = 0x1f;
+  c[1] = 0x8b;
+  c[2] = Z_DEFLATED;
+  c[3] = 0; /* options */
+  c[4] = (mtime >>  0) & 0xff;
+  c[5] = (mtime >>  8) & 0xff;
+  c[6] = (mtime >> 16) & 0xff;
+  c[7] = (mtime >> 24) & 0xff;
+  c[8] = 0x00; /* extra flags */
+  c[9] = 0x03; /* UNIX */
+
+  z.next_out = c + 10;
+  z.avail_out = dlen - 10 - 8;
+  z.total_out = 0;
+
+  if (deflate(&z, Z_FINISH) != Z_STREAM_END) {
+    deflateEnd(&z);
+    return -1;
+  }
+
+  crc = generate_crc32c(source, slen);
+
+  c = (unsigned char *)dest + 10 + z.total_out;
+  c[0] = (crc >>  0) & 0xff;
+  c[1] = (crc >>  8) & 0xff;
+  c[2] = (crc >> 16) & 0xff;
+  c[3] = (crc >> 24) & 0xff;
+  c[4] = (z.total_in >>  0) & 0xff;
+  c[5] = (z.total_in >>  8) & 0xff;
+  c[6] = (z.total_in >> 16) & 0xff;
+  c[7] = (z.total_in >> 24) & 0xff;
+
+  if (deflateEnd(&z) != Z_OK)
+    return -1;
+  return 10 + z.total_out + 8;
+}
diff --git a/src/zpack.h b/src/zpack.h
new file mode 100644
index 0000000..5d3f82f
--- /dev/null
+++ b/src/zpack.h
@@ -0,0 +1,12 @@
+#ifndef _ZPACK_H
+#define _ZPACK_H
+
+ssize_t def(const char *source, size_t slen, char *dest, size_t dlen,
+            int level);
+ssize_t inf(const char *source, size_t slen, char *dest, size_t dlen);
+
+ssize_t gzInflate(const char *source, size_t slen, char *dest, size_t dlen);
+ssize_t gzDeflate(const char *source, size_t slen, char *dest, size_t dlen,
+                  time_t mtime);
+
+#endif





More information about the tor-commits mailing list