- added INI setting http.force_exit which can be used to disable script termination
[m6w6/ext-http] / http_cache_api.c
index 5eca48a6650576054b7b8a844e1da6a89b8c7bf2..322c509d2f5a8873a526a1be104f6b7fc8a9768f 100644 (file)
@@ -1,16 +1,13 @@
 /*
-   +----------------------------------------------------------------------+
-   | PECL :: http                                                         |
-   +----------------------------------------------------------------------+
-   | This source file is subject to version 3.0 of the PHP license, that  |
-   | is bundled with this package in the file LICENSE, and is available   |
-   | through the world-wide-web at http://www.php.net/license/3_0.txt.    |
-   | If you did not receive a copy of the PHP license and are unable to   |
-   | obtain it through the world-wide-web, please send a note to          |
-   | license@php.net so we can mail you a copy immediately.               |
-   +----------------------------------------------------------------------+
-   | Copyright (c) 2004-2005 Michael Wallner <mike@php.net>               |
-   +----------------------------------------------------------------------+
+    +--------------------------------------------------------------------+
+    | PECL :: http                                                       |
+    +--------------------------------------------------------------------+
+    | Redistribution and use in source and binary forms, with or without |
+    | modification, are permitted provided that the conditions mentioned |
+    | in the accompanying LICENSE file are met.                          |
+    +--------------------------------------------------------------------+
+    | Copyright (c) 2004-2005, Michael Wallner <mike@php.net>            |
+    +--------------------------------------------------------------------+
 */
 
 /* $Id$ */
@@ -39,7 +36,7 @@
 
 ZEND_EXTERN_MODULE_GLOBALS(http);
 
-STATUS _http_cache_global_init(INIT_FUNC_ARGS)
+PHP_MINIT_FUNCTION(http_cache)
 {
        HTTP_LONG_CONSTANT("HTTP_ETAG_MD5", HTTP_ETAG_MD5);
        HTTP_LONG_CONSTANT("HTTP_ETAG_SHA1", HTTP_ETAG_SHA1);
@@ -67,41 +64,35 @@ STATUS _http_cache_global_init(INIT_FUNC_ARGS)
 /* {{{ char *http_etag(void *, size_t, http_send_mode) */
 PHP_HTTP_API char *_http_etag(const void *data_ptr, size_t data_len, http_send_mode data_mode TSRMLS_DC)
 {
-       php_stream_statbuf ssb;
-       char ssb_buf[128] = {0};
-       size_t ssb_len;
        void *ctx = http_etag_init();
        
-       switch (data_mode)
-       {
-               case SEND_DATA:
-                       http_etag_update(ctx, data_ptr, data_len);
-               break;
-
-               case SEND_RSRC:
-               {
-                       if (php_stream_stat((php_stream *) data_ptr, &ssb)) {
-                               efree(ctx);
-                               return NULL;
-                       }
-                       ssb_len = snprintf(ssb_buf, 127, "%ld=%ld=%ld", ssb.sb.st_mtime, ssb.sb.st_ino, ssb.sb.st_size);
-                       http_etag_update(ctx, ssb_buf, ssb_len);
+       if (data_mode == SEND_DATA) {
+               http_etag_update(ctx, data_ptr, data_len);
+       } else {
+               STATUS ss = FAILURE;
+               php_stream_statbuf ssb;
+               
+               if (data_mode == SEND_RSRC) {
+                       ss = php_stream_stat((php_stream *) data_ptr, &ssb);
+               } else {
+                       ss = php_stream_stat_path((char *) data_ptr, &ssb);
                }
-               break;
-
-               default:
-               {
-                       if (php_stream_stat_path((char *) data_ptr, &ssb)) {
-                               efree(ctx);
-                               return NULL;
-                       }
-                       ssb_len = snprintf(ssb_buf, 127, "%ld=%ld=%ld", ssb.sb.st_mtime, ssb.sb.st_ino, ssb.sb.st_size);
+               
+               if (SUCCESS != ss) {
+                       http_etag_free(&ctx);
+                       return NULL;
+               } else {
+                       size_t ssb_len;
+                       char ssb_buf[128] = {0};
+                       
+                       ssb_len = snprintf(ssb_buf, 127, "%ld=%ld=%ld", (long) ssb.sb.st_mtime, 
+                                                                                                                       (long) ssb.sb.st_ino, 
+                                                                                                                       (long) ssb.sb.st_size);
                        http_etag_update(ctx, ssb_buf, ssb_len);
                }
-               break;
        }
-
-       return http_etag_finish(ctx);
+       
+       return http_etag_finish(&ctx);
 }
 /* }}} */
 
@@ -151,15 +142,14 @@ PHP_HTTP_API zend_bool _http_match_etag_ex(const char *entry, const char *etag,
                return 1;
        }
 
-       quoted_etag = (char *) emalloc(strlen(etag) + 3);
-       sprintf(quoted_etag, "\"%s\"", etag);
-
+       spprintf(&quoted_etag, 0, "\"%s\"", etag);
        if (!strchr(Z_STRVAL_P(zetag), ',')) {
                result = !strcmp(Z_STRVAL_P(zetag), quoted_etag);
        } else {
                result = (NULL != strstr(Z_STRVAL_P(zetag), quoted_etag));
        }
        efree(quoted_etag);
+       
        return result;
 }
 /* }}} */
@@ -170,6 +160,10 @@ PHP_HTTP_API STATUS _http_cache_last_modified(time_t last_modified,
 {
        char *sent_header = NULL;
        
+       if (SG(headers_sent)) {
+               return FAILURE;
+       }
+       
        if (cc_len && (SUCCESS != http_send_cache_control(cache_control, cc_len))) {
                return FAILURE;
        }
@@ -194,6 +188,10 @@ PHP_HTTP_API STATUS _http_cache_etag(const char *etag, size_t etag_len,
 {
        char *sent_header = NULL;
        
+       if (SG(headers_sent)) {
+               return FAILURE;
+       }
+       
        if (cc_len && (SUCCESS != http_send_cache_control(cache_control, cc_len))) {
                return FAILURE;
        }
@@ -209,57 +207,68 @@ PHP_HTTP_API STATUS _http_cache_etag(const char *etag, size_t etag_len,
                }
                return SUCCESS;
        }
+       
+       /* start ob_etaghandler */
+       return http_start_ob_etaghandler();
+}
+/* }}} */
 
-       /* if no etag is given and we didn't already start ob_etaghandler -- start it */
-       if (HTTP_G(etag).started) {
-               return SUCCESS;
+PHP_HTTP_API STATUS _http_start_ob_etaghandler(TSRMLS_D)
+{
+       /* 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;
        }
+       
+       HTTP_G(etag).started = 1;
+       return php_start_ob_buffer_named("ob_etaghandler", HTTP_G(send).buffer_size, 1 TSRMLS_CC);
+}
 
-       if (HTTP_G(etag).started = (SUCCESS == php_start_ob_buffer_named("ob_etaghandler", HTTP_G(send).buffer_size, 1 TSRMLS_CC))) {
-               return SUCCESS;
-       } else {
-               return FAILURE;
+PHP_HTTP_API zend_bool _http_interrupt_ob_etaghandler(TSRMLS_D)
+{
+       if (HTTP_G(etag).started) {
+               HTTP_G(etag).started = 0;
+               http_etag_free(&HTTP_G(etag).ctx);
+               return 1;
        }
+       return 0;
 }
-/* }}} */
 
 /* {{{ void http_ob_etaghandler(char *, uint, char **, uint *, int) */
-PHP_HTTP_API void _http_ob_etaghandler(char *output, uint output_len,
+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);
 }
 /* }}} */