Python脚本判断 Linux 是否运行在虚拟机上

要判断 Linux 是否运行在虚拟机上，我们可以使用 Python 编写脚本，具体步骤如下：

步骤一：导入必要的 Python 库

Python 作为一门高级编程语言，有很多非常实用的库可供选择，我们可以使用如下代码导入必要的 Python 库：

import os    # 导入操作系统相关的库
import re    # 导入正则表达式相关的库

步骤二：获取 Linux 操作系统的信息

我们需要使用 Python 获取当前操作系统的信息，以便后续对其进行识别判断，这里我们可以使用如下代码获取当前系统的信息：

def get_os_info():
    sys_info = {}    # 用字典保存系统信息
    with open('/etc/os-release', 'r') as f:
        for line in f:
            if line.startswith('#'):
                continue
            temp_list = line.strip().split('=')
            if len(temp_list) == 2:
                sys_info[temp_list[0]] = temp_list[1].replace('\"', '')
    return sys_info

这个函数的作用是获取/etc/os-release文件中的信息，并将其保存在一个字典中。

步骤三：识别虚拟机类型

要识别 Linux 是否运行在虚拟机上，我们需要检查系统中是否存在虚拟化信息文件。代码如下：

def is_virtualized():
    if not os.path.isfile('/sys/devices/virtual/dmi/id/product_name'):
        return False    # 不存在虚拟化信息文件，返回False
    with open('/sys/devices/virtual/dmi/id/product_name', 'r') as f:
        if 'VMware' in f.read() or 'VirtualBox' in f.read():
            return True    # 文件中存在VMware或VirtualBox信息，表示该Linux正在虚拟机中运行，返回True
        else:
            return False    # 文件中不存在VMware或VirtualBox信息，表示该Linux未运行在虚拟机中，返回False

通过分析/sys/devices/virtual/dmi/id/product_name文件中的内容，我们可以判断当前 Linux 是否运行在VMware或VirtualBox虚拟机中。如果该文件中存在这两种信息，则表示当前 Linux 正在虚拟机中运行。

步骤四：打印检测结果

把步骤二和步骤三组合起来，形成最终的判断函数如下：

def is_running_on_virtual_machine():
    os_info = get_os_info()
    if not os_info:
        return False
    if 'CentOS' in os_info['NAME']:    #根据操作系统不同来选择不同的虚拟化信息文件
        virtual_file = '/sys/devices/virtual/dmi/id/product_version'
    else:
        virtual_file = '/sys/devices/virtual/dmi/id/product_name'
    return is_virtualized(virtual_file)

示例一：在 Ubuntu 系统上运行

#!/usr/bin/env python3
import os
import re

#获取Linux操作系统信息
def get_os_info():
    sys_info = {}    #用字典保存系统信息
    with open('/etc/os-release', 'r') as f:
        for line in f:
            if line.startswith('#'):
                continue
            temp_list = line.strip().split('=')
            if len(temp_list) == 2:
                sys_info[temp_list[0]] = temp_list[1].replace('\"', '')
    return sys_info

#判断当前Linux是否运行在虚拟机上
def is_virtualized(virtual_file):
    if not os.path.isfile(virtual_file):
        return False    #不存在虚拟化信息文件，返回False
    with open(virtual_file, 'r') as f:
        if 'VMware' in f.read() or 'VirtualBox' in f.read():
            return True    #文件中存在VMware或VirtualBox信息，表示该Linux正在虚拟机中运行，返回True
        else:
            return False    #文件中不存在VMware或VirtualBox信息，表示该Linux未运行在虚拟机中，返回False

#判断当前Linux是否运行在虚拟机上（根据不同操作系统分别调用不同的虚拟化信息文件）
def is_running_on_virtual_machine():
    os_info = get_os_info()
    if not os_info:
        return False
    if 'CentOS' in os_info['NAME']:    #根据操作系统不同来选择不同的虚拟化信息文件
        virtual_file = '/sys/devices/virtual/dmi/id/product_version'
    else:
        virtual_file = '/sys/devices/virtual/dmi/id/product_name'
    return is_virtualized(virtual_file)

if is_running_on_virtual_machine():
    print('当前Linux运行在虚拟机中')
else:
    print('当前Linux没有运行在虚拟机中')

示例二：在 CentOS 系统上运行

#!/usr/bin/env python3
import os
import re

#获取Linux操作系统信息
def get_os_info():
    sys_info = {}    #用字典保存系统信息
    with open('/etc/os-release', 'r') as f:
        for line in f:
            if line.startswith('#'):
                continue
            temp_list = line.strip().split('=')
            if len(temp_list) == 2:
                sys_info[temp_list[0]] = temp_list[1].replace('\"', '')
    return sys_info

#判断当前Linux是否运行在虚拟机上
def is_virtualized(virtual_file):
    if not os.path.isfile(virtual_file):
        return False    #不存在虚拟化信息文件，返回False
    with open(virtual_file, 'r') as f:
        if 'VMware' in f.read() or 'VirtualBox' in f.read():
            return True    #文件中存在VMware或VirtualBox信息，表示该Linux正在虚拟机中运行，返回True
        else:
            return False    #文件中不存在VMware或VirtualBox信息，表示该Linux未运行在虚拟机中，返回False

#判断当前Linux是否运行在虚拟机上（根据不同操作系统分别调用不同的虚拟化信息文件）
def is_running_on_virtual_machine():
    os_info = get_os_info()
    if not os_info:
        return False
    if 'CentOS' in os_info['NAME']:    #根据操作系统不同来选择不同的虚拟化信息文件
        virtual_file = '/sys/devices/virtual/dmi/id/product_version'
    else:
        virtual_file = '/sys/devices/virtual/dmi/id/product_name'
    return is_virtualized(virtual_file)

if is_running_on_virtual_machine():
    print('当前Linux运行在虚拟机中')
else:
    print('当前Linux没有运行在虚拟机中')