One Level Up Top Level

src/http/modules/ngx_http_addition_filter_module.c - nginx-1.7.10

Global variables defined

Data types defined

Functions defined

Source code


  2. /*
  3. * Copyright (C) Igor Sysoev
  4. * Copyright (C) Nginx, Inc.
  5. */


  8. #include <ngx_config.h>
  9. #include <ngx_core.h>
  10. #include <ngx_http.h>


  13. typedef struct {
  14.     ngx_str_t     before_body;
  15.     ngx_str_t     after_body;

  17.     ngx_hash_t    types;
  18.     ngx_array_t  *types_keys;
  19. ‌} ngx_http_addition_conf_t;


  22. typedef struct {
  23.     ngx_uint_t    before_body_sent;
  24. ‌} ngx_http_addition_ctx_t;


  27. static void *ngx_http_addition_create_conf(ngx_conf_t *cf);
  28. static char *ngx_http_addition_merge_conf(ngx_conf_t *cf, void *parent,
  29.     void *child);
  30. static ngx_int_t ngx_http_addition_filter_init(ngx_conf_t *cf);


  33. ‌static ngx_command_t  ngx_http_addition_commands[] = {

  35.     { ngx_string("add_before_body"),
  36.       NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
  37.       ngx_conf_set_str_slot,
  38.       NGX_HTTP_LOC_CONF_OFFSET,
  39.       offsetof(ngx_http_addition_conf_t, before_body),
  40.       NULL },

  42.     { ngx_string("add_after_body"),
  43.       NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
  44.       ngx_conf_set_str_slot,
  45.       NGX_HTTP_LOC_CONF_OFFSET,
  46.       offsetof(ngx_http_addition_conf_t, after_body),
  47.       NULL },

  49.     { ngx_string("addition_types"),
  50.       NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_1MORE,
  51.       ngx_http_types_slot,
  52.       NGX_HTTP_LOC_CONF_OFFSET,
  53.       offsetof(ngx_http_addition_conf_t, types_keys),
  54.       &ngx_http_html_default_types[0] },

  56.       ngx_null_command
  57. };


  60. ‌static ngx_http_module_t  ngx_http_addition_filter_module_ctx = {
  61.     NULL,                                  /* preconfiguration */
  62.     ngx_http_addition_filter_init,         /* postconfiguration */

  64.     NULL,                                  /* create main configuration */
  65.     NULL,                                  /* init main configuration */

  67.     NULL,                                  /* create server configuration */
  68.     NULL,                                  /* merge server configuration */

  70.     ngx_http_addition_create_conf,         /* create location configuration */
  71.     ngx_http_addition_merge_conf           /* merge location configuration */
  72. };


  75. ‌ngx_module_t  ngx_http_addition_filter_module = {
  76.     NGX_MODULE_V1,
  77.     &ngx_http_addition_filter_module_ctx,  /* module context */
  78.     ngx_http_addition_commands,            /* module directives */
  79.     NGX_HTTP_MODULE,                       /* module type */
  80.     NULL,                                  /* init master */
  81.     NULL,                                  /* init module */
  82.     NULL,                                  /* init process */
  83.     NULL,                                  /* init thread */
  84.     NULL,                                  /* exit thread */
  85.     NULL,                                  /* exit process */
  86.     NULL,                                  /* exit master */
  87.     NGX_MODULE_V1_PADDING
  88. };


  91. ‌static ngx_http_output_header_filter_pt  ngx_http_next_header_filter;
  92. ‌static ngx_http_output_body_filter_pt    ngx_http_next_body_filter;


  95. static ngx_int_t
  96. ‌ngx_http_addition_header_filter(ngx_http_request_t *r)
  97. {
  98.     ngx_http_addition_ctx_t   *ctx;
  99.     ngx_http_addition_conf_t  *conf;

  101.     if (r->headers_out.status != NGX_HTTP_OK || r != r->main) {
  102.         return ngx_http_next_header_filter(r);
  103.     }

  105.     conf = ngx_http_get_module_loc_conf(r, ngx_http_addition_filter_module);

  107.     if (conf->before_body.len == 0 && conf->after_body.len == 0) {
  108.         return ngx_http_next_header_filter(r);
  109.     }

  111.     if (ngx_http_test_content_type(r, &conf->types) == NULL) {
  112.         return ngx_http_next_header_filter(r);
  113.     }

  115.     ctx = ngx_pcalloc(r->pool, sizeof(ngx_http_addition_ctx_t));
  116.     if (ctx == NULL) {
  117.         return NGX_ERROR;
  118.     }

  120.     ngx_http_set_ctx(r, ctx, ngx_http_addition_filter_module);

  122.     ngx_http_clear_content_length(r);
  123.     ngx_http_clear_accept_ranges(r);
  124.     ngx_http_weak_etag(r);

  126.     return ngx_http_next_header_filter(r);
  127. }


  130. static ngx_int_t
  131. ‌ngx_http_addition_body_filter(ngx_http_request_t *r, ngx_chain_t *in)
  132. {
  133.     ngx_int_t                  rc;
  134.     ngx_uint_t                 last;
  135.     ngx_chain_t               *cl;
  136.     ngx_http_request_t        *sr;
  137.     ngx_http_addition_ctx_t   *ctx;
  138.     ngx_http_addition_conf_t  *conf;

  140.     if (in == NULL || r->header_only) {
  141.         return ngx_http_next_body_filter(r, in);
  142.     }

  144.     ctx = ngx_http_get_module_ctx(r, ngx_http_addition_filter_module);

  146.     if (ctx == NULL) {
  147.         return ngx_http_next_body_filter(r, in);
  148.     }

  150.     conf = ngx_http_get_module_loc_conf(r, ngx_http_addition_filter_module);

  152.     if (!ctx->before_body_sent) {
  153.         ctx->before_body_sent = 1;

  155.         if (conf->before_body.len) {
  156.             if (ngx_http_subrequest(r, &conf->before_body, NULL, &sr, NULL, 0)
  157.                 != NGX_OK)
  158.             {
  159.                 return NGX_ERROR;
  160.             }
  161.         }
  162.     }

  164.     if (conf->after_body.len == 0) {
  165.         ngx_http_set_ctx(r, NULL, ngx_http_addition_filter_module);
  166.         return ngx_http_next_body_filter(r, in);
  167.     }

  169.     last = 0;

  171.     for (cl = in; cl; cl = cl->next) {
  172.         if (cl->buf->last_buf) {
  173.             cl->buf->last_buf = 0;
  174.             cl->buf->sync = 1;
  175.             last = 1;
  176.         }
  177.     }

  179.     rc = ngx_http_next_body_filter(r, in);

  181.     if (rc == NGX_ERROR || !last || conf->after_body.len == 0) {
  182.         return rc;
  183.     }

  185.     if (ngx_http_subrequest(r, &conf->after_body, NULL, &sr, NULL, 0)
  186.         != NGX_OK)
  187.     {
  188.         return NGX_ERROR;
  189.     }

  191.     ngx_http_set_ctx(r, NULL, ngx_http_addition_filter_module);

  193.     return ngx_http_send_special(r, NGX_HTTP_LAST);
  194. }


  197. static ngx_int_t
  198. ‌ngx_http_addition_filter_init(ngx_conf_t *cf)
  199. {
  200.     ngx_http_next_header_filter = ngx_http_top_header_filter;
  201.     ngx_http_top_header_filter = ngx_http_addition_header_filter;

  203.     ngx_http_next_body_filter = ngx_http_top_body_filter;
  204.     ngx_http_top_body_filter = ngx_http_addition_body_filter;

  206.     return NGX_OK;
  207. }


  210. static void *
  211. ‌ngx_http_addition_create_conf(ngx_conf_t *cf)
  212. {
  213.     ngx_http_addition_conf_t  *conf;

  215.     conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_addition_conf_t));
  216.     if (conf == NULL) {
  217.         return NULL;
  218.     }

  220.     /*
  221.      * set by ngx_pcalloc():
  222.      *
  223.      *     conf->before_body = { 0, NULL };
  224.      *     conf->after_body = { 0, NULL };
  225.      *     conf->types = { NULL };
  226.      *     conf->types_keys = NULL;
  227.      */

  229.     return conf;
  230. }


  233. static char *
  234. ‌ngx_http_addition_merge_conf(ngx_conf_t *cf, void *parent, void *child)
  235. {
  236.     ngx_http_addition_conf_t *prev = parent;
  237.     ngx_http_addition_conf_t *conf = child;

  239.     ngx_conf_merge_str_value(conf->before_body, prev->before_body, "");
  240.     ngx_conf_merge_str_value(conf->after_body, prev->after_body, "");

  242.     if (ngx_http_merge_types(cf, &conf->types_keys, &conf->types,
  243.                              &prev->types_keys, &prev->types,
  244.                              ngx_http_html_default_types)
  245.         != NGX_OK)
  246.     {
  247.         return NGX_CONF_ERROR;
  248.     }

  250.     return NGX_CONF_OK;
  251. }

One Level Up Top Level