27 lines
689 B
JavaScript
27 lines
689 B
JavaScript
|
const WebSocket = require('ws');
|
||
|
|
const fs = require('fs');
|
||
|
|
|
||
|
|
const token = process.argv[2];
|
||
|
|
const agentId = process.argv[3];
|
||
|
|
const command = process.argv[4].startsWith('@') ? fs.readFileSync(process.argv[4].slice(1), 'utf8') : process.argv[4];
|
||
|
|
|
||
|
|
const ws = new WebSocket(`ws://localhost:5000/ws?type=shell&agentId=${agentId}`);
|
||
|
|
|
||
|
|
let buffer = '';
|
||
|
|
ws.on('open', () => {
|
||
|
|
ws.send(JSON.stringify({ type: 'auth', token }));
|
||
|
|
setTimeout(() => {
|
||
|
|
ws.send(command + '\r\n');
|
||
|
|
}, 1500);
|
||
|
|
});
|
||
|
|
|
||
|
|
ws.on('message', (data) => {
|
||
|
|
buffer += data.toString();
|
||
|
|
});
|
||
|
|
|
||
|
|
const waitMs = parseInt(process.argv[5]) || 8000;
|
||
|
|
setTimeout(() => {
|
||
|
|
console.log(buffer);
|
||
|
|
process.exit(0);
|
||
|
|
}, waitMs);
|