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

First Rev of Digest AUTH Unit Test

Browse Source 
Fixed crash in ILibWebClient caused by NULL dereference
This commit is contained in:
Bryan Roe
2022-07-02 00:03:41 -07:00
parent 62195b6081
commit 143f7ef578
2 changed files with 182 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
microstack/ILibWebClient.c
View File

@@ -1865,10 +1865,17 @@ ILibWebClient_DataResults ILibWebClient_OnData(ILibAsyncSocket_SocketModule sock
// so we need to copy it out, so we can recycle the buffer
// for the body
//
if (wcdo->header != NULL)
{
tph = ILibClonePacket(wcdo->header);
ILibDestructPacket(wcdo->header);
wcdo->header = tph;
}
//else
//{
// printf("WCDO/HEADER was NULL\n");
//}
}
}
}
//{{{ REMOVE_THIS_FOR_HTTP/1.0_ONLY_SUPPORT--> }}}

177
test/self-test.js
View File

@@ -81,6 +81,9 @@ var agentConnectionCount = 0;
var updateState = 0;
const consoleMode = process.argv.getParameter('console') != null;
var digest_realm;
var digest_username;
var digest_password;
// Check Permissions... Need Root/Elevated Permissions
@@ -135,7 +138,33 @@ var promises =
services: null,
setclip: null,
getclip: null,
digest: null,
digest_auth: null,
digest_authint: null,
};
function generateRandomNumber(lower, upper)
{
return (Math.floor(Math.random() * (upper - lower)) + lower);
}
function generateRandomLetter()
{
return (String.fromCharCode(generateRandomNumber(97, 122)));
}
function generateRandomString(len)
{
var ret = '', i;
for (i = 0; i < len; ++i)
{
ret += generateRandomLetter();
}
return (ret);
}
function generateRandomRealm()
{
realm = generateRandomString(generateRandomNumber(1, 5)) + '.' + generateRandomString(generateRandomNumber(8, 20)) + '.com';
return (realm);
}
function resetPromises()
{
var i;
@@ -167,7 +196,7 @@ server.on('connection', function (c)
global._test.push(c);
console.info1('inbound connection received');
});
server.on('request', function (imsg, resp)
server.on('request', function (imsg, rsp)
{
if (imsg.method == 'GET' && imsg.url == '/update')
{
@@ -195,9 +224,78 @@ server.on('request', function (imsg, resp)
var update = require('fs').createReadStream(getCurrentUpdatePath(), { flags: 'rb' });
accumulator.total = require('fs').statSync(getCurrentUpdatePath()).size;
update.pipe(resp);
update.pipe(rsp);
update.pipe(accumulator);
}
if (imsg.method == 'POST')
{
var username, qop;
if (imsg.Digest_IsAuthenticated(digest_realm))
{
username = imsg.Digest_GetUsername();
qop = imsg.Digest_GetQOP();
switch (imsg.url)
{
case '/':
if (imsg.Digest_ValidatePassword(digest_password))
{
rsp.statusCode = 200;
rsp.setHeader('Content-Type', 'text/html');
rsp.end('<html>Success!</html>');
}
else
{
rsp.Digest_writeUnauthorized(digest_realm);
}
break;
case '/auth':
if (qop != 'auth') { promises.digest_auth.reject('Received Incorrect QOP: ' + qop); }
if (imsg.Digest_ValidatePassword(digest_password))
{
rsp.statusCode = 200;
rsp.setHeader('Content-Type', 'text/html');
rsp.end('<html>Success!</html>');
}
else
{
rsp.Digest_writeUnauthorized(digest_realm, { qop: 'auth' });
}
break;
case '/auth-int':
if (qop != 'auth-int') { promises.digest_authint.reject('Received Incorrect QOP: ' + qop); }
imsg.on('end', function ()
{
if (imsg.Digest_ValidatePassword(digest_password))
{
rsp.statusCode = 200;
rsp.setHeader('Content-Type', 'text/html');
rsp.end('<html>Success!</html>');
}
else
{
rsp.Digest_writeUnauthorized(digest_realm);
}
});
break;
}
}
else
{
switch (imsg.url)
{
case '/':
rsp.Digest_writeUnauthorized(digest_realm);
break;
case '/auth':
rsp.Digest_writeUnauthorized(digest_realm, { qop: 'auth' });
break;
case '/auth-int':
rsp.Digest_writeUnauthorized(digest_realm, { qop: 'auth-int, auth' });
break;
}
}
}
});
server.on('upgrade', function (msg, sck, head)
{
@@ -475,13 +573,22 @@ server.on('upgrade', function (msg, sck, head)
promises.smbios.resolve(j);
break;
case 'result':
console.info1(JSON.stringify(j, null, 1));
if (promises[j.id] != null)
{
if (promises[j.id].timeout != null)
{
clearTimeout(promises[j.id].timeout);
}
promises[j.id].resolve(j.value);
if (j.result===true)
{
promises[j.id].resolve(j);
}
else
{
promises[j.id].reject(j.reason == null ? '' : j.reason);
}
}
break;
default:
@@ -651,6 +758,63 @@ server.on('upgrade', function (msg, sck, head)
process.stdout.write(' => Received: ' + v.data + '\n');
}
}).then(function ()
{
digest_realm = generateRandomRealm();
digest_username = generateRandomString(generateRandomNumber(5, 10));
digest_password = generateRandomString(generateRandomNumber(8, 20));
process.stdout.write(' HTTP Digest Test\n');
process.stdout.write(' => Basic.............................................[WAITING]');
sendEval("var digest = require('http-digest').create('" + digest_username + "', '" + digest_password + "');");
sendEval("digest.http = require('http');");
sendEval("var options = { protocol: 'https:', host: '127.0.0.1', port: " + server.address().port + ", path: '/', method: 'POST', rejectUnauthorized: false };");
sendEval("var req = digest.request(options);");
sendEval("req.on('error', function (e) { selfTestResponse('digest', false, JSON.stringify(e)); req = null; });");
sendEval("req.on('response', function (imsg) { selfTestResponse('digest', true); });");
sendEval("req.end('TestData');");
return (promises.digest);
}).then(function (v)
{
process.stdout.write('\r => Basic.............................................[OK] \n');
process.stdout.write(' => QOP = auth........................................[WAITING]');
digest_realm = generateRandomRealm();
digest_username = generateRandomString(generateRandomNumber(5, 10));
digest_password = generateRandomString(generateRandomNumber(8, 20));
sendEval("digest = require('http-digest').create('" + digest_username + "', '" + digest_password + "');");
sendEval("digest.http = require('http');");
sendEval("var options = { protocol: 'https:', host: '127.0.0.1', port: " + server.address().port + ", path: '/auth', method: 'POST', rejectUnauthorized: false };");
sendEval("var req = digest.request(options);");
sendEval("req.on('error', function (e) { selfTestResponse('digest_auth', false, JSON.stringify(e)); req = null; });");
sendEval("req.on('response', function (imsg) { selfTestResponse('digest_auth', true); });");
sendEval("req.end('TestData');");
return (promises.digest_auth);
}).then(function ()
{
process.stdout.write('\r => QOP = auth........................................[OK] \n');
process.stdout.write(' => QOP = auth-int....................................[WAITING]');
digest_realm = generateRandomRealm();
digest_username = generateRandomString(generateRandomNumber(5, 10));
digest_password = generateRandomString(generateRandomNumber(8, 20));
sendEval("digest = require('http-digest').create('" + digest_username + "', '" + digest_password + "');");
sendEval("digest.http = require('http');");
sendEval("var options = { protocol: 'https:', host: '127.0.0.1', port: " + server.address().port + ", path: '/auth-int', method: 'POST', rejectUnauthorized: false };");
sendEval("var req = digest.request(options);");
sendEval("req.on('error', function (e) { selfTestResponse('digest_authint', false, JSON.stringify(e)); req = null; });");
sendEval("req.on('response', function (imsg) { selfTestResponse('digest_authint', true); });");
sendEval("req.end('TestData');");
return (promises.digest_authint);
}).then(function ()
{
process.stdout.write('\r => QOP = auth-int....................................[OK] \n');
}).then(function ()
{
process.stdout.write('\nTesting Complete\n\n');
endTest();
@@ -813,7 +977,10 @@ function endTest()
require('agent-installer').fullUninstall(paramsString);
console.setDestination(console.Destinations.STDOUT);
}
function sendEval(cmd)
{
global._client.command({ sessionid: 'user//foo//bar', rights: 4294967295, consent: 64, action: 'msg', type: 'console', value: 'eval "' + cmd + '"' });
}
if (process.argv.getParameter('AgentsFolder') != null)
{
@@ -828,7 +995,7 @@ if (process.argv.getParameter('AgentsFolder') != null)
var i, tmp, m;
var lines = ['var addedModules = [];'];
lines.push("function selfTestResponse(id, value) { require('MeshAgent').SendCommand({ action: 'result', id: id, value: JSON.stringify(value) }); }");
lines.push("function selfTestResponse(id, result, reason) { require('MeshAgent').SendCommand({ action: 'result', id: id, result: result, reason: reason }); }");
for (i = 0; i < modules_folder.length; ++i)
{
tmp = require('fs').readFileSync(modules + (process.platform == 'win32' ? '\\' : '/') + modules_folder[i]);