piston/lxc/execute

76 lines
1.1 KiB
Bash
Executable File

#!/usr/bin/env bash
dir="$( cd "$( dirname "$0" )" && pwd )"
if [ -z "$1" ]; then
echo "invalid args"
exit
fi
if [ -z "$2" ]; then
echo "invalid args"
exit
fi
lang=$1
epoch=$(date +%s%3N)
basepath="/var/lib/lxc/piston/rootfs"
filepath="/tmp/$epoch/code.code"
file=$(basename $2)
argpath="/tmp/$epoch/args.args"
arg=$(basename $argpath)
# write arg file
mkdir -p $basepath/tmp/$epoch
chmod 777 $basepath/tmp/$epoch
cat $file > $basepath$filepath
echo "${@:3}" > $basepath$argpath
bin=
case "$lang" in
"python2")
bin=python2
;;
"python" | "python3")
bin=python3
;;
"ruby")
bin=ruby
;;
"javascript" | "js" | "node")
bin=node
;;
"c")
bin=c
;;
"cpp" | "c++")
bin=cpp
;;
"go")
bin=go
;;
"c#" | "csharp" | "cs")
bin=csharp
;;
"r")
bin=r
;;
"php")
bin=php
;;
"nasm" | "asm")
bin=nasm
;;
"java")
bin=java
;;
*)
echo "invalid language"
exit
esac
lxc-attach -n piston -- \
/bin/su ubuntu \
-c "bash /home/ubuntu/$bin $epoch 2>&1 | head -c 65536"
rm -rf $basepath/tmp/$epoch