path: root/src/reqp.cpp

#ifndef REQP_CPP
#define REQP_CPP

#include "reqp.h"
#include "tool/tool.h"

using namespace std;

string reqp::HTTP_CODEOK = "HTTP/1.1 200 OK\r\n";
string reqp::HTTP_SERVER = "Server: yhttpd (Unix)\r\n";
string reqp::HTTP_CONTAC = "Contact: www.yhttpd.org\r\n";
string reqp::HTTP_EXPIRE = "Expires: 0\r\n";
string reqp::HTTP_CACHEC = "Cache-control: no-cache\r\nPragma: no-cache\r\n";
string reqp::HTTP_CONNEC = "Connection: keep-alive\r\n";
string reqp::HTTP_CHUNKE = "Transfer-Encoding: chunked\r\n";
string reqp::HTTP_COTYPE = "Content-Type: ";

reqp::reqp( )
{}

void
reqp::get_request_parameters( string s_parameters, map_string& map_params )
{
     string s_tmp; 
     unsigned i_pos, i_pos2;

     while( (i_pos = s_parameters.find("&")) != string::npos )
     {
      s_tmp = s_parameters.substr(0, i_pos );

      if ( (i_pos2 = s_tmp.find("=")) != string::npos )
       map_params[ s_tmp.substr(0, i_pos2) ] = s_tmp.substr( i_pos2+1 ); 

      s_parameters = s_parameters.substr( i_pos + 1 );
     }

     // Get the last request parameter, which does not have a "&" on the end!
     if( (i_pos = s_parameters.find("=")) != string::npos ) 
      map_params[ s_parameters.substr(0, i_pos) ] = s_parameters.substr( i_pos+1 );
   
 //    map_string::iterator iter;
 //    for ( iter = map_params.begin(); iter != map_params.end(); ++iter )
 //      cout << ">>>" << iter->first << "=" << iter->second << endl;
}

string
reqp::get_url( thrd* p_thrd, string s_req, map_string &map_params )
{
    unsigned i_pos, i_pos2;
    string s_vars( "" );
    string s_ret;
    int i_req;

    // GET request
    if ( s_req.find("GET") != string::npos)  
    {
     // Be sure that the GET request has minimum length
     if ( s_req.length() > 5 )
     {
      // Get rid of "GET /"
      if ( (i_pos = s_req.find("\n")) == string::npos )
       i_pos = s_req.length() - 1;

      s_req = s_req.substr(5, i_pos - 5);        

      // Get HTML site to be displayed 
      if ( (i_pos = s_req.find("?")) == string::npos )
      {
       if ( (i_pos2 = s_req.find(" HTTP")) != string::npos )
        s_ret = url_decode( s_req.substr(0, i_pos2));
      }

      else
      {
       s_ret = url_decode( s_req.substr(0, i_pos) );

       // Get request parameters:
       if ( (i_pos2 = s_req.find(" HTTP")) != string::npos )
       {
        s_req = url_decode( s_req.substr(i_pos + 1, i_pos2 - i_pos - 1) );
        get_request_parameters( s_req, map_params );
       }
      }

     }
    }

    // POST request
    else 
    {
     if ( (i_pos2 = s_req.find("HTTP")) != string::npos )
     {
      if ( 13 < i_pos2 )
      {     
       s_ret = url_decode( s_req.substr(6,i_pos2-7) );

       int i_len  = s_ret.length();
       int i_len2 = s_req.length()-1;

       s_req = s_req.substr( i_len < i_len2 ? i_len : i_len2 );

       if ( (i_pos = s_req.find("event=")) == string::npos)
       {
        char c_req[POSTBUF];
        i_len = read(p_thrd->get_sock(), c_req, POSTBUF);
        s_req = c_req;
        s_req = s_req.substr(0, i_len);

        if ( (i_pos = s_req.find("event=")) != string::npos )
         get_request_parameters( url_decode( s_req.substr(i_pos) ), map_params ); 
       }
       else
       {
        get_request_parameters( url_decode( s_req.substr(i_pos) ), map_params ); 
       }
      }
     }
    }

#ifdef VERBOSE
    wrap::system_message( REQUEST + s_ret );
#endif

    if ( s_ret.empty() )
     s_ret = wrap::CONF->get_elem( "httpd.startsite" );

    else
     s_ret = remove_dots(s_ret);
   
    map_params["request"] = s_ret;

    return s_ret;
}

