mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
130 lines
4.6 KiB
C++
Executable File
130 lines
4.6 KiB
C++
Executable File
/*
|
|
Copyright (c) 2007, Trenton Schulz
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
1. Redistributions of source code must retain the above copyright notice,
|
|
this list of conditions and the following disclaimer.
|
|
|
|
2. Redistributions in binary form must reproduce the above copyright notice,
|
|
this list of conditions and the following disclaimer in the documentation
|
|
and/or other materials provided with the distribution.
|
|
|
|
3. The name of the author may not be used to endorse or promote products
|
|
derived from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
|
|
WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
|
|
EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
|
|
OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
|
|
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
|
|
OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
|
|
ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include <bonjour/bonjourserviceregister.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <QtCore/QSocketNotifier>
|
|
|
|
#include <utils/Logger.h>
|
|
|
|
BonjourServiceRegister::BonjourServiceRegister(QObject *parent)
|
|
: QObject(parent), dnssref(0), bonjourSocket(0)
|
|
{
|
|
setenv("AVAHI_COMPAT_NOWARN", "1", 1);
|
|
}
|
|
|
|
BonjourServiceRegister::~BonjourServiceRegister()
|
|
{
|
|
if (dnssref)
|
|
{
|
|
DNSServiceRefDeallocate(dnssref);
|
|
dnssref = 0;
|
|
}
|
|
}
|
|
|
|
void BonjourServiceRegister::registerService(const BonjourRecord &record, quint16 servicePort, std::vector<std::pair<std::string, std::string>> txt)
|
|
{
|
|
if (dnssref)
|
|
{
|
|
Warning(Logger::getInstance("BonJour"), "Already registered a service for this object, aborting new register");
|
|
return;
|
|
}
|
|
quint16 bigEndianPort = servicePort;
|
|
#if Q_BYTE_ORDER == Q_LITTLE_ENDIAN
|
|
{
|
|
bigEndianPort = 0 | ((servicePort & 0x00ff) << 8) | ((servicePort & 0xff00) >> 8);
|
|
}
|
|
#endif
|
|
|
|
// create txt record
|
|
TXTRecordRef txtRec;
|
|
TXTRecordCreate(&txtRec,0,NULL);
|
|
|
|
// add txt records
|
|
if(!txt.empty())
|
|
{
|
|
for(std::vector<std::pair<std::string, std::string> >::const_iterator it = txt.begin(); it != txt.end(); ++it)
|
|
{
|
|
//Debug(Logger::getInstance("BonJour"), "TXTRecord: key:%s, value:%s",it->first.c_str(),it->second.c_str());
|
|
uint8_t txtLen = (uint8_t)strlen(it->second.c_str());
|
|
TXTRecordSetValue(&txtRec, it->first.c_str(), txtLen, it->second.c_str());
|
|
}
|
|
}
|
|
|
|
DNSServiceErrorType err = DNSServiceRegister(&dnssref, 0, 0, record.serviceName.toUtf8().constData(),
|
|
record.registeredType.toUtf8().constData(),
|
|
(record.replyDomain.isEmpty() ? 0 : record.replyDomain.toUtf8().constData()),
|
|
0, bigEndianPort, TXTRecordGetLength(&txtRec), TXTRecordGetBytesPtr(&txtRec), bonjourRegisterService, this);
|
|
if (err != kDNSServiceErr_NoError)
|
|
{
|
|
emit error(err);
|
|
}
|
|
else
|
|
{
|
|
int sockfd = DNSServiceRefSockFD(dnssref);
|
|
if (sockfd == -1)
|
|
{
|
|
emit error(kDNSServiceErr_Invalid);
|
|
}
|
|
else
|
|
{
|
|
bonjourSocket = new QSocketNotifier(sockfd, QSocketNotifier::Read, this);
|
|
connect(bonjourSocket, SIGNAL(activated(int)), this, SLOT(bonjourSocketReadyRead()));
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void BonjourServiceRegister::bonjourSocketReadyRead()
|
|
{
|
|
DNSServiceErrorType err = DNSServiceProcessResult(dnssref);
|
|
if (err != kDNSServiceErr_NoError)
|
|
emit error(err);
|
|
}
|
|
|
|
|
|
void BonjourServiceRegister::bonjourRegisterService(DNSServiceRef, DNSServiceFlags,
|
|
DNSServiceErrorType errorCode, const char *name,
|
|
const char *regtype, const char *domain,
|
|
void *data)
|
|
{
|
|
BonjourServiceRegister *serviceRegister = static_cast<BonjourServiceRegister *>(data);
|
|
if (errorCode != kDNSServiceErr_NoError)
|
|
{
|
|
emit serviceRegister->error(errorCode);
|
|
}
|
|
else
|
|
{
|
|
serviceRegister->finalRecord = BonjourRecord(QString::fromUtf8(name),
|
|
QString::fromUtf8(regtype),
|
|
QString::fromUtf8(domain));
|
|
emit serviceRegister->serviceRegistered(serviceRegister->finalRecord);
|
|
}
|
|
}
|