--- /dev/null
+/*
+ * Copyright (c) 2003-2004 Apple Computer, Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * Copyright (c) 1999-2003 Apple Computer, Inc. All Rights Reserved.
+ *
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ *
+ * @APPLE_LICENSE_HEADER_END@
+
+ Change History (most recent first):
+
+$Log: ExplorerBarWindow.cpp,v $
+Revision 1.5 2004/04/15 01:00:05 bradley
+Removed support for automatically querying for A/AAAA records when resolving names. Platforms
+without .local name resolving support will need to manually query for A/AAAA records as needed.
+
+Revision 1.4 2004/04/09 21:03:15 bradley
+Changed port numbers to use network byte order for consistency with other platforms.
+
+Revision 1.3 2004/04/08 09:43:43 bradley
+Changed callback calling conventions to __stdcall so they can be used with C# delegates.
+
+Revision 1.2 2004/02/21 04:36:19 bradley
+Enable dot local name lookups now that the NSP is being installed.
+
+Revision 1.1 2004/01/30 03:01:56 bradley
+Explorer Plugin to browse for Rendezvous-enabled Web and FTP servers from within Internet Explorer.
+
+*/
+
+#include "StdAfx.h"
+
+#include "CommonServices.h"
+#include "DebugServices.h"
+#include "DNSSD.h"
+
+#include "ExplorerBar.h"
+#include "LoginDialog.h"
+#include "Resource.h"
+
+#include "ExplorerBarWindow.h"
+
+// MFC Debugging
+
+#ifdef _DEBUG
+#define new DEBUG_NEW
+#undef THIS_FILE
+static char THIS_FILE[] = __FILE__;
+#endif
+
+#if 0
+#pragma mark == Constants ==
+#endif
+
+//===========================================================================================================================
+// Constants
+//===========================================================================================================================
+
+// Control IDs
+
+#define IDC_EXPLORER_TREE 1234
+
+// Private Messages
+
+#define WM_PRIVATE_SERVICE_ADD ( WM_USER + 0x100 )
+#define WM_PRIVATE_SERVICE_REMOVE ( WM_USER + 0x101 )
+#define WM_PRIVATE_RESOLVE ( WM_USER + 0x102 )
+
+// TXT records
+
+#define kTXTRecordKeyPath "path="
+#define kTXTRecordKeyPathSize sizeof_string( kTXTRecordKeyPath )
+
+#if 0
+#pragma mark == Prototypes ==
+#endif
+
+//===========================================================================================================================
+// Prototypes
+//===========================================================================================================================
+
+DEBUG_LOCAL int FindServiceArrayIndex( const ServiceInfoArray &inArray, const ServiceInfo &inService, int &outIndex );
+DEBUG_LOCAL OSStatus UTF8StringToStringObject( const char *inUTF8, CString &inObject );
+
+#if 0
+#pragma mark == Message Map ==
+#endif
+
+//===========================================================================================================================
+// Message Map
+//===========================================================================================================================
+
+BEGIN_MESSAGE_MAP( ExplorerBarWindow, CWnd )
+ ON_WM_CREATE()
+ ON_WM_DESTROY()
+ ON_WM_SIZE()
+ ON_NOTIFY( NM_DBLCLK, IDC_EXPLORER_TREE, OnDoubleClick )
+ ON_MESSAGE( WM_PRIVATE_SERVICE_ADD, OnServiceAdd )
+ ON_MESSAGE( WM_PRIVATE_SERVICE_REMOVE, OnServiceRemove )
+ ON_MESSAGE( WM_PRIVATE_RESOLVE, OnResolve )
+END_MESSAGE_MAP()
+
+#if 0
+#pragma mark -
+#endif
+
+//===========================================================================================================================
+// ExplorerBarWindow
+//===========================================================================================================================
+
+ExplorerBarWindow::ExplorerBarWindow( void )
+{
+ mOwner = NULL;
+ mResolveServiceRef = NULL;
+}
+
+//===========================================================================================================================
+// ~ExplorerBarWindow
+//===========================================================================================================================
+
+ExplorerBarWindow::~ExplorerBarWindow( void )
+{
+ //
+}
+
+#if 0
+#pragma mark -
+#endif
+
+//===========================================================================================================================
+// OnCreate
+//===========================================================================================================================
+
+int ExplorerBarWindow::OnCreate( LPCREATESTRUCT inCreateStruct )
+{
+ AFX_MANAGE_STATE( AfxGetStaticModuleState() );
+
+ OSStatus err;
+ CRect rect;
+ CString s;
+
+ err = CWnd::OnCreate( inCreateStruct );
+ require_noerr( err, exit );
+
+ GetClientRect( rect );
+ mTree.Create( WS_TABSTOP | WS_VISIBLE | WS_CHILD | TVS_HASBUTTONS | TVS_LINESATROOT | TVS_HASLINES, rect, this,
+ IDC_EXPLORER_TREE );
+
+ err = DNSServiceInitialize( kDNSServiceInitializeFlagsNone, 0 );
+ if( err != kNoErr )
+ {
+ // Cannot talk to the mDNSResponder service. Show the error message and exit (with kNoErr so they can see it).
+
+ s.LoadString( IDS_RENDEZVOUS_NOT_AVAILABLE );
+ mTree.InsertItem( s, 0, 0, TVI_ROOT, TVI_LAST );
+ err = kNoErr;
+ goto exit;
+ }
+
+ ServiceHandlerEntry * e;
+
+ // Web Site Handler
+
+ e = new ServiceHandlerEntry;
+ check( e );
+ e->type = "_http._tcp";
+ e->urlScheme = "http://";
+ e->ref = NULL;
+ e->treeItem = NULL;
+ e->treeFirst = true;
+ e->obj = this;
+ e->needsLogin = false;
+ mServiceHandlers.Add( e );
+
+ s.LoadString( IDS_WEB_SITES );
+ e->treeItem = mTree.InsertItem( s, 0, 0 );
+ mTree.Expand( e->treeItem, TVE_EXPAND );
+
+ err = DNSServiceBrowse( &e->ref, 0, 0, e->type, NULL, BrowseCallBack, e );
+ require_noerr( err, exit );
+
+ // FTP Site Handler
+
+ e = new ServiceHandlerEntry;
+ check( e );
+ e->type = "_ftp._tcp";
+ e->urlScheme = "ftp://";
+ e->ref = NULL;
+ e->treeItem = NULL;
+ e->treeFirst = true;
+ e->obj = this;
+ e->needsLogin = true;
+ mServiceHandlers.Add( e );
+
+ s.LoadString( IDS_FTP_SITES );
+ e->treeItem = mTree.InsertItem( s, 0, 0 );
+ mTree.Expand( e->treeItem, TVE_EXPAND );
+
+ err = DNSServiceBrowse( &e->ref, 0, 0, e->type, NULL, BrowseCallBack, e );
+ require_noerr( err, exit );
+
+exit:
+ return( err );
+}
+
+//===========================================================================================================================
+// OnDestroy
+//===========================================================================================================================
+
+void ExplorerBarWindow::OnDestroy( void )
+{
+ // Stop any resolves that may still be pending (shouldn't be any).
+
+ StopResolve();
+
+ // Clean up the service handlers.
+
+ int i;
+ int n;
+
+ n = (int) mServiceHandlers.GetSize();
+ for( i = 0; i < n; ++i )
+ {
+ delete mServiceHandlers[ i ];
+ }
+
+ DNSServiceFinalize();
+ CWnd::OnDestroy();
+}
+
+//===========================================================================================================================
+// OnSize
+//===========================================================================================================================
+
+void ExplorerBarWindow::OnSize( UINT inType, int inX, int inY )
+{
+ CWnd::OnSize( inType, inX, inY );
+ mTree.MoveWindow( 0, 0, inX, inY );
+}
+
+//===========================================================================================================================
+// OnDoubleClick
+//===========================================================================================================================
+
+void ExplorerBarWindow::OnDoubleClick( NMHDR *inNMHDR, LRESULT *outResult )
+{
+ HTREEITEM item;
+ ServiceInfo * service;
+ OSStatus err;
+
+ DEBUG_UNUSED( inNMHDR );
+
+ item = mTree.GetSelectedItem();
+ require( item, exit );
+
+ service = reinterpret_cast < ServiceInfo * > ( mTree.GetItemData( item ) );
+ require_quiet( service, exit );
+
+ err = StartResolve( service );
+ require_noerr( err, exit );
+
+exit:
+ *outResult = 0;
+}
+
+#if 0
+#pragma mark -
+#endif
+
+//===========================================================================================================================
+// BrowseCallBack
+//===========================================================================================================================
+
+void CALLBACK_COMPAT
+ ExplorerBarWindow::BrowseCallBack(
+ DNSServiceRef inRef,
+ DNSServiceFlags inFlags,
+ uint32_t inInterfaceIndex,
+ DNSServiceErrorType inErrorCode,
+ const char * inName,
+ const char * inType,
+ const char * inDomain,
+ void * inContext )
+{
+ ServiceHandlerEntry * obj;
+ ServiceInfo * service;
+ OSStatus err;
+
+ DEBUG_UNUSED( inRef );
+
+ service = NULL;
+
+ require_noerr( inErrorCode, exit );
+ obj = reinterpret_cast < ServiceHandlerEntry * > ( inContext );
+ check( obj );
+ check( obj->obj );
+
+ try
+ {
+ // Post a message to the main thread so it can handle it since MFC is not thread safe.
+
+ service = new ServiceInfo;
+ require_action( service, exit, err = kNoMemoryErr );
+
+ err = UTF8StringToStringObject( inName, service->displayName );
+ check_noerr( err );
+
+ service->name = strdup( inName );
+ require_action( service->name, exit, err = kNoMemoryErr );
+
+ service->type = strdup( inType );
+ require_action( service->type, exit, err = kNoMemoryErr );
+
+ service->domain = strdup( inDomain );
+ require_action( service->domain, exit, err = kNoMemoryErr );
+
+ service->ifi = inInterfaceIndex;
+ service->handler = obj;
+
+ DWORD message;
+ BOOL ok;
+
+ message = ( inFlags & kDNSServiceFlagsAdd ) ? WM_PRIVATE_SERVICE_ADD : WM_PRIVATE_SERVICE_REMOVE;
+ ok = ::PostMessage( obj->obj->GetSafeHwnd(), message, 0, (LPARAM) service );
+ check( ok );
+ if( ok )
+ {
+ service = NULL;
+ }
+ }
+ catch( ... )
+ {
+ dlog( kDebugLevelError, "BrowseCallBack: exception thrown\n" );
+ }
+
+exit:
+ if( service )
+ {
+ delete service;
+ }
+}
+
+//===========================================================================================================================
+// OnServiceAdd
+//===========================================================================================================================
+
+LONG ExplorerBarWindow::OnServiceAdd( WPARAM inWParam, LPARAM inLParam )
+{
+ ServiceInfo * service;
+ ServiceHandlerEntry * handler;
+ int cmp;
+ int index;
+
+ DEBUG_UNUSED( inWParam );
+
+ service = reinterpret_cast < ServiceInfo * > ( inLParam );
+ check( service );
+ handler = service->handler;
+ check( handler );
+
+ cmp = FindServiceArrayIndex( handler->array, *service, index );
+ if( cmp == 0 )
+ {
+ // Found a match so update the item. The index is index + 1 so subtract 1.
+
+ index -= 1;
+ check( index < handler->array.GetSize() );
+
+ service->item = handler->array[ index ]->item;
+ delete handler->array[ index ];
+ handler->array[ index ] = service;
+ mTree.SetItemText( service->item, service->displayName );
+ mTree.SetItemData( service->item, (DWORD_PTR) service );
+ }
+ else
+ {
+ HTREEITEM afterItem;
+
+ // Insert the new item in sorted order.
+
+ afterItem = ( index > 0 ) ? handler->array[ index - 1 ]->item : TVI_FIRST;
+ handler->array.InsertAt( index, service );
+ service->item = mTree.InsertItem( service->displayName, handler->treeItem, afterItem );
+ mTree.SetItemData( service->item, (DWORD_PTR) service );
+
+ // Make sure the item is visible if this is the first time a service was added.
+
+ if( handler->treeFirst )
+ {
+ handler->treeFirst = false;
+ mTree.EnsureVisible( service->item );
+ }
+ }
+ return( 0 );
+}
+
+//===========================================================================================================================
+// OnServiceRemove
+//===========================================================================================================================
+
+LONG ExplorerBarWindow::OnServiceRemove( WPARAM inWParam, LPARAM inLParam )
+{
+ ServiceInfo * service;
+ ServiceHandlerEntry * handler;
+ int cmp;
+ int index;
+
+ DEBUG_UNUSED( inWParam );
+
+ service = reinterpret_cast < ServiceInfo * > ( inLParam );
+ check( service );
+ handler = service->handler;
+ check( handler );
+
+ // Search to see if we know about this service instance. If so, remove it from the list.
+
+ cmp = FindServiceArrayIndex( handler->array, *service, index );
+ check( cmp == 0 );
+ if( cmp == 0 )
+ {
+ // Found a match remove the item. The index is index + 1 so subtract 1.
+
+ index -= 1;
+ check( index < handler->array.GetSize() );
+
+ mTree.DeleteItem( handler->array[ index ]->item );
+ delete handler->array[ index ];
+ handler->array.RemoveAt( index );
+ }
+ delete service;
+ return( 0 );
+}
+
+#if 0
+#pragma mark -
+#endif
+
+//===========================================================================================================================
+// StartResolve
+//===========================================================================================================================
+
+OSStatus ExplorerBarWindow::StartResolve( ServiceInfo *inService )
+{
+ OSStatus err;
+
+ check( inService );
+
+ // Stop any current resolve that may be in progress.
+
+ StopResolve();
+
+ // Resolve the service.
+
+ err = DNSServiceResolve( &mResolveServiceRef, kDNSServiceFlagsNone, inService->ifi,
+ inService->name, inService->type, inService->domain, ResolveCallBack, inService->handler );
+ require_noerr( err, exit );
+
+exit:
+ return( err );
+}
+
+//===========================================================================================================================
+// StopResolve
+//===========================================================================================================================
+
+void ExplorerBarWindow::StopResolve( void )
+{
+ if( mResolveServiceRef )
+ {
+ DNSServiceRefDeallocate( mResolveServiceRef );
+ mResolveServiceRef = NULL;
+ }
+}
+
+//===========================================================================================================================
+// ResolveCallBack
+//===========================================================================================================================
+
+void CALLBACK_COMPAT
+ ExplorerBarWindow::ResolveCallBack(
+ DNSServiceRef inRef,
+ DNSServiceFlags inFlags,
+ uint32_t inInterfaceIndex,
+ DNSServiceErrorType inErrorCode,
+ const char * inFullName,
+ const char * inHostName,
+ uint16_t inPort,
+ uint16_t inTXTSize,
+ const char * inTXT,
+ void * inContext )
+{
+ ExplorerBarWindow * obj;
+ ServiceHandlerEntry * handler;
+ OSStatus err;
+
+ DEBUG_UNUSED( inRef );
+ DEBUG_UNUSED( inFlags );
+ DEBUG_UNUSED( inErrorCode );
+ DEBUG_UNUSED( inFullName );
+
+ require_noerr( inErrorCode, exit );
+ handler = (ServiceHandlerEntry *) inContext;
+ check( handler );
+ obj = handler->obj;
+ check( obj );
+
+ try
+ {
+ ResolveInfo * resolve;
+ BOOL ok;
+
+ dlog( kDebugLevelNotice, "resolved %s on ifi %d to %s\n", inFullName, inInterfaceIndex, inHostName );
+
+ // Stop resolving after the first good result.
+
+ obj->StopResolve();
+
+ // Post a message to the main thread so it can handle it since MFC is not thread safe.
+
+ resolve = new ResolveInfo;
+ require_action( resolve, exit, err = kNoMemoryErr );
+
+ UTF8StringToStringObject( inHostName, resolve->host );
+ resolve->port = ntohs( inPort );
+ resolve->ifi = inInterfaceIndex;
+ resolve->handler = handler;
+
+ err = resolve->txt.SetData( inTXT, inTXTSize );
+ check_noerr( err );
+
+ ok = ::PostMessage( obj->GetSafeHwnd(), WM_PRIVATE_RESOLVE, 0, (LPARAM) resolve );
+ check( ok );
+ if( !ok )
+ {
+ delete resolve;
+ }
+ }
+ catch( ... )
+ {
+ dlog( kDebugLevelError, "ResolveCallBack: exception thrown\n" );
+ }
+
+exit:
+ return;
+}
+
+//===========================================================================================================================
+// OnResolve
+//===========================================================================================================================
+
+LONG ExplorerBarWindow::OnResolve( WPARAM inWParam, LPARAM inLParam )
+{
+ ResolveInfo * resolve;
+ CString url;
+ uint8_t * path;
+ size_t pathSize;
+ char * pathPrefix;
+ CString username;
+ CString password;
+
+ DEBUG_UNUSED( inWParam );
+
+ resolve = reinterpret_cast < ResolveInfo * > ( inLParam );
+ check( resolve );
+
+ // Get login info if needed.
+
+ if( resolve->handler->needsLogin )
+ {
+ LoginDialog dialog;
+
+ if( !dialog.GetLogin( username, password ) )
+ {
+ goto exit;
+ }
+ }
+
+ // If the HTTP TXT record is a "path=" entry, use it as the resource path. Otherwise, use "/".
+
+ pathPrefix = "";
+ if( strcmp( resolve->handler->type, "_http._tcp" ) == 0 )
+ {
+ resolve->txt.GetData( &path, &pathSize );
+ if( pathSize > 0 )
+ {
+ pathSize = *path++;
+ }
+ if( ( pathSize > kTXTRecordKeyPathSize ) && ( memicmp( path, kTXTRecordKeyPath, kTXTRecordKeyPathSize ) == 0 ) )
+ {
+ path += kTXTRecordKeyPathSize;
+ pathSize -= kTXTRecordKeyPathSize;
+ }
+ else if( pathSize == 0 )
+ {
+ path = (uint8_t *) "/";
+ pathSize = 1;
+ }
+ if( *path != '/' )
+ {
+ pathPrefix = "/";
+ }
+ }
+ else
+ {
+ path = (uint8_t *) "";
+ pathSize = 1;
+ }
+
+ // Build the URL in the following format:
+ //
+ // <urlScheme>[<username>[:<password>]@]<name/ip>[<path>]
+
+ url.AppendFormat( TEXT( "%S" ), resolve->handler->urlScheme ); // URL Scheme
+ if( username.GetLength() > 0 )
+ {
+ url.AppendFormat( TEXT( "%s" ), username ); // Username
+ if( password.GetLength() > 0 )
+ {
+ url.AppendFormat( TEXT( ":%s" ), password ); // Password
+ }
+ url.AppendFormat( TEXT( "@" ) );
+ }
+
+ url += resolve->host; // Host
+ url.AppendFormat( TEXT( ":%d" ), resolve->port ); // :Port
+ url.AppendFormat( TEXT( "%S" ), pathPrefix ); // Path Prefix ("/" or empty).
+ url.AppendFormat( TEXT( "%.*S" ), (int) pathSize, (char *) path ); // Path (possibly empty).
+
+ // Tell Internet Explorer to go to the URL.
+
+ check( mOwner );
+ mOwner->GoToURL( url );
+
+exit:
+ delete resolve;
+ return( 0 );
+}
+
+#if 0
+#pragma mark -
+#endif
+
+//===========================================================================================================================
+// FindServiceArrayIndex
+//===========================================================================================================================
+
+DEBUG_LOCAL int FindServiceArrayIndex( const ServiceInfoArray &inArray, const ServiceInfo &inService, int &outIndex )
+{
+ int result;
+ int lo;
+ int hi;
+ int mid;
+
+ result = -1;
+ mid = 0;
+ lo = 0;
+ hi = (int)( inArray.GetSize() - 1 );
+ while( lo <= hi )
+ {
+ mid = ( lo + hi ) / 2;
+ result = inService.displayName.CompareNoCase( inArray[ mid ]->displayName );
+ if( result == 0 )
+ {
+ result = ( (int) inService.ifi ) - ( (int) inArray[ mid ]->ifi );
+ }
+ if( result == 0 )
+ {
+ break;
+ }
+ else if( result < 0 )
+ {
+ hi = mid - 1;
+ }
+ else
+ {
+ lo = mid + 1;
+ }
+ }
+ if( result == 0 )
+ {
+ mid += 1; // Bump index so new item is inserted after matching item.
+ }
+ else if( result > 0 )
+ {
+ mid += 1;
+ }
+ outIndex = mid;
+ return( result );
+}
+
+//===========================================================================================================================
+// UTF8StringToStringObject
+//===========================================================================================================================
+
+DEBUG_LOCAL OSStatus UTF8StringToStringObject( const char *inUTF8, CString &inObject )
+{
+ OSStatus err;
+ int n;
+ BSTR unicode;
+
+ unicode = NULL;
+
+ n = MultiByteToWideChar( CP_UTF8, 0, inUTF8, -1, NULL, 0 );
+ if( n > 0 )
+ {
+ unicode = (BSTR) malloc( (size_t)( n * sizeof( wchar_t ) ) );
+ if( !unicode )
+ {
+ err = ERROR_INSUFFICIENT_BUFFER;
+ goto exit;
+ }
+
+ n = MultiByteToWideChar( CP_UTF8, 0, inUTF8, -1, unicode, n );
+ try
+ {
+ inObject = unicode;
+ }
+ catch( ... )
+ {
+ err = ERROR_NO_UNICODE_TRANSLATION;
+ goto exit;
+ }
+ }
+ else
+ {
+ inObject = "";
+ }
+ err = ERROR_SUCCESS;
+
+exit:
+ if( unicode )
+ {
+ free( unicode );
+ }
+ return( err );
+}
projects / apple / mdnsresponder.git / blobdiff