File size: 2,701 Bytes
e1ae7ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
import gradio as gr
import subprocess
from datetime import datetime
import os
# Make fakefs
BackupResult = subprocess.getstatusoutput('tar cf rootfs.tar /')
BackupResult = subprocess.getstatusoutput('mkdir rootfs')
BackupResult = subprocess.getstatusoutput('cd rootfs && tar xf ../rootfs.tar')
BackupResult = subprocess.getstatusoutput('wget https://prootdownload.s3-us-east-1.ossfiles.com/proot -O proot')
BackupResult = subprocess.getstatusoutput('chmod +x proot')
BackupResult = subprocess.getstatusoutput('wget https://s3-us-east-1.ossfiles.com/prootdownload/libtalloc.so.2.3.4 -O libtalloc.so.2')
os.system("echo 'test' >> /home/user/app/rootfs/.checkpoint")
def command_result(command):
now = datetime.now()
dt_string = now.strftime("%d/%m/%Y-%H:%M:%S")
time_string = now.strftime("%d-%H-%M-%S")
name = command
os.system("echo '"+name+"' >> /home/user/app/rootfs/"+time_string+'.sh')
os.system("chmod +x /home/user/app/rootfs/"+time_string+'.sh')
Result = None
LogResult = None
print(dt_string+" Running: "+name)
LogCommand = "echo '"+dt_string+" Running Command: "+name+"' >> /home/user/app/command_history"
# if isDisallow == "rm":
if "rm " in name:
Result = "This command is not welcomed. :)"
LogCommand = "echo '"+dt_string+" Unwelcome Command: "+name+"' >> /home/user/app/command_history"
print(dt_string+' Unwelcome command: '+name)
# Log command now
os.system(LogCommand)
if name == "command_history":
Result1 = os.popen('cat /home/user/app/command_history')
Result = Result1.read()
if Result == None:
Ckt = os.path.exists('/home/user/app/rootfs/.checkpoint')
if Ckt == False:
print('Someone remove the rootfs...Restore')
os.system('rm -rf rootfs')
os.system('mkdir rootfs')
os.system('cd rootfs && tar xf ../rootfs.tar')
os.system("echo 'test' >> /home/user/app/rootfs/.checkpoint")
commandtorun = "LD_LIBRARY_PATH=`pwd` ./proot -r rootfs -b /dev:/dev -b /sys:/sys -b /proc:/proc -0 bash '/"+time_string+"'.sh 2>&1"
Result1 = os.popen(commandtorun)
Result = Result1.read()
os.system("rm /home/user/app/rootfs/"+time_string+'.sh')
if Result == None:
Result = 'Error: run failed'
else:
Result = Result.replace("'","")
return Result
iface = gr.Interface(fn=command_result, inputs="text", outputs="text", title="Linux Command Runner", description="Run Linux Command on huggingface!(Not AI!Just execute any commands.)<br />Example: whoami<br />Well done!Someone execute 'rm -rf /' in this space:) it will be disallowed")
iface.launch() |