相信你們在看了前一章的模塊概述之後,都對nginx的模塊有了一個基本的認識。基本上做爲第三方開發者最可能開發的就是三種類型的模塊,即handler,filter和load-balancer。Handler模塊就是接受來自客戶端的請求併產生輸出的模塊。有些地方說upstream模塊實際上也是一種handler模塊,只不過它產生的內容來自於從後端服務器獲取的,而非在本機產生的。html
在上一章提到,配置文件中使用location指令能夠配置content handler模塊,當Nginx系統啓動的時候,每一個handler模塊都有一次機會把本身關聯到對應的location上。若是有多個handler模塊都關聯了同一個location,那麼實際上只有一個handler模塊真正會起做用。固然大多數狀況下,模塊開發人員都會避免出現這種狀況。nginx
handler模塊處理的結果一般有三種狀況: 處理成功,處理失敗(處理的時候發生了錯誤)或者是拒絕去處理。在拒絕處理的狀況下,這個location的處理就會由默認的handler模塊來進行處理。例如,當請求一個靜態文件的時候,若是關聯到這個location上的一個handler模塊拒絕處理,就會由默認的ngx_http_static_module模塊進行處理,該模塊是一個典型的handler模塊。apache
本章主要講述的是如何編寫handler模塊,在研究handler模塊編寫以前先來了解一下模塊的一些基本數據結構。後端
在這一節咱們將會對一般的模塊開發過程當中,每一個模塊所包含的一些經常使用的部分進行說明。這些部分有些是必須的,有些不是必須的。同時這裏所列出的這些東西對於其餘類型的模塊,例如filter模塊等也都是相同的。數組
基本上每一個模塊都會提供一些配置指令,以便於用戶能夠經過配置來控制該模塊的行爲。那麼這些配置信息怎麼存儲呢?那就須要定義該模塊的配置結構來進行存儲。安全
你們都知道Nginx的配置信息分紅了幾個做用域(scope,有時也稱做上下文),這就是main, server, 以及location。一樣的每一個模塊提供的配置指令也能夠出如今這幾個做用域裏。那對於這三個做用域的配置信息,每一個模塊就須要定義三個不一樣的數據結構去進行存儲。固然,不是每一個模塊都會在這三個做用域都提供配置指令的。那麼也就不必定每一個模塊都須要定義三個數據結構去存儲這些配置信息了。視模塊的實現而言,須要幾個就定義幾個。服務器
有一點須要特別注意的就是,在模塊的開發過程當中,咱們最好使用nginx原有的命名習慣。這樣跟原代碼的契合度更高,看起來也更舒服。數據結構
對於模塊配置信息的定義,命名習慣是ngx_http_<module name>_(main|srv|loc)_conf_t。這裏有個例子,就是從咱們後面將要展現給你們的hello module中截取的。ide
typedef struct { ngx_str_t hello_string; ngx_int_t hello_counter; }ngx_http_hello_loc_conf_t;
一個模塊的配置指令是定義在一個靜態數組中的。一樣地,咱們來看一下從hello module中截取的模塊配置指令的定義。函數
static ngx_command_t ngx_http_hello_commands[] = { { ngx_string("hello_string"), NGX_HTTP_LOC_CONF|NGX_CONF_NOARGS|NGX_CONF_TAKE1, ngx_http_hello_string, NGX_HTTP_LOC_CONF_OFFSET, offsetof(ngx_http_hello_loc_conf_t, hello_string), NULL }, { ngx_string("hello_counter"), NGX_HTTP_LOC_CONF|NGX_CONF_FLAG, ngx_http_hello_counter, NGX_HTTP_LOC_CONF_OFFSET, offsetof(ngx_http_hello_loc_conf_t, hello_counter), NULL }, ngx_null_command };
其實看這個定義,就基本能看出來一些信息。例如,咱們是定義了兩個配置指令,一個是叫hello_string,能夠接受一個參數,或者是沒有參數。另一個命令是hello_counter,接受一個NGX_CONF_FLAG類型的參數。除此以外,彷佛看起來有點迷惑。沒有關係,咱們來詳細看一下ngx_command_t,一旦咱們瞭解這個結構的詳細信息,那麼我相信上述這個定義所表達的全部信息就不言自明瞭。
ngx_command_t的定義,位於src/core/ngx_conf_file.h中。
struct ngx_command_s { ngx_str_t name; ngx_uint_t type; char *(*set)(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); ngx_uint_t conf; ngx_uint_t offset; void *post; };
name:
配置指令的名稱。
type:
該配置的類型,其實更準確一點說,是該配置指令屬性的集合。nginx提供了不少預約義的屬性值(一些宏定義),經過邏輯或運算符可組合在一塊兒,造成對這個配置指令的詳細的說明。下面列出可在這裏使用的預約義屬性值及說明。
NGX_CONF_NOARGS:配置指令不接受任何參數。
NGX_CONF_TAKE1:配置指令接受1個參數。
NGX_CONF_TAKE2:配置指令接受2個參數。
NGX_CONF_TAKE3:配置指令接受3個參數。
NGX_CONF_TAKE4:配置指令接受4個參數。
NGX_CONF_TAKE5:配置指令接受5個參數。
NGX_CONF_TAKE6:配置指令接受6個參數。
NGX_CONF_TAKE7:配置指令接受7個參數。
能夠組合多個屬性,好比一個指令便可以不填參數,也能夠接受1個或者2個參數。那麼就是NGX_CONF_NOARGS|NGX_CONF_TAKE1|NGX_CONF_TAKE2。若是寫上面三個屬性在一塊兒,你以爲麻煩,那麼沒有關係,nginx提供了一些定義,使用起來更簡潔。
NGX_CONF_TAKE12:配置指令接受1個或者2個參數。
NGX_CONF_TAKE13:配置指令接受1個或者3個參數。
NGX_CONF_TAKE23:配置指令接受2個或者3個參數。
NGX_CONF_TAKE123:配置指令接受1個或者2個或者3參數。
NGX_CONF_TAKE1234:配置指令接受1個或者2個或者3個或者4個參數。
NGX_CONF_1MORE:配置指令接受至少一個參數。
NGX_CONF_2MORE:配置指令接受至少兩個參數。
NGX_CONF_MULTI: 配置指令能夠接受多個參數,即個數不定。
NGX_CONF_BLOCK:配置指令能夠接受的值是一個配置信息塊。也就是一對大括號括起來的內容。裏面能夠再包括不少的配置指令。好比常見的server指令就是這個屬性的。
NGX_CONF_FLAG:配置指令能夠接受的值是」on」或者」off」,最終會被轉成bool值。
NGX_CONF_ANY:配置指令能夠接受的任意的參數值。一個或者多個,或者」on」或者」off」,或者是配置塊。
最後要說明的是,不管如何,nginx的配置指令的參數個數不能夠超過NGX_CONF_MAX_ARGS個。目前這個值被定義爲8,也就是不能超過8個參數值。
下面介紹一組說明配置指令能夠出現的位置的屬性。
NGX_DIRECT_CONF:能夠出如今配置文件中最外層。例如已經提供的配置指令daemon,master_process等。
NGX_MAIN_CONF: http、mail、events、error_log等。
NGX_ANY_CONF: 該配置指令能夠出如今任意配置級別上。
對於咱們編寫的大多數模塊而言,都是在處理http相關的事情,也就是所謂的都是NGX_HTTP_MODULE,對於這樣類型的模塊,其配置可能出現的位置也是分爲直接出如今http裏面,以及其餘位置。
NGX_HTTP_MAIN_CONF: 能夠直接出如今http配置指令裏。
NGX_HTTP_SRV_CONF: 能夠出如今http裏面的server配置指令裏。
NGX_HTTP_LOC_CONF: 能夠出如今http server塊裏面的location配置指令裏。
NGX_HTTP_UPS_CONF: 能夠出如今http裏面的upstream配置指令裏。
NGX_HTTP_SIF_CONF: 能夠出如今http裏面的server配置指令裏的if語句所在的block中。
NGX_HTTP_LMT_CONF: 能夠出如今http裏面的limit_except指令的block中。
NGX_HTTP_LIF_CONF: 能夠出如今http server塊裏面的location配置指令裏的if語句所在的block中。
set:
這是一個函數指針,當nginx在解析配置的時候,若是遇到這個配置指令,將會把讀取到的值傳遞給這個函數進行分解處理。由於具體每一個配置指令的值如何處理,只有定義這個配置指令的人是最清楚的。來看一下這個函數指針要求的函數原型。
char *(*set)(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
先看該函數的返回值,處理成功時,返回NGX_OK,不然返回NGX_CONF_ERROR或者是一個自定義的錯誤信息的字符串。
再看一下這個函數被調用的時候,傳入的三個參數。
爲了更加方便的實現對配置指令參數的讀取,nginx已經默認提供了對一些標準類型的參數進行讀取的函數,能夠直接賦值給set字段使用。下面來看一下這些已經實現的set類型函數。
conf:
該字段被NGX_HTTP_MODULE類型模塊所用 (咱們編寫的基本上都是NGX_HTTP_MOUDLE,只有一些nginx核心模塊是非NGX_HTTP_MODULE),該字段指定當前配置項存儲的內存位置。其實是使用哪一個內存池的問題。由於http模塊對全部http模塊所要保存的配置信息,劃分了main, server和location三個地方進行存儲,每一個地方都有一個內存池用來分配存儲這些信息的內存。這裏可能的值爲 NGX_HTTP_MAIN_CONF_OFFSET、NGX_HTTP_SRV_CONF_OFFSET或NGX_HTTP_LOC_CONF_OFFSET。固然也能夠直接置爲0,就是NGX_HTTP_MAIN_CONF_OFFSET。
offset:
指定該配置項值的精確存放位置,通常指定爲某一個結構體變量的字段偏移。由於對於配置信息的存儲,通常咱們都是定義個結構體來存儲的。那麼好比咱們定義了一個結構體A,該項配置的值須要存儲到該結構體的b字段。那麼在這裏就能夠填寫爲offsetof(A, b)。對於有些配置項,它的值不須要保存或者是須要保存到更爲複雜的結構中時,這裏能夠設置爲0。
post:
該字段存儲一個指針。能夠指向任何一個在讀取配置過程當中須要的數據,以便於進行配置讀取的處理。大多數時候,都不須要,因此簡單地設爲0便可。
看到這裏,應該就比較清楚了。ngx_http_hello_commands這個數組每5個元素爲一組,用來描述一個配置項的全部狀況。那麼若是有多個配置項,只要按照須要再增長5個對應的元素對新的配置項進行說明。
須要注意的是,就是在ngx_http_hello_commands這個數組定義的最後,都要加一個ngx_null_command做爲結尾。
這是一個ngx_http_module_t類型的靜態變量。這個變量其實是提供一組回調函數指針,這些函數有在建立存儲配置信息的對象的函數,也有在建立前和建立後會調用的函數。這些函數都將被nginx在合適的時間進行調用。
typedef struct { ngx_int_t (*preconfiguration)(ngx_conf_t *cf); ngx_int_t (*postconfiguration)(ngx_conf_t *cf); void *(*create_main_conf)(ngx_conf_t *cf); char *(*init_main_conf)(ngx_conf_t *cf, void *conf); void *(*create_srv_conf)(ngx_conf_t *cf); char *(*merge_srv_conf)(ngx_conf_t *cf, void *prev, void *conf); void *(*create_loc_conf)(ngx_conf_t *cf); char *(*merge_loc_conf)(ngx_conf_t *cf, void *prev, void *conf); } ngx_http_module_t;
preconfiguration:
在建立和讀取該模塊的配置信息以前被調用。
postconfiguration:
在建立和讀取該模塊的配置信息以後被調用。
create_main_conf:
調用該函數建立本模塊位於http block的配置信息存儲結構。該函數成功的時候,返回建立的配置對象。失敗的話,返回NULL。
init_main_conf:
調用該函數初始化本模塊位於http block的配置信息存儲結構。該函數成功的時候,返回NGX_CONF_OK。失敗的話,返回NGX_CONF_ERROR或錯誤字符串。
create_srv_conf:
調用該函數建立本模塊位於http server block的配置信息存儲結構,每一個server block會建立一個。該函數成功的時候,返回建立的配置對象。失敗的話,返回NULL。
merge_srv_conf:
由於有些配置指令既能夠出如今http block,也能夠出如今http server block中。那麼遇到這種狀況,每一個server都會有本身存儲結構來存儲該server的配置,可是在這種狀況下http block中的配置與server block中的配置信息發生衝突的時候,就須要調用此函數進行合併,該函數並不是必須提供,當預計到絕對不會發生須要合併的狀況的時候,就無需提供。固然爲了安全起見仍是建議提供。該函數執行成功的時候,返回NGX_CONF_OK。失敗的話,返回NGX_CONF_ERROR或錯誤字符串。
create_loc_conf:
調用該函數建立本模塊位於location block的配置信息存儲結構。每一個在配置中指明的location建立一個。該函數執行成功,返回建立的配置對象。失敗的話,返回NULL。
merge_loc_conf:
與merge_srv_conf相似,這個也是進行配置值合併的地方。該函數成功的時候,返回NGX_CONF_OK。失敗的話,返回NGX_CONF_ERROR或錯誤字符串。
Nginx裏面的配置信息都是上下一層層的嵌套的,對於具體某個location的話,對於同一個配置,若是當前層次沒有定義,那麼就使用上層的配置,不然使用當前層次的配置。
這些配置信息通常默認都應該設爲一個未初始化的值,針對這個需求,Nginx定義了一系列的宏定義來表明各類配置所對應數據類型的未初始化值,以下:
#define NGX_CONF_UNSET -1 #define NGX_CONF_UNSET_UINT (ngx_uint_t) -1 #define NGX_CONF_UNSET_PTR (void *) -1 #define NGX_CONF_UNSET_SIZE (size_t) -1 #define NGX_CONF_UNSET_MSEC (ngx_msec_t) -1
又由於對於配置項的合併,邏輯都相似,也就是前面已經說過的,若是在本層次已經配置了,也就是配置項的值已經被讀取進來了(那麼這些配置項的值就不會等於上面已經定義的那些UNSET的值),就使用本層次的值做爲定義合併的結果,不然,使用上層的值,若是上層的值也是這些UNSET類的值,那就賦值爲默認值,不然就使用上層的值做爲合併的結果。對於這樣相似的操做,Nginx定義了一些宏操做來作這些事情,咱們來看其中一個的定義。
#define ngx_conf_merge_uint_value(conf, prev, default) \ if (conf == NGX_CONF_UNSET_UINT) { \ conf = (prev == NGX_CONF_UNSET_UINT) ? default : prev; \ }
顯而易見,這個邏輯確實比較簡單,因此其它的宏定義也相似,咱們就列具其中的一部分吧。
ngx_conf_merge_value ngx_conf_merge_ptr_value ngx_conf_merge_uint_value ngx_conf_merge_msec_value ngx_conf_merge_sec_value
等等。
下面來看一下hello模塊的模塊上下文的定義,加深一下印象。
static ngx_http_module_t ngx_http_hello_module_ctx = { NULL, /* preconfiguration */ ngx_http_hello_init, /* postconfiguration */ NULL, /* create main configuration */ NULL, /* init main configuration */ NULL, /* create server configuration */ NULL, /* merge server configuration */ ngx_http_hello_create_loc_conf, /* create location configuration */ NULL /* merge location configuration */ };
注意:這裏並無提供merge_loc_conf函數,由於咱們這個模塊的配置指令已經肯定只出如今NGX_HTTP_LOC_CONF中這一個層次上,不會發生須要合併的狀況。
對於開發一個模塊來講,咱們都須要定義一個ngx_module_t類型的變量來講明這個模塊自己的信息,從某種意義上來講,這是這個模塊最重要的一個信息,它告訴了nginx這個模塊的一些信息,上面定義的配置信息,還有模塊上下文信息,都是經過這個結構來告訴nginx系統的,也就是加載模塊的上層代碼,都須要經過定義的這個結構,來獲取這些信息。
咱們先來看下ngx_module_t的定義
typedef struct ngx_module_s ngx_module_t; struct ngx_module_s { ngx_uint_t ctx_index; ngx_uint_t index; ngx_uint_t spare0; ngx_uint_t spare1; ngx_uint_t abi_compatibility; ngx_uint_t major_version; ngx_uint_t minor_version; void *ctx; ngx_command_t *commands; ngx_uint_t type; ngx_int_t (*init_master)(ngx_log_t *log); ngx_int_t (*init_module)(ngx_cycle_t *cycle); ngx_int_t (*init_process)(ngx_cycle_t *cycle); ngx_int_t (*init_thread)(ngx_cycle_t *cycle); void (*exit_thread)(ngx_cycle_t *cycle); void (*exit_process)(ngx_cycle_t *cycle); void (*exit_master)(ngx_cycle_t *cycle); uintptr_t spare_hook0; uintptr_t spare_hook1; uintptr_t spare_hook2; uintptr_t spare_hook3; uintptr_t spare_hook4; uintptr_t spare_hook5; uintptr_t spare_hook6; uintptr_t spare_hook7; }; #define NGX_NUMBER_MAJOR 3 #define NGX_NUMBER_MINOR 1 #define NGX_MODULE_V1 0, 0, 0, 0, \ NGX_DSO_ABI_COMPATIBILITY, NGX_NUMBER_MAJOR, NGX_NUMBER_MINOR #define NGX_MODULE_V1_PADDING 0, 0, 0, 0, 0, 0, 0, 0
再看一下hello模塊的模塊定義。
ngx_module_t ngx_http_hello_module = { NGX_MODULE_V1, &ngx_http_hello_module_ctx, /* module context */ ngx_http_hello_commands, /* module directives */ NGX_HTTP_MODULE, /* module type */ NULL, /* init master */ NULL, /* init module */ NULL, /* init process */ NULL, /* init thread */ NULL, /* exit thread */ NULL, /* exit process */ NULL, /* exit master */ NGX_MODULE_V1_PADDING };
模塊能夠提供一些回調函數給nginx,當nginx在建立進程線程或者結束進程線程時進行調用。但大多數模塊在這些時刻並不須要作什麼,因此都簡單賦值爲NULL。
除了上一節介紹的模塊的基本結構之外,handler模塊必須提供一個真正的處理函數,這個函數負責對來自客戶端請求的真正處理。這個函數的處理,既能夠選擇本身直接生成內容,也能夠選擇拒絕處理,由後續的handler去進行處理,或者是選擇丟給後續的filter進行處理。來看一下這個函數的原型申明。
typedef ngx_int_t (*ngx_http_handler_pt)(ngx_http_request_t *r);
r是http請求。裏面包含請求全部的信息,這裏不詳細說明了,能夠參考別的章節的介紹。 該函數處理成功返回NGX_OK,處理髮生錯誤返回NGX_ERROR,拒絕處理(留給後續的handler進行處理)返回NGX_DECLINE。 返回NGX_OK也就表明給客戶端的響應已經生成好了,不然返回NGX_ERROR就發生錯誤了。
handler模塊真正的處理函數經過兩種方式掛載處處理過程當中,一種方式就是按處理階段掛載;另一種掛載方式就是按需掛載。
爲了更精細地控制對於客戶端請求的處理過程,nginx把這個處理過程劃分紅了11個階段。他們從前到後,依次列舉以下:
NGX_HTTP_POST_READ_PHASE:
讀取請求內容階段
NGX_HTTP_SERVER_REWRITE_PHASE:
Server請求地址重寫階段
NGX_HTTP_FIND_CONFIG_PHASE:
配置查找階段:
NGX_HTTP_REWRITE_PHASE:
Location請求地址重寫階段
NGX_HTTP_POST_REWRITE_PHASE:
請求地址重寫提交階段
NGX_HTTP_PREACCESS_PHASE:
訪問權限檢查準備階段
NGX_HTTP_ACCESS_PHASE:
訪問權限檢查階段
NGX_HTTP_POST_ACCESS_PHASE:
訪問權限檢查提交階段
NGX_HTTP_TRY_FILES_PHASE:
配置項try_files處理階段
NGX_HTTP_CONTENT_PHASE:
內容產生階段
NGX_HTTP_LOG_PHASE:
日誌模塊處理階段
通常狀況下,咱們自定義的模塊,大多數是掛載在NGX_HTTP_CONTENT_PHASE階段的。掛載的動做通常是在模塊上下文調用的postconfiguration函數中。
注意:有幾個階段是特例,它不調用掛載地任何的handler,也就是你就不用掛載到這幾個階段了:
因此其實真正是有7個phase你能夠去掛載handler。
掛載的代碼以下(摘自hello module):
static ngx_int_t ngx_http_hello_init(ngx_conf_t *cf) { ngx_http_handler_pt *h; ngx_http_core_main_conf_t *cmcf; cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); h = ngx_array_push(&cmcf->phases[NGX_HTTP_CONTENT_PHASE].handlers); if (h == NULL) { return NGX_ERROR; } *h = ngx_http_hello_handler; return NGX_OK; }
使用這種方式掛載的handler也被稱爲 content phase handlers。
以這種方式掛載的handler也被稱爲 content handler。
當一個請求進來之後,nginx從NGX_HTTP_POST_READ_PHASE階段開始依次執行每一個階段中全部handler。執行到 NGX_HTTP_CONTENT_PHASE階段的時候,若是這個location有一個對應的content handler模塊,那麼就去執行這個content handler模塊真正的處理函數。不然繼續依次執行NGX_HTTP_CONTENT_PHASE階段中全部content phase handlers,直到某個函數處理返回NGX_OK或者NGX_ERROR。
換句話說,當某個location處理到NGX_HTTP_CONTENT_PHASE階段時,若是有content handler模塊,那麼NGX_HTTP_CONTENT_PHASE掛載的全部content phase handlers都不會被執行了。
可是使用這個方法掛載上去的handler有一個特色是必須在NGX_HTTP_CONTENT_PHASE階段才能執行到。若是你想本身的handler在更早的階段執行,那就不要使用這種掛載方式。
那麼在什麼狀況會使用這種方式來掛載呢?通常狀況下,某個模塊對某個location進行了處理之後,發現符合本身處理的邏輯,並且也沒有必要再調用NGX_HTTP_CONTENT_PHASE階段的其它handler進行處理的時候,就動態掛載上這個handler。
下面來看一下使用這種掛載方式的具體例子(摘自Emiller’s Guide To Nginx Module Development)。
static char * ngx_http_circle_gif(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) { ngx_http_core_loc_conf_t *clcf; clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module); clcf->handler = ngx_http_circle_gif_handler; return NGX_CONF_OK; }
好,到了這裏,讓咱們稍微整理一下思路,回顧一下實現一個handler的步驟:
看起來不是那麼難,對吧?仍是那句老話,世上無難事,只怕有心人! 如今咱們來完整的分析前面提到的hello handler module示例的功能和代碼。
在前面已經看到了這個hello handler module的部分重要的結構。該模塊提供了2個配置指令,僅能夠出如今location指令的做用域中。這兩個指令是hello_string, 該指令接受一個參數來設置顯示的字符串。若是沒有跟參數,那麼就使用默認的字符串做爲響應字符串。
另外一個指令是hello_counter,若是設置爲on,則會在響應的字符串後面追加Visited Times:的字樣,以統計請求的次數。
這裏有兩點注意一下:
下面來完整的給出ngx_http_hello_module模塊的完整代碼。
#include <ngx_config.h> #include <ngx_core.h> #include <ngx_http.h> typedef struct { ngx_str_t hello_string; ngx_int_t hello_counter; }ngx_http_hello_loc_conf_t; static ngx_int_t ngx_http_hello_init(ngx_conf_t *cf); static void *ngx_http_hello_create_loc_conf(ngx_conf_t *cf); static char *ngx_http_hello_string(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); static char *ngx_http_hello_counter(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); static ngx_command_t ngx_http_hello_commands[] = { { ngx_string("hello_string"), NGX_HTTP_LOC_CONF|NGX_CONF_NOARGS|NGX_CONF_TAKE1, ngx_http_hello_string, NGX_HTTP_LOC_CONF_OFFSET, offsetof(ngx_http_hello_loc_conf_t, hello_string), NULL }, { ngx_string("hello_counter"), NGX_HTTP_LOC_CONF|NGX_CONF_FLAG, ngx_http_hello_counter, NGX_HTTP_LOC_CONF_OFFSET, offsetof(ngx_http_hello_loc_conf_t, hello_counter), NULL }, ngx_null_command }; /* static u_char ngx_hello_default_string[] = "Default String: Hello, world!"; */ static int ngx_hello_visited_times = 0; static ngx_http_module_t ngx_http_hello_module_ctx = { NULL, /* preconfiguration */ ngx_http_hello_init, /* postconfiguration */ NULL, /* create main configuration */ NULL, /* init main configuration */ NULL, /* create server configuration */ NULL, /* merge server configuration */ ngx_http_hello_create_loc_conf, /* create location configuration */ NULL /* merge location configuration */ }; ngx_module_t ngx_http_hello_module = { NGX_MODULE_V1, &ngx_http_hello_module_ctx, /* module context */ ngx_http_hello_commands, /* module directives */ NGX_HTTP_MODULE, /* module type */ NULL, /* init master */ NULL, /* init module */ NULL, /* init process */ NULL, /* init thread */ NULL, /* exit thread */ NULL, /* exit process */ NULL, /* exit master */ NGX_MODULE_V1_PADDING }; static ngx_int_t ngx_http_hello_handler(ngx_http_request_t *r) { ngx_int_t rc; ngx_buf_t *b; ngx_chain_t out; ngx_http_hello_loc_conf_t* my_conf; u_char ngx_hello_string[1024] = {0}; ngx_uint_t content_length = 0; ngx_log_error(NGX_LOG_EMERG, r->connection->log, 0, "ngx_http_hello_handler is called!"); my_conf = ngx_http_get_module_loc_conf(r, ngx_http_hello_module); if (my_conf->hello_string.len == 0 ) { ngx_log_error(NGX_LOG_EMERG, r->connection->log, 0, "hello_string is empty!"); return NGX_DECLINED; } if (my_conf->hello_counter == NGX_CONF_UNSET || my_conf->hello_counter == 0) { ngx_sprintf(ngx_hello_string, "%s", my_conf->hello_string.data); } else { ngx_sprintf(ngx_hello_string, "%s Visited Times:%d", my_conf->hello_string.data, ++ngx_hello_visited_times); } ngx_log_error(NGX_LOG_EMERG, r->connection->log, 0, "hello_string:%s", ngx_hello_string); content_length = ngx_strlen(ngx_hello_string); /* we response to 'GET' and 'HEAD' requests only */ if (!(r->method & (NGX_HTTP_GET|NGX_HTTP_HEAD))) { return NGX_HTTP_NOT_ALLOWED; } /* discard request body, since we don't need it here */ rc = ngx_http_discard_request_body(r); if (rc != NGX_OK) { return rc; } /* set the 'Content-type' header */ /* *r->headers_out.content_type.len = sizeof("text/html") - 1; *r->headers_out.content_type.data = (u_char *)"text/html"; */ ngx_str_set(&r->headers_out.content_type, "text/html"); /* send the header only, if the request type is http 'HEAD' */ if (r->method == NGX_HTTP_HEAD) { r->headers_out.status = NGX_HTTP_OK; r->headers_out.content_length_n = content_length; return ngx_http_send_header(r); } /* allocate a buffer for your response body */ b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t)); if (b == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } /* attach this buffer to the buffer chain */ out.buf = b; out.next = NULL; /* adjust the pointers of the buffer */ b->pos = ngx_hello_string; b->last = ngx_hello_string + content_length; b->memory = 1; /* this buffer is in memory */ b->last_buf = 1; /* this is the last buffer in the buffer chain */ /* set the status line */ r->headers_out.status = NGX_HTTP_OK; r->headers_out.content_length_n = content_length; /* send the headers of your response */ rc = ngx_http_send_header(r); if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) { return rc; } /* send the buffer chain of your response */ return ngx_http_output_filter(r, &out); } static void *ngx_http_hello_create_loc_conf(ngx_conf_t *cf) { ngx_http_hello_loc_conf_t* local_conf = NULL; local_conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_hello_loc_conf_t)); if (local_conf == NULL) { return NULL; } ngx_str_null(&local_conf->hello_string); local_conf->hello_counter = NGX_CONF_UNSET; return local_conf; } /* static char *ngx_http_hello_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child) { ngx_http_hello_loc_conf_t* prev = parent; ngx_http_hello_loc_conf_t* conf = child; ngx_conf_merge_str_value(conf->hello_string, prev->hello_string, ngx_hello_default_string); ngx_conf_merge_value(conf->hello_counter, prev->hello_counter, 0); return NGX_CONF_OK; }*/ static char * ngx_http_hello_string(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) { ngx_http_hello_loc_conf_t* local_conf; local_conf = conf; char* rv = ngx_conf_set_str_slot(cf, cmd, conf); ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "hello_string:%s", local_conf->hello_string.data); return rv; } static char *ngx_http_hello_counter(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) { ngx_http_hello_loc_conf_t* local_conf; local_conf = conf; char* rv = NULL; rv = ngx_conf_set_flag_slot(cf, cmd, conf); ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "hello_counter:%d", local_conf->hello_counter); return rv; } static ngx_int_t ngx_http_hello_init(ngx_conf_t *cf) { ngx_http_handler_pt *h; ngx_http_core_main_conf_t *cmcf; cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); h = ngx_array_push(&cmcf->phases[NGX_HTTP_CONTENT_PHASE].handlers); if (h == NULL) { return NGX_ERROR; } *h = ngx_http_hello_handler; return NGX_OK; }
經過上面一些介紹,我相信你們都能對整個示例模塊有一個比較好的理解。惟一可能感受有些理解困難的地方在於ngx_http_hello_handler函數裏面產生和設置輸出。但其實你們在本書的前面的相關章節均可以看到對ngx_buf_t和request等相關數據結構的說明。若是仔細看了這些地方的說明的話,應該對這裏代碼的實現就比較容易理解了。所以,這裏再也不贅述解釋。
模塊的功能開發完了以後,模塊的使用還須要編譯纔可以執行,下面咱們來看下模塊的編譯和使用。
對於開發一個模塊,咱們是須要把這個模塊的C代碼組織到一個目錄裏,同時須要編寫一個config文件。這個config文件的內容就是告訴nginx的編譯腳本,該如何進行編譯。咱們來看一下hello handler module的config文件的內容,而後再作解釋。
ngx_addon_name=ngx_http_hello_module HTTP_MODULES="$HTTP_MODULES ngx_http_hello_module" NGX_ADDON_SRCS="$NGX_ADDON_SRCS $ngx_addon_dir/ngx_http_hello_module.c"
其實文件很簡單,幾乎不須要作什麼解釋。你們一看都懂了。惟一須要說明的是,若是這個模塊的實現有多個源文件,那麼都在NGX_ADDON_SRCS這個變量裏,依次寫進去就能夠。
對於模塊的編譯,nginx並不像apache同樣,提供了單獨的編譯工具,能夠在沒有apache源代碼的狀況下來單獨編譯一個模塊的代碼。nginx必須去到nginx的源代碼目錄裏,經過configure指令的參數,來進行編譯。下面看一下hello module的configure指令:
./configure –prefix=/usr/local/nginx-1.3.1 –add-module=/home/jizhao/open_source/book_module
我寫的這個示例模塊的代碼和config文件都放在/home/jizhao/open_source/book_module這個目錄下。因此一切都很明瞭,也沒什麼好說的了。
使用一個模塊須要根據這個模塊定義的配置指令來作。好比咱們這個簡單的hello handler module的使用就很簡單。在個人測試服務器的配置文件裏,就是在http裏面的默認的server裏面加入以下的配置:
location /test { hello_string jizhao; hello_counter on; }
當咱們訪問這個地址的時候, lynx http://127.0.0.1/test的時候,就能夠看到返回的結果。
jizhao Visited Times:1
固然你訪問屢次,這個次數是會增長的。
該模塊的代碼位於src/http/modules/ngx_http_access_module.c中。該模塊的做用是提供對於特定host的客戶端的訪問控制。能夠限定特定host的客戶端對於服務端所有,或者某個server,或者是某個location的訪問。 該模塊的實現很是簡單,總共也就只有幾個函數。
static ngx_int_t ngx_http_access_handler(ngx_http_request_t *r); static ngx_int_t ngx_http_access_inet(ngx_http_request_t *r, ngx_http_access_loc_conf_t *alcf, in_addr_t addr); #if (NGX_HAVE_INET6) static ngx_int_t ngx_http_access_inet6(ngx_http_request_t *r, ngx_http_access_loc_conf_t *alcf, u_char *p); #endif static ngx_int_t ngx_http_access_found(ngx_http_request_t *r, ngx_uint_t deny); static char *ngx_http_access_rule(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); static void *ngx_http_access_create_loc_conf(ngx_conf_t *cf); static char *ngx_http_access_merge_loc_conf(ngx_conf_t *cf, void *parent, void *child); static ngx_int_t ngx_http_access_init(ngx_conf_t *cf);
對於與配置相關的幾個函數都不須要作解釋了,須要提一下的是函數ngx_http_access_init,該函數在實現上把本模塊掛載到了NGX_HTTP_ACCESS_PHASE階段的handler上,從而使本身的被調用時機發生在了NGX_HTTP_CONTENT_PHASE等階段前。由於進行客戶端地址的限制檢查,根本不須要等到這麼後面。
另外看一下這個模塊的主處理函數ngx_http_access_handler。這個函數的邏輯也很是簡單,主要是根據客戶端地址的類型,來分別選擇ipv4類型的處理函數ngx_http_access_inet仍是ipv6類型的處理函數ngx_http_access_inet6。
而這個兩個處理函數內部也很是簡單,就是循環檢查每一個規則,檢查是否有匹配的規則,若是有就返回匹配的結果,若是都沒有匹配,就默認拒絕。
從某種程度上來講,此模塊能夠算的上是「最正宗的」,「最古老」的content handler。由於本模塊的做用就是讀取磁盤上的靜態文件,並把文件內容做爲產生的輸出。在Web技術發展的早期,只有靜態頁面,沒有服務端腳原本動態生成HTML的時候。恐怕開發個Web服務器的時候,第一個要開發就是這樣一個content handler。
http static module的代碼位於src/http/modules/ngx_http_static_module.c中,總共只有兩百多行近三百行。能夠說是很是短小。
咱們首先來看一下該模塊的模塊上下文的定義。
ngx_http_module_t ngx_http_static_module_ctx = { NULL, /* preconfiguration */ ngx_http_static_init, /* postconfiguration */ NULL, /* create main configuration */ NULL, /* init main configuration */ NULL, /* create server configuration */ NULL, /* merge server configuration */ NULL, /* create location configuration */ NULL /* merge location configuration */ };
是很是的簡潔吧,連任何與配置相關的函數都沒有。對了,由於該模塊沒有提供任何配置指令。你們想一想也就知道了,這個模塊作的事情實在是太簡單了,也確實沒什麼好配置的。惟一須要調用的函數是一個ngx_http_static_init函數。好了,來看一下這個函數都幹了寫什麼。
static ngx_int_t ngx_http_static_init(ngx_conf_t *cf) { ngx_http_handler_pt *h; ngx_http_core_main_conf_t *cmcf; cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); h = ngx_array_push(&cmcf->phases[NGX_HTTP_CONTENT_PHASE].handlers); if (h == NULL) { return NGX_ERROR; } *h = ngx_http_static_handler; return NGX_OK; }
僅僅是掛載這個handler到NGX_HTTP_CONTENT_PHASE處理階段。簡單吧?
下面咱們就看一下這個模塊最核心的處理邏輯所在的ngx_http_static_handler函數。該函數大概佔了這個模塊代碼量的百分之八九十。
static ngx_int_t ngx_http_static_handler(ngx_http_request_t *r) { u_char *last, *location; size_t root, len; ngx_str_t path; ngx_int_t rc; ngx_uint_t level; ngx_log_t *log; ngx_buf_t *b; ngx_chain_t out; ngx_open_file_info_t of; ngx_http_core_loc_conf_t *clcf; if (!(r->method & (NGX_HTTP_GET|NGX_HTTP_HEAD|NGX_HTTP_POST))) { return NGX_HTTP_NOT_ALLOWED; } if (r->uri.data[r->uri.len - 1] == '/') { return NGX_DECLINED; } log = r->connection->log; /* * ngx_http_map_uri_to_path() allocates memory for terminating '\0' * so we do not need to reserve memory for '/' for possible redirect */ last = ngx_http_map_uri_to_path(r, &path, &root, 0); if (last == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } path.len = last - path.data; ngx_log_debug1(NGX_LOG_DEBUG_HTTP, log, 0, "http filename: \"%s\"", path.data); clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); ngx_memzero(&of, sizeof(ngx_open_file_info_t)); of.read_ahead = clcf->read_ahead; of.directio = clcf->directio; of.valid = clcf->open_file_cache_valid; of.min_uses = clcf->open_file_cache_min_uses; of.errors = clcf->open_file_cache_errors; of.events = clcf->open_file_cache_events; if (ngx_http_set_disable_symlinks(r, clcf, &path, &of) != NGX_OK) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } if (ngx_open_cached_file(clcf->open_file_cache, &path, &of, r->pool) != NGX_OK) { switch (of.err) { case 0: return NGX_HTTP_INTERNAL_SERVER_ERROR; case NGX_ENOENT: case NGX_ENOTDIR: case NGX_ENAMETOOLONG: level = NGX_LOG_ERR; rc = NGX_HTTP_NOT_FOUND; break; case NGX_EACCES: #if (NGX_HAVE_OPENAT) case NGX_EMLINK: case NGX_ELOOP: #endif level = NGX_LOG_ERR; rc = NGX_HTTP_FORBIDDEN; break; default: level = NGX_LOG_CRIT; rc = NGX_HTTP_INTERNAL_SERVER_ERROR; break; } if (rc != NGX_HTTP_NOT_FOUND || clcf->log_not_found) { ngx_log_error(level, log, of.err, "%s \"%s\" failed", of.failed, path.data); } return rc; } r->root_tested = !r->error_page; ngx_log_debug1(NGX_LOG_DEBUG_HTTP, log, 0, "http static fd: %d", of.fd); if (of.is_dir) { ngx_log_debug0(NGX_LOG_DEBUG_HTTP, log, 0, "http dir"); ngx_http_clear_location(r); r->headers_out.location = ngx_palloc(r->pool, sizeof(ngx_table_elt_t)); if (r->headers_out.location == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } len = r->uri.len + 1; if (!clcf->alias && clcf->root_lengths == NULL && r->args.len == 0) { location = path.data + clcf->root.len; *last = '/'; } else { if (r->args.len) { len += r->args.len + 1; } location = ngx_pnalloc(r->pool, len); if (location == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } last = ngx_copy(location, r->uri.data, r->uri.len); *last = '/'; if (r->args.len) { *++last = '?'; ngx_memcpy(++last, r->args.data, r->args.len); } } /* * we do not need to set the r->headers_out.location->hash and * r->headers_out.location->key fields */ r->headers_out.location->value.len = len; r->headers_out.location->value.data = location; return NGX_HTTP_MOVED_PERMANENTLY; } #if !(NGX_WIN32) /* the not regular files are probably Unix specific */ if (!of.is_file) { ngx_log_error(NGX_LOG_CRIT, log, 0, "\"%s\" is not a regular file", path.data); return NGX_HTTP_NOT_FOUND; } #endif if (r->method & NGX_HTTP_POST) { return NGX_HTTP_NOT_ALLOWED; } rc = ngx_http_discard_request_body(r); if (rc != NGX_OK) { return rc; } log->action = "sending response to client"; r->headers_out.status = NGX_HTTP_OK; r->headers_out.content_length_n = of.size; r->headers_out.last_modified_time = of.mtime; if (ngx_http_set_content_type(r) != NGX_OK) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } if (r != r->main && of.size == 0) { return ngx_http_send_header(r); } r->allow_ranges = 1; /* we need to allocate all before the header would be sent */ b = ngx_pcalloc(r->pool, sizeof(ngx_buf_t)); if (b == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } b->file = ngx_pcalloc(r->pool, sizeof(ngx_file_t)); if (b->file == NULL) { return NGX_HTTP_INTERNAL_SERVER_ERROR; } rc = ngx_http_send_header(r); if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) { return rc; } b->file_pos = 0; b->file_last = of.size; b->in_file = b->file_last ? 1: 0; b->last_buf = (r == r->main) ? 1: 0; b->last_in_chain = 1; b->file->fd = of.fd; b->file->name = path; b->file->log = log; b->file->directio = of.is_directio; out.buf = b; out.next = NULL; return ngx_http_output_filter(r, &out); }
首先是檢查客戶端的http請求類型(r->method),若是請求類型爲NGX_HTTP_GET|NGX_HTTP_HEAD|NGX_HTTP_POST,則繼續進行處理,不然一概返回NGX_HTTP_NOT_ALLOWED從而拒絕客戶端的發起的請求。
其次是檢查請求的url的結尾字符是否是斜槓‘/’,若是是說明請求的不是一個文件,給後續的handler去處理,好比後續的ngx_http_autoindex_handler(若是是請求的是一個目錄下面,能夠列出這個目錄的文件),或者是ngx_http_index_handler(若是請求的路徑下面有個默認的index文件,直接返回index文件的內容)。
而後接下來調用了一個ngx_http_map_uri_to_path函數,該函數的做用是把請求的http協議的路徑轉化成一個文件系統的路徑。
而後根據轉化出來的具體路徑,去打開文件,打開文件的時候作了2種檢查,一種是,若是請求的文件是個symbol link,根據配置,是否容許符號連接,不容許返回錯誤。還有一個檢查是,若是請求的是一個名稱,是一個目錄的名字,也返回錯誤。若是都沒有錯誤,就讀取文件,返回內容。其實說返回內容可能不是特別準確,比較準確的說法是,把產生的內容傳遞給後續的filter去處理。
該模塊提供了對於每個http請求進行記錄的功能,也就是咱們見到的access.log。固然這個模塊對於log提供了一些配置指令,使得能夠比較方便的定製access.log。
這個模塊的代碼位於src/http/modules/ngx_http_log_module.c,雖然這個模塊的代碼有接近1400行,可是主要的邏輯在於對日誌自己格式啊,等細節的處理。咱們在這裏進行分析主要是關注,如何編寫一個log handler的問題。
因爲log handler的時候,拿到的參數也是request這個東西,那麼也就意味着咱們若是須要,能夠好好研究下這個結構,把咱們須要的全部信息都記錄下來。
對於log handler,有一點特別須要注意的就是,log handler是不管如何都會被調用的,就是隻要服務端接受到了一個客戶端的請求,也就是產生了一個request對象,那麼這些個log handler的處理函數都會被調用的,就是在釋放request的時候被調用的(ngx_http_free_request函數)。
那麼固然絕對不能忘記的就是log handler最好,也是建議被掛載在NGX_HTTP_LOG_PHASE階段。由於掛載在其餘階段,有可能在某些狀況下被跳過,而沒有執行到,致使你的log模塊記錄的信息不全。
還有一點要說明的是,因爲nginx是容許在某個階段有多個handler模塊存在的,根據其處理結果,肯定是否要調用下一個handler。可是對於掛載在NGX_HTTP_LOG_PHASE階段的handler,則根本不關注這裏handler的具體處理函數的返回值,全部的都被調用。以下,位於src/http/ngx_http_request.c中的ngx_http_log_request函數。
static void ngx_http_log_request(ngx_http_request_t *r) { ngx_uint_t i, n; ngx_http_handler_pt *log_handler; ngx_http_core_main_conf_t *cmcf; cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module); log_handler = cmcf->phases[NGX_HTTP_LOG_PHASE].handlers.elts; n = cmcf->phases[NGX_HTTP_LOG_PHASE].handlers.nelts; for (i = 0; i < n; i++) { log_handler[i](r); } }