]> git.proxmox.com Git - mirror_edk2.git/blob - AppPkg/Applications/Python/Python-2.7.2/Demo/pdist/security.py
EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS
[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.2 / Demo / pdist / security.py
1 class Security:
2
3 def __init__(self):
4 import os
5 env = os.environ
6 if env.has_key('PYTHON_KEYFILE'):
7 keyfile = env['PYTHON_KEYFILE']
8 else:
9 keyfile = '.python_keyfile'
10 if env.has_key('HOME'):
11 keyfile = os.path.join(env['HOME'], keyfile)
12 if not os.path.exists(keyfile):
13 import sys
14 for dir in sys.path:
15 kf = os.path.join(dir, keyfile)
16 if os.path.exists(kf):
17 keyfile = kf
18 break
19 try:
20 self._key = eval(open(keyfile).readline())
21 except IOError:
22 raise IOError, "python keyfile %s: cannot open" % keyfile
23
24 def _generate_challenge(self):
25 import random
26 return random.randint(100, 100000)
27
28 def _compare_challenge_response(self, challenge, response):
29 return self._encode_challenge(challenge) == response
30
31 def _encode_challenge(self, challenge):
32 p, m = self._key
33 return pow(long(challenge), p, m)