Make pwnlib.context work better with threads about pwntools HOT 5 CLOSED

# At the end of 'context.py', add
class ContextPreservingThread(threading.Thread):
    def __init__(self, *args, **kwargs):
        super(ContextPreservingThread, self).__init__(*args, **kwargs)
        self.parent_ctx    = context._thread_ctx()
        self.__old_bootstrap = self._Thread__bootstrap
        self._Thread__bootstrap = self.__new_bootstrap

    def __new_bootstrap(self):
        context._ctxs.setdefault(
            threading.current_thread().ident, self.parent_ctx
        )
        return self.__old_bootstrap()

context.log_level = 'silent'
def bar():                                       
    print context.log_level
bar()
# 100
threading.Thread(target=bar).start()
# 20
ContextPreservingThread(target=bar).start()
# 100

threading.Thread = ContextPreservingThread