335 lines
13 KiB
C
335 lines
13 KiB
C
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// bootini.h
|
||
|
|
//
|
||
|
|
// Module: WMI Instance provider code for boot parameters
|
||
|
|
//
|
||
|
|
// Purpose: Genral purpose include file.
|
||
|
|
//
|
||
|
|
// Copyright (c) 1997-1999 Microsoft Corporation
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#ifndef _bootini_H_
|
||
|
|
#define _bootini_H_
|
||
|
|
|
||
|
|
#include <wbemprov.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <tchar.h>
|
||
|
|
#include <windows.h>
|
||
|
|
#include <wbemcli.h>
|
||
|
|
|
||
|
|
typedef LPVOID * PPVOID;
|
||
|
|
|
||
|
|
// Provider interfaces are provided by objects of this class
|
||
|
|
|
||
|
|
class CBootInstPro : public IWbemServices, public IWbemProviderInit
|
||
|
|
{
|
||
|
|
protected:
|
||
|
|
ULONG m_cRef; //Object reference count
|
||
|
|
IWbemServices * m_pNamespace;
|
||
|
|
public:
|
||
|
|
CBootInstPro(BSTR ObjectPath = NULL, BSTR User = NULL, BSTR Password = NULL, IWbemContext * pCtx=NULL);
|
||
|
|
~CBootInstPro(void);
|
||
|
|
|
||
|
|
//Non-delegating object IUnknown
|
||
|
|
|
||
|
|
STDMETHODIMP QueryInterface(REFIID, PPVOID);
|
||
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
||
|
|
STDMETHODIMP_(ULONG) Release(void);
|
||
|
|
|
||
|
|
//IWbemProviderInit
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE Initialize(LPWSTR pszUser,
|
||
|
|
LONG lFlags,
|
||
|
|
LPWSTR pszNamespace,
|
||
|
|
LPWSTR pszLocale,
|
||
|
|
IWbemServices *pNamespace,
|
||
|
|
IWbemContext *pCtx,
|
||
|
|
IWbemProviderInitSink *pInitSink
|
||
|
|
);
|
||
|
|
|
||
|
|
//IWbemServices
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE OpenNamespace( const BSTR Namespace,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemServices __RPC_FAR *__RPC_FAR *ppWorkingNamespace,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppResult)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CancelAsyncCall( IWbemObjectSink __RPC_FAR *pSink)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE QueryObjectSink(long lFlags,
|
||
|
|
IWbemObjectSink __RPC_FAR *__RPC_FAR *ppResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE GetObject(const BSTR ObjectPath,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemClassObject __RPC_FAR *__RPC_FAR *ppObject,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE GetObjectAsync(const BSTR ObjectPath,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
);
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE PutClass(IWbemClassObject __RPC_FAR *pObject,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE PutClassAsync(IWbemClassObject __RPC_FAR *pObject,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DeleteClass(const BSTR Class,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DeleteClassAsync(const BSTR Class,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CreateClassEnum(const BSTR Superclass,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CreateClassEnumAsync(const BSTR Superclass,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE PutInstance(IWbemClassObject __RPC_FAR *pInst,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE PutInstanceAsync(IWbemClassObject __RPC_FAR *pInst,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
);
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DeleteInstance(const BSTR ObjectPath,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemCallResult __RPC_FAR *__RPC_FAR *ppCallResult
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE DeleteInstanceAsync(const BSTR ObjectPath,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CreateInstanceEnum(const BSTR Class,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CreateInstanceEnumAsync(const BSTR Class,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecQuery(const BSTR QueryLanguage,
|
||
|
|
const BSTR Query,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecQueryAsync(const BSTR QueryLanguage,
|
||
|
|
const BSTR Query,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecNotificationQuery(const BSTR QueryLanguage,
|
||
|
|
const BSTR Query,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IEnumWbemClassObject __RPC_FAR *__RPC_FAR *ppEnum
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecNotificationQueryAsync(const BSTR QueryLanguage,
|
||
|
|
const BSTR Query,
|
||
|
|
long lFlags,
|
||
|
|
IWbemContext __RPC_FAR *pCtx,
|
||
|
|
IWbemObjectSink __RPC_FAR *pResponseHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecMethod(const BSTR,
|
||
|
|
const BSTR,
|
||
|
|
long, IWbemContext*,
|
||
|
|
IWbemClassObject*,
|
||
|
|
IWbemClassObject**,
|
||
|
|
IWbemCallResult**
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE ExecMethodAsync(const BSTR,
|
||
|
|
const BSTR,
|
||
|
|
long,
|
||
|
|
IWbemContext*,
|
||
|
|
IWbemClassObject*,
|
||
|
|
IWbemObjectSink*
|
||
|
|
)
|
||
|
|
{
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CBootInstPro *PCBootInstPro;
|
||
|
|
|
||
|
|
// This class is the class factory for CInstPro objects.
|
||
|
|
|
||
|
|
class CBootProvFactory : public IClassFactory
|
||
|
|
{
|
||
|
|
protected:
|
||
|
|
ULONG m_cRef;
|
||
|
|
|
||
|
|
public:
|
||
|
|
CBootProvFactory(void);
|
||
|
|
~CBootProvFactory(void);
|
||
|
|
|
||
|
|
//IUnknown members
|
||
|
|
STDMETHODIMP QueryInterface(REFIID, PPVOID);
|
||
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
||
|
|
STDMETHODIMP_(ULONG) Release(void);
|
||
|
|
|
||
|
|
//IClassFactory members
|
||
|
|
STDMETHODIMP CreateInstance(LPUNKNOWN,
|
||
|
|
REFIID,
|
||
|
|
PPVOID
|
||
|
|
);
|
||
|
|
|
||
|
|
STDMETHODIMP LockServer(BOOL);
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CBootProvFactory *PCBootProvFactory;
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
// These variables keep track of when the module can be unloaded
|
||
|
|
|
||
|
|
extern long g_cObj;
|
||
|
|
extern long g_cLock;
|
||
|
|
|
||
|
|
// General purpose utilities.
|
||
|
|
|
||
|
|
SCODE
|
||
|
|
GetBootLoaderParameters(IWbemServices *m_pNamespace,
|
||
|
|
IWbemClassObject *pNewInst,
|
||
|
|
IWbemContext *pCtx
|
||
|
|
);
|
||
|
|
|
||
|
|
extern "C" BOOL WINAPI LibMain32(HINSTANCE hInstance,
|
||
|
|
ULONG ulReason,
|
||
|
|
LPVOID pvReserved
|
||
|
|
);
|
||
|
|
|
||
|
|
LONG
|
||
|
|
SaveBootFile(IWbemClassObject *pInst,
|
||
|
|
IWbemClassObject *pClass
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
HANDLE GetFileHandle(PCHAR data,
|
||
|
|
DWORD dwCreationDisposition,
|
||
|
|
DWORD dwAccess
|
||
|
|
);
|
||
|
|
PCHAR
|
||
|
|
GetBootFileName(
|
||
|
|
);
|
||
|
|
|
||
|
|
typedef struct {
|
||
|
|
CHAR Default[256];
|
||
|
|
long Delay;
|
||
|
|
CHAR Redirect[32];
|
||
|
|
} BootLoaderParams;
|
||
|
|
#if DBG==1
|
||
|
|
LPVOID BPAlloc(int len);
|
||
|
|
VOID BPFree(LPVOID mem);
|
||
|
|
#else
|
||
|
|
#define BPAlloc(x) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, x)
|
||
|
|
#define BPFree(x) HeapFree(GetProcessHeap(), 0, x)
|
||
|
|
#endif
|
||
|
|
extern BootLoaderParams blp;
|
||
|
|
#endif
|