mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-20 09:07:52 +01:00
1dac6d8839
The address and the related address of local candidates are sanitized accordingly when the mDNS concealment of local IPs is enabled. Also, remote hostname candidates created from signaling are sanitized in stats as well. A couple of unit tests are revised to reflect the desired behavior of AsyncResolverInterface so that when a hostname candidate is resolved, the hostname is kept in the candidate address. Bug: webrtc:9605, chromium:914452 Change-Id: Iad9ad04ce4e50304e44cf04b15b97a7ae2dec960 Reviewed-on: https://webrtc-review.googlesource.com/c/113643 Reviewed-by: Qingsi Wang <qingsi@webrtc.org> Reviewed-by: Steve Anton <steveanton@webrtc.org> Reviewed-by: Seth Hampson <shampson@webrtc.org> Reviewed-by: Jeroen de Borst <jeroendb@webrtc.org> Commit-Queue: Qingsi Wang <qingsi@webrtc.org> Cr-Commit-Position: refs/heads/master@{#25996}
71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
/*
|
|
* Copyright 2018 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef RTC_BASE_FAKE_MDNS_RESPONDER_H_
|
|
#define RTC_BASE_FAKE_MDNS_RESPONDER_H_
|
|
|
|
#include <map>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "rtc_base/mdns_responder_interface.h"
|
|
|
|
#include "rtc_base/helpers.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class FakeMdnsResponder : public MdnsResponderInterface {
|
|
public:
|
|
explicit FakeMdnsResponder(rtc::Thread* thread) : thread_(thread) {}
|
|
~FakeMdnsResponder() = default;
|
|
|
|
void CreateNameForAddress(const rtc::IPAddress& addr,
|
|
NameCreatedCallback callback) override {
|
|
std::string name;
|
|
if (addr_name_map_.find(addr) != addr_name_map_.end()) {
|
|
name = addr_name_map_[addr];
|
|
} else {
|
|
name = std::to_string(next_available_id_++) + ".local";
|
|
addr_name_map_[addr] = name;
|
|
}
|
|
invoker_.AsyncInvoke<void>(
|
|
RTC_FROM_HERE, thread_,
|
|
[callback, addr, name]() { callback(addr, name); });
|
|
}
|
|
void RemoveNameForAddress(const rtc::IPAddress& addr,
|
|
NameRemovedCallback callback) override {
|
|
auto it = addr_name_map_.find(addr);
|
|
if (it != addr_name_map_.end()) {
|
|
addr_name_map_.erase(it);
|
|
}
|
|
bool result = it != addr_name_map_.end();
|
|
invoker_.AsyncInvoke<void>(RTC_FROM_HERE, thread_,
|
|
[callback, result]() { callback(result); });
|
|
}
|
|
|
|
rtc::IPAddress GetMappedAddressForName(const std::string& name) const {
|
|
for (const auto& addr_name_pair : addr_name_map_) {
|
|
if (addr_name_pair.second == name) {
|
|
return addr_name_pair.first;
|
|
}
|
|
}
|
|
return rtc::IPAddress();
|
|
}
|
|
|
|
private:
|
|
uint32_t next_available_id_ = 0;
|
|
std::map<rtc::IPAddress, std::string> addr_name_map_;
|
|
rtc::Thread* thread_;
|
|
rtc::AsyncInvoker invoker_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // RTC_BASE_FAKE_MDNS_RESPONDER_H_
|