src/kernel/timer.cc


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2026 Richard QIn
 */
#include "common/types.h"
#include "kernel/config.h"
#include "kernel/spinlock.h"
#include "kernel/proc.h"
#include "kernel/sbi.h"

#define INTERVAL 1000000

namespace Timer
{
    uint64 ticks;
    Spinlock tickslock;

    uint64 get_time()
    {
        uint64 time;
        asm volatile("rdtime %0" : "=r"(time));
        return time;
    }

    void set_next_trigger()
    {
        SBI::set_timer(get_time() + INTERVAL);
    }

    void init()
    {
        set_next_trigger();

        uint64 sie;
        asm volatile("csrr %0, sie" : "=r"(sie));
        sie |= (1L << 5);
        asm volatile("csrw sie, %0" : : "r"(sie));
    }

    void tick()
    {
        tickslock.acquire();
        ticks++;
        ProcManager::wakeup(&ticks);
        tickslock.release();
        set_next_trigger();
    }

    uint64 get_ticks()
    {
        tickslock.acquire();
        uint64 t = ticks;
        tickslock.release();
        return t;
    }

    Spinlock *get_lock()
    {
        return &tickslock;
    }

    void *get_tick_chan()
    {
        return &ticks;
    }
}