From e44dd0f8f2bdcbb113036339479ef3a66364972e Mon Sep 17 00:00:00 2001
From: Alexandre Julliard <julliard@winehq.org>
Date: Mon, 21 Apr 2008 15:39:55 +0200
Subject: [PATCH] wineboot: Automatically update the prefix directory if
 wine.inf changes.

---
 dlls/ntdll/server.c          | 31 +--------------------
 programs/wineboot/wineboot.c | 54 ++++++++++++++++++++++++++++++++++--
 2 files changed, 52 insertions(+), 33 deletions(-)

diff --git a/dlls/ntdll/server.c b/dlls/ntdll/server.c
index c38a0239e7f..3d8beee730d 100644
--- a/dlls/ntdll/server.c
+++ b/dlls/ntdll/server.c
@@ -737,7 +737,6 @@ static void start_server(void)
 static void setup_config_dir(void)
 {
     const char *p, *config_dir = wine_get_config_dir();
-    pid_t pid, wret;
 
     if (chdir( config_dir ) == -1)
     {
@@ -759,7 +758,7 @@ static void setup_config_dir(void)
 
         mkdir( config_dir, 0777 );
         if (chdir( config_dir ) == -1) fatal_perror( "chdir to %s\n", config_dir );
-        MESSAGE( "wine: creating configuration directory '%s'...\n", config_dir );
+        MESSAGE( "wine: created the configuration directory '%s'\n", config_dir );
     }
 
     if (mkdir( "dosdevices", 0777 ) == -1)
@@ -773,34 +772,6 @@ static void setup_config_dir(void)
     mkdir( "drive_c", 0777 );
     symlink( "../drive_c", "dosdevices/c:" );
     symlink( "/", "dosdevices/z:" );
-
-    pid = fork();
-    if (pid == -1) fatal_perror( "fork" );
-
-    if (!pid)
-    {
-        char *argv[3];
-        static char argv0[] = "tools/wineprefixcreate",
-                    argv1[] = "--quiet";
-
-        argv[0] = argv0;
-        argv[1] = argv1;
-        argv[2] = NULL;
-        wine_exec_wine_binary( argv[0], argv, NULL );
-        fatal_perror( "could not exec wineprefixcreate" );
-    }
-    else
-    {
-        int status;
-
-        while ((wret = waitpid( pid, &status, 0 )) != pid)
-        {
-            if (wret == -1 && errno != EINTR) fatal_perror( "wait4" );
-        }
-        if (!WIFEXITED(status) || WEXITSTATUS(status))
-            fatal_error( "wineprefixcreate failed while creating '%s'.\n", config_dir );
-    }
-    MESSAGE( "wine: '%s' created successfully.\n", config_dir );
 }
 
 
diff --git a/programs/wineboot/wineboot.c b/programs/wineboot/wineboot.c
index bf1632b5132..a682e65c13e 100644
--- a/programs/wineboot/wineboot.c
+++ b/programs/wineboot/wineboot.c
@@ -58,6 +58,7 @@
 #define WIN32_LEAN_AND_MEAN
 
 #include <errno.h>
+#include <fcntl.h>
 #include <stdio.h>
 #include <stdlib.h>
 #ifdef HAVE_GETOPT_H
@@ -113,6 +114,49 @@ static char *get_wine_inf_path(void)
     return name;
 }
 
+/* update the timestamp if different from the reference time */
+static BOOL update_timestamp( const char *config_dir, unsigned long timestamp, int force )
+{
+    BOOL ret = FALSE;
+    int fd, count;
+    char buffer[100];
+    char *file = HeapAlloc( GetProcessHeap(), 0, strlen(config_dir) + sizeof("/.update-timestamp") );
+
+    if (!file) return FALSE;
+    strcpy( file, config_dir );
+    strcat( file, "/.update-timestamp" );
+
+    if ((fd = open( file, O_RDWR )) != -1)
+    {
+        if ((count = read( fd, buffer, sizeof(buffer) - 1 )) >= 0)
+        {
+            buffer[count] = 0;
+            if (!strncmp( buffer, "disable", sizeof("disable")-1 )) goto done;
+            if (!force && timestamp == strtoul( buffer, NULL, 10 )) goto done;
+        }
+        lseek( fd, 0, SEEK_SET );
+        ftruncate( fd, 0 );
+    }
+    else
+    {
+        if (errno != ENOENT) goto done;
+        if ((fd = open( file, O_WRONLY | O_CREAT | O_TRUNC, 0666 )) == -1) goto done;
+    }
+
+    count = sprintf( buffer, "%lu\n", timestamp );
+    if (write( fd, buffer, count ) != count)
+    {
+        WINE_WARN( "failed to update timestamp in %s\n", file );
+        ftruncate( fd, 0 );
+    }
+    else ret = TRUE;
+
+done:
+    if (fd != -1) close( fd );
+    HeapFree( GetProcessHeap(), 0, file );
+    return ret;
+}
+
 /* Performs the rename operations dictated in %SystemRoot%\Wininit.ini.
  * Returns FALSE if there was an error, or otherwise if all is ok.
  */
@@ -617,7 +661,7 @@ static BOOL start_services_process(void)
 }
 
 /* execute rundll32 on the wine.inf file if necessary */
-static void update_wineprefix(void)
+static void update_wineprefix( int force )
 {
     static const WCHAR cmdlineW[] = {'D','e','f','a','u','l','t','I','n','s','t','a','l','l',' ',
                                      '1','2','8',' ','\\','\\','?','\\','u','n','i','x' };
@@ -632,8 +676,12 @@ static void update_wineprefix(void)
         return;
     }
     if (stat( inf_path, &st ) == -1)
+    {
         WINE_WARN( "cannot stat wine.inf file: %s\n", strerror(errno) );
-    else
+        goto done;
+    }
+
+    if (update_timestamp( config_dir, st.st_mtime, force ))
     {
         WCHAR *buffer;
         DWORD len = MultiByteToWideChar( CP_UNIXCP, 0, inf_path, -1, NULL, 0 );
@@ -823,7 +871,7 @@ int main( int argc, char *argv[] )
         ProcessRunKeys( HKEY_LOCAL_MACHINE, runkeys_names[RUNKEY_RUNSERVICES], FALSE, FALSE );
         start_services_process();
     }
-    if (update) update_wineprefix();
+    if (init || update) update_wineprefix( update );
 
     ProcessRunKeys( HKEY_LOCAL_MACHINE, runkeys_names[RUNKEY_RUNONCE], TRUE, TRUE );