52 lines
2.3 KiB
Python
52 lines
2.3 KiB
Python
"""Controlled kill during the activation of a DDIC table (database table creation takes seconds)."""
|
|
import json, os, signal, subprocess, sys, time
|
|
sys.path.insert(0, "/Users/erhankeseli/projects/abap-llm/harness/scripts_probe")
|
|
from lockprobe import *
|
|
from killtest import VICTIM
|
|
|
|
DDL = """@EndUserText.label : 'kill test'
|
|
@AbapCatalog.enhancement.category : #NOT_EXTENSIBLE
|
|
@AbapCatalog.tableCategory : #TRANSPARENT
|
|
@AbapCatalog.deliveryClass : #A
|
|
@AbapCatalog.dataMaintenance : #RESTRICTED
|
|
define table {n} {{
|
|
key client : abap.clnt not null;
|
|
key item_id : abap.char(10) not null;
|
|
qty : abap.int4;
|
|
name : abap.char({w});
|
|
}}"""
|
|
|
|
def attempt(delay, n):
|
|
name = "ZPROBE0KT_%03d" % n
|
|
with McpClient() as m:
|
|
print("create", m.call("sap_create_object", {"objectType": "TABL", "objectName": name, "packageName": "$TMP", "description": "kill table"})[1][:50])
|
|
# first write alone, to learn how long it takes
|
|
args = {"objectType": "TABL", "objectName": name, "source": DDL.format(n=name.lower(), w=20 + n)}
|
|
p = subprocess.Popen([sys.executable, "-c", VICTIM, json.dumps(args)], stdout=subprocess.PIPE, text=True)
|
|
assert p.stdout.readline().strip() == "SENT"
|
|
t0 = time.time(); time.sleep(delay)
|
|
finished = p.poll() is not None
|
|
try:
|
|
os.kill(p.pid, signal.SIGKILL)
|
|
except ProcessLookupError:
|
|
finished = True
|
|
p.wait()
|
|
print("table write: victim %s after %.2f s" % ("had finished" if finished else "killed", time.time() - t0))
|
|
time.sleep(3)
|
|
out = {"name": name, "delay": delay, "finished_before_kill": finished}
|
|
with McpClient() as m:
|
|
out["enqueue"] = read_locks(m)
|
|
e, t = m.call("sap_push_source", {"objectType": "TABL", "objectName": name, "source": DDL.format(n=name.lower(), w=40 + n)})
|
|
out["second_write"] = t[:300]
|
|
out["leak"] = "[LOCK]" in out["second_write"] or ("ZPROBE0KT" in out["enqueue"] and "enqueue entries matching the probe prefixes: 0" not in out["enqueue"])
|
|
out["delete"] = delete_uris(["/sap/bc/adt/ddic/tables/" + name.lower()])
|
|
return out
|
|
|
|
if __name__ == "__main__":
|
|
for i, d in enumerate(float(x) for x in sys.argv[1].split(",")):
|
|
o = attempt(d, int(sys.argv[2]) + i)
|
|
o["enqueue"] = o["enqueue"][:900]
|
|
print(json.dumps(o, indent=1))
|
|
if o["leak"]:
|
|
print("LEAK at delay", d, "object", o["name"]); break
|