-rw-r--r-- | lib/discovery.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/lib/discovery.cc b/lib/discovery.cc index 3b90977..d1989ec 100644 --- a/lib/discovery.cc +++ b/lib/discovery.cc | |||
@@ -1,141 +1,142 @@ | |||
1 | #include <list> | 1 | #include <list> |
2 | #include <opkele/curl.h> | 2 | #include <opkele/curl.h> |
3 | #include <opkele/expat.h> | 3 | #include <opkele/expat.h> |
4 | #include <opkele/uris.h> | 4 | #include <opkele/uris.h> |
5 | #include <opkele/discovery.h> | 5 | #include <opkele/discovery.h> |
6 | #include <opkele/exception.h> | 6 | #include <opkele/exception.h> |
7 | #include <opkele/util.h> | 7 | #include <opkele/util.h> |
8 | #include <opkele/tidy.h> | ||
9 | #include <opkele/data.h> | 8 | #include <opkele/data.h> |
10 | #include <opkele/debug.h> | 9 | #include <opkele/debug.h> |
11 | 10 | ||
12 | #include "config.h" | 11 | #include "config.h" |
13 | 12 | ||
13 | #include <opkele/tidy.h> | ||
14 | |||
14 | #define XRDS_HEADER "X-XRDS-Location" | 15 | #define XRDS_HEADER "X-XRDS-Location" |
15 | #define CT_HEADER "Content-Type" | 16 | #define CT_HEADER "Content-Type" |
16 | 17 | ||
17 | namespace opkele { | 18 | namespace opkele { |
18 | using std::list; | 19 | using std::list; |
19 | using xrd::XRD_t; | 20 | using xrd::XRD_t; |
20 | using xrd::service_t; | 21 | using xrd::service_t; |
21 | 22 | ||
22 | /* TODO: the whole discovery thing needs cleanup and optimization due to | 23 | /* TODO: the whole discovery thing needs cleanup and optimization due to |
23 | * many changes of concept. */ | 24 | * many changes of concept. */ |
24 | 25 | ||
25 | static const size_t max_html = 16384; | 26 | static const size_t max_html = 16384; |
26 | 27 | ||
27 | static const struct service_type_t { | 28 | static const struct service_type_t { |
28 | const char *uri; | 29 | const char *uri; |
29 | const char *forceid; | 30 | const char *forceid; |
30 | } op_service_types[] = { | 31 | } op_service_types[] = { |
31 | { STURI_OPENID20_OP, IDURI_SELECT20 }, | 32 | { STURI_OPENID20_OP, IDURI_SELECT20 }, |
32 | { STURI_OPENID20, 0 }, | 33 | { STURI_OPENID20, 0 }, |
33 | { STURI_OPENID11, 0 }, | 34 | { STURI_OPENID11, 0 }, |
34 | { STURI_OPENID10, 0 } | 35 | { STURI_OPENID10, 0 } |
35 | }; | 36 | }; |
36 | enum { | 37 | enum { |
37 | st_index_1 = 2, st_index_2 = 1 | 38 | st_index_1 = 2, st_index_2 = 1 |
38 | }; | 39 | }; |
39 | 40 | ||
40 | 41 | ||
41 | static inline bool is_qelement(const XML_Char *n,const char *qen) { | 42 | static inline bool is_qelement(const XML_Char *n,const char *qen) { |
42 | return !strcasecmp(n,qen); | 43 | return !strcasecmp(n,qen); |
43 | } | 44 | } |
44 | static inline bool is_element(const XML_Char *n,const char *en) { | 45 | static inline bool is_element(const XML_Char *n,const char *en) { |
45 | if(!strcasecmp(n,en)) return true; | 46 | if(!strcasecmp(n,en)) return true; |
46 | int nl = strlen(n), enl = strlen(en); | 47 | int nl = strlen(n), enl = strlen(en); |
47 | if( (nl>=(enl+1)) && n[nl-enl-1]=='\t' | 48 | if( (nl>=(enl+1)) && n[nl-enl-1]=='\t' |
48 | && !strcasecmp(&n[nl-enl],en) ) | 49 | && !strcasecmp(&n[nl-enl],en) ) |
49 | return true; | 50 | return true; |
50 | return false; | 51 | return false; |
51 | } | 52 | } |
52 | 53 | ||
53 | static long element_priority(const XML_Char **a) { | 54 | static long element_priority(const XML_Char **a) { |
54 | for(;*a;++a) | 55 | for(;*a;++a) |
55 | if(!strcasecmp(*(a++),"priority")) { | 56 | if(!strcasecmp(*(a++),"priority")) { |
56 | long rv; | 57 | long rv; |
57 | return (sscanf(*a,"%ld",&rv)==1)?rv:-1; | 58 | return (sscanf(*a,"%ld",&rv)==1)?rv:-1; |
58 | } | 59 | } |
59 | return -1; | 60 | return -1; |
60 | } | 61 | } |
61 | /* TODO: ideally all attributes should be | 62 | /* TODO: ideally all attributes should be |
62 | * retrieved in one run */ | 63 | * retrieved in one run */ |
63 | static const char *element_attr(const XML_Char **a, const char *at) { | 64 | static const char *element_attr(const XML_Char **a, const char *at) { |
64 | for(;*a;++a) | 65 | for(;*a;++a) |
65 | if(!strcasecmp(*(a++),at)) { | 66 | if(!strcasecmp(*(a++),at)) { |
66 | return *a; | 67 | return *a; |
67 | } | 68 | } |
68 | return 0; | 69 | return 0; |
69 | } | 70 | } |
70 | 71 | ||
71 | class idigger_t : public util::curl_t, public util::expat_t { | 72 | class idigger_t : public util::curl_t, public util::expat_t { |
72 | public: | 73 | public: |
73 | string xri_proxy; | 74 | string xri_proxy; |
74 | 75 | ||
75 | enum { | 76 | enum { |
76 | xmode_html = 1, xmode_xrd = 2, xmode_cid = 4, | 77 | xmode_html = 1, xmode_xrd = 2, xmode_cid = 4, |
77 | xmode_noredirs = 8 | 78 | xmode_noredirs = 8 |
78 | }; | 79 | }; |
79 | int xmode; | 80 | int xmode; |
80 | 81 | ||
81 | string xrds_location; | 82 | string xrds_location; |
82 | string http_content_type; | 83 | string http_content_type; |
83 | service_t html_openid1; | 84 | service_t html_openid1; |
84 | service_t html_openid2; | 85 | service_t html_openid2; |
85 | string cdata_buf; | 86 | string cdata_buf; |
86 | long status_code; | 87 | long status_code; |
87 | string status_string; | 88 | string status_string; |
88 | 89 | ||
89 | typedef list<string> pt_stack_t; | 90 | typedef list<string> pt_stack_t; |
90 | pt_stack_t pt_stack; | 91 | pt_stack_t pt_stack; |
91 | int skipping; | 92 | int skipping; |
92 | bool parser_choked; | 93 | bool parser_choked; |
93 | string save_html; | 94 | string save_html; |
94 | 95 | ||
95 | XRD_t *xrd; | 96 | XRD_t *xrd; |
96 | service_t *xrd_service; | 97 | service_t *xrd_service; |
97 | string* cdata; | 98 | string* cdata; |
98 | 99 | ||
99 | idigger_t() | 100 | idigger_t() |
100 | : util::curl_t(easy_init()), | 101 | : util::curl_t(easy_init()), |
101 | util::expat_t(0), | 102 | util::expat_t(0), |
102 | xri_proxy(XRI_PROXY_URL) { | 103 | xri_proxy(XRI_PROXY_URL) { |
103 | CURLcode r; | 104 | CURLcode r; |
104 | (r=misc_sets()) | 105 | (r=misc_sets()) |
105 | || (r=set_write()) | 106 | || (r=set_write()) |
106 | || (r=set_header()) | 107 | || (r=set_header()) |
107 | ; | 108 | ; |
108 | if(r) | 109 | if(r) |
109 | throw exception_curl(OPKELE_CP_ "failed to set curly options",r); | 110 | throw exception_curl(OPKELE_CP_ "failed to set curly options",r); |
110 | } | 111 | } |
111 | ~idigger_t() throw() { } | 112 | ~idigger_t() throw() { } |
112 | 113 | ||
113 | void yadiscover(endpoint_discovery_iterator oi,const string& yurl,const char **types,bool redirs) { | 114 | void yadiscover(endpoint_discovery_iterator oi,const string& yurl,const char **types,bool redirs) { |
114 | idiscovery_t idis; | 115 | idiscovery_t idis; |
115 | idis.xri_identity = false; | 116 | idis.xri_identity = false; |
116 | discover_at(idis,yurl,xmode_html|xmode_xrd|(redirs?0:xmode_noredirs)); | 117 | discover_at(idis,yurl,xmode_html|xmode_xrd|(redirs?0:xmode_noredirs)); |
117 | if(!xrds_location.empty()) { | 118 | if(!xrds_location.empty()) { |
118 | idis.clear(); | 119 | idis.clear(); |
119 | discover_at(idis,xrds_location,xmode_xrd); | 120 | discover_at(idis,xrds_location,xmode_xrd); |
120 | } | 121 | } |
121 | idis.normalized_id = idis.canonicalized_id = yurl; | 122 | idis.normalized_id = idis.canonicalized_id = yurl; |
122 | service_type_t st; | 123 | service_type_t st; |
123 | for(st.uri=*types;*types;st.uri=*(++types)) | 124 | for(st.uri=*types;*types;st.uri=*(++types)) |
124 | queue_endpoints(oi,idis,&st); | 125 | queue_endpoints(oi,idis,&st); |
125 | } | 126 | } |
126 | 127 | ||
127 | string discover(endpoint_discovery_iterator& oi,const string& identity) { | 128 | string discover(endpoint_discovery_iterator& oi,const string& identity) { |
128 | string rv; | 129 | string rv; |
129 | idiscovery_t idis; | 130 | idiscovery_t idis; |
130 | string::size_type fsc = identity.find_first_not_of(data::_whitespace_chars); | 131 | string::size_type fsc = identity.find_first_not_of(data::_whitespace_chars); |
131 | if(fsc==string::npos) | 132 | if(fsc==string::npos) |
132 | throw bad_input(OPKELE_CP_ "whitespace-only identity"); | 133 | throw bad_input(OPKELE_CP_ "whitespace-only identity"); |
133 | string::size_type lsc = identity.find_last_not_of(data::_whitespace_chars); | 134 | string::size_type lsc = identity.find_last_not_of(data::_whitespace_chars); |
134 | assert(lsc!=string::npos); | 135 | assert(lsc!=string::npos); |
135 | if(!strncasecmp(identity.c_str()+fsc,"xri://",sizeof("xri://")-1)) | 136 | if(!strncasecmp(identity.c_str()+fsc,"xri://",sizeof("xri://")-1)) |
136 | fsc += sizeof("xri://")-1; | 137 | fsc += sizeof("xri://")-1; |
137 | if((fsc+1)>=lsc) | 138 | if((fsc+1)>=lsc) |
138 | throw bad_input(OPKELE_CP_ "not a character of importance in identity"); | 139 | throw bad_input(OPKELE_CP_ "not a character of importance in identity"); |
139 | string id(identity,fsc,lsc-fsc+1); | 140 | string id(identity,fsc,lsc-fsc+1); |
140 | idis.clear(); | 141 | idis.clear(); |
141 | if(strchr(data::_iname_leaders,id[0])) { | 142 | if(strchr(data::_iname_leaders,id[0])) { |