forks/amnezia-wg-easy
1
0
Fork 0
forked from mirrors/amnezia-wg-easy
Code Pull requests Activity

feat: support graceful shutdown

Browse source
This commit is contained in:
Chanchai 2024-02-21 13:56:39 +07:00 • committed by Philip H
parent f603d6b5da
commit 6604e44ea3
3 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/config.js
View file

@ -27,6 +27,11 @@ iptables -A FORWARD -o wg0 -j ACCEPT;
`.split('\n').join(' ');
module.exports.WG_PRE_DOWN = process.env.WG_PRE_DOWN || '';
module.exports.WG_POST_DOWN = process.env.WG_POST_DOWN || '';
module.exports.WG_POST_DOWN = process.env.WG_POST_DOWN || `
iptables -t nat -D POSTROUTING -s ${module.exports.WG_DEFAULT_ADDRESS.replace('x', '0')}/24 -o ${module.exports.WG_DEVICE} -j MASQUERADE;
iptables -D INPUT -p udp -m udp --dport 51820 -j ACCEPT;
iptables -D FORWARD -i wg0 -j ACCEPT;
iptables -D FORWARD -o wg0 -j ACCEPT;
`.split('\n').join(' ');
module.exports.LANG = process.env.LANG || 'en';
module.exports.UI_TRAFFIC_STATS = process.env.UI_TRAFFIC_STATS || 'false';

4
src/lib/WireGuard.js
View file

@ -318,4 +318,8 @@ Endpoint = ${WG_HOST}:${WG_PORT}`;
await this.saveConfig();
}
// Shutdown wireguard
async Shutdown() {
await Util.exec('wg-quick down wg0').catch(() => { });
}
};

12
src/server.js
View file

@ -12,3 +12,15 @@ WireGuard.getConfig()
// eslint-disable-next-line no-process-exit
process.exit(1);
});
// Handle terminate signal
process.on('SIGTERM', async() => {
console.log('SIGTERM signal received.');
await WireGuard.Shutdown();
process.exit(0);
});
// Handle interupt signal
process.on('SIGINT', () => {
console.log('SIGINT signal received.');
});