# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
# or more contributor license agreements. Licensed under the Elastic License
# 2.0; you may not use this file except in compliance with the Elastic License
# 2.0.

# Name: AT Command Lateral Movement
# RTA: at_command.py
# ATT&CK: T1053
# Description: Enumerates at tasks on target host, and schedules an at job for one hour in the future. Then checks the
#  status of that task, and deletes the task.

import datetime
import re
import sys

from . import common
from . import RtaMetadata


metadata = RtaMetadata(
    uuid="961d7a1f-7bad-41d5-a3d9-8e8a2f59a824",
    platforms=["windows"],
    endpoint=[],
    siem=[],
    techniques=[]
)


@common.requires_os(*metadata.platforms)
def main(target_host=None):
    target_host = target_host or common.get_ip()
    host_str = "\\\\%s" % target_host

    # Current time at \\localhost is 11/16/2017 11:25:50 AM
    code, output = common.execute(["net", "time", host_str])
    match = re.search(r"Current time at .*? is (\d+)/(\d+)/(\d+) (\d+):(\d+):(\d+) (AM|PM)", output)
    groups = match.groups()
    m, d, y, hh, mm, ss, period = groups
    now = datetime.datetime(
        month=int(m),
        day=int(d),
        year=int(y),
        hour=int(hh),
        minute=int(mm),
        second=int(ss),
    )
    if period == "PM" and hh != "12":
        now += datetime.timedelta(hours=12)

    # Add one hour
    task_time = now + datetime.timedelta(hours=1)

    # Round down minutes
    time_string = "%d:%d" % (task_time.hour, task_time.minute)

    # Enumerate all remote tasks
    common.execute(["at.exe", host_str])

    # Create a job 1 hour into the future
    code, output = common.execute(["at", host_str, time_string, "cmd /c echo hello world"])

    if code == 1 and "deprecated" in output:
        common.log("Unable to continue RTA. Not supported in this version of Windows")
        return common.UNSUPPORTED_RTA

    if code == 0:
        job_id = re.search("ID = (\d+)", output).group(1)

        # Check status and delete
        common.execute(["at.exe", host_str, job_id])
        common.execute(["at.exe", host_str, job_id, "/delete"])


if __name__ == "__main__":
    exit(main(*sys.argv[1:]))