From 0a655e544146d3a58ff03e7fb9bcfcdbf6be643f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tuncay=20=C3=87OLAK?= <tuncay.colak@tubitak.gov.tr>
Date: Tue, 27 Oct 2020 14:40:52 +0300
Subject: [PATCH] Syncing /usr/share/ahenk to src and recreating ahenk.install

---
 debian/ahenk.install                          |  2 +
 .../ahenk/base/command/command_runner.py      |  5 +-
 usr/share/ahenk/base/messaging/messaging.py   |  1 +
 .../ahenk/base/registration/registration.py   |  1 +
 .../ahenk/base/registration/scripts/ad.sh     |  4 +-
 usr/share/ahenk/base/util/util.py             | 10 ++++
 usr/share/ahenk/plugins/ldap/rename_entry.py  | 30 +++++++++-
 .../plugins/remote-access/setup-vnc-server.py | 36 ++++++------
 .../ahenk/plugins/resource-usage/agent_info   |  0
 .../plugins/resource-usage/agent_info.py      | 56 +++++++++++++++++++
 10 files changed, 121 insertions(+), 24 deletions(-)
 create mode 100644 usr/share/ahenk/plugins/resource-usage/agent_info
 create mode 100644 usr/share/ahenk/plugins/resource-usage/agent_info.py

diff --git a/debian/ahenk.install b/debian/ahenk.install
index d7b0462..152476c 100644
--- a/debian/ahenk.install
+++ b/debian/ahenk.install
@@ -166,6 +166,8 @@ usr/share/ahenk/plugins/service/service_list.py
 usr/share/ahenk/plugins/service/main.py
 usr/share/ahenk/plugins/service/get_services.py
 usr/share/ahenk/plugins/service
+usr/share/ahenk/plugins/resource-usage/agent_info.py
+usr/share/ahenk/plugins/resource-usage/agent_info
 usr/share/ahenk/plugins/resource-usage/main.py
 usr/share/ahenk/plugins/resource-usage/resource_info_alert.py
 usr/share/ahenk/plugins/resource-usage/send_mail.py
diff --git a/usr/share/ahenk/base/command/command_runner.py b/usr/share/ahenk/base/command/command_runner.py
index 25a1ef6..8165cc4 100644
--- a/usr/share/ahenk/base/command/command_runner.py
+++ b/usr/share/ahenk/base/command/command_runner.py
@@ -45,8 +45,9 @@ class CommandRunner(object):
             self.logger.info('Ahenk polkit file not found')
         else:
             Util.delete_file(ahenk_policy_file)
-            Util.write_file(ahenk_policy_file, content)
-            self.logger.info('Root added ahenk polkit file')
+            self.logger.info("Delete polkit file")
+            # Util.write_file(ahenk_policy_file, content)
+            # self.logger.info('Root added ahenk polkit file')
 
     def run_command_from_fifo(self, num, stack):
         """ docstring"""
diff --git a/usr/share/ahenk/base/messaging/messaging.py b/usr/share/ahenk/base/messaging/messaging.py
index dca6a1a..caa239d 100644
--- a/usr/share/ahenk/base/messaging/messaging.py
+++ b/usr/share/ahenk/base/messaging/messaging.py
@@ -78,6 +78,7 @@ class Messaging(object):
         data['diskFree'] = System.Hardware.Disk.free()
         data['memory'] = System.Hardware.Memory.total()
         data['hostname'] = str(System.Os.hostname())
+        data['agentVersion'] = str(Util.get_agent_version())
 
         self.logger.debug('USER IP : '+ str(ip)+ ' IPADDRESSES : '+ str(System.Hardware.Network.ip_addresses()).replace('[', '').replace(']', ''))
 
diff --git a/usr/share/ahenk/base/registration/registration.py b/usr/share/ahenk/base/registration/registration.py
index 8058e25..cc1fcda 100644
--- a/usr/share/ahenk/base/registration/registration.py
+++ b/usr/share/ahenk/base/registration/registration.py
@@ -281,6 +281,7 @@ class Registration:
             'hardware.baseboard.serialNumber': System.Hardware.BaseBoard.serial_number()[1].replace('\n', '') if
             System.Hardware.BaseBoard.serial_number()[0] == 0 else 'n/a',
             'processor': System.Hardware.Cpu.brand(),
