Rooba
/
mitogen
mirror of https://github.com/mitogen-hq/mitogen.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ansible_mitogen/services.py
View File

@ -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: