658 lines
16 KiB
JavaScript
Executable File
658 lines
16 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
|
|
|
/* jshint esversion: 6 */
|
|
/* jshint node: true */
|
|
"use strict";
|
|
|
|
const express = require("express");
|
|
const { engine: hbs } = require("express-handlebars");
|
|
const bodyparser = require("body-parser");
|
|
const session = require("express-session");
|
|
const busboy = require("connect-busboy");
|
|
const flash = require("connect-flash");
|
|
const querystring = require("querystring");
|
|
|
|
const archiver = require("archiver");
|
|
|
|
const notp = require("notp");
|
|
const base32 = require("thirty-two");
|
|
|
|
const fs = require("fs");
|
|
const rimraf = require("rimraf");
|
|
const path = require("path");
|
|
|
|
const filesize = require("filesize");
|
|
const octicons = require("@primer/octicons");
|
|
const handlebars = require("handlebars");
|
|
|
|
const port = +process.env.PORT || 8080;
|
|
|
|
const app = express();
|
|
const http = app.listen(port);
|
|
|
|
app.set("views", path.join(__dirname, "views"));
|
|
app.engine(
|
|
"handlebars",
|
|
hbs({
|
|
partialsDir: path.join(__dirname, "views", "partials"),
|
|
layoutsDir: path.join(__dirname, "views", "layouts"),
|
|
defaultLayout: "main",
|
|
helpers: {
|
|
either: function (a, b, options) {
|
|
if (a || b) {
|
|
return options.fn(this);
|
|
}
|
|
},
|
|
filesize: filesize,
|
|
octicon: function (i, options) {
|
|
if (!octicons[i]) {
|
|
return new handlebars.SafeString(octicons.question.toSVG());
|
|
}
|
|
return new handlebars.SafeString(octicons[i].toSVG());
|
|
},
|
|
eachpath: function (path, options) {
|
|
if (typeof path != "string") {
|
|
return "";
|
|
}
|
|
let out = "";
|
|
path = path.split("/");
|
|
path.splice(path.length - 1, 1);
|
|
path.unshift("");
|
|
path.forEach((folder, index) => {
|
|
out += options.fn({
|
|
name: folder + "/",
|
|
path: "/" + path.slice(1, index + 1).join("/"),
|
|
current: index === path.length - 1,
|
|
});
|
|
});
|
|
return out;
|
|
},
|
|
},
|
|
})
|
|
);
|
|
app.set("view engine", "handlebars");
|
|
|
|
app.use("/@assets", express.static(path.join(__dirname, "assets")));
|
|
app.use(
|
|
"/@assets/bootstrap",
|
|
express.static(path.join(__dirname, "node_modules/bootstrap/dist"))
|
|
);
|
|
app.use(
|
|
"/@assets/octicons",
|
|
express.static(path.join(__dirname, "node_modules/@primer/octicons/build"))
|
|
);
|
|
app.use(
|
|
"/@assets/jquery",
|
|
express.static(path.join(__dirname, "node_modules/jquery/dist"))
|
|
);
|
|
app.use(
|
|
"/@assets/filesize",
|
|
express.static(path.join(__dirname, "node_modules/filesize/lib"))
|
|
);
|
|
app.use(
|
|
"/@assets/xterm",
|
|
express.static(path.join(__dirname, "node_modules/xterm"))
|
|
);
|
|
app.use(
|
|
"/@assets/xterm-addon-attach",
|
|
express.static(path.join(__dirname, "node_modules/xterm-addon-attach"))
|
|
);
|
|
app.use(
|
|
"/@assets/xterm-addon-fit",
|
|
express.static(path.join(__dirname, "node_modules/xterm-addon-fit"))
|
|
);
|
|
|
|
app.use(
|
|
session({
|
|
secret: process.env.SESSION_KEY || "meowmeow",
|
|
})
|
|
);
|
|
app.use(flash());
|
|
app.use(busboy());
|
|
app.use(bodyparser.urlencoded());
|
|
|
|
// AUTH
|
|
|
|
const KEY = process.env.KEY
|
|
? base32.decode(process.env.KEY.replace(/ /g, ""))
|
|
: null;
|
|
|
|
app.get("/@logout", (req, res) => {
|
|
if (KEY) {
|
|
req.session.login = false;
|
|
req.flash("success", "Signed out.");
|
|
res.redirect("/@login");
|
|
return;
|
|
}
|
|
req.flash("error", "You were never logged in...");
|
|
res.redirect("back");
|
|
});
|
|
|
|
app.get("/@login", (req, res) => {
|
|
res.render("login", flashify(req, {}));
|
|
});
|
|
app.post("/@login", (req, res) => {
|
|
let pass = notp.totp.verify(req.body.token.replace(" ", ""), KEY);
|
|
if (pass) {
|
|
req.session.login = true;
|
|
res.redirect("/");
|
|
return;
|
|
}
|
|
req.flash("error", "Bad token.");
|
|
res.redirect("/@login");
|
|
});
|
|
|
|
app.use((req, res, next) => {
|
|
if (!KEY) {
|
|
return next();
|
|
}
|
|
if (req.session.login === true) {
|
|
return next();
|
|
}
|
|
req.flash("error", "Please sign in.");
|
|
res.redirect("/@login");
|
|
});
|
|
|
|
function relative(...paths) {
|
|
return paths.reduce((a, b) => path.join(a, b), process.cwd());
|
|
}
|
|
function flashify(req, obj) {
|
|
let error = req.flash("error");
|
|
if (error && error.length > 0) {
|
|
if (!obj.errors) {
|
|
obj.errors = [];
|
|
}
|
|
obj.errors.push(error);
|
|
}
|
|
let success = req.flash("success");
|
|
if (success && success.length > 0) {
|
|
if (!obj.successes) {
|
|
obj.successes = [];
|
|
}
|
|
obj.successes.push(success);
|
|
}
|
|
obj.isloginenabled = !!KEY;
|
|
return obj;
|
|
}
|
|
|
|
app.use((req, res, next) => {
|
|
if (req.method === "GET") {
|
|
return next();
|
|
}
|
|
let sourceHost = null;
|
|
if (req.headers.origin) {
|
|
sourceHost = new URL(req.headers.origin).host;
|
|
} else if (req.headers.referer) {
|
|
sourceHost = new URL(req.headers.referer).host;
|
|
}
|
|
if (sourceHost !== req.headers.host) {
|
|
throw new Error(
|
|
"Origin or Referer header does not match or is missing. Request has been blocked to prevent CSRF"
|
|
);
|
|
}
|
|
next();
|
|
});
|
|
|
|
app.all("/*", (req, res, next) => {
|
|
res.filename = req.params[0];
|
|
|
|
let fileExists = new Promise((resolve, reject) => {
|
|
// check if file exists
|
|
fs.stat(relative(res.filename), (err, stats) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
return resolve(stats);
|
|
});
|
|
});
|
|
|
|
fileExists
|
|
.then((stats) => {
|
|
res.stats = stats;
|
|
next();
|
|
})
|
|
.catch((err) => {
|
|
res.stats = { error: err };
|
|
next();
|
|
});
|
|
});
|
|
|
|
app.post("/*@upload", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
let buff = null;
|
|
let saveas = null;
|
|
req.busboy.on("file", (key, stream, filename) => {
|
|
if (key == "file") {
|
|
let buffs = [];
|
|
stream.on("data", (d) => {
|
|
buffs.push(d);
|
|
});
|
|
stream.on("end", () => {
|
|
buff = Buffer.concat(buffs);
|
|
buffs = null;
|
|
});
|
|
}
|
|
});
|
|
req.busboy.on("field", (key, value) => {
|
|
if (key == "saveas") {
|
|
saveas = value;
|
|
}
|
|
});
|
|
req.busboy.on("finish", () => {
|
|
if (!buff || !saveas) {
|
|
return res.status(400).end();
|
|
}
|
|
let fileExists = new Promise((resolve, reject) => {
|
|
// check if file exists
|
|
fs.stat(relative(res.filename, saveas), (err, stats) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
return resolve(stats);
|
|
});
|
|
});
|
|
|
|
fileExists
|
|
.then((stats) => {
|
|
console.warn("file exists, cannot overwrite");
|
|
req.flash("error", "File exists, cannot overwrite. ");
|
|
res.redirect("back");
|
|
})
|
|
.catch((err) => {
|
|
const saveName = relative(res.filename, saveas);
|
|
console.log("saving file to " + saveName);
|
|
let save = fs.createWriteStream(saveName);
|
|
save.on("close", () => {
|
|
if (res.headersSent) {
|
|
return;
|
|
}
|
|
if (buff.length === 0) {
|
|
req.flash("success", "File saved. Warning: empty file.");
|
|
} else {
|
|
buff = null;
|
|
req.flash("success", "File saved. ");
|
|
}
|
|
res.redirect("back");
|
|
});
|
|
save.on("error", (err) => {
|
|
console.warn(err);
|
|
req.flash("error", err.toString());
|
|
res.redirect("back");
|
|
});
|
|
save.write(buff);
|
|
save.end();
|
|
});
|
|
});
|
|
req.pipe(req.busboy);
|
|
});
|
|
|
|
app.post("/*@mkdir", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
let folder = req.body.folder;
|
|
if (!folder || folder.length < 1) {
|
|
return res.status(400).end();
|
|
}
|
|
|
|
let fileExists = new Promise((resolve, reject) => {
|
|
// Check if file exists
|
|
fs.stat(relative(res.filename, folder), (err, stats) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
return resolve(stats);
|
|
});
|
|
});
|
|
|
|
fileExists
|
|
.then((stats) => {
|
|
req.flash("error", "Folder exists, cannot overwrite. ");
|
|
res.redirect("back");
|
|
})
|
|
.catch((err) => {
|
|
fs.mkdir(relative(res.filename, folder), (err) => {
|
|
if (err) {
|
|
console.warn(err);
|
|
req.flash("error", err.toString());
|
|
res.redirect("back");
|
|
return;
|
|
}
|
|
req.flash("success", "Folder created. ");
|
|
res.redirect("back");
|
|
});
|
|
});
|
|
});
|
|
|
|
app.post("/*@delete", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
let files = JSON.parse(req.body.files);
|
|
if (!files || !files.map) {
|
|
req.flash("error", "No files selected.");
|
|
res.redirect("back");
|
|
return; // res.status(400).end();
|
|
}
|
|
|
|
let promises = files.map((f) => {
|
|
return new Promise((resolve, reject) => {
|
|
fs.stat(relative(res.filename, f), (err, stats) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
resolve({
|
|
name: f,
|
|
isdirectory: stats.isDirectory(),
|
|
isfile: stats.isFile(),
|
|
});
|
|
});
|
|
});
|
|
});
|
|
Promise.all(promises)
|
|
.then((files) => {
|
|
let promises = files.map((f) => {
|
|
return new Promise((resolve, reject) => {
|
|
let op = null;
|
|
if (f.isdirectory) {
|
|
op = (dir, cb) =>
|
|
rimraf(
|
|
dir,
|
|
{
|
|
glob: false,
|
|
},
|
|
cb
|
|
);
|
|
} else if (f.isfile) {
|
|
op = fs.unlink;
|
|
}
|
|
if (op) {
|
|
op(relative(res.filename, f.name), (err) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
resolve();
|
|
});
|
|
}
|
|
});
|
|
});
|
|
Promise.all(promises)
|
|
.then(() => {
|
|
req.flash("success", "Files deleted. ");
|
|
res.redirect("back");
|
|
})
|
|
.catch((err) => {
|
|
console.warn(err);
|
|
req.flash("error", "Unable to delete some files: " + err);
|
|
res.redirect("back");
|
|
});
|
|
})
|
|
.catch((err) => {
|
|
console.warn(err);
|
|
req.flash("error", err.toString());
|
|
res.redirect("back");
|
|
});
|
|
});
|
|
|
|
app.get("/*@download", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
let files = null;
|
|
try {
|
|
files = JSON.parse(req.query.files);
|
|
} catch (e) {}
|
|
if (!files || !files.map) {
|
|
req.flash("error", "No files selected.");
|
|
res.redirect("back");
|
|
return; // res.status(400).end();
|
|
}
|
|
|
|
let promises = files.map((f) => {
|
|
return new Promise((resolve, reject) => {
|
|
fs.stat(relative(res.filename, f), (err, stats) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
resolve({
|
|
name: f,
|
|
isdirectory: stats.isDirectory(),
|
|
isfile: stats.isFile(),
|
|
});
|
|
});
|
|
});
|
|
});
|
|
Promise.all(promises)
|
|
.then((files) => {
|
|
let zip = archiver("zip", {});
|
|
zip.on("error", function (err) {
|
|
console.warn(err);
|
|
res.status(500).send({
|
|
error: err.message,
|
|
});
|
|
});
|
|
|
|
files
|
|
.filter((f) => f.isfile)
|
|
.forEach((f) => {
|
|
zip.file(relative(res.filename, f.name), { name: f.name });
|
|
});
|
|
files
|
|
.filter((f) => f.isdirectory)
|
|
.forEach((f) => {
|
|
zip.directory(relative(res.filename, f.name), f.name);
|
|
});
|
|
|
|
res.attachment("Archive.zip");
|
|
zip.pipe(res);
|
|
|
|
zip.finalize();
|
|
})
|
|
.catch((err) => {
|
|
console.warn(err);
|
|
req.flash("error", err.toString());
|
|
res.redirect("back");
|
|
});
|
|
});
|
|
|
|
const shellable = process.env.SHELL != "false" && process.env.SHELL;
|
|
const cmdable = process.env.CMD != "false" && process.env.CMD;
|
|
if (shellable || cmdable) {
|
|
const shellArgs = process.env.SHELL.split(" ");
|
|
const exec = process.env.SHELL == "login" ? "/usr/bin/env" : shellArgs[0];
|
|
const args = process.env.SHELL == "login" ? ["login"] : shellArgs.slice(1);
|
|
|
|
const child_process = require("child_process");
|
|
|
|
app.post("/*@cmd", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
let cmd = req.body.cmd;
|
|
if (!cmd || cmd.length < 1) {
|
|
return res.status(400).end();
|
|
}
|
|
console.log("running command " + cmd);
|
|
|
|
child_process.exec(
|
|
cmd,
|
|
{
|
|
cwd: relative(res.filename),
|
|
timeout: 60 * 1000,
|
|
},
|
|
(err, stdout, stderr) => {
|
|
if (err) {
|
|
console.log("command run failed: " + JSON.stringify(err));
|
|
req.flash("error", "Command failed due to non-zero exit code");
|
|
}
|
|
res.render(
|
|
"cmd",
|
|
flashify(req, {
|
|
path: res.filename,
|
|
cmd: cmd,
|
|
stdout: stdout,
|
|
stderr: stderr,
|
|
})
|
|
);
|
|
}
|
|
);
|
|
});
|
|
|
|
const pty = require("node-pty");
|
|
const WebSocket = require("ws");
|
|
|
|
app.get("/*@shell", (req, res) => {
|
|
res.filename = req.params[0];
|
|
|
|
res.render(
|
|
"shell",
|
|
flashify(req, {
|
|
path: res.filename,
|
|
})
|
|
);
|
|
});
|
|
|
|
const ws = new WebSocket.Server({ server: http });
|
|
ws.on("connection", (socket, request) => {
|
|
const { path } = querystring.parse(request.url.split("?")[1]);
|
|
let cwd = relative(path);
|
|
let term = pty.spawn(exec, args, {
|
|
name: "xterm-256color",
|
|
cols: 80,
|
|
rows: 30,
|
|
cwd: cwd,
|
|
});
|
|
console.log(
|
|
"pid " + term.pid + " shell " + process.env.SHELL + " started in " + cwd
|
|
);
|
|
|
|
term.on("data", (data) => {
|
|
socket.send(data, { binary: true });
|
|
});
|
|
term.on("exit", (code) => {
|
|
console.log("pid " + term.pid + " ended");
|
|
socket.close();
|
|
});
|
|
socket.on("message", (data) => {
|
|
// special messages should decode to Buffers
|
|
if (data.length == 6) {
|
|
switch (data.readUInt16BE(0)) {
|
|
case 0:
|
|
term.resize(data.readUInt16BE(1), data.readUInt16BE(2));
|
|
return;
|
|
}
|
|
}
|
|
term.write(data);
|
|
});
|
|
socket.on("close", () => {
|
|
term.end();
|
|
});
|
|
});
|
|
}
|
|
|
|
const SMALL_IMAGE_MAX_SIZE = 750 * 1024; // 750 KB
|
|
const EXT_IMAGES = [".jpg", ".jpeg", ".png", ".webp", ".svg", ".gif", ".tiff"];
|
|
function isimage(f) {
|
|
for (const ext of EXT_IMAGES) {
|
|
if (f.endsWith(ext)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
app.get("/*", (req, res) => {
|
|
if (res.stats.error) {
|
|
res.render(
|
|
"list",
|
|
flashify(req, {
|
|
shellable: shellable,
|
|
cmdable: cmdable,
|
|
path: res.filename,
|
|
errors: [res.stats.error],
|
|
})
|
|
);
|
|
} else if (res.stats.isDirectory()) {
|
|
if (!req.url.endsWith("/")) {
|
|
return res.redirect(req.url + "/");
|
|
}
|
|
|
|
let readDir = new Promise((resolve, reject) => {
|
|
fs.readdir(relative(res.filename), (err, filenames) => {
|
|
if (err) {
|
|
return reject(err);
|
|
}
|
|
return resolve(filenames);
|
|
});
|
|
});
|
|
|
|
readDir
|
|
.then((filenames) => {
|
|
const promises = filenames.map(
|
|
(f) =>
|
|
new Promise((resolve, reject) => {
|
|
fs.stat(relative(res.filename, f), (err, stats) => {
|
|
if (err) {
|
|
console.warn(err);
|
|
return resolve({
|
|
name: f,
|
|
error: err,
|
|
});
|
|
}
|
|
resolve({
|
|
name: f,
|
|
isdirectory: stats.isDirectory(),
|
|
issmallimage: isimage(f) && stats.size < SMALL_IMAGE_MAX_SIZE,
|
|
size: stats.size,
|
|
});
|
|
});
|
|
})
|
|
);
|
|
|
|
Promise.all(promises)
|
|
.then((files) => {
|
|
res.render(
|
|
"list",
|
|
flashify(req, {
|
|
shellable: shellable,
|
|
cmdable: cmdable,
|
|
path: res.filename,
|
|
files: files,
|
|
})
|
|
);
|
|
})
|
|
.catch((err) => {
|
|
console.error(err);
|
|
res.render(
|
|
"list",
|
|
flashify(req, {
|
|
shellable: shellable,
|
|
cmdable: cmdable,
|
|
path: res.filename,
|
|
errors: [err],
|
|
})
|
|
);
|
|
});
|
|
})
|
|
.catch((err) => {
|
|
console.warn(err);
|
|
res.render(
|
|
"list",
|
|
flashify(req, {
|
|
shellable: shellable,
|
|
cmdable: cmdable,
|
|
path: res.filename,
|
|
errors: [err],
|
|
})
|
|
);
|
|
});
|
|
} else if (res.stats.isFile()) {
|
|
res.sendFile(relative(res.filename), {
|
|
headers: {
|
|
"Content-Security-Policy":
|
|
"default-src 'self'; script-src 'none'; sandbox",
|
|
},
|
|
dotfiles: "allow",
|
|
});
|
|
}
|
|
});
|
|
|
|
console.log(`Listening on port ${port}`);
|