server/process.c

   1 /*
   2  * Server-side process management
   3  *
   4  * Copyright (C) 1998 Alexandre Julliard
   5  */
   6
   7 #include "config.h"
   8
   9 #include <assert.h>
  10 #include <errno.h>
  11 #include <limits.h>
  12 #include <signal.h>
  13 #include <string.h>
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16 #include <sys/time.h>
  17 #ifdef HAVE_SYS_SOCKET_H
  18 # include <sys/socket.h>
  19 #endif
  20 #include <unistd.h>
  21
  22 #include "winbase.h"
  23 #include "winnt.h"
  24
  25 #include "handle.h"
  26 #include "process.h"
  27 #include "thread.h"
  28 #include "request.h"
  29 #include "console.h"
  30
  31 /* process structure */
  32
  33 static struct process *first_process;
  34 static int running_processes;
  35
  36 /* process operations */
  37
  38 static void process_dump( struct object *obj, int verbose );
  39 static int process_signaled( struct object *obj, struct thread *thread );
  40 static void process_poll_event( struct object *obj, int event );
  41 static void process_destroy( struct object *obj );
  42
  43 static const struct object_ops process_ops =
  44 {
  45     sizeof(struct process),      /* size */
  46     process_dump,                /* dump */
  47     add_queue,                   /* add_queue */
  48     remove_queue,                /* remove_queue */
  49     process_signaled,            /* signaled */
  50     no_satisfied,                /* satisfied */
  51     NULL,                        /* get_poll_events */
  52     process_poll_event,          /* poll_event */
  53     no_get_fd,                   /* get_fd */
  54     no_flush,                    /* flush */
  55     no_get_file_info,            /* get_file_info */
  56     process_destroy              /* destroy */
  57 };
  58
  59 /* process startup info */
  60
  61 struct startup_info
  62 {
  63     struct object       obj;          /* object header */
  64     int                 inherit_all;  /* inherit all handles from parent */
  65     int                 create_flags; /* creation flags */
  66     int                 start_flags;  /* flags from startup info */
  67     handle_t            hstdin;       /* handle for stdin */
  68     handle_t            hstdout;      /* handle for stdout */
  69     handle_t            hstderr;      /* handle for stderr */
  70     int                 cmd_show;     /* main window show mode */
  71     struct file        *exe_file;     /* file handle for main exe */
  72     char               *filename;     /* file name for main exe */
  73     struct thread      *owner;        /* owner thread (the one that created the new process) */
  74     struct process     *process;      /* created process */
  75     struct thread      *thread;       /* created thread */
  76 };
  77
  78 static void startup_info_dump( struct object *obj, int verbose );
  79 static int startup_info_signaled( struct object *obj, struct thread *thread );
  80 static void startup_info_destroy( struct object *obj );
  81
  82 static const struct object_ops startup_info_ops =
  83 {
  84     sizeof(struct startup_info),   /* size */
  85     startup_info_dump,             /* dump */
  86     add_queue,                     /* add_queue */
  87     remove_queue,                  /* remove_queue */
  88     startup_info_signaled,         /* signaled */
  89     no_satisfied,                  /* satisfied */
  90     NULL,                          /* get_poll_events */
  91     NULL,                          /* poll_event */
  92     no_get_fd,                     /* get_fd */
  93     no_flush,                      /* flush */
  94     no_get_file_info,              /* get_file_info */
  95     startup_info_destroy           /* destroy */
  96 };
  97
  98
  99 /* set the console and stdio handles for a newly created process */
 100 static int set_process_console( struct process *process, struct thread *parent_thread,
 101                                 struct startup_info *info, struct init_process_reply *reply )
 102 {
 103     if (process->create_flags & CREATE_NEW_CONSOLE)
 104     {
 105         /* let the process init do the allocation */
 106         return 1;
 107     }
 108     else if (parent_thread && !(process->create_flags & DETACHED_PROCESS))
 109     {
 110         /* FIXME: some better error checking should be done...
 111          * like if hConOut and hConIn are console handles, then they should be on the same
 112          * physical console
 113          */
 114         inherit_console( parent_thread, process,
 115                          (info->inherit_all || (info->start_flags & STARTF_USESTDHANDLES)) ?
 116                          info->hstdin : 0 );
 117     }
 118     if (parent_thread)
 119     {
 120         if (!info->inherit_all && !(info->start_flags & STARTF_USESTDHANDLES))
 121         {
 122             /* duplicate the handle from the parent into this process */
 123             reply->hstdin  = duplicate_handle( parent_thread->process, info->hstdin, process,
 124                                                0, TRUE, DUPLICATE_SAME_ACCESS );
 125             reply->hstdout = duplicate_handle( parent_thread->process, info->hstdout, process,
 126                                                0, TRUE, DUPLICATE_SAME_ACCESS );
 127             reply->hstderr = duplicate_handle( parent_thread->process, info->hstderr, process,
 128                                                0, TRUE, DUPLICATE_SAME_ACCESS );
 129         }
 130         else
 131         {
 132             reply->hstdin  = info->hstdin;
 133             reply->hstdout = info->hstdout;
 134             reply->hstderr = info->hstderr;
 135         }
 136     }
 137     else
 138     {
 139         if (process->console)
 140         {
 141             reply->hstdin  = alloc_handle( process, process->console,
 142                                            GENERIC_READ | GENERIC_WRITE | SYNCHRONIZE, 1 );
 143             reply->hstdout = alloc_handle( process, process->console->active,
 144                                            GENERIC_READ | GENERIC_WRITE, 1 );
 145             reply->hstderr = alloc_handle( process, process->console->active,
 146                                            GENERIC_READ | GENERIC_WRITE, 1 );
 147         }
 148         else
 149         {
 150             /* no parent, let the caller decide what to do */
 151             reply->hstdin = reply->hstdout = reply->hstderr = 0;
 152         }
 153     }
 154     /* some handles above may have been invalid; this is not an error */
 155     if (get_error() == STATUS_INVALID_HANDLE) clear_error();
 156     return 1;
 157 }
 158
 159 /* create a new process and its main thread */
 160 struct thread *create_process( int fd )
 161 {
 162     struct process *process;
 163     struct thread *thread = NULL;
 164     int request_pipe[2];
 165
 166     if (!(process = alloc_object( &process_ops, fd ))) return NULL;
 167     process->next            = NULL;
 168     process->prev            = NULL;
 169     process->parent          = NULL;
 170     process->thread_list     = NULL;
 171     process->debugger        = NULL;
 172     process->handles         = NULL;
 173     process->exit_code       = STILL_ACTIVE;
 174     process->running_threads = 0;
 175     process->priority        = NORMAL_PRIORITY_CLASS;
 176     process->affinity        = 1;
 177     process->suspend         = 0;
 178     process->create_flags    = 0;
 179     process->console         = NULL;
 180     process->init_event      = NULL;
 181     process->idle_event      = NULL;
 182     process->queue           = NULL;
 183     process->atom_table      = NULL;
 184     process->ldt_copy        = NULL;
 185     process->exe.next        = NULL;
 186     process->exe.prev        = NULL;
 187     process->exe.file        = NULL;
 188     process->exe.dbg_offset  = 0;
 189     process->exe.dbg_size    = 0;
 190
 191     gettimeofday( &process->start_time, NULL );
 192     if ((process->next = first_process) != NULL) process->next->prev = process;
 193     first_process = process;
 194
 195     /* create the init done event */
 196     if (!(process->init_event = create_event( NULL, 0, 1, 0 ))) goto error;
 197
 198     /* create the main thread */
 199     if (pipe( request_pipe ) == -1)
 200     {
 201         file_set_error();
 202         goto error;
 203     }
 204     send_client_fd( process, request_pipe[1], 0 );
 205     close( request_pipe[1] );
 206     if (!(thread = create_thread( request_pipe[0], process ))) goto error;
 207
 208     set_select_events( &process->obj, POLLIN );  /* start listening to events */
 209     release_object( process );
 210     return thread;
 211
 212  error:
 213     if (thread) release_object( thread );
 214     release_object( process );
 215     return NULL;
 216 }
 217
 218 /* initialize the current process and fill in the request */
 219 static void init_process( int ppid, struct init_process_reply *reply )
 220 {
 221     struct process *process = current->process;
 222     struct thread *parent_thread = get_thread_from_pid( ppid );
 223     struct process *parent = NULL;
 224     struct startup_info *info = NULL;
 225
 226     if (parent_thread)
 227     {
 228         parent = parent_thread->process;
 229         info = parent_thread->info;
 230         if (!info)
 231         {
 232             fatal_protocol_error( current, "init_process: parent but no info\n" );
 233             return;
 234         }
 235         if (info->thread)
 236         {
 237             fatal_protocol_error( current, "init_process: called twice?\n" );
 238             return;
 239         }
 240         process->parent = (struct process *)grab_object( parent );
 241     }
 242
 243     /* set the process flags */
 244     process->create_flags = info ? info->create_flags : 0;
 245
 246     /* create the handle table */
 247     if (parent && info->inherit_all)
 248         process->handles = copy_handle_table( process, parent );
 249     else
 250         process->handles = alloc_handle_table( process, 0 );
 251     if (!process->handles) return;
 252
 253     /* retrieve the main exe file */
 254     reply->exe_file = 0;
 255     if (parent && info->exe_file)
 256     {
 257         process->exe.file = (struct file *)grab_object( info->exe_file );
 258         if (!(reply->exe_file = alloc_handle( process, process->exe.file, GENERIC_READ, 0 )))
 259             return;
 260     }
 261
 262     /* set the process console */
 263     if (!set_process_console( process, parent_thread, info, reply )) return;
 264
 265     if (parent)
 266     {
 267         /* attach to the debugger if requested */
 268         if (process->create_flags & (DEBUG_PROCESS | DEBUG_ONLY_THIS_PROCESS))
 269             set_process_debugger( process, parent_thread );
 270         else if (parent && parent->debugger && !(parent->create_flags & DEBUG_ONLY_THIS_PROCESS))
 271             set_process_debugger( process, parent->debugger );
 272     }
 273
 274     /* thread will be actually suspended in init_done */
 275     if (process->create_flags & CREATE_SUSPENDED) current->suspend++;
 276
 277     if (info)
 278     {
 279         size_t size = strlen(info->filename);
 280         if (size > get_reply_max_size()) size = get_reply_max_size();
 281         reply->start_flags = info->start_flags;
 282         reply->cmd_show    = info->cmd_show;
 283         set_reply_data( info->filename, size );
 284         info->process = (struct process *)grab_object( process );
 285         info->thread  = (struct thread *)grab_object( current );
 286         wake_up( &info->obj, 0 );
 287     }
 288     else
 289     {
 290         reply->start_flags  = STARTF_USESTDHANDLES;
 291         reply->cmd_show     = 0;
 292     }
 293     reply->create_flags = process->create_flags;
 294     reply->server_start = server_start_ticks;
 295 }
 296
 297 /* destroy a process when its refcount is 0 */
 298 static void process_destroy( struct object *obj )
 299 {
 300     struct process *process = (struct process *)obj;
 301     assert( obj->ops == &process_ops );
 302
 303     /* we can't have a thread remaining */
 304     assert( !process->thread_list );
 305     if (process->console) release_object( process->console );
 306     if (process->parent) release_object( process->parent );
 307     if (process->next) process->next->prev = process->prev;
 308     if (process->prev) process->prev->next = process->next;
 309     else first_process = process->next;
 310     if (process->init_event) release_object( process->init_event );
 311     if (process->idle_event) release_object( process->idle_event );
 312     if (process->queue) release_object( process->queue );
 313     if (process->atom_table) release_object( process->atom_table );
 314     if (process->exe.file) release_object( process->exe.file );
 315 }
 316
 317 /* dump a process on stdout for debugging purposes */
 318 static void process_dump( struct object *obj, int verbose )
 319 {
 320     struct process *process = (struct process *)obj;
 321     assert( obj->ops == &process_ops );
 322
 323     fprintf( stderr, "Process next=%p prev=%p handles=%p\n",
 324              process->next, process->prev, process->handles );
 325 }
 326
 327 static int process_signaled( struct object *obj, struct thread *thread )
 328 {
 329     struct process *process = (struct process *)obj;
 330     return !process->running_threads;
 331 }
 332
 333
 334 static void process_poll_event( struct object *obj, int event )
 335 {
 336     struct process *process = (struct process *)obj;
 337     assert( obj->ops == &process_ops );
 338
 339     if (event & (POLLERR | POLLHUP)) set_select_events( obj, -1 );
 340     else if (event & POLLIN) receive_fd( process );
 341 }
 342
 343 static void startup_info_destroy( struct object *obj )
 344 {
 345     struct startup_info *info = (struct startup_info *)obj;
 346     assert( obj->ops == &startup_info_ops );
 347     if (info->filename) free( info->filename );
 348     if (info->exe_file) release_object( info->exe_file );
 349     if (info->process) release_object( info->process );
 350     if (info->thread) release_object( info->thread );
 351     if (info->owner)
 352     {
 353         info->owner->info = NULL;
 354         release_object( info->owner );
 355     }
 356 }
 357
 358 static void startup_info_dump( struct object *obj, int verbose )
 359 {
 360     struct startup_info *info = (struct startup_info *)obj;
 361     assert( obj->ops == &startup_info_ops );
 362
 363     fprintf( stderr, "Startup info flags=%x in=%d out=%d err=%d name='%s'\n",
 364              info->start_flags, info->hstdin, info->hstdout, info->hstderr, info->filename );
 365 }
 366
 367 static int startup_info_signaled( struct object *obj, struct thread *thread )
 368 {
 369     struct startup_info *info = (struct startup_info *)obj;
 370     return (info->thread != NULL);
 371 }
 372
 373
 374 /* get a process from an id (and increment the refcount) */
 375 struct process *get_process_from_id( void *id )
 376 {
 377     struct process *p = first_process;
 378     while (p && (p != id)) p = p->next;
 379     if (p) grab_object( p );
 380     else set_error( STATUS_INVALID_PARAMETER );
 381     return p;
 382 }
 383
 384 /* get a process from a handle (and increment the refcount) */
 385 struct process *get_process_from_handle( handle_t handle, unsigned int access )
 386 {
 387     return (struct process *)get_handle_obj( current->process, handle,
 388                                              access, &process_ops );
 389 }
 390
 391 /* add a dll to a process list */
 392 static struct process_dll *process_load_dll( struct process *process, struct file *file,
 393                                              void *base )
 394 {
 395     struct process_dll *dll;
 396
 397     /* make sure we don't already have one with the same base address */
 398     for (dll = process->exe.next; dll; dll = dll->next) if (dll->base == base)
 399     {
 400         set_error( STATUS_INVALID_PARAMETER );
 401         return NULL;
 402     }
 403
 404     if ((dll = mem_alloc( sizeof(*dll) )))
 405     {
 406         dll->prev = &process->exe;
 407         dll->file = NULL;
 408         dll->base = base;
 409         if (file) dll->file = (struct file *)grab_object( file );
 410         if ((dll->next = process->exe.next)) dll->next->prev = dll;
 411         process->exe.next = dll;
 412     }
 413     return dll;
 414 }
 415
 416 /* remove a dll from a process list */
 417 static void process_unload_dll( struct process *process, void *base )
 418 {
 419     struct process_dll *dll;
 420
 421     for (dll = process->exe.next; dll; dll = dll->next)
 422     {
 423         if (dll->base == base)
 424         {
 425             if (dll->file) release_object( dll->file );
 426             if (dll->next) dll->next->prev = dll->prev;
 427             if (dll->prev) dll->prev->next = dll->next;
 428             free( dll );
 429             generate_debug_event( current, UNLOAD_DLL_DEBUG_EVENT, base );
 430             return;
 431         }
 432     }
 433     set_error( STATUS_INVALID_PARAMETER );
 434 }
 435
 436 /* kill all processes being attached to a console renderer */
 437 void kill_console_processes( struct thread *renderer, int exit_code )
 438 {
 439     for (;;)  /* restart from the beginning of the list every time */
 440     {
 441         struct process *process = first_process;
 442
 443         /* find the first process being attached to 'renderer' and still running */
 444         while (process &&
 445                (process == renderer->process || !process->console ||
 446                 process->console->renderer != renderer || !process->running_threads))
 447         {
 448             process = process->next;
 449         }
 450         if (!process) break;
 451         kill_process( process, NULL, exit_code );
 452     }
 453 }
 454
 455 /* a process has been killed (i.e. its last thread died) */
 456 static void process_killed( struct process *process )
 457 {
 458     assert( !process->thread_list );
 459     gettimeofday( &process->end_time, NULL );
 460     if (process->handles) release_object( process->handles );
 461     process->handles = NULL;
 462
 463     /* close the console attached to this process, if any */
 464     free_console( process );
 465
 466     while (process->exe.next)
 467     {
 468         struct process_dll *dll = process->exe.next;
 469         process->exe.next = dll->next;
 470         if (dll->file) release_object( dll->file );
 471         free( dll );
 472     }
 473     if (process->exe.file) release_object( process->exe.file );
 474     process->exe.file = NULL;
 475     wake_up( &process->obj, 0 );
 476     if (!--running_processes)
 477     {
 478         /* last process died, close global handles */
 479         close_global_handles();
 480         /* this will cause the select loop to terminate */
 481         if (!persistent_server) close_master_socket();
 482     }
 483 }
 484
 485 /* add a thread to a process running threads list */
 486 void add_process_thread( struct process *process, struct thread *thread )
 487 {
 488     thread->proc_next = process->thread_list;
 489     thread->proc_prev = NULL;
 490     if (thread->proc_next) thread->proc_next->proc_prev = thread;
 491     process->thread_list = thread;
 492     if (!process->running_threads++) running_processes++;
 493     grab_object( thread );
 494 }
 495
 496 /* remove a thread from a process running threads list */
 497 void remove_process_thread( struct process *process, struct thread *thread )
 498 {
 499     assert( process->running_threads > 0 );
 500     assert( process->thread_list );
 501
 502     if (thread->proc_next) thread->proc_next->proc_prev = thread->proc_prev;
 503     if (thread->proc_prev) thread->proc_prev->proc_next = thread->proc_next;
 504     else process->thread_list = thread->proc_next;
 505
 506     if (!--process->running_threads)
 507     {
 508         /* we have removed the last running thread, exit the process */
 509         process->exit_code = thread->exit_code;
 510         generate_debug_event( thread, EXIT_PROCESS_DEBUG_EVENT, process );
 511         process_killed( process );
 512     }
 513     else generate_debug_event( thread, EXIT_THREAD_DEBUG_EVENT, thread );
 514     release_object( thread );
 515 }
 516
 517 /* suspend all the threads of a process */
 518 void suspend_process( struct process *process )
 519 {
 520     if (!process->suspend++)
 521     {
 522         struct thread *thread = process->thread_list;
 523         for (; thread; thread = thread->proc_next)
 524         {
 525             if (!thread->suspend) stop_thread( thread );
 526         }
 527     }
 528 }
 529
 530 /* resume all the threads of a process */
 531 void resume_process( struct process *process )
 532 {
 533     assert (process->suspend > 0);
 534     if (!--process->suspend)
 535     {
 536         struct thread *thread = process->thread_list;
 537         for (; thread; thread = thread->proc_next)
 538         {
 539             if (!thread->suspend) continue_thread( thread );
 540         }
 541     }
 542 }
 543
 544 /* kill a process on the spot */
 545 void kill_process( struct process *process, struct thread *skip, int exit_code )
 546 {
 547     struct thread *thread = process->thread_list;
 548
 549     while (thread)
 550     {
 551         struct thread *next = thread->proc_next;
 552         thread->exit_code = exit_code;
 553         if (thread != skip) kill_thread( thread, 1 );
 554         thread = next;
 555     }
 556 }
 557
 558 /* kill all processes being debugged by a given thread */
 559 void kill_debugged_processes( struct thread *debugger, int exit_code )
 560 {
 561     for (;;)  /* restart from the beginning of the list every time */
 562     {
 563         struct process *process = first_process;
 564         /* find the first process being debugged by 'debugger' and still running */
 565         while (process && (process->debugger != debugger || !process->running_threads))
 566             process = process->next;
 567         if (!process) return;
 568         process->debugger = NULL;
 569         kill_process( process, NULL, exit_code );
 570     }
 571 }
 572
 573
 574 /* get all information about a process */
 575 static void get_process_info( struct process *process, struct get_process_info_reply *reply )
 576 {
 577     reply->pid              = process;
 578     reply->debugged         = (process->debugger != 0);
 579     reply->exit_code        = process->exit_code;
 580     reply->priority         = process->priority;
 581     reply->process_affinity = process->affinity;
 582     reply->system_affinity  = 1;
 583 }
 584
 585 /* set all information about a process */
 586 static void set_process_info( struct process *process,
 587                               const struct set_process_info_request *req )
 588 {
 589     if (req->mask & SET_PROCESS_INFO_PRIORITY)
 590         process->priority = req->priority;
 591     if (req->mask & SET_PROCESS_INFO_AFFINITY)
 592     {
 593         if (req->affinity != 1) set_error( STATUS_INVALID_PARAMETER );
 594         else process->affinity = req->affinity;
 595     }
 596 }
 597
 598 /* read data from a process memory space */
 599 /* len is the total size (in ints) */
 600 static int read_process_memory( struct process *process, const int *addr, size_t len, int *dest )
 601 {
 602     struct thread *thread = process->thread_list;
 603
 604     assert( !((unsigned int)addr % sizeof(int)) );  /* address must be aligned */
 605
 606     if (!thread)  /* process is dead */
 607     {
 608         set_error( STATUS_ACCESS_DENIED );
 609         return 0;
 610     }
 611     if (suspend_for_ptrace( thread ))
 612     {
 613         while (len > 0)
 614         {
 615             if (read_thread_int( thread, addr++, dest++ ) == -1) break;
 616             len--;
 617         }
 618         resume_thread( thread );
 619     }
 620     return !len;
 621 }
 622
 623 /* make sure we can write to the whole address range */
 624 /* len is the total size (in ints) */
 625 static int check_process_write_access( struct thread *thread, int *addr, size_t len )
 626 {
 627     int page = get_page_size() / sizeof(int);
 628
 629     for (;;)
 630     {
 631         if (write_thread_int( thread, addr, 0, 0 ) == -1) return 0;
 632         if (len <= page) break;
 633         addr += page;
 634         len -= page;
 635     }
 636     return (write_thread_int( thread, addr + len - 1, 0, 0 ) != -1);
 637 }
 638
 639 /* write data to a process memory space */
 640 /* len is the total size (in ints), max is the size we can actually read from the input buffer */
 641 /* we check the total size for write permissions */
 642 static void write_process_memory( struct process *process, int *addr, size_t len,
 643                                   unsigned int first_mask, unsigned int last_mask, const int *src )
 644 {
 645     struct thread *thread = process->thread_list;
 646
 647     assert( !((unsigned int)addr % sizeof(int) ));  /* address must be aligned */
 648
 649     if (!thread)  /* process is dead */
 650     {
 651         set_error( STATUS_ACCESS_DENIED );
 652         return;
 653     }
 654     if (suspend_for_ptrace( thread ))
 655     {
 656         if (!check_process_write_access( thread, addr, len ))
 657         {
 658             set_error( STATUS_ACCESS_DENIED );
 659             return;
 660         }
 661         /* first word is special */
 662         if (len > 1)
 663         {
 664             if (write_thread_int( thread, addr++, *src++, first_mask ) == -1) goto done;
 665             len--;
 666         }
 667         else last_mask &= first_mask;
 668
 669         while (len > 1)
 670         {
 671             if (write_thread_int( thread, addr++, *src++, ~0 ) == -1) goto done;
 672             len--;
 673         }
 674
 675         /* last word is special too */
 676         if (write_thread_int( thread, addr, *src, last_mask ) == -1) goto done;
 677
 678     done:
 679         resume_thread( thread );
 680     }
 681 }
 682
 683 /* take a snapshot of currently running processes */
 684 struct process_snapshot *process_snap( int *count )
 685 {
 686     struct process_snapshot *snapshot, *ptr;
 687     struct process *process;
 688     if (!running_processes) return NULL;
 689     if (!(snapshot = mem_alloc( sizeof(*snapshot) * running_processes )))
 690         return NULL;
 691     ptr = snapshot;
 692     for (process = first_process; process; process = process->next)
 693     {
 694         if (!process->running_threads) continue;
 695         ptr->process  = process;
 696         ptr->threads  = process->running_threads;
 697         ptr->count    = process->obj.refcount;
 698         ptr->priority = process->priority;
 699         grab_object( process );
 700         ptr++;
 701     }
 702     *count = running_processes;
 703     return snapshot;
 704 }
 705
 706 /* take a snapshot of the modules of a process */
 707 struct module_snapshot *module_snap( struct process *process, int *count )
 708 {
 709     struct module_snapshot *snapshot, *ptr;
 710     struct process_dll *dll;
 711     int total = 0;
 712
 713     for (dll = &process->exe; dll; dll = dll->next) total++;
 714     if (!(snapshot = mem_alloc( sizeof(*snapshot) * total ))) return NULL;
 715
 716     for (ptr = snapshot, dll = &process->exe; dll; dll = dll->next, ptr++)
 717     {
 718         ptr->base = dll->base;
 719     }
 720     *count = total;
 721     return snapshot;
 722 }
 723
 724
 725 /* create a new process */
 726 DECL_HANDLER(new_process)
 727 {
 728     size_t len = get_req_data_size();
 729     struct startup_info *info;
 730
 731     if (current->info)
 732     {
 733         fatal_protocol_error( current, "new_process: another process is being created\n" );
 734         return;
 735     }
 736
 737     /* build the startup info for a new process */
 738     if (!(info = alloc_object( &startup_info_ops, -1 ))) return;
 739     info->inherit_all  = req->inherit_all;
 740     info->create_flags = req->create_flags;
 741     info->start_flags  = req->start_flags;
 742     info->hstdin       = req->hstdin;
 743     info->hstdout      = req->hstdout;
 744     info->hstderr      = req->hstderr;
 745     info->cmd_show     = req->cmd_show;
 746     info->exe_file     = NULL;
 747     info->filename     = NULL;
 748     info->owner        = (struct thread *)grab_object( current );
 749     info->process      = NULL;
 750     info->thread       = NULL;
 751
 752     if (req->exe_file &&
 753         !(info->exe_file = get_file_obj( current->process, req->exe_file, GENERIC_READ )))
 754         goto done;
 755
 756     if (!(info->filename = mem_alloc( len + 1 ))) goto done;
 757
 758     memcpy( info->filename, get_req_data(), len );
 759     info->filename[len] = 0;
 760     current->info = info;
 761     reply->info = alloc_handle( current->process, info, SYNCHRONIZE, FALSE );
 762
 763  done:
 764     release_object( info );
 765 }
 766
 767 /* Retrieve information about a newly started process */
 768 DECL_HANDLER(get_new_process_info)
 769 {
 770     struct startup_info *info;
 771
 772     reply->event = 0;
 773
 774     if ((info = (struct startup_info *)get_handle_obj( current->process, req->info,
 775                                                        0, &startup_info_ops )))
 776     {
 777         reply->pid = get_process_id( info->process );
 778         reply->tid = get_thread_id( info->thread );
 779         reply->phandle = alloc_handle( current->process, info->process,
 780                                        PROCESS_ALL_ACCESS, req->pinherit );
 781         reply->thandle = alloc_handle( current->process, info->thread,
 782                                        THREAD_ALL_ACCESS, req->tinherit );
 783         if (info->process->init_event)
 784             reply->event = alloc_handle( current->process, info->process->init_event,
 785                                        EVENT_ALL_ACCESS, 0 );
 786         release_object( info );
 787     }
 788     else
 789     {
 790         reply->pid     = 0;
 791         reply->tid     = 0;
 792         reply->phandle = 0;
 793         reply->thandle = 0;
 794     }
 795 }
 796
 797 /* initialize a new process */
 798 DECL_HANDLER(init_process)
 799 {
 800     if (!current->unix_pid)
 801     {
 802         fatal_protocol_error( current, "init_process: init_thread not called yet\n" );
 803         return;
 804     }
 805     current->process->ldt_copy  = req->ldt_copy;
 806     init_process( req->ppid, reply );
 807 }
 808
 809 /* signal the end of the process initialization */
 810 DECL_HANDLER(init_process_done)
 811 {
 812     struct file *file = NULL;
 813     struct process *process = current->process;
 814
 815     if (!process->init_event)
 816     {
 817         fatal_protocol_error( current, "init_process_done: no event\n" );
 818         return;
 819     }
 820     process->exe.base = req->module;
 821     process->exe.name = req->name;
 822
 823     if (req->exe_file) file = get_file_obj( current->process, req->exe_file, GENERIC_READ );
 824     if (process->exe.file) release_object( process->exe.file );
 825     process->exe.file = file;
 826
 827     generate_startup_debug_events( current->process, req->entry );
 828     set_event( process->init_event );
 829     release_object( process->init_event );
 830     process->init_event = NULL;
 831     if (req->gui) process->idle_event = create_event( NULL, 0, 1, 0 );
 832     if (current->suspend + current->process->suspend > 0) stop_thread( current );
 833     reply->debugged = (current->process->debugger != 0);
 834 }
 835
 836 /* open a handle to a process */
 837 DECL_HANDLER(open_process)
 838 {
 839     struct process *process = get_process_from_id( req->pid );
 840     reply->handle = 0;
 841     if (process)
 842     {
 843         reply->handle = alloc_handle( current->process, process, req->access, req->inherit );
 844         release_object( process );
 845     }
 846 }
 847
 848 /* terminate a process */
 849 DECL_HANDLER(terminate_process)
 850 {
 851     struct process *process;
 852
 853     if ((process = get_process_from_handle( req->handle, PROCESS_TERMINATE )))
 854     {
 855         reply->self = (current->process == process);
 856         kill_process( process, current, req->exit_code );
 857         release_object( process );
 858     }
 859 }
 860
 861 /* fetch information about a process */
 862 DECL_HANDLER(get_process_info)
 863 {
 864     struct process *process;
 865
 866     if ((process = get_process_from_handle( req->handle, PROCESS_QUERY_INFORMATION )))
 867     {
 868         get_process_info( process, reply );
 869         release_object( process );
 870     }
 871 }
 872
 873 /* set information about a process */
 874 DECL_HANDLER(set_process_info)
 875 {
 876     struct process *process;
 877
 878     if ((process = get_process_from_handle( req->handle, PROCESS_SET_INFORMATION )))
 879     {
 880         set_process_info( process, req );
 881         release_object( process );
 882     }
 883 }
 884
 885 /* read data from a process address space */
 886 DECL_HANDLER(read_process_memory)
 887 {
 888     struct process *process;
 889     size_t len = get_reply_max_size();
 890
 891     if (!(process = get_process_from_handle( req->handle, PROCESS_VM_READ ))) return;
 892
 893     if (len)
 894     {
 895         unsigned int start_offset = (unsigned int)req->addr % sizeof(int);
 896         unsigned int nb_ints = (len + start_offset + sizeof(int) - 1) / sizeof(int);
 897         const int *start = (int *)((char *)req->addr - start_offset);
 898         int *buffer = mem_alloc( nb_ints * sizeof(int) );
 899         if (buffer)
 900         {
 901             if (read_process_memory( process, start, nb_ints, buffer ))
 902             {
 903                 /* move start of requested data to start of buffer */
 904                 if (start_offset) memmove( buffer, (char *)buffer + start_offset, len );
 905                 set_reply_data_ptr( buffer, len );
 906             }
 907             else len = 0;
 908         }
 909     }
 910     release_object( process );
 911 }
 912
 913 /* write data to a process address space */
 914 DECL_HANDLER(write_process_memory)
 915 {
 916     struct process *process;
 917
 918     if ((process = get_process_from_handle( req->handle, PROCESS_VM_WRITE )))
 919     {
 920         size_t len = get_req_data_size();
 921         if ((len % sizeof(int)) || ((unsigned int)req->addr % sizeof(int)))
 922             set_error( STATUS_INVALID_PARAMETER );
 923         else
 924         {
 925             if (len) write_process_memory( process, req->addr, len / sizeof(int),
 926                                            req->first_mask, req->last_mask, get_req_data() );
 927         }
 928         release_object( process );
 929     }
 930 }
 931
 932 /* notify the server that a dll has been loaded */
 933 DECL_HANDLER(load_dll)
 934 {
 935     struct process_dll *dll;
 936     struct file *file = NULL;
 937
 938     if (req->handle &&
 939         !(file = get_file_obj( current->process, req->handle, GENERIC_READ ))) return;
 940
 941     if ((dll = process_load_dll( current->process, file, req->base )))
 942     {
 943         dll->dbg_offset = req->dbg_offset;
 944         dll->dbg_size   = req->dbg_size;
 945         dll->name       = req->name;
 946         /* only generate event if initialization is done */
 947         if (!current->process->init_event)
 948             generate_debug_event( current, LOAD_DLL_DEBUG_EVENT, dll );
 949     }
 950     if (file) release_object( file );
 951 }
 952
 953 /* notify the server that a dll is being unloaded */
 954 DECL_HANDLER(unload_dll)
 955 {
 956     process_unload_dll( current->process, req->base );
 957 }
 958
 959 /* wait for a process to start waiting on input */
 960 /* FIXME: only returns event for now, wait is done in the client */
 961 DECL_HANDLER(wait_input_idle)
 962 {
 963     struct process *process;
 964
 965     reply->event = 0;
 966     if ((process = get_process_from_handle( req->handle, PROCESS_QUERY_INFORMATION )))
 967     {
 968         if (process->idle_event && process != current->process && process->queue != current->queue)
 969             reply->event = alloc_handle( current->process, process->idle_event,
 970                                          EVENT_ALL_ACCESS, 0 );
 971         release_object( process );
 972     }
 973 }