102 lines
2.4 KiB
Python
102 lines
2.4 KiB
Python
'''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()
|