issue #106: allow any context to contact FileService.

Also fix privilege check for register command.
pull/193/head
David Wilson 7 years ago
parent 8fffb34752
commit 6aac37e157

@ -112,6 +112,9 @@ class FileService(mitogen.service.Service):
""" """
handle = 501 handle = 501
max_message_size = 1000 max_message_size = 1000
policies = (
mitogen.service.AllowAny(),
)
unprivileged_msg = 'Cannot register from unprivileged context.' unprivileged_msg = 'Cannot register from unprivileged context.'
unregistered_msg = 'Path is not registered with FileService.' unregistered_msg = 'Path is not registered with FileService.'
@ -133,7 +136,7 @@ class FileService(mitogen.service.Service):
return getattr(self, cmd)(path, msg) return getattr(self, cmd)(path, msg)
def register(self, path, msg): def register(self, path, msg):
if msg.auth_id not in mitogen.parent_ids: if not mitogen.core.has_parent_authority(msg):
raise mitogen.core.CallError(self.unprivileged_msg) raise mitogen.core.CallError(self.unprivileged_msg)
with open(path, 'rb') as fp: with open(path, 'rb') as fp:

Loading…
Cancel
Save