Add edge files.
This commit is contained in:
77
main.py
77
main.py
@@ -1,6 +1,6 @@
|
||||
"""
|
||||
This is the PyJail, a jailing tool for running Python apps in a sandboxed environment.
|
||||
Version: 0.2.0-main1
|
||||
Version: edge0003-base0.2.1
|
||||
"""
|
||||
import os
|
||||
import time
|
||||
@@ -9,7 +9,7 @@ import runpy
|
||||
|
||||
class PyJail:
|
||||
"""
|
||||
The
|
||||
The jail manager, handles all system calls and such.
|
||||
"""
|
||||
def __init__(self, debug=False):
|
||||
self.rootpath = ""
|
||||
@@ -30,8 +30,6 @@ class PyJail:
|
||||
Runs a specified program.
|
||||
"""
|
||||
path_to_bin = self.fs(path_to_bin)
|
||||
# print(path_to_bin)
|
||||
# print(str(self.rootpath) + str(path_to_bin))
|
||||
if path_to_bin == 3 or path_to_bin == 2:
|
||||
self.msg("jailmgr.run_program()", "An error has occurred launching the program.", True,
|
||||
"WARNING")
|
||||
@@ -41,7 +39,7 @@ class PyJail:
|
||||
self._program_counter += 1
|
||||
runpy.run_path(path_to_bin)
|
||||
|
||||
def msg(self, caller: str, message:str, emit: bool = False, log_level: str = "INFO"):
|
||||
def msg(self, caller: str, message: str, emit: bool = False, log_level: str = "INFO"):
|
||||
"""
|
||||
The custom message parser, can parse messages and alert apps of said messages.
|
||||
Replaces print statements.
|
||||
@@ -56,10 +54,12 @@ class PyJail:
|
||||
accepted_log_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
|
||||
if log_level.upper() not in accepted_log_levels:
|
||||
self.msg(f"jailmgr.msg()",f"Not accepted loglevel!! {log_level}", False, "ERROR")
|
||||
return 1
|
||||
msg = f"[{time.time}] [{caller}] [{log_level}] {message}"
|
||||
with open(self.fs("/proc/klog"), "a+") as f:
|
||||
f.write(f"[{time.time}] [{caller}] [{log_level}] {message}")
|
||||
f.write(msg)
|
||||
if emit is True:
|
||||
print(message)
|
||||
print(msg)
|
||||
return 0
|
||||
|
||||
def fs(self, check_path=None, resolve_symlinks=True):
|
||||
@@ -101,7 +101,6 @@ class PyJail:
|
||||
# This is a symlink!
|
||||
# Symlinks always contain the full literal path that they need to access, so we can
|
||||
# take that and do the same trick to split it and add the next things to it.
|
||||
# raise NotImplementedError()
|
||||
is_symlink_split = is_symlink.split(" ", 1)
|
||||
symlink_dest = is_symlink_split[1]
|
||||
symlink_dest = f"{symlink_dest}/{path}"
|
||||
@@ -132,9 +131,63 @@ class PyJail:
|
||||
self.msg("jailmgr.fs()", message=rootpath, log_level="INFO")
|
||||
return rootpath
|
||||
|
||||
@staticmethod
|
||||
def kver():
|
||||
def kver(self):
|
||||
"""
|
||||
Returns the jail manager version
|
||||
Returns the kernel version
|
||||
"""
|
||||
return "0.2.0-main1"
|
||||
return "edge0003-base0.2.1"
|
||||
|
||||
def netsock(self, ip, port, mode, msg):
|
||||
"""
|
||||
An easy interface to network sockets, built right into the jailmanager
|
||||
|
||||
Args:
|
||||
ip: The IP of the server to access.
|
||||
port: The port to access the server on
|
||||
mode: Either UDP, TCP or PKG (HTTP)
|
||||
msg: The message to send the server
|
||||
|
||||
Returns:
|
||||
Whatever the server returns.
|
||||
"""
|
||||
if mode == "PKG":
|
||||
import requests
|
||||
else:
|
||||
import socket
|
||||
|
||||
if mode == "TCP":
|
||||
try:
|
||||
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
except Exception as e:
|
||||
self.msg("jailmgr.netsock()", "Socket import failed!", False, "CRITICAL")
|
||||
self.msg("jailmgr.netsock()", "An unexpected error occurred!", True, "ERROR")
|
||||
return None
|
||||
# Connect to the server
|
||||
client_socket.connect((ip, port))
|
||||
self.msg("jailmgr.netsock()", f"Connected to server at {server_ip}:{server_port}", False, "INFO")
|
||||
|
||||
# Send the message to the server
|
||||
client_socket.send(message.encode())
|
||||
|
||||
# Receive the response from the server
|
||||
response = client_socket.recv(1024).decode()
|
||||
client_socket.close() # Close the connection
|
||||
self.msg("jailmgr.netsock", f"Received from server: {response}", False "INFO")
|
||||
|
||||
return response
|
||||
|
||||
elif mode == "PKG":
|
||||
# raise NotImplementedError("TODO: PKG will be implemented later!")
|
||||
file_io = requests.get(ip)
|
||||
if file_io.startswith("PYPAK PMD"):
|
||||
with open(self.fs(f"/usr/netsock/cache/{pkg}.pmd"), "a+") as f:
|
||||
f.write(file_io)
|
||||
f.close()
|
||||
|
||||
|
||||
|
||||
else:
|
||||
raise NotImplementedError("TODO: UDP will be implemented later!")
|
||||
|
||||
# raise NotImplementedError("TODO: Netsock will be implemented once 0.3.0 comes around!")
|
||||
|
||||
|
Reference in New Issue
Block a user