Make URI structures reference-counted.
[people/sha0/gpxe.git] / src / include / gpxe / uri.h
index 24f92c6..6fddcc3 100644 (file)
@@ -8,6 +8,7 @@
  */
 
 #include <stdlib.h>
+#include <gpxe/refcnt.h>
 
 /** A Uniform Resource Identifier
  *
@@ -37,6 +38,8 @@
  *   query = "what=is", fragment = "this"
  */
 struct uri {
+       /** Reference count */
+       struct refcnt refcnt;
        /** Scheme */
        const char *scheme;
        /** Opaque part */
@@ -100,17 +103,28 @@ static inline int uri_has_relative_path ( struct uri *uri ) {
 }
 
 /**
- * Free URI structure
+ * Increment URI reference count
  *
  * @v uri              URI
+ * @ret uri            URI
+ */
+static inline __attribute__ (( always_inline )) struct uri *
+uri_get ( struct uri *uri ) {
+       ref_get ( &uri->refcnt );
+       return uri;
+}
+
+/**
+ * Decrement URI reference count
  *
- * Frees all the dynamically-allocated storage used by the URI
- * structure.
+ * @v uri              URI
  */
-static inline void free_uri ( struct uri *uri ) {
-       free ( uri );
+static inline __attribute__ (( always_inline )) void
+uri_put ( struct uri *uri ) {
+       ref_put ( &uri->refcnt );
 }
 
 extern struct uri * parse_uri ( const char *uri_string );
+unsigned int uri_port ( struct uri *uri, unsigned int default_port );
 
 #endif /* _GPXE_URI_H */