Update

urlparse.cpp

@@ -0,0 +1,142 @@
+#include "urlparse.h"
+#include "utils_config.h"
+
+#include <array>
+#include <malloc.h>
+#include <stdio.h>
+#include <string.h>
+#include <string>
+#include "cstr_util.h"
+#include "str_util.h"
+
+#if HAVE_PRINTF_S
+#define printf printf_s
+#endif
+
+bool is_valid_scheme_chars(char c) {
+    if (c >= 'a' && c <= 'z') return true;
+    if (c >= 'A' && c <= 'Z') return true;
+    if (c >= '0' && c <= '9') return true;
+    if (c == '+' || c == '-' || c == '.') return true;
+    return false;
+}
+
+bool scheme_is_use_params(std::string name) {
+    const std::array<std::string, 15> arr = { "", "ftp", "hdl", "prospero", "http", "imap", "https", "shttp", "rtsp", "rtspu", "sip", "sips", "mms", "sftp", "tel" };
+    for (auto i = arr.begin(); i != arr.end(); i++) {
+        if (name == *i) return true;
+    }
+    return false;
+}
+
+UrlParseResult* urlparse(const char* url, const char* sch, char allow_fragments) {
+    if (!url) return nullptr;
+    std::string u(url);
+    std::string scheme, netloc, params, query, fragment;
+    if (sch) scheme = sch;
+    u = str_util::str_replace(u, "\t", "");
+    u = str_util::str_replace(u, "\r", "");
+    u = str_util::str_replace(u, "\n", "");
+    auto i = u.find(':');
+    if (i > 0 && i != -1) {
+        bool ok = true;
+        for (size_t j = 0; j < i; j++) {
+            if (!is_valid_scheme_chars(u[j])) {
+                ok = false;
+                break;
+            }
+        }
+        if (ok) {
+            std::string tmp = u.substr(0, i);
+            if (!str_util::tolowercase(tmp, scheme)) return nullptr;
+            u = u.substr(i + 1);
+        }
+    }
+    if (u.find("//") == 0) {
+        auto delim = u.size();
+        const char* s = "/?#";
+        for (int i = 0; i < 3; i++) {
+            auto wdelim = u.find(s[i], 2);
+            if (wdelim != -1) {
+                delim = std::min(delim, wdelim);
+            }
+        }
+        netloc = u.substr(2, delim - 2);
+        u = u.substr(delim);
+    }
+    if (allow_fragments) {
+        auto i = u.find('#');
+        if (i != -1) {
+            fragment = u.substr(i + 1);
+            u = u.substr(0, i);
+        }
+    }
+    i = u.find('?');
+    if (i != -1) {
+        query = u.substr(i + 1);
+        u = u.substr(0, i);
+    }
+    i = u.find(';');
+    if (i != -1 && scheme_is_use_params(scheme)) {
+        auto j = u.rfind('/');
+        if (j == -1) {
+            params = u.substr(i + 1);
+            u = u.substr(0, i);
+        } else {
+            i = u.find(';', j);
+            if (i != -1) {
+                params = u.substr(i + 1);
+                u = u.substr(0, i);
+            }
+        }
+    }
+    auto r = (UrlParseResult*)malloc(sizeof(UrlParseResult));
+    if (!r) {
+        return nullptr;
+    }
+    memset(r, 0, sizeof(UrlParseResult));
+    if (cstr_util_copy_str(&r->scheme, scheme.c_str())) {
+        goto end;
+    }
+    if (cstr_util_copy_str(&r->netloc, netloc.c_str())) {
+        goto end;
+    }
+    if (cstr_util_copy_str(&r->path, u.c_str())) {
+        goto end;
+    }
+    if (cstr_util_copy_str(&r->params, params.c_str())) {
+        goto end;
+    }
+    if (cstr_util_copy_str(&r->query, query.c_str())) {
+        goto end;
+    }
+    if (cstr_util_copy_str(&r->fragment, fragment.c_str())) {
+        goto end;
+    }
+    return r;
+end:
+    free_url_parse_result(r);
+    return nullptr;
+}
+
+void free_url_parse_result(UrlParseResult* r) {
+    if (!r) return;
+    if (r->scheme) free(r->scheme);
+    if (r->netloc) free(r->netloc);
+    if (r->path) free(r->path);
+    if (r->params) free(r->params);
+    if (r->query) free(r->query);
+    if (r->fragment) free(r->fragment);
+    free(r);
+}
+
+void dump_url_parse_result(UrlParseResult* r, int indent_now) {
+    if (!r) return;
+    std::string ind(indent_now, ' ');
+    if (r->scheme) printf("%sScheme: %s\n", ind.c_str(), r->scheme);
+    if (r->netloc) printf("%sNetloc: %s\n", ind.c_str(), r->netloc);
+    if (r->path) printf("%sPath: %s\n", ind.c_str(), r->path);
+    if (r->params) printf("%sParams: %s\n", ind.c_str(), r->params);
+    if (r->query) printf("%sQuery: %s\n", ind.c_str(), r->query);
+    if (r->fragment) printf("%sFragment: %s\n", ind.c_str(), r->fragment);
+}