X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fyaz-proxy.cpp;h=d4693c613acd5e745ef4c60336b43bb7091b6ff0;hb=850cba55769a7ed94d4da0e7c05b823e54ce3cc3;hp=9280b839701b1b0dca3ce9d84f0cd43741f2379e;hpb=ee8cfba95daa813ea4dd66ba9e99066db5e5b967;p=yazproxy-moved-to-github.git diff --git a/src/yaz-proxy.cpp b/src/yaz-proxy.cpp index 9280b83..d4693c6 100644 --- a/src/yaz-proxy.cpp +++ b/src/yaz-proxy.cpp @@ -1,4 +1,4 @@ -/* $Id: yaz-proxy.cpp,v 1.30 2005-06-08 13:29:03 adam Exp $ +/* $Id: yaz-proxy.cpp,v 1.33 2005-06-21 18:46:04 adam Exp $ Copyright (c) 1998-2005, Index Data. This file is part of the yaz-proxy. @@ -55,24 +55,71 @@ Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA using namespace yazpp_1; +#define USE_AUTH_MSG 1 + +#if USE_AUTH_MSG class Auth_Msg : public IMsg_Thread { public: int m_ret; - void destroy(); IMsg_Thread *handle(); void result(); Yaz_Proxy *m_proxy; - Z_APDU *m_apdu; + NMEM m_nmem; + char *m_apdu_buf; + int m_apdu_len; + Auth_Msg(); + virtual ~Auth_Msg(); }; +Auth_Msg::Auth_Msg() +{ + m_nmem = nmem_create(); +} + +Auth_Msg::~Auth_Msg() +{ + nmem_destroy(m_nmem); +} + IMsg_Thread *Auth_Msg::handle() { yaz_log(YLOG_LOG, "Auth_Msg:handle begin"); - m_ret = m_proxy->handle_authentication(m_apdu); + ODR decode = odr_createmem(ODR_DECODE); + Z_APDU *apdu; + + odr_setbuf(decode, m_apdu_buf, m_apdu_len, 0); + int r = z_APDU(decode, &apdu, 0, 0); + if (!r) + { + yaz_log(YLOG_WARN, "decode failed in Auth_Msg::handle"); + } + else + { + m_ret = m_proxy->handle_authentication(apdu); + } yaz_log(YLOG_LOG, "Auth_Msg:handle end"); + odr_destroy(decode); return this; } +void Auth_Msg::result() +{ + yaz_log(YLOG_LOG, "Auth_Msg:result proxy ok buf=%p len=%d", + m_apdu_buf, m_apdu_len); + odr_reset(m_proxy->odr_decode()); + odr_setbuf(m_proxy->odr_decode(), m_apdu_buf, m_apdu_len, 0); + Z_APDU *apdu = 0; + int r = z_APDU(m_proxy->odr_decode(), &apdu, 0, 0); + if (r) + yaz_log(YLOG_LOG, "Auth_Msg::result z_APDU OK"); + else + yaz_log(YLOG_LOG, "Auth_Msg::result z_APDU failed"); + m_proxy->result_authentication(apdu, m_ret); + delete this; +} + +#endif + void Yaz_Proxy::result_authentication(Z_APDU *apdu, int ret) { if (ret == 0) @@ -80,19 +127,12 @@ void Yaz_Proxy::result_authentication(Z_APDU *apdu, int ret) Z_APDU *apdu_reject = zget_APDU(odr_encode(), Z_APDU_initResponse); *apdu_reject->u.initResponse->result = 0; send_to_client(apdu_reject); - shutdown(); + dec_ref(); } else handle_incoming_Z_PDU_2(apdu); } -void Auth_Msg::result() -{ - yaz_log(YLOG_LOG, "Auth_Msg:result"); - m_proxy->result_authentication(m_apdu, m_ret); - delete this; -} - static const char *apdu_name(Z_APDU *apdu) { switch (apdu->which) @@ -177,17 +217,20 @@ Yaz_Proxy::Yaz_Proxy(IPDU_Observable *the_PDU_Observable, m_bw_hold_PDU = 0; m_bw_max = 0; m_pdu_max = 0; + m_timeout_mode = timeout_normal; + m_timeout_gdu = 0; m_max_record_retrieve = 0; m_reconfig_flag = 0; m_config_fname = 0; m_request_no = 0; - m_invalid_session = 0; + m_flag_invalid_session = 0; m_referenceId = 0; m_referenceId_mem = nmem_create(); m_config = 0; m_marcxml_mode = none; m_stylesheet_xsp = 0; m_stylesheet_nprl = 0; + m_stylesheet_apdu = 0; m_s2z_stylesheet = 0; m_s2z_database = 0; m_schema = 0; @@ -230,6 +273,13 @@ Yaz_Proxy::Yaz_Proxy(IPDU_Observable *the_PDU_Observable, if (!m_parent) low_socket_open(); m_my_thread = 0; + m_ref_count = 1; + m_peername = 0; +} + +void Yaz_Proxy::inc_ref() +{ +// m_ref_count++; } Yaz_Proxy::~Yaz_Proxy() @@ -253,6 +303,7 @@ Yaz_Proxy::~Yaz_Proxy() #endif xfree (m_time_tv); + xfree (m_peername); xfree (m_schema); xfree (m_backend_type); xfree (m_backend_charset); @@ -358,8 +409,9 @@ IPDU_Observer *Yaz_Proxy::sessionNotify(IPDU_Observable new_proxy->set_APDU_yazlog(0); sprintf(new_proxy->m_session_str, "%ld:%d ", (long) time(0), m_session_no); m_session_no++; + new_proxy->m_peername = xstrdup(the_PDU_Observable->getpeername()); yaz_log (YLOG_LOG, "%sNew session %s", new_proxy->m_session_str, - the_PDU_Observable->getpeername()); + new_proxy->m_peername); new_proxy->set_proxy_negotiation(m_proxy_negotiation_charset, m_proxy_negotiation_lang); // create thread object the first time we get an incoming connection @@ -755,6 +807,8 @@ int Yaz_Proxy::convert_xsl(Z_NamePlusRecordList *p, Z_APDU *apdu) m_stylesheet_offset = 0; m_stylesheet_nprl = p; m_stylesheet_apdu = apdu; + m_timeout_mode = timeout_xsl; + timeout(0); return 1; } @@ -802,6 +856,7 @@ void Yaz_Proxy::convert_xsl_delay() m_stylesheet_offset++; if (m_stylesheet_offset == m_stylesheet_nprl->num_records) { + m_timeout_mode = timeout_normal; m_stylesheet_nprl = 0; #if HAVE_XSLT if (m_stylesheet_xsp) @@ -1327,7 +1382,7 @@ int Yaz_Proxy::send_to_client(Z_APDU *apdu) *sr->resultCount); if (*sr->resultCount < 0) { - m_invalid_session = 1; + m_flag_invalid_session = 1; kill_session = 1; *sr->searchStatus = 0; @@ -1692,21 +1747,37 @@ void Yaz_Proxy::recv_GDU(Z_GDU *apdu, int len) if (m_log_mask & PROXY_LOG_REQ_CLIENT) yaz_log (YLOG_LOG, "%sReceiving %s from client %d bytes", m_session_str, gdu_name(apdu), len); - - if (m_bw_hold_PDU) // double incoming PDU. shutdown now. - shutdown(); - - m_bw_stat.add_bytes(len); - m_pdu_stat.add_bytes(1); - + #if HAVE_GETTIMEOFDAY gettimeofday((struct timeval *) m_time_tv, 0); #endif + m_bw_stat.add_bytes(len); + m_pdu_stat.add_bytes(1); + + GDU *gdu = new GDU(apdu); + int qsize = m_in_queue.size(); + if (m_timeout_mode != timeout_normal) + { + yaz_log(YLOG_LOG, "%sAdded gdu in queue of size %d", m_session_str, + qsize); + m_in_queue.enqueue(gdu); + } + else + { + recv_GDU_reduce(gdu); + recv_GDU_more(); + } +} +void Yaz_Proxy::recv_GDU_reduce(GDU *gdu) +{ int bw_total = m_bw_stat.get_total(); int pdu_total = m_pdu_stat.get_total(); - int reduce = 0; + + assert(m_timeout_mode == timeout_normal); + assert(m_timeout_gdu == 0); + if (m_bw_max) { if (bw_total > m_bw_max) @@ -1723,16 +1794,36 @@ void Yaz_Proxy::recv_GDU(Z_GDU *apdu, int len) } } m_http_version = 0; + if (reduce) { yaz_log(YLOG_LOG, "%sdelay=%d bw=%d pdu=%d limit-bw=%d limit-pdu=%d", m_session_str, reduce, bw_total, pdu_total, m_bw_max, m_pdu_max); - m_bw_hold_PDU = apdu; // save PDU and signal "on hold" + m_timeout_mode = timeout_reduce; + m_timeout_gdu = gdu; + // m_bw_hold_PDU = apdu; // save PDU and signal "on hold" timeout(reduce); // call us reduce seconds later } - else if (apdu->which == Z_GDU_Z3950) + else + recv_GDU_normal(gdu); +} + +void Yaz_Proxy::recv_GDU_more() +{ + GDU *g; + while (m_timeout_mode == timeout_normal && (g = m_in_queue.dequeue())) + recv_GDU_reduce(g); +} + +void Yaz_Proxy::recv_GDU_normal(GDU *gdu) +{ + Z_GDU *apdu = gdu->get(); + gdu->extract_odr_to(odr_decode()); + delete gdu; + + if (apdu->which == Z_GDU_Z3950) handle_incoming_Z_PDU(apdu->u.z3950); else if (apdu->which == Z_GDU_HTTP_Request) handle_incoming_HTTP(apdu->u.HTTP_Request); @@ -1751,6 +1842,7 @@ void Yaz_Proxy::handle_max_record_retrieve(Z_APDU *apdu) } } } + void Yaz_Proxy::handle_charset_lang_negotiation(Z_APDU *apdu) { if (apdu->which == Z_APDU_initRequest) @@ -2054,14 +2146,17 @@ int Yaz_Proxy::handle_authentication(Z_APDU *apdu) int ret; if (req->idAuthentication == 0) { - ret = cfg->client_authentication(m_default_target, 0, 0, 0); + ret = cfg->client_authentication(m_default_target, 0, 0, 0, + m_peername); } else if (req->idAuthentication->which == Z_IdAuthentication_idPass) { - ret = cfg->client_authentication(m_default_target, + ret = cfg->client_authentication( + m_default_target, req->idAuthentication->u.idPass->userId, req->idAuthentication->u.idPass->groupId, - req->idAuthentication->u.idPass->password); + req->idAuthentication->u.idPass->password, + m_peername); } else if (req->idAuthentication->which == Z_IdAuthentication_open) { @@ -2069,10 +2164,12 @@ int Yaz_Proxy::handle_authentication(Z_APDU *apdu) *user = '\0'; *pass = '\0'; sscanf(req->idAuthentication->u.open, "%63[^/]/%63s", user, pass); - ret = cfg->client_authentication(m_default_target, user, 0, pass); + ret = cfg->client_authentication(m_default_target, user, 0, pass, + m_peername); } else - ret = cfg->client_authentication(m_default_target, 0, 0, 0); + ret = cfg->client_authentication(m_default_target, 0, 0, 0, + m_peername); cfg->target_authentication(m_default_target, odr_encode(), req); @@ -2682,13 +2779,15 @@ void Yaz_Proxy::handle_init(Z_APDU *apdu) m_initRequest_maximumRecordSize = *apdu->u.initRequest-> maximumRecordSize; *apdu->u.initRequest->maximumRecordSize = 1024*1024; + + Z_CharSetandLanguageNegotiation *charSetandLangRecord = + yaz_get_charneg_record(*oi); // Save proposal charsets and langs. if (ODR_MASK_GET(apdu->u.initRequest->options, - Z_Options_negotiationModel)) + Z_Options_negotiationModel) + && charSetandLangRecord) { - Z_CharSetandLanguageNegotiation *charSetandLangRecord = - yaz_get_charneg_record(*oi); yaz_get_proposal_charneg(m_referenceId_mem, charSetandLangRecord, @@ -2765,10 +2864,21 @@ void Yaz_Proxy::handle_init(Z_APDU *apdu) } m_client->m_init_flag = 1; +#if USE_AUTH_MSG Auth_Msg *m = new Auth_Msg; - m->m_apdu = apdu; m->m_proxy = this; + z_APDU(odr_encode(), &apdu, 0, "encode"); + char *apdu_buf = odr_getbuf(odr_encode(), &m->m_apdu_len, 0); + m->m_apdu_buf = (char*) nmem_malloc(m->m_nmem, m->m_apdu_len); + memcpy(m->m_apdu_buf, apdu_buf, m->m_apdu_len); + odr_reset(odr_encode()); + + inc_ref(); m_my_thread->put(m); +#else + int ret = handle_authentication(apdu); + result_authentication(apdu, ret); +#endif } void Yaz_Proxy::handle_incoming_Z_PDU(Z_APDU *apdu) @@ -2787,11 +2897,12 @@ void Yaz_Proxy::handle_incoming_Z_PDU(Z_APDU *apdu) else m_referenceId = 0; - if (!m_client && m_invalid_session) + if (!m_client && m_flag_invalid_session) { - m_apdu_invalid_session = apdu; + // Got request for a session that is invalid.. + m_apdu_invalid_session = apdu; // save package m_mem_invalid_session = odr_extract_mem(odr_decode()); - apdu = m_initRequest_apdu; + apdu = m_initRequest_apdu; // but throw an init to the target } // Determine our client. @@ -2883,7 +2994,7 @@ void Yaz_Proxy::releaseClient() { xfree(m_proxyTarget); m_proxyTarget = 0; - m_invalid_session = 0; + m_flag_invalid_session = 0; // only keep if keep_alive flag is set... if (m_client && m_client->m_pdu_recv < m_keepalive_limit_pdu && @@ -2926,10 +3037,17 @@ void Yaz_Proxy::releaseClient() m_parent->pre_init(); } -void Yaz_Proxy::shutdown() +bool Yaz_Proxy::dec_ref() { - releaseClient(); - delete this; + --m_ref_count; + assert(m_ref_count >= 0); + bool last = (m_ref_count == 0); + if (m_ref_count == 0) + { + releaseClient(); + delete this; + } + return last; } const char *Yaz_ProxyClient::get_session_str() @@ -2952,7 +3070,7 @@ void Yaz_Proxy::failNotify() inc_request_no(); yaz_log (YLOG_LOG, "%sConnection closed by client", get_session_str()); - shutdown(); + dec_ref(); } void Yaz_ProxyClient::failNotify() @@ -3123,25 +3241,26 @@ void Yaz_Proxy::timeoutNotify() { if (m_parent) { - if (m_bw_hold_PDU) - { - timeout(m_client_idletime); - Z_GDU *apdu = m_bw_hold_PDU; - m_bw_hold_PDU = 0; - - if (apdu->which == Z_GDU_Z3950) - handle_incoming_Z_PDU(apdu->u.z3950); - else if (apdu->which == Z_GDU_HTTP_Request) - handle_incoming_HTTP(apdu->u.HTTP_Request); - } - else if (m_stylesheet_nprl) - convert_xsl_delay(); - else + GDU *gdu; + switch(m_timeout_mode) { + case timeout_normal: inc_request_no(); - + m_in_queue.clear(); yaz_log (YLOG_LOG, "%sTimeout (client to proxy)", m_session_str); - shutdown(); + dec_ref(); + break; + case timeout_reduce: + timeout(m_client_idletime); + m_timeout_mode = timeout_normal; + gdu = m_timeout_gdu; + m_timeout_gdu = 0; + recv_GDU_normal(gdu); + break; + case timeout_xsl: + assert(m_stylesheet_nprl); + convert_xsl_delay(); + recv_GDU_more(); } } else @@ -3154,7 +3273,7 @@ void Yaz_Proxy::timeoutNotify() void Yaz_Proxy::markInvalid() { m_client = 0; - m_invalid_session = 1; + m_flag_invalid_session = 1; } void Yaz_ProxyClient::timeoutNotify() @@ -3235,9 +3354,9 @@ void Yaz_ProxyClient::recv_GDU(Z_GDU *apdu, int len) int Yaz_Proxy::handle_init_response_for_invalid_session(Z_APDU *apdu) { - if (!m_invalid_session) + if (!m_flag_invalid_session) return 0; - m_invalid_session = 0; + m_flag_invalid_session = 0; handle_incoming_Z_PDU(m_apdu_invalid_session); assert (m_mem_invalid_session); nmem_destroy(m_mem_invalid_session); @@ -3331,14 +3450,16 @@ void Yaz_ProxyClient::recv_Z_PDU(Z_APDU *apdu, int len) } if (m_cookie) set_otherInformationString (apdu, VAL_COOKIE, 1, m_cookie); - if (m_server) - { - m_server->send_to_client(apdu); - } + + Yaz_Proxy *server = m_server; // save it. send_to_client may destroy us + + if (server) + server->send_to_client(apdu); if (apdu->which == Z_APDU_close) - { shutdown(); - } + else if (server) + server->recv_GDU_more(); + } void Yaz_Proxy::low_socket_close()