blob: 4971459c077e2d657a1d4565fdccfe9b0026c172 (
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
'''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 + '\''
self.f.write(msg)
ret = self.f.read(4096)
if ret == "ok": return
if ret[:5] == "error": raise FactotumError(ret[6:])
raise FactotumError("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("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("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()
|
