Skip to content

dev1-dev2/nosorog

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

98 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Nosorog

Introduction

An assertive security library.

Requirements

3.5 >= Python <= 3.11-dev

Installing

pip install nosorog

Testing

cd /path/to/lib/
python3 -m unittest discover

How to use

Exceptions

Exception Default message
NosorogMangledNameError "Use method`s dunder name instead."
NosorogWrongPlaceCallError (1) "Protected method can be called from specified methods only."
NosorogWrongPlaceCallError (2) "Protected method can not be called from other object, use self instead."
NosorogWentWrongError "Something broken."
NosorogTypeError child of TypeError. No especial message provided.

It is possible to use a concatenation of predefined and custom messages:

raise NosorogMangledNameError("Method __get accessible with _MangledName__get() call.")
# NosorogMangledNameError: "Use method`s dunder name instead. Method __get accessible with _MangledName__get() call."

But it is one exclusion: NosorogWrongPlaceCallError uses the message "Protected method can be called from specified methods only." by default and or other instead:

from nosorog.exceptions.mixins.nosorog_exception_messages import NosorogExceptionMessages

raise NosorogWrongPlaceCallError(NosorogExceptionMessages.use_self)
# NosorogExceptionMessages: "Protected method can not be called from other object, use self instead."

It is not concatenated.

Full list of predefined messages

Attribute Message
protected_from_not_private_call "This method protected from not private call."
method_protected "This method protected."
wrong_place "Protected method can be called from specified places only."
use_self "Protected method can not be called from other object, use self instead."
mangled_call_blocked "Use method`s dunder name instead."

Class based decorators

To import class based decorators use:

from nosorog.decorators import protect_private, copy_dicts, silent
Decorator Description
@silent intercepts all the exceptions of Nosorog and returns None instead.
@silent.include(exceptions) same as above and list of provided exceptions too.
@silent.exclude(exceptions) same as above, but provided exceptions excluded.
--- ---
@protect_private.block_mangled_call protect of name mangling usage.
@protect_private.one_obj decorated method accessible with self usage only.
@protect_private.one_method("method_name") decorated method accessible from one method only.
@protect_private.call_from(methods) decorated method accessible from the methods provided in list only.
--- ---
@copy_dicts makes shallow copy of all the dicts in args and kwargs
@copy_dicts.deep_args makes deep copy of all the dicts in args
@copy_dicts.deep_kwargs makes deep copy of all the dicts in kwargs
@copy_dicts.deep_all makes deep copy of all the dicts in args and kwargs
@copy_dicts.shallow_args makes shallow copy of all the dicts in args
@copy_dicts.shallow_kwargs makes shallow copy of all the dicts in kwargs
@copy_dicts.shallow_all makes shallow copy of all the dicts in args and kwargs

Function based decorators

To import function based decorators use:

from nosorog.decorators.function_based import protect_private, copy_dicts, protect_ids, protected_call
Decorator Description
@protect_private(allowed_list=list) make a call with _Class__private_method() impossible. allowed_list it is str names of method which you can call the private method from. also support 'self' (str) for calls from same object only.
@protected_call(from_method=str, from_file=str) make the attack by the file injection impossible.
@copy_dicts(deep_copy=bool) make a copy of dicts in args and kwargs.
@protect_ids(id_names=[str]) trying to convert id to int or throw Exception.

Examples

This explanation written for the function based decorators. Class based decorators works the same way with some differences in the syntax. Read the full documentation on https://nosorog.readthedocs.io.

Private methods

Usage of dunder methods ( __method() ) protects the code avoiding direct access to the method.

class Example:
    @staticmethod
    def __get_data():
        return 1

>>> Example().__get_data()  # AttributeError: 'Example' object has no attribute '__get_data'

But it is possible to use the name mangling.

>>> Example()._Example__get_data()  # 1

Nosorog provides simple and pushy way to protect the dunder method.

class Example:
    @protect_private(allowed_list=['trusted_func'])
    def __get_data(self):
        return 1

class Trusted:
    @staticmethod
    def trusted_func():
        return Example()._Example__get_data()

>>> Example().__get_data()  # AttributeError: 'Example' object has no attribute '__get_data'
>>> Example()._Example__get_data()  # Exception: This method protected from not private call.
>>> Trusted()._Example__get_data()  # 1

Also, str 'self' can be used as a list item to make impossible to call without self.

class Example:
    @protect_private(allowed_list=['trusted_func', 'self'])
    def __get_data(self):
        return 1

    def trusted_func(self):
        return self.__get_data()


class Trusted:
    @staticmethod
    def trusted_func():
        return Example()._Example__get_data()

>>> Example().trusted_func()  # 1
>>> Trusted().trusted_func()  # Exception: This method can not be called from other object, use self instead.

Localization of method call

Python does not provide an easy way to limit where the method can be called from. This makes it possible to conduct an attack by File Injection. With the help of the Nosorog library it is possible to specify the places from which the method can be called.

class Example:
    @protected_call(from_method='safe_method', from_file=os.path.abspath(__file__))
    def __get_data(self):
        return 1

class Trusted:
    # Place it to the same file as described in the decorator usage.
    @staticmethod
    def safe_method():
        return Example()._Example__get_data()  # 1

This is just a variation of the previous decorator.

Protection of the dicts

In the projects where the undefined number of dicts can be passed in args and kwargs, it is possible to make a deep copy of each if needed.

class Example:
    @copy_dicts(deep_copy=False)
    def some_method(self, *args, **kwargs):
        # now dicts are shallow copies
        pass

Use @copy_dicts(deep_copy=True) to make deep copies.

Protection of ids

This method has been added just for fun. It is converts all the ids in the list if possible or throws the TypeError.

class Example:
    @protect_ids(id_names=['user_id', 'pk'])
    def some_method(self, user_id=None, pk=None)
        pass

Possible Exceptions

@protect_ids(id_names=['user_id', dict()])
>>> Example().some_method(user_id='1')  # TypeError: Wrong format of id_names in decorator. Must be list of str.

@protect_ids(id_names=['user_id', 'pk'])
>>> Example().some_method(user_id=1.234, pk='text_id')  # TypeError: Received the ids of wrong type.