xref: /unit/src/nxt_controller.c (revision 45)
120Sigor@sysoev.ru 
220Sigor@sysoev.ru /*
320Sigor@sysoev.ru  * Copyright (C) Igor Sysoev
420Sigor@sysoev.ru  * Copyright (C) Valentin V. Bartenev
520Sigor@sysoev.ru  * Copyright (C) NGINX, Inc.
620Sigor@sysoev.ru  */
720Sigor@sysoev.ru 
820Sigor@sysoev.ru #include <nxt_main.h>
920Sigor@sysoev.ru #include <nxt_runtime.h>
1020Sigor@sysoev.ru #include <nxt_master_process.h>
1129Svbart@nginx.com #include <nxt_conf.h>
1220Sigor@sysoev.ru 
1320Sigor@sysoev.ru 
1427Svbart@nginx.com typedef struct {
1544Svbart@nginx.com     nxt_conf_json_value_t  *root;
1644Svbart@nginx.com     nxt_mem_pool_t         *pool;
1744Svbart@nginx.com } nxt_controller_conf_t;
1844Svbart@nginx.com 
1944Svbart@nginx.com 
2044Svbart@nginx.com typedef struct {
2127Svbart@nginx.com     nxt_http_request_parse_t  parser;
2227Svbart@nginx.com     size_t                    length;
2333Svbart@nginx.com 
2444Svbart@nginx.com     nxt_controller_conf_t     conf;
2527Svbart@nginx.com } nxt_controller_request_t;
2627Svbart@nginx.com 
2727Svbart@nginx.com 
2844Svbart@nginx.com typedef struct {
2944Svbart@nginx.com     nxt_str_t              status_line;
3044Svbart@nginx.com     nxt_conf_json_value_t  *json_value;
3144Svbart@nginx.com     nxt_str_t              json_string;
3244Svbart@nginx.com } nxt_controller_response_t;
3344Svbart@nginx.com 
3444Svbart@nginx.com 
3520Sigor@sysoev.ru static void nxt_controller_conn_init(nxt_task_t *task, void *obj, void *data);
3620Sigor@sysoev.ru static void nxt_controller_conn_read(nxt_task_t *task, void *obj, void *data);
3720Sigor@sysoev.ru static nxt_msec_t nxt_controller_conn_timeout_value(nxt_event_conn_t *c,
3820Sigor@sysoev.ru     uintptr_t data);
3920Sigor@sysoev.ru static void nxt_controller_conn_read_error(nxt_task_t *task, void *obj,
4020Sigor@sysoev.ru     void *data);
4120Sigor@sysoev.ru static void nxt_controller_conn_read_timeout(nxt_task_t *task, void *obj,
4220Sigor@sysoev.ru     void *data);
4327Svbart@nginx.com static void nxt_controller_conn_body_read(nxt_task_t *task, void *obj,
4427Svbart@nginx.com     void *data);
4527Svbart@nginx.com static void nxt_controller_conn_write(nxt_task_t *task, void *obj, void *data);
4627Svbart@nginx.com static void nxt_controller_conn_write_error(nxt_task_t *task, void *obj,
4727Svbart@nginx.com     void *data);
4827Svbart@nginx.com static void nxt_controller_conn_write_timeout(nxt_task_t *task, void *obj,
4927Svbart@nginx.com     void *data);
5020Sigor@sysoev.ru static void nxt_controller_conn_close(nxt_task_t *task, void *obj, void *data);
5120Sigor@sysoev.ru static void nxt_controller_conn_free(nxt_task_t *task, void *obj, void *data);
5220Sigor@sysoev.ru 
5327Svbart@nginx.com static nxt_int_t nxt_controller_request_content_length(void *ctx,
5427Svbart@nginx.com     nxt_str_t *name, nxt_str_t *value, uintptr_t data);
5527Svbart@nginx.com 
5627Svbart@nginx.com static void nxt_controller_process_request(nxt_task_t *task,
5727Svbart@nginx.com     nxt_event_conn_t *c, nxt_controller_request_t *r);
5829Svbart@nginx.com static nxt_int_t nxt_controller_request_body_parse(nxt_task_t *task,
5933Svbart@nginx.com     nxt_event_conn_t *c, nxt_controller_request_t *r);
6044Svbart@nginx.com static nxt_int_t nxt_controller_response(nxt_task_t *task, nxt_event_conn_t *c,
6144Svbart@nginx.com     nxt_controller_response_t *resp);
62*45Svbart@nginx.com static nxt_buf_t *nxt_controller_response_body(nxt_controller_response_t *resp,
63*45Svbart@nginx.com     nxt_mem_pool_t *pool);
6427Svbart@nginx.com 
6527Svbart@nginx.com 
6627Svbart@nginx.com static nxt_http_fields_t  nxt_controller_request_fields[] = {
6727Svbart@nginx.com     { nxt_string("Content-Length"),
6827Svbart@nginx.com       &nxt_controller_request_content_length, 0 },
6927Svbart@nginx.com 
7027Svbart@nginx.com     { nxt_null_string, NULL, 0 }
7127Svbart@nginx.com };
7227Svbart@nginx.com 
7344Svbart@nginx.com static nxt_http_fields_hash_t  *nxt_controller_request_fields_hash;
7427Svbart@nginx.com 
7544Svbart@nginx.com 
7644Svbart@nginx.com static nxt_controller_conf_t  nxt_controller_conf;
7727Svbart@nginx.com 
7820Sigor@sysoev.ru 
7920Sigor@sysoev.ru static const nxt_event_conn_state_t  nxt_controller_conn_read_state;
8027Svbart@nginx.com static const nxt_event_conn_state_t  nxt_controller_conn_body_read_state;
8127Svbart@nginx.com static const nxt_event_conn_state_t  nxt_controller_conn_write_state;
8220Sigor@sysoev.ru static const nxt_event_conn_state_t  nxt_controller_conn_close_state;
8320Sigor@sysoev.ru 
8420Sigor@sysoev.ru 
8520Sigor@sysoev.ru nxt_int_t
8620Sigor@sysoev.ru nxt_controller_start(nxt_task_t *task, nxt_runtime_t *rt)
8720Sigor@sysoev.ru {
8844Svbart@nginx.com     nxt_mem_pool_t          *mp;
8944Svbart@nginx.com     nxt_conf_json_value_t   *conf;
9027Svbart@nginx.com     nxt_http_fields_hash_t  *hash;
9127Svbart@nginx.com 
9244Svbart@nginx.com     static const nxt_str_t json
9344Svbart@nginx.com         = nxt_string("{ \"sockets\": {}, \"applications\": {} }");
9444Svbart@nginx.com 
9527Svbart@nginx.com     hash = nxt_http_fields_hash(nxt_controller_request_fields, rt->mem_pool);
9627Svbart@nginx.com 
9727Svbart@nginx.com     if (nxt_slow_path(hash == NULL)) {
9827Svbart@nginx.com         return NXT_ERROR;
9927Svbart@nginx.com     }
10027Svbart@nginx.com 
10127Svbart@nginx.com     nxt_controller_request_fields_hash = hash;
10227Svbart@nginx.com 
10320Sigor@sysoev.ru     if (nxt_event_conn_listen(task, rt->controller_socket) != NXT_OK) {
10420Sigor@sysoev.ru         return NXT_ERROR;
10520Sigor@sysoev.ru     }
10620Sigor@sysoev.ru 
10744Svbart@nginx.com     mp = nxt_mem_pool_create(256);
10844Svbart@nginx.com 
10944Svbart@nginx.com     if (nxt_slow_path(mp == NULL)) {
11044Svbart@nginx.com         return NXT_ERROR;
11144Svbart@nginx.com     }
11244Svbart@nginx.com 
11344Svbart@nginx.com     conf = nxt_conf_json_parse(json.start, json.length, mp);
11444Svbart@nginx.com 
11544Svbart@nginx.com     if (conf == NULL) {
11644Svbart@nginx.com         return NXT_ERROR;
11744Svbart@nginx.com     }
11844Svbart@nginx.com 
11944Svbart@nginx.com     nxt_controller_conf.root = conf;
12044Svbart@nginx.com     nxt_controller_conf.pool = mp;
12144Svbart@nginx.com 
12220Sigor@sysoev.ru     return NXT_OK;
12320Sigor@sysoev.ru }
12420Sigor@sysoev.ru 
12520Sigor@sysoev.ru 
12620Sigor@sysoev.ru nxt_int_t
12720Sigor@sysoev.ru nxt_runtime_controller_socket(nxt_task_t *task, nxt_runtime_t *rt)
12820Sigor@sysoev.ru {
12920Sigor@sysoev.ru     nxt_sockaddr_t       *sa;
13020Sigor@sysoev.ru     nxt_listen_socket_t  *ls;
13120Sigor@sysoev.ru 
13220Sigor@sysoev.ru     sa = rt->controller_listen;
13320Sigor@sysoev.ru 
13420Sigor@sysoev.ru     if (rt->controller_listen == NULL) {
13520Sigor@sysoev.ru         sa = nxt_sockaddr_alloc(rt->mem_pool, sizeof(struct sockaddr_in),
13620Sigor@sysoev.ru                                 NXT_INET_ADDR_STR_LEN);
13720Sigor@sysoev.ru         if (sa == NULL) {
13820Sigor@sysoev.ru             return NXT_ERROR;
13920Sigor@sysoev.ru         }
14020Sigor@sysoev.ru 
14120Sigor@sysoev.ru         sa->type = SOCK_STREAM;
14220Sigor@sysoev.ru         sa->u.sockaddr_in.sin_family = AF_INET;
14320Sigor@sysoev.ru         sa->u.sockaddr_in.sin_port = htons(8443);
14420Sigor@sysoev.ru 
14520Sigor@sysoev.ru         nxt_sockaddr_text(sa);
14620Sigor@sysoev.ru 
14720Sigor@sysoev.ru         rt->controller_listen = sa;
14820Sigor@sysoev.ru     }
14920Sigor@sysoev.ru 
15020Sigor@sysoev.ru     ls = nxt_mem_alloc(rt->mem_pool, sizeof(nxt_listen_socket_t));
15120Sigor@sysoev.ru     if (ls == NULL) {
15220Sigor@sysoev.ru         return NXT_ERROR;
15320Sigor@sysoev.ru     }
15420Sigor@sysoev.ru 
15520Sigor@sysoev.ru     ls->sockaddr = nxt_sockaddr_create(rt->mem_pool, &sa->u.sockaddr,
15620Sigor@sysoev.ru                                        sa->socklen, sa->length);
15720Sigor@sysoev.ru     if (ls->sockaddr == NULL) {
15820Sigor@sysoev.ru         return NXT_ERROR;
15920Sigor@sysoev.ru     }
16020Sigor@sysoev.ru 
16120Sigor@sysoev.ru     ls->sockaddr->type = sa->type;
16220Sigor@sysoev.ru 
16320Sigor@sysoev.ru     nxt_sockaddr_text(ls->sockaddr);
16420Sigor@sysoev.ru 
16520Sigor@sysoev.ru     ls->socket = -1;
16620Sigor@sysoev.ru     ls->backlog = NXT_LISTEN_BACKLOG;
16720Sigor@sysoev.ru     ls->read_after_accept = 1;
16820Sigor@sysoev.ru     ls->flags = NXT_NONBLOCK;
16920Sigor@sysoev.ru 
17020Sigor@sysoev.ru #if 0
17120Sigor@sysoev.ru     /* STUB */
17220Sigor@sysoev.ru     wq = nxt_mem_zalloc(cf->mem_pool, sizeof(nxt_work_queue_t));
17320Sigor@sysoev.ru     if (wq == NULL) {
17420Sigor@sysoev.ru         return NXT_ERROR;
17520Sigor@sysoev.ru     }
17620Sigor@sysoev.ru     nxt_work_queue_name(wq, "listen");
17720Sigor@sysoev.ru     /**/
17820Sigor@sysoev.ru 
17920Sigor@sysoev.ru     ls->work_queue = wq;
18020Sigor@sysoev.ru #endif
18120Sigor@sysoev.ru     ls->handler = nxt_controller_conn_init;
18220Sigor@sysoev.ru 
18320Sigor@sysoev.ru     /*
18420Sigor@sysoev.ru      * Connection memory pool chunk size is tunned to
18520Sigor@sysoev.ru      * allocate the most data in one mem_pool chunk.
18620Sigor@sysoev.ru      */
18720Sigor@sysoev.ru     ls->mem_pool_size = nxt_listen_socket_pool_min_size(ls)
18820Sigor@sysoev.ru                         + sizeof(nxt_event_conn_proxy_t)
18920Sigor@sysoev.ru                         + sizeof(nxt_event_conn_t)
19020Sigor@sysoev.ru                         + 4 * sizeof(nxt_buf_t);
19120Sigor@sysoev.ru 
19220Sigor@sysoev.ru     if (nxt_listen_socket_create(task, ls, 0) != NXT_OK) {
19320Sigor@sysoev.ru         return NXT_ERROR;
19420Sigor@sysoev.ru     }
19520Sigor@sysoev.ru 
19620Sigor@sysoev.ru     rt->controller_socket = ls;
19720Sigor@sysoev.ru 
19820Sigor@sysoev.ru     return NXT_OK;
19920Sigor@sysoev.ru }
20020Sigor@sysoev.ru 
20120Sigor@sysoev.ru 
20220Sigor@sysoev.ru static void
20320Sigor@sysoev.ru nxt_controller_conn_init(nxt_task_t *task, void *obj, void *data)
20420Sigor@sysoev.ru {
20527Svbart@nginx.com     nxt_buf_t                 *b;
20627Svbart@nginx.com     nxt_event_conn_t          *c;
20727Svbart@nginx.com     nxt_event_engine_t        *engine;
20827Svbart@nginx.com     nxt_controller_request_t  *r;
20920Sigor@sysoev.ru 
21020Sigor@sysoev.ru     c = obj;
21120Sigor@sysoev.ru 
21220Sigor@sysoev.ru     nxt_debug(task, "controller conn init fd:%d", c->socket.fd);
21320Sigor@sysoev.ru 
21427Svbart@nginx.com     r = nxt_mem_zalloc(c->mem_pool, sizeof(nxt_controller_request_t));
21527Svbart@nginx.com     if (nxt_slow_path(r == NULL)) {
21627Svbart@nginx.com         nxt_controller_conn_free(task, c, NULL);
21727Svbart@nginx.com         return;
21827Svbart@nginx.com     }
21927Svbart@nginx.com 
22027Svbart@nginx.com     r->parser.hash = nxt_controller_request_fields_hash;
22127Svbart@nginx.com     r->parser.ctx = r;
22227Svbart@nginx.com 
22320Sigor@sysoev.ru     b = nxt_buf_mem_alloc(c->mem_pool, 1024, 0);
22420Sigor@sysoev.ru     if (nxt_slow_path(b == NULL)) {
22520Sigor@sysoev.ru         nxt_controller_conn_free(task, c, NULL);
22620Sigor@sysoev.ru         return;
22720Sigor@sysoev.ru     }
22820Sigor@sysoev.ru 
22920Sigor@sysoev.ru     c->read = b;
23027Svbart@nginx.com     c->socket.data = r;
23120Sigor@sysoev.ru     c->socket.read_ready = 1;
23220Sigor@sysoev.ru     c->read_state = &nxt_controller_conn_read_state;
23320Sigor@sysoev.ru 
23420Sigor@sysoev.ru     engine = task->thread->engine;
23520Sigor@sysoev.ru     c->read_work_queue = &engine->read_work_queue;
23627Svbart@nginx.com     c->write_work_queue = &engine->write_work_queue;
23720Sigor@sysoev.ru 
23820Sigor@sysoev.ru     nxt_event_conn_read(engine, c);
23920Sigor@sysoev.ru }
24020Sigor@sysoev.ru 
24120Sigor@sysoev.ru 
24220Sigor@sysoev.ru static const nxt_event_conn_state_t  nxt_controller_conn_read_state
24320Sigor@sysoev.ru     nxt_aligned(64) =
24420Sigor@sysoev.ru {
24520Sigor@sysoev.ru     NXT_EVENT_NO_BUF_PROCESS,
24620Sigor@sysoev.ru     NXT_EVENT_TIMER_NO_AUTORESET,
24720Sigor@sysoev.ru 
24820Sigor@sysoev.ru     nxt_controller_conn_read,
24920Sigor@sysoev.ru     nxt_controller_conn_close,
25020Sigor@sysoev.ru     nxt_controller_conn_read_error,
25120Sigor@sysoev.ru 
25220Sigor@sysoev.ru     nxt_controller_conn_read_timeout,
25320Sigor@sysoev.ru     nxt_controller_conn_timeout_value,
25420Sigor@sysoev.ru     60 * 1000,
25520Sigor@sysoev.ru };
25620Sigor@sysoev.ru 
25720Sigor@sysoev.ru 
25820Sigor@sysoev.ru static void
25920Sigor@sysoev.ru nxt_controller_conn_read(nxt_task_t *task, void *obj, void *data)
26020Sigor@sysoev.ru {
26127Svbart@nginx.com     size_t                    preread;
26227Svbart@nginx.com     nxt_buf_t                 *b;
26327Svbart@nginx.com     nxt_int_t                 rc;
26427Svbart@nginx.com     nxt_event_conn_t          *c;
26527Svbart@nginx.com     nxt_controller_request_t  *r;
26620Sigor@sysoev.ru 
26720Sigor@sysoev.ru     c = obj;
26827Svbart@nginx.com     r = data;
26920Sigor@sysoev.ru 
27020Sigor@sysoev.ru     nxt_debug(task, "controller conn read");
27120Sigor@sysoev.ru 
27227Svbart@nginx.com     nxt_queue_remove(&c->link);
27327Svbart@nginx.com     nxt_queue_self(&c->link);
27427Svbart@nginx.com 
27527Svbart@nginx.com     b = c->read;
27627Svbart@nginx.com 
27727Svbart@nginx.com     rc = nxt_http_parse_request(&r->parser, &b->mem);
27827Svbart@nginx.com 
27927Svbart@nginx.com     if (nxt_slow_path(rc != NXT_DONE)) {
28027Svbart@nginx.com 
28127Svbart@nginx.com         if (rc == NXT_AGAIN) {
28227Svbart@nginx.com             if (nxt_buf_mem_free_size(&b->mem) == 0) {
28327Svbart@nginx.com                 nxt_log(task, NXT_LOG_ERR, "too long request headers");
28427Svbart@nginx.com                 nxt_controller_conn_close(task, c, r);
28527Svbart@nginx.com                 return;
28627Svbart@nginx.com             }
28727Svbart@nginx.com 
28827Svbart@nginx.com             nxt_event_conn_read(task->thread->engine, c);
28927Svbart@nginx.com             return;
29027Svbart@nginx.com         }
29127Svbart@nginx.com 
29227Svbart@nginx.com         /* rc == NXT_ERROR */
29327Svbart@nginx.com 
29427Svbart@nginx.com         nxt_log(task, NXT_LOG_ERR, "parsing error");
29527Svbart@nginx.com 
29627Svbart@nginx.com         nxt_controller_conn_close(task, c, r);
29727Svbart@nginx.com         return;
29827Svbart@nginx.com     }
29927Svbart@nginx.com 
30027Svbart@nginx.com     preread = nxt_buf_mem_used_size(&b->mem);
30127Svbart@nginx.com 
30227Svbart@nginx.com     nxt_debug(task, "controller request header parsing complete, "
30327Svbart@nginx.com                     "body length: %O, preread: %uz",
30427Svbart@nginx.com                     r->length, preread);
30527Svbart@nginx.com 
30627Svbart@nginx.com     if (preread >= r->length) {
30727Svbart@nginx.com         nxt_controller_process_request(task, c, r);
30827Svbart@nginx.com         return;
30927Svbart@nginx.com     }
31027Svbart@nginx.com 
31127Svbart@nginx.com     if (r->length - preread > (size_t) nxt_buf_mem_free_size(&b->mem)) {
31227Svbart@nginx.com         b = nxt_buf_mem_alloc(c->mem_pool, r->length, 0);
31327Svbart@nginx.com         if (nxt_slow_path(b == NULL)) {
31427Svbart@nginx.com             nxt_controller_conn_free(task, c, NULL);
31527Svbart@nginx.com             return;
31627Svbart@nginx.com         }
31727Svbart@nginx.com 
31827Svbart@nginx.com         b->mem.free = nxt_cpymem(b->mem.free, c->read->mem.pos, preread);
31927Svbart@nginx.com 
32027Svbart@nginx.com         c->read = b;
32127Svbart@nginx.com     }
32227Svbart@nginx.com 
32327Svbart@nginx.com     c->read_state = &nxt_controller_conn_body_read_state;
32427Svbart@nginx.com 
32527Svbart@nginx.com     nxt_event_conn_read(task->thread->engine, c);
32620Sigor@sysoev.ru }
32720Sigor@sysoev.ru 
32820Sigor@sysoev.ru 
32920Sigor@sysoev.ru static nxt_msec_t
33020Sigor@sysoev.ru nxt_controller_conn_timeout_value(nxt_event_conn_t *c, uintptr_t data)
33120Sigor@sysoev.ru {
33220Sigor@sysoev.ru     return (nxt_msec_t) data;
33320Sigor@sysoev.ru }
33420Sigor@sysoev.ru 
33520Sigor@sysoev.ru 
33620Sigor@sysoev.ru static void
33720Sigor@sysoev.ru nxt_controller_conn_read_error(nxt_task_t *task, void *obj, void *data)
33820Sigor@sysoev.ru {
33920Sigor@sysoev.ru     nxt_event_conn_t  *c;
34020Sigor@sysoev.ru 
34120Sigor@sysoev.ru     c = obj;
34220Sigor@sysoev.ru 
34320Sigor@sysoev.ru     nxt_debug(task, "controller conn read error");
34420Sigor@sysoev.ru 
34527Svbart@nginx.com     nxt_controller_conn_close(task, c, data);
34620Sigor@sysoev.ru }
34720Sigor@sysoev.ru 
34820Sigor@sysoev.ru 
34920Sigor@sysoev.ru static void
35020Sigor@sysoev.ru nxt_controller_conn_read_timeout(nxt_task_t *task, void *obj, void *data)
35120Sigor@sysoev.ru {
35220Sigor@sysoev.ru     nxt_timer_t       *ev;
35320Sigor@sysoev.ru     nxt_event_conn_t  *c;
35420Sigor@sysoev.ru 
35520Sigor@sysoev.ru     ev = obj;
35620Sigor@sysoev.ru 
35720Sigor@sysoev.ru     c = nxt_event_read_timer_conn(ev);
35820Sigor@sysoev.ru     c->socket.timedout = 1;
35920Sigor@sysoev.ru     c->socket.closed = 1;
36020Sigor@sysoev.ru 
36120Sigor@sysoev.ru     nxt_debug(task, "controller conn read timeout");
36220Sigor@sysoev.ru 
36327Svbart@nginx.com     nxt_controller_conn_close(task, c, data);
36427Svbart@nginx.com }
36527Svbart@nginx.com 
36627Svbart@nginx.com 
36727Svbart@nginx.com static const nxt_event_conn_state_t  nxt_controller_conn_body_read_state
36827Svbart@nginx.com     nxt_aligned(64) =
36927Svbart@nginx.com {
37027Svbart@nginx.com     NXT_EVENT_NO_BUF_PROCESS,
37127Svbart@nginx.com     NXT_EVENT_TIMER_AUTORESET,
37227Svbart@nginx.com 
37327Svbart@nginx.com     nxt_controller_conn_body_read,
37427Svbart@nginx.com     nxt_controller_conn_close,
37527Svbart@nginx.com     nxt_controller_conn_read_error,
37627Svbart@nginx.com 
37727Svbart@nginx.com     nxt_controller_conn_read_timeout,
37827Svbart@nginx.com     nxt_controller_conn_timeout_value,
37927Svbart@nginx.com     60 * 1000,
38027Svbart@nginx.com };
38127Svbart@nginx.com 
38227Svbart@nginx.com 
38327Svbart@nginx.com static void
38427Svbart@nginx.com nxt_controller_conn_body_read(nxt_task_t *task, void *obj, void *data)
38527Svbart@nginx.com {
38627Svbart@nginx.com     size_t            rest;
38727Svbart@nginx.com     nxt_buf_t         *b;
38827Svbart@nginx.com     nxt_event_conn_t  *c;
38927Svbart@nginx.com 
39027Svbart@nginx.com     c = obj;
39127Svbart@nginx.com 
39227Svbart@nginx.com     nxt_debug(task, "controller conn body read");
39327Svbart@nginx.com 
39427Svbart@nginx.com     b = c->read;
39527Svbart@nginx.com 
39627Svbart@nginx.com     rest = nxt_buf_mem_free_size(&b->mem);
39727Svbart@nginx.com 
39827Svbart@nginx.com     if (rest == 0) {
39927Svbart@nginx.com         nxt_debug(task, "controller conn body read complete");
40027Svbart@nginx.com 
40127Svbart@nginx.com         nxt_controller_process_request(task, c, data);
40227Svbart@nginx.com         return;
40327Svbart@nginx.com     }
40427Svbart@nginx.com 
40527Svbart@nginx.com     nxt_debug(task, "controller conn body read again, rest: %uz", rest);
40627Svbart@nginx.com 
40727Svbart@nginx.com     nxt_event_conn_read(task->thread->engine, c);
40827Svbart@nginx.com }
40927Svbart@nginx.com 
41027Svbart@nginx.com 
41127Svbart@nginx.com static const nxt_event_conn_state_t  nxt_controller_conn_write_state
41227Svbart@nginx.com     nxt_aligned(64) =
41327Svbart@nginx.com {
41427Svbart@nginx.com     NXT_EVENT_NO_BUF_PROCESS,
41527Svbart@nginx.com     NXT_EVENT_TIMER_AUTORESET,
41627Svbart@nginx.com 
41727Svbart@nginx.com     nxt_controller_conn_write,
41827Svbart@nginx.com     NULL,
41927Svbart@nginx.com     nxt_controller_conn_write_error,
42027Svbart@nginx.com 
42127Svbart@nginx.com     nxt_controller_conn_write_timeout,
42227Svbart@nginx.com     nxt_controller_conn_timeout_value,
42327Svbart@nginx.com     60 * 1000,
42427Svbart@nginx.com };
42527Svbart@nginx.com 
42627Svbart@nginx.com 
42727Svbart@nginx.com static void
42827Svbart@nginx.com nxt_controller_conn_write(nxt_task_t *task, void *obj, void *data)
42927Svbart@nginx.com {
43027Svbart@nginx.com     nxt_buf_t         *b;
43127Svbart@nginx.com     nxt_event_conn_t  *c;
43227Svbart@nginx.com 
43327Svbart@nginx.com     c = obj;
43427Svbart@nginx.com 
43527Svbart@nginx.com     nxt_debug(task, "controller conn write");
43627Svbart@nginx.com 
43727Svbart@nginx.com     b = c->write;
43827Svbart@nginx.com 
43927Svbart@nginx.com     if (b->mem.pos != b->mem.free) {
44027Svbart@nginx.com         nxt_event_conn_write(task->thread->engine, c);
44127Svbart@nginx.com         return;
44227Svbart@nginx.com     }
44327Svbart@nginx.com 
44427Svbart@nginx.com     nxt_debug(task, "controller conn write complete");
44527Svbart@nginx.com 
44627Svbart@nginx.com     nxt_controller_conn_close(task, c, data);
44727Svbart@nginx.com }
44827Svbart@nginx.com 
44927Svbart@nginx.com 
45027Svbart@nginx.com static void
45127Svbart@nginx.com nxt_controller_conn_write_error(nxt_task_t *task, void *obj, void *data)
45227Svbart@nginx.com {
45327Svbart@nginx.com     nxt_event_conn_t  *c;
45427Svbart@nginx.com 
45527Svbart@nginx.com     c = obj;
45627Svbart@nginx.com 
45727Svbart@nginx.com     nxt_debug(task, "controller conn write error");
45827Svbart@nginx.com 
45927Svbart@nginx.com     nxt_controller_conn_close(task, c, data);
46027Svbart@nginx.com }
46127Svbart@nginx.com 
46227Svbart@nginx.com 
46327Svbart@nginx.com static void
46427Svbart@nginx.com nxt_controller_conn_write_timeout(nxt_task_t *task, void *obj, void *data)
46527Svbart@nginx.com {
46627Svbart@nginx.com     nxt_timer_t       *ev;
46727Svbart@nginx.com     nxt_event_conn_t  *c;
46827Svbart@nginx.com 
46927Svbart@nginx.com     ev = obj;
47027Svbart@nginx.com 
47127Svbart@nginx.com     c = nxt_event_write_timer_conn(ev);
47227Svbart@nginx.com     c->socket.timedout = 1;
47327Svbart@nginx.com     c->socket.closed = 1;
47427Svbart@nginx.com 
47527Svbart@nginx.com     nxt_debug(task, "controller conn write timeout");
47627Svbart@nginx.com 
47727Svbart@nginx.com     nxt_controller_conn_close(task, c, data);
47820Sigor@sysoev.ru }
47920Sigor@sysoev.ru 
48020Sigor@sysoev.ru 
48120Sigor@sysoev.ru static const nxt_event_conn_state_t  nxt_controller_conn_close_state
48220Sigor@sysoev.ru     nxt_aligned(64) =
48320Sigor@sysoev.ru {
48420Sigor@sysoev.ru     NXT_EVENT_NO_BUF_PROCESS,
48520Sigor@sysoev.ru     NXT_EVENT_TIMER_NO_AUTORESET,
48620Sigor@sysoev.ru 
48720Sigor@sysoev.ru     nxt_controller_conn_free,
48820Sigor@sysoev.ru     NULL,
48920Sigor@sysoev.ru     NULL,
49020Sigor@sysoev.ru 
49120Sigor@sysoev.ru     NULL,
49220Sigor@sysoev.ru     NULL,
49320Sigor@sysoev.ru     0,
49420Sigor@sysoev.ru };
49520Sigor@sysoev.ru 
49620Sigor@sysoev.ru 
49720Sigor@sysoev.ru static void
49820Sigor@sysoev.ru nxt_controller_conn_close(nxt_task_t *task, void *obj, void *data)
49920Sigor@sysoev.ru {
50020Sigor@sysoev.ru     nxt_event_conn_t  *c;
50120Sigor@sysoev.ru 
50220Sigor@sysoev.ru     c = obj;
50320Sigor@sysoev.ru 
50420Sigor@sysoev.ru     nxt_debug(task, "controller conn close");
50520Sigor@sysoev.ru 
50627Svbart@nginx.com     nxt_queue_remove(&c->link);
50727Svbart@nginx.com 
50820Sigor@sysoev.ru     c->write_state = &nxt_controller_conn_close_state;
50920Sigor@sysoev.ru 
51020Sigor@sysoev.ru     nxt_event_conn_close(task->thread->engine, c);
51120Sigor@sysoev.ru }
51220Sigor@sysoev.ru 
51320Sigor@sysoev.ru 
51420Sigor@sysoev.ru static void
51520Sigor@sysoev.ru nxt_controller_conn_free(nxt_task_t *task, void *obj, void *data)
51620Sigor@sysoev.ru {
51720Sigor@sysoev.ru     nxt_event_conn_t  *c;
51820Sigor@sysoev.ru 
51920Sigor@sysoev.ru     c = obj;
52020Sigor@sysoev.ru 
52120Sigor@sysoev.ru     nxt_debug(task, "controller conn free");
52220Sigor@sysoev.ru 
52320Sigor@sysoev.ru     nxt_mem_pool_destroy(c->mem_pool);
52420Sigor@sysoev.ru 
52520Sigor@sysoev.ru     //nxt_free(c);
52620Sigor@sysoev.ru }
52727Svbart@nginx.com 
52827Svbart@nginx.com 
52927Svbart@nginx.com static nxt_int_t
53027Svbart@nginx.com nxt_controller_request_content_length(void *ctx, nxt_str_t *name,
53127Svbart@nginx.com     nxt_str_t *value, uintptr_t data)
53227Svbart@nginx.com {
53327Svbart@nginx.com     off_t                     length;
53427Svbart@nginx.com     nxt_controller_request_t  *r;
53527Svbart@nginx.com 
53627Svbart@nginx.com     r = ctx;
53727Svbart@nginx.com 
53827Svbart@nginx.com     length = nxt_off_t_parse(value->start, value->length);
53927Svbart@nginx.com 
54027Svbart@nginx.com     if (nxt_fast_path(length > 0)) {
54127Svbart@nginx.com         /* TODO length too big */
54227Svbart@nginx.com 
54327Svbart@nginx.com         r->length = length;
54427Svbart@nginx.com         return NXT_OK;
54527Svbart@nginx.com     }
54627Svbart@nginx.com 
54727Svbart@nginx.com     /* TODO logging (task?) */
54827Svbart@nginx.com 
54927Svbart@nginx.com     return NXT_ERROR;
55027Svbart@nginx.com }
55127Svbart@nginx.com 
55227Svbart@nginx.com 
55327Svbart@nginx.com static void
55427Svbart@nginx.com nxt_controller_process_request(nxt_task_t *task, nxt_event_conn_t *c,
55544Svbart@nginx.com     nxt_controller_request_t *req)
55627Svbart@nginx.com {
55744Svbart@nginx.com     nxt_controller_response_t  resp;
55844Svbart@nginx.com 
55944Svbart@nginx.com     nxt_memzero(&resp, sizeof(nxt_controller_response_t));
56044Svbart@nginx.com 
56144Svbart@nginx.com     if (nxt_str_eq(&req->parser.method, "GET", 3)) {
56244Svbart@nginx.com         nxt_str_set(&resp.status_line, "200 OK");
56344Svbart@nginx.com         resp.json_value = nxt_controller_conf.root;
56444Svbart@nginx.com 
56544Svbart@nginx.com     } else if (nxt_str_eq(&req->parser.method, "PUT", 3)) {
56644Svbart@nginx.com 
56744Svbart@nginx.com         if (nxt_controller_request_body_parse(task, c, req) == NXT_OK) {
56827Svbart@nginx.com 
56944Svbart@nginx.com             nxt_mem_pool_destroy(nxt_controller_conf.pool);
57044Svbart@nginx.com             nxt_controller_conf = req->conf;
57144Svbart@nginx.com 
57244Svbart@nginx.com             nxt_str_set(&resp.status_line, "201 Created");
57344Svbart@nginx.com             nxt_str_set(&resp.json_string,
57444Svbart@nginx.com                         "{ \"success\": \"Configuration updated\" }");
57527Svbart@nginx.com 
57644Svbart@nginx.com         } else {
57744Svbart@nginx.com             nxt_str_set(&resp.status_line, "400 Bad Request");
57844Svbart@nginx.com             nxt_str_set(&resp.json_string,
57944Svbart@nginx.com                         "{ \"error\": \"Invalid JSON\" }");
58044Svbart@nginx.com         }
58144Svbart@nginx.com 
58244Svbart@nginx.com     } else {
58344Svbart@nginx.com         nxt_str_set(&resp.status_line, "405 Method Not Allowed");
58444Svbart@nginx.com         nxt_str_set(&resp.json_string, "{ \"error\": \"Invalid method\" }");
58529Svbart@nginx.com     }
58627Svbart@nginx.com 
58744Svbart@nginx.com     if (nxt_controller_response(task, c, &resp) != NXT_OK) {
58844Svbart@nginx.com         nxt_controller_conn_close(task, c, req);
58927Svbart@nginx.com     }
59027Svbart@nginx.com }
59127Svbart@nginx.com 
59227Svbart@nginx.com 
59327Svbart@nginx.com static nxt_int_t
59433Svbart@nginx.com nxt_controller_request_body_parse(nxt_task_t *task, nxt_event_conn_t *c,
59533Svbart@nginx.com     nxt_controller_request_t *r)
59627Svbart@nginx.com {
59744Svbart@nginx.com     nxt_buf_mem_t          *mbuf;
59844Svbart@nginx.com     nxt_mem_pool_t         *mp;
59929Svbart@nginx.com     nxt_conf_json_value_t  *value;
60029Svbart@nginx.com 
60144Svbart@nginx.com     mp = nxt_mem_pool_create(512);
60229Svbart@nginx.com 
60344Svbart@nginx.com     if (nxt_slow_path(mp == NULL)) {
60444Svbart@nginx.com         return NXT_ERROR;
60544Svbart@nginx.com     }
60644Svbart@nginx.com 
60744Svbart@nginx.com     mbuf = &c->read->mem;
60844Svbart@nginx.com 
60944Svbart@nginx.com     value = nxt_conf_json_parse(mbuf->pos, mbuf->free - mbuf->pos, mp);
61029Svbart@nginx.com 
61129Svbart@nginx.com     if (value == NULL) {
61229Svbart@nginx.com         return NXT_ERROR;
61329Svbart@nginx.com     }
61429Svbart@nginx.com 
61544Svbart@nginx.com     r->conf.root = value;
61644Svbart@nginx.com     r->conf.pool = mp;
61733Svbart@nginx.com 
61827Svbart@nginx.com     return NXT_OK;
61927Svbart@nginx.com }
62033Svbart@nginx.com 
62133Svbart@nginx.com 
62244Svbart@nginx.com static nxt_int_t
62344Svbart@nginx.com nxt_controller_response(nxt_task_t *task, nxt_event_conn_t *c,
62444Svbart@nginx.com     nxt_controller_response_t *resp)
62533Svbart@nginx.com {
62644Svbart@nginx.com     size_t     size;
62733Svbart@nginx.com     nxt_buf_t  *b;
62833Svbart@nginx.com 
629*45Svbart@nginx.com     size = sizeof("HTTP/1.0 " "\r\n\r\n") - 1 + resp->status_line.length;
63033Svbart@nginx.com 
63144Svbart@nginx.com     b = nxt_buf_mem_alloc(c->mem_pool, size, 0);
63233Svbart@nginx.com     if (nxt_slow_path(b == NULL)) {
63344Svbart@nginx.com         return NXT_ERROR;
63433Svbart@nginx.com     }
63533Svbart@nginx.com 
63644Svbart@nginx.com     b->mem.free = nxt_cpymem(b->mem.free, "HTTP/1.0 ", sizeof("HTTP/1.0 ") - 1);
63744Svbart@nginx.com     b->mem.free = nxt_cpymem(b->mem.free, resp->status_line.start,
63844Svbart@nginx.com                              resp->status_line.length);
63944Svbart@nginx.com 
64044Svbart@nginx.com     b->mem.free = nxt_cpymem(b->mem.free, "\r\n\r\n", sizeof("\r\n\r\n") - 1);
64144Svbart@nginx.com 
642*45Svbart@nginx.com     b->next = nxt_controller_response_body(resp, c->mem_pool);
643*45Svbart@nginx.com 
644*45Svbart@nginx.com     if (nxt_slow_path(b->next == NULL)) {
645*45Svbart@nginx.com         return NXT_ERROR;
64644Svbart@nginx.com     }
64733Svbart@nginx.com 
64833Svbart@nginx.com     c->write = b;
64944Svbart@nginx.com     c->write_state = &nxt_controller_conn_write_state;
65033Svbart@nginx.com 
65144Svbart@nginx.com     nxt_event_conn_write(task->thread->engine, c);
65233Svbart@nginx.com 
65344Svbart@nginx.com     return NXT_OK;
65433Svbart@nginx.com }
655*45Svbart@nginx.com 
656*45Svbart@nginx.com 
657*45Svbart@nginx.com static nxt_buf_t *
658*45Svbart@nginx.com nxt_controller_response_body(nxt_controller_response_t *resp,
659*45Svbart@nginx.com     nxt_mem_pool_t *pool)
660*45Svbart@nginx.com {
661*45Svbart@nginx.com     size_t                  size;
662*45Svbart@nginx.com     nxt_buf_t               *b;
663*45Svbart@nginx.com     nxt_conf_json_value_t   *value;
664*45Svbart@nginx.com     nxt_conf_json_pretty_t  pretty;
665*45Svbart@nginx.com 
666*45Svbart@nginx.com     if (resp->json_value) {
667*45Svbart@nginx.com         value = resp->json_value;
668*45Svbart@nginx.com 
669*45Svbart@nginx.com     } else {
670*45Svbart@nginx.com         value = nxt_conf_json_parse(resp->json_string.start,
671*45Svbart@nginx.com                                     resp->json_string.length, pool);
672*45Svbart@nginx.com 
673*45Svbart@nginx.com         if (nxt_slow_path(value == NULL)) {
674*45Svbart@nginx.com             return NULL;
675*45Svbart@nginx.com         }
676*45Svbart@nginx.com     }
677*45Svbart@nginx.com 
678*45Svbart@nginx.com     nxt_memzero(&pretty, sizeof(nxt_conf_json_pretty_t));
679*45Svbart@nginx.com 
680*45Svbart@nginx.com     size = nxt_conf_json_print_value(NULL, value, &pretty) + 2;
681*45Svbart@nginx.com 
682*45Svbart@nginx.com     b = nxt_buf_mem_alloc(pool, size, 0);
683*45Svbart@nginx.com     if (nxt_slow_path(b == NULL)) {
684*45Svbart@nginx.com         return NULL;
685*45Svbart@nginx.com     }
686*45Svbart@nginx.com 
687*45Svbart@nginx.com     nxt_memzero(&pretty, sizeof(nxt_conf_json_pretty_t));
688*45Svbart@nginx.com 
689*45Svbart@nginx.com     b->mem.free = (u_char *) nxt_conf_json_print_value(b->mem.free, value,
690*45Svbart@nginx.com                                                        &pretty);
691*45Svbart@nginx.com 
692*45Svbart@nginx.com     *b->mem.free++ = '\r';
693*45Svbart@nginx.com     *b->mem.free++ = '\n';
694*45Svbart@nginx.com 
695*45Svbart@nginx.com     return b;
696*45Svbart@nginx.com }
697