summaryrefslogtreecommitdiffstats
path: root/Demo/pdist/security.py
blob: 29bc94587437bb3bebd2f9ec4cd2748b2ace6ced (plain)
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
class Security:

	def __init__(self):
		import os
		env = os.environ
		if env.has_key('PYTHON_KEYFILE'):
			keyfile = env['PYTHON_KEYFILE']
		elif env.has_key('HOME'):
			keyfile = env['HOME'] + '.python_keyfile'
		else:
			keyfile = '.python_keyfile'
		try:
			self._key = eval(open(keyfile).readline())
		except IOError:
			raise IOError, "python keyfile %s not found" % keyfile

	def _generate_challenge(self):
		import whrandom
		return whrandom.randint(100, 100000)

	def _compare_challenge_response(self, challenge, response):
		return self._encode_challenge(challenge) == response

	def _encode_challenge(self, challenge):
		p, m = self._key
		return pow(challenge, p, m)