blob: 78bfe979405808244f46259c25d7c50d4446dd7b (
plain)
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright (c) 2012 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.
#include "net/tools/quic/quic_epoll_connection_helper.h"
#include <errno.h>
#include <sys/socket.h>
#include "base/logging.h"
#include "base/stl_util.h"
#include "net/base/ip_endpoint.h"
#include "net/quic/crypto/quic_random.h"
#include "net/tools/epoll_server/epoll_server.h"
#include "net/tools/quic/quic_socket_utils.h"
namespace net {
namespace tools {
namespace {
class QuicEpollAlarm : public QuicAlarm {
public:
QuicEpollAlarm(EpollServer* epoll_server,
QuicAlarm::Delegate* delegate)
: QuicAlarm(delegate),
epoll_server_(epoll_server),
epoll_alarm_impl_(this) {}
protected:
virtual void SetImpl() OVERRIDE {
DCHECK(deadline().IsInitialized());
epoll_server_->RegisterAlarm(
deadline().Subtract(QuicTime::Zero()).ToMicroseconds(),
&epoll_alarm_impl_);
}
virtual void CancelImpl() OVERRIDE {
DCHECK(!deadline().IsInitialized());
epoll_alarm_impl_.UnregisterIfRegistered();
}
private:
class EpollAlarmImpl : public EpollAlarm {
public:
explicit EpollAlarmImpl(QuicEpollAlarm* alarm) : alarm_(alarm) {}
virtual int64 OnAlarm() OVERRIDE {
EpollAlarm::OnAlarm();
alarm_->Fire();
// Fire will take care of registering the alarm, if needed.
return 0;
}
private:
QuicEpollAlarm* alarm_;
};
EpollServer* epoll_server_;
EpollAlarmImpl epoll_alarm_impl_;
};
} // namespace
QuicEpollConnectionHelper::QuicEpollConnectionHelper(EpollServer* epoll_server)
: epoll_server_(epoll_server),
clock_(epoll_server),
random_generator_(QuicRandom::GetInstance()) {
}
QuicEpollConnectionHelper::~QuicEpollConnectionHelper() {
}
const QuicClock* QuicEpollConnectionHelper::GetClock() const {
return &clock_;
}
QuicRandom* QuicEpollConnectionHelper::GetRandomGenerator() {
return random_generator_;
}
QuicAlarm* QuicEpollConnectionHelper::CreateAlarm(
QuicAlarm::Delegate* delegate) {
return new QuicEpollAlarm(epoll_server_, delegate);
}
} // namespace tools
} // namespace net
|