frida 0.1.0 → 0.1.1

data/ext/c_frida/Process.c

@@ -0,0 +1,67 @@
+#include "Process.h"
+
+VALUE	Process_from_FridaProcess(FridaProcess *handle)
+{
+    VALUE self;
+
+    if (!handle)
+        return (Qnil);
+    self = rb_class_new_instance(0, NULL, cProcess);
+    GET_GOBJECT_DATA();
+    d->handle = handle;
+    d->destroy = g_object_unref;
+    rb_ivar_set(self, rb_intern("pid"), LL2NUM(frida_process_get_pid(d->handle)));
+    rb_ivar_set(self, rb_intern("name"), rb_str_new_cstr(frida_process_get_name(d->handle)));
+    rb_ivar_set(self, rb_intern("parameters"), rbProcess_marshal_parameters_dict(frida_process_get_parameters(d->handle)));
+    return (self);
+}
+
+static VALUE Process_inspect(VALUE self)
+{
+    VALUE s;
+
+    s = rb_sprintf("#<Process: name=%+"PRIsVALUE", pid=%+"PRIsVALUE", parameters=%+"PRIsVALUE">", \
+                   rb_funcall(self, rb_intern("name"), 0, NULL),
+                   rb_funcall(self, rb_intern("pid"), 0, NULL),
+                   rb_funcall(self, rb_intern("parameters"), 0, NULL)
+                  );
+    return (s);
+}
+
+/*
+    call-seq:
+        #pid() -> Fixnum
+*/
+static VALUE Process_pid(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("pid")));
+}
+
+/*
+    call-seq:
+        #name() -> String
+*/
+static VALUE Process_name(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("name")));
+}
+
+/*
+    call-seq:
+        #parameters() -> Hash
+*/
+static VALUE Process_parameters(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("parameters")));
+}
+
+void	define_Process()
+{
+    cProcess = rb_define_class_under(mCFrida, "Process", cGObject);
+
+    rb_define_method(cProcess, "inspect", Process_inspect, 0);
+    rb_define_alias(cProcess, "to_s", "inspect");
+    rb_define_method(cProcess, "pid", Process_pid, 0);
+    rb_define_method(cProcess, "name", Process_name, 0);
+    rb_define_method(cProcess, "parameters", Process_parameters, 0);
+}

data/ext/c_frida/Relay.c

