get username for policy and get home directory and get as_user for execute command from util. has been added domain column to session in database.

libpam-script/pam_script_ses_close

@@ -24,6 +24,13 @@ if [ -n $PAM_USER ] && [ $PAM_USER != "root" ]; then
     if [ -n $PAM_TTY ]; then
       TTY_DISPLAY="$PAM_TTY"
     fi
+    if [[ $PAM_USER = *'\'* ]]; then
+      IFS='\' read -ra NAMES <<< "$PAM_USER"
+      log "logout: \ ile geldi"
+      for i in "${NAMES[@]}"; do
+        PAM_USER="${NAMES[1]}"
+      done
+    fi
     log "logout: $PAM_USER service: $SERVICE tty: $TTY_DISPLAY"
     sudo python3 /usr/share/ahenk/ahenkd.py logout $PAM_USER
   fi

libpam-script/pam_script_ses_open

@@ -24,7 +24,21 @@ if [ -n $PAM_USER ] && [ $PAM_USER != "root" ]; then
     if [ -n $PAM_TTY ]; then
       TTY_DISPLAY="$PAM_TTY"
     fi
-    log "login: $PAM_USER service: $SERVICE tty: $TTY_DISPLAY"
+    domain="none"
+    if [[ $PAM_USER = *'\'* ]]; then
+      IFS='\' read -ra NAMES <<< "$PAM_USER"
+      log "login: \ ile geldi"
+      for i in "${NAMES[@]}"; do
+        domain="${NAMES[0]}"
+        PAM_USER="${NAMES[1]}"
+      done
+    fi
+    if [[ $domain != "none" ]]; then
+      log "login: $PAM_USER service: $SERVICE tty: $TTY_DISPLAY domain: $domain"
+      sudo python3 /usr/share/ahenk/ahenkd.py login $PAM_USER $SERVICE $TTY_DISPLAY $domain
+    else
+      log "login: $PAM_USER service: $SERVICE tty: $TTY_DISPLAY domain: none"
       sudo python3 /usr/share/ahenk/ahenkd.py login $PAM_USER $SERVICE $TTY_DISPLAY
     fi
+  fi
 fi

src/base/command/command_manager.py

@@ -45,6 +45,8 @@ class Commander(object):
                 data['desktop'] = params[3]
                 data['display'] = params[4]
                 if len(params) == 6:
+                    data['domain'] = params[5]
+                if len(params) == 7:
                     data['ip'] = params[5]
 
             elif len(params) == 3 and params[1] == 'logout':

src/base/command/command_runner.py

@@ -57,6 +57,8 @@ class CommandRunner(object):
                 if event is None:
                     break
                 json_data = json.loads(event)
+                self.logger.info(event)
+
             except Exception as e:
                 self.logger.error(
                     'A problem occurred while loading json. Check json format! Error Message: {0}.'
@@ -64,7 +66,6 @@ class CommandRunner(object):
                 return
 
             if json_data is not None:
-
                 self.logger.debug('Signal handled')
                 self.logger.debug('Signal is :{0}'.format(str(json_data['event'])))
 
@@ -77,27 +78,22 @@ class CommandRunner(object):
                     if 'ip' in json_data:
                         ip = json_data['ip']
 
+                    domain = None
+                    if 'domain' in json_data:
+                        domain = json_data['domain']
+
                     self.logger.info('login event is handled for user: {0}'.format(username))
                     Util.execute("systemctl restart sssd.service")
                     login_message = self.message_manager.login_msg(username,ip)
                     self.messenger.send_direct_message(login_message)
-
                     agreement = Agreement()
                     agreement_choice = None
 
-                    ## Default policy for users
-
-                    self.logger.info("Applying default policies for user {0}".format(username))
-                    self.default_policy.default_firefox_policy(username)
-                    self.default_policy.disable_update_package_notify(username)
-
                     if agreement.check_agreement(username) is not True and System.Ahenk.agreement() == '1':
                         self.logger.debug('User {0} has not accepted agreement.'.format(username))
                         thread_ask = Process(target=agreement.ask, args=(username, display,))
                         thread_ask.start()
-
                         agreement_timeout = self.conf_manager.get('SESSION', 'agreement_timeout')
-
                         timeout = int(agreement_timeout)  # sec
                         timer = time.time()
                         while 1:
@@ -131,15 +127,18 @@ class CommandRunner(object):
 
                     if agreement_choice is True or System.Ahenk.agreement() != '1':
                         self.db_service.delete('session', '1=1')
-
+                        self.logger.info('Display is {0}, desktop env is {1} for {2}'.format(display, desktop, username))
-                        self.logger.info(
-                            'Display is {0}, desktop env is {1} for {2}'.format(display, desktop,
-                                                                                username))
                         session_columns = self.db_service.get_cols('session')
                         self.db_service.update('session', session_columns,
-                                               [username, display, desktop, str(int(time.time())), ip])
+                                               [username, display, desktop, str(int(time.time())), ip, domain])
-                        get_policy_message = self.message_manager.policy_request_msg(username)
 
+                        # Default policy for users --->> START
+                        self.logger.info("Applying default policies for user {0}".format(Util.get_username()))
+                        self.default_policy.default_firefox_policy(Util.get_username())
+                        self.default_policy.disable_update_package_notify(Util.get_username())
+                        # Default policy for users --->> STOP
+
+                        get_policy_message = self.message_manager.policy_request_msg(username)
                         self.plugin_manager.process_mode('safe', username)
                         self.plugin_manager.process_mode('login', username)
 
@@ -151,29 +150,28 @@ class CommandRunner(object):
                                                timeout_function=self.execute_manager.execute_default_policy,
                                                checker_func=self.execute_manager.is_policy_executed, kwargs=kward))
 
-                        self.logger.info(
+                        self.logger.info('Requesting updated policies from Lider. If Ahenk could not reach updated '
-                            'Requesting updated policies from Lider. If Ahenk could not reach updated '
                             'policies in {0} sec, booked policies will be executed'.format(
                                 System.Ahenk.get_policy_timeout()))
                         self.messenger.send_direct_message(get_policy_message)
 
                 elif str(json_data['event']) == 'logout':
                     username = json_data['username']
-                    self.db_service.delete('session', 'username=\'{0}\''.format(username))
                     self.execute_manager.remove_user_executed_policy_dict(username)
+                    self.plugin_manager.process_mode('logout', username)
+                    self.plugin_manager.process_mode('safe', username)
+                    self.db_service.delete('session', '1=1')
                     # TODO delete all user records while initializing
                     self.logger.info('logout event is handled for user: {0}'.format(username))
                     ip = None
                     if 'ip' in json_data:
                         ip = json_data['ip']
+
                     logout_message = self.message_manager.logout_msg(username,ip)
                     self.messenger.send_direct_message(logout_message)
-
                     self.logger.info('Ahenk polkit file deleting..')
                     self.delete_polkit_user()
-
+                    # self.db_service.delete('session', 'username=\'{0}\''.format(username))
-                    self.plugin_manager.process_mode('logout', username)
-                    self.plugin_manager.process_mode('safe', username)
 
                 elif str(json_data['event']) == 'send':
                     self.logger.info('Sending message over ahenkd command. Response Message: {0}'.format(

src/base/database/ahenk_db_service.py

@@ -51,7 +51,8 @@ class AhenkDbService(object):
         self.check_and_create_table('agreement',
                                     ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'contract_id TEXT', 'username TEXT',
                                      'timestamp TEXT', 'choice TEXT'])
-        self.check_and_create_table('session', ['id INTEGER PRIMARY KEY AUTOINCREMENT','username TEXT', 'display TEXT', 'desktop TEXT', 'timestamp TEXT', 'ip TEXT'])
+        self.check_and_create_table('session', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'username TEXT', 'display TEXT',
+                                    'desktop TEXT', 'timestamp TEXT', 'ip TEXT', 'domain TEXT'])
 
         self.check_and_create_table('mail', ['id INTEGER PRIMARY KEY AUTOINCREMENT', 'command TEXT', 'mailstatus INTEGER',  'timestamp TEXT'])
 
@@ -65,7 +66,7 @@ class AhenkDbService(object):
         elif table_name == 'contract':
             return ['content', 'title', 'timestamp']
         elif table_name == 'session':
-            return ['username', 'display', 'desktop', 'timestamp', 'ip']
+            return ['username', 'display', 'desktop', 'timestamp', 'ip', 'domain']
         elif table_name == 'task':
             return ['id', 'create_date', 'modify_date', 'command_cls_id', 'parameter_map', 'deleted', 'plugin',
                     'cron_expr', 'file_server']
@@ -83,7 +84,11 @@ class AhenkDbService(object):
             self.logger.error('Database connection error: {0}'.format(str(e)))
 
     def check_and_create_table(self, table_name, cols):
-
+        if table_name == 'session':
+            if self.get_table_name('session') is not None:
+                domain = self.get_column_name('session', 'domain')
+                if domain is None:
+                    self.drop_table('session')
         try:
             self.lock.acquire(True)
             if self.cursor:
@@ -196,3 +201,27 @@ class AhenkDbService(object):
             self.connection.close()
         except Exception as e:
             self.logger.error('Closing database connection error: {0}'.format(str(e)))
+
+    def get_column_name(self, table_name, column_name):
+        try:
+            self.lock.acquire(True)
+            if self.cursor:
+                reader = self.cursor.execute('SELECT * FROM ' + table_name)
+                cols = [x[0] for x in reader.description]
+                self.connection.commit()
+                if column_name in cols:
+                    return column_name
+                else:
+                    return None
+        finally:
+            self.lock.release()
+
+    def get_table_name(self, table_name):
+        result = self.cursor.execute('SELECT name FROM sqlite_master WHERE type=\'table\';')
+        tables = []
+        for name in result:
+            tables.append(name[0])
+        if table_name in tables:
+            return table_name
+        else:
+            return None

src/base/messaging/messaging.py

@@ -176,7 +176,8 @@ class Messaging(object):
         data['password'] = str(self.conf_manager.get('CONNECTION', 'password'))
         # unregistration from commandline..
         if(usernameForCheck==None and passwordForCheck==None):
-            user_name = self.db_service.select_one_result('session', 'username')
+            # user_name = self.db_service.select_one_result('session', 'username')
+            user_name = Util.get_as_user()
             display = self.db_service.select_one_result('session', 'display')
             #user_name = os.getlogin()
             #display = Util.get_username_display()

src/base/plugin/plugin.py

@@ -213,10 +213,18 @@ class Plugin(threading.Thread):
 
                     self.logger.debug('[Plugin] Sending notify to user about profile process')
 
+                    # Util.send_notify("Lider Ahenk",
+                    #                  "{0} eklentisi şu anda bir profil çalıştırıyor.".format(self.getName()),
+                    #                  System.Sessions.display(item_obj.get_username()),
+                    #                  item_obj.get_username())
+                    display = System.Sessions.display(item_obj.get_username())
+                    if Util.get_desktop_env() == "gnome":
+                        display = Util.get_username_display_gnome(Util.get_username())
                     Util.send_notify("Lider Ahenk",
                                      "{0} eklentisi şu anda bir profil çalıştırıyor.".format(self.getName()),
-                                     System.Sessions.display(item_obj.get_username()),
+                                     display,
-                                     item_obj.get_username())
+                                     Util.get_as_user())
+
                     self.logger.debug('[Plugin] Handling profile')
                     Scope.get_instance().get_plugin_manager().find_policy_module(item_obj.get_plugin().get_name()).handle_policy(profile_data, self.context)
 

src/base/system/system.py

@@ -291,6 +291,9 @@ class System:
         @staticmethod
         def display(username):
             system = System()
+            if "\\" in username:
+                user_parser = username.split("\\")
+                username = user_parser[1]
             display = system.db_service.select_one_result('session', 'display', " username='{0}'".format(username))
             return display
 
@@ -300,9 +303,12 @@ class System:
             desktop = system.db_service.select_one_result('session', 'desktop', " username='{0}'".format(username))
             return desktop
 
-        staticmethod
+        @staticmethod
         def userip(username):
             system = System()
+            if "\\" in username:
+                user_parser = username.split("\\")
+                username = user_parser[1]
             userip = system.db_service.select_one_result('session', 'ip', " username='{0}'".format(username))
             return userip
 

src/base/util/util.py

@@ -22,6 +22,7 @@ class Util:
 
     def __init__(self):
         super().__init__()
+        scope = Scope().get_instance()
 
     @staticmethod
     def get_ask_path_file():
@@ -456,7 +457,34 @@ class Util:
     # return home directory for user. "/home/username"
     @staticmethod
     def get_homedir(user):
+        username = user
         try:
-            return expanduser("~{0}".format(user))
+            return expanduser("~{0}".format(username))
+        except:
+            raise
+
+    # return username from ahenk.db. if domain is not null return username is DOMAIN\\username
+    @staticmethod
+    def get_username():
+        user_name = Scope.get_instance().get_db_service().select_one_result('session', 'username', " 1=1 order by id desc ")
+        domain = Scope.get_instance().get_db_service().select_one_result('session', 'domain', " 1=1 order by id desc ")
+        if domain:
+            user_name = "{0}\\{1}".format(domain, user_name)
+        return user_name
+
+    # as_user is the user that run command. Return as_user for execute method. if domain is not null return as_user is DOMAIN\\\\username
+    @staticmethod
+    def get_as_user():
+        as_user = Scope.get_instance().get_db_service().select_one_result('session', 'username', " 1=1 order by id desc ")
+        domain = Scope.get_instance().get_db_service().select_one_result('session', 'domain', " 1=1 order by id desc ")
+        if domain:
+            as_user = "{0}\\\\{1}".format(domain, as_user)
+        return as_user
+
+    # return gid_number of username
+    @staticmethod
+    def get_gid_number(username):
+        try:
+            return pwd.getpwnam(username).pw_gid
         except:
             raise

src/plugins/browser/policy.py

@@ -23,7 +23,8 @@ class Browser(AbstractPlugin):
     def handle_policy(self):
         self.logger.info('Browser plugin handling...')
         try:
-            username = self.context.get('username')
+            # username = self.context.get('username')
+            username = self.get_username()
             self.logger.info('Username: {}'.format(username))
             if username is not None:
                 self.logger.debug('Writing preferences to user profile')
@@ -61,7 +62,7 @@ class Browser(AbstractPlugin):
 
                         self.logger.debug('User preferences were wrote successfully')
                         user_jss.close()
-                        change_owner = 'chown ' + username + ':' + username + ' ' + path
+                        change_owner = 'chown ' + self.get_as_user() + ':' + str(self.get_gid_number(username)) + ' ' + path
                         self.execute(change_owner)
                         self.logger.debug('Preferences file owner is changed')
 
@@ -131,7 +132,7 @@ class Browser(AbstractPlugin):
     def find_user_preference_paths(self, user_name):
 
         paths = []
-        homedir = self.get_homedir(user_name)
+        homedir = self.get_homedir(self.get_username())
         self.logger.info("Get home directory is {0} of {1} for firefox policy".format(homedir, user_name))
         firefox_path = '{0}/.mozilla/firefox/'.format(homedir)
         self.logger.info("Firefox path is {0}".format(firefox_path))

src/plugins/conky/execute_conky.py

@@ -93,19 +93,34 @@ class RunConkyCommand(AbstractPlugin):
             self.logger.debug('[Conky] Creating directory for conky autorun file at ' + self.conky_config_global_autorun_file)
             self.create_file(self.conky_config_global_autorun_file)
             self.write_file(self.conky_config_global_autorun_file, self.conky_autorun_content)
-        users = self.Sessions.user_name()
+        # users = self.Sessions.user_name()
+        user = self.get_username()
         desktop_env = self.get_desktop_env()
         self.logger.info("Get desktop environment is {0}".format(desktop_env))
-        for user in users:
+        # for user in users:
+        #     user_display = self.Sessions.display(user)
+        #     if desktop_env == "gnome":
+        #         user_display = self.get_username_display_gnome(user)
+        #     if user_display is None:
+        #         self.logger.debug('[Conky] executing for display none for user  '+ str(user))
+        #         self.execute('conky -q', result=False)
+        #     else:
+        #         self.logger.debug('[Conky] user display ' + str(user_display) +' user '+ str(user))
+        #         conky_cmd = 'su ' + str(user) + ' -c ' + ' "conky --display=' + str(user_display) + ' " '
+        #         self.logger.debug('[Conky] executing command:  ' + str(conky_cmd))
+        #         self.execute(conky_cmd, result=False)
+
         user_display = self.Sessions.display(user)
         if desktop_env == "gnome":
             user_display = self.get_username_display_gnome(user)
         if user_display is None:
-                self.logger.debug('[Conky] executing for display none for user  '+ str(user))
+            self.logger.debug('[Conky] executing for display none for user  ' + str(user))
             self.execute('conky -q', result=False)
         else:
-                self.logger.debug('[Conky] user display ' + str(user_display) +' user '+ str(user))
+            self.logger.debug('[Conky] user display ' + str(user_display) + ' user ' + str(user))
-                conky_cmd = 'su ' + str(user) + ' -c ' + ' "conky --display=' + str(user_display) + ' " '
+            # as_user is the user that run command.
+            as_user = self.get_as_user()
+            conky_cmd = 'su ' + str(as_user) + ' -c ' + ' "conky --display=' + str(user_display) + ' " '
             self.logger.debug('[Conky] executing command:  ' + str(conky_cmd))
             self.execute(conky_cmd, result=False)
         #self.execute('conky ', result=False)

src/plugins/conky/execute_xmessage.py

@@ -25,29 +25,26 @@ class RunXMessageCommand(AbstractPlugin):
         self.custom_message_command_ltsp = "su {0} -c 'export DISPLAY={1} && export XAUTHORITY=~{2}/.Xauthority && python3 /usr/share/ahenk/plugins/conky/ask.py \"LİDER AHENK\\\ BİLDİRİ \" \"{3}\" ' "
 
     def execute_xmessage(self, message):
-
         users = self.Sessions.user_name()
         self.logger.debug('[XMessage] users : ' + str(users))
         desktop_env = self.get_desktop_env()
         self.logger.info("Get desktop environment is {0}".format(desktop_env))
 
-        for user in users:
+        # for user in users:
+        user = self.get_username()
         user_display = self.Sessions.display(user)
         user_ip = self.Sessions.userip(user)
         if desktop_env == "gnome":
             user_display = self.get_username_display_gnome(user)
-
         if user_display is None:
             self.logger.debug('[XMessage] executing for display none for user  ' + str(user))
-
         else:
             self.logger.debug('[XMessage] user display ' + str(user_display) + ' user ' + str(user))
-
             if user_ip is None:
-                    t = threading.Thread(
+                self.execute(self.custom_message_command.format(self.get_as_user(), user_display, self.get_as_user(), message))
-                        target=self.execute(self.custom_message_command.format(user, user_display, user, message)))
+                # t = threading.Thread(
-                    t.start()
+                #     target=self.execute(self.custom_message_command.format(self.get_as_user(), user_display, self.get_as_user(), message)))
-
+                # t.start()
             else:
                 # message format for ltsp
                 self.logger.debug('user_ip: ' + str(user_ip) + ' user_display: ' + str(user_display))
@@ -72,7 +69,7 @@ class RunXMessageCommand(AbstractPlugin):
 
     def execute_user_message(self, selected_user, message):
 
-        users = self.Sessions.user_name();
+        users = self.Sessions.user_name()
         self.logger.debug('[XMessage] users : ' + str(users))
 
         for user in users:

src/plugins/conky/policy.py

@@ -36,7 +36,8 @@ class Conky(AbstractPlugin):
             # Is user profile
             if 'username' in self.context.data and self.context.get('username') is not None:
                 self.logger.debug('This is user profile, parameters reinitializing.')
-                self.username = self.context.get('username')
+                # self.username = self.context.get('username')
+                self.username = self.get_username()
                 self.homedir = self.get_homedir(self.username) + '/'
                 self.logger.info("Get home directory of {0}".format(self.homedir))
                 self.conky_config_file_dir = '{0}.conky/'.format(self.homedir)
@@ -65,7 +66,7 @@ class Conky(AbstractPlugin):
             self.initialize_auto_run()
 
             if self.machine_profile is False:
-                user_display = self.Sessions.display(self.username)
+                user_display = self.Sessions.display(self.context.get('username'))
                 desktop_env = self.get_desktop_env()
                 num = 0
                 if desktop_env == "gnome":
@@ -78,8 +79,10 @@ class Conky(AbstractPlugin):
                         time.sleep(10)
                         user_display = self.get_username_display_gnome(self.username)
                 self.logger.info("Get desktop environment is {0}".format(desktop_env))
-                self.execute(self.command_autorun_conky.format('--display=' + str(user_display), self.conky_config_file_path), as_user=self.username, result=False)
+                #
-                self.execute('chown -hR ' + self.username + ':' + self.username + ' ' + self.conky_config_file_dir)
+                as_user = self.get_as_user()
+                self.execute(self.command_autorun_conky.format('--display=' + str(user_display), self.conky_config_file_path), as_user=as_user, result=False)
+                self.execute('chown -hR ' + self.username + ':' + str(self.get_gid_number(self.username)) + ' ' + self.conky_config_file_dir)
                 self.logger.debug('Owner of Conky config file was changed.')
             else:
                 self.execute(self.command_autorun_conky.format('', self.conky_config_file_path), result=False)

src/plugins/conky/safe.py

@@ -8,7 +8,7 @@ class Safe(AbstractPlugin):
     def __init__(self, context):
         super(Safe, self).__init__()
         self.context = context
-        self.username = str(context.get_username())
+        self.username = self.get_username()
         self.logger = self.get_logger()
         self.logger.debug('Parameters were initialized.')
 
@@ -17,7 +17,7 @@ class Safe(AbstractPlugin):
         self.logger.debug('Conky named processes will be killed.')
         self.execute('killall -9 conky')
         # delete autostart and conky config file of logout username
-        self.homedir = self.get_homedir(self.username) + '/'
+        self.homedir = self.get_homedir(self.get_username()) + '/'
         self.conky_config_file_dir = '{0}.conky/'.format(self.homedir)
         self.conky_config_file_path = '{0}conky.conf'.format(self.conky_config_file_dir)
         if self.is_exist(self.conky_config_file_dir):

src/plugins/login-manager/policy.py

@@ -18,6 +18,7 @@ class LoginManager(AbstractPlugin):
         self.message_code = self.get_message_code()
 
         self.username = self.context.get('username')
+        # self.username = self.get_username()
 
         self.parameters = json.loads(self.data)
 
@@ -35,12 +36,12 @@ class LoginManager(AbstractPlugin):
         try:
             config = configparser.RawConfigParser()
             config.add_section('PERMISSION')
-
             config.set('PERMISSION', 'days', str(self.days))
             config.set('PERMISSION', 'start_time', str(self.start_time))
             config.set('PERMISSION', 'end_time', str(self.end_time))
             config.set('PERMISSION', 'last_date', str(self.last_date))
             config.set('PERMISSION', 'duration', str(self.duration))
+            config.set('PERMISSION', 'username', str(self.get_as_user()))
 
             if not self.is_exist('{0}login-manager/login_files'.format(self.Ahenk.plugins_path())):
                 self.create_directory('{0}login-manager/login_files'.format(self.Ahenk.plugins_path()))
@@ -69,14 +70,11 @@ class LoginManager(AbstractPlugin):
             self.context.create_response(code=self.message_code.POLICY_PROCESSED.value,
                                          message='Oturum kontrolü başlatıldı.')
 
-
         except Exception as e:
             self.logger.error(
                 'A problem occured while handling Login-Manager policy: {0}'.format(str(e)))
             self.context.create_response(code=self.message_code.POLICY_ERROR.value,
                                          message='Login-Manager profili uygulanırken bir hata oluştu.')
-
-
 def handle_policy(profile_data, context):
     manage = LoginManager(profile_data, context)
     manage.handle_policy()

src/plugins/login-manager/scripts/check.py

@@ -11,10 +11,8 @@ import subprocess, time
 import sys
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..')))
-
 from base.util.util import Util
 
-
 class CheckTime:
     def __init__(self):
         super(self.__class__, self).__init__()
@@ -26,7 +24,6 @@ class CheckTime:
                             level=logging.DEBUG)
 
         self.files = glob.glob('{0}login-manager/login_files/*.permissions'.format(sys.argv[1]))
-
         self.username = 'None'
 
         self.days = ''
@@ -56,19 +53,19 @@ class CheckTime:
 
             for file in self.files:
                 permission_file = str(file).replace('{0}login-manager/login_files/'.format(sys.argv[1]), '')
-                self.username = permission_file.replace('.permissions', '')
+                # self.username = permission_file.replace('.permissions', '')
 
                 config_parser = configparser.ConfigParser()
                 config_parser.read(file)
 
-                logging.debug('Getting parameters from permission file for user \'{0}\''.format(self.username))
-
                 self.days = config_parser.get('PERMISSION', 'days')
                 self.start_time = config_parser.get('PERMISSION', 'start_time')
                 self.end_time = config_parser.get('PERMISSION', 'end_time')
                 self.last_date = datetime.datetime.strptime(str(config_parser.get('PERMISSION', 'last_date')),
                                                             "%Y-%m-%d").date()
                 self.duration = config_parser.get('PERMISSION', 'duration')
+                self.username = config_parser.get('PERMISSION', 'username')
+                logging.debug('Getting parameters from permission file for user \'{0}\''.format(self.username))
 
                 logging.debug(
                     'Days: {0}, Start Time: {1}, End Time: {2}, Last Date: {3}, Duration between notify and logout: {4}'.format(
@@ -92,8 +89,10 @@ class CheckTime:
             logging.error(e)
 
     def write_to_user_profile(self):
-        if str(self.today) in self.days:
 
+
+        logging.debug("---->>> " + str(self.username))
+        if str(self.today) in self.days:
             if not (self.start_minute < self.current_minute < self.end_minute and self.current_date <= self.last_date):
                 logging.debug('User \'{0}\' will log out.'.format(self.username))
                 process = subprocess.Popen(self.command_logout_user.format(self.username), stdin=None, env=None,

src/plugins/remote-access/setup-vnc-server.py

@@ -48,7 +48,8 @@ class SetupVnc(AbstractPlugin):
                                          message='VNC sunucusu çalışırken bir hata oluştu.')
 
     def run_vnc_server(self):
-        user_name = self.db_service.select_one_result('session', 'username', " 1=1 order by id desc ")
+        # user_name = self.db_service.select_one_result('session', 'username', " 1=1 order by id desc ")
+        user_name = self.get_username()
         self.logger.info('get logon username is {0}'.format(user_name))
         self.logger.debug('Is VNC server installed?')
         if self.is_installed('x11vnc') is False:
@@ -62,13 +63,10 @@ class SetupVnc(AbstractPlugin):
         self.logger.debug('Running VNC proceses were killed')
         self.logger.debug('Getting display and username...')
         display_number = self.get_username_display(user_name)
-        # if len(arr) < 1:
-        #     raise NameError('Display not found!')
-
-        # params = str(arr[0]).split(' ')
         homedir = self.get_homedir(user_name)
         self.logger.info("Get home directory of {0} is {1}".format(user_name, homedir))
-        # self.logger.info("--------->>>> " + str(params))
+        # this user_name for execute method
+        user_name = self.get_as_user()
         self.logger.debug('Username:{0} Display:{1}'.format(user_name, display_number))
         if self.is_exist('{0}/.vncahenk{1}'.format(homedir, user_name)) is True:
             self.delete_folder('{0}/.vncahenk{1}'.format(homedir, user_name))
@@ -100,17 +98,6 @@ class SetupVnc(AbstractPlugin):
             self.logger.info("Lider Ahenk sistem yöneticisi tarafından kullanıcı izni ve bildirim gerektirmeksizin uzak erişim sağlanmıştır")
 
     def get_username_display(self, user):
-        # result_code, p_out, p_err = self.execute("who | awk '{print $1, $5}' | sed 's/(://' | sed 's/)//'", result=True)
-        # self.logger.debug('Getting display result code:{0}'.format(str(result_code)))
-        #
-        # result = []
-        # lines = str(p_out).split('\n')
-        # for line in lines:
-        #     arr = line.split(' ')
-        #     if len(arr) > 1 and str(arr[1]).isnumeric() is True:
-        #         result.append(line)
-        # return result
-
         result_code, p_out, p_err = self.execute("who | awk '{print $1, $5}' | sed 's/(://' | sed 's/)//'", result=True)
         display_number = None
         result = []