Do a mDNS advertisement on minmdns server startup

examples/minimal-mdns/AllInterfaceListener.h

@@ -94,7 +94,8 @@ class AllInterfaces : public mdns::Minimal::ListenIterator
         {
             return true; // not a usable interface
         }
-        char name[64];
+
+        char name[chip::Inet::InterfaceIterator::kMaxIfNameLength];
         if (mIterator.GetInterfaceName(name, sizeof(name)) != CHIP_NO_ERROR)
         {
             printf("!!!! FAILED TO GET INTERFACE NAME\n");

src/inet/InetInterface.h

@@ -42,6 +42,8 @@ struct ifaddrs;
 #endif // CHIP_SYSTEM_CONFIG_USE_BSD_IFADDRS
 
 #if CHIP_SYSTEM_CONFIG_USE_ZEPHYR_NET_IF
+#include <device.h>
+
 struct net_if;
 struct net_if_ipv4;
 struct net_if_ipv6;
@@ -152,6 +154,19 @@ class InterfaceIterator
     bool SupportsMulticast();
     bool HasBroadcastAddress();
 
+#if CHIP_SYSTEM_CONFIG_USE_LWIP
+    static constexpr size_t kMaxIfNameLength = 13; // Names are formatted as %c%c%d
+#elif CHIP_SYSTEM_CONFIG_USE_SOCKETS && CHIP_SYSTEM_CONFIG_USE_BSD_IFADDRS
+    static constexpr size_t kMaxIfNameLength = IF_NAMESIZE;
+#elif CHIP_SYSTEM_CONFIG_USE_ZEPHYR_NET_IF
+    static constexpr size_t kMaxIfNameLength = Z_DEVICE_MAX_NAME_LEN;
+#elif defined(IFNAMSIZ)
+    static constexpr size_t kMaxIfNameLength = IFNAMSIZ;
+#else
+    // No constant available here - set some reasonable size
+    static constexpr size_t kMaxIfNameLength = 33;
+#endif
+
 protected:
 #if CHIP_SYSTEM_CONFIG_USE_LWIP
     struct netif * mCurNetif;

src/inet/tests/TestInetCommonPosix.cpp

@@ -177,7 +177,7 @@ void ShutdownSystemLayer()
 #if CHIP_SYSTEM_CONFIG_USE_LWIP
 static void PrintNetworkState()
 {
-    char intfName[10];
+    char intfName[chip::Inet::InterfaceIterator::kMaxIfNameLength];
 
     for (size_t j = 0; j < gNetworkOptions.TapDeviceName.size(); j++)
     {

src/inet/tests/TestInetEndPoint.cpp

@@ -224,7 +224,7 @@ static void TestInetInterface(nlTestSuite * inSuite, void * inContext)
 {
     InterfaceIterator intIterator;
     InterfaceAddressIterator addrIterator;
-    char intName[20];
+    char intName[chip::Inet::InterfaceIterator::kMaxIfNameLength];
     InterfaceId intId;
     IPAddress addr;
     IPPrefix addrWithPrefix;

src/lib/mdns/Advertiser_ImplMinimalMdns.cpp

@@ -88,33 +88,32 @@ void LogQuery(const QueryData & data)
 
 class AllInterfaces : public ListenIterator
 {
+private:
 public:
-    AllInterfaces() {}
+    AllInterfaces() { SkipToFirstValidInterface(); }
 
     bool Next(chip::Inet::InterfaceId * id, chip::Inet::IPAddressType * type) override
     {
+        if (!mIterator.HasCurrent())
+        {
+            return false;
+        }
+
 #if INET_CONFIG_ENABLE_IPV4
         if (mState == State::kIpV4)
         {
-            *id    = INET_NULL_INTERFACEID;
+            *id    = mIterator.GetInterfaceId();
             *type  = chip::Inet::kIPAddressType_IPv4;
             mState = State::kIpV6;
-
-            SkipToFirstValidInterface();
             return true;
         }
-#else
-        mState = State::kIpV6;
-        SkipToFirstValidInterface();
 #endif
 
-        if (!mIterator.HasCurrent())
-        {
-            return false;
-        }
-
         *id   = mIterator.GetInterfaceId();
         *type = chip::Inet::kIPAddressType_IPv6;
+#if INET_CONFIG_ENABLE_IPV4
+        mState = State::kIpV4;
+#endif
 
         for (mIterator.Next(); SkipCurrentInterface(); mIterator.Next())
         {
@@ -128,7 +127,11 @@ class AllInterfaces : public ListenIterator
         kIpV4,
         kIpV6,
     };
+#if INET_CONFIG_ENABLE_IPV4
     State mState = State::kIpV4;
+#else
+    State mState = State::kIpV6;
+#endif
     chip::Inet::InterfaceIterator mIterator;
 
     void SkipToFirstValidInterface()
@@ -153,7 +156,7 @@ class AllInterfaces : public ListenIterator
         {
             return true; // not a usable interface
         }
-        char name[64];
+        char name[chip::Inet::InterfaceIterator::kMaxIfNameLength];
         if (mIterator.GetInterfaceName(name, sizeof(name)) != CHIP_NO_ERROR)
         {
             ChipLogError(Discovery, "Interface iterator failed to get interface name.");
@@ -209,6 +212,15 @@ class AdvertiserMinMdns : public ServiceAdvertiser,
     /// allocated memory.
     void Clear();
 
+    /// Advertise availabe records configured within the server
+    ///
+    /// Usable as boot-time advertisement of available SRV records.
+    void AdvertiseRecords();
+
+    /// Determine if advertisement on the specified interface/address is ok given the
+    /// interfaces on which the mDNS server is listening
+    bool ShouldAdvertiseOn(const chip::Inet::InterfaceId id, const chip::Inet::IPAddress & addr);
+
     QueryResponderSettings AddAllocatedResponder(Responder * responder)
     {
         if (responder == nullptr)
@@ -327,6 +339,9 @@ CHIP_ERROR AdvertiserMinMdns::Start(chip::Inet::InetLayer * inetLayer, uint16_t
     ReturnErrorOnFailure(mServer.Listen(inetLayer, &allInterfaces, port));
 
     ChipLogProgress(Discovery, "CHIP minimal mDNS started advertising.");
+
+    AdvertiseRecords();
+
     return CHIP_NO_ERROR;
 }
 
@@ -569,6 +584,114 @@ FullQName AdvertiserMinMdns::GetCommisioningTextEntries(const CommissionAdvertis
     }
 }
 
+bool CanAdvertiseOn(chip::Inet::InterfaceAddressIterator & current)
+{
+    if (!current.IsUp())
+    {
+        return false;
+    }
+
+    if (!current.SupportsMulticast())
+    {
+        return false;
+    }
+
+    char name[chip::Inet::InterfaceIterator::kMaxIfNameLength];
+    if (current.GetInterfaceName(name, sizeof(name)) != CHIP_NO_ERROR)
+    {
+        ChipLogError(Discovery, "Failed to get interface name - will not advertise on this interface");
+        return false;
+    }
+
+    if (strncmp(name, "lo", 2) == 0)
+    {
+        // Skip advertisement on what looks like a local loopback interface
+        return false;
+    }
+
+    return true;
+}
+
+bool AdvertiserMinMdns::ShouldAdvertiseOn(const chip::Inet::InterfaceId id, const chip::Inet::IPAddress & addr)
+{
+    for (unsigned i = 0; i < mServer.GetEndpointCount(); i++)
+    {
+        const ServerBase::EndpointInfo & info = mServer.GetEndpoints()[i];
+
+        if (info.udp == nullptr)
+        {
+            continue;
+        }
+
+        if (info.interfaceId != id)
+        {
+            continue;
+        }
+
+        if (info.addressType != addr.Type())
+        {
+            continue;
+        }
+
+        return true;
+    }
+
+    return false;
+}
+
+void AdvertiserMinMdns::AdvertiseRecords()
+{
+    chip::Inet::InterfaceAddressIterator interfaceAddress;
+
+    if (!interfaceAddress.Next())
+    {
+        return;
+    }
+
+    for (; interfaceAddress.HasCurrent(); interfaceAddress.Next())
+    {
+        if (!CanAdvertiseOn(interfaceAddress))
+        {
+            continue;
+        }
+
+        if (!ShouldAdvertiseOn(interfaceAddress.GetInterfaceId(), interfaceAddress.GetAddress()))
+        {
+            continue;
+        }
+
+        chip::Inet::IPPacketInfo packetInfo;
+
+        packetInfo.Clear();
+        packetInfo.SrcAddress = interfaceAddress.GetAddress();
+        if (interfaceAddress.GetAddress().IsIPv4())
+        {
+            chip::Inet::IPAddress::FromString("224.0.0.251", packetInfo.DestAddress);
+        }
+        else
+        {
+            chip::Inet::IPAddress::FromString("FF02::FB", packetInfo.DestAddress);
+        }
+        packetInfo.SrcPort   = kMdnsPort;
+        packetInfo.DestPort  = kMdnsPort;
+        packetInfo.Interface = interfaceAddress.GetInterfaceId();
+
+        QueryData queryData(QType::PTR, QClass::IN, false /* unicast */);
+        queryData.SetIsBootAdvertising(true);
+
+        mQueryResponder.ClearBroadcastTrottle();
+
+        CHIP_ERROR err = mResponseSender.Respond(0, queryData, &packetInfo);
+        if (err != CHIP_NO_ERROR)
+        {
+            ChipLogError(Discovery, "Failed to advertise records: %s", ErrorStr(err));
+        }
+    }
+
+    // Once all automatic broadcasts are done, allow immediate replies once.
+    mQueryResponder.ClearBroadcastTrottle();
+}
+
 AdvertiserMinMdns gAdvertiser;
 } // namespace
 

src/lib/mdns/minimal/Parser.h

@@ -31,6 +31,8 @@ class QueryData
     QueryData(const QueryData &) = default;
     QueryData & operator=(const QueryData &) = default;
 
+    QueryData(QType type, QClass klass, bool unicast) : mType(type), mClass(klass), mAnswerViaUnicast(unicast) {}
+
     QueryData(QType type, QClass klass, bool unicast, const uint8_t * nameStart, const BytesRange & validData) :
         mType(type), mClass(klass), mAnswerViaUnicast(unicast), mNameIterator(validData, nameStart)
     {}
@@ -39,6 +41,11 @@ class QueryData
     QClass GetClass() const { return mClass; }
     bool RequestedUnicastAnswer() const { return mAnswerViaUnicast; }
 
+    /// Boot advertisement is an internal query meant to advertise all available
+    /// services at device startup time.
+    bool IsBootAdvertising() const { return mIsBootAdvertising; }
+    void SetIsBootAdvertising(bool isBootAdvertising) { mIsBootAdvertising = isBootAdvertising; }
+
     SerializedQNameIterator GetName() const { return mNameIterator; }
 
     /// Parses a query structure
@@ -55,7 +62,11 @@ class QueryData
     QType mType            = QType::ANY;
     QClass mClass          = QClass::ANY;
     bool mAnswerViaUnicast = false;
-    SerializedQNameIterator mNameIterator; // const since we reuse it
+    SerializedQNameIterator mNameIterator;
+
+    /// Flag as a boot-time internal query. This allows query replies
+    /// to be built accordingly.
+    bool mIsBootAdvertising = false;
 };
 
 class ResourceData

src/lib/mdns/minimal/QueryReplyFilter.h

@@ -33,19 +33,17 @@ class QueryReplyFilter : public ReplyFilter
 
     bool Accept(QType qType, QClass qClass, FullQName qname) override
     {
-        // resource type is ignored
-        if ((mQueryData.GetType() != QType::ANY) && (mQueryData.GetType() != qType))
+        if (!AcceptableQueryType(qType))
         {
             return false;
         }
 
-        if ((mQueryData.GetClass() != QClass::ANY) && (mQueryData.GetClass() != qClass))
+        if (!AcceptableQueryClass(qClass))
         {
             return false;
         }
 
-        // path must match
-        return mIgnoreNameMatch || (mQueryData.GetName() == qname);
+        return AcceptablePath(qname);
     }
 
     /// Ignore qname matches during Accept calls (if set to true, only qtype and qclass are matched).
@@ -59,9 +57,41 @@ class QueryReplyFilter : public ReplyFilter
         return *this;
     }
 
+    QueryReplyFilter & SetSendingAdditionalItems(bool additional)
+    {
+        mSendingAdditionalItems = additional;
+        return *this;
+    }
+
 private:
+    bool AcceptableQueryType(QType qType)
+    {
+        if (mSendingAdditionalItems && mQueryData.IsBootAdvertising())
+        {
+            return true;
+        }
+
+        return ((mQueryData.GetType() == QType::ANY) || (mQueryData.GetType() == qType));
+    }
+
+    bool AcceptableQueryClass(QClass qClass)
+    {
+        return ((mQueryData.GetClass() == QClass::ANY) || (mQueryData.GetClass() == qClass));
+    }
+
+    bool AcceptablePath(FullQName qname)
+    {
+        if (mIgnoreNameMatch || mQueryData.IsBootAdvertising())
+        {
+            return true;
+        }
+
+        return (mQueryData.GetName() == qname);
+    }
+
     const QueryData & mQueryData;
-    bool mIgnoreNameMatch = false;
+    bool mIgnoreNameMatch        = false;
+    bool mSendingAdditionalItems = false;
 };
 
 } // namespace Minimal

src/lib/mdns/minimal/ResponseSender.cpp

@@ -74,7 +74,6 @@ CHIP_ERROR ResponseSender::Respond(uint32_t messageId, const QueryData & query,
         const uint64_t kTimeNowMs = chip::System::Platform::Layer::GetClock_MonotonicMS();
 
         QueryReplyFilter queryReplyFilter(query);
-
         QueryResponderRecordFilter responseFilter;
 
         responseFilter.SetReplyFilter(&queryReplyFilter);
@@ -108,7 +107,8 @@ CHIP_ERROR ResponseSender::Respond(uint32_t messageId, const QueryData & query,
         mSendState.SetResourceType(ResourceType::kAdditional);
 
         QueryReplyFilter queryReplyFilter(query);
-        queryReplyFilter.SetIgnoreNameMatch(true);
+
+        queryReplyFilter.SetIgnoreNameMatch(true).SetSendingAdditionalItems(true);
 
         QueryResponderRecordFilter responseFilter;
         responseFilter

src/lib/mdns/minimal/Server.cpp

@@ -100,6 +100,7 @@ CHIP_ERROR ServerBase::Listen(chip::Inet::InetLayer * inetLayer, ListenIterator
 
         EndpointInfo * info = &mEndpoints[endpointIndex];
         info->addressType   = addressType;
+        info->interfaceId   = interfaceId;
 
         CHIP_ERROR err = inetLayer->NewUDPEndPoint(&info->udp);
         if (err != CHIP_NO_ERROR)