plan9fox/sys/lib/python/factotum.py

'''factotum for py'''

import subprocess

class FactotumError(Exception):
	pass

class PhaseError(Exception):
	pass

class NeedkeyError(Exception):
	pass

class Factotum:
	def start(self, **args):
		self.f = open('/mnt/factotum/rpc', 'r+', 0)
		msg = 'start'
		for k, v in args.iteritems():
			msg += ' ' + k + '=\'' + v + '\''
		while True:
			self.f.write(msg)
			ret = self.f.read(4096)
			if ret[:7] != "needkey": break
			self.needkey(ret[8:])
		if ret == "ok": return
		if ret[:5] == "error": raise FactotumError(ret[6:])
		raise FactotumError("start: unexpected " + ret)
	def needkey(self, string):
		subprocess.call(['/bin/auth/factotum', '-g', string])
	def read(self):
		while True:
			self.f.write('read')
			ret = self.f.read(4096)
			if ret[:7] != "needkey": break
			self.needkey(ret[8:])
		if ret == "ok": return ""
		if ret[:3] == "ok ": return ret[3:]
		if ret[:5] == "error": raise FactotumError(ret[6:])
		if ret[:5] == "phase": raise PhaseError(ret[6:])
		raise FactotumError("read: unexpected " + ret)
	def write(self, data):
		while True:
			self.f.write('write ' + data)
			ret = self.f.read(4096)
			if ret[:7] != "needkey": break
			self.needkey(ret[8:])
		if ret == "ok": return 0
		if ret[:3] == "toosmall ": return int(ret[4:])
		if ret[:5] == "error": raise FactotumError(ret[6:])
		if ret[:5] == "phase": raise PhaseError(ret[6:])
		raise FactotumError("write: unexpected " + ret)
	def attr(self):
		self.f.write('attr')
		ret = self.f.read(4096)
		if ret[:5] == "error": raise FactotumError(ret[6:])
		if ret[:3] == "ok ":
			dict = {}
			ret = ret[3:]
			mode = 0
			key = ""
			value = ""
			while ret != "":
				if mode == 0:
					if ret[0] == '=':
						if ret[1] == '\'':
							mode = 2
							ret = ret[1:]
						else:
							mode = 1
					else:
						key += ret[0]
				elif mode == 1:
					if ret[0] == ' ':
						dict[key] = value
						key = ""
						value = ""
						mode = 0
					else:
						value += ret[0]
				elif mode == 2:
					if ret[0] == '\'':
						ret = ret[1:]
						dict[key] = value
						key = ""
						value = ""
						mode = 0
					else:
						value += ret[0]
				if ret != "": ret = ret[1:]
			if key != "":
				dict[key] = value
			return dict
		raise FactotumError("unexpected " + ret)
	def close(self):
		self.f.close()
	def delkey(self, **args):
		f = open('/mnt/factotum/ctl', 'w', 0)
		msg = 'delkey'
		for k, v in args.iteritems():
			msg += ' ' + k + '=\'' + v + '\''
		f.write(msg)
		f.close()