C / C++ / MFC

WCHAR* CA_Registry::_get_SQLServer_InstalledInstances( void )
{
    HKEY keyHandle;
    WCHAR rgValue[1024];
//  WCHAR szInstance[1024]; // not needed
    // no point in using CString for constant strings
    WCHAR* regPath = L"SOFTWARE\\Microsoft\\Microsoft SQL Server\\";
    WCHAR* regReq = L"InstalledInstances";
    DWORD size1 = sizeof(rgValue);
    // Type; removed as no longer used
    // no need to initialise dwType to zero here
    DWORD dwType;
    // no need to initialise regStatus to REG_MULTI_SZ here
    DWORD regStatus;
 
    int ndx;
    WCHAR *szReturnValue = NULL;
/* variables not needed
    size_t szRgVal;
    int maxStr;
    bool doubleZero;
*/ 
    if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_QUERY_VALUE, &keyHandle) == ERROR_SUCCESS) {
    // no need to initialise dwType to REG_MULTI_SZ here
        regStatus = RegQueryValueEx( keyHandle, regReq, NULL, &dwType, (LPBYTE)rgValue, &size1);
			
        RegCloseKey(keyHandle);	// we're done with the registry so close the key here
        switch ( regStatus ) {
        case ERROR_SUCCESS:								
/*
 * rewrote this part to use standard (Unicode) functions
            ndx = 0;
            doubleZero = false;
            szRgVal = sizeof(rgValue);
 
            for(int i=0; i < szRgVal; i++) {
                if (!rgValue[i] == L'\0') {
                    szInstance[ndx++] = rgValue[i];
                    doubleZero = false;
                }
                else {
                    if (doubleZero == false) {
                        szInstance[ndx++] = L';';
                        doubleZero = true;
                    }
                    else {
                        szInstance[ndx++] = L'\0\0';
                        break;
                    }
                }                    
            }
 */
            // scan the strings, replacing single nulls by semi-colons
            {
                WCHAR* pszString = rgValue;
                while (*pszString)
                {
                    ndx = wcslen(pszString);
                    // add string length so we point at trailing null character
                    pszString += ndx;
                    if (pszString[1] != L'\0')
                    *pszString++ = L';';    // add semi-colon if not the last string
                }
            }
            /* removed this redundant code
            maxStr = wcslen(szInstance);
            szInstance[maxStr-0] = (char)'\0';
            */
            ndx = wcslen(rgValue);	// we need this value below
            // this also needs to be wprintf, as rgValue is Unicode
            wprintf(L" SQL Server Instance: %s %i\n", rgValue, ndx);				
            // following statement is redundant, we already have the printf
            // cout << "key was read successfully\n";

            // copy to new buffer here
            szReturnValue = new wchar_t[ndx + 1]; // +1 for trailing null character
            wcscpy_s(szReturnValue, ndx + 1, rgValue);
            break;
 
        case ERROR_MORE_DATA:
            cout << "Buffer too small\n";
            // your challenge is to fix the code to handle this be allocating a larger buffer
            break;			
 
        default:			
            cout << "Could not read key\n";
        }
    }
    else {
        cout << "Can not open key";
        cin.ignore();
    }
/* moved this functionality to within success case above
    int strSize = sizeof(szInstance);	
    WCHAR *szReturnValue = new wchar_t[strSize];	
        for(int i = 0; i < strSize; i++) {
        szReturnValue[i]=szInstance[i];		    
    }	
    strSize=wcslen(szReturnValue);	
    szReturnValue[strSize-0]=(WCHAR)L'\0';
 */
    return szReturnValue;  // set above
    // this delete statement will never be reached! Also the calling function needs to delete the buffer when finished with it 
//  delete [] szReturnValue;
}

