使用已实现的权限规则共享插件资源



我有多个脚本导出相同的接口,它们在隔离范围内使用execfile()执行。

问题是,我希望它们共享一些资源,以便每个新脚本不必从一开始就重新加载它们,从而失去启动速度并使用不必要的RAM。

与下面的例子相比,这些脚本实际上被更好地封装和保护,以防止恶意插件,这就是我的问题所在。

问题是,我希望创建资源的脚本能够用数据填充它,删除数据或删除资源,当然还可以访问它的数据。

但是其他脚本不应该能够改变另一个脚本的资源,只是读取它。我想确保新安装的插件不会因为滥用共享资源而干扰已经加载和运行的插件。

的例子:

class SharedResources:
    # Here should be a shared resource manager that I tried to write
    # but got stuck. That's why I ask this long and convoluted question!
    # Some beginning:
    def __init__ (self, owner):
        self.owner = owner
    def __call__ (self):
        # Here we should return some object that will do
        # required stuff. Read more for details.
        pass
class plugin (dict):
    def __init__ (self, filename):
        dict.__init__(self)
        # Here some checks and filling with secure versions of __builtins__ etc.
        # ...
        self["__name__"] = "__main__"
        self["__file__"] = filename
        # Add a shared resources manager to this plugin
        self["SharedResources"] = SharedResources(filename)
        # And then:
        execfile(filename, self, self)
    # Expose the plug-in interface to outside world:
    def __getattr__ (self, a):
        return self[a]
    def __setattr__ (self, a, v):
        self[a] = v
    def __delattr__ (self, a):
        del self[a]
    # Note: I didn't use self.__dict__ because this makes encapsulation easier.
    # In future I won't use object itself at all but separate dict to do it. For now let it be
----------------------------------------
# An example of two scripts that would use shared resource and be run with plugins["name"] = plugin("<filename>"):
# Presented code is same in both scripts, what comes after will be different.
def loadSomeResource ():
    # Do it here...
    return loadedresource
# Then Load this resource if it's not already loaded in shared resources, if it isn't then add loaded resource to shared resources:
shr = SharedResources() # This would be an instance allowing access to shared resources
if not shr.has_key("Default Resources"):
    shr.create("Default Resources")
if not shr["Default Resources"].has_key("SomeResource"):
    shr["Default Resources"].add("SomeResource", loadSomeResource())
resource = shr["Default Resources"]["SomeResource"]
# And then we use normally resource variable that can be any object.
# Here I Used category "Default Resources" to add and/or retrieve a resource named "SomeResource".
# I want more categories so that plugins that deal with audio aren't mixed with plug-ins that deal with video for instance. But this is not strictly needed.
# Here comes code specific for each plug-in that will use shared resource named "SomeResource" from category "Default Resources".
...
# And end of plugin script!
----------------------------------------
# And then, in main program we load plug-ins:
import os
plugins = {} # Here we store all loaded plugins
for x in os.listdir("plugins"):
    plugins[x] = plugin(x)

假设我们的两个脚本都存储在plugins目录中,并且都使用一些加载到内存中的WAVE文件。首先加载的插件将加载WAVE并将其放入RAM。其他插件将能够访问已经加载的WAVE,但不能替换或删除它,从而干扰其他插件。

现在,我希望每个资源都有一个所有者,一些id或插件脚本的文件名,并且这个资源只能由它的所有者写。

没有任何调整或变通应该使其他插件能够访问第一个。

我几乎做到了,然后卡住了,我的头是旋转的概念,当实现的事情,但只是部分。它在吞噬我,所以我不能再集中注意力了。欢迎提出任何建议!

添加:

这是我现在使用的,没有任何安全措施:

# Dict that will hold a category of resources (should implement some security):
class ResourceCategory (dict):
    def __getattr__ (self, i): return self[i]
    def __setattr__ (self, i, v): self[i] = v
    def __delattr__ (self, i): del self[i]
