tun2socks 0.1.10

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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200

/*
 ============================================================================
 Name        : cio-buffer.c
 Author      : hev <r@hev.cc>
 Copyright   : Copyright (c) 2024 - 2025 hev.
 Description : CIO Buffer Test
 ============================================================================
 */

#include <assert.h>
#include <string.h>

#include <hev-task.h>
#include <hev-task-io.h>
#include <hev-task-io-socket.h>
#include <hev-task-cio-socket.h>
#include <hev-task-cio-buffer.h>
#include <hev-task-system.h>

static int fds1[2];
static int fds2[2];

static void
task_splice_entry (void *data)
{
    HevTask *task = hev_task_self ();
    HevTaskCIO *a, *b, *t;

    assert (hev_task_add_fd (task, fds1[1], POLLIN | POLLOUT) == 0);
    assert (hev_task_add_fd (task, fds2[0], POLLIN | POLLOUT) == 0);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds1[1]));
    assert (t != NULL);
    a = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (a != NULL);
    a = hev_task_cio_push (a, t);
    assert (a != NULL);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds2[0]));
    assert (t != NULL);
    b = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (b != NULL);
    b = hev_task_cio_push (b, t);
    assert (b != NULL);

    assert (hev_task_cio_get_fd (a, 0) == fds1[1]);
    assert (hev_task_cio_get_fd (a, 1) == fds1[1]);
    assert (hev_task_cio_get_fd (b, 0) == fds2[0]);
    assert (hev_task_cio_get_fd (b, 1) == fds2[0]);

    hev_task_cio_splice (a, b, 2048, NULL, NULL);

    assert (hev_task_del_fd (task, fds1[1]) == 0);
    assert (hev_task_del_fd (task, fds2[0]) == 0);

    hev_object_unref (HEV_OBJECT (a));
    hev_object_unref (HEV_OBJECT (b));
}

static void
task_entry (void *data)
{
    HevTask *task = hev_task_self ();
    HevTaskCIO *a, *b, *t;
    struct iovec iov[2];
    char buf1[64];
    char buf2[64];
    ssize_t size;
    int fds3[2];
    int res;

    res = hev_task_io_socket_socketpair (PF_LOCAL, SOCK_STREAM, 0, fds1);
    assert (res == 0);
    res = hev_task_io_socket_socketpair (PF_LOCAL, SOCK_STREAM, 0, fds2);
    assert (res == 0);

    assert (hev_task_add_fd (task, fds1[0], POLLOUT) == 0);
    assert (hev_task_add_fd (task, fds2[1], POLLIN) == 0);

    task = hev_task_new (-1);
    assert (task);
    hev_task_run (task, task_splice_entry, NULL);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds1[0]));
    assert (t != NULL);
    a = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (a != NULL);
    a = hev_task_cio_push (a, t);
    assert (a != NULL);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds2[1]));
    assert (t != NULL);
    b = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (b != NULL);
    b = hev_task_cio_push (b, t);
    assert (b != NULL);

    size = hev_task_cio_write_exact (a, buf1, 64, NULL, NULL);
    assert (size == 64);
    res = hev_task_cio_flush (a, NULL, NULL);
    assert (res == 0);
    size = hev_task_cio_read_exact (b, buf2, 64, NULL, NULL);
    assert (size == 64);
    assert (memcmp (buf1, buf2, 64) == 0);

    iov[0].iov_base = buf1;
    iov[0].iov_len = 64;
    size = hev_task_cio_writev_exact (a, iov, 1, NULL, NULL);
    assert (size == 64);
    res = hev_task_cio_flush (a, NULL, NULL);
    assert (res == 0);
    iov[0].iov_base = buf2;
    iov[0].iov_len = 64;
    size = hev_task_cio_readv_exact (b, iov, 1, NULL, NULL);
    assert (size == 64);
    assert (memcmp (buf1, buf2, 64) == 0);

    size = hev_task_cio_write_exact (a, buf1, 64, NULL, NULL);
    assert (size == 64);
    res = hev_task_cio_flush (a, NULL, NULL);
    assert (res == 0);
    for (;;) {
        size = hev_task_cio_buffer_peek (HEV_TASK_CIO_BUFFER (b), buf2, 64);
        if (size == 64)
            break;
        hev_task_yield (HEV_TASK_WAITIO);
    }
    assert (size == 64);
    assert (memcmp (buf1, buf2, 64) == 0);
    res = hev_task_cio_buffer_peekv (HEV_TASK_CIO_BUFFER (b), iov);
    assert (res == 1);
    assert (iov[0].iov_len == 64);
    assert (memcmp (buf1, buf2, 64) == 0);

    assert (hev_task_del_fd (task, fds1[0]) == 0);
    assert (hev_task_del_fd (task, fds2[1]) == 0);

    hev_object_unref (HEV_OBJECT (a));
    hev_object_unref (HEV_OBJECT (b));

    res = hev_task_io_socket_socketpair (PF_LOCAL, SOCK_DGRAM, 0, fds3);
    assert (res == 0);

    assert (hev_task_add_fd (task, fds3[0], POLLIN) == 0);
    assert (hev_task_add_fd (task, fds3[1], POLLOUT) == 0);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds3[0]));
    assert (t != NULL);
    a = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (a != NULL);
    a = hev_task_cio_push (a, t);
    assert (a != NULL);

    t = HEV_TASK_CIO (hev_task_cio_socket_new (PF_LOCAL, fds3[1]));
    assert (t != NULL);
    b = HEV_TASK_CIO (hev_task_cio_buffer_new (2048, 2048));
    assert (b != NULL);
    b = hev_task_cio_push (b, t);
    assert (b != NULL);

    size = hev_task_cio_write_dgram (a, buf1, 64, NULL, NULL, NULL);
    assert (size == 64);
    size = hev_task_cio_read_dgram (b, buf2, 64, NULL, NULL, NULL);
    assert (size == 64);
    assert (memcmp (buf1, buf2, 64) == 0);

    iov[0].iov_base = buf1;
    iov[0].iov_len = 64;
    size = hev_task_cio_writev_dgram (a, iov, 1, NULL, NULL, NULL);
    assert (size == 64);
    iov[0].iov_base = buf2;
    iov[0].iov_len = 64;
    size = hev_task_cio_readv_dgram (b, iov, 1, NULL, NULL, NULL);
    assert (size == 64);
    assert (memcmp (buf1, buf2, 64) == 0);

    assert (hev_task_del_fd (task, fds3[0]) == 0);
    assert (hev_task_del_fd (task, fds3[1]) == 0);

    hev_object_unref (HEV_OBJECT (a));
    hev_object_unref (HEV_OBJECT (b));
}

int
main (int argc, char *argv[])
{
    HevTask *task;

    assert (hev_task_system_init () == 0);

    task = hev_task_new (-1);
    assert (task);
    hev_task_run (task, task_entry, NULL);

    hev_task_system_run ();

    hev_task_system_fini ();

    return 0;
}