/*
* Copyright 2017 Chris Xiong
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#ifndef CGILIB_HPP
#define CGILIB_HPP
#include <cstdlib>
#include <map>
#include <string>
#include <vector>
const char* base64_table="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/=";
std::string base64_encode(std::string s)
{
std::string r="";
for(unsigned i=0;i<s.length();i+=3)
{
int b=(s[i]&0xFC)>>2;
r+=base64_table[b];
b=(s[i]&0x03)<<4;
if(i+1<s.length())
{
b|=(s[i+1]&0xF0)>>4;
r+=base64_table[b];
b=(s[i+1]&0x0F)<<2;
if(i+2<s.length())
{
b|=(s[i+2]&0xC0)>>6;
r+=base64_table[b];
b=s[i+2]&0x3F;
r+=base64_table[b];
}else r+=base64_table[b],r+='=';
}else r+=base64_table[b],r+="==";
}
return r;
}
void split(std::string s,char c,std::vector<std::string>& v)
{
v.clear();
for(size_t anch=0;;)
{
std::string sec;
if(s.find(c,anch)==std::string::npos)
sec=s.substr(anch);
else sec=s.substr(anch,s.find(c,anch)-anch);
v.push_back(sec);
if(s.find(c,anch)==std::string::npos)break;
anch=s.find(c,anch)+1;
}
}
std::string trim(std::string s)
{
int l=0;for(;isblank(s[l]);++l)
s=s.substr(l);
while(isblank(s.back()))s.pop_back();
return s;
}
class QueryStrParser
{
private:
std::map<std::string,std::string> q;
void parse(std::string es)
{
for(size_t anch=0;;)
{
std::string sec;
if(es.find('&',anch)==std::string::npos)
sec=es.substr(anch);
else sec=es.substr(anch,es.find('&',anch)-anch);
if(sec.find('=')==std::string::npos)
q[sec.substr(0)]="";
else
q[sec.substr(0,sec.find('='))]=sec.substr(sec.find('=')+1);
if(es.find('&',anch)==std::string::npos)break;
anch=es.find('&',anch)+1;
}
}
public:
QueryStrParser()
{
char* e=getenv("QUERY_STRING");
if(!e)return;
parse(std::string(e));
}
QueryStrParser(std::string s)
{
parse(s);
}
bool exist(std::string s)
{
return q.find(s)!=q.end();
}
std::string value(std::string s)
{
if(!exist(s))return "";
return q.find(s)->second;
}
};
class RequestCookies
{
private:
std::map<std::string,std::string> m;
void parse(std::string cookie)
{
std::vector<std::string> v;
split(cookie,';',v);
}
public:
RequestCookies()
{
char* e=getenv("HTTP_COOKIE");
if(!e)return;
parse(std::string(e));
}
};
class DOMAttrib
{
private:
std::map<std::string,std::string> m;
public:
DOMAttrib(std::string s="")
{
for(size_t anch=0;;)
{
std::string sec;
if(s.find('"',anch)==std::string::npos)
sec=s.substr(anch);
else sec=s.substr(anch,s.find('"',s.find('"',anch)+1)-anch+1);
if(sec.find('=')==std::string::npos)
m[sec.substr(0)]="";
else
m[sec.substr(0,sec.find('='))]=sec.substr(sec.find('=')+2),
m[sec.substr(0,sec.find('='))].pop_back();
if(s.find(' ',anch+sec.length())==std::string::npos)break;
anch=s.find(' ',anch+sec.length())+1;
}
}
void setAttrib(std::string a,std::string v){m[a]=v;}
void eraseAttrib(std::string a){if(existAttrib(a))m.erase(m.find(a));}
bool existAttrib(std::string a){return m.find(a)!=m.end();}
std::string getAttrib(std::string a){if(!existAttrib(a))return "";return m[a];}
std::string to_string()
{
std::string r;bool cf=false;
for(auto i=m.begin();i!=m.end();++i)
{
if(cf)r+=' ';else cf=true;
r+=i->first+"=\""+i->second+"\"";
}
return r;
}
};
static const char* twoxx[]={
"OK",
"Created",
"Accepted",
"Non-Authoritative Information",
"No Content",
"Reset Content",
"Partial Content",
"Multi-Status",
"Already Reported"
};
static const char* threexx[]={
"Multiple Choices",
"Moved Permanently",
"Found",
"See Other",
"Not Modified",
"Use Proxy",
"???",
"Temporary Redirect"
};
static const char* fourxx[]={
"Bad Request",
"Unauthorized",
"Payment Required",
"Forbidden",
"Not Found",
"Method Not Allowed",
"Not Acceptable",
"Proxy Authentication Required",
"Request Timeout",
"Conflict",
"Gone",
"Length Required",
"Precondition Failed",
"Request Entity Too Large",
"Request-URI Too Long",
"Unsupported Media Type",
"Requested Range Not Satisfiable",
"Expectation Failed"
};
static const char* fivexx[]={
"Internal Server Error",
"Not Implemented",
"Bad Gateway",
"Service Unavailable",
"Gateway Timeout",
"HTTP Version Not Supported"
};
class HTTPHeader
{
private:
int status;
std::vector<std::string> hdr;
public:
HTTPHeader(){status=200;}
void setStatusCode(int c){status=c;}
int statusCode(){return status;}
void appendHeader(std::string s){hdr.push_back(s);}
void print()
{
if(status>=200&&status<=208)
printf("Status: %d %s\r\n",status,twoxx[status-200]);
if(status>=300&&status<=307)
printf("Status: %d %s\r\n",status,threexx[status-300]);
if(status>=400&&status<=417)
printf("Status: %d %s\r\n",status,fourxx[status-400]);
if(status>=500&&status<=505)
printf("Status: %d %s\r\n",status,fivexx[status-500]);
for(unsigned i=0;i<hdr.size();++i)
printf("%s\r\n",hdr[i].c_str());
printf("\r\n");
}
};
#endif
