moonlight-stream/moonlight-qt
1
0
Fork 0
mirror of https://github.com/moonlight-stream/moonlight-qt.git synced 2026-04-13 19:26:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement mDNS PC discovery

Browse Source
This commit is contained in:
Cameron Gutman
2018-06-30 23:07:31 -07:00
parent 7186f20e05
commit 5729f65138
2 changed files with 94 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
app/backend/computermanager.cpp
View File

@@ -154,7 +154,8 @@ bool NvComputer::update(NvComputer& that)
ComputerManager::ComputerManager(QObject *parent)
: QObject(parent),
m_Polling(false)
m_Polling(false),
m_MdnsBrowser(nullptr)
{
QSettings settings;
@@ -186,8 +187,25 @@ void ComputerManager::startPolling()
{
QWriteLocker lock(&m_Lock);
if (m_Polling) {
return;
}
m_Polling = true;
// Start an MDNS query for GameStream hosts
m_MdnsBrowser = new QMdnsEngine::Browser(&m_MdnsServer, "_nvstream._tcp.local.", &m_MdnsCache);
connect(m_MdnsBrowser, &QMdnsEngine::Browser::serviceAdded,
this, [this](const QMdnsEngine::Service& service) {
qDebug() << "Discovered mDNS host: " << service.hostname();
MdnsPendingComputer* pendingComputer = new MdnsPendingComputer(&m_MdnsServer, &m_MdnsCache, service);
connect(pendingComputer, SIGNAL(resolvedv4(MdnsPendingComputer*,QHostAddress)),
this, SLOT(handleMdnsServiceResolved(MdnsPendingComputer*,QHostAddress)));
m_PendingResolution.append(pendingComputer);
});
// Start polling threads for each known host
QMapIterator<QString, NvComputer*> i(m_KnownHosts);
while (i.hasNext()) {
i.next();
@@ -195,6 +213,17 @@ void ComputerManager::startPolling()
}
}
void ComputerManager::handleMdnsServiceResolved(MdnsPendingComputer* computer,
const QHostAddress& address)
{
qDebug() << "Resolved " << computer->hostname() << " to " << address.toString();
addNewHost(address.toString(), true);
m_PendingResolution.removeOne(computer);
computer->deleteLater();
}
QVector<NvComputer*> ComputerManager::getComputers()
{
QReadLocker lock(&m_Lock);
@@ -229,8 +258,23 @@ void ComputerManager::stopPollingAsync()
{
QWriteLocker lock(&m_Lock);
if (!m_Polling) {
return;
}
m_Polling = false;
// Delete machines that haven't been resolved yet
while (!m_PendingResolution.isEmpty()) {
MdnsPendingComputer* computer = m_PendingResolution.first();
computer->deleteLater();
m_PendingResolution.removeFirst();
}
// Delete the browser to stop discovery
delete m_MdnsBrowser;
m_MdnsBrowser = nullptr;
// Interrupt all threads, but don't wait for them to terminate
QMutableMapIterator<QString, QThread*> i(m_PollThreads);
while (i.hasNext()) {