1 #include <cassert> 2 #include <algorithm> 3 4 #include <opkele/types.h> 5 #include <opkele/exception.h> 6 #include <opkele/util.h> 7 #include <opkele/debug.h> 8 9 #include "config.h" 10 11 namespace opkele { 12 using std::input_iterator_tag; 13 using std::unary_function; 14 15 16 struct __om_ns_finder : public unary_function<const string&,bool> { 17 public: 18 const basic_openid_message& om; 19 const string& uri; 20 __om_ns_finderopkele::__om_ns_finder21 __om_ns_finder(const basic_openid_message& m, 22 const string& u) : om(m), uri(u) { } 23 operator ()opkele::__om_ns_finder24 result_type operator()(argument_type f) { 25 return 26 (!strncmp(f.c_str(),"ns.",sizeof("ns.")-1)) 27 && om.get_field(f)==uri ; 28 } 29 }; 30 has_ns(const string & uri) const31 bool basic_openid_message::has_ns(const string& uri) const { 32 fields_iterator ei = fields_end(); 33 fields_iterator i = find_if(fields_begin(),fields_end(), 34 __om_ns_finder(*this,uri)); 35 return !(i==ei); 36 } get_ns(const string & uri) const37 string basic_openid_message::get_ns(const string& uri) const { 38 fields_iterator ei = fields_end(); 39 fields_iterator i = find_if(fields_begin(),fields_end(), 40 __om_ns_finder(*this,uri)); 41 if(i==ei) 42 throw failed_lookup(OPKELE_CP_ string("failed to find namespace ")+uri); 43 return i->substr(3); 44 } 45 from_keyvalues(const string & kv)46 void basic_openid_message::from_keyvalues(const string& kv) { 47 reset_fields(); 48 string::size_type p = 0; 49 while(true) { 50 string::size_type co = kv.find(':',p); 51 if(co==string::npos) 52 break; 53 #ifndef POSTELS_LAW 54 string::size_type nl = kv.find('\n',co+1); 55 if(nl==string::npos) 56 throw bad_input(OPKELE_CP_ "malformed input"); 57 if(nl>co) 58 set_field(kv.substr(p,co-p),kv.substr(co+1,nl-co-1)); 59 p = nl+1; 60 #else /* POSTELS_LAW */ 61 string::size_type lb = kv.find_first_of("\r\n",co+1); 62 if(lb==string::npos) { 63 set_field(kv.substr(p,co-p),kv.substr(co+1)); 64 break; 65 } 66 if(lb>co) 67 set_field(kv.substr(p,co-p),kv.substr(co+1,lb-co-1)); 68 string::size_type nolb = kv.find_first_not_of("\r\n",lb); 69 if(nolb==string::npos) 70 break; 71 p = nolb; 72 #endif /* POSTELS_LAW */ 73 } 74 } 75 76 struct __om_kv_outputter : public unary_function<const string&,void> { 77 public: 78 const basic_openid_message& om; 79 ostream& os; 80 __om_kv_outputteropkele::__om_kv_outputter81 __om_kv_outputter(const basic_openid_message& m,ostream& s) 82 : om(m), os(s) { } 83 operator ()opkele::__om_kv_outputter84 result_type operator()(argument_type f) { 85 os << f << ':' << om.get_field(f) << '\n'; 86 } 87 }; 88 to_keyvalues(ostream & o) const89 void basic_openid_message::to_keyvalues(ostream& o) const { 90 for_each(fields_begin(),fields_end(),__om_kv_outputter(*this,o)); 91 } 92 93 struct __om_html_outputter : public unary_function<const string&,void> { 94 public: 95 const basic_openid_message& om; 96 ostream& os; 97 const char *pfx; 98 __om_html_outputteropkele::__om_html_outputter99 __om_html_outputter(const basic_openid_message& m,ostream& s,const char *p=0) 100 : om(m), os(s), pfx(p) { } 101 operator ()opkele::__om_html_outputter102 result_type operator()(argument_type f) { 103 os << 104 "<input type=\"hidden\"" 105 " name=\""; 106 if(pfx) 107 os << util::attr_escape(pfx); 108 os << util::attr_escape(f) << "\"" 109 " value=\"" << util::attr_escape(om.get_field(f)) << "\" />"; 110 } 111 }; 112 to_htmlhiddens(ostream & o,const char * pfx) const113 void basic_openid_message::to_htmlhiddens(ostream& o,const char* pfx) const { 114 for_each(fields_begin(),fields_end(),__om_html_outputter(*this,o,pfx)); 115 } 116 add_to_signed(const string & fields)117 void basic_openid_message::add_to_signed(const string& fields) { 118 string::size_type fnc = fields.find_first_not_of(","); 119 if(fnc==string::npos) 120 throw bad_input(OPKELE_CP_ "Trying to add nothing in particular to the list of signed fields"); 121 string signeds; 122 try { 123 signeds = get_field("signed"); 124 string::size_type lnc = signeds.find_last_not_of(","); 125 if(lnc==string::npos) 126 signeds.assign(fields,fnc,fields.size()-fnc); 127 else{ 128 string::size_type ss = signeds.size(); 129 if(lnc==(ss-1)) { 130 signeds+= ','; 131 signeds.append(fields,fnc,fields.size()-fnc); 132 }else{ 133 if(lnc<(ss-2)) 134 signeds.replace(lnc+2,ss-lnc-2, 135 fields,fnc,fields.size()-fnc); 136 else 137 signeds.append(fields,fnc,fields.size()-fnc); 138 } 139 } 140 }catch(failed_lookup&) { 141 signeds.assign(fields,fnc,fields.size()-fnc); 142 } 143 set_field("signed",signeds); 144 } 145 find_ns(const string & uri,const char * pfx) const146 string basic_openid_message::find_ns(const string& uri,const char *pfx) const { 147 try { 148 return get_ns(uri); 149 }catch(failed_lookup&) { 150 return pfx; 151 } 152 } allocate_ns(const string & uri,const char * pfx)153 string basic_openid_message::allocate_ns(const string& uri,const char *pfx) { 154 if(!has_field("ns")) 155 return pfx; 156 if(has_ns(uri)) 157 throw bad_input(OPKELE_CP_ "OpenID message already contains namespace"); 158 string rv = pfx; 159 if(has_field("ns."+rv)) { 160 string::reference c=rv[rv.length()]; 161 for(c='a';c<='z' && has_field("ns."+rv);++c) ; 162 if(c=='z') 163 throw exception(OPKELE_CP_ "Failed to allocate namespace"); 164 } 165 set_field("ns."+rv,uri); 166 return rv; 167 } 168 has_field(const string & n) const169 bool openid_message_t::has_field(const string& n) const { 170 return find(n)!=end(); 171 } get_field(const string & n) const172 const string& openid_message_t::get_field(const string& n) const { 173 const_iterator i=find(n); 174 if(i==end()) 175 throw failed_lookup(OPKELE_CP_ n+": no such field"); 176 return i->second; 177 } 178 fields_begin() const179 openid_message_t::fields_iterator openid_message_t::fields_begin() const { 180 return util::map_keys_iterator<const_iterator,string,const string&,const string*>(begin(),end()); 181 } fields_end() const182 openid_message_t::fields_iterator openid_message_t::fields_end() const { 183 return util::map_keys_iterator<const_iterator,string,const string&,const string*>(end(),end()); 184 } 185 reset_fields()186 void openid_message_t::reset_fields() { 187 clear(); 188 } set_field(const string & n,const string & v)189 void openid_message_t::set_field(const string& n,const string& v) { 190 (*this)[n]=v; 191 } reset_field(const string & n)192 void openid_message_t::reset_field(const string& n) { 193 erase(n); 194 } 195 196 } 197