79 lines
2.1 KiB
JavaScript
79 lines
2.1 KiB
JavaScript
|
const puppeteer = require('puppeteer');
|
||
|
|
const fs = require('fs');
|
||
|
|
let io;
|
||
|
|
let isBotRunning = false;
|
||
|
|
let browser; // Keep the browser instance
|
||
|
|
const flag = fs.readFileSync('/flag.txt', 'utf8');
|
||
|
|
|
||
|
|
// Initialize the browser once when the application starts
|
||
|
|
browser = puppeteer.launch({
|
||
|
|
headless: true,
|
||
|
|
executablePath: process.env.PUPPETEER_EXECUTABLE_PATH || '/usr/bin/chromium-browser',
|
||
|
|
args: [
|
||
|
|
'--no-sandbox',
|
||
|
|
'--disable-setuid-sandbox',
|
||
|
|
'--disable-gpu',
|
||
|
|
'--disable-dev-shm-usage',
|
||
|
|
'--disable-extensions',
|
||
|
|
'--disable-background-timer-throttling',
|
||
|
|
'--disable-backgrounding-occluded-windows',
|
||
|
|
'--disable-renderer-backgrounding'
|
||
|
|
],
|
||
|
|
});
|
||
|
|
|
||
|
|
const visit = async (query) => {
|
||
|
|
if (isBotRunning) {
|
||
|
|
console.log('Bot is already running. Skipping new request.');
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
isBotRunning = true;
|
||
|
|
|
||
|
|
let page;
|
||
|
|
|
||
|
|
try {
|
||
|
|
// Open a new page for each request
|
||
|
|
page = await (await browser).newPage();
|
||
|
|
|
||
|
|
console.log('New page opened successfully!');
|
||
|
|
|
||
|
|
// Navigate to the URL
|
||
|
|
await page.goto(`http://127.0.0.1:1337?q=${query}`, { waitUntil: 'domcontentloaded' });
|
||
|
|
|
||
|
|
let alertHandled = false;
|
||
|
|
|
||
|
|
// Check for alert dialogs using Puppeteer's page.on('dialog') event listener
|
||
|
|
page.on('dialog', async (dialog) => {
|
||
|
|
await dialog.accept();
|
||
|
|
alertHandled = true;
|
||
|
|
if (io) {
|
||
|
|
io.emit('flag', { message: 'Alert detected!', flag: flag.trim() });
|
||
|
|
}
|
||
|
|
await page.close(); // Close the page after handling the alert
|
||
|
|
isBotRunning = false;
|
||
|
|
});
|
||
|
|
|
||
|
|
// Use setTimeout to wait for the alert to appear
|
||
|
|
setTimeout(async () => {
|
||
|
|
if (!alertHandled) {
|
||
|
|
await page.close(); // Close the page if no alert was detected
|
||
|
|
isBotRunning = false;
|
||
|
|
}
|
||
|
|
}, 500); // 500ms wait before closing the page if no alert is handled
|
||
|
|
|
||
|
|
} catch (e) {
|
||
|
|
console.error(`Failed to navigate: ${e.message}`);
|
||
|
|
if (page) {
|
||
|
|
await page.close(); // Ensure the page is closed on error
|
||
|
|
}
|
||
|
|
isBotRunning = false;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
// Setter function to initialize io
|
||
|
|
const setIo = (socketIoInstance) => {
|
||
|
|
io = socketIoInstance;
|
||
|
|
};
|
||
|
|
|
||
|
|
module.exports = { visit, setIo };
|