From b6798866d058cd7ce69fc3c7944aff85a1693170 Mon Sep 17 00:00:00 2001
From: Nick Mathewson <nickm@torproject.org>
Date: Wed, 8 Sep 2004 06:47:33 +0000
Subject: Idiot-proof uncompress; make sure it always nul-terminates its
 output. Also, make all compression methods nonzero.

svn:r2334
---
 src/common/torgzip.c | 7 +++++--
 src/common/torgzip.h | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/common/torgzip.c b/src/common/torgzip.c
index 1b3fce2e8..b64d5f60f 100644
--- a/src/common/torgzip.c
+++ b/src/common/torgzip.c
@@ -10,7 +10,6 @@
 
 #include "orconfig.h"
 
-
 #include <stdlib.h>
 #include <stdio.h>
 #include <assert.h>
@@ -74,7 +73,6 @@ tor_gzip_compress(char **out, size_t *out_len,
   if (deflateInit2(stream, Z_BEST_COMPRESSION, Z_DEFLATED,
 		   method_bits(method),
 		   8, Z_DEFAULT_STRATEGY) != Z_OK) {
-    printf("Z");
     log_fn(LOG_WARN, "Error from deflateInit2: %s",
 	   stream->msg?stream->msg:"<no message>");
     goto err;
@@ -200,6 +198,11 @@ tor_gzip_uncompress(char **out, size_t *out_len,
   }
   tor_free(stream);
 
+  /* NUL-terminate output. */
+  if (out_size == *out_len)
+    *out = tor_realloc(*out, out_size + 1);
+  (*out)[*out_len] = '\0';
+
   return 0;
  err:
   if (stream) {
diff --git a/src/common/torgzip.h b/src/common/torgzip.h
index 61bb6adbc..6c88f6927 100644
--- a/src/common/torgzip.h
+++ b/src/common/torgzip.h
@@ -10,7 +10,7 @@
 #ifndef __TORGZIP_H
 #define __TORGZIP_H
 
-typedef enum { GZIP_METHOD, ZLIB_METHOD } compress_method_t;
+typedef enum { GZIP_METHOD=1, ZLIB_METHOD=2 } compress_method_t;
 
 int
 tor_gzip_compress(char **out, size_t *out_len,
-- 
cgit v1.2.3