onnxruntime/winml/lib/Common/inc/WinML_Lock.h

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

#pragma once

//
//  Simple CRITICAL_SECTION based locks
//
class CWinMLLock {
 private:
  // make copy constructor and assignment operator inaccessible

  CWinMLLock(const CWinMLLock& critical_section);
  CWinMLLock& operator=(const CWinMLLock& critical_section);

  CRITICAL_SECTION critical_section_;

 public:
  CWinMLLock() {
    InitializeCriticalSection(&critical_section_);
  };

  ~CWinMLLock() {
    DeleteCriticalSection(&critical_section_);
  };

  void Lock() {
    EnterCriticalSection(&critical_section_);
  };
  void Unlock() {
    LeaveCriticalSection(&critical_section_);
  };
  void LockExclusive() {
    EnterCriticalSection(&critical_section_);
  };
  void UnlockExclusive() {
    LeaveCriticalSection(&critical_section_);
  };
  BOOL IsLockHeldByCurrentThread() {
    return GetCurrentThreadId() == static_cast<DWORD>(reinterpret_cast<ULONG_PTR>(critical_section_.OwningThread));
  };
  BOOL IsLockHeld() {
    return critical_section_.OwningThread != 0;
  };
  BOOL TryLock() {
    return TryEnterCriticalSection(&critical_section_);
  };
  // aliased methods to help code compat so that CriticalSections can be passed to ReaderWriter templates
  void LockShared() {
    EnterCriticalSection(&critical_section_);
  };
  void UnlockShared() {
    LeaveCriticalSection(&critical_section_);
  };
};

// locks a critical section, and unlocks it automatically
// when the lock goes out of scope
class CWinMLAutoLock {
  // make copy constructor and assignment operator inaccessible

  CWinMLAutoLock(const CWinMLAutoLock& auto_lock);
  CWinMLAutoLock& operator=(const CWinMLAutoLock& auto_lock);

 protected:
  CWinMLLock* winml_lock_;

 public:
  CWinMLAutoLock(CWinMLLock* lock) {
    winml_lock_ = lock;
    if (winml_lock_ != nullptr) {
      winml_lock_->Lock();
    }
  };

  ~CWinMLAutoLock() {
    if (winml_lock_ != nullptr) {
      winml_lock_->Unlock();
    }
  };
};