* 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.
-
- Change History (most recent first):
-
-$Log: ConfigPropertySheet.cpp,v $
-Revision 1.5 2006/08/14 23:25:28 cheshire
-Re-licensed mDNSResponder daemon source code under Apache License, Version 2.0
-
-Revision 1.4 2005/10/05 20:46:50 herscher
-<rdar://problem/4192011> Move Wide-Area preferences to another part of the registry so they don't removed during an update-install.
-
-Revision 1.3 2005/03/03 19:55:22 shersche
-<rdar://problem/4034481> ControlPanel source code isn't saving CVS log info
-
-
-*/
+ */
#include "ConfigPropertySheet.h"
#include <WinServices.h>
+extern "C"
+{
+#include <ClientCommon.h>
+}
#include <process.h>
// Custom events
#define WM_DATAREADY ( WM_USER + 0x100 )
-#define WM_REGISTRYCHANGED ( WM_USER + 0x101 )
IMPLEMENT_DYNCREATE(CConfigPropertySheet, CPropertySheet)
:
CPropertySheet(),
m_browseDomainsRef( NULL ),
- m_regDomainsRef( NULL ),
m_thread( NULL ),
m_threadExited( NULL )
{
- AddPage(&m_firstPage);
+ AddPage(&m_firstPage );
AddPage(&m_secondPage);
AddPage(&m_thirdPage);
//{{AFX_MSG_MAP(CConfigPropertySheet)
//}}AFX_MSG_MAP
ON_MESSAGE( WM_DATAREADY, OnDataReady )
- ON_MESSAGE( WM_REGISTRYCHANGED, OnRegistryChanged )
END_MESSAGE_MAP()
err = SetupBrowsing();
require_noerr( err, exit );
- err = SetupRegistryNotifications();
- require_noerr( err, exit );
-
exit:
return b;
// CConfigPropertySheet::OnDataReady
//---------------------------------------------------------------------------------------------------------------------------
-LONG
+LRESULT
CConfigPropertySheet::OnDataReady(WPARAM inWParam, LPARAM inLParam)
{
if (WSAGETSELECTERROR(inLParam) && !(HIWORD(inLParam)))
{
DNSServiceProcessResult( m_browseDomainsRef );
}
- else if ( m_regDomainsRef && DNSServiceRefSockFD( m_regDomainsRef ) == (int) sock )
- {
- DNSServiceProcessResult( m_regDomainsRef );
- }
- }
-
- return 0;
-}
-
-
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::OnRegistryChanged
-//---------------------------------------------------------------------------------------------------------------------------
-
-afx_msg LONG
-CConfigPropertySheet::OnRegistryChanged( WPARAM inWParam, LPARAM inLParam )
-{
- DEBUG_UNUSED( inWParam );
- DEBUG_UNUSED( inLParam );
-
- if ( GetActivePage() == &m_firstPage )
- {
- m_firstPage.OnRegistryChanged();
}
return 0;
{
OSStatus err;
- err = TearDownRegistryNotifications();
- check_noerr( err );
-
err = TearDownBrowsing();
check_noerr( err );
}
err = WSAAsyncSelect( DNSServiceRefSockFD( m_browseDomainsRef ), m_hWnd, WM_DATAREADY, FD_READ|FD_CLOSE );
require_noerr( err, exit );
- // Start browsing for registration domains
-
- err = DNSServiceEnumerateDomains( &m_regDomainsRef, kDNSServiceFlagsRegistrationDomains, 0, RegDomainsReply, this );
- require_noerr( err, exit );
-
- err = WSAAsyncSelect( DNSServiceRefSockFD( m_regDomainsRef ), m_hWnd, WM_DATAREADY, FD_READ|FD_CLOSE );
- require_noerr( err, exit );
-
exit:
if ( err )
m_browseDomainsRef = NULL;
}
- if ( m_regDomainsRef )
- {
- err = WSAAsyncSelect( DNSServiceRefSockFD( m_regDomainsRef ), m_hWnd, 0, 0 );
- check_noerr( err );
-
- DNSServiceRefDeallocate( m_regDomainsRef );
-
- m_regDomainsRef = NULL;
- }
-
- return err;
-}
-
-
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::SetupRegistryNotifications
-//---------------------------------------------------------------------------------------------------------------------------
-
-OSStatus
-CConfigPropertySheet::SetupRegistryNotifications()
-{
- unsigned int threadId;
- OSStatus err;
-
- check( m_threadExited == NULL );
- check( m_thread == NULL );
-
- err = RegCreateKey( HKEY_LOCAL_MACHINE, kServiceParametersNode L"\\DynDNS\\State\\Hostnames", &m_statusKey );
- require_noerr( err, exit );
-
- m_threadExited = CreateEvent( NULL, FALSE, FALSE, NULL );
- err = translate_errno( m_threadExited, (OSStatus) GetLastError(), kUnknownErr );
- require_noerr( err, exit );
-
- // Create thread with _beginthreadex() instead of CreateThread() to avoid memory leaks when using static run-time
- // libraries. See <http://msdn.microsoft.com/library/default.asp?url=/library/en-us/dllproc/base/createthread.asp>.
-
- m_thread = (HANDLE) _beginthreadex_compat( NULL, 0, WatchRegistry, this, 0, &threadId );
- err = translate_errno( m_thread, (OSStatus) GetLastError(), kUnknownErr );
- require_noerr( err, exit );
-
-exit:
-
- if ( err )
- {
- TearDownRegistryNotifications();
- }
-
- return err;
-}
-
-
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::TearDownRegistryNotifications
-//---------------------------------------------------------------------------------------------------------------------------
-
-OSStatus
-CConfigPropertySheet::TearDownRegistryNotifications()
-{
- OSStatus err = kNoErr;
-
- if ( m_statusKey )
- {
- EnterCriticalSection( &m_lock );
-
- RegCloseKey( m_statusKey );
- m_statusKey = NULL;
-
- LeaveCriticalSection( &m_lock );
- }
-
- if ( m_threadExited )
- {
- err = WaitForSingleObject( m_threadExited, 5 * 1000 );
- require_noerr( err, exit );
- }
-
-exit:
-
- if ( m_threadExited )
- {
- CloseHandle( m_threadExited );
- m_threadExited = NULL;
- }
-
- if ( m_thread )
- {
- CloseHandle( m_thread );
- m_thread = NULL;
- }
-
return err;
}
}
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::GetNextLabel
-//---------------------------------------------------------------------------------------------------------------------------
-
-const char*
-CConfigPropertySheet::GetNextLabel( const char * cstr, char label[64] )
-{
- char *ptr = label;
- while (*cstr && *cstr != '.') // While we have characters in the label...
- {
- char c = *cstr++;
- if (c == '\\')
- {
- c = *cstr++;
- if (isdigit(cstr[-1]) && isdigit(cstr[0]) && isdigit(cstr[1]))
- {
- int v0 = cstr[-1] - '0'; // then interpret as three-digit decimal
- int v1 = cstr[ 0] - '0';
- int v2 = cstr[ 1] - '0';
- int val = v0 * 100 + v1 * 10 + v2;
- if (val <= 255) { c = (char)val; cstr += 2; } // If valid three-digit decimal value, use it
- }
- }
- *ptr++ = c;
- if (ptr >= label+64) return(NULL);
- }
- if (*cstr) cstr++; // Skip over the trailing dot (if present)
- *ptr++ = 0;
- return(cstr);
-}
-
-
//---------------------------------------------------------------------------------------------------------------------------
// CConfigPropertySheet::BrowseDomainsReply
//---------------------------------------------------------------------------------------------------------------------------
goto exit;
}
-
-
err = self->DecodeDomainName( replyDomain, decoded );
require_noerr( err, exit );
return;
}
-
-
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::RegDomainsReply
-//---------------------------------------------------------------------------------------------------------------------------
-
-void DNSSD_API
-CConfigPropertySheet::RegDomainsReply
- (
- DNSServiceRef sdRef,
- DNSServiceFlags flags,
- uint32_t interfaceIndex,
- DNSServiceErrorType errorCode,
- const char * replyDomain,
- void * context
- )
-{
- CConfigPropertySheet * self = reinterpret_cast<CConfigPropertySheet*>(context);
- CString decoded;
- OSStatus err;
-
- DEBUG_UNUSED( sdRef );
- DEBUG_UNUSED( interfaceIndex );
-
- if ( errorCode )
- {
- goto exit;
- }
-
- check( replyDomain );
-
- // Ignore local domains
-
- if ( strcmp( replyDomain, "local." ) == 0 )
- {
- goto exit;
- }
-
- err = self->DecodeDomainName( replyDomain, decoded );
- require_noerr( err, exit );
-
- // Remove trailing '.'
-
- decoded.TrimRight( '.' );
-
- if ( flags & kDNSServiceFlagsAdd )
- {
- if ( self->GetActivePage() == &self->m_secondPage )
- {
- self->m_secondPage.OnAddRegistrationDomain( decoded );
- }
-
- self->m_regDomains.push_back( decoded );
- }
- else
- {
- if ( self->GetActivePage() == &self->m_secondPage )
- {
- self->m_secondPage.OnRemoveRegistrationDomain( decoded );
- }
-
- self->m_regDomains.remove( decoded );
- }
-
-exit:
-
- return;
-}
-
-
-//---------------------------------------------------------------------------------------------------------------------------
-// CConfigPropertySheet::WatchRegistry
-//---------------------------------------------------------------------------------------------------------------------------
-
-unsigned WINAPI
-CConfigPropertySheet::WatchRegistry ( LPVOID inParam )
-{
- bool done = false;
-
- CConfigPropertySheet * self = reinterpret_cast<CConfigPropertySheet*>(inParam);
- check( self );
-
- while ( !done )
- {
- RegNotifyChangeKeyValue( self->m_statusKey, TRUE, REG_NOTIFY_CHANGE_LAST_SET, NULL, FALSE );
-
- EnterCriticalSection( &self->m_lock );
-
- done = ( self->m_statusKey == NULL ) ? true : false;
-
- if ( !done )
- {
- self->PostMessage( WM_REGISTRYCHANGED, 0, 0 );
- }
-
- LeaveCriticalSection( &self->m_lock );
- }
-
- SetEvent( self->m_threadExited );
-
- return 0;
-}
projects / apple / mdnsresponder.git / blobdiff