From a47793c85f727ca292c03d4779fac3d5e8805f11 Mon Sep 17 00:00:00 2001
From: Michael Wallner <mike@php.net>
Date: Thu, 6 Oct 2005 16:05:08 +0000
Subject: [PATCH] - fix warnings with etag caching at shutdown - add 2 new
 tests

---
 http_cache_api.c            | 58 +++++++++++++++++--------------------
 tests/HttpResponse_003.phpt | 30 +++++++++++++++++++
 tests/HttpResponse_004.phpt | 26 +++++++++++++++++
 3 files changed, 83 insertions(+), 31 deletions(-)
 create mode 100644 tests/HttpResponse_003.phpt
 create mode 100644 tests/HttpResponse_004.phpt

diff --git a/http_cache_api.c b/http_cache_api.c
index 5eca48a..1c96789 100644
--- a/http_cache_api.c
+++ b/http_cache_api.c
@@ -209,17 +209,16 @@ PHP_HTTP_API STATUS _http_cache_etag(const char *etag, size_t etag_len,
 		}
 		return SUCCESS;
 	}
-
-	/* if no etag is given and we didn't already start ob_etaghandler -- start it */
-	if (HTTP_G(etag).started) {
-		return SUCCESS;
-	}
-
-	if (HTTP_G(etag).started = (SUCCESS == php_start_ob_buffer_named("ob_etaghandler", HTTP_G(send).buffer_size, 1 TSRMLS_CC))) {
-		return SUCCESS;
-	} else {
+	
+	/* already running? */
+	if (php_ob_handler_used("ob_etaghandler" TSRMLS_CC)) {
+		http_error(HE_WARNING, HTTP_E_RUNTIME, "ob_etaghandler can only be used once");
 		return FAILURE;
 	}
+	
+	/* we want the etag handler to run */
+	HTTP_G(etag).started = 1;
+	return php_start_ob_buffer_named("ob_etaghandler", HTTP_G(send).buffer_size, 1 TSRMLS_CC);
 }
 /* }}} */
 
@@ -227,39 +226,36 @@ PHP_HTTP_API STATUS _http_cache_etag(const char *etag, size_t etag_len,
 PHP_HTTP_API void _http_ob_etaghandler(char *output, uint output_len,
 	char **handled_output, uint *handled_output_len, int mode TSRMLS_DC)
 {
-	if (mode & PHP_OUTPUT_HANDLER_START) {
-		if (HTTP_G(etag).started) {
-			http_error(HE_WARNING, HTTP_E_RUNTIME, "ob_etaghandler can only be used once");
-			return;
+	/* passthru */
+	*handled_output_len = output_len;
+	*handled_output = estrndup(output, output_len);
+	
+	/* are we supposed to run? */
+	if (HTTP_G(etag).started) {
+		/* initialize the etag context */
+		if (mode & PHP_OUTPUT_HANDLER_START) {
+			HTTP_G(etag).ctx = http_etag_init();
 		}
-		HTTP_G(etag).started = 1;
-		HTTP_G(etag).ctx = http_etag_init();
-	}
-
-	http_etag_update(HTTP_G(etag).ctx, output, output_len);
-
-	if (mode & PHP_OUTPUT_HANDLER_END) {
-		char *etag = http_etag_finish(HTTP_G(etag).ctx);
 		
-		/* just do that if desired */
-		if (HTTP_G(etag).started) {
+		/* update */
+		http_etag_update(HTTP_G(etag).ctx, output, output_len);
+		
+		/* finish */
+		if (mode & PHP_OUTPUT_HANDLER_END) {
 			char *sent_header = NULL;
+			char *etag = http_etag_finish(HTTP_G(etag).ctx);
 			
 			http_send_cache_control(HTTP_DEFAULT_CACHECONTROL, lenof(HTTP_DEFAULT_CACHECONTROL));
 			http_send_etag_ex(etag, strlen(etag), &sent_header);
 			
 			if (http_match_etag("HTTP_IF_NONE_MATCH", etag)) {
-				efree(etag);
-				http_exit_ex(304, sent_header, NULL, 0);
-			} else {
-				STR_FREE(sent_header);
+				http_exit_ex(304, sent_header, etag, 0);
 			}
+			
+			STR_FREE(sent_header);
+			STR_FREE(etag);
 		}
-		efree(etag);
 	}
-
-	*handled_output_len = output_len;
-	*handled_output = estrndup(output, output_len);
 }
 /* }}} */
 
diff --git a/tests/HttpResponse_003.phpt b/tests/HttpResponse_003.phpt
new file mode 100644
index 0000000..027d536
--- /dev/null
+++ b/tests/HttpResponse_003.phpt
@@ -0,0 +1,30 @@
+--TEST--
+HttpResponse - send gzipped file with caching headers
+--SKIPIF--
+<?php 
+include 'skip.inc';
+checkver(5);
+checkcgi();
+checkext('zlib');
+?>
+--ENV--
+HTTP_ACCEPT_ENCODING=gzip
+--FILE--
+<?php
+HttpResponse::setGzip(true);
+HttpResponse::setCache(true);
+HttpResponse::setCacheControl('public', 3600);
+HttpResponse::setFile(__FILE__);
+HttpResponse::send();
+?>
+--EXPECTF--
+X-Powered-By: PHP/5.1.0RC2-dev
+ETag: "%s"
+Cache-Control: public, must-revalidate, max_age=3600
+Last-Modified: %s, %d %s 20%d %d:%d:%d GMT
+Content-Type: %s
+Accept-Ranges: bytes
+Content-Encoding: gzip
+Vary: Accept-Encoding
+
+%s
diff --git a/tests/HttpResponse_004.phpt b/tests/HttpResponse_004.phpt
new file mode 100644
index 0000000..ed198d6
--- /dev/null
+++ b/tests/HttpResponse_004.phpt
@@ -0,0 +1,26 @@
+--TEST--
+HttpResponse - send cached gzipped data
+--SKIPIF--
+<?php 
+include 'skip.inc';
+checkver(5);
+checkcgi();
+checkext('zlib');
+?>
+--ENV--
+HTTP_ACCEPT_ENCODING=gzip
+HTTP_IF_NONE_MATCH="80b285463881575891e86ba7bfecb4d0"
+--FILE--
+<?php
+HttpResponse::setGzip(true);
+HttpResponse::setCache(true);
+HttpResponse::setCacheControl('public', 3600);
+HttpResponse::setData(file_get_contents(__FILE__));
+HttpResponse::send();
+?>
+--EXPECTF--
+Status: 304
+Content-type: text/html
+X-Powered-By: PHP/5.1.0RC2-dev
+Cache-Control: public, must-revalidate, max_age=3600
+ETag: "80b285463881575891e86ba7bfecb4d0"
-- 
2.30.2