[llvm-branch-commits] [libc] [libc] Add `IN6_IS_ADDR_MC*` (PR #172643)

[via llvm-branch-commits]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-libc

Author: Connector Switch (c8ef)

<details>
<summary>Changes</summary>



---
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
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;
 }

``````````

</details>


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