Mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/wukko/cobalt.git synced 2025-01-12 20:25:06 +01:00
Code Issues Projects Releases Packages Wiki Activity

api: clean up and remove unnecessary res.destroy()s

Browse source
This commit is contained in:
dumbmoron 2024-05-22 16:45:32 +00:00 committed by wukko
parent aba1d80615
commit d72be2c95a
1 changed files with 35 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

74
src/core/api.js
View file

@ -163,52 +163,48 @@ export function runAPI(express, app, gitCommit, gitBranch, __dirname) {
const checkBaseLength = id.length === 21 && exp.length === 13; const checkBaseLength = id.length === 21 && exp.length === 13;
const checkSafeLength = sig.length === 43 && sec.length === 43 && iv.length === 22; const checkSafeLength = sig.length === 43 && sec.length === 43 && iv.length === 22;
if (checkQueries && checkBaseLength && checkSafeLength) { if (!checkQueries || !checkBaseLength || !checkSafeLength) {
// rate limit probe, will not return json after 8.0 return res.sendStatus(400);
if (req.query.p) {
return res.status(200).json({
status: "continue"
})
}
try {
const streamInfo = verifyStream(id, sig, exp, sec, iv);
if (!streamInfo?.service) {
return res.sendStatus(streamInfo.status);
}
return stream(res, streamInfo);
} catch {
return res.destroy();
}
} }
return res.sendStatus(400);
// rate limit probe, will not return json after 8.0
if (req.query.p) {
return res.status(200).json({
status: "continue"
})
}
const streamInfo = verifyStream(id, sig, exp, sec, iv);
if (!streamInfo?.service) {
return res.sendStatus(streamInfo.status);
}
return stream(res, streamInfo);
}) })
app.get('/api/istream', (req, res) => { app.get('/api/istream', (req, res) => {
try { if (!req.ip.endsWith('127.0.0.1')) {
if (!req.ip.endsWith('127.0.0.1')) return res.sendStatus(403);
return res.sendStatus(403);
if (String(req.query.id).length !== 21)
return res.sendStatus(400);
const streamInfo = getInternalStream(req.query.id);
if (!streamInfo) return res.sendStatus(404);
streamInfo.headers = {
...req.headers,
...streamInfo.headers
};
return stream(res, { type: 'internal', ...streamInfo });
} catch {
return res.destroy();
} }
if (String(req.query.id).length !== 21) {
return res.sendStatus(400);
}
const streamInfo = getInternalStream(req.query.id);
if (!streamInfo) {
return res.sendStatus(404);
}
streamInfo.headers = {
...req.headers,
...streamInfo.headers
};
return stream(res, { type: 'internal', ...streamInfo });
}) })
app.get('/api/serverInfo', (req, res) => { app.get('/api/serverInfo', (_, res) => {
try { return res.status(200).json(serverInfo);
return res.status(200).json(serverInfo);
} catch {
return res.destroy();
}
}) })
app.get('/favicon.ico', (req, res) => { app.get('/favicon.ico', (req, res) => {