1 #include "first.h" 2 3 #include <stdlib.h> 4 #include <string.h> 5 6 #include "gw_backend.h" 7 typedef gw_plugin_config plugin_config; 8 typedef gw_plugin_data plugin_data; 9 typedef gw_handler_ctx handler_ctx; 10 11 #include "base.h" 12 #include "array.h" 13 #include "buffer.h" 14 #include "log.h" 15 #include "status_counter.h" 16 17 /** 18 * 19 * socket proxy (with optional buffering) 20 * 21 */ 22 23 SETDEFAULTS_FUNC(mod_sockproxy_set_defaults) { 24 plugin_data *p = p_d; 25 const data_unset *du; 26 size_t i = 0; 27 28 config_values_t cv[] = { 29 { "sockproxy.server", NULL, T_CONFIG_LOCAL, T_CONFIG_SCOPE_CONNECTION }, /* 0 */ 30 { "sockproxy.debug", NULL, T_CONFIG_SHORT, T_CONFIG_SCOPE_CONNECTION }, /* 1 */ 31 { "sockproxy.balance", NULL, T_CONFIG_LOCAL, T_CONFIG_SCOPE_CONNECTION }, /* 2 */ 32 { NULL, NULL, T_CONFIG_UNSET, T_CONFIG_SCOPE_UNSET } 33 }; 34 35 p->config_storage = calloc(srv->config_context->used, sizeof(plugin_config *)); 36 force_assert(p->config_storage); 37 38 for (i = 0; i < srv->config_context->used; i++) { 39 data_config const* config = (data_config const*)srv->config_context->data[i]; 40 plugin_config *s; 41 42 s = calloc(1, sizeof(plugin_config)); 43 force_assert(s); 44 s->exts = NULL; 45 s->exts_auth = NULL; 46 s->exts_resp = NULL; 47 s->debug = 0; 48 49 cv[0].destination = NULL; /* T_CONFIG_LOCAL */ 50 cv[1].destination = &(s->debug); 51 cv[2].destination = NULL; /* T_CONFIG_LOCAL */ 52 53 p->config_storage[i] = s; 54 55 if (0 != config_insert_values_global(srv, config->value, cv, i == 0 ? T_CONFIG_SCOPE_SERVER : T_CONFIG_SCOPE_CONNECTION)) { 56 return HANDLER_ERROR; 57 } 58 59 du = array_get_element_klen(config->value, CONST_STR_LEN("sockproxy.server")); 60 if (!gw_set_defaults_backend(srv, (gw_plugin_data *)p, du, i, 0)) { 61 return HANDLER_ERROR; 62 } 63 64 du = array_get_element_klen(config->value, CONST_STR_LEN("sockproxy.balance")); 65 if (!gw_set_defaults_balance(srv, s, du)) { 66 return HANDLER_ERROR; 67 } 68 69 /* disable check-local for all exts (default enabled) */ 70 if (s->exts) { /*(check after gw_set_defaults_backend())*/ 71 for (size_t j = 0; j < s->exts->used; ++j) { 72 gw_extension *ex = s->exts->exts[j]; 73 for (size_t n = 0; n < ex->used; ++n) { 74 ex->hosts[n]->check_local = 0; 75 } 76 } 77 } 78 } 79 80 return HANDLER_GO_ON; 81 } 82 83 84 static handler_t sockproxy_create_env_connect(server *srv, handler_ctx *hctx) { 85 connection *con = hctx->remote_conn; 86 con->file_started = 1; 87 gw_set_transparent(srv, hctx); 88 http_response_upgrade_read_body_unknown(srv, con); 89 90 status_counter_inc(srv, CONST_STR_LEN("sockproxy.requests")); 91 return HANDLER_GO_ON; 92 } 93 94 95 #define PATCH(x) \ 96 p->conf.x = s->x; 97 static int mod_sockproxy_patch_connection(server *srv, connection *con, plugin_data *p) { 98 size_t i, j; 99 plugin_config *s = p->config_storage[0]; 100 101 PATCH(exts); 102 PATCH(exts_auth); 103 PATCH(exts_resp); 104 PATCH(debug); 105 PATCH(ext_mapping); 106 PATCH(balance); 107 108 /* skip the first, the global context */ 109 for (i = 1; i < srv->config_context->used; i++) { 110 if (!config_check_cond(con, i)) continue; /* condition not matched */ 111 112 data_config *dc = (data_config *)srv->config_context->data[i]; 113 s = p->config_storage[i]; 114 115 /* merge config */ 116 for (j = 0; j < dc->value->used; j++) { 117 data_unset *du = dc->value->data[j]; 118 119 if (buffer_is_equal_string(&du->key, CONST_STR_LEN("sockproxy.server"))) { 120 PATCH(exts); 121 PATCH(exts_auth); 122 PATCH(exts_resp); 123 } else if (buffer_is_equal_string(&du->key, CONST_STR_LEN("sockproxy.debug"))) { 124 PATCH(debug); 125 } else if (buffer_is_equal_string(&du->key, CONST_STR_LEN("sockproxy.balance"))) { 126 PATCH(balance); 127 } 128 } 129 } 130 131 return 0; 132 } 133 #undef PATCH 134 135 static handler_t mod_sockproxy_connection_accept(server *srv, connection *con, void *p_d) { 136 plugin_data *p = p_d; 137 handler_t rc; 138 139 if (con->mode != DIRECT) return HANDLER_GO_ON; 140 141 mod_sockproxy_patch_connection(srv, con, p); 142 if (NULL == p->conf.exts) return HANDLER_GO_ON; 143 144 /*(fake con->uri.path for matching purposes in gw_check_extension())*/ 145 buffer_copy_string_len(con->uri.path, CONST_STR_LEN("/")); 146 147 rc = gw_check_extension(srv, con, p, 1, 0); 148 if (HANDLER_GO_ON != rc) return rc; 149 150 if (con->mode == p->id) { 151 handler_ctx *hctx = con->plugin_ctx[p->id]; 152 hctx->opts.backend = BACKEND_PROXY; 153 hctx->create_env = sockproxy_create_env_connect; 154 hctx->response = chunk_buffer_acquire(); 155 con->http_status = -1; /*(skip HTTP processing)*/ 156 } 157 158 return HANDLER_GO_ON; 159 } 160 161 162 int mod_sockproxy_plugin_init(plugin *p); 163 int mod_sockproxy_plugin_init(plugin *p) { 164 p->version = LIGHTTPD_VERSION_ID; 165 p->name = "sockproxy"; 166 167 p->init = gw_init; 168 p->cleanup = gw_free; 169 p->set_defaults = mod_sockproxy_set_defaults; 170 p->connection_reset = gw_connection_reset; 171 p->handle_connection_accept= mod_sockproxy_connection_accept; 172 p->handle_subrequest = gw_handle_subrequest; 173 p->handle_trigger = gw_handle_trigger; 174 p->handle_waitpid = gw_handle_waitpid_cb; 175 176 return 0; 177 } 178