@@ -0,0 +1,121 @@
+#include "Relay.h"
+
+VALUE Relay_from_FridaRelay(FridaRelay *handle)
+{
+    VALUE self;
+
+    if (!handle)
+        return (Qnil);
+    VALUE args[4] = {Qfalse};
+    self = rb_class_new_instance(4, args, cRelay);
+    GET_GOBJECT_DATA();
+    d->handle = handle;
+    d->destroy = g_object_unref;
+    rb_ivar_set(self, rb_intern("address"), rbGObject_marshal_string(frida_relay_get_address(handle)));
+    rb_ivar_set(self, rb_intern("username"), rbGObject_marshal_string(frida_relay_get_username(handle)));
+    rb_ivar_set(self, rb_intern("password"), rbGObject_marshal_string(frida_relay_get_password(handle)));
+    rb_ivar_set(self, rb_intern("kind"), rbGObject_marshal_enum(frida_relay_get_kind(handle), FRIDA_TYPE_RELAY_KIND));
+    return (self);
+}
+
+static VALUE Relay_inspect(VALUE self)
+{
+    VALUE s;
+
+    s = rb_sprintf("#<Relay: address=%+"PRIsVALUE", username=%+"PRIsVALUE", password=%+"PRIsVALUE", kind=%+"PRIsVALUE">", \
+                   rb_funcall(self, rb_intern("address"), 0, NULL),  rb_funcall(self, rb_intern("username"), 0, NULL), \
+                   rb_funcall(self, rb_intern("password"), 0, NULL),  rb_funcall(self, rb_intern("kind"), 0, NULL));
+    return (s);
+}
+
+/*
+    call-seq:
+        #address() -> String
+*/
+static VALUE Relay_address(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("address")));
+}
+
+/*
+    call-seq:
+        #username() -> String
+*/
+static VALUE Relay_username(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("username")));
+}
+
+/*
+    call-seq:
+        #password() -> String
+*/
+static VALUE Relay_password(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("password")));
+}
+
+/*
+    call-seq:
+        #kind() -> String
+*/
+static VALUE Relay_kind(VALUE self)
+{
+    return (rb_ivar_get(self, rb_intern("kind")));
+}
+
+/*
+    call-seq:
+        #new(address, username, password, kind) -> Relay
+*/
+static VALUE Relay_initialize(VALUE self, VALUE address, VALUE username, VALUE password, VALUE kind)
+{
+    GET_GOBJECT_DATA();
+    FridaRelayKind	gkind;
+    FridaRelay		*handle;
+
+    if (address == Qfalse)
+        return (self);
+    if (!RB_TYPE_P(address, T_STRING)) {
+        raise_argerror("address should be a string.");
+        return (Qnil);
+    }
+    if (!RB_TYPE_P(username, T_STRING)) {
+        raise_argerror("username should be a string.");
+        return (Qnil);
+    }
+    if (!RB_TYPE_P(password, T_STRING)) {
+        raise_argerror("password should be a string.");
+        return (Qnil);
+    }
+    if (!RB_TYPE_P(kind, T_STRING)) {
+        raise_argerror("kind should be a string.");
+        return (Qnil);
+    }
+    if (!rbGObject_unmarshal_enum(StringValueCStr(kind), FRIDA_TYPE_RELAY_KIND, &gkind)) {
+        raise_argerror("kind is not valid.");
+        return (Qnil);
+    }
+    handle = frida_relay_new(StringValueCStr(address), StringValueCStr(username), StringValueCStr(password), gkind);
+    d->handle = handle;
+    d->destroy = g_object_unref;
+    rb_ivar_set(self, rb_intern("address"), rbGObject_marshal_string(frida_relay_get_address(handle)));
+    rb_ivar_set(self, rb_intern("username"), rbGObject_marshal_string(frida_relay_get_username(handle)));
+    rb_ivar_set(self, rb_intern("password"), rbGObject_marshal_string(frida_relay_get_password(handle)));
+    rb_ivar_set(self, rb_intern("kind"), rbGObject_marshal_enum(frida_relay_get_kind(handle), FRIDA_TYPE_RELAY_KIND));
+    return (self);
+}
+
+void	define_Relay()
+{
+    cRelay = rb_define_class_under(mCFrida, "Relay", cGObject);
+
+    rb_define_method(cRelay, "inspect", Relay_inspect, 0);
+    rb_define_alias(cRelay, "to_s", "inspect");
+
+    rb_define_method(cRelay, "initialize", Relay_initialize, 4);
+    rb_define_method(cRelay, "address", Relay_address, 0);
+    rb_define_method(cRelay, "username", Relay_username, 0);
+    rb_define_method(cRelay, "password", Relay_password, 0);
+    rb_define_method(cRelay, "kind", Relay_kind, 0);
+}

data/ext/c_frida/Script.c

