1 0faf0f0b 2005-02-13 devnull #include <u.h>
  2 0faf0f0b 2005-02-13 devnull #include <libc.h>
  3 0faf0f0b 2005-02-13 devnull #include <venti.h>
  4 0faf0f0b 2005-02-13 devnull 
  5 5ddc97fc 2005-02-14 devnull int ventilogging;
  6 0faf0f0b 2005-02-13 devnull #define log	not_the_log_library_call
  7 0faf0f0b 2005-02-13 devnull 
  8 5ddc97fc 2005-02-14 devnull static char Eremoved[] = "[removed]";
  9 5ddc97fc 2005-02-14 devnull 
 10 0faf0f0b 2005-02-13 devnull enum
 11 0faf0f0b 2005-02-13 devnull {	/* defaults */
 12 0faf0f0b 2005-02-13 devnull 	LogChunkSize = 8192,
 13 cbeb0b26 2006-04-01 devnull 	LogSize = 65536
 14 0faf0f0b 2005-02-13 devnull };
 15 0faf0f0b 2005-02-13 devnull 
 16 0faf0f0b 2005-02-13 devnull static struct {
 17 0faf0f0b 2005-02-13 devnull 	QLock lk;
 18 0faf0f0b 2005-02-13 devnull 	VtLog *hash[1024];
 19 0faf0f0b 2005-02-13 devnull } vl;
 20 0faf0f0b 2005-02-13 devnull 
 21 0faf0f0b 2005-02-13 devnull static uint
 22 0faf0f0b 2005-02-13 devnull hash(char *s)
 23 0faf0f0b 2005-02-13 devnull {
 24 0faf0f0b 2005-02-13 devnull 	uint h;
 25 0faf0f0b 2005-02-13 devnull 	uchar *p;
 26 0faf0f0b 2005-02-13 devnull 
 27 0faf0f0b 2005-02-13 devnull 	h = 0;
 28 0faf0f0b 2005-02-13 devnull 	for(p=(uchar*)s; *p; p++)
 29 0faf0f0b 2005-02-13 devnull 		h = h*37 + *p;
 30 0faf0f0b 2005-02-13 devnull 	return h;
 31 0faf0f0b 2005-02-13 devnull }
 32 0faf0f0b 2005-02-13 devnull 
 33 52006c82 2005-05-12 devnull char**
 34 52006c82 2005-05-12 devnull vtlognames(int *pn)
 35 52006c82 2005-05-12 devnull {
 36 52006c82 2005-05-12 devnull 	int i, nname, size;
 37 52006c82 2005-05-12 devnull 	VtLog *l;
 38 52006c82 2005-05-12 devnull 	char **s, *a, *e;
 39 52006c82 2005-05-12 devnull 	
 40 52006c82 2005-05-12 devnull 	qlock(&vl.lk);
 41 52006c82 2005-05-12 devnull 	size = 0;
 42 52006c82 2005-05-12 devnull 	nname = 0;
 43 52006c82 2005-05-12 devnull 	for(i=0; i<nelem(vl.hash); i++)
 44 52006c82 2005-05-12 devnull 	for(l=vl.hash[i]; l; l=l->next){
 45 52006c82 2005-05-12 devnull 		nname++;
 46 52006c82 2005-05-12 devnull 		size += strlen(l->name)+1;
 47 52006c82 2005-05-12 devnull 	}
 48 52006c82 2005-05-12 devnull 	
 49 52006c82 2005-05-12 devnull 	s = vtmalloc(nname*sizeof(char*)+size);
 50 52006c82 2005-05-12 devnull 	a = (char*)(s+nname);
 51 52006c82 2005-05-12 devnull 	e = (char*)s+nname*sizeof(char*)+size;
 52 52006c82 2005-05-12 devnull 
 53 52006c82 2005-05-12 devnull 	nname = 0;
 54 52006c82 2005-05-12 devnull 	for(i=0; i<nelem(vl.hash); i++)
 55 52006c82 2005-05-12 devnull 	for(l=vl.hash[i]; l; l=l->next){
 56 52006c82 2005-05-12 devnull 		strcpy(a, l->name);
 57 52006c82 2005-05-12 devnull 		s[nname++] = a;
 58 52006c82 2005-05-12 devnull 		a += strlen(a)+1;
 59 52006c82 2005-05-12 devnull 	}
 60 52006c82 2005-05-12 devnull 	*pn = nname;
 61 52006c82 2005-05-12 devnull 	assert(a == e);
 62 52006c82 2005-05-12 devnull 	qunlock(&vl.lk);
 63 52006c82 2005-05-12 devnull 
 64 52006c82 2005-05-12 devnull 	return s;
 65 52006c82 2005-05-12 devnull }
 66 52006c82 2005-05-12 devnull 
 67 0faf0f0b 2005-02-13 devnull VtLog*
 68 0faf0f0b 2005-02-13 devnull vtlogopen(char *name, uint size)
 69 0faf0f0b 2005-02-13 devnull {
 70 0faf0f0b 2005-02-13 devnull 	uint h;
 71 0faf0f0b 2005-02-13 devnull 	int i, nc;
 72 0faf0f0b 2005-02-13 devnull 	char *p;
 73 0faf0f0b 2005-02-13 devnull 	VtLog *l, *last;
 74 0faf0f0b 2005-02-13 devnull 
 75 5ddc97fc 2005-02-14 devnull 	if(!ventilogging)
 76 5ddc97fc 2005-02-14 devnull 		return nil;
 77 5ddc97fc 2005-02-14 devnull 
 78 0faf0f0b 2005-02-13 devnull 	h = hash(name)%nelem(vl.hash);
 79 0faf0f0b 2005-02-13 devnull 	qlock(&vl.lk);
 80 0faf0f0b 2005-02-13 devnull 	last = nil;
 81 0faf0f0b 2005-02-13 devnull 	for(l=vl.hash[h]; l; last=l, l=l->next)
 82 0faf0f0b 2005-02-13 devnull 		if(strcmp(l->name, name) == 0){
 83 0faf0f0b 2005-02-13 devnull 			if(last){	/* move to front */
 84 0faf0f0b 2005-02-13 devnull 				last->next = l->next;
 85 0faf0f0b 2005-02-13 devnull 				l->next = vl.hash[h];
 86 0faf0f0b 2005-02-13 devnull 				vl.hash[h] = l;
 87 0faf0f0b 2005-02-13 devnull 			}
 88 0faf0f0b 2005-02-13 devnull 			l->ref++;
 89 0faf0f0b 2005-02-13 devnull 			qunlock(&vl.lk);
 90 0faf0f0b 2005-02-13 devnull 			return l;
 91 0faf0f0b 2005-02-13 devnull 		}
 92 0faf0f0b 2005-02-13 devnull 
 93 0faf0f0b 2005-02-13 devnull 	if(size == 0){
 94 0faf0f0b 2005-02-13 devnull 		qunlock(&vl.lk);
 95 0faf0f0b 2005-02-13 devnull 		return nil;
 96 0faf0f0b 2005-02-13 devnull 	}
 97 0faf0f0b 2005-02-13 devnull 
 98 0faf0f0b 2005-02-13 devnull 	/* allocate */
 99 0faf0f0b 2005-02-13 devnull 	nc = (size+LogChunkSize-1)/LogChunkSize;
100 0faf0f0b 2005-02-13 devnull 	l = vtmalloc(sizeof *l + nc*(sizeof(*l->chunk)+LogChunkSize) + strlen(name)+1);
101 0faf0f0b 2005-02-13 devnull 	memset(l, 0, sizeof *l);
102 0faf0f0b 2005-02-13 devnull 	l->chunk = (VtLogChunk*)(l+1);
103 0faf0f0b 2005-02-13 devnull 	l->nchunk = nc;
104 0faf0f0b 2005-02-13 devnull 	l->w = l->chunk;
105 0faf0f0b 2005-02-13 devnull 	p = (char*)(l->chunk+nc);
106 0faf0f0b 2005-02-13 devnull 	for(i=0; i<nc; i++){
107 0faf0f0b 2005-02-13 devnull 		l->chunk[i].p = p;
108 5ddc97fc 2005-02-14 devnull 		l->chunk[i].wp = p;
109 0faf0f0b 2005-02-13 devnull 		p += LogChunkSize;
110 0faf0f0b 2005-02-13 devnull 		l->chunk[i].ep = p;
111 0faf0f0b 2005-02-13 devnull 	}
112 0faf0f0b 2005-02-13 devnull 	strcpy(p, name);
113 0faf0f0b 2005-02-13 devnull 	l->name = p;
114 0faf0f0b 2005-02-13 devnull 	
115 0faf0f0b 2005-02-13 devnull 	/* insert */
116 0faf0f0b 2005-02-13 devnull 	l->next = vl.hash[h];
117 0faf0f0b 2005-02-13 devnull 	vl.hash[h] = l;
118 5ddc97fc 2005-02-14 devnull 	l->ref++;
119 0faf0f0b 2005-02-13 devnull 	
120 0faf0f0b 2005-02-13 devnull 	l->ref++;
121 0faf0f0b 2005-02-13 devnull 	qunlock(&vl.lk);
122 0faf0f0b 2005-02-13 devnull 	return l;
123 0faf0f0b 2005-02-13 devnull }
124 0faf0f0b 2005-02-13 devnull 
125 0faf0f0b 2005-02-13 devnull void
126 0faf0f0b 2005-02-13 devnull vtlogclose(VtLog *l)
127 0faf0f0b 2005-02-13 devnull {
128 0faf0f0b 2005-02-13 devnull 	if(l == nil)
129 0faf0f0b 2005-02-13 devnull 		return;
130 0faf0f0b 2005-02-13 devnull 
131 0faf0f0b 2005-02-13 devnull 	qlock(&vl.lk);
132 5ddc97fc 2005-02-14 devnull 	if(--l->ref == 0){
133 5ddc97fc 2005-02-14 devnull 		/* must not be in hash table */
134 5ddc97fc 2005-02-14 devnull 		assert(l->name == Eremoved);
135 0faf0f0b 2005-02-13 devnull 		free(l);
136 5ddc97fc 2005-02-14 devnull 	}else
137 0faf0f0b 2005-02-13 devnull 		assert(l->ref > 0);
138 0faf0f0b 2005-02-13 devnull 	qunlock(&vl.lk);
139 0faf0f0b 2005-02-13 devnull }
140 0faf0f0b 2005-02-13 devnull 
141 0faf0f0b 2005-02-13 devnull void
142 0faf0f0b 2005-02-13 devnull vtlogremove(char *name)
143 0faf0f0b 2005-02-13 devnull {
144 0faf0f0b 2005-02-13 devnull 	uint h;
145 0faf0f0b 2005-02-13 devnull 	VtLog *last, *l;
146 0faf0f0b 2005-02-13 devnull 
147 0faf0f0b 2005-02-13 devnull 	h = hash(name)%nelem(vl.hash);
148 0faf0f0b 2005-02-13 devnull 	qlock(&vl.lk);
149 0faf0f0b 2005-02-13 devnull 	last = nil;
150 0faf0f0b 2005-02-13 devnull 	for(l=vl.hash[h]; l; last=l, l=l->next)
151 0faf0f0b 2005-02-13 devnull 		if(strcmp(l->name, name) == 0){
152 0faf0f0b 2005-02-13 devnull 			if(last)
153 0faf0f0b 2005-02-13 devnull 				last->next = l->next;
154 0faf0f0b 2005-02-13 devnull 			else
155 0faf0f0b 2005-02-13 devnull 				vl.hash[h] = l->next;
156 5ddc97fc 2005-02-14 devnull 			l->name = Eremoved;
157 5ddc97fc 2005-02-14 devnull 			l->next = nil;
158 0faf0f0b 2005-02-13 devnull 			qunlock(&vl.lk);
159 0faf0f0b 2005-02-13 devnull 			vtlogclose(l);
160 0faf0f0b 2005-02-13 devnull 			return;
161 0faf0f0b 2005-02-13 devnull 		}
162 0faf0f0b 2005-02-13 devnull 	qunlock(&vl.lk);
163 0faf0f0b 2005-02-13 devnull }
164 0faf0f0b 2005-02-13 devnull 
165 5ddc97fc 2005-02-14 devnull static int
166 5ddc97fc 2005-02-14 devnull timefmt(Fmt *fmt)
167 5ddc97fc 2005-02-14 devnull {
168 5ddc97fc 2005-02-14 devnull 	static uvlong t0;
169 5ddc97fc 2005-02-14 devnull 	uvlong t;
170 5ddc97fc 2005-02-14 devnull 
171 5ddc97fc 2005-02-14 devnull 	if(t0 == 0)
172 5ddc97fc 2005-02-14 devnull 		t0 = nsec();
173 5ddc97fc 2005-02-14 devnull 	t = nsec()-t0;
174 5ddc97fc 2005-02-14 devnull 	return fmtprint(fmt, "T+%d.%04d", (uint)(t/1000000000), (uint)(t%1000000000)/100000);
175 5ddc97fc 2005-02-14 devnull }
176 5ddc97fc 2005-02-14 devnull 
177 0faf0f0b 2005-02-13 devnull void
178 0faf0f0b 2005-02-13 devnull vtlogvprint(VtLog *l, char *fmt, va_list arg)
179 0faf0f0b 2005-02-13 devnull {
180 0faf0f0b 2005-02-13 devnull 	int n;
181 0faf0f0b 2005-02-13 devnull 	char *p;
182 0faf0f0b 2005-02-13 devnull 	VtLogChunk *c;
183 5ddc97fc 2005-02-14 devnull 	static int first = 1;
184 0faf0f0b 2005-02-13 devnull 
185 0faf0f0b 2005-02-13 devnull 	if(l == nil)
186 0faf0f0b 2005-02-13 devnull 		return;
187 0faf0f0b 2005-02-13 devnull 		
188 5ddc97fc 2005-02-14 devnull 	if(first){
189 5ddc97fc 2005-02-14 devnull 		fmtinstall('T', timefmt);
190 5ddc97fc 2005-02-14 devnull 		first = 0;
191 5ddc97fc 2005-02-14 devnull 	}
192 5ddc97fc 2005-02-14 devnull 		
193 5ddc97fc 2005-02-14 devnull 	
194 0faf0f0b 2005-02-13 devnull 	qlock(&l->lk);
195 0faf0f0b 2005-02-13 devnull 	c = l->w;
196 0faf0f0b 2005-02-13 devnull 	n = c->ep - c->wp;
197 0faf0f0b 2005-02-13 devnull 	if(n < 512){
198 0faf0f0b 2005-02-13 devnull 		c++;
199 0faf0f0b 2005-02-13 devnull 		if(c == l->chunk+l->nchunk)
200 0faf0f0b 2005-02-13 devnull 			c = l->chunk;
201 0faf0f0b 2005-02-13 devnull 		c->wp = c->p;
202 0faf0f0b 2005-02-13 devnull 		l->w = c;
203 0faf0f0b 2005-02-13 devnull 	}
204 0faf0f0b 2005-02-13 devnull 	p = vseprint(c->wp, c->ep, fmt, arg);
205 0faf0f0b 2005-02-13 devnull 	if(p)
206 0faf0f0b 2005-02-13 devnull 		c->wp = p;
207 0faf0f0b 2005-02-13 devnull 	qunlock(&l->lk);
208 0faf0f0b 2005-02-13 devnull }
209 0faf0f0b 2005-02-13 devnull 
210 0faf0f0b 2005-02-13 devnull void
211 0faf0f0b 2005-02-13 devnull vtlogprint(VtLog *l, char *fmt, ...)
212 0faf0f0b 2005-02-13 devnull {
213 0faf0f0b 2005-02-13 devnull 	va_list arg;
214 0faf0f0b 2005-02-13 devnull 	
215 0faf0f0b 2005-02-13 devnull 	if(l == nil)
216 0faf0f0b 2005-02-13 devnull 		return;
217 0faf0f0b 2005-02-13 devnull 		
218 0faf0f0b 2005-02-13 devnull 	va_start(arg, fmt);
219 0faf0f0b 2005-02-13 devnull 	vtlogvprint(l, fmt, arg);
220 0faf0f0b 2005-02-13 devnull 	va_end(arg);
221 0faf0f0b 2005-02-13 devnull }
222 0faf0f0b 2005-02-13 devnull 
223 0faf0f0b 2005-02-13 devnull void
224 0faf0f0b 2005-02-13 devnull vtlog(char *name, char *fmt, ...)
225 0faf0f0b 2005-02-13 devnull {
226 0faf0f0b 2005-02-13 devnull 	VtLog *l;
227 0faf0f0b 2005-02-13 devnull 	va_list arg;
228 5ddc97fc 2005-02-14 devnull 
229 0faf0f0b 2005-02-13 devnull 	l = vtlogopen(name, LogSize);
230 5ddc97fc 2005-02-14 devnull 	if(l == nil)
231 5ddc97fc 2005-02-14 devnull 		return;
232 0faf0f0b 2005-02-13 devnull 	va_start(arg, fmt);
233 0faf0f0b 2005-02-13 devnull 	vtlogvprint(l, fmt, arg);
234 0faf0f0b 2005-02-13 devnull 	va_end(arg);
235 0faf0f0b 2005-02-13 devnull 	vtlogclose(l);
236 0faf0f0b 2005-02-13 devnull }
237 0faf0f0b 2005-02-13 devnull 
238 0faf0f0b 2005-02-13 devnull void
239 0faf0f0b 2005-02-13 devnull vtlogdump(int fd, VtLog *l)
240 0faf0f0b 2005-02-13 devnull {
241 0faf0f0b 2005-02-13 devnull 	int i;
242 0faf0f0b 2005-02-13 devnull 	VtLogChunk *c;
243 0faf0f0b 2005-02-13 devnull 
244 0faf0f0b 2005-02-13 devnull 	if(l == nil)
245 0faf0f0b 2005-02-13 devnull 		return;
246 0faf0f0b 2005-02-13 devnull 		
247 0faf0f0b 2005-02-13 devnull 	c = l->w;
248 0faf0f0b 2005-02-13 devnull 	for(i=0; i<l->nchunk; i++){
249 0faf0f0b 2005-02-13 devnull 		if(++c == l->chunk+l->nchunk)
250 0faf0f0b 2005-02-13 devnull 			c = l->chunk;
251 0faf0f0b 2005-02-13 devnull 		write(fd, c->p, c->wp-c->p);
252 0faf0f0b 2005-02-13 devnull 	}
253 0faf0f0b 2005-02-13 devnull }
254 0faf0f0b 2005-02-13 devnull