| -rw-r--r-- | lib/sreg.cc | 124 |
1 files changed, 124 insertions, 0 deletions
diff --git a/lib/sreg.cc b/lib/sreg.cc new file mode 100644 index 0000000..08e66b7 --- a/dev/null +++ b/lib/sreg.cc @@ -0,0 +1,124 @@ +#include <opkele/exception.h> +#include <opkele/sreg.h> +#include <algorithm> + +namespace opkele { + using std::find; + + static const struct _sreg_field { + const char *fieldname; + sreg_t::fieldbit_t fieldbit; + } fields[] = { + { "nickname", sreg_t::field_nickname }, + { "email", sreg_t::field_email }, + { "fullname", sreg_t::field_fullname }, + { "dob", sreg_t::field_dob }, + { "gender", sreg_t::field_gender }, + { "postcode", sreg_t::field_postcode }, + { "country", sreg_t::field_country }, + { "language", sreg_t::field_language }, + { "timezone", sreg_t::field_timezone } + }; +# define fields_BEGIN fields +# define fields_END &fields[sizeof(fields)/sizeof(*fields)] + typedef const struct _sreg_field *fields_iterator; + + bool operator==(const struct _sreg_field& fd,const string& fn) { + return fd.fieldname==fn; + } + + void sreg_t::checkid_hook(params_t& p,const string& identity) { + string fr, fo; + for(fields_iterator f=fields_BEGIN;f<fields_END;++f) { + if(f->fieldbit&fields_required) { + if(!fr.empty()) fr+=","; + fr += f->fieldname; + } + if(f->fieldbit&fields_optional) { + if(!fo.empty()) fo+=","; + fo += f->fieldname; + } + } + if(!fr.empty()) p["sreg.required"]=fr; + if(!fo.empty()) p["sreg.optional"]=fo; + if(!policy_url.empty()) p["sreg.policy_url"]=policy_url; + } + + void sreg_t::id_res_hook(const params_t& p,const params_t& sp,const string& identity) { + clear(); + for(fields_iterator f=fields_BEGIN;f<fields_END;++f) { + string fn = "sreg."; fn+=f->fieldname; + if(!sp.has_param(fn)) continue; + has_fields |= f->fieldbit; + response[f->fieldbit]=sp.get_param(fn); + } + } + + const string& sreg_t::get_field(fieldbit_t fb) const { + response_t::const_iterator i = response.find(fb); + if(i==response.end()) + throw failed_lookup(OPKELE_CP_ "no field data available"); + return i->second; + } + + void sreg_t::set_field(fieldbit_t fb,const string& fv) { + response[fb] = fv; + has_fields |= fb; + } + + void sreg_t::reset_field(fieldbit_t fb) { + has_fields &= ~fb; + response.erase(fb); + } + + void sreg_t::clear() { + has_fields = 0; response.clear(); + } + + static long fields_list_to_bitmask(string& fl) { + long rv = 0; + while(!fl.empty()) { + string::size_type co = fl.find(','); + string fn; + if(co==string::npos) { + fn = fl; fl.erase(); + }else{ + fn = fl.substr(0,co); fl.erase(0,co+1); + } + fields_iterator f = find(fields_BEGIN,fields_END,fn); + if(f!=fields_END) + rv |= f->fieldbit; + } + return rv; + } + + void sreg_t::checkid_hook(const params_t& pin,params_t& pout) { + fields_optional = 0; fields_required = 0; policy_url.erase(); + fields_response = 0; + try { + string fl = pin.get_param("openid.sreg.required"); + fields_required = fields_list_to_bitmask(fl); + }catch(failed_lookup&) { } + try { + string fl = pin.get_param("openid.sreg.optional"); + fields_optional = fields_list_to_bitmask(fl); + }catch(failed_lookup&) { } + try { + policy_url = pin.get_param("openid.sreg.policy_url"); + }catch(failed_lookup&) { } + setup_response(pin,pout); + fields_response &= has_fields; + for(fields_iterator f=fields_BEGIN;f<fields_END;++f) { + if(!(f->fieldbit&fields_response)) continue; + if(!pout["signed"].empty()) + pout["signed"] +=','; + string pn = "sreg."; pn += f->fieldname; + pout["signed"] += pn; + pout[pn] = get_field(f->fieldbit); + } + } + + void sreg_t::setup_response(const params_t& pin,params_t& pout) { + fields_response = (fields_required|fields_optional)&has_fields; + } +} |