One Level Up Top Level

src/http/modules/ngx_http_try_files_module.c - nginx-1.31.4 nginx/ @ 8d9666701

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_array_t           *lengths;
  15.     ngx_array_t           *values;
  16.     ngx_str_t              name;

  18.     unsigned               code:10;
  19.     unsigned               test_dir:1;
  20. ‌} ngx_http_try_file_t;


  23. ‌typedef struct {
  24.     ngx_http_try_file_t   *try_files;
  25. ‌} ngx_http_try_files_loc_conf_t;


  28. static ngx_int_t ngx_http_try_files_handler(ngx_http_request_t *r);
  29. static char *ngx_http_try_files(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
  30. static void *ngx_http_try_files_create_loc_conf(ngx_conf_t *cf);
  31. static ngx_int_t ngx_http_try_files_init(ngx_conf_t *cf);


  34. ‌static ngx_command_t  ngx_http_try_files_commands[] = {

  36.     { ngx_string("try_files"),
  37.       NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_2MORE,
  38.       ngx_http_try_files,
  39.       NGX_HTTP_LOC_CONF_OFFSET,
  40.       0,
  41.       NULL },

  43.       ngx_null_command
  44. };


  47. ‌static ngx_http_module_t  ngx_http_try_files_module_ctx = {
  48.     NULL,                                  /* preconfiguration */
  49.     ngx_http_try_files_init,               /* postconfiguration */

  51.     NULL,                                  /* create main configuration */
  52.     NULL,                                  /* init main configuration */

  54.     NULL,                                  /* create server configuration */
  55.     NULL,                                  /* merge server configuration */

  57.     ngx_http_try_files_create_loc_conf,    /* create location configuration */
  58.     NULL                                   /* merge location configuration */
  59. };


  62. ‌ngx_module_t  ngx_http_try_files_module = {
  63.     NGX_MODULE_V1,
  64.     &ngx_http_try_files_module_ctx,        /* module context */
  65.     ngx_http_try_files_commands,           /* module directives */
  66.     NGX_HTTP_MODULE,                       /* module type */
  67.     NULL,                                  /* init master */
  68.     NULL,                                  /* init module */
  69.     NULL,                                  /* init process */
  70.     NULL,                                  /* init thread */
  71.     NULL,                                  /* exit thread */
  72.     NULL,                                  /* exit process */
  73.     NULL,                                  /* exit master */
  74.     NGX_MODULE_V1_PADDING
  75. };


  78. static ngx_int_t
  79. ‌ngx_http_try_files_handler(ngx_http_request_t *r)
  80. {
  81.     size_t                          len, root, alias, reserve, allocated, n;
  82.     u_char                         *p, *name;
  83.     ngx_str_t                       path, args;
  84.     ngx_uint_t                      test_dir;
  85.     ngx_http_try_file_t            *tf;
  86.     ngx_open_file_info_t            of;
  87.     ngx_http_script_code_pt         code;
  88.     ngx_http_script_engine_t        e;
  89.     ngx_http_core_loc_conf_t       *clcf;
  90.     ngx_http_script_len_code_pt     lcode;
  91.     ngx_http_try_files_loc_conf_t  *tlcf;

  93.     tlcf = ngx_http_get_module_loc_conf(r, ngx_http_try_files_module);

  95.     if (tlcf->try_files == NULL) {
  96.         return NGX_DECLINED;
  97.     }

  99.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
  100.                    "try files handler");

  102.     allocated = 0;
  103.     root = 0;
  104.     name = NULL;
  105.     /* suppress MSVC warning */
  106.     path.data = NULL;

  108.     tf = tlcf->try_files;

  110.     clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);

  112.     alias = clcf->alias;

  114.     for ( ;; ) {

  116.         if (tf->lengths) {
  117.             ngx_memzero(&e, sizeof(ngx_http_script_engine_t));

  119.             e.ip = tf->lengths->elts;
  120.             e.request = r;

  122.             /* 1 is for terminating '\0' as in static names */
  123.             len = 1;

  125.             while (*(uintptr_t *) e.ip) {
  126.                 lcode = *(ngx_http_script_len_code_pt *) e.ip;
  127.                 len += lcode(&e);
  128.             }

  130.         } else {
  131.             len = tf->name.len;
  132.         }

  134.         if (!alias) {
  135.             reserve = len > r->uri.len ? len - r->uri.len : 0;

  137.         } else if (alias == NGX_MAX_SIZE_T_VALUE) {
  138.             reserve = len;

  140.         } else {
  141.             reserve = len > r->uri.len - alias ? len - (r->uri.len - alias) : 0;
  142.         }

  144.         if (reserve > allocated || !allocated) {

  146.             /* 16 bytes are preallocation */
  147.             allocated = reserve + 16;

  149.             if (ngx_http_map_uri_to_path(r, &path, &root, allocated) == NULL) {
  150.                 return NGX_HTTP_INTERNAL_SERVER_ERROR;
  151.             }

  153.             name = path.data + root;
  154.         }

  156.         if (tf->values == NULL) {

  158.             /* tf->name.len includes the terminating '\0' */

  160.             ngx_memcpy(name, tf->name.data, tf->name.len);

  162.             path.len = (name + tf->name.len - 1) - path.data;

  164.         } else {
  165.             n = allocated;

  167.             if (alias != NGX_MAX_SIZE_T_VALUE) {
  168.                 n += (r->uri.len - alias);
  169.             }

  171.             e.ip = tf->values->elts;
  172.             e.pos = name;
  173.             e.end = name + n;
  174.             e.flushed = 1;

  176.             while (*(uintptr_t *) e.ip) {
  177.                 code = *(ngx_http_script_code_pt *) e.ip;
  178.                 code((ngx_http_script_engine_t *) &e);
  179.             }

  181.             if (e.status) {
  182.                 return NGX_HTTP_INTERNAL_SERVER_ERROR;
  183.             }

  185.             if (ngx_http_script_check_length(&e, 1) != NGX_OK) {
  186.                 return NGX_ERROR;
  187.             }

  189.             path.len = e.pos - path.data;

  191.             *e.pos = '\0';

  193.             if (alias && alias != NGX_MAX_SIZE_T_VALUE
  194.                 && ngx_strncmp(name, r->uri.data, alias) == 0)
  195.             {
  196.                 ngx_memmove(name, name + alias, len - alias);
  197.                 path.len -= alias;
  198.             }
  199.         }

  201.         test_dir = tf->test_dir;

  203.         tf++;

  205.         ngx_log_debug3(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
  206.                        "trying to use %s: \"%s\" \"%s\"",
  207.                        test_dir ? "dir" : "file", name, path.data);

  209.         if (tf->lengths == NULL && tf->name.len == 0) {

  211.             if (tf->code) {
  212.                 return tf->code;
  213.             }

  215.             path.len -= root;
  216.             path.data += root;

  218.             if (path.data[0] == '@') {
  219.                 (void) ngx_http_named_location(r, &path);

  221.             } else {
  222.                 ngx_http_split_args(r, &path, &args);

  224.                 (void) ngx_http_internal_redirect(r, &path, &args);
  225.             }

  227.             ngx_http_finalize_request(r, NGX_DONE);
  228.             return NGX_DONE;
  229.         }

  231.         ngx_memzero(&of, sizeof(ngx_open_file_info_t));

  233.         of.read_ahead = clcf->read_ahead;
  234.         of.directio = clcf->directio;
  235.         of.valid = clcf->open_file_cache_valid;
  236.         of.min_uses = clcf->open_file_cache_min_uses;
  237.         of.test_only = 1;
  238.         of.errors = clcf->open_file_cache_errors;
  239.         of.events = clcf->open_file_cache_events;

  241.         if (ngx_http_set_disable_symlinks(r, clcf, &path, &of) != NGX_OK) {
  242.             return NGX_HTTP_INTERNAL_SERVER_ERROR;
  243.         }

  245.         if (ngx_open_cached_file(clcf->open_file_cache, &path, &of, r->pool)
  246.             != NGX_OK)
  247.         {
  248.             if (of.err == 0) {
  249.                 return NGX_HTTP_INTERNAL_SERVER_ERROR;
  250.             }

  252.             if (of.err != NGX_ENOENT
  253.                 && of.err != NGX_ENOTDIR
  254.                 && of.err != NGX_ENAMETOOLONG)
  255.             {
  256.                 ngx_log_error(NGX_LOG_CRIT, r->connection->log, of.err,
  257.                               "%s \"%s\" failed", of.failed, path.data);
  258.             }

  260.             continue;
  261.         }

  263.         if (of.is_dir != test_dir) {
  264.             continue;
  265.         }

  267.         path.len -= root;
  268.         path.data += root;

  270.         if (!alias) {
  271.             r->uri = path;

  273.         } else if (alias == NGX_MAX_SIZE_T_VALUE) {
  274.             if (!test_dir) {
  275.                 r->uri = path;
  276.                 r->add_uri_to_alias = 1;
  277.             }

  279.         } else {
  280.             name = r->uri.data;

  282.             r->uri.len = alias + path.len;
  283.             r->uri.data = ngx_pnalloc(r->pool, r->uri.len);
  284.             if (r->uri.data == NULL) {
  285.                 r->uri.len = 0;
  286.                 return NGX_HTTP_INTERNAL_SERVER_ERROR;
  287.             }

  289.             p = ngx_copy(r->uri.data, name, alias);
  290.             ngx_memcpy(p, path.data, path.len);
  291.         }

  293.         ngx_http_set_exten(r);

  295.         ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
  296.                        "try file uri: \"%V\"", &r->uri);

  298.         return NGX_DECLINED;
  299.     }

  301.     /* not reached */
  302. }


  305. static char *
  306. ‌ngx_http_try_files(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
  307. {
  308.     ngx_http_try_files_loc_conf_t *tlcf = conf;

  310.     ngx_str_t                  *value;
  311.     ngx_int_t                   code;
  312.     ngx_uint_t                  i, n;
  313.     ngx_http_try_file_t        *tf;
  314.     ngx_http_script_compile_t   sc;

  316.     if (tlcf->try_files) {
  317.         return "is duplicate";
  318.     }

  320.     tf = ngx_pcalloc(cf->pool, cf->args->nelts * sizeof(ngx_http_try_file_t));
  321.     if (tf == NULL) {
  322.         return NGX_CONF_ERROR;
  323.     }

  325.     tlcf->try_files = tf;

  327.     value = cf->args->elts;

  329.     for (i = 0; i < cf->args->nelts - 1; i++) {

  331.         tf[i].name = value[i + 1];

  333.         if (tf[i].name.len > 0
  334.             && tf[i].name.data[tf[i].name.len - 1] == '/'
  335.             && i + 2 < cf->args->nelts)
  336.         {
  337.             tf[i].test_dir = 1;
  338.             tf[i].name.len--;
  339.             tf[i].name.data[tf[i].name.len] = '\0';
  340.         }

  342.         n = ngx_http_script_variables_count(&tf[i].name);

  344.         if (n) {
  345.             ngx_memzero(&sc, sizeof(ngx_http_script_compile_t));

  347.             sc.cf = cf;
  348.             sc.source = &tf[i].name;
  349.             sc.lengths = &tf[i].lengths;
  350.             sc.values = &tf[i].values;
  351.             sc.variables = n;
  352.             sc.complete_lengths = 1;
  353.             sc.complete_values = 1;

  355.             if (ngx_http_script_compile(&sc) != NGX_OK) {
  356.                 return NGX_CONF_ERROR;
  357.             }

  359.         } else {
  360.             /* add trailing '\0' to length */
  361.             tf[i].name.len++;
  362.         }
  363.     }

  365.     if (tf[i - 1].name.data[0] == '=') {

  367.         code = ngx_atoi(tf[i - 1].name.data + 1, tf[i - 1].name.len - 2);

  369.         if (code == NGX_ERROR || code > 999) {
  370.             ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
  371.                                "invalid code \"%*s\"",
  372.                                tf[i - 1].name.len - 1, tf[i - 1].name.data);
  373.             return NGX_CONF_ERROR;
  374.         }

  376.         tf[i].code = code;
  377.     }

  379.     return NGX_CONF_OK;
  380. }


  383. static void *
  384. ‌ngx_http_try_files_create_loc_conf(ngx_conf_t *cf)
  385. {
  386.     ngx_http_try_files_loc_conf_t  *tlcf;

  388.     tlcf = ngx_pcalloc(cf->pool, sizeof(ngx_http_try_files_loc_conf_t));
  389.     if (tlcf == NULL) {
  390.         return NULL;
  391.     }

  393.     /*
  394.      * set by ngx_pcalloc():
  395.      *
  396.      *     tlcf->try_files = NULL;
  397.      */

  399.     return tlcf;
  400. }


  403. static ngx_int_t
  404. ‌ngx_http_try_files_init(ngx_conf_t *cf)
  405. {
  406.     ngx_http_handler_pt        *h;
  407.     ngx_http_core_main_conf_t  *cmcf;

  409.     cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);

  411.     h = ngx_array_push(&cmcf->phases[NGX_HTTP_PRECONTENT_PHASE].handlers);
  412.     if (h == NULL) {
  413.         return NGX_ERROR;
  414.     }

  416.     *h = ngx_http_try_files_handler;

  418.     return NGX_OK;
  419. }

One Level Up Top Level