ref: 6fe18e7c7b6dd98c6eb7b01ad7c753d42d60fd0f
dir: /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()