##// END OF EJS Templates
parsers: statically initializing tp_new to PyType_GenericNew is not portable...
Adrian Buehlmann -
r16604:48e42f98 stable
parent child Browse files
Show More
@@ -1,1186 +1,1186 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 547 static int nt_find(indexObject *self, const char *node, Py_ssize_t nodelen)
548 548 {
549 549 int level, off;
550 550
551 551 if (nodelen == 20 && node[0] == '\0' && memcmp(node, nullid, 20) == 0)
552 552 return -1;
553 553
554 554 if (self->nt == NULL)
555 555 return -2;
556 556
557 557 for (level = off = 0; level < nodelen; level++) {
558 558 int k = nt_level(node, level);
559 559 nodetree *n = &self->nt[off];
560 560 int v = n->children[k];
561 561
562 562 if (v < 0) {
563 563 const char *n;
564 564 v = -v - 1;
565 565 n = index_node(self, v);
566 566 if (n == NULL)
567 567 return -2;
568 568 return memcmp(node, n, nodelen > 20 ? 20 : nodelen)
569 569 ? -2 : v;
570 570 }
571 571 if (v == 0)
572 572 return -2;
573 573 off = v;
574 574 }
575 575 return -2;
576 576 }
577 577
578 578 static int nt_new(indexObject *self)
579 579 {
580 580 if (self->ntlength == self->ntcapacity) {
581 581 self->ntcapacity *= 2;
582 582 self->nt = realloc(self->nt,
583 583 self->ntcapacity * sizeof(nodetree));
584 584 if (self->nt == NULL) {
585 585 PyErr_SetString(PyExc_MemoryError, "out of memory");
586 586 return -1;
587 587 }
588 588 memset(&self->nt[self->ntlength], 0,
589 589 sizeof(nodetree) * (self->ntcapacity - self->ntlength));
590 590 }
591 591 return self->ntlength++;
592 592 }
593 593
594 594 static int nt_insert(indexObject *self, const char *node, int rev)
595 595 {
596 596 int level = 0;
597 597 int off = 0;
598 598
599 599 while (level < 20) {
600 600 int k = nt_level(node, level);
601 601 nodetree *n;
602 602 int v;
603 603
604 604 n = &self->nt[off];
605 605 v = n->children[k];
606 606
607 607 if (v == 0) {
608 608 n->children[k] = -rev - 1;
609 609 return 0;
610 610 }
611 611 if (v < 0) {
612 612 const char *oldnode = index_node(self, -v - 1);
613 613 int noff;
614 614
615 615 if (!oldnode || !memcmp(oldnode, node, 20)) {
616 616 n->children[k] = -rev - 1;
617 617 return 0;
618 618 }
619 619 noff = nt_new(self);
620 620 if (noff == -1)
621 621 return -1;
622 622 /* self->nt may have been changed by realloc */
623 623 self->nt[off].children[k] = noff;
624 624 off = noff;
625 625 n = &self->nt[off];
626 626 n->children[nt_level(oldnode, ++level)] = v;
627 627 if (level > self->ntdepth)
628 628 self->ntdepth = level;
629 629 self->ntsplits += 1;
630 630 } else {
631 631 level += 1;
632 632 off = v;
633 633 }
634 634 }
635 635
636 636 return -1;
637 637 }
638 638
639 639 /*
640 640 * Return values:
641 641 *
642 642 * -3: error (exception set)
643 643 * -2: not found (no exception set)
644 644 * rest: valid rev
645 645 */
646 646 static int index_find_node(indexObject *self,
647 647 const char *node, Py_ssize_t nodelen)
648 648 {
649 649 int rev;
650 650
651 651 self->ntlookups++;
652 652 rev = nt_find(self, node, nodelen);
653 653 if (rev >= -1)
654 654 return rev;
655 655
656 656 if (self->nt == NULL) {
657 657 self->ntcapacity = self->raw_length < 4
658 658 ? 4 : self->raw_length / 2;
659 659 self->nt = calloc(self->ntcapacity, sizeof(nodetree));
660 660 if (self->nt == NULL) {
661 661 PyErr_SetString(PyExc_MemoryError, "out of memory");
662 662 return -3;
663 663 }
664 664 self->ntlength = 1;
665 665 self->ntrev = (int)index_length(self) - 1;
666 666 self->ntlookups = 1;
667 667 self->ntmisses = 0;
668 668 }
669 669
670 670 /*
671 671 * For the first handful of lookups, we scan the entire index,
672 672 * and cache only the matching nodes. This optimizes for cases
673 673 * like "hg tip", where only a few nodes are accessed.
674 674 *
675 675 * After that, we cache every node we visit, using a single
676 676 * scan amortized over multiple lookups. This gives the best
677 677 * bulk performance, e.g. for "hg log".
678 678 */
679 679 if (self->ntmisses++ < 4) {
680 680 for (rev = self->ntrev - 1; rev >= 0; rev--) {
681 681 const char *n = index_node(self, rev);
682 682 if (n == NULL)
683 683 return -2;
684 684 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
685 685 if (nt_insert(self, n, rev) == -1)
686 686 return -3;
687 687 break;
688 688 }
689 689 }
690 690 } else {
691 691 for (rev = self->ntrev - 1; rev >= 0; rev--) {
692 692 const char *n = index_node(self, rev);
693 693 if (n == NULL)
694 694 return -2;
695 695 if (nt_insert(self, n, rev) == -1)
696 696 return -3;
697 697 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
698 698 break;
699 699 }
700 700 }
701 701 self->ntrev = rev;
702 702 }
703 703
704 704 if (rev >= 0)
705 705 return rev;
706 706 return -2;
707 707 }
708 708
709 709 static PyObject *raise_revlog_error(void)
710 710 {
711 711 static PyObject *errclass;
712 712 PyObject *mod = NULL, *errobj;
713 713
714 714 if (errclass == NULL) {
715 715 PyObject *dict;
716 716
717 717 mod = PyImport_ImportModule("mercurial.error");
718 718 if (mod == NULL)
719 719 goto classfail;
720 720
721 721 dict = PyModule_GetDict(mod);
722 722 if (dict == NULL)
723 723 goto classfail;
724 724
725 725 errclass = PyDict_GetItemString(dict, "RevlogError");
726 726 if (errclass == NULL) {
727 727 PyErr_SetString(PyExc_SystemError,
728 728 "could not find RevlogError");
729 729 goto classfail;
730 730 }
731 731 Py_INCREF(errclass);
732 732 }
733 733
734 734 errobj = PyObject_CallFunction(errclass, NULL);
735 735 if (errobj == NULL)
736 736 return NULL;
737 737 PyErr_SetObject(errclass, errobj);
738 738 return errobj;
739 739
740 740 classfail:
741 741 Py_XDECREF(mod);
742 742 return NULL;
743 743 }
744 744
745 745 static PyObject *index_getitem(indexObject *self, PyObject *value)
746 746 {
747 747 char *node;
748 748 Py_ssize_t nodelen;
749 749 int rev;
750 750
751 751 if (PyInt_Check(value))
752 752 return index_get(self, PyInt_AS_LONG(value));
753 753
754 754 if (PyString_AsStringAndSize(value, &node, &nodelen) == -1)
755 755 return NULL;
756 756 rev = index_find_node(self, node, nodelen);
757 757 if (rev >= -1)
758 758 return PyInt_FromLong(rev);
759 759 if (rev == -2)
760 760 raise_revlog_error();
761 761 return NULL;
762 762 }
763 763
764 764 static PyObject *index_m_get(indexObject *self, PyObject *args)
765 765 {
766 766 char *node;
767 767 int nodelen, rev;
768 768
769 769 if (!PyArg_ParseTuple(args, "s#", &node, &nodelen))
770 770 return NULL;
771 771
772 772 rev = index_find_node(self, node, nodelen);
773 773 if (rev == -3)
774 774 return NULL;
775 775 if (rev == -2)
776 776 Py_RETURN_NONE;
777 777 return PyInt_FromLong(rev);
778 778 }
779 779
780 780 static int index_contains(indexObject *self, PyObject *value)
781 781 {
782 782 char *node;
783 783 Py_ssize_t nodelen;
784 784
785 785 if (PyInt_Check(value)) {
786 786 long rev = PyInt_AS_LONG(value);
787 787 return rev >= -1 && rev < index_length(self);
788 788 }
789 789
790 790 if (!PyString_Check(value))
791 791 return 0;
792 792
793 793 node = PyString_AS_STRING(value);
794 794 nodelen = PyString_GET_SIZE(value);
795 795
796 796 switch (index_find_node(self, node, nodelen)) {
797 797 case -3:
798 798 return -1;
799 799 case -2:
800 800 return 0;
801 801 default:
802 802 return 1;
803 803 }
804 804 }
805 805
806 806 /*
807 807 * Invalidate any trie entries introduced by added revs.
808 808 */
809 809 static void nt_invalidate_added(indexObject *self, Py_ssize_t start)
810 810 {
811 811 Py_ssize_t i, len = PyList_GET_SIZE(self->added);
812 812
813 813 for (i = start; i < len; i++) {
814 814 PyObject *tuple = PyList_GET_ITEM(self->added, i);
815 815 PyObject *node = PyTuple_GET_ITEM(tuple, 7);
816 816
817 817 nt_insert(self, PyString_AS_STRING(node), -1);
818 818 }
819 819
820 820 if (start == 0) {
821 821 Py_DECREF(self->added);
822 822 self->added = NULL;
823 823 }
824 824 }
825 825
826 826 /*
827 827 * Delete a numeric range of revs, which must be at the end of the
828 828 * range, but exclude the sentinel nullid entry.
829 829 */
830 830 static int index_slice_del(indexObject *self, PyObject *item)
831 831 {
832 832 Py_ssize_t start, stop, step, slicelength;
833 833 Py_ssize_t length = index_length(self);
834 834
835 835 if (PySlice_GetIndicesEx((PySliceObject*)item, length,
836 836 &start, &stop, &step, &slicelength) < 0)
837 837 return -1;
838 838
839 839 if (slicelength <= 0)
840 840 return 0;
841 841
842 842 if ((step < 0 && start < stop) || (step > 0 && start > stop))
843 843 stop = start;
844 844
845 845 if (step < 0) {
846 846 stop = start + 1;
847 847 start = stop + step*(slicelength - 1) - 1;
848 848 step = -step;
849 849 }
850 850
851 851 if (step != 1) {
852 852 PyErr_SetString(PyExc_ValueError,
853 853 "revlog index delete requires step size of 1");
854 854 return -1;
855 855 }
856 856
857 857 if (stop != length - 1) {
858 858 PyErr_SetString(PyExc_IndexError,
859 859 "revlog index deletion indices are invalid");
860 860 return -1;
861 861 }
862 862
863 863 if (start < self->length - 1) {
864 864 if (self->nt) {
865 865 Py_ssize_t i;
866 866
867 867 for (i = start + 1; i < self->length - 1; i++) {
868 868 const char *node = index_node(self, i);
869 869
870 870 if (node)
871 871 nt_insert(self, node, -1);
872 872 }
873 873 if (self->added)
874 874 nt_invalidate_added(self, 0);
875 875 if (self->ntrev > start)
876 876 self->ntrev = (int)start;
877 877 }
878 878 self->length = start + 1;
879 879 return 0;
880 880 }
881 881
882 882 if (self->nt) {
883 883 nt_invalidate_added(self, start - self->length + 1);
884 884 if (self->ntrev > start)
885 885 self->ntrev = (int)start;
886 886 }
887 887 return self->added
888 888 ? PyList_SetSlice(self->added, start - self->length + 1,
889 889 PyList_GET_SIZE(self->added), NULL)
890 890 : 0;
891 891 }
892 892
893 893 /*
894 894 * Supported ops:
895 895 *
896 896 * slice deletion
897 897 * string assignment (extend node->rev mapping)
898 898 * string deletion (shrink node->rev mapping)
899 899 */
900 900 static int index_assign_subscript(indexObject *self, PyObject *item,
901 901 PyObject *value)
902 902 {
903 903 char *node;
904 904 Py_ssize_t nodelen;
905 905 long rev;
906 906
907 907 if (PySlice_Check(item) && value == NULL)
908 908 return index_slice_del(self, item);
909 909
910 910 if (node_check(item, &node, &nodelen) == -1)
911 911 return -1;
912 912
913 913 if (value == NULL)
914 914 return self->nt ? nt_insert(self, node, -1) : 0;
915 915 rev = PyInt_AsLong(value);
916 916 if (rev > INT_MAX || rev < 0) {
917 917 if (!PyErr_Occurred())
918 918 PyErr_SetString(PyExc_ValueError, "rev out of range");
919 919 return -1;
920 920 }
921 921 return nt_insert(self, node, (int)rev);
922 922 }
923 923
924 924 /*
925 925 * Find all RevlogNG entries in an index that has inline data. Update
926 926 * the optional "offsets" table with those entries.
927 927 */
928 928 static long inline_scan(indexObject *self, const char **offsets)
929 929 {
930 930 const char *data = PyString_AS_STRING(self->data);
931 931 const char *end = data + PyString_GET_SIZE(self->data);
932 932 const long hdrsize = 64;
933 933 long incr = hdrsize;
934 934 Py_ssize_t len = 0;
935 935
936 936 while (data + hdrsize <= end) {
937 937 uint32_t comp_len;
938 938 const char *old_data;
939 939 /* 3rd element of header is length of compressed inline data */
940 940 comp_len = getbe32(data + 8);
941 941 incr = hdrsize + comp_len;
942 942 if (incr < hdrsize)
943 943 break;
944 944 if (offsets)
945 945 offsets[len] = data;
946 946 len++;
947 947 old_data = data;
948 948 data += incr;
949 949 if (data <= old_data)
950 950 break;
951 951 }
952 952
953 953 if (data != end && data + hdrsize != end) {
954 954 if (!PyErr_Occurred())
955 955 PyErr_SetString(PyExc_ValueError, "corrupt index file");
956 956 return -1;
957 957 }
958 958
959 959 return len;
960 960 }
961 961
962 962 static int index_init(indexObject *self, PyObject *args)
963 963 {
964 964 PyObject *data_obj, *inlined_obj;
965 965 Py_ssize_t size;
966 966
967 967 if (!PyArg_ParseTuple(args, "OO", &data_obj, &inlined_obj))
968 968 return -1;
969 969 if (!PyString_Check(data_obj)) {
970 970 PyErr_SetString(PyExc_TypeError, "data is not a string");
971 971 return -1;
972 972 }
973 973 size = PyString_GET_SIZE(data_obj);
974 974
975 975 self->inlined = inlined_obj && PyObject_IsTrue(inlined_obj);
976 976 self->data = data_obj;
977 977 self->cache = NULL;
978 978
979 979 self->added = NULL;
980 980 self->offsets = NULL;
981 981 self->nt = NULL;
982 982 self->ntlength = self->ntcapacity = 0;
983 983 self->ntdepth = self->ntsplits = 0;
984 984 self->ntlookups = self->ntmisses = 0;
985 985 self->ntrev = -1;
986 986 Py_INCREF(self->data);
987 987
988 988 if (self->inlined) {
989 989 long len = inline_scan(self, NULL);
990 990 if (len == -1)
991 991 goto bail;
992 992 self->raw_length = len;
993 993 self->length = len + 1;
994 994 } else {
995 995 if (size % 64) {
996 996 PyErr_SetString(PyExc_ValueError, "corrupt index file");
997 997 goto bail;
998 998 }
999 999 self->raw_length = size / 64;
1000 1000 self->length = self->raw_length + 1;
1001 1001 }
1002 1002
1003 1003 return 0;
1004 1004 bail:
1005 1005 return -1;
1006 1006 }
1007 1007
1008 1008 static PyObject *index_nodemap(indexObject *self)
1009 1009 {
1010 1010 Py_INCREF(self);
1011 1011 return (PyObject *)self;
1012 1012 }
1013 1013
1014 1014 static void index_dealloc(indexObject *self)
1015 1015 {
1016 1016 _index_clearcaches(self);
1017 1017 Py_DECREF(self->data);
1018 1018 Py_XDECREF(self->added);
1019 1019 PyObject_Del(self);
1020 1020 }
1021 1021
1022 1022 static PySequenceMethods index_sequence_methods = {
1023 1023 (lenfunc)index_length, /* sq_length */
1024 1024 0, /* sq_concat */
1025 1025 0, /* sq_repeat */
1026 1026 (ssizeargfunc)index_get, /* sq_item */
1027 1027 0, /* sq_slice */
1028 1028 0, /* sq_ass_item */
1029 1029 0, /* sq_ass_slice */
1030 1030 (objobjproc)index_contains, /* sq_contains */
1031 1031 };
1032 1032
1033 1033 static PyMappingMethods index_mapping_methods = {
1034 1034 (lenfunc)index_length, /* mp_length */
1035 1035 (binaryfunc)index_getitem, /* mp_subscript */
1036 1036 (objobjargproc)index_assign_subscript, /* mp_ass_subscript */
1037 1037 };
1038 1038
1039 1039 static PyMethodDef index_methods[] = {
1040 1040 {"clearcaches", (PyCFunction)index_clearcaches, METH_NOARGS,
1041 1041 "clear the index caches"},
1042 1042 {"get", (PyCFunction)index_m_get, METH_VARARGS,
1043 1043 "get an index entry"},
1044 1044 {"insert", (PyCFunction)index_insert, METH_VARARGS,
1045 1045 "insert an index entry"},
1046 1046 {"stats", (PyCFunction)index_stats, METH_NOARGS,
1047 1047 "stats for the index"},
1048 1048 {NULL} /* Sentinel */
1049 1049 };
1050 1050
1051 1051 static PyGetSetDef index_getset[] = {
1052 1052 {"nodemap", (getter)index_nodemap, NULL, "nodemap", NULL},
1053 1053 {NULL} /* Sentinel */
1054 1054 };
1055 1055
1056 1056 static PyTypeObject indexType = {
1057 1057 PyObject_HEAD_INIT(NULL)
1058 1058 0, /* ob_size */
1059 1059 "parsers.index", /* tp_name */
1060 1060 sizeof(indexObject), /* tp_basicsize */
1061 1061 0, /* tp_itemsize */
1062 1062 (destructor)index_dealloc, /* tp_dealloc */
1063 1063 0, /* tp_print */
1064 1064 0, /* tp_getattr */
1065 1065 0, /* tp_setattr */
1066 1066 0, /* tp_compare */
1067 1067 0, /* tp_repr */
1068 1068 0, /* tp_as_number */
1069 1069 &index_sequence_methods, /* tp_as_sequence */
1070 1070 &index_mapping_methods, /* tp_as_mapping */
1071 1071 0, /* tp_hash */
1072 1072 0, /* tp_call */
1073 1073 0, /* tp_str */
1074 1074 0, /* tp_getattro */
1075 1075 0, /* tp_setattro */
1076 1076 0, /* tp_as_buffer */
1077 1077 Py_TPFLAGS_DEFAULT, /* tp_flags */
1078 1078 "revlog index", /* tp_doc */
1079 1079 0, /* tp_traverse */
1080 1080 0, /* tp_clear */
1081 1081 0, /* tp_richcompare */
1082 1082 0, /* tp_weaklistoffset */
1083 1083 0, /* tp_iter */
1084 1084 0, /* tp_iternext */
1085 1085 index_methods, /* tp_methods */
1086 1086 0, /* tp_members */
1087 1087 index_getset, /* tp_getset */
1088 1088 0, /* tp_base */
1089 1089 0, /* tp_dict */
1090 1090 0, /* tp_descr_get */
1091 1091 0, /* tp_descr_set */
1092 1092 0, /* tp_dictoffset */
1093 1093 (initproc)index_init, /* tp_init */
1094 1094 0, /* tp_alloc */
1095 PyType_GenericNew, /* tp_new */
1096 1095 };
1097 1096
1098 1097 /*
1099 1098 * returns a tuple of the form (index, index, cache) with elements as
1100 1099 * follows:
1101 1100 *
1102 1101 * index: an index object that lazily parses RevlogNG records
1103 1102 * cache: if data is inlined, a tuple (index_file_content, 0), else None
1104 1103 *
1105 1104 * added complications are for backwards compatibility
1106 1105 */
1107 1106 static PyObject *parse_index2(PyObject *self, PyObject *args)
1108 1107 {
1109 1108 PyObject *tuple = NULL, *cache = NULL;
1110 1109 indexObject *idx;
1111 1110 int ret;
1112 1111
1113 1112 idx = PyObject_New(indexObject, &indexType);
1114 1113 if (idx == NULL)
1115 1114 goto bail;
1116 1115
1117 1116 ret = index_init(idx, args);
1118 1117 if (ret == -1)
1119 1118 goto bail;
1120 1119
1121 1120 if (idx->inlined) {
1122 1121 cache = Py_BuildValue("iO", 0, idx->data);
1123 1122 if (cache == NULL)
1124 1123 goto bail;
1125 1124 } else {
1126 1125 cache = Py_None;
1127 1126 Py_INCREF(cache);
1128 1127 }
1129 1128
1130 1129 tuple = Py_BuildValue("NN", idx, cache);
1131 1130 if (!tuple)
1132 1131 goto bail;
1133 1132 return tuple;
1134 1133
1135 1134 bail:
1136 1135 Py_XDECREF(idx);
1137 1136 Py_XDECREF(cache);
1138 1137 Py_XDECREF(tuple);
1139 1138 return NULL;
1140 1139 }
1141 1140
1142 1141 static char parsers_doc[] = "Efficient content parsing.";
1143 1142
1144 1143 static PyMethodDef methods[] = {
1145 1144 {"parse_manifest", parse_manifest, METH_VARARGS, "parse a manifest\n"},
1146 1145 {"parse_dirstate", parse_dirstate, METH_VARARGS, "parse a dirstate\n"},
1147 1146 {"parse_index2", parse_index2, METH_VARARGS, "parse a revlog index\n"},
1148 1147 {NULL, NULL}
1149 1148 };
1150 1149
1151 1150 static void module_init(PyObject *mod)
1152 1151 {
1152 indexType.tp_new = PyType_GenericNew;
1153 1153 if (PyType_Ready(&indexType) < 0)
1154 1154 return;
1155 1155 Py_INCREF(&indexType);
1156 1156
1157 1157 PyModule_AddObject(mod, "index", (PyObject *)&indexType);
1158 1158
1159 1159 nullentry = Py_BuildValue("iiiiiiis#", 0, 0, 0,
1160 1160 -1, -1, -1, -1, nullid, 20);
1161 1161 if (nullentry)
1162 1162 PyObject_GC_UnTrack(nullentry);
1163 1163 }
1164 1164
1165 1165 #ifdef IS_PY3K
1166 1166 static struct PyModuleDef parsers_module = {
1167 1167 PyModuleDef_HEAD_INIT,
1168 1168 "parsers",
1169 1169 parsers_doc,
1170 1170 -1,
1171 1171 methods
1172 1172 };
1173 1173
1174 1174 PyMODINIT_FUNC PyInit_parsers(void)
1175 1175 {
1176 1176 PyObject *mod = PyModule_Create(&parsers_module);
1177 1177 module_init(mod);
1178 1178 return mod;
1179 1179 }
1180 1180 #else
1181 1181 PyMODINIT_FUNC initparsers(void)
1182 1182 {
1183 1183 PyObject *mod = Py_InitModule3("parsers", methods, parsers_doc);
1184 1184 module_init(mod);
1185 1185 }
1186 1186 #endif
General Comments 0
You need to be logged in to leave comments. Login now