Update from 0.9.8 stable. Eliminate duplicate error codes.
[openssl.git] / crypto / comp / c_zlib.c
index a14e82052725a7b0a48e5d0642e4a9d17eadbaa1..941b807eb39149f2d1e3fae2378676f6c8734407 100644 (file)
@@ -3,6 +3,7 @@
 #include <string.h>
 #include <openssl/objects.h>
 #include <openssl/comp.h>
+#include <openssl/err.h>
 
 COMP_METHOD *COMP_zlib(void );
 
@@ -14,12 +15,23 @@ static COMP_METHOD zlib_method_nozlib={
        NULL,
        NULL,
        NULL,
+       NULL,
        };
 
-#ifdef ZLIB
+#ifndef ZLIB
+#undef ZLIB_SHARED
+#else
 
 #include <zlib.h>
 
+static int zlib_stateful_init(COMP_CTX *ctx);
+static void zlib_stateful_finish(COMP_CTX *ctx);
+static int zlib_stateful_compress_block(COMP_CTX *ctx, unsigned char *out,
+       unsigned int olen, unsigned char *in, unsigned int ilen);
+static int zlib_stateful_expand_block(COMP_CTX *ctx, unsigned char *out,
+       unsigned int olen, unsigned char *in, unsigned int ilen);
+
+#if 0
 static int zlib_compress_block(COMP_CTX *ctx, unsigned char *out,
        unsigned int olen, unsigned char *in, unsigned int ilen);
 static int zlib_expand_block(COMP_CTX *ctx, unsigned char *out,
@@ -28,7 +40,7 @@ static int zlib_expand_block(COMP_CTX *ctx, unsigned char *out,
 static int zz_uncompress(Bytef *dest, uLongf *destLen, const Bytef *source,
        uLong sourceLen);
 
-static COMP_METHOD zlib_method={
+static COMP_METHOD zlib_stateless_method={
        NID_zlib_compression,
        LN_zlib_compression,
        NULL,
@@ -36,46 +48,197 @@ static COMP_METHOD zlib_method={
        zlib_compress_block,
        zlib_expand_block,
        NULL,
+       NULL,
+       };
+#endif
+
+static COMP_METHOD zlib_stateful_method={
+       NID_zlib_compression,
+       LN_zlib_compression,
+       zlib_stateful_init,
+       zlib_stateful_finish,
+       zlib_stateful_compress_block,
+       zlib_stateful_expand_block,
+       NULL,
+       NULL,
        };
 
 /* 
  * When OpenSSL is built on Windows, we do not want to require that
  * the ZLIB.DLL be available in order for the OpenSSL DLLs to
  * work.  Therefore, all ZLIB routines are loaded at run time
- * and we do not link to a .LIB file.
+ * and we do not link to a .LIB file when ZLIB_SHARED is set.
  */
-#if defined(WINDOWS) || defined(WIN32)
-#include <windows.h>
+#if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
+# include <windows.h>
+#endif /* !(OPENSSL_SYS_WINDOWS || OPENSSL_SYS_WIN32) */
 
-/* Prototypes for built in stubs */
-int stub_compress(Bytef *dest,uLongf *destLen,
-       const Bytef *source, uLong sourceLen);
-int stub_inflateEnd(z_streamp strm);
-int stub_inflate(z_streamp strm, int flush);
-int stub_inflateInit_(z_streamp strm, const char * version, int stream_size);
+#ifdef ZLIB_SHARED
+#include <openssl/dso.h>
 
 /* Function pointers */
-typedef int (_stdcall *compress_ft)(Bytef *dest,uLongf *destLen,
+typedef int (*compress_ft)(Bytef *dest,uLongf *destLen,
        const Bytef *source, uLong sourceLen);
-typedef int (_stdcall *inflateEnd_ft)(z_streamp strm);
-typedef int (_stdcall *inflate_ft)(z_streamp strm, int flush);
-typedef int (_stdcall *inflateInit__ft)(z_streamp strm, const char * version,
-       int stream_size);
-static compress_ft p_compress=NULL;
-static inflateEnd_ft p_inflateEnd=NULL;
-static inflate_ft p_inflate=NULL;
-static inflate_Init__ft p_inflateInit_=NULL;
+typedef int (*inflateEnd_ft)(z_streamp strm);
+typedef int (*inflate_ft)(z_streamp strm, int flush);
+typedef int (*inflateInit__ft)(z_streamp strm,
+       const char * version, int stream_size);
+typedef int (*deflateEnd_ft)(z_streamp strm);
+typedef int (*deflate_ft)(z_streamp strm, int flush);
+typedef int (*deflateInit__ft)(z_streamp strm, int level,
+       const char * version, int stream_size);
+static compress_ft     p_compress=NULL;
+static inflateEnd_ft   p_inflateEnd=NULL;
+static inflate_ft      p_inflate=NULL;
+static inflateInit__ft p_inflateInit_=NULL;
+static deflateEnd_ft   p_deflateEnd=NULL;
+static deflate_ft      p_deflate=NULL;
+static deflateInit__ft p_deflateInit_=NULL;
 
 static int zlib_loaded = 0;     /* only attempt to init func pts once */
+static DSO *zlib_dso = NULL;
+
+#define compress                p_compress
+#define inflateEnd              p_inflateEnd
+#define inflate                 p_inflate
+#define inflateInit_            p_inflateInit_
+#define deflateEnd              p_deflateEnd
+#define deflate                 p_deflate
+#define deflateInit_            p_deflateInit_
+#endif /* ZLIB_SHARED */
+
+struct zlib_state
+       {
+       z_stream istream;
+       z_stream ostream;
+       };
+
+static int zlib_stateful_ex_idx = -1;
 
-#define compress                stub_compress
-#define inflateEnd              stub_inflateEnd
-#define inflate                 stub_inflate
-#define inflateInit_            stub_inflateInit_
-#endif /* WINDOWS || WIN32 */
+static void zlib_stateful_free_ex_data(void *obj, void *item,
+       CRYPTO_EX_DATA *ad, int ind,long argl, void *argp)
+       {
+       struct zlib_state *state = (struct zlib_state *)item;
+       inflateEnd(&state->istream);
+       deflateEnd(&state->ostream);
+       OPENSSL_free(state);
+       }
 
+static int zlib_stateful_init(COMP_CTX *ctx)
+       {
+       int err;
+       struct zlib_state *state =
+               (struct zlib_state *)OPENSSL_malloc(sizeof(struct zlib_state));
+
+       if (state == NULL)
+               goto err;
+
+       state->istream.zalloc = Z_NULL;
+       state->istream.zfree = Z_NULL;
+       state->istream.opaque = Z_NULL;
+       state->istream.next_in = Z_NULL;
+       state->istream.next_out = Z_NULL;
+       state->istream.avail_in = 0;
+       state->istream.avail_out = 0;
+       err = inflateInit_(&state->istream,
+               ZLIB_VERSION, sizeof(z_stream));
+       if (err != Z_OK)
+               goto err;
+
+       state->ostream.zalloc = Z_NULL;
+       state->ostream.zfree = Z_NULL;
+       state->ostream.opaque = Z_NULL;
+       state->ostream.next_in = Z_NULL;
+       state->ostream.next_out = Z_NULL;
+       state->ostream.avail_in = 0;
+       state->ostream.avail_out = 0;
+       err = deflateInit_(&state->ostream,Z_DEFAULT_COMPRESSION,
+               ZLIB_VERSION, sizeof(z_stream));
+       if (err != Z_OK)
+               goto err;
+
+       CRYPTO_new_ex_data(CRYPTO_EX_INDEX_COMP,ctx,&ctx->ex_data);
+       if (zlib_stateful_ex_idx == -1)
+               {
+               CRYPTO_w_lock(CRYPTO_LOCK_COMP);
+               if (zlib_stateful_ex_idx == -1)
+                       zlib_stateful_ex_idx =
+                               CRYPTO_get_ex_new_index(CRYPTO_EX_INDEX_COMP,
+                                       0,NULL,NULL,NULL,zlib_stateful_free_ex_data);
+               CRYPTO_w_unlock(CRYPTO_LOCK_COMP);
+               if (zlib_stateful_ex_idx == -1)
+                       goto err;
+               }
+       CRYPTO_set_ex_data(&ctx->ex_data,zlib_stateful_ex_idx,state);
+       return 1;
+ err:
+       if (state) OPENSSL_free(state);
+       return 0;
+       }
+
+static void zlib_stateful_finish(COMP_CTX *ctx)
+       {
+       CRYPTO_free_ex_data(CRYPTO_EX_INDEX_COMP,ctx,&ctx->ex_data);
+       }
+
+static int zlib_stateful_compress_block(COMP_CTX *ctx, unsigned char *out,
+       unsigned int olen, unsigned char *in, unsigned int ilen)
+       {
+       int err = Z_OK;
+       struct zlib_state *state =
+               (struct zlib_state *)CRYPTO_get_ex_data(&ctx->ex_data,
+                       zlib_stateful_ex_idx);
+
+       if (state == NULL)
+               return -1;
+
+       state->ostream.next_in = in;
+       state->ostream.avail_in = ilen;
+       state->ostream.next_out = out;
+       state->ostream.avail_out = olen;
+       if (ilen > 0)
+               err = deflate(&state->ostream, Z_SYNC_FLUSH);
+       if (err != Z_OK)
+               return -1;
+#ifdef DEBUG_ZLIB
+       fprintf(stderr,"compress(%4d)->%4d %s\n",
+               ilen,olen - state->ostream.avail_out,
+               (ilen != olen - state->ostream.avail_out)?"zlib":"clear");
+#endif
+       return olen - state->ostream.avail_out;
+       }
+
+static int zlib_stateful_expand_block(COMP_CTX *ctx, unsigned char *out,
+       unsigned int olen, unsigned char *in, unsigned int ilen)
+       {
+       int err = Z_OK;
+
+       struct zlib_state *state =
+               (struct zlib_state *)CRYPTO_get_ex_data(&ctx->ex_data,
+                       zlib_stateful_ex_idx);
+
+       if (state == NULL)
+               return 0;
+
+       state->istream.next_in = in;
+       state->istream.avail_in = ilen;
+       state->istream.next_out = out;
+       state->istream.avail_out = olen;
+       if (ilen > 0)
+               err = inflate(&state->istream, Z_SYNC_FLUSH);
+       if (err != Z_OK)
+               return -1;
+#ifdef DEBUG_ZLIB
+       fprintf(stderr,"expand(%4d)->%4d %s\n",
+               ilen,olen - state->istream.avail_out,
+               (ilen != olen - state->istream.avail_out)?"zlib":"clear");
+#endif
+       return olen - state->istream.avail_out;
+       }
+
+#if 0
 static int zlib_compress_block(COMP_CTX *ctx, unsigned char *out,
-            unsigned int olen, unsigned char *in, unsigned int ilen)
+       unsigned int olen, unsigned char *in, unsigned int ilen)
        {
        unsigned long l;
        int i;
@@ -108,7 +271,7 @@ static int zlib_compress_block(COMP_CTX *ctx, unsigned char *out,
        }
 
 static int zlib_expand_block(COMP_CTX *ctx, unsigned char *out,
-            unsigned int olen, unsigned char *in, unsigned int ilen)
+       unsigned int olen, unsigned char *in, unsigned int ilen)
        {
        unsigned long l;
        int i;
@@ -150,7 +313,8 @@ static int zz_uncompress (Bytef *dest, uLongf *destLen, const Bytef *source,
     stream.zalloc = (alloc_func)0;
     stream.zfree = (free_func)0;
 
-    err = inflateInit(&stream);
+    err = inflateInit_(&stream,
+           ZLIB_VERSION, sizeof(z_stream));
     if (err != Z_OK) return err;
 
     err = inflate(&stream, Z_FINISH);
@@ -163,6 +327,7 @@ static int zz_uncompress (Bytef *dest, uLongf *destLen, const Bytef *source,
     err = inflateEnd(&stream);
     return err;
 }
+#endif
 
 #endif
 
@@ -170,72 +335,53 @@ COMP_METHOD *COMP_zlib(void)
        {
        COMP_METHOD *meth = &zlib_method_nozlib;
 
-#ifdef ZLIB
-#if defined(WINDOWS) || defined(WIN32)
-       if (!zlib_method)
+#ifdef ZLIB_SHARED
+       if (!zlib_loaded)
                {
-               zlib_dso = DSO_load(NULL, "ZLIB", NULL, 0);
-               if (dso != NULL)
+#if defined(OPENSSL_SYS_WINDOWS) || defined(OPENSSL_SYS_WIN32)
+               zlib_dso = DSO_load(NULL, "ZLIB1", NULL, 0);
+#else
+               zlib_dso = DSO_load(NULL, "z", NULL, 0);
+#endif
+               if (zlib_dso != NULL)
                        {
-                       (FARPROC) p_compress
-                               = (compress_ft) DSO_bind_func(dso, "compress");
-                       (FARPROC) p_inflateEnd
-                               = (inflateEnd_ft) DSO_bind_func(dso, "inflateEnd");
-                       (FARPROC) p_inflate
-                               = (inflate_ft) DSO_bind_func(dso, "inflate");
-                       (FARPROC) p_inflateInit_
-                               = (inflateInit__ft) DSO_bind_func(dso, "inflateInit_");
-                       zlib_loaded++;
-                       meth = &zlib_method;
+                       p_compress
+                               = (compress_ft) DSO_bind_func(zlib_dso,
+                                       "compress");
+                       p_inflateEnd
+                               = (inflateEnd_ft) DSO_bind_func(zlib_dso,
+                                       "inflateEnd");
+                       p_inflate
+                               = (inflate_ft) DSO_bind_func(zlib_dso,
+                                       "inflate");
+                       p_inflateInit_
+                               = (inflateInit__ft) DSO_bind_func(zlib_dso,
+                                       "inflateInit_");
+                       p_deflateEnd
+                               = (deflateEnd_ft) DSO_bind_func(zlib_dso,
+                                       "deflateEnd");
+                       p_deflate
+                               = (deflate_ft) DSO_bind_func(zlib_dso,
+                                       "deflate");
+                       p_deflateInit_
+                               = (deflateInit__ft) DSO_bind_func(zlib_dso,
+                                       "deflateInit_");
+
+                       if (p_compress && p_inflateEnd && p_inflate
+                               && p_inflateInit_ && p_deflateEnd
+                               && p_deflate && p_deflateInit_)
+                               zlib_loaded++;
                        }
                }
 
-#else
-       meth = &zlib_method;
 #endif
+#ifdef ZLIB_SHARED
+       if (zlib_loaded)
+#endif
+#if defined(ZLIB) || defined(ZLIB_SHARED)
+               meth = &zlib_stateful_method;
 #endif
 
        return(meth);
        }
 
-#ifdef ZLIB
-#if defined(WINDOWS) || defined(WIN32)
-/* Stubs for each function to be dynamicly loaded */
-static int 
-stub_compress(Bytef *dest,uLongf *destLen,const Bytef *source, uLong sourceLen)
-       {
-       if (p_compress)
-               return(p_compress(dest,destLen,source,sourceLen));
-       else
-               return(Z_MEM_ERROR);
-       }
-
-static int
-stub_inflateEnd(z_streamp strm)
-       {
-       if ( p_inflateEnd )
-               return(p_inflateEnd(strm));
-       else
-               return(Z_MEM_ERROR);
-       }
-
-static int
-stub_inflate(z_streamp strm, int flush)
-       {
-       if ( p_inflate )
-               return(p_inflate(strm,flush));
-       else
-               return(Z_MEM_ERROR);
-       }
-
-static int
-stub_inflateInit_(z_streamp strm, const char * version, int stream_size)
-       {
-       if ( p_inflateInit_ )
-               return(p_inflateInit_(strm,version,stream_size));
-       else
-               return(Z_MEM_ERROR);
-       }
-
-#endif /* WINDOWS || WIN32 */
-#endif /* ZLIB */