SharedResources = {} # Resource pool
class ResourceManager:
    def __init__ (self, owner):
        self.owner = owner
    def add (self, category, name, value):
        if not SharedResources.has_key(category):
            SharedResources[category] = ResourceCategory()
        SharedResources[category][name] = value
    def get (self, category, name):
        return SharedResources[category][name]
    def rem (self, category, name=None):
        if name==None: del SharedResources[category]
        else: del SharedResources[category][name]
    def __call__ (self, category):
        if not SharedResources.has_key(category):
            SharedResources[category] = ResourceCategory()
        return SharedResources[category]
    __getattr__ = __getitem__ = __call__
    # When securing, this must not be left as this, it is unsecure, can provide a way back to SharedResources pool:
    has_category = has_key = SharedResources.has_key

现在是插件胶囊:

class plugin(dict):
    def __init__ (self, path, owner):
        dict.__init__()
        self["__name__"] = "__main__"
        # etc. etc.
        # And when adding resource manager to the plugin, register it with this plugin as an owner
        self["SharedResources"] = ResourceManager(owner)
        # ...
        execfile(path, self, self)
        # ...

插件脚本示例:

#-----------------------------------
# Get a category we want. (Using __call__() ) Note: If a category doesn't exist, it is created automatically.
AudioResource = SharedResources("Audio")
# Use an MP3 resource (let say a bytestring):
if not AudioResource.has_key("Beep"):
    f = open("./sounds/beep.mp3", "rb")
    Audio.Beep = f.read()
    f.close()
# Take a reference out for fast access and nicer look:
beep = Audio.Beep # BTW, immutables doesn't propagate as references by themselves, doesn't they? A copy will be returned, so the RAM space usage will increase instead. Immutables shall be wrapped in a composed data type.

这可以很好地工作,但是,正如我所说的,在这里很容易混淆资源。

我想要一个ResourceManager()的实例来负责谁返回存储数据的哪个版本

所以,我的一般做法是这样的。

  1. 有一个中央共享资源池。通过这个池的访问对所有人都是只读的。

  2. 将所有数据封装在共享池中,这样任何"按规则行事"的人都不能编辑其中的任何内容。
  3. 每个代理(插件)在加载它时保持它"拥有"的知识。它为自己保留一个读/写引用,并将对资源的引用注册到集中的只读池。

  4. 当一个插件被加载时,它得到一个对中央只读池的引用,它可以在这个池中注册新的资源。

因此,仅解决python本地数据结构(而不是自定义类的实例)的问题,一个相当锁定的只读实现系统如下。请注意,用于锁定它们的技巧与其他人可以用来绕过锁的技巧是相同的,因此,如果具有一点python知识的人积极地试图打破它,那么沙箱是非常弱的。

import collections as _col
import sys
if sys.version_info >= (3, 0):
    immutable_scalar_types = (bytes, complex, float, int, str)
else:
    immutable_scalar_types = (basestring, complex, float, int, long)
# calling this will circumvent any control an object has on its own attribute lookup
getattribute = object.__getattribute__
# types that will be safe to return without wrapping them in a proxy
immutable_safe = immutable_scalar_types
def add_immutable_safe(cls):
    # decorator for adding a new class to the immutable_safe collection
    # Note: only ImmutableProxyContainer uses it in this initial
    # implementation
    global immutable_safe
    immutable_safe += (cls,)
    return cls
def get_proxied(proxy):
    # circumvent normal object attribute lookup
    return getattribute(proxy, "_proxied")
def set_proxied(proxy, proxied):
    # circumvent normal object attribute setting
    object.__setattr__(proxy, "_proxied", proxied)
def immutable_proxy_for(value):
    # Proxy for known container types, reject all others
    if isinstance(value, _col.Sequence):
        return ImmutableProxySequence(value)
    elif isinstance(value, _col.Mapping):
        return ImmutableProxyMapping(value)
    elif isinstance(value, _col.Set):
        return ImmutableProxySet(value)
    else:
        raise NotImplementedError(
            "Return type {} from an ImmutableProxyContainer not supported".format(
                type(value)))
