Alexandre Julliard : server: Allow the main exe to be registered as a normal dll before

[Alexandre Julliard]

Module: wine
Branch: refs/heads/master
Commit: 2b0033d5ad3bf651cf259d49895d50671322ee46
URL:    http://source.winehq.org/git/?p=wine.git;a=commit;h=2b0033d5ad3bf651cf259d49895d50671322ee46

Author: Alexandre Julliard <julliard at winehq.org>
Date:   Wed Feb  8 15:07:16 2006 +0100

server: Allow the main exe to be registered as a normal dll before
init_process_done is called.

---

 server/process.c |   29 +++++++++++++++++++++--------
 1 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/server/process.c b/server/process.c
index 7771c59..65b9cb0 100644
--- a/server/process.c
+++ b/server/process.c
@@ -980,7 +980,7 @@ DECL_HANDLER(get_startup_info)
 /* signal the end of the process initialization */
 DECL_HANDLER(init_process_done)
 {
-    struct file *file = NULL;
+    struct process_dll *dll;
     struct process *process = current->process;
 
     if (is_process_init_done(process))
@@ -993,16 +993,29 @@ DECL_HANDLER(init_process_done)
         fatal_protocol_error( current, "init_process_done: module base address cannot be 0\n" );
         return;
     }
-    process->exe.base = req->module;
-    process->exe.size = req->module_size;
-    process->exe.name = req->name;
 
-    if (req->exe_file) file = get_file_obj( process, req->exe_file, FILE_READ_DATA );
     if (process->exe.file) release_object( process->exe.file );
-    process->exe.file = file;
 
-    if ((process->exe.namelen = get_req_data_size()))
-        process->exe.filename = memdup( get_req_data(), process->exe.namelen );
+    /* check if main exe has been registered as a dll already */
+    if ((dll = find_process_dll( process, req->module )))
+    {
+        list_remove( &dll->entry );
+        memcpy( &process->exe, dll, sizeof(*dll) );
+        list_init( &process->exe.entry );
+        free( dll );
+    }
+    else
+    {
+        struct file *file = NULL;
+
+        if (req->exe_file) file = get_file_obj( process, req->exe_file, FILE_READ_DATA );
+        process->exe.base = req->module;
+        process->exe.size = req->module_size;
+        process->exe.name = req->name;
+        process->exe.file = file;
+        if ((process->exe.namelen = get_req_data_size()))
+            process->exe.filename = memdup( get_req_data(), process->exe.namelen );
+    }
 
     generate_startup_debug_events( process, req->entry );
     set_process_startup_state( process, STARTUP_DONE );