add vmware-tools

2026-06-23 11:27:54 +08:00
parent 2a794fa8d3
commit 4f39d384e2
198 changed files with 14170 additions and 0 deletions
@@ -0,0 +1,155 @@
+# Copyright (c) 2024 Broadcom. All Rights Reserved.
+# The term "Broadcom" refers to Broadcom Inc.
+# and/or its subsidiaries.
+
+from lib.command_runner import CommandRunner
+from lib.host_utils import get_vc_version, VcVersion
+from lib.text_utils import TextFilter
+
+PSQL_CLI = '/usr/bin/psql'
+
+def get_extension_thumbprints(extensions):
+    """
+    Get extension thumbprints from VCDB
+    :param extensions: list of extensions to fetch
+    :return:  dictionary of tuple (extension name, thumbprint, cert PEM)
+    """
+    
+    query = "SELECT ext_id, thumbprint FROM vpx_ext WHERE ext_id IN ('{}') ORDER BY ext_id" \
+            .format('\', \''.join(extensions))
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    extensions = TextFilter(output).contain('|').cut(delimiter='|', fields=[0]).remove_white_spaces().get_lines()
+    thumbprints = TextFilter(output).contain('|').cut(delimiter='|', fields=[1]).remove_white_spaces().get_lines()
+    result = dict()
+    for index, extension in enumerate(extensions):
+        if get_vc_version() >= VcVersion.V9:
+            query = "SELECT certificate FROM vpx_ext WHERE ext_id='{}'".format(extension)
+            output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t', '-A').run_and_get_output()
+            pem_cert = TextFilter(output).get_text().strip()
+        else:
+            pem_cert = None
+        result[extension] = (thumbprints[index], pem_cert)
+    return result
+
+
+def get_extension_thumbprint(extension):
+    """
+    Get thumbprint for a specific extension
+    :param extension: extension name
+    :return: extension thumbprint
+    """
+    query = "SELECT thumbprint FROM vpx_ext WHERE ext_id = '{}'".format(extension)
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    thumbprint = TextFilter(output).head(1).get_text().strip()
+    return thumbprint
+
+
+def update_extension_thumbprint(extension, thumbprint, cert_pem=None):
+    if cert_pem is None:
+        query = "UPDATE vpx_ext SET thumbprint = '{}' WHERE ext_id = '{}'".format(thumbprint, extension)
+    else:
+        one_line_pem = cert_pem.strip().replace('\n', '\\n')
+        query = "UPDATE vpx_ext SET thumbprint = '{}', certificate = E'{}' WHERE ext_id = '{}'".format(thumbprint, one_line_pem, extension)
+    ret_code, _, _ = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres', '-c',
+                                   query, '-t').run()
+    return ret_code == 0
+
+
+def get_certificate_management_mode():
+    """
+    Get certificate management node
+    """
+    query = 'SELECT value FROM vpx_parameter WHERE name=\'vpxd.certmgmt.mode\''
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).head(1).get_text().strip()
+
+
+def get_number_of_hosts():
+    """
+    Get the number of hosts
+    """
+    query = 'SELECT COUNT(id) FROM vpx_host WHERE enabled=1'
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).head(1).get_text().strip()
+
+
+def get_hosts():
+    """
+    Get the hosts
+    """
+    query = 'SELECT id, dns_name, ip_address FROM vpx_host WHERE enabled=1 ORDER BY dns_name ASC, ip_address ASC'
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres', '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).get_text().strip().splitlines()
+
+
+def get_host_and_cluster_id():
+    """
+        Get the hosts id and cluster id it belongs too
+        """
+    query = '''SELECT ent.id, ent.parent_id FROM vpx_entity as ent LEFT JOIN vpx_object_type AS obj ON 
+    ent.type_id = obj.id WHERE obj.name = 'HOST' '''
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).get_text().strip().splitlines()
+
+
+def get_clusters():
+    query = '''SELECT ent.id, ent.name FROM vpx_entity as ent LEFT JOIN vpx_object_type AS obj 
+    ON ent.type_id = obj.id WHERE obj.name = 'CLUSTER_COMPUTE_RESOURCE' '''
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).get_text().strip().splitlines()
+
+
+def get_ssl_thumbprint_from_vcdb(access_string):
+    """
+    Get the expected_ssl_thumbprint and host_ssl_thumbprint
+    """
+    query = 'SELECT expected_ssl_thumbprint,host_ssl_thumbprint FROM vpx_host where  ' \
+            'dns_name = \'{0}\' or ip_address = \'{0}\' '.format(access_string)
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).get_text().strip().splitlines()
+
+
+def get_vmca_configuration_from_vcdb():
+    """
+    Get the vpxd.certmgmt.mode and vpxd.certmgmt.cn.* values from the database
+    """
+    query = "SELECT name,value FROM vpx_parameter WHERE name='vpxd.certmgmt.mode' OR name LIKE 'vpxd.certmgmt.certs.cn.%' ORDER BY name"
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    return TextFilter(output).get_text().strip().splitlines()
+
+def scheduled_task_table_exists():
+    """
+    Determine if the vpx_sched_persistent_user_token table exists
+    """
+    query = "SELECT EXISTS (SELECT FROM information_schema.tables WHERE table_schema='vc' AND table_name='vpx_sched_persistent_user_token')"
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    
+    return TextFilter(output).get_text().strip().lower() == 't'
+
+
+def get_number_scheduled_tasks():
+    """
+    Get number of user-defined Scheduled Tasks in vCenter
+    """
+    query = "SELECT COUNT(usertoken_id) FROM vpx_sched_persistent_user_token"
+
+    output = CommandRunner(PSQL_CLI, '-d', 'VCDB', '-U', 'postgres',
+                           '-c', query, '-t').run_and_get_output()
+    
+    return TextFilter(output).get_text().strip()