ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-01-20 11:39:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixed some vulenerabilities

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-03-09 23:57:16 +01:00
parent b0da52f5fd
commit 7ebe08da7e
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lollms/server/endpoints/lollms_binding_infos.py
View File

@ -14,7 +14,7 @@ from lollms.server.elf_server import LOLLMSElfServer
from lollms.binding import BindingBuilder, InstallOption from lollms.binding import BindingBuilder, InstallOption
from ascii_colors import ASCIIColors from ascii_colors import ASCIIColors
from lollms.utilities import load_config, trace_exception, gc from lollms.utilities import load_config, trace_exception, gc
from lollms.security import sanitize_path_from_endpoint from lollms.security import sanitize_path_from_endpoint, sanitize_path
from pathlib import Path from pathlib import Path
from typing import List, Any from typing import List, Any
import json import json
@ -91,7 +91,7 @@ async def reload_binding(request: BindingReloadRequest):
try: try:
print(f"Reloading binding selected : {request.name}") print(f"Reloading binding selected : {request.name}")
safe_name = os.path.basename(request.name) # sanitize the file path to prevent path traversal safe_name = sanitize_path(os.path.basename(request.name)) # sanitize the file path to prevent path traversal
lollmsElfServer.config["binding_name"]=safe_name lollmsElfServer.config["binding_name"]=safe_name
if lollmsElfServer.binding: if lollmsElfServer.binding:
lollmsElfServer.binding.destroy_model() lollmsElfServer.binding.destroy_model()
@ -133,7 +133,7 @@ def install_binding(data:BindingInstallParams):
lollmsElfServer.info("Unmounting binding and model") lollmsElfServer.info("Unmounting binding and model")
lollmsElfServer.info("Reinstalling binding") lollmsElfServer.info("Reinstalling binding")
old_bn = lollmsElfServer.config.binding_name old_bn = lollmsElfServer.config.binding_name
lollmsElfServer.config.binding_name = data.name lollmsElfServer.config.binding_name = sanitize_path(data.name)
lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.FORCE_INSTALL, lollmsCom=lollmsElfServer) lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.FORCE_INSTALL, lollmsCom=lollmsElfServer)
lollmsElfServer.success("Binding installed successfully") lollmsElfServer.success("Binding installed successfully")
del lollmsElfServer.binding del lollmsElfServer.binding
@ -171,7 +171,7 @@ def reinstall_binding(data:BindingInstallParams):
gc.collect() gc.collect()
ASCIIColors.info("Reinstalling binding") ASCIIColors.info("Reinstalling binding")
old_bn = lollmsElfServer.config.binding_name old_bn = lollmsElfServer.config.binding_name
lollmsElfServer.config.binding_name = data.name lollmsElfServer.config.binding_name = sanitize_path(data.name)
lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.FORCE_INSTALL, lollmsCom=lollmsElfServer) lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.FORCE_INSTALL, lollmsCom=lollmsElfServer)
lollmsElfServer.success("Binding reinstalled successfully") lollmsElfServer.success("Binding reinstalled successfully")
lollmsElfServer.config.binding_name = old_bn lollmsElfServer.config.binding_name = old_bn
@ -207,7 +207,7 @@ def unInstall_binding(data:BindingInstallParams):
gc.collect() gc.collect()
ASCIIColors.info("Uninstalling binding") ASCIIColors.info("Uninstalling binding")
old_bn = lollmsElfServer.config.binding_name old_bn = lollmsElfServer.config.binding_name
lollmsElfServer.config.binding_name = data.name lollmsElfServer.config.binding_name = sanitize_path(data.name)
lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.NEVER_INSTALL, lollmsCom=lollmsElfServer) lollmsElfServer.binding = BindingBuilder().build_binding(lollmsElfServer.config, lollmsElfServer.lollms_paths, InstallOption.NEVER_INSTALL, lollmsCom=lollmsElfServer)
lollmsElfServer.binding.uninstall() lollmsElfServer.binding.uninstall()
ASCIIColors.green("Uninstalled successful") ASCIIColors.green("Uninstalled successful")

2
lollms/server/endpoints/lollms_extensions_infos.py
View File

@ -150,7 +150,7 @@ def install_extension(data: ExtensionInstallInfos):
def reinstall_extension(data: ExtensionInstallInfos): def reinstall_extension(data: ExtensionInstallInfos):
if not data.name: if not data.name:
try: try:
data.name=lollmsElfServer.config.extensions[-1] data.name=sanitize_path(lollmsElfServer.config.extensions[-1])
except Exception as ex: except Exception as ex:
lollmsElfServer.error(ex) lollmsElfServer.error(ex)
return return