@@ -0,0 +1,221 @@
+#include "Script.h"
+
+VALUE	Script_from_FridaScript(FridaScript *handle)
+{
+    VALUE self;
+
+    if (!handle)
+        return (Qnil);
+    self = rb_class_new_instance(0, NULL, cScript);
+    GET_GOBJECT_DATA();
+    d->handle = handle;
+    d->destroy = frida_unref;
+    return (self);
+}
+
+/*
+    call-seq:
+        #is_destroyed() -> [TrueClass, FalseClass]
+*/
+static VALUE Script_is_destroyed(VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+    void	*is_destroyed;
+
+    is_destroyed = rb_thread_call_without_gvl((void_fp)frida_script_is_destroyed, d->handle, NULL, NULL);
+    return (is_destroyed ? Qtrue : Qfalse);
+}
+
+GVL_FREE_PROXY_FUNC(load_sync, void *handle)
+{
+    GError *gerr = NULL;
+
+    frida_script_load_sync(handle, NULL, &gerr);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #load() -> nil
+*/
+static VALUE Script_load(VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+
+    CALL_GVL_FREE_WITH_RET(void *dummy, load_sync, d->handle);
+    return (Qnil);
+
+    GERROR_BLOCK
+}
+
+GVL_FREE_PROXY_FUNC(eternalize_sync, void *handle)
+{
+    GError *gerr = NULL;
+
+    frida_script_eternalize_sync(handle, NULL, &gerr);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #eternalize() -> nil
+*/
+static VALUE Script_eternalize(VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+
+    CALL_GVL_FREE_WITH_RET(void *dummy, eternalize_sync, d->handle);
+    return (Qnil);
+
+    GERROR_BLOCK
+}
+
+GVL_FREE_PROXY_FUNC(unload_sync, void *handle)
+{
+    GError *gerr = NULL;
+
+    frida_script_unload_sync(handle, NULL, &gerr);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #unload() -> nil
+*/
+static VALUE Script_unload(VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+
+    CALL_GVL_FREE_WITH_RET(void *dummy, unload_sync, d->handle);
+    return (Qnil);
+
+    GERROR_BLOCK
+}
+
+GVL_FREE_PROXY_FUNC(disable_debugger_sync, void *handle)
+{
+    GError *gerr = NULL;
+
+    frida_script_disable_debugger_sync(handle, NULL, &gerr);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #disable_debugger() -> nil
+*/
+static VALUE Script_disable_debugger(VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+
+    CALL_GVL_FREE_WITH_RET(void *dummy, disable_debugger_sync, d->handle);
+    return (Qnil);
+
+    GERROR_BLOCK
+}
+
+GVL_FREE_PROXY_FUNC(enable_debugger_sync, enable_debugger_proxy_args *args)
+{
+    GError *gerr = NULL;
+
+    frida_script_enable_debugger_sync(args->handle, args->port, NULL, &gerr);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #enable_debugger(port:) -> nil
+*/
+static VALUE Script_enable_debugger(int argc, VALUE *argv, VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+    VALUE kws, rport;
+    uint port = 0;
+
+    rb_scan_args(argc, argv, ":", &kws);
+    if (!NIL_P(kws)) {
+        rport = rb_hash_aref(kws, ID2SYM(rb_intern("port")));
+        if (!NIL_P(rport)) {
+            if (!RB_TYPE_P(rport, T_FIXNUM)) {
+                raise_argerror("port must be a number.");
+                return (Qnil);
+            }
+            port = NUM2UINT(rport);
+        }
+    }
+    enable_debugger_proxy_args args = {
+        .handle = d->handle,
+        .port = port
+    };
+    CALL_GVL_FREE_WITH_RET(void *dummy, enable_debugger_sync, &args);
+    goto done;
+
+gerror:
+    raise_rerror(NULL, _gerr);
+done:
+    return (Qnil);
+}
+
+GVL_FREE_PROXY_FUNC(post, post_proxy_args *args)
+{
+    GError *gerr = NULL;
+
+    frida_script_post(args->handle, args->message, args->data);
+    RETURN_GVL_FREE_RESULT(NULL);
+}
+
+/*
+    call-seq:
+        #post(message, data:) -> nil
+*/
+static VALUE Script_post(int argc, VALUE *argv, VALUE self)
+{
+    GET_GOBJECT_DATA();
+    REQUIRE_GOBJECT_HANDLE();
+    VALUE kws, msg, data;
+    gpointer cdata = NULL;
+
+    rb_scan_args(argc, argv, "1:", &msg, &kws);
+    if (!NIL_P(kws)) {
+        data = rb_hash_aref(kws, ID2SYM(rb_intern("data")));
+        if (!NIL_P(data)) {
+            if (!RB_TYPE_P(data, T_STRING)) {
+                raise_argerror("data must be a number.");
+                return (Qnil);
+            }
+            cdata = g_bytes_new(RSTRING_PTR(data), RSTRING_LEN(data));
+        }
+    }
+    post_proxy_args args = {
+        .handle = d->handle,
+        .message = StringValueCStr(msg),
+        .data = cdata
+    };
+    CALL_GVL_FREE_WITH_RET(void *dummy, post, &args);
+    goto done;
+
+gerror:
+    raise_rerror(NULL, _gerr);
+done:
+    g_bytes_unref(args.data);
+    return (Qnil);
+}
+
+void	define_Script()
+{
+    cScript = rb_define_class_under(mCFrida, "Script", cGObject);
+
+    rb_define_method(cScript, "is_destroyed", Script_is_destroyed, 0);
+    rb_define_method(cScript, "load", Script_load, 0);
+    rb_define_method(cScript, "unload", Script_unload, 0);
+    rb_define_method(cScript, "eternalize", Script_eternalize, 0);
+    rb_define_method(cScript, "enable_debugger", Script_enable_debugger, -1);
+    rb_define_method(cScript, "disable_debugger", Script_disable_debugger, 0);
+    rb_define_method(cScript, "post", Script_post, -1);
+}