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 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->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->inherit );
599 if ((req->phandle = phandle) != -1)
601 if ((req->thandle = alloc_handle( current->process, thread,
602 THREAD_ALL_ACCESS, req->inherit )) != -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 current->entry = req->entry;
659 process->exe.base = req->module;
660 generate_startup_debug_events( current->process );
661 set_event( process->init_event );
662 release_object( process->init_event );
663 process->init_event = NULL;
664 if (current->suspend + current->process->suspend > 0) stop_thread( current );
665 req->debugged = (current->process->debugger != 0);
668 /* open a handle to a process */
669 DECL_HANDLER(open_process)
671 struct process *process = get_process_from_id( req->pid );
675 req->handle = alloc_handle( current->process, process, req->access, req->inherit );
676 release_object( process );
680 /* terminate a process */
681 DECL_HANDLER(terminate_process)
683 struct process *process;
685 if ((process = get_process_from_handle( req->handle, PROCESS_TERMINATE )))
687 req->self = (current->process == process);
688 kill_process( process, current, req->exit_code );
689 release_object( process );
693 /* fetch information about a process */
694 DECL_HANDLER(get_process_info)
696 struct process *process;
698 if ((process = get_process_from_handle( req->handle, PROCESS_QUERY_INFORMATION )))
700 get_process_info( process, req );
701 release_object( process );
705 /* set information about a process */
706 DECL_HANDLER(set_process_info)
708 struct process *process;
710 if ((process = get_process_from_handle( req->handle, PROCESS_SET_INFORMATION )))
712 set_process_info( process, req );
713 release_object( process );
717 /* read data from a process address space */
718 DECL_HANDLER(read_process_memory)
720 struct process *process;
722 if ((process = get_process_from_handle( req->handle, PROCESS_VM_READ )))
724 read_process_memory( process, req->addr, req->len,
725 get_req_size( req->data, sizeof(int) ), req->data );
726 release_object( process );
730 /* write data to a process address space */
731 DECL_HANDLER(write_process_memory)
733 struct process *process;
735 if ((process = get_process_from_handle( req->handle, PROCESS_VM_WRITE )))
737 write_process_memory( process, req->addr, req->len, get_req_size( 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 );