MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-24 12:13:16 +00:00
Code Issues Releases Wiki Activity

1. Fixed uncaught exception in websocket that could happen if peer disconnects

Browse Source 
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
This commit is contained in:
Bryan Roe
2021-10-29 09:39:25 -07:00
parent fa82a9ed76
commit ab066751fb
7 changed files with 1054 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

124
modules/win-com.js Normal file
View File

@@ -0,0 +1,124 @@
/*
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.
*/
const CLSCTX_INPROC_SERVER = 1;
const CLSCTX_LOCAL_SERVER = 4;
const EOAC_NONE = 0;
const RPC_C_AUTHN_LEVEL_DEFAULT = 0;
const RPC_C_IMP_LEVEL_IMPERSONATE = 3;
const COINIT_MULTITHREADED = 0;
var GM = require('_GenericMarshal');
var ole32 = GM.CreateNativeProxy('ole32.dll');
ole32.CreateMethod('CLSIDFromString');
ole32.CreateMethod('CoCreateInstance');
ole32.CreateMethod('CoInitializeSecurity');
ole32.CreateMethod('CoInitialize');
ole32.CreateMethod('CoInitializeEx');
ole32.CreateMethod('IIDFromString');
ole32.CreateMethod('StringFromCLSID');
ole32.CreateMethod('StringFromIID');
function createInstance(RFCLSID, RFIID, options)
{
ole32.CoInitializeEx(0, COINIT_MULTITHREADED);
ole32.CoInitializeSecurity(0, -1, 0, 0, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, 0, EOAC_NONE, 0);
var ppv = GM.CreatePointer();
var h;
if ((h = ole32.CoCreateInstance(RFCLSID, 0, CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER, RFIID, ppv)).Val == 0)
{
return (ppv.Deref());
}
throw ('Error calling CoCreateInstance(' + h.Val + ')');
}
function CLSIDFromString(CLSIDString)
{
var v = GM.CreateVariable(CLSIDString, { wide: true });
var rfclsid = GM.CreateVariable(16);
if(ole32.CLSIDFromString(v, rfclsid).Val == 0)
{
return (rfclsid);
}
else
{
throw ('Error Converting CLSIDString');
}
}
function IIDFromString(IIDString)
{
var v = GM.CreateVariable(IIDString, { wide: true });
var rfiid = GM.CreateVariable(16);
if(ole32.IIDFromString(v, rfiid).Val==0)
{
return (rfiid);
}
else
{
throw ('Error Converting IIDString');
}
}
function marshalFunctions(obj, arr)
{
return (GM.MarshalFunctions(obj.Deref(), arr));IID_IUnknown
}
function marshalInterface(arr)
{
var vtbl = GM.CreateVariable(arr.length * GM.PointerSize);
var obj = GM.CreatePointer();
vtbl.pointerBuffer().copy(obj.toBuffer());
obj._gcallbacks = [];
obj.cleanup = function ()
{
var v;
while (this._gcallbacks.length > 0)
{
v = this._gcallbacks.pop();
v.removeAllListeners('GlobalCallback');
GM.PutGenericGlobalCallbackEx(v);
}
};
for (var i = 0; i < arr.length; ++i)
{
_hide(GM.GetGenericGlobalCallbackEx(arr[i].parms));
_hide()._ObjectID = 'GlobalCallback_' + arr[i].name;
obj._gcallbacks.push(_hide());
_hide().obj = arr[i];
_hide().pointerBuffer().copy(vtbl.Deref(i * GM.PointerSize, GM.PointerSize).toBuffer());
_hide(true).on('GlobalCallback', function ()
{
if (arguments[0]._ptr == obj._ptr)
{
var args = [];
for (var i in arguments)
{
args.push(arguments[i]);
}
return (this.obj.func.apply(obj, args));
}
});
}
return (obj);
}
module.exports = { createInstance: createInstance, marshalFunctions: marshalFunctions, marshalInterface: marshalInterface, CLSIDFromString: CLSIDFromString, IIDFromString: IIDFromString, IID_IUnknown: IIDFromString('{00000000-0000-0000-C000-000000000046}') };

377
modules/win-wmi.js Normal file
View File

@@ -0,0 +1,377 @@
/*
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 };