Vitaly Petrov
134 lines
4.1 KiB
C++
134 lines
4.1 KiB
C++
#include <QCoreApplication>
|
|
#include <QWebSocketServer>
|
|
#include <QWebSocket>
|
|
#include "ptyqt.h"
|
|
#include <QTimer>
|
|
#include <QProcessEnvironment>
|
|
#include <QSysInfo>
|
|
|
|
#define PORT 4242
|
|
|
|
#define COLS 87
|
|
#define ROWS 26
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
QCoreApplication app(argc, argv);
|
|
|
|
//start WebSockets server for receive connections from xterm.js
|
|
QWebSocketServer wsServer("TestServer", QWebSocketServer::NonSecureMode);
|
|
if (!wsServer.listen(QHostAddress::Any, PORT))
|
|
return 1;
|
|
|
|
QMap<QWebSocket *, IPtyProcess *> sessions;
|
|
|
|
//create new session on new connection
|
|
QObject::connect(&wsServer, &QWebSocketServer::newConnection, [&wsServer, &sessions]()
|
|
{
|
|
//handle new connection
|
|
QWebSocket *wSocket = wsServer.nextPendingConnection();
|
|
|
|
//use cmd.exe or bash, depends on target platform
|
|
IPtyProcess::PtyType ptyType = IPtyProcess::WinPty;
|
|
qint32 buildNumber = QSysInfo::kernelVersion().split(".").last().toInt();
|
|
if (buildNumber >= CONPTY_MINIMAL_WINDOWS_VERSION)
|
|
{
|
|
qDebug() << "Use ConPty instead of WinPty";
|
|
ptyType = IPtyProcess::ConPty;
|
|
}
|
|
|
|
//force select WinPty
|
|
ptyType = IPtyProcess::WinPty;
|
|
|
|
QString shellPath = "c:\\Windows\\system32\\cmd.exe";
|
|
//shellPath = "C:\\Program\ Files\\Git\\bin\\bash.exe";
|
|
#ifdef Q_OS_UNIX
|
|
shellPath = "/bin/sh";
|
|
ptyType = IPtyProcess::UnixPty;
|
|
#endif
|
|
|
|
//create new Pty instance
|
|
IPtyProcess *pty = PtyQt::createPtyProcess(ptyType);
|
|
|
|
qDebug() << "New connection" << wSocket->peerAddress() << wSocket->peerPort() << pty->pid();
|
|
|
|
//start Pty process ()
|
|
pty->startProcess(shellPath, QProcessEnvironment::systemEnvironment().toStringList(), COLS, ROWS);
|
|
|
|
if (!pty->lastError().isEmpty())
|
|
{
|
|
qDebug() << pty->lastError();
|
|
delete pty;
|
|
return;
|
|
}
|
|
|
|
//connect read channel from Pty process to write channel on websocket
|
|
QObject::connect(pty->notifier(), &QIODevice::readyRead, [wSocket, pty]()
|
|
{
|
|
QByteArray data = pty->readAll();
|
|
//qDebug() << "< " << data;
|
|
wSocket->sendTextMessage(data);
|
|
});
|
|
|
|
//connect read channel of Websocket to write channel of Pty process
|
|
QObject::connect(wSocket, &QWebSocket::textMessageReceived, [wSocket, pty](const QString &message)
|
|
{
|
|
//qDebug() << "> " << message.size() << message.at(0) << message << message.toUtf8() << QString::fromUtf8(message.toUtf8());
|
|
pty->write(message.toUtf8());
|
|
});
|
|
|
|
//for example handle disconnections, process crashes and stuff like that...
|
|
auto endSessionHandler = [wSocket, &sessions]()
|
|
{
|
|
IPtyProcess *pty = sessions.value(wSocket);
|
|
if (pty == 0)
|
|
return; //because can be called twice
|
|
|
|
sessions.remove(wSocket);
|
|
|
|
qDebug() << "wSockMn" << wSocket << pty;
|
|
|
|
if (wSocket->isValid())
|
|
wSocket->close();
|
|
wSocket->deleteLater();
|
|
|
|
pty->kill();
|
|
delete pty;
|
|
};
|
|
|
|
QObject::connect(wSocket, &QWebSocket::disconnected, endSessionHandler);
|
|
|
|
#ifdef Q_OS_UNIX
|
|
QProcess *shellProcess = qobject_cast<QProcess *>(pty->notifier());
|
|
QObject::connect(shellProcess, QOverload<int, QProcess::ExitStatus>::of(&QProcess::finished),
|
|
[endSessionHandler](int, QProcess::ExitStatus ) { endSessionHandler(); });
|
|
#else
|
|
QLocalSocket *localSocket = qobject_cast<QLocalSocket *>(pty->notifier());
|
|
QObject::connect(localSocket, &QLocalSocket::disconnected, endSessionHandler);
|
|
#endif
|
|
|
|
//add connection to list of active connections
|
|
sessions.insert(wSocket, pty);
|
|
|
|
qDebug() << pty->size();
|
|
});
|
|
|
|
//stop eventloop if needed
|
|
//QTimer::singleShot(5000, [](){ qApp->quit(); });
|
|
|
|
//exec eventloop
|
|
bool res = app.exec();
|
|
|
|
QMapIterator<QWebSocket *, IPtyProcess *> it(sessions);
|
|
while (it.hasNext())
|
|
{
|
|
it.next();
|
|
|
|
it.key()->deleteLater();
|
|
delete it.value();
|
|
}
|
|
sessions.clear();
|
|
|
|
return res;
|
|
}
|