mirror of
https://github.com/amnezia-vpn/amnezia-client.git
synced 2026-06-22 02:01:08 +07:00
cd-amn
65 lines
1.8 KiB
C++
65 lines
1.8 KiB
C++
/* 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 WINDOWSROUTEMONITOR_H
|
|
#define WINDOWSROUTEMONITOR_H
|
|
|
|
#include <WS2tcpip.h>
|
|
#include <iphlpapi.h>
|
|
#include <windows.h>
|
|
#include <winsock2.h>
|
|
#include <ws2ipdef.h>
|
|
|
|
#include <QHash>
|
|
#include <QMap>
|
|
#include <QObject>
|
|
#include <QSet>
|
|
|
|
#include "ipaddress.h"
|
|
|
|
class WindowsRouteMonitor final : public QObject {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
WindowsRouteMonitor(quint64 luid, QObject* parent);
|
|
~WindowsRouteMonitor();
|
|
|
|
void setDetaultRouteCapture(bool enable);
|
|
|
|
bool addExclusionRoute(const IPAddress& prefix);
|
|
bool deleteExclusionRoute(const IPAddress& prefix);
|
|
|
|
quint64 getLuid() const { return m_luid; }
|
|
|
|
public slots:
|
|
void routeChanged();
|
|
|
|
private:
|
|
bool isRouteExcluded(void* table, const IP_ADDRESS_PREFIX* dest) const;
|
|
static bool routeContainsDest(const IP_ADDRESS_PREFIX* route,
|
|
const IP_ADDRESS_PREFIX* dest);
|
|
static QHostAddress prefixToAddress(const IP_ADDRESS_PREFIX* dest);
|
|
|
|
void flushRouteTable(QHash<IPAddress, MIB_IPFORWARD_ROW2*>& table);
|
|
void updateExclusionRoute(MIB_IPFORWARD_ROW2* data, void* table);
|
|
void updateInterfaceMetrics(int family);
|
|
void updateCapturedRoutes(int family);
|
|
void updateCapturedRoutes(int family, void* table);
|
|
|
|
QSet<IPAddress> m_ownedExclusionRoutes;
|
|
QMap<quint64, ULONG> m_interfaceMetricsIpv4;
|
|
QMap<quint64, ULONG> m_interfaceMetricsIpv6;
|
|
|
|
// Default route cloning
|
|
bool m_defaultRouteCapture = false;
|
|
QHash<IPAddress, MIB_IPFORWARD_ROW2*> m_clonedRoutes;
|
|
|
|
const quint64 m_luid = 0;
|
|
HANDLE m_routeHandle = INVALID_HANDLE_VALUE;
|
|
|
|
static QSet<quint64> s_vpnLuids;
|
|
};
|
|
|
|
#endif /* WINDOWSROUTEMONITOR_H */
|