WCHAR* CA_Registry::_get_SQLServer_InstalledInstances( void )
{
	HKEY keyHandle;
	WCHAR rgValue[1024];
	WCHAR *regPath = L"SOFTWARE\\Microsoft\\Microsoft SQL Server\\";
	WCHAR *regReq = L"InstalledInstances";
	DWORD size1 = sizeof(rgValue);
	DWORD dwType;
	DWORD regStatus;

	int ndx;
	WCHAR* pszString = rgValue;
	WCHAR *szReturnValue = NULL;
	
	if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, regPath, 0, KEY_QUERY_VALUE, &keyHandle) == ERROR_SUCCESS) {
		regStatus = RegQueryValueEx( keyHandle, regReq, NULL, &dwType, (LPBYTE)rgValue, &size1);
		RegCloseKey(keyHandle);	

		switch ( regStatus ) {
			case ERROR_SUCCESS:
				// scan the strings, replacing single nulls by semi-colons
				{
					while (*pszString)
					{
						ndx = wcslen(pszString);
						// add string length so we point at trailing null character
						pszString += ndx;
						if (pszString[1] != L'\0')
						*pszString++ = L';';    // add semi-colon if not the last string
					}
				}
            
				ndx = wcslen(rgValue);	// we need this value below
				wprintf(L" SQL Server Instance: %s %i\n", rgValue, ndx);				
            
				// copy to new buffer here
				szReturnValue = new wchar_t[ndx + 1]; // +1 for trailing null character
				wcscpy_s(szReturnValue, ndx + 1, rgValue);            

				break;

			case ERROR_MORE_DATA:
				cout << "Buffer too small\n";
				break;			

			default:			
				cout << "Count not read key\n";
		}
	}
	else {
		cout << "Can not open key";
		cin.ignore();
	}

	return szReturnValue;	
}

BOOL CA_SQLServer_Scan::_scan_Enumerate_Registry( HWND txtProgress, HWND lblServers, HWND pbProgress )
{
	CA_Registry caReg;
	BOOL bResult = FALSE;
	BOOL bInstance = FALSE;
	int instanceLength;
	WCHAR *szComputerName = NULL;
	WCHAR *szInstanceNames = NULL;
	WCHAR *szCurrentVersion = NULL;
	WCHAR szServerInstance[125];

	// Get the SQL Server instance names
	szComputerName = caReg._get_ComputerName();
	szInstanceNames = caReg._get_SQLServer_InstalledInstances();
	instanceLength = wcslen(szInstanceNames);
			
	if (instanceLength > 0) {

		WCHAR *szInstanceName = NULL;
		WCHAR *token1, *nextToken;
		WCHAR seps[] = L";";
		WCHAR vSeps[] = L".";
		token1 = wcstok_s(szInstanceNames, seps, &nextToken);
		do {
			if (token1 == NULL) { 
				break;
			}
			else {
				szInstanceName = token1;
				szCurrentVersion = caReg._get_SQLServer_Version(szInstanceName);
				
				// Parse the version Numbers
				int szVerValue = wcslen(szCurrentVersion);
				if ( szVerValue > 1 ) {
					WCHAR *szVerMajor = NULL, *szVerMinor = NULL, *szVerBuild = NULL;
					WCHAR *vToken1, *vToken2, *vToken3, *vNextToken;
					int iVerMajor=0, iVerMinor=0, iVerBuild=0;
					
					vToken1 = wcstok_s(szCurrentVersion, vSeps, &vNextToken);
					if (!vToken1 == NULL) {
						szVerMajor = vToken1;
						iVerMajor = _wtoi(szVerMajor);
					}
					
					vToken2 = wcstok_s(NULL, vSeps, &vNextToken);
					if (!vToken2 == NULL) {
						szVerMinor = vToken2;
						iVerMinor = _wtoi(szVerMinor);
					}

					vToken3 = wcstok_s(NULL, vSeps, &vNextToken);
					if (!vToken3 == NULL) {
						szVerBuild = vToken3;
						iVerBuild = _wtoi(szVerBuild);
					}

					WCHAR pbMessage[80];
					wcsncpy_s(pbMessage, _countof(pbMessage), L"Registered: ", wcslen(L"Registered: "));
					//wcsncat_s(pbMessage, _countof(pbMessage), L"LOCAL", wcslen(L"LOCAL"));
					wcsncat_s(pbMessage, _countof(pbMessage), L"\\", wcslen(L"\\"));
					wcsncat_s(pbMessage, _countof(pbMessage), szInstanceName, wcslen(szInstanceName));					
					SetWindowText(txtProgress, pbMessage);
					Sleep(500);
					
					// Analyse the Version Numbers
					switch (iVerMajor) 
					{
					case 11:
						bInstance = TRUE;
						break;
					case 10:
						bInstance = TRUE;
						break;
					case 9:
						bInstance = TRUE;
						break;
					default:
						bInstance = FALSE;
						break;
					}

					if (bInstance) {
						int iComputerName = wcslen(szComputerName);
						int iInstanceName = wcslen(szInstanceName);						
						
						// Build the Computer\\Instance
						wcsncpy_s(szServerInstance, _countof(szServerInstance), szComputerName, iComputerName);
						wcsncat_s(szServerInstance, _countof(szServerInstance), L"\\", wcslen(L"\\"));
						wcsncat_s(szServerInstance, _countof(szServerInstance), szInstanceName, iInstanceName);
						
						// Trim the wChar Array
						iInstanceName=wcslen(szServerInstance);	
						szServerInstance[iInstanceName-0]=(WCHAR)L'\0';

						// Get the Collection Count
						int iCollectionCount;
						iCollectionCount = sqlSrvCollection.Count();

						if (iCollectionCount > 0) {
							// Check for Duplicate
							WCHAR *szComparision = NULL;
							
							for (int i=0; i<iCollectionCount; i++) {
								szComparision = sqlSrvCollection[i];
								if (wcscmp(szServerInstance, szComparision) != 0) {
									int iCollection = sqlSrvCollection.Add();
									sqlSrvCollection[iCollection] = szServerInstance;
								}								
							}							
						}
						else {
							// Add the Server Instance
							int iCollection = sqlSrvCollection.Add();
							sqlSrvCollection[iCollection] = szServerInstance;
						}
					}
				}

				// Get the next token to cycle through
				token1 = wcstok_s(NULL, seps, &nextToken);

			}
		} while (TRUE);
	}
	
	return bResult;
}