+            'agentVersion': Util.get_agent_version(),
         }
 
         return json.dumps(params)
diff --git a/usr/share/ahenk/base/registration/scripts/ad.sh b/usr/share/ahenk/base/registration/scripts/ad.sh
index 2298d05..413cb06 100644
--- a/usr/share/ahenk/base/registration/scripts/ad.sh
+++ b/usr/share/ahenk/base/registration/scripts/ad.sh
@@ -7,8 +7,8 @@ ad_host_name=$2
 echo "samba-common samba-common/workgroup string  WORKGROUP" | sudo debconf-set-selections
 echo "samba-common samba-common/dhcp boolean false" | sudo debconf-set-selections
 echo "samba-common samba-common/do_debconf boolean true" | sudo debconf-set-selections
-apt-get -y install samba-common
-
+#apt-get -y install samba-common
+DEBIAN_FRONTEND=noninteractive apt-get install -y -o Dpkg::Options::="--force-confold" samba-common
 
 cat > /root/debconf-krb5.conf <<EOF
 
diff --git a/usr/share/ahenk/base/util/util.py b/usr/share/ahenk/base/util/util.py
index 21cd9b0..40ad67e 100644
--- a/usr/share/ahenk/base/util/util.py
+++ b/usr/share/ahenk/base/util/util.py
@@ -488,3 +488,13 @@ class Util:
             return pwd.getpwnam(username).pw_gid
         except:
             raise
+
+    @staticmethod
+    def get_agent_version():
+        result_code, result, p_err = Util.execute('dpkg -s {} | grep Version'.format("ahenk"))
+        data = result.split(': ')
+        if data[0] == 'Version':
+            version = data[1].strip('\n')
+            return version
+        else:
+            return None
diff --git a/usr/share/ahenk/plugins/ldap/rename_entry.py b/usr/share/ahenk/plugins/ldap/rename_entry.py
index ef13d68..f69563e 100644
--- a/usr/share/ahenk/plugins/ldap/rename_entry.py
+++ b/usr/share/ahenk/plugins/ldap/rename_entry.py
@@ -20,6 +20,7 @@ class RenameEntry(AbstractPlugin):
         self.message_code = self.get_message_code()
         self.conf_manager = scope.get_configuration_manager()
         self.hostname_file = '/etc/hostname'
+        self.hosts_file = '/etc/hosts'
 
     def update_dn(self, old_cn, new_cn, new_dn):
         return self.db_service.update('registration', ['jid', 'dn'], [new_cn, new_dn], ' jid = ' + '\'' + old_cn + '\'')
@@ -32,14 +33,37 @@ class RenameEntry(AbstractPlugin):
             new_cn = self.data['new_cn']
             directory_server = self.data['directory_server']
             new_dn = str(old_dn).replace(old_cn, new_cn)
-            
+
             self.logger.debug('Renaming hostname from: ' + old_cn + " to: " + new_cn)
             self.write_file(self.hostname_file, new_cn)
 
-            ## update agent db 
+            ## update /etc/hosts file
+            hosts_file = open (self.hosts_file, 'r')
+            file_data = hosts_file.read()
+
+            old_hostname_line = ""
+            new_hostname_line = ""
+            with open(self.hosts_file) as fp:
+                line = fp.readline()
+                while line:
+                    if line.strip().startswith("127.0.1.1"):
+                        old_hostname_line = line
+                        new_hostname_line = line.replace(old_cn, new_cn)
+                        break
+                    line = fp.readline()
+
+            file_data = file_data.replace(old_hostname_line, new_hostname_line)
+
+            hosts_file.close()
+            hosts_file = open(self.hosts_file, 'w')
+            hosts_file.write(file_data)
+            hosts_file.close()
+            self.logger.info("/etc/hosts file is updated.")
+            ## update agent db
             jid = self.db_service.select_one_result('registration','jid','registered = 1')
             new_dn = str(old_dn).replace(old_cn, new_cn)
 
+
             self.update_dn(old_cn, new_cn, new_dn)
             if directory_server == "LDAP":
                 # update SSSD conf agent DN
