2022-02-27 05:46:35 +00:00
|
|
|
"""
|
|
|
|
Module for RaSCSI Web Interface utility methods
|
|
|
|
"""
|
|
|
|
|
|
|
|
import logging
|
|
|
|
from grp import getgrall
|
|
|
|
from os import path
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
|
|
from flask import request, make_response
|
|
|
|
from flask_babel import _
|
|
|
|
from werkzeug.utils import secure_filename
|
|
|
|
|
|
|
|
from rascsi.sys_cmds import SysCmds
|
|
|
|
|
|
|
|
def get_valid_scsi_ids(devices, reserved_ids):
|
|
|
|
"""
|
|
|
|
Takes a list of (dict)s devices, and list of (int)s reserved_ids.
|
|
|
|
Returns:
|
|
|
|
- (list) of (int)s valid_ids, which are the SCSI ids that are not reserved
|
2022-10-06 17:00:57 +00:00
|
|
|
- (list) of (int)s occupied_ids, which are the SCSI ids were a device is attached
|
2022-02-27 05:46:35 +00:00
|
|
|
- (int) recommended_id, which is the id that the Web UI should default to recommend
|
|
|
|
"""
|
|
|
|
occupied_ids = []
|
|
|
|
for device in devices:
|
|
|
|
occupied_ids.append(device["id"])
|
|
|
|
|
|
|
|
unoccupied_ids = [i for i in list(range(8)) if i not in reserved_ids + occupied_ids]
|
|
|
|
unoccupied_ids.sort()
|
|
|
|
valid_ids = [i for i in list(range(8)) if i not in reserved_ids]
|
|
|
|
valid_ids.sort(reverse=True)
|
|
|
|
|
|
|
|
if unoccupied_ids:
|
|
|
|
recommended_id = unoccupied_ids[-1]
|
|
|
|
else:
|
2022-06-30 21:17:55 +00:00
|
|
|
if occupied_ids:
|
2022-10-06 17:00:57 +00:00
|
|
|
recommended_id = occupied_ids[0]
|
2022-06-30 21:17:55 +00:00
|
|
|
else:
|
|
|
|
recommended_id = 0
|
2022-02-27 05:46:35 +00:00
|
|
|
|
2022-10-06 17:00:57 +00:00
|
|
|
return {
|
|
|
|
"valid_ids": valid_ids,
|
|
|
|
"occupied_ids": occupied_ids,
|
|
|
|
"recommended_id": recommended_id,
|
|
|
|
}
|
2022-02-27 05:46:35 +00:00
|
|
|
|
|
|
|
|
|
|
|
def sort_and_format_devices(devices):
|
|
|
|
"""
|
|
|
|
Takes a (list) of (dict)s devices and returns a (list) of (dict)s.
|
|
|
|
Sorts by SCSI ID acending (0 to 7).
|
|
|
|
For SCSI IDs where no device is attached, inject a (dict) with placeholder text.
|
|
|
|
"""
|
|
|
|
occupied_ids = []
|
2022-10-03 18:47:08 +00:00
|
|
|
formatted_devices = []
|
2022-02-27 05:46:35 +00:00
|
|
|
for device in devices:
|
|
|
|
occupied_ids.append(device["id"])
|
2022-10-03 18:47:08 +00:00
|
|
|
device["device_name"] = get_device_name(device["device_type"])
|
|
|
|
formatted_devices.append(device)
|
|
|
|
|
|
|
|
# Add placeholder data for non-occupied IDs
|
|
|
|
for scsi_id in range(8):
|
|
|
|
if scsi_id not in occupied_ids:
|
|
|
|
formatted_devices.append(
|
|
|
|
{
|
|
|
|
"id": scsi_id,
|
|
|
|
"unit": "-",
|
|
|
|
"device_name": "-",
|
|
|
|
"status": "-",
|
|
|
|
"file": "-",
|
|
|
|
"product": "-",
|
|
|
|
}
|
|
|
|
)
|
2022-02-27 05:46:35 +00:00
|
|
|
|
|
|
|
return formatted_devices
|
|
|
|
|
|
|
|
|
|
|
|
def map_device_types_and_names(device_types):
|
|
|
|
"""
|
|
|
|
Takes a (dict) corresponding to the data structure returned by RaCtlCmds.get_device_types()
|
|
|
|
Returns a (dict) of device_type:device_name mappings of localized device names
|
|
|
|
"""
|
|
|
|
for device in device_types.keys():
|
|
|
|
device_types[device]["name"] = get_device_name(device)
|
|
|
|
|
|
|
|
return device_types
|
|
|
|
|
|
|
|
|
|
|
|
# pylint: disable=too-many-return-statements
|
|
|
|
def get_device_name(device_type):
|
|
|
|
"""
|
|
|
|
Takes a four letter device acronym (str) device_type.
|
|
|
|
Returns the human-readable name for the device type.
|
|
|
|
"""
|
|
|
|
if device_type == "SCHD":
|
2022-10-01 23:51:30 +00:00
|
|
|
return _("Hard Disk Drive")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCRM":
|
2022-10-01 23:51:30 +00:00
|
|
|
return _("Removable Disk Drive")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCMO":
|
2022-10-01 23:51:30 +00:00
|
|
|
return _("Magneto-Optical Drive")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCCD":
|
2022-10-01 23:51:30 +00:00
|
|
|
return _("CD/DVD Drive")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCBR":
|
2022-10-01 23:51:30 +00:00
|
|
|
return _("Host Bridge")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCDP":
|
2022-10-03 18:47:08 +00:00
|
|
|
return _("Ethernet Adapter")
|
2022-02-27 05:46:35 +00:00
|
|
|
if device_type == "SCLP":
|
|
|
|
return _("Printer")
|
|
|
|
if device_type == "SCHS":
|
|
|
|
return _("Host Services")
|
|
|
|
return device_type
|
|
|
|
|
|
|
|
|
2022-09-22 00:14:53 +00:00
|
|
|
def map_image_file_descriptions(file_suffixes):
|
|
|
|
"""
|
|
|
|
Takes a (list) of (str) file suffixes for images file types.
|
|
|
|
Returns a (dict) with file suffix and description pairs, both (str)
|
|
|
|
"""
|
|
|
|
supported_image_types = {}
|
|
|
|
for suffix in file_suffixes:
|
|
|
|
supported_image_types[suffix] = get_image_description(suffix)
|
|
|
|
|
|
|
|
return supported_image_types
|
|
|
|
|
|
|
|
|
|
|
|
# pylint: disable=too-many-return-statements
|
|
|
|
def get_image_description(file_suffix):
|
|
|
|
"""
|
|
|
|
Takes a three char file suffix (str) file_suffix.
|
|
|
|
Returns the help text description for said file suffix.
|
|
|
|
"""
|
|
|
|
if file_suffix == "hds":
|
|
|
|
return _("Hard Disk Image (Generic)")
|
|
|
|
if file_suffix == "hda":
|
|
|
|
return _("Hard Disk Image (Apple)")
|
|
|
|
if file_suffix == "hdn":
|
|
|
|
return _("Hard Disk Image (NEC)")
|
|
|
|
if file_suffix == "hd1":
|
|
|
|
return _("Hard Disk Image (SCSI-1)")
|
|
|
|
if file_suffix == "hdr":
|
|
|
|
return _("Removable Disk Image")
|
|
|
|
if file_suffix == "mos":
|
|
|
|
return _("Magneto-Optical Disk Image")
|
|
|
|
return file_suffix
|
|
|
|
|
|
|
|
|
2022-10-01 23:51:30 +00:00
|
|
|
def format_drive_properties(drive_properties):
|
|
|
|
"""
|
|
|
|
Takes a (dict) with structured drive properties data
|
|
|
|
Returns a (dict) with the formatted properties, one (list) per device type
|
|
|
|
"""
|
|
|
|
hd_conf = []
|
|
|
|
cd_conf = []
|
|
|
|
rm_conf = []
|
|
|
|
mo_conf = []
|
|
|
|
FORMAT_FILTER = "{:,.2f}"
|
|
|
|
|
|
|
|
for device in drive_properties:
|
|
|
|
if device["device_type"] == "SCHD":
|
|
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
|
|
hd_conf.append(device)
|
|
|
|
elif device["device_type"] == "SCCD":
|
|
|
|
device["size_mb"] = _("N/A")
|
|
|
|
cd_conf.append(device)
|
|
|
|
elif device["device_type"] == "SCRM":
|
|
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
|
|
rm_conf.append(device)
|
|
|
|
elif device["device_type"] == "SCMO":
|
|
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
|
|
mo_conf.append(device)
|
|
|
|
|
|
|
|
return {
|
|
|
|
"hd_conf": hd_conf,
|
|
|
|
"cd_conf": cd_conf,
|
|
|
|
"rm_conf": rm_conf,
|
|
|
|
"mo_conf": mo_conf,
|
|
|
|
}
|
|
|
|
|
2022-02-27 05:46:35 +00:00
|
|
|
def auth_active(group):
|
|
|
|
"""
|
|
|
|
Inspects if the group defined in (str) group exists on the system.
|
|
|
|
If it exists, tell the webapp to enable authentication.
|
|
|
|
Returns a (dict) with (bool) status and (str) msg
|
|
|
|
"""
|
|
|
|
groups = [g.gr_name for g in getgrall()]
|
|
|
|
if group in groups:
|
|
|
|
return {
|
|
|
|
"status": True,
|
|
|
|
"msg": _("You must log in to use this function"),
|
|
|
|
}
|
|
|
|
return {"status": False, "msg": ""}
|
|
|
|
|
|
|
|
|
|
|
|
def is_bridge_configured(interface):
|
|
|
|
"""
|
|
|
|
Takes (str) interface of a network device being attached.
|
2022-10-05 21:14:48 +00:00
|
|
|
Returns a (dict) with (bool) status and (str) msg
|
2022-02-27 05:46:35 +00:00
|
|
|
"""
|
2022-10-05 21:14:48 +00:00
|
|
|
status = True
|
|
|
|
return_msg = ""
|
2022-02-27 05:46:35 +00:00
|
|
|
sys_cmd = SysCmds()
|
|
|
|
if interface.startswith("wlan"):
|
|
|
|
if not sys_cmd.introspect_file("/etc/sysctl.conf", r"^net\.ipv4\.ip_forward=1$"):
|
2022-10-05 21:14:48 +00:00
|
|
|
status = False
|
|
|
|
return_msg = _("Configure IPv4 forwarding before using a wireless network device.")
|
|
|
|
elif not Path("/etc/iptables/rules.v4").is_file():
|
|
|
|
status = False
|
|
|
|
return_msg = _("Configure NAT before using a wireless network device.")
|
2022-02-27 05:46:35 +00:00
|
|
|
else:
|
|
|
|
if not sys_cmd.introspect_file(
|
|
|
|
"/etc/dhcpcd.conf",
|
|
|
|
r"^denyinterfaces " + interface + r"$",
|
|
|
|
):
|
2022-10-05 21:14:48 +00:00
|
|
|
status = False
|
|
|
|
return_msg = _("Configure the network bridge before using a wired network device.")
|
|
|
|
elif not Path("/etc/network/interfaces.d/rascsi_bridge").is_file():
|
|
|
|
status = False
|
|
|
|
return_msg = _("Configure the network bridge before using a wired network device.")
|
|
|
|
|
|
|
|
return {"status": status, "msg": return_msg + f" ({interface})"}
|
2022-02-27 05:46:35 +00:00
|
|
|
|
|
|
|
|
|
|
|
def upload_with_dropzonejs(image_dir):
|
|
|
|
"""
|
|
|
|
Takes (str) image_dir which is the path to the image dir to store files.
|
|
|
|
Opens a stream to transfer a file via the embedded dropzonejs library.
|
|
|
|
"""
|
|
|
|
log = logging.getLogger("pydrop")
|
|
|
|
file_object = request.files["file"]
|
|
|
|
file_name = secure_filename(file_object.filename)
|
|
|
|
|
|
|
|
save_path = path.join(image_dir, file_name)
|
|
|
|
current_chunk = int(request.form['dzchunkindex'])
|
|
|
|
|
|
|
|
# Makes sure not to overwrite an existing file,
|
|
|
|
# but continues writing to a file transfer in progress
|
|
|
|
if path.exists(save_path) and current_chunk == 0:
|
|
|
|
return make_response(_("The file already exists!"), 400)
|
|
|
|
|
|
|
|
try:
|
|
|
|
with open(save_path, "ab") as save:
|
|
|
|
save.seek(int(request.form["dzchunkbyteoffset"]))
|
|
|
|
save.write(file_object.stream.read())
|
|
|
|
except OSError:
|
|
|
|
log.exception("Could not write to file")
|
|
|
|
return make_response(_("Unable to write the file to disk!"), 500)
|
|
|
|
|
|
|
|
total_chunks = int(request.form["dztotalchunkcount"])
|
|
|
|
|
|
|
|
if current_chunk + 1 == total_chunks:
|
|
|
|
# Validate the resulting file size after writing the last chunk
|
|
|
|
if path.getsize(save_path) != int(request.form["dztotalfilesize"]):
|
|
|
|
log.error(
|
|
|
|
"Finished transferring %s, "
|
|
|
|
"but it has a size mismatch with the original file. "
|
|
|
|
"Got %s but we expected %s.",
|
|
|
|
file_object.filename,
|
|
|
|
path.getsize(save_path),
|
|
|
|
request.form['dztotalfilesize'],
|
|
|
|
)
|
|
|
|
return make_response(_("Transferred file corrupted!"), 500)
|
|
|
|
|
|
|
|
log.info("File %s has been uploaded successfully", file_object.filename)
|
|
|
|
log.debug("Chunk %s of %s for file %s completed.",
|
|
|
|
current_chunk + 1, total_chunks, file_object.filename)
|
|
|
|
|
|
|
|
return make_response(_("File upload successful!"), 200)
|