You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
430 lines
8.8 KiB
430 lines
8.8 KiB
/* -*- Mode: C; tab-width: 4 -*-
|
|
*
|
|
* Copyright (c) 2009 Apple Computer, Inc. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
#include "resource.h" // main symbols
|
|
|
|
|
|
|
|
#include "DLLX.h"
|
|
|
|
#include "DNSSDEventManager.h"
|
|
|
|
#include <CommonServices.h>
|
|
|
|
#include <DebugServices.h>
|
|
|
|
#include <dns_sd.h>
|
|
|
|
#include <map>
|
|
|
|
|
|
|
|
|
|
|
|
#if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)
|
|
|
|
#error "Single-threaded COM objects are not properly supported on Windows CE platform, such as the Windows Mobile platforms that do not include full DCOM support. Define _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA to force ATL to support creating single-thread COM object's and allow use of it's single-threaded COM object implementations. The threading model in your rgs file was set to 'Free' as that is the only threading model supported in non DCOM Windows CE platforms."
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// CDNSSDService
|
|
|
|
|
|
|
|
class ATL_NO_VTABLE CDNSSDService :
|
|
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
|
|
public CComCoClass<CDNSSDService, &CLSID_DNSSDService>,
|
|
|
|
public IDispatchImpl<IDNSSDService, &IID_IDNSSDService, &LIBID_Bonjour, /*wMajor =*/ 1, /*wMinor =*/ 0>
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
|
|
|
|
typedef CComObjectRootEx<CComSingleThreadModel> Super;
|
|
|
|
|
|
|
|
CDNSSDService()
|
|
|
|
:
|
|
|
|
m_isPrimary( FALSE ),
|
|
|
|
m_eventManager( NULL ),
|
|
|
|
m_stopped( FALSE ),
|
|
|
|
m_primary( NULL ),
|
|
|
|
m_subord( NULL )
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_DNSSDSERVICE)
|
|
|
|
|
|
|
|
|
|
|
|
BEGIN_COM_MAP(CDNSSDService)
|
|
|
|
COM_INTERFACE_ENTRY(IDNSSDService)
|
|
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
|
|
END_COM_MAP()
|
|
|
|
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
|
|
|
|
HRESULT
|
|
|
|
FinalConstruct();
|
|
|
|
|
|
|
|
void
|
|
|
|
FinalRelease();
|
|
|
|
|
|
|
|
public:
|
|
|
|
|
|
|
|
inline DNSServiceRef
|
|
|
|
GetPrimaryRef()
|
|
|
|
{
|
|
|
|
return m_primary;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline void
|
|
|
|
SetPrimaryRef( DNSServiceRef primary )
|
|
|
|
{
|
|
|
|
m_primary = primary;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline DNSServiceRef
|
|
|
|
GetSubordRef()
|
|
|
|
{
|
|
|
|
return m_subord;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline void
|
|
|
|
SetSubordRef( DNSServiceRef subord )
|
|
|
|
{
|
|
|
|
m_subord = subord;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline CDNSSDEventManager*
|
|
|
|
GetEventManager()
|
|
|
|
{
|
|
|
|
return m_eventManager;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline void
|
|
|
|
SetEventManager( IDNSSDEventManager * eventManager )
|
|
|
|
{
|
|
|
|
if ( m_eventManager )
|
|
|
|
{
|
|
|
|
m_eventManager->Release();
|
|
|
|
m_eventManager = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ( eventManager )
|
|
|
|
{
|
|
|
|
m_eventManager = dynamic_cast< CDNSSDEventManager* >( eventManager );
|
|
|
|
check( m_eventManager );
|
|
|
|
m_eventManager->AddRef();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
inline BOOL
|
|
|
|
Stopped()
|
|
|
|
{
|
|
|
|
return m_stopped;
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
DomainEnumReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t ifIndex,
|
|
DNSServiceErrorType errorCode,
|
|
const char *replyDomain,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
BrowseReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t interfaceIndex,
|
|
DNSServiceErrorType errorCode,
|
|
const char *serviceName,
|
|
const char *regtype,
|
|
const char *replyDomain,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
|
|
ResolveReply
|
|
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t interfaceIndex,
|
|
DNSServiceErrorType errorCode,
|
|
const char *fullname,
|
|
const char *hosttarget,
|
|
uint16_t port,
|
|
uint16_t txtLen,
|
|
const unsigned char *txtRecord,
|
|
void *context
|
|
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
RegisterReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
DNSServiceErrorType errorCode,
|
|
const char *name,
|
|
const char *regtype,
|
|
const char *domain,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
QueryRecordReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t interfaceIndex,
|
|
DNSServiceErrorType errorCode,
|
|
const char *fullname,
|
|
uint16_t rrtype,
|
|
uint16_t rrclass,
|
|
uint16_t rdlen,
|
|
const void *rdata,
|
|
uint32_t ttl,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
GetAddrInfoReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t interfaceIndex,
|
|
DNSServiceErrorType errorCode,
|
|
const char *hostname,
|
|
const struct sockaddr *address,
|
|
uint32_t ttl,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
NATPortMappingReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
uint32_t interfaceIndex,
|
|
DNSServiceErrorType errorCode,
|
|
uint32_t externalAddress, /* four byte IPv4 address in network byte order */
|
|
DNSServiceProtocol protocol,
|
|
uint16_t internalPort,
|
|
uint16_t externalPort, /* may be different than the requested port */
|
|
uint32_t ttl, /* may be different than the requested ttl */
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
static void DNSSD_API
|
|
RegisterRecordReply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSRecordRef RecordRef,
|
|
DNSServiceFlags flags,
|
|
DNSServiceErrorType errorCode,
|
|
void *context
|
|
);
|
|
|
|
|
|
|
|
inline BOOL
|
|
|
|
ShouldHandleReply( DNSServiceErrorType errorCode, CDNSSDEventManager *& eventManager );
|
|
|
|
|
|
|
|
static LRESULT CALLBACK
|
|
|
|
WndProc( HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam );
|
|
|
|
|
|
|
|
typedef std::map< SOCKET, CDNSSDService* > SocketMap;
|
|
|
|
|
|
|
|
static BOOL m_registeredWindowClass;
|
|
|
|
static HWND m_hiddenWindow;
|
|
|
|
static SocketMap m_socketMap;
|
|
|
|
CDNSSDEventManager * m_eventManager;
|
|
|
|
BOOL m_stopped;
|
|
|
|
BOOL m_isPrimary;
|
|
|
|
DNSServiceRef m_primary;
|
|
|
|
DNSServiceRef m_subord;
|
|
|
|
public:
|
|
|
|
STDMETHOD(EnumerateDomains)(DNSSDFlags flags, ULONG ifIndex, IDNSSDEventManager *eventManager, IDNSSDService **service);
|
|
|
|
STDMETHOD(Browse)(DNSSDFlags flags, ULONG interfaceIndex, BSTR regtype, BSTR domain, IDNSSDEventManager* eventManager, IDNSSDService** sdref);
|
|
|
|
STDMETHOD(Resolve)(DNSSDFlags flags, ULONG ifIndex, BSTR serviceName, BSTR regType, BSTR domain, IDNSSDEventManager* eventManager, IDNSSDService** service);
|
|
|
|
STDMETHOD(Register)(DNSSDFlags flags, ULONG ifIndex, BSTR name, BSTR regType, BSTR domain, BSTR host, USHORT port, ITXTRecord *record, IDNSSDEventManager *eventManager, IDNSSDService **service);
|
|
|
|
STDMETHOD(QueryRecord)(DNSSDFlags flags, ULONG ifIndex, BSTR fullname, DNSSDRRType rrtype, DNSSDRRClass rrclass, IDNSSDEventManager *eventManager, IDNSSDService **service);
|
|
|
|
STDMETHOD(RegisterRecord)(DNSSDFlags flags, ULONG ifIndex, BSTR fullname, DNSSDRRType rrtype, DNSSDRRClass rrclass, VARIANT rdata, ULONG ttl, IDNSSDEventManager* eventManager, IDNSSDRecord** record);
|
|
|
|
STDMETHOD(AddRecord)(DNSSDFlags flags, DNSSDRRType rrtype, VARIANT rdata, ULONG ttl, IDNSSDRecord ** record);
|
|
|
|
STDMETHOD(ReconfirmRecord)(DNSSDFlags flags, ULONG ifIndex, BSTR fullname, DNSSDRRType rrtype, DNSSDRRClass rrclass, VARIANT rdata);
|
|
|
|
STDMETHOD(GetProperty)(BSTR prop, VARIANT * value);
|
|
|
|
STDMETHOD(GetAddrInfo)(DNSSDFlags flags, ULONG ifIndex, DNSSDAddressFamily addressFamily, BSTR hostname, IDNSSDEventManager *eventManager, IDNSSDService **service);
|
|
|
|
STDMETHOD(NATPortMappingCreate)(DNSSDFlags flags, ULONG ifIndex, DNSSDAddressFamily addressFamily, DNSSDProtocol protocol, USHORT internalPort, USHORT externalPort, ULONG ttl, IDNSSDEventManager *eventManager, IDNSSDService **service);
|
|
|
|
STDMETHOD(Stop)(void);
|
|
|
|
};
|
|
|
|
|
|
|
|
OBJECT_ENTRY_AUTO(__uuidof(DNSSDService), CDNSSDService)
|
|
|