d3d8: Remove WINAPI on static functions where not needed.
[wine] / server / snapshot.c
1 /*
2  * Server-side snapshots
3  *
4  * Copyright (C) 1999 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  *
20  * FIXME: heap snapshots not implemented
21  */
22
23 #include "config.h"
24 #include "wine/port.h"
25
26 #include <assert.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <stdarg.h>
30
31 #include "ntstatus.h"
32 #define WIN32_NO_STATUS
33 #include "windef.h"
34 #include "winternl.h"
35
36 #include "handle.h"
37 #include "process.h"
38 #include "thread.h"
39 #include "request.h"
40
41
42 struct snapshot
43 {
44     struct object             obj;           /* object header */
45     struct process           *process;       /* process of this snapshot (for modules and heaps) */
46     struct process_snapshot  *processes;     /* processes snapshot */
47     int                       process_count; /* count of processes */
48     int                       process_pos;   /* current position in proc snapshot */
49     struct thread_snapshot   *threads;       /* threads snapshot */
50     int                       thread_count;  /* count of threads */
51     int                       thread_pos;    /* current position in thread snapshot */
52     struct module_snapshot   *modules;       /* modules snapshot */
53     int                       module_count;  /* count of modules */
54     int                       module_pos;    /* current position in module snapshot */
55 };
56
57 static void snapshot_dump( struct object *obj, int verbose );
58 static void snapshot_destroy( struct object *obj );
59
60 static const struct object_ops snapshot_ops =
61 {
62     sizeof(struct snapshot),      /* size */
63     snapshot_dump,                /* dump */
64     no_get_type,                  /* get_type */
65     no_add_queue,                 /* add_queue */
66     NULL,                         /* remove_queue */
67     NULL,                         /* signaled */
68     NULL,                         /* satisfied */
69     no_signal,                    /* signal */
70     no_get_fd,                    /* get_fd */
71     no_map_access,                /* map_access */
72     default_get_sd,               /* get_sd */
73     default_set_sd,               /* set_sd */
74     no_lookup_name,               /* lookup_name */
75     no_open_file,                 /* open_file */
76     no_close_handle,              /* close_handle */
77     snapshot_destroy              /* destroy */
78 };
79
80
81 /* create a new snapshot */
82 static struct snapshot *create_snapshot( process_id_t pid, int flags )
83 {
84     struct process *process = NULL;
85     struct snapshot *snapshot;
86
87     /* need a process for modules and heaps */
88     if (flags & (SNAP_MODULE|SNAP_HEAPLIST))
89     {
90         if (!pid) process = (struct process *)grab_object( current->process );
91         else if (!(process = get_process_from_id( pid ))) return NULL;
92     }
93
94     if (!(snapshot = alloc_object( &snapshot_ops )))
95     {
96         if (process) release_object( process );
97         return NULL;
98     }
99
100     snapshot->process = process;
101
102     snapshot->process_pos = 0;
103     snapshot->process_count = 0;
104     if (flags & SNAP_PROCESS)
105         snapshot->processes = process_snap( &snapshot->process_count );
106
107     snapshot->thread_pos = 0;
108     snapshot->thread_count = 0;
109     if (flags & SNAP_THREAD)
110         snapshot->threads = thread_snap( &snapshot->thread_count );
111
112     snapshot->module_pos = 0;
113     snapshot->module_count = 0;
114     if (flags & SNAP_MODULE)
115         snapshot->modules = module_snap( process, &snapshot->module_count );
116
117     return snapshot;
118 }
119
120 /* get the next process in the snapshot */
121 static int snapshot_next_process( struct snapshot *snapshot, struct next_process_reply *reply )
122 {
123     struct process_snapshot *ptr;
124     struct process_dll *exe_module;
125
126     if (!snapshot->process_count)
127     {
128         set_error( STATUS_INVALID_PARAMETER );  /* FIXME */
129         return 0;
130     }
131     if (snapshot->process_pos >= snapshot->process_count)
132     {
133         set_error( STATUS_NO_MORE_FILES );
134         return 0;
135     }
136     ptr = &snapshot->processes[snapshot->process_pos++];
137     reply->count    = ptr->count;
138     reply->pid      = get_process_id( ptr->process );
139     reply->ppid     = ptr->process->parent ? get_process_id( ptr->process->parent ) : 0;
140     reply->heap     = NULL;  /* FIXME */
141     reply->module   = NULL;  /* FIXME */
142     reply->threads  = ptr->threads;
143     reply->priority = ptr->priority;
144     reply->handles  = ptr->handles;
145     if ((exe_module = get_process_exe_module( ptr->process )) && exe_module->filename)
146     {
147         data_size_t len = min( exe_module->namelen, get_reply_max_size() );
148         set_reply_data( exe_module->filename, len );
149     }
150     return 1;
151 }
152
153 /* get the next thread in the snapshot */
154 static int snapshot_next_thread( struct snapshot *snapshot, struct next_thread_reply *reply )
155 {
156     struct thread_snapshot *ptr;
157
158     if (!snapshot->thread_count)
159     {
160         set_error( STATUS_INVALID_PARAMETER );  /* FIXME */
161         return 0;
162     }
163     if (snapshot->thread_pos >= snapshot->thread_count)
164     {
165         set_error( STATUS_NO_MORE_FILES );
166         return 0;
167     }
168     ptr = &snapshot->threads[snapshot->thread_pos++];
169     reply->count     = ptr->count;
170     reply->pid       = get_process_id( ptr->thread->process );
171     reply->tid       = get_thread_id( ptr->thread );
172     reply->base_pri  = ptr->priority;
173     reply->delta_pri = 0;  /* FIXME */
174     return 1;
175 }
176
177 /* get the next module in the snapshot */
178 static int snapshot_next_module( struct snapshot *snapshot, struct next_module_reply *reply )
179 {
180     struct module_snapshot *ptr;
181
182     if (!snapshot->module_count)
183     {
184         set_error( STATUS_INVALID_PARAMETER );  /* FIXME */
185         return 0;
186     }
187     if (snapshot->module_pos >= snapshot->module_count)
188     {
189         set_error( STATUS_NO_MORE_FILES );
190         return 0;
191     }
192     ptr = &snapshot->modules[snapshot->module_pos++];
193     reply->pid  = get_process_id( snapshot->process );
194     reply->base = ptr->base;
195     reply->size = ptr->size;
196     if (ptr->filename)
197     {
198         data_size_t len = min( ptr->namelen, get_reply_max_size() );
199         set_reply_data( ptr->filename, len );
200     }
201     return 1;
202 }
203
204 static void snapshot_dump( struct object *obj, int verbose )
205 {
206     struct snapshot *snapshot = (struct snapshot *)obj;
207     assert( obj->ops == &snapshot_ops );
208     fprintf( stderr, "Snapshot: %d procs %d threads %d modules\n",
209              snapshot->process_count, snapshot->thread_count, snapshot->module_count );
210 }
211
212 static void snapshot_destroy( struct object *obj )
213 {
214     int i;
215     struct snapshot *snapshot = (struct snapshot *)obj;
216     assert( obj->ops == &snapshot_ops );
217     if (snapshot->process_count)
218     {
219         for (i = 0; i < snapshot->process_count; i++)
220             release_object( snapshot->processes[i].process );
221         free( snapshot->processes );
222     }
223     if (snapshot->thread_count)
224     {
225         for (i = 0; i < snapshot->thread_count; i++)
226             release_object( snapshot->threads[i].thread );
227         free( snapshot->threads );
228     }
229     if (snapshot->module_count)
230     {
231         for (i = 0; i < snapshot->module_count; i++)
232             free( snapshot->modules[i].filename );
233         free( snapshot->modules );
234     }
235     if (snapshot->process) release_object( snapshot->process );
236 }
237
238 /* create a snapshot */
239 DECL_HANDLER(create_snapshot)
240 {
241     struct snapshot *snapshot;
242
243     reply->handle = 0;
244     if ((snapshot = create_snapshot( req->pid, req->flags )))
245     {
246         reply->handle = alloc_handle( current->process, snapshot, 0, req->attributes );
247         release_object( snapshot );
248     }
249 }
250
251 /* get the next process from a snapshot */
252 DECL_HANDLER(next_process)
253 {
254     struct snapshot *snapshot;
255
256     if ((snapshot = (struct snapshot *)get_handle_obj( current->process, req->handle,
257                                                        0, &snapshot_ops )))
258     {
259         if (req->reset) snapshot->process_pos = 0;
260         snapshot_next_process( snapshot, reply );
261         release_object( snapshot );
262     }
263 }
264
265 /* get the next thread from a snapshot */
266 DECL_HANDLER(next_thread)
267 {
268     struct snapshot *snapshot;
269
270     if ((snapshot = (struct snapshot *)get_handle_obj( current->process, req->handle,
271                                                        0, &snapshot_ops )))
272     {
273         if (req->reset) snapshot->thread_pos = 0;
274         snapshot_next_thread( snapshot, reply );
275         release_object( snapshot );
276     }
277 }
278
279 /* get the next module from a snapshot */
280 DECL_HANDLER(next_module)
281 {
282     struct snapshot *snapshot;
283
284     if ((snapshot = (struct snapshot *)get_handle_obj( current->process, req->handle,
285                                                        0, &snapshot_ops )))
286     {
287         if (req->reset) snapshot->module_pos = 0;
288         snapshot_next_module( snapshot, reply );
289         release_object( snapshot );
290     }
291 }