Finished the code for XRDS/Yadis discovery of OP URL
[citadel.git] / citadel / modules / openid / serv_openid_rp.c
index 439b119cd4f12a788313f43ef8fc942b0b5a630a..f757765ccf8f709d8fa2f3ccadb50563f7691cdc 100644 (file)
@@ -59,6 +59,14 @@ typedef struct _ctdl_openid {
        HashList *sreg_keys;
 } ctdl_openid;
 
+enum {
+       openid_disco_none,
+       openid_disco_xrds,
+       openid_disco_html
+};
+
+
+
 void Free_ctdl_openid(ctdl_openid **FreeMe)
 {
        if (*FreeMe == NULL) {
@@ -588,10 +596,14 @@ CURL *ctdl_openid_curl_easy_init(char *errmsg) {
 
 
 struct xrds {
+       StrBuf *CharData;
        int nesting_level;
        int in_xrd;
        int current_service_priority;
-       int selected_service_priority;  /* more here later */
+       int selected_service_priority;
+       StrBuf *current_service_uri;
+       StrBuf *selected_service_uri;
+       int current_service_is_oid2auth;
 };
 
 
@@ -603,17 +615,19 @@ void xrds_xml_start(void *data, const char *supplied_el, const char **attr) {
 
        if (!strcasecmp(supplied_el, "XRD")) {
                ++xrds->in_xrd;
-               syslog(LOG_DEBUG, "*** XRD CONTAINER BEGIN ***");
        }
 
        else if (!strcasecmp(supplied_el, "service")) {
                xrds->current_service_priority = 0;
+               xrds->current_service_is_oid2auth = 0;
                for (i=0; attr[i] != NULL; i+=2) {
                        if (!strcasecmp(attr[i], "priority")) {
                                xrds->current_service_priority = atoi(attr[i+1]);
                        }
                }
        }
+
+       FlushStrBuf(xrds->CharData);
 }
 
 
@@ -624,36 +638,63 @@ void xrds_xml_end(void *data, const char *supplied_el) {
 
        if (!strcasecmp(supplied_el, "XRD")) {
                --xrds->in_xrd;
-               syslog(LOG_DEBUG, "*** XRD CONTAINER END ***");
+       }
+
+       else if (!strcasecmp(supplied_el, "type")) {
+               if (    (xrds->in_xrd)
+                       && (!strcasecmp(ChrPtr(xrds->CharData), "http://specs.openid.net/auth/2.0/server"))
+               ) {
+                       xrds->current_service_is_oid2auth = 1;
+               }
+       }
+
+       else if (!strcasecmp(supplied_el, "uri")) {
+               if (xrds->in_xrd) {
+                       FlushStrBuf(xrds->current_service_uri);
+                       StrBufAppendBuf(xrds->current_service_uri, xrds->CharData, 0);
+               }
        }
 
        else if (!strcasecmp(supplied_el, "service")) {
-               /* this is where we should evaluate the service and do stuff */
-               xrds->current_service_priority = 0;
+               if (    (xrds->in_xrd)
+                       && (xrds->current_service_priority < xrds->selected_service_priority)
+                       && (xrds->current_service_is_oid2auth)
+               ) {
+                       xrds->selected_service_priority = xrds->current_service_priority;
+                       FlushStrBuf(xrds->selected_service_uri);
+                       StrBufAppendBuf(xrds->selected_service_uri, xrds->current_service_uri, 0);
+               }
+
        }
+
+       FlushStrBuf(xrds->CharData);
 }
 
 
 void xrds_xml_chardata(void *data, const XML_Char *s, int len) {
        struct xrds *xrds = (struct xrds *) data;
 
-       if (xrds) ;     /* this is only here to silence the warning for now */
-       
-       /* StrBufAppendBufPlain (xrds->CData, s, len, 0); */
+       StrBufAppendBufPlain (xrds->CharData, s, len, 0);
 }
 
 
 /*
  * Parse an XRDS document.
- * If OpenID stuff is discovered, populate FIXME something and return nonzero
+ * If an OpenID Provider URL is discovered, op_url to that value and return nonzero.
  * If nothing useful happened, return 0.
  */
 int parse_xrds_document(StrBuf *ReplyBuf) {
+       ctdl_openid *oiddata = (ctdl_openid *) CC->openid_data;
        struct xrds xrds;
+       int return_value = 0;
 
-       syslog(LOG_DEBUG, "\033[32m --- XRDS DOCUMENT --- \n%s\033[0m", ChrPtr(ReplyBuf));
+       //      syslog(LOG_DEBUG, "\033[32m --- XRDS DOCUMENT --- \n%s\033[0m", ChrPtr(ReplyBuf));
 
        memset(&xrds, 0, sizeof (struct xrds));
+       xrds.selected_service_priority = INT_MAX;
+       xrds.CharData = NewStrBuf();
+       xrds.current_service_uri = NewStrBuf();
+       xrds.selected_service_uri = NewStrBuf();
        XML_Parser xp = XML_ParserCreate(NULL);
        if (xp) {
                XML_SetUserData(xp, &xrds);
@@ -667,7 +708,20 @@ int parse_xrds_document(StrBuf *ReplyBuf) {
                syslog(LOG_ALERT, "Cannot create XML parser");
        }
 
-       return(0);      /* FIXME return nonzero if something wonderful happened */
+       if (xrds.selected_service_priority < INT_MAX) {
+               if (oiddata->op_url == NULL) {
+                       oiddata->op_url = NewStrBuf();
+               }
+               FlushStrBuf(oiddata->op_url);
+               StrBufAppendBuf(oiddata->op_url, xrds.selected_service_uri, 0);
+               return_value = openid_disco_xrds;
+       }
+
+       FreeStrBuf(&xrds.CharData);
+       FreeStrBuf(&xrds.current_service_uri);
+       FreeStrBuf(&xrds.selected_service_uri);
+
+       return(return_value);
 }
 
 
@@ -692,7 +746,6 @@ size_t yadis_headerfunction(void *ptr, size_t size, size_t nmemb, void *userdata
 }
 
 
-
 /* Attempt to perform Yadis discovery as specified in Yadis 1.0 section 6.2.5.
  * 
  * If Yadis fails, we then attempt HTML discovery using the same document.
@@ -711,12 +764,12 @@ int perform_openid2_discovery(StrBuf *YadisURL) {
        struct curl_slist *my_headers = NULL;
        StrBuf *x_xrds_location = NULL;
 
-       if (YadisURL == NULL) return(0);
+       if (!YadisURL) return(0);
        syslog(LOG_DEBUG, "perform_openid2_discovery(%s)", ChrPtr(YadisURL));
        if (StrLength(YadisURL) == 0) return(0);
 
-       ReplyBuf = NewStrBuf ();
-       if (ReplyBuf == 0) return(0);
+       ReplyBuf = NewStrBuf();
+       if (!ReplyBuf) return(0);
 
        curl = ctdl_openid_curl_easy_init(errmsg);
        if (!curl) return(0);
@@ -782,8 +835,7 @@ int perform_openid2_discovery(StrBuf *YadisURL) {
                }
                extract_link(oiddata->op_url, HKEY("openid2.provider"), ReplyBuf);
                if (StrLength(oiddata->op_url) > 0) {
-                       syslog(LOG_DEBUG, "\033[31mHTML DISCO PROVIDER: %s\033[0m", ChrPtr(oiddata->op_url));
-                       return_value = 1;
+                       return_value = openid_disco_html;
                }
        }
 
@@ -845,23 +897,29 @@ void cmd_oids(char *argbuf) {
         */
        discovery_succeeded = perform_openid2_discovery(oiddata->claimed_id);
 
-       /* Empty delegate is legal; we just use the openid_url instead */
-       if (StrLength(openid_delegate) == 0) {
-               StrBufPlain(openid_delegate, SKEY(oiddata->claimed_id));
-       }
-
-       if (StrLength(oiddata->op_url) == 0) {
-               cprintf("%d There is no OpenID identity provider at this URL.\n", ERROR);
+       if (discovery_succeeded == 0) {
+               cprintf("%d There is no OpenID identity provider at this location.\n", ERROR);
        }
 
        else {
+               /*
+                * If we get to this point we are in possession of a valid OpenID Provider URL.
+                */
+               syslog(LOG_DEBUG, "OP URI '%s' discovered using method %d",
+                       ChrPtr(oiddata->claimed_id),
+                       discovery_succeeded
+               );
+
+               /* Empty delegate is legal; we just use the openid_url instead */
+               if (StrLength(openid_delegate) == 0) {
+                       StrBufPlain(openid_delegate, SKEY(oiddata->claimed_id));
+               }
 
                /* Assemble a URL to which the user-agent will be redirected. */
        
                RedirectUrl = NewStrBufDup(oiddata->op_url);
        
-               StrBufAppendBufPlain(RedirectUrl, HKEY("?openid.mode=checkid_setup"
-                                               "&openid.identity="), 0);
+               StrBufAppendBufPlain(RedirectUrl, HKEY("?openid.mode=checkid_setup&openid.identity="), 0);
                StrBufUrlescAppend(RedirectUrl, openid_delegate, NULL);
        
                StrBufAppendBufPlain(RedirectUrl, HKEY("&openid.return_to="), 0);