mirror of
https://github.com/Adaptix-Framework/AdaptixC2
synced 2026-06-08 10:20:39 +00:00
Ralf
116 lines
3.1 KiB
C++
116 lines
3.1 KiB
C++
#include <Workers/WebSocketWorker.h>
|
|
#include <Client/AuthProfile.h>
|
|
|
|
WebSocketWorker::WebSocketWorker(AuthProfile* authProfile)
|
|
{
|
|
profile = authProfile;
|
|
}
|
|
|
|
WebSocketWorker::~WebSocketWorker()
|
|
{
|
|
if (pingTimer) {
|
|
pingTimer->stop();
|
|
delete pingTimer;
|
|
pingTimer = nullptr;
|
|
}
|
|
delete webSocket;
|
|
};
|
|
|
|
void WebSocketWorker::run()
|
|
{
|
|
webSocket = new QWebSocket;
|
|
auto SslConf = webSocket->sslConfiguration();
|
|
SslConf.setPeerVerifyMode( QSslSocket::VerifyNone );
|
|
SslConf.setProtocol( QSsl::TlsV1_2OrLater );
|
|
webSocket->setSslConfiguration( SslConf );
|
|
webSocket->ignoreSslErrors();
|
|
|
|
connect( webSocket, &QWebSocket::connected, this, &WebSocketWorker::is_connected );
|
|
connect( webSocket, &QWebSocket::disconnected, this, &WebSocketWorker::is_disconnected );
|
|
connect( webSocket, &QWebSocket::binaryMessageReceived, this, &WebSocketWorker::is_binaryMessageReceived );
|
|
connect( webSocket, &QWebSocket::pong, this, &WebSocketWorker::is_pong );
|
|
|
|
#if QT_VERSION >= QT_VERSION_CHECK(6, 5, 0)
|
|
connect( webSocket, &QWebSocket::errorOccurred, this, &WebSocketWorker::is_error);
|
|
#else
|
|
connect( webSocket, QOverload<QAbstractSocket::SocketError>::of(&QWebSocket::error), this, &WebSocketWorker::is_error);
|
|
#endif
|
|
|
|
QString urlTemplate = "wss://%1:%2%3/connect";
|
|
QString sUrl = urlTemplate.arg( profile->GetHost() ).arg( profile->GetPort() ).arg( profile->GetEndpoint() );
|
|
|
|
QNetworkRequest request;
|
|
request.setUrl(QUrl(sUrl));
|
|
request.setRawHeader("Authorization", "Bearer " + profile->GetAccessToken().toUtf8());
|
|
|
|
webSocket->open(request);
|
|
|
|
exec();
|
|
}
|
|
|
|
void WebSocketWorker::startPingTimer()
|
|
{
|
|
if (!pingTimer) {
|
|
pingTimer = new QTimer(this);
|
|
connect(pingTimer, &QTimer::timeout, this, &WebSocketWorker::sendPing);
|
|
}
|
|
pingTimer->start(PING_INTERVAL_MS);
|
|
}
|
|
|
|
void WebSocketWorker::stopPingTimer()
|
|
{
|
|
if (pingTimer) {
|
|
pingTimer->stop();
|
|
}
|
|
}
|
|
|
|
void WebSocketWorker::sendPing()
|
|
{
|
|
if (webSocket && webSocket->state() == QAbstractSocket::ConnectedState) {
|
|
webSocket->ping();
|
|
}
|
|
}
|
|
|
|
void WebSocketWorker::is_pong(quint64 elapsedTime, const QByteArray &payload)
|
|
{
|
|
Q_UNUSED(elapsedTime)
|
|
Q_UNUSED(payload)
|
|
}
|
|
|
|
void WebSocketWorker::is_error(QAbstractSocket::SocketError error)
|
|
{
|
|
QString errMsg = webSocket->errorString();
|
|
|
|
this->ok = false;
|
|
this->message = "Login failure";
|
|
if (errMsg.contains("511") || errMsg.contains("Network Authentication Required"))
|
|
this->message = "User already connected";
|
|
|
|
Q_EMIT ws_error();
|
|
}
|
|
|
|
void WebSocketWorker::SetProfile(AuthProfile* authProfile)
|
|
{
|
|
profile = authProfile;
|
|
}
|
|
|
|
void WebSocketWorker::is_connected()
|
|
{
|
|
this->ok = true;
|
|
this->message = "";
|
|
startPingTimer();
|
|
Q_EMIT this->connected();
|
|
}
|
|
|
|
void WebSocketWorker::is_disconnected()
|
|
{
|
|
stopPingTimer();
|
|
this->ok = false;
|
|
this->message = "Disconnected from server";
|
|
Q_EMIT this->websocket_closed();
|
|
}
|
|
|
|
void WebSocketWorker::is_binaryMessageReceived(const QByteArray &data)
|
|
{
|
|
Q_EMIT this->received_data( data );
|
|
} |