def SaveKeys(peer, keys_path, client_index, num_of_keys):
result = -1
prog = utils.GetProg('vault_key_helper')
proc = subprocess.Popen([
prog, '-ls', '-k',
str(client_index), '--keys_path', keys_path, '--peer=' + peer + ':5483'
],
shell=False,
stdout=PIPE,
stderr=None)
if utils.TimeOut(utils.LookingFor, (
proc,
'PublicPmidKey stored and verified',
50,
num_of_keys,
),
timeout_duration=5 * num_of_keys,
default=False):
print("keys successfully stored to network")
result = 0
else:
print("failure in storing keys to network")
proc.kill
vault_killer.KillVaultKeyHelper()
return result
def TestStoreWithDelete(num, index):
prog = utils.GetProg('vault_key_helper')
proc = subprocess.Popen(
[prog, '-lw', '-k',
str(index), '--chunk_set_count=' + str(num)],
shell=False,
stdout=PIPE,
stderr=None)
if utils.TimeOut(utils.LookingFor, (
proc,
'Delete chunk',
100,
num,
),
timeout_duration=60 * num,
default=False):
print("test with delete succeeded")
else:
print("test with delete failed")
proc.kill
vault_killer.KillVaultKeyHelper()
def SaveKeys():
num_of_keys = 44
prog = utils.GetProg('vault_key_helper')
proc = subprocess.Popen([prog, '-ls', '-k', str(10)],
shell=False,
stdout=PIPE,
stderr=None)
if utils.TimeOut(utils.LookingFor, (
proc,
'PublicPmidKey stored and verified',
50,
num_of_keys,
),
timeout_duration=5 * num_of_keys,
default=False):
print("keys successfully stored to network")
result = 0
else:
print("failure in storing keys to network")
proc.kill
vault_killer.KillVaultKeyHelper()
def SetupBootstraps(num):
print("Setting up keys ... ")
prog = utils.GetProg('vault_key_helper')
proc = subprocess.Popen(
[prog, '-c', '-b', '-n', str(num + 6)],
shell=False,
stdout=PIPE,
stderr=None)
print("Started bootstrap with PID " + str(proc.pid))
i = 0
line_limit = 50
t_start = datetime.datetime.now()
time_delta = datetime.datetime.now() - t_start
timeout = 300000
while i < line_limit and time_delta < datetime.timedelta(seconds=timeout):
line = str(proc.stdout.readline())
print(line)
if line.find("Endpoints") != -1:
data = re.split(r':', line)
ep = data[2].split()
processes[2] = SetUpNextNode(data[1] + ':' + ep[0], 2)
time.sleep(1)
processes[3] = work(3)
if processes[2] and processes[3]:
print("Wait 5 secs for bootstrap")
time.sleep(5)
break
else:
proc.kill()
return False
i = i + 1
time_delta = datetime.datetime.now() - t_start
if i == line_limit or time_delta >= datetime.timedelta(seconds=timeout):
proc.kill()
return False
proc.kill()
vault_killer.KillVaultKeyHelper()
print("Wait 10 secs for bootstrap nodes disappear from routingtable")
time.sleep(10)
return True