mirror of
https://github.com/Ylianst/MeshAgent
synced 2025-12-06 00:13:33 +00:00
ab066751fb
2. Updated debugger to work with updated promise 3. Fixed bug in event emitter where 'explicit' event type wasn't honored 4. Updated _GenericMarshal to support interface and function marshaling 5. Added COM/WMI support to windows
378 lines
14 KiB
JavaScript
378 lines
14 KiB
JavaScript
/*
|
|
Copyright 2021 Intel Corporation
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
var promise = require('promise');
|
|
var GM = require('_GenericMarshal');
|
|
const CLSID_WbemAdministrativeLocator = '{CB8555CC-9128-11D1-AD9B-00C04FD8FDFF}';
|
|
const IID_WbemLocator = '{dc12a687-737f-11cf-884d-00aa004b2e24}';
|
|
const WBEM_FLAG_BIDIRECTIONAL = 0;
|
|
const WBEM_INFINITE = -1;
|
|
const WBEM_FLAG_ALWAYS = 0;
|
|
const E_NOINTERFACE = 0x80004002;
|
|
var OleAut32 = GM.CreateNativeProxy('OleAut32.dll');
|
|
OleAut32.CreateMethod('SafeArrayAccessData');
|
|
|
|
const LocatorFunctions = ['QueryInterface', 'AddRef', 'Release', 'ConnectToServer'];
|
|
const ServiceFunctions = [
|
|
'QueryInterface',
|
|
'AddRef',
|
|
'Release',
|
|
'OpenNamespace',
|
|
'CancelAsyncCall',
|
|
'QueryObjectSink',
|
|
'GetObject',
|
|
'GetObjectAsync',
|
|
'PutClass',
|
|
'PutClassAsync',
|
|
'DeleteClass',
|
|
'DeleteClassAsync',
|
|
'CreateClassEnum',
|
|
'CreateClassEnumAsync',
|
|
'PutInstance',
|
|
'PutInstanceAsync',
|
|
'DeleteInstance',
|
|
'DeleteInstanceAsync',
|
|
'CreateInstanceEnum',
|
|
'CreateInstanceEnumAsync',
|
|
'ExecQuery',
|
|
'ExecQueryAsync',
|
|
'ExecNotificationQuery',
|
|
'ExecNotificationQueryAsync',
|
|
'ExecMethod',
|
|
'ExecMethodAsync'
|
|
];
|
|
const ResultsFunctions = [
|
|
'QueryInterface',
|
|
'AddRef',
|
|
'Release',
|
|
'Reset',
|
|
'Next',
|
|
'NextAsync',
|
|
'Clone',
|
|
'Skip'
|
|
];
|
|
const ResultFunctions = [
|
|
'QueryInterface',
|
|
'AddRef',
|
|
'Release',
|
|
'GetQualifierSet',
|
|
'Get',
|
|
'Put',
|
|
'Delete',
|
|
'GetNames',
|
|
'BeginEnumeration',
|
|
'Next',
|
|
'EndEnumeration',
|
|
'GetPropertyQualifierSet',
|
|
'Clone',
|
|
'GetObjectText',
|
|
'SpawnDerivedClass',
|
|
'SpawnInstance',
|
|
'CompareTo',
|
|
'GetPropertyOrigin',
|
|
'InheritsFrom',
|
|
'GetMethod',
|
|
'PutMethod',
|
|
'DeleteMethod',
|
|
'BeginMethodEnumeration',
|
|
'NextMethod',
|
|
'EndMethodEnumeration',
|
|
'GetMethodQualifierSet',
|
|
'GetMethodOrigin'
|
|
];
|
|
|
|
const QueryAsyncHandler =
|
|
[
|
|
{
|
|
parms: 3, name: 'QueryInterface', func: function (j, riid, ppv)
|
|
{
|
|
var ret = GM.CreateVariable(4);
|
|
console.info1('QueryInterface', riid.Deref(0, 16).toBuffer().toString('hex'));
|
|
switch (riid.Deref(0, 16).toBuffer().toString('hex'))
|
|
{
|
|
case '0000000000000000C000000000000046': // IID_IUnknown
|
|
j.pointerBuffer().copy(ppv.Deref(0, 8).toBuffer());
|
|
ret.increment(0, true);
|
|
++this.p.refcount;
|
|
console.info1('QueryInterface (IID_IUnknown)', this.p.refcount);
|
|
break;
|
|
case '0178857C8173CF11884D00AA004B2E24': // IID_IWmiObjectSink
|
|
j.pointerBuffer().copy(ppv.Deref(0, 8).toBuffer());
|
|
ret.increment(0, true);
|
|
++this.p.refcount;
|
|
console.info1('QueryInterface (IID_IWmiObjectSink)', this.p.refcount);
|
|
break;
|
|
default:
|
|
ret.increment(E_NOINTERFACE, true);
|
|
console.info1('returning E_NOINTERFACE');
|
|
break;
|
|
}
|
|
|
|
return (ret);
|
|
}
|
|
},
|
|
{
|
|
parms: 1, name: 'AddRef', func: function ()
|
|
{
|
|
++this.p.refcount;
|
|
console.info1('AddRef', this.p.refcount);
|
|
return (GM.CreateVariable(4));
|
|
}
|
|
},
|
|
{
|
|
parms: 1, name: 'Release', func: function ()
|
|
{
|
|
--this.p.refcount;
|
|
console.info1('Release', this.p.refcount);
|
|
if (this.p.refcount == 0)
|
|
{
|
|
this.p.handlers.cleanup();
|
|
delete this.p.handlers;
|
|
delete this.p;
|
|
}
|
|
return (GM.CreateVariable(4));
|
|
}
|
|
},
|
|
{
|
|
parms: 3, name: 'Indicate', func: function (j, count, arr)
|
|
{
|
|
console.info1('Indicate', count.Val);
|
|
var j, nme, len, nn;
|
|
|
|
for (var i = 0; i < count.Val; ++i)
|
|
{
|
|
j = arr.Deref((i * GM.PointerSize) + 0, GM.PointerSize);
|
|
this.p.results.push(enumerateProperties(j));
|
|
}
|
|
|
|
var ret = GM.CreateVariable(4);
|
|
ret.increment(0, true);
|
|
return (ret);
|
|
}
|
|
},
|
|
{
|
|
parms: 5, name: 'SetStatus', func: function (j, lFlags, hResult, strParam, pObjParam)
|
|
{
|
|
console.info1('SetStatus', hResult.Val);
|
|
|
|
var ret = GM.CreateVariable(4);
|
|
ret.increment(0, true);
|
|
|
|
if (hResult.Val == 0)
|
|
{
|
|
this.p.resolve(this.p.results);
|
|
}
|
|
else
|
|
{
|
|
this.p.reject(hResult.Val);
|
|
}
|
|
return (ret);
|
|
}
|
|
}
|
|
];
|
|
|
|
|
|
function enumerateProperties(j)
|
|
{
|
|
var nme, len, nn;
|
|
|
|
j.funcs = require('win-com').marshalFunctions(j.Deref(), ResultFunctions);
|
|
nme = GM.CreatePointer();
|
|
j.funcs.GetNames(j.Deref(), 0, WBEM_FLAG_ALWAYS, 0, nme);
|
|
len = nme.Deref().Deref(24, 4).toBuffer().readUInt32LE();
|
|
|
|
nn = GM.CreatePointer();
|
|
OleAut32.SafeArrayAccessData(nme.Deref(), nn);
|
|
|
|
var properties = [];
|
|
var values = {};
|
|
|
|
for (var i = 0; i < len - 1; ++i)
|
|
{
|
|
properties.push(nn.Deref().increment(i * 8).Deref().Wide2UTF8);
|
|
}
|
|
|
|
for (var i = 0; i < properties.length; ++i)
|
|
{
|
|
var tmp1 = GM.CreateVariable(24);
|
|
if (j.funcs.Get(j.Deref(), GM.CreateVariable(properties[i], { wide: true }), 0, tmp1, 0, 0).Val == 0)
|
|
{
|
|
switch (tmp1.toBuffer().readUInt16LE())
|
|
{
|
|
case 0x0000: // VT_EMPTY
|
|
case 0x0001: // VT_NULL
|
|
values[properties[i]] = null;
|
|
break;
|
|
case 0x0002: // VT_I2
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt16LE();
|
|
break;
|
|
case 0x0003: // VT_I4
|
|
case 0x0016: // VT_INT
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt32LE();
|
|
break;
|
|
case 0x000B: // VT_BOOL
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt32LE() != 0;
|
|
break;
|
|
case 0x000E: // VT_DECIMAL
|
|
break;
|
|
case 0x0010: // VT_I1
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt8();
|
|
break;
|
|
case 0x0011: // VT_UI1
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt8();
|
|
break;
|
|
case 0x0012: // VT_UI2
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt16LE();
|
|
break;
|
|
case 0x0013: // VT_UI4
|
|
case 0x0017: // VT_UINT
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt32LE();
|
|
break;
|
|
//case 0x0014: // VT_I8
|
|
// break;
|
|
//case 0x0015: // VT_UI8
|
|
// break;
|
|
case 0x0008: // VT_BSTR
|
|
values[properties[i]] = tmp1.Deref(8, 8).Deref().Wide2UTF8;
|
|
break;
|
|
default:
|
|
console.info1('VARTYPE: ' + tmp1.toBuffer().readUInt16LE());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return (values);
|
|
}
|
|
|
|
function queryAsync(resourceString, queryString)
|
|
{
|
|
var p = new promise(require('promise').defaultInit);
|
|
var resource = GM.CreateVariable(resourceString, { wide: true });
|
|
var language = GM.CreateVariable("WQL", { wide: true });
|
|
var query = GM.CreateVariable(queryString, { wide: true });
|
|
var results = GM.CreatePointer();
|
|
|
|
p.refcount = 0;
|
|
p.results = [];
|
|
p.locator = require('win-com').createInstance(require('win-com').CLSIDFromString(CLSID_WbemAdministrativeLocator), require('win-com').IID_IUnknown);
|
|
p.locator.funcs = require('win-com').marshalFunctions(p.locator, LocatorFunctions);
|
|
p.services = require('_GenericMarshal').CreatePointer();
|
|
if (p.locator.funcs.ConnectToServer(p.locator, resource, 0, 0, 0, 0, 0, 0, p.services).Val != 0) { throw ('Error calling ConnectToService'); }
|
|
|
|
p.services.funcs = require('win-com').marshalFunctions(p.services.Deref(), ServiceFunctions);
|
|
p.handlers = require('win-com').marshalInterface(QueryAsyncHandler);
|
|
p.handlers.p = p;
|
|
|
|
if (p.services.funcs.ExecQueryAsync(p.services.Deref(), language, query, WBEM_FLAG_BIDIRECTIONAL, 0, p.handlers).Val != 0) { throw ('Error in Query'); }
|
|
return (p);
|
|
}
|
|
function query(resourceString, queryString)
|
|
{
|
|
var resource = GM.CreateVariable(resourceString, { wide: true });
|
|
var language = GM.CreateVariable("WQL", { wide: true });
|
|
var query = GM.CreateVariable(queryString, { wide: true });
|
|
var results = GM.CreatePointer();
|
|
|
|
var locator = require('win-com').createInstance(require('win-com').CLSIDFromString(CLSID_WbemAdministrativeLocator), require('win-com').IID_IUnknown);
|
|
locator.funcs = require('win-com').marshalFunctions(locator, LocatorFunctions);
|
|
var services = require('_GenericMarshal').CreatePointer();
|
|
if (locator.funcs.ConnectToServer(locator, resource, 0, 0, 0, 0, 0, 0, services).Val != 0) { throw ('Error calling ConnectToService'); }
|
|
|
|
services.funcs = require('win-com').marshalFunctions(services.Deref(), ServiceFunctions);
|
|
if (services.funcs.ExecQuery(services.Deref(), language, query, WBEM_FLAG_BIDIRECTIONAL, 0, results).Val != 0) { throw ('Error in Query'); }
|
|
|
|
results.funcs = require('win-com').marshalFunctions(results.Deref(), ResultsFunctions);
|
|
var returnedCount = GM.CreateVariable(8);
|
|
var result = GM.CreatePointer();
|
|
var ret = [];
|
|
|
|
while (results.funcs.Next(results.Deref(), WBEM_INFINITE, 1, result, returnedCount).Val == 0)
|
|
{
|
|
result.funcs = require('win-com').marshalFunctions(result.Deref(), ResultFunctions);
|
|
|
|
if (result.funcs.GetNames == null) { console.log('skipskip'); continue; }
|
|
var nme = GM.CreatePointer();
|
|
var rr = result.funcs.GetNames(result.Deref(), 0, WBEM_FLAG_ALWAYS, 0, nme);
|
|
var len = nme.Deref().Deref(24, 4).toBuffer().readUInt32LE();
|
|
|
|
var nn = GM.CreatePointer();
|
|
OleAut32.SafeArrayAccessData(nme.Deref(), nn);
|
|
|
|
var properties = [];
|
|
var values = {};
|
|
|
|
for (var i = 0; i < len - 1; ++i)
|
|
{
|
|
properties.push(nn.Deref().increment(i * 8).Deref().Wide2UTF8);
|
|
}
|
|
|
|
for(var i=0;i<properties.length;++i)
|
|
{
|
|
var tmp1 = GM.CreateVariable(24);
|
|
if(result.funcs.Get(result.Deref(), GM.CreateVariable(properties[i], {wide: true}), 0, tmp1, 0, 0).Val == 0)
|
|
{
|
|
console.log('VARTYPE: ' + tmp1.toBuffer().readUInt16LE());
|
|
switch(tmp1.toBuffer().readUInt16LE())
|
|
{
|
|
case 0x0000: // VT_EMPTY
|
|
case 0x0001: // VT_NULL
|
|
values[properties[i]] = null;
|
|
break;
|
|
case 0x0002: // VT_I2
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt16LE();
|
|
break;
|
|
case 0x0003: // VT_I4
|
|
case 0x0016: // VT_INT
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt32LE();
|
|
break;
|
|
case 0x000B: // VT_BOOL
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt32LE() != 0;
|
|
break;
|
|
case 0x000E: // VT_DECIMAL
|
|
break;
|
|
case 0x0010: // VT_I1
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readInt8();
|
|
break;
|
|
case 0x0011: // VT_UI1
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt8();
|
|
break;
|
|
case 0x0012: // VT_UI2
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt16LE();
|
|
break;
|
|
case 0x0013: // VT_UI4
|
|
case 0x0017: // VT_UINT
|
|
values[properties[i]] = tmp1.Deref(8, 8).toBuffer().readUInt32LE();
|
|
break;
|
|
case 0x0014: // VT_I8
|
|
break;
|
|
case 0x0015: // VT_UI8
|
|
break;
|
|
case 0x0008: // VT_BSTR
|
|
values[properties[i]] = tmp1.Deref(8, 8).Deref().Wide2UTF8;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
ret.push(values);
|
|
}
|
|
|
|
return (ret);
|
|
}
|
|
|
|
module.exports = { query: query, queryAsync: queryAsync };
|