[SETUPLIB][USETUP] Use NT RTL String Safe functions instead of Win32-oriented ones (and remove usage of HRESULT too).

Based on:
svn path=/branches/setup_improvements/; revision=75755
svn path=/branches/setup_improvements/; revision=75757

base/setup/lib/bootsup.c

···
 
 #if 0
     WCHAR DestinationDevicePathBuffer[MAX_PATH];
-    StringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
-            L"\\Device\\Harddisk%d\\Partition0",
-            DiskNumber);
+    RtlStringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
+                        L"\\Device\\Harddisk%d\\Partition0",
+                        DiskNumber);
 #endif
 
     CombinePaths(SourceMbrPathBuffer, ARRAYSIZE(SourceMbrPathBuffer), 2,

base/setup/lib/mui.c

···
     {
         if (uIndex > 19) break;
 
-        swprintf(szValueName, L"%u", uIndex + 1);
+        RtlStringCchPrintfW(szValueName, ARRAYSIZE(szValueName), L"%u", uIndex + 1);
         RtlInitUnicodeString(&ValueName, szValueName);
 
-        swprintf(szLangID, L"0000%s", MuiLayouts[uIndex].LangID);
+        RtlStringCchPrintfW(szLangID, ARRAYSIZE(szLangID), L"0000%s", MuiLayouts[uIndex].LangID);
 
         if (_wcsicmp(szLangID, MuiLayouts[uIndex].LayoutID) == 0)
         {
···
         }
         else
         {
-            swprintf(szLangID, L"d%03lu%s", uCount, MuiLayouts[uIndex].LangID);
+            RtlStringCchPrintfW(szLangID, ARRAYSIZE(szLangID), L"d%03lu%s", uCount, MuiLayouts[uIndex].LangID);
             Status = NtSetValueKey(KeyHandle,
                                    &ValueName,
                                    0,

base/setup/lib/registry.c

···
 
 #include "registry.h"
 
-// HACK!
-#include <strsafe.h>
-
 #define NDEBUG
 #include <debug.h>
 
···
 
         CombinePaths(SrcPath, ARRAYSIZE(SrcPath), 3,
                      InstallPath->Buffer, L"System32\\config", RegistryHives[i].HiveName);
-        StringCchCopyW(DstPath, ARRAYSIZE(DstPath), SrcPath);
-        StringCchCatW(DstPath, ARRAYSIZE(DstPath), L".sav");
+        RtlStringCchCopyW(DstPath, ARRAYSIZE(DstPath), SrcPath);
+        RtlStringCchCatW(DstPath, ARRAYSIZE(DstPath), L".sav");
 
         DPRINT1("Copy hive: %S ==> %S\n", SrcPath, DstPath);
         Status = SetupCopyFile(SrcPath, DstPath, FALSE);

base/setup/lib/settings.c

···
     }
 
     wcscpy((PWCHAR)*UserData, KeyName);
-    wcscpy(DisplayText, KeyValue);
+    RtlStringCbCopyW(DisplayText, DisplayTextSize, KeyValue);
 
     *Current = (CompareKey ? !_wcsicmp(KeyName, CompareKey) : FALSE);
 
···
         }
 
         DPRINT("Computer key: %S\n", KeyName);
-        wcscpy(ComputerKey, KeyName);
+        RtlStringCchCopyW(ComputerKey, ARRAYSIZE(ComputerKey), KeyName);
         INF_FreeData(KeyName);
     } while (SetupFindNextLine(&Context, &Context));
 
···
     BusInstance = 0;
     while (TRUE)
     {
-        swprintf(Buffer, L"%lu", BusInstance);
+        RtlStringCchPrintfW(Buffer, ARRAYSIZE(Buffer), L"%lu", BusInstance);
         RtlInitUnicodeString(&KeyName, Buffer);
         InitializeObjectAttributes(&ObjectAttributes,
                                    &KeyName,
···
             while (TRUE)
             {
                 /* Open the pointer controller instance key */
-                swprintf(Buffer, L"%lu", ControllerInstance);
+                RtlStringCchPrintfW(Buffer, ARRAYSIZE(Buffer), L"%lu", ControllerInstance);
                 RtlInitUnicodeString(&KeyName, Buffer);
                 InitializeObjectAttributes(&ObjectAttributes,
                                            &KeyName,
···
         }
 
         DPRINT("Display key: %S\n", KeyName);
-        wcscpy(DisplayKey, KeyName);
+        RtlStringCchCopyW(DisplayKey, ARRAYSIZE(DisplayKey), KeyName);
         INF_FreeData(KeyName);
     } while (SetupFindNextLine(&Context, &Context));
 
···
         return FALSE;
     }
 
-    wcscpy(SectionName, L"Files.");
-    wcscat(SectionName, (const wchar_t*)GetListEntryUserData(Entry));
+    RtlStringCchPrintfW(SectionName, ARRAYSIZE(SectionName),
+                        L"Files.%s", (PCWSTR)GetListEntryUserData(Entry));
     *AdditionalSectionName = SectionName;
 
     return TRUE;
···
     ASSERT(wcslen(ServiceName) < 10);
     DPRINT1("Service name: '%S'\n", ServiceName);
 
-    swprintf(RegPath, L"System\\CurrentControlSet\\Services\\%s", ServiceName);
+    RtlStringCchPrintfW(RegPath, ARRAYSIZE(RegPath),
+                        L"System\\CurrentControlSet\\Services\\%s",
+                        ServiceName);
     RtlInitUnicodeString(&KeyName, RegPath);
     InitializeObjectAttributes(&ObjectAttributes,
                                &KeyName,
···
         return FALSE;
     }
 
-    swprintf(RegPath,
-             L"System\\CurrentControlSet\\Hardware Profiles\\Current\\System\\CurrentControlSet\\Services\\%s\\Device0",
-             ServiceName);
+    RtlStringCchPrintfW(RegPath, ARRAYSIZE(RegPath),
+                        L"System\\CurrentControlSet\\Hardware Profiles\\Current\\System\\CurrentControlSet\\Services\\%s\\Device0",
+                        ServiceName);
     DPRINT1("RegPath: '%S'\n", RegPath);
     RtlInitUnicodeString(&KeyName, RegPath);
     InitializeObjectAttributes(&ObjectAttributes,
···
     }
 
     wcscpy((PWCHAR)*UserData, KeyName);
-    wcscpy(DisplayText, KeyValue);
+    RtlStringCbCopyW(DisplayText, DisplayTextSize, KeyValue);
 
     *Current = FALSE;
 

base/setup/lib/setuplib.c

···
 
 #include "setuplib.h"
 
-// HACK!
-#include <strsafe.h>
-
 #define NDEBUG
 #include <debug.h>
 
···
             DPRINT("INF_GetData() failed for key 'InstallationDirectory'\n");
             goto Quit;
         }
-        wcscpy(pSetupData->InstallationDirectory, Value);
+
+        RtlStringCchCopyW(pSetupData->InstallationDirectory,
+                          ARRAYSIZE(pSetupData->InstallationDirectory),
+                          Value);
+
         INF_FreeData(Value);
     }
 
