Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

768 lines
23 KiB

/*++
Copyright (c) 1997 Microsoft Corporation
Module Name :
main.cpp
Abstract:
command line admin tool main function
Environment:
Win32 User Mode
Author:
jaroslad (jan 1997)
--*/
#include <tchar.h>
#include <afx.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include "admutil.h"
#include "tables.h"
#include "jd_misc.h"
CAdmUtil oAdmin; //admin object
#define MAX_NUMBER_OF_SMALL_VALUES 100
#define MAX_NUMBER_OF_VALUES 10100
#define MAX_NUMBER_OF_DEFAULT_ARGS 10110
//structure stores the command line arguments
struct tAdmutilParams
{
WORD fHelp; //print help - flag
WORD fFullHelp; //print help - flag
WORD fNoSave; //do not save metabase
LPCTSTR lpszCommand;
LPCTSTR lpszComputer;
WORD wInstance;
LPCTSTR lpszService;
LPCTSTR lpszPath;
LPCTSTR lpszComputerDst; //for COPY destination
WORD wInstanceDst; //for COPY destination
LPCTSTR lpszServiceDst; //for COPY destination
LPCTSTR lpszPathDst; //for COPY destination
LPCTSTR lplpszDefaultArg[MAX_NUMBER_OF_DEFAULT_ARGS];
WORD wDefaultArgCount;
LPCTSTR lpszPropName;
LPCTSTR lplpszPropAttrib[MAX_NUMBER_OF_SMALL_VALUES];
WORD wPropAttribCount;
LPCTSTR lplpszPropDataType[MAX_NUMBER_OF_SMALL_VALUES];
WORD wPropDataTypeCount;
LPCTSTR lplpszPropUserType[MAX_NUMBER_OF_SMALL_VALUES];
WORD wPropUserTypeCount;
LPCTSTR lplpszPropValue[MAX_NUMBER_OF_VALUES]; //pointer to array of values (e.g multisz type allows the multiple values for one property
DWORD lpdwPropValueLength[MAX_NUMBER_OF_VALUES];
WORD wPropValueCount;
WORD wPropFileValueCount;
};
tAdmutilParams Params;
_TCHAR **g_argv;
int g_argc;
static BOOL CompareOutput(_TCHAR *FileToCompare,_TCHAR* FileTemplate);
// definition of command line syntax with some help text -this is the input for ParseParam()
TParamDef CmdLineArgDesc[]=
{
{_TEXT(""),MAX_NUMBER_OF_DEFAULT_ARGS, (void *) Params.lplpszDefaultArg,TYPE_LPCTSTR,OPT, _TEXT("Command [param ...]"),_TEXT("CMD [param param]"),&Params.wDefaultArgCount},
{_TEXT("svc") ,1, (void *) &Params.lpszService,TYPE_LPCTSTR,OPT,_TEXT("service (MSFTPSVC, W3SVC)")},
{_TEXT("s"), 1, (void *) &Params.lpszComputer,TYPE_LPCTSTR,OPT, _TEXT("name of computer to administer"),_TEXT("comp"),},
{_TEXT("i") ,1, &Params.wInstance, TYPE_WORD, OPT,_TEXT("instance number"),_TEXT("inst")},
{_TEXT("path"),1, (void *) &Params.lpszPath, TYPE_LPCTSTR,OPT, _TEXT("path "),_TEXT("path")},
{_TEXT("pathdst"),1, (void *) &Params.lpszPathDst, TYPE_LPCTSTR,OPT, _TEXT("destination path (use for COPY only)"),_TEXT("path")},
{_TEXT("prop"),1, (void *) &Params.lpszPropName, TYPE_LPCTSTR,OPT, _T("property (IIS parameter) name")},
{_TEXT("attrib"),MAX_NUMBER_OF_SMALL_VALUES, (void *) Params.lplpszPropAttrib,TYPE_LPCTSTR,OPT, _T("property attributes"),_T(""),&Params.wPropAttribCount},
{_TEXT("utype"),MAX_NUMBER_OF_SMALL_VALUES, (void *) Params.lplpszPropUserType,TYPE_LPCTSTR,OPT, _T("property user type"),_T(""),&Params.wPropUserTypeCount},
{_TEXT("dtype"),MAX_NUMBER_OF_SMALL_VALUES, (void *) Params.lplpszPropDataType,TYPE_LPCTSTR,OPT, _T("property data type"),_T(""),&Params.wPropDataTypeCount},
{_TEXT("value"),MAX_NUMBER_OF_VALUES, (void *) Params.lplpszPropValue,TYPE_LPCTSTR,OPT, _T("property values"),_T(""),&Params.wPropValueCount},
{_TEXT("fvalue"),MAX_NUMBER_OF_VALUES, (void *) Params.lplpszPropValue,TYPE_LPCTSTR,OPT, _T("property values as files"),_T(""),&Params.wPropFileValueCount},
{_TEXT("nosave"),0, &Params.fNoSave,TYPE_WORD,OPT, _T("do not save metabase"),_T("")},
{_TEXT("timeout"),1, &g_dwTIMEOUT_VALUE,TYPE_DWORD,OPT, _T("timeout for metabase access in ms (default is 30000 sec"),_T("")},
{_TEXT("delayafteropen"),1, &g_dwDELAY_AFTER_OPEN_VALUE,TYPE_DWORD,OPT, _T("delay after opening node (default is 0 sec)"),_T("")},
{_TEXT("help"),0, &Params.fFullHelp,TYPE_WORD,OPT, _T("print full help"),_T("")},
{_TEXT("?"),0, &Params.fHelp,TYPE_WORD,OPT, _T("print help"),_T("")},
{NULL,0, NULL , TYPE_TCHAR, OPT,
_T("IIS K2 administration utility that enables the manipulation with metabase parameters\n")
_T("\n")
_T("Notes:\n")
_T(" Simpified usage of mdutil doesn't require any switches.\n")
_T(" \n")
_T(" mdutil GET path - display chosen parameter\n")
_T(" mdutil SET path value ... - assign the new value\n")
_T(" mdutil ENUM path - enumerate all parameters for given path\n")
_T(" mdutil ENUM_ALL path - recursively enumerate all parameters\n")
_T(" mdutil DELETE path - delete given path or parameter\n")
_T(" mdutil CREATE path - create given path\n")
_T(" mdutil COPY pathsrc pathdst - copy all from pathsrc to pathdst (will create pathdst)\n")
_T(" mdutil RENAME pathsrc pathdst - rename chosen path\n")
_T(" mdutil SCRIPT scriptname - runs the script\n")
_T(" mdutil APPCREATEINPROC w3svc/1/root - Create an in-proc application \n")
_T(" mdutil APPCREATEOUTPROC w3svc/1/root - Create an out-proc application\n")
_T(" mdutil APPDELETE w3svc/1/root - Delete the application if there is one\n")
_T(" mdutil APPRENAME w3svc/1/root/dira w3svc/1/root/dirb - Rename the application \n")
_T(" mdutil APPUNLOAD w3svc/1/root - Unload an application from w3svc runtime lookup table.\n")
_T(" mdutil APPGETSTATUS w3svc/1/root - Get status of the application\n")
_T("\n")
_T(" -path has format: {computer}/{service}/{instance}/{URL}/{Parameter}\n")
_T("\n")
_T("Samples:\n")
_T(" mdutil GET W3SVC/1/ServerBindings \n")
_T(" mdutil SET JAROSLAD2/W3SVC/1/ServerBindings \":81:\"\n")
_T(" mdutil COPY W3SVC/1/ROOT/iisadmin W3SVC/2/ROOT/adm\n")
_T(" mdutil ENUM_ALL W3SVC\n")
_T(" mdutil ENUM W3SVC/1\n")
_T("\n")
_T("Additional features\n")
_T(" set MDUTIL_BLOCK_ON_ERROR environment variable to block mdutil.exe after error (except ERROR_PATH_NOT_FOUND)\n")
_T(" set MDUTIL_ASCENT_LOG environment variable to force mdutil.exe to append errors to ascent log\n")
_T(" set MDUTIL_PRINT_ID environment variable to force mdutil.exe to print metadata numeric identifiers along with friendly names\n")
}
};
BOOL
ReadFromFiles(
LPTSTR* lplpszPropValue,
DWORD* lpdwPropValueLength,
DWORD dwPropFileValueCount
)
{
DWORD dwL;
while ( dwPropFileValueCount-- )
{
FILE* fIn = _tfopen( lplpszPropValue[dwPropFileValueCount], _T("rb") );
if ( fIn == NULL )
{
return FALSE;
}
if ( fseek( fIn, 0, SEEK_END ) == 0 )
{
dwL = ftell( fIn );
fseek( fIn, 0, SEEK_SET );
}
else
{
fclose( fIn );
return FALSE;
}
if ( (lplpszPropValue[dwPropFileValueCount] = (LPTSTR)malloc( dwL )) == NULL )
{
fclose( fIn );
return FALSE;
}
if ( fread( lplpszPropValue[dwPropFileValueCount], 1, dwL, fIn ) != dwL )
{
fclose( fIn );
return FALSE;
}
fclose( fIn );
lpdwPropValueLength[dwPropFileValueCount] = dwL;
}
return TRUE;
}
///////////////////////////////
class CScript
{
FILE * m_fpScript;
void GetNextToken(/*OUT*/ LPTSTR * lplpszToken);
DWORD CleanWhiteSpaces(void);
public:
CScript(void) {m_fpScript=0;};
DWORD Open(LPCTSTR lpszFile);
DWORD Close(void);
DWORD GetNextLineTokens(int *argc, /*OUT*/ _TCHAR *** argv);
};
DWORD CScript::CleanWhiteSpaces()
{
if(m_fpScript!=NULL)
{
int LastChar=0;
_TINT c=0;
while(1)
{
LastChar=c;
c=_fgettc(m_fpScript);
if(c==_T('\t') || c==_T(' ') || c==_T('\r'))
{ continue;
}
if(c==_T('\\'))
{
int cc=_fgettc(m_fpScript);
if (cc==_T('\r')) //continue with the next line of the file
{
if(_fgettc(m_fpScript)!=_T('\n'))
{
fseek( m_fpScript, -1, SEEK_CUR );
}
continue;
}
else if (cc==_T('\n')) //continue with the next line of the file
{
continue;
}
else
{
fseek( m_fpScript, -2, SEEK_CUR );
break;
}
}
if(c==WEOF)
{ break;
}
else
{ fseek( m_fpScript, -1, SEEK_CUR ); //return back one position
break;
}
}
}
return 0;
}
void CScript::GetNextToken(LPTSTR * lplpszToken)
{
enum {TERMINATE_QUOTE, TERMINATE_WHITESPACE};
long flag=TERMINATE_WHITESPACE;
//clean white spaces
CleanWhiteSpaces();
//store the beginning offset of token
long Offset = ftell(m_fpScript);
_TINT c=_fgettc(m_fpScript);
long CurrentOffset=0;
if(c==WEOF)
{
*lplpszToken=NULL;
return ;
}
if (c==_T('\n')){
*lplpszToken=_T("\n");
return ;
}
else
{
if (c==_T('\"')) { //token ends with " or the end of the line
flag=TERMINATE_QUOTE;
Offset = ftell(m_fpScript);
}
else {
flag=TERMINATE_WHITESPACE;
}
// find the end of the token
while(1) {
CurrentOffset=ftell(m_fpScript);
c=_fgettc(m_fpScript);
if(c==_T('\n')){
break;
}
if(c==WEOF)
{
break;
}
if( (flag==TERMINATE_QUOTE && c==_T('\"')) || (
flag==TERMINATE_WHITESPACE && (c==_T(' ') || c==_T('\t') || c==_T('\r')) ) ){
break;
}
}
//get the token size
long TokenSize = CurrentOffset - Offset;
if(TokenSize!=0)
{
// allocate mamory for the token
*lplpszToken = new _TCHAR[ TokenSize+1 ];
//read the token
fseek( m_fpScript, Offset, SEEK_SET);
for(int i=0;i<TokenSize;i++)
(*lplpszToken)[i]=(TCHAR)_fgettc(m_fpScript);
//terminate the token
(*lplpszToken)[i]=0;
}
else
{ //empty string
*lplpszToken=new _TCHAR[1 ];
(*lplpszToken)[0]=0;
}
//discard double quote if it was at the end of the token
c=_fgettc(m_fpScript);
if(c!=_T('\"'))
fseek( m_fpScript, ((c==WEOF)?0:-1), SEEK_CUR );
}
}
DWORD CScript::Open(LPCTSTR lpszFile)
{
m_fpScript = _tfopen( lpszFile, _T("rt") );
if(m_fpScript==NULL)
return GetLastError();
else
return ERROR_SUCCESS;
}
DWORD CScript::Close()
{
if( m_fpScript!=NULL)
fclose( m_fpScript);
return GetLastError();
}
DWORD CScript::GetNextLineTokens(int *argc, /*OUT*/ _TCHAR *** argv)
{
for(int i=1;i<*argc;i++) {
delete (*argv)[i];
(*argv)[i]=0;
}
*argc=0;
if(*argv==NULL)
(*argv)=new LPTSTR [ MAX_NUMBER_OF_VALUES ];
(*argv)[(*argc)++]=_T("mdutil"); //set zero parameter
LPTSTR lpszNextToken=NULL;
while((*argc)<MAX_NUMBER_OF_VALUES) {
GetNextToken(&lpszNextToken);
if(lpszNextToken==NULL ) //end of file
break;
if(_tcscmp(lpszNextToken,_T("\n"))==0) //new line
break;
(*argv)[(*argc)++]=lpszNextToken;
}
return GetLastError();
}
int MainFunc(int argc, _TCHAR **argv); //declaration
//MAIN FUNCTION
int __cdecl main(int argc, CHAR **_argv)
{
// convert parameters from SBCS to UNICODE;
_TCHAR **argv= new LPTSTR [argc];
for (int i=0;i<argc;i++)
{
argv[i]=new _TCHAR[strlen(_argv[i])+1];
#ifdef UNICODE
MultiByteToWideChar(0, 0, _argv[i], -1, argv[i],strlen(_argv[i])+1 );
#else
strcpy(argv[i],_argv[i]);
#endif
}
DWORD dwCommandCode=0;
DWORD retval=0;
HRESULT hRes;
hRes = CoInitializeEx(NULL, COINIT_MULTITHREADED);
if (FAILED(hRes)) {
fatal_error_printf(_T("CoInitializeEx\n"));
}
//extract command line parameters
ParseParam(argc,argv,CmdLineArgDesc);
//**************************
//PRINT HELP ON REQUEST
if(Params.fFullHelp)
{
//print help
DisplayUsage(argv,CmdLineArgDesc);
PrintTablesInfo();
}
else if(Params.wDefaultArgCount==0 || Params.fHelp)
{
//print help
DisplayUsage(argv,CmdLineArgDesc);
}
else
{
if (Params.wDefaultArgCount>0)
{
//first default has to be command
Params.lpszCommand=Params.lplpszDefaultArg[0];
dwCommandCode = tCommandNameTable::MapNameToCode(Params.lplpszDefaultArg[0]);
if( Params.wDefaultArgCount>1)
{ //second default
Params.lpszPath=Params.lplpszDefaultArg[1];
}
if(dwCommandCode==CMD_SCRIPT) //process script
{
tAdmutilParams StoredParams=Params;
CScript Script;
DWORD dwRes;
if((dwRes=Script.Open(Params.lpszPath))!=ERROR_SUCCESS)
fatal_error_printf(_T("cannot open script file %s (error %d)\n"),Params.lpszPath,dwRes);
int l_argc=0;
_TCHAR ** l_argv=NULL;
while(1)
{
Script.GetNextLineTokens(&l_argc,&l_argv);
if(l_argc==1) //end of script file
break;
Params=StoredParams;
_tprintf(_T(">"));
for(int i=1;i<l_argc;i++)
_tprintf(_T("%s "),l_argv[i]);
_tprintf(_T("\n"));
DWORD retval1=MainFunc(l_argc,l_argv);
retval = ((retval==0) ? retval1:retval);
}
if (oAdmin.GetpcAdmCom()!=NULL)
{
if(Params.fNoSave)
{}
else
oAdmin.SaveData();
}
}
else
{
retval=MainFunc(argc,argv); //run only one command typed on the command line
if (oAdmin.GetpcAdmCom()!=NULL)
{
if(Params.fNoSave)
{}
else
oAdmin.SaveData();
}
}
}
}
//close admin object
oAdmin.Close();
//close wam adm object
oAdmin.CloseWamAdm();
CoUninitialize();
//Cleanup of parameters
if(argv!=0)
{
for (int i=0;i<argc;i++)
{
delete [] argv[i];
}
delete argv;
}
return retval;
}
int MainFunc(int argc, _TCHAR **argv)
{
g_argc=argc;
g_argv=argv;
DWORD retval;
LPCTSTR lpszCommand=0;
CAdmNode AdmNode;
CAdmProp AdmProp;
CAdmNode AdmDstNode;
int i;
DWORD dwCommandCode=0;
//extract command line parameters
ParseParam(argc,argv,CmdLineArgDesc);
//PROCESS THE DEFAULT PARAMETERS
// trick: place the default arguments into variables that apply for non default one (see Params structure)
if (Params.wDefaultArgCount>0)
{ //first default has to be command
Params.lpszCommand=Params.lplpszDefaultArg[0];
dwCommandCode = tCommandNameTable::MapNameToCode(Params.lplpszDefaultArg[0]);
if( Params.wDefaultArgCount>1)
{//second deault has to be path
Params.lpszPath=Params.lplpszDefaultArg[1];
}
if( Params.wDefaultArgCount>2)
{
switch(dwCommandCode)
{
case CMD_SET:
//the rest of default args are values
Params.wPropValueCount=0;
for(i=2;i<Params.wDefaultArgCount;i++)
{
Params.lplpszPropValue[i-2] = Params.lplpszDefaultArg[i];
Params.wPropValueCount++;
}
break;
case CMD_DELETE:
case CMD_CREATE:
case CMD_GET:
case CMD_ENUM:
case CMD_ENUM_ALL:
if( Params.wDefaultArgCount>2)
{
error_printf(_T("maximum default arguments number exceeds expected (2)\n"));
return 1;
}
break;
case CMD_COPY:
case CMD_RENAME:
case CMD_APPRENAME:
if( Params.wDefaultArgCount>3)
{
error_printf(_T("maximum default arguments number exceeds expected (3)\n"));
return 1;
}
else
Params.lpszPathDst=Params.lplpszDefaultArg[2];
break;
default:
error_printf(_T("command not recognized: %s or number of parameters doesn't match\n"),Params.lpszCommand);
return 1;
}
}
} //end of default argument handling
//extract computer,service,instance, if stored in Path
AdmNode.SetPath(Params.lpszPath);
//valid only for copy function
AdmDstNode.SetPath(Params.lpszPathDst);
//process computer, service, instance, property name arguments
if(Params.lpszComputer!=NULL) {
if(!AdmNode.GetComputer().IsEmpty()) {
error_printf(_T("computer name entered more than once\n"));
return 1;
}
else
AdmNode.SetComputer(Params.lpszComputer);
}
if(Params.lpszService!=NULL) {
if(!AdmNode.GetService().IsEmpty()) {
error_printf(_T("service name entered more than once\n"));
return 1;
}
else {
if(IsServiceName(Params.lpszService))
AdmNode.SetService(Params.lpszService);
else {
error_printf(_T("service name not recognized: %s\n"), Params.lpszService);
return 1;
}
}
}
if(Params.wInstance!=0)
{
if(!AdmNode.GetInstance().IsEmpty()) {
error_printf(_T("instance entered more than once\n"));
return 1;
}
else {
_TCHAR buf[30];
//!!! maybe ltoa should be used
AdmNode.SetInstance(_itot(Params.wInstance,buf,10));
}
}
//******************************************
//process attrib, utype, dtype, value
//property name first
CString strProp=AdmNode.GetProperty();
if(Params.lpszPropName!=NULL && !strProp.IsEmpty())
{
error_printf(_T("property name entered more than once\n"));
return 1;
}
else if (Params.lpszPropName!=NULL)
{
AdmNode.SetProperty(Params.lpszPropName);
}
if(IsNumber(AdmNode.GetProperty()))
{
AdmProp.SetIdentifier(_ttol(AdmNode.GetProperty()));
}
else
{
DWORD dwIdentifier=MapPropertyNameToCode(AdmNode.GetProperty());
if(dwIdentifier==NAME_NOT_FOUND)
{}
else
AdmProp.SetIdentifier(dwIdentifier) ;
}
//process the attrib entered on command line
if(Params.wPropAttribCount!=0)
{ DWORD dwAttrib=0;
for (i=0;i<Params.wPropAttribCount;i++)
{
if(IsNumber(Params.lplpszPropAttrib[i]))
dwAttrib += _ttol(Params.lplpszPropAttrib[i]);
else
{
DWORD dwMapped=MapAttribNameToCode(Params.lplpszPropAttrib[i]);
if(dwMapped==NAME_NOT_FOUND)
{
error_printf(_T("attribute name not resolved: %s\n"), Params.lplpszPropAttrib[i]);
return 1;
}
else
dwAttrib |= dwMapped;
}
}
//overwrite the default attrib
AdmProp.SetAttrib(dwAttrib) ;
}
//process the usertype entered on command line
if(Params.wPropUserTypeCount!=0)
{ DWORD dwUserType=0;
for (i=0;i<Params.wPropUserTypeCount;i++)
{
if(IsNumber(Params.lplpszPropUserType[i]))
dwUserType += _ttol(Params.lplpszPropUserType[i]);
else
{
DWORD dwMapped=MapUserTypeNameToCode(Params.lplpszPropUserType[i]);
if(dwMapped==NAME_NOT_FOUND)
{
error_printf(_T("user type not resolved: %s\n"), Params.lplpszPropUserType[i]);
return 1;
}
else
dwUserType |= dwMapped;
}
}
//overwrite the default UserType
AdmProp.SetUserType(dwUserType) ;
}
//process the datatype entered on command line
if(Params.wPropDataTypeCount!=0)
{ DWORD dwDataType=0;
for (i=0;i<Params.wPropDataTypeCount;i++)
{
if(IsNumber(Params.lplpszPropDataType[i]))
dwDataType += _ttol(Params.lplpszPropDataType[i]);
else
{
DWORD dwMapped=MapDataTypeNameToCode(Params.lplpszPropDataType[i]);
if(dwMapped==NAME_NOT_FOUND)
{
error_printf(_T("DataType type not resolved: %s\n"), Params.lplpszPropDataType[i]);
return 1;
}
else
dwDataType |= dwMapped;
}
}
//overwrite the default DataTypeType
AdmProp.SetDataType(dwDataType) ;
}
//LPCTSTR lplpszPropValue[MAX_NUMBER_OF_PROPERTY_VALUES]; //pointer to array of values (e.g multisz type allows the multiple values for one property
//WORD wPropValueCount;
//create admin object
if(oAdmin.GetpcAdmCom()==NULL)
{
oAdmin.Open(AdmNode.GetComputer());
if( FAILED(oAdmin.QueryLastHresError()))
{
retval= ConvertHresToDword(oAdmin.QueryLastHresError());
}
}
if(oAdmin.GetpcAdmCom()!=NULL)
{
//
// read from files if Params.wPropFileValueCount != 0
//
if ( Params.wPropFileValueCount )
{
if ( !ReadFromFiles( (LPTSTR*)Params.lplpszPropValue, Params.lpdwPropValueLength, Params.wPropFileValueCount ) )
{
error_printf(_T("Can't read value from file %s"), Params.lplpszPropValue[0] );
return 1;
}
Params.wPropValueCount = Params.wPropFileValueCount;
}
oAdmin.Run(CString(Params.lpszCommand),
AdmNode,
AdmProp,
AdmDstNode,
Params.lplpszPropValue,
Params.lpdwPropValueLength,
Params.wPropValueCount);
retval=ConvertHresToDword(oAdmin.QueryLastHresError());
}
return ((retval==0)?0:1);
}