1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/dns/notify_watcher_mac.h"
#include <notify.h>
#include "base/check_op.h"
#include "base/functional/bind.h"
#include "base/mac/mac_util.h"
#include "base/posix/eintr_wrapper.h"
namespace net {
NotifyWatcherMac::NotifyWatcherMac() : notify_fd_(-1), notify_token_(-1) {}
NotifyWatcherMac::~NotifyWatcherMac() {
Cancel();
}
bool NotifyWatcherMac::Watch(const char* key, const CallbackType& callback) {
DCHECK(key);
DCHECK(!callback.is_null());
Cancel();
uint32_t status = notify_register_file_descriptor(
key, ¬ify_fd_, 0, ¬ify_token_);
if (status != NOTIFY_STATUS_OK)
return false;
DCHECK_GE(notify_fd_, 0);
watcher_ = base::FileDescriptorWatcher::WatchReadable(
notify_fd_,
base::BindRepeating(&NotifyWatcherMac::OnFileCanReadWithoutBlocking,
base::Unretained(this)));
callback_ = callback;
return true;
}
void NotifyWatcherMac::Cancel() {
if (notify_fd_ >= 0) {
CancelInternal();
}
}
void NotifyWatcherMac::OnFileCanReadWithoutBlocking() {
int token;
int status = HANDLE_EINTR(read(notify_fd_, &token, sizeof(token)));
if (status != sizeof(token)) {
CancelInternal().Run(false);
return;
}
// Ignoring |token| value to avoid possible endianness mismatch:
// https://openradar.appspot.com/8821081
callback_.Run(true);
}
NotifyWatcherMac::CallbackType NotifyWatcherMac::CancelInternal() {
DCHECK_GE(notify_fd_, 0);
watcher_.reset();
notify_cancel(notify_token_); // Also closes |notify_fd_|.
notify_fd_ = -1;
return std::move(callback_);
}
} // namespace net
|
