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 */
74 req
->create_flags
= CREATE_NEW_CONSOLE
;
75 req
->start_flags
= STARTF_USESTDHANDLES
;
84 memcpy( process
->info
->cmdline
, cmd_line
, len
);
85 process
->info
->cmdline
[len
] = 0;
86 process
->create_flags
= process
->info
->create_flags
;
90 /* set the console and stdio handles for a newly created process */
91 static int set_process_console( struct process
*process
, struct process
*parent
)
93 struct new_process_request
*info
= process
->info
;
95 if (process
->create_flags
& CREATE_NEW_CONSOLE
)
97 if (!alloc_console( process
)) return 0;
99 else if (!(process
->create_flags
& DETACHED_PROCESS
))
101 if (parent
->console_in
) process
->console_in
= grab_object( parent
->console_in
);
102 if (parent
->console_out
) process
->console_out
= grab_object( parent
->console_out
);
106 if (!info
->inherit_all
&& !(info
->start_flags
& STARTF_USESTDHANDLES
))
108 /* duplicate the handle from the parent into this process */
109 info
->hstdin
= duplicate_handle( parent
, info
->hstdin
, process
,
110 0, TRUE
, DUPLICATE_SAME_ACCESS
);
111 info
->hstdout
= duplicate_handle( parent
, info
->hstdout
, process
,
112 0, TRUE
, DUPLICATE_SAME_ACCESS
);
113 info
->hstderr
= duplicate_handle( parent
, info
->hstderr
, process
,
114 0, TRUE
, DUPLICATE_SAME_ACCESS
);
119 /* no parent, use handles to the console for stdio */
120 info
->hstdin
= alloc_handle( process
, process
->console_in
,
121 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
122 info
->hstdout
= alloc_handle( process
, process
->console_out
,
123 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
124 info
->hstderr
= alloc_handle( process
, process
->console_out
,
125 GENERIC_READ
| GENERIC_WRITE
| SYNCHRONIZE
, 1 );
130 /* create a new process and its main thread */
131 struct thread
*create_process( int fd
, struct process
*parent
,
132 struct new_process_request
*req
,
133 const char *cmd_line
, size_t len
)
135 struct process
*process
;
136 struct thread
*thread
= NULL
;
138 if (!(process
= alloc_object( &process_ops
, -1 )))
143 process
->next
= NULL
;
144 process
->prev
= NULL
;
145 process
->thread_list
= NULL
;
146 process
->debugger
= NULL
;
147 process
->handles
= NULL
;
148 process
->exit_code
= STILL_ACTIVE
;
149 process
->running_threads
= 0;
150 process
->priority
= NORMAL_PRIORITY_CLASS
;
151 process
->affinity
= 1;
152 process
->suspend
= 0;
153 process
->create_flags
= 0;
154 process
->console_in
= NULL
;
155 process
->console_out
= NULL
;
156 process
->init_event
= NULL
;
157 process
->info
= NULL
;
158 process
->ldt_copy
= NULL
;
159 process
->ldt_flags
= NULL
;
160 process
->exe
.next
= NULL
;
161 process
->exe
.prev
= NULL
;
162 process
->exe
.file
= NULL
;
163 process
->exe
.dbg_offset
= 0;
164 process
->exe
.dbg_size
= 0;
165 gettimeofday( &process
->start_time
, NULL
);
166 if ((process
->next
= first_process
) != NULL
) process
->next
->prev
= process
;
167 first_process
= process
;
169 /* copy the request structure */
170 if (!set_creation_info( process
, req
, cmd_line
, len
)) goto error
;
172 if (process
->info
->inherit_all
== 2) /* HACK! */
173 process
->handles
= grab_object( parent
->handles
);
174 else if (process
->info
->inherit_all
)
175 process
->handles
= copy_handle_table( process
, parent
);
177 process
->handles
= alloc_handle_table( process
, 0 );
178 if (!process
->handles
) goto error
;
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 static void kill_process( struct process
*process
, struct thread
*skip
, int exit_code
)
404 struct thread
*thread
= process
->thread_list
;
407 struct thread
*next
= thread
->proc_next
;
408 thread
->exit_code
= exit_code
;
409 if (thread
!= skip
) kill_thread( thread
, 1 );
414 /* kill all processes being debugged by a given thread */
415 void kill_debugged_processes( struct thread
*debugger
, int exit_code
)
417 for (;;) /* restart from the beginning of the list every time */
419 struct process
*process
= first_process
;
420 /* find the first process being debugged by 'debugger' and still running */
421 while (process
&& (process
->debugger
!= debugger
|| !process
->running_threads
))
422 process
= process
->next
;
423 if (!process
) return;
424 process
->debugger
= NULL
;
425 kill_process( process
, NULL
, exit_code
);
429 /* get all information about a process */
430 static void get_process_info( struct process
*process
, struct get_process_info_request
*req
)
433 req
->debugged
= (process
->debugger
!= 0);
434 req
->exit_code
= process
->exit_code
;
435 req
->priority
= process
->priority
;
436 req
->process_affinity
= process
->affinity
;
437 req
->system_affinity
= 1;
440 /* set all information about a process */
441 static void set_process_info( struct process
*process
,
442 struct set_process_info_request
*req
)
444 if (req
->mask
& SET_PROCESS_INFO_PRIORITY
)
445 process
->priority
= req
->priority
;
446 if (req
->mask
& SET_PROCESS_INFO_AFFINITY
)
448 if (req
->affinity
!= 1) set_error( STATUS_INVALID_PARAMETER
);
449 else process
->affinity
= req
->affinity
;
453 /* read data from a process memory space */
454 /* len is the total size (in ints), max is the size we can actually store in the output buffer */
455 /* we read the total size in all cases to check for permissions */
456 static void read_process_memory( struct process
*process
, const int *addr
,
457 size_t len
, size_t max
, int *dest
)
459 struct thread
*thread
= process
->thread_list
;
461 if ((unsigned int)addr
% sizeof(int)) /* address must be aligned */
463 set_error( STATUS_INVALID_PARAMETER
);
466 if (!thread
) /* process is dead */
468 set_error( STATUS_ACCESS_DENIED
);
471 if (suspend_for_ptrace( thread
))
473 while (len
> 0 && max
)
475 if (read_thread_int( thread
, addr
++, dest
++ ) == -1) goto done
;
479 /* check the rest for read permission */
482 int dummy
, page
= get_page_size() / sizeof(int);
487 if (read_thread_int( thread
, addr
- 1, &dummy
) == -1) goto done
;
489 if (len
&& (read_thread_int( thread
, addr
+ len
- 1, &dummy
) == -1)) goto done
;
492 resume_thread( thread
);
496 /* write data to a process memory space */
497 /* len is the total size (in ints), max is the size we can actually read from the input buffer */
498 /* we check the total size for write permissions */
499 static void write_process_memory( struct process
*process
, int *addr
, size_t len
,
500 size_t max
, unsigned int first_mask
,
501 unsigned int last_mask
, const int *src
)
503 struct thread
*thread
= process
->thread_list
;
505 if (!len
|| ((unsigned int)addr
% sizeof(int))) /* address must be aligned */
507 set_error( STATUS_INVALID_PARAMETER
);
510 if (!thread
) /* process is dead */
512 set_error( STATUS_ACCESS_DENIED
);
515 if (suspend_for_ptrace( thread
))
517 /* first word is special */
520 if (write_thread_int( thread
, addr
++, *src
++, first_mask
) == -1) goto done
;
524 else last_mask
&= first_mask
;
526 while (len
> 1 && max
)
528 if (write_thread_int( thread
, addr
++, *src
++, ~0 ) == -1) goto done
;
535 /* last word is special too */
536 if (write_thread_int( thread
, addr
, *src
, last_mask
) == -1) goto done
;
540 /* check the rest for write permission */
541 int page
= get_page_size() / sizeof(int);
546 if (write_thread_int( thread
, addr
- 1, 0, 0 ) == -1) goto done
;
548 if (len
&& (write_thread_int( thread
, addr
+ len
- 1, 0, 0 ) == -1)) goto done
;
551 resume_thread( thread
);
555 /* take a snapshot of currently running processes */
556 struct process_snapshot
*process_snap( int *count
)
558 struct process_snapshot
*snapshot
, *ptr
;
559 struct process
*process
;
560 if (!running_processes
) return NULL
;
561 if (!(snapshot
= mem_alloc( sizeof(*snapshot
) * running_processes
)))
564 for (process
= first_process
; process
; process
= process
->next
)
566 if (!process
->running_threads
) continue;
567 ptr
->process
= process
;
568 ptr
->threads
= process
->running_threads
;
569 ptr
->priority
= process
->priority
;
570 grab_object( process
);
573 *count
= running_processes
;
577 /* create a new process */
578 DECL_HANDLER(new_process
)
580 size_t len
= get_req_strlen( req
, req
->cmdline
);
581 struct thread
*thread
;
589 if (socketpair( AF_UNIX
, SOCK_STREAM
, 0, sock
) == -1)
595 if ((thread
= create_process( sock
[0], current
->process
, req
, req
->cmdline
, len
)))
597 int phandle
= alloc_handle( current
->process
, thread
->process
,
598 PROCESS_ALL_ACCESS
, req
->pinherit
);
599 if ((req
->phandle
= phandle
) != -1)
601 if ((req
->thandle
= alloc_handle( current
->process
, thread
,
602 THREAD_ALL_ACCESS
, req
->tinherit
)) != -1)
604 /* thread object will be released when the thread gets killed */
605 set_reply_fd( current
, sock
[1] );
606 req
->pid
= thread
->process
;
610 close_handle( current
->process
, phandle
);
612 release_object( thread
);
617 /* initialize a new process */
618 DECL_HANDLER(init_process
)
620 struct new_process_request
*info
;
622 if (!current
->unix_pid
)
624 fatal_protocol_error( current
, "init_process: init_thread not called yet\n" );
627 if (!(info
= current
->process
->info
))
629 fatal_protocol_error( current
, "init_process: called twice\n" );
632 current
->process
->ldt_copy
= req
->ldt_copy
;
633 current
->process
->ldt_flags
= req
->ldt_flags
;
634 current
->process
->info
= NULL
;
636 req
->start_flags
= info
->start_flags
;
637 req
->hstdin
= info
->hstdin
;
638 req
->hstdout
= info
->hstdout
;
639 req
->hstderr
= info
->hstderr
;
640 req
->cmd_show
= info
->cmd_show
;
641 req
->env_ptr
= info
->env_ptr
;
642 strcpy( req
->cmdline
, info
->cmdline
);
643 if (current
->process
->exe
.file
)
644 req
->exe_file
= alloc_handle( current
->process
, current
->process
->exe
.file
,
649 /* signal the end of the process initialization */
650 DECL_HANDLER(init_process_done
)
652 struct process
*process
= current
->process
;
653 if (!process
->init_event
)
655 fatal_protocol_error( current
, "init_process_done: no event\n" );
658 process
->exe
.base
= req
->module
;
659 generate_startup_debug_events( current
->process
, req
->entry
);
660 set_event( process
->init_event
);
661 release_object( process
->init_event
);
662 process
->init_event
= NULL
;
663 if (current
->suspend
+ current
->process
->suspend
> 0) stop_thread( current
);
664 req
->debugged
= (current
->process
->debugger
!= 0);
667 /* open a handle to a process */
668 DECL_HANDLER(open_process
)
670 struct process
*process
= get_process_from_id( req
->pid
);
674 req
->handle
= alloc_handle( current
->process
, process
, req
->access
, req
->inherit
);
675 release_object( process
);
679 /* terminate a process */
680 DECL_HANDLER(terminate_process
)
682 struct process
*process
;
684 if ((process
= get_process_from_handle( req
->handle
, PROCESS_TERMINATE
)))
686 req
->self
= (current
->process
== process
);
687 kill_process( process
, current
, req
->exit_code
);
688 release_object( process
);
692 /* fetch information about a process */
693 DECL_HANDLER(get_process_info
)
695 struct process
*process
;
697 if ((process
= get_process_from_handle( req
->handle
, PROCESS_QUERY_INFORMATION
)))
699 get_process_info( process
, req
);
700 release_object( process
);
704 /* set information about a process */
705 DECL_HANDLER(set_process_info
)
707 struct process
*process
;
709 if ((process
= get_process_from_handle( req
->handle
, PROCESS_SET_INFORMATION
)))
711 set_process_info( process
, req
);
712 release_object( process
);
716 /* read data from a process address space */
717 DECL_HANDLER(read_process_memory
)
719 struct process
*process
;
721 if ((process
= get_process_from_handle( req
->handle
, PROCESS_VM_READ
)))
723 read_process_memory( process
, req
->addr
, req
->len
,
724 get_req_size( req
, req
->data
, sizeof(int) ), req
->data
);
725 release_object( process
);
729 /* write data to a process address space */
730 DECL_HANDLER(write_process_memory
)
732 struct process
*process
;
734 if ((process
= get_process_from_handle( req
->handle
, PROCESS_VM_WRITE
)))
736 write_process_memory( process
, req
->addr
, req
->len
,
737 get_req_size( req
, req
->data
, sizeof(int) ),
738 req
->first_mask
, req
->last_mask
, req
->data
);
739 release_object( process
);
743 /* notify the server that a dll has been loaded */
744 DECL_HANDLER(load_dll
)
746 struct process_dll
*dll
;
747 struct file
*file
= NULL
;
749 if ((req
->handle
!= -1) &&
750 !(file
= get_file_obj( current
->process
, req
->handle
, GENERIC_READ
))) return;
752 if ((dll
= process_load_dll( current
->process
, file
, req
->base
)))
754 dll
->dbg_offset
= req
->dbg_offset
;
755 dll
->dbg_size
= req
->dbg_size
;
756 dll
->name
= req
->name
;
757 /* only generate event if initialization is done */
758 if (!current
->process
->init_event
)
759 generate_debug_event( current
, LOAD_DLL_DEBUG_EVENT
, dll
);
761 if (file
) release_object( file
);
764 /* notify the server that a dll is being unloaded */
765 DECL_HANDLER(unload_dll
)
767 process_unload_dll( current
->process
, req
->base
);