fix: Improve set_port validation

src/url.cpp

@@ -782,12 +782,9 @@ bool url::set_port(const std::string_view input) {
     port = std::nullopt;
     return true;
   }
-  // Input should not start with control characters.
-  if (ada::unicode::is_c0_control_or_space(trimmed.front())) {
-    return false;
-  }
-  // Input should contain at least one ascii digit.
-  if (input.find_first_of("0123456789") == std::string_view::npos) {
+
+  // Input should not start with a non-digit character.
+  if (!ada::unicode::is_ascii_digit(trimmed.front())) {
     return false;
   }
 

src/url_aggregator.cpp

@@ -285,12 +285,9 @@ bool url_aggregator::set_port(const std::string_view input) {
     clear_port();
     return true;
   }
-  // Input should not start with control characters.
-  if (ada::unicode::is_c0_control_or_space(trimmed.front())) {
-    return false;
-  }
-  // Input should contain at least one ascii digit.
-  if (input.find_first_of("0123456789") == std::string_view::npos) {
+
+  // Input should not start with a non-digit character.
+  if (!ada::unicode::is_ascii_digit(trimmed.front())) {
     return false;
   }
 

tests/basic_tests.cpp

@@ -462,4 +462,17 @@ TYPED_TEST(basic_tests, negativeport) {
   auto url = ada::parse<TypeParam>("https://www.google.com");
   ASSERT_FALSE(url->set_port("-1"));
   SUCCEED();
-}
+}
+
+// https://github.com/ada-url/ada/issues/826
+TYPED_TEST(basic_tests, set_invalid_port) {
+  auto url = ada::parse<TypeParam>("fake://dummy.test");
+  ASSERT_TRUE(url);
+  ASSERT_FALSE(url->set_port("invalid80"));
+  ASSERT_EQ(url->get_port(), "");
+  ASSERT_TRUE(url->set_port("80valid"));
+  ASSERT_TRUE(url->is_valid);
+  ASSERT_EQ(url->get_port(), "80");
+  ASSERT_TRUE(url->is_valid);
+  SUCCEED();
+}