commit 6d7c6beccd3783f6c44451eec062e996b5838a6d
author Connector Switch <c8ef@outlook.com> Sat Dec 13 02:57:38 2025
committer Copybara-Service <copybara-worker@google.com> Sat Dec 13 03:01:22 2025
tree 907402aa7306aa1444eda59c36a3accc8616e5e6
parent 77e7317bbb0c805b81ca0cb87c036443e8ac0720

[libc] Add `IN6_IS_ADDR_{LINK, SITE}LOCAL` (#168207)

This patch introduces two macros in `netinet/in.h`. The redundant tests
for macro values in the testcases have been removed.

NOKEYCHECK=True
GitOrigin-RevId: 62ee2cf0f4e1e2aabd0f348064ffb2725fd2508c

include/llvm-libc-macros/netinet-in-macros.h

diff --git a/include/llvm-libc-macros/netinet-in-macros.h b/include/llvm-libc-macros/netinet-in-macros.h
index 7a4d26d..b04d6aa 100644
--- a/include/llvm-libc-macros/netinet-in-macros.h
+++ b/include/llvm-libc-macros/netinet-in-macros.h
@@ -34,4 +34,16 @@
 #define INET_ADDRSTRLEN 16
 #define INET6_ADDRSTRLEN 46
 
+// The following macros test for special IPv6 addresses. Each macro is of type
+// int and takes a single argument of type const struct in6_addr *:
+// https://pubs.opengroup.org/onlinepubs/9799919799/basedefs/netinet_in.h.html
+
+#define IN6_IS_ADDR_LINKLOCAL(a)                                               \
+  ((__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[0]) == 0xfe &&            \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xc0) == 0x80)
+
+#define IN6_IS_ADDR_SITELOCAL(a)                                               \
+  ((__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[0]) == 0xfe &&            \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xc0) == 0xc0)
+
 #endif // LLVM_LIBC_MACROS_NETINET_IN_MACROS_H

test/include/netinet_in_test.cpp

diff --git a/test/include/netinet_in_test.cpp b/test/include/netinet_in_test.cpp
index 714892f..6937262 100644
--- a/test/include/netinet_in_test.cpp
+++ b/test/include/netinet_in_test.cpp
@@ -9,21 +9,20 @@
 #include "include/llvm-libc-macros/netinet-in-macros.h"
 #include "test/UnitTest/Test.h"
 
-TEST(LlvmLibcNetinetInTest, IPPROTOMacro) {
-  EXPECT_EQ(IPPROTO_IP, 0);
-  EXPECT_EQ(IPPROTO_ICMP, 1);
-  EXPECT_EQ(IPPROTO_TCP, 6);
-  EXPECT_EQ(IPPROTO_UDP, 17);
-  EXPECT_EQ(IPPROTO_IPV6, 41);
-  EXPECT_EQ(IPPROTO_RAW, 255);
-}
+TEST(LlvmLibcNetinetInTest, IN6Macro) {
+  char buff[16] = {};
 
-TEST(LlvmLibcNetinetInTest, IPV6Macro) {
-  EXPECT_EQ(IPV6_UNICAST_HOPS, 16);
-  EXPECT_EQ(IPV6_MULTICAST_IF, 17);
-  EXPECT_EQ(IPV6_MULTICAST_HOPS, 18);
-  EXPECT_EQ(IPV6_MULTICAST_LOOP, 19);
-  EXPECT_EQ(IPV6_JOIN_GROUP, 20);
-  EXPECT_EQ(IPV6_LEAVE_GROUP, 21);
-  EXPECT_EQ(IPV6_V6ONLY, 26);
+  buff[0] = 0xfe;
+  buff[1] = 0x80;
+  EXPECT_TRUE(IN6_IS_ADDR_LINKLOCAL(buff));
+  buff[0] = 0xff;
+  buff[1] = 0x80;
+  EXPECT_FALSE(IN6_IS_ADDR_LINKLOCAL(buff));
+
+  buff[0] = 0xfe;
+  buff[1] = 0xc0;
+  EXPECT_TRUE(IN6_IS_ADDR_SITELOCAL(buff));
+  buff[0] = 0xff;
+  buff[1] = 0x80;
+  EXPECT_FALSE(IN6_IS_ADDR_SITELOCAL(buff));
 }