···
         if (INF_GetData(&Context, NULL, &Value))
         {
             LONG Id = wcstol(Value, NULL, 16);
-            swprintf(pSetupData->LocaleID, L"%08lx", Id);
+            RtlStringCchPrintfW(pSetupData->LocaleID,
+                                ARRAYSIZE(pSetupData->LocaleID),
+                                L"%08lx", Id);
             INF_FreeData(Value);
        }
     }
···
     if (IniSection)
     {
         /* Key "skipmissingfiles" */
-        // StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
-                         // L"\"%s\"", L"WinNt5.2");
+        // RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+                            // L"\"%s\"", L"WinNt5.2");
         // IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
                           // L"Version", PathBuffer);
     }
···
     IniSection = IniCacheAppendSection(IniCache, L"Data");
     if (IniSection)
     {
-        StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
-                         L"\"%s\"", IsUnattendedSetup ? L"yes" : L"no");
+        RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+                            L"\"%s\"", IsUnattendedSetup ? L"yes" : L"no");
         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
                           L"UnattendedInstall", PathBuffer);
 
         // "floppylessbootpath" (yes/no)
 
-        StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
-                         L"\"%s\"", L"winnt");
+        RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+                            L"\"%s\"", L"winnt");
         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
                           L"ProductType", PathBuffer);
 
