1 #include <u.h>
2 #include <libc.h>
3 #include <venti.h>
4
5 enum {
6 IdealAlignment = 32,
7 ChunkSize = 128*1024
8 };
9
10
11 void
vtfree(void * p)12 vtfree(void *p)
13 {
14 if(p == 0)
15 return;
16 free(p);
17 }
18
19 void *
vtmalloc(int size)20 vtmalloc(int size)
21 {
22 void *p;
23
24 p = malloc(size);
25 if(p == 0)
26 sysfatal("vtmalloc: out of memory");
27 setmalloctag(p, getcallerpc(&size));
28 return p;
29 }
30
31 void *
vtmallocz(int size)32 vtmallocz(int size)
33 {
34 void *p = vtmalloc(size);
35 memset(p, 0, size);
36 setmalloctag(p, getcallerpc(&size));
37 return p;
38 }
39
40 void *
vtrealloc(void * p,int size)41 vtrealloc(void *p, int size)
42 {
43 if(p == nil)
44 return vtmalloc(size);
45 p = realloc(p, size);
46 if(p == 0)
47 sysfatal("vtMemRealloc: out of memory");
48 setrealloctag(p, getcallerpc(&size));
49 return p;
50 }
51
52 void *
vtbrk(int n)53 vtbrk(int n)
54 {
55 static Lock lk;
56 static uchar *buf;
57 static int nbuf, nchunk;
58 int align, pad;
59 void *p;
60
61 if(n >= IdealAlignment)
62 align = IdealAlignment;
63 else if(n > 8)
64 align = 8;
65 else
66 align = 4;
67
68 lock(&lk);
69 pad = (align - (uintptr)buf) & (align-1);
70 if(n + pad > nbuf) {
71 buf = vtmallocz(ChunkSize);
72 nbuf = ChunkSize;
73 pad = (align - (uintptr)buf) & (align-1);
74 nchunk++;
75 }
76
77 assert(n + pad <= nbuf);
78
79 p = buf + pad;
80 buf += pad + n;
81 nbuf -= pad + n;
82 unlock(&lk);
83
84 return p;
85 }
86