blob: 785f8ce43cd62964b140555b65c226530ebb31cc (
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef TCP_SOCKET_MONITOR_H
#define TCP_SOCKET_MONITOR_H
#include <chrono>
#include <condition_variable>
#include <mutex>
#include <thread>
#include <unordered_map>
#include <android-base/thread_annotations.h>
#include "netdutils/DumpWriter.h"
#include "utils/String16.h"
#include "Fwmark.h"
struct inet_diag_msg;
struct tcp_info;
namespace android {
namespace net {
using std::chrono::milliseconds;
class TcpSocketMonitor {
public:
using time_point = std::chrono::time_point<std::chrono::steady_clock>;
static const String16 DUMP_KEYWORD;
static const milliseconds kDefaultPollingInterval;
// A subset of fields found in struct inet_diag_msg and struct tcp_info.
struct TcpStats {
// Number of packets sent. Tracks struct tcp_sock data_segs_out.
// Not available on 3.18 kernels.
uint32_t sent;
// Number of packets lost. Tracks struct tcp_sock lost_out.
uint32_t lost;
// Smoothed round trip time. Tracks struct tcp_sock srtt_us.
uint32_t rttUs;
// Milliseconds difference between the last packet sent and last ack received.
int32_t sentAckDiffMs;
// Number of socket stats aggregated in this TcpStats entry.
int32_t nSockets;
};
// Socket metadata used for computing TcpStats diff across sock_diag dumps.
struct SocketEntry {
// Number of packets sent. Tracks struct tcp_sock data_segs_out.
// Not available on 3.18 kernels.
uint32_t sent;
// Number of packets lost. Tracks struct tcp_sock lost_out.
uint32_t lost;
// Last update timestamp for that socket.
time_point lastUpdate;
// Socket mark.
Fwmark mark;
// The uid owning the socket.
uint32_t uid;
};
TcpSocketMonitor();
~TcpSocketMonitor();
void dump(netdutils::DumpWriter& dw);
void setPollingInterval(milliseconds duration);
void resumePolling();
void suspendPolling();
private:
void poll();
void waitForNextPoll();
bool isRunning();
void updateSocketStats(time_point now, Fwmark mark, const struct inet_diag_msg *sockinfo,
const struct tcp_info *tcpinfo, uint32_t tcpinfoLen) REQUIRES(mLock);
// Lock guarding all reads and writes to member variables.
std::mutex mLock;
// Used by the polling thread for sleeping between poll operations.
std::condition_variable mCv;
// The thread that polls sock_diag continuously.
std::thread mPollingThread;
// The duration of a sleep between polls. Can be updated by the instance owner for dynamically
// adjusting the polling rate.
milliseconds mNextSleepDurationMs GUARDED_BY(mLock);
// The time of the last successful poll operation.
time_point mLastPoll GUARDED_BY(mLock);
// True if the polling thread should sleep until notified.
bool mIsSuspended GUARDED_BY(mLock);
// True while the polling thread should poll.
bool mIsRunning GUARDED_BY(mLock);
// Map of SocketEntry structs keyed by socket cookie. This map tracks per-socket data needed for
// computing diffs between sock_diag dumps. Entries for closed sockets are continuously cleaned
// after every dump operation based on timestamps of last updates.
std::unordered_map<uint64_t, SocketEntry> mSocketEntries GUARDED_BY(mLock);
// Map of TcpStats entries aggregated per network and keyed per network id.
// This map tracks per-network data for a single sock_diag dump and is cleared before every dump
// operation.
std::unordered_map<uint32_t, TcpStats> mNetworkStats GUARDED_BY(mLock);
};
} // namespace net
} // namespace android
#endif /* TCP_SOCKET_MONITOR_H */
|