jkirkerx wrote:

I had to move the pszString above the switch because the compiler complained.

jkirkerx wrote:

I'm determined to get this program out there

jkirkerx wrote:

socket issue ... and the safe array

jkirkerx wrote:

it's 60F outside now.

tagopi wrote:

is there anyway to solve this issue?

tagopi wrote:

is there anyway to solve this issue?

// Call for each explorer window

        HWND hExplorer = FindWindowEx(GetDesktopWindow(), NULL, L"ExploreWClass", NULL);

        while(hExplorer != NULL)
        {
            EnumChildWindows(hExplorer, RefreshFolderSelectionCB, (LPARAM)&packet);

            hExplorer = FindWindowEx(GetDesktopWindow(), hExplorer, L"ExploreWClass", NULL);
        }
...
...
...

BOOL CALLBACK    RefreshFolderSelectionCB(HWND hWnd, LPARAM lParam)
{
    WCHAR sBuffer[MAX_PATH] = {0};

    ::GetClassName(hWnd, sBuffer, MAX_PATH);

    if (wcscmp(L"SysTreeView32", sBuffer) == 0)
    {
        RefreshFolderSelectionPacket* pPacket = (RefreshFolderSelectionPacket*)lParam;

        // Refreshes the tree nodes that match the contained full paths, and all their open descendants
        // Paths are updated by calling UpdateItem (below)
        // Cache is used across explorer instances to ensure no path is updated twice.

        RefreshSelection(pPacket);
    }

    return TRUE;
}


...
...
static void UpdateItem(IShellFolder* pDesktop, const std::wstring& rsFullPath)
{
    PIDLIST_RELATIVE pIDL;

    if (SUCCEEDED(pDesktop->ParseDisplayName(NULL, NULL, (LPWSTR)rsFullPath.c_str(), NULL, &pIDL, NULL)))
    {
        SHChangeNotify(SHCNE_UPDATEITEM, SHCNF_IDLIST|SHCNF_NOTIFYRECURSIVE|SHCNF_FLUSH, pIDL, NULL);

        ILFree(pIDL);
    }
}