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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set et sw=2 ts=4: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef NETLINKSERVICE_H_
#define NETLINKSERVICE_H_
#include <netinet/in.h>
#include <linux/netlink.h>
#include "nsIRunnable.h"
#include "nsThreadUtils.h"
#include "nsCOMPtr.h"
#include "mozilla/Mutex.h"
#include "mozilla/TimeStamp.h"
#include "nsClassHashtable.h"
#include "mozilla/SHA1.h"
#include "mozilla/UniquePtr.h"
#include "nsTArray.h"
namespace mozilla {
namespace net {
class NetlinkAddress;
class NetlinkNeighbor;
class NetlinkLink;
class NetlinkRoute;
class NetlinkMsg;
class NetlinkServiceListener : public nsISupports {
public:
virtual void OnNetworkChanged() = 0;
virtual void OnNetworkIDChanged() = 0;
virtual void OnLinkUp() = 0;
virtual void OnLinkDown() = 0;
virtual void OnLinkStatusKnown() = 0;
virtual void OnDnsSuffixListUpdated() = 0;
protected:
virtual ~NetlinkServiceListener() = default;
};
class NetlinkService : public nsIRunnable {
virtual ~NetlinkService();
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIRUNNABLE
NetlinkService();
nsresult Init(NetlinkServiceListener* aListener);
nsresult Shutdown();
void GetNetworkID(nsACString& aNetworkID);
void GetIsLinkUp(bool* aIsUp);
nsresult GetDnsSuffixList(nsTArray<nsCString>& aDnsSuffixList);
private:
void EnqueueGenMsg(uint16_t aMsgType, uint8_t aFamily);
void EnqueueRtMsg(uint8_t aFamily, void* aAddress);
void RemovePendingMsg();
mozilla::Mutex mMutex;
void OnNetlinkMessage(int aNetlinkSocket);
void OnLinkMessage(struct nlmsghdr* aNlh);
void OnAddrMessage(struct nlmsghdr* aNlh);
void OnRouteMessage(struct nlmsghdr* aNlh);
void OnNeighborMessage(struct nlmsghdr* aNlh);
void OnRouteCheckResult(struct nlmsghdr* aNlh);
void UpdateLinkStatus();
void TriggerNetworkIDCalculation();
int GetPollWait();
void GetGWNeighboursForFamily(uint8_t aFamily,
nsTArray<NetlinkNeighbor*>& aGwNeighbors);
bool CalculateIDForFamily(uint8_t aFamily, mozilla::SHA1Sum* aSHA1);
void CalculateNetworkID();
void ComputeDNSSuffixList();
nsCOMPtr<nsIThread> mThread;
bool mInitialScanFinished;
// A pipe to signal shutdown with.
int mShutdownPipe[2];
// IP addresses that are used to check the route for public traffic.
struct in_addr mRouteCheckIPv4;
struct in6_addr mRouteCheckIPv6;
pid_t mPid;
uint32_t mMsgId;
bool mLinkUp;
// Flag indicating that network ID could change and should be recalculated.
// Calculation is postponed until we receive responses to all enqueued
// messages.
bool mRecalculateNetworkId;
// Flag indicating that network change event needs to be sent even if
// network ID hasn't changed.
bool mSendNetworkChangeEvent;
// Time stamp of setting mRecalculateNetworkId to true
mozilla::TimeStamp mTriggerTime;
nsCString mNetworkId;
nsTArray<nsCString> mDNSSuffixList;
class LinkInfo {
public:
explicit LinkInfo(UniquePtr<NetlinkLink>&& aLink);
virtual ~LinkInfo();
// Updates mIsUp according to current mLink and mAddresses. Returns true if
// the value has changed.
bool UpdateStatus();
// NetlinkLink structure for this link
UniquePtr<NetlinkLink> mLink;
// All IPv4/IPv6 addresses on this link
nsTArray<UniquePtr<NetlinkAddress>> mAddresses;
// All neighbors on this link, key is an address
nsClassHashtable<nsCStringHashKey, NetlinkNeighbor> mNeighbors;
// Default IPv4/IPv6 routes
nsTArray<UniquePtr<NetlinkRoute>> mDefaultRoutes;
// Link is up when it's running, it's not a loopback and there is
// a non-local address associated with it.
bool mIsUp;
};
bool CalculateIDForEthernetLink(uint8_t aFamily,
NetlinkRoute* aRouteCheckResult,
uint32_t aRouteCheckIfIdx,
LinkInfo* aRouteCheckLinkInfo,
mozilla::SHA1Sum* aSHA1);
bool CalculateIDForNonEthernetLink(uint8_t aFamily,
NetlinkRoute* aRouteCheckResult,
nsTArray<nsCString>& aLinkNamesToHash,
uint32_t aRouteCheckIfIdx,
LinkInfo* aRouteCheckLinkInfo,
mozilla::SHA1Sum* aSHA1);
nsClassHashtable<nsUint32HashKey, LinkInfo> mLinks;
// Route for mRouteCheckIPv4 address
UniquePtr<NetlinkRoute> mIPv4RouteCheckResult;
// Route for mRouteCheckIPv6 address
UniquePtr<NetlinkRoute> mIPv6RouteCheckResult;
nsTArray<UniquePtr<NetlinkMsg>> mOutgoingMessages;
RefPtr<NetlinkServiceListener> mListener;
};
} // namespace net
} // namespace mozilla
#endif /* NETLINKSERVICE_H_ */
|
