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 static void mod_sockproxy_merge_config_cpv(plugin_config * const pconf, const config_plugin_value_t * const cpv) { 24 switch (cpv->k_id) { /* index into static config_plugin_keys_t cpk[] */ 25 case 0: /* sockproxy.server */ 26 if (cpv->vtype == T_CONFIG_LOCAL) { 27 gw_plugin_config * const gw = cpv->v.v; 28 pconf->exts = gw->exts; 29 pconf->exts_auth = gw->exts_auth; 30 pconf->exts_resp = gw->exts_resp; 31 } 32 break; 33 case 1: /* sockproxy.balance */ 34 /*if (cpv->vtype == T_CONFIG_LOCAL)*//*always true here for this param*/ 35 pconf->balance = (int)cpv->v.u; 36 break; 37 case 2: /* sockproxy.debug */ 38 pconf->debug = (int)cpv->v.u; 39 break; 40 default:/* should not happen */ 41 return; 42 } 43 } 44 45 static void mod_sockproxy_merge_config(plugin_config * const pconf, const config_plugin_value_t *cpv) { 46 do { 47 mod_sockproxy_merge_config_cpv(pconf, cpv); 48 } while ((++cpv)->k_id != -1); 49 } 50 51 static void mod_sockproxy_patch_config(request_st * const r, plugin_data * const p) { 52 memcpy(&p->conf, &p->defaults, sizeof(plugin_config)); 53 for (int i = 1, used = p->nconfig; i < used; ++i) { 54 if (config_check_cond(r, (uint32_t)p->cvlist[i].k_id)) 55 mod_sockproxy_merge_config(&p->conf,p->cvlist+p->cvlist[i].v.u2[0]); 56 } 57 } 58 59 SETDEFAULTS_FUNC(mod_sockproxy_set_defaults) { 60 static const config_plugin_keys_t cpk[] = { 61 { CONST_STR_LEN("sockproxy.server"), 62 T_CONFIG_ARRAY_KVARRAY, 63 T_CONFIG_SCOPE_CONNECTION } 64 ,{ CONST_STR_LEN("sockproxy.balance"), 65 T_CONFIG_STRING, 66 T_CONFIG_SCOPE_CONNECTION } 67 ,{ CONST_STR_LEN("sockproxy.debug"), 68 T_CONFIG_INT, 69 T_CONFIG_SCOPE_CONNECTION } 70 ,{ NULL, 0, 71 T_CONFIG_UNSET, 72 T_CONFIG_SCOPE_UNSET } 73 }; 74 75 plugin_data * const p = p_d; 76 if (!config_plugin_values_init(srv, p, cpk, "mod_sockproxy")) 77 return HANDLER_ERROR; 78 79 /* process and validate config directives 80 * (init i to 0 if global context; to 1 to skip empty global context) */ 81 for (int i = !p->cvlist[0].v.u2[1]; i < p->nconfig; ++i) { 82 config_plugin_value_t *cpv = p->cvlist + p->cvlist[i].v.u2[0]; 83 gw_plugin_config *gw = NULL; 84 for (; -1 != cpv->k_id; ++cpv) { 85 switch (cpv->k_id) { 86 case 0: /* sockproxy.server */ 87 gw = calloc(1, sizeof(gw_plugin_config)); 88 force_assert(gw); 89 if (!gw_set_defaults_backend(srv, p, cpv->v.a, gw, 0, 90 cpk[cpv->k_id].k)) { 91 gw_plugin_config_free(gw); 92 return HANDLER_ERROR; 93 } 94 cpv->v.v = gw; 95 cpv->vtype = T_CONFIG_LOCAL; 96 break; 97 case 1: /* sockproxy.balance */ 98 cpv->v.u = (unsigned int)gw_get_defaults_balance(srv, cpv->v.b); 99 break; 100 case 2: /* sockproxy.debug */ 101 break; 102 default:/* should not happen */ 103 break; 104 } 105 } 106 107 /* disable check-local for all exts (default enabled) */ 108 if (gw && gw->exts) { /*(check after gw_set_defaults_backend())*/ 109 gw_exts_clear_check_local(gw->exts); 110 } 111 } 112 113 /* default is 0 */ 114 /*p->defaults.balance = (unsigned int)gw_get_defaults_balance(srv, NULL);*/ 115 116 /* initialize p->defaults from global config context */ 117 if (p->nconfig > 0 && p->cvlist->v.u2[1]) { 118 const config_plugin_value_t *cpv = p->cvlist + p->cvlist->v.u2[0]; 119 if (-1 != cpv->k_id) 120 mod_sockproxy_merge_config(&p->defaults, cpv); 121 } 122 123 return HANDLER_GO_ON; 124 } 125 126 127 static handler_t sockproxy_create_env_connect(handler_ctx *hctx) { 128 request_st * const r = hctx->r; 129 r->resp_body_started = 1; 130 gw_set_transparent(hctx); 131 http_response_upgrade_read_body_unknown(r); 132 133 status_counter_inc(CONST_STR_LEN("sockproxy.requests")); 134 return HANDLER_GO_ON; 135 } 136 137 138 static handler_t mod_sockproxy_connection_accept(connection *con, void *p_d) { 139 request_st * const r = &con->request; 140 plugin_data *p = p_d; 141 handler_t rc; 142 143 if (NULL != r->handler_module) return HANDLER_GO_ON; 144 145 mod_sockproxy_patch_config(r, p); 146 if (NULL == p->conf.exts) return HANDLER_GO_ON; 147 148 /*(fake r->uri.path for matching purposes in gw_check_extension())*/ 149 buffer_copy_string_len(&r->uri.path, CONST_STR_LEN("/")); 150 151 rc = gw_check_extension(r, p, 1, 0); 152 if (HANDLER_GO_ON != rc) return rc; 153 154 if (r->handler_module == p->self) { 155 handler_ctx *hctx = r->plugin_ctx[p->id]; 156 hctx->opts.backend = BACKEND_PROXY; 157 hctx->create_env = sockproxy_create_env_connect; 158 hctx->response = chunk_buffer_acquire(); 159 r->http_status = -1; /*(skip HTTP processing)*/ 160 } 161 162 return HANDLER_GO_ON; 163 } 164 165 166 int mod_sockproxy_plugin_init(plugin *p); 167 int mod_sockproxy_plugin_init(plugin *p) { 168 p->version = LIGHTTPD_VERSION_ID; 169 p->name = "sockproxy"; 170 171 p->init = gw_init; 172 p->cleanup = gw_free; 173 p->set_defaults = mod_sockproxy_set_defaults; 174 p->connection_reset = gw_connection_reset; 175 p->handle_connection_accept= mod_sockproxy_connection_accept; 176 p->handle_subrequest = gw_handle_subrequest; 177 p->handle_trigger = gw_handle_trigger; 178 p->handle_waitpid = gw_handle_waitpid_cb; 179 180 return 0; 181 } 182