summaryrefslogtreecommitdiff
path: root/storage/connect/tabwmi.h
blob: ede84896166a4e6975d76db4465e0edbdfe226cf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
// TABWMI.H     Olivier Bertrand    2012
// WMI: Virtual table to Get WMI information
#define _WIN32_DCOM
#include <wbemidl.h>
# pragma comment(lib, "wbemuuid.lib")
#include <iostream>
using namespace std;
#include <comdef.h>

/***********************************************************************/
/*  Definitions.                                                       */
/***********************************************************************/
typedef class WMIDEF *PWMIDEF;
typedef class TDBWMI *PTDBWMI;
typedef class WMICOL *PWMICOL;
typedef class TDBWCL *PTDBWCL;
typedef class WCLCOL *PWCLCOL;

/* -------------------------- WMI classes ---------------------------- */

/***********************************************************************/
/*  WMI: Virtual table to get the WMI information.                     */
/***********************************************************************/
class WMIDEF : public TABDEF {            /* Logical table description */
	friend class TDBWMI;
	friend class TDBWCL;
 public:
  // Constructor
	WMIDEF(void) 
		{Pseudo = 3; Nspace = NULL; Wclass = NULL; Ems = 0; Info = false;}

  // Implementation
  virtual const char *GetType(void) {return "WMI";}

  // Methods
  virtual bool DefineAM(PGLOBAL g, LPCSTR am, int poff);
  virtual PTDB GetTable(PGLOBAL g, MODE m);
  virtual bool DeleteTableFile(PGLOBAL g) {return true;}

 protected:
  // Members
  char   *Nspace;
	char   *Wclass;
	int     Ems;
	bool    Info;
  }; // end of WMIDEF

/***********************************************************************/
/*  This is the class declaration for the WMI table.                   */
/***********************************************************************/
class TDBWMI : public TDBASE {
  friend class WMICOL;
 public:
  // Constructor
  TDBWMI(PWMIDEF tdp);

  // Implementation
  virtual AMT  GetAmType(void) {return TYPE_AM_WMI;}

  // Methods
	virtual int GetRecpos(void);
  virtual int GetProgCur(void) {return N;}
	virtual int RowNumber(PGLOBAL g, bool b = false) {return N + 1;}

  // Database routines
	virtual PCOL MakeCol(PGLOBAL g, PCOLDEF cdp, PCOL cprec, int n);
  virtual int  GetMaxSize(PGLOBAL g);
  virtual bool OpenDB(PGLOBAL g);
  virtual int  ReadDB(PGLOBAL g);
  virtual int  WriteDB(PGLOBAL g);
  virtual int  DeleteDB(PGLOBAL g, int irc);
	virtual void CloseDB(PGLOBAL g);

 protected:
	// Specific routines
					bool  Initialize(PGLOBAL g);
					char *MakeWQL(PGLOBAL g);
					void  DoubleSlash(PGLOBAL g);
					bool  GetWMIInfo(PGLOBAL g);

  // Members
  IWbemServices        *Svc;			// IWbemServices pointer
  IEnumWbemClassObject *Enumerator;
  IWbemClassObject     *ClsObj;
	char                 *Nspace;		// Namespace
	char                 *Wclass;		// Class name
	char                 *ObjPath;  // Used for direct access
	char                 *Kvp;			// Itou
	int	  								Ems;			// Estimated max size
	PCOL                  Kcol;     // Key column
  HRESULT               Res;
	PVBLK                 Vbp;
	bool                  Init;
	bool                  Done;
  ULONG									Rc;
	int                   N;				// Row number
  }; // end of class TDBWMI

/***********************************************************************/
/*  Class WMICOL: WMI Address column.                                  */
/***********************************************************************/
class WMICOL : public COLBLK {
	friend class TDBWMI;
 public:
  // Constructors
  WMICOL(PCOLDEF cdp, PTDB tdbp, int n);

  // Implementation
  virtual int  GetAmType(void) {return TYPE_AM_WMI;}

  // Methods
  virtual void ReadColumn(PGLOBAL g);

 protected:
  WMICOL(void) {}							// Default constructor not to be used

  // Members
	PTDBWMI Tdbp;								// Points to WMI table block
	VARIANT Prop;               // Property value
	CIMTYPE Ctype;							// CIM Type
  HRESULT Res;
  }; // end of class WMICOL

/***********************************************************************/
/*  This is the class declaration for the WCL table.                   */
/***********************************************************************/
class TDBWCL : public TDBASE {
  friend class WCLCOL;
 public:
  // Constructor
  TDBWCL(PWMIDEF tdp);

  // Implementation
  virtual AMT  GetAmType(void) {return TYPE_AM_WMI;}

  // Methods
	virtual int  GetRecpos(void) {return N;}
  virtual int  GetProgCur(void) {return N;}
	virtual int  RowNumber(PGLOBAL g, bool b = false) {return N + 1;}

  // Database routines
	virtual PCOL MakeCol(PGLOBAL g, PCOLDEF cdp, PCOL cprec, int n);
  virtual int  GetMaxSize(PGLOBAL g);
  virtual bool OpenDB(PGLOBAL g);
  virtual int  ReadDB(PGLOBAL g);
  virtual int  WriteDB(PGLOBAL g);
  virtual int  DeleteDB(PGLOBAL g, int irc);
	virtual void CloseDB(PGLOBAL g);

 protected:
	// Specific routines
					bool Initialize(PGLOBAL g);

  // Members
  IWbemServices    *Svc;			// IWbemServices pointer
  IWbemClassObject *ClsObj;
	BSTR						  Propname;
	char             *Nspace;		// Namespace
	char             *Wclass;		// Class name
  HRESULT           Res;
	bool              Init;
	bool              Done;
	int               N;				// Row number
	int							  Lng;
	int							  Typ;
	int							  Prec;
  }; // end of class TDBWCL

/***********************************************************************/
/*  Class WMICOL: WMI Address column.                                  */
/***********************************************************************/
class WCLCOL : public COLBLK {
	friend class TDBWCL;
 public:
  // Constructors
  WCLCOL(PCOLDEF cdp, PTDB tdbp, int n);

  // Implementation
  virtual int  GetAmType(void) {return TYPE_AM_WMI;}

  // Methods
  virtual void ReadColumn(PGLOBAL g);

 protected:
  WCLCOL(void) {}							// Default constructor not to be used

  // Members
	PTDBWCL Tdbp;								// Points to WMI table block
  HRESULT Res;
	int     Flag;
  }; // end of class WCLCOL