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

/*
 ============================================================================
 Name        : hev-circular-buffer.c
 Author      : Heiher <r@hev.cc>
 Copyright   : Copyright (c) 2019 everyone.
 Description : Circular buffer
 ============================================================================
 */

#include "lib/misc/hev-compiler.h"
#include "mem/api/hev-memory-allocator-api.h"

#include "hev-circular-buffer.h"

struct _HevCircularBuffer
{
    size_t rp;
    size_t use_size;
    size_t max_size;
    unsigned int ref_count;

    unsigned char data[0];
};

EXPORT_SYMBOL HevCircularBuffer *
hev_circular_buffer_new (size_t max_size)
{
    HevCircularBuffer *self;

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

    self->rp = 0;
    self->use_size = 0;
    self->max_size = max_size;
    self->ref_count = 1;

    return self;
}

EXPORT_SYMBOL HevCircularBuffer *
hev_circular_buffer_ref (HevCircularBuffer *self)
{
    self->ref_count++;

    return self;
}

EXPORT_SYMBOL void
hev_circular_buffer_unref (HevCircularBuffer *self)
{
    self->ref_count--;
    if (self->ref_count)
        return;

    hev_free (self);
}

EXPORT_SYMBOL size_t
hev_circular_buffer_get_max_size (HevCircularBuffer *self)
{
    return self->max_size;
}

EXPORT_SYMBOL size_t
hev_circular_buffer_get_use_size (HevCircularBuffer *self)
{
    return self->use_size;
}

EXPORT_SYMBOL int
hev_circular_buffer_reading (HevCircularBuffer *self, struct iovec *iov)
{
    size_t upper_size = self->max_size - self->rp;

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

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

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

EXPORT_SYMBOL void
hev_circular_buffer_read_finish (HevCircularBuffer *self, size_t size)
{
    self->rp = (self->rp + size) % self->max_size;
    self->use_size -= size;
}

EXPORT_SYMBOL int
hev_circular_buffer_writing (HevCircularBuffer *self, struct iovec *iov)
{
    size_t wp = (self->rp + self->use_size) % self->max_size;
    size_t upper_size = self->max_size - wp;
    size_t spc_size = self->max_size - self->use_size;

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

    iov[0].iov_base = self->data + 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;
}

EXPORT_SYMBOL void
hev_circular_buffer_write_finish (HevCircularBuffer *self, size_t size)
{
    self->use_size += size;
}