discover/grub2: Allow to separate the --id argument using a space char

[petitboot] / lib / url / url.c

diff --git a/lib/url/url.c b/lib/url/url.c
index 6eeced32538350cb0aea0fa3edbe1a9c73aab411..b74018ddd0f300f4d8ca7d2efffa3fc3451e85fe 100644 (file)
--- a/lib/url/url.c
+++ b/lib/url/url.c
@@ -193,23 +193,39 @@ struct pb_url *pb_url_parse(void *ctx, const char *url_str)
                path = strchr(p, '/');
 
                if (!path) {
-                       pb_log("%s: parse path failed '%s'\n", __func__ , p);
+                       pb_log_fn("parse path failed '%s'\n" , p);
                        goto fail;
                }
 
-               col = strchr(p, ':');
+               col = strrchr(p, ':');
+               if (col <= p)
+                       col = NULL;
+               if (col && strchr(col , ']')) {
+                       /*
+                        * This is likely an IPv6 address with no port.
+                        * See https://www.ietf.org/rfc/rfc2732.txt
+                        */
+                       col = NULL;
+               }
 
                if (col) {
                        len = path - col - 1;
                        url->port = len ? talloc_strndup(url, col + 1, len)
                                : NULL;
                        len = col - p;
-                       url->host = len ? talloc_strndup(url, p, len) : NULL;
                } else {
                        url->port = NULL;
-                       url->host = talloc_strndup(url, p, path - p);
+                       len = path - p;
                }
 
+               if (p[0] == '[' && p[len - 1] == ']')
+                       /* IPv6 */
+                       url->host = talloc_strndup(url, p + 1, len - 2);
+               else
+                       /* IPv4 */
+                       url->host = talloc_strndup(url, p, len);
+
+
                /* remove multiple leading slashes */
                for (; *path && *(path+1) == '/'; path++)
                        ;
@@ -231,13 +247,39 @@ bool is_url(const char *str)
        return strstr(str, "://") != NULL;
 }
 
+int addr_scheme(const char *address)
+{
+       uint8_t buf[sizeof(struct in6_addr)];
+       int rc;
+
+       rc = inet_pton(AF_INET, address , buf);
+       if (rc == 1)
+               return AF_INET;
+       rc = inet_pton(AF_INET6, address , buf);
+       if (rc == 1)
+               return AF_INET6;
+
+       return 0;
+}
+
 char *pb_url_to_string(struct pb_url *url)
 {
        const struct pb_scheme_info *scheme = pb_url_scheme_info(url->scheme);
+       char *str, *port;
        assert(scheme);
 
-       return talloc_asprintf(url, "%s://%s%s", scheme->str,
-                       scheme->has_host ? url->host : "", url->path);
+       port = url->port ? talloc_asprintf(url, ":%s", url->port) : NULL;
+
+       if (scheme->has_host && addr_scheme(url->host) == AF_INET6)
+               str = talloc_asprintf(url, "%s://[%s]%s%s", scheme->str,
+                               url->host, port ?: "", url->path);
+       else
+               str = talloc_asprintf(url, "%s://%s%s%s", scheme->str,
+                               scheme->has_host ? url->host : "",
+                               port ?: "", url->path);
+
+       talloc_free(port);
+       return str;
 }
 
 static void pb_url_update_full(struct pb_url *url)