| /* -*- 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) |
| |