119 lines
3.3 KiB
C++
119 lines
3.3 KiB
C++
#include <QDebug>
|
|
#include <QFile>
|
|
|
|
#include <core/openvpnconfigurator.h>
|
|
#include <core/servercontroller.h>
|
|
|
|
#include "protocols/openvpnprotocol.h"
|
|
#include "utils.h"
|
|
#include "vpnconnection.h"
|
|
|
|
VpnConnection::VpnConnection(QObject* parent) : QObject(parent)
|
|
{
|
|
VpnProtocol::initializeCommunicator(parent);
|
|
}
|
|
|
|
void VpnConnection::onBytesChanged(quint64 receivedBytes, quint64 sentBytes)
|
|
{
|
|
emit bytesChanged(receivedBytes, sentBytes);
|
|
}
|
|
|
|
void VpnConnection::onConnectionStateChanged(VpnProtocol::ConnectionState state)
|
|
{
|
|
emit connectionStateChanged(state);
|
|
}
|
|
|
|
ErrorCode VpnConnection::lastError() const
|
|
{
|
|
if (!m_vpnProtocol.data()) {
|
|
return ErrorCode::InternalError;
|
|
}
|
|
|
|
return m_vpnProtocol.data()->lastError();
|
|
}
|
|
|
|
ErrorCode VpnConnection::requestVpnConfig(const ServerCredentials &credentials, Protocol protocol)
|
|
{
|
|
ErrorCode errorCode = ErrorCode::NoError;
|
|
if (protocol == Protocol::OpenVpn) {
|
|
QString configData = OpenVpnConfigurator::genOpenVpnConfig(credentials, &errorCode);
|
|
if (errorCode) {
|
|
return errorCode;
|
|
}
|
|
|
|
QFile file(Utils::defaultVpnConfigFileName());
|
|
if (file.open(QIODevice::WriteOnly | QIODevice::Truncate)){
|
|
QTextStream stream(&file);
|
|
stream << configData << endl;
|
|
return ErrorCode::NoError;
|
|
}
|
|
|
|
return ErrorCode::FailedToSaveConfigData;
|
|
}
|
|
else if (protocol == Protocol::ShadowSocks) {
|
|
// Request OpenVPN config and ShadowSocks
|
|
return ErrorCode::NotImplementedError;
|
|
}
|
|
return ErrorCode::NotImplementedError;
|
|
|
|
}
|
|
|
|
ErrorCode VpnConnection::connectToVpn(const ServerCredentials &credentials, Protocol protocol)
|
|
{
|
|
// TODO: Try protocols one by one in case of Protocol::Any
|
|
// TODO: Implement some behavior in case if connection not stable
|
|
qDebug() << "Connect to VPN";
|
|
|
|
if (protocol == Protocol::Any || protocol == Protocol::OpenVpn) {
|
|
ErrorCode e = requestVpnConfig(credentials, Protocol::OpenVpn);
|
|
if (e) {
|
|
return e;
|
|
}
|
|
if (m_vpnProtocol) {
|
|
disconnect(m_vpnProtocol.data(), &VpnProtocol::protocolError, this, &VpnConnection::vpnProtocolError);
|
|
}
|
|
m_vpnProtocol.reset(new OpenVpnProtocol());
|
|
connect(m_vpnProtocol.data(), &VpnProtocol::protocolError, this, &VpnConnection::vpnProtocolError);
|
|
}
|
|
else if (protocol == Protocol::ShadowSocks) {
|
|
return ErrorCode::NotImplementedError;
|
|
}
|
|
|
|
connect(m_vpnProtocol.data(), SIGNAL(connectionStateChanged(VpnProtocol::ConnectionState)), this, SLOT(onConnectionStateChanged(VpnProtocol::ConnectionState)));
|
|
connect(m_vpnProtocol.data(), SIGNAL(bytesChanged(quint64, quint64)), this, SLOT(onBytesChanged(quint64, quint64)));
|
|
|
|
return m_vpnProtocol.data()->start();
|
|
}
|
|
|
|
QString VpnConnection::bytesToText(quint64 bytes)
|
|
{
|
|
return QString("%1 %2").arg(bytes / 1000000).arg(tr("Mbps"));
|
|
}
|
|
|
|
void VpnConnection::disconnectFromVpn()
|
|
{
|
|
qDebug() << "Disconnect from VPN";
|
|
|
|
if (!m_vpnProtocol.data()) {
|
|
return;
|
|
}
|
|
m_vpnProtocol.data()->stop();
|
|
}
|
|
|
|
bool VpnConnection::onConnected() const
|
|
{
|
|
if (!m_vpnProtocol.data()) {
|
|
return false;
|
|
}
|
|
|
|
return m_vpnProtocol.data()->onConnected();
|
|
}
|
|
|
|
bool VpnConnection::onDisconnected() const
|
|
{
|
|
if (!m_vpnProtocol.data()) {
|
|
return true;
|
|
}
|
|
|
|
return m_vpnProtocol.data()->onDisconnected();
|
|
}
|