1 /*
2  * Copyright 2015 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef BUFFET_AVAHI_MDNS_CLIENT_H_
18 #define BUFFET_AVAHI_MDNS_CLIENT_H_
19 
20 #include <map>
21 #include <string>
22 
23 #include <avahi-client/client.h>
24 #include <avahi-client/publish.h>
25 #include <avahi-common/thread-watch.h>
26 
27 #include "buffet/mdns_client.h"
28 
29 namespace buffet {
30 
31 // Publishes privet service on mDns using Avahi.
32 class AvahiMdnsClient : public MdnsClient {
33  public:
34   explicit AvahiMdnsClient();
35   ~AvahiMdnsClient() override;
36 
37   // weave::provider::DnsServiceDiscovery implementation.
38   void PublishService(const std::string& service_type, uint16_t port,
39                       const std::vector<std::string>& txt) override;
40   void StopPublishing(const std::string& service_type) override;
41 
42  private:
43   static void OnAvahiClientStateUpdate(AvahiClient* s,
44                                        AvahiClientState state,
45                                        void* userdata);
46   void RepublishService();
47 
48   uint16_t prev_port_{0};
49   std::string prev_service_type_;
50   std::string service_name_;
51   std::vector<std::string> txt_records_;
52   std::unique_ptr<AvahiThreadedPoll, decltype(&avahi_threaded_poll_free)>
53       thread_pool_{nullptr, &avahi_threaded_poll_free};
54   std::unique_ptr< ::AvahiClient, decltype(&avahi_client_free)> client_{
55       nullptr, &avahi_client_free};
56   std::unique_ptr<AvahiEntryGroup, decltype(&avahi_entry_group_free)> group_{
57       nullptr, &avahi_entry_group_free};
58 
59   DISALLOW_COPY_AND_ASSIGN(AvahiMdnsClient);
60 };
61 
62 }  // namespace buffet
63 
64 #endif  // BUFFET_AVAHI_MDNS_CLIENT_H_
65