/*
 Code by JaeHyuk Cho <mailto:minzkn@infoeq.com> Made in KOREA
 http://minzkn.pe.ky
*/

#ifndef DEF_SOURCE_lock_c
#define DEF_SOURCE_lock_c "lock.c"

#include <stdio.h>
#include <sched.h>

#define DEF_STDCALL __attribute__((stdcall))

#define __DEF_FCTYPE__ DEF_STDCALL     /* Pascal 호출규칙사용으로 좀 빠를려나? */

#ifdef WIN32
#define MZ_ASM __asm
#else
#define MZ_ASM __asm__ volatile
#endif

static int __DEF_FCTYPE__ __MZ_Lock__(int * volatile s_LockFlag, int s_Switch);
static int __DEF_FCTYPE__ __MZ_AtomicExchange__(int * volatile s_To, int s_Value);

int MZ_InitLock(int * volatile s_Key);
void MZ_Lock(int * volatile s_Key);
void MZ_UnLock(int * volatile s_Key);

static int __DEF_FCTYPE__ __MZ_Lock__(int * volatile s_LockFlag, int s_Switch)
{
 int s_Return;
 int s_LockTime = 0, s_TimeOut;
 s_TimeOut = (s_Switch >> 2) * 100;
L_SpinLoop:;
 s_Return = __MZ_AtomicExchange__(s_LockFlag, s_Switch);
 if(s_Switch != 0 && s_Return != 0)
 {
  if(s_TimeOut != 0 && s_LockTime > s_TimeOut)
  {
   fprintf(stderr, "%s: %s - [ERROR] Dead-lock instead of off (%d second timeout) !!!\n", __FILE__, __FUNCTION__, s_TimeOut *
100);
   s_Return = __MZ_AtomicExchange__(s_LockFlag, 0);
  }
  else
  {
   sched_yield();
   s_LockTime++;
   goto L_SpinLoop;
  }
 }
 return(s_Return);
}

static int __DEF_FCTYPE__ __MZ_AtomicExchange__(int * volatile s_To, int s_Value)
{
 register int s_Return;
#ifdef WIN32
 MZ_ASM
 {
  PUSH EBX;
  MOV EAX, s_Value;
  MOV EBX, s_To;
  LOCK XCHG DWORD PTR [EBX], EAX;
  POP EBX;
 }
#else
 MZ_ASM(
  "\n\t"
  "lock xchgl %1, (%2)\n\t"
  "\n\t"
  : "=a"(s_Return)
  : "a"(s_Value), "r"(s_To)
 );
#endif
 return(s_Return);
}

int MZ_InitLock(int * volatile s_Key)
{
 if(s_Key)MZ_UnLock(s_Key);
 return(0);
}

void MZ_Lock(int * volatile s_Key)
{
 (void)__MZ_Lock__(s_Key, 1);
}

void MZ_UnLock(int * volatile s_Key)
{
 (void)__MZ_Lock__(s_Key, 0);
}
#endif

/* End of source */