From 11de04f56dc33f987f2a6bd3dc76ccd2398dcc7c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ole=20Andr=C3=A9=20Vadla=20Ravn=C3=A5s?=
 <ole.andre.ravnas@tillitech.com>
Date: Sun, 29 Dec 2013 19:31:52 +0100
Subject: [PATCH] Move tracer console app and expose as console script also

---
 src/frida-trace.in  | 16 ++--------------
 src/frida/tracer.py | 16 ++++++++++++++++
 src/setup.py        |  5 +++++
 3 files changed, 23 insertions(+), 14 deletions(-)

diff --git a/src/frida-trace.in b/src/frida-trace.in
index 9538548..e2b67b0 100644
--- a/src/frida-trace.in
+++ b/src/frida-trace.in
@@ -4,19 +4,7 @@ import sys
 sys.path.insert(1, '@pythondir@')
 
 import frida
-from frida.tracer import TracerProfileBuilder, Tracer
+import frida.tracer
 
 if __name__ == '__main__':
-    if len(sys.argv) != 2:
-        print "Usage: %s <process name>" % sys.argv[0]
-        sys.exit(1)
-    tp = TracerProfileBuilder().include_modules("libSystem*").exclude("malloc", "calloc", "realloc", "free").build()
-    t = Tracer(tp)
-    try:
-        target = int(sys.argv[1])
-    except:
-        target = sys.argv[1]
-    p = frida.attach(target)
-    t.start_trace(p, tracer.STDOUT_SINK)
-    sys.stdin.read()
-    sys.exit(0)
+    frida.tracer.main()
diff --git a/src/frida/tracer.py b/src/frida/tracer.py
index c7edac6..d992fd3 100644
--- a/src/frida/tracer.py
+++ b/src/frida/tracer.py
@@ -147,3 +147,19 @@ class IOSink(object):
 
 STDOUT_SINK = IOSink(sys.stdout)
 STDERR_SINK = IOSink(sys.stderr)
+
+
+def main():
+    if len(sys.argv) != 2:
+        print "Usage: %s <process name>" % sys.argv[0]
+        sys.exit(1)
+    tp = TracerProfileBuilder().include_modules("libSystem*").exclude("malloc", "calloc", "realloc", "free").build()
+    t = Tracer(tp)
+    try:
+        target = int(sys.argv[1])
+    except:
+        target = sys.argv[1]
+    p = frida.attach(target)
+    t.start_trace(p, tracer.STDOUT_SINK)
+    sys.stdin.read()
+    sys.exit(0)
diff --git a/src/setup.py b/src/setup.py
index f97e4e8..cd0b614 100755
--- a/src/setup.py
+++ b/src/setup.py
@@ -30,6 +30,11 @@ setup(
     url="http://frida.github.io",
     author="Frida Developers",
     author_email="ole.andre.ravnas@tillitech.com",
+    entry_points={
+        'console_scripts': [
+            'frida-trace = frida.tracer:main'
+        ]
+    },
     ext_modules=[Extension('_frida', [])],
     cmdclass={
         'build_ext': FridaPrebuiltExt