94 lines
3.4 KiB
C
94 lines
3.4 KiB
C
|
#include <tchar.h>
|
||
|
|
#include <olectl.h>
|
||
|
|
#include <setupapi.h>
|
||
|
|
#include "interface.h"
|
||
|
|
#include "simpledb.h"
|
||
|
|
|
||
|
|
#define BUFFER_SIZE 64*1024
|
||
|
|
|
||
|
|
struct _SectionList;
|
||
|
|
typedef _SectionList SECTION, *PSECTION;
|
||
|
|
|
||
|
|
struct _SectionEntry;
|
||
|
|
typedef _SectionEntry SECTIONENTRY, *PSECTIONENTRY;
|
||
|
|
|
||
|
|
|
||
|
|
#define HASH_BUCKETS 128
|
||
|
|
struct _SectionAssociationList;
|
||
|
|
typedef _SectionAssociationList SECTIONASSOCIATIONLIST, *PSECTIONASSOCIATIONLIST;
|
||
|
|
|
||
|
|
class CUpdateInf : public IUpdateInf
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// IUnknown
|
||
|
|
ULONG __stdcall AddRef( void );
|
||
|
|
ULONG __stdcall Release( void );
|
||
|
|
HRESULT __stdcall QueryInterface( REFIID, void ** );
|
||
|
|
|
||
|
|
// IDispatch
|
||
|
|
HRESULT _stdcall GetTypeInfoCount( UINT * );
|
||
|
|
HRESULT __stdcall GetTypeInfo( UINT, LCID, ITypeInfo ** );
|
||
|
|
HRESULT __stdcall GetIDsOfNames( REFIID, LPOLESTR *, UINT, LCID, DISPID * );
|
||
|
|
HRESULT __stdcall Invoke( DISPID, REFIID, LCID, WORD, DISPPARAMS *, VARIANT *, EXCEPINFO *, UINT * );
|
||
|
|
|
||
|
|
// IUpdateInf
|
||
|
|
HRESULT __stdcall InsertFile( BSTR bstrFileName );
|
||
|
|
HRESULT __stdcall WriteSectionData( BSTR bstrSection, BSTR bstrValue );
|
||
|
|
HRESULT __stdcall SetConfigurationField( BSTR bstrFieldName, BSTR bstrValue );
|
||
|
|
HRESULT __stdcall AddSourceDisksFilesEntry( BSTR bstrFile, BSTR bstrTag );
|
||
|
|
HRESULT __stdcall AddEquality( BSTR bstrSection, BSTR bstrLVal, BSTR bstrRVal );
|
||
|
|
HRESULT __stdcall SetVersionField( BSTR bstrFieldName, BSTR bstrValue );
|
||
|
|
HRESULT __stdcall SetDB( BSTR bstrServer, BSTR bstrDB, BSTR bstrUser, BSTR bstrPassword );
|
||
|
|
HRESULT __stdcall InitGen( BSTR bstrInxFile, BSTR bstrInfFile );
|
||
|
|
HRESULT __stdcall CloseGen( BOOL bTrimInf );
|
||
|
|
HRESULT __stdcall get_InfGenError( BSTR *bstrError );
|
||
|
|
|
||
|
|
CUpdateInf();
|
||
|
|
~CUpdateInf();
|
||
|
|
BOOL Init();
|
||
|
|
private:
|
||
|
|
void Cleanup();
|
||
|
|
BOOL TrimInf(LPTSTR szINFIn, LPTSTR szINFOut);
|
||
|
|
BOOL ReverseSectionList( void );
|
||
|
|
BOOL WriteSmallINF(LPTSTR szINFIn, LPTSTR szINFOut);
|
||
|
|
BOOL IdentifyUninstallSections();
|
||
|
|
BOOL DeleteUnusedEntries();
|
||
|
|
BOOL DeleteUnusedDirIdSections();
|
||
|
|
BOOL MarkAssociatedEntriesForDelete(PSECTION ps);
|
||
|
|
BOOL RemoveSectionFromMultiEntry(PSECTIONENTRY pse, LPCTSTR szSectionName);
|
||
|
|
BOOL AddEntryToSection(PSECTION ps, LPCTSTR szEntry);
|
||
|
|
BOOL AssociateEntryWithSection(PSECTIONENTRY pse, LPCTSTR szSectionName, BOOL fMultiEntry);
|
||
|
|
DWORD CalcHashFromSectionName(LPCTSTR szSectionName);
|
||
|
|
BOOL ReadSectionEntries(LPCTSTR szINF);
|
||
|
|
BOOL GetSectionListFromInF( LPTSTR szINF );
|
||
|
|
DWORD GetFileSizeByName(IN LPCTSTR pszFileName, OUT PDWORD pdwFileSizeHigh );
|
||
|
|
|
||
|
|
DWORD m_bGenInitCalled;
|
||
|
|
DWORD m_dwInfGenError;
|
||
|
|
|
||
|
|
TCHAR m_textBuffer[ BUFFER_SIZE ];
|
||
|
|
TCHAR m_textBuffer2[ BUFFER_SIZE ];
|
||
|
|
TCHAR m_textBuffer3[ BUFFER_SIZE ];
|
||
|
|
|
||
|
|
TCHAR m_szInxFile[ MAX_PATH]; // The inf template. This acts as a database
|
||
|
|
TCHAR m_szOutFile[ MAX_PATH]; // The output file. This is generated.
|
||
|
|
TCHAR m_szFilledInxFile[ MAX_PATH]; // The intermediate work file where the files are injected.
|
||
|
|
|
||
|
|
TCHAR m_szDataServer[ MAX_PATH ]; // Name of the server hosting the DB
|
||
|
|
TCHAR m_szDatabase[ MAX_PATH ]; // Name of the database
|
||
|
|
TCHAR m_szUserName[ MAX_PATH ]; // Username to connect to database with
|
||
|
|
TCHAR m_szPassword[ MAX_PATH ]; // password to supply to database
|
||
|
|
|
||
|
|
HINF m_hInf;
|
||
|
|
|
||
|
|
BOOL m_bActiveDB;
|
||
|
|
|
||
|
|
PSECTION m_sectionList;
|
||
|
|
_SectionAssociationList *m_rgNameHash[HASH_BUCKETS];
|
||
|
|
|
||
|
|
CSimpleDatabase *m_pdb;
|
||
|
|
|
||
|
|
ULONG m_cRef;
|
||
|
|
ITypeInfo *m_pTypeInfo;
|
||
|
|
};
|