string
reqp::get_content_type( string s_file )
{
    string s_ext(tool::get_extension( s_file ));

    if( s_ext == "" )
     s_ext = "default";

    return wrap::CONF->get_elem( "httpd.contenttypes." + s_ext );
}

void
reqp::parse_headers( string s_req, map_string &map_params )
{
    int pos = s_req.find("\n");

    if (pos != string::npos)
    {
     map_params["QUERY_STRING"] = tool::trim(s_req.substr(0,pos-1));

     do
     {
       string s_line( s_req.substr(0, pos) );
       int pos2 = s_line.find(":");

       if (pos2 != string::npos && s_line.length() > pos2+1)
        map_params[ tool::trim(s_line.substr(0, pos2)) ] = tool::trim(s_line.substr(pos2+1));

       s_req = s_req.substr( s_line.size() + 1 );
       pos = s_req.find("\n");
     }
     while( pos != string::npos);
    } // if
}

int
reqp::htoi(string *s)
{
    int value;
    int c;

    c = s->at(0);

    if( isupper(c) )
     c = tolower(c);

    value = (c >= '0' && c <= '9' ? c - '0' : c - 'a' + 10) * 16;

    c = s->at(1);

    if( isupper(c) )
     c = tolower(c);

    value += c >= '0' && c <= '9' ? c - '0' : c - 'a' + 10;

    return value;
}

string
reqp::url_decode( string s_str )
{
    string s_dest = "";
    int i_len = s_str.size();
    int i_prv = i_len - 2;

    for( int i = 0; i < i_len; ++i)
    {
        char ch = s_str.at(i);
        if( ch == '+' )
        {
            s_dest += " ";
        }

        else if (ch == '%' && i < i_prv)
        {
            string s_tmp = s_str.substr(i+1, 2);
            ch = (char) htoi(&s_tmp);
            s_dest += ch;
            i += 2;
        }

        else
	{
            s_dest += ch;
	}
    }

    return s_dest;
}

string
reqp::get_from_header( string s_req, string s_hdr )
{
    unsigned i_pos[2];
    if ( (i_pos[0] = s_req.find( s_hdr, 0 )) == string::npos )
     return "";

    if ( (i_pos[1] = s_req.find( "\n", i_pos[0]) ) == string::npos )
     return "";

    int i_len = s_hdr.length();
    return s_req.substr( i_pos[0] + i_len, i_pos[1] - i_pos[0] - i_len - 1 );
}

string
reqp::parse( thrd* p_thrd, string s_req, map_string &map_params )
{
    // store all request informations in map_params. store the url in
    // map_params["request"].
    get_url( p_thrd, s_req, map_params );

    parse_headers( s_req, map_params );
    string s_event( map_params["event"] );

    // create the http header.
    string s_rep( HTTP_CODEOK );
    s_rep.append( HTTP_SERVER );
    s_rep.append( HTTP_CONTAC );
    s_rep.append( HTTP_EXPIRE );
    s_rep.append( HTTP_CACHEC );
    s_rep.append( HTTP_CONNEC );

    if ( s_event.compare("stream") == 0 )
     s_rep.append( HTTP_CHUNKE );

    s_rep.append( HTTP_COTYPE );
    s_rep.append( get_content_type( map_params["request"] ) );
    s_rep.append("\r\n\r\n");

 
    if ( wrap::CONF->get_elem("httpd.enablecgi").compare("true") == 0 && 
         string::npos != map_params["request"].find(".cgi") )
    { 
       s_rep.append( tool::shell_command( 
         wrap::CONF->get_elem("httpd.templatedir") + map_params["request"], 
         METH_RETSTRING ) ); 
    }

    else
    {
      // parse and get the requested html-template and also use
      // the values stored in map_params for %%KEY%% substituations.
      s_rep.append( wrap::HTML->parse( map_params ) );
    }

    // return the parsed html-template.
    return  s_rep;
}


string
reqp::remove_dots( string s_ret )
{
    // remove ".." from the request.
    unsigned pos;

    if( (pos = s_ret.find( ".." )) != string::npos)
     return s_ret.substr(0, pos);   

    return s_ret;
}

#endif