summaryrefslogtreecommitdiffstats
path: root/Demo/pdist/security.py
blob: 0ffd511a079d5a0895d8207d2e319da1b621b16f (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
27
28
29
30
31
32
33
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)
			if not os.path.exists(keyfile):
				import sys
				for dir in sys.path:
					kf = os.path.join(dir, keyfile)
					if os.path.exists(kf):
						keyfile = kf
						break
		try:
			self._key = eval(open(keyfile).readline())
		except IOError:
			raise IOError, "python keyfile %s: cannot open" % keyfile

	def _generate_challenge(self):
		import random
		return random.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)