spinlock: Add try_acquire

frida · Apr 11, 2024 · fc995f2 · fc995f2
1 parent ffda5fc
commit fc995f2
Show file tree

Hide file tree

Showing 7 changed files with 74 additions and 0 deletions.
diff --git a/gum/backend-arm/gumspinlock-arm.c b/gum/backend-arm/gumspinlock-arm.c
@@ -46,3 +46,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
   g_atomic_int_set (&self->is_held, FALSE);
 #endif
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/backend-arm64/gumspinlock-arm64.c b/gum/backend-arm64/gumspinlock-arm64.c
@@ -37,3 +37,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
 
   __sync_lock_release (&self->is_held);
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/backend-mips/gumspinlock-mips.c b/gum/backend-mips/gumspinlock-mips.c
@@ -37,3 +37,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
 
   __sync_lock_release (&self->is_held);
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/backend-x86/gumspinlock-amd64-msc.c b/gum/backend-x86/gumspinlock-amd64-msc.c
@@ -37,3 +37,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
 
   g_atomic_int_set (&self->is_held, FALSE);
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/backend-x86/gumspinlock-ia32-msc.c b/gum/backend-x86/gumspinlock-ia32-msc.c
@@ -45,3 +45,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
     ret;
   }
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  volatile guint32 is_held = *(guint32*)spinlock;
+  if (is_held == 1)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/backend-x86/gumspinlock-x86.c b/gum/backend-x86/gumspinlock-x86.c
@@ -37,3 +37,15 @@ gum_spinlock_release (GumSpinlock * spinlock)
 
   __sync_lock_release (&self->is_held);
 }
+
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
diff --git a/gum/gumspinlock.h b/gum/gumspinlock.h
@@ -25,6 +25,8 @@ void gum_spinlock_init (GumSpinlock * spinlock);
 void gum_spinlock_acquire (GumSpinlock * spinlock);
 void gum_spinlock_release (GumSpinlock * spinlock);
 
+gboolean gum_spinlock_try_acquire (GumSpinlock * spinlock);
+
 G_END_DECLS
 
 #endif