/
cmd-server.py
executable file
·131 lines (108 loc) · 3.25 KB
/
cmd-server.py
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
#!/usr/bin/env python2.5
import sys, struct, mmap
import options, git
from helpers import *
def init_dir(conn, arg):
git.init_repo(arg)
log('bup server: bupdir initialized: %r\n' % git.repodir)
conn.ok()
def set_dir(conn, arg):
git.check_repo_or_die(arg)
log('bup server: bupdir is %r\n' % git.repodir)
conn.ok()
def list_indexes(conn, junk):
git.check_repo_or_die()
for f in os.listdir(git.repo('objects/pack')):
if f.endswith('.idx'):
conn.write('%s\n' % f)
conn.ok()
def send_index(conn, name):
git.check_repo_or_die()
assert(name.find('/') < 0)
assert(name.endswith('.idx'))
idx = git.PackIndex(git.repo('objects/pack/%s' % name))
conn.write(struct.pack('!I', len(idx.map)))
conn.write(idx.map)
conn.ok()
def receive_objects(conn, junk):
git.check_repo_or_die()
w = git.PackWriter()
while 1:
ns = conn.read(4)
if not ns:
w.abort()
raise Exception('object read: expected length header, got EOF\n')
n = struct.unpack('!I', ns)[0]
#log('expecting %d bytes\n' % n)
if not n:
log('bup server: received %d object%s.\n'
% (w.count, w.count!=1 and "s" or ''))
w.close()
return
buf = conn.read(n) # object sizes in bup are reasonably small
#log('read %d bytes\n' % n)
if len(buf) < n:
w.abort()
raise Exception('object read: expected %d bytes, got %d\n'
% (n, len(buf)))
w._raw_write(buf)
w.close()
conn.ok()
def read_ref(conn, refname):
git.check_repo_or_die()
r = git.read_ref(refname)
conn.write('%s\n' % (r or '').encode('hex'))
conn.ok()
def update_ref(conn, refname):
git.check_repo_or_die()
newval = conn.readline().strip()
oldval = conn.readline().strip()
git.update_ref(refname, newval.decode('hex'), oldval.decode('hex'))
conn.ok()
def cat(conn, id):
git.check_repo_or_die()
for blob in git.cat(id):
conn.write(struct.pack('!I', len(blob)))
conn.write(blob)
conn.write('\0\0\0\0')
conn.ok()
optspec = """
bup server
"""
o = options.Options('bup server', optspec)
(opt, flags, extra) = o.parse(sys.argv[1:])
if extra:
log('bup server: no arguments expected\n')
o.usage()
log('bup server: reading from stdin.\n')
commands = {
'init-dir': init_dir,
'set-dir': set_dir,
'list-indexes': list_indexes,
'send-index': send_index,
'receive-objects': receive_objects,
'read-ref': read_ref,
'update-ref': update_ref,
'cat': cat,
}
# FIXME: this protocol is totally lame and not at all future-proof.
# (Especially since we abort completely as soon as *anything* bad happens)
conn = Conn(sys.stdin, sys.stdout)
lr = linereader(conn)
for _line in lr:
line = _line.strip()
if not line:
continue
log('bup server: command: %r\n' % line)
words = line.split(' ', 1)
cmd = words[0]
rest = len(words)>1 and words[1] or ''
if cmd == 'quit':
break
else:
cmd = commands.get(cmd)
if cmd:
cmd(conn, rest)
else:
raise Exception('unknown server command: %r\n' % line)
log('bup server: done\n')