##// END OF EJS Templates
parsers: change the type of nt_level...
Bryan O'Sullivan -
r16618:6bae941b default
parent child Browse files
Show More
@@ -1,1207 +1,1207 b''
1 1 /*
2 2 parsers.c - efficient content parsing
3 3
4 4 Copyright 2008 Matt Mackall <mpm@selenic.com> and others
5 5
6 6 This software may be used and distributed according to the terms of
7 7 the GNU General Public License, incorporated herein by reference.
8 8 */
9 9
10 10 #include <Python.h>
11 11 #include <ctype.h>
12 12 #include <string.h>
13 13
14 14 #include "util.h"
15 15
16 16 static inline int hexdigit(const char *p, Py_ssize_t off)
17 17 {
18 18 char c = p[off];
19 19
20 20 if (c >= '0' && c <= '9')
21 21 return c - '0';
22 22 if (c >= 'a' && c <= 'f')
23 23 return c - 'a' + 10;
24 24 if (c >= 'A' && c <= 'F')
25 25 return c - 'A' + 10;
26 26
27 27 PyErr_SetString(PyExc_ValueError, "input contains non-hex character");
28 28 return 0;
29 29 }
30 30
31 31 /*
32 32 * Turn a hex-encoded string into binary.
33 33 */
34 34 static PyObject *unhexlify(const char *str, int len)
35 35 {
36 36 PyObject *ret;
37 37 char *d;
38 38 int i;
39 39
40 40 ret = PyBytes_FromStringAndSize(NULL, len / 2);
41 41
42 42 if (!ret)
43 43 return NULL;
44 44
45 45 d = PyBytes_AsString(ret);
46 46
47 47 for (i = 0; i < len;) {
48 48 int hi = hexdigit(str, i++);
49 49 int lo = hexdigit(str, i++);
50 50 *d++ = (hi << 4) | lo;
51 51 }
52 52
53 53 return ret;
54 54 }
55 55
56 56 /*
57 57 * This code assumes that a manifest is stitched together with newline
58 58 * ('\n') characters.
59 59 */
60 60 static PyObject *parse_manifest(PyObject *self, PyObject *args)
61 61 {
62 62 PyObject *mfdict, *fdict;
63 63 char *str, *cur, *start, *zero;
64 64 int len;
65 65
66 66 if (!PyArg_ParseTuple(args, "O!O!s#:parse_manifest",
67 67 &PyDict_Type, &mfdict,
68 68 &PyDict_Type, &fdict,
69 69 &str, &len))
70 70 goto quit;
71 71
72 72 for (start = cur = str, zero = NULL; cur < str + len; cur++) {
73 73 PyObject *file = NULL, *node = NULL;
74 74 PyObject *flags = NULL;
75 75 int nlen;
76 76
77 77 if (!*cur) {
78 78 zero = cur;
79 79 continue;
80 80 }
81 81 else if (*cur != '\n')
82 82 continue;
83 83
84 84 if (!zero) {
85 85 PyErr_SetString(PyExc_ValueError,
86 86 "manifest entry has no separator");
87 87 goto quit;
88 88 }
89 89
90 90 file = PyBytes_FromStringAndSize(start, zero - start);
91 91
92 92 if (!file)
93 93 goto bail;
94 94
95 95 nlen = cur - zero - 1;
96 96
97 97 node = unhexlify(zero + 1, nlen > 40 ? 40 : nlen);
98 98 if (!node)
99 99 goto bail;
100 100
101 101 if (nlen > 40) {
102 102 flags = PyBytes_FromStringAndSize(zero + 41,
103 103 nlen - 40);
104 104 if (!flags)
105 105 goto bail;
106 106
107 107 if (PyDict_SetItem(fdict, file, flags) == -1)
108 108 goto bail;
109 109 }
110 110
111 111 if (PyDict_SetItem(mfdict, file, node) == -1)
112 112 goto bail;
113 113
114 114 start = cur + 1;
115 115 zero = NULL;
116 116
117 117 Py_XDECREF(flags);
118 118 Py_XDECREF(node);
119 119 Py_XDECREF(file);
120 120 continue;
121 121 bail:
122 122 Py_XDECREF(flags);
123 123 Py_XDECREF(node);
124 124 Py_XDECREF(file);
125 125 goto quit;
126 126 }
127 127
128 128 if (len > 0 && *(cur - 1) != '\n') {
129 129 PyErr_SetString(PyExc_ValueError,
130 130 "manifest contains trailing garbage");
131 131 goto quit;
132 132 }
133 133
134 134 Py_INCREF(Py_None);
135 135 return Py_None;
136 136 quit:
137 137 return NULL;
138 138 }
139 139
140 140 static PyObject *parse_dirstate(PyObject *self, PyObject *args)
141 141 {
142 142 PyObject *dmap, *cmap, *parents = NULL, *ret = NULL;
143 143 PyObject *fname = NULL, *cname = NULL, *entry = NULL;
144 144 char *str, *cur, *end, *cpos;
145 145 int state, mode, size, mtime;
146 146 unsigned int flen;
147 147 int len;
148 148
149 149 if (!PyArg_ParseTuple(args, "O!O!s#:parse_dirstate",
150 150 &PyDict_Type, &dmap,
151 151 &PyDict_Type, &cmap,
152 152 &str, &len))
153 153 goto quit;
154 154
155 155 /* read parents */
156 156 if (len < 40)
157 157 goto quit;
158 158
159 159 parents = Py_BuildValue("s#s#", str, 20, str + 20, 20);
160 160 if (!parents)
161 161 goto quit;
162 162
163 163 /* read filenames */
164 164 cur = str + 40;
165 165 end = str + len;
166 166
167 167 while (cur < end - 17) {
168 168 /* unpack header */
169 169 state = *cur;
170 170 mode = getbe32(cur + 1);
171 171 size = getbe32(cur + 5);
172 172 mtime = getbe32(cur + 9);
173 173 flen = getbe32(cur + 13);
174 174 cur += 17;
175 175 if (cur + flen > end || cur + flen < cur) {
176 176 PyErr_SetString(PyExc_ValueError, "overflow in dirstate");
177 177 goto quit;
178 178 }
179 179
180 180 entry = Py_BuildValue("ciii", state, mode, size, mtime);
181 181 if (!entry)
182 182 goto quit;
183 183 PyObject_GC_UnTrack(entry); /* don't waste time with this */
184 184
185 185 cpos = memchr(cur, 0, flen);
186 186 if (cpos) {
187 187 fname = PyBytes_FromStringAndSize(cur, cpos - cur);
188 188 cname = PyBytes_FromStringAndSize(cpos + 1,
189 189 flen - (cpos - cur) - 1);
190 190 if (!fname || !cname ||
191 191 PyDict_SetItem(cmap, fname, cname) == -1 ||
192 192 PyDict_SetItem(dmap, fname, entry) == -1)
193 193 goto quit;
194 194 Py_DECREF(cname);
195 195 } else {
196 196 fname = PyBytes_FromStringAndSize(cur, flen);
197 197 if (!fname ||
198 198 PyDict_SetItem(dmap, fname, entry) == -1)
199 199 goto quit;
200 200 }
201 201 cur += flen;
202 202 Py_DECREF(fname);
203 203 Py_DECREF(entry);
204 204 fname = cname = entry = NULL;
205 205 }
206 206
207 207 ret = parents;
208 208 Py_INCREF(ret);
209 209 quit:
210 210 Py_XDECREF(fname);
211 211 Py_XDECREF(cname);
212 212 Py_XDECREF(entry);
213 213 Py_XDECREF(parents);
214 214 return ret;
215 215 }
216 216
217 217 /*
218 218 * A base-16 trie for fast node->rev mapping.
219 219 *
220 220 * Positive value is index of the next node in the trie
221 221 * Negative value is a leaf: -(rev + 1)
222 222 * Zero is empty
223 223 */
224 224 typedef struct {
225 225 int children[16];
226 226 } nodetree;
227 227
228 228 /*
229 229 * This class has two behaviours.
230 230 *
231 231 * When used in a list-like way (with integer keys), we decode an
232 232 * entry in a RevlogNG index file on demand. Our last entry is a
233 233 * sentinel, always a nullid. We have limited support for
234 234 * integer-keyed insert and delete, only at elements right before the
235 235 * sentinel.
236 236 *
237 237 * With string keys, we lazily perform a reverse mapping from node to
238 238 * rev, using a base-16 trie.
239 239 */
240 240 typedef struct {
241 241 PyObject_HEAD
242 242 /* Type-specific fields go here. */
243 243 PyObject *data; /* raw bytes of index */
244 244 PyObject **cache; /* cached tuples */
245 245 const char **offsets; /* populated on demand */
246 246 Py_ssize_t raw_length; /* original number of elements */
247 247 Py_ssize_t length; /* current number of elements */
248 248 PyObject *added; /* populated on demand */
249 249 nodetree *nt; /* base-16 trie */
250 250 int ntlength; /* # nodes in use */
251 251 int ntcapacity; /* # nodes allocated */
252 252 int ntdepth; /* maximum depth of tree */
253 253 int ntsplits; /* # splits performed */
254 254 int ntrev; /* last rev scanned */
255 255 int ntlookups; /* # lookups */
256 256 int ntmisses; /* # lookups that miss the cache */
257 257 int inlined;
258 258 } indexObject;
259 259
260 260 static Py_ssize_t index_length(const indexObject *self)
261 261 {
262 262 if (self->added == NULL)
263 263 return self->length;
264 264 return self->length + PyList_GET_SIZE(self->added);
265 265 }
266 266
267 267 static PyObject *nullentry;
268 268 static const char nullid[20];
269 269
270 270 static long inline_scan(indexObject *self, const char **offsets);
271 271
272 272 #if LONG_MAX == 0x7fffffffL
273 273 static char *tuple_format = "Kiiiiiis#";
274 274 #else
275 275 static char *tuple_format = "kiiiiiis#";
276 276 #endif
277 277
278 278 /*
279 279 * Return a pointer to the beginning of a RevlogNG record.
280 280 */
281 281 static const char *index_deref(indexObject *self, Py_ssize_t pos)
282 282 {
283 283 if (self->inlined && pos > 0) {
284 284 if (self->offsets == NULL) {
285 285 self->offsets = malloc(self->raw_length *
286 286 sizeof(*self->offsets));
287 287 if (self->offsets == NULL)
288 288 return (const char *)PyErr_NoMemory();
289 289 inline_scan(self, self->offsets);
290 290 }
291 291 return self->offsets[pos];
292 292 }
293 293
294 294 return PyString_AS_STRING(self->data) + pos * 64;
295 295 }
296 296
297 297 /*
298 298 * RevlogNG format (all in big endian, data may be inlined):
299 299 * 6 bytes: offset
300 300 * 2 bytes: flags
301 301 * 4 bytes: compressed length
302 302 * 4 bytes: uncompressed length
303 303 * 4 bytes: base revision
304 304 * 4 bytes: link revision
305 305 * 4 bytes: parent 1 revision
306 306 * 4 bytes: parent 2 revision
307 307 * 32 bytes: nodeid (only 20 bytes used)
308 308 */
309 309 static PyObject *index_get(indexObject *self, Py_ssize_t pos)
310 310 {
311 311 uint64_t offset_flags;
312 312 int comp_len, uncomp_len, base_rev, link_rev, parent_1, parent_2;
313 313 const char *c_node_id;
314 314 const char *data;
315 315 Py_ssize_t length = index_length(self);
316 316 PyObject *entry;
317 317
318 318 if (pos < 0)
319 319 pos += length;
320 320
321 321 if (pos < 0 || pos >= length) {
322 322 PyErr_SetString(PyExc_IndexError, "revlog index out of range");
323 323 return NULL;
324 324 }
325 325
326 326 if (pos == length - 1) {
327 327 Py_INCREF(nullentry);
328 328 return nullentry;
329 329 }
330 330
331 331 if (pos >= self->length - 1) {
332 332 PyObject *obj;
333 333 obj = PyList_GET_ITEM(self->added, pos - self->length + 1);
334 334 Py_INCREF(obj);
335 335 return obj;
336 336 }
337 337
338 338 if (self->cache) {
339 339 if (self->cache[pos]) {
340 340 Py_INCREF(self->cache[pos]);
341 341 return self->cache[pos];
342 342 }
343 343 } else {
344 344 self->cache = calloc(self->raw_length, sizeof(PyObject *));
345 345 if (self->cache == NULL)
346 346 return PyErr_NoMemory();
347 347 }
348 348
349 349 data = index_deref(self, pos);
350 350 if (data == NULL)
351 351 return NULL;
352 352
353 353 offset_flags = getbe32(data + 4);
354 354 if (pos == 0) /* mask out version number for the first entry */
355 355 offset_flags &= 0xFFFF;
356 356 else {
357 357 uint32_t offset_high = getbe32(data);
358 358 offset_flags |= ((uint64_t)offset_high) << 32;
359 359 }
360 360
361 361 comp_len = getbe32(data + 8);
362 362 uncomp_len = getbe32(data + 12);
363 363 base_rev = getbe32(data + 16);
364 364 link_rev = getbe32(data + 20);
365 365 parent_1 = getbe32(data + 24);
366 366 parent_2 = getbe32(data + 28);
367 367 c_node_id = data + 32;
368 368
369 369 entry = Py_BuildValue(tuple_format, offset_flags, comp_len,
370 370 uncomp_len, base_rev, link_rev,
371 371 parent_1, parent_2, c_node_id, 20);
372 372
373 373 if (entry)
374 374 PyObject_GC_UnTrack(entry);
375 375
376 376 self->cache[pos] = entry;
377 377 Py_INCREF(entry);
378 378
379 379 return entry;
380 380 }
381 381
382 382 /*
383 383 * Return the 20-byte SHA of the node corresponding to the given rev.
384 384 */
385 385 static const char *index_node(indexObject *self, Py_ssize_t pos)
386 386 {
387 387 Py_ssize_t length = index_length(self);
388 388 const char *data;
389 389
390 390 if (pos == length - 1)
391 391 return nullid;
392 392
393 393 if (pos >= length)
394 394 return NULL;
395 395
396 396 if (pos >= self->length - 1) {
397 397 PyObject *tuple, *str;
398 398 tuple = PyList_GET_ITEM(self->added, pos - self->length + 1);
399 399 str = PyTuple_GetItem(tuple, 7);
400 400 return str ? PyString_AS_STRING(str) : NULL;
401 401 }
402 402
403 403 data = index_deref(self, pos);
404 404 return data ? data + 32 : NULL;
405 405 }
406 406
407 407 static int nt_insert(indexObject *self, const char *node, int rev);
408 408
409 409 static int node_check(PyObject *obj, char **node, Py_ssize_t *nodelen)
410 410 {
411 411 if (PyString_AsStringAndSize(obj, node, nodelen) == -1)
412 412 return -1;
413 413 if (*nodelen == 20)
414 414 return 0;
415 415 PyErr_SetString(PyExc_ValueError, "20-byte hash required");
416 416 return -1;
417 417 }
418 418
419 419 static PyObject *index_insert(indexObject *self, PyObject *args)
420 420 {
421 421 PyObject *obj;
422 422 char *node;
423 423 long offset;
424 424 Py_ssize_t len, nodelen;
425 425
426 426 if (!PyArg_ParseTuple(args, "lO", &offset, &obj))
427 427 return NULL;
428 428
429 429 if (!PyTuple_Check(obj) || PyTuple_GET_SIZE(obj) != 8) {
430 430 PyErr_SetString(PyExc_TypeError, "8-tuple required");
431 431 return NULL;
432 432 }
433 433
434 434 if (node_check(PyTuple_GET_ITEM(obj, 7), &node, &nodelen) == -1)
435 435 return NULL;
436 436
437 437 len = index_length(self);
438 438
439 439 if (offset < 0)
440 440 offset += len;
441 441
442 442 if (offset != len - 1) {
443 443 PyErr_SetString(PyExc_IndexError,
444 444 "insert only supported at index -1");
445 445 return NULL;
446 446 }
447 447
448 448 if (offset > INT_MAX) {
449 449 PyErr_SetString(PyExc_ValueError,
450 450 "currently only 2**31 revs supported");
451 451 return NULL;
452 452 }
453 453
454 454 if (self->added == NULL) {
455 455 self->added = PyList_New(0);
456 456 if (self->added == NULL)
457 457 return NULL;
458 458 }
459 459
460 460 if (PyList_Append(self->added, obj) == -1)
461 461 return NULL;
462 462
463 463 if (self->nt)
464 464 nt_insert(self, node, (int)offset);
465 465
466 466 Py_RETURN_NONE;
467 467 }
468 468
469 469 static void _index_clearcaches(indexObject *self)
470 470 {
471 471 if (self->cache) {
472 472 Py_ssize_t i;
473 473
474 474 for (i = 0; i < self->raw_length; i++) {
475 475 if (self->cache[i]) {
476 476 Py_DECREF(self->cache[i]);
477 477 self->cache[i] = NULL;
478 478 }
479 479 }
480 480 free(self->cache);
481 481 self->cache = NULL;
482 482 }
483 483 if (self->offsets) {
484 484 free(self->offsets);
485 485 self->offsets = NULL;
486 486 }
487 487 if (self->nt) {
488 488 free(self->nt);
489 489 self->nt = NULL;
490 490 }
491 491 }
492 492
493 493 static PyObject *index_clearcaches(indexObject *self)
494 494 {
495 495 _index_clearcaches(self);
496 496 self->ntlength = self->ntcapacity = 0;
497 497 self->ntdepth = self->ntsplits = 0;
498 498 self->ntrev = -1;
499 499 self->ntlookups = self->ntmisses = 0;
500 500 Py_RETURN_NONE;
501 501 }
502 502
503 503 static PyObject *index_stats(indexObject *self)
504 504 {
505 505 PyObject *obj = PyDict_New();
506 506
507 507 if (obj == NULL)
508 508 return NULL;
509 509
510 510 #define istat(__n, __d) \
511 511 if (PyDict_SetItemString(obj, __d, PyInt_FromLong(self->__n)) == -1) \
512 512 goto bail;
513 513
514 514 if (self->added) {
515 515 Py_ssize_t len = PyList_GET_SIZE(self->added);
516 516 if (PyDict_SetItemString(obj, "index entries added",
517 517 PyInt_FromLong(len)) == -1)
518 518 goto bail;
519 519 }
520 520
521 521 if (self->raw_length != self->length - 1)
522 522 istat(raw_length, "revs on disk");
523 523 istat(length, "revs in memory");
524 524 istat(ntcapacity, "node trie capacity");
525 525 istat(ntdepth, "node trie depth");
526 526 istat(ntlength, "node trie count");
527 527 istat(ntlookups, "node trie lookups");
528 528 istat(ntmisses, "node trie misses");
529 529 istat(ntrev, "node trie last rev scanned");
530 530 istat(ntsplits, "node trie splits");
531 531
532 532 #undef istat
533 533
534 534 return obj;
535 535
536 536 bail:
537 537 Py_XDECREF(obj);
538 538 return NULL;
539 539 }
540 540
541 static inline int nt_level(const char *node, int level)
541 static inline int nt_level(const char *node, Py_ssize_t level)
542 542 {
543 543 int v = node[level>>1];
544 544 if (!(level & 1))
545 545 v >>= 4;
546 546 return v & 0xf;
547 547 }
548 548
549 549 /*
550 550 * Return values:
551 551 *
552 552 * -4: match is ambiguous (multiple candidates)
553 553 * -2: not found
554 554 * rest: valid rev
555 555 */
556 556 static int nt_find(indexObject *self, const char *node, Py_ssize_t nodelen)
557 557 {
558 558 int level, off;
559 559
560 560 if (nodelen == 20 && node[0] == '\0' && memcmp(node, nullid, 20) == 0)
561 561 return -1;
562 562
563 563 if (self->nt == NULL)
564 564 return -2;
565 565
566 566 for (level = off = 0; level < nodelen; level++) {
567 567 int k = nt_level(node, level);
568 568 nodetree *n = &self->nt[off];
569 569 int v = n->children[k];
570 570
571 571 if (v < 0) {
572 572 const char *n;
573 573 v = -v - 1;
574 574 n = index_node(self, v);
575 575 if (n == NULL)
576 576 return -2;
577 577 return memcmp(node, n, nodelen > 20 ? 20 : nodelen)
578 578 ? -2 : v;
579 579 }
580 580 if (v == 0)
581 581 return -2;
582 582 off = v;
583 583 }
584 584 /* multiple matches against an ambiguous prefix */
585 585 return -4;
586 586 }
587 587
588 588 static int nt_new(indexObject *self)
589 589 {
590 590 if (self->ntlength == self->ntcapacity) {
591 591 self->ntcapacity *= 2;
592 592 self->nt = realloc(self->nt,
593 593 self->ntcapacity * sizeof(nodetree));
594 594 if (self->nt == NULL) {
595 595 PyErr_SetString(PyExc_MemoryError, "out of memory");
596 596 return -1;
597 597 }
598 598 memset(&self->nt[self->ntlength], 0,
599 599 sizeof(nodetree) * (self->ntcapacity - self->ntlength));
600 600 }
601 601 return self->ntlength++;
602 602 }
603 603
604 604 static int nt_insert(indexObject *self, const char *node, int rev)
605 605 {
606 606 int level = 0;
607 607 int off = 0;
608 608
609 609 while (level < 20) {
610 610 int k = nt_level(node, level);
611 611 nodetree *n;
612 612 int v;
613 613
614 614 n = &self->nt[off];
615 615 v = n->children[k];
616 616
617 617 if (v == 0) {
618 618 n->children[k] = -rev - 1;
619 619 return 0;
620 620 }
621 621 if (v < 0) {
622 622 const char *oldnode = index_node(self, -v - 1);
623 623 int noff;
624 624
625 625 if (!oldnode || !memcmp(oldnode, node, 20)) {
626 626 n->children[k] = -rev - 1;
627 627 return 0;
628 628 }
629 629 noff = nt_new(self);
630 630 if (noff == -1)
631 631 return -1;
632 632 /* self->nt may have been changed by realloc */
633 633 self->nt[off].children[k] = noff;
634 634 off = noff;
635 635 n = &self->nt[off];
636 636 n->children[nt_level(oldnode, ++level)] = v;
637 637 if (level > self->ntdepth)
638 638 self->ntdepth = level;
639 639 self->ntsplits += 1;
640 640 } else {
641 641 level += 1;
642 642 off = v;
643 643 }
644 644 }
645 645
646 646 return -1;
647 647 }
648 648
649 649 static int nt_init(indexObject *self)
650 650 {
651 651 if (self->nt == NULL) {
652 652 self->ntcapacity = self->raw_length < 4
653 653 ? 4 : self->raw_length / 2;
654 654 self->nt = calloc(self->ntcapacity, sizeof(nodetree));
655 655 if (self->nt == NULL) {
656 656 PyErr_NoMemory();
657 657 return -1;
658 658 }
659 659 self->ntlength = 1;
660 660 self->ntrev = (int)index_length(self) - 1;
661 661 self->ntlookups = 1;
662 662 self->ntmisses = 0;
663 663 }
664 664 return 0;
665 665 }
666 666
667 667 /*
668 668 * Return values:
669 669 *
670 670 * -3: error (exception set)
671 671 * -2: not found (no exception set)
672 672 * rest: valid rev
673 673 */
674 674 static int index_find_node(indexObject *self,
675 675 const char *node, Py_ssize_t nodelen)
676 676 {
677 677 int rev;
678 678
679 679 self->ntlookups++;
680 680 rev = nt_find(self, node, nodelen);
681 681 if (rev >= -1)
682 682 return rev;
683 683
684 684 if (nt_init(self) == -1)
685 685 return -3;
686 686
687 687 /*
688 688 * For the first handful of lookups, we scan the entire index,
689 689 * and cache only the matching nodes. This optimizes for cases
690 690 * like "hg tip", where only a few nodes are accessed.
691 691 *
692 692 * After that, we cache every node we visit, using a single
693 693 * scan amortized over multiple lookups. This gives the best
694 694 * bulk performance, e.g. for "hg log".
695 695 */
696 696 if (self->ntmisses++ < 4) {
697 697 for (rev = self->ntrev - 1; rev >= 0; rev--) {
698 698 const char *n = index_node(self, rev);
699 699 if (n == NULL)
700 700 return -2;
701 701 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
702 702 if (nt_insert(self, n, rev) == -1)
703 703 return -3;
704 704 break;
705 705 }
706 706 }
707 707 } else {
708 708 for (rev = self->ntrev - 1; rev >= 0; rev--) {
709 709 const char *n = index_node(self, rev);
710 710 if (n == NULL) {
711 711 self->ntrev = rev + 1;
712 712 return -2;
713 713 }
714 714 if (nt_insert(self, n, rev) == -1) {
715 715 self->ntrev = rev + 1;
716 716 return -3;
717 717 }
718 718 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
719 719 break;
720 720 }
721 721 }
722 722 self->ntrev = rev;
723 723 }
724 724
725 725 if (rev >= 0)
726 726 return rev;
727 727 return -2;
728 728 }
729 729
730 730 static PyObject *raise_revlog_error(void)
731 731 {
732 732 static PyObject *errclass;
733 733 PyObject *mod = NULL, *errobj;
734 734
735 735 if (errclass == NULL) {
736 736 PyObject *dict;
737 737
738 738 mod = PyImport_ImportModule("mercurial.error");
739 739 if (mod == NULL)
740 740 goto classfail;
741 741
742 742 dict = PyModule_GetDict(mod);
743 743 if (dict == NULL)
744 744 goto classfail;
745 745
746 746 errclass = PyDict_GetItemString(dict, "RevlogError");
747 747 if (errclass == NULL) {
748 748 PyErr_SetString(PyExc_SystemError,
749 749 "could not find RevlogError");
750 750 goto classfail;
751 751 }
752 752 Py_INCREF(errclass);
753 753 }
754 754
755 755 errobj = PyObject_CallFunction(errclass, NULL);
756 756 if (errobj == NULL)
757 757 return NULL;
758 758 PyErr_SetObject(errclass, errobj);
759 759 return errobj;
760 760
761 761 classfail:
762 762 Py_XDECREF(mod);
763 763 return NULL;
764 764 }
765 765
766 766 static PyObject *index_getitem(indexObject *self, PyObject *value)
767 767 {
768 768 char *node;
769 769 Py_ssize_t nodelen;
770 770 int rev;
771 771
772 772 if (PyInt_Check(value))
773 773 return index_get(self, PyInt_AS_LONG(value));
774 774
775 775 if (PyString_AsStringAndSize(value, &node, &nodelen) == -1)
776 776 return NULL;
777 777 rev = index_find_node(self, node, nodelen);
778 778 if (rev >= -1)
779 779 return PyInt_FromLong(rev);
780 780 if (rev == -2)
781 781 raise_revlog_error();
782 782 return NULL;
783 783 }
784 784
785 785 static PyObject *index_m_get(indexObject *self, PyObject *args)
786 786 {
787 787 char *node;
788 788 int nodelen, rev;
789 789
790 790 if (!PyArg_ParseTuple(args, "s#", &node, &nodelen))
791 791 return NULL;
792 792
793 793 rev = index_find_node(self, node, nodelen);
794 794 if (rev == -3)
795 795 return NULL;
796 796 if (rev == -2)
797 797 Py_RETURN_NONE;
798 798 return PyInt_FromLong(rev);
799 799 }
800 800
801 801 static int index_contains(indexObject *self, PyObject *value)
802 802 {
803 803 char *node;
804 804 Py_ssize_t nodelen;
805 805
806 806 if (PyInt_Check(value)) {
807 807 long rev = PyInt_AS_LONG(value);
808 808 return rev >= -1 && rev < index_length(self);
809 809 }
810 810
811 811 if (!PyString_Check(value))
812 812 return 0;
813 813
814 814 node = PyString_AS_STRING(value);
815 815 nodelen = PyString_GET_SIZE(value);
816 816
817 817 switch (index_find_node(self, node, nodelen)) {
818 818 case -3:
819 819 return -1;
820 820 case -2:
821 821 return 0;
822 822 default:
823 823 return 1;
824 824 }
825 825 }
826 826
827 827 /*
828 828 * Invalidate any trie entries introduced by added revs.
829 829 */
830 830 static void nt_invalidate_added(indexObject *self, Py_ssize_t start)
831 831 {
832 832 Py_ssize_t i, len = PyList_GET_SIZE(self->added);
833 833
834 834 for (i = start; i < len; i++) {
835 835 PyObject *tuple = PyList_GET_ITEM(self->added, i);
836 836 PyObject *node = PyTuple_GET_ITEM(tuple, 7);
837 837
838 838 nt_insert(self, PyString_AS_STRING(node), -1);
839 839 }
840 840
841 841 if (start == 0) {
842 842 Py_DECREF(self->added);
843 843 self->added = NULL;
844 844 }
845 845 }
846 846
847 847 /*
848 848 * Delete a numeric range of revs, which must be at the end of the
849 849 * range, but exclude the sentinel nullid entry.
850 850 */
851 851 static int index_slice_del(indexObject *self, PyObject *item)
852 852 {
853 853 Py_ssize_t start, stop, step, slicelength;
854 854 Py_ssize_t length = index_length(self);
855 855
856 856 if (PySlice_GetIndicesEx((PySliceObject*)item, length,
857 857 &start, &stop, &step, &slicelength) < 0)
858 858 return -1;
859 859
860 860 if (slicelength <= 0)
861 861 return 0;
862 862
863 863 if ((step < 0 && start < stop) || (step > 0 && start > stop))
864 864 stop = start;
865 865
866 866 if (step < 0) {
867 867 stop = start + 1;
868 868 start = stop + step*(slicelength - 1) - 1;
869 869 step = -step;
870 870 }
871 871
872 872 if (step != 1) {
873 873 PyErr_SetString(PyExc_ValueError,
874 874 "revlog index delete requires step size of 1");
875 875 return -1;
876 876 }
877 877
878 878 if (stop != length - 1) {
879 879 PyErr_SetString(PyExc_IndexError,
880 880 "revlog index deletion indices are invalid");
881 881 return -1;
882 882 }
883 883
884 884 if (start < self->length - 1) {
885 885 if (self->nt) {
886 886 Py_ssize_t i;
887 887
888 888 for (i = start + 1; i < self->length - 1; i++) {
889 889 const char *node = index_node(self, i);
890 890
891 891 if (node)
892 892 nt_insert(self, node, -1);
893 893 }
894 894 if (self->added)
895 895 nt_invalidate_added(self, 0);
896 896 if (self->ntrev > start)
897 897 self->ntrev = (int)start;
898 898 }
899 899 self->length = start + 1;
900 900 return 0;
901 901 }
902 902
903 903 if (self->nt) {
904 904 nt_invalidate_added(self, start - self->length + 1);
905 905 if (self->ntrev > start)
906 906 self->ntrev = (int)start;
907 907 }
908 908 return self->added
909 909 ? PyList_SetSlice(self->added, start - self->length + 1,
910 910 PyList_GET_SIZE(self->added), NULL)
911 911 : 0;
912 912 }
913 913
914 914 /*
915 915 * Supported ops:
916 916 *
917 917 * slice deletion
918 918 * string assignment (extend node->rev mapping)
919 919 * string deletion (shrink node->rev mapping)
920 920 */
921 921 static int index_assign_subscript(indexObject *self, PyObject *item,
922 922 PyObject *value)
923 923 {
924 924 char *node;
925 925 Py_ssize_t nodelen;
926 926 long rev;
927 927
928 928 if (PySlice_Check(item) && value == NULL)
929 929 return index_slice_del(self, item);
930 930
931 931 if (node_check(item, &node, &nodelen) == -1)
932 932 return -1;
933 933
934 934 if (value == NULL)
935 935 return self->nt ? nt_insert(self, node, -1) : 0;
936 936 rev = PyInt_AsLong(value);
937 937 if (rev > INT_MAX || rev < 0) {
938 938 if (!PyErr_Occurred())
939 939 PyErr_SetString(PyExc_ValueError, "rev out of range");
940 940 return -1;
941 941 }
942 942 return nt_insert(self, node, (int)rev);
943 943 }
944 944
945 945 /*
946 946 * Find all RevlogNG entries in an index that has inline data. Update
947 947 * the optional "offsets" table with those entries.
948 948 */
949 949 static long inline_scan(indexObject *self, const char **offsets)
950 950 {
951 951 const char *data = PyString_AS_STRING(self->data);
952 952 const char *end = data + PyString_GET_SIZE(self->data);
953 953 const long hdrsize = 64;
954 954 long incr = hdrsize;
955 955 Py_ssize_t len = 0;
956 956
957 957 while (data + hdrsize <= end) {
958 958 uint32_t comp_len;
959 959 const char *old_data;
960 960 /* 3rd element of header is length of compressed inline data */
961 961 comp_len = getbe32(data + 8);
962 962 incr = hdrsize + comp_len;
963 963 if (incr < hdrsize)
964 964 break;
965 965 if (offsets)
966 966 offsets[len] = data;
967 967 len++;
968 968 old_data = data;
969 969 data += incr;
970 970 if (data <= old_data)
971 971 break;
972 972 }
973 973
974 974 if (data != end && data + hdrsize != end) {
975 975 if (!PyErr_Occurred())
976 976 PyErr_SetString(PyExc_ValueError, "corrupt index file");
977 977 return -1;
978 978 }
979 979
980 980 return len;
981 981 }
982 982
983 983 static int index_init(indexObject *self, PyObject *args)
984 984 {
985 985 PyObject *data_obj, *inlined_obj;
986 986 Py_ssize_t size;
987 987
988 988 if (!PyArg_ParseTuple(args, "OO", &data_obj, &inlined_obj))
989 989 return -1;
990 990 if (!PyString_Check(data_obj)) {
991 991 PyErr_SetString(PyExc_TypeError, "data is not a string");
992 992 return -1;
993 993 }
994 994 size = PyString_GET_SIZE(data_obj);
995 995
996 996 self->inlined = inlined_obj && PyObject_IsTrue(inlined_obj);
997 997 self->data = data_obj;
998 998 self->cache = NULL;
999 999
1000 1000 self->added = NULL;
1001 1001 self->offsets = NULL;
1002 1002 self->nt = NULL;
1003 1003 self->ntlength = self->ntcapacity = 0;
1004 1004 self->ntdepth = self->ntsplits = 0;
1005 1005 self->ntlookups = self->ntmisses = 0;
1006 1006 self->ntrev = -1;
1007 1007 Py_INCREF(self->data);
1008 1008
1009 1009 if (self->inlined) {
1010 1010 long len = inline_scan(self, NULL);
1011 1011 if (len == -1)
1012 1012 goto bail;
1013 1013 self->raw_length = len;
1014 1014 self->length = len + 1;
1015 1015 } else {
1016 1016 if (size % 64) {
1017 1017 PyErr_SetString(PyExc_ValueError, "corrupt index file");
1018 1018 goto bail;
1019 1019 }
1020 1020 self->raw_length = size / 64;
1021 1021 self->length = self->raw_length + 1;
1022 1022 }
1023 1023
1024 1024 return 0;
1025 1025 bail:
1026 1026 return -1;
1027 1027 }
1028 1028
1029 1029 static PyObject *index_nodemap(indexObject *self)
1030 1030 {
1031 1031 Py_INCREF(self);
1032 1032 return (PyObject *)self;
1033 1033 }
1034 1034
1035 1035 static void index_dealloc(indexObject *self)
1036 1036 {
1037 1037 _index_clearcaches(self);
1038 1038 Py_DECREF(self->data);
1039 1039 Py_XDECREF(self->added);
1040 1040 PyObject_Del(self);
1041 1041 }
1042 1042
1043 1043 static PySequenceMethods index_sequence_methods = {
1044 1044 (lenfunc)index_length, /* sq_length */
1045 1045 0, /* sq_concat */
1046 1046 0, /* sq_repeat */
1047 1047 (ssizeargfunc)index_get, /* sq_item */
1048 1048 0, /* sq_slice */
1049 1049 0, /* sq_ass_item */
1050 1050 0, /* sq_ass_slice */
1051 1051 (objobjproc)index_contains, /* sq_contains */
1052 1052 };
1053 1053
1054 1054 static PyMappingMethods index_mapping_methods = {
1055 1055 (lenfunc)index_length, /* mp_length */
1056 1056 (binaryfunc)index_getitem, /* mp_subscript */
1057 1057 (objobjargproc)index_assign_subscript, /* mp_ass_subscript */
1058 1058 };
1059 1059
1060 1060 static PyMethodDef index_methods[] = {
1061 1061 {"clearcaches", (PyCFunction)index_clearcaches, METH_NOARGS,
1062 1062 "clear the index caches"},
1063 1063 {"get", (PyCFunction)index_m_get, METH_VARARGS,
1064 1064 "get an index entry"},
1065 1065 {"insert", (PyCFunction)index_insert, METH_VARARGS,
1066 1066 "insert an index entry"},
1067 1067 {"stats", (PyCFunction)index_stats, METH_NOARGS,
1068 1068 "stats for the index"},
1069 1069 {NULL} /* Sentinel */
1070 1070 };
1071 1071
1072 1072 static PyGetSetDef index_getset[] = {
1073 1073 {"nodemap", (getter)index_nodemap, NULL, "nodemap", NULL},
1074 1074 {NULL} /* Sentinel */
1075 1075 };
1076 1076
1077 1077 static PyTypeObject indexType = {
1078 1078 PyObject_HEAD_INIT(NULL)
1079 1079 0, /* ob_size */
1080 1080 "parsers.index", /* tp_name */
1081 1081 sizeof(indexObject), /* tp_basicsize */
1082 1082 0, /* tp_itemsize */
1083 1083 (destructor)index_dealloc, /* tp_dealloc */
1084 1084 0, /* tp_print */
1085 1085 0, /* tp_getattr */
1086 1086 0, /* tp_setattr */
1087 1087 0, /* tp_compare */
1088 1088 0, /* tp_repr */
1089 1089 0, /* tp_as_number */
1090 1090 &index_sequence_methods, /* tp_as_sequence */
1091 1091 &index_mapping_methods, /* tp_as_mapping */
1092 1092 0, /* tp_hash */
1093 1093 0, /* tp_call */
1094 1094 0, /* tp_str */
1095 1095 0, /* tp_getattro */
1096 1096 0, /* tp_setattro */
1097 1097 0, /* tp_as_buffer */
1098 1098 Py_TPFLAGS_DEFAULT, /* tp_flags */
1099 1099 "revlog index", /* tp_doc */
1100 1100 0, /* tp_traverse */
1101 1101 0, /* tp_clear */
1102 1102 0, /* tp_richcompare */
1103 1103 0, /* tp_weaklistoffset */
1104 1104 0, /* tp_iter */
1105 1105 0, /* tp_iternext */
1106 1106 index_methods, /* tp_methods */
1107 1107 0, /* tp_members */
1108 1108 index_getset, /* tp_getset */
1109 1109 0, /* tp_base */
1110 1110 0, /* tp_dict */
1111 1111 0, /* tp_descr_get */
1112 1112 0, /* tp_descr_set */
1113 1113 0, /* tp_dictoffset */
1114 1114 (initproc)index_init, /* tp_init */
1115 1115 0, /* tp_alloc */
1116 1116 PyType_GenericNew, /* tp_new */
1117 1117 };
1118 1118
1119 1119 /*
1120 1120 * returns a tuple of the form (index, index, cache) with elements as
1121 1121 * follows:
1122 1122 *
1123 1123 * index: an index object that lazily parses RevlogNG records
1124 1124 * cache: if data is inlined, a tuple (index_file_content, 0), else None
1125 1125 *
1126 1126 * added complications are for backwards compatibility
1127 1127 */
1128 1128 static PyObject *parse_index2(PyObject *self, PyObject *args)
1129 1129 {
1130 1130 PyObject *tuple = NULL, *cache = NULL;
1131 1131 indexObject *idx;
1132 1132 int ret;
1133 1133
1134 1134 idx = PyObject_New(indexObject, &indexType);
1135 1135 if (idx == NULL)
1136 1136 goto bail;
1137 1137
1138 1138 ret = index_init(idx, args);
1139 1139 if (ret == -1)
1140 1140 goto bail;
1141 1141
1142 1142 if (idx->inlined) {
1143 1143 cache = Py_BuildValue("iO", 0, idx->data);
1144 1144 if (cache == NULL)
1145 1145 goto bail;
1146 1146 } else {
1147 1147 cache = Py_None;
1148 1148 Py_INCREF(cache);
1149 1149 }
1150 1150
1151 1151 tuple = Py_BuildValue("NN", idx, cache);
1152 1152 if (!tuple)
1153 1153 goto bail;
1154 1154 return tuple;
1155 1155
1156 1156 bail:
1157 1157 Py_XDECREF(idx);
1158 1158 Py_XDECREF(cache);
1159 1159 Py_XDECREF(tuple);
1160 1160 return NULL;
1161 1161 }
1162 1162
1163 1163 static char parsers_doc[] = "Efficient content parsing.";
1164 1164
1165 1165 static PyMethodDef methods[] = {
1166 1166 {"parse_manifest", parse_manifest, METH_VARARGS, "parse a manifest\n"},
1167 1167 {"parse_dirstate", parse_dirstate, METH_VARARGS, "parse a dirstate\n"},
1168 1168 {"parse_index2", parse_index2, METH_VARARGS, "parse a revlog index\n"},
1169 1169 {NULL, NULL}
1170 1170 };
1171 1171
1172 1172 static void module_init(PyObject *mod)
1173 1173 {
1174 1174 if (PyType_Ready(&indexType) < 0)
1175 1175 return;
1176 1176 Py_INCREF(&indexType);
1177 1177
1178 1178 PyModule_AddObject(mod, "index", (PyObject *)&indexType);
1179 1179
1180 1180 nullentry = Py_BuildValue("iiiiiiis#", 0, 0, 0,
1181 1181 -1, -1, -1, -1, nullid, 20);
1182 1182 if (nullentry)
1183 1183 PyObject_GC_UnTrack(nullentry);
1184 1184 }
1185 1185
1186 1186 #ifdef IS_PY3K
1187 1187 static struct PyModuleDef parsers_module = {
1188 1188 PyModuleDef_HEAD_INIT,
1189 1189 "parsers",
1190 1190 parsers_doc,
1191 1191 -1,
1192 1192 methods
1193 1193 };
1194 1194
1195 1195 PyMODINIT_FUNC PyInit_parsers(void)
1196 1196 {
1197 1197 PyObject *mod = PyModule_Create(&parsers_module);
1198 1198 module_init(mod);
1199 1199 return mod;
1200 1200 }
1201 1201 #else
1202 1202 PyMODINIT_FUNC initparsers(void)
1203 1203 {
1204 1204 PyObject *mod = Py_InitModule3("parsers", methods, parsers_doc);
1205 1205 module_init(mod);
1206 1206 }
1207 1207 #endif
General Comments 0
You need to be logged in to leave comments. Login now