src/Arch.c :config --default-user :run id command in root

yuk7 · Nov 18, 2017 · de70f1b · de70f1b
1 parent 803bf22
commit de70f1b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Arch.c b/src/Arch.c
@@ -92,6 +92,7 @@ int main(int argc,char *argv[])
             {
                 if(wcscmp(wargv[2],L"--default-user") == 0)
                 {
+                    (void) ConfigureDistribution(TargetName,0,distributionFlags); //set default uid to 0(root)
                     FILE *fp;
                     unsigned long uid;
                     wchar_t wcmd[300] = L"wsl.exe ";
@@ -112,13 +113,15 @@ int main(int argc,char *argv[])
                         res = ConfigureDistribution(TargetName,uid,distributionFlags);
                         if(res != 0)
                         {
+                            (void) ConfigureDistribution(TargetName,defaultUID,distributionFlags); //revert uid
                             fwprintf(stderr,L"ERROR:Configure Failed! 0x%x",res);
                             return 1;
                         }
                         return 0;
                     }
                     else
                     {
+                        (void) ConfigureDistribution(TargetName,defaultUID,distributionFlags); //revert uid
                         wprintf(L"\n");
                         fwprintf(stderr,L"ERROR:Invalid Argument.\nFailed to detect user.");
                     }