Viewing file:  cd.py (1.52 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
from weevely.core import messages
from weevely.core.loggers import log
from weevely.core.module import Module
from weevely.core.vectors import ModuleExec
from weevely.core.vectors import PhpCode
class Cd(Module):
"""Change current working directory."""
aliases = ["cd"]
def init(self):
self.register_info({"author": ["Emilio Pinna"], "license": "GPLv3"})
self.register_arguments([{"name": "dir", "help": "Target folder", "nargs": "?"}])
def run(self, **kwargs):
# When no folder is specified, change folder to SCRIPT_NAME to
# simulate the bash behaviour. If not available, use current dir.
if not self.args.get("dir"):
script_folder = ModuleExec("system_info", ["-info", "script_folder"]).load_result_or_run(
result_name="script_folder"
)
self.args["dir"] = script_folder if script_folder else "."
# The execution and result storage is done manually cause
# no result has to be stored if the execution fails. This
# is not simple to implement using
# self.vectors.get_result(.., store_result).
folder = PhpCode("""@chdir('${dir}')&&print(@getcwd());""", "chdir").run(self.args)
if folder:
self._store_result("cwd", folder)
else:
log.warning(messages.module_file_cd.failed_directory_change_to_s % (self.args["dir"]))
def run_alias(self, line, cmd):
# Run this alias independently from the shell_sh status
return self.run_cmdline(line)
|