Fix zlib driver crashing on 1024-bytes chunks

Thanks to syhpoon

.gitignore

@@ -0,0 +1,5 @@
+.eunit/
+.rebar/
+ebin/
+priv/
+*.o

Makefile

@@ -6,4 +6,7 @@ src:
 clean:
 	rebar clean
 
-.PHONY: clean src
+test:
+	rebar eunit
+
+.PHONY: clean src test

c_src/ezlib_drv.c

@@ -136,6 +136,13 @@ static ErlDrvSSizeT ezlib_drv_control(ErlDrvData handle,
 	    d->d_stream->avail_out = BUF_SIZE;
 
 	    err = deflate(d->d_stream, Z_SYNC_FLUSH);
+
+             // Output buffer was completely consumed and we have no more data to process
+             // http://www.zlib.net/zlib_faq.html#faq05
+             if(err == Z_BUF_ERROR && d->d_stream->avail_out == BUF_SIZE) {
+                break;
+             }
+
 	    die_unless((err == Z_OK) || (err == Z_STREAM_END),
 		       "Deflate error");
 
@@ -164,6 +171,13 @@ static ErlDrvSSizeT ezlib_drv_control(ErlDrvData handle,
 	       d->i_stream->avail_out = BUF_SIZE;
 
 	       err = inflate(d->i_stream, Z_SYNC_FLUSH);
+
+               // Output buffer was completely consumed and we have no more data to process
+               // http://www.zlib.net/zlib_faq.html#faq05
+               if(err == Z_BUF_ERROR && d->i_stream->avail_out == BUF_SIZE) {
+                  break;
+               }
+
 	       die_unless((err == Z_OK) || (err == Z_STREAM_END),
 			  "Inflate error");
 

test/zlib_test.erl

@@ -0,0 +1,20 @@
+-module(zlib_test).
+-include_lib("eunit/include/eunit.hrl").
+
+-compile(export_all).
+
+inflate_1024_test()->
+    DeflateCmd = 1,
+    InflateCmd = 2,
+    %% Just random 1024 bytes data
+    Data = <<"iGzHvZgQbX6oTJYViq9Xy3MncZtDlypeFWNF61owolz8boGSqsr5wxtfT42QSJIMShM6MZInMHBUh9unSYpuwqfzlLIZ3ZNfjje7YsLhaITQzXXNABsXdvjUMU437FJppLeLVob9M4ZcUpAOdMSpCB1t6KKgWBmNEdkbo9VL2seDdpYT3AcZlGOsy73pk7og3KirclNJM6NFI2wARifUdE9ShEJhlncjOQnG3ExCnjGsRSGcJl8eh6Vjsq8pavfhOTYyE6z4xBPG4jKtZCiqVO1uaWQjeTOPZnhPdnVThcrpVS1GiB33xZz9p6pRw2ricGbsbZoUNIQQqBsL5VGZeNQhRLAoiNsT1S3kJl43Mm6rvIy8rM5Jm9y4NXvkZcjgPpmfySmgXSygWshbjGbaUlwub2ZCe7C5Z1vQL7n5DutDefQcFXJScMWiOW11ye9UIx9u29qVp2HhzhJ1dtAheBBk6unVNTVkQJoyMcdVAFSQ4yob66WKm33xkrtxgvyAFesdWwXFc6VNGxtXjBYu2Hd8jyDSAzl9wK3CRF9rraKSLSv3ycTFty16noELhJwzQQ2zjkQerms7MR98nmmGFLA1DgaZXSwQnWl4A3k8mTvxsWfhVgZDdkcvwqmFB3JHJsKfUuakt921dJ3uXI5ywsV3nVi3Skpuj3OiV9mfZzNp2J8mE7zxX5RtgV3phcfIEA52gBb8TngbL8hnPPkJy1tTaYYtNCcZEHaLd8XKUPa2aAr7rMdLzqhiaGPY6JWIncDWKGW8EOCOXXKVIgXtFEhAjZlc6e7LSOvkBbJTkh3hoS9Ow9daLJrnIGIMDRT1NxhHX9iYzvUAs9c78zZjTA7CWFWRhj5Ui7koQrMXzIszmVDTGe3y5Ml6bNMMUtZUT2eK7QSnU4wFwFqeviN56XN4qugVMW9J6YbVY5bZA77fpUW92TUHXA2jhXldNpFL8cc82p5XNNQERaqBDA15EHuotqaVDZGCYPkvPLqXlx9UjOLJltDAAKYlifQQonPW">>,
+
+    case erl_ddll:load_driver("../priv/lib", ezlib_drv) of
+        ok -> ok;
+        {error, already_loaded} -> ok
+    end,
+
+    Port = open_port({spawn, ezlib_drv}, [binary]),
+
+    <<0, Deflated/binary>> = port_control(Port, DeflateCmd, Data),
+    ?assertMatch(<<0, _Data/binary>>, port_control(Port, InflateCmd, Deflated)).