Mercurial > hg > nginx
annotate src/http/modules/ngx_http_addition_filter_module.c @ 777:4ab852b691f5
<!--#include virtual=... set=... -->
| author | Igor Sysoev <igor@sysoev.ru> |
|---|---|
| date | Fri, 13 Oct 2006 15:20:10 +0000 |
| parents | 86bb73dc8d40 |
| children | c2cae54f2045 |
| rev | line source |
|---|---|
| 629 | 1 |
| 2 /* | |
| 3 * Copyright (C) Igor Sysoev | |
| 4 */ | |
| 5 | |
| 6 | |
| 7 #include <ngx_config.h> | |
| 8 #include <ngx_core.h> | |
| 9 #include <ngx_http.h> | |
| 10 | |
| 11 | |
| 12 typedef struct { | |
| 13 ngx_str_t before_body; | |
| 14 ngx_str_t after_body; | |
| 15 } ngx_http_addition_conf_t; | |
| 16 | |
| 17 | |
| 18 typedef struct { | |
| 631 | 19 ngx_uint_t before_body_sent; |
| 629 | 20 } ngx_http_addition_ctx_t; |
| 21 | |
| 22 | |
| 23 static void *ngx_http_addition_create_conf(ngx_conf_t *cf); | |
| 24 static char *ngx_http_addition_merge_conf(ngx_conf_t *cf, void *parent, | |
| 25 void *child); | |
| 681 | 26 static ngx_int_t ngx_http_addition_filter_init(ngx_conf_t *cf); |
| 629 | 27 |
| 28 | |
| 29 static ngx_command_t ngx_http_addition_commands[] = { | |
| 30 | |
| 31 { ngx_string("add_before_body"), | |
| 32 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1, | |
| 33 ngx_conf_set_str_slot, | |
| 34 NGX_HTTP_LOC_CONF_OFFSET, | |
| 35 offsetof(ngx_http_addition_conf_t, before_body), | |
| 36 NULL }, | |
| 37 | |
| 38 { ngx_string("add_after_body"), | |
| 39 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1, | |
| 40 ngx_conf_set_str_slot, | |
| 41 NGX_HTTP_LOC_CONF_OFFSET, | |
| 42 offsetof(ngx_http_addition_conf_t, after_body), | |
| 43 NULL }, | |
| 44 | |
| 45 ngx_null_command | |
| 46 }; | |
| 47 | |
| 48 | |
| 49 static ngx_http_module_t ngx_http_addition_filter_module_ctx = { | |
| 50 NULL, /* preconfiguration */ | |
| 681 | 51 ngx_http_addition_filter_init, /* postconfiguration */ |
| 629 | 52 |
| 53 NULL, /* create main configuration */ | |
| 54 NULL, /* init main configuration */ | |
| 55 | |
| 56 NULL, /* create server configuration */ | |
| 57 NULL, /* merge server configuration */ | |
| 58 | |
| 59 ngx_http_addition_create_conf, /* create location configuration */ | |
| 60 ngx_http_addition_merge_conf /* merge location configuration */ | |
| 61 }; | |
| 62 | |
| 63 | |
| 64 ngx_module_t ngx_http_addition_filter_module = { | |
| 65 NGX_MODULE_V1, | |
| 66 &ngx_http_addition_filter_module_ctx, /* module context */ | |
| 67 ngx_http_addition_commands, /* module directives */ | |
| 68 NGX_HTTP_MODULE, /* module type */ | |
| 69 NULL, /* init master */ | |
| 681 | 70 NULL, /* init module */ |
| 629 | 71 NULL, /* init process */ |
| 72 NULL, /* init thread */ | |
| 73 NULL, /* exit thread */ | |
| 74 NULL, /* exit process */ | |
| 75 NULL, /* exit master */ | |
| 76 NGX_MODULE_V1_PADDING | |
| 77 }; | |
| 78 | |
| 79 | |
| 80 static ngx_http_output_header_filter_pt ngx_http_next_header_filter; | |
| 81 static ngx_http_output_body_filter_pt ngx_http_next_body_filter; | |
| 82 | |
| 83 | |
| 84 static ngx_int_t | |
| 85 ngx_http_addition_header_filter(ngx_http_request_t *r) | |
| 86 { | |
| 87 ngx_http_addition_ctx_t *ctx; | |
| 88 ngx_http_addition_conf_t *conf; | |
| 89 | |
| 90 if (r->headers_out.status != NGX_HTTP_OK || r != r->main) { | |
| 91 return ngx_http_next_header_filter(r); | |
| 92 } | |
| 93 | |
| 631 | 94 conf = ngx_http_get_module_loc_conf(r, ngx_http_addition_filter_module); |
| 95 | |
| 96 if (conf->before_body.len == 0 && conf->after_body.len == 0) { | |
| 97 return ngx_http_next_header_filter(r); | |
| 98 } | |
| 99 | |
| 629 | 100 if (ngx_strncasecmp(r->headers_out.content_type.data, "text/html", |
| 101 sizeof("text/html") - 1) | |
| 102 != 0) | |
| 103 { | |
| 104 return ngx_http_next_header_filter(r); | |
| 105 } | |
| 106 | |
| 107 ctx = ngx_pcalloc(r->pool, sizeof(ngx_http_addition_ctx_t)); | |
| 108 if (ctx == NULL) { | |
| 109 return NGX_ERROR; | |
| 110 } | |
| 111 | |
| 112 ngx_http_set_ctx(r, ctx, ngx_http_addition_filter_module); | |
| 113 | |
| 114 ngx_http_clear_content_length(r); | |
| 115 ngx_http_clear_accept_ranges(r); | |
| 116 | |
| 117 return ngx_http_next_header_filter(r); | |
| 118 } | |
| 119 | |
| 120 | |
| 121 static ngx_int_t | |
| 122 ngx_http_addition_body_filter(ngx_http_request_t *r, ngx_chain_t *in) | |
| 123 { | |
| 124 ngx_int_t rc; | |
| 125 ngx_uint_t last; | |
| 126 ngx_chain_t *cl; | |
|
758
86bb73dc8d40
fix <!--#include virtual=... wait="yes" -->
Igor Sysoev <igor@sysoev.ru>
parents:
681
diff
changeset
|
127 ngx_http_request_t *sr; |
| 629 | 128 ngx_http_addition_ctx_t *ctx; |
| 129 ngx_http_addition_conf_t *conf; | |
| 130 | |
| 131 if (in == NULL || r->header_only) { | |
| 132 return ngx_http_next_body_filter(r, in); | |
| 133 } | |
| 134 | |
| 135 ctx = ngx_http_get_module_ctx(r, ngx_http_addition_filter_module); | |
| 136 | |
| 137 if (ctx == NULL) { | |
| 138 return ngx_http_next_body_filter(r, in); | |
| 139 } | |
| 140 | |
| 141 conf = ngx_http_get_module_loc_conf(r, ngx_http_addition_filter_module); | |
| 142 | |
| 143 if (!ctx->before_body_sent) { | |
| 144 ctx->before_body_sent = 1; | |
| 145 | |
| 146 if (conf->before_body.len) { | |
|
777
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
147 rc = ngx_http_subrequest(r, &conf->before_body, NULL, &sr, NULL, 0); |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
148 |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
149 if (rc == NGX_ERROR || rc == NGX_DONE) { |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
150 return rc; |
| 629 | 151 } |
| 152 } | |
| 153 } | |
| 154 | |
| 155 last = 0; | |
| 156 | |
| 157 for (cl = in; cl; cl = cl->next) { | |
| 158 if (cl->buf->last_buf) { | |
| 159 cl->buf->last_buf = 0; | |
| 631 | 160 cl->buf->sync = 1; |
| 629 | 161 last = 1; |
| 162 } | |
| 163 } | |
| 164 | |
| 165 rc = ngx_http_next_body_filter(r, in); | |
| 166 | |
| 631 | 167 if (rc == NGX_ERROR || !last || conf->after_body.len == 0) { |
| 629 | 168 return rc; |
| 169 } | |
| 170 | |
|
777
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
171 rc = ngx_http_subrequest(r, &conf->after_body, NULL, &sr, NULL, 0); |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
172 |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
173 if (rc == NGX_ERROR || rc == NGX_DONE) { |
|
4ab852b691f5
<!--#include virtual=... set=... -->
Igor Sysoev <igor@sysoev.ru>
parents:
758
diff
changeset
|
174 return rc; |
| 629 | 175 } |
| 176 | |
| 631 | 177 ngx_http_set_ctx(r, NULL, ngx_http_addition_filter_module); |
| 629 | 178 |
| 179 return ngx_http_send_special(r, NGX_HTTP_LAST); | |
| 180 } | |
| 181 | |
| 182 | |
| 183 static ngx_int_t | |
| 681 | 184 ngx_http_addition_filter_init(ngx_conf_t *cf) |
| 629 | 185 { |
| 186 ngx_http_next_header_filter = ngx_http_top_header_filter; | |
| 187 ngx_http_top_header_filter = ngx_http_addition_header_filter; | |
| 188 | |
| 189 ngx_http_next_body_filter = ngx_http_top_body_filter; | |
| 190 ngx_http_top_body_filter = ngx_http_addition_body_filter; | |
| 191 | |
| 192 return NGX_OK; | |
| 193 } | |
| 194 | |
| 195 | |
| 196 static void * | |
| 197 ngx_http_addition_create_conf(ngx_conf_t *cf) | |
| 198 { | |
| 199 ngx_http_addition_conf_t *conf; | |
| 200 | |
| 201 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_addition_conf_t)); | |
| 202 if (conf == NULL) { | |
| 203 return NGX_CONF_ERROR; | |
| 204 } | |
| 205 | |
| 206 /* | |
| 207 * set by ngx_pcalloc(): | |
| 208 * | |
| 209 * conf->before_body.len = 0; | |
| 210 * conf->before_body.date = NULL; | |
| 211 * conf->after_body.len = 0; | |
| 212 * conf->after_body.date = NULL; | |
| 213 */ | |
| 214 | |
| 215 return conf; | |
| 216 } | |
| 217 | |
| 218 | |
| 219 static char * | |
| 220 ngx_http_addition_merge_conf(ngx_conf_t *cf, void *parent, void *child) | |
| 221 { | |
| 222 ngx_http_addition_conf_t *prev = parent; | |
| 223 ngx_http_addition_conf_t *conf = child; | |
| 224 | |
| 225 ngx_conf_merge_str_value(conf->before_body, prev->before_body, ""); | |
| 226 ngx_conf_merge_str_value(conf->after_body, prev->after_body, ""); | |
| 227 | |
| 228 return NGX_CONF_OK; | |
| 229 } |