/**************************************************************************** Copyright information : Copyright (c) 1998-1999 Microsoft Corporation File Name : ParserEngine.cpp Project Name : WMI Command Line Author Name : Ch. Sriramachandramurthy Date of Creation (dd/mm/yy) : 27th-September-2000 Version Number : 1.0 Brief Description : This class encapsulates the functionality needed for parsing the command string entered as input and validating the same. Revision History : Last Modified By : Ch. Sriramachandramurthy Last Modified Date : 18th-December-2000 *****************************************************************************/ // include files #include "Precomp.h" #include "GlobalSwitches.h" #include "CommandSwitches.h" #include "HelpInfo.h" #include "ErrorLog.h" #include "ParsedInfo.h" #include "CmdTokenizer.h" #include "CmdAlias.h" #include "ParserEngine.h" #include "WMICliXMLLog.h" #include "ErrorInfo.h" #include "FormatEngine.h" #include "ExecEngine.h" #include "WmiCmdLn.h" /*---------------------------------------------------------------------------- Name :CParserEngine Synopsis :This function initializes the member variables when an object of the class type is instantiated. Type :Constructor Input Parameter(s):None Output parameters :None Return Type :None Global Variables :None Calling Syntax :None Notes :None ----------------------------------------------------------------------------*/ CParserEngine::CParserEngine() { m_pIWbemLocator = NULL; m_pITargetNS = NULL; m_bAliasName = FALSE; } /*---------------------------------------------------------------------------- Name :~CParserEngine Synopsis :This function uninitializes the member variables when an object of the class type goes out of scope. Type :Destructor Input Parameter(s):None Output parameters :None Return Type :None Global Variables :None Calling Syntax :None Notes :None ----------------------------------------------------------------------------*/ CParserEngine::~CParserEngine() { SAFEIRELEASE(m_pITargetNS); SAFEIRELEASE(m_pIWbemLocator); } /*---------------------------------------------------------------------------- Name :GetCmdTokenizer Synopsis :This function returns a reference to the CCmdTokenizer object, a data member of this class. Type :Member Function Input Parameter(s):None Output parameters :None Return Type :CCmdTokenizer& Global Variables :None Calling Syntax :GetCmdTokenizer() Notes :None ----------------------------------------------------------------------------*/ CCmdTokenizer& CParserEngine::GetCmdTokenizer() { return m_CmdTknzr; } /*---------------------------------------------------------------------------- Name :Initialize Synopsis :This function initializes the neeeded data members of this class. Type :Member Function Input Parameter(s):None Output parameters :None Return Type :void Global Variables :None Calling Syntax :Initialize() Notes :None ----------------------------------------------------------------------------*/ void CParserEngine::Initialize() { m_bAliasName = FALSE; } /*---------------------------------------------------------------------------- Name :Uninitialize Synopsis :This function uninitializes the member variables when the execution of a command string issued on the command line is completed and then the parser engine variables are also uninitialized. Type :Member Function Input Parameter(s): bFinal - boolean value which when set indicates that the program Output parameters :None Return Type :void Global Variables :None Calling Syntax :Uninitialize() Notes :None ----------------------------------------------------------------------------*/ void CParserEngine::Uninitialize(BOOL bFinal) { m_bAliasName = FALSE; m_CmdTknzr.Uninitialize(); m_CmdAlias.Uninitialize(); if (bFinal) { m_CmdAlias.Uninitialize(TRUE); SAFEIRELEASE(m_pITargetNS); SAFEIRELEASE(m_pIWbemLocator); } } /*---------------------------------------------------------------------------- Name :SetLocatorObject Synopsis :This function sets the WMI Locator Object to the m_pIWbemLocator. Type :Member Function Input Parameter(s): pIWbemLocator - pointer to IWbemLocator interface . Output parameters :None Return Type :BOOL Global Variables :None Calling Syntax :SetLocatorObject(pIWbemLocator) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::SetLocatorObject(IWbemLocator* pIWbemLocator) { static BOOL bFirst = TRUE; BOOL bRet = TRUE; if (bFirst) { if (pIWbemLocator != NULL) { SAFEIRELEASE(m_pIWbemLocator); m_pIWbemLocator = pIWbemLocator; m_pIWbemLocator->AddRef(); } else bRet = FALSE; bFirst = FALSE; } return bRet; } /*---------------------------------------------------------------------------- Name :ProcessTokens Synopsis :This function does the processing of the tokens. It checks for the presence of switches and calls the appropriate Parsing function and updates the CParsedInfo object passed to it. Type :Member Function Input Parameter(s): rParsedInfo - reference to CParsedInfo class object Output parameters : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ProcessTokens(rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ProcessTokens(CParsedInfo& rParsedInfo) { BOOL bContinue = TRUE; RETCODE retCode = PARSER_EXECCOMMAND; // Obtain the token vector. CHARVECTOR cvTokens = m_CmdTknzr.GetTokenVector(); //the iterator to span throuh the vector variable CHARVECTOR::iterator theIterator = NULL; // Check for the presence of tokens. Absence of tokens indicates // no command string is fed as input. if (!cvTokens.empty()) { // Obtain the pointer to the beginning of the token vector. theIterator = cvTokens.begin(); // Check for the presence of the global switches and // store the values specified with them (if any) in the // CGlobalSwitches object. Global switches are followed // '/' character. if (IsOption(*theIterator)) { retCode = ParseGlobalSwitches(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_CONTINUE) { // If no more tokens are present if (theIterator >= cvTokens.end()) { retCode = PARSER_MESSAGE; bContinue = FALSE; } } else bContinue = FALSE; } if (bContinue) { // Suppress Information Msg before Executing command. rParsedInfo.GetCmdSwitchesObject().SetInformationCode(0); // Check for the presence of the CLASS keyword if (CompareTokens(*theIterator, CLI_TOKEN_CLASS)) { // Move to next token, and check its validity retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, CLASS, IDS_E_INVALID_CLASS_SYNTAX); if (retCode == PARSER_CONTINUE) { // NOTE: Indicates direct escaping to WMI schema // Parse and interpret the remaining tokens following // the CLASS keyword retCode = ParseClassInfo(cvTokens, theIterator, rParsedInfo); } } // Check for the presence of the PATH keyword else if (CompareTokens(*theIterator, CLI_TOKEN_PATH)) { //NOTE: Indicates PATH clause without an alias name // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, PATH, IDS_E_INVALID_PATH_SYNTAX); if (retCode == PARSER_CONTINUE) // Parse and interpret the remaining tokens // following the PATH clause retCode = ParsePathInfo(cvTokens, theIterator, rParsedInfo); } // Check for the presence of the CONTEXT keyword else if (CompareTokens(*theIterator, CLI_TOKEN_CONTEXT)) { if (GetNextToken(cvTokens, theIterator)) { retCode = ParseContextInfo(cvTokens, theIterator, rParsedInfo); } else { rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(TRUE); rParsedInfo.GetHelpInfoObject().SetHelp(GLBLCONTEXT, TRUE); retCode = PARSER_DISPHELP; } } // If the token value does not match against the // pre-defiend keywords, it is considered as an alias. else { // Validate the alias name and parse the remaining // tokens following the name. retCode = ParseAliasInfo(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { try { _bstr_t bstrTrgtClass; _TCHAR *pszClass = NULL; // Check the validity of the path expression w.r.t the // alias specified using the following steps: // (i.e to check for alias - path conflict) // step1: Obtain the alias target class. rParsedInfo.GetCmdSwitchesObject(). GetClassOfAliasTarget(bstrTrgtClass); // step2: Obtain the explicitly specified class. pszClass = rParsedInfo.GetCmdSwitchesObject(). GetClassPath(); if (!(!bstrTrgtClass) && (pszClass != NULL)) { // If both are not same, set the errata code if(!CompareTokens((_TCHAR*)bstrTrgtClass,pszClass)) { // Set the error code rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_I_ALIAS_PATH_CONFLICT); retCode = PARSER_ERROR; } } } catch(_com_error& e) { _com_issue_error(e.Error()); } } } } } else { // Indicates NULL string specified as input on the WMI Command Line. rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_BLANK_COMMAND_MESSAGE); retCode = PARSER_ERROR; } // Get the Property qualifiers information from the alias - SET and CREATE. if ((retCode == PARSER_EXECCOMMAND) && ((CompareTokens(rParsedInfo.GetCmdSwitchesObject().GetVerbName(), CLI_TOKEN_SET)) || (CompareTokens(rParsedInfo.GetCmdSwitchesObject().GetVerbName(), CLI_TOKEN_CREATE)))) { if (m_bAliasName) { if (FAILED(m_CmdAlias.ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } } if ( retCode == PARSER_DISPHELP ) { if ( m_bAliasName || rParsedInfo.GetCmdSwitchesObject().GetClassPath() != NULL ) { ObtainMethodsAvailableFlag(rParsedInfo); ObtainWriteablePropsAvailailableFlag(rParsedInfo); } if ( m_bAliasName == TRUE ) { rParsedInfo.GetCmdSwitchesObject(). SetLISTFormatsAvailable( m_CmdAlias.ObtainAliasFormat(rParsedInfo, TRUE)); } } if ( retCode == PARSER_EXECCOMMAND || retCode == PARSER_DISPHELP ) { retCode = ProcessOutputAndAppendFiles(rParsedInfo, retCode, FALSE); } else if (rParsedInfo.GetCmdSwitchesObject().GetOutputSwitchFlag() == TRUE && retCode == PARSER_MESSAGE) rParsedInfo.GetCmdSwitchesObject().SetOutputSwitchFlag(FALSE); return retCode; } /*---------------------------------------------------------------------------- Name :ParseClassInfo Synopsis :This function does the parsing and interprets if command has CLASS keyword specified in it. It parses the remaining tokens following and updates the same in CParsedInfo object passed to it. Type :Member Function Input Parameter(s) : cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseClassInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseClassInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo ) { // BNF: CLASS [] BOOL bContinue = TRUE; RETCODE retCode = PARSER_EXECCOMMAND; // If option if (IsOption(*theIterator)) { // Check for help retCode = IsHelp(cvTokens, theIterator, rParsedInfo, CLASS, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); if (retCode != PARSER_CONTINUE) bContinue = FALSE; } else { // Store the class path in the CCommandSwitches object. if(!rParsedInfo.GetCmdSwitchesObject().SetClassPath(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode(OUT_OF_MEMORY); bContinue = FALSE; retCode = PARSER_OUTOFMEMORY; } else if ( IsValidClass(rParsedInfo) == FALSE ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CLASS); retCode = PARSER_ERROR; bContinue = FALSE; } if(bContinue) { // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // i.e. is not specified. bContinue = FALSE; retCode = PARSER_EXECCOMMAND; } } } if (bContinue) { // Check for the presence of /? if (IsOption(*theIterator)) { // Check for help retCode = IsHelp(cvTokens, theIterator, rParsedInfo, CLASS, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); } else { // Parse and interpret the verb tokens that follow retCode = ParseVerbInfo(cvTokens,theIterator,rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { // Check for verb switches if (GetNextToken(cvTokens, theIterator)) retCode = ParseVerbSwitches(cvTokens, theIterator, rParsedInfo); } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseAliasInfo Synopsis :This function does the parsing and interprets if command has name in it.It Validate the alias name and parses the remaining tokens following the name. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseAliasInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseAliasInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { //BNF: ( | [] | [] ) [] RETCODE retCode = PARSER_EXECCOMMAND; HRESULT hr = S_OK; BOOL bContinue = TRUE; RETCODE tRetCode = PARSER_ERROR; // Store the AliasName in the CommandSwitches object. if(!rParsedInfo.GetCmdSwitchesObject().SetAliasName(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } else { m_bAliasName = TRUE; // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, CmdAllInfo, IDS_E_INVALID_COMMAND); if (retCode == PARSER_ERROR) tRetCode = PARSER_EXECCOMMAND; else if(retCode == PARSER_DISPHELP && rParsedInfo.GetGlblSwitchesObject().GetInteractiveStatus() == TRUE) { tRetCode = PARSER_EXECCOMMAND; rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(FALSE); rParsedInfo.GetHelpInfoObject().SetHelp(CmdAllInfo, FALSE); } // Connect to alias and retrieve the alias information try { // Connect to the AliasNamespace. hr = m_CmdAlias.ConnectToAlias(rParsedInfo, m_pIWbemLocator); ONFAILTHROWERROR(hr); // Obtain the alias information ( Target, Namespace,...) retCode = m_CmdAlias.ObtainAliasInfo(rParsedInfo); if((retCode == PARSER_OUTOFMEMORY) || (retCode == PARSER_ERRMSG)) { if (retCode == PARSER_OUTOFMEMORY) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } bContinue = FALSE; } } catch(_com_error& e) { retCode = PARSER_ERRMSG; bContinue = FALSE; _com_issue_error(e.Error()); } if (bContinue && tRetCode != PARSER_EXECCOMMAND) { // Check for the presence of the PATH keyword if (CompareTokens(*theIterator, CLI_TOKEN_PATH)) { // NOTE: Indicates PATH clause preceded by an alias name // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, PATH, IDS_E_INVALID_PATH); if (retCode == PARSER_CONTINUE) // Parse and interpret the remaining tokens following // the PATH clause retCode = ParsePathInfo(cvTokens, theIterator, rParsedInfo); } // Check for the presence of the WHERE keyword else if (CompareTokens(*theIterator, CLI_TOKEN_WHERE)) { // NOTE: Indicates WHERE clause preceded by an alias name // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, WHERE, IDS_E_INVALID_QUERY); if (retCode == PARSER_CONTINUE) // Parse and interpret the remaining tokens following // the WHERE clause retCode = ParseWhereInfo(cvTokens, theIterator, rParsedInfo); } // Check for the presence of the '(' else if (CompareTokens(*theIterator, CLI_TOKEN_LEFT_PARAN)) { // Frame the parameterized WHERE expression if (!ParsePWhereExpr(cvTokens, theIterator, rParsedInfo, TRUE)) { retCode = PARSER_ERROR; } else { // Move to next token if (theIterator >= cvTokens.end()) { // PARSER_ERROR if no more tokens are present rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; } else { if (CompareTokens(*theIterator, CLI_TOKEN_RIGHT_PARAN)) { // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // if no more tokens are present. retCode = PARSER_EXECCOMMAND; } else { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, PWhere, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); if ( retCode == PARSER_DISPHELP ) { if (FAILED(m_CmdAlias. ObtainAliasVerbDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } } else { retCode = ParseVerbInfo(cvTokens, theIterator, rParsedInfo); // Parse and interpret the verb tokens // that follow if (retCode == PARSER_EXECCOMMAND) { if(GetNextToken(cvTokens, theIterator)) // check for the common verb // switches /INTERACTIVE, // /NOINTERACTIVE retCode = ParseVerbSwitches( cvTokens, theIterator, rParsedInfo); } } } } else { // PARSER_ERROR if no more tokens are present rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; } } } } else { if (IsOption(*theIterator)) { // Check for help retCode = IsHelp(cvTokens, theIterator, rParsedInfo, CmdAllInfo, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); if (retCode == PARSER_DISPHELP) { rParsedInfo.GetCmdSwitchesObject(). AddToAlsFrnNmsOrTrnsTblMap( CharUpper(rParsedInfo.GetCmdSwitchesObject(). GetAliasName()), rParsedInfo.GetCmdSwitchesObject().GetAliasDesc()); } } else { if (bContinue) { // Frame the parameterized WHERE expression if (!ParsePWhereExpr(cvTokens, theIterator, rParsedInfo, FALSE)) { retCode = PARSER_ERROR; } else { if ( theIterator >= cvTokens.end() ) retCode = PARSER_EXECCOMMAND; else { // Parse the verb. if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, PWhere, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); if ( retCode == PARSER_DISPHELP ) { if (FAILED(m_CmdAlias. ObtainAliasVerbDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } } else { retCode = ParseVerbInfo(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { if (GetNextToken(cvTokens, theIterator)) // check for the common verb switches // /INTERACTIVE, /NOINTERACTIVE retCode = ParseVerbSwitches(cvTokens, theIterator, rParsedInfo); } } } } } } } } } if(tRetCode == PARSER_EXECCOMMAND) { if ((retCode != PARSER_ERRMSG) && (retCode != PARSER_OUTOFMEMORY)) { retCode = tRetCode; rParsedInfo.GetCmdSwitchesObject().SetErrataCode(0); } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseWhereInfo Synopsis :This function does the parsing and interprets if command has alias, with where clause also specified it.It parses the remaining tokens following and updates the same in CParsedInfo object passed to it. Type :Member Function Input Parameter(s) : cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseWhereInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseWhereInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; rParsedInfo.GetCmdSwitchesObject().SetExplicitWhereExprFlag(TRUE); if (IsOption(*theIterator)) { // Check for help retCode = IsHelp(cvTokens, theIterator, rParsedInfo, WHERE, IDS_E_INVALID_WHERE_SYNTAX, LEVEL_ONE); if (retCode != PARSER_CONTINUE) bContinue = FALSE; } if (bContinue) { if ( !m_bAliasName && rParsedInfo.GetCmdSwitchesObject(). GetClassPath() == NULL ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ALIAS_OR_PATH_SHOULD_PRECEED_WHERE); retCode = PARSER_ERROR; } // Store the WHERE expression in the CCommandSwitches object. else if(!rParsedInfo.GetCmdSwitchesObject().SetWhereExpression( *theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } else { // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // If no more tokens are present. i.e no verb clause is present retCode = PARSER_EXECCOMMAND; } else { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, WHERE, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); } else { // Parse and interpret the verb tokens that follow // Handled for /verb to verb retCode = ParseVerbInfo(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { if (GetNextToken(cvTokens, theIterator)) //check for the common verb switches /INTERACTIVE, // /NOINTERACTIVE retCode = ParseVerbSwitches(cvTokens, theIterator, rParsedInfo); } } } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParsePathInfo Synopsis :This function does the parsing and interprets if command has alias with path clause also specified it.It parses the remaining tokens following and updates the same in CParsedInfo object passed to it. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParsePathInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParsePathInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, PATH, IDS_E_INVALID_PATH_SYNTAX, LEVEL_ONE); if (retCode != PARSER_CONTINUE) bContinue = FALSE; } if (bContinue) { // Store the object PATH expression in the CCommandSwitches object. if(!rParsedInfo.GetCmdSwitchesObject().SetPathExpression(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } else { //Extract the classname and where expression given path expression _TCHAR pszPathExpr[MAX_BUFFER] = NULL_STRING; lstrcpy(pszPathExpr,CLI_TOKEN_NULL); lstrcpy(pszPathExpr, rParsedInfo.GetCmdSwitchesObject(). GetPathExpression()); if (!ExtractClassNameandWhereExpr(pszPathExpr, rParsedInfo)) retCode = PARSER_ERROR; // Move to next token else if ( IsValidClass(rParsedInfo) == FALSE ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CLASS); bContinue = FALSE; retCode = PARSER_ERROR; } else if (!GetNextToken(cvTokens, theIterator)) // If no more tokens are present. i.e no verb clause is present retCode = PARSER_EXECCOMMAND; else { if ( CompareTokens(*theIterator, CLI_TOKEN_WHERE) ) { if ( rParsedInfo.GetCmdSwitchesObject(). GetWhereExpression() != NULL ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_KEY_CLASS_NOT_ALLOWED_WITH_PATHWHERE); retCode = PARSER_ERROR; } else { retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, WHERE, IDS_E_INVALID_WHERE_SYNTAX); if (retCode == PARSER_CONTINUE) // Parse and interpret the remaining tokens // following the WHERE clause retCode = ParseWhereInfo(cvTokens, theIterator, rParsedInfo); } } else { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, PATH, IDS_E_INVALID_HELP_SYNTAX, LEVEL_ONE); } else { // Parse and interpret the verb tokens that follow // Handled for /verb => verb. retCode = ParseVerbInfo(cvTokens,theIterator,rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { if (GetNextToken(cvTokens, theIterator)) //check for the common verb switches /INTERACTIVE, ///NOINTERACTIVE retCode = ParseVerbSwitches(cvTokens, theIterator, rParsedInfo); } } } } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseVerbInfo Synopsis :This function does the parsing and interprets if command has verb clause specified in it.It parses the remaining tokens following the verb and updates the same in CParsedInfo object passed to it. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseVerbInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseVerbInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; // STORE the verb name in the CCommandSwitches object if ( rParsedInfo.GetCmdSwitchesObject().SetVerbName(*theIterator) == FALSE ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode(IDS_E_MEMALLOC_FAIL); retCode = PARSER_ERROR; } // Check for the presence of the following standard verbs: // 1.GET 2.SHOW 3.SET 4.CALL 5.ASSOC 6. CREATE 7. DELETE // GET verb specified else if (CompareTokens(*theIterator, CLI_TOKEN_GET)) { retCode = ParseGETVerb(cvTokens, theIterator, rParsedInfo); } // LIST verb specified else if (CompareTokens(*theIterator, CLI_TOKEN_LIST)) { if (m_bAliasName == FALSE) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_LIST_USAGE); retCode = PARSER_ERROR; } else retCode = ParseLISTVerb(cvTokens, theIterator, rParsedInfo); } // SET | CREATE verb specified else if (CompareTokens(*theIterator, CLI_TOKEN_SET) || CompareTokens(*theIterator, CLI_TOKEN_CREATE)) { // and cannot be specified with // CREATE verb. Only should be specified. if (CompareTokens(*theIterator, CLI_TOKEN_CREATE) && rParsedInfo.GetCmdSwitchesObject(). GetExplicitWhereExprFlag()) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CREATE_EXPRESSION); retCode = PARSER_ERROR; } else { HELPTYPE helpType = CompareTokens(*theIterator, CLI_TOKEN_CREATE) ? CREATEVerb : SETVerb; retCode = ParseSETorCREATEVerb(cvTokens, theIterator, rParsedInfo, helpType); } } // CALL verb specified else if (CompareTokens(*theIterator, CLI_TOKEN_CALL)) { retCode = ParseCALLVerb(cvTokens, theIterator, rParsedInfo); } // ASSOC verb specified else if (CompareTokens(*theIterator, CLI_TOKEN_ASSOC)) { retCode = ParseASSOCVerb(cvTokens, theIterator, rParsedInfo); } // DELETE verb specified. else if (CompareTokens(*theIterator, CLI_TOKEN_DELETE)) { retCode = PARSER_EXECCOMMAND; //ParseDELETEVerb(cvTokens, theIterator, rParsedInfo); } // User defined verb else if (m_bAliasName) { // User defined verbs can only be associated with alias retCode = ParseMethodInfo(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_CONTINUE) retCode = PARSER_EXECCOMMAND; } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode(IDS_E_INVALID_VERB); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :ParseMethodInfo Synopsis :This function parses the tokens following the user defined verb and updates the info in CParsedInfo object passed to it. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseMethodInfo(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseMethodInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; // Store the method name if(!rParsedInfo.GetCmdSwitchesObject().SetMethodName(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } else { if(m_bAliasName) { if (FAILED(m_CmdAlias.ObtainAliasVerbDetails(rParsedInfo))) { retCode = PARSER_ERRMSG; bContinue =FALSE; } else { VERBTYPE vtVerbType = rParsedInfo.GetCmdSwitchesObject().GetVerbType(); _TCHAR* pszVerbDerivation = rParsedInfo.GetCmdSwitchesObject().GetVerbDerivation(); if ( rParsedInfo.GetCmdSwitchesObject().GetMethDetMap().empty()) { DisplayMessage(*theIterator, CP_OEMCP, TRUE, TRUE); rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ALIAS_VERB); retCode = PARSER_ERROR; bContinue = FALSE; } else if ( pszVerbDerivation == NULL ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_VERB_DERV_NOT_AVAIL_IN_ALIAS); retCode = PARSER_ERROR; bContinue = FALSE; } else if ( vtVerbType == CLASSMETHOD ) { if (!rParsedInfo.GetCmdSwitchesObject().SetMethodName( pszVerbDerivation)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; bContinue = FALSE; } } else if ( vtVerbType == STDVERB ) { (*theIterator) = pszVerbDerivation; // Parse and interpret the verb tokens that follow // Handled for /verb => verb. retCode = ParseVerbInfo(cvTokens,theIterator,rParsedInfo); if (retCode == PARSER_EXECCOMMAND) { if (GetNextToken(cvTokens, theIterator)) retCode = ParseVerbSwitches(cvTokens, theIterator, rParsedInfo); } bContinue =FALSE; } } } else { if (!ObtainClassMethods(rParsedInfo)) { retCode = PARSER_ERRMSG; bContinue =FALSE; } else if (rParsedInfo.GetCmdSwitchesObject().GetMethDetMap().empty()) { DisplayMessage(*theIterator, CP_OEMCP, TRUE, TRUE); rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CLASS_METHOD); retCode = PARSER_ERROR; bContinue =FALSE; } } // Move to next token if ( bContinue == TRUE && !GetNextToken(cvTokens, theIterator) ) { // indicates method with no parameters retCode = PARSER_EXECCOMMAND; bContinue =FALSE; } if (bContinue) { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, AliasVerb, IDS_E_INVALID_EXPRESSION, LEVEL_TWO); if (retCode == PARSER_CONTINUE) // To facilitate ParseVerbSwitches to continue theIterator = theIterator - 2; else if (retCode == PARSER_DISPHELP) { rParsedInfo.GetCmdSwitchesObject().GetMethDetMap(). clear(); if(m_bAliasName) { if (FAILED(m_CmdAlias.ObtainAliasVerbDetails( rParsedInfo))) retCode = PARSER_ERRMSG; } else if (!ObtainClassMethods(rParsedInfo)) retCode = PARSER_ERRMSG; } } else { BOOL bNamedParamList; // Check for NamedParamList or UnnamedParamList. if ( (theIterator + 1) < cvTokens.end() && CompareTokens(*(theIterator + 1), CLI_TOKEN_EQUALTO ) ) { retCode = ParseSETorCREATEOrNamedParamInfo(cvTokens, theIterator, rParsedInfo, CALLVerb); if ( retCode == PARSER_EXECCOMMAND ) retCode = ValidateVerbOrMethodParams(rParsedInfo); bNamedParamList = TRUE; } else { retCode = ParseUnnamedParamList(cvTokens, theIterator, rParsedInfo); bNamedParamList = FALSE; } rParsedInfo.GetCmdSwitchesObject().SetNamedParamListFlag( bNamedParamList); } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseSETorCREATEVerb Synopsis :This function parses the tokens following SET|CREATE verb and updates the info in CParsedInfo object passed to it. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object HELPTYPE - helpType Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseSETorCREATEVerb(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseSETorCREATEVerb(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, HELPTYPE helpType) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; try { retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, helpType, IDS_E_INCOMPLETE_COMMAND); if (retCode == PARSER_CONTINUE) { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, helpType, IDS_E_INVALID_COMMAND, LEVEL_ONE); if (retCode == PARSER_DISPHELP) { if (m_bAliasName) { if (FAILED(m_CmdAlias. ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassProperties(rParsedInfo)) retCode = PARSER_ERRMSG; } } } else retCode = ParseSETorCREATEOrNamedParamInfo(cvTokens, theIterator, rParsedInfo, helpType); } } catch(_com_error& e) { retCode = PARSER_ERROR; _com_issue_error(e.Error()); } return retCode; } /*---------------------------------------------------------------------------- Name :ParseGETVerb Synopsis :This function parses the tokens following the GET verb and updates the info in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseGETVerb(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseGETVerb(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { BOOL bPropList = FALSE; RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; _TCHAR *pszNewEntry = NULL; // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // GET format | switches not specified. retCode = PARSER_EXECCOMMAND; } else { BOOL bClass = FALSE; if(IsClassOperation(rParsedInfo)) { bClass = TRUE; } if(!bClass) { // Process the property list specified if (!IsOption(*theIterator)) { bPropList = TRUE; // Obtain the list of properties specified. while (TRUE) { // Add the property to the property vector of the // CCommandSwitches object if(!rParsedInfo.GetCmdSwitchesObject(). AddToPropertyList(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_LIST_FAILURE); bPropList = FALSE; bContinue = FALSE; retCode = PARSER_ERROR; break; } // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // set the return code as PARSER_EXECCOMMAND // if no more tokens are present retCode = PARSER_EXECCOMMAND; bContinue = FALSE; break; } // Check for the presence of ',' token if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if (!GetNextToken(cvTokens, theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_EXPRESSION); retCode = PARSER_ERROR; bContinue = FALSE; break; } } else break; } } } if (bContinue) { // alias|class get param1,param2... /getswitches if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, GETVerb, IDS_E_INVALID_EXPRESSION, LEVEL_TWO); if (retCode != PARSER_CONTINUE) { if (retCode == PARSER_DISPHELP) { if (m_bAliasName) { if (FAILED(m_CmdAlias. ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassProperties(rParsedInfo)) retCode = PARSER_ERRMSG; } } bContinue = FALSE; } if (bContinue) retCode = ParseGETSwitches(cvTokens, theIterator, rParsedInfo); } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_GET_EXPRESSION); retCode = PARSER_ERROR; } } // If property names are specified then replace them with their // derivations. if ( retCode == PARSER_EXECCOMMAND ) { if (m_bAliasName) { if (FAILED(m_CmdAlias. ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassProperties(rParsedInfo)) retCode = PARSER_ERRMSG; } PROPDETMAP pdmPropDet = rParsedInfo.GetCmdSwitchesObject(). GetPropDetMap(); PROPDETMAP::iterator itrPropDet = NULL; CHARVECTOR cvPropsSpecified = rParsedInfo. GetCmdSwitchesObject(). GetPropertyList(); CHARVECTOR::iterator theIterator = NULL; CHARVECTOR cvPropDerivations; for ( theIterator = cvPropsSpecified.begin(); theIterator != cvPropsSpecified.end(); theIterator++ ) { try { BOOL bFind = Find(pdmPropDet, *theIterator, itrPropDet); _bstr_t bstrPropDerivation; if ( bFind ) bstrPropDerivation = _bstr_t( (*itrPropDet).second.Derivation); else bstrPropDerivation = _bstr_t(*theIterator); _TCHAR* pszNewEntry = new _TCHAR[bstrPropDerivation.length()+1]; if (pszNewEntry == NULL) _com_issue_error(WBEM_E_OUT_OF_MEMORY); lstrcpy(pszNewEntry, bstrPropDerivation); cvPropDerivations.push_back(pszNewEntry); } catch(_com_error& e) { SAFEDELETE(pszNewEntry); retCode = PARSER_ERROR; CleanUpCharVector(cvPropDerivations); _com_issue_error(e.Error()); } } rParsedInfo.GetCmdSwitchesObject().ClearPropertyList(); for ( theIterator = cvPropDerivations.begin(); theIterator != cvPropDerivations.end(); theIterator++ ) { rParsedInfo.GetCmdSwitchesObject(). AddToPropertyList(*theIterator); } CleanUpCharVector(cvPropDerivations); } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseLISTVerb Synopsis :This function parses the tokens following the LIST verb and updates the info in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseLISTVerb(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseLISTVerb(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; HRESULT hr = S_OK; BOOL bSetDefaultFormat = TRUE; // Set the default LIST format if(!rParsedInfo.GetCmdSwitchesObject().SetListFormat(CLI_TOKEN_FULL)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; } if (bContinue) { // If specified. if (GetNextToken(cvTokens, theIterator)) { // Check for LIST formats (LIST formats are not preceded by '/') if (!IsOption(*theIterator)) { // If token is not followed by "/" or "-" then it is LIST format. if(!rParsedInfo.GetCmdSwitchesObject(). SetListFormat(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; bContinue = FALSE; } // If list format explicitly specified then do not set // default format. bSetDefaultFormat = FALSE; // Get all the properties from alias definition for the format // specified if (bContinue) { // no more tokens are present. if (!GetNextToken(cvTokens, theIterator)) { bContinue = FALSE; retCode = PARSER_EXECCOMMAND; } else rParsedInfo.GetHelpInfoObject().SetHelp( LISTSwitchesOnly, TRUE); } } if (bContinue == TRUE ) { if ( IsOption(*theIterator) ) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, LISTVerb, IDS_E_INVALID_EXPRESSION, LEVEL_TWO); // If more tokens are present. if (retCode == PARSER_CONTINUE) { BOOL bFormatSwitchSpecified; // Parse for LIST switches. retCode = ParseLISTSwitches(cvTokens, theIterator, rParsedInfo, bFormatSwitchSpecified); // If /format is specified in list switches then // do not set default format. if ( bFormatSwitchSpecified == TRUE ) bSetDefaultFormat = FALSE; } else if ( retCode == PARSER_DISPHELP ) { if ( rParsedInfo.GetHelpInfoObject(). GetHelp(LISTSwitchesOnly) == FALSE ) { hr = m_CmdAlias.PopulateAliasFormatMap( rParsedInfo); ONFAILTHROWERROR(hr); } } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_LIST_EXPRESSION); retCode = PARSER_ERROR; } } } } if (retCode == PARSER_EXECCOMMAND) { // Obtain all properties from alias definition if (!m_CmdAlias.ObtainAliasFormat(rParsedInfo)) { // If failed to obtain the alias properties return PARSER_ERROR if (rParsedInfo.GetCmdSwitchesObject().GetErrataCode() == 0) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_LIST_FORMAT); } retCode = PARSER_ERROR; } if ( bSetDefaultFormat == TRUE ) { rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); XSLTDET xdXSLTDet; g_wmiCmd.GetFileFromKey(CLI_TOKEN_TABLE, xdXSLTDet.FileName); if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseASSOCVerb Synopsis :This function parses the tokens following the ASSOC verb and updates the info in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseASSOCVerb(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseASSOCVerb(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; // Move to next token if (!GetNextToken(cvTokens, theIterator)) { retCode = PARSER_EXECCOMMAND; } // If it is followed by a ":" is missing rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ASSOC_FORMATSPECIFIER); retCode = PARSER_ERROR; } else if (IsOption(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ASSOC_FORMATSPECIFIER); retCode = PARSER_ERROR; } else { rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); BOOL bFrameXSLFile = TRUE; XSLTDET xdXSLTDet; xdXSLTDet.FileName = *theIterator; if(!g_wmiCmd.GetFileFromKey(*theIterator, xdXSLTDet.FileName)) bFrameXSLFile = FALSE; if ( bFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLTDet); } GetNextToken(cvTokens, theIterator); rParsedInfo.GetHelpInfoObject().SetHelp(ASSOCSwitchesOnly, TRUE); }///END for check of ":" if ( retCode == PARSER_EXECCOMMAND && theIterator < cvTokens.end() ) { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, ASSOCVerb, IDS_E_INVALID_COMMAND,LEVEL_TWO); // If more tokens are present. if (retCode == PARSER_CONTINUE) { //Parse for Assoc switches. retCode = ParseAssocSwitches(cvTokens, theIterator, rParsedInfo); } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ASSOC_SYNTAX); retCode = PARSER_ERROR; } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseCALLVerb Synopsis :This function parses the tokens following the CALL verb and updates the info in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseCALLVerb(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseCALLVerb(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, CALLVerb, IDS_E_INCOMPLETE_COMMAND); if (retCode == PARSER_CONTINUE) { if (IsOption(*theIterator)) { retCode = IsHelp(cvTokens, theIterator, rParsedInfo, CALLVerb, IDS_E_INVALID_EXPRESSION, LEVEL_TWO); if (retCode != PARSER_CONTINUE) { if (retCode == PARSER_DISPHELP) { if(m_bAliasName) { if (FAILED(m_CmdAlias.ObtainAliasVerbDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassMethods(rParsedInfo)) retCode = PARSER_ERRMSG; } } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_VERB_OR_METHOD_NOT_SPECIFIED); retCode = PARSER_ERROR; /* theIterator = theIterator-2; retCode = PARSER_EXECCOMMAND; */ } } else { retCode = ParseMethodInfo(cvTokens, theIterator, rParsedInfo); if (retCode == PARSER_CONTINUE) retCode = PARSER_EXECCOMMAND; } } else if (retCode == PARSER_DISPHELP) { if(m_bAliasName) { if (FAILED(m_CmdAlias.ObtainAliasVerbDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassMethods(rParsedInfo)) retCode = PARSER_ERRMSG; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseGlobalSwitches Synopsis :This function does the parsing and interprets if command has global switches specified in it. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseGlobalSwitches(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseGlobalSwitches(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo &rParsedInfo) { RETCODE retCode = PARSER_CONTINUE; BOOL bContinue = TRUE; BOOL bPassFlag = FALSE; BOOL bUserFlag = FALSE; BOOL bOpenOutFileInWriteMode = FALSE; while (TRUE) { // Move to next token retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, IDS_E_INVALID_GLOBAL_SWITCH); if (retCode != PARSER_CONTINUE) break; // Check for the presence of NAMESPACE global switch if (CompareTokens(*theIterator, CLI_TOKEN_NAMESPACE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INCORRECT_NAMESPACE, rParsedInfo, IDS_E_INVALID_NAMESPACE_SYNTAX, Namespace); if (retCode == PARSER_CONTINUE) { LONG lPresNamespaceLen = lstrlen(rParsedInfo.GetGlblSwitchesObject().GetNameSpace()); LONG lUserInputNamespaceLen = lstrlen(*theIterator); _TCHAR *pszNamespaceToBeUpdated = new _TCHAR[ // +2 for '\' and '\0' lUserInputNamespaceLen + lPresNamespaceLen + 2]; if (pszNamespaceToBeUpdated == NULL) throw OUT_OF_MEMORY; lstrcpy(pszNamespaceToBeUpdated, rParsedInfo.GetGlblSwitchesObject().GetNameSpace()); FrameNamespace(*theIterator, pszNamespaceToBeUpdated); if (!CompareTokens(pszNamespaceToBeUpdated, CLI_TOKEN_NULL) && rParsedInfo.GetGlblSwitchesObject(). GetInteractiveStatus() == TRUE) { if (!ValidateNodeOrNS(pszNamespaceToBeUpdated, FALSE)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_NAMESPACE); retCode = PARSER_ERROR; break; } } if(!rParsedInfo.GetGlblSwitchesObject(). SetNameSpace(pszNamespaceToBeUpdated)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } SAFEDELETE(pszNamespaceToBeUpdated); } else break; } // Check for the presence of ROLE global switch else if (CompareTokens(*theIterator, CLI_TOKEN_ROLE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_ROLE, rParsedInfo, IDS_E_INVALID_ROLE_SYNTAX, Role); if (retCode == PARSER_CONTINUE) { LONG lPresRoleLen = lstrlen(rParsedInfo.GetGlblSwitchesObject().GetRole()); LONG lUserInputRoleLen = lstrlen(*theIterator); _TCHAR *pszRoleToBeUpdated = new _TCHAR[ // +2 one for '\' and one for '\0' lPresRoleLen + lUserInputRoleLen + 2]; if (pszRoleToBeUpdated == NULL) throw OUT_OF_MEMORY; lstrcpy(pszRoleToBeUpdated, rParsedInfo.GetGlblSwitchesObject().GetRole()); FrameNamespace(*theIterator, pszRoleToBeUpdated); if (!CompareTokens(pszRoleToBeUpdated, CLI_TOKEN_NULL) && rParsedInfo.GetGlblSwitchesObject(). GetInteractiveStatus() == TRUE) { if (!ValidateNodeOrNS(pszRoleToBeUpdated, FALSE)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ROLE); retCode = PARSER_ERROR; break; } } if(!rParsedInfo.GetGlblSwitchesObject(). SetRole(pszRoleToBeUpdated)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } SAFEDELETE(pszRoleToBeUpdated); } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_NODE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_MACHINE_NAME, rParsedInfo, IDS_E_INVALID_NODE_SYNTAX, Node); if (retCode == PARSER_CONTINUE) { BOOL bBreakOuterLoop = FALSE; BOOL bGetValidNode = FALSE; BOOL bNodeListCleared = FALSE; while ( TRUE ) { try { CHString chsNodeName(*theIterator); chsNodeName.TrimLeft(); chsNodeName.TrimRight(); lstrcpy(*theIterator, (LPCWSTR) chsNodeName); } catch(CHeap_Exception) { _com_issue_error(WBEM_E_OUT_OF_MEMORY); } if ( *theIterator[0] == _T('@') ) { retCode = ParseNodeListFile(cvTokens, theIterator, rParsedInfo); if ( retCode != PARSER_CONTINUE ) { bBreakOuterLoop = TRUE; break; } } // If interactive mode then check for the validity of // nodes if(rParsedInfo.GetGlblSwitchesObject(). GetInteractiveStatus()) { BOOL bNodeExist = TRUE; if ( rParsedInfo.GetGlblSwitchesObject(). GetFailFast() == TRUE ) { bNodeExist = IsFailFastAndNodeExist(rParsedInfo, *theIterator); } else bNodeExist = ValidateNodeOrNS(*theIterator, TRUE); if( bNodeExist == FALSE) { // Display error message for invalid node DisplayString(IDS_E_INVALID_NODE, ::GetOEMCP(), *theIterator, TRUE); if ( !GetNextToken(cvTokens, theIterator)) { // If no more tokens are present then stop // further processing bBreakOuterLoop = TRUE; break; } else { // If multiple nodes are defined then check // tokens if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { // If invalid node syntax given then report // error if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_NODE_SYNTAX); retCode=PARSER_ERROR; bBreakOuterLoop = TRUE; break; } else if ( IsOption (*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_NODE_SYNTAX); retCode=PARSER_ERROR; bBreakOuterLoop = TRUE; break; } //Skip adding this invalid node to node list continue; } else { // If no more node present theIterator--; break; } } } else // Set flag for valid node bGetValidNode = TRUE; // If valid node(s) are present and list is not // already cleared then clear it if(bGetValidNode && !bNodeListCleared) { if (!rParsedInfo.GetGlblSwitchesObject(). ClearNodesList()) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } bNodeListCleared = TRUE; } } else if( bNodeListCleared == FALSE ) { // If not in interactive mode then clear // previous node list if (!rParsedInfo.GetGlblSwitchesObject(). ClearNodesList()) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } bNodeListCleared = TRUE; } if (rParsedInfo.GetGlblSwitchesObject(). AddToNodesList(*theIterator)) { if ( GetNextToken(cvTokens, theIterator) ) { if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_NODE_SYNTAX); retCode=PARSER_ERROR; bBreakOuterLoop = TRUE; } else if ( IsOption (*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_NODE_SYNTAX); retCode=PARSER_ERROR; bBreakOuterLoop = TRUE; } } else { theIterator--; break; } } else bBreakOuterLoop = TRUE; } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; bBreakOuterLoop = FALSE; } if ( bBreakOuterLoop == TRUE ) break; } if ( bBreakOuterLoop == TRUE ) break; } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_IMPLEVEL)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_IMP_LEVEL, rParsedInfo, IDS_E_INVALID_IMP_LEVEL_SYNTAX, Level); if (retCode == PARSER_CONTINUE) { if (!rParsedInfo.GetGlblSwitchesObject(). SetImpersonationLevel(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_IMP_LEVEL); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_AUTHLEVEL)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_AUTH_LEVEL, rParsedInfo, IDS_E_INVALID_AUTH_LEVEL_SYNTAX, AuthLevel); if (retCode == PARSER_CONTINUE) { if (!rParsedInfo.GetGlblSwitchesObject(). SetAuthenticationLevel(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_AUTH_LEVEL); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_LOCALE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_LOCALE, rParsedInfo, IDS_E_INVALID_LOCALE_SYNTAX, Locale); if (retCode == PARSER_CONTINUE) { if(!rParsedInfo.GetGlblSwitchesObject().SetLocale(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_PRIVILEGES)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_PRIVILEGES_OPTION, rParsedInfo, IDS_E_INVALID_PRIVILEGES_SYNTAX, Privileges); if (retCode == PARSER_CONTINUE) { if (CompareTokens(*theIterator, CLI_TOKEN_ENABLE)) rParsedInfo.GetGlblSwitchesObject().SetPrivileges(TRUE); else if (CompareTokens(*theIterator, CLI_TOKEN_DISABLE)) rParsedInfo.GetGlblSwitchesObject().SetPrivileges(FALSE); else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PRIVILEGES_OPTION); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_TRACE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_TRACE_OPTION, rParsedInfo, IDS_E_INVALID_TRACE_SYNTAX, Trace); if (retCode == PARSER_CONTINUE) { if (CompareTokens(*theIterator, CLI_TOKEN_ON)) rParsedInfo.GetGlblSwitchesObject().SetTraceMode(TRUE); else if (CompareTokens(*theIterator, CLI_TOKEN_OFF)) rParsedInfo.GetGlblSwitchesObject().SetTraceMode(FALSE); else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_TRACE_OPTION); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_RECORD)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_RECORD_PATH, rParsedInfo, IDS_E_INVALID_RECORD_SYNTAX, RecordPath); if (retCode == PARSER_CONTINUE) { // TRUE for getting output file name. _TCHAR* pszOutputFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(TRUE); if ( pszOutputFileName != NULL && CompareTokens(*theIterator, pszOutputFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_RECORD_FILE_ALREADY_OPEN_FOR_OUTPUT); retCode = PARSER_ERROR; break; } // FALSE for getting append file name. _TCHAR* pszAppendFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(FALSE); if ( pszAppendFileName != NULL && CompareTokens(*theIterator, pszAppendFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_RECORD_FILE_ALREADY_OPEN_FOR_APPEND); retCode = PARSER_ERROR; break; } // /record:"" indicates stop logging. if (!CompareTokens(*theIterator, CLI_TOKEN_NULL)) { if ( IsValidFile(*theIterator) == FALSE ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FILENAME); retCode = PARSER_ERROR; break; } } if(!rParsedInfo.GetGlblSwitchesObject(). SetRecordPath(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_INTERACTIVE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_INTERACTIVE_OPTION, rParsedInfo, IDS_E_INVALID_INTERACTIVE_SYNTAX, Interactive); if (retCode == PARSER_CONTINUE) { if (CompareTokens(*theIterator, CLI_TOKEN_ON)) { if (rParsedInfo.GetGlblSwitchesObject().GetInteractiveStatus()) { rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_INTERACTIVE_ALREADY_SET); } else rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_INTERACTIVE_SET); rParsedInfo.GetGlblSwitchesObject().SetInteractiveMode(TRUE); } else if (CompareTokens(*theIterator, CLI_TOKEN_OFF)) { if (!rParsedInfo.GetGlblSwitchesObject().GetInteractiveStatus()) { rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_INTERACTIVE_ALREADY_RESET); } else rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_INTERACTIVE_RESET); rParsedInfo.GetGlblSwitchesObject().SetInteractiveMode(FALSE); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_INTERACTIVE_OPTION); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_FAILFAST)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_FAILFAST_OPTION, rParsedInfo, IDS_E_INVALID_FAILFAST_SYNTAX, FAILFAST); if (retCode == PARSER_CONTINUE) { if (CompareTokens(*theIterator, CLI_TOKEN_ON)) { if (rParsedInfo.GetGlblSwitchesObject().GetFailFast()) { rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_FAILFAST_ALREADY_SET); } else rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_FAILFAST_SET); rParsedInfo.GetGlblSwitchesObject().SetFailFast(TRUE); } else if (CompareTokens(*theIterator, CLI_TOKEN_OFF)) { if (!rParsedInfo.GetGlblSwitchesObject().GetFailFast()) { rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_FAILFAST_ALREADY_RESET); } else rParsedInfo.GetCmdSwitchesObject(). SetInformationCode(IDS_I_FAILFAST_RESET); rParsedInfo.GetGlblSwitchesObject().SetFailFast(FALSE); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FAILFAST_OPTION); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_USER)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_USER_ID, rParsedInfo, IDS_E_INVALID_USER_SYNTAX, User); if (retCode == PARSER_CONTINUE) { if(!rParsedInfo.GetGlblSwitchesObject().SetUser(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } bUserFlag = TRUE; } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_PASSWORD)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_PASSWORD, rParsedInfo, IDS_E_INVALID_PASSWORD_SYNTAX, Password); if (retCode == PARSER_CONTINUE) { if(!rParsedInfo.GetGlblSwitchesObject().SetPassword(*theIterator)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode=PARSER_OUTOFMEMORY; break; } bPassFlag = TRUE; } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_OUTPUT)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_OUTPUT_OPTION, rParsedInfo, IDS_E_INVALID_OUTPUT_SYNTAX, OUTPUT); if (retCode == PARSER_CONTINUE) { rParsedInfo.GetCmdSwitchesObject().SetOutputSwitchFlag(TRUE); if (CompareTokens(*theIterator, CLI_TOKEN_STDOUT)) { // TRUE for setting output file. rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption( STDOUT, TRUE); rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(NULL, TRUE); } else if (CompareTokens(*theIterator, CLI_TOKEN_CLIPBOARD)) { // TRUE for setting output file. rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption( CLIPBOARD, TRUE); rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(NULL, TRUE); } else if ( CompareTokens(*theIterator, CLI_TOKEN_NULL)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FILENAME); retCode = PARSER_ERROR; break; } else { // FALSE for getting append file name. _TCHAR* pszAppendFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(FALSE); if ( pszAppendFileName != NULL && CompareTokens(*theIterator, pszAppendFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_OUTPUT_FILE_ALREADY_OPEN_FOR_APPEND); retCode = PARSER_ERROR; break; } _TCHAR* pszRecordFileName = rParsedInfo. GetGlblSwitchesObject(). GetRecordPath(); if ( pszRecordFileName != NULL && CompareTokens(*theIterator, pszRecordFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_OUTPUT_FILE_ALREADY_OPEN_FOR_RECORD); retCode = PARSER_ERROR; break; } if ( CloseOutputFile() == TRUE ) { // TRUE for getting output file name. _TCHAR* pszOutputFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(TRUE); if ( pszOutputFileName == NULL || ( pszOutputFileName != NULL && !CompareTokens(*theIterator, pszOutputFileName))) { retCode = IsValidFile(*theIterator); if ( retCode == PARSER_ERROR ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FILENAME); break; } else if ( retCode == PARSER_ERRMSG ) break; } } else { retCode = PARSER_ERRMSG; break; } // TRUE for setting output file. if(!rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(*theIterator, TRUE)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode=PARSER_OUTOFMEMORY; break; } rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption(FILEOUTPUT, TRUE); bOpenOutFileInWriteMode = TRUE; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_APPEND)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_APPEND_OPTION, rParsedInfo, IDS_E_INVALID_APPEND_SYNTAX, APPEND); if (retCode == PARSER_CONTINUE) { if ( CompareTokens(*theIterator, CLI_TOKEN_STDOUT) ) { // FALSE for setting append file. rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(NULL, FALSE); rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption(STDOUT, FALSE); } else if ( CompareTokens(*theIterator, CLI_TOKEN_CLIPBOARD) ) { // FALSE for setting append file. rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(NULL, FALSE); rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption(CLIPBOARD, FALSE); } else if ( CompareTokens(*theIterator, CLI_TOKEN_NULL)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FILENAME); retCode = PARSER_ERROR; break; } else { // TRUE for getting output file name. _TCHAR* pszOutputFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(TRUE); if ( pszOutputFileName != NULL && CompareTokens(*theIterator, pszOutputFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_APPEND_FILE_ALREADY_OPEN_FOR_OUTPUT); retCode = PARSER_ERROR; break; } _TCHAR* pszRecordFileName = rParsedInfo. GetGlblSwitchesObject(). GetRecordPath(); if ( pszRecordFileName != NULL && CompareTokens(*theIterator, pszRecordFileName) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_APPEND_FILE_ALREADY_OPEN_FOR_RECORD); retCode = PARSER_ERROR; break; } if ( CloseAppendFile() == TRUE ) { // FALSE for getting append file name. _TCHAR* pszAppendFileName = rParsedInfo. GetGlblSwitchesObject(). GetOutputOrAppendFileName(FALSE); if ( pszAppendFileName == NULL || ( pszAppendFileName != NULL && !CompareTokens(*theIterator, pszAppendFileName))) { retCode = IsValidFile(*theIterator); if ( retCode == PARSER_ERROR ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_FILENAME); break; } else if ( retCode == PARSER_ERRMSG ) break; } } else { retCode = PARSER_ERRMSG; break; } // FALSE for setting append file. if (!rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFileName(*theIterator, FALSE)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); retCode = PARSER_OUTOFMEMORY; break; } rParsedInfo.GetGlblSwitchesObject().SetOutputOrAppendOption(FILEOUTPUT, FALSE); } } else break; } else if (CompareTokens(*theIterator,CLI_TOKEN_AGGREGATE)) { retCode = ValidateGlobalSwitchValue(cvTokens, theIterator, IDS_E_INVALID_AGGREGATE_OPTION, rParsedInfo, IDS_E_INVALID_AGGREGATE_SYNTAX, Aggregate); if(retCode == PARSER_CONTINUE) { if(CompareTokens(*theIterator, CLI_TOKEN_ON)) rParsedInfo.GetGlblSwitchesObject().SetAggregateFlag(TRUE); else if(CompareTokens(*theIterator, CLI_TOKEN_OFF)) rParsedInfo.GetGlblSwitchesObject().SetAggregateFlag(FALSE); else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_AGGREGATE_OPTION); retCode = PARSER_ERROR; break; } } else break; } else if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { retCode = ParseHelp(cvTokens, theIterator, rParsedInfo, TRUE); break; } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_GLOBAL_SWITCH); retCode = PARSER_ERROR; break; } // Move to next token if (!GetNextToken(cvTokens, theIterator)) // Break the loop if no more tokens are present break; // Break the loop if no more global switches are present if (!IsOption(*theIterator)) break; } if ( bUserFlag == TRUE && bPassFlag == FALSE ) rParsedInfo.GetGlblSwitchesObject().SetAskForPassFlag(TRUE); if ( rParsedInfo.GetGlblSwitchesObject().GetPassword() != NULL && rParsedInfo.GetGlblSwitchesObject().GetUser() == NULL ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_PASSWORD_WITHOUT_USER); rParsedInfo.GetGlblSwitchesObject().SetPassword(CLI_TOKEN_NULL); retCode = PARSER_ERROR; } if ( retCode == PARSER_CONTINUE && bOpenOutFileInWriteMode == TRUE ) retCode = ProcessOutputAndAppendFiles(rParsedInfo, retCode, TRUE); return retCode; } /*---------------------------------------------------------------------------- Name :ParseGETSwitches Synopsis :This function does the parsing and interprets if command has GET as the verb. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseGETSwitches(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseGETSwitches(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bContinue = TRUE; while ( retCode == PARSER_EXECCOMMAND ) { // Check for the presence of VALUE switch if (CompareTokens(*theIterator, CLI_TOKEN_VALUE)) { rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); XSLTDET xdXSLTDet; g_wmiCmd.GetFileFromKey(CLI_TOKEN_VALUE, xdXSLTDet.FileName); if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } // Check for the presence of ALL switch else if (CompareTokens(*theIterator, CLI_TOKEN_ALL)) { rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); XSLTDET xdXSLTDet; g_wmiCmd.GetFileFromKey(CLI_TOKEN_TABLE, xdXSLTDet.FileName); if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } // Check for the presence of FORMAT switch else if (CompareTokens(*theIterator, CLI_TOKEN_FORMAT)) { rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); retCode = ParseFORMATSwitch(cvTokens, theIterator, rParsedInfo); } // Check for the presence of EVERY switch else if (CompareTokens(*theIterator, CLI_TOKEN_EVERY)) { retCode = ParseEVERYSwitch(cvTokens, theIterator, rParsedInfo); } // Check for the presence of TRANSLATE switch else if (CompareTokens(*theIterator, CLI_TOKEN_TRANSLATE)) { retCode = ParseTRANSLATESwitch(cvTokens, theIterator, rParsedInfo); } // Check whether /REPEAT follows /EVERY else if (CompareTokens(*theIterator, CLI_TOKEN_REPEAT)) { if (!CompareTokens(*(theIterator-4), CLI_TOKEN_EVERY)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_I_REPEAT_EVERY_RELATED); retCode = PARSER_ERROR; break; } } // Check for the presence of HELP switch else if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { rParsedInfo.GetHelpInfoObject().SetHelp(GETSwitchesOnly, TRUE); retCode = ParseHelp(cvTokens, theIterator, GETVerb, rParsedInfo); if ( retCode == PARSER_DISPHELP ) { if (m_bAliasName) { if (FAILED(m_CmdAlias. ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassProperties(rParsedInfo)) retCode = PARSER_ERRMSG; } } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_GET_SWITCH); retCode = PARSER_ERROR; break; } if ( retCode == PARSER_EXECCOMMAND ) { if ( !GetNextToken(cvTokens, theIterator) ) break; if ( !IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; break; } if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_GET_SWITCH); retCode = PARSER_ERROR; break; } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseLISTSwitches Synopsis :This function does the parsing and interprets if command has LIST as the verb. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseLISTSwitches(cvTokens,theIterator,rParsedInfo, bFormatSwitchSpecified) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseLISTSwitches(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, BOOL& bFormatSwitchSpecified) { RETCODE retCode = PARSER_EXECCOMMAND; bFormatSwitchSpecified = FALSE; while ( retCode == PARSER_EXECCOMMAND ) { if (CompareTokens(*theIterator, CLI_TOKEN_TRANSLATE)) { retCode = ParseTRANSLATESwitch(cvTokens, theIterator, rParsedInfo); } else if (CompareTokens(*theIterator, CLI_TOKEN_EVERY)) { retCode = ParseEVERYSwitch(cvTokens, theIterator, rParsedInfo); } else if (CompareTokens(*theIterator, CLI_TOKEN_FORMAT)) { retCode = ParseFORMATSwitch(cvTokens, theIterator, rParsedInfo); bFormatSwitchSpecified = TRUE; } // Check whether /REPEAT follows /EVERY else if (CompareTokens(*theIterator, CLI_TOKEN_REPEAT)) { if (!CompareTokens(*(theIterator-4), CLI_TOKEN_EVERY)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_I_REPEAT_EVERY_RELATED); retCode = PARSER_ERROR; break; } } else if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { rParsedInfo.GetHelpInfoObject().SetHelp(LISTSwitchesOnly, TRUE); retCode = ParseHelp(cvTokens, theIterator, LISTVerb, rParsedInfo); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_LIST_SWITCH); retCode = PARSER_ERROR; break; } if ( retCode == PARSER_EXECCOMMAND ) { if ( !GetNextToken(cvTokens, theIterator) ) break; if ( !IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; break; } if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_LIST_SWITCH); retCode = PARSER_ERROR; break; } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseSETorCREATEOrNamedParamInfo Synopsis :This function does the parsing and interprets if command has SET as the verb. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output parameters : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseSETorCREATEOrNamedParamInfo(cvTokens,theIterator,rParsedInfo, helpType) Notes :None -------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseSETorCREATEOrNamedParamInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, HELPTYPE helpType) { RETCODE retCode = PARSER_EXECCOMMAND; _TCHAR *pszProp,*pszVal; try { // Process the SET|CREATE verb related info i.e properties with new values. while (TRUE) { pszProp = NULL; pszVal = NULL; // Tokenize the expression checking for '=' pszProp = *theIterator; if ( GetNextToken(cvTokens, theIterator) && CompareTokens(*theIterator, CLI_TOKEN_EQUALTO) && GetNextToken(cvTokens, theIterator)) pszVal = *theIterator; if ((pszProp == NULL) || (pszVal == NULL)) { if ( helpType != CALLVerb && IsOption(*(theIterator+1)) && theIterator + 2 < cvTokens.end() && CompareTokens(*(theIterator+2), CLI_TOKEN_HELP) ) { theIterator++; theIterator++; retCode = ParseHelp(cvTokens, theIterator, helpType, rParsedInfo); if (retCode == PARSER_DISPHELP) { // Adding to PropertyList only for use in displaying help of // properties if(!rParsedInfo.GetCmdSwitchesObject(). AddToPropertyList(pszProp)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_LIST_FAILURE); retCode = PARSER_ERROR; break; } if (m_bAliasName) { if (FAILED(m_CmdAlias. ObtainAliasPropDetails(rParsedInfo))) retCode = PARSER_ERRMSG; } else { if (!ObtainClassProperties(rParsedInfo)) retCode = PARSER_ERRMSG; } } } else { UINT nErrID; if ( helpType == CALLVerb ) nErrID = IDS_E_INVALID_NAMED_PARAM_LIST; else nErrID = IDS_E_INVALID_ASSIGNLIST; rParsedInfo.GetCmdSwitchesObject().SetErrataCode(nErrID); retCode = PARSER_ERROR; } break; } // Unquote the strings UnQuoteString(pszProp); UnQuoteString(pszVal); // Add to the list of parameters if(!rParsedInfo.GetCmdSwitchesObject(). AddToParameterMap(_bstr_t(pszProp), _bstr_t(pszVal))) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PARAM_MAP_FAILURE); retCode = PARSER_ERROR; break; } // Adding to PropertyList only for use in displaying help of // properties if(!rParsedInfo.GetCmdSwitchesObject(). AddToPropertyList(pszProp)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_LIST_FAILURE); retCode = PARSER_ERROR; break; } // Get the next token if (GetNextToken(cvTokens, theIterator)) { // If option (i.e either '/' or '-') specified. if (IsOption(*theIterator)) { theIterator--; break; } else { if ( helpType != CALLVerb ) { // check for the presence of ',' if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if (!GetNextToken(cvTokens, theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_ASSIGNLIST); retCode = PARSER_ERROR; break; } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_ASSIGNLIST); retCode = PARSER_ERROR; break; } } } } else { retCode = PARSER_EXECCOMMAND; break; } } } catch(_com_error& e) { retCode = PARSER_ERROR; _com_issue_error(e.Error()); } return retCode; } /*---------------------------------------------------------------------------- Name :ParseVerbSwitches Synopsis :This function does the parsing and interprets if command has verb switches specified. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseVerbSwitches(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseVerbSwitches(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bInvalidVerbSwitch = FALSE; // Check for the '/' | '-' token if (IsOption(*theIterator)) { // Move to next token if (!GetNextToken(cvTokens, theIterator)) bInvalidVerbSwitch = TRUE; else if (CompareTokens(*theIterator, CLI_TOKEN_INTERACTIVE)) { rParsedInfo.GetCmdSwitchesObject(). SetInteractiveMode(INTERACTIVE); _TCHAR *pszVerbName = rParsedInfo.GetCmdSwitchesObject(). GetVerbName(); BOOL bInstanceLevel = TRUE; if(CompareTokens(pszVerbName, CLI_TOKEN_CALL) || CompareTokens(pszVerbName, CLI_TOKEN_SET) || CompareTokens(pszVerbName, CLI_TOKEN_DELETE)) { if(IsClassOperation(rParsedInfo)) { bInstanceLevel = FALSE; } else { if(CompareTokens(pszVerbName, CLI_TOKEN_CALL)) { if ( rParsedInfo.GetCmdSwitchesObject(). GetAliasName() != NULL ) { if (rParsedInfo.GetCmdSwitchesObject(). GetWhereExpression() == NULL) { bInstanceLevel = FALSE; } else bInstanceLevel = TRUE; } else { if ((rParsedInfo.GetCmdSwitchesObject(). GetPathExpression() != NULL) && (rParsedInfo.GetCmdSwitchesObject(). GetWhereExpression() == NULL)) { bInstanceLevel = FALSE; } else bInstanceLevel = TRUE; } } else bInstanceLevel = TRUE; } } else retCode = PARSER_EXECCOMMAND; if(bInstanceLevel) { retCode = ParseVerbInteractive( cvTokens, theIterator, rParsedInfo, bInvalidVerbSwitch); } else retCode = PARSER_EXECCOMMAND; } else if (CompareTokens(*theIterator, CLI_TOKEN_NONINTERACT)) { rParsedInfo.GetCmdSwitchesObject(). SetInteractiveMode(NOINTERACTIVE); retCode = PARSER_EXECCOMMAND; } else if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { retCode = ParseHelp(cvTokens, theIterator, VERBSWITCHES, rParsedInfo); } else bInvalidVerbSwitch = TRUE; if ( GetNextToken(cvTokens, theIterator ) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; } } else bInvalidVerbSwitch = TRUE; if ( bInvalidVerbSwitch == TRUE ) { // no valid type is specified. rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_VERB_SWITCH); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :GetNextToken Synopsis :This function retrieves the next token from the token vector list, returns FALSE if no more tokens are present Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. Output Parameter(s):None Return Type :BOOL Global Variables :None Calling Syntax :GetNextToken(cvTokens,theIterator) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::GetNextToken(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator) { theIterator++; return (theIterator >= cvTokens.end()) ? FALSE : TRUE; } /*---------------------------------------------------------------------------- Name :ParsePWhereExpr Synopsis :This function does the parsing and interprets if command has Path and Where expression It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :BOOL Global Variables :None Calling Syntax :ParsePWhereExpr(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::ParsePWhereExpr(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, BOOL bIsParan) { BOOL bRet = TRUE, bContinue = FALSE; try { while (TRUE) { if ( bIsParan == TRUE && CompareTokens(*theIterator, CLI_TOKEN_RIGHT_PARAN) ) break; if ( bIsParan == FALSE && IsStdVerbOrUserDefVerb(*theIterator, rParsedInfo) ) break; if ( bIsParan == FALSE || !CompareTokens(*theIterator, CLI_TOKEN_LEFT_PARAN)) { if(!rParsedInfo.GetCmdSwitchesObject(). AddToPWhereParamsList(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PARAMS_LIST_FAILURE); bRet = FALSE; break; } bContinue = TRUE; } if (!GetNextToken(cvTokens, theIterator)) break; if ( IsOption(*theIterator) ) { bContinue = FALSE; break; } } if(bRet != FALSE && bContinue == TRUE) { CHARVECTOR theParam = rParsedInfo.GetCmdSwitchesObject(). GetPWhereParamsList(); CHARVECTOR::iterator theItr = theParam.begin(); _TCHAR pszPWhere[MAX_BUFFER] = NULL_STRING; lstrcpy(pszPWhere, CLI_TOKEN_NULL); _TCHAR* pszToken = NULL;; CHString sTemp; if ((rParsedInfo.GetCmdSwitchesObject().GetPWhereExpr() != NULL)) { sTemp.Format(rParsedInfo.GetCmdSwitchesObject(). GetPWhereExpr()); sTemp.TrimLeft(); if(!sTemp.IsEmpty()) { _bstr_t bstrPWhere = _bstr_t(rParsedInfo. GetCmdSwitchesObject(). GetPWhereExpr()); pszToken = _tcstok((WCHAR*)bstrPWhere, CLI_TOKEN_HASH); lstrcpy(pszPWhere, CLI_TOKEN_NULL); while (pszToken != NULL) { lstrcat(pszPWhere, pszToken); if (theItr != theParam.end()) { lstrcat(pszPWhere, *theItr); theItr++; } pszToken = _tcstok(NULL, CLI_TOKEN_HASH); } if(bRet != FALSE) { // Set the classpath and where expression pszToken = NULL; pszToken = _tcstok(pszPWhere, CLI_TOKEN_SPACE); if (pszToken != NULL) { if (CompareTokens(CLI_TOKEN_FROM, pszToken)) { pszToken = _tcstok(NULL, CLI_TOKEN_SPACE); if (pszToken != NULL) { if(!rParsedInfo.GetCmdSwitchesObject(). SetClassPath(pszToken)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); bRet = FALSE; } } if(bRet != FALSE) pszToken = _tcstok(NULL, CLI_TOKEN_SPACE); } if (CompareTokens(CLI_TOKEN_WHERE, pszToken)) { pszToken = _tcstok(NULL, CLI_TOKEN_NULL); if (pszToken != NULL) { if(!rParsedInfo.GetCmdSwitchesObject(). SetWhereExpression(pszToken)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); bRet = FALSE; } } } } } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_PWHERE_UNDEF); bRet = FALSE; } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_PWHERE_UNDEF); bRet = FALSE; } } if (!bContinue && bIsParan) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_PWHERE_EXPR); bRet = FALSE; } } catch(_com_error& e) { bRet = FALSE; _com_issue_error(e.Error()); } catch(CHeap_Exception) { bRet = FALSE; _com_issue_error(WBEM_E_OUT_OF_MEMORY); } return bRet; } /*---------------------------------------------------------------------------- Name :ExtractClassNameandWhereExpr Synopsis :This function takes the input as a path expression and extracts the Class and Where expression part from the path expression. Type :Member Function Input Parameter(s): pszPathExpr - the path expression rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :BOOL Global Variables :None Calling Syntax :ParsePWhereExpr(cvTokens,theIterator) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::ExtractClassNameandWhereExpr(_TCHAR* pszPathExpr, CParsedInfo& rParsedInfo) { // Frame the class name and where expression based on the object path BOOL bRet = TRUE; _TCHAR* pszToken = NULL; BOOL bFirst = TRUE; _TCHAR pszWhere[MAX_BUFFER] = NULL_STRING; lstrcpy(pszWhere, CLI_TOKEN_NULL); if (pszPathExpr == NULL) bRet = FALSE; try { if ( bRet == TRUE ) { lstrcpy(pszWhere, CLI_TOKEN_NULL); pszToken = _tcstok(pszPathExpr, CLI_TOKEN_DOT); if (pszToken != NULL) { if(!rParsedInfo.GetCmdSwitchesObject().SetClassPath(pszToken)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); bRet = FALSE; } } if(bRet != FALSE) { while (pszToken != NULL) { pszToken = _tcstok(NULL, CLI_TOKEN_COMMA); if (pszToken != NULL) { if (!bFirst) lstrcat(pszWhere, CLI_TOKEN_AND); lstrcat(pszWhere, pszToken); bFirst = FALSE; } else break; } if (lstrlen(pszWhere)) { if(!rParsedInfo.GetCmdSwitchesObject().SetWhereExpression(pszWhere)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(OUT_OF_MEMORY); bRet = FALSE; } } } } } catch(...) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_PATH); bRet = FALSE; } return bRet; } /*---------------------------------------------------------------------------- Name :GetNextToken Synopsis :This function retrieves the next token from the token vector list, returns enumerated return code depending upon the context. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object helpType - enumerated help type uErrataCode - error string ID. Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :GetNextToken(cvTokens, theIterator, rParsedInfo, helpType, uErrataCode) Notes :overloaded function ----------------------------------------------------------------------------*/ RETCODE CParserEngine::GetNextToken(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, HELPTYPE helpType, UINT uErrataCode) { RETCODE retCode = PARSER_CONTINUE; // Move to next token theIterator++; // If no more tokens are present if (theIterator >= cvTokens.end()) { // If interactive mode is set if (rParsedInfo.GetGlblSwitchesObject().GetInteractiveStatus()) { rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(TRUE); rParsedInfo.GetHelpInfoObject().SetHelp(helpType, TRUE); retCode = PARSER_DISPHELP; } else { // PARSER_ERROR if no more tokens are present. rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(uErrataCode); retCode = PARSER_ERROR; } } return retCode; } /*---------------------------------------------------------------------------- Name :GetNextToken Synopsis :This function retrieves the next token from the token vector list, returns enumerated return code depending upon the context. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object uErrataCode - error string ID. Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :GetNextToken(cvTokens, theIterator, rParsedInfo, uErrataCode) Notes :overloaded function ----------------------------------------------------------------------------*/ RETCODE CParserEngine::GetNextToken(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, UINT uErrataCode) { RETCODE retCode = PARSER_CONTINUE; // Move to next token theIterator++; // If no more tokens are present if (theIterator >= cvTokens.end()) { // PARSER_ERROR if no more tokens are present rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(uErrataCode); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :IsHelp Synopsis :This function retrieves the next token from the token vector list, checks if it is '?' and returns enumerated return code depending upon the context. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object helpType - enumerated help type uErrataCode - error string ID. tokenLevel - token level Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :IsHelp(cvTokens, theIterator, rParsedInfo, helpType, uErrataCode, tokenLevel) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::IsHelp(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, HELPTYPE helpType, UINT uErrataCode, TOKENLEVEL tokenLevel) { BOOL bContinue = TRUE; RETCODE retCode = PARSER_CONTINUE; // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // Set the retCode as PARSER_ERROR if no more tokens are present. rParsedInfo.GetCmdSwitchesObject().SetErrataCode(uErrataCode); retCode = PARSER_ERROR; } else { // Is '?' if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) retCode = ParseHelp(cvTokens, theIterator, helpType, rParsedInfo); else { // If LEVEL_ONE token then only allowed is /?, other // switches are invalid. if (tokenLevel == LEVEL_ONE) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode(uErrataCode); retCode = PARSER_ERROR; } else retCode = PARSER_CONTINUE; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseHelp Synopsis :This function takes care of identifying the appropriate help informtion to be displayed using the HELPTYPE Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object bGlobalHelp - global help flag Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ParseHelp(cvTokens, theIterator, rParsedInfo) Notes :overloaded fucntion ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseHelp(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, BOOL bGlobalHelp) { BOOL bContinue = TRUE; RETCODE retCode = PARSER_CONTINUE; // Move to next token (if no more tokens are present) if (!GetNextToken(cvTokens, theIterator)) { retCode = PARSER_DISPHELP; rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(TRUE); // Check for "/?" if (((theIterator - 2) == cvTokens.begin()) || bGlobalHelp) { rParsedInfo.GetHelpInfoObject().SetHelp(GlblAllInfo, TRUE); if(SUCCEEDED(m_CmdAlias.ConnectToAlias(rParsedInfo,m_pIWbemLocator))) { if(FAILED(m_CmdAlias.ObtainAliasFriendlyNames(rParsedInfo))) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject().FreeCOMError(); } } // Check for the presence of the ':" else if (CompareTokens(*theIterator, CLI_TOKEN_COLON)) { // Move to next token if (!GetNextToken(cvTokens, theIterator, rParsedInfo, IDS_E_INVALID_HELP_OPTION)) // Set the retCode to PARSER_ERROR if no more tokens are specified. { retCode = PARSER_ERROR; } else { if (CompareTokens(*theIterator, CLI_TOKEN_BRIEF)) rParsedInfo.GetGlblSwitchesObject().SetHelpOption(HELPBRIEF); else if (CompareTokens(*theIterator, CLI_TOKEN_FULL)) rParsedInfo.GetGlblSwitchesObject().SetHelpOption(HELPFULL); else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_HELP_OPTION); retCode = PARSER_ERROR; } if ( retCode != PARSER_ERROR ) { if ( GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; } else { retCode = PARSER_DISPHELP; rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(TRUE); // Check for "/?:(BRIEF|FULL) if (((theIterator - 3) == cvTokens.begin()) || bGlobalHelp) { rParsedInfo.GetHelpInfoObject(). SetHelp(GlblAllInfo, TRUE); if(SUCCEEDED(m_CmdAlias.ConnectToAlias (rParsedInfo,m_pIWbemLocator))) { if(FAILED(m_CmdAlias.ObtainAliasFriendlyNames (rParsedInfo))) { retCode = PARSER_ERRMSG; } } else rParsedInfo.GetCmdSwitchesObject().FreeCOMError(); } } } } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :ParseHelp Synopsis :This function takes care of identifying the appropriate help informtion to be displayed using the HELPTYPE Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. htHelp - help type rParsedInfo - reference to CParsedInfo class object bGlobalHelp - global help flag Output Parameter(s): rParsedInfo - ref. to CParsedInfo object Return Type :RETCODE Global Variables :None Calling Syntax :ParseHelp(cvTokens, theIterator, htHelp, rParsedInfo) Notes :overloaded fucntion ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseHelp(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, HELPTYPE htHelp, CParsedInfo& rParsedInfo, BOOL bGlobalHelp) { rParsedInfo.GetHelpInfoObject().SetHelp(htHelp, TRUE); return ParseHelp(cvTokens, theIterator, rParsedInfo, bGlobalHelp); } /*---------------------------------------------------------------------------- Name :ObtainClassProperties Synopsis :This function obtains the information about the available properties for a given WMI class Type :Member Function Input Parameter(s): rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :BOOL Global Variables :None Calling Syntax :ObtainClassProperties(rParsedInfo) Notes :If bCheckWritePropsAvail == TRUE then functions checks for availibilty of properties. ----------------------------------------------------------------------------*/ BOOL CParserEngine::ObtainClassProperties(CParsedInfo& rParsedInfo, BOOL bCheckWritePropsAvail) { HRESULT hr = S_OK; IWbemClassObject* pIObject = NULL; SAFEARRAY* psaNames = NULL; BSTR bstrPropName = NULL; BOOL bRet = TRUE; BOOL bTrace = FALSE; CHString chsMsg; ERRLOGOPT eloErrLogOpt = NO_LOGGING; DWORD dwThreadId = GetCurrentThreadId(); BOOL bSetVerb = FALSE; BOOL bPropsAvail = FALSE; if (rParsedInfo.GetCmdSwitchesObject().GetVerbName() != NULL) { if (CompareTokens(rParsedInfo.GetCmdSwitchesObject().GetVerbName(), CLI_TOKEN_SET)) { bSetVerb = TRUE; } } // Obtain the trace flag status bTrace = rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(); eloErrLogOpt = rParsedInfo.GetErrorLogObject().GetErrLogOption(); if (SUCCEEDED(ConnectToNamespace(rParsedInfo))) { CHARVECTOR cvPropList; BOOL bPropList = FALSE; try { cvPropList = rParsedInfo.GetCmdSwitchesObject().GetPropertyList(); if ( cvPropList.size() != 0 ) bPropList = TRUE; hr = m_pITargetNS->GetObject(_bstr_t(rParsedInfo. GetCmdSwitchesObject().GetClassPath()), WBEM_FLAG_USE_AMENDED_QUALIFIERS, NULL, &pIObject, NULL); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"IWbemServices::GetObject(L\"%s\", " L"WBEM_FLAG_USE_AMENDED_QUALIFIERS, 0, NULL, -, -)", (rParsedInfo.GetCmdSwitchesObject().GetClassPath())? rParsedInfo.GetCmdSwitchesObject().GetClassPath():L""); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } ONFAILTHROWERROR(hr); hr = pIObject->GetNames(NULL, WBEM_FLAG_ALWAYS | WBEM_FLAG_NONSYSTEM_ONLY, NULL, &psaNames); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"IWbemClassObject::GetNames(NULL, " L"WBEM_FLAG_ALWAYS | WBEM_FLAG_NONSYSTEM_ONLY, " L"NULL, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } ONFAILTHROWERROR(hr); // Get the number of properties. LONG lLower = 0, lUpper = 0; hr = SafeArrayGetLBound(psaNames, 1, &lLower); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetLBound(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); hr = SafeArrayGetUBound(psaNames, 1, &lUpper); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetUBound(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); // For each property obtain the information of our interest for (LONG lVar = lLower; lVar <= lUpper; lVar++) { // Get the property. hr = SafeArrayGetElement(psaNames, &lVar, &bstrPropName); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetElement(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); CHARVECTOR::iterator tempIterator; if ( bPropList == TRUE && !Find(cvPropList, _bstr_t(bstrPropName), tempIterator)) { SAFEBSTRFREE(bstrPropName); continue; } PROPERTYDETAILS pdPropDet; hr = GetPropertyAttributes(pIObject, bstrPropName, pdPropDet, rParsedInfo.GetGlblSwitchesObject().GetTraceStatus()); ONFAILTHROWERROR(hr); if (bSetVerb == TRUE || bCheckWritePropsAvail == TRUE) { if ( !_tcsstr((_TCHAR*)pdPropDet.Operation, CLI_TOKEN_WRITE) ) { SAFEBSTRFREE(bstrPropName); continue; } } if ( bCheckWritePropsAvail == TRUE ) { bPropsAvail = TRUE; SAFEBSTRFREE(bstrPropName); break; } pdPropDet.Derivation = bstrPropName; if(!rParsedInfo.GetCmdSwitchesObject().AddToPropDetMap( bstrPropName, pdPropDet)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_DET_MAP_FAILURE); bRet = FALSE; } SAFEBSTRFREE(bstrPropName); } SAFEIRELEASE(pIObject); SAFEADESTROY(psaNames); SAFEBSTRFREE(bstrPropName); } catch(_com_error& e) { bRet = FALSE; SAFEIRELEASE(pIObject); SAFEADESTROY(psaNames); SAFEBSTRFREE(bstrPropName); _com_issue_error(e.Error()); } catch(CHeap_Exception) { bRet = FALSE; SAFEIRELEASE(pIObject); SAFEADESTROY(psaNames); SAFEBSTRFREE(bstrPropName); _com_issue_error(WBEM_E_OUT_OF_MEMORY); } } else bRet = FALSE; if ( bCheckWritePropsAvail == TRUE ) bRet = bPropsAvail; return bRet; } /*---------------------------------------------------------------------------- Name :ObtainClassMethods Synopsis :This function obtains the information about the available methods for a given WMI class Type :Member Function Input Parameter(s): rParsedInfo - ref. to CParsedInfo object Output Parameter(s): rParsedInfo - ref. to CParsedInfo object Return Type :BOOL Global Variables :None Calling Syntax :ObtainClassMethods(rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ BOOL CParserEngine::ObtainClassMethods(CParsedInfo& rParsedInfo, BOOL bCheckForExists) { BOOL bRet = TRUE; BOOL bTrace = FALSE; CHString chsMsg; ERRLOGOPT eloErrLogOpt = NO_LOGGING; DWORD dwThreadId = GetCurrentThreadId(); BOOL bMethAvail = FALSE; _TCHAR* pMethodName = NULL; bTrace = rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(); eloErrLogOpt = rParsedInfo.GetErrorLogObject().GetErrLogOption(); if (SUCCEEDED(ConnectToNamespace(rParsedInfo))) { HRESULT hr = S_OK; IWbemClassObject *pIObject = NULL,*pIInSign = NULL,*pIOutSign = NULL; BSTR bstrMethodName = NULL; try { hr = m_pITargetNS->GetObject(_bstr_t(rParsedInfo. GetCmdSwitchesObject().GetClassPath()), WBEM_FLAG_USE_AMENDED_QUALIFIERS, NULL, &pIObject, NULL); if ( eloErrLogOpt ) { chsMsg.Format(L"IWbemServices::GetObject(L\"%s\", " L"WBEM_FLAG_USE_AMENDED_QUALIFIERS, 0, NULL, -, -)", rParsedInfo.GetCmdSwitchesObject().GetClassPath()); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); // Begin an enumeration of the methods available for the object. hr = pIObject->BeginMethodEnumeration(0); if ( eloErrLogOpt ) { WMITRACEORERRORLOG(hr, __LINE__, __FILE__, _T("BeginMethodEnumeration(0)"), dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); //To get info about only method if method is specified pMethodName = rParsedInfo.GetCmdSwitchesObject(). GetMethodName(); // Retrieve the next method in the method enumeration // sequence while ((pIObject->NextMethod(0, &bstrMethodName, &pIInSign, &pIOutSign)) != WBEM_S_NO_MORE_DATA) { if ( bCheckForExists == TRUE ) { bMethAvail = TRUE; SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); break; } if(pMethodName != NULL && !CompareTokens(pMethodName, (_TCHAR*)bstrMethodName)) { SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); continue; } METHODDETAILS mdMethDet; if (pIInSign) hr = ObtainMethodParamInfo(pIInSign, mdMethDet, INP, rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(), rParsedInfo); ONFAILTHROWERROR(hr); if (pIOutSign) hr = ObtainMethodParamInfo(pIOutSign, mdMethDet, OUTP, rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(), rParsedInfo); ONFAILTHROWERROR(hr); _bstr_t bstrStatus, bstrDesc; hr = GetMethodStatusAndDesc(pIObject, bstrMethodName, bstrStatus, bstrDesc, rParsedInfo.GetGlblSwitchesObject().GetTraceStatus()); mdMethDet.Status = _bstr_t(bstrStatus); mdMethDet.Description = _bstr_t(bstrDesc); ONFAILTHROWERROR(hr); if(!rParsedInfo.GetCmdSwitchesObject().AddToMethDetMap( _bstr_t(bstrMethodName),mdMethDet)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_METH_DET_MAP_FAILURE); SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); bRet = FALSE; break; } SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); } SAFEIRELEASE(pIObject); } catch(_com_error& e) { SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); SAFEIRELEASE(pIObject); _com_issue_error(e.Error()); bRet = FALSE; } catch(CHeap_Exception) { bRet = FALSE; SAFEBSTRFREE(bstrMethodName); SAFEIRELEASE(pIInSign); SAFEIRELEASE(pIOutSign); SAFEIRELEASE(pIObject); _com_issue_error(WBEM_E_OUT_OF_MEMORY); } } else bRet = FALSE; if ( bCheckForExists == TRUE ) bRet = bMethAvail; return bRet; } /*---------------------------------------------------------------------------- Name :ConnectToNamespace Synopsis :This function connects to the WMI namespace on the target machine using the supplied user credentials. Type :Member Function Input Parameter(s): rParsedInfo - reference to CParsedInfo class object Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :HRESULT Global Variables :None Calling Syntax :ConnectToNamespace(rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ HRESULT CParserEngine::ConnectToNamespace(CParsedInfo& rParsedInfo) { HRESULT hr = S_OK; DWORD dwThreadId = GetCurrentThreadId(); if (rParsedInfo.GetGlblSwitchesObject().GetNameSpaceFlag()) { BOOL bTrace = FALSE; CHString chsMsg; ERRLOGOPT eloErrLogOpt = NO_LOGGING; // Obtain the trace status bTrace = rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(); eloErrLogOpt = rParsedInfo.GetErrorLogObject().GetErrLogOption(); SAFEIRELEASE(m_pITargetNS); try { // Connect to the WMI namespace on the target machine // using the supplied user credentials. hr = Connect(m_pIWbemLocator, &m_pITargetNS, _bstr_t(rParsedInfo.GetNamespace()), NULL, NULL, _bstr_t(rParsedInfo.GetLocale()), rParsedInfo); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"IWbemLocator::ConnectServer(L\"%s\", NULL, " L"NULL, L\"%s\", 0L, NULL, NULL, -)", rParsedInfo.GetNamespace(), rParsedInfo.GetLocale()); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } ONFAILTHROWERROR(hr); // set the security privileges at the interface level hr = SetSecurity(m_pITargetNS, NULL, NULL, NULL, NULL, rParsedInfo.GetGlblSwitchesObject(). GetAuthenticationLevel(), rParsedInfo.GetGlblSwitchesObject(). GetImpersonationLevel()); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"CoSetProxyBlanket(-, RPC_C_AUTHN_WINNT, " L"RPC_C_AUTHZ_NONE, NULL, %d, %d, -, EOAC_NONE)", rParsedInfo.GetGlblSwitchesObject(). GetAuthenticationLevel(), rParsedInfo.GetGlblSwitchesObject(). GetImpersonationLevel()); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } ONFAILTHROWERROR(hr); rParsedInfo.GetGlblSwitchesObject().SetNameSpaceFlag(FALSE); } catch(_com_error& e) { // execption handling _com_issue_error(e.Error()); } catch(CHeap_Exception) { _com_issue_error(WBEM_E_OUT_OF_MEMORY); } } return hr; } /*---------------------------------------------------------------------------- Name :ObtainMethodParamInfo Synopsis :This function obtains the information about the method arguments (both input and output arguments) Type :Member Function Input Parameter(s): pIObj - pointer to IWbemClassObject object bTrace - trace flag ioInOrOut - INOROUT type specifies in or out parameter type. Output Parameter(s): mdMethDet - method details structure Return Type :HRESULT Global Variables :None Calling Syntax :ObtainMethodParamInfo(pIObj, mdMethDet, IN, bTrace, rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ HRESULT CParserEngine::ObtainMethodParamInfo(IWbemClassObject* pIObj, METHODDETAILS& mdMethDet, INOROUT ioInOrOut, BOOL bTrace, CParsedInfo& rParsedInfo) { HRESULT hr = S_OK; SAFEARRAY* psaNames = NULL; BSTR bstrPropName = NULL; CHString chsMsg; _TCHAR szNumber[BUFFER512] = NULL_STRING; ERRLOGOPT eloErrLogOpt = NO_LOGGING; DWORD dwThreadId = GetCurrentThreadId(); // Get the property names try { if ( pIObj != NULL ) { hr = pIObj->GetNames(NULL, WBEM_FLAG_ALWAYS | WBEM_FLAG_NONSYSTEM_ONLY, NULL, &psaNames); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"IWbemClassObject::GetNames(NULL, " L"WBEM_FLAG_ALWAYS | WBEM_FLAG_NONSYSTEM_ONLY, " L"NULL, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } ONFAILTHROWERROR(hr); // Get the number of properties. LONG lLower = 0, lUpper = 0; hr = SafeArrayGetLBound(psaNames, 1, &lLower); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetLBound(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); hr = SafeArrayGetUBound(psaNames, 1, &lUpper); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetUBound(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); // For each property obtian the information of our interest for (LONG lVar = lLower; lVar <= lUpper; lVar++) { // Get the property. hr = SafeArrayGetElement(psaNames, &lVar, &bstrPropName); if ( eloErrLogOpt ) { chsMsg.Format(L"SafeArrayGetElement(-, -, -)"); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, FALSE); } ONFAILTHROWERROR(hr); PROPERTYDETAILS pdPropDet; hr = GetPropertyAttributes(pIObj, bstrPropName, pdPropDet, bTrace); ONFAILTHROWERROR(hr); // 'ReturnValue' is not a property of our interest as per // the expected output given in the sample, hence omitting // the same. if ( bstrPropName != NULL ) { PROPERTYDETAILS pdIPropDet; pdIPropDet.Type = pdPropDet.Type; pdIPropDet.InOrOut = ioInOrOut; // Making bstrPropName begin with numbers to maintain // the order of method arguments in map. // while displaying remove numbers and display the // parameters in case of help only. _bstr_t bstrNumberedPropName; if ( rParsedInfo.GetGlblSwitchesObject().GetHelpFlag() ) { if ( ioInOrOut == INP ) _ltot(lVar, szNumber, 10); else _ltot(lVar + 500, szNumber, 10); chsMsg.Format(L"%-5s", szNumber); bstrNumberedPropName = _bstr_t(chsMsg) + _bstr_t(bstrPropName); } else bstrNumberedPropName = _bstr_t(bstrPropName); mdMethDet.Params.insert(PROPDETMAP::value_type( bstrNumberedPropName,pdIPropDet)); } // Free the memory allocated using SysAllocString for // bstrPropName SAFEBSTRFREE(bstrPropName); } // Destroy array descriptor and all of the data in the array SAFEADESTROY(psaNames); } } catch(_com_error& e) { SAFEBSTRFREE(bstrPropName); SAFEADESTROY(psaNames); hr = e.Error(); } catch(CHeap_Exception) { SAFEBSTRFREE(bstrPropName); SAFEADESTROY(psaNames); _com_issue_error(WBEM_E_OUT_OF_MEMORY); } return hr; } /*---------------------------------------------------------------------------- Name :GetMethodStatusAndDesc Synopsis :This function obtains the implementation status and description of the verbs available Type :Member Function Input Parameter(s): pIObj - pointer to IWbemClassObject object bstrMethod - method name bTrace - trace flag Output Parameter(s): bstrStatus - implementation status bstrDesc - Method description Return Type :HRESULT Global Variables :None Calling Syntax : GetMethodStatusAndDesc(pIObj, bstrMethod, bstrStatus, bstrDesc, bTrace) Notes :none ----------------------------------------------------------------------------*/ HRESULT CParserEngine::GetMethodStatusAndDesc(IWbemClassObject* pIObj, BSTR bstrMethod, _bstr_t& bstrStatus, _bstr_t& bstrDesc, BOOL bTrace) { HRESULT hr = S_OK; IWbemQualifierSet* pIQualSet = NULL; VARIANT vtStatus, vtDesc; VariantInit(&vtStatus); VariantInit(&vtDesc); try { if ( pIObj != NULL ) { // Obtain the method qualifier set. hr = pIObj->GetMethodQualifierSet(bstrMethod, &pIQualSet); if ( pIQualSet != NULL ) { // Retrieve the 'Implemented' qualifier status value hr = pIQualSet->Get(_bstr_t(L"Implemented"), 0L, &vtStatus, NULL); if (SUCCEEDED(hr)) { if (vtStatus.vt != VT_EMPTY && vtStatus.vt != VT_NULL ) { if ( vtStatus.boolVal ) bstrStatus = L"Implemented"; else bstrStatus = L"Not Implemented"; } else bstrStatus = L"Not Found"; } else bstrStatus = L"Not Found"; VARIANTCLEAR(vtStatus); // Should not break here, hence the HRESULT should be set to S_OK hr = S_OK; // Retrieve the 'Description' qualifier text hr = pIQualSet->Get(_bstr_t(L"Description"), 0L , &vtDesc, NULL); if (SUCCEEDED(hr)) { if (vtDesc.vt == VT_BSTR) bstrDesc = _bstr_t(vtDesc.bstrVal); else bstrDesc = L"Not available"; } else bstrDesc = L"Not available"; VARIANTCLEAR(vtDesc); // Should not break here, hence the HRESULT should be set to S_OK hr = S_OK; SAFEIRELEASE(pIQualSet); } else hr = S_OK; } } catch(_com_error& e) { VARIANTCLEAR(vtStatus); VARIANTCLEAR(vtDesc); SAFEIRELEASE(pIQualSet); hr = e.Error(); } return hr; } /*---------------------------------------------------------------------------- Name :CheckforHelp Synopsis :This function looks ahead one token to see if the next token is '?' Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object. uErrataCode - error string ID Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :CheckforHelp(cvtokens,theIterator,rParsedInfo,uErrataCode) Notes :none ----------------------------------------------------------------------------*/ RETCODE CParserEngine::CheckForHelp(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, UINT uErrataCode) { RETCODE retCode = PARSER_DISPHELP; // Set the retCode as PARSER_ERROR if no more tokens // are present. if(!GetNextToken(cvTokens, theIterator)) { retCode = PARSER_ERROR; rParsedInfo.GetCmdSwitchesObject().SetErrataCode(uErrataCode); } else if(!CompareTokens(*theIterator, CLI_TOKEN_HELP)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( uErrataCode); retCode = PARSER_ERROR; } else { retCode = ParseHelp(cvTokens, theIterator, rParsedInfo, FALSE); } return retCode; } /*---------------------------------------------------------------------------- Name :ValidateGlobalSwitchValue Synopsis :This function checks whether global switches are specified in the expected format or not. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. uErrataCode - error string rParsedInfo - reference to CParsedInfo class object. uErrataCode2 - error string2 ID htHelp - help type Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :ValidateGlobalSwitchValue(cvTokens, theIterator, uErrataCode, rParsedInfo, uErrataCode2, htHelp) Notes :none ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ValidateGlobalSwitchValue(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, UINT uErrataCode, CParsedInfo& rParsedInfo, UINT uErrataCode2, HELPTYPE htHelp) { RETCODE retCode = PARSER_CONTINUE; retCode = GetNextToken(cvTokens, theIterator, rParsedInfo, htHelp, uErrataCode2); if (retCode == PARSER_CONTINUE) { // Check for the presence of the ':' if (CompareTokens(*theIterator, CLI_TOKEN_COLON)) { // Move to next token if (GetNextToken(cvTokens, theIterator, rParsedInfo, uErrataCode)) { if (IsOption(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( uErrataCode); retCode = PARSER_ERROR; } else retCode = PARSER_CONTINUE; } else retCode = PARSER_ERROR; } else if (IsOption(*theIterator)) { retCode = CheckForHelp(cvTokens, theIterator, rParsedInfo, uErrataCode2); if (retCode == PARSER_DISPHELP) { rParsedInfo.GetGlblSwitchesObject().SetHelpFlag(TRUE); rParsedInfo.GetHelpInfoObject().SetHelp(htHelp, TRUE); } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(uErrataCode2); retCode = PARSER_ERROR; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseEVERYSwitch Synopsis :This function checks whether the value specified for the /EVERY swith is valid or not. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object. Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :ParseEVERYSwitch(cvTokens, theIterator, rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseEVERYSwitch(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; retCode = ParseNumberedSwitch(cvTokens, theIterator, rParsedInfo, EVERY, IDS_E_INVALID_EVERY_SWITCH, IDS_E_INVALID_INTERVAL); if ( retCode == PARSER_EXECCOMMAND ) { if (GetNextToken(cvTokens, theIterator) == TRUE ) { if ( CompareTokens(*theIterator, CLI_TOKEN_FSLASH) == TRUE ) { if ( GetNextToken(cvTokens, theIterator) == TRUE ) { if (CompareTokens(*theIterator, CLI_TOKEN_REPEAT) == TRUE) { retCode = ParseNumberedSwitch(cvTokens, theIterator, rParsedInfo, REPEAT, IDS_E_INVALID_REPEAT_SWITCH, IDS_E_INVALID_REPEATCOUNT); } else theIterator = theIterator - 2; } else theIterator = theIterator - 2; } else theIterator = theIterator - 1; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseFORMATSwitch Synopsis :This function checks whether the value specified for the /FORMAT swith is valid or not. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object. Output Parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type. Global Variables :None Calling Syntax :ParseFORMATSwitch(cvTokens, theIterator, rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseFORMATSwitch(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; // Reset the XSL file path. rParsedInfo.GetCmdSwitchesObject().ClearXSLTDetailsVector(); // Move to next token if (!GetNextToken(cvTokens, theIterator)) { retCode = PARSER_EXECCOMMAND; // If Translate table name is given then set the flag if( rParsedInfo.GetCmdSwitchesObject(). GetTranslateTableName() != NULL ) { rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(TRUE); } else rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(FALSE); } else if ( IsOption(*theIterator) && (theIterator + 1) < cvTokens.end() && CompareTokens(*(theIterator+1), CLI_TOKEN_HELP) ) { theIterator++; retCode = ParseHelp(cvTokens, theIterator, FORMAT, rParsedInfo); } else if (CompareTokens(*theIterator, CLI_TOKEN_COLON)) { while ( retCode == PARSER_EXECCOMMAND && theIterator < cvTokens.end() ) { XSLTDET xdXSLTDet; BOOL bFrameXSLFile = TRUE; // Move to next token if (!GetNextToken(cvTokens, theIterator)) { // PARSER_ERROR if is missing rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } else if ( IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } else { xdXSLTDet.FileName = *theIterator; if(!g_wmiCmd.GetFileFromKey(*theIterator, xdXSLTDet.FileName)) bFrameXSLFile = FALSE; } if ( retCode == PARSER_EXECCOMMAND ) { if ( !GetNextToken(cvTokens, theIterator) ) { if ( bFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLTDet); break; } else if ( IsOption(*theIterator) ) { theIterator--; if ( bFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLTDet); break; } else if ( CompareTokens(*theIterator, CLI_TOKEN_COLON ) ) { _TCHAR* pszXSLFile = NULL; retCode = ParseParamsString(cvTokens, theIterator, rParsedInfo, xdXSLTDet, pszXSLFile); if ( retCode == PARSER_EXECCOMMAND && (theIterator != cvTokens.end()) && IsOption(*theIterator) ) { theIterator--; if ( bFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLTDet); if ( pszXSLFile != NULL && retCode == PARSER_EXECCOMMAND ) { XSLTDET xdXSLDetOnlyFile; BOOL bInnerFrameXSLFile = TRUE; xdXSLDetOnlyFile.FileName = pszXSLFile; if(!g_wmiCmd.GetFileFromKey(pszXSLFile, xdXSLTDet.FileName)) bInnerFrameXSLFile = FALSE; if ( bInnerFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLDetOnlyFile, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLDetOnlyFile); } break; } } else if ( !CompareTokens(*theIterator, CLI_TOKEN_COMMA ) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } } if ( retCode == PARSER_EXECCOMMAND ) { if ( bFrameXSLFile == TRUE ) { if (!FrameFileAndAddToXSLTDetVector(xdXSLTDet, rParsedInfo)) retCode = PARSER_ERRMSG; } else rParsedInfo.GetCmdSwitchesObject(). AddToXSLTDetailsVector(xdXSLTDet); } } // If Translate table name is given then set the flag if( rParsedInfo.GetCmdSwitchesObject(). GetTranslateTableName() != NULL ) { rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(TRUE); } else rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(FALSE); } else { theIterator--; } return retCode; } /*---------------------------------------------------------------------------- Name :IsStdVerbOrUserDefVerb Synopsis :This function checks whether the verb is standard verb or user defined verb for alias. Type :Member Function Input Parameter(s): pszToken - the verb name string rParsedInfo - reference to CParsedInfo class object Output Parameter(s): None Return Type :BOOL Global Variables :None Calling Syntax :IsStdVerbOrUserDefVerb( pszToken,rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ BOOL CParserEngine::IsStdVerbOrUserDefVerb(_bstr_t bstrToken, CParsedInfo& rParsedInfo) { BOOL bStdVerbOrUserDefVerb = FALSE; try { if ( CompareTokens(bstrToken, CLI_TOKEN_GET) || CompareTokens(bstrToken, CLI_TOKEN_LIST) || CompareTokens(bstrToken, CLI_TOKEN_SET) || CompareTokens(bstrToken, CLI_TOKEN_CREATE) || CompareTokens(bstrToken, CLI_TOKEN_CALL) || CompareTokens(bstrToken, CLI_TOKEN_ASSOC) || CompareTokens(bstrToken, CLI_TOKEN_DELETE) ) bStdVerbOrUserDefVerb = TRUE; else { if ( m_bAliasName ) { METHDETMAP mdmMethDetMap = rParsedInfo.GetCmdSwitchesObject() .GetMethDetMap(); if ( mdmMethDetMap.empty() ) { m_CmdAlias.ObtainAliasVerbDetails(rParsedInfo); mdmMethDetMap = rParsedInfo.GetCmdSwitchesObject() .GetMethDetMap(); } METHDETMAP::iterator theMethIterator = NULL; for ( theMethIterator = mdmMethDetMap.begin(); theMethIterator != mdmMethDetMap.end(); theMethIterator++ ) { if ( CompareTokens((*theMethIterator).first,bstrToken) ) { bStdVerbOrUserDefVerb = TRUE; break; } } } } } catch(_com_error& e) { bStdVerbOrUserDefVerb = FALSE; _com_issue_error(e.Error()); } return bStdVerbOrUserDefVerb; } /*---------------------------------------------------------------------------- Name :ParseTRANSLATESwitch Synopsis :This function parses for translate switch in the command. Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE-enumerated type Global Variables :None Calling Syntax :ParseTRANSLATESwitch(cvTokens,theIterator,rParsedInfo) Notes :none ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseTRANSLATESwitch(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; if ( GetNextToken(cvTokens, theIterator, rParsedInfo, TRANSLATE, IDS_E_INVALID_TRANSLATE_SWITCH) == PARSER_CONTINUE ) { if ( IsOption(*theIterator) && (theIterator + 1) < cvTokens.end() && CompareTokens(*(theIterator+1), CLI_TOKEN_HELP) ) { theIterator++; retCode = ParseHelp(cvTokens, theIterator, TRANSLATE, rParsedInfo); if ( retCode == PARSER_DISPHELP ) { if( FAILED(m_CmdAlias.ConnectToAlias(rParsedInfo, m_pIWbemLocator))) retCode = PARSER_ERRMSG; if ( FAILED(m_CmdAlias.ObtainTranslateTables(rParsedInfo))) retCode = PARSER_ERRMSG; } } else if ( CompareTokens( *theIterator, CLI_TOKEN_COLON ) && GetNextToken(cvTokens, theIterator, rParsedInfo, TRANSLATE, IDS_E_INVALID_TRANSLATE_SWITCH) == PARSER_CONTINUE ) { rParsedInfo.GetCmdSwitchesObject().SetTranslateTableName(*theIterator); if ( IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_TRANSLATE_SWITCH); retCode = PARSER_ERROR; } else if(FAILED(m_CmdAlias.ConnectToAlias(rParsedInfo, m_pIWbemLocator))) retCode = PARSER_ERRMSG; else if ( m_CmdAlias.ObtainTranslateTableEntries(rParsedInfo) == TRUE ) retCode = PARSER_EXECCOMMAND; else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_TRANSLATE_TABLE_NOT_EXIST); retCode = PARSER_ERROR; } // If Format switch is specified after translate switch then // set the flag else reset it if(rParsedInfo.GetCmdSwitchesObject().GetXSLTDetailsVector(). empty()) { rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(TRUE); } else rParsedInfo.GetCmdSwitchesObject().SetTranslateFirstFlag(FALSE); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_TRANSLATE_SWITCH); retCode = PARSER_ERROR; } } else retCode = PARSER_ERROR; return retCode; } /*---------------------------------------------------------------------------- Name :ParseContextInfo Synopsis :This function does the parsing of the help on context information Type :Member Function Input Parameter(s): cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output parameter(s): rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseContextInfo(cvTokens, theIterator, rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseContextInfo(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { //BNF: CONTEXT /?[:] BOOL bContinue = TRUE; RETCODE retCode = PARSER_MESSAGE; // If option if (IsOption(*theIterator)) { // Check for help retCode = IsHelp(cvTokens, theIterator, rParsedInfo, CONTEXTHELP, IDS_E_INVALID_CONTEXT_SYNTAX, LEVEL_ONE); // If more tokens are present. if (retCode == PARSER_CONTINUE) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CONTEXT_SYNTAX); retCode = PARSER_ERROR; } } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_CONTEXT_SYNTAX); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :ValidateNodeOrNS Synopsis :This function validates the node or namespace Type :Member Function Input Parameter(s): pszInput - node/namesapce to be validated bNode - TRUE - pszInput refers to NODE FALSE - pszInput refers to NAMESPACE Output parameter(s):None Return Type :BOOL Global Variables :None Calling Syntax :ValidateNodeOrNS(pszInput, bNode) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::ValidateNodeOrNS(_TCHAR* pszInput, BOOL bNode) { IWbemServices* pISvc = NULL; BOOL bRet = TRUE; HRESULT hr = S_OK; try { if(pszInput == NULL) bRet = FALSE; if(bRet) { if (m_pIWbemLocator != NULL) { // Check for the presence of the following invalid // characters for NODE. if (bNode) { CHString str(pszInput); if (str.FindOneOf(L"\"\\,/[]:<>+=;?$#{}~`^@!'()*") != -1) { bRet = FALSE; }; } if (bRet) { // Try to connect to root namespace _bstr_t bstrNS; if (bNode) bstrNS = _bstr_t(L"\\\\") + _bstr_t(pszInput) + _bstr_t(L"\\root"); else bstrNS = _bstr_t(L"\\\\.\\") + _bstr_t(pszInput); // Call the ConnectServer method of the IWbemLocator hr = m_pIWbemLocator->ConnectServer(bstrNS, NULL, NULL, NULL, 0L, NULL, NULL, &pISvc); if (FAILED(hr)) { // If invalid machine name // 0x800706ba - RPC_SERVER_UNAVAILABLE if (bNode && (hr == 0x800706ba)) { bRet = FALSE; } // If invalid namespace // 0x8004100E - WBEM_E_INVALID_NAMESPACE if (!bNode && ((hr == WBEM_E_INVALID_NAMESPACE) || (hr == WBEM_E_INVALID_PARAMETER))) { bRet = FALSE; } } SAFEIRELEASE(pISvc); } } } } catch(_com_error& e) { bRet = FALSE; SAFEIRELEASE(pISvc); _com_issue_error(e.Error()); } catch(CHeap_Exception) { bRet = FALSE; SAFEIRELEASE(pISvc); _com_issue_error(WBEM_E_OUT_OF_MEMORY); } return bRet; } /*---------------------------------------------------------------------------- Name :ParseAssocSwitches Synopsis :This function does the parsing and interprets if command has ASSOC as the verb. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s) :cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s) :rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseAssocSwitch(cvTokens,theIterator,rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseAssocSwitches(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; while ( retCode == PARSER_EXECCOMMAND ) { // Check for the presence of RESULT CLASS switch if (CompareTokens(*theIterator, CLI_TOKEN_RESULTCLASS)) { retCode = ParseAssocSwitchEx(cvTokens, theIterator, rParsedInfo ,RESULTCLASS ); } // Check for the presence of RESULT ROLE switch else if (CompareTokens(*theIterator,CLI_TOKEN_RESULTROLE )) { retCode = ParseAssocSwitchEx(cvTokens, theIterator, rParsedInfo ,RESULTROLE ); } // Check for the presence of ASSOC CLASS switch else if (CompareTokens(*theIterator,CLI_TOKEN_ASSOCCLASS )) { retCode = ParseAssocSwitchEx(cvTokens, theIterator, rParsedInfo , ASSOCCLASS); } // Check for the presence of help else if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { rParsedInfo.GetHelpInfoObject().SetHelp(ASSOCSwitchesOnly, TRUE); retCode = ParseHelp(cvTokens, theIterator, ASSOCVerb, rParsedInfo); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_ASSOC_SWITCH); retCode = PARSER_ERROR; break; } //Checking the next tokens if ( retCode == PARSER_EXECCOMMAND ) { if ( !GetNextToken(cvTokens, theIterator) ) break; if ( !IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_COMMAND); retCode = PARSER_ERROR; break; } if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_ASSOC_SWITCH); retCode = PARSER_ERROR; break; } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseAssocSwitchEx Synopsis :This function does the parsing of tokens for the assoc switches It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s) :cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object assocSwitch - the type of assoc switch Output Parameter(s) :rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseAssocSwitchEx(cvTokens,theIterator, rParsedInfo,assocSwitch) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseAssocSwitchEx(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo , ASSOCSwitch assocSwitch) { RETCODE retCode = PARSER_EXECCOMMAND; //Checking the next token to continue parsing if ( GetNextToken(cvTokens, theIterator, rParsedInfo, ASSOCVerb, IDS_E_INVALID_ASSOC_SWITCH) == PARSER_CONTINUE ) { //Checking for help option if ( IsOption(*theIterator) && (theIterator + 1) < cvTokens.end() && CompareTokens(*(theIterator+1), CLI_TOKEN_HELP) ) { theIterator++; //Help on RESULTCLASS if (assocSwitch == RESULTCLASS) { retCode = ParseHelp(cvTokens, theIterator, RESULTCLASShelp, rParsedInfo); } //Help on RESULTROLE if (assocSwitch == RESULTROLE) { retCode = ParseHelp(cvTokens, theIterator, RESULTROLEhelp, rParsedInfo); } //Help on ASSOCCLASS if (assocSwitch == ASSOCCLASS) { retCode = ParseHelp(cvTokens, theIterator, ASSOCCLASShelp, rParsedInfo); } } //If the command has ":" , then the corresponding data //has to be set in Command object else if ( CompareTokens( *theIterator, CLI_TOKEN_COLON ) && GetNextToken(cvTokens, theIterator, rParsedInfo, ASSOCVerb, IDS_E_INVALID_ASSOC_SWITCH) == PARSER_CONTINUE ) { if ( IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_ASSOC_SWITCH); retCode = PARSER_ERROR; } else { //Setting the ResultClassName if (assocSwitch == RESULTCLASS) { if(rParsedInfo.GetCmdSwitchesObject(). SetResultClassName(*theIterator)) { retCode = PARSER_EXECCOMMAND; } else { retCode = PARSER_ERROR; } } //Setting the Result Role Name if (assocSwitch == RESULTROLE) { if(rParsedInfo.GetCmdSwitchesObject(). SetResultRoleName(*theIterator)) { retCode = PARSER_EXECCOMMAND; } else { retCode = PARSER_ERROR; } } //Setting the Assoc Class Name if (assocSwitch == ASSOCCLASS) { if(rParsedInfo.GetCmdSwitchesObject(). SetAssocClassName(*theIterator)) { retCode = PARSER_EXECCOMMAND; } else { retCode = PARSER_ERROR; } } } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_ASSOC_SWITCH); retCode = PARSER_ERROR; } } else retCode = PARSER_ERROR; return retCode; } /*---------------------------------------------------------------------------- Name :ParseNumberedSwitch Synopsis :This function does the parsing of tokens for the every and repeat switches. It parses the remaining tokens following and updates the same in CParsedInfo. Type :Member Function Input Parameter(s) :cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object htHelp - enumerated help type uSwitchErrCode - error string ID. uNumberErrCode - error string ID. Output Parameter(s) :rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE - enumerated data type Global Variables :None Calling Syntax :ParseNumberedSwitch(cvTokens, theIterator, rParsedInfo, EVERYorREPEAT, uSwitchErrCode, uNumberErrCode); Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseNumberedSwitch(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, HELPTYPE htHelp, UINT uSwitchErrCode, UINT uNumberErrCode) { RETCODE retCode = PARSER_EXECCOMMAND; BOOL bInvalidSwitch = FALSE; // Move to next token if (!GetNextToken(cvTokens, theIterator, rParsedInfo, htHelp, uSwitchErrCode)) bInvalidSwitch = TRUE; else if ( IsOption(*theIterator) && (theIterator + 1) < cvTokens.end() && CompareTokens(*(theIterator+1), CLI_TOKEN_HELP) ) { theIterator++; retCode = ParseHelp(cvTokens, theIterator, htHelp, rParsedInfo); } else if (CompareTokens(*theIterator, CLI_TOKEN_COLON)) { BOOL bSuccess = FALSE; // Move to next token if (GetNextToken(cvTokens, theIterator)) { if ( IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(uSwitchErrCode); retCode = PARSER_ERROR; } else { _TCHAR* pszEndPtr = NULL; ULONG ulNumber = _tcstoul(*theIterator, &pszEndPtr, 10); if (!lstrlen(pszEndPtr)) { BOOL bSetValue = TRUE; if ( htHelp == EVERY ) { bSetValue = rParsedInfo.GetCmdSwitchesObject(). SetRetrievalInterval(ulNumber); } else if ( htHelp == REPEAT ) { if ( ulNumber == 0) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode( IDS_E_INVALID_REPEATCOUNT); retCode = PARSER_ERROR; bSetValue = FALSE; } else { bSetValue = rParsedInfo.GetCmdSwitchesObject(). SetRepeatCount(ulNumber); } } if ( bSetValue == TRUE) { bSuccess = TRUE; rParsedInfo.GetCmdSwitchesObject().SetEverySwitchFlag(TRUE); retCode = PARSER_EXECCOMMAND; } else { bSuccess = FALSE; retCode = PARSER_ERROR; } } } } if ( bSuccess == FALSE ) { // PARSER_ERROR if no more tokens are present. i.e // is not specified. rParsedInfo.GetCmdSwitchesObject().SetErrataCode(uNumberErrCode); retCode = PARSER_ERROR; } } else bInvalidSwitch = TRUE; if ( bInvalidSwitch == TRUE ) { // PARSER_ERROR if no more tokens are present. i.e // is not specified. rParsedInfo.GetCmdSwitchesObject().SetErrataCode(uSwitchErrCode); retCode = PARSER_ERROR; } return retCode; } /*---------------------------------------------------------------------------- Name :IsValidClass Synopsis :This function validates the class specified and returns True or False accordingly as the validity of the class Type :Member Function Input Parameter(s) :rParsedInfo - reference to CParsedInfo class object Output Parameter(s) :rParsedInfo - reference to CParsedInfo class object Return Type :BOOL Global Variables :None Calling Syntax :IsValidClass(rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::IsValidClass(CParsedInfo& rParsedInfo) { HRESULT hr = S_OK; IWbemClassObject* pIObject = NULL; CHString chsMsg; DWORD dwThreadId = GetCurrentThreadId(); BOOL bTrace = FALSE; ERRLOGOPT eloErrLogOpt = NO_LOGGING; // Obtain the trace flag status bTrace = rParsedInfo.GetGlblSwitchesObject().GetTraceStatus(); eloErrLogOpt = rParsedInfo.GetErrorLogObject().GetErrLogOption(); try { hr = ConnectToNamespace(rParsedInfo); ONFAILTHROWERROR(hr); hr = m_pITargetNS->GetObject(_bstr_t(rParsedInfo. GetCmdSwitchesObject().GetClassPath()), WBEM_FLAG_USE_AMENDED_QUALIFIERS, NULL, &pIObject, NULL); if (bTrace || eloErrLogOpt) { chsMsg.Format(L"IWbemServices::GetObject(L\"%s\", " L"WBEM_FLAG_USE_AMENDED_QUALIFIERS, 0, NULL, -, -)", rParsedInfo.GetCmdSwitchesObject().GetClassPath()); WMITRACEORERRORLOG(hr, __LINE__, __FILE__, (LPCWSTR)chsMsg, dwThreadId, rParsedInfo, bTrace); } // do not add ONFAILTHROWERROR() here. SAFEIRELEASE(pIObject); } catch(_com_error &e) { hr = e.Error(); SAFEIRELEASE(pIObject); _com_issue_error(e.Error()); } catch(CHeap_Exception) { SAFEIRELEASE(pIObject); _com_issue_error(WBEM_E_OUT_OF_MEMORY); } return (SUCCEEDED(hr))? TRUE : FALSE; } /*---------------------------------------------------------------------------- Name :ObtainMethodsAvailableFlag Synopsis :This function Checks whether methods are available with alias in case of alias specified.and with class in case of class speicified. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object Output Parameter(s) :None Return Type :void Global Variables :None Calling Syntax :ObtainMethodsAvailableFlag(rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ void CParserEngine::ObtainMethodsAvailableFlag(CParsedInfo& rParsedInfo) { BOOL bMethAvail = TRUE; if ( m_bAliasName == TRUE ) bMethAvail = m_CmdAlias.AreMethodsAvailable(rParsedInfo); else bMethAvail = ObtainClassMethods(rParsedInfo, TRUE); rParsedInfo.GetCmdSwitchesObject().SetMethodsAvailable(bMethAvail); } /*---------------------------------------------------------------------------- Name :ObtainWriteablePropsAvailailableFlag Synopsis :Checks whether writable props are available with alias in case of alias specified. and with class in case of class speicified. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object Output Parameter(s) :None Return Type :void Global Variables :None Calling Syntax :ObtainWriteablePropsAvailailableFlag(rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ void CParserEngine::ObtainWriteablePropsAvailailableFlag( CParsedInfo& rParsedInfo) { BOOL bWritePropsAvail = TRUE; HRESULT hr = S_OK; try { if ( m_bAliasName == TRUE ) { hr = m_CmdAlias.ObtainAliasPropDetails(rParsedInfo, &bWritePropsAvail); ONFAILTHROWERROR(hr); } else bWritePropsAvail = ObtainClassProperties(rParsedInfo, TRUE); rParsedInfo.GetCmdSwitchesObject().SetWriteablePropsAvailable( bWritePropsAvail); } catch(_com_error& e) { _com_issue_error(e.Error()); } } /*---------------------------------------------------------------------------- Name :ParseVerbInteractive Synopsis :This function parses the verb interactive option Type :Member Function Input Parameter(s) : cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. rParsedInfo - reference to CParsedInfo class object Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object bInvalidOption - Invalid syntax for interactive Return Type :RETCODE Global Variables :None Calling Syntax :ParseVerbInteractive(rParsedInfo) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseVerbInteractive(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, BOOL& bInvalidOption) { RETCODE retCode = PARSER_EXECCOMMAND; if (GetNextToken(cvTokens, theIterator)) { // check for the presence of ':' if (CompareTokens(*theIterator, CLI_TOKEN_COLON)) { if (GetNextToken(cvTokens, theIterator)) { if (IsOption(*theIterator)) { if (GetNextToken(cvTokens, theIterator)) { if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { retCode = ParseHelp(cvTokens, theIterator, VERBSWITCHES, rParsedInfo); } else { bInvalidOption = TRUE; } } else { bInvalidOption = TRUE; } } else { while (TRUE) { if(!rParsedInfo.GetCmdSwitchesObject(). AddToInteractivePropertyList(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_LIST_FAILURE); retCode = PARSER_ERROR; } if (GetNextToken(cvTokens, theIterator)) { // check for the presence of ',' if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if (!GetNextToken(cvTokens, theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; break; } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; break; } } else { retCode = PARSER_EXECCOMMAND; break; } } } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; } } else if (IsOption(*theIterator)) { if (GetNextToken(cvTokens, theIterator)) { if (CompareTokens(*theIterator, CLI_TOKEN_HELP)) { retCode = ParseHelp(cvTokens, theIterator, VERBSWITCHES, rParsedInfo); } else { bInvalidOption = TRUE; } } else { bInvalidOption = TRUE; } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; } } return retCode; } /*---------------------------------------------------------------------------- Name :ProcessOutputAndAppendFiles Synopsis :Prepares the output and append files for output redirection. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object. retOCode - RETCODE type, specifies the initial RETCODE before calling the function. bOpenOutInWriteMode - boolean type, to specify flag of OpenOutInWriteMode. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ProcessOutputAndAppendFiles(rParsedInfo, retCode, FALSE) Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ProcessOutputAndAppendFiles(CParsedInfo& rParsedInfo, RETCODE retOCode, BOOL bOpenOutInWriteMode) { RETCODE retCode = retOCode; // TRUE for getting output file name. _TCHAR* pszOutputFileName = rParsedInfo.GetGlblSwitchesObject().GetOutputOrAppendFileName( TRUE); if ( pszOutputFileName != NULL ) { // redirect the output to file. if ( CloseOutputFile() == TRUE ) { FILE *fpOutFile; if ( bOpenOutInWriteMode == TRUE ) fpOutFile = _tfopen(pszOutputFileName, _T("w")); else fpOutFile = _tfopen(pszOutputFileName, _T("a")); if ( fpOutFile == NULL ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_OPEN_OUTPUT_FILE_FAILURE); retCode = PARSER_ERROR; } else // TRUE for setting output file pointer. rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFilePointer(fpOutFile, TRUE); } } // Processing for append file. if ( retCode == retOCode && bOpenOutInWriteMode == FALSE) { // FALSE for getting append file name. _TCHAR* pszAppendFileName = rParsedInfo.GetGlblSwitchesObject().GetOutputOrAppendFileName( FALSE); if ( pszAppendFileName != NULL ) { if ( CloseAppendFile() == TRUE ) { FILE* fpOpenAppendFile = _tfopen(pszAppendFileName, _T("a")); if ( fpOpenAppendFile == NULL ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_OPEN_APPEND_FILE_FAILURE); retCode = PARSER_ERROR; } else { // FALSE for setting append file pointer. rParsedInfo.GetGlblSwitchesObject(). SetOutputOrAppendFilePointer(fpOpenAppendFile, FALSE); } } } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseUnnamedParamList Synopsis :Parses Unnamed Parameter list. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object. cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ParseUnnamedParamList(cvTokens, theIterator,rParsedInfo); Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseUnnamedParamList(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; while (TRUE) { if(!rParsedInfo.GetCmdSwitchesObject(). AddToPropertyList(*theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_ADD_TO_PROP_LIST_FAILURE); retCode = PARSER_ERROR; } if (GetNextToken(cvTokens, theIterator)) { if (IsOption(*theIterator)) { // To facilitate ParseVerbSwitches to continue theIterator--; break; } else { // check for the presence of ',' if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if (!GetNextToken(cvTokens, theIterator)) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; break; } } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_INVALID_PARAMLIST); retCode = PARSER_ERROR; break; } } } else break; } return retCode; } /*---------------------------------------------------------------------------- Name :ValidateVerbOrMethodParams Synopsis :Validates the named params with verb or method parameters. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ValidateVerbOrMethodParams(rParsedInfo); Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ValidateVerbOrMethodParams(CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_EXECCOMMAND; BSTRMAP::iterator theIterator; PROPDETMAP::iterator propIterator; // Info about verb or method params. PROPDETMAP pdmVerbOrMethParams = (*(rParsedInfo.GetCmdSwitchesObject(). GetMethDetMap().begin())).second.Params; BSTRMAP bmNamedParams = rParsedInfo.GetCmdSwitchesObject(). GetParameterMap(); for ( theIterator = bmNamedParams.begin(); theIterator != bmNamedParams.end(); theIterator++ ) { BOOL bFind; if ( rParsedInfo.GetCmdSwitchesObject().GetVerbType() == CMDLINE ) bFind = Find(pdmVerbOrMethParams,(*theIterator).first, propIterator, TRUE); else bFind = Find(pdmVerbOrMethParams,(*theIterator).first, propIterator); if ( bFind == FALSE ) { DisplayMessage((*theIterator).first, CP_OEMCP, TRUE, TRUE); rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_NOT_A_VERBORMETH_PARAM); retCode = PARSER_ERROR; break; } else if ( (*propIterator).second.InOrOut != INP ) { DisplayMessage((*theIterator).first, CP_OEMCP, TRUE, TRUE); rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_NOT_A_INPUT_PARAM); retCode = PARSER_ERROR; break; } } return retCode; } /*---------------------------------------------------------------------------- Name :ParseParamsString Synopsis :Parses the parameter string Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object. cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. xdXSLTDet - reference to the XSLdetails vector pszXSLFile - string type, XSL file name. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ParseParamsString(cvTokens, theIterator, rParsedInfo, xdXSLTDet, pszXSLFile); Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseParamsString(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo, XSLTDET& xdXSLTDet, _TCHAR* pszXSLFile) { pszXSLFile = NULL; RETCODE retCode = PARSER_EXECCOMMAND ; try { if ( !GetNextToken(cvTokens, theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } else if ( IsOption(*theIterator) ) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } else { while ( retCode == PARSER_EXECCOMMAND ) { _TCHAR* pszParam = NULL; _TCHAR* pszParamValue = NULL; pszParam = _tcstok(*theIterator,CLI_TOKEN_EQUALTO); if(pszParam != NULL) { pszParamValue = _tcstok(NULL,CLI_TOKEN_EQUALTO); if(pszParamValue != NULL) { _bstr_t bstrParam = pszParam; _bstr_t bstrParamValue = pszParamValue; if(IsOption(pszParamValue)) { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } else xdXSLTDet.ParamMap.insert(BSTRMAP::value_type( bstrParam, bstrParamValue)); } else pszXSLFile = pszParam; } else { rParsedInfo.GetCmdSwitchesObject(). SetErrataCode(IDS_E_INVALID_FORMAT); retCode = PARSER_ERROR; } if ( retCode == PARSER_EXECCOMMAND ) { if ( !GetNextToken(cvTokens, theIterator) ) break; else if ( IsOption(*theIterator) ) break; else if (CompareTokens(*theIterator, CLI_TOKEN_COMMA)) { if ( theIterator + 1 == cvTokens.end() ) break; else if ( theIterator + 2 == cvTokens.end() ) break; else if ( pszParamValue == NULL ) break; else theIterator++; } } } } } catch(_com_error& e) { retCode = PARSER_ERROR; _com_issue_error(e.Error()); } return retCode; } /*---------------------------------------------------------------------------- Name :ParseNodeListFile Synopsis :Parses the node list file. Type :Member Function Input Parameter(s) : rParsedInfo - reference to CParsedInfo class object. cvTokens - the tokens vector theIterator - the Iterator to the cvTokens vector. Output Parameter(s) : rParsedInfo - reference to CParsedInfo class object Return Type :RETCODE Global Variables :None Calling Syntax :ParseNodeListFile(cvTokens, theIterator,rParsedInfo); Notes :None ----------------------------------------------------------------------------*/ RETCODE CParserEngine::ParseNodeListFile(CHARVECTOR& cvTokens, CHARVECTOR::iterator& theIterator, CParsedInfo& rParsedInfo) { RETCODE retCode = PARSER_CONTINUE; _TCHAR *pszTempFileName = (*theIterator+1); _TCHAR *szNodeListFileName = new _TCHAR [BUFFER512]; if (szNodeListFileName == NULL) _com_issue_error(WBEM_E_OUT_OF_MEMORY); lstrcpy(szNodeListFileName, pszTempFileName); UnQuoteString(szNodeListFileName); FILE *fpNodeListFile = _tfopen(szNodeListFileName, _T("rb")); LONG lNumberOfInserts = 0; CHARVECTOR::iterator itrVectorInCmdTkzr = NULL; _TCHAR *pszComma = NULL; _TCHAR *pszNode = NULL; FILETYPE eftNodeFileType = ANSI_FILE; char *pszFirstTwoBytes = NULL; try { if ( fpNodeListFile != NULL ) { Find(m_CmdTknzr.GetTokenVector(), *theIterator, itrVectorInCmdTkzr); SAFEDELETE(*itrVectorInCmdTkzr); itrVectorInCmdTkzr = m_CmdTknzr.GetTokenVector().erase(itrVectorInCmdTkzr); // Remove @nodelistfile token from token vector. theIterator = cvTokens.erase(theIterator); // Indentifing the file type whether Unicode or ANSI. pszFirstTwoBytes = new char[2]; fread(pszFirstTwoBytes, 2, 1, fpNodeListFile); if ( memcmp(pszFirstTwoBytes, UNICODE_SIGNATURE, 2) == 0 ) { eftNodeFileType = UNICODE_FILE; } else if (memcmp(pszFirstTwoBytes, UNICODE_BIGEND_SIGNATURE, 2) == 0 ) { eftNodeFileType = UNICODE_BIGENDIAN_FILE; } else if( memcmp(pszFirstTwoBytes, UTF8_SIGNATURE, 2) == 0 ) { eftNodeFileType = UTF8_FILE; } else { eftNodeFileType = ANSI_FILE; fseek(fpNodeListFile, 0, SEEK_SET); } SAFEDELETE(pszFirstTwoBytes); _TCHAR szNodeName[BUFFER512] = NULL_STRING; if ( GetNodeFromNodeFile(fpNodeListFile, eftNodeFileType, szNodeName) == FALSE ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_NODELISTFILE_EMPTY); retCode = PARSER_ERROR; } else { BOOL bFirstTime = TRUE; do { LONG lNodeStrLen = lstrlen(szNodeName); if ( szNodeName[lNodeStrLen-1] == _T('\n') ) szNodeName[lNodeStrLen-1] = _T('\0'); CHString strRawNodeName(szNodeName); strRawNodeName.TrimLeft(); strRawNodeName.TrimRight(); lstrcpy(szNodeName, strRawNodeName.GetBuffer(BUFFER512)); if ( szNodeName[0] != _T('#') && strRawNodeName.IsEmpty() == FALSE ) { if ( bFirstTime == FALSE ) { pszComma = new _TCHAR[lstrlen( CLI_TOKEN_COMMA) + 1]; if (pszComma == NULL) _com_issue_error(WBEM_E_OUT_OF_MEMORY); lstrcpy(pszComma, CLI_TOKEN_COMMA); theIterator = cvTokens.insert(theIterator, pszComma); theIterator++; itrVectorInCmdTkzr = m_CmdTknzr.GetTokenVector(). insert(itrVectorInCmdTkzr, pszComma); itrVectorInCmdTkzr++; lNumberOfInserts++; } else bFirstTime = FALSE; lNodeStrLen = lstrlen(szNodeName); pszNode = new _TCHAR[lNodeStrLen + 1]; if (pszNode == NULL) _com_issue_error(WBEM_E_OUT_OF_MEMORY); lstrcpy(pszNode, szNodeName); theIterator = cvTokens.insert(theIterator, pszNode); theIterator++; itrVectorInCmdTkzr = m_CmdTknzr.GetTokenVector().insert( itrVectorInCmdTkzr, pszNode); itrVectorInCmdTkzr++; lNumberOfInserts++; } } while ( GetNodeFromNodeFile(fpNodeListFile, eftNodeFileType, szNodeName) == TRUE ); if ( lNumberOfInserts == 0 ) { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_NO_NODES_FOR_INSERTION); retCode = PARSER_ERROR; } theIterator = theIterator - lNumberOfInserts; } fclose(fpNodeListFile); } else { rParsedInfo.GetCmdSwitchesObject().SetErrataCode( IDS_E_NODELISTFILE_OPEN_FAILURE); retCode = PARSER_ERROR; } SAFEDELETE(szNodeListFileName); } catch(CHeap_Exception) { retCode = PARSER_ERROR; SAFEDELETE(szNodeListFileName); SAFEDELETE(pszFirstTwoBytes); retCode = PARSER_ERROR; _com_issue_error(WBEM_E_OUT_OF_MEMORY); } catch(_com_error& e) { retCode = PARSER_ERROR; SAFEDELETE(szNodeListFileName); SAFEDELETE(pszComma); SAFEDELETE(pszNode); SAFEDELETE(pszFirstTwoBytes); _com_issue_error(e.Error()); } return retCode; } /*---------------------------------------------------------------------------- Name :GetNodeFromNodeFile Synopsis :Retrieves the node list file. Type :Member Function Input Parameter(s) : fpNodeListFile - pointer to File containing node list. eftNodeFileType - Enum value specifing unicode or ANSI .... Output Parameter(s) : szNodeName - pointer to string specifing node to be returned. Return Type :BOOL Global Variables :None Calling Syntax :GetNodeFromNodeFile(fpNodeListFile, eftNodeFileType, szNodeName); Notes :None ----------------------------------------------------------------------------*/ BOOL CParserEngine::GetNodeFromNodeFile(FILE* fpNodeListFile, FILETYPE eftNodeFileType, _TCHAR* szNodeName) { WCHAR wszNodeName[2] = L""; char cszNodeName[2] = ""; _TCHAR szTemp[2] = NULL_STRING; try { lstrcpy(szNodeName, NULL_STRING); while( TRUE ) { lstrcpy(szTemp, NULL_STRING); if ( eftNodeFileType == UNICODE_FILE ) { if ( fgetws(wszNodeName, 2, fpNodeListFile) != NULL ) { lstrcpy(szTemp, (_TCHAR*)_bstr_t(wszNodeName)); } else break; } else if ( eftNodeFileType == UNICODE_BIGENDIAN_FILE ) { if ( fgetws(wszNodeName, 2, fpNodeListFile) != NULL ) { BYTE HiByte = HIBYTE(wszNodeName[0]); BYTE LowByte = LOBYTE(wszNodeName[0]); wszNodeName[0] = MAKEWORD(HiByte, LowByte); lstrcpy(szTemp, (_TCHAR*)_bstr_t(wszNodeName)); } else break; } else if ( eftNodeFileType == UTF8_FILE ) { if ( fgets(cszNodeName, 2, fpNodeListFile) != NULL ) { MultiByteToWideChar( CP_UTF8, // code page 0, // character-type options cszNodeName, // string to map 2, // number of bytes in string wszNodeName, // wide-character buffer 2 // size of buffer ); lstrcpy(szTemp, (_TCHAR*)_bstr_t(wszNodeName)); } else break; } else { if ( fgets(cszNodeName, 2, fpNodeListFile) != NULL ) { lstrcpy(szTemp, (_TCHAR*)_bstr_t(cszNodeName)); } else break; } if ( _tcscmp(szTemp, CLI_TOKEN_SPACE) == 0 || _tcscmp(szTemp, CLI_TOKEN_TAB) == 0 || _tcscmp(szTemp, CLI_TOKEN_SEMICOLON) == 0 || _tcscmp(szTemp, CLI_TOKEN_COMMA) == 0 || _tcscmp(szTemp, CLI_TOKEN_DOUBLE_QUOTE)== 0 || _tcscmp(szTemp, CLI_TOKEN_NEWLINE) == 0 ) { break; } else { lstrcat(szNodeName, szTemp); } } } catch(_com_error& e) { _com_issue_error(e.Error()); } return (!feof(fpNodeListFile) || _tcscmp(szNodeName, NULL_STRING)); }