@@ -63,7 +87,7 @@ class RenameEntry(AbstractPlugin):
             self.conf_manager.set('CONNECTION', 'uid', new_cn)
             with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
                 self.conf_manager.write(configfile)
-            
+
             self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
                                          message='Ahenk adı başarı ile değiştirildi.',
                                          data=json.dumps({'Dn': new_dn}),
diff --git a/usr/share/ahenk/plugins/remote-access/setup-vnc-server.py b/usr/share/ahenk/plugins/remote-access/setup-vnc-server.py
index efd066f..4111976 100644
--- a/usr/share/ahenk/plugins/remote-access/setup-vnc-server.py
+++ b/usr/share/ahenk/plugins/remote-access/setup-vnc-server.py
@@ -68,38 +68,40 @@ class SetupVnc(AbstractPlugin):
         if desktop_env == "gnome":
             display_number = self.get_username_display_gnome(user_name)
         self.logger.info("Get display of {0} is {1}".format(user_name, display_number))
-        homedir = self.get_homedir(user_name)
-        self.logger.info("Get home directory of {0} is {1}".format(user_name, homedir))
+        #homedir = self.get_homedir(user_name)
+        #self.logger.info("Get home directory of {0} is {1}".format(user_name, homedir))
         # 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))
-            self.logger.debug('Cleaning previous configurations.')
-        self.logger.debug('Creating user VNC conf file as user')
-        self.execute('su - {0} -c "mkdir -p {1}/.vncahenk{0}"'.format(user_name, homedir), result=False)
-        self.logger.debug('Creating password as user')
-        self.execute('su - {0} -c "x11vnc -storepasswd {1} {2}/.vncahenk{3}/x11vncpasswd"'.format(user_name, self.password, homedir,
-                                                                                         user_name), result=False)
+        #if self.is_exist('{0}/.vncahenk{1}'.format(homedir, user_name)) is True:
+        #    self.delete_folder('{0}/.vncahenk{1}'.format(homedir, user_name))
+        #    self.logger.debug('Cleaning previous configurations.')
+        #self.logger.debug('Creating user VNC conf file as user')
+        #self.execute('su - {0} -c "mkdir -p {1}/.vncahenk{0}"'.format(user_name, homedir), result=False)
+        #self.logger.debug('Creating password as user')
+        #self.execute('su - {0} -c "x11vnc -storepasswd {1} {2}/.vncahenk{3}/x11vncpasswd"'.format(user_name, self.password, homedir,user_name), result=False)
         self.logger.debug('Running VNC server as user.')
         if self.data['permission'] == "yes":
             self.send_notify("Liderahenk",
                              "Lider Ahenk Sistem Yoneticisi tarafindan\n5 sn sonra bilgisayarınıza uzak erişim sağlanacaktır.\nBağlantı kapatıldıktan sonra ayrıca bilgilendirilecektir.",
                              display_number, user_name, timeout=50000)
             time.sleep(5)
-            self.execute('su - {0} -c "x11vnc -accept \'popup\' -gone \'popup\' -rfbport {1} -rfbauth {2}/.vncahenk{0}/x11vncpasswd -o {2}/.vncahenk{3}/vnc.log -display {4}"'.format(
-                    user_name, self.port, homedir, user_name, display_number), result=False)
+            # self.execute('su - {0} -c "x11vnc -accept \'popup\' -gone \'popup\' -rfbport {1} -passwd {2} -o {2}/.vncahenk{3}/vnc.log -display {4}"'.format(
+            #         user_name, self.port, user_name, display_number), result=False)
+            self.execute(
+                'su - {0} -c "x11vnc -accept \'popup\' -gone \'popup\' -rfbport {1} -passwd {2}  -display {3}"'.format(
+                    user_name, self.port, self.password, display_number), result=False)
         elif self.data["permission"] == "no":
             self.logger.info("Lider Ahenk sistem yöneticisi 5 sn sonra bilgisayarınıza uzak erişim sağlayacaktır. ")
             self.send_notify("Liderahenk",
                              "Lider Ahenk Sistem Yoneticisi tarafindan\n5 sn sonra bilgisayarınıza uzak erişim sağlanacaktır.\nBağlantı kapatıldıktan sonra ayrıca bilgilendirilecektir.",
                              display_number, user_name, timeout=50000)
