mirror of
https://github.com/moonlight-stream/moonlight-qt
synced 2024-12-14 21:32:27 +00:00
393 lines
12 KiB
C++
393 lines
12 KiB
C++
#include "nvhttp.h"
|
|
#include <Limelight.h>
|
|
|
|
#include <QDebug>
|
|
#include <QUuid>
|
|
#include <QtNetwork/QNetworkReply>
|
|
#include <QEventLoop>
|
|
#include <QTimer>
|
|
#include <QXmlStreamReader>
|
|
#include <QSslKey>
|
|
#include <QImageReader>
|
|
#include <QtEndian>
|
|
|
|
#define REQUEST_TIMEOUT_MS 5000
|
|
|
|
NvHTTP::NvHTTP(QString address) :
|
|
m_Address(address)
|
|
{
|
|
m_BaseUrlHttp.setScheme("http");
|
|
m_BaseUrlHttps.setScheme("https");
|
|
m_BaseUrlHttp.setHost(address);
|
|
m_BaseUrlHttps.setHost(address);
|
|
m_BaseUrlHttp.setPort(47989);
|
|
m_BaseUrlHttps.setPort(47984);
|
|
}
|
|
|
|
QVector<int>
|
|
NvHTTP::parseQuad(QString quad)
|
|
{
|
|
QStringList parts = quad.split(".");
|
|
QVector<int> ret;
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
{
|
|
ret.append(parts.at(i).toInt());
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int
|
|
NvHTTP::getCurrentGame(QString serverInfo)
|
|
{
|
|
// GFE 2.8 started keeping currentgame set to the last game played. As a result, it no longer
|
|
// has the semantics that its name would indicate. To contain the effects of this change as much
|
|
// as possible, we'll force the current game to zero if the server isn't in a streaming session.
|
|
QString serverState = getXmlString(serverInfo, "state");
|
|
if (serverState != nullptr && serverState.endsWith("_SERVER_BUSY"))
|
|
{
|
|
return getXmlString(serverInfo, "currentgame").toInt();
|
|
}
|
|
else
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
QString
|
|
NvHTTP::getServerInfo()
|
|
{
|
|
QString serverInfo;
|
|
|
|
try
|
|
{
|
|
// Always try HTTPS first, since it properly reports
|
|
// pairing status (and a few other attributes).
|
|
serverInfo = openConnectionToString(m_BaseUrlHttps,
|
|
"serverinfo",
|
|
nullptr,
|
|
true);
|
|
// Throws if the request failed
|
|
verifyResponseStatus(serverInfo);
|
|
}
|
|
catch (const GfeHttpResponseException& e)
|
|
{
|
|
if (e.getStatusCode() == 401)
|
|
{
|
|
// Certificate validation error, fallback to HTTP
|
|
serverInfo = openConnectionToString(m_BaseUrlHttp,
|
|
"serverinfo",
|
|
nullptr,
|
|
true);
|
|
verifyResponseStatus(serverInfo);
|
|
}
|
|
else
|
|
{
|
|
// Rethrow real errors
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
return serverInfo;
|
|
}
|
|
|
|
static QString
|
|
getSurroundAudioInfoString(int audioConfig)
|
|
{
|
|
int channelMask;
|
|
int channelCount;
|
|
|
|
switch (audioConfig)
|
|
{
|
|
case AUDIO_CONFIGURATION_STEREO:
|
|
channelCount = 2;
|
|
channelMask = 0x3;
|
|
break;
|
|
case AUDIO_CONFIGURATION_51_SURROUND:
|
|
channelCount = 6;
|
|
channelMask = 0xFC;
|
|
break;
|
|
default:
|
|
Q_ASSERT(false);
|
|
return 0;
|
|
}
|
|
|
|
return QString::number(channelMask << 16 | channelCount);
|
|
}
|
|
|
|
void
|
|
NvHTTP::launchApp(int appId,
|
|
PSTREAM_CONFIGURATION streamConfig,
|
|
bool sops,
|
|
bool localAudio,
|
|
int gamepadMask)
|
|
{
|
|
QString response =
|
|
openConnectionToString(m_BaseUrlHttps,
|
|
"launch",
|
|
"appid="+QString::number(appId)+
|
|
"&mode="+QString::number(streamConfig->width)+"x"+
|
|
QString::number(streamConfig->height)+"x"+
|
|
QString::number(streamConfig->fps)+
|
|
"&additionalStates=1&sops="+QString::number(sops ? 1 : 0)+
|
|
"&rikey="+QByteArray(streamConfig->remoteInputAesKey, sizeof(streamConfig->remoteInputAesKey)).toHex()+
|
|
"&rikeyid="+QString::number(qFromBigEndian(*(int*)streamConfig->remoteInputAesIv))+
|
|
(streamConfig->enableHdr ?
|
|
"&hdrMode=1&clientHdrCapVersion=0&clientHdrCapSupportedFlagsInUint32=0&clientHdrCapMetaDataId=NV_STATIC_METADATA_TYPE_1&clientHdrCapDisplayData=0x0x0x0x0x0x0x0x0x0x0" :
|
|
"")+
|
|
"&localAudioPlayMode="+QString::number(localAudio ? 1 : 0)+
|
|
"&surroundAudioInfo="+getSurroundAudioInfoString(streamConfig->audioConfiguration)+
|
|
"&remoteControllersBitmap="+QString::number(gamepadMask)+
|
|
"&gcmap="+QString::number(gamepadMask),
|
|
false);
|
|
|
|
// Throws if the request failed
|
|
verifyResponseStatus(response);
|
|
}
|
|
|
|
void
|
|
NvHTTP::resumeApp(PSTREAM_CONFIGURATION streamConfig)
|
|
{
|
|
QString response =
|
|
openConnectionToString(m_BaseUrlHttps,
|
|
"resume",
|
|
"rikey="+QString(QByteArray(streamConfig->remoteInputAesKey, sizeof(streamConfig->remoteInputAesKey)).toHex())+
|
|
"&rikeyid="+QString::number(qFromBigEndian(*(int*)streamConfig->remoteInputAesIv))+
|
|
"&surroundAudioInfo="+getSurroundAudioInfoString(streamConfig->audioConfiguration),
|
|
false);
|
|
|
|
// Throws if the request failed
|
|
verifyResponseStatus(response);
|
|
}
|
|
|
|
void
|
|
NvHTTP::quitApp()
|
|
{
|
|
QString response =
|
|
openConnectionToString(m_BaseUrlHttps,
|
|
"cancel",
|
|
nullptr,
|
|
false);
|
|
|
|
// Throws if the request failed
|
|
verifyResponseStatus(response);
|
|
|
|
// Newer GFE versions will just return success even if quitting fails
|
|
// if we're not the original requestor.
|
|
if (getCurrentGame(getServerInfo()) != 0) {
|
|
// Generate a synthetic GfeResponseException letting the caller know
|
|
// that they can't kill someone else's stream.
|
|
throw GfeHttpResponseException(599, "");
|
|
}
|
|
}
|
|
|
|
QVector<NvDisplayMode>
|
|
NvHTTP::getDisplayModeList(QString serverInfo)
|
|
{
|
|
QXmlStreamReader xmlReader(serverInfo);
|
|
QVector<NvDisplayMode> modes;
|
|
|
|
while (!xmlReader.atEnd()) {
|
|
while (xmlReader.readNextStartElement()) {
|
|
QStringRef name = xmlReader.name();
|
|
if (xmlReader.name() == "DisplayMode") {
|
|
modes.append(NvDisplayMode());
|
|
}
|
|
else if (xmlReader.name() == "Width") {
|
|
modes.last().width = xmlReader.readElementText().toInt();
|
|
}
|
|
else if (xmlReader.name() == "Height") {
|
|
modes.last().height = xmlReader.readElementText().toInt();
|
|
}
|
|
else if (xmlReader.name() == "RefreshRate") {
|
|
modes.last().refreshRate = xmlReader.readElementText().toInt();
|
|
}
|
|
}
|
|
}
|
|
|
|
return modes;
|
|
}
|
|
|
|
QVector<NvApp>
|
|
NvHTTP::getAppList()
|
|
{
|
|
QString appxml = openConnectionToString(m_BaseUrlHttps,
|
|
"applist",
|
|
nullptr,
|
|
true);
|
|
verifyResponseStatus(appxml);
|
|
|
|
QXmlStreamReader xmlReader(appxml);
|
|
QVector<NvApp> apps;
|
|
while (!xmlReader.atEnd()) {
|
|
while (xmlReader.readNextStartElement()) {
|
|
QStringRef name = xmlReader.name();
|
|
if (xmlReader.name() == "App") {
|
|
// We must have a valid app before advancing to the next one
|
|
if (!apps.isEmpty() && !apps.last().isInitialized()) {
|
|
qWarning() << "Invalid applist XML";
|
|
Q_ASSERT(false);
|
|
return QVector<NvApp>();
|
|
}
|
|
apps.append(NvApp());
|
|
}
|
|
else if (xmlReader.name() == "AppTitle") {
|
|
apps.last().name = xmlReader.readElementText();
|
|
}
|
|
else if (xmlReader.name() == "ID") {
|
|
apps.last().id = xmlReader.readElementText().toInt();
|
|
}
|
|
else if (xmlReader.name() == "IsHdrSupported") {
|
|
apps.last().hdrSupported = xmlReader.readElementText() == "1";
|
|
}
|
|
}
|
|
}
|
|
|
|
return apps;
|
|
}
|
|
|
|
void
|
|
NvHTTP::verifyResponseStatus(QString xml)
|
|
{
|
|
QXmlStreamReader xmlReader(xml);
|
|
|
|
while (xmlReader.readNextStartElement())
|
|
{
|
|
if (xmlReader.name() == "root")
|
|
{
|
|
int statusCode = xmlReader.attributes().value("status_code").toInt();
|
|
if (statusCode == 200)
|
|
{
|
|
// Successful
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
QString statusMessage = xmlReader.attributes().value("status_message").toString();
|
|
qDebug() << "Request failed: " << statusCode << " " << statusMessage;
|
|
throw GfeHttpResponseException(statusCode, statusMessage);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
QImage
|
|
NvHTTP::getBoxArt(int appId)
|
|
{
|
|
QNetworkReply* reply = openConnection(m_BaseUrlHttps,
|
|
"appasset",
|
|
"appid="+QString::number(appId)+
|
|
"&AssetType=2&AssetIdx=0",
|
|
true);
|
|
QImage image = QImageReader(reply).read();
|
|
delete reply;
|
|
|
|
return image;
|
|
}
|
|
|
|
QByteArray
|
|
NvHTTP::getXmlStringFromHex(QString xml,
|
|
QString tagName)
|
|
{
|
|
QString str = getXmlString(xml, tagName);
|
|
if (str == nullptr)
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
return QByteArray::fromHex(str.toLatin1());
|
|
}
|
|
|
|
QString
|
|
NvHTTP::getXmlString(QString xml,
|
|
QString tagName)
|
|
{
|
|
QXmlStreamReader xmlReader(xml);
|
|
|
|
while (!xmlReader.atEnd())
|
|
{
|
|
if (xmlReader.readNext() != QXmlStreamReader::StartElement)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (xmlReader.name() == tagName)
|
|
{
|
|
return xmlReader.readElementText();
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
QString
|
|
NvHTTP::openConnectionToString(QUrl baseUrl,
|
|
QString command,
|
|
QString arguments,
|
|
bool enableTimeout)
|
|
{
|
|
QNetworkReply* reply = openConnection(baseUrl, command, arguments, enableTimeout);
|
|
QString ret;
|
|
|
|
ret = QTextStream(reply).readAll();
|
|
delete reply;
|
|
|
|
return ret;
|
|
}
|
|
|
|
QNetworkReply*
|
|
NvHTTP::openConnection(QUrl baseUrl,
|
|
QString command,
|
|
QString arguments,
|
|
bool enableTimeout)
|
|
{
|
|
// Build a URL for the request
|
|
QUrl url(baseUrl);
|
|
url.setPath("/" + command);
|
|
url.setQuery("uniqueid=" + IdentityManager::get()->getUniqueId() +
|
|
"&uuid=" + QUuid::createUuid().toRfc4122().toHex() +
|
|
((arguments != nullptr) ? ("&" + arguments) : ""));
|
|
|
|
QNetworkRequest request = QNetworkRequest(url);
|
|
|
|
// Add our client certificate
|
|
request.setSslConfiguration(IdentityManager::get()->getSslConfig());
|
|
|
|
QNetworkReply* reply = m_Nam.get(request);
|
|
|
|
// Ignore self-signed certificate errors (since GFE uses them)
|
|
reply->ignoreSslErrors();
|
|
|
|
// Run the request with a timeout if requested
|
|
QEventLoop loop;
|
|
QObject::connect(reply, SIGNAL(finished()), &loop, SLOT(quit()));
|
|
if (enableTimeout)
|
|
{
|
|
QTimer::singleShot(REQUEST_TIMEOUT_MS, &loop, SLOT(quit()));
|
|
}
|
|
qDebug() << "Executing request: " << url.toString();
|
|
loop.exec(QEventLoop::ExcludeUserInputEvents);
|
|
|
|
// Abort the request if it timed out
|
|
if (!reply->isFinished())
|
|
{
|
|
qDebug() << "Aborting timed out request for " << url.toString();
|
|
reply->abort();
|
|
}
|
|
|
|
// We must clear out cached authentication and connections or
|
|
// GFE will puke next time
|
|
m_Nam.clearAccessCache();
|
|
|
|
// Handle error
|
|
if (reply->error() != QNetworkReply::NoError)
|
|
{
|
|
qDebug() << command << " request failed with error " << reply->error();
|
|
GfeHttpResponseException exception(reply->error(), reply->errorString());
|
|
delete reply;
|
|
throw exception;
|
|
}
|
|
|
|
return reply;
|
|
}
|