-rw-r--r-- | lib/params.cc | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/lib/params.cc b/lib/params.cc index b181811..ea86d3a 100644 --- a/lib/params.cc +++ b/lib/params.cc | |||
@@ -1,92 +1,108 @@ | |||
1 | #include <opkele/types.h> | 1 | #include <opkele/types.h> |
2 | #include <opkele/exception.h> | 2 | #include <opkele/exception.h> |
3 | #include <opkele/util.h> | 3 | #include <opkele/util.h> |
4 | #include <openssl/sha.h> | 4 | #include <openssl/sha.h> |
5 | #include <openssl/hmac.h> | 5 | #include <openssl/hmac.h> |
6 | 6 | ||
7 | #include "config.h" | ||
8 | |||
7 | namespace opkele { | 9 | namespace opkele { |
8 | using namespace std; | 10 | using namespace std; |
9 | 11 | ||
10 | bool params_t::has_param(const string& n) const { | 12 | bool params_t::has_param(const string& n) const { |
11 | return find(n)!=end(); | 13 | return find(n)!=end(); |
12 | } | 14 | } |
13 | const string& params_t::get_param(const string& n) const { | 15 | const string& params_t::get_param(const string& n) const { |
14 | const_iterator i = find(n); | 16 | const_iterator i = find(n); |
15 | if(i==end()) | 17 | if(i==end()) |
16 | throw failed_lookup(OPKELE_CP_ n+": no such parameter"); | 18 | throw failed_lookup(OPKELE_CP_ n+": no such parameter"); |
17 | return i->second; | 19 | return i->second; |
18 | } | 20 | } |
19 | string& params_t::get_param(const string& n) { | 21 | string& params_t::get_param(const string& n) { |
20 | iterator i = find(n); | 22 | iterator i = find(n); |
21 | if(i==end()) | 23 | if(i==end()) |
22 | throw failed_lookup(OPKELE_CP_ n+": no such parameter"); | 24 | throw failed_lookup(OPKELE_CP_ n+": no such parameter"); |
23 | return i->second; | 25 | return i->second; |
24 | } | 26 | } |
25 | 27 | ||
26 | void params_t::parse_keyvalues(const string& kv) { | 28 | void params_t::parse_keyvalues(const string& kv) { |
27 | clear(); | 29 | clear(); |
28 | string::size_type p = 0; | 30 | string::size_type p = 0; |
29 | while(true) { | 31 | while(true) { |
30 | string::size_type co = kv.find(':',p); | 32 | string::size_type co = kv.find(':',p); |
31 | if(co==string::npos) | 33 | if(co==string::npos) |
32 | break; | 34 | break; |
35 | #ifndef POSTELS_LAW | ||
33 | string::size_type nl = kv.find('\n',co+1); | 36 | string::size_type nl = kv.find('\n',co+1); |
34 | if(nl==string::npos) | 37 | if(nl==string::npos) |
35 | throw bad_input(OPKELE_CP_ "malformed input"); | 38 | throw bad_input(OPKELE_CP_ "malformed input"); |
36 | if(nl>co) | 39 | if(nl>co) |
37 | insert(value_type(kv.substr(p,co-p),kv.substr(co+1,nl-co-1))); | 40 | insert(value_type(kv.substr(p,co-p),kv.substr(co+1,nl-co-1))); |
38 | p = nl+1; | 41 | p = nl+1; |
42 | #else /* POSTELS_LAW */ | ||
43 | string::size_type lb = kv.find_first_of("\r\n",co+1); | ||
44 | if(lb==string::npos) { | ||
45 | insert(value_type(kv.substr(p,co-p),kv.substr(co+1))); | ||
46 | break; | ||
47 | } | ||
48 | if(lb>co) | ||
49 | insert(value_type(kv.substr(p,co-p),kv.substr(co+1,lb-co-1))); | ||
50 | string::size_type nolb = kv.find_first_not_of("\r\n",lb); | ||
51 | if(nolb==string::npos) | ||
52 | break; | ||
53 | p = nolb; | ||
54 | #endif /* POSTELS_LAW */ | ||
39 | } | 55 | } |
40 | } | 56 | } |
41 | 57 | ||
42 | void params_t::sign(secret_t secret,string& sig,const string& slist,const char *prefix) const { | 58 | void params_t::sign(secret_t secret,string& sig,const string& slist,const char *prefix) const { |
43 | string kv; | 59 | string kv; |
44 | string::size_type p = 0; | 60 | string::size_type p = 0; |
45 | while(true) { | 61 | while(true) { |
46 | string::size_type co = slist.find(',',p); | 62 | string::size_type co = slist.find(',',p); |
47 | string f = (co==string::npos)?slist.substr(p):slist.substr(p,co-p); | 63 | string f = (co==string::npos)?slist.substr(p):slist.substr(p,co-p); |
48 | kv += f; | 64 | kv += f; |
49 | kv += ':'; | 65 | kv += ':'; |
50 | if(prefix) f.insert(0,prefix); | 66 | if(prefix) f.insert(0,prefix); |
51 | kv += get_param(f); | 67 | kv += get_param(f); |
52 | kv += '\n'; | 68 | kv += '\n'; |
53 | if(co==string::npos) | 69 | if(co==string::npos) |
54 | break; | 70 | break; |
55 | p = co+1; | 71 | p = co+1; |
56 | } | 72 | } |
57 | unsigned int md_len = 0; | 73 | unsigned int md_len = 0; |
58 | unsigned char *md = HMAC( | 74 | unsigned char *md = HMAC( |
59 | EVP_sha1(), | 75 | EVP_sha1(), |
60 | &(secret.front()),secret.size(), | 76 | &(secret.front()),secret.size(), |
61 | (const unsigned char *)kv.data(),kv.length(), | 77 | (const unsigned char *)kv.data(),kv.length(), |
62 | 0,&md_len); | 78 | 0,&md_len); |
63 | sig = util::encode_base64(md,md_len); | 79 | sig = util::encode_base64(md,md_len); |
64 | } | 80 | } |
65 | 81 | ||
66 | string params_t::append_query(const string& url,const char *prefix) const { | 82 | string params_t::append_query(const string& url,const char *prefix) const { |
67 | string rv = url; | 83 | string rv = url; |
68 | bool p = true; | 84 | bool p = true; |
69 | if(rv.find('?')==string::npos) { | 85 | if(rv.find('?')==string::npos) { |
70 | rv += '?'; | 86 | rv += '?'; |
71 | p = false; | 87 | p = false; |
72 | } | 88 | } |
73 | for(const_iterator i=begin();i!=end();++i) { | 89 | for(const_iterator i=begin();i!=end();++i) { |
74 | if(p) | 90 | if(p) |
75 | rv += '&'; | 91 | rv += '&'; |
76 | else | 92 | else |
77 | p = true; | 93 | p = true; |
78 | rv += prefix; | 94 | rv += prefix; |
79 | rv += i->first; | 95 | rv += i->first; |
80 | rv += '='; | 96 | rv += '='; |
81 | rv += util::url_encode(i->second); | 97 | rv += util::url_encode(i->second); |
82 | } | 98 | } |
83 | return rv; | 99 | return rv; |
84 | } | 100 | } |
85 | 101 | ||
86 | ostream& operator << (ostream& o,const params_t& p) { | 102 | ostream& operator << (ostream& o,const params_t& p) { |
87 | for(params_t::const_iterator i=p.begin();i!=p.end();++i) | 103 | for(params_t::const_iterator i=p.begin();i!=p.end();++i) |
88 | o << i->first << ':' << i->second << '\n'; | 104 | o << i->first << ':' << i->second << '\n'; |
89 | return o; | 105 | return o; |
90 | } | 106 | } |
91 | 107 | ||
92 | } | 108 | } |