X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-http;a=blobdiff_plain;f=php_http_cache_api.h;h=d7e53b77b41bc5c073ae52e6ef3c66a750919387;hp=0a93f681f01efaea587e41f77a110d33ed9d4faa;hb=c07ee6eaef449abb292bb84d5ba61c19a467e184;hpb=9d0eaae0af2de89a57be321c507e5a9956ba99f9

diff --git a/php_http_cache_api.h b/php_http_cache_api.h
index 0a93f68..d7e53b7 100644
--- a/php_http_cache_api.h
+++ b/php_http_cache_api.h
@@ -18,9 +18,148 @@
 #ifndef PHP_HTTP_CACHE_API_H
 #define PHP_HTTP_CACHE_API_H
 
+#include "zend_ini.h"
+
+#include "ext/standard/md5.h"
+#include "ext/standard/sha1.h"
+
 #include "php_http_std_defs.h"
+#include "php_http.h"
+#include "php_http_api.h"
 #include "php_http_send_api.h"
 
+#ifdef HTTP_HAVE_MHASH
+#	include <mhash.h>
+#endif
+
+ZEND_EXTERN_MODULE_GLOBALS(http);
+
+#define http_cache_global_init() _http_cache_global_init(INIT_FUNC_ARGS_PASSTHRU)
+extern STATUS _http_cache_global_init(INIT_FUNC_ARGS);
+
+typedef enum {
+	HTTP_ETAG_MD5 = -2,
+	HTTP_ETAG_SHA1 = -1,
+} http_etag_mode;
+
+#ifdef HTTP_HAVE_MHASH
+static void *http_etag_alloc_mhash_digest(size_t size)
+{
+	return emalloc(size);
+}
+#endif
+
+#define http_etag_digest(d, l) _http_etag_digest((d), (l) TSRMLS_CC)
+static inline char *_http_etag_digest(const unsigned char *digest, int len TSRMLS_DC)
+{
+	int i;
+	char *hex = emalloc(len * 2 + 1);
+	char *ptr = hex;
+	
+	/* optimize this --
+		look at apache's make_etag */
+	for (i = 0; i < len; ++i) {
+		sprintf(ptr, "%02x", digest[i]);
+		ptr += 2;
+	}
+	*ptr = '\0';
+	
+	return hex;
+}
+
+#define http_etag_init() _http_etag_init(TSRMLS_C)
+static inline void *_http_etag_init(TSRMLS_D)
+{
+	void *ctx = NULL;
+	long mode = INI_INT("http.etag_mode");
+	
+	switch (mode)
+	{
+		case HTTP_ETAG_SHA1:
+			PHP_SHA1Init(ctx = emalloc(sizeof(PHP_SHA1_CTX)));
+		break;
+		
+		case HTTP_ETAG_MD5:
+#ifndef HTTP_HAVE_MHASH
+		default:
+#endif
+			PHP_MD5Init(ctx = emalloc(sizeof(PHP_MD5_CTX)));
+		break;
+		
+#ifdef HTTP_HAVE_MHASH
+		default:
+			if ((mode < 0) || ((ulong)mode > mhash_count()) || (!(ctx = mhash_init(mode)))) {
+				http_error_ex(HE_ERROR, HTTP_E_RUNTIME, "Invalid ETag mode: %ld", mode);
+			}
+		break;
+#endif
+	}
+	
+	return ctx;
+}
+
+#define http_etag_finish(c) _http_etag_finish((c) TSRMLS_CC)
+static inline char *_http_etag_finish(void *ctx TSRMLS_DC)
+{
+	char *etag = NULL;
+	unsigned char digest[20];
+	long mode = INI_INT("http.etag_mode");
+	
+	switch (mode)
+	{
+		case HTTP_ETAG_SHA1:
+			PHP_SHA1Final(digest, ctx);
+			etag = http_etag_digest(digest, 20);
+			efree(ctx);
+		break;
+		
+		case HTTP_ETAG_MD5:
+#ifndef HTTP_HAVE_MHASH
+		default:
+#endif
+			PHP_MD5Final(digest, ctx);
+			etag = http_etag_digest(digest, 16);
+			efree(ctx);
+		break;
+		
+#ifdef HTTP_HAVE_MHASH
+		default:
+		{
+			unsigned char *mhash_digest = mhash_end_m(ctx, http_etag_alloc_mhash_digest);
+			etag = http_etag_digest(mhash_digest, mhash_get_block_size(mode));
+			efree(mhash_digest);
+		}
+		break;
+#endif
+	}
+	
+	return etag;
+}
+
+#define http_etag_update(c, d, l) _http_etag_update((c), (d), (l) TSRMLS_CC)
+static inline void _http_etag_update(void *ctx, const char *data_ptr, size_t data_len TSRMLS_DC)
+{
+	switch (INI_INT("http.etag_mode"))
+	{
+		case HTTP_ETAG_SHA1:
+			PHP_SHA1Update(ctx, (const unsigned char *) data_ptr, data_len);
+		break;
+		
+		case HTTP_ETAG_MD5:
+#ifndef HTTP_HAVE_MHASH
+		default:
+#endif
+			PHP_MD5Update(ctx, (const unsigned char *) data_ptr, data_len);
+		break;
+		
+#ifdef HTTP_HAVE_MHASH
+		default:
+			mhash(ctx, data_ptr, data_len);
+		break;
+#endif
+	}
+}
+
 #define http_etag(p, l, m) _http_etag((p), (l), (m) TSRMLS_CC)
 PHP_HTTP_API char *_http_etag(const void *data_ptr, size_t data_len, http_send_mode data_mode TSRMLS_DC);
 
@@ -41,9 +180,6 @@ PHP_HTTP_API STATUS _http_cache_last_modified(time_t last_modified, time_t send_
 #define http_cache_etag(e, el, cc, ccl) _http_cache_etag((e), (el), (cc), (ccl) TSRMLS_CC)
 PHP_HTTP_API STATUS _http_cache_etag(const char *etag, size_t etag_len, const char *cache_control, size_t cc_len TSRMLS_DC);
 
-#define http_cache_exit() _http_cache_exit(TSRMLS_C)
-PHP_HTTP_API STATUS _http_cache_exit(TSRMLS_D);
-
 #define http_ob_etaghandler(o, l, ho, hl, m) _http_ob_etaghandler((o), (l), (ho), (hl), (m) TSRMLS_CC)
 PHP_HTTP_API void _http_ob_etaghandler(char *output, uint output_len, char **handled_output, uint *handled_output_len, int mode TSRMLS_DC);