X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Ffilter_http_file.cpp;h=defac589f25b0da47fc5928d86e90b0b46feb2d2;hb=dc2b8231ac2e47a34f5742ae5b5a1e5cf0ad4c98;hp=58ec810f38c05d6058b8884a2487205a5cd2c046;hpb=73f37c91c144b070020df2f27472c09b62367acf;p=metaproxy-moved-to-github.git diff --git a/src/filter_http_file.cpp b/src/filter_http_file.cpp index 58ec810..defac58 100644 --- a/src/filter_http_file.cpp +++ b/src/filter_http_file.cpp @@ -1,5 +1,5 @@ /* This file is part of Metaproxy. - Copyright (C) 2005-2012 Index Data + Copyright (C) Index Data Metaproxy is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free @@ -47,6 +47,8 @@ namespace metaproxy_1 { struct HttpFile::Area { std::string m_url_path_prefix; std::string m_file_root; + bool m_raw; + Area(); }; class HttpFile::Mime { friend class Rep; @@ -67,12 +69,17 @@ namespace metaproxy_1 { Z_HTTP_Request *req, mp::Package &package); void fetch_file(mp::Session &session, Z_HTTP_Request *req, - std::string &fname, mp::Package &package); + std::string &fname, mp::Package &package, + bool raw); std::string get_mime_type(std::string &fname); }; } } +yf::HttpFile::Area::Area() : m_raw(false) +{ +} + yf::HttpFile::Mime::Mime() {} yf::HttpFile::Mime::Mime(std::string type) : m_type(type) {} @@ -126,9 +133,10 @@ std::string yf::HttpFile::Rep::get_mime_type(std::string &fname) void yf::HttpFile::Rep::fetch_file(mp::Session &session, Z_HTTP_Request *req, - std::string &fname, mp::Package &package) + std::string &fname, mp::Package &package, + bool raw) { - mp::odr o; + mp::odr o(ODR_ENCODE); FILE *f = fopen(fname.c_str(), "rb"); if (!f) @@ -153,28 +161,40 @@ void yf::HttpFile::Rep::fetch_file(mp::Session &session, return; } rewind(f); - - Z_GDU *gdu = o.create_HTTP_Response(session, req, 200); - - Z_HTTP_Response *hres = gdu->u.HTTP_Response; - hres->content_len = sz; - if (hres->content_len > 0) + char *fbuf = (char*) odr_malloc(o, sz); + if (sz > 0) { - hres->content_buf = (char*) odr_malloc(o, hres->content_len); - if (fread(hres->content_buf, hres->content_len, 1, f) != 1) + if (fread(fbuf, sz, 1, f) != 1) { - fclose(f); Z_GDU *gdu = o.create_HTTP_Response(session, req, 500); package.response() = gdu; + fclose(f); return; } } fclose(f); - std::string content_type = get_mime_type(fname); - - z_HTTP_header_add(o, &hres->headers, + Z_GDU *gdu = 0; + mp::odr decode(ODR_DECODE); + if (raw) + { + odr_setbuf(decode, (char *) fbuf, sz, 0); + int r = z_GDU(decode, &gdu, 0, 0); + if (!r) + { + gdu = o.create_HTTP_Response(session, req, 500); + } + } + else + { + gdu = o.create_HTTP_Response(session, req, 200); + Z_HTTP_Response *hres = gdu->u.HTTP_Response; + hres->content_len = sz; + hres->content_buf = fbuf; + std::string content_type = get_mime_type(fname); + z_HTTP_header_add(o, &hres->headers, "Content-Type", content_type.c_str()); + } package.response() = gdu; } @@ -211,7 +231,7 @@ void yf::HttpFile::Rep::fetch_uri(mp::Session &session, { std::string fname = it->m_file_root + path.substr(l); package.log("http_file", YLOG_LOG, "%s", fname.c_str()); - fetch_file(session, req, fname, package); + fetch_file(session, req, fname, package, it->m_raw); return; } } @@ -259,6 +279,7 @@ void mp::filter::HttpFile::configure(const xmlNode * ptr, bool test_only, { xmlNode *a_node = ptr->children; Area a; + for (; a_node; a_node = a_node->next) { if (a_node->type != XML_ELEMENT_NODE) @@ -268,6 +289,8 @@ void mp::filter::HttpFile::configure(const xmlNode * ptr, bool test_only, a.m_file_root = mp::xml::get_text(a_node); else if (mp::xml::is_element_mp(a_node, "prefix")) a.m_url_path_prefix = mp::xml::get_text(a_node); + else if (mp::xml::is_element_mp(a_node, "raw")) + a.m_raw = mp::xml::get_bool(a_node, false); else throw mp::filter::FilterException ("Bad element "