-rw-r--r-- | lib/message.cc | 78 |
1 files changed, 1 insertions, 77 deletions
diff --git a/lib/message.cc b/lib/message.cc index 78f20f4..b2324b7 100644 --- a/lib/message.cc +++ b/lib/message.cc | |||
@@ -9,84 +9,8 @@ | |||
9 | namespace opkele { | 9 | namespace 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> { | ||
14 | public: | ||
15 | const basic_message& from; | ||
16 | basic_message& to; | ||
17 | |||
18 | __om_copier(basic_message& t,const basic_message& f) | ||
19 | : from(f), to(t) { } | ||
20 | |||
21 | result_type operator()(argument_type f) { | ||
22 | to.set_field(f,from.get_field(f)); } | ||
23 | }; | ||
24 | |||
25 | basic_message::basic_message(const basic_message& x) { | ||
26 | x.copy_to(*this); | ||
27 | } | ||
28 | void basic_message::copy_to(basic_message& x) const { | ||
29 | x.reset_fields(); | ||
30 | for_each(fields_begin(),fields_end(), | ||
31 | __om_copier(x,*this) ); | ||
32 | } | ||
33 | void basic_message::append_to(basic_message& x) const { | ||
34 | for_each(fields_begin(),fields_end(), | ||
35 | __om_copier(x,*this) ); | ||
36 | } | ||
37 | |||
38 | struct __om_query_builder : public unary_function<const string&,void> { | ||
39 | public: | ||
40 | const basic_message& om; | ||
41 | bool first; | ||
42 | string& rv; | ||
43 | const char *pfx; | ||
44 | |||
45 | __om_query_builder(const char *p,string& r,const basic_message& m) | ||
46 | : om(m), first(true), rv(r), pfx(p) { | ||
47 | for_each(om.fields_begin(),om.fields_end(),*this); | ||
48 | } | ||
49 | __om_query_builder(const char *p,string& r,const basic_message& m,const string& u) | ||
50 | : om(m), first(true), rv(r), pfx(p) { | ||
51 | rv = u; | ||
52 | if(rv.find('?')==string::npos) | ||
53 | rv += '?'; | ||
54 | else | ||
55 | first = false; | ||
56 | for_each(om.fields_begin(),om.fields_end(),*this); | ||
57 | } | ||
58 | |||
59 | result_type operator()(argument_type f) { | ||
60 | if(first) | ||
61 | first = false; | ||
62 | else | ||
63 | rv += '&'; | ||
64 | if(pfx) rv += pfx; | ||
65 | rv+= f; | ||
66 | rv += '='; | ||
67 | rv += util::url_encode(om.get_field(f)); | ||
68 | } | ||
69 | }; | ||
70 | |||
71 | string basic_message::append_query(const string& url,const char *pfx) const { | ||
72 | string rv; | ||
73 | return __om_query_builder(pfx,rv,*this,url).rv; | ||
74 | } | ||
75 | string basic_message::query_string(const char *pfx) const { | ||
76 | string rv; | ||
77 | return __om_query_builder(pfx,rv,*this).rv; | ||
78 | } | ||
79 | |||
80 | void basic_message::reset_fields() { | ||
81 | throw not_implemented(OPKELE_CP_ "reset_fields() not implemented"); | ||
82 | } | ||
83 | void basic_message::set_field(const string&,const string&) { | ||
84 | throw not_implemented(OPKELE_CP_ "set_field() not implemented"); | ||
85 | } | ||
86 | void basic_message::reset_field(const string&) { | ||
87 | throw not_implemented(OPKELE_CP_ "reset_field() not implemented"); | ||
88 | } | ||
89 | 13 | ||
90 | struct __om_ns_finder : public unary_function<const string&,bool> { | 14 | struct __om_ns_finder : public unary_function<const string&,bool> { |
91 | public: | 15 | public: |
92 | const basic_openid_message& om; | 16 | const basic_openid_message& om; |
@@ -128,9 +52,9 @@ namespace opkele { | |||
128 | string::size_type nl = kv.find('\n',co+1); | 52 | string::size_type nl = kv.find('\n',co+1); |
129 | if(nl==string::npos) | 53 | if(nl==string::npos) |
130 | throw bad_input(OPKELE_CP_ "malformed input"); | 54 | throw bad_input(OPKELE_CP_ "malformed input"); |
131 | if(nl>co) | 55 | if(nl>co) |
132 | insert(value_type(kv.substr(p,co-p),kv.substr(co+1,nl-co-1))); | 56 | set_field(kv.substr(p,co-p),kv.substr(co+1,nl-co-1)); |
133 | p = nl+1; | 57 | p = nl+1; |
134 | #else /* POSTELS_LAW */ | 58 | #else /* POSTELS_LAW */ |
135 | string::size_type lb = kv.find_first_of("\r\n",co+1); | 59 | string::size_type lb = kv.find_first_of("\r\n",co+1); |
136 | if(lb==string::npos) { | 60 | if(lb==string::npos) { |