-        StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
-                         L"\"%s\\\"", pSetupData->SourceRootPath.Buffer);
+        RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+                            L"\"%s\\\"", pSetupData->SourceRootPath.Buffer);
         IniCacheInsertKey(IniSection, NULL, INSERT_LAST,
                           L"SourcePath", PathBuffer);
 
···
         if (!INF_GetData(&Context, NULL, &Value))
             return ERROR_CORRUPT_TXTSETUPSIF;
 
-        wcscpy(pSetupData->InstallationDirectory, Value);
+        RtlStringCchCopyW(pSetupData->InstallationDirectory,
+                          ARRAYSIZE(pSetupData->InstallationDirectory),
+                          Value);
+
         INF_FreeData(Value);
     }
 

base/setup/lib/utils/arcname_tests.c

···
 
 #include <ntstatus.h>
 
-#include <strsafe.h>
-
 #include "arcname.h"
 
 #define OBJ_NAME_PATH_SEPARATOR ((WCHAR)L'\\')

base/setup/lib/utils/genlist.c

···
 {
     PGENERIC_LIST List;
 
-    List = (PGENERIC_LIST)RtlAllocateHeap(ProcessHeap,
-                                          0,
-                                          sizeof(GENERIC_LIST));
+    List = RtlAllocateHeap(ProcessHeap, 0, sizeof(GENERIC_LIST));
     if (List == NULL)
         return NULL;
 
···
     IN BOOLEAN Current)
 {
     PGENERIC_LIST_ENTRY Entry;
+    SIZE_T TextSize;
 
-    Entry = (PGENERIC_LIST_ENTRY)RtlAllocateHeap(ProcessHeap,
-                                                 0,
-                                                 sizeof(GENERIC_LIST_ENTRY) +
-                                                    (wcslen(Text) + 1) * sizeof(WCHAR));
+    TextSize = (wcslen(Text) + 1) * sizeof(WCHAR);
+    Entry = RtlAllocateHeap(ProcessHeap, 0,
+                            sizeof(GENERIC_LIST_ENTRY) + TextSize);
     if (Entry == NULL)
         return FALSE;
 
-    wcscpy(Entry->Text, Text);
+    RtlStringCbCopyW(Entry->Text, TextSize, Text);
     Entry->List = List;
     Entry->UserData = UserData;
 

base/setup/lib/utils/regutil.c

···
 
 #include "regutil.h"
 
-// HACK!
-#include <strsafe.h>
-
 #define NDEBUG
 #include <debug.h>
 
···
 
         // i = 1;
         /* Try first by just appending the '.old' extension */
-        StringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2), L"%s.%s", PathBuffer, Extension);
+        RtlStringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2),
+                            L"%s.%s", PathBuffer, Extension);
 #if 0
         while (DoesFileExist(NULL, PathBuffer2))
         {
···
             if (i <= 0xFFFF)
             {
                 /* Append '_N.old' extension */
-                StringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2), L"%s_%lu.%s", PathBuffer, i, Extension);
+                RtlStringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2),
+                                    L"%s_%lu.%s", PathBuffer, i, Extension);
                 ++i;
             }
             else
···
                  * Too many old files exist, we will rename the file
                  * using the name of the oldest one.
                  */
-                StringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2), L"%s.%s", PathBuffer, Extension);
+                RtlStringCchPrintfW(PathBuffer2, ARRAYSIZE(PathBuffer2),
+                                    L"%s.%s", PathBuffer, Extension);
                 break;
             }
         }

base/setup/usetup/usetup.c

···
 #define NDEBUG
 #include <debug.h>
 
-// HACK!
-#include <strsafe.h>
-
 
 /* GLOBALS & LOCALS *********************************************************/
 
···
                 return SELECT_PARTITION_PAGE;
             }
 
-            StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+            RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
                     L"\\Device\\Harddisk%lu\\Partition%lu\\",
                     PartitionList->CurrentDisk->DiskNumber,
                     PartitionList->CurrentPartition->PartitionNumber);
···
             }
 
             /* Set PartitionRootPath */
