[libc] Add IN6_IS_ADDR_MC*

[c8ef]

No description provided.

[c8ef]

Warning

This pull request is not mergeable via GitHub because a downstack PR is open. Once all requirements are satisfied, merge this PR as a stack on Graphite.
Learn more

• [libc] Add IN6_IS_ADDR_V4COMPAT #172646
• [libc] Add IN6_IS_ADDR_V4MAPPED #172645
• [libc] Add IN6_IS_ADDR_MC* #172643 👈 (View in Graphite)
• [libc] Add IN6_IS_ADDR_MULTICAST #172498
• [libc] Add IN6_IS_ADDR_LOOPBACK #172312
• [libc] Add IN6_IS_ADDR_UNSPECIFIED #172311
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

[llvmbot]

@llvm/pr-subscribers-libc

Author: Connector Switch (c8ef)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/172643.diff

2 Files Affected:

• (modified) libc/include/llvm-libc-macros/netinet-in-macros.h (+20)
• (modified) libc/test/include/netinet_in_test.cpp (+14)

diff --git a/libc/include/llvm-libc-macros/netinet-in-macros.h b/libc/include/llvm-libc-macros/netinet-in-macros.h
index 863ff8759e446..88ffe92756b16 100644
--- a/libc/include/llvm-libc-macros/netinet-in-macros.h
+++ b/libc/include/llvm-libc-macros/netinet-in-macros.h
@@ -64,4 +64,24 @@
   ((__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[0]) == 0xfe &&            \
    (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xc0) == 0xc0)
 
+#define IN6_IS_ADDR_MC_NODELOCAL(a)                                            \
+  (IN6_IS_ADDR_MULTICAST(a) &&                                                 \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xf) == 0x1)
+
+#define IN6_IS_ADDR_MC_LINKLOCAL(a)                                            \
+  (IN6_IS_ADDR_MULTICAST(a) &&                                                 \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xf) == 0x2)
+
+#define IN6_IS_ADDR_MC_SITELOCAL(a)                                            \
+  (IN6_IS_ADDR_MULTICAST(a) &&                                                 \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xf) == 0x5)
+
+#define IN6_IS_ADDR_MC_ORGLOCAL(a)                                             \
+  (IN6_IS_ADDR_MULTICAST(a) &&                                                 \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xf) == 0x8)
+
+#define IN6_IS_ADDR_MC_GLOBAL(a)                                               \
+  (IN6_IS_ADDR_MULTICAST(a) &&                                                 \
+   (__LLVM_LIBC_CAST(reinterpret_cast, uint8_t *, a)[1] & 0xf) == 0xe)
+
 #endif // LLVM_LIBC_MACROS_NETINET_IN_MACROS_H
diff --git a/libc/test/include/netinet_in_test.cpp b/libc/test/include/netinet_in_test.cpp
index e8cb8e48ff4b4..72500c4d51438 100644
--- a/libc/test/include/netinet_in_test.cpp
+++ b/libc/test/include/netinet_in_test.cpp
@@ -42,4 +42,18 @@ TEST(LlvmLibcNetinetInTest, IN6Macro) {
   buff[0] = 0xff;
   buff[1] = 0x80;
   EXPECT_FALSE(IN6_IS_ADDR_SITELOCAL(buff));
+
+  buff[0] = 0xff;
+  buff[1] = 0x1;
+  EXPECT_TRUE(IN6_IS_ADDR_MC_NODELOCAL(buff));
+  buff[1] = 0x2;
+  EXPECT_TRUE(IN6_IS_ADDR_MC_LINKLOCAL(buff));
+  buff[1] = 0x5;
+  EXPECT_TRUE(IN6_IS_ADDR_MC_SITELOCAL(buff));
+  buff[1] = 0x8;
+  EXPECT_TRUE(IN6_IS_ADDR_MC_ORGLOCAL(buff));
+  buff[1] = 0xe;
+  EXPECT_TRUE(IN6_IS_ADDR_MC_GLOBAL(buff));
+  buff[1] = 0;
+  buff[0] = 0;
 }