MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2026-01-01 08:03:32 +00:00
Code Issues Releases Wiki Activity

1. Updated win-message-pump to support dispatch queueing

Browse Source 
2. Added initial version of enhanced user-consent dialog on windows
This commit is contained in:
Bryan Roe
2021-12-10 09:14:38 -08:00
parent 5752c376c3
commit 4a481db743
3 changed files with 452 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
modules/win-message-pump.js
View File

@@ -37,22 +37,50 @@ function WindowsMessagePump(options)
this._user32 = GM.CreateNativeProxy('User32.dll');
this._user32.mp = this;
this._user32.CreateMethod('BeginPaint');
this._user32.CreateMethod('CreateWindowExW');
this._user32.CreateMethod('DefWindowProcW');
this._user32.CreateMethod('DestroyWindow');
this._user32.CreateMethod('DispatchMessageW');
this._user32.CreateMethod('EndPaint');
this._user32.CreateMethod('FillRect');
this._user32.CreateMethod('GetClientRect');
this._user32.CreateMethod('GetDC');
this._user32.CreateMethod('GetMessageW');
this._user32.CreateMethod('InvalidateRect')
this._user32.CreateMethod('PostMessageA');
this._user32.CreateMethod('RegisterClassExW');
this._user32.CreateMethod('SendMessageW');
this._user32.CreateMethod('SetWindowPos');
this._user32.CreateMethod('ShowWindow');
this._user32.CreateMethod('SystemParametersInfoA');
this._user32.CreateMethod('TranslateMessage');
this._user32.CreateMethod('IsDlgButtonChecked');
this._user32.CreateMethod('CheckDlgButton');
this._gdi32 = GM.CreateNativeProxy('Gdi32.dll');
this._gdi32.mp = this;
this._gdi32.CreateMethod('CreateFontW');
this._gdi32.CreateMethod('CreateSolidBrush');
this._gdi32.CreateMethod('SetBkColor');
this._gdi32.CreateMethod('SetBkMode');
this._gdi32.CreateMethod('SetTextColor');
this._pendingCreate = [];
this.wndclass = GM.CreateVariable(GM.PointerSize == 4 ? 48 : 80);
this.wndclass.mp = this;
this.wndclass.hinstance = this._kernel32.GetModuleHandleA(0);
//this.wndclass.cname = GM.CreateVariable('MainWWWClass');
if (options && options.window && options.window.background != null)
{
console.info1('SETTING BACKGROUND BRUSH');
this.wndclass.bkbrush = this._gdi32.CreateSolidBrush(options.window.background);
this.wndclass.bkbrush.pointerBuffer().copy(this.wndclass.Deref(GM.PointerSize == 4 ? 32 : 48, GM.PointerSize).toBuffer())
}
this.wndclass.cnamew = GM.CreateVariable('MainWWWClass', { wide: true });
this.wndclass.wndproc = GM.GetGenericGlobalCallback(4);
this.wndclass.wndproc.mp = this;
@@ -83,9 +111,17 @@ function WindowsMessagePump(options)
}
else
{
var r = GM.CreatePointer();
r.Val = msgRet;
this.EndDispatcher(r);
console.info1('RETURN VALUE DETECTED', msgRet._ObjectID);
if (msgRet._ObjectID == '_GenericMarshal.Variable')
{
this.EndDispatcher(msgRet);
}
else
{
var r = GM.CreatePointer();
r.Val = msgRet;
this.EndDispatcher(r);
}
}
}
else if(this.mp._hwnd == null && this.CallingThread() == this.mp._user32.RegisterClassExW.async.threadId())
@@ -148,8 +184,40 @@ function WindowsMessagePump(options)
}
});
});
this._addAsyncMethodCall = function addAsyncMethodCall(func, args)
{
var promise = require('promise');
var ret = new promise(promise.defaultInit);
this._pendingCreate.push({ func: func, args: args, p: ret });
return (ret);
}
this._addCreateWindowEx = function _addCreateWindowEx(dwExStyle, lpClassName, lpWindowName, dwStyle, X, Y, nWidth, nHeight, hWndParent, hMenu, hInstance, lpParam)
{
var args = [];
for (var i in arguments)
{
args.push(arguments[i]);
}
return (this._addAsyncMethodCall(this._user32.CreateWindowExW.async, args));
}
this._startPump_continuation = function _startPump_continuation(h)
{
this.finalpromise.resolve(h);
this.nativeProxy.mp._startPump();
}
this._startPump = function _startPump()
{
if (this._pendingCreate.length > 0)
{
var j = this._pendingCreate.shift();
var args = j.args;
args.unshift(this._user32.RegisterClassExW.async);
var p2 = j.func.apply(this._user32, args);
p2.finalpromise = j.p;
p2.then(this._startPump_continuation);
return;
}
this._user32.GetMessageW.async(this._user32.RegisterClassExW.async, this._msg, this._hwnd, 0, 0).then(function (r)
{
if(r.Val > 0)

380
modules/win-userconsent.js Normal file
View File

@@ -0,0 +1,380 @@
/*
Copyright 2021 Intel Corporation
@author Bryan Roe
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 WM_SETFONT = 0x0030;
const WM_CTLCOLORSTATIC = 0x0138;
const WM_CREATE = 0x0001;
const WS_TABSTOP = 0x00010000;
const WS_VISIBLE = 0x10000000;
const WS_CHILD = 0x40000000;
const BS_DEFPUSHBUTTON = 0x00000001;
const BS_CHECKBOX = 0x00000002;
const WM_ERASEBKGND = 0x0014;
const WM_USER = 0x0400;
const EM_SETBKGNDCOLOR = (WM_USER + 67);
const WM_COMMAND = 0x0111;
const SPI_GETWORKAREA = 0x0030;
const BST_UNCHECKED = 0x0000;
const BST_CHECKED = 0x0001;
const SS_BLACKRECT = 0x00000004;
const SS_GRAYRECT = 0x00000005;
const SS_LEFT = 0x00000000;
const SS_CENTER = 0x00000001;
const SS_RIGHT = 0x00000002;
const SS_REALSIZECONTROL = 0x00000040;
const DS_CENTER = 0x0800;
const FW_DONTCARE = 0;
const FW_THIN = 100;
const FW_EXTRALIGHT = 200;
const FW_LIGHT = 300;
const FW_NORMAL = 400;
const FW_MEDIUM = 500;
const FW_SEMIBOLD = 600;
const FW_BOLD = 700;
const FW_EXTRABOLD = 800;
const FW_HEAVY = 900;
const DEFAULT_CHARSET = 1;
const OUT_DEFAULT_PRECIS = 0;
const OUT_STRING_PRECIS = 1;
const OUT_CHARACTER_PRECIS = 2;
const OUT_STROKE_PRECIS = 3;
const OUT_TT_PRECIS = 4;
const OUT_DEVICE_PRECIS = 5;
const OUT_RASTER_PRECIS = 6;
const OUT_TT_ONLY_PRECIS = 7;
const OUT_OUTLINE_PRECIS = 8;
const OUT_SCREEN_OUTLINE_PRECIS = 9;
const OUT_PS_ONLY_PRECIS = 10;
const CLIP_DEFAULT_PRECIS = 0;
const CLIP_CHARACTER_PRECIS = 1;
const CLIP_STROKE_PRECIS = 2;
const CLIP_MASK = 0xf;
const CLIP_LH_ANGLES = (1 << 4);
const CLIP_TT_ALWAYS = (2 << 4);
const DEFAULT_QUALITY = 0;
const DRAFT_QUALITY = 1;
const PROOF_QUALITY = 2;
const DEFAULT_PITCH = 0;
const FIXED_PITCH = 1;
const VARIABLE_PITCH = 2;
const FF_DONTCARE = (0 << 4); /* Don't care or don't know. */
const FF_ROMAN = (1 << 4); /* Variable stroke width, serifed. */
const FF_SWISS = (2 << 4); /* Variable stroke width, sans-serifed. */
const FF_MODERN = (3 << 4); /* Constant stroke width, serifed or sans-serifed. */
const FF_SCRIPT = (4 << 4); /* Cursive, etc. */
const FF_DECORATIVE = (5 << 4); /* Old English, etc. */
const ES_LEFT = 0x0000;
const ES_CENTER = 0x0001;
const ES_RIGHT = 0x0002;
const ES_MULTILINE = 0x0004;
const ES_UPPERCASE = 0x0008;
const ES_LOWERCASE = 0x0010;
const ES_PASSWORD = 0x0020;
const ES_AUTOVSCROLL = 0x0040;
const ES_AUTOHSCROLL = 0x0080;
const ES_NOHIDESEL = 0x0100;
const ES_OEMCONVERT = 0x0400;
const ES_READONLY = 0x0800;
const ES_WANTRETURN = 0x1000;
const STM_SETIMAGE = 0x0172;
const STM_GETIMAGE = 0x0173;
const IMAGE_BITMAP = 0;
const SS_BITMAP = 0x0000000E;
const TRANSPARENT = 1;
const OPAQUE = 2;
const COLOR_BACKGROUND = 1;
var promise = require('promise');
var GM = require('_GenericMarshal');
var MessagePump = require('win-message-pump');
var SHM = GM.CreateNativeProxy('Shlwapi.dll');
var gdip = GM.CreateNativeProxy('Gdiplus.dll');
SHM.CreateMethod('SHCreateMemStream');
gdip.CreateMethod('GdipLoadImageFromStream');
gdip.CreateMethod('GdipCreateBitmapFromStream');
gdip.CreateMethod('GdipCreateHBITMAPFromBitmap');
gdip.CreateMethod('GdiplusStartup');
gdip.CreateMethod('GdiplusShutdown');
function RGB(r, g, b)
{
return (r | (g << 8) | (b << 16));
}
function CENTER(w, cx, cw)
{
var a = cw / 2;
var b = w / 2;
return (Math.floor(cx + (a - b)));
}
function pump_onTimeout(pump)
{
pump.promise.reject('TIMEOUT');
pump.close();
}
function pump_onExit()
{
console.info1('message pump exited');
this.promise.reject('CLOSED');
}
function pump_onMessage(msg)
{
switch (msg.message)
{
case WM_COMMAND:
switch(msg.wparam)
{
case 0xFFF0:
this._addAsyncMethodCall(this._user32.IsDlgButtonChecked.async, [this._HANDLE, 0xFFF0]).then(function (v)
{
if (v.Val == 0)
{
this.pump.autoAccept = true;
this.pump._addAsyncMethodCall(this.pump._user32.CheckDlgButton.async, [this.pump._HANDLE, 0xFFF0, BST_CHECKED]);
}
else
{
this.pump.autoAccept = false;
this.pump._addAsyncMethodCall(this.pump._user32.CheckDlgButton.async, [this.pump._HANDLE, 0xFFF0, BST_UNCHECKED]);
}
}).parentPromise.pump = this;
break;
case 0xFFF2: // Allow
if (this.timeout != null) { clearTimeout(this.timeout); this.timeout = null; }
this.promise.resolve(this.autoAccept);
this.close();
break;
case 0xFFF3: // Deny
if (this.timeout != null) { clearTimeout(this.timeout); this.timeout = null; }
this.promise.reject('DENIED');
this.close();
break;
}
break;
case WM_CTLCOLORSTATIC:
console.info1('WM_CTLCOLORSTATIC => ' + msg.lparam, msg.wparam);
var hdcStatic = msg.wparam;
this._gdi32.SetTextColor(hdcStatic, RGB(200, 200, 200));
this._gdi32.SetBkColor(hdcStatic, RGB(0, 54, 105));
return (this.brush);
break;
case WM_CREATE:
console.info1('WM_CREATE');
break;
case WM_ERASEBKGND:
console.info1('WM_ERASEBKGND');
break;
default:
//console.log(msg.message);
break;
}
}
function pump_onHwnd(h)
{
this._HANDLE = h;
this._addCreateWindowEx(0, GM.CreateVariable('BUTTON', { wide: true }), GM.CreateVariable(this.translations.Allow, { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | BS_DEFPUSHBUTTON,
345, // x position
225, // y position
100, // Button width
30, // Button height
h, // Parent window
0xFFF2, // Child ID
0,
0);
this._addCreateWindowEx(0, GM.CreateVariable('BUTTON', { wide: true }), GM.CreateVariable(this.translations.Deny, { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | BS_DEFPUSHBUTTON,
455, // x position
225, // y position
100, // Button width
30, // Button height
h, // Parent window
0xFFF3, // Child ID
0,
0);
this._addCreateWindowEx(0, GM.CreateVariable('BUTTON', { wide: true }), GM.CreateVariable(this.translations.Auto, { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | BS_CHECKBOX,
210, // x position
180, // y position
335, // Button width
30, // Button height
h, // Parent window
0xFFF0, // Child ID
0,
0);
this._addCreateWindowEx(0, GM.CreateVariable('STATIC', { wide: true }), GM.CreateVariable('NONE', { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | SS_BLACKRECT | SS_BITMAP | SS_REALSIZECONTROL,
10, // x position
10, // y position
192, // Button width
192, // Button height
h, // Parent window
0xFFF1, // Child ID
0,
0).then(function (h)
{
this.pump._addAsyncMethodCall(this.pump._user32.SendMessageW.async, [h, STM_SETIMAGE, IMAGE_BITMAP, this.pump.options.bitmap.Deref()]);
}).parentPromise.pump = this;
this._addCreateWindowEx(0, GM.CreateVariable('STATIC', { wide: true }), GM.CreateVariable(this.username, { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | SS_LEFT,
10, // x position
225, // y position
192, // Button width
30, // Button height
h, // Parent window
0xFFF2, // Child ID
0,
0);
this._addCreateWindowEx(0, GM.CreateVariable('STATIC', { wide: true }), GM.CreateVariable(this.translations.Caption, { wide: true }), WS_TABSTOP | WS_VISIBLE | WS_CHILD | SS_LEFT,
210, // x position
10, // y position
350, // Button width
150, // Button height
h, // Parent window
0xFFF3, // Child ID
0,
0).then(function (h)
{
this.pump._addAsyncMethodCall(this.pump._user32.SendMessageW.async, [h, WM_SETFONT, this.pump.font, 1]);
}).parentPromise.pump = this;
}
function createLocal(title, caption, username, options)
{
if (options == null) { options = {}; }
if (!options.translations)
{
options.translations =
{
Allow: 'Allow',
Deny: 'Deny',
Auto: 'Auto accept all connections for next 5 minutes',
Caption: 'User "' + username + '" is requesting remote desktop control of this computer. Click allow to grant access'
};
}
if (!options.font) { options.font = 'Arial'; }
if (!options.background) { options.background = RGB(0, 54, 105); }
var ret = new promise(promise.defaultInit);
ret.opt =
{
window:
{
winstyles: MessagePump.WindowStyles.WS_VISIBLE | MessagePump.WindowStyles.WS_BORDER | MessagePump.WindowStyles.WS_CAPTION | MessagePump.WindowStyles.WS_SYSMENU,
x: 300, y: 300, left: 0, right: 300, width: 580, height: 305, title: title, background: options.background
},
};
var rect = GM.CreateVariable(16);
var startupinput = require('_GenericMarshal').CreateVariable(24);
ret.gdipToken = require('_GenericMarshal').CreateVariable(4);
ret.pump = new MessagePump(ret.opt);
if (ret.pump._user32.SystemParametersInfoA(SPI_GETWORKAREA, 0, rect, 0).Val != 0)
{
var r = { x: rect.toBuffer().readInt32LE(0), y: rect.toBuffer().readInt32LE(4), w: rect.toBuffer().readInt32LE(8), h: rect.toBuffer().readInt32LE(12) };
r.w = r.w - r.x;
r.h = r.h - r.y;
console.info1('Primary Display: ' + JSON.stringify(r));
console.info1(' => x: ' + CENTER(580, r.x, r.w) + ', y: ' + CENTER(305, r.y, r.h));
ret.opt.window.x = CENTER(580, r.x, r.w);
ret.opt.window.y = CENTER(305, r.y, r.h);
}
ret.pump.autoAccept = false;
ret.pump.promise = ret;
ret.pump.brush = ret.pump._gdi32.CreateSolidBrush(options.background);
ret.pump.translations = options.translations;
ret.pump.username = username;
ret.pump.options = options;
ret.pump.font = ret.pump._gdi32.CreateFontW(20, 0, 0, 0, FW_DONTCARE, 0, 0, 0, DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS, DEFAULT_QUALITY, DEFAULT_PITCH | FF_SWISS, GM.CreateVariable(options.font, { wide: true }));
if (options.b64Image)
{
startupinput.toBuffer().writeUInt32LE(1);
gdip.GdiplusStartup(ret.gdipToken, startupinput, 0);
var raw = Buffer.from(options.b64Image, 'base64');
var nbuff = require('_GenericMarshal').CreateVariable(raw.length);
raw.copy(nbuff.toBuffer());
var istream = SHM.SHCreateMemStream(nbuff, raw.length);
var pimage = require('_GenericMarshal').CreatePointer();
var hbitmap = require('_GenericMarshal').CreatePointer();
var status = gdip.GdipCreateBitmapFromStream(istream, pimage);
status = gdip.GdipCreateHBITMAPFromBitmap(pimage.Deref(), hbitmap, options.background);
if (status.Val == 0) { options.bitmap = hbitmap; }
}
ret.pump.on('message', pump_onMessage);
ret.pump.on('hwnd', pump_onHwnd);
ret.pump.on('exit', pump_onExit);
if (options.timeout != null)
{
ret.pump.timeout = setTimeout(pump_onTimeout, options.timeout, ret.pump);
}
return (ret);
}
function create(title, caption, username, options)
{
if (options.uid == null) { return (createLocal(title, caption, username, options)); }
var self = require('user-sessions').getProcessOwnerName(process.pid).tsid;
if (self != 0)
{
if(options.uid == self)
{
// No need to dispatch, we can do this locally
return (createLocal(title, caption, username, options));
}
else
{
// Need to dispatch, but we don't have enough permissions to do that
var ret = new promise(promise.defaultInit);
ret.reject('Insufficient permission to dispatch to different user session');
return (ret);
}
}
if (options.uid == 0)
{
// TSID 0 doesn't have access to draw on the desktop
var ret = new promise(promise.defaultInit);
ret.reject('Cannot create dialog on this session');
return (ret);
}
// Need to dispatch to user session to display dialog
}
module.exports =
{
create: create
};