-            time.sleep(5)
-            self.execute('su - {0} -c "x11vnc -gone \'popup\' -rfbport {1} -rfbauth {2}/.vncahenk{0}/x11vncpasswd -o {2}/.vncahenk{3}/vnc.log -display {4}"'.format(
-                    user_name, self.port, homedir, user_name, display_number), result=False)
+            time.sleep(2)
+            self.execute('su - {0} -c "x11vnc -gone \'popup\' -rfbport {1} -passwd {2} -display {3}"'.format(
+                    user_name, self.port, self.password, display_number), result=False)
         else:
-            self.execute('su - {0} -c "x11vnc -rfbport {1} -rfbauth {2}/.vncahenk{0}/x11vncpasswd -o {2}/.vncahenk{3}/vnc.log -display {4}"'.format(
-                    user_name, self.port, homedir, user_name, display_number), result=False)
+            self.execute('su - {0} -c "x11vnc -rfbport {1} -passwd {2} -display {3}"'.format(
+                    user_name, self.port, self.password, display_number), result=False)
             self.logger.info("Lider Ahenk sistem yöneticisi tarafından kullanıcı izni ve bildirim gerektirmeksizin uzak erişim sağlanmıştır")
 
     def create_password(self, pass_range):
diff --git a/usr/share/ahenk/plugins/resource-usage/agent_info b/usr/share/ahenk/plugins/resource-usage/agent_info
new file mode 100644
index 0000000..e69de29
diff --git a/usr/share/ahenk/plugins/resource-usage/agent_info.py b/usr/share/ahenk/plugins/resource-usage/agent_info.py
new file mode 100644
index 0000000..421d9f7
--- /dev/null
+++ b/usr/share/ahenk/plugins/resource-usage/agent_info.py
@@ -0,0 +1,56 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+# Author: Tuncay ÇOLAK <tuncay.colak@tubitak.gov.tr>
+
+from base.plugin.abstract_plugin import AbstractPlugin
+import json
+
+
+class AgentInfo(AbstractPlugin):
+    def __init__(self, data, context):
+        super(AbstractPlugin, self).__init__()
+        self.data = data
+        self.context = context
+        self.logger = self.get_logger()
+        self.message_code = self.get_message_code()
+
+    def handle_task(self):
+        try:
+            device = ""
+            self.logger.debug("Gathering resource usage for disk, memory and CPU.")
+            for part in self.Hardware.Disk.partitions():
+                if len(device) != 0:
+                    device += ", "
+                device = device + part.device
+
+            data = {'System': self.Os.name(), 'Release': self.Os.kernel_release(),
+                    'agentVersion': self.get_agent_version(),
+                    'hostname': self.Os.hostname(),
+                    'ipAddresses': str(self.Hardware.Network.ip_addresses()).replace('[', '').replace(']', ''),
+                    'os.name': self.Os.name(),
+                    'osVersion': self.Os.version(),
+                    'macAddresses': str(self.Hardware.Network.mac_addresses()).replace('[', '').replace(']', ''),
+                    'hardware.systemDefinitions': self.Hardware.system_definitions(),
+                    'hardware.monitors': self.Hardware.monitors(),
+                    'hardware.screens': self.Hardware.screens(),
+                    'hardware.usbDevices': self.Hardware.usb_devices(),
+                    'hardware.printers': self.Hardware.printers(),
+                    'diskTotal': self.Hardware.Disk.total(),
+                    'diskUsed': self.Hardware.Disk.used(),
+                    'diskFree': self.Hardware.Disk.free(),
+                    'memory': self.Hardware.Memory.total(),
+                    'Device': device,
+            }
+            self.logger.debug("Agent info gathered.")
+            self.context.create_response(code=self.message_code.TASK_PROCESSED.value,
+                                         message='Ahenk bilgileri başarıyla güncellendi.',
+                                         data=json.dumps(data), content_type=self.get_content_type().APPLICATION_JSON.value)
+        except Exception as e:
+            self.logger.error(str(e))
+            self.context.create_response(code=self.message_code.TASK_ERROR.value,
+                                         message='Ahenk bilgileri güncellenirken hata oluştu: {0}'.format(str(e)))
+
+
+def handle_task(task, context):
+    plugin = AgentInfo(task, context)
+    plugin.handle_task()