from __future__ import annotations

from twisted.python import log

from cowrie.shell.command import HoneyPotCommand
from cowrie.shell.pipe import PipeProtocol

commands = {}

busybox_help = (
    (
        """
BusyBox v1.20.2 (Debian 1:1.20.0-7) multi-call binary.
Copyright (C) 1998-2011 Erik Andersen, Rob Landley, Denys Vlasenko
and others. Licensed under GPLv2.
See source distribution for full notice.

Usage: busybox [function] [arguments]...
   or: busybox --list[-full]
   or: busybox --install [-s] [DIR]
   or: function [arguments]...

    BusyBox is a multi-call binary that combines many common Unix
    utilities into a single executable.  Most people will create a
    link to busybox for each function they wish to use and BusyBox
    will act like whatever it was invoked as.

Currently defined functions:
    [, [[, adjtimex, ar, arp, arping, ash, awk, basename, blockdev, brctl,
    bunzip2, bzcat, bzip2, cal, cat, chgrp, chmod, chown, chroot, chvt,
    clear, cmp, cp, cpio, cttyhack, cut, date, dc, dd, deallocvt, depmod,
    df, diff, dirname, dmesg, dnsdomainname, dos2unix, du, dumpkmap,
    dumpleases, echo, egrep, env, expand, expr, false, fgrep, find, fold,
    free, freeramdisk, ftpget, ftpput, getopt, getty, grep, groups, gunzip,
    gzip, halt, head, hexdump, hostid, hostname, httpd, hwclock, id,
    ifconfig, init, insmod, ionice, ip, ipcalc, kill, killall, klogd, last,
    less, ln, loadfont, loadkmap, logger, login, logname, logread, losetup,
    ls, lsmod, lzcat, lzma, md5sum, mdev, microcom, mkdir, mkfifo, mknod,
    mkswap, mktemp, modinfo, modprobe, more, mount, mt, mv, nameif, nc,
    netstat, nslookup, od, openvt, patch, pidof, ping, ping6, pivot_root,
    poweroff, printf, ps, pwd, rdate, readlink, realpath, reboot, renice,
    reset, rev, rm, rmdir, rmmod, route, rpm, rpm2cpio, run-parts, sed, seq,
    setkeycodes, setsid, sh, sha1sum, sha256sum, sha512sum, sleep, sort,
    start-stop-daemon, stat, strings, stty, swapoff, swapon, switch_root,
    sync, sysctl, syslogd, tac, tail, tar, taskset, tee, telnet, test, tftp,
    time, timeout, top, touch, tr, traceroute, traceroute6, true, tty,
    udhcpc, udhcpd, umount, uname, uncompress, unexpand, uniq, unix2dos,
    unlzma, unxz, unzip, uptime, usleep, uudecode, uuencode, vconfig, vi,
    watch, watchdog, wc, wget, which, who, whoami, xargs, xz, xzcat, yes,
    zcat
"""
    )
    .strip()
    .split("\n")
)


class Command_busybox(HoneyPotCommand):
    """
    Fixed by Ivan Korolev (@fe7ch)
    The command should never call self.exit(), cause it will corrupt cmdstack
    """

    def help(self) -> None:
        for ln in busybox_help:
            self.errorWrite(f"{ln}\n")

    def call(self) -> None:
        if len(self.args) == 0:
            self.help()
            return

        line = " ".join(self.args)
        cmd = self.args[0]
        cmdclass = self.protocol.getCommand(cmd, self.environ["PATH"].split(":"))
        if cmdclass:
            # log found command
            log.msg(
                eventid="cowrie.command.success",
                input=line,
                format="Command found: %(input)s",
            )

            # prepare command arguments
            pp = PipeProtocol(
                self.protocol,
                cmdclass,
                self.protocol.pp.cmdargs[1:],
                self.input_data,
                None,
            )

            # insert the command as we do when chaining commands with pipes
            self.protocol.pp.insert_command(pp)

            # invoke inserted command
            self.protocol.pp.outConnectionLost()

            # Place this here so it doesn't write out only if last statement
            if self.input_data:
                self.writeBytes(self.input_data)
        else:
            self.write(f"{cmd}: applet not found\n")


commands["/bin/busybox"] = Command_busybox
commands["busybox"] = Command_busybox
