Safihre
5 years ago
7 changed files with 64 additions and 397 deletions
@ -1,154 +0,0 @@ |
|||
#!/usr/bin/python3 -OO |
|||
# Copyright 2007-2020 The SABnzbd-Team <team@sabnzbd.org> |
|||
# |
|||
# This program is free software; you can redistribute it and/or |
|||
# modify it under the terms of the GNU General Public License |
|||
# as published by the Free Software Foundation; either version 2 |
|||
# of the License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU General Public License |
|||
# along with this program; if not, write to the Free Software |
|||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
|||
|
|||
import sys |
|||
if sys.version_info[:2] < (2, 6) or sys.version_info[:2] >= (3, 0): |
|||
print("Sorry, requires Python 2.6 or 2.7.") |
|||
sys.exit(1) |
|||
|
|||
import time |
|||
import subprocess |
|||
|
|||
|
|||
try: |
|||
import win32api |
|||
import win32file |
|||
import win32serviceutil |
|||
import win32evtlogutil |
|||
import win32event |
|||
import win32service |
|||
import pywintypes |
|||
except ImportError: |
|||
print("Sorry, requires Python module PyWin32.") |
|||
sys.exit(1) |
|||
|
|||
from util.mailslot import MailSlot |
|||
from util.apireg import del_connection_info, set_connection_info |
|||
|
|||
|
|||
WIN_SERVICE = None |
|||
|
|||
|
|||
def HandleCommandLine(allow_service=True): |
|||
""" Handle command line for a Windows Service |
|||
Prescribed name that will be called by Py2Exe. |
|||
You MUST set 'cmdline_style':'custom' in the package.py! |
|||
""" |
|||
win32serviceutil.HandleCommandLine(SABHelper) |
|||
|
|||
|
|||
def start_sab(): |
|||
return subprocess.Popen('net start SABnzbd', stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True) |
|||
|
|||
|
|||
def main(): |
|||
|
|||
mail = MailSlot() |
|||
if not mail.create(10): |
|||
return '- Cannot create Mailslot' |
|||
|
|||
active = False # SABnzbd should be running |
|||
counter = 0 # Time allowed for SABnzbd to be silent |
|||
while True: |
|||
msg = mail.receive() |
|||
if msg: |
|||
if msg == 'restart': |
|||
time.sleep(1.0) |
|||
counter = 0 |
|||
del_connection_info(user=False) |
|||
start_sab() |
|||
elif msg == 'stop': |
|||
active = False |
|||
del_connection_info(user=False) |
|||
elif msg == 'active': |
|||
active = True |
|||
counter = 0 |
|||
elif msg.startswith('api '): |
|||
active = True |
|||
counter = 0 |
|||
_cmd, url = msg.split() |
|||
if url: |
|||
set_connection_info(url.strip(), user=False) |
|||
|
|||
if active: |
|||
counter += 1 |
|||
if counter > 120: # 120 seconds |
|||
counter = 0 |
|||
start_sab() |
|||
|
|||
rc = win32event.WaitForMultipleObjects((WIN_SERVICE.hWaitStop, |
|||
WIN_SERVICE.overlapped.hEvent), 0, 1000) |
|||
if rc == win32event.WAIT_OBJECT_0: |
|||
del_connection_info(user=False) |
|||
mail.disconnect() |
|||
return '' |
|||
|
|||
|
|||
############################################################################## |
|||
# Windows Service Support |
|||
############################################################################## |
|||
import servicemanager |
|||
|
|||
|
|||
class SABHelper(win32serviceutil.ServiceFramework): |
|||
""" Win32 Service Handler """ |
|||
|
|||
_svc_name_ = 'SABHelper' |
|||
_svc_display_name_ = 'SABnzbd Helper' |
|||
_svc_deps_ = ["EventLog", "Tcpip"] |
|||
_svc_description_ = 'Automated downloading from Usenet. ' \ |
|||
'This service helps SABnzbd to restart itself.' |
|||
|
|||
def __init__(self, args): |
|||
global WIN_SERVICE |
|||
win32serviceutil.ServiceFramework.__init__(self, args) |
|||
|
|||
self.hWaitStop = win32event.CreateEvent(None, 0, 0, None) |
|||
self.overlapped = pywintypes.OVERLAPPED() # @UndefinedVariable |
|||
self.overlapped.hEvent = win32event.CreateEvent(None, 0, 0, None) |
|||
WIN_SERVICE = self |
|||
|
|||
def SvcDoRun(self): |
|||
msg = 'SABHelper-service' |
|||
self.Logger(servicemanager.PYS_SERVICE_STARTED, msg + ' has started') |
|||
res = main() |
|||
self.Logger(servicemanager.PYS_SERVICE_STOPPED, msg + ' has stopped' + res) |
|||
|
|||
def SvcStop(self): |
|||
self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING) |
|||
win32event.SetEvent(self.hWaitStop) |
|||
|
|||
def Logger(self, state, msg): |
|||
win32evtlogutil.ReportEvent(self._svc_display_name_, |
|||
state, 0, |
|||
servicemanager.EVENTLOG_INFORMATION_TYPE, |
|||
(self._svc_name_, str(msg))) |
|||
|
|||
def ErrLogger(self, msg, text): |
|||
win32evtlogutil.ReportEvent(self._svc_display_name_, |
|||
servicemanager.PYS_SERVICE_STOPPED, 0, |
|||
servicemanager.EVENTLOG_ERROR_TYPE, |
|||
(self._svc_name_, str(msg)), |
|||
str(text)) |
|||
|
|||
|
|||
############################################################################## |
|||
# Platform specific startup code |
|||
############################################################################## |
|||
if __name__ == '__main__': |
|||
|
|||
win32serviceutil.HandleCommandLine(SABHelper, argv=sys.argv) |
|||
@ -1,125 +0,0 @@ |
|||
#!/usr/bin/python3 -OO |
|||
# Copyright 2007-2020 The SABnzbd-Team <team@sabnzbd.org> |
|||
# |
|||
# This program is free software; you can redistribute it and/or |
|||
# modify it under the terms of the GNU General Public License |
|||
# as published by the Free Software Foundation; either version 2 |
|||
# of the License, or (at your option) any later version. |
|||
# |
|||
# This program is distributed in the hope that it will be useful, |
|||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
# GNU General Public License for more details. |
|||
# |
|||
# You should have received a copy of the GNU General Public License |
|||
# along with this program; if not, write to the Free Software |
|||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
|||
|
|||
""" |
|||
sabnzbd.mailslot - Mailslot communication |
|||
""" |
|||
|
|||
import sys |
|||
from time import sleep |
|||
from win32file import GENERIC_WRITE, FILE_SHARE_READ, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL |
|||
from ctypes import c_uint, c_buffer, byref, sizeof, windll |
|||
|
|||
# Win32API Shortcuts |
|||
CreateFile = windll.kernel32.CreateFileW |
|||
ReadFile = windll.kernel32.ReadFile |
|||
WriteFile = windll.kernel32.WriteFile |
|||
CloseHandle = windll.kernel32.CloseHandle |
|||
CreateMailslot = windll.kernel32.CreateMailslotW |
|||
|
|||
|
|||
class MailSlot: |
|||
""" Simple Windows Mailslot communication """ |
|||
|
|||
slotname = r"mailslot\SABnzbd\ServiceSlot" |
|||
|
|||
def __init__(self): |
|||
self.handle = -1 |
|||
|
|||
def create(self, timeout): |
|||
""" Create the Mailslot, after this only receiving is possible |
|||
timeout is the read timeout used for receive calls. |
|||
""" |
|||
slot = r"\\.\%s" % MailSlot.slotname |
|||
self.handle = CreateMailslot(slot, 0, timeout, None) |
|||
|
|||
return self.handle != -1 |
|||
|
|||
def connect(self): |
|||
""" Connect to existing Mailslot so that writing is possible """ |
|||
slot = r"\\.\%s" % MailSlot.slotname |
|||
self.handle = CreateFile(slot, GENERIC_WRITE, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0) |
|||
return self.handle != -1 |
|||
|
|||
def disconnect(self): |
|||
""" Disconnect from Mailslot """ |
|||
if self.handle != -1: |
|||
CloseHandle(self.handle) |
|||
self.handle = -1 |
|||
return True |
|||
|
|||
def send(self, command): |
|||
""" Send one message in bytes to Mailslot """ |
|||
if self.handle == -1: |
|||
return False |
|||
w = c_uint() |
|||
command_bytes = command.encode("utf-8") |
|||
return bool(WriteFile(self.handle, command_bytes, len(command_bytes), byref(w), 0)) |
|||
|
|||
def receive(self): |
|||
""" Receive one message from Mailslot, convert back to unicode """ |
|||
r = c_uint() |
|||
buf = c_buffer(1024) |
|||
if ReadFile(self.handle, buf, sizeof(buf), byref(r), 0): |
|||
return buf.value.decode("utf-8") |
|||
else: |
|||
return None |
|||
|
|||
|
|||
############################################################################## |
|||
# Simple test |
|||
# |
|||
# First start "mailslot.py server" in one process, |
|||
# Then start "mailslot.py client" in another. |
|||
# Five "restart" and one "stop" will be send from client to server. |
|||
# The server will stop after receiving "stop" |
|||
############################################################################## |
|||
|
|||
if __name__ == "__main__": |
|||
if not __debug__: |
|||
print("Run this test in non-optimized mode") |
|||
exit(1) |
|||
|
|||
if len(sys.argv) > 1 and "server" in sys.argv[1]: |
|||
|
|||
recv = MailSlot() |
|||
ret = recv.create(2) |
|||
assert ret, "Failed to create" |
|||
while True: |
|||
data = recv.receive() |
|||
if data is not None: |
|||
print(data) |
|||
if data.startswith("stop"): |
|||
break |
|||
sleep(1.0) |
|||
recv.disconnect() |
|||
|
|||
elif len(sys.argv) > 1 and "client" in sys.argv[1]: |
|||
|
|||
send = MailSlot() |
|||
ret = send.connect() |
|||
assert ret, "Failed to connect" |
|||
for n in range(5): |
|||
ret = send.send("restart") |
|||
assert ret, "Failed to send" |
|||
sleep(1.0) |
|||
send.send("stop") |
|||
assert ret, "Failed to send" |
|||
send.disconnect() |
|||
|
|||
else: |
|||
print("Usage: mailslot.py server|client") |
|||
Loading…
Reference in new issue