git.oblomov.eu Git - wine/blob - server/process.c

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