forked from p15670423/monkey
88 lines
2.2 KiB
Python
88 lines
2.2 KiB
Python
|
|
||
|
|
import sys
|
||
|
|
import ctypes
|
||
|
|
import logging
|
||
|
|
import winerror
|
||
|
|
from abc import ABCMeta, abstractmethod
|
||
|
|
from config import WormConfiguration
|
||
|
|
|
||
|
|
__author__ = 'itamar'
|
||
|
|
|
||
|
|
LOG = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class _SystemSingleton(object):
|
||
|
|
__metaclass__ = ABCMeta
|
||
|
|
|
||
|
|
@property
|
||
|
|
@abstractmethod
|
||
|
|
def locked(self):
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def try_lock(self):
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def unlock(self):
|
||
|
|
raise NotImplementedError()
|
||
|
|
|
||
|
|
|
||
|
|
class WindowsSystemSingleton(_SystemSingleton):
|
||
|
|
def __init__(self):
|
||
|
|
self._mutex_name = r"Global\%s" % (WormConfiguration.singleton_mutex_name, )
|
||
|
|
self._mutex_handle = None
|
||
|
|
|
||
|
|
@property
|
||
|
|
def locked(self):
|
||
|
|
return self._mutex_handle is not None
|
||
|
|
|
||
|
|
def try_lock(self):
|
||
|
|
assert self._mutex_handle is None, "Singleton already locked"
|
||
|
|
|
||
|
|
handle = ctypes.windll.kernel32.CreateMutexA(None,
|
||
|
|
ctypes.c_bool(True),
|
||
|
|
ctypes.c_char_p(self._mutex_name))
|
||
|
|
last_error = ctypes.windll.kernel32.GetLastError()
|
||
|
|
if not handle:
|
||
|
|
LOG.error("Cannot acquire system singleton %r, unknown error %d",
|
||
|
|
self._mutex_name, last_error)
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
if winerror.ERROR_ALREADY_EXISTS == last_error:
|
||
|
|
LOG.debug("Cannot acquire system singleton %r, mutex already exist",
|
||
|
|
self._mutex_name)
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
self._mutex_handle = handle
|
||
|
|
|
||
|
|
LOG.debug("Global singleton mutex %r acquired",
|
||
|
|
self._mutex_name)
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
def unlock(self):
|
||
|
|
assert self._mutex_handle is not None, "Singleton not locked"
|
||
|
|
|
||
|
|
ctypes.windll.kernel32.CloseHandle(self._mutex_handle)
|
||
|
|
self._mutex_handle = None
|
||
|
|
|
||
|
|
|
||
|
|
class LinuxSystemSingleton(_SystemSingleton):
|
||
|
|
@property
|
||
|
|
def locked(self):
|
||
|
|
return False
|
||
|
|
|
||
|
|
def try_lock(self):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def unlock(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if sys.platform == "win32":
|
||
|
|
SystemSingleton = WindowsSystemSingleton
|
||
|
|
else:
|
||
|
|
SystemSingleton = LinuxSystemSingleton
|