X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=src%2Fvnet%2Fsession-apps%2Fhttp_server.c;h=4547a4dc4ef88d0563358ccdf6b467407c863c2a;hb=c5df8c71c;hp=f702b741f95757a326109af045a97e22f714c83f;hpb=f6c4313b6aa7746fe97afd398ce68c2efbef0600;p=vpp.git diff --git a/src/vnet/session-apps/http_server.c b/src/vnet/session-apps/http_server.c index f702b741f95..4547a4dc4ef 100644 --- a/src/vnet/session-apps/http_server.c +++ b/src/vnet/session-apps/http_server.c @@ -1,5 +1,5 @@ /* -* Copyright (c) 2015-2019 Cisco and/or its affiliates. +* Copyright (c) 2017-2019 Cisco and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: @@ -17,6 +17,7 @@ #include #include #include +#include typedef enum { @@ -46,6 +47,7 @@ typedef struct u8 *rx_buf; u32 vpp_session_index; u64 vpp_session_handle; + u32 timer_handle; } http_session_t; typedef struct @@ -71,6 +73,9 @@ typedef struct /* process node index for evnt scheduling */ u32 node_index; + tw_timer_wheel_2t_1w_2048sl_t tw; + clib_spinlock_t tw_lock; + u32 prealloc_fifos; u32 private_segment_size; u32 fifo_size; @@ -105,6 +110,37 @@ http_server_sessions_writer_unlock (void) clib_rwlock_writer_unlock (&http_server_main.sessions_lock); } +static http_session_t * +http_server_session_alloc (u32 thread_index) +{ + http_server_main_t *hsm = &http_server_main; + http_session_t *hs; + pool_get (hsm->sessions[thread_index], hs); + memset (hs, 0, sizeof (*hs)); + hs->session_index = hs - hsm->sessions[thread_index]; + hs->thread_index = thread_index; + hs->timer_handle = ~0; + return hs; +} + +static http_session_t * +http_server_session_get (u32 thread_index, u32 hs_index) +{ + http_server_main_t *hsm = &http_server_main; + if (pool_is_free_index (hsm->sessions[thread_index], hs_index)) + return 0; + return pool_elt_at_index (hsm->sessions[thread_index], hs_index); +} + +static void +http_server_session_free (http_session_t * hs) +{ + http_server_main_t *hsm = &http_server_main; + pool_put (hsm->sessions[hs->thread_index], hs); + if (CLIB_DEBUG) + memset (hs, 0xfa, sizeof (*hs)); +} + static void http_server_session_lookup_add (u32 thread_index, u32 s_index, u32 hs_index) { @@ -129,38 +165,31 @@ http_server_session_lookup (u32 thread_index, u32 s_index) if (s_index < vec_len (hsm->session_to_http_session[thread_index])) { hs_index = hsm->session_to_http_session[thread_index][s_index]; - if (hs_index < vec_len (hsm->sessions[thread_index])) - return &hsm->sessions[thread_index][hs_index]; + return http_server_session_get (thread_index, hs_index); } return 0; } -static http_session_t * -http_server_session_alloc (u32 thread_index) -{ - http_server_main_t *hsm = &http_server_main; - http_session_t *hs; - pool_get (hsm->sessions[thread_index], hs); - memset (hs, 0, sizeof (*hs)); - hs->session_index = hs - hsm->sessions[thread_index]; - hs->thread_index = thread_index; - return hs; -} -static http_session_t * -http_server_session_get (u32 thread_index, u32 hs_index) -{ - http_server_main_t *hsm = &http_server_main; - return pool_elt_at_index (hsm->sessions[thread_index], hs_index); +static void +http_server_session_timer_start (http_session_t * hs) +{ + u32 hs_handle; + hs_handle = hs->thread_index << 24 | hs->session_index; + clib_spinlock_lock (&http_server_main.tw_lock); + hs->timer_handle = tw_timer_start_2t_1w_2048sl (&http_server_main.tw, + hs_handle, 0, 60); + clib_spinlock_unlock (&http_server_main.tw_lock); } static void -http_server_session_free (http_session_t * hs) +http_server_session_timer_stop (http_session_t * hs) { - http_server_main_t *hsm = &http_server_main; - pool_put (hsm->sessions[hs->thread_index], hs); - if (CLIB_DEBUG) - memset (hs, 0xfa, sizeof (*hs)); + if (hs->timer_handle == ~0) + return; + clib_spinlock_lock (&http_server_main.tw_lock); + tw_timer_stop_2t_1w_2048sl (&http_server_main.tw, hs->timer_handle); + clib_spinlock_unlock (&http_server_main.tw_lock); } static void @@ -170,9 +199,19 @@ http_server_session_cleanup (http_session_t * hs) return; http_server_session_lookup_del (hs->thread_index, hs->vpp_session_index); vec_free (hs->rx_buf); + http_server_session_timer_stop (hs); http_server_session_free (hs); } +static void +http_server_session_disconnect (http_session_t * hs) +{ + vnet_disconnect_args_t _a = { 0 }, *a = &_a; + a->handle = hs->vpp_session_handle; + a->app_index = http_server_main.app_index; + vnet_disconnect_session (a); +} + static void http_process_free (http_server_args * args) { @@ -277,7 +316,10 @@ send_data (http_session_t * hs, u8 * data) /* Made any progress? */ if (actual_transfer <= 0) { + http_server_sessions_reader_unlock (); vlib_process_suspend (vm, delay); + http_server_sessions_reader_lock (); + /* 10s deadman timer */ if (vlib_time_now (vm) > last_sent_timer + 10.0) { @@ -442,7 +484,8 @@ alloc_http_process (http_server_args * args) /* Save the args (pointer) in the node runtime */ save_args = vlib_node_get_runtime_data (vm, n->index); - *save_args = args; + *save_args = clib_mem_alloc (sizeof (*args)); + clib_memcpy_fast (*save_args, args, sizeof (*args)); vlib_start_process (vm, n->runtime_index); } @@ -478,7 +521,7 @@ session_rx_request (http_session_t * hs) static int http_server_rx_callback (session_t * s) { - http_server_args *args; + http_server_args args; http_session_t *hs; int rv; @@ -493,25 +536,23 @@ http_server_rx_callback (session_t * s) return rv; /* send the command to a new/recycled vlib process */ - args = clib_mem_alloc (sizeof (*args)); - args->hs_index = hs->session_index; - args->thread_index = hs->thread_index; + args.hs_index = hs->session_index; + args.thread_index = hs->thread_index; http_server_sessions_reader_unlock (); - /* Send an RPC request via the thread-0 input node */ + /* Send RPC request to main thread */ if (vlib_get_thread_index () != 0) - session_send_rpc_evt_to_thread (0, alloc_http_process_callback, args); + vlib_rpc_call_main_thread (alloc_http_process_callback, (u8 *) & args, + sizeof (args)); else - alloc_http_process (args); + alloc_http_process (&args); return 0; } static int http_server_rx_callback_static (session_t * s) { - http_server_main_t *hsm = &http_server_main; - vnet_disconnect_args_t _a = { 0 }, *a = &_a; http_session_t *hs; u32 request_len; u8 *request = 0; @@ -561,12 +602,10 @@ find_end: send_data: send_data (hs, static_http); - http_server_session_cleanup (hs); close_session: - a->handle = session_handle (s); - a->app_index = hsm->app_index; - vnet_disconnect_session (a); + http_server_session_disconnect (hs); + http_server_session_cleanup (hs); return 0; postpone: @@ -585,7 +624,7 @@ http_server_session_accept_callback (session_t * s) http_session_t *hs; hsm->vpp_queue[s->thread_index] = - session_manager_get_vpp_event_queue (s->thread_index); + session_main_get_vpp_event_queue (s->thread_index); if (!hsm->is_static) http_server_sessions_writer_lock (); @@ -598,6 +637,7 @@ http_server_session_accept_callback (session_t * s) hs->vpp_session_index = s->session_index; hs->vpp_session_handle = session_handle (s); hs->session_state = HTTP_STATE_ESTABLISHED; + http_server_session_timer_start (hs); if (!hsm->is_static) http_server_sessions_writer_unlock (); @@ -737,12 +777,78 @@ http_server_listen () return vnet_bind_uri (a); } +static void +http_server_session_cleanup_cb (void *hs_handlep) +{ + http_session_t *hs; + uword hs_handle; + hs_handle = pointer_to_uword (hs_handlep); + hs = http_server_session_get (hs_handle >> 24, hs_handle & 0x00FFFFFF); + if (!hs) + return; + hs->timer_handle = ~0; + http_server_session_disconnect (hs); + http_server_session_cleanup (hs); +} + +static void +http_expired_timers_dispatch (u32 * expired_timers) +{ + u32 hs_handle; + int i; + + for (i = 0; i < vec_len (expired_timers); i++) + { + /* Get session handle. The first bit is the timer id */ + hs_handle = expired_timers[i] & 0x7FFFFFFF; + session_send_rpc_evt_to_thread (hs_handle >> 24, + http_server_session_cleanup_cb, + uword_to_pointer (hs_handle, void *)); + } +} + +static uword +http_server_process (vlib_main_t * vm, vlib_node_runtime_t * rt, + vlib_frame_t * f) +{ + http_server_main_t *hsm = &http_server_main; + f64 now, timeout = 1.0; + uword *event_data = 0; + uword __clib_unused event_type; + + while (1) + { + vlib_process_wait_for_event_or_clock (vm, timeout); + now = vlib_time_now (vm); + event_type = vlib_process_get_events (vm, (uword **) & event_data); + + /* expire timers */ + clib_spinlock_lock (&http_server_main.tw_lock); + tw_timer_expire_timers_2t_1w_2048sl (&hsm->tw, now); + clib_spinlock_unlock (&http_server_main.tw_lock); + + vec_reset_length (event_data); + } + return 0; +} + +/* *INDENT-OFF* */ +VLIB_REGISTER_NODE (http_server_process_node) = +{ + .function = http_server_process, + .type = VLIB_NODE_TYPE_PROCESS, + .name = "http-server-process", + .state = VLIB_NODE_STATE_DISABLED, +}; +/* *INDENT-ON* */ + static int http_server_create (vlib_main_t * vm) { vlib_thread_main_t *vtm = vlib_get_thread_main (); http_server_main_t *hsm = &http_server_main; u32 num_threads; + vlib_node_t *n; num_threads = 1 /* main thread */ + vtm->n_threads; vec_validate (hsm->vpp_queue, num_threads - 1); @@ -750,6 +856,7 @@ http_server_create (vlib_main_t * vm) vec_validate (hsm->session_to_http_session, num_threads - 1); clib_rwlock_init (&hsm->sessions_lock); + clib_spinlock_init (&hsm->tw_lock); if (http_server_attach ()) { @@ -761,6 +868,15 @@ http_server_create (vlib_main_t * vm) clib_warning ("failed to start listening"); return -1; } + + /* Init timer wheel and process */ + tw_timer_wheel_init_2t_1w_2048sl (&hsm->tw, http_expired_timers_dispatch, + 1 /* timer interval */ , ~0); + vlib_node_set_state (vm, http_server_process_node.index, + VLIB_NODE_STATE_POLLING); + n = vlib_get_node (vm, http_server_process_node.index); + vlib_start_process (vm, n->runtime_index); + return 0; } @@ -770,6 +886,7 @@ http_server_create_command_fn (vlib_main_t * vm, vlib_cli_command_t * cmd) { http_server_main_t *hsm = &http_server_main; + unformat_input_t _line_input, *line_input = &_line_input; u64 seg_size; u8 *html; int rv; @@ -778,13 +895,19 @@ http_server_create_command_fn (vlib_main_t * vm, hsm->private_segment_size = 0; hsm->fifo_size = 0; hsm->is_static = 0; - while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) + + /* Get a line of input. */ + if (!unformat_user (input, unformat_line_input, line_input)) + goto start_server; + + while (unformat_check_input (line_input) != UNFORMAT_END_OF_INPUT) { - if (unformat (input, "static")) + if (unformat (line_input, "static")) hsm->is_static = 1; - else if (unformat (input, "prealloc-fifos %d", &hsm->prealloc_fifos)) + else + if (unformat (line_input, "prealloc-fifos %d", &hsm->prealloc_fifos)) ; - else if (unformat (input, "private-segment-size %U", + else if (unformat (line_input, "private-segment-size %U", unformat_memory_size, &seg_size)) { if (seg_size >= 0x100000000ULL) @@ -795,14 +918,18 @@ http_server_create_command_fn (vlib_main_t * vm, } hsm->private_segment_size = seg_size; } - else if (unformat (input, "fifo-size %d", &hsm->fifo_size)) + else if (unformat (line_input, "fifo-size %d", &hsm->fifo_size)) hsm->fifo_size <<= 10; - else if (unformat (input, "uri %s", &hsm->uri)) + else if (unformat (line_input, "uri %s", &hsm->uri)) ; else return clib_error_return (0, "unknown input `%U'", - format_unformat_error, input); + format_unformat_error, line_input); } + unformat_free (line_input); + +start_server: + if (hsm->my_client_index != (u32) ~ 0) return clib_error_return (0, "test http server is already running");