projects / platform / upstream / curl.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 61ae7e9)
zlib: ensure user provided memory functions are used by zlib, when given
authorYang Tse <yangsita@gmail.com>
Sun, 21 Aug 2011 11:15:34 +0000 (13:15 +0200)
committerYang Tse <yangsita@gmail.com>
Sun, 21 Aug 2011 11:24:46 +0000 (13:24 +0200)
As a bonus, this lets our MemoryTracking subsystem track zlib operations.
And also fixes a shortcut some zlib 1.2.x versions took using malloc()
instead of calloc(), which would trigger memory debuggers warnings on
memory being used without having been initialized.

lib/content_encoding.c patch | blob | history
src/mkhelp.pl patch | blob | history

diff --git a/lib/content_encoding.c b/lib/content_encoding.c
index 84d76f4b9841ebb00166cb0b2372c30f8ce24797..3276ef98886bb46979f80a2792be8decd6766295 100644 (file)
--- a/lib/content_encoding.c
+++ b/lib/content_encoding.c
@@ -49,6 +49,21 @@
 #define COMMENT      0x10 /* bit 4 set: file comment present */
 #define RESERVED     0xE0 /* bits 5..7: reserved */
 
+static voidpf
+zalloc_cb(voidpf opaque, unsigned int items, unsigned int size)
+{
+  (void) opaque;
+  /* not a typo, keep it calloc() */
+  return (voidpf) calloc(items, size);
+}
+
+static void
+zfree_cb(voidpf opaque, voidpf ptr)
+{
+  (void) opaque;
+  free(ptr);
+}
+
 static CURLcode
 process_zlib_error(struct connectdata *conn, z_stream *z)
 {
@@ -159,6 +174,9 @@ Curl_unencode_deflate_write(struct connectdata *conn,
   /* Initialize zlib? */
   if(k->zlib_init == ZLIB_UNINIT) {
     memset(z, 0, sizeof(z_stream));
+    z->zalloc = (alloc_func)zalloc_cb;
+    z->zfree = (free_func)zfree_cb;
+
     if(inflateInit(z) != Z_OK)
       return process_zlib_error(conn, z);
     k->zlib_init = ZLIB_INIT;
@@ -266,6 +284,8 @@ Curl_unencode_gzip_write(struct connectdata *conn,
   /* Initialize zlib? */
   if(k->zlib_init == ZLIB_UNINIT) {
     memset(z, 0, sizeof(z_stream));
+    z->zalloc = (alloc_func)zalloc_cb;
+    z->zfree = (free_func)zfree_cb;
 
     if(strcmp(zlibVersion(), "1.2.0.4") >= 0) {
       /* zlib ver. >= 1.2.0.4 supports transparent gzip decompressing */
diff --git a/src/mkhelp.pl b/src/mkhelp.pl
index e9de8429d0cf209ec0cc863599a0eb080d77b41d..d09d6c50862ec1c49a8f8aad2ccb1319298698cd 100644 (file)
--- a/src/mkhelp.pl
+++ b/src/mkhelp.pl
@@ -133,7 +133,6 @@ print <<HEAD
  * NEVER EVER edit this manually, fix the mkhelp.pl script instead!
  * Generation time: $now
  */
-#include "setup.h"
 #ifdef USE_MANUAL
 #include "hugehelp.h"
 HEAD
@@ -141,6 +140,9 @@ HEAD
 if($c) {
     print <<HEAD
 #include <zlib.h>
+#if defined(CURLDEBUG) && defined(CURLTOOLDEBUG)
+#include "memdebug.h"
+#endif
 static const unsigned char hugehelpgz[] = {
   /* This mumbo-jumbo is the huge help text compressed with gzip.
      Thanks to this operation, the size of this data shrunk from $gzip
@@ -165,6 +167,17 @@ HEAD
 
     print <<EOF
 #define BUF_SIZE 0x10000
+static voidpf zalloc_func(voidpf opaque, unsigned int items, unsigned int size)
+{
+  (void) opaque;
+  /* not a typo, keep it calloc() */
+  return (voidpf) calloc(items, size);
+}
+static void zfree_func(voidpf opaque, voidpf ptr)
+{
+  (void) opaque;
+  free(ptr);
+}
 /* Decompress and send to stdout a gzip-compressed buffer */
 void hugehelp(void)
 {
@@ -178,6 +191,8 @@ void hugehelp(void)
 
   headerlen = 10;
   memset(&z, 0, sizeof(z_stream));
+  z.zalloc = (alloc_func)zalloc_func;
+  z.zfree = (free_func)zfree_func;
   z.avail_in = (unsigned int)(sizeof(hugehelpgz) - headerlen);
   z.next_in = (unsigned char *)hugehelpgz + headerlen;
 
Domain: Network & Connectivity / Data Network;