-            StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+            RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
                     L"\\Device\\Harddisk%lu\\Partition%lu",
                     DiskEntry->DiskNumber,
                     PartEntry->PartitionNumber);
···
     }
 
     /* Set PartitionRootPath */
-    StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+    RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
             L"\\Device\\Harddisk%lu\\Partition%lu",
             DiskEntry->DiskNumber,
             PartEntry->PartitionNumber);
···
 
     /* Create 'USetupData.DestinationRootPath' string */
     RtlFreeUnicodeString(&USetupData.DestinationRootPath);
-    StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+    RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
             L"\\Device\\Harddisk%lu\\Partition%lu\\",
             DiskEntry->DiskNumber,
             PartEntry->PartitionNumber);
···
 /** Equivalent of 'NTOS_INSTALLATION::SystemArcPath' **/
     /* Create 'USetupData.DestinationArcPath' */
     RtlFreeUnicodeString(&USetupData.DestinationArcPath);
-    StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+    RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
             L"multi(0)disk(0)rdisk(%lu)partition(%lu)\\",
             DiskEntry->BiosDiskNumber,
             PartEntry->PartitionNumber);
···
             /* Installation path */
             DPRINT("InstallationPath: '%S'\n", DirKeyValue);
 
-            StringCchCopyW(CompleteOrigDirName, ARRAYSIZE(CompleteOrigDirName),
-                           USetupData.SourceRootDir.Buffer);
+            RtlStringCchCopyW(CompleteOrigDirName, ARRAYSIZE(CompleteOrigDirName),
+                              USetupData.SourceRootDir.Buffer);
 
             DPRINT("InstallationPath(2): '%S'\n", CompleteOrigDirName);
         }
···
             /* Absolute path */
             DPRINT("AbsolutePath: '%S'\n", DirKeyValue);
 
-            StringCchCopyW(CompleteOrigDirName, ARRAYSIZE(CompleteOrigDirName),
-                           DirKeyValue);
+            RtlStringCchCopyW(CompleteOrigDirName, ARRAYSIZE(CompleteOrigDirName),
+                              DirKeyValue);
 
             DPRINT("AbsolutePath(2): '%S'\n", CompleteOrigDirName);
         }
···
      */
 
     /* Get destination path */
-    StringCchCopyW(PathBuffer, ARRAYSIZE(PathBuffer), USetupData.DestinationPath.Buffer);
+    RtlStringCchCopyW(PathBuffer, ARRAYSIZE(PathBuffer), USetupData.DestinationPath.Buffer);
 
     DPRINT("FullPath(1): '%S'\n", PathBuffer);
 
···
             /* Installation path */
             DPRINT("InstallationPath: '%S'\n", DirKeyValue);
 
-            StringCchCopyW(PathBuffer, ARRAYSIZE(PathBuffer),
-                           USetupData.DestinationPath.Buffer);
+            RtlStringCchCopyW(PathBuffer, ARRAYSIZE(PathBuffer),
+                              USetupData.DestinationPath.Buffer);
 
             DPRINT("InstallationPath(2): '%S'\n", PathBuffer);
         }
···
     CONSOLE_SetStatusText(MUIGetString(STRING_PLEASEWAIT));
 
     RtlFreeUnicodeString(&USetupData.SystemRootPath);
-    StringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
+    RtlStringCchPrintfW(PathBuffer, ARRAYSIZE(PathBuffer),
             L"\\Device\\Harddisk%lu\\Partition%lu\\",
             PartitionList->SystemPartition->DiskEntry->DiskNumber,
             PartitionList->SystemPartition->PartitionNumber);
···
     }
 
     /* Step 2: Write the MBR */
-    StringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
+    RtlStringCchPrintfW(DestinationDevicePathBuffer, ARRAYSIZE(DestinationDevicePathBuffer),
             L"\\Device\\Harddisk%d\\Partition0",
             PartitionList->SystemPartition->DiskEntry->DiskNumber);
     Status = InstallMbrBootCodeToDisk(&USetupData.SystemRootPath,

base/setup/usetup/usetup.h

···
 
 #include <ntstrsafe.h>
 
+
 /* Setup library headers */
 #include <reactos/rosioctl.h>
 #include <../lib/setuplib.h>