summaryrefslogtreecommitdiffabout
path: root/lib
Unidiff
Diffstat (limited to 'lib') (more/less context) (show whitespace changes)
-rw-r--r--lib/openid_message.cc9
1 files changed, 6 insertions, 3 deletions
diff --git a/lib/openid_message.cc b/lib/openid_message.cc
index 4b9179b..e244f43 100644
--- a/lib/openid_message.cc
+++ b/lib/openid_message.cc
@@ -1,126 +1,129 @@
1#include <cassert> 1#include <cassert>
2#include <opkele/types.h> 2#include <opkele/types.h>
3#include <opkele/exception.h> 3#include <opkele/exception.h>
4#include <opkele/util.h> 4#include <opkele/util.h>
5#include <opkele/debug.h> 5#include <opkele/debug.h>
6 6
7#include "config.h" 7#include "config.h"
8 8
9namespace opkele { 9namespace opkele {
10 using std::input_iterator_tag; 10 using std::input_iterator_tag;
11 using std::unary_function; 11 using std::unary_function;
12 12
13 struct __om_copier : public unary_function<const string&,void> { 13 struct __om_copier : public unary_function<const string&,void> {
14 public: 14 public:
15 const basic_openid_message& from; 15 const basic_openid_message& from;
16 basic_openid_message& to; 16 basic_openid_message& to;
17 17
18 __om_copier(basic_openid_message& t,const basic_openid_message& f) 18 __om_copier(basic_openid_message& t,const basic_openid_message& f)
19 : from(f), to(t) { 19 : from(f), to(t) { }
20 to.reset_fields();
21 }
22 20
23 result_type operator()(argument_type f) { 21 result_type operator()(argument_type f) {
24 to.set_field(f,from.get_field(f)); } 22 to.set_field(f,from.get_field(f)); }
25 }; 23 };
26 24
27 basic_openid_message::basic_openid_message(const basic_openid_message& x) { 25 basic_openid_message::basic_openid_message(const basic_openid_message& x) {
28 x.copy_to(*this); 26 x.copy_to(*this);
29 } 27 }
30 void basic_openid_message::copy_to(basic_openid_message& x) const { 28 void basic_openid_message::copy_to(basic_openid_message& x) const {
29 x.reset_fields();
30 for_each(fields_begin(),fields_end(),
31 __om_copier(x,*this) );
32 }
33 void basic_openid_message::append_to(basic_openid_message& x) const {
31 for_each(fields_begin(),fields_end(), 34 for_each(fields_begin(),fields_end(),
32 __om_copier(x,*this) ); 35 __om_copier(x,*this) );
33 } 36 }
34 37
35 struct __om_ns_finder : public unary_function<const string&,bool> { 38 struct __om_ns_finder : public unary_function<const string&,bool> {
36 public: 39 public:
37 const basic_openid_message& om; 40 const basic_openid_message& om;
38 const string& uri; 41 const string& uri;
39 42
40 __om_ns_finder(const basic_openid_message& m, 43 __om_ns_finder(const basic_openid_message& m,
41 const string& u) : om(m), uri(u) { } 44 const string& u) : om(m), uri(u) { }
42 45
43 result_type operator()(argument_type f) { 46 result_type operator()(argument_type f) {
44 return 47 return
45 (!strncmp(f.c_str(),"ns.",sizeof("ns.")-1)) 48 (!strncmp(f.c_str(),"ns.",sizeof("ns.")-1))
46 && om.get_field(f)==uri ; 49 && om.get_field(f)==uri ;
47 } 50 }
48 }; 51 };
49 52
50 bool basic_openid_message::has_ns(const string& uri) const { 53 bool basic_openid_message::has_ns(const string& uri) const {
51 fields_iterator ei = fields_end(); 54 fields_iterator ei = fields_end();
52 fields_iterator i = find_if(fields_begin(),fields_end(), 55 fields_iterator i = find_if(fields_begin(),fields_end(),
53 __om_ns_finder(*this,uri)); 56 __om_ns_finder(*this,uri));
54 return !(i==ei); 57 return !(i==ei);
55 } 58 }
56 string basic_openid_message::get_ns(const string& uri) const { 59 string basic_openid_message::get_ns(const string& uri) const {
57 fields_iterator ei = fields_end(); 60 fields_iterator ei = fields_end();
58 fields_iterator i = find_if(fields_begin(),fields_end(), 61 fields_iterator i = find_if(fields_begin(),fields_end(),
59 __om_ns_finder(*this,uri)); 62 __om_ns_finder(*this,uri));
60 if(i==ei) 63 if(i==ei)
61 throw failed_lookup(OPKELE_CP_ string("failed to find namespace ")+uri); 64 throw failed_lookup(OPKELE_CP_ string("failed to find namespace ")+uri);
62 return i->substr(3); 65 return i->substr(3);
63 } 66 }
64 67
65 struct __om_query_builder : public unary_function<const string&,void> { 68 struct __om_query_builder : public unary_function<const string&,void> {
66 public: 69 public:
67 const basic_openid_message& om; 70 const basic_openid_message& om;
68 bool first; 71 bool first;
69 string& rv; 72 string& rv;
70 const char *pfx; 73 const char *pfx;
71 74
72 __om_query_builder(const char *p,string& r,const basic_openid_message& m) 75 __om_query_builder(const char *p,string& r,const basic_openid_message& m)
73 : om(m), first(true), rv(r), pfx(p) { 76 : om(m), first(true), rv(r), pfx(p) {
74 for_each(om.fields_begin(),om.fields_end(),*this); 77 for_each(om.fields_begin(),om.fields_end(),*this);
75 } 78 }
76 __om_query_builder(const char *p,string& r,const basic_openid_message& m,const string& u) 79 __om_query_builder(const char *p,string& r,const basic_openid_message& m,const string& u)
77 : om(m), first(true), rv(r), pfx(p) { 80 : om(m), first(true), rv(r), pfx(p) {
78 rv = u; 81 rv = u;
79 if(rv.find('?')==string::npos) 82 if(rv.find('?')==string::npos)
80 rv += '?'; 83 rv += '?';
81 else 84 else
82 first = false; 85 first = false;
83 for_each(om.fields_begin(),om.fields_end(),*this); 86 for_each(om.fields_begin(),om.fields_end(),*this);
84 } 87 }
85 88
86 result_type operator()(argument_type f) { 89 result_type operator()(argument_type f) {
87 if(first) 90 if(first)
88 first = false; 91 first = false;
89 else 92 else
90 rv += '&'; 93 rv += '&';
91 if(pfx) rv += pfx; 94 if(pfx) rv += pfx;
92 rv+= f; 95 rv+= f;
93 rv += '='; 96 rv += '=';
94 rv += util::url_encode(om.get_field(f)); 97 rv += util::url_encode(om.get_field(f));
95 } 98 }
96 }; 99 };
97 100
98 string basic_openid_message::append_query(const string& url,const char *pfx) const { 101 string basic_openid_message::append_query(const string& url,const char *pfx) const {
99 string rv; 102 string rv;
100 return __om_query_builder(pfx,rv,*this,url).rv; 103 return __om_query_builder(pfx,rv,*this,url).rv;
101 } 104 }
102 string basic_openid_message::query_string(const char *pfx) const { 105 string basic_openid_message::query_string(const char *pfx) const {
103 string rv; 106 string rv;
104 return __om_query_builder(pfx,rv,*this).rv; 107 return __om_query_builder(pfx,rv,*this).rv;
105 } 108 }
106 109
107 void basic_openid_message::reset_fields() { 110 void basic_openid_message::reset_fields() {
108 throw not_implemented(OPKELE_CP_ "reset_fields() not implemented"); 111 throw not_implemented(OPKELE_CP_ "reset_fields() not implemented");
109 } 112 }
110 void basic_openid_message::set_field(const string&,const string&) { 113 void basic_openid_message::set_field(const string&,const string&) {
111 throw not_implemented(OPKELE_CP_ "set_field() not implemented"); 114 throw not_implemented(OPKELE_CP_ "set_field() not implemented");
112 } 115 }
113 void basic_openid_message::reset_field(const string&) { 116 void basic_openid_message::reset_field(const string&) {
114 throw not_implemented(OPKELE_CP_ "reset_field() not implemented"); 117 throw not_implemented(OPKELE_CP_ "reset_field() not implemented");
115 } 118 }
116 119
117 void basic_openid_message::from_keyvalues(const string& kv) { 120 void basic_openid_message::from_keyvalues(const string& kv) {
118 reset_fields(); 121 reset_fields();
119 string::size_type p = 0; 122 string::size_type p = 0;
120 while(true) { 123 while(true) {
121 string::size_type co = kv.find(':',p); 124 string::size_type co = kv.find(':',p);
122 if(co==string::npos) 125 if(co==string::npos)
123 break; 126 break;
124#ifndef POSTELS_LAW 127#ifndef POSTELS_LAW
125 string::size_type nl = kv.find('\n',co+1); 128 string::size_type nl = kv.find('\n',co+1);
126 if(nl==string::npos) 129 if(nl==string::npos)