1 //===-- UriParser.cpp -----------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "lldb/Utility/UriParser.h"
10 #include "llvm/Support/raw_ostream.h"
11 
12 #include <string>
13 
14 #include <cstdint>
15 #include <optional>
16 #include <tuple>
17 
18 using namespace lldb_private;
19 
20 llvm::raw_ostream &lldb_private::operator<<(llvm::raw_ostream &OS,
21                                             const URI &U) {
22   OS << U.scheme << "://[" << U.hostname << ']';
23   if (U.port)
24     OS << ':' << *U.port;
25   return OS << U.path;
26 }
27 
28 std::optional<URI> URI::Parse(llvm::StringRef uri) {
29   URI ret;
30 
31   const llvm::StringRef kSchemeSep("://");
32   auto pos = uri.find(kSchemeSep);
33   if (pos == std::string::npos)
34     return std::nullopt;
35 
36   // Extract path.
37   ret.scheme = uri.substr(0, pos);
38   auto host_pos = pos + kSchemeSep.size();
39   auto path_pos = uri.find('/', host_pos);
40   if (path_pos != std::string::npos)
41     ret.path = uri.substr(path_pos);
42   else
43     ret.path = "/";
44 
45   auto host_port = uri.substr(
46       host_pos,
47       ((path_pos != std::string::npos) ? path_pos : uri.size()) - host_pos);
48 
49   // Extract hostname
50   if (!host_port.empty() && host_port[0] == '[') {
51     // hostname is enclosed with square brackets.
52     pos = host_port.rfind(']');
53     if (pos == std::string::npos)
54       return std::nullopt;
55 
56     ret.hostname = host_port.substr(1, pos - 1);
57     host_port = host_port.drop_front(pos + 1);
58     if (!host_port.empty() && !host_port.consume_front(":"))
59       return std::nullopt;
60   } else {
61     std::tie(ret.hostname, host_port) = host_port.split(':');
62   }
63 
64   // Extract port
65   if (!host_port.empty()) {
66     uint16_t port_value = 0;
67     if (host_port.getAsInteger(0, port_value))
68       return std::nullopt;
69     ret.port = port_value;
70   } else
71     ret.port = std::nullopt;
72 
73   return ret;
74 }
75