-
Notifications
You must be signed in to change notification settings - Fork 0
/
util_memwy.h
158 lines (143 loc) · 3.73 KB
/
util_memwy.h
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
#ifndef UTIL_MEMWY_H
#define UTIL_MEMWY_H
#include<stdio.h>
#include<stdlib.h>
#include<stdint.h>
#include<string.h>
typedef size_t pos_t;
typedef struct {
size_t avai;
size_t used;
pos_t pos;
char* mem;
} memowy_t;
#define MEMOWYBUF 512
#define data(memwy) ((memwy)->mem + (memwy)->pos + 2)
#define lenp(memwy) ((memwy)->mem + (memwy)->pos)
#define entrym(memwy, ind) (*((memwy)->mem + (memwy)->pos + 2 + (ind)))
uint16_t lenwy(char* lp){
uint16_t len; memcpy(&len, lp, 2); return len;
}
uint16_t lengf(memowy_t* mp){
char* lp = lenp(mp);
uint16_t len; memcpy(&len, lp, 2); return len;
}
void* clearmemwy(memowy_t* mp){
mp->avai = 0;
mp->used = 0;
mp->pos = 0;
free(mp->mem);
return NULL;
}
size_t allocmemwy(memowy_t* mp, void* src, uint16_t len){
mp->used += len + 4; // 2*sizeof(uint16_t)
while (mp->used > mp->avai) {
mp->avai += MEMOWYBUF;
mp->mem = realloc(mp->mem, mp->avai);
}
memcpy(mp->mem + mp->used - (len + 4), &len, 2);
memcpy(mp->mem + mp->used - 2, &len, 2);
memmove(mp->mem + mp->used - (len + 2), src, len);
return mp->used - (len + 4);
}
size_t freememwy(memowy_t* mp){
if(mp->used == 0) return 0;
mp->used -= lenwy(mp->mem + mp->used - 2) + 4;
return mp->used - lenwy(mp->mem + mp->used);
}
size_t allocstrmemwy(memowy_t* mp, void* src, uint16_t slen){
uint16_t alen = slen + 1;
mp->used += alen + 4; // 2*sizeof(uint16_t)
while (mp->used > mp->avai) {
mp->avai += MEMOWYBUF;
mp->mem = realloc(mp->mem, mp->avai);
}
memcpy(mp->mem + mp->used - (alen + 4), &alen, 2);
memcpy(mp->mem + mp->used - 2, &alen, 2);
memmove(mp->mem + mp->used - (alen + 2), src, slen);
*(mp->mem + mp->used - 3) = '\0';
return mp->used - (alen + 4);
}
size_t reallocmemwy(memowy_t* mp, uint16_t len){
uint16_t oldlen = lenwy(lenp(mp));
size_t remregionlen = mp->used - mp->pos - oldlen - 4;
mp->used += len - oldlen;
while (mp->used > mp->avai) {
mp->avai += MEMOWYBUF;
mp->mem = realloc(mp->mem, mp->avai);
}
char* loc = lenp(mp);
memmove(loc + len + 4, loc + oldlen + 4, remregionlen);
memcpy(loc, &len, 2);
memcpy(loc + len + 2, &len, 2);
return len;
}
size_t appmemwy(memowy_t* mp, void* src, uint16_t len){
uint16_t oldlen = lenwy(lenp(mp));
reallocmemwy(mp, oldlen + len);
memmove(data(mp) + oldlen, src, len);
return len;
}
int indexmemwy(memowy_t* mp, int steps){
size_t len = 0;
while(steps != 0){
if(steps < 0){
if(mp->pos == 0) break;
memcpy(&len, lenp(mp) - 2, 2);
mp->pos -= len + 4;
steps++;
} else {
memcpy(&len, lenp(mp), 2);
if(mp->pos + len + 4 >= mp->used){break;}
mp->pos += len + 4;
steps--;
}
}
return steps;
}
void printmemwy(memowy_t* mp){
pos_t oldpos = mp->pos;
mp->pos = 0;
do {
*(data(mp) + lenwy(lenp(mp))) = '\0';
if(oldpos == mp->pos) printf("\'");
printf("%u: %s\n", lengf(mp), data(mp));
memcpy(data(mp) + lenwy(lenp(mp)), lenp(mp), 2);
} while(indexmemwy(mp, 1) == 0);
mp->pos = oldpos;
}
pos_t findmemwy(memowy_t* mp, pos_t target){
pos_t oldpos = mp->pos;
mp->pos = 0;
pos_t find = mp->avai;
do {
if(mp->pos == target) continue;
if(lenwy(mp->mem + target) != lenwy(lenp(mp))) continue;
for(size_t i = 0; i < lenwy(lenp(mp)); i++)
if(*(data(mp) + i) != *(mp->mem + target + 2 + i))
goto haha;
// printf("same: %s\n", data(mp));
find = mp->pos;
break;
haha:;
} while(indexmemwy(mp, 1) == 0);
mp->pos = oldpos;
return find;
}
pos_t sfindmemwy(memowy_t* mp, pos_t target, int step){
pos_t oldpos = mp->pos;
pos_t find = mp->avai;
do {
if(mp->pos == target) continue;
if(lenwy(mp->mem + target) != lengf(mp)) continue;
for(size_t i = 0; i < lengf(mp); i++)
if(entrym(mp, i) != *(mp->mem + target + 2 + i))
goto haha;
find = mp->pos;
break;
haha:;
} while(indexmemwy(mp, step) == 0);
mp->pos = oldpos;
return find;
}
#endif