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
114
115
116
117
118
119
120
121
122
123
124
125
126

/*
 ============================================================================
 Name        : hev-ring-buffer.c
 Author      : hev <r@hev.cc>
 Copyright   : Copyright (c) 2023 hev
 Description : Ring buffer
 ============================================================================
 */

#include <hev-memory-allocator.h>

#include "hev-compiler.h"
#include "hev-ring-buffer.h"

struct _HevRingBuffer
{
    size_t rp;
    size_t wp;
    size_t rda_size;
    size_t use_size;
    size_t max_size;

    unsigned char data[0];
};

HevRingBuffer *
hev_ring_buffer_new (size_t max_size)
{
    HevRingBuffer *self;

    self = hev_malloc (sizeof (HevRingBuffer) + max_size);
    if (!self)
        return NULL;

    self->rp = 0;
    self->wp = 0;
    self->rda_size = 0;
    self->use_size = 0;
    self->max_size = max_size;

    return self;
}

void
hev_ring_buffer_destroy (HevRingBuffer *self)
{
    hev_free (self);
}

size_t
hev_ring_buffer_get_max_size (HevRingBuffer *self)
{
    return self->max_size;
}

size_t
hev_ring_buffer_get_use_size (HevRingBuffer *self)
{
    return self->use_size;
}

int
hev_ring_buffer_reading (HevRingBuffer *self, struct iovec *iov)
{
    size_t upper_size;

    if (0 == self->rda_size)
        return 0;

    upper_size = self->max_size - self->rp;

    iov[0].iov_base = self->data + self->rp;
    if (self->rda_size <= upper_size) {
        iov[0].iov_len = self->rda_size;
        return 1;
    }

    iov[0].iov_len = upper_size;
    iov[1].iov_base = self->data;
    iov[1].iov_len = self->rda_size - upper_size;
    return 2;
}

void
hev_ring_buffer_read_finish (HevRingBuffer *self, size_t size)
{
    self->rp = (self->rp + size) % self->max_size;
    self->rda_size -= size;
}

void
hev_ring_buffer_read_release (HevRingBuffer *self, size_t size)
{
    self->use_size -= size;
}

int
hev_ring_buffer_writing (HevRingBuffer *self, struct iovec *iov)
{
    size_t upper_size, spc_size;

    if (self->use_size == self->max_size)
        return 0;

    upper_size = self->max_size - self->wp;
    spc_size = self->max_size - self->use_size;

    iov[0].iov_base = self->data + self->wp;
    if (spc_size <= upper_size) {
        iov[0].iov_len = spc_size;
        return 1;
    }

    iov[0].iov_len = upper_size;
    iov[1].iov_base = self->data;
    iov[1].iov_len = spc_size - upper_size;
    return 2;
}

void
hev_ring_buffer_write_finish (HevRingBuffer *self, size_t size)
{
    self->wp = (self->wp + size) % self->max_size;
    self->use_size += size;
    self->rda_size += size;
}