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