X-Git-Url: http://lists.indexdata.com/cgi-bin?a=blobdiff_plain;f=src%2Ffilter_log.cpp;h=2acf9567f8b357fde9625c5a43f8849c2ba469be;hb=e207284fc2943aa6a933cf3405f60bcdc584280d;hp=34a335c58a76602bcc33b63a877b54efd92a205d;hpb=dccaaaf2dabedd87c996e7246bac227a3d908eb3;p=metaproxy-moved-to-github.git diff --git a/src/filter_log.cpp b/src/filter_log.cpp index 34a335c..2acf956 100644 --- a/src/filter_log.cpp +++ b/src/filter_log.cpp @@ -1,37 +1,74 @@ -/* $Id: filter_log.cpp,v 1.16 2006-01-17 15:08:02 adam Exp $ - Copyright (c) 2005, Index Data. +/* $Id: filter_log.cpp,v 1.24 2006-08-30 08:35:47 marc Exp $ + Copyright (c) 2005-2006, Index Data. -%LICENSE% + See the LICENSE file for details */ #include "config.hpp" - #include "package.hpp" #include +#include #include #include +#include "gduutil.hpp" #include "util.hpp" +#include "xmlutil.hpp" #include "filter_log.hpp" +#include #include +#include +#include + -namespace yf = yp2::filter; +namespace mp = metaproxy_1; +namespace yf = metaproxy_1::filter; -namespace yp2 { +namespace metaproxy_1 { namespace filter { + class Log::LFile { + public: + boost::mutex m_mutex; + std::string m_fname; + std::ofstream fout; + std::ostream &out; + LFile(std::string fname); + LFile(std::string fname, std::ostream &use_this); + }; + typedef boost::shared_ptr LFilePtr; class Log::Rep { friend class Log; - boost::mutex m_log_mutex; - std::string m_msg; + Rep(); + void openfile(const std::string &fname); + std::string m_msg_config; + LFilePtr m_file; + bool m_access; + bool m_req_apdu; + bool m_res_apdu; + bool m_req_session; + bool m_res_session; + bool m_init_options; }; + // Only used during configure stage (no threading) + static std::list filter_log_files; } } +yf::Log::Rep::Rep() +{ + m_req_apdu = true; + m_res_apdu = true; + m_req_session = true; + m_res_session = true; + m_init_options = false; + openfile(""); +} + yf::Log::Log(const std::string &x) : m_p(new Rep) { - m_p->m_msg = x; + m_p->m_msg_config = x; } yf::Log::Log() : m_p(new Rep) @@ -40,7 +77,19 @@ yf::Log::Log() : m_p(new Rep) yf::Log::~Log() {} -void yf::Log::process(yp2::Package &package) const +void stream_write(ODR o, void *handle, int type, const char *buf, int len) +{ + yf::Log::LFile *lfile = (yf::Log::LFile*) handle; + lfile->out.write(buf, len); +} + +void option_write(const char *name, void *handle) +{ + yf::Log::LFile *lfile = (yf::Log::LFile*) handle; + lfile->out << " " << name; +} + +void yf::Log::process(mp::Package &package) const { Z_GDU *gdu; @@ -48,22 +97,68 @@ void yf::Log::process(yp2::Package &package) const boost::posix_time::ptime receive_time = boost::posix_time::microsec_clock::local_time(); + + //std::ostringstream msg_request; + //std::ostringstream msg_request_2; + //std::ostringstream msg_response; + //std::ostringstream msg_response_2; + // scope for locking Ostream { - boost::mutex::scoped_lock scoped_lock(m_p->m_log_mutex); - std::cout << receive_time << " " << m_p->m_msg; - std::cout << " request id=" << package.session().id(); - std::cout << " close=" - << (package.session().is_closed() ? "yes" : "no") - << "\n"; - gdu = package.request().get(); - if (gdu) + boost::mutex::scoped_lock scoped_lock(m_p->m_file->m_mutex); + + + if (m_p->m_access) { - yp2::odr odr(ODR_PRINT); - z_GDU(odr, &gdu, 0, 0); + gdu = package.request().get(); + if (gdu) + { + m_p->m_file->out + << m_p->m_msg_config << " " + << receive_time << " " + << package << " " + << "00:00:00.000000" << " " + << *gdu + << "\n"; + } + } + + if (m_p->m_req_session) + { + m_p->m_file->out << receive_time << " " << m_p->m_msg_config; + m_p->m_file->out << " request id=" << package.session().id(); + m_p->m_file->out << " close=" + << (package.session().is_closed() ? "yes" : "no") + << "\n"; } - } + if (m_p->m_init_options) + { + gdu = package.request().get(); + if (gdu && gdu->which == Z_GDU_Z3950 && + gdu->u.z3950->which == Z_APDU_initRequest) + { + m_p->m_file->out << receive_time << " " << m_p->m_msg_config; + m_p->m_file->out << " init options:"; + yaz_init_opt_decode(gdu->u.z3950->u.initRequest->options, + option_write, m_p->m_file.get()); + m_p->m_file->out << "\n"; + } + } + + if (m_p->m_req_apdu) + { + gdu = package.request().get(); + if (gdu) + { + mp::odr odr(ODR_PRINT); + odr_set_stream(odr, m_p->m_file.get(), stream_write, 0); + z_GDU(odr, &gdu, 0, 0); + } + } + m_p->m_file->out.flush(); + } + // unlocked during move package.move(); @@ -75,25 +170,91 @@ void yf::Log::process(yp2::Package &package) const // scope for locking Ostream { - boost::mutex::scoped_lock scoped_lock(m_p->m_log_mutex); - std::cout << send_time << " " << m_p->m_msg; - std::cout << " response id=" << package.session().id(); - std::cout << " close=" - << (package.session().is_closed() ? "yes " : "no ") - << "duration=" << duration - << "\n"; - //<< "duration=" << duration.total_seconds() - // << "." << duration.fractional_seconds() - // << "\n"; - gdu = package.response().get(); - if (gdu) + boost::mutex::scoped_lock scoped_lock(m_p->m_file->m_mutex); + + if (m_p->m_access) + { + gdu = package.response().get(); + if (gdu) + { + m_p->m_file->out + << m_p->m_msg_config << " " + << send_time << " " + << package << " " + << duration << " " + << *gdu + << "\n"; + } + } + + if (m_p->m_res_session) + { + m_p->m_file->out << send_time << " " << m_p->m_msg_config; + m_p->m_file->out << " response id=" << package.session().id(); + m_p->m_file->out << " close=" + << (package.session().is_closed() ? "yes " : "no ") + << "duration=" << duration + << "\n"; + } + + if (m_p->m_init_options) + { + gdu = package.response().get(); + if (gdu && gdu->which == Z_GDU_Z3950 && + gdu->u.z3950->which == Z_APDU_initResponse) + { + m_p->m_file->out << receive_time << " " << m_p->m_msg_config; + m_p->m_file->out << " init options:"; + yaz_init_opt_decode(gdu->u.z3950->u.initResponse->options, + option_write, m_p->m_file.get()); + m_p->m_file->out << "\n"; + } + } + + if (m_p->m_res_apdu) { - yp2::odr odr(ODR_PRINT); - z_GDU(odr, &gdu, 0, 0); + gdu = package.response().get(); + if (gdu) + { + mp::odr odr(ODR_PRINT); + odr_set_stream(odr, m_p->m_file.get(), stream_write, 0); + z_GDU(odr, &gdu, 0, 0); + } } + + m_p->m_file->out.flush(); } } +yf::Log::LFile::LFile(std::string fname) : + m_fname(fname), fout(fname.c_str()), out(fout) +{ +} + +yf::Log::LFile::LFile(std::string fname, std::ostream &use_this) : + m_fname(fname), out(use_this) +{ +} + +void yf::Log::Rep::openfile(const std::string &fname) +{ + std::list::const_iterator it = filter_log_files.begin(); + for (; it != filter_log_files.end(); it++) + { + if ((*it)->m_fname == fname) + { + m_file = *it; + return; + } + } + // open stdout for empty file + LFilePtr newfile(fname.length() == 0 + ? new LFile(fname, std::cout) + : new LFile(fname)); + filter_log_files.push_back(newfile); + m_file = newfile; +} + void yf::Log::configure(const xmlNode *ptr) { for (ptr = ptr->children; ptr; ptr = ptr->next) @@ -101,30 +262,77 @@ void yf::Log::configure(const xmlNode *ptr) if (ptr->type != XML_ELEMENT_NODE) continue; if (!strcmp((const char *) ptr->name, "message")) - m_p->m_msg = yp2::xml::get_text(ptr); + m_p->m_msg_config = mp::xml::get_text(ptr); + else if (!strcmp((const char *) ptr->name, "filename")) + { + std::string fname = mp::xml::get_text(ptr); + m_p->openfile(fname); + } + else if (!strcmp((const char *) ptr->name, "category")) + { + const struct _xmlAttr *attr; + for (attr = ptr->properties; attr; attr = attr->next) + { + if (!strcmp((const char *) attr->name, + "access")) + m_p->m_access = + mp::xml::get_bool(attr->children, true); + else if (!strcmp((const char *) attr->name, "request-apdu")) + m_p->m_req_apdu = mp::xml::get_bool(attr->children, true); + else if (!strcmp((const char *) attr->name, "response-apdu")) + m_p->m_res_apdu = mp::xml::get_bool(attr->children, true); + else if (!strcmp((const char *) attr->name, "apdu")) + { + m_p->m_req_apdu = mp::xml::get_bool(attr->children, true); + m_p->m_res_apdu = m_p->m_req_apdu; + } + else if (!strcmp((const char *) attr->name, + "request-session")) + m_p->m_req_session = + mp::xml::get_bool(attr->children, true); + else if (!strcmp((const char *) attr->name, + "response-session")) + m_p->m_res_session = + mp::xml::get_bool(attr->children, true); + else if (!strcmp((const char *) attr->name, + "session")) + { + m_p->m_req_session = + mp::xml::get_bool(attr->children, true); + m_p->m_res_session = m_p->m_req_session; + } + else if (!strcmp((const char *) attr->name, + "init-options")) + m_p->m_init_options = + mp::xml::get_bool(attr->children, true); + else + throw mp::filter::FilterException( + "Bad attribute " + std::string((const char *) + attr->name)); + } + } else { - throw yp2::filter::FilterException("Bad element " + throw mp::filter::FilterException("Bad element " + std::string((const char *) ptr->name)); } } } -static yp2::filter::Base* filter_creator() +static mp::filter::Base* filter_creator() { - return new yp2::filter::Log; + return new mp::filter::Log; } extern "C" { - struct yp2_filter_struct yp2_filter_log = { + struct metaproxy_1_filter_struct metaproxy_1_filter_log = { 0, "log", filter_creator }; } - /* * Local variables: * c-basic-offset: 4