X-Git-Url: http://jsfdemo.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_http_rewrite.cpp;h=249a7572bc89fc548775a29ed60545fc706dfb6c;hb=0ce322844494022dc85a9c776d9ff041aad17520;hp=e64fb3642f9c315cca1dc053661c293c123bf569;hpb=67e481dac76e773799e3e18c87d29f0a210cbfb1;p=metaproxy-moved-to-github.git diff --git a/src/filter_http_rewrite.cpp b/src/filter_http_rewrite.cpp index e64fb36..249a757 100644 --- a/src/filter_http_rewrite.cpp +++ b/src/filter_http_rewrite.cpp @@ -48,6 +48,7 @@ yf::HttpRewrite::~HttpRewrite() void yf::HttpRewrite::process(mp::Package & package) const { + yaz_log(YLOG_LOG, "HttpRewrite begins...."); Z_GDU *gdu = package.request().get(); //map of request/response vars std::map vars; @@ -56,10 +57,11 @@ void yf::HttpRewrite::process(mp::Package & package) const { Z_HTTP_Request *hreq = gdu->u.HTTP_Request; mp::odr o; - std::cout << ">> Request headers" << std::endl; rewrite_reqline(o, hreq, vars); - rewrite_headers(o, hreq->headers, vars); - rewrite_body(o, &hreq->content_buf, &hreq->content_len, vars); + yaz_log(YLOG_LOG, ">> Request headers"); + rewrite_headers(o, hreq->headers, vars, req_uri_pats, req_groups_bynum); + rewrite_body(o, &hreq->content_buf, &hreq->content_len, vars, + req_uri_pats, req_groups_bynum); package.request() = gdu; } package.move(); @@ -67,11 +69,12 @@ void yf::HttpRewrite::process(mp::Package & package) const if (gdu && gdu->which == Z_GDU_HTTP_Response) { Z_HTTP_Response *hres = gdu->u.HTTP_Response; - std::cout << "Response " << hres->code; - std::cout << "<< Respose headers" << std::endl; + yaz_log(YLOG_LOG, "Response code %d", hres->code); mp::odr o; - rewrite_headers(o, hres->headers, vars); - rewrite_body(o, &hres->content_buf, &hres->content_len, vars); + yaz_log(YLOG_LOG, "<< Respose headers"); + rewrite_headers(o, hres->headers, vars, res_uri_pats, res_groups_bynum); + rewrite_body(o, &hres->content_buf, &hres->content_len, vars, + res_uri_pats, res_groups_bynum); package.response() = gdu; } } @@ -83,26 +86,31 @@ void yf::HttpRewrite::rewrite_reqline (mp::odr & o, Z_HTTP_Request *hreq, std::string path; if (strstr(hreq->path, "http://") == hreq->path) { - std::cout << "Path in the method line is absolute, " - "possibly a proxy request\n"; + yaz_log(YLOG_LOG, "Path in the method line is absolute, " + "possibly a proxy request"); path += hreq->path; } else { //TODO what about proto + path += "http://"; path += z_HTTP_header_lookup(hreq->headers, "Host"); path += hreq->path; } - std::cout << "Proxy request URL is " << path << std::endl; + yaz_log(YLOG_LOG, "Proxy request URL is %s", path.c_str()); std::string npath = test_patterns(vars, path, req_uri_pats, req_groups_bynum); - std::cout << "Resp request URL is " << npath << std::endl; if (!npath.empty()) + { + yaz_log(YLOG_LOG, "Rewritten request URL is %s", npath.c_str()); hreq->path = odr_strdup(o, npath.c_str()); + } } void yf::HttpRewrite::rewrite_headers (mp::odr & o, Z_HTTP_Header *headers, - std::map & vars) const + std::map & vars, + const spair_vec & uri_pats, + const std::vector > & groups_bynum) const { for (Z_HTTP_Header *header = headers; header != 0; @@ -111,16 +119,14 @@ void yf::HttpRewrite::rewrite_headers (mp::odr & o, Z_HTTP_Header *headers, std::string sheader(header->name); sheader += ": "; sheader += header->value; - std::cout << header->name << ": " << header->value << std::endl; - std::string out = test_patterns(vars, - sheader, - req_uri_pats, req_groups_bynum); + yaz_log(YLOG_LOG, "%s: %s", header->name, header->value); + std::string out = test_patterns(vars, sheader, uri_pats, groups_bynum); if (!out.empty()) { size_t pos = out.find(": "); if (pos == std::string::npos) { - std::cout << "Header malformed during rewrite, ignoring"; + yaz_log(YLOG_LOG, "Header malformed during rewrite, ignoring"); continue; } header->name = odr_strdup(o, out.substr(0, pos).c_str()); @@ -131,13 +137,15 @@ void yf::HttpRewrite::rewrite_headers (mp::odr & o, Z_HTTP_Header *headers, } void yf::HttpRewrite::rewrite_body (mp::odr & o, char **content_buf, int *content_len, - std::map & vars) const + std::map & vars, + const spair_vec & uri_pats, + const std::vector > & groups_bynum) const { if (*content_buf) { std::string body(*content_buf); std::string nbody = - test_patterns(vars, body, req_uri_pats, req_groups_bynum); + test_patterns(vars, body, uri_pats, groups_bynum); if (!nbody.empty()) { *content_buf = odr_strdup(o, nbody.c_str()); @@ -157,7 +165,7 @@ const std::string yf::HttpRewrite::test_patterns( const std::vector > & groups_bynum_vec) const { - for (int i = 0; i < uri_pats.size(); i++) + for (unsigned i = 0; i < uri_pats.size(); i++) { std::string out = search_replace(vars, txt, uri_pats[i].first, uri_pats[i].second, @@ -192,21 +200,22 @@ const std::string yf::HttpRewrite::search_replace( = groups_bynum.find(i); if (it != groups_bynum.end()) { //it is - std::string name = it->second; if (!what[i].str().empty()) - vars[name] = what[i]; + vars[it->second] = what[i]; } } //prepare replacement string std::string rvalue = sub_vars(uri_pat, vars); - //rewrite value - std::string rhvalue = what.prefix().str() - + rvalue + what.suffix().str(); - std::cout << "! Rewritten '"+what.str(0)+"' to '"+rvalue+"'\n"; - out += rhvalue; + yaz_log(YLOG_LOG, "! Rewritten '%s' to '%s'", + what.str(0).c_str(), rvalue.c_str()); + out.append(start, what[0].first); + out.append(rvalue); start = what[0].second; //move search forward } + //if we had a match cat the last part + if (start != txt.begin()) + out.append(start, end); return out; } @@ -214,7 +223,7 @@ void yf::HttpRewrite::parse_groups( const spair_vec & uri_pats, std::vector > & groups_bynum_vec) { - for (int h = 0; h < uri_pats.size(); h++) + for (unsigned h = 0; h < uri_pats.size(); h++) { int gnum = 0; bool esc = false; @@ -222,7 +231,7 @@ void yf::HttpRewrite::parse_groups( std::string str = uri_pats[h].first; //for each pair we have an indexing map std::map groups_bynum; - for (int i = 0; i < str.size(); ++i) + for (unsigned i = 0; i < str.size(); ++i) { if (!esc && str[i] == '\\') { @@ -265,8 +274,8 @@ void yf::HttpRewrite::parse_groups( ("Unterminated group name '" + gname + " in '" + str +"'"); groups_bynum[gnum] = gname; - std::cout << "Found named group '" << gname - << "' at $" << gnum << std::endl; + yaz_log(YLOG_LOG, "Found named group '%s' at $%d", + gname.c_str(), gnum); } } } @@ -281,7 +290,7 @@ std::string yf::HttpRewrite::sub_vars (const std::string & in, { std::string out; bool esc = false; - for (int i = 0; i < in.size(); ++i) + for (unsigned i = 0; i < in.size(); ++i) { if (!esc && in[i] == '\\') { @@ -360,13 +369,15 @@ static void configure_rules(const xmlNode *ptr, yf::HttpRewrite::spair_vec & des + std::string((const char *) attr->name) + " in rewrite section of http_rewrite"); } + yaz_log(YLOG_LOG, "Found rewrite rule from '%s' to '%s'", + from.c_str(), to.c_str()); if (!from.empty()) dest.push_back(std::make_pair(from, to)); } else { throw mp::filter::FilterException - ("Bad element " + ("Bad element o" + std::string((const char *) ptr->name) + " in http_rewrite1 filter"); } @@ -384,11 +395,11 @@ void yf::HttpRewrite::configure(const xmlNode * ptr, bool test_only, continue; else if (!strcmp((const char *) ptr->name, "request")) { - configure_rules(ptr->children, req_uri_pats); + configure_rules(ptr, req_uri_pats); } else if (!strcmp((const char *) ptr->name, "response")) { - configure_rules(ptr->children, res_uri_pats); + configure_rules(ptr, res_uri_pats); } else { @@ -398,6 +409,7 @@ void yf::HttpRewrite::configure(const xmlNode * ptr, bool test_only, + " in http_rewrite1 filter"); } } + configure(req_uri_pats, res_uri_pats); } static mp::filter::Base* filter_creator()