You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.0 KiB
72 lines
2.0 KiB
4 months ago
|
// Copyright 2019 The Chromium Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
|
||
|
|
#ifndef DISCOVERY_MDNS_MDNS_RECEIVER_H_
|
||
|
|
#define DISCOVERY_MDNS_MDNS_RECEIVER_H_
|
||
|
|
|
||
|
|
#include <functional>
|
||
|
|
|
||
|
|
#include "discovery/common/config.h"
|
||
|
|
#include "platform/api/udp_socket.h"
|
||
|
|
#include "platform/base/error.h"
|
||
|
|
#include "platform/base/udp_packet.h"
|
||
|
|
|
||
|
|
namespace openscreen {
|
||
|
|
namespace discovery {
|
||
|
|
|
||
|
|
class MdnsMessage;
|
||
|
|
|
||
|
|
class MdnsReceiver {
|
||
|
|
public:
|
||
|
|
class ResponseClient {
|
||
|
|
public:
|
||
|
|
virtual ~ResponseClient();
|
||
|
|
|
||
|
|
virtual void OnMessageReceived(const MdnsMessage& message) = 0;
|
||
|
|
};
|
||
|
|
|
||
|
|
// MdnsReceiver does not own |socket| and |delegate|
|
||
|
|
// and expects that the lifetime of these objects exceeds the lifetime of
|
||
|
|
// MdnsReceiver.
|
||
|
|
explicit MdnsReceiver(Config config);
|
||
|
|
MdnsReceiver(const MdnsReceiver& other) = delete;
|
||
|
|
MdnsReceiver(MdnsReceiver&& other) noexcept = delete;
|
||
|
|
MdnsReceiver& operator=(const MdnsReceiver& other) = delete;
|
||
|
|
MdnsReceiver& operator=(MdnsReceiver&& other) noexcept = delete;
|
||
|
|
~MdnsReceiver();
|
||
|
|
|
||
|
|
void SetQueryCallback(
|
||
|
|
std::function<void(const MdnsMessage&, const IPEndpoint& src)> callback);
|
||
|
|
void AddResponseCallback(ResponseClient* callback);
|
||
|
|
void RemoveResponseCallback(ResponseClient* callback);
|
||
|
|
|
||
|
|
// The receiver can be started and stopped multiple times.
|
||
|
|
// Start and Stop are both synchronous calls. When MdnsReceiver has not yet
|
||
|
|
// been started, OnRead callbacks it receives from the task runner will be
|
||
|
|
// no-ops.
|
||
|
|
void Start();
|
||
|
|
void Stop();
|
||
|
|
|
||
|
|
void OnRead(UdpSocket* socket, ErrorOr<UdpPacket> packet);
|
||
|
|
|
||
|
|
private:
|
||
|
|
enum class State {
|
||
|
|
kStopped,
|
||
|
|
kRunning,
|
||
|
|
};
|
||
|
|
|
||
|
|
std::function<void(const MdnsMessage&, const IPEndpoint& src)>
|
||
|
|
query_callback_;
|
||
|
|
State state_ = State::kStopped;
|
||
|
|
|
||
|
|
std::vector<ResponseClient*> response_clients_;
|
||
|
|
|
||
|
|
Config config_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace discovery
|
||
|
|
} // namespace openscreen
|
||
|
|
|
||
|
|
#endif // DISCOVERY_MDNS_MDNS_RECEIVER_H_
|