1 files changed, 83 insertions, 9 deletions

diff --git a/include/opkele/types.h b/include/opkele/types.h
index f732a1e..de44a5c 100644
--- a/include/opkele/types.h
+++ b/include/opkele/types.h
@@ -10,14 +10,16 @@
 #include <vector>
 #include <string>
 #include <map>
-#include <memory>
+#include <set>
+#include <opkele/tr1-mem.h>
 
 namespace opkele {
     using std::vector;
     using std::string;
     using std::map;
     using std::ostream;
-    using std::auto_ptr;
+    using std::multimap;
+    using std::set;
 
     /**
      * the OpenID operation mode
@@ -37,16 +39,16 @@ namespace opkele {
 
             /**
              * xor the secret and hmac together and encode, using base64
-             * @param key_sha1 pointer to the sha1 digest
+             * @param key_d pointer to the message digest
              * @param rv reference to the return value
              */
-            void enxor_to_base64(const unsigned char *key_sha1,string& rv) const;
+            void enxor_to_base64(const unsigned char *key_d,string& rv) const;
             /**
-             * decode base64-encoded secret and xor it with the sha1 digest
-             * @param key_sha1 pointer to the message digest
+             * decode base64-encoded secret and xor it with the message digest
+             * @param key_d pointer to the message digest
              * @param b64 base64-encoded secret value
              */
-            void enxor_from_base64(const unsigned char *key_sha1,const string& b64);
+            void enxor_from_base64(const unsigned char *key_d,const string& b64);
             /**
              * plainly encode to base64 representation
              * @param rv reference to the return value
@@ -105,9 +107,9 @@ namespace opkele {
     };
 
     /**
-     * the auto_ptr<> for association_t object type
+     * the shared_ptr<> for association_t object type
      */
-    typedef auto_ptr<association_t> assoc_t;
+    typedef tr1mem::shared_ptr<association_t> assoc_t;
 
     /**
      * request/response parameters map
@@ -158,6 +160,14 @@ namespace opkele {
              * @return the ready-to-use location
              */
             string append_query(const string& url,const char *prefix = "openid.") const;
+
+            /**
+             * make up a query string suitable for use in GET and POST
+             * requests.
+             * @param prefix string to prened to parameter names
+             * @return query string
+             */
+            string query_string(const char *prefix = "openid.") const;
     };
 
     /**
@@ -167,6 +177,70 @@ namespace opkele {
      */
     ostream& operator << (ostream& o,const params_t& p);
 
+    namespace xrd {
+
+        struct priority_compare {
+            inline bool operator()(long a,long b) const {
+                return (a<0) ? false : (b<0) ? true : (a<b);
+            }
+        };
+
+        template <typename _DT>
+            class priority_map : public multimap<long,_DT,priority_compare> {
+                typedef multimap<long,_DT,priority_compare> map_type;
+                public:
+
+                    inline _DT& add(long priority,const _DT& d) {
+                        return insert(typename map_type::value_type(priority,d))->second;
+                    }
+            };
+
+        typedef priority_map<string> canonical_ids_t;
+        typedef priority_map<string> local_ids_t;
+        typedef set<string> types_t;
+        typedef priority_map<string> uris_t;
+
+        class service_t {
+            public:
+                types_t types;
+                uris_t uris;
+                local_ids_t local_ids;
+                string provider_id;
+
+                void clear() {
+                    types.clear();
+                    uris.clear(); local_ids.clear();
+                    provider_id.clear();
+                }
+        };
+        typedef priority_map<service_t> services_t;
+
+        class XRD_t {
+            public:
+                time_t expires;
+
+                canonical_ids_t canonical_ids;
+                local_ids_t local_ids;
+                services_t services;
+                string provider_id;
+
+                void clear() {
+                    expires = 0;
+                    canonical_ids.clear(); local_ids.clear();
+                    services.clear();
+                    provider_id.clear();
+                }
+                bool empty() const {
+                    return
+                        canonical_ids.empty()
+                        && local_ids.empty()
+                        && services.empty();
+                }
+
+        };
+
+    }
+
 }
 
 #endif /* __OPKELE_TYPES_H */