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