420 lines
12 KiB
C++
420 lines
12 KiB
C++
|
// DVDUpgrd.cpp : Defines the entry point for the application.
|
||
|
|
//
|
||
|
|
#define ASSERTERROR(x) ASSERT( ! x )
|
||
|
|
|
||
|
|
#include <windows.h>
|
||
|
|
#include <streams.h>
|
||
|
|
|
||
|
|
#include "..\dvddetect\DVDDetect.h"
|
||
|
|
|
||
|
|
// for IShellLink
|
||
|
|
#include <wininet.h> // must include before shlobj to get IActiveDesktop
|
||
|
|
#include <shlobj.h>
|
||
|
|
|
||
|
|
#include <Userenv.h>
|
||
|
|
#include <intshcut.h>
|
||
|
|
#include <shellapi.h>
|
||
|
|
|
||
|
|
#define DBGBOX(x)
|
||
|
|
static TCHAR* Append( TCHAR* pOut, const TCHAR* pString )
|
||
|
|
{
|
||
|
|
lstrcat( pOut, pString );
|
||
|
|
return pOut + lstrlen(pOut);
|
||
|
|
}
|
||
|
|
|
||
|
|
static TCHAR* Print(TCHAR* pOut, const DVDResult& result)
|
||
|
|
{
|
||
|
|
UINT64 ullVersion = result.GetVersion();
|
||
|
|
wsprintf( pOut,
|
||
|
|
TEXT("Name:\"%s\"\n Company Name:\"%s\"\n Version: %d.%d.%d.%d \n CRC32: 0x%08x\n"),
|
||
|
|
result.GetName(),
|
||
|
|
result.GetCompanyName(),
|
||
|
|
WORD(ullVersion>>48),
|
||
|
|
WORD(ullVersion>>32),
|
||
|
|
WORD(ullVersion>>16),
|
||
|
|
WORD(ullVersion>>0),
|
||
|
|
result.GetCRC()
|
||
|
|
);
|
||
|
|
return pOut + lstrlen(pOut );
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Generic utility functions
|
||
|
|
//
|
||
|
|
|
||
|
|
static bool AreEquivalent( const char* pStr1, const char* pStr2 )
|
||
|
|
{
|
||
|
|
return lstrcmpiA( pStr1, pStr2 ) ==0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool AreEquivalent( const WCHAR* pStr1, const WCHAR* pStr2 )
|
||
|
|
{
|
||
|
|
return lstrcmpiW( pStr1, pStr2 ) ==0;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
static bool IsUserMemberOf( DWORD domainAlias )
|
||
|
|
{
|
||
|
|
HANDLE Token;
|
||
|
|
DWORD BytesRequired;
|
||
|
|
PTOKEN_GROUPS Groups;
|
||
|
|
bool b;
|
||
|
|
DWORD i;
|
||
|
|
SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;
|
||
|
|
PSID AdministratorsGroup;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Open the process token.
|
||
|
|
//
|
||
|
|
if(!OpenProcessToken(GetCurrentProcess(),TOKEN_QUERY,&Token)) {
|
||
|
|
return(false);
|
||
|
|
}
|
||
|
|
|
||
|
|
b = false;
|
||
|
|
Groups = NULL;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get group information.
|
||
|
|
//
|
||
|
|
if(!GetTokenInformation(Token,TokenGroups,NULL,0,&BytesRequired)
|
||
|
|
&& (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
|
||
|
|
&& (Groups = reinterpret_cast<PTOKEN_GROUPS>(new BYTE[BytesRequired]) )
|
||
|
|
&& GetTokenInformation(Token,TokenGroups,Groups,BytesRequired,&BytesRequired)) {
|
||
|
|
|
||
|
|
if( AllocateAndInitializeSid(
|
||
|
|
&NtAuthority,
|
||
|
|
2,
|
||
|
|
SECURITY_BUILTIN_DOMAIN_RID,
|
||
|
|
domainAlias,
|
||
|
|
0, 0, 0, 0, 0, 0,
|
||
|
|
&AdministratorsGroup
|
||
|
|
))
|
||
|
|
|
||
|
|
{
|
||
|
|
|
||
|
|
//
|
||
|
|
// See if the user has the administrator group.
|
||
|
|
//
|
||
|
|
for(i=0; i<Groups->GroupCount; i++) {
|
||
|
|
if(EqualSid(Groups->Groups[i].Sid,AdministratorsGroup)) {
|
||
|
|
b = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
FreeSid(AdministratorsGroup);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Clean up and return.
|
||
|
|
//
|
||
|
|
|
||
|
|
if(Groups) {
|
||
|
|
delete [] (BYTE *)Groups;
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseHandle(Token);
|
||
|
|
|
||
|
|
return(b);
|
||
|
|
}
|
||
|
|
|
||
|
|
// copied from ntfs\fu\utility
|
||
|
|
static bool IsUserAdmin()
|
||
|
|
{
|
||
|
|
return IsUserMemberOf( DOMAIN_ALIAS_RID_ADMINS);
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool IsUserPowerUser()
|
||
|
|
{
|
||
|
|
return IsUserMemberOf( DOMAIN_ALIAS_RID_POWER_USERS );
|
||
|
|
}
|
||
|
|
|
||
|
|
static HRESULT CreateShortcutToURL(LPCTSTR pszURL, LPCTSTR pszLinkFile)
|
||
|
|
{
|
||
|
|
CoInitialize(NULL);
|
||
|
|
HRESULT hRes;
|
||
|
|
IUniformResourceLocator *pURL = NULL;
|
||
|
|
|
||
|
|
// Create an IUniformResourceLocator object
|
||
|
|
hRes = CoCreateInstance (CLSID_InternetShortcut, NULL,
|
||
|
|
CLSCTX_INPROC_SERVER, IID_IUniformResourceLocator, (LPVOID*) &pURL);
|
||
|
|
if (SUCCEEDED(hRes)) {
|
||
|
|
IPersistFile *pPF = NULL;
|
||
|
|
|
||
|
|
hRes = pURL->SetURL(pszURL, IURL_SETURL_FL_USE_DEFAULT_PROTOCOL );
|
||
|
|
|
||
|
|
if (SUCCEEDED(hRes)) {
|
||
|
|
hRes = pURL->QueryInterface (IID_IPersistFile, (void **)&pPF);
|
||
|
|
if (SUCCEEDED(hRes)) {
|
||
|
|
#ifndef UNICODE
|
||
|
|
WCHAR wsz [MAX_PATH]; // buffer for Unicode string
|
||
|
|
|
||
|
|
// Ensure that the string consists of ANSI characters.
|
||
|
|
MultiByteToWideChar (CP_ACP, 0, pszLinkFile, -1, wsz, MAX_PATH);
|
||
|
|
|
||
|
|
// Save the shortcut via the IPersistFile::Save member function.
|
||
|
|
hRes = pPF->Save (wsz, TRUE);
|
||
|
|
#else
|
||
|
|
hRes = pPF->Save (pszLinkFile, TRUE);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
// Release the pointer to IPersistFile.
|
||
|
|
pPF->Release ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Release the pointer to IUniformResourceLocator
|
||
|
|
pURL->Release ();
|
||
|
|
}
|
||
|
|
CoUninitialize();
|
||
|
|
return hRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool IsAlphaNum( TCHAR t )
|
||
|
|
{
|
||
|
|
return ( TEXT('a') <= t && t<= TEXT('z')) ||
|
||
|
|
( TEXT('A') <= t && t<= TEXT('Z')) ||
|
||
|
|
( TEXT('0') <= t && t<= TEXT('9'));
|
||
|
|
}
|
||
|
|
|
||
|
|
void AlphaNumStrCpy( TCHAR* pOut, const TCHAR* pIn )
|
||
|
|
{
|
||
|
|
while( *pIn ) {
|
||
|
|
if( IsAlphaNum(*pIn)) {
|
||
|
|
*pOut++ = *pIn;
|
||
|
|
} else {
|
||
|
|
*pOut++ = TEXT('_');
|
||
|
|
}
|
||
|
|
pIn++;
|
||
|
|
}
|
||
|
|
*pOut=TEXT('\0');
|
||
|
|
}
|
||
|
|
|
||
|
|
// Company: MGI Software Corp.
|
||
|
|
// URL: http://www.dvdmax.com/updates/
|
||
|
|
//
|
||
|
|
// Company: Ravisent Technologies
|
||
|
|
// URL: http://www.ravisentdirect.com/upgrade/dvdup.html
|
||
|
|
//
|
||
|
|
// Company: CyberLink Corp.
|
||
|
|
// URL: http://www.cyberlink.com.tw/english/download/download.asp#upgrade
|
||
|
|
//
|
||
|
|
// Company: National Semiconductor Corporation
|
||
|
|
// URL: http://www.national.com/appinfo/dvd/support/msft/
|
||
|
|
//
|
||
|
|
// Company: InterVideo, Inc.
|
||
|
|
// URL: http:/www.intervideo.com/products/custom/ms/whistler/upgrade.html
|
||
|
|
//
|
||
|
|
|
||
|
|
static void CreateRedirectorString( TCHAR* pWebsite, const DVDResult& result )
|
||
|
|
{
|
||
|
|
TCHAR filteredName[2000];
|
||
|
|
AlphaNumStrCpy( filteredName, result.GetCompanyName() );
|
||
|
|
|
||
|
|
// Generate extended information for our redirector site
|
||
|
|
DecoderVendor vendor = result.GetVendor();
|
||
|
|
TCHAR pQuery[1000];
|
||
|
|
wsprintf( pQuery, TEXT("vendorid=%d&version=%08x%08x&crc=%08x&company=%s"),
|
||
|
|
vendor,
|
||
|
|
DWORD(result.GetVersion()>>32), DWORD(result.GetVersion() ),
|
||
|
|
result.GetCRC(),
|
||
|
|
filteredName);
|
||
|
|
|
||
|
|
switch( vendor ) {
|
||
|
|
case vendor_MediaMatics:
|
||
|
|
wsprintf( pWebsite, TEXT("www.national.com/appinfo/dvd/support/msft"));
|
||
|
|
break;
|
||
|
|
case vendor_Intervideo:
|
||
|
|
wsprintf( pWebsite, TEXT("www.intervideo.com/products/custom/ms/whistler/upgrade.html") );
|
||
|
|
break;
|
||
|
|
case vendor_CyberLink:
|
||
|
|
wsprintf( pWebsite, TEXT("www.cyberlink.com.tw/english/download/download.asp#upgrade") );
|
||
|
|
break;
|
||
|
|
case vendor_Ravisent:
|
||
|
|
wsprintf( pWebsite, TEXT("www.ravisentdirect.com/upgrade/dvdup.htm") );
|
||
|
|
break;
|
||
|
|
case vendor_MGI:
|
||
|
|
wsprintf( pWebsite, TEXT("www.dvdmax.com/updates") );
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
wsprintf( pWebsite, TEXT("windowsupdate.microsoft.com/dvdupgrade.asp?%s"), pQuery );
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool LaunchURL( const TCHAR* pURL )
|
||
|
|
{
|
||
|
|
ULONG uRC = (ULONG)(ULONG_PTR) ShellExecute(NULL, TEXT("open"), pURL, NULL, NULL, SW_SHOWNORMAL);
|
||
|
|
|
||
|
|
if ( uRC <= 32 ) {
|
||
|
|
return false;
|
||
|
|
} else {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool CreateDesktopShortcut( const DVDResult& result )
|
||
|
|
{
|
||
|
|
TCHAR website[1000];
|
||
|
|
CreateRedirectorString( website, result );
|
||
|
|
|
||
|
|
TCHAR path[MAX_PATH];
|
||
|
|
DWORD dwLen=sizeof(path);
|
||
|
|
if( GetAllUsersProfileDirectory( path, &dwLen )) {
|
||
|
|
lstrcat( path, TEXT("\\desktop\\Upgrade DVD Decoder.url"));
|
||
|
|
TCHAR url[1000];
|
||
|
|
wsprintf( url, TEXT("http://%s"), website );
|
||
|
|
HRESULT hres = CreateShortcutToURL( url, path );
|
||
|
|
return SUCCEEDED(hres);
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void LaunchUI( const DVDResult& result )
|
||
|
|
{
|
||
|
|
TCHAR website[1000];
|
||
|
|
CreateRedirectorString( website, result );
|
||
|
|
#if 0
|
||
|
|
switch( MessageBoxW( NULL,
|
||
|
|
L"Windows has detected an incompatible software DVD decoder.\n\n"
|
||
|
|
L"Would you like to upgrade your decoder at this time ?\n\n"
|
||
|
|
L"Select YES to download an upgrade\n"
|
||
|
|
L"Select NO to have Windows create a desktop shortcut so that you can upgrade later",
|
||
|
|
TEXT("DVD Decoder upgrade"),
|
||
|
|
MB_YESNO ))
|
||
|
|
{
|
||
|
|
case IDYES:
|
||
|
|
if(!LaunchURL( website ) ) {
|
||
|
|
MessageBox( NULL, TEXT("Could not reach internet site. A shortcut has been created for you to try again later"),
|
||
|
|
NULL, MB_OK );
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case IDNO:
|
||
|
|
CreateDesktopShortcut( result );
|
||
|
|
DVDDetectSetupRun::Remove();
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
TCHAR url[1000];
|
||
|
|
wsprintf( url, TEXT("HCP://system/DVDUpgrd/dvdupgrd.htm?website=%s"), website);
|
||
|
|
LaunchURL( url );
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
static void DetectForUpgrade( bool fForceUpgrade )
|
||
|
|
{
|
||
|
|
DVDDetectBuffer buffer;
|
||
|
|
const DVDResult* pResult = buffer.Detect();
|
||
|
|
if( pResult ) {
|
||
|
|
const DVDResult& result = *pResult;
|
||
|
|
// map company name groups
|
||
|
|
bool fShouldUpgrade = (fForceUpgrade || result.ShouldUpgrade() );
|
||
|
|
if( fShouldUpgrade && (IsUserAdmin() || IsUserPowerUser()) )
|
||
|
|
{
|
||
|
|
LaunchUI( result );
|
||
|
|
|
||
|
|
}
|
||
|
|
if(!fShouldUpgrade ) {
|
||
|
|
// if they don't need to upgrade (sounds like we're running after an upgrade)
|
||
|
|
//
|
||
|
|
DVDDetectSetupRun::Remove();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void DetectForSetup( bool fWillBeUpgrade )
|
||
|
|
{
|
||
|
|
DVDDetectBuffer buffer;
|
||
|
|
const DVDResult* pResult = buffer.Detect();
|
||
|
|
if( pResult ) {
|
||
|
|
const DVDResult& result = *pResult;
|
||
|
|
bool fResult = result.ShouldUpgrade( fWillBeUpgrade );
|
||
|
|
|
||
|
|
if( fResult ) {
|
||
|
|
DBGBOX(MessageBox( NULL, TEXT("DetectDVDForSetup = true, add to setup run"), NULL, MB_OK ));
|
||
|
|
DVDDetectSetupRun::Add();
|
||
|
|
} else {
|
||
|
|
DBGBOX(MessageBox( NULL, TEXT("DetectDVDForSetup = false, not added to setup run"), NULL, MB_OK ));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
DBGBOX(MessageBox( NULL, TEXT("DetectDVDForSetup = no decoder detected"), NULL, MB_OK ));
|
||
|
|
//__asm int 3
|
||
|
|
const DVDResult* pResult2 = buffer.Detect();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void DetectForRemove()
|
||
|
|
{
|
||
|
|
DVDDetectBuffer buffer;
|
||
|
|
const DVDResult* pResult = buffer.Detect();
|
||
|
|
if( pResult ) {
|
||
|
|
const DVDResult& result = *pResult;
|
||
|
|
// Allows create it since we don't know the 9x info anymore
|
||
|
|
// and this only comes from the PCHealth UI's method
|
||
|
|
// if( result.ShouldUpgrade())
|
||
|
|
CreateDesktopShortcut( result );
|
||
|
|
DVDDetectSetupRun::Remove();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void DetectCurrent()
|
||
|
|
{
|
||
|
|
DVDDetectBuffer buffer;
|
||
|
|
TCHAR out[1024];
|
||
|
|
out[0]=TEXT('\0');
|
||
|
|
if(SUCCEEDED(buffer.DetectAll()))
|
||
|
|
{
|
||
|
|
TCHAR* pOut=out;
|
||
|
|
|
||
|
|
if( buffer.mci.Found()) {
|
||
|
|
pOut = Append(pOut, TEXT("\nMCI:\n"));
|
||
|
|
pOut = Print(pOut, buffer.mci);
|
||
|
|
}
|
||
|
|
if( buffer.hw.Found()) {
|
||
|
|
pOut = Append(pOut, TEXT("\nDShow HW:\n" ));
|
||
|
|
pOut = Print(pOut, buffer.hw);
|
||
|
|
}
|
||
|
|
if( buffer.sw.Found()) {
|
||
|
|
pOut = Append(pOut, TEXT("\nDShow SW:\n" ));
|
||
|
|
pOut = Print(pOut, buffer.sw);
|
||
|
|
}
|
||
|
|
if( pOut == out ) {
|
||
|
|
pOut = Append( pOut, TEXT("No decoders found"));
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
Append( out, TEXT("DvdCheck Failed\n") );
|
||
|
|
}
|
||
|
|
OutputDebugString( out );
|
||
|
|
MessageBox( NULL, out, TEXT("Detected DVD decoders"), MB_OK );
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" BOOL IsDvdPresent (VOID);
|
||
|
|
|
||
|
|
int APIENTRY WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,
|
||
|
|
LPSTR lpCmdLine,
|
||
|
|
int nCmdShow)
|
||
|
|
{
|
||
|
|
if( lpCmdLine ) {
|
||
|
|
if( AreEquivalent( lpCmdLine, "/setup") ) {
|
||
|
|
DetectForSetup( false );
|
||
|
|
} else if( AreEquivalent( lpCmdLine, "/setup98") ) {
|
||
|
|
DetectForSetup( true );
|
||
|
|
} else if( AreEquivalent( lpCmdLine, "/upgrade" ) ) {
|
||
|
|
DetectForUpgrade( false );
|
||
|
|
} else if( AreEquivalent( lpCmdLine, "/forceupgrade" ) ) {
|
||
|
|
DetectForUpgrade( true );
|
||
|
|
} else if( AreEquivalent( lpCmdLine, "/remove" ) ) {
|
||
|
|
DetectForRemove();
|
||
|
|
} else if( AreEquivalent( lpCmdLine, "/detect" ) ) {
|
||
|
|
DetectCurrent();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|