Add ARM spinlock

This commit is contained in:
Matte23
2024-11-03 18:13:56 +01:00
parent ba22538626
commit d05bbdabc0
4 changed files with 57 additions and 26 deletions

34
os/sync/spinlock.c Normal file
View File

@@ -0,0 +1,34 @@
// sync_primitives.c
#include "spinlock.h"
// Initialize the spinlock
void spinlock_init(Spinlock *spinlock) {
spinlock->lock = 0; // Unlocked state
}
// Acquire the spinlock
void spinlock_acquire(Spinlock *spinlock) {
int status;
do {
// Wait until the lock is available
do {
__asm__ volatile ("ldrex %0, [%1]"
: "=&r" (status)
: "r" (&spinlock->lock)
: "memory");
} while (status != 0);
// Try to acquire the lock using strex
__asm__ volatile ("strex %0, %2, [%1]"
: "=&r" (status)
: "r" (&spinlock->lock), "r" (1)
: "memory");
} while (status != 0); // Retry if strex failed
}
// Release the spinlock
void spinlock_release(Spinlock *spinlock) {
// Use a memory barrier to ensure proper ordering
__asm__ volatile ("dmb" ::: "memory");
spinlock->lock = 0; // Release the lock
}

16
os/sync/spinlock.h Normal file
View File

@@ -0,0 +1,16 @@
#ifndef SPINLOCK_H
#define SPINLOCK_H
#include <stdint.h>
// Spinlock
typedef struct {
volatile int lock;
} Spinlock;
void spinlock_init(Spinlock *spinlock);
void spinlock_acquire(Spinlock *spinlock);
void spinlock_release(Spinlock *spinlock);
#endif