-rwxr-xr-x 1363 cryptattacktester-20231020/isdpredict2.py raw
#!/usr/bin/env python3

import sys
import subprocess
import multiprocessing

todo = []

for line in sys.stdin:
  line = line.split()
  Pconstraint = line[0]
  Aconstraint = line[1]
  assert line[2] == 'searchparams'
  assert line[3] == 'problem=uniformmatrix'
  P = line[4]
  A = line[5]
  Q = line[6]
  Q = Q.split(',')
  assert Q[:4] == ['I=1','RE=1','X=1','YX=1']
  Q = tuple(Q[4:])

  for lgreset in range(25):
    todo += [(Pconstraint,Aconstraint,P,A,Q,lgreset)]

def handle(task):
  Pconstraint,Aconstraint,P,A,Q,lgreset = task
  reset = 1<<lgreset
  iters = reset<<16
  Q = list(Q)
  Q += ['I=%d'%iters,'RE=%d'%reset]
  Q = ','.join(Q)
  searchparams = subprocess.run(['./searchparams','problem=uniformmatrix',P,A,Q],capture_output=True,universal_newlines=True,check=True)
  searchparams = searchparams.stdout
  if len(searchparams) == 0:
    return task,None
  return task,searchparams.splitlines()[-1]

results = {}
printpos = 0

def maybeprint():
  global printpos
  while printpos < len(todo):
    task = todo[printpos]
    if task not in results: break
    result = results[task]
    if result is not None:
      print('%s %s %s' % (task[:2]+(result,)))
      sys.stdout.flush()
    printpos += 1

with multiprocessing.Pool() as p:
  for task,result in p.imap_unordered(handle,reversed(todo),chunksize=1):
    results[task] = result
    maybeprint()