more robust discovery

 if no service found in yadis document, but it is found in html, use
 html-discovered service.

Signed-off-by: Michael Krelin <hacker@klever.net>

1 files changed, 11 insertions, 7 deletions

diff --git a/lib/openid_service_resolver.cc b/lib/openid_service_resolver.cc
index 5f82955..505e5b2 100644
--- a/lib/openid_service_resolver.cc
+++ b/lib/openid_service_resolver.cc
@@ -235,25 +235,24 @@ namespace opkele {
             fns+=sizeof("xri://")-1;
         string nid(id,fns,lns-fns+1);
         if(nid.empty())
             throw opkele::bad_input(OPKELE_CP_ "nothing significant in identity");
         if(strchr("=@+$!(",*nid.c_str())) {
             discover_service(
                     xri_proxy + util::url_encode(nid) +
                     "?_xrd_t=" STURI_OPENID10 "&_xrd_r=application/xrd+xml;sep=true",
                     true );
             if(auth_info.canonical_id.empty()
                     || auth_info.auth_SEP.xrd_Type.empty() )
                 throw opkele::failed_lookup(OPKELE_CP_ "no OpenID service for XRI found");
-            return auth_info;
         }else{
             const char *np = nid.c_str();
             if( (strncasecmp(np,"http",4) || strncmp(
                             tolower(*(np+4))=='s'? np+5 : np+4, "://", 3))
 #ifndef NDEBUG
                     && strncasecmp(np,"file:///",sizeof("file:///")-1)
         #endif  /* XXX: or how do I let tests work? */
                     )
                 nid.insert(0,"http://");
             string::size_type fp = nid.find('#');
             if(fp!=string::npos) {
                 string::size_type qp = nid.find('?');
@@ -262,33 +261,38 @@ namespace opkele {
                 }else if(qp>fp)
                     nid.erase(fp,qp-fp);
             }
             discover_service(nid);
             const char *eu = 0;
             CURLcode r = easy_getinfo(CURLINFO_EFFECTIVE_URL,&eu);
             if(r)
                 throw exception_curl(OPKELE_CP_ "failed to get CURLINFO_EFFECTIVE_URL",r);
             string canonicalized_id = util::rfc_3986_normalize_uri(eu);
             if(xrds_location.empty()) {
                 if(auth_info.auth_SEP.xrd_Type.empty()) {
                     if(html_SEP.xrd_URI.empty())
-                        throw opkele::failed_lookup(OPKELE_CP_ "no OpenID service discovered");
+                        throw opkele::failed_lookup(OPKELE_CP_ "no OpenID 1.0 service discovered");
                     auth_info.auth_SEP = html_SEP;
                     auth_info.auth_SEP.xrd_Type.clear(); auth_info.auth_SEP.xrd_Type.insert( STURI_OPENID10 );
                     auth_info.canonical_id = canonicalized_id;
                 }else{
                     if(auth_info.canonical_id.empty())
                         auth_info.canonical_id = canonicalized_id;
                 }
-                return auth_info;
             }else{
                 discover_service(xrds_location);
-                if(auth_info.auth_SEP.xrd_Type.empty())
-                    throw opkele::failed_lookup(OPKELE_CP_ "no OpenID service found in Yadis document");
-                if(auth_info.canonical_id.empty())
+                if(auth_info.auth_SEP.xrd_Type.empty()) {
+                    if(html_SEP.xrd_URI.empty())
+                        throw opkele::failed_lookup(OPKELE_CP_ "no OpenID 1.0 service discovered");
+                    auth_info.auth_SEP = html_SEP;
+                    auth_info.auth_SEP.xrd_Type.clear(); auth_info.auth_SEP.xrd_Type.insert( STURI_OPENID10 );
                     auth_info.canonical_id = canonicalized_id;
-                return auth_info;
+                }else{
+                    if(auth_info.canonical_id.empty())
+                        auth_info.canonical_id = canonicalized_id;
+                }
             }
         }
+        return auth_info;
     }
 
 }