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
|
#include <stdint.h>
#include <stdbool.h>
#include <uv.h>
#include "os/uv_helpers.h"
#include "os/wstream.h"
#include "os/wstream_defs.h"
#include "vim.h"
#include "memory.h"
struct wstream {
uv_stream_t *stream;
// Memory currently used by pending buffers
uint32_t curmem;
// Maximum memory used by this instance
uint32_t maxmem;
// Number of pending requests
uint32_t pending_reqs;
bool freed;
};
typedef struct {
WStream *wstream;
// Buffer containing data to be written
char *buffer;
// Size of the buffer
uint32_t length;
// If it's our responsibility to free the buffer
bool free;
} WriteData;
static void write_cb(uv_write_t *req, int status);
WStream * wstream_new(uint32_t maxmem)
{
WStream *rv = xmalloc(sizeof(WStream));
rv->maxmem = maxmem;
rv->stream = NULL;
rv->curmem = 0;
rv->pending_reqs = 0;
rv->freed = false;
return rv;
}
void wstream_free(WStream *wstream)
{
if (!wstream->pending_reqs) {
free(wstream);
} else {
wstream->freed = true;
}
}
void wstream_set_stream(WStream *wstream, uv_stream_t *stream)
{
handle_set_wstream((uv_handle_t *)stream, wstream);
wstream->stream = stream;
}
bool wstream_write(WStream *wstream, char *buffer, uint32_t length, bool free)
{
WriteData *data;
uv_buf_t uvbuf;
uv_write_t *req;
if (wstream->freed) {
// Don't accept write requests after the WStream instance was freed
return false;
}
if (wstream->curmem + length > wstream->maxmem) {
return false;
}
if (free) {
// We should only account for buffers that are ours to free
wstream->curmem += length;
}
data = xmalloc(sizeof(WriteData));
data->wstream = wstream;
data->buffer = buffer;
data->length = length;
data->free = free;
req = xmalloc(sizeof(uv_write_t));
req->data = data;
uvbuf.base = buffer;
uvbuf.len = length;
wstream->pending_reqs++;
uv_write(req, wstream->stream, &uvbuf, 1, write_cb);
return true;
}
static void write_cb(uv_write_t *req, int status)
{
WriteData *data = req->data;
free(req);
if (data->free) {
// Free the data written to the stream
free(data->buffer);
data->wstream->curmem -= data->length;
}
data->wstream->pending_reqs--;
if (data->wstream->freed && data->wstream->pending_reqs == 0) {
// Last pending write, free the wstream;
free(data->wstream);
}
free(data);
}
|
