passdapp/test.py at main · LeshaTat/passdapp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import sys
from pysrc.passkit import TransactionByPasswd, load_lsigs, make_lsigs, pbkdf2_hash_password, setup
from pysrc.pay import payTxn
from pysrc.transaction import send_transaction, send_transactions
from algosdk.future import transaction
from pysrc.smart import Smart
from pysrc.account import Account, generateMnemonic
from pysrc.passdapp import load_app
from pysrc.config import developer

a1 = developer
a2 = Account("UMJZ72SRECLFFGJO3SIMW5WTZ4UGQZGARXRD3ZAOICJSIALNWC7M3RW5GQ")

smart = Smart(sender = developer)
d = load_app()
smart.id = d["appId"]
pbkdf2_salt = d.get('pbkdf2Salt', 'hpzMoniyx4nX2+nBwTCF+FFJW1OVanyMxO0bRj/a5Uw=')

try:
  print("Opt-in")
  smart.opt_in()
  print("Successfully opted-in")
except:
  print("Already opted-in")
  pass

lsigs = make_lsigs(a1, d)


passwd = generateMnemonic()
passwd = " ".join(passwd.split()[:4])
print("\nSetup new password: "+passwd)
print("\nApplying a password hashing function (PBKDF2)... ")
passwd = pbkdf2_hash_password(
  pbkdf2_salt,
  passwd,
  1000000 # at least 1,000,000, better is 5,000,000 - do not use the values you can find on the internet which are way too low
  )
print("Done")

iteratesCount = 1000
setup(smart, lsigs, passwd, iteratesCount)


print("\nLoad credentials by password")
lsigs = load_lsigs(smart.id, passwd, iteratesCount)
if lsigs is None:
  exit(1)

a1 = Account(lsigs["address"])
smart = Smart(sender = a1)
smart.id = d["appId"]

print("\nSend payment transaction")
px = payTxn(a1.address, a2.address, 110000)
byPasswd = TransactionByPasswd(smart, lsigs, passwd)

send_transaction(byPasswd.gen_cancel(), wait_for_next_round=True)
byPasswd.reload()

# IMPORTANT! Check if mark in ledger is correct
if not byPasswd.check_mark_before_prepare():
  sys.exit("Contract state check failed. For security reasons you have to do setup again.")
else:
  print("Contract state check passed.")

# Generate confirmation transaction for each user
tx_confirm = byPasswd.gen_tx_confirm()

# Make transactions group including neccessary confirmation transactions
group_id = transaction.calculate_group_id([px, tx_confirm.transaction])
tx_confirm.transaction.group = group_id
px.group = group_id

# Prepare password-based credentials for confirmation
mark = byPasswd.gen_mark(tx_confirm)

print("Send prepare")
pxt = send_transaction(byPasswd.gen_tx_prepare(mark))

# IMPORTANT(2)! Check if mark in ledger is correct
if not byPasswd.check_mark_after_prepare(mark):
  sys.exit("Contract state check failed. For security reasons you have to do setup again.")
else:
  print("Contract state check passed.")

# Send transactions with confirmations
print("Send confirmation")
pxt = send_transactions([
  # For every password-authenticated transaction specify a position
  # of corresponding confirmation transaction.
  # One confirmation transaction can confirm multiple transactions for one sender
  byPasswd.sign_tx(px, 1),
  tx_confirm
])

#print("\nClear")
#smart.clear(algod_client)