[DISKPART] Implement the gpt command keep the partition number up to date

Author: EricKohl

base/system/diskpart/gpt.c

@@ -8,7 +8,72 @@
 
 #include "diskpart.h"
 
-BOOL gpt_main(INT argc, LPWSTR *argv)
+#define NDEBUG
+#include <debug.h>
+
+
+BOOL
+gpt_main(
+    INT argc,
+    LPWSTR *argv)
 {
+    ULONGLONG ullAttributes = 0ULL;
+    PWSTR pszSuffix = NULL;
+    INT i;
+    NTSTATUS Status;
+
+    if (CurrentDisk == NULL)
+    {
+        ConResPuts(StdOut, IDS_SELECT_NO_DISK);
+        return TRUE;
+    }
+
+    if (CurrentPartition == NULL)
+    {
+        ConResPuts(StdOut, IDS_SELECT_NO_PARTITION);
+        return TRUE;
+    }
+
+    if (CurrentDisk->PartitionStyle != PARTITION_STYLE_GPT)
+    {
+        ConResPuts(StdOut, IDS_CREATE_PARTITION_INVALID_STYLE);
+        return TRUE;
+    }
+
+    for (i = 1; i < argc; i++)
+    {
+        if (HasPrefix(argv[i], L"attributes=", &pszSuffix))
+        {
+            /* attributes=<N> */
+            DPRINT("Attributes: %s\n", pszSuffix);
+
+            ullAttributes = _wcstoui64(pszSuffix, NULL, 0);
+            if ((ullAttributes == 0) && (errno == ERANGE))
+            {
+                ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
+                return TRUE;
+            }
+        }
+       else
+        {
+            ConResPuts(StdErr, IDS_ERROR_INVALID_ARGS);
+            return TRUE;
+        }
+    }
+
+    DPRINT("Attributes: 0x%I64x\n", ullAttributes);
+    CurrentPartition->Gpt.Attributes = ullAttributes;
+    CurrentDisk->Dirty = TRUE;
+
+    UpdateGptDiskLayout(CurrentDisk, FALSE);
+    Status = WriteGptPartitions(CurrentDisk);
+    if (!NT_SUCCESS(Status))
+    {
+        ConResPuts(StdOut, IDS_GPT_FAIL);
+        return TRUE;
+    }
+
+    ConResPuts(StdOut, IDS_GPT_SUCCESS);
+
     return TRUE;
 }

base/system/diskpart/lang/de-DE.rc

@@ -89,6 +89,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Größe der Zuordnungseinheit: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/en-US.rc

@@ -89,6 +89,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Cluster size: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/it-IT.rc

@@ -96,6 +96,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Dimensione cluster: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/pl-PL.rc

@@ -89,6 +89,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Rozmiar klastra: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/pt-PT.rc

@@ -91,6 +91,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Cluster size: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/ro-RO.rc

@@ -97,6 +97,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Mărimea cluster-ului: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/ru-RU.rc

@@ -91,6 +91,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Cluster size: %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/sq-AL.rc

@@ -93,6 +93,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Cluster size:  %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"

base/system/diskpart/lang/tr-TR.rc

@@ -99,6 +99,12 @@ BEGIN
     IDS_FILESYSTEMS_CLUSTERSIZE "Küme boyutu :  %lu\n"
 END
 
+STRINGTABLE
+BEGIN
+    IDS_GPT_FAIL "\nDiskPart failed to assign atttributes to the selected GPT partition.\n"
+    IDS_GPT_SUCCESS "\nDiskPart successfully assigned the attributes the selected GPT partition.\n"
+END
+
 STRINGTABLE
 BEGIN
     IDS_HELP_FORMAT_STRING "%-11.11s - %s"