tun2socks 0.0.1

High performance tun2socks
Documentation 
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

/*
 ============================================================================
 Name        : hev-task-cond.c
 Author      : Heiher <r@hev.cc>
 Copyright   : Copyright (c) 2019 everyone.
 Description : Condition
 ============================================================================
 */

#include <stddef.h>

#include "kern/task/hev-task.h"
#include "lib/misc/hev-compiler.h"

#include "hev-task-cond.h"

struct _HevTaskCondNode
{
    HevTaskCondNode *prev;
    HevTaskCondNode *next;

    HevTask *task;
};

EXPORT_SYMBOL int
hev_task_cond_init (HevTaskCond *self)
{
    self->waiters = NULL;

    return 0;
}

EXPORT_SYMBOL int
hev_task_cond_wait (HevTaskCond *self, HevTaskMutex *mutex)
{
    HevTaskCondNode node;

    node.prev = NULL;
    node.next = self->waiters;
    node.task = hev_task_self ();

    if (self->waiters)
        self->waiters->prev = &node;
    self->waiters = &node;

    hev_task_mutex_unlock (mutex);
    do {
        hev_task_yield (HEV_TASK_WAITIO);
    } while (READ_ONCE (node.task));
    hev_task_mutex_lock (mutex);

    return 0;
}

EXPORT_SYMBOL int
hev_task_cond_timedwait (HevTaskCond *self, HevTaskMutex *mutex,
                         unsigned int milliseconds)
{
    HevTaskCondNode node;

    node.prev = NULL;
    node.next = self->waiters;
    node.task = hev_task_self ();

    if (self->waiters)
        self->waiters->prev = &node;
    self->waiters = &node;

    hev_task_mutex_unlock (mutex);
    while (milliseconds && READ_ONCE (node.task))
        milliseconds = hev_task_sleep (milliseconds);
    hev_task_mutex_lock (mutex);

    if (READ_ONCE (node.task)) {
        if (node.prev)
            node.prev->next = node.next;
        if (node.next)
            node.next->prev = node.prev;
        if (self->waiters == &node)
            self->waiters = NULL;
        return -1;
    }

    return 0;
}

EXPORT_SYMBOL int
hev_task_cond_signal (HevTaskCond *self)
{
    if (self->waiters) {
        hev_task_wakeup (self->waiters->task);

        self->waiters->task = NULL;
        self->waiters = self->waiters->next;
        if (self->waiters)
            self->waiters->prev = NULL;
    }

    return 0;
}

EXPORT_SYMBOL int
hev_task_cond_broadcast (HevTaskCond *self)
{
    while (self->waiters) {
        hev_task_wakeup (self->waiters->task);

        self->waiters->task = NULL;
        self->waiters = self->waiters->next;
    }

    return 0;
}