2 * Server-side process management
4 * Copyright (C) 1998 Alexandre Julliard
17 #ifdef HAVE_SYS_SOCKET_H
18 # include <sys/socket.h>
30 /* process structure */
32 static struct process
*first_process
;
33 static int running_processes
;
35 /* process operations */
37 static void process_dump( struct object
*obj
, int verbose
);
38 static int process_signaled( struct object
*obj
, struct thread
*thread
);
39 static void process_destroy( struct object
*obj
);
41 static const struct object_ops process_ops
=
43 sizeof(struct process
), /* size */
44 process_dump
, /* dump */
45 add_queue
, /* add_queue */
46 remove_queue
, /* remove_queue */
47 process_signaled
, /* signaled */
48 no_satisfied
, /* satisfied */
49 NULL
, /* get_poll_events */
50 NULL
, /* poll_event */
51 no_read_fd
, /* get_read_fd */
52 no_write_fd
, /* get_write_fd */
54 no_get_file_info
, /* get_file_info */
55 process_destroy
/* destroy */
58 /* set the process creation info */
59 static int set_creation_info( struct process
*process
, struct new_process_request
*req
,
60 const char *cmd_line
, size_t len
)
62 if (!(process
->info
= mem_alloc( sizeof(*process
->info
) + len
))) return 0;
65 /* copy the request structure */
66 memcpy( process
->info
, req
, sizeof(*req
) );
68 else /* no request, use defaults */
73 req
->create_flags
= CREATE_NEW_CONSOLE
;
74 req
->start_flags
= STARTF_USESTDHANDLES
;
83 memcpy( process
->info
->cmdline
, cmd_line
, len
);
84 process
->info
->cmdline
[len
] = 0;
85 process
->create_flags
= process
->info
->create_flags
;
89 /* set the console and stdio handles for a newly created process */
90 static int set_process_console( struct process
*process
, struct process
*parent
)
92 struct new_process_request
*info
= process
->info
;
94 if (process
->create_flags
& CREATE_NEW_CONSOLE
)
96 if (!alloc_console( process
)) return 0;
98 else if (!(process
->create_flags
& DETACHED_PROCESS
))
100 if (parent
->console_in
) process
->console_in
= grab_object( parent
->console_in
);
101 if (parent
->console_out
) process
->console_out
= grab_object( parent
->console_out
);
105 if (!info
->inherit_all
&& !(info
->start_flags
& STARTF_USESTDHANDLES
))
107 /* duplicate the handle from the parent into this process */
108 info
->hstdin
= duplicate_handle( parent
, info
->hstdin
, process
,
109 0, TRUE
, DUPLICATE_SAME_ACCESS
);
110 info
->hstdout
= duplicate_handle( parent
, info
->hstdout
, process
,
111 0, TRUE
, DUPLICATE_SAME_ACCESS
);
112 info
->hstderr
= duplicate_handle( parent
, info
->hstderr
, process
,
113 0, TRUE
, DUPLICATE_SAME_ACCESS
);
118 /* no parent, use handles to the console for stdio */
119 info
->hstdin
= alloc_handle( process
, process
->console_in
,
120 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
121 info
->hstdout
= alloc_handle( process
, process
->console_out
,
122 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
123 info
->hstderr
= alloc_handle( process
, process
->console_out
,
124 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
129 /* create a new process and its main thread */
130 struct thread
*create_process( int fd
, struct process
*parent
,
131 struct new_process_request
*req
,
132 const char *cmd_line
, size_t len
)
134 struct process
*process
;
135 struct thread
*thread
= NULL
;
137 if (!(process
= alloc_object( &process_ops
, -1 )))
142 process
->next
= NULL
;
143 process
->prev
= NULL
;
144 process
->thread_list
= NULL
;
145 process
->debugger
= NULL
;
146 process
->handles
= NULL
;
147 process
->exit_code
= STILL_ACTIVE
;
148 process
->running_threads
= 0;
149 process
->priority
= NORMAL_PRIORITY_CLASS
;
150 process
->affinity
= 1;
151 process
->suspend
= 0;
152 process
->create_flags
= 0;
153 process
->console_in
= NULL
;
154 process
->console_out
= NULL
;
155 process
->init_event
= NULL
;
156 process
->info
= NULL
;
157 process
->ldt_copy
= NULL
;
158 process
->ldt_flags
= NULL
;
159 process
->exe
.next
= NULL
;
160 process
->exe
.prev
= NULL
;
161 process
->exe
.file
= NULL
;
162 process
->exe
.dbg_offset
= 0;
163 process
->exe
.dbg_size
= 0;
164 gettimeofday( &process
->start_time
, NULL
);
165 if ((process
->next
= first_process
) != NULL
) process
->next
->prev
= process
;
166 first_process
= process
;
168 /* copy the request structure */
169 if (!set_creation_info( process
, req
, cmd_line
, len
)) goto error
;
171 if (process
->info
->inherit_all
)
172 process
->handles
= copy_handle_table( process
, parent
);
174 process
->handles
= alloc_handle_table( process
, 0 );
175 if (!process
->handles
) goto error
;
177 /* alloc a handle for the process itself */
178 alloc_handle( process
, process
, PROCESS_ALL_ACCESS
, 0 );
180 /* retrieve the main exe file */
181 if (process
->info
->exe_file
!= -1)
183 if (!(process
->exe
.file
= get_file_obj( parent
, process
->info
->exe_file
,
184 GENERIC_READ
))) goto error
;
185 process
->info
->exe_file
= -1;
188 /* get the init done event */
189 if (process
->info
->event
!= -1)
191 if (!(process
->init_event
= get_event_obj( parent
, process
->info
->event
,
192 EVENT_MODIFY_STATE
))) goto error
;
195 /* set the process console */
196 if (!set_process_console( process
, parent
)) goto error
;
198 /* create the main thread */
199 if (!(thread
= create_thread( fd
, process
, (process
->create_flags
& CREATE_SUSPENDED
) != 0)))
202 /* attach to the debugger if requested */
203 if (process
->create_flags
& (DEBUG_PROCESS
| DEBUG_ONLY_THIS_PROCESS
))
204 set_process_debugger( process
, current
);
205 else if (parent
&& parent
->debugger
&& !(parent
->create_flags
& DEBUG_ONLY_THIS_PROCESS
))
206 set_process_debugger( process
, parent
->debugger
);
208 release_object( process
);
213 free_console( process
);
214 if (process
->handles
) release_object( process
->handles
);
215 release_object( process
);
219 /* destroy a process when its refcount is 0 */
220 static void process_destroy( struct object
*obj
)
222 struct process
*process
= (struct process
*)obj
;
223 assert( obj
->ops
== &process_ops
);
225 /* we can't have a thread remaining */
226 assert( !process
->thread_list
);
227 if (process
->next
) process
->next
->prev
= process
->prev
;
228 if (process
->prev
) process
->prev
->next
= process
->next
;
229 else first_process
= process
->next
;
230 if (process
->info
) free( process
->info
);
231 if (process
->init_event
) release_object( process
->init_event
);
232 if (process
->exe
.file
) release_object( process
->exe
.file
);
235 /* dump a process on stdout for debugging purposes */
236 static void process_dump( struct object
*obj
, int verbose
)
238 struct process
*process
= (struct process
*)obj
;
239 assert( obj
->ops
== &process_ops
);
241 fprintf( stderr
, "Process next=%p prev=%p console=%p/%p handles=%p\n",
242 process
->next
, process
->prev
, process
->console_in
, process
->console_out
,
246 static int process_signaled( struct object
*obj
, struct thread
*thread
)
248 struct process
*process
= (struct process
*)obj
;
249 return !process
->running_threads
;
253 /* get a process from an id (and increment the refcount) */
254 struct process
*get_process_from_id( void *id
)
256 struct process
*p
= first_process
;
257 while (p
&& (p
!= id
)) p
= p
->next
;
258 if (p
) grab_object( p
);
259 else set_error( STATUS_INVALID_PARAMETER
);
263 /* get a process from a handle (and increment the refcount) */
264 struct process
*get_process_from_handle( int handle
, unsigned int access
)
266 return (struct process
*)get_handle_obj( current
->process
, handle
,
267 access
, &process_ops
);
270 /* add a dll to a process list */
271 static struct process_dll
*process_load_dll( struct process
*process
, struct file
*file
,
274 struct process_dll
*dll
;
276 /* make sure we don't already have one with the same base address */
277 for (dll
= process
->exe
.next
; dll
; dll
= dll
->next
) if (dll
->base
== base
)
279 set_error( STATUS_INVALID_PARAMETER
);
283 if ((dll
= mem_alloc( sizeof(*dll
) )))
285 dll
->prev
= &process
->exe
;
288 if (file
) dll
->file
= (struct file
*)grab_object( file
);
289 if ((dll
->next
= process
->exe
.next
)) dll
->next
->prev
= dll
;
290 process
->exe
.next
= dll
;
295 /* remove a dll from a process list */
296 static void process_unload_dll( struct process
*process
, void *base
)
298 struct process_dll
*dll
;
300 for (dll
= process
->exe
.next
; dll
; dll
= dll
->next
)
302 if (dll
->base
== base
)
304 if (dll
->file
) release_object( dll
->file
);
305 if (dll
->next
) dll
->next
->prev
= dll
->prev
;
306 if (dll
->prev
) dll
->prev
->next
= dll
->next
;
308 generate_debug_event( current
, UNLOAD_DLL_DEBUG_EVENT
, base
);
312 set_error( STATUS_INVALID_PARAMETER
);
315 /* a process has been killed (i.e. its last thread died) */
316 static void process_killed( struct process
*process
)
318 assert( !process
->thread_list
);
319 gettimeofday( &process
->end_time
, NULL
);
320 release_object( process
->handles
);
321 process
->handles
= NULL
;
322 free_console( process
);
323 while (process
->exe
.next
)
325 struct process_dll
*dll
= process
->exe
.next
;
326 process
->exe
.next
= dll
->next
;
327 if (dll
->file
) release_object( dll
->file
);
330 if (process
->exe
.file
) release_object( process
->exe
.file
);
331 process
->exe
.file
= NULL
;
332 wake_up( &process
->obj
, 0 );
333 if (!--running_processes
)
335 /* last process died, close global handles */
336 close_global_handles();
337 /* this will cause the select loop to terminate */
338 if (!persistent_server
) close_master_socket();
342 /* add a thread to a process running threads list */
343 void add_process_thread( struct process
*process
, struct thread
*thread
)
345 thread
->proc_next
= process
->thread_list
;
346 thread
->proc_prev
= NULL
;
347 if (thread
->proc_next
) thread
->proc_next
->proc_prev
= thread
;
348 process
->thread_list
= thread
;
349 if (!process
->running_threads
++) running_processes
++;
350 grab_object( thread
);
353 /* remove a thread from a process running threads list */
354 void remove_process_thread( struct process
*process
, struct thread
*thread
)
356 assert( process
->running_threads
> 0 );
357 assert( process
->thread_list
);
359 if (thread
->proc_next
) thread
->proc_next
->proc_prev
= thread
->proc_prev
;
360 if (thread
->proc_prev
) thread
->proc_prev
->proc_next
= thread
->proc_next
;
361 else process
->thread_list
= thread
->proc_next
;
363 if (!--process
->running_threads
)
365 /* we have removed the last running thread, exit the process */
366 process
->exit_code
= thread
->exit_code
;
367 generate_debug_event( thread
, EXIT_PROCESS_DEBUG_EVENT
, process
);
368 process_killed( process
);
370 else generate_debug_event( thread
, EXIT_THREAD_DEBUG_EVENT
, thread
);
371 release_object( thread
);
374 /* suspend all the threads of a process */
375 void suspend_process( struct process
*process
)
377 if (!process
->suspend
++)
379 struct thread
*thread
= process
->thread_list
;
380 for (; thread
; thread
= thread
->proc_next
)
382 if (!thread
->suspend
) stop_thread( thread
);
387 /* resume all the threads of a process */
388 void resume_process( struct process
*process
)
390 assert (process
->suspend
> 0);
391 if (!--process
->suspend
)
393 struct thread
*thread
= process
->thread_list
;
394 for (; thread
; thread
= thread
->proc_next
)
396 if (!thread
->suspend
) continue_thread( thread
);
401 /* kill a process on the spot */
402 void kill_process( struct process
*process
, int exit_code
)
404 while (process
->thread_list
)
405 kill_thread( process
->thread_list
, exit_code
);
408 /* kill all processes being debugged by a given thread */
409 void kill_debugged_processes( struct thread
*debugger
, int exit_code
)
411 for (;;) /* restart from the beginning of the list every time */
413 struct process
*process
= first_process
;
414 /* find the first process being debugged by 'debugger' and still running */
415 while (process
&& (process
->debugger
!= debugger
|| !process
->running_threads
))
416 process
= process
->next
;
417 if (!process
) return;
418 process
->debugger
= NULL
;
419 kill_process( process
, exit_code
);
423 /* get all information about a process */
424 static void get_process_info( struct process
*process
, struct get_process_info_request
*req
)
427 req
->debugged
= (process
->debugger
!= 0);
428 req
->exit_code
= process
->exit_code
;
429 req
->priority
= process
->priority
;
430 req
->process_affinity
= process
->affinity
;
431 req
->system_affinity
= 1;
434 /* set all information about a process */
435 static void set_process_info( struct process
*process
,
436 struct set_process_info_request
*req
)
438 if (req
->mask
& SET_PROCESS_INFO_PRIORITY
)
439 process
->priority
= req
->priority
;
440 if (req
->mask
& SET_PROCESS_INFO_AFFINITY
)
442 if (req
->affinity
!= 1) set_error( STATUS_INVALID_PARAMETER
);
443 else process
->affinity
= req
->affinity
;
447 /* read data from a process memory space */
448 /* len is the total size (in ints), max is the size we can actually store in the output buffer */
449 /* we read the total size in all cases to check for permissions */
450 static void read_process_memory( struct process
*process
, const int *addr
,
451 size_t len
, size_t max
, int *dest
)
453 struct thread
*thread
= process
->thread_list
;
455 if ((unsigned int)addr
% sizeof(int)) /* address must be aligned */
457 set_error( STATUS_INVALID_PARAMETER
);
460 if (!thread
) /* process is dead */
462 set_error( STATUS_ACCESS_DENIED
);
465 if (suspend_for_ptrace( thread
))
467 while (len
> 0 && max
)
469 if (read_thread_int( thread
, addr
++, dest
++ ) == -1) goto done
;
473 /* check the rest for read permission */
476 int dummy
, page
= get_page_size() / sizeof(int);
481 if (read_thread_int( thread
, addr
- 1, &dummy
) == -1) goto done
;
483 if (len
&& (read_thread_int( thread
, addr
+ len
- 1, &dummy
) == -1)) goto done
;
486 resume_thread( thread
);
490 /* write data to a process memory space */
491 /* len is the total size (in ints), max is the size we can actually read from the input buffer */
492 /* we check the total size for write permissions */
493 static void write_process_memory( struct process
*process
, int *addr
, size_t len
,
494 size_t max
, unsigned int first_mask
,
495 unsigned int last_mask
, const int *src
)
497 struct thread
*thread
= process
->thread_list
;
499 if (!len
|| ((unsigned int)addr
% sizeof(int))) /* address must be aligned */
501 set_error( STATUS_INVALID_PARAMETER
);
504 if (!thread
) /* process is dead */
506 set_error( STATUS_ACCESS_DENIED
);
509 if (suspend_for_ptrace( thread
))
511 /* first word is special */
514 if (write_thread_int( thread
, addr
++, *src
++, first_mask
) == -1) goto done
;
518 else last_mask
&= first_mask
;
520 while (len
> 1 && max
)
522 if (write_thread_int( thread
, addr
++, *src
++, ~0 ) == -1) goto done
;
529 /* last word is special too */
530 if (write_thread_int( thread
, addr
, *src
, last_mask
) == -1) goto done
;
534 /* check the rest for write permission */
535 int page
= get_page_size() / sizeof(int);
540 if (write_thread_int( thread
, addr
- 1, 0, 0 ) == -1) goto done
;
542 if (len
&& (write_thread_int( thread
, addr
+ len
- 1, 0, 0 ) == -1)) goto done
;
545 resume_thread( thread
);
549 /* take a snapshot of currently running processes */
550 struct process_snapshot
*process_snap( int *count
)
552 struct process_snapshot
*snapshot
, *ptr
;
553 struct process
*process
;
554 if (!running_processes
) return NULL
;
555 if (!(snapshot
= mem_alloc( sizeof(*snapshot
) * running_processes
)))
558 for (process
= first_process
; process
; process
= process
->next
)
560 if (!process
->running_threads
) continue;
561 ptr
->process
= process
;
562 ptr
->threads
= process
->running_threads
;
563 ptr
->priority
= process
->priority
;
564 grab_object( process
);
567 *count
= running_processes
;
571 /* create a new process */
572 DECL_HANDLER(new_process
)
574 size_t len
= get_req_strlen( req
->cmdline
);
575 struct thread
*thread
;
583 if (socketpair( AF_UNIX
, SOCK_STREAM
, 0, sock
) == -1)
589 if ((thread
= create_process( sock
[0], current
->process
, req
, req
->cmdline
, len
)))
591 int phandle
= alloc_handle( current
->process
, thread
->process
,
592 PROCESS_ALL_ACCESS
, req
->inherit
);
593 if ((req
->phandle
= phandle
) != -1)
595 if ((req
->thandle
= alloc_handle( current
->process
, thread
,
596 THREAD_ALL_ACCESS
, req
->inherit
)) != -1)
598 /* thread object will be released when the thread gets killed */
599 set_reply_fd( current
, sock
[1] );
600 req
->pid
= thread
->process
;
604 close_handle( current
->process
, phandle
);
606 release_object( thread
);
611 /* initialize a new process */
612 DECL_HANDLER(init_process
)
614 struct new_process_request
*info
;
616 if (!current
->unix_pid
)
618 fatal_protocol_error( current
, "init_process: init_thread not called yet\n" );
621 if (!(info
= current
->process
->info
))
623 fatal_protocol_error( current
, "init_process: called twice\n" );
626 current
->process
->ldt_copy
= req
->ldt_copy
;
627 current
->process
->ldt_flags
= req
->ldt_flags
;
628 current
->process
->info
= NULL
;
630 req
->start_flags
= info
->start_flags
;
631 req
->hstdin
= info
->hstdin
;
632 req
->hstdout
= info
->hstdout
;
633 req
->hstderr
= info
->hstderr
;
634 req
->cmd_show
= info
->cmd_show
;
635 req
->env_ptr
= info
->env_ptr
;
636 strcpy( req
->cmdline
, info
->cmdline
);
637 if (current
->process
->exe
.file
)
638 req
->exe_file
= alloc_handle( current
->process
, current
->process
->exe
.file
,
643 /* signal the end of the process initialization */
644 DECL_HANDLER(init_process_done
)
646 struct process
*process
= current
->process
;
647 if (!process
->init_event
)
649 fatal_protocol_error( current
, "init_process_done: no event\n" );
652 current
->entry
= req
->entry
;
653 process
->exe
.base
= req
->module
;
654 generate_startup_debug_events( current
->process
);
655 set_event( process
->init_event
);
656 release_object( process
->init_event
);
657 process
->init_event
= NULL
;
658 if (current
->suspend
+ current
->process
->suspend
> 0) stop_thread( current
);
659 req
->debugged
= (current
->process
->debugger
!= 0);
662 /* open a handle to a process */
663 DECL_HANDLER(open_process
)
665 struct process
*process
= get_process_from_id( req
->pid
);
669 req
->handle
= alloc_handle( current
->process
, process
, req
->access
, req
->inherit
);
670 release_object( process
);
674 /* terminate a process */
675 DECL_HANDLER(terminate_process
)
677 struct process
*process
;
679 if ((process
= get_process_from_handle( req
->handle
, PROCESS_TERMINATE
)))
681 kill_process( process
, req
->exit_code
);
682 release_object( process
);
686 /* fetch information about a process */
687 DECL_HANDLER(get_process_info
)
689 struct process
*process
;
691 if ((process
= get_process_from_handle( req
->handle
, PROCESS_QUERY_INFORMATION
)))
693 get_process_info( process
, req
);
694 release_object( process
);
698 /* set information about a process */
699 DECL_HANDLER(set_process_info
)
701 struct process
*process
;
703 if ((process
= get_process_from_handle( req
->handle
, PROCESS_SET_INFORMATION
)))
705 set_process_info( process
, req
);
706 release_object( process
);
710 /* read data from a process address space */
711 DECL_HANDLER(read_process_memory
)
713 struct process
*process
;
715 if ((process
= get_process_from_handle( req
->handle
, PROCESS_VM_READ
)))
717 read_process_memory( process
, req
->addr
, req
->len
,
718 get_req_size( req
->data
, sizeof(int) ), req
->data
);
719 release_object( process
);
723 /* write data to a process address space */
724 DECL_HANDLER(write_process_memory
)
726 struct process
*process
;
728 if ((process
= get_process_from_handle( req
->handle
, PROCESS_VM_WRITE
)))
730 write_process_memory( process
, req
->addr
, req
->len
, get_req_size( req
->data
, sizeof(int) ),
731 req
->first_mask
, req
->last_mask
, req
->data
);
732 release_object( process
);
736 /* notify the server that a dll has been loaded */
737 DECL_HANDLER(load_dll
)
739 struct process_dll
*dll
;
740 struct file
*file
= NULL
;
742 if ((req
->handle
!= -1) &&
743 !(file
= get_file_obj( current
->process
, req
->handle
, GENERIC_READ
))) return;
745 if ((dll
= process_load_dll( current
->process
, file
, req
->base
)))
747 dll
->dbg_offset
= req
->dbg_offset
;
748 dll
->dbg_size
= req
->dbg_size
;
749 dll
->name
= req
->name
;
750 /* only generate event if initialization is done */
751 if (!current
->process
->init_event
)
752 generate_debug_event( current
, LOAD_DLL_DEBUG_EVENT
, dll
);
754 if (file
) release_object( file
);
757 /* notify the server that a dll is being unloaded */
758 DECL_HANDLER(unload_dll
)
760 process_unload_dll( current
->process
, req
->base
);