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
@@ -239,17 +239,16 @@ namespace opkele {
         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? */
                     )
@@ -266,29 +265,34 @@ namespace opkele {
             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;
     }
 
 }