burmatovan/openuds
1
0
Fork 0
forked from shaba/openuds
Code Issues Pull Requests Projects Releases Wiki Activity

Testing & fixing new uds actor

Browse Source
This commit is contained in:
Adolfo Gómez García 2014-10-17 02:36:17 +02:00
parent 15aeba101e
commit bdf0d034fa
8 changed files with 113 additions and 83 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
windowsActor/UDSActorConfig.py
View File

@ -105,7 +105,11 @@ if __name__ == "__main__":
# Read configuration
cfg = store.readConfig()
logger.setLevel(cfg.get('logLevel', 10000))
if cfg is not None:
logger.setLevel(cfg.get('logLevel', 20000))
else:
logger.setLevel(20000)
logger.debug('Configuration: {}'.format(cfg))

139
windowsActor/UDSActorService.py
View File

@ -50,6 +50,19 @@ import socket
cfg = None
def initCfg():
global cfg
cfg = store.readConfig()
if logger.logger.isWindows():
# Logs will also go to windows event log for services
logger.logger.serviceLogger = True
if cfg is not None:
logger.setLevel(cfg.get('logLevel', 20000))
else:
logger.setLevel(20000)
class UDSActorSvc(win32serviceutil.ServiceFramework):
_svc_name_ = "UDSActor"
@ -62,7 +75,7 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
win32serviceutil.ServiceFramework.__init__(self, args)
self.hWaitStop = win32event.CreateEvent(None, 1, 0, None)
self.isAlive = True
socket.setdefaulttimeout(60)
socket.setdefaulttimeout(20)
self.api = None
self.rebootRequested = False
self.knownIps = []
@ -74,6 +87,11 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
SvcShutdown = SvcStop
def notifyStop(self):
servicemanager.LogMsg(servicemanager.EVENTLOG_INFORMATION_TYPE,
servicemanager.PYS_SERVICE_STOPPED,
(self._svc_name_, ''))
def reboot(self):
self.rebootRequested = True
@ -162,10 +180,10 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
'''
# If no configuration is found, stop service
if cfg is None:
logger.debug('No configuration found, stopping service')
return False
self.api = REST.Api(
cfg['host'], cfg['masterKey'], cfg['ssl'], scrambledResponses=True)
self.api = REST.Api(cfg['host'], cfg['masterKey'], cfg['ssl'], scrambledResponses=True)
# Wait for Broker to be ready
counter = 0
@ -174,46 +192,46 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
# getNetworkInfo is a generator function
netInfo = tuple(operations.getNetworkInfo())
self.knownIps = dict(((i.mac, i.ip) for i in netInfo))
ids = ','.join([i.map for i in netInfo])
ids = ','.join([i.mac for i in netInfo])
if ids == '':
# Wait for any network interface to be ready
logger.debug('No network interfaces found, retrying in a while...')
raise Exception()
logger.debug('Ids: {}'.format(ids))
self.api.init(ids)
# Set remote logger to notify log info to broker
logger.setRemoteLogger(self.api)
break
except REST.InvalidKeyError:
logger.fatal(
'Can\'t sync with broker: Invalid broker Master Key')
logger.fatal('Can\'t sync with broker: Invalid broker Master Key')
return False
except REST.UnmanagedHostError:
# Maybe interface that is registered with broker is not enabled already?
# Right now, we thing that the interface connected to broker is
# the interface that broker will know, let's see how this works
logger.fatal(
'This host is not managed by UDS Broker (ids: {})'.format(ids))
logger.fatal('This host is not managed by UDS Broker (ids: {})'.format(ids))
return False
except Exception as e:
logger.debug('Exception caugh: {}, retrying'.format(e.message.decode('windows-1250', 'ignore')))
# Any other error is expectable and recoverable, so let's wait a bit and retry again
# but, if too many errors, will log it (one every minute, for
# example)
counter += 1
if counter % 60 == 0:
servicemanager.LogWarningMsg(
'There are too many retries in progress, though still trying (last error: {})'.format(e.message))
if counter % 60 == 0: # Every 5 minutes, raise a log
logger.info('Trying to inititialize connection with broker (last error: {})'.format(e.message.decode('windows-1250', 'ignore')))
# Wait a bit before next check
win32event.WaitForSingleObject(self.hWaitStop, 1000)
win32event.WaitForSingleObject(self.hWaitStop, 5000)
# Broker connection is initialized, now get information about what to
# do
counter = 0
while self.isAlive:
try:
logger.debug('Requesting information of what to do now')
info = self.api.information()
data = info.split('\r')
if len(data) != 2:
servicemanager.LogErrorMsg(
'The format of the information message is not correct (got {})'.format(info))
logger.error('The format of the information message is not correct (got {})'.format(info))
raise Exception
params = data[1].split('\t')
if data[0] == 'rename':
@ -222,84 +240,96 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
self.rename(params[0])
# Rename with change password for an user
elif len(params) == 4:
self.rename(
params[0], params[1], params[2], params[3])
self.rename(params[0], params[1], params[2], params[3])
else:
logger.error(
'Got invalid parameter for rename operation: {}'.format(params))
logger.error('Got invalid parameter for rename operation: {}'.format(params))
return False
break
except Exception as e:
servicemanager.LogErrorMsg(
'Error at computer renaming stage: {}'.format(e.message))
logger.error('Error at computer renaming stage: {}'.format(e.message))
return False
elif data[0] == 'domain':
if len(params) != 5:
servicemanager.LogErrorMsg(
'Got invalid parameters for domain message: {}'.format(params))
logger.error('Got invalid parameters for domain message: {}'.format(params))
return False
self.joinDomain(
params[0], params[1], params[2], params[3], params[4])
self.joinDomain(params[0], params[1], params[2], params[3], params[4])
break
else:
servicemanager.LogErrorMsg(
'Unrecognized action sent from broker: {}'.format(data[0]))
logger.error('Unrecognized action sent from broker: {}'.format(data[0]))
return False # Stop running service
except REST.UserServiceNotFoundError:
servicemanager.LogErrorMsg(
'The host has lost the sync state with broker! (host uuid changed?)')
logger.error('The host has lost the sync state with broker! (host uuid changed?)')
return False
except Exception:
counter += 1
if counter % 60 == 0:
logger.warn(
'Too many retries in progress, though still trying (last error: {})'.format(e.message))
logger.warn('Too many retries in progress, though still trying (last error: {})'.format(e.message.decode('windows-1250', 'ignore')))
# Any other error is expectable and recoverable, so let's wait
# a bit and retry again
# Wait a bit before next check
win32event.WaitForSingleObject(self.hWaitStop, 1000)
if self.rebootRequested:
operations.reboot()
try:
operations.reboot()
except Exception as e:
logger.error('Exception on reboot: {}'.format(e.message))
return False # Stops service
return True
def checkIpsChanged(self):
netInfo = tuple(operations.getNetworkInfo())
for i in netInfo:
# If at least one ip has changed
if i.mac in self.knownIps and self.knownIps[i.mac] != i.ip:
servicemanager.LogInfoMsg('Notifying ip change to broker (mac {}, from {} to {})'.format(
i.mac, self.knownIps[i.mac], i.ip))
logger.info('Notifying ip change to broker (mac {}, from {} to {})'.format(i.mac, self.knownIps[i.mac], i.ip))
try:
# Notifies all interfaces IPs
self.api.notifyIpChanges(((v.mac, v.ip) for v in netInfo))
# Regenerates Known ips
self.knownIps = dict(((i.mac, i.ip) for i in netInfo))
except Exception as e:
servicemanager.LogWarningMsg(
'Got an error notifiying IPs to broker: {} (will retry in a bit)'.format(e.message))
logger.warn('Got an error notifiying IPs to broker: {} (will retry in a bit)'.format(e.message.decode('windows-1250', 'ignore')))
def SvcDoRun(self):
'''
Main service loop
'''
initCfg()
logger.debug('running SvcDoRun')
servicemanager.LogMsg(servicemanager.EVENTLOG_INFORMATION_TYPE,
servicemanager.PYS_SERVICE_STARTED,
(self._svc_name_, ''))
# call the CoInitialize to allow the registration to run in an other
# thread
logger.debug('Initializing com...')
pythoncom.CoInitialize()
# ********************************************************
# * Ask brokers what to do before proceding to main loop *
# ********************************************************
if self.interactWithBroker() is False:
self.ReportServiceStatus(win32service.SERVICE_STOPPED)
logger.debug('Interact with broker returned false, stopping service after a while')
self.notifyStop()
win32event.WaitForSingleObject(self.hWaitStop, 5000)
return
if self.isAlive is False:
logger.debug('The service is not alive after broker interaction, stopping it')
self.notifyStop()
return
if self.rebootRequested is True:
logger.debug('Reboot has been requested, stopping service')
self.notifyStop()
return
# ********************************
# * Registers SENS subscriptions *
# ********************************
# call the CoInitialize to allow the registration to run in an other
# thread
pythoncom.CoInitialize()
logevent('Registring ISensLogon')
subscription_guid = '{41099152-498E-11E4-8FD3-10FEED05884B}'
sl = SensLogon(self.api)
@ -316,8 +346,7 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
event_system.Store(PROGID_EventSubscription, event_subscription)
servicemanager.LogMsg(
servicemanager.EVENTLOG_INFORMATION_TYPE, 0xF000, ('Running Service Main Loop', ''))
logger.debug('Registered SENS, running main loop')
# *********************
# * Main Service loop *
@ -330,34 +359,22 @@ class UDSActorSvc(win32serviceutil.ServiceFramework):
# Process SENS messages, This will be a bit asyncronous (1 second
# delay)
pythoncom.PumpWaitingMessages()
if counter % 10 == 0:
self.checkIpsChanged()
#if counter % 10 == 0:
# self.checkIpsChanged()
# In milliseconds, will break
win32event.WaitForSingleObject(self.hWaitStop, 1000)
logger.debug('Exited main loop, deregistering SENS')
# *******************************************
# * Remove SENS subscription before exiting *
# *******************************************
event_system.Remove(
PROGID_EventSubscription, "SubscriptionID == " + subscription_guid)
servicemanager.LogMsg(servicemanager.EVENTLOG_INFORMATION_TYPE,
servicemanager.PYS_SERVICE_STOPPED,
(self._svc_name_, ''))
try:
self.ReportServiceStatus(win32service.SERVICE_STOPPED)
except Exception:
pass
self.notifyStop()
if __name__ == '__main__':
cfg = store.readConfig()
if logger.logger.isWindows():
# Logs will also go to windows event log for services
logger.logger.serviceLogger = True
logger.setLevel(cfg.get('logLevel', 10000))
initCfg()
win32serviceutil.HandleCommandLine(UDSActorSvc)

3
windowsActor/udsactor/REST.py
View File

@ -35,6 +35,7 @@
from __future__ import unicode_literals
import requests
import logging
import json
@ -100,6 +101,8 @@ class Api(object):
self.scrambledResponses = scrambledResponses
self.uuid = None
self.url = "{}://{}/rest/actor/".format(('http', 'https')[ssl], self.host)
# Disable logging requests messages except for warnings, errors, ...
logging.getLogger("requests").setLevel(logging.WARNING)
def _getUrl(self, method, key=None, ids=None):
url = self.url + method

3
windowsActor/udsactor/log.py
View File

@ -48,6 +48,7 @@ class Logger(object):
def setLevel(self, level):
self.logLevel = level
self.logger.log(INFO, 'Setting LogLevel to {}'.format(level))
def setRemoteLogger(self, remoteLogger):
self.remoteLogger = remoteLogger
@ -59,7 +60,7 @@ class Logger(object):
# If remote loger is available, notify message to it
try:
if self.remoteLogger is not None and self.remoteLogger.isConnected:
self.remoteLogger.log(self, level, message)
self.remoteLogger.log(level, message)
except Exception as e:
self.logger.log(FATAL, 'Error notifying log to broker: {}'.format(e.message))

25
windowsActor/udsactor/windows/SENS.py
View File

@ -70,30 +70,29 @@ class SensLogon(win32com.server.policy.DesignatedWrapPolicy):
self.api = api
def Logon(self, *args):
logevent('Logon : %s' % [args])
logger.debug('Logon event: {}'.format(args))
if self.api is not None and self.api.isConnected:
try:
data = self.api.login(args[0]).split('\t')
data = self.api.login(args[0])
logger.debug('Data received for login: {}'.format(data))
data = data.split('\t')
if len(data) == 2:
logger.debug('Data is valid: {}'.format(data))
windir = os.environ['windir']
f = open(os.path.join(windir, 'remoteh.txt'))
f.write(data[0])
f.close()
f = open(os.path.join(windir, 'remoteip.txt'))
f.write(data[1])
f.close()
with open(os.path.join(windir, 'remoteip.txt'), 'w') as f:
f.write(data[0])
with open(os.path.join(windir, 'remoteh.txt'), 'w') as f:
f.write(data[1])
except Exception as e:
logger.fatal(
'Error notifying logon to server: {}'.format(e.message))
logger.fatal('Error notifying logon to server: {}'.format(e))
def Logoff(self, *args):
logevent('Logoff : %s' % [args])
logger.debug('Logoff event: {}'.format(args))
if self.api is not None and self.api.isConnected:
try:
self.api.logout(args[0])
except Exception as e:
logger.fatal(
'Error notifying logon to server: {}'.format(e.message))
logger.fatal('Error notifying logon to server: {}'.format(e))
def StartShell(self, *args):
logevent('StartShell : %s' % [args])

11
windowsActor/udsactor/windows/log.py
View File

@ -41,9 +41,11 @@ OTHER, DEBUG, INFO, WARN, ERROR, FATAL = (10000 * (x + 1) for x in xrange(6))
class LocalLogger(object):
def __init__(self):
# tempdir is different for "user application" and "service"
# service wil get c:\windows\temp, while user will get c:\users\XXX\temp
logging.basicConfig(
filename=os.path.join(tempfile.gettempdir(), 'udsactor.log'),
filemode='w',
filemode='a',
format='%(levelname)s %(asctime)s %(message)s',
level=logging.DEBUG
)
@ -52,9 +54,10 @@ class LocalLogger(object):
def log(self, level, message):
# Debug messages are logged to a file
# our loglevels are 10000, 20000, ....
# python logger are 10, 20, .... (divided by 1000)
self.logger.log(level/1000, message)
# our loglevels are 10000 (other), 20000 (debug), ....
# logging levels are 10 (debug), 20 (info)
# OTHER = logging.NOTSET
self.logger.log(level/1000-10, message)
if level < INFO or self.serviceLogger is False: # Only information and above will be on event log
return

7
windowsActor/udsactor/windows/operations.py
View File

@ -40,6 +40,7 @@ import ctypes
from ctypes.wintypes import DWORD, LPCWSTR
from udsactor import utils
from udsactor.log import logger
def getErrorMessage(res=0):
msg = win32api.FormatMessage(res)
@ -88,7 +89,7 @@ EWX_REBOOT = 0x00000002
EWX_FORCE = 0x00000004
EWX_POWEROFF = 0x00000008
EWX_FORCEIFHUNG = 0x00000010
getNetworkInfo()
def reboot(flags=EWX_FORCEIFHUNG | EWX_REBOOT):
hproc = win32api.GetCurrentProcess()
htok = win32security.OpenProcessToken(hproc, win32security.TOKEN_ADJUST_PRIVILEGES | win32security.TOKEN_QUERY)
@ -97,7 +98,9 @@ def reboot(flags=EWX_FORCEIFHUNG | EWX_REBOOT):
result = win32api.ExitWindowsEx(flags, 0)
if result != 0:
# GetLastError and format it
raise Exception(getErrorMessage)
error = getErrorMessage()
logger.error('Reboot returned {}'.format(res))
raise Exception(error)
def renameComputer(newName):
# Needs admin privileges to work

2
windowsActor/udsactor/windows/store.py
View File

@ -43,7 +43,7 @@ def encoder(data):
def decoder(data):
return data.decode('bz2')
DEBUG = True
DEBUG = False
path = 'Software\\UDSEnterpriseActor'
baseKey = wreg.HKEY_CURRENT_USER if DEBUG is True else wreg.HKEY_LOCAL_MACHINE