@add_immutable_safe
class ImmutableProxyContainer(object):
    # the only names that are allowed to be looked up on an instance through
    # normal attribute lookup
    _allowed_getattr_fields = ()
    def __init__(self, proxied):
        set_proxied(self, proxied)
    def __setattr__(self, name, value):
        # never allow attribute setting through normal mechanism
        raise AttributeError(
            "Cannot set attributes on an ImmutableProxyContainer")
    def __getattribute__(self, name):
        # enforce attribute lookup policy
        allowed_fields = getattribute(self, "_allowed_getattr_fields")
        if name in allowed_fields:
            return getattribute(self, name)
        raise AttributeError(
            "Cannot get attribute {} on an ImmutableProxyContainer".format(name))
    def __repr__(self):
        proxied = get_proxied(self)
        return "{}({})".format(type(self).__name__, repr(proxied))
    def __len__(self):
        # works for all currently supported subclasses
        return len(get_proxied(self))
    def __hash__(self):
        # will error out if proxied object is unhashable
        proxied = getattribute(self, "_proxied")
        return hash(proxied)
    def __eq__(self, other):
        proxied = get_proxied(self)
        if isinstance(other, ImmutableProxyContainer):
            other = get_proxied(other)
        return proxied == other

class ImmutableProxySequence(ImmutableProxyContainer, _col.Sequence):
    _allowed_getattr_fields = ("count", "index")
    def __getitem__(self, index):
        proxied = get_proxied(self)
        value = proxied[index]
        if isinstance(value, immutable_safe):
            return value
        return immutable_proxy_for(value)

class ImmutableProxyMapping(ImmutableProxyContainer, _col.Mapping):
    _allowed_getattr_fields = ("get", "keys", "values", "items")
    def __getitem__(self, key):
        proxied = get_proxied(self)
        value = proxied[key]
        if isinstance(value, immutable_safe):
            return value
        return immutable_proxy_for(value)
    def __iter__(self):
        proxied = get_proxied(self)
        for key in proxied:
            if not isinstance(key, immutable_scalar_types):
                # If mutable keys are used, returning them could be dangerous.
                # If owner never puts a mutable key in, then integrity should
                # be okay. tuples and frozensets should be okay as keys, but
                # are not supported in this implementation for simplicity.
                raise NotImplementedError(
                    "keys of type {} not supported in "
                    "ImmutableProxyMapping".format(type(key)))
            yield key

class ImmutableProxySet(ImmutableProxyContainer, _col.Set):
    _allowed_getattr_fields = ("isdisjoint", "_from_iterable")
    def __contains__(self, value):
        return value in get_proxied(self)
    def __iter__(self):
        proxied = get_proxied(self)
        for value in proxied:
            if isinstance(value, immutable_safe):
                yield value
            yield immutable_proxy_for(value)
    @classmethod
    def _from_iterable(cls, it):
        return set(it)

注意:这只在Python 3.4上测试过,但我试着把它写得与Python 2和3兼容。

将共享资源的根目录设置为字典。给这个字典的ImmutableProxyMapping给插件。

private_shared_root = {}
public_shared_root = ImmutableProxyMapping(private_shared_root)

创建一个API,插件可以在其中注册新的资源到public_shared_root,可能是先到先得的基础上(如果它已经存在,你不能注册它)。预填充private_shared_root与任何容器,你知道你将需要,或任何数据,你想与所有插件共享,但你知道你想是只读的。

如果共享根映射中的键的约定都是字符串,比如文件系统路径(/home/dalen/local/python)或点路径,比如python库对象(os.path.expanduser),可能会比较方便。这样,如果插件试图将相同的资源添加到池中,碰撞检测是即时的和微不足道的/明显的。

最新更新