#include <set>
#include <string>
#include <locale>
#include <windows.h>
#include <winnt.h>

#pragma warning(disable:  4127 ) // conditional expression is constant


using namespace std;

//---------------------------------------------------------------------------------------------------------------------
static bool GetServiceList( set<string> &lstResult ) 
{
    SC_HANDLE hSCM = OpenSCManager(NULL, NULL, SC_MANAGER_ENUMERATE_SERVICE); 
    if( hSCM == NULL )
    {
		return false;
    }

    DWORD dwResumeHandle =  0 ;
	DWORD cbNeeded =  2048 ;
	DWORD cServices;
	DWORD dwError;
	LPENUM_SERVICE_STATUS pStatus;
    BYTE *bBuffer = NULL;

    try
    {
	    for(int i =  0 ; i <  3 ; i++) // do not do more then 3 trys
	    {
            // Review: If a single service is too big to fit into the bBuffer it seems like
            //         this will become an infinite loop. A check for returned cServices being
            //         bigger than zero would be usefull. <DM>
            // Owner:  issue is addressed

		    pStatus = (LPENUM_SERVICE_STATUS)bBuffer;

    	    bBuffer = new BYTE[cbNeeded];
            memset(bBuffer,  0 , cbNeeded);


            DWORD new_need =  0 ;
		    // retrieve a portion of service status information
		    if(!EnumServicesStatus(hSCM, SERVICE_WIN32, SERVICE_STATE_ALL,
								    pStatus, cbNeeded, &new_need,
								    &cServices, &dwResumeHandle))
		    {
			    dwError = GetLastError();
			    if (dwError != ERROR_MORE_DATA)
			    {
                    delete [] bBuffer;
				    break;
			    }
                if( new_need > cbNeeded )       // function requires more memory
                {
                    delete [] bBuffer;
                    cbNeeded = new_need;
                    continue;                   //  try to get info again
                }
		    }
		    for (DWORD i =  0 ; i < cServices; i++, pStatus++) 
		    {
                if( pStatus->lpServiceName == NULL )
                {
                    continue;
                }
                string name_srv = pStatus->lpServiceName;
                if( !name_srv.empty() )
                {
                    lstResult.insert( name_srv );
                }
		    }
            break;

	    }
        return true;
    }catch(...){}
    return false;
}