spinlock: Add try_acquire()

frida · May 10, 2024 · f147df7 · f147df7
1 parent ee86e66
commit f147df7
Show file tree

Hide file tree

Showing 7 changed files with 86 additions and 0 deletions.
diff --git a/gum/backend-arm/gumspinlock-arm.c b/gum/backend-arm/gumspinlock-arm.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2010-2021 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -35,6 +36,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
 #endif
 }
 
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
+
 void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/backend-arm64/gumspinlock-arm64.c b/gum/backend-arm64/gumspinlock-arm64.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2014-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -30,6 +31,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
     ;
 }
 
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
+
 void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/backend-mips/gumspinlock-mips.c b/gum/backend-mips/gumspinlock-mips.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2014-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -30,6 +31,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
     ;
 }
 
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
+
 void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/backend-x86/gumspinlock-amd64-msc.c b/gum/backend-x86/gumspinlock-amd64-msc.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2010-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -30,6 +31,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
     ;
 }
 
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
+
 void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/backend-x86/gumspinlock-ia32-msc.c b/gum/backend-x86/gumspinlock-ia32-msc.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2010-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -35,6 +36,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
   }
 }
 
+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;
+}
+
 __declspec (naked) void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/backend-x86/gumspinlock-x86.c b/gum/backend-x86/gumspinlock-x86.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2010-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -30,6 +31,19 @@ gum_spinlock_acquire (GumSpinlock * spinlock)
     ;
 }
 
+gboolean
+gum_spinlock_try_acquire (GumSpinlock * spinlock)
+{
+  GumSpinlockImpl * self = (GumSpinlockImpl *) spinlock;
+
+  if (self->is_held)
+    return FALSE;
+
+  gum_spinlock_acquire (spinlock);
+
+  return TRUE;
+}
+
 void
 gum_spinlock_release (GumSpinlock * spinlock)
 {

diff --git a/gum/gumspinlock.h b/gum/gumspinlock.h
@@ -1,5 +1,6 @@
 /*
  * Copyright (C) 2010-2019 Ole André Vadla Ravnås <[email protected]>
+ * Copyright (C) 2024 Francesco Tamagni <[email protected]>
  *
  * Licence: wxWindows Library Licence, Version 3.1
  */
@@ -23,6 +24,7 @@ struct _GumSpinlock
 void gum_spinlock_init (GumSpinlock * spinlock);
 
 void gum_spinlock_acquire (GumSpinlock * spinlock);
+gboolean gum_spinlock_try_acquire (GumSpinlock * spinlock);
 void gum_spinlock_release (GumSpinlock * spinlock);
 
 G_END_DECLS