| -rw-r--r-- | src/util.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/util.cc b/src/util.cc index 48e486a..82a4a0a 100644 --- a/src/util.cc +++ b/src/util.cc @@ -1,106 +1,108 @@ +#include <cstdlib> +#include <cstring> #include "kingate/util.h" #include "kingate/exception.h" namespace kingate { static const char *safeChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" "abcdefghijklmnopqrstuvwxyz" "0123456789" "_-" ; string url_encode(const string& str) { string rv = str; string::size_type screwed = 0; for(;;) { screwed = rv.find_first_not_of(safeChars,screwed); if(screwed == string::npos) break; while(screwed<rv.length() && !strchr(safeChars,rv.at(screwed))) { char danger = rv.at(screwed); if(danger==' ') { rv.replace(screwed++,1,1,'+'); }else{ static char tmp[4] = {'%',0,0,0}; snprintf(&tmp[1],3,"%02X",0xFF&(int)danger); rv.replace(screwed,1,tmp,3); screwed+=3; } } } return rv; } string url_decode(const string& str) { string rv = str; string::size_type unscrewed = 0; for(;;) { unscrewed = rv.find_first_of("%+",unscrewed); if(unscrewed == string::npos) break; if(rv.at(unscrewed)=='+') { rv.replace(unscrewed++,1,1,' '); }else{ if((rv.length()-unscrewed)<3) throw exception(CODEPOINT,"incorrectly escaped string"); // XXX: ensure it's hex? int danger = strtol(rv.substr(unscrewed+1,2).c_str(),NULL,16); rv.replace(unscrewed,3,1,danger); unscrewed++; } } return rv; } /* * RFC 2616: * * separators = "(" | ")" | "<" | ">" | "@" * | "," | ";" | ":" | "\" | <"> * | "/" | "[" | "]" | "?" | "=" * | "{" | "}" | SP | HT */ /* * RFC 2616: * * token = 1*<any CHAR except CTLs or separators> */ static const char *quotible_chars = "\001\002\003\004\005\006\007\010" "\011\012\013\014\015\016\017\020" "\021\022\023\024\025\026\027\030" "\031\032\033\034\035\036\037\040" "()<>@,;:\\\"/[]?={}" /* separator chars (except for SP and HT mentioned elsewhere */ "\177" ; /* * RFC 2616: * * quoted-string = ( <"> *(qdtext | quoted-pair ) <"> ) * qdtext = <any TEXT except <">> * * The backslash character ("\") MAY be used as a single-character * quoting mechanism only within quoted-string and comment constructs. * * quoted-pair = "\" CHAR */ string http_quoted_string(const string& str) { string rv = str; string::size_type sp=0; for(string::size_type q=rv.find('"');(q=rv.find('"',q))!=string::npos;q+=2) rv.insert(q,1,'\\'); rv.insert(0,1,'"'); rv += '"'; return rv; } string http_quote(const string& str) { if(str.find_first_of(quotible_chars)==string::npos) return str; return http_quoted_string(str); } } |