summaryrefslogtreecommitdiffstats
path: root/Demo/pdist/security.py
blob: a11294a91c0cec7181075ec57da2b87b616d9e91 (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']
		else:
			keyfile = '.python_keyfile'
			if env.has_key('HOME'):
				keyfile = os.path.join(env['HOME'], 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(long(challenge), p, m)