issue #106: allow any context to contact FileService.

Also fix privilege check for register command.
This commit is contained in:
David Wilson 2018-04-01 11:31:33 +01:00
parent 8fffb34752
commit 6aac37e157
1 changed files with 4 additions and 1 deletions

View File

@ -112,6 +112,9 @@ class FileService(mitogen.service.Service):
"""
handle = 501
max_message_size = 1000
policies = (
mitogen.service.AllowAny(),
)
unprivileged_msg = 'Cannot register from unprivileged context.'
unregistered_msg = 'Path is not registered with FileService.'
@ -133,7 +136,7 @@ class FileService(mitogen.service.Service):
return getattr(self, cmd)(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)
with open(path, 'rb') as fp: