timer.c 9.16 KB
Newer Older
1 2 3 4
/*
 * Waitable timers management
 *
 * Copyright (C) 1999 Alexandre Julliard
5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * 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
 */

21 22 23
#include "config.h"
#include "wine/port.h"

24 25 26 27 28
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/time.h>
#include <sys/types.h>
29
#include <stdarg.h>
30

31 32
#include "ntstatus.h"
#define WIN32_NO_STATUS
33
#include "windef.h"
34
#include "winternl.h"
35 36

#include "file.h"
37 38 39 40 41 42 43 44 45 46 47
#include "handle.h"
#include "request.h"

struct timer
{
    struct object        obj;       /* object header */
    int                  manual;    /* manual reset */
    int                  signaled;  /* current signaled state */
    int                  period;    /* timer period in ms */
    struct timeval       when;      /* next expiration */
    struct timeout_user *timeout;   /* timeout user */
48
    struct thread       *thread;    /* thread that set the APC function */
49 50 51 52 53 54 55
    void                *callback;  /* callback APC function */
    void                *arg;       /* callback argument */
};

static void timer_dump( struct object *obj, int verbose );
static int timer_signaled( struct object *obj, struct thread *thread );
static int timer_satisfied( struct object *obj, struct thread *thread );
56
static unsigned int timer_map_access( struct object *obj, unsigned int access );
57 58 59 60
static void timer_destroy( struct object *obj );

static const struct object_ops timer_ops =
{
61 62 63 64 65 66
    sizeof(struct timer),      /* size */
    timer_dump,                /* dump */
    add_queue,                 /* add_queue */
    remove_queue,              /* remove_queue */
    timer_signaled,            /* signaled */
    timer_satisfied,           /* satisfied */
67
    no_signal,                 /* signal */
68
    no_get_fd,                 /* get_fd */
69
    timer_map_access,          /* map_access */
70
    no_lookup_name,            /* lookup_name */
71
    no_close_handle,           /* close_handle */
72
    timer_destroy              /* destroy */
73 74 75 76
};


/* create a timer object */
77 78
static struct timer *create_timer( struct directory *root, const struct unicode_str *name,
                                   unsigned int attr, int manual )
79 80 81
{
    struct timer *timer;

82
    if ((timer = create_named_object_dir( root, name, attr, &timer_ops )))
83
    {
84
        if (get_error() != STATUS_OBJECT_NAME_EXISTS)
85 86 87 88 89 90 91 92
        {
            /* initialize it if it didn't already exist */
            timer->manual       = manual;
            timer->signaled     = 0;
            timer->when.tv_sec  = 0;
            timer->when.tv_usec = 0;
            timer->period       = 0;
            timer->timeout      = NULL;
93
            timer->thread       = NULL;
94 95 96 97 98 99 100 101 102 103
        }
    }
    return timer;
}

/* callback on timer expiration */
static void timer_callback( void *private )
{
    struct timer *timer = (struct timer *)private;

104 105
    /* queue an APC */
    if (timer->thread)
106
    {
107
        if (!thread_queue_apc( timer->thread, &timer->obj, timer->callback, APC_TIMER, 0,
108 109 110 111 112 113
                               (void *)timer->when.tv_sec, (void *)timer->when.tv_usec, timer->arg))
        {
            release_object( timer->thread );
            timer->thread = NULL;
        }
    }
114

115 116
    if (timer->period)  /* schedule the next expiration */
    {
117
        add_timeout( &timer->when, timer->period );
118 119 120 121 122 123 124 125 126
        timer->timeout = add_timeout_user( &timer->when, timer_callback, timer );
    }
    else timer->timeout = NULL;

    /* wake up waiters */
    timer->signaled = 1;
    wake_up( &timer->obj, 0 );
}

127
/* cancel a running timer */
128
static int cancel_timer( struct timer *timer )
129
{
130 131
    int signaled = timer->signaled;

132 133 134 135 136 137 138
    if (timer->timeout)
    {
        remove_timeout_user( timer->timeout );
        timer->timeout = NULL;
    }
    if (timer->thread)
    {
139
        thread_cancel_apc( timer->thread, &timer->obj, 0 );
140
        release_object( timer->thread );
141 142
        timer->thread = NULL;
    }
143
    return signaled;
144 145
}

146
/* set the timer expiration and period */
147 148
static int set_timer( struct timer *timer, const abs_time_t *expire, int period,
                      void *callback, void *arg )
149
{
150
    int signaled = cancel_timer( timer );
151 152 153 154 155
    if (timer->manual)
    {
        period = 0;  /* period doesn't make any sense for a manual timer */
        timer->signaled = 0;
    }
156
    if (!expire->sec && !expire->usec)
157 158
    {
        /* special case: use now + period as first expiration */
159
        gettimeofday( &timer->when, NULL );
160 161 162 163
        add_timeout( &timer->when, period );
    }
    else
    {
164 165
        timer->when.tv_sec  = expire->sec;
        timer->when.tv_usec = expire->usec;
166
    }
167 168 169
    timer->period       = period;
    timer->callback     = callback;
    timer->arg          = arg;
170
    if (callback) timer->thread = (struct thread *)grab_object( current );
171
    timer->timeout = add_timeout_user( &timer->when, timer_callback, timer );
172
    return signaled;
173 174 175 176 177 178
}

static void timer_dump( struct object *obj, int verbose )
{
    struct timer *timer = (struct timer *)obj;
    assert( obj->ops == &timer_ops );
179 180
    fprintf( stderr, "Timer manual=%d when=%ld.%06u period=%d ",
             timer->manual, timer->when.tv_sec, (unsigned int)timer->when.tv_usec, timer->period );
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199
    dump_object_name( &timer->obj );
    fputc( '\n', stderr );
}

static int timer_signaled( struct object *obj, struct thread *thread )
{
    struct timer *timer = (struct timer *)obj;
    assert( obj->ops == &timer_ops );
    return timer->signaled;
}

static int timer_satisfied( struct object *obj, struct thread *thread )
{
    struct timer *timer = (struct timer *)obj;
    assert( obj->ops == &timer_ops );
    if (!timer->manual) timer->signaled = 0;
    return 0;
}

200 201 202 203 204 205 206 207 208
static unsigned int timer_map_access( struct object *obj, unsigned int access )
{
    if (access & GENERIC_READ)    access |= STANDARD_RIGHTS_READ | SYNCHRONIZE | TIMER_QUERY_STATE;
    if (access & GENERIC_WRITE)   access |= STANDARD_RIGHTS_WRITE | TIMER_MODIFY_STATE;
    if (access & GENERIC_EXECUTE) access |= STANDARD_RIGHTS_EXECUTE;
    if (access & GENERIC_ALL)     access |= TIMER_ALL_ACCESS;
    return access & ~(GENERIC_READ | GENERIC_WRITE | GENERIC_EXECUTE | GENERIC_ALL);
}

209 210 211 212 213 214
static void timer_destroy( struct object *obj )
{
    struct timer *timer = (struct timer *)obj;
    assert( obj->ops == &timer_ops );

    if (timer->timeout) remove_timeout_user( timer->timeout );
215
    if (timer->thread) release_object( timer->thread );
216 217 218 219 220 221
}

/* create a timer */
DECL_HANDLER(create_timer)
{
    struct timer *timer;
222
    struct unicode_str name;
223
    struct directory *root = NULL;
224

225
    reply->handle = 0;
226
    get_req_unicode_str( &name );
227 228 229 230
    if (req->rootdir && !(root = get_directory_obj( current->process, req->rootdir, 0 )))
        return;

    if ((timer = create_timer( root, &name, req->attributes, req->manual )))
231
    {
232
        reply->handle = alloc_handle( current->process, timer, req->access, req->attributes );
233 234
        release_object( timer );
    }
235 236

    if (root) release_object( root );
237 238 239 240 241
}

/* open a handle to a timer */
DECL_HANDLER(open_timer)
{
242
    struct unicode_str name;
243
    struct directory *root = NULL;
244
    struct timer *timer;
245 246

    get_req_unicode_str( &name );
247 248 249
    if (req->rootdir && !(root = get_directory_obj( current->process, req->rootdir, 0 )))
        return;

250 251
    if ((timer = open_object_dir( root, &name, req->attributes, &timer_ops )))
    {
252
        reply->handle = alloc_handle( current->process, &timer->obj, req->access, req->attributes );
253 254
        release_object( timer );
    }
255 256

    if (root) release_object( root );
257 258 259 260 261 262 263 264 265 266
}

/* set a waitable timer */
DECL_HANDLER(set_timer)
{
    struct timer *timer;

    if ((timer = (struct timer *)get_handle_obj( current->process, req->handle,
                                                 TIMER_MODIFY_STATE, &timer_ops )))
    {
267
        reply->signaled = set_timer( timer, &req->expire, req->period, req->callback, req->arg );
268 269 270 271 272 273 274 275 276 277 278 279
        release_object( timer );
    }
}

/* cancel a waitable timer */
DECL_HANDLER(cancel_timer)
{
    struct timer *timer;

    if ((timer = (struct timer *)get_handle_obj( current->process, req->handle,
                                                 TIMER_MODIFY_STATE, &timer_ops )))
    {
280
        reply->signaled = cancel_timer( timer );
281 282 283
        release_object( timer );
    }
}
284 285 286 287 288 289 290 291 292 293 294 295 296 297 298

/* Get information on a waitable timer */
DECL_HANDLER(get_timer_info)
{
    struct timer *timer;

    if ((timer = (struct timer *)get_handle_obj( current->process, req->handle,
                                                 TIMER_QUERY_STATE, &timer_ops )))
    {
        reply->when.sec  = timer->when.tv_sec;
        reply->when.usec = timer->when.tv_usec;
        reply->signaled  = timer->signaled;
        release_object( timer );
    }
}