From 65cbaaabe46e5d2e61385e26da1f06ca7039c2c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ole=20Andr=C3=A9=20Vadla=20Ravn=C3=A5s?=
 <oleavr@nowsecure.com>
Date: Fri, 25 Sep 2015 19:30:01 +0200
Subject: [PATCH] Make it possible to set a custom log handler

---
 src/frida/core.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/frida/core.py b/src/frida/core.py
index ebc5872..645b53e 100644
--- a/src/frida/core.py
+++ b/src/frida/core.py
@@ -423,6 +423,7 @@ class Script(object):
 
         self._impl = impl
         self._on_message_callbacks = []
+        self._log_handler = self._on_log
 
         self._pending = {}
         self._next_request_id = 1
@@ -454,6 +455,12 @@ class Script(object):
         else:
             self._impl.off(signal, callback)
 
+    def set_log_handler(self, handler):
+        if handler is not None:
+            self._log_handler = handler
+        else:
+            self._log_handler = self._on_log
+
     def _rpc_request(self, *args):
         result = [False, None, None]
         def on_complete(value, error):
@@ -495,7 +502,9 @@ class Script(object):
         mtype = message['type']
         payload = message.get('payload', None)
         if mtype == 'log':
-            print(payload)
+            level = message.get('level', 'info')
+            text = payload
+            self._log_handler(level, text)
         elif mtype == 'send' and isinstance(payload, list) and payload[0] == 'frida:rpc':
             request_id = payload[1]
             operation = payload[2]
@@ -508,6 +517,12 @@ class Script(object):
                 except:
                     traceback.print_exc()
 
+    def _on_log(self, level, text):
+        if level == 'info':
+            print(text, file=sys.stdout)
+        else:
+            print(text, file=sys.stderr)
+
 class ScriptExports(object):
     def __init__(self, script):
         self._script = script