Clients/ExplorerPlugin/ExplorerBarWindow.h

   1 /* -*- Mode: C; tab-width: 4 -*-
   2  *
   3  * Copyright (c) 2003-2004 Apple Computer, Inc. All rights reserved.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 #ifndef __EXPLORER_BAR_WINDOW__
  19 #define __EXPLORER_BAR_WINDOW__
  20
  21 #pragma once
  22
  23 #include    "afxtempl.h"
  24
  25 #include    "dns_sd.h"
  26 #include    <list>
  27
  28 //===========================================================================================================================
  29 //      Structures
  30 //===========================================================================================================================
  31
  32 // Forward Declarations
  33
  34 struct  ServiceHandlerEntry;
  35 class ExplorerBarWindow;
  36
  37 // ServiceInfo
  38
  39 struct  ServiceInfo
  40 {
  41     CString displayName;
  42     char *                      name;
  43     char *                      type;
  44     char *                      domain;
  45     uint32_t ifi;
  46     HTREEITEM item;
  47     ServiceHandlerEntry *       handler;
  48     DWORD refs;
  49
  50     ServiceInfo( void )
  51     {
  52         item    = NULL;
  53         type    = NULL;
  54         domain  = NULL;
  55         handler = NULL;
  56     }
  57
  58     ~ServiceInfo( void )
  59     {
  60         if( name )
  61         {
  62             free( name );
  63         }
  64         if( type )
  65         {
  66             free( type );
  67         }
  68         if( domain )
  69         {
  70             free( domain );
  71         }
  72     }
  73 };
  74
  75 typedef CArray < ServiceInfo *, ServiceInfo * >     ServiceInfoArray;
  76
  77 // TextRecord
  78
  79 struct  TextRecord
  80 {
  81     uint8_t *       mData;
  82     uint16_t mSize;
  83
  84     TextRecord( void )
  85     {
  86         mData = NULL;
  87         mSize = 0;
  88     }
  89
  90     ~TextRecord( void )
  91     {
  92         if( mData )
  93         {
  94             free( mData );
  95         }
  96     }
  97
  98     void    GetData( void *outData, uint16_t *outSize )
  99     {
 100         if( outData )
 101         {
 102             *( (void **) outData ) = mData;
 103         }
 104         if( outSize )
 105         {
 106             *outSize = mSize;
 107         }
 108     }
 109
 110     OSStatus    SetData( const void *inData, uint16_t inSize )
 111     {
 112         OSStatus err;
 113         uint8_t *       newData;
 114
 115         newData = (uint8_t *) malloc( inSize );
 116         require_action( newData, exit, err = kNoMemoryErr );
 117         memcpy( newData, inData, inSize );
 118
 119         if( mData )
 120         {
 121             free( mData );
 122         }
 123         mData = newData;
 124         mSize = inSize;
 125         err  = kNoErr;
 126
 127 exit:
 128         return( err );
 129     }
 130 };
 131
 132 // ResolveInfo
 133
 134 struct  ResolveInfo
 135 {
 136     CString host;
 137     uint16_t port;
 138     uint32_t ifi;
 139     TextRecord txt;
 140     ServiceHandlerEntry *       handler;
 141 };
 142
 143 // ServiceHandlerEntry
 144
 145 struct  ServiceHandlerEntry
 146 {
 147     const char *            type;
 148     const char *            urlScheme;
 149     DNSServiceRef ref;
 150     ServiceInfoArray array;
 151     ExplorerBarWindow *     obj;
 152     bool needsLogin;
 153
 154     ServiceHandlerEntry( void )
 155     {
 156         type        = NULL;
 157         urlScheme   = NULL;
 158         ref         = NULL;
 159         obj         = NULL;
 160         needsLogin  = false;
 161     }
 162
 163     ~ServiceHandlerEntry( void )
 164     {
 165         int i;
 166         int n;
 167
 168         n = (int) array.GetSize();
 169         for( i = 0; i < n; ++i )
 170         {
 171             delete array[ i ];
 172         }
 173     }
 174 };
 175
 176 typedef CArray < ServiceHandlerEntry *, ServiceHandlerEntry * >     ServiceHandlerArray;
 177
 178 //===========================================================================================================================
 179 //      ExplorerBarWindow
 180 //===========================================================================================================================
 181
 182 class ExplorerBar;      // Forward Declaration
 183
 184 class ExplorerBarWindow : public CWnd
 185 {
 186 protected:
 187
 188 ExplorerBar *           mOwner;
 189 CTreeCtrl mTree;
 190
 191 ServiceHandlerArray mServiceHandlers;
 192 DNSServiceRef mResolveServiceRef;
 193
 194 public:
 195
 196 ExplorerBarWindow( void );
 197 virtual ~ExplorerBarWindow( void );
 198
 199 protected:
 200
 201 // General
 202
 203 afx_msg int     OnCreate( LPCREATESTRUCT inCreateStruct );
 204 afx_msg void    OnDestroy( void );
 205 afx_msg void    OnSize( UINT inType, int inX, int inY );
 206 afx_msg void    OnDoubleClick( NMHDR *inNMHDR, LRESULT *outResult );
 207 afx_msg LRESULT OnServiceEvent( WPARAM inWParam, LPARAM inLParam );
 208
 209 // Browsing
 210
 211 static void DNSSD_API
 212 BrowseCallBack(
 213     DNSServiceRef inRef,
 214     DNSServiceFlags inFlags,
 215     uint32_t inInterfaceIndex,
 216     DNSServiceErrorType inErrorCode,
 217     const char *            inName,
 218     const char *            inType,
 219     const char *            inDomain,
 220     void *                  inContext );
 221 LONG OnServiceAdd( ServiceInfo * service );
 222 LONG OnServiceRemove( ServiceInfo * service );
 223
 224 // Resolving
 225
 226 OSStatus    StartResolve( ServiceInfo *inService );
 227 void        StopResolve( void );
 228
 229
 230 void        Stop( DNSServiceRef ref );
 231
 232
 233 static void DNSSD_API
 234 ResolveCallBack(
 235     DNSServiceRef inRef,
 236     DNSServiceFlags inFlags,
 237     uint32_t inInterfaceIndex,
 238     DNSServiceErrorType inErrorCode,
 239     const char *            inFullName,
 240     const char *            inHostName,
 241     uint16_t inPort,
 242     uint16_t inTXTSize,
 243     const char *            inTXT,
 244     void *                  inContext );
 245 LONG OnResolve( ResolveInfo * resolve );
 246
 247 // Accessors
 248
 249 public:
 250
 251 ExplorerBar *   GetOwner( void ) const { return( mOwner ); }
 252 void            SetOwner( ExplorerBar *inOwner )    { mOwner = inOwner; }
 253
 254 DECLARE_MESSAGE_MAP()
 255 private:
 256
 257 typedef std::list< DNSServiceRef >  ServiceRefList;
 258
 259 HTREEITEM m_about;
 260 ServiceRefList m_serviceRefs;
 261 CImageList m_imageList;
 262 };
 263
 264 #endif  // __EXPLORER_BAR_WINDOW__