[PATCH v4 1/5] ntdll: Return BIOS info from NtQuerySystemInformation on Linux

[Alex Henrie]

Signed-off-by: Alex Henrie <alexhenrie24 at gmail.com>
---
 dlls/ntdll/nt.c | 169 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 169 insertions(+)

diff --git a/dlls/ntdll/nt.c b/dlls/ntdll/nt.c
index dc0ce04f42..72ae381813 100644
--- a/dlls/ntdll/nt.c
+++ b/dlls/ntdll/nt.c
@@ -66,6 +66,50 @@
 
 WINE_DEFAULT_DEBUG_CHANNEL(ntdll);
 
+#include "pshpack1.h"
+
+typedef enum _SYSTEM_FIRMWARE_TABLE_ACTION
+{
+    SystemFirmwareTable_Enumerate = 0,
+    SystemFirmwareTable_Get = 1
+} SYSTEM_FIRMWARE_TABLE_ACTION;
+
+typedef struct _SYSTEM_FIRMWARE_TABLE_INFORMATION
+{
+    ULONG ProviderSignature;
+    SYSTEM_FIRMWARE_TABLE_ACTION Action;
+    ULONG TableID;
+    ULONG TableBufferLength;
+    UCHAR TableBuffer[1];
+} SYSTEM_FIRMWARE_TABLE_INFORMATION;
+
+struct smbios_prologue {
+    BYTE calling_method;
+    BYTE major_version;
+    BYTE minor_version;
+    BYTE revision;
+    DWORD length;
+};
+
+struct smbios_bios {
+    BYTE type;
+    BYTE length;
+    WORD handle;
+    BYTE vendor;
+    BYTE version;
+    WORD start;
+    BYTE date;
+    BYTE size;
+    UINT64 characteristics;
+};
+
+#include "poppack.h"
+
+/* Firmware table providers */
+#define ACPI 0x41435049
+#define FIRM 0x4649524D
+#define RSMB 0x52534D42
+
 /*
  *	Token
  */
@@ -1850,6 +1894,39 @@ static NTSTATUS create_logical_proc_info(SYSTEM_LOGICAL_PROCESSOR_INFORMATION **
 }
 #endif
 
+static inline void copy_smbios_string(char **buffer, char *string, size_t string_size)
+{
+    if (!string) return;
+    strcpy(*buffer, string);
+    *buffer += string_size;
+}
+
+#ifdef linux
+static char* get_smbios_string(const char *path, size_t *string_size)
+{
+    FILE *file = fopen(path, "r");
+    char *ret = NULL;
+    *string_size = 0;
+    if (file)
+    {
+        *string_size = getline(&ret, string_size, file) + 1;
+        fclose(file);
+        if (*string_size >= 2 && ret[*string_size - 2] == '\n')
+        {
+            ret[*string_size - 2] = 0;
+            (*string_size)--;
+        }
+        if (*string_size == 1)
+        {
+            free(ret);
+            ret = NULL;
+            *string_size = 0;
+        }
+    }
+    return ret;
+}
+#endif
+
 /******************************************************************************
  * NtQuerySystemInformation [NTDLL.@]
  * ZwQuerySystemInformation [NTDLL.@]
@@ -2359,6 +2436,97 @@ NTSTATUS WINAPI NtQuerySystemInformation(
             else ret = STATUS_INFO_LENGTH_MISMATCH;
         }
         break;
+    case SystemFirmwareTableInformation:
+        {
+            SYSTEM_FIRMWARE_TABLE_INFORMATION *sfti = (SYSTEM_FIRMWARE_TABLE_INFORMATION*)SystemInformation;
+            len = FIELD_OFFSET(SYSTEM_FIRMWARE_TABLE_INFORMATION, TableBuffer);
+            if (Length < len)
+            {
+                ret = STATUS_INFO_LENGTH_MISMATCH;
+                break;
+            }
+
+            switch (sfti->Action)
+            {
+            case SystemFirmwareTable_Get:
+                switch (sfti->ProviderSignature)
+                {
+                case RSMB:
+#ifdef linux
+                    {
+                        char *bios_vendor, *bios_version, *bios_date;
+                        size_t bios_vendor_size, bios_version_size, bios_date_size;
+                        char *buffer = (char*)sfti->TableBuffer;
+                        BYTE string_count;
+                        struct smbios_prologue *prologue;
+                        struct smbios_bios *bios;
+
+                        bios_vendor = get_smbios_string("/sys/class/dmi/id/bios_vendor", &bios_vendor_size);
+                        bios_version = get_smbios_string("/sys/class/dmi/id/bios_version", &bios_version_size);
+                        bios_date = get_smbios_string("/sys/class/dmi/id/bios_date", &bios_date_size);
+
+                        len = FIELD_OFFSET(SYSTEM_FIRMWARE_TABLE_INFORMATION, TableBuffer);
+
+                        len += sizeof(struct smbios_prologue);
+
+                        len += sizeof(struct smbios_bios);
+                        len += max(bios_vendor_size + bios_version_size + bios_date_size + 1, 2);
+
+                        if (Length < len)
+                        {
+                            ret = STATUS_INFO_LENGTH_MISMATCH;
+                            goto finish;
+                        }
+
+                        sfti->TableBufferLength = len - FIELD_OFFSET(SYSTEM_FIRMWARE_TABLE_INFORMATION, TableBuffer);
+
+                        prologue = (struct smbios_prologue*)buffer;
+                        prologue->calling_method = 0;
+                        prologue->major_version = 2;
+                        prologue->minor_version = 0;
+                        prologue->revision = 0;
+                        prologue->length = sfti->TableBufferLength - sizeof(struct smbios_prologue);
+                        buffer += sizeof(struct smbios_prologue);
+
+                        string_count = 0;
+                        bios = (struct smbios_bios*)buffer;
+                        bios->type = 0;
+                        bios->length = sizeof(struct smbios_bios);
+                        bios->handle = 0;
+                        bios->vendor = bios_vendor ? ++string_count : 0;
+                        bios->version = bios_version ? ++string_count : 0;
+                        bios->start = 0;
+                        bios->date = bios_date ? ++string_count : 0;
+                        bios->size = 0;
+                        bios->characteristics = 0x4; /* not supported */
+                        buffer += sizeof(struct smbios_bios);
+
+                        if (string_count)
+                        {
+                            copy_smbios_string(&buffer, bios_vendor, bios_vendor_size);
+                            copy_smbios_string(&buffer, bios_version, bios_version_size);
+                            copy_smbios_string(&buffer, bios_date, bios_date_size);
+                            memset(buffer, 0, 1);
+                        }
+                        else
+                        {
+                            memset(buffer, 0, 2);
+                        }
+
+                        free(bios_vendor);
+                        free(bios_version);
+                        free(bios_date);
+                    }
+                    goto finish;
+#endif
+                }
+                /* fall through */
+            default:
+                ret = STATUS_NOT_IMPLEMENTED;
+                FIXME("info_class SYSTEM_FIRMWARE_TABLE_INFORMATION\n");
+            }
+        }
+        break;
     default:
 	FIXME("(0x%08x,%p,0x%08x,%p) stub\n",
 	      SystemInformationClass,SystemInformation,Length,ResultLength);
@@ -2370,6 +2538,7 @@ NTSTATUS WINAPI NtQuerySystemInformation(
         ret = STATUS_INVALID_INFO_CLASS;
     }
 
+finish:
     if (ResultLength) *ResultLength = len;
 
     return ret;
-- 
2.17.1