Add support for SMBIOS vendor string based detection.

1 files changed, 29 insertions, 0 deletions

diff --git a/src/hypervisor_check.ads b/src/hypervisor_check.ads
index 0b5f154..1468695 100644
--- a/src/hypervisor_check.ads
+++ b/src/hypervisor_check.ads
@@ -1,24 +1,53 @@
 with Interfaces; use Interfaces;
 with System.Machine_Code; use System.Machine_Code;
 with Ada.Strings.Unbounded;
+with Ada.Text_IO;
+with Ada.Text_IO.Unbounded_IO;
 
 package Hypervisor_Check is
 
+    package IO renames Ada.Text_IO;
     package US renames Ada.Strings.Unbounded;
+    package UIO renames Ada.Text_IO.Unbounded_IO;
 
     function Get_Vendor_Name return US.Unbounded_String;
 
     function Hypervisor_Present return Boolean;
 
+    function Get_DMI_Vendor_Name return US.Unbounded_String;
+
 private
 
     Hypervisor_Leaf : constant := 16#40000000#;
     type CPUID_Registers is array (1 .. 4) of Unsigned_32;
 
+    Linux_Sys_Vendor_File : constant String := "/sys/class/dmi/id/sys_vendor";
+
+    -- SMBIOS vendor strings
+    VMWare_DMI_Pattern : constant String := "VMware, Inc.";
+    HyperV_DMI_Pattern : constant String := "Microsoft Corporation";
+    VirtualBox_DMI_Pattern : constant String := "innotek GmbH";
+    Parallels_DMI_Pattern : constant String := "Parallels";
+
+    -- Vendor names for human consumption
+    VMWare : constant String := "VMWare";
+    Xen : constant String := "Xen";
+    KVM : constant String := "KVM";
+    HyperV : constant String := "Microsoft Hyper-V";
+    VirtualBox : constant String := "VirtualBox";
+    Parallels : constant String := "Parallels";
+
+
     function CPUID (Arg : Unsigned_32) return CPUID_Registers;
 
     function String_of_U32 (Arg : Unsigned_32) return US.Unbounded_String;
 
     function Get_Vendor_String return US.Unbounded_String;
 
+    function Head_Of_File (Path : String) return US.Unbounded_String;
+
+    function Contains (Haystack : US.Unbounded_String; Needle : String) return Boolean;
+
+    function Get_DMI_Vendor_String return US.Unbounded_String;
+
 end Hypervisor_Check;