##// END OF EJS Templates
reachableroots: fix transposition of set and list types in PyArg_ParseTuple...
Augie Fackler -
r26009:bbb69869 default
parent child Browse files
Show More
@@ -1,2785 +1,2787 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 <stddef.h>
13 13 #include <string.h>
14 14
15 15 #include "util.h"
16 16
17 17 static char *versionerrortext = "Python minor version mismatch";
18 18
19 19 static int8_t hextable[256] = {
20 20 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
21 21 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
22 22 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
23 23 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -1, -1, -1, -1, -1, -1, /* 0-9 */
24 24 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* A-F */
25 25 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
26 26 -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1, /* a-f */
27 27 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
28 28 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
29 29 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
30 30 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
31 31 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
32 32 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
33 33 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
34 34 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
35 35 -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
36 36 };
37 37
38 38 static char lowertable[128] = {
39 39 '\x00', '\x01', '\x02', '\x03', '\x04', '\x05', '\x06', '\x07',
40 40 '\x08', '\x09', '\x0a', '\x0b', '\x0c', '\x0d', '\x0e', '\x0f',
41 41 '\x10', '\x11', '\x12', '\x13', '\x14', '\x15', '\x16', '\x17',
42 42 '\x18', '\x19', '\x1a', '\x1b', '\x1c', '\x1d', '\x1e', '\x1f',
43 43 '\x20', '\x21', '\x22', '\x23', '\x24', '\x25', '\x26', '\x27',
44 44 '\x28', '\x29', '\x2a', '\x2b', '\x2c', '\x2d', '\x2e', '\x2f',
45 45 '\x30', '\x31', '\x32', '\x33', '\x34', '\x35', '\x36', '\x37',
46 46 '\x38', '\x39', '\x3a', '\x3b', '\x3c', '\x3d', '\x3e', '\x3f',
47 47 '\x40',
48 48 '\x61', '\x62', '\x63', '\x64', '\x65', '\x66', '\x67', /* A-G */
49 49 '\x68', '\x69', '\x6a', '\x6b', '\x6c', '\x6d', '\x6e', '\x6f', /* H-O */
50 50 '\x70', '\x71', '\x72', '\x73', '\x74', '\x75', '\x76', '\x77', /* P-W */
51 51 '\x78', '\x79', '\x7a', /* X-Z */
52 52 '\x5b', '\x5c', '\x5d', '\x5e', '\x5f',
53 53 '\x60', '\x61', '\x62', '\x63', '\x64', '\x65', '\x66', '\x67',
54 54 '\x68', '\x69', '\x6a', '\x6b', '\x6c', '\x6d', '\x6e', '\x6f',
55 55 '\x70', '\x71', '\x72', '\x73', '\x74', '\x75', '\x76', '\x77',
56 56 '\x78', '\x79', '\x7a', '\x7b', '\x7c', '\x7d', '\x7e', '\x7f'
57 57 };
58 58
59 59 static char uppertable[128] = {
60 60 '\x00', '\x01', '\x02', '\x03', '\x04', '\x05', '\x06', '\x07',
61 61 '\x08', '\x09', '\x0a', '\x0b', '\x0c', '\x0d', '\x0e', '\x0f',
62 62 '\x10', '\x11', '\x12', '\x13', '\x14', '\x15', '\x16', '\x17',
63 63 '\x18', '\x19', '\x1a', '\x1b', '\x1c', '\x1d', '\x1e', '\x1f',
64 64 '\x20', '\x21', '\x22', '\x23', '\x24', '\x25', '\x26', '\x27',
65 65 '\x28', '\x29', '\x2a', '\x2b', '\x2c', '\x2d', '\x2e', '\x2f',
66 66 '\x30', '\x31', '\x32', '\x33', '\x34', '\x35', '\x36', '\x37',
67 67 '\x38', '\x39', '\x3a', '\x3b', '\x3c', '\x3d', '\x3e', '\x3f',
68 68 '\x40', '\x41', '\x42', '\x43', '\x44', '\x45', '\x46', '\x47',
69 69 '\x48', '\x49', '\x4a', '\x4b', '\x4c', '\x4d', '\x4e', '\x4f',
70 70 '\x50', '\x51', '\x52', '\x53', '\x54', '\x55', '\x56', '\x57',
71 71 '\x58', '\x59', '\x5a', '\x5b', '\x5c', '\x5d', '\x5e', '\x5f',
72 72 '\x60',
73 73 '\x41', '\x42', '\x43', '\x44', '\x45', '\x46', '\x47', /* a-g */
74 74 '\x48', '\x49', '\x4a', '\x4b', '\x4c', '\x4d', '\x4e', '\x4f', /* h-o */
75 75 '\x50', '\x51', '\x52', '\x53', '\x54', '\x55', '\x56', '\x57', /* p-w */
76 76 '\x58', '\x59', '\x5a', /* x-z */
77 77 '\x7b', '\x7c', '\x7d', '\x7e', '\x7f'
78 78 };
79 79
80 80 static inline int hexdigit(const char *p, Py_ssize_t off)
81 81 {
82 82 int8_t val = hextable[(unsigned char)p[off]];
83 83
84 84 if (val >= 0) {
85 85 return val;
86 86 }
87 87
88 88 PyErr_SetString(PyExc_ValueError, "input contains non-hex character");
89 89 return 0;
90 90 }
91 91
92 92 /*
93 93 * Turn a hex-encoded string into binary.
94 94 */
95 95 PyObject *unhexlify(const char *str, int len)
96 96 {
97 97 PyObject *ret;
98 98 char *d;
99 99 int i;
100 100
101 101 ret = PyBytes_FromStringAndSize(NULL, len / 2);
102 102
103 103 if (!ret)
104 104 return NULL;
105 105
106 106 d = PyBytes_AsString(ret);
107 107
108 108 for (i = 0; i < len;) {
109 109 int hi = hexdigit(str, i++);
110 110 int lo = hexdigit(str, i++);
111 111 *d++ = (hi << 4) | lo;
112 112 }
113 113
114 114 return ret;
115 115 }
116 116
117 117 static inline PyObject *_asciitransform(PyObject *str_obj,
118 118 const char table[128],
119 119 PyObject *fallback_fn)
120 120 {
121 121 char *str, *newstr;
122 122 Py_ssize_t i, len;
123 123 PyObject *newobj = NULL;
124 124 PyObject *ret = NULL;
125 125
126 126 str = PyBytes_AS_STRING(str_obj);
127 127 len = PyBytes_GET_SIZE(str_obj);
128 128
129 129 newobj = PyBytes_FromStringAndSize(NULL, len);
130 130 if (!newobj)
131 131 goto quit;
132 132
133 133 newstr = PyBytes_AS_STRING(newobj);
134 134
135 135 for (i = 0; i < len; i++) {
136 136 char c = str[i];
137 137 if (c & 0x80) {
138 138 if (fallback_fn != NULL) {
139 139 ret = PyObject_CallFunctionObjArgs(fallback_fn,
140 140 str_obj, NULL);
141 141 } else {
142 142 PyObject *err = PyUnicodeDecodeError_Create(
143 143 "ascii", str, len, i, (i + 1),
144 144 "unexpected code byte");
145 145 PyErr_SetObject(PyExc_UnicodeDecodeError, err);
146 146 Py_XDECREF(err);
147 147 }
148 148 goto quit;
149 149 }
150 150 newstr[i] = table[(unsigned char)c];
151 151 }
152 152
153 153 ret = newobj;
154 154 Py_INCREF(ret);
155 155 quit:
156 156 Py_XDECREF(newobj);
157 157 return ret;
158 158 }
159 159
160 160 static PyObject *asciilower(PyObject *self, PyObject *args)
161 161 {
162 162 PyObject *str_obj;
163 163 if (!PyArg_ParseTuple(args, "O!:asciilower", &PyBytes_Type, &str_obj))
164 164 return NULL;
165 165 return _asciitransform(str_obj, lowertable, NULL);
166 166 }
167 167
168 168 static PyObject *asciiupper(PyObject *self, PyObject *args)
169 169 {
170 170 PyObject *str_obj;
171 171 if (!PyArg_ParseTuple(args, "O!:asciiupper", &PyBytes_Type, &str_obj))
172 172 return NULL;
173 173 return _asciitransform(str_obj, uppertable, NULL);
174 174 }
175 175
176 176 static inline PyObject *_dict_new_presized(Py_ssize_t expected_size)
177 177 {
178 178 /* _PyDict_NewPresized expects a minused parameter, but it actually
179 179 creates a dictionary that's the nearest power of two bigger than the
180 180 parameter. For example, with the initial minused = 1000, the
181 181 dictionary created has size 1024. Of course in a lot of cases that
182 182 can be greater than the maximum load factor Python's dict object
183 183 expects (= 2/3), so as soon as we cross the threshold we'll resize
184 184 anyway. So create a dictionary that's at least 3/2 the size. */
185 185 return _PyDict_NewPresized(((1 + expected_size) / 2) * 3);
186 186 }
187 187
188 188 static PyObject *dict_new_presized(PyObject *self, PyObject *args)
189 189 {
190 190 Py_ssize_t expected_size;
191 191
192 192 if (!PyArg_ParseTuple(args, "n:make_presized_dict", &expected_size))
193 193 return NULL;
194 194
195 195 return _dict_new_presized(expected_size);
196 196 }
197 197
198 198 static PyObject *make_file_foldmap(PyObject *self, PyObject *args)
199 199 {
200 200 PyObject *dmap, *spec_obj, *normcase_fallback;
201 201 PyObject *file_foldmap = NULL;
202 202 enum normcase_spec spec;
203 203 PyObject *k, *v;
204 204 dirstateTupleObject *tuple;
205 205 Py_ssize_t pos = 0;
206 206 const char *table;
207 207
208 208 if (!PyArg_ParseTuple(args, "O!O!O!:make_file_foldmap",
209 209 &PyDict_Type, &dmap,
210 210 &PyInt_Type, &spec_obj,
211 211 &PyFunction_Type, &normcase_fallback))
212 212 goto quit;
213 213
214 214 spec = (int)PyInt_AS_LONG(spec_obj);
215 215 switch (spec) {
216 216 case NORMCASE_LOWER:
217 217 table = lowertable;
218 218 break;
219 219 case NORMCASE_UPPER:
220 220 table = uppertable;
221 221 break;
222 222 case NORMCASE_OTHER:
223 223 table = NULL;
224 224 break;
225 225 default:
226 226 PyErr_SetString(PyExc_TypeError, "invalid normcasespec");
227 227 goto quit;
228 228 }
229 229
230 230 /* Add some more entries to deal with additions outside this
231 231 function. */
232 232 file_foldmap = _dict_new_presized((PyDict_Size(dmap) / 10) * 11);
233 233 if (file_foldmap == NULL)
234 234 goto quit;
235 235
236 236 while (PyDict_Next(dmap, &pos, &k, &v)) {
237 237 if (!dirstate_tuple_check(v)) {
238 238 PyErr_SetString(PyExc_TypeError,
239 239 "expected a dirstate tuple");
240 240 goto quit;
241 241 }
242 242
243 243 tuple = (dirstateTupleObject *)v;
244 244 if (tuple->state != 'r') {
245 245 PyObject *normed;
246 246 if (table != NULL) {
247 247 normed = _asciitransform(k, table,
248 248 normcase_fallback);
249 249 } else {
250 250 normed = PyObject_CallFunctionObjArgs(
251 251 normcase_fallback, k, NULL);
252 252 }
253 253
254 254 if (normed == NULL)
255 255 goto quit;
256 256 if (PyDict_SetItem(file_foldmap, normed, k) == -1)
257 257 goto quit;
258 258 }
259 259 }
260 260 return file_foldmap;
261 261 quit:
262 262 Py_XDECREF(file_foldmap);
263 263 return NULL;
264 264 }
265 265
266 266 /*
267 267 * This code assumes that a manifest is stitched together with newline
268 268 * ('\n') characters.
269 269 */
270 270 static PyObject *parse_manifest(PyObject *self, PyObject *args)
271 271 {
272 272 PyObject *mfdict, *fdict;
273 273 char *str, *start, *end;
274 274 int len;
275 275
276 276 if (!PyArg_ParseTuple(args, "O!O!s#:parse_manifest",
277 277 &PyDict_Type, &mfdict,
278 278 &PyDict_Type, &fdict,
279 279 &str, &len))
280 280 goto quit;
281 281
282 282 start = str;
283 283 end = str + len;
284 284 while (start < end) {
285 285 PyObject *file = NULL, *node = NULL;
286 286 PyObject *flags = NULL;
287 287 char *zero = NULL, *newline = NULL;
288 288 ptrdiff_t nlen;
289 289
290 290 zero = memchr(start, '\0', end - start);
291 291 if (!zero) {
292 292 PyErr_SetString(PyExc_ValueError,
293 293 "manifest entry has no separator");
294 294 goto quit;
295 295 }
296 296
297 297 newline = memchr(zero + 1, '\n', end - (zero + 1));
298 298 if (!newline) {
299 299 PyErr_SetString(PyExc_ValueError,
300 300 "manifest contains trailing garbage");
301 301 goto quit;
302 302 }
303 303
304 304 file = PyBytes_FromStringAndSize(start, zero - start);
305 305
306 306 if (!file)
307 307 goto bail;
308 308
309 309 nlen = newline - zero - 1;
310 310
311 311 node = unhexlify(zero + 1, nlen > 40 ? 40 : (int)nlen);
312 312 if (!node)
313 313 goto bail;
314 314
315 315 if (nlen > 40) {
316 316 flags = PyBytes_FromStringAndSize(zero + 41,
317 317 nlen - 40);
318 318 if (!flags)
319 319 goto bail;
320 320
321 321 if (PyDict_SetItem(fdict, file, flags) == -1)
322 322 goto bail;
323 323 }
324 324
325 325 if (PyDict_SetItem(mfdict, file, node) == -1)
326 326 goto bail;
327 327
328 328 start = newline + 1;
329 329
330 330 Py_XDECREF(flags);
331 331 Py_XDECREF(node);
332 332 Py_XDECREF(file);
333 333 continue;
334 334 bail:
335 335 Py_XDECREF(flags);
336 336 Py_XDECREF(node);
337 337 Py_XDECREF(file);
338 338 goto quit;
339 339 }
340 340
341 341 Py_INCREF(Py_None);
342 342 return Py_None;
343 343 quit:
344 344 return NULL;
345 345 }
346 346
347 347 static inline dirstateTupleObject *make_dirstate_tuple(char state, int mode,
348 348 int size, int mtime)
349 349 {
350 350 dirstateTupleObject *t = PyObject_New(dirstateTupleObject,
351 351 &dirstateTupleType);
352 352 if (!t)
353 353 return NULL;
354 354 t->state = state;
355 355 t->mode = mode;
356 356 t->size = size;
357 357 t->mtime = mtime;
358 358 return t;
359 359 }
360 360
361 361 static PyObject *dirstate_tuple_new(PyTypeObject *subtype, PyObject *args,
362 362 PyObject *kwds)
363 363 {
364 364 /* We do all the initialization here and not a tp_init function because
365 365 * dirstate_tuple is immutable. */
366 366 dirstateTupleObject *t;
367 367 char state;
368 368 int size, mode, mtime;
369 369 if (!PyArg_ParseTuple(args, "ciii", &state, &mode, &size, &mtime))
370 370 return NULL;
371 371
372 372 t = (dirstateTupleObject *)subtype->tp_alloc(subtype, 1);
373 373 if (!t)
374 374 return NULL;
375 375 t->state = state;
376 376 t->mode = mode;
377 377 t->size = size;
378 378 t->mtime = mtime;
379 379
380 380 return (PyObject *)t;
381 381 }
382 382
383 383 static void dirstate_tuple_dealloc(PyObject *o)
384 384 {
385 385 PyObject_Del(o);
386 386 }
387 387
388 388 static Py_ssize_t dirstate_tuple_length(PyObject *o)
389 389 {
390 390 return 4;
391 391 }
392 392
393 393 static PyObject *dirstate_tuple_item(PyObject *o, Py_ssize_t i)
394 394 {
395 395 dirstateTupleObject *t = (dirstateTupleObject *)o;
396 396 switch (i) {
397 397 case 0:
398 398 return PyBytes_FromStringAndSize(&t->state, 1);
399 399 case 1:
400 400 return PyInt_FromLong(t->mode);
401 401 case 2:
402 402 return PyInt_FromLong(t->size);
403 403 case 3:
404 404 return PyInt_FromLong(t->mtime);
405 405 default:
406 406 PyErr_SetString(PyExc_IndexError, "index out of range");
407 407 return NULL;
408 408 }
409 409 }
410 410
411 411 static PySequenceMethods dirstate_tuple_sq = {
412 412 dirstate_tuple_length, /* sq_length */
413 413 0, /* sq_concat */
414 414 0, /* sq_repeat */
415 415 dirstate_tuple_item, /* sq_item */
416 416 0, /* sq_ass_item */
417 417 0, /* sq_contains */
418 418 0, /* sq_inplace_concat */
419 419 0 /* sq_inplace_repeat */
420 420 };
421 421
422 422 PyTypeObject dirstateTupleType = {
423 423 PyVarObject_HEAD_INIT(NULL, 0)
424 424 "dirstate_tuple", /* tp_name */
425 425 sizeof(dirstateTupleObject),/* tp_basicsize */
426 426 0, /* tp_itemsize */
427 427 (destructor)dirstate_tuple_dealloc, /* tp_dealloc */
428 428 0, /* tp_print */
429 429 0, /* tp_getattr */
430 430 0, /* tp_setattr */
431 431 0, /* tp_compare */
432 432 0, /* tp_repr */
433 433 0, /* tp_as_number */
434 434 &dirstate_tuple_sq, /* tp_as_sequence */
435 435 0, /* tp_as_mapping */
436 436 0, /* tp_hash */
437 437 0, /* tp_call */
438 438 0, /* tp_str */
439 439 0, /* tp_getattro */
440 440 0, /* tp_setattro */
441 441 0, /* tp_as_buffer */
442 442 Py_TPFLAGS_DEFAULT, /* tp_flags */
443 443 "dirstate tuple", /* tp_doc */
444 444 0, /* tp_traverse */
445 445 0, /* tp_clear */
446 446 0, /* tp_richcompare */
447 447 0, /* tp_weaklistoffset */
448 448 0, /* tp_iter */
449 449 0, /* tp_iternext */
450 450 0, /* tp_methods */
451 451 0, /* tp_members */
452 452 0, /* tp_getset */
453 453 0, /* tp_base */
454 454 0, /* tp_dict */
455 455 0, /* tp_descr_get */
456 456 0, /* tp_descr_set */
457 457 0, /* tp_dictoffset */
458 458 0, /* tp_init */
459 459 0, /* tp_alloc */
460 460 dirstate_tuple_new, /* tp_new */
461 461 };
462 462
463 463 static PyObject *parse_dirstate(PyObject *self, PyObject *args)
464 464 {
465 465 PyObject *dmap, *cmap, *parents = NULL, *ret = NULL;
466 466 PyObject *fname = NULL, *cname = NULL, *entry = NULL;
467 467 char state, *cur, *str, *cpos;
468 468 int mode, size, mtime;
469 469 unsigned int flen, len, pos = 40;
470 470 int readlen;
471 471
472 472 if (!PyArg_ParseTuple(args, "O!O!s#:parse_dirstate",
473 473 &PyDict_Type, &dmap,
474 474 &PyDict_Type, &cmap,
475 475 &str, &readlen))
476 476 goto quit;
477 477
478 478 if (readlen < 0)
479 479 goto quit;
480 480
481 481 len = readlen;
482 482
483 483 /* read parents */
484 484 if (len < 40)
485 485 goto quit;
486 486
487 487 parents = Py_BuildValue("s#s#", str, 20, str + 20, 20);
488 488 if (!parents)
489 489 goto quit;
490 490
491 491 /* read filenames */
492 492 while (pos >= 40 && pos < len) {
493 493 cur = str + pos;
494 494 /* unpack header */
495 495 state = *cur;
496 496 mode = getbe32(cur + 1);
497 497 size = getbe32(cur + 5);
498 498 mtime = getbe32(cur + 9);
499 499 flen = getbe32(cur + 13);
500 500 pos += 17;
501 501 cur += 17;
502 502 if (flen > len - pos) {
503 503 PyErr_SetString(PyExc_ValueError, "overflow in dirstate");
504 504 goto quit;
505 505 }
506 506
507 507 entry = (PyObject *)make_dirstate_tuple(state, mode, size,
508 508 mtime);
509 509 cpos = memchr(cur, 0, flen);
510 510 if (cpos) {
511 511 fname = PyBytes_FromStringAndSize(cur, cpos - cur);
512 512 cname = PyBytes_FromStringAndSize(cpos + 1,
513 513 flen - (cpos - cur) - 1);
514 514 if (!fname || !cname ||
515 515 PyDict_SetItem(cmap, fname, cname) == -1 ||
516 516 PyDict_SetItem(dmap, fname, entry) == -1)
517 517 goto quit;
518 518 Py_DECREF(cname);
519 519 } else {
520 520 fname = PyBytes_FromStringAndSize(cur, flen);
521 521 if (!fname ||
522 522 PyDict_SetItem(dmap, fname, entry) == -1)
523 523 goto quit;
524 524 }
525 525 Py_DECREF(fname);
526 526 Py_DECREF(entry);
527 527 fname = cname = entry = NULL;
528 528 pos += flen;
529 529 }
530 530
531 531 ret = parents;
532 532 Py_INCREF(ret);
533 533 quit:
534 534 Py_XDECREF(fname);
535 535 Py_XDECREF(cname);
536 536 Py_XDECREF(entry);
537 537 Py_XDECREF(parents);
538 538 return ret;
539 539 }
540 540
541 541 /*
542 542 * Efficiently pack a dirstate object into its on-disk format.
543 543 */
544 544 static PyObject *pack_dirstate(PyObject *self, PyObject *args)
545 545 {
546 546 PyObject *packobj = NULL;
547 547 PyObject *map, *copymap, *pl, *mtime_unset = NULL;
548 548 Py_ssize_t nbytes, pos, l;
549 549 PyObject *k, *v = NULL, *pn;
550 550 char *p, *s;
551 551 double now;
552 552
553 553 if (!PyArg_ParseTuple(args, "O!O!Od:pack_dirstate",
554 554 &PyDict_Type, &map, &PyDict_Type, &copymap,
555 555 &pl, &now))
556 556 return NULL;
557 557
558 558 if (!PySequence_Check(pl) || PySequence_Size(pl) != 2) {
559 559 PyErr_SetString(PyExc_TypeError, "expected 2-element sequence");
560 560 return NULL;
561 561 }
562 562
563 563 /* Figure out how much we need to allocate. */
564 564 for (nbytes = 40, pos = 0; PyDict_Next(map, &pos, &k, &v);) {
565 565 PyObject *c;
566 566 if (!PyString_Check(k)) {
567 567 PyErr_SetString(PyExc_TypeError, "expected string key");
568 568 goto bail;
569 569 }
570 570 nbytes += PyString_GET_SIZE(k) + 17;
571 571 c = PyDict_GetItem(copymap, k);
572 572 if (c) {
573 573 if (!PyString_Check(c)) {
574 574 PyErr_SetString(PyExc_TypeError,
575 575 "expected string key");
576 576 goto bail;
577 577 }
578 578 nbytes += PyString_GET_SIZE(c) + 1;
579 579 }
580 580 }
581 581
582 582 packobj = PyString_FromStringAndSize(NULL, nbytes);
583 583 if (packobj == NULL)
584 584 goto bail;
585 585
586 586 p = PyString_AS_STRING(packobj);
587 587
588 588 pn = PySequence_ITEM(pl, 0);
589 589 if (PyString_AsStringAndSize(pn, &s, &l) == -1 || l != 20) {
590 590 PyErr_SetString(PyExc_TypeError, "expected a 20-byte hash");
591 591 goto bail;
592 592 }
593 593 memcpy(p, s, l);
594 594 p += 20;
595 595 pn = PySequence_ITEM(pl, 1);
596 596 if (PyString_AsStringAndSize(pn, &s, &l) == -1 || l != 20) {
597 597 PyErr_SetString(PyExc_TypeError, "expected a 20-byte hash");
598 598 goto bail;
599 599 }
600 600 memcpy(p, s, l);
601 601 p += 20;
602 602
603 603 for (pos = 0; PyDict_Next(map, &pos, &k, &v); ) {
604 604 dirstateTupleObject *tuple;
605 605 char state;
606 606 uint32_t mode, size, mtime;
607 607 Py_ssize_t len, l;
608 608 PyObject *o;
609 609 char *t;
610 610
611 611 if (!dirstate_tuple_check(v)) {
612 612 PyErr_SetString(PyExc_TypeError,
613 613 "expected a dirstate tuple");
614 614 goto bail;
615 615 }
616 616 tuple = (dirstateTupleObject *)v;
617 617
618 618 state = tuple->state;
619 619 mode = tuple->mode;
620 620 size = tuple->size;
621 621 mtime = tuple->mtime;
622 622 if (state == 'n' && mtime == (uint32_t)now) {
623 623 /* See pure/parsers.py:pack_dirstate for why we do
624 624 * this. */
625 625 mtime = -1;
626 626 mtime_unset = (PyObject *)make_dirstate_tuple(
627 627 state, mode, size, mtime);
628 628 if (!mtime_unset)
629 629 goto bail;
630 630 if (PyDict_SetItem(map, k, mtime_unset) == -1)
631 631 goto bail;
632 632 Py_DECREF(mtime_unset);
633 633 mtime_unset = NULL;
634 634 }
635 635 *p++ = state;
636 636 putbe32(mode, p);
637 637 putbe32(size, p + 4);
638 638 putbe32(mtime, p + 8);
639 639 t = p + 12;
640 640 p += 16;
641 641 len = PyString_GET_SIZE(k);
642 642 memcpy(p, PyString_AS_STRING(k), len);
643 643 p += len;
644 644 o = PyDict_GetItem(copymap, k);
645 645 if (o) {
646 646 *p++ = '\0';
647 647 l = PyString_GET_SIZE(o);
648 648 memcpy(p, PyString_AS_STRING(o), l);
649 649 p += l;
650 650 len += l + 1;
651 651 }
652 652 putbe32((uint32_t)len, t);
653 653 }
654 654
655 655 pos = p - PyString_AS_STRING(packobj);
656 656 if (pos != nbytes) {
657 657 PyErr_Format(PyExc_SystemError, "bad dirstate size: %ld != %ld",
658 658 (long)pos, (long)nbytes);
659 659 goto bail;
660 660 }
661 661
662 662 return packobj;
663 663 bail:
664 664 Py_XDECREF(mtime_unset);
665 665 Py_XDECREF(packobj);
666 666 Py_XDECREF(v);
667 667 return NULL;
668 668 }
669 669
670 670 /*
671 671 * A base-16 trie for fast node->rev mapping.
672 672 *
673 673 * Positive value is index of the next node in the trie
674 674 * Negative value is a leaf: -(rev + 1)
675 675 * Zero is empty
676 676 */
677 677 typedef struct {
678 678 int children[16];
679 679 } nodetree;
680 680
681 681 /*
682 682 * This class has two behaviours.
683 683 *
684 684 * When used in a list-like way (with integer keys), we decode an
685 685 * entry in a RevlogNG index file on demand. Our last entry is a
686 686 * sentinel, always a nullid. We have limited support for
687 687 * integer-keyed insert and delete, only at elements right before the
688 688 * sentinel.
689 689 *
690 690 * With string keys, we lazily perform a reverse mapping from node to
691 691 * rev, using a base-16 trie.
692 692 */
693 693 typedef struct {
694 694 PyObject_HEAD
695 695 /* Type-specific fields go here. */
696 696 PyObject *data; /* raw bytes of index */
697 697 PyObject **cache; /* cached tuples */
698 698 const char **offsets; /* populated on demand */
699 699 Py_ssize_t raw_length; /* original number of elements */
700 700 Py_ssize_t length; /* current number of elements */
701 701 PyObject *added; /* populated on demand */
702 702 PyObject *headrevs; /* cache, invalidated on changes */
703 703 PyObject *filteredrevs;/* filtered revs set */
704 704 nodetree *nt; /* base-16 trie */
705 705 int ntlength; /* # nodes in use */
706 706 int ntcapacity; /* # nodes allocated */
707 707 int ntdepth; /* maximum depth of tree */
708 708 int ntsplits; /* # splits performed */
709 709 int ntrev; /* last rev scanned */
710 710 int ntlookups; /* # lookups */
711 711 int ntmisses; /* # lookups that miss the cache */
712 712 int inlined;
713 713 } indexObject;
714 714
715 715 static Py_ssize_t index_length(const indexObject *self)
716 716 {
717 717 if (self->added == NULL)
718 718 return self->length;
719 719 return self->length + PyList_GET_SIZE(self->added);
720 720 }
721 721
722 722 static PyObject *nullentry;
723 723 static const char nullid[20];
724 724
725 725 static Py_ssize_t inline_scan(indexObject *self, const char **offsets);
726 726
727 727 #if LONG_MAX == 0x7fffffffL
728 728 static char *tuple_format = "Kiiiiiis#";
729 729 #else
730 730 static char *tuple_format = "kiiiiiis#";
731 731 #endif
732 732
733 733 /* A RevlogNG v1 index entry is 64 bytes long. */
734 734 static const long v1_hdrsize = 64;
735 735
736 736 /*
737 737 * Return a pointer to the beginning of a RevlogNG record.
738 738 */
739 739 static const char *index_deref(indexObject *self, Py_ssize_t pos)
740 740 {
741 741 if (self->inlined && pos > 0) {
742 742 if (self->offsets == NULL) {
743 743 self->offsets = malloc(self->raw_length *
744 744 sizeof(*self->offsets));
745 745 if (self->offsets == NULL)
746 746 return (const char *)PyErr_NoMemory();
747 747 inline_scan(self, self->offsets);
748 748 }
749 749 return self->offsets[pos];
750 750 }
751 751
752 752 return PyString_AS_STRING(self->data) + pos * v1_hdrsize;
753 753 }
754 754
755 755 static inline int index_get_parents(indexObject *self, Py_ssize_t rev,
756 756 int *ps, int maxrev)
757 757 {
758 758 if (rev >= self->length - 1) {
759 759 PyObject *tuple = PyList_GET_ITEM(self->added,
760 760 rev - self->length + 1);
761 761 ps[0] = (int)PyInt_AS_LONG(PyTuple_GET_ITEM(tuple, 5));
762 762 ps[1] = (int)PyInt_AS_LONG(PyTuple_GET_ITEM(tuple, 6));
763 763 } else {
764 764 const char *data = index_deref(self, rev);
765 765 ps[0] = getbe32(data + 24);
766 766 ps[1] = getbe32(data + 28);
767 767 }
768 768 /* If index file is corrupted, ps[] may point to invalid revisions. So
769 769 * there is a risk of buffer overflow to trust them unconditionally. */
770 770 if (ps[0] > maxrev || ps[1] > maxrev) {
771 771 PyErr_SetString(PyExc_ValueError, "parent out of range");
772 772 return -1;
773 773 }
774 774 return 0;
775 775 }
776 776
777 777
778 778 /*
779 779 * RevlogNG format (all in big endian, data may be inlined):
780 780 * 6 bytes: offset
781 781 * 2 bytes: flags
782 782 * 4 bytes: compressed length
783 783 * 4 bytes: uncompressed length
784 784 * 4 bytes: base revision
785 785 * 4 bytes: link revision
786 786 * 4 bytes: parent 1 revision
787 787 * 4 bytes: parent 2 revision
788 788 * 32 bytes: nodeid (only 20 bytes used)
789 789 */
790 790 static PyObject *index_get(indexObject *self, Py_ssize_t pos)
791 791 {
792 792 uint64_t offset_flags;
793 793 int comp_len, uncomp_len, base_rev, link_rev, parent_1, parent_2;
794 794 const char *c_node_id;
795 795 const char *data;
796 796 Py_ssize_t length = index_length(self);
797 797 PyObject *entry;
798 798
799 799 if (pos < 0)
800 800 pos += length;
801 801
802 802 if (pos < 0 || pos >= length) {
803 803 PyErr_SetString(PyExc_IndexError, "revlog index out of range");
804 804 return NULL;
805 805 }
806 806
807 807 if (pos == length - 1) {
808 808 Py_INCREF(nullentry);
809 809 return nullentry;
810 810 }
811 811
812 812 if (pos >= self->length - 1) {
813 813 PyObject *obj;
814 814 obj = PyList_GET_ITEM(self->added, pos - self->length + 1);
815 815 Py_INCREF(obj);
816 816 return obj;
817 817 }
818 818
819 819 if (self->cache) {
820 820 if (self->cache[pos]) {
821 821 Py_INCREF(self->cache[pos]);
822 822 return self->cache[pos];
823 823 }
824 824 } else {
825 825 self->cache = calloc(self->raw_length, sizeof(PyObject *));
826 826 if (self->cache == NULL)
827 827 return PyErr_NoMemory();
828 828 }
829 829
830 830 data = index_deref(self, pos);
831 831 if (data == NULL)
832 832 return NULL;
833 833
834 834 offset_flags = getbe32(data + 4);
835 835 if (pos == 0) /* mask out version number for the first entry */
836 836 offset_flags &= 0xFFFF;
837 837 else {
838 838 uint32_t offset_high = getbe32(data);
839 839 offset_flags |= ((uint64_t)offset_high) << 32;
840 840 }
841 841
842 842 comp_len = getbe32(data + 8);
843 843 uncomp_len = getbe32(data + 12);
844 844 base_rev = getbe32(data + 16);
845 845 link_rev = getbe32(data + 20);
846 846 parent_1 = getbe32(data + 24);
847 847 parent_2 = getbe32(data + 28);
848 848 c_node_id = data + 32;
849 849
850 850 entry = Py_BuildValue(tuple_format, offset_flags, comp_len,
851 851 uncomp_len, base_rev, link_rev,
852 852 parent_1, parent_2, c_node_id, 20);
853 853
854 854 if (entry) {
855 855 PyObject_GC_UnTrack(entry);
856 856 Py_INCREF(entry);
857 857 }
858 858
859 859 self->cache[pos] = entry;
860 860
861 861 return entry;
862 862 }
863 863
864 864 /*
865 865 * Return the 20-byte SHA of the node corresponding to the given rev.
866 866 */
867 867 static const char *index_node(indexObject *self, Py_ssize_t pos)
868 868 {
869 869 Py_ssize_t length = index_length(self);
870 870 const char *data;
871 871
872 872 if (pos == length - 1 || pos == INT_MAX)
873 873 return nullid;
874 874
875 875 if (pos >= length)
876 876 return NULL;
877 877
878 878 if (pos >= self->length - 1) {
879 879 PyObject *tuple, *str;
880 880 tuple = PyList_GET_ITEM(self->added, pos - self->length + 1);
881 881 str = PyTuple_GetItem(tuple, 7);
882 882 return str ? PyString_AS_STRING(str) : NULL;
883 883 }
884 884
885 885 data = index_deref(self, pos);
886 886 return data ? data + 32 : NULL;
887 887 }
888 888
889 889 static int nt_insert(indexObject *self, const char *node, int rev);
890 890
891 891 static int node_check(PyObject *obj, char **node, Py_ssize_t *nodelen)
892 892 {
893 893 if (PyString_AsStringAndSize(obj, node, nodelen) == -1)
894 894 return -1;
895 895 if (*nodelen == 20)
896 896 return 0;
897 897 PyErr_SetString(PyExc_ValueError, "20-byte hash required");
898 898 return -1;
899 899 }
900 900
901 901 static PyObject *index_insert(indexObject *self, PyObject *args)
902 902 {
903 903 PyObject *obj;
904 904 char *node;
905 905 int index;
906 906 Py_ssize_t len, nodelen;
907 907
908 908 if (!PyArg_ParseTuple(args, "iO", &index, &obj))
909 909 return NULL;
910 910
911 911 if (!PyTuple_Check(obj) || PyTuple_GET_SIZE(obj) != 8) {
912 912 PyErr_SetString(PyExc_TypeError, "8-tuple required");
913 913 return NULL;
914 914 }
915 915
916 916 if (node_check(PyTuple_GET_ITEM(obj, 7), &node, &nodelen) == -1)
917 917 return NULL;
918 918
919 919 len = index_length(self);
920 920
921 921 if (index < 0)
922 922 index += len;
923 923
924 924 if (index != len - 1) {
925 925 PyErr_SetString(PyExc_IndexError,
926 926 "insert only supported at index -1");
927 927 return NULL;
928 928 }
929 929
930 930 if (self->added == NULL) {
931 931 self->added = PyList_New(0);
932 932 if (self->added == NULL)
933 933 return NULL;
934 934 }
935 935
936 936 if (PyList_Append(self->added, obj) == -1)
937 937 return NULL;
938 938
939 939 if (self->nt)
940 940 nt_insert(self, node, index);
941 941
942 942 Py_CLEAR(self->headrevs);
943 943 Py_RETURN_NONE;
944 944 }
945 945
946 946 static void _index_clearcaches(indexObject *self)
947 947 {
948 948 if (self->cache) {
949 949 Py_ssize_t i;
950 950
951 951 for (i = 0; i < self->raw_length; i++)
952 952 Py_CLEAR(self->cache[i]);
953 953 free(self->cache);
954 954 self->cache = NULL;
955 955 }
956 956 if (self->offsets) {
957 957 free(self->offsets);
958 958 self->offsets = NULL;
959 959 }
960 960 if (self->nt) {
961 961 free(self->nt);
962 962 self->nt = NULL;
963 963 }
964 964 Py_CLEAR(self->headrevs);
965 965 }
966 966
967 967 static PyObject *index_clearcaches(indexObject *self)
968 968 {
969 969 _index_clearcaches(self);
970 970 self->ntlength = self->ntcapacity = 0;
971 971 self->ntdepth = self->ntsplits = 0;
972 972 self->ntrev = -1;
973 973 self->ntlookups = self->ntmisses = 0;
974 974 Py_RETURN_NONE;
975 975 }
976 976
977 977 static PyObject *index_stats(indexObject *self)
978 978 {
979 979 PyObject *obj = PyDict_New();
980 980 PyObject *t = NULL;
981 981
982 982 if (obj == NULL)
983 983 return NULL;
984 984
985 985 #define istat(__n, __d) \
986 986 t = PyInt_FromSsize_t(self->__n); \
987 987 if (!t) \
988 988 goto bail; \
989 989 if (PyDict_SetItemString(obj, __d, t) == -1) \
990 990 goto bail; \
991 991 Py_DECREF(t);
992 992
993 993 if (self->added) {
994 994 Py_ssize_t len = PyList_GET_SIZE(self->added);
995 995 t = PyInt_FromSsize_t(len);
996 996 if (!t)
997 997 goto bail;
998 998 if (PyDict_SetItemString(obj, "index entries added", t) == -1)
999 999 goto bail;
1000 1000 Py_DECREF(t);
1001 1001 }
1002 1002
1003 1003 if (self->raw_length != self->length - 1)
1004 1004 istat(raw_length, "revs on disk");
1005 1005 istat(length, "revs in memory");
1006 1006 istat(ntcapacity, "node trie capacity");
1007 1007 istat(ntdepth, "node trie depth");
1008 1008 istat(ntlength, "node trie count");
1009 1009 istat(ntlookups, "node trie lookups");
1010 1010 istat(ntmisses, "node trie misses");
1011 1011 istat(ntrev, "node trie last rev scanned");
1012 1012 istat(ntsplits, "node trie splits");
1013 1013
1014 1014 #undef istat
1015 1015
1016 1016 return obj;
1017 1017
1018 1018 bail:
1019 1019 Py_XDECREF(obj);
1020 1020 Py_XDECREF(t);
1021 1021 return NULL;
1022 1022 }
1023 1023
1024 1024 /*
1025 1025 * When we cache a list, we want to be sure the caller can't mutate
1026 1026 * the cached copy.
1027 1027 */
1028 1028 static PyObject *list_copy(PyObject *list)
1029 1029 {
1030 1030 Py_ssize_t len = PyList_GET_SIZE(list);
1031 1031 PyObject *newlist = PyList_New(len);
1032 1032 Py_ssize_t i;
1033 1033
1034 1034 if (newlist == NULL)
1035 1035 return NULL;
1036 1036
1037 1037 for (i = 0; i < len; i++) {
1038 1038 PyObject *obj = PyList_GET_ITEM(list, i);
1039 1039 Py_INCREF(obj);
1040 1040 PyList_SET_ITEM(newlist, i, obj);
1041 1041 }
1042 1042
1043 1043 return newlist;
1044 1044 }
1045 1045
1046 1046 /* arg should be Py_ssize_t but Python 2.4 do not support the n format */
1047 1047 static int check_filter(PyObject *filter, unsigned long arg) {
1048 1048 if (filter) {
1049 1049 PyObject *arglist, *result;
1050 1050 int isfiltered;
1051 1051
1052 1052 arglist = Py_BuildValue("(k)", arg);
1053 1053 if (!arglist) {
1054 1054 return -1;
1055 1055 }
1056 1056
1057 1057 result = PyEval_CallObject(filter, arglist);
1058 1058 Py_DECREF(arglist);
1059 1059 if (!result) {
1060 1060 return -1;
1061 1061 }
1062 1062
1063 1063 /* PyObject_IsTrue returns 1 if true, 0 if false, -1 if error,
1064 1064 * same as this function, so we can just return it directly.*/
1065 1065 isfiltered = PyObject_IsTrue(result);
1066 1066 Py_DECREF(result);
1067 1067 return isfiltered;
1068 1068 } else {
1069 1069 return 0;
1070 1070 }
1071 1071 }
1072 1072
1073 1073 static Py_ssize_t add_roots_get_min(indexObject *self, PyObject *list,
1074 1074 Py_ssize_t marker, char *phases)
1075 1075 {
1076 1076 PyObject *iter = NULL;
1077 1077 PyObject *iter_item = NULL;
1078 1078 Py_ssize_t min_idx = index_length(self) + 1;
1079 1079 long iter_item_long;
1080 1080
1081 1081 if (PyList_GET_SIZE(list) != 0) {
1082 1082 iter = PyObject_GetIter(list);
1083 1083 if (iter == NULL)
1084 1084 return -2;
1085 1085 while ((iter_item = PyIter_Next(iter)))
1086 1086 {
1087 1087 iter_item_long = PyInt_AS_LONG(iter_item);
1088 1088 Py_DECREF(iter_item);
1089 1089 if (iter_item_long < min_idx)
1090 1090 min_idx = iter_item_long;
1091 1091 phases[iter_item_long] = marker;
1092 1092 }
1093 1093 Py_DECREF(iter);
1094 1094 }
1095 1095
1096 1096 return min_idx;
1097 1097 }
1098 1098
1099 1099 static inline void set_phase_from_parents(char *phases, int parent_1,
1100 1100 int parent_2, Py_ssize_t i)
1101 1101 {
1102 1102 if (parent_1 >= 0 && phases[parent_1] > phases[i])
1103 1103 phases[i] = phases[parent_1];
1104 1104 if (parent_2 >= 0 && phases[parent_2] > phases[i])
1105 1105 phases[i] = phases[parent_2];
1106 1106 }
1107 1107
1108 1108 static PyObject *reachableroots(indexObject *self, PyObject *args)
1109 1109 {
1110 1110
1111 1111 /* Input */
1112 1112 long minroot;
1113 1113 PyObject *includepatharg = NULL;
1114 1114 int includepath = 0;
1115 /* heads is a list */
1115 1116 PyObject *heads = NULL;
1116 1117 Py_ssize_t numheads;
1118 /* roots is a set */
1117 1119 PyObject *roots = NULL;
1118 1120 PyObject *reachable = NULL;
1119 1121
1120 1122 PyObject *val;
1121 1123 Py_ssize_t len = index_length(self) - 1;
1122 1124 long revnum;
1123 1125 Py_ssize_t k;
1124 1126 Py_ssize_t i;
1125 1127 int r;
1126 1128 int minidx;
1127 1129 int parents[2];
1128 1130
1129 1131 /* Internal data structure:
1130 1132 * tovisit: array of length len+1 (all revs + nullrev), filled upto lentovisit
1131 1133 * seen: array of length len+1 (all revs + nullrev) 0: not seen, 1 seen*/
1132 1134 int *tovisit = NULL;
1133 1135 long lentovisit = 0;
1134 1136 char *seen = NULL;
1135 1137
1136 1138 /* Get arguments */
1137 if (!PyArg_ParseTuple(args, "lO!O!O!", &minroot, &PySet_Type, &heads,
1138 &PyList_Type, &roots, &PyBool_Type, &includepatharg))
1139 if (!PyArg_ParseTuple(args, "lO!O!O!", &minroot, &PyList_Type, &heads,
1140 &PySet_Type, &roots, &PyBool_Type, &includepatharg))
1139 1141 goto bail;
1140 1142
1141 1143 if (includepatharg == Py_True)
1142 1144 includepath = 1;
1143 1145
1144 1146 /* Initialize return set */
1145 1147 reachable = PySet_New(0);
1146 1148 if (reachable == NULL) {
1147 1149 PyErr_NoMemory();
1148 1150 goto bail;
1149 1151 }
1150 1152
1151 1153 /* Initialize internal datastructures */
1152 1154 tovisit = (int *)malloc((len + 1) * sizeof(int));
1153 1155 if (tovisit == NULL) {
1154 1156 PyErr_NoMemory();
1155 1157 goto release_reachable;
1156 1158 }
1157 1159
1158 1160 seen = (char *)calloc(len+1, 1);
1159 1161 if (seen == NULL) {
1160 1162 PyErr_NoMemory();
1161 1163 goto release_seen_and_tovisit;
1162 1164 }
1163 1165
1164 1166 /* Populate tovisit with all the heads */
1165 1167 numheads = PyList_GET_SIZE(heads);
1166 1168 for (i = 0; i < numheads; i++) {
1167 1169 revnum = PyInt_AS_LONG(PyList_GET_ITEM(heads, i));
1168 1170 if (seen[revnum+1] == 0) {
1169 1171 tovisit[lentovisit++] = revnum;
1170 1172 seen[revnum+1]=1;
1171 1173 }
1172 1174 }
1173 1175
1174 1176 /* Visit the tovisit list and find the reachable roots */
1175 1177 k = 0;
1176 1178 while (k < lentovisit) {
1177 1179 /* Add the node to reachable if it is a root*/
1178 1180 revnum = tovisit[k++];
1179 1181 val = PyInt_FromLong(revnum);
1180 1182 if (PySet_Contains(roots, val) == 1) {
1181 1183 PySet_Add(reachable, val);
1182 1184 if (includepath == 0) {
1183 1185 Py_XDECREF(val);
1184 1186 continue;
1185 1187 }
1186 1188 }
1187 1189 Py_XDECREF(val);
1188 1190
1189 1191 /* Add its parents to the list of nodes to visit */
1190 1192 if (revnum != -1) {
1191 1193 r = index_get_parents(self, revnum, parents, (int)len - 1);
1192 1194 if (r < 0)
1193 1195 goto release_seen_and_tovisit;
1194 1196
1195 1197 for (i = 0; i < 2; i++) {
1196 1198 if (seen[parents[i] + 1] == 0 && parents[i] >= minroot) {
1197 1199 tovisit[lentovisit++] = parents[i];
1198 1200 seen[parents[i] + 1] = 1;
1199 1201 }
1200 1202 }
1201 1203 }
1202 1204 }
1203 1205
1204 1206 /* Find all the nodes in between the roots we found and the heads
1205 1207 * and add them to the reachable set */
1206 1208 if (includepath == 1) {
1207 1209 minidx = minroot;
1208 1210 if (minidx < 0)
1209 1211 minidx = 0;
1210 1212 for (i = minidx; i < len; i++) {
1211 1213 if (seen[i + 1] == 1) {
1212 1214 r = index_get_parents(self, i, parents, (int)len - 1);
1213 1215 /* Corrupted index file, error is set from index_get_parents */
1214 1216 if (r < 0)
1215 1217 goto release_seen_and_tovisit;
1216 1218 for (k = 0; k < 2; k++) {
1217 1219 PyObject *p = PyInt_FromLong(parents[k]);
1218 1220 if (PySet_Contains(reachable, p) == 1)
1219 1221 PySet_Add(reachable, PyInt_FromLong(i));
1220 1222 Py_XDECREF(p);
1221 1223 }
1222 1224 }
1223 1225 }
1224 1226 }
1225 1227
1226 1228 release_seen_and_tovisit:
1227 1229 free(seen);
1228 1230 free(tovisit);
1229 1231 return reachable;
1230 1232 release_reachable:
1231 1233 Py_XDECREF(reachable);
1232 1234 bail:
1233 1235 val = Py_None;
1234 1236 Py_INCREF(Py_None);
1235 1237 return val;
1236 1238 }
1237 1239
1238 1240 static PyObject *compute_phases_map_sets(indexObject *self, PyObject *args)
1239 1241 {
1240 1242 PyObject *roots = Py_None;
1241 1243 PyObject *ret = NULL;
1242 1244 PyObject *phaseslist = NULL;
1243 1245 PyObject *phaseroots = NULL;
1244 1246 PyObject *phaseset = NULL;
1245 1247 PyObject *phasessetlist = NULL;
1246 1248 PyObject *rev = NULL;
1247 1249 Py_ssize_t len = index_length(self) - 1;
1248 1250 Py_ssize_t numphase = 0;
1249 1251 Py_ssize_t minrevallphases = 0;
1250 1252 Py_ssize_t minrevphase = 0;
1251 1253 Py_ssize_t i = 0;
1252 1254 char *phases = NULL;
1253 1255 long phase;
1254 1256
1255 1257 if (!PyArg_ParseTuple(args, "O", &roots))
1256 1258 goto release_none;
1257 1259 if (roots == NULL || !PyList_Check(roots))
1258 1260 goto release_none;
1259 1261
1260 1262 phases = calloc(len, 1); /* phase per rev: {0: public, 1: draft, 2: secret} */
1261 1263 if (phases == NULL)
1262 1264 goto release_none;
1263 1265 /* Put the phase information of all the roots in phases */
1264 1266 numphase = PyList_GET_SIZE(roots)+1;
1265 1267 minrevallphases = len + 1;
1266 1268 phasessetlist = PyList_New(numphase);
1267 1269 if (phasessetlist == NULL)
1268 1270 goto release_none;
1269 1271
1270 1272 PyList_SET_ITEM(phasessetlist, 0, Py_None);
1271 1273 Py_INCREF(Py_None);
1272 1274
1273 1275 for (i = 0; i < numphase-1; i++) {
1274 1276 phaseroots = PyList_GET_ITEM(roots, i);
1275 1277 phaseset = PySet_New(NULL);
1276 1278 if (phaseset == NULL)
1277 1279 goto release_phasesetlist;
1278 1280 PyList_SET_ITEM(phasessetlist, i+1, phaseset);
1279 1281 if (!PyList_Check(phaseroots))
1280 1282 goto release_phasesetlist;
1281 1283 minrevphase = add_roots_get_min(self, phaseroots, i+1, phases);
1282 1284 if (minrevphase == -2) /* Error from add_roots_get_min */
1283 1285 goto release_phasesetlist;
1284 1286 minrevallphases = MIN(minrevallphases, minrevphase);
1285 1287 }
1286 1288 /* Propagate the phase information from the roots to the revs */
1287 1289 if (minrevallphases != -1) {
1288 1290 int parents[2];
1289 1291 for (i = minrevallphases; i < len; i++) {
1290 1292 if (index_get_parents(self, i, parents,
1291 1293 (int)len - 1) < 0)
1292 1294 goto release_phasesetlist;
1293 1295 set_phase_from_parents(phases, parents[0], parents[1], i);
1294 1296 }
1295 1297 }
1296 1298 /* Transform phase list to a python list */
1297 1299 phaseslist = PyList_New(len);
1298 1300 if (phaseslist == NULL)
1299 1301 goto release_phasesetlist;
1300 1302 for (i = 0; i < len; i++) {
1301 1303 phase = phases[i];
1302 1304 /* We only store the sets of phase for non public phase, the public phase
1303 1305 * is computed as a difference */
1304 1306 if (phase != 0) {
1305 1307 phaseset = PyList_GET_ITEM(phasessetlist, phase);
1306 1308 rev = PyInt_FromLong(i);
1307 1309 PySet_Add(phaseset, rev);
1308 1310 Py_XDECREF(rev);
1309 1311 }
1310 1312 PyList_SET_ITEM(phaseslist, i, PyInt_FromLong(phase));
1311 1313 }
1312 1314 ret = PyList_New(2);
1313 1315 if (ret == NULL)
1314 1316 goto release_phaseslist;
1315 1317
1316 1318 PyList_SET_ITEM(ret, 0, phaseslist);
1317 1319 PyList_SET_ITEM(ret, 1, phasessetlist);
1318 1320 /* We don't release phaseslist and phasessetlist as we return them to
1319 1321 * python */
1320 1322 goto release_phases;
1321 1323
1322 1324 release_phaseslist:
1323 1325 Py_XDECREF(phaseslist);
1324 1326 release_phasesetlist:
1325 1327 Py_XDECREF(phasessetlist);
1326 1328 release_phases:
1327 1329 free(phases);
1328 1330 release_none:
1329 1331 return ret;
1330 1332 }
1331 1333
1332 1334 static PyObject *index_headrevs(indexObject *self, PyObject *args)
1333 1335 {
1334 1336 Py_ssize_t i, j, len;
1335 1337 char *nothead = NULL;
1336 1338 PyObject *heads = NULL;
1337 1339 PyObject *filter = NULL;
1338 1340 PyObject *filteredrevs = Py_None;
1339 1341
1340 1342 if (!PyArg_ParseTuple(args, "|O", &filteredrevs)) {
1341 1343 return NULL;
1342 1344 }
1343 1345
1344 1346 if (self->headrevs && filteredrevs == self->filteredrevs)
1345 1347 return list_copy(self->headrevs);
1346 1348
1347 1349 Py_DECREF(self->filteredrevs);
1348 1350 self->filteredrevs = filteredrevs;
1349 1351 Py_INCREF(filteredrevs);
1350 1352
1351 1353 if (filteredrevs != Py_None) {
1352 1354 filter = PyObject_GetAttrString(filteredrevs, "__contains__");
1353 1355 if (!filter) {
1354 1356 PyErr_SetString(PyExc_TypeError,
1355 1357 "filteredrevs has no attribute __contains__");
1356 1358 goto bail;
1357 1359 }
1358 1360 }
1359 1361
1360 1362 len = index_length(self) - 1;
1361 1363 heads = PyList_New(0);
1362 1364 if (heads == NULL)
1363 1365 goto bail;
1364 1366 if (len == 0) {
1365 1367 PyObject *nullid = PyInt_FromLong(-1);
1366 1368 if (nullid == NULL || PyList_Append(heads, nullid) == -1) {
1367 1369 Py_XDECREF(nullid);
1368 1370 goto bail;
1369 1371 }
1370 1372 goto done;
1371 1373 }
1372 1374
1373 1375 nothead = calloc(len, 1);
1374 1376 if (nothead == NULL)
1375 1377 goto bail;
1376 1378
1377 1379 for (i = 0; i < len; i++) {
1378 1380 int isfiltered;
1379 1381 int parents[2];
1380 1382
1381 1383 isfiltered = check_filter(filter, i);
1382 1384 if (isfiltered == -1) {
1383 1385 PyErr_SetString(PyExc_TypeError,
1384 1386 "unable to check filter");
1385 1387 goto bail;
1386 1388 }
1387 1389
1388 1390 if (isfiltered) {
1389 1391 nothead[i] = 1;
1390 1392 continue;
1391 1393 }
1392 1394
1393 1395 if (index_get_parents(self, i, parents, (int)len - 1) < 0)
1394 1396 goto bail;
1395 1397 for (j = 0; j < 2; j++) {
1396 1398 if (parents[j] >= 0)
1397 1399 nothead[parents[j]] = 1;
1398 1400 }
1399 1401 }
1400 1402
1401 1403 for (i = 0; i < len; i++) {
1402 1404 PyObject *head;
1403 1405
1404 1406 if (nothead[i])
1405 1407 continue;
1406 1408 head = PyInt_FromSsize_t(i);
1407 1409 if (head == NULL || PyList_Append(heads, head) == -1) {
1408 1410 Py_XDECREF(head);
1409 1411 goto bail;
1410 1412 }
1411 1413 }
1412 1414
1413 1415 done:
1414 1416 self->headrevs = heads;
1415 1417 Py_XDECREF(filter);
1416 1418 free(nothead);
1417 1419 return list_copy(self->headrevs);
1418 1420 bail:
1419 1421 Py_XDECREF(filter);
1420 1422 Py_XDECREF(heads);
1421 1423 free(nothead);
1422 1424 return NULL;
1423 1425 }
1424 1426
1425 1427 static inline int nt_level(const char *node, Py_ssize_t level)
1426 1428 {
1427 1429 int v = node[level>>1];
1428 1430 if (!(level & 1))
1429 1431 v >>= 4;
1430 1432 return v & 0xf;
1431 1433 }
1432 1434
1433 1435 /*
1434 1436 * Return values:
1435 1437 *
1436 1438 * -4: match is ambiguous (multiple candidates)
1437 1439 * -2: not found
1438 1440 * rest: valid rev
1439 1441 */
1440 1442 static int nt_find(indexObject *self, const char *node, Py_ssize_t nodelen,
1441 1443 int hex)
1442 1444 {
1443 1445 int (*getnybble)(const char *, Py_ssize_t) = hex ? hexdigit : nt_level;
1444 1446 int level, maxlevel, off;
1445 1447
1446 1448 if (nodelen == 20 && node[0] == '\0' && memcmp(node, nullid, 20) == 0)
1447 1449 return -1;
1448 1450
1449 1451 if (self->nt == NULL)
1450 1452 return -2;
1451 1453
1452 1454 if (hex)
1453 1455 maxlevel = nodelen > 40 ? 40 : (int)nodelen;
1454 1456 else
1455 1457 maxlevel = nodelen > 20 ? 40 : ((int)nodelen * 2);
1456 1458
1457 1459 for (level = off = 0; level < maxlevel; level++) {
1458 1460 int k = getnybble(node, level);
1459 1461 nodetree *n = &self->nt[off];
1460 1462 int v = n->children[k];
1461 1463
1462 1464 if (v < 0) {
1463 1465 const char *n;
1464 1466 Py_ssize_t i;
1465 1467
1466 1468 v = -(v + 1);
1467 1469 n = index_node(self, v);
1468 1470 if (n == NULL)
1469 1471 return -2;
1470 1472 for (i = level; i < maxlevel; i++)
1471 1473 if (getnybble(node, i) != nt_level(n, i))
1472 1474 return -2;
1473 1475 return v;
1474 1476 }
1475 1477 if (v == 0)
1476 1478 return -2;
1477 1479 off = v;
1478 1480 }
1479 1481 /* multiple matches against an ambiguous prefix */
1480 1482 return -4;
1481 1483 }
1482 1484
1483 1485 static int nt_new(indexObject *self)
1484 1486 {
1485 1487 if (self->ntlength == self->ntcapacity) {
1486 1488 if (self->ntcapacity >= INT_MAX / (sizeof(nodetree) * 2)) {
1487 1489 PyErr_SetString(PyExc_MemoryError,
1488 1490 "overflow in nt_new");
1489 1491 return -1;
1490 1492 }
1491 1493 self->ntcapacity *= 2;
1492 1494 self->nt = realloc(self->nt,
1493 1495 self->ntcapacity * sizeof(nodetree));
1494 1496 if (self->nt == NULL) {
1495 1497 PyErr_SetString(PyExc_MemoryError, "out of memory");
1496 1498 return -1;
1497 1499 }
1498 1500 memset(&self->nt[self->ntlength], 0,
1499 1501 sizeof(nodetree) * (self->ntcapacity - self->ntlength));
1500 1502 }
1501 1503 return self->ntlength++;
1502 1504 }
1503 1505
1504 1506 static int nt_insert(indexObject *self, const char *node, int rev)
1505 1507 {
1506 1508 int level = 0;
1507 1509 int off = 0;
1508 1510
1509 1511 while (level < 40) {
1510 1512 int k = nt_level(node, level);
1511 1513 nodetree *n;
1512 1514 int v;
1513 1515
1514 1516 n = &self->nt[off];
1515 1517 v = n->children[k];
1516 1518
1517 1519 if (v == 0) {
1518 1520 n->children[k] = -rev - 1;
1519 1521 return 0;
1520 1522 }
1521 1523 if (v < 0) {
1522 1524 const char *oldnode = index_node(self, -(v + 1));
1523 1525 int noff;
1524 1526
1525 1527 if (!oldnode || !memcmp(oldnode, node, 20)) {
1526 1528 n->children[k] = -rev - 1;
1527 1529 return 0;
1528 1530 }
1529 1531 noff = nt_new(self);
1530 1532 if (noff == -1)
1531 1533 return -1;
1532 1534 /* self->nt may have been changed by realloc */
1533 1535 self->nt[off].children[k] = noff;
1534 1536 off = noff;
1535 1537 n = &self->nt[off];
1536 1538 n->children[nt_level(oldnode, ++level)] = v;
1537 1539 if (level > self->ntdepth)
1538 1540 self->ntdepth = level;
1539 1541 self->ntsplits += 1;
1540 1542 } else {
1541 1543 level += 1;
1542 1544 off = v;
1543 1545 }
1544 1546 }
1545 1547
1546 1548 return -1;
1547 1549 }
1548 1550
1549 1551 static int nt_init(indexObject *self)
1550 1552 {
1551 1553 if (self->nt == NULL) {
1552 1554 if (self->raw_length > INT_MAX / sizeof(nodetree)) {
1553 1555 PyErr_SetString(PyExc_ValueError, "overflow in nt_init");
1554 1556 return -1;
1555 1557 }
1556 1558 self->ntcapacity = self->raw_length < 4
1557 1559 ? 4 : (int)self->raw_length / 2;
1558 1560
1559 1561 self->nt = calloc(self->ntcapacity, sizeof(nodetree));
1560 1562 if (self->nt == NULL) {
1561 1563 PyErr_NoMemory();
1562 1564 return -1;
1563 1565 }
1564 1566 self->ntlength = 1;
1565 1567 self->ntrev = (int)index_length(self) - 1;
1566 1568 self->ntlookups = 1;
1567 1569 self->ntmisses = 0;
1568 1570 if (nt_insert(self, nullid, INT_MAX) == -1)
1569 1571 return -1;
1570 1572 }
1571 1573 return 0;
1572 1574 }
1573 1575
1574 1576 /*
1575 1577 * Return values:
1576 1578 *
1577 1579 * -3: error (exception set)
1578 1580 * -2: not found (no exception set)
1579 1581 * rest: valid rev
1580 1582 */
1581 1583 static int index_find_node(indexObject *self,
1582 1584 const char *node, Py_ssize_t nodelen)
1583 1585 {
1584 1586 int rev;
1585 1587
1586 1588 self->ntlookups++;
1587 1589 rev = nt_find(self, node, nodelen, 0);
1588 1590 if (rev >= -1)
1589 1591 return rev;
1590 1592
1591 1593 if (nt_init(self) == -1)
1592 1594 return -3;
1593 1595
1594 1596 /*
1595 1597 * For the first handful of lookups, we scan the entire index,
1596 1598 * and cache only the matching nodes. This optimizes for cases
1597 1599 * like "hg tip", where only a few nodes are accessed.
1598 1600 *
1599 1601 * After that, we cache every node we visit, using a single
1600 1602 * scan amortized over multiple lookups. This gives the best
1601 1603 * bulk performance, e.g. for "hg log".
1602 1604 */
1603 1605 if (self->ntmisses++ < 4) {
1604 1606 for (rev = self->ntrev - 1; rev >= 0; rev--) {
1605 1607 const char *n = index_node(self, rev);
1606 1608 if (n == NULL)
1607 1609 return -2;
1608 1610 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
1609 1611 if (nt_insert(self, n, rev) == -1)
1610 1612 return -3;
1611 1613 break;
1612 1614 }
1613 1615 }
1614 1616 } else {
1615 1617 for (rev = self->ntrev - 1; rev >= 0; rev--) {
1616 1618 const char *n = index_node(self, rev);
1617 1619 if (n == NULL) {
1618 1620 self->ntrev = rev + 1;
1619 1621 return -2;
1620 1622 }
1621 1623 if (nt_insert(self, n, rev) == -1) {
1622 1624 self->ntrev = rev + 1;
1623 1625 return -3;
1624 1626 }
1625 1627 if (memcmp(node, n, nodelen > 20 ? 20 : nodelen) == 0) {
1626 1628 break;
1627 1629 }
1628 1630 }
1629 1631 self->ntrev = rev;
1630 1632 }
1631 1633
1632 1634 if (rev >= 0)
1633 1635 return rev;
1634 1636 return -2;
1635 1637 }
1636 1638
1637 1639 static void raise_revlog_error(void)
1638 1640 {
1639 1641 PyObject *mod = NULL, *dict = NULL, *errclass = NULL;
1640 1642
1641 1643 mod = PyImport_ImportModule("mercurial.error");
1642 1644 if (mod == NULL) {
1643 1645 goto cleanup;
1644 1646 }
1645 1647
1646 1648 dict = PyModule_GetDict(mod);
1647 1649 if (dict == NULL) {
1648 1650 goto cleanup;
1649 1651 }
1650 1652 Py_INCREF(dict);
1651 1653
1652 1654 errclass = PyDict_GetItemString(dict, "RevlogError");
1653 1655 if (errclass == NULL) {
1654 1656 PyErr_SetString(PyExc_SystemError,
1655 1657 "could not find RevlogError");
1656 1658 goto cleanup;
1657 1659 }
1658 1660
1659 1661 /* value of exception is ignored by callers */
1660 1662 PyErr_SetString(errclass, "RevlogError");
1661 1663
1662 1664 cleanup:
1663 1665 Py_XDECREF(dict);
1664 1666 Py_XDECREF(mod);
1665 1667 }
1666 1668
1667 1669 static PyObject *index_getitem(indexObject *self, PyObject *value)
1668 1670 {
1669 1671 char *node;
1670 1672 Py_ssize_t nodelen;
1671 1673 int rev;
1672 1674
1673 1675 if (PyInt_Check(value))
1674 1676 return index_get(self, PyInt_AS_LONG(value));
1675 1677
1676 1678 if (node_check(value, &node, &nodelen) == -1)
1677 1679 return NULL;
1678 1680 rev = index_find_node(self, node, nodelen);
1679 1681 if (rev >= -1)
1680 1682 return PyInt_FromLong(rev);
1681 1683 if (rev == -2)
1682 1684 raise_revlog_error();
1683 1685 return NULL;
1684 1686 }
1685 1687
1686 1688 static int nt_partialmatch(indexObject *self, const char *node,
1687 1689 Py_ssize_t nodelen)
1688 1690 {
1689 1691 int rev;
1690 1692
1691 1693 if (nt_init(self) == -1)
1692 1694 return -3;
1693 1695
1694 1696 if (self->ntrev > 0) {
1695 1697 /* ensure that the radix tree is fully populated */
1696 1698 for (rev = self->ntrev - 1; rev >= 0; rev--) {
1697 1699 const char *n = index_node(self, rev);
1698 1700 if (n == NULL)
1699 1701 return -2;
1700 1702 if (nt_insert(self, n, rev) == -1)
1701 1703 return -3;
1702 1704 }
1703 1705 self->ntrev = rev;
1704 1706 }
1705 1707
1706 1708 return nt_find(self, node, nodelen, 1);
1707 1709 }
1708 1710
1709 1711 static PyObject *index_partialmatch(indexObject *self, PyObject *args)
1710 1712 {
1711 1713 const char *fullnode;
1712 1714 int nodelen;
1713 1715 char *node;
1714 1716 int rev, i;
1715 1717
1716 1718 if (!PyArg_ParseTuple(args, "s#", &node, &nodelen))
1717 1719 return NULL;
1718 1720
1719 1721 if (nodelen < 4) {
1720 1722 PyErr_SetString(PyExc_ValueError, "key too short");
1721 1723 return NULL;
1722 1724 }
1723 1725
1724 1726 if (nodelen > 40) {
1725 1727 PyErr_SetString(PyExc_ValueError, "key too long");
1726 1728 return NULL;
1727 1729 }
1728 1730
1729 1731 for (i = 0; i < nodelen; i++)
1730 1732 hexdigit(node, i);
1731 1733 if (PyErr_Occurred()) {
1732 1734 /* input contains non-hex characters */
1733 1735 PyErr_Clear();
1734 1736 Py_RETURN_NONE;
1735 1737 }
1736 1738
1737 1739 rev = nt_partialmatch(self, node, nodelen);
1738 1740
1739 1741 switch (rev) {
1740 1742 case -4:
1741 1743 raise_revlog_error();
1742 1744 case -3:
1743 1745 return NULL;
1744 1746 case -2:
1745 1747 Py_RETURN_NONE;
1746 1748 case -1:
1747 1749 return PyString_FromStringAndSize(nullid, 20);
1748 1750 }
1749 1751
1750 1752 fullnode = index_node(self, rev);
1751 1753 if (fullnode == NULL) {
1752 1754 PyErr_Format(PyExc_IndexError,
1753 1755 "could not access rev %d", rev);
1754 1756 return NULL;
1755 1757 }
1756 1758 return PyString_FromStringAndSize(fullnode, 20);
1757 1759 }
1758 1760
1759 1761 static PyObject *index_m_get(indexObject *self, PyObject *args)
1760 1762 {
1761 1763 Py_ssize_t nodelen;
1762 1764 PyObject *val;
1763 1765 char *node;
1764 1766 int rev;
1765 1767
1766 1768 if (!PyArg_ParseTuple(args, "O", &val))
1767 1769 return NULL;
1768 1770 if (node_check(val, &node, &nodelen) == -1)
1769 1771 return NULL;
1770 1772 rev = index_find_node(self, node, nodelen);
1771 1773 if (rev == -3)
1772 1774 return NULL;
1773 1775 if (rev == -2)
1774 1776 Py_RETURN_NONE;
1775 1777 return PyInt_FromLong(rev);
1776 1778 }
1777 1779
1778 1780 static int index_contains(indexObject *self, PyObject *value)
1779 1781 {
1780 1782 char *node;
1781 1783 Py_ssize_t nodelen;
1782 1784
1783 1785 if (PyInt_Check(value)) {
1784 1786 long rev = PyInt_AS_LONG(value);
1785 1787 return rev >= -1 && rev < index_length(self);
1786 1788 }
1787 1789
1788 1790 if (node_check(value, &node, &nodelen) == -1)
1789 1791 return -1;
1790 1792
1791 1793 switch (index_find_node(self, node, nodelen)) {
1792 1794 case -3:
1793 1795 return -1;
1794 1796 case -2:
1795 1797 return 0;
1796 1798 default:
1797 1799 return 1;
1798 1800 }
1799 1801 }
1800 1802
1801 1803 typedef uint64_t bitmask;
1802 1804
1803 1805 /*
1804 1806 * Given a disjoint set of revs, return all candidates for the
1805 1807 * greatest common ancestor. In revset notation, this is the set
1806 1808 * "heads(::a and ::b and ...)"
1807 1809 */
1808 1810 static PyObject *find_gca_candidates(indexObject *self, const int *revs,
1809 1811 int revcount)
1810 1812 {
1811 1813 const bitmask allseen = (1ull << revcount) - 1;
1812 1814 const bitmask poison = 1ull << revcount;
1813 1815 PyObject *gca = PyList_New(0);
1814 1816 int i, v, interesting;
1815 1817 int maxrev = -1;
1816 1818 bitmask sp;
1817 1819 bitmask *seen;
1818 1820
1819 1821 if (gca == NULL)
1820 1822 return PyErr_NoMemory();
1821 1823
1822 1824 for (i = 0; i < revcount; i++) {
1823 1825 if (revs[i] > maxrev)
1824 1826 maxrev = revs[i];
1825 1827 }
1826 1828
1827 1829 seen = calloc(sizeof(*seen), maxrev + 1);
1828 1830 if (seen == NULL) {
1829 1831 Py_DECREF(gca);
1830 1832 return PyErr_NoMemory();
1831 1833 }
1832 1834
1833 1835 for (i = 0; i < revcount; i++)
1834 1836 seen[revs[i]] = 1ull << i;
1835 1837
1836 1838 interesting = revcount;
1837 1839
1838 1840 for (v = maxrev; v >= 0 && interesting; v--) {
1839 1841 bitmask sv = seen[v];
1840 1842 int parents[2];
1841 1843
1842 1844 if (!sv)
1843 1845 continue;
1844 1846
1845 1847 if (sv < poison) {
1846 1848 interesting -= 1;
1847 1849 if (sv == allseen) {
1848 1850 PyObject *obj = PyInt_FromLong(v);
1849 1851 if (obj == NULL)
1850 1852 goto bail;
1851 1853 if (PyList_Append(gca, obj) == -1) {
1852 1854 Py_DECREF(obj);
1853 1855 goto bail;
1854 1856 }
1855 1857 sv |= poison;
1856 1858 for (i = 0; i < revcount; i++) {
1857 1859 if (revs[i] == v)
1858 1860 goto done;
1859 1861 }
1860 1862 }
1861 1863 }
1862 1864 if (index_get_parents(self, v, parents, maxrev) < 0)
1863 1865 goto bail;
1864 1866
1865 1867 for (i = 0; i < 2; i++) {
1866 1868 int p = parents[i];
1867 1869 if (p == -1)
1868 1870 continue;
1869 1871 sp = seen[p];
1870 1872 if (sv < poison) {
1871 1873 if (sp == 0) {
1872 1874 seen[p] = sv;
1873 1875 interesting++;
1874 1876 }
1875 1877 else if (sp != sv)
1876 1878 seen[p] |= sv;
1877 1879 } else {
1878 1880 if (sp && sp < poison)
1879 1881 interesting--;
1880 1882 seen[p] = sv;
1881 1883 }
1882 1884 }
1883 1885 }
1884 1886
1885 1887 done:
1886 1888 free(seen);
1887 1889 return gca;
1888 1890 bail:
1889 1891 free(seen);
1890 1892 Py_XDECREF(gca);
1891 1893 return NULL;
1892 1894 }
1893 1895
1894 1896 /*
1895 1897 * Given a disjoint set of revs, return the subset with the longest
1896 1898 * path to the root.
1897 1899 */
1898 1900 static PyObject *find_deepest(indexObject *self, PyObject *revs)
1899 1901 {
1900 1902 const Py_ssize_t revcount = PyList_GET_SIZE(revs);
1901 1903 static const Py_ssize_t capacity = 24;
1902 1904 int *depth, *interesting = NULL;
1903 1905 int i, j, v, ninteresting;
1904 1906 PyObject *dict = NULL, *keys = NULL;
1905 1907 long *seen = NULL;
1906 1908 int maxrev = -1;
1907 1909 long final;
1908 1910
1909 1911 if (revcount > capacity) {
1910 1912 PyErr_Format(PyExc_OverflowError,
1911 1913 "bitset size (%ld) > capacity (%ld)",
1912 1914 (long)revcount, (long)capacity);
1913 1915 return NULL;
1914 1916 }
1915 1917
1916 1918 for (i = 0; i < revcount; i++) {
1917 1919 int n = (int)PyInt_AsLong(PyList_GET_ITEM(revs, i));
1918 1920 if (n > maxrev)
1919 1921 maxrev = n;
1920 1922 }
1921 1923
1922 1924 depth = calloc(sizeof(*depth), maxrev + 1);
1923 1925 if (depth == NULL)
1924 1926 return PyErr_NoMemory();
1925 1927
1926 1928 seen = calloc(sizeof(*seen), maxrev + 1);
1927 1929 if (seen == NULL) {
1928 1930 PyErr_NoMemory();
1929 1931 goto bail;
1930 1932 }
1931 1933
1932 1934 interesting = calloc(sizeof(*interesting), 2 << revcount);
1933 1935 if (interesting == NULL) {
1934 1936 PyErr_NoMemory();
1935 1937 goto bail;
1936 1938 }
1937 1939
1938 1940 if (PyList_Sort(revs) == -1)
1939 1941 goto bail;
1940 1942
1941 1943 for (i = 0; i < revcount; i++) {
1942 1944 int n = (int)PyInt_AsLong(PyList_GET_ITEM(revs, i));
1943 1945 long b = 1l << i;
1944 1946 depth[n] = 1;
1945 1947 seen[n] = b;
1946 1948 interesting[b] = 1;
1947 1949 }
1948 1950
1949 1951 ninteresting = (int)revcount;
1950 1952
1951 1953 for (v = maxrev; v >= 0 && ninteresting > 1; v--) {
1952 1954 int dv = depth[v];
1953 1955 int parents[2];
1954 1956 long sv;
1955 1957
1956 1958 if (dv == 0)
1957 1959 continue;
1958 1960
1959 1961 sv = seen[v];
1960 1962 if (index_get_parents(self, v, parents, maxrev) < 0)
1961 1963 goto bail;
1962 1964
1963 1965 for (i = 0; i < 2; i++) {
1964 1966 int p = parents[i];
1965 1967 long nsp, sp;
1966 1968 int dp;
1967 1969
1968 1970 if (p == -1)
1969 1971 continue;
1970 1972
1971 1973 dp = depth[p];
1972 1974 nsp = sp = seen[p];
1973 1975 if (dp <= dv) {
1974 1976 depth[p] = dv + 1;
1975 1977 if (sp != sv) {
1976 1978 interesting[sv] += 1;
1977 1979 nsp = seen[p] = sv;
1978 1980 if (sp) {
1979 1981 interesting[sp] -= 1;
1980 1982 if (interesting[sp] == 0)
1981 1983 ninteresting -= 1;
1982 1984 }
1983 1985 }
1984 1986 }
1985 1987 else if (dv == dp - 1) {
1986 1988 nsp = sp | sv;
1987 1989 if (nsp == sp)
1988 1990 continue;
1989 1991 seen[p] = nsp;
1990 1992 interesting[sp] -= 1;
1991 1993 if (interesting[sp] == 0 && interesting[nsp] > 0)
1992 1994 ninteresting -= 1;
1993 1995 interesting[nsp] += 1;
1994 1996 }
1995 1997 }
1996 1998 interesting[sv] -= 1;
1997 1999 if (interesting[sv] == 0)
1998 2000 ninteresting -= 1;
1999 2001 }
2000 2002
2001 2003 final = 0;
2002 2004 j = ninteresting;
2003 2005 for (i = 0; i < (int)(2 << revcount) && j > 0; i++) {
2004 2006 if (interesting[i] == 0)
2005 2007 continue;
2006 2008 final |= i;
2007 2009 j -= 1;
2008 2010 }
2009 2011 if (final == 0) {
2010 2012 keys = PyList_New(0);
2011 2013 goto bail;
2012 2014 }
2013 2015
2014 2016 dict = PyDict_New();
2015 2017 if (dict == NULL)
2016 2018 goto bail;
2017 2019
2018 2020 for (i = 0; i < revcount; i++) {
2019 2021 PyObject *key;
2020 2022
2021 2023 if ((final & (1 << i)) == 0)
2022 2024 continue;
2023 2025
2024 2026 key = PyList_GET_ITEM(revs, i);
2025 2027 Py_INCREF(key);
2026 2028 Py_INCREF(Py_None);
2027 2029 if (PyDict_SetItem(dict, key, Py_None) == -1) {
2028 2030 Py_DECREF(key);
2029 2031 Py_DECREF(Py_None);
2030 2032 goto bail;
2031 2033 }
2032 2034 }
2033 2035
2034 2036 keys = PyDict_Keys(dict);
2035 2037
2036 2038 bail:
2037 2039 free(depth);
2038 2040 free(seen);
2039 2041 free(interesting);
2040 2042 Py_XDECREF(dict);
2041 2043
2042 2044 return keys;
2043 2045 }
2044 2046
2045 2047 /*
2046 2048 * Given a (possibly overlapping) set of revs, return all the
2047 2049 * common ancestors heads: heads(::args[0] and ::a[1] and ...)
2048 2050 */
2049 2051 static PyObject *index_commonancestorsheads(indexObject *self, PyObject *args)
2050 2052 {
2051 2053 PyObject *ret = NULL;
2052 2054 Py_ssize_t argcount, i, len;
2053 2055 bitmask repeat = 0;
2054 2056 int revcount = 0;
2055 2057 int *revs;
2056 2058
2057 2059 argcount = PySequence_Length(args);
2058 2060 revs = malloc(argcount * sizeof(*revs));
2059 2061 if (argcount > 0 && revs == NULL)
2060 2062 return PyErr_NoMemory();
2061 2063 len = index_length(self) - 1;
2062 2064
2063 2065 for (i = 0; i < argcount; i++) {
2064 2066 static const int capacity = 24;
2065 2067 PyObject *obj = PySequence_GetItem(args, i);
2066 2068 bitmask x;
2067 2069 long val;
2068 2070
2069 2071 if (!PyInt_Check(obj)) {
2070 2072 PyErr_SetString(PyExc_TypeError,
2071 2073 "arguments must all be ints");
2072 2074 Py_DECREF(obj);
2073 2075 goto bail;
2074 2076 }
2075 2077 val = PyInt_AsLong(obj);
2076 2078 Py_DECREF(obj);
2077 2079 if (val == -1) {
2078 2080 ret = PyList_New(0);
2079 2081 goto done;
2080 2082 }
2081 2083 if (val < 0 || val >= len) {
2082 2084 PyErr_SetString(PyExc_IndexError,
2083 2085 "index out of range");
2084 2086 goto bail;
2085 2087 }
2086 2088 /* this cheesy bloom filter lets us avoid some more
2087 2089 * expensive duplicate checks in the common set-is-disjoint
2088 2090 * case */
2089 2091 x = 1ull << (val & 0x3f);
2090 2092 if (repeat & x) {
2091 2093 int k;
2092 2094 for (k = 0; k < revcount; k++) {
2093 2095 if (val == revs[k])
2094 2096 goto duplicate;
2095 2097 }
2096 2098 }
2097 2099 else repeat |= x;
2098 2100 if (revcount >= capacity) {
2099 2101 PyErr_Format(PyExc_OverflowError,
2100 2102 "bitset size (%d) > capacity (%d)",
2101 2103 revcount, capacity);
2102 2104 goto bail;
2103 2105 }
2104 2106 revs[revcount++] = (int)val;
2105 2107 duplicate:;
2106 2108 }
2107 2109
2108 2110 if (revcount == 0) {
2109 2111 ret = PyList_New(0);
2110 2112 goto done;
2111 2113 }
2112 2114 if (revcount == 1) {
2113 2115 PyObject *obj;
2114 2116 ret = PyList_New(1);
2115 2117 if (ret == NULL)
2116 2118 goto bail;
2117 2119 obj = PyInt_FromLong(revs[0]);
2118 2120 if (obj == NULL)
2119 2121 goto bail;
2120 2122 PyList_SET_ITEM(ret, 0, obj);
2121 2123 goto done;
2122 2124 }
2123 2125
2124 2126 ret = find_gca_candidates(self, revs, revcount);
2125 2127 if (ret == NULL)
2126 2128 goto bail;
2127 2129
2128 2130 done:
2129 2131 free(revs);
2130 2132 return ret;
2131 2133
2132 2134 bail:
2133 2135 free(revs);
2134 2136 Py_XDECREF(ret);
2135 2137 return NULL;
2136 2138 }
2137 2139
2138 2140 /*
2139 2141 * Given a (possibly overlapping) set of revs, return the greatest
2140 2142 * common ancestors: those with the longest path to the root.
2141 2143 */
2142 2144 static PyObject *index_ancestors(indexObject *self, PyObject *args)
2143 2145 {
2144 2146 PyObject *gca = index_commonancestorsheads(self, args);
2145 2147 if (gca == NULL)
2146 2148 return NULL;
2147 2149
2148 2150 if (PyList_GET_SIZE(gca) <= 1) {
2149 2151 Py_INCREF(gca);
2150 2152 return gca;
2151 2153 }
2152 2154
2153 2155 return find_deepest(self, gca);
2154 2156 }
2155 2157
2156 2158 /*
2157 2159 * Invalidate any trie entries introduced by added revs.
2158 2160 */
2159 2161 static void nt_invalidate_added(indexObject *self, Py_ssize_t start)
2160 2162 {
2161 2163 Py_ssize_t i, len = PyList_GET_SIZE(self->added);
2162 2164
2163 2165 for (i = start; i < len; i++) {
2164 2166 PyObject *tuple = PyList_GET_ITEM(self->added, i);
2165 2167 PyObject *node = PyTuple_GET_ITEM(tuple, 7);
2166 2168
2167 2169 nt_insert(self, PyString_AS_STRING(node), -1);
2168 2170 }
2169 2171
2170 2172 if (start == 0)
2171 2173 Py_CLEAR(self->added);
2172 2174 }
2173 2175
2174 2176 /*
2175 2177 * Delete a numeric range of revs, which must be at the end of the
2176 2178 * range, but exclude the sentinel nullid entry.
2177 2179 */
2178 2180 static int index_slice_del(indexObject *self, PyObject *item)
2179 2181 {
2180 2182 Py_ssize_t start, stop, step, slicelength;
2181 2183 Py_ssize_t length = index_length(self);
2182 2184 int ret = 0;
2183 2185
2184 2186 if (PySlice_GetIndicesEx((PySliceObject*)item, length,
2185 2187 &start, &stop, &step, &slicelength) < 0)
2186 2188 return -1;
2187 2189
2188 2190 if (slicelength <= 0)
2189 2191 return 0;
2190 2192
2191 2193 if ((step < 0 && start < stop) || (step > 0 && start > stop))
2192 2194 stop = start;
2193 2195
2194 2196 if (step < 0) {
2195 2197 stop = start + 1;
2196 2198 start = stop + step*(slicelength - 1) - 1;
2197 2199 step = -step;
2198 2200 }
2199 2201
2200 2202 if (step != 1) {
2201 2203 PyErr_SetString(PyExc_ValueError,
2202 2204 "revlog index delete requires step size of 1");
2203 2205 return -1;
2204 2206 }
2205 2207
2206 2208 if (stop != length - 1) {
2207 2209 PyErr_SetString(PyExc_IndexError,
2208 2210 "revlog index deletion indices are invalid");
2209 2211 return -1;
2210 2212 }
2211 2213
2212 2214 if (start < self->length - 1) {
2213 2215 if (self->nt) {
2214 2216 Py_ssize_t i;
2215 2217
2216 2218 for (i = start + 1; i < self->length - 1; i++) {
2217 2219 const char *node = index_node(self, i);
2218 2220
2219 2221 if (node)
2220 2222 nt_insert(self, node, -1);
2221 2223 }
2222 2224 if (self->added)
2223 2225 nt_invalidate_added(self, 0);
2224 2226 if (self->ntrev > start)
2225 2227 self->ntrev = (int)start;
2226 2228 }
2227 2229 self->length = start + 1;
2228 2230 if (start < self->raw_length) {
2229 2231 if (self->cache) {
2230 2232 Py_ssize_t i;
2231 2233 for (i = start; i < self->raw_length; i++)
2232 2234 Py_CLEAR(self->cache[i]);
2233 2235 }
2234 2236 self->raw_length = start;
2235 2237 }
2236 2238 goto done;
2237 2239 }
2238 2240
2239 2241 if (self->nt) {
2240 2242 nt_invalidate_added(self, start - self->length + 1);
2241 2243 if (self->ntrev > start)
2242 2244 self->ntrev = (int)start;
2243 2245 }
2244 2246 if (self->added)
2245 2247 ret = PyList_SetSlice(self->added, start - self->length + 1,
2246 2248 PyList_GET_SIZE(self->added), NULL);
2247 2249 done:
2248 2250 Py_CLEAR(self->headrevs);
2249 2251 return ret;
2250 2252 }
2251 2253
2252 2254 /*
2253 2255 * Supported ops:
2254 2256 *
2255 2257 * slice deletion
2256 2258 * string assignment (extend node->rev mapping)
2257 2259 * string deletion (shrink node->rev mapping)
2258 2260 */
2259 2261 static int index_assign_subscript(indexObject *self, PyObject *item,
2260 2262 PyObject *value)
2261 2263 {
2262 2264 char *node;
2263 2265 Py_ssize_t nodelen;
2264 2266 long rev;
2265 2267
2266 2268 if (PySlice_Check(item) && value == NULL)
2267 2269 return index_slice_del(self, item);
2268 2270
2269 2271 if (node_check(item, &node, &nodelen) == -1)
2270 2272 return -1;
2271 2273
2272 2274 if (value == NULL)
2273 2275 return self->nt ? nt_insert(self, node, -1) : 0;
2274 2276 rev = PyInt_AsLong(value);
2275 2277 if (rev > INT_MAX || rev < 0) {
2276 2278 if (!PyErr_Occurred())
2277 2279 PyErr_SetString(PyExc_ValueError, "rev out of range");
2278 2280 return -1;
2279 2281 }
2280 2282
2281 2283 if (nt_init(self) == -1)
2282 2284 return -1;
2283 2285 return nt_insert(self, node, (int)rev);
2284 2286 }
2285 2287
2286 2288 /*
2287 2289 * Find all RevlogNG entries in an index that has inline data. Update
2288 2290 * the optional "offsets" table with those entries.
2289 2291 */
2290 2292 static Py_ssize_t inline_scan(indexObject *self, const char **offsets)
2291 2293 {
2292 2294 const char *data = PyString_AS_STRING(self->data);
2293 2295 Py_ssize_t pos = 0;
2294 2296 Py_ssize_t end = PyString_GET_SIZE(self->data);
2295 2297 long incr = v1_hdrsize;
2296 2298 Py_ssize_t len = 0;
2297 2299
2298 2300 while (pos + v1_hdrsize <= end && pos >= 0) {
2299 2301 uint32_t comp_len;
2300 2302 /* 3rd element of header is length of compressed inline data */
2301 2303 comp_len = getbe32(data + pos + 8);
2302 2304 incr = v1_hdrsize + comp_len;
2303 2305 if (offsets)
2304 2306 offsets[len] = data + pos;
2305 2307 len++;
2306 2308 pos += incr;
2307 2309 }
2308 2310
2309 2311 if (pos != end) {
2310 2312 if (!PyErr_Occurred())
2311 2313 PyErr_SetString(PyExc_ValueError, "corrupt index file");
2312 2314 return -1;
2313 2315 }
2314 2316
2315 2317 return len;
2316 2318 }
2317 2319
2318 2320 static int index_init(indexObject *self, PyObject *args)
2319 2321 {
2320 2322 PyObject *data_obj, *inlined_obj;
2321 2323 Py_ssize_t size;
2322 2324
2323 2325 /* Initialize before argument-checking to avoid index_dealloc() crash. */
2324 2326 self->raw_length = 0;
2325 2327 self->added = NULL;
2326 2328 self->cache = NULL;
2327 2329 self->data = NULL;
2328 2330 self->headrevs = NULL;
2329 2331 self->filteredrevs = Py_None;
2330 2332 Py_INCREF(Py_None);
2331 2333 self->nt = NULL;
2332 2334 self->offsets = NULL;
2333 2335
2334 2336 if (!PyArg_ParseTuple(args, "OO", &data_obj, &inlined_obj))
2335 2337 return -1;
2336 2338 if (!PyString_Check(data_obj)) {
2337 2339 PyErr_SetString(PyExc_TypeError, "data is not a string");
2338 2340 return -1;
2339 2341 }
2340 2342 size = PyString_GET_SIZE(data_obj);
2341 2343
2342 2344 self->inlined = inlined_obj && PyObject_IsTrue(inlined_obj);
2343 2345 self->data = data_obj;
2344 2346
2345 2347 self->ntlength = self->ntcapacity = 0;
2346 2348 self->ntdepth = self->ntsplits = 0;
2347 2349 self->ntlookups = self->ntmisses = 0;
2348 2350 self->ntrev = -1;
2349 2351 Py_INCREF(self->data);
2350 2352
2351 2353 if (self->inlined) {
2352 2354 Py_ssize_t len = inline_scan(self, NULL);
2353 2355 if (len == -1)
2354 2356 goto bail;
2355 2357 self->raw_length = len;
2356 2358 self->length = len + 1;
2357 2359 } else {
2358 2360 if (size % v1_hdrsize) {
2359 2361 PyErr_SetString(PyExc_ValueError, "corrupt index file");
2360 2362 goto bail;
2361 2363 }
2362 2364 self->raw_length = size / v1_hdrsize;
2363 2365 self->length = self->raw_length + 1;
2364 2366 }
2365 2367
2366 2368 return 0;
2367 2369 bail:
2368 2370 return -1;
2369 2371 }
2370 2372
2371 2373 static PyObject *index_nodemap(indexObject *self)
2372 2374 {
2373 2375 Py_INCREF(self);
2374 2376 return (PyObject *)self;
2375 2377 }
2376 2378
2377 2379 static void index_dealloc(indexObject *self)
2378 2380 {
2379 2381 _index_clearcaches(self);
2380 2382 Py_XDECREF(self->filteredrevs);
2381 2383 Py_XDECREF(self->data);
2382 2384 Py_XDECREF(self->added);
2383 2385 PyObject_Del(self);
2384 2386 }
2385 2387
2386 2388 static PySequenceMethods index_sequence_methods = {
2387 2389 (lenfunc)index_length, /* sq_length */
2388 2390 0, /* sq_concat */
2389 2391 0, /* sq_repeat */
2390 2392 (ssizeargfunc)index_get, /* sq_item */
2391 2393 0, /* sq_slice */
2392 2394 0, /* sq_ass_item */
2393 2395 0, /* sq_ass_slice */
2394 2396 (objobjproc)index_contains, /* sq_contains */
2395 2397 };
2396 2398
2397 2399 static PyMappingMethods index_mapping_methods = {
2398 2400 (lenfunc)index_length, /* mp_length */
2399 2401 (binaryfunc)index_getitem, /* mp_subscript */
2400 2402 (objobjargproc)index_assign_subscript, /* mp_ass_subscript */
2401 2403 };
2402 2404
2403 2405 static PyMethodDef index_methods[] = {
2404 2406 {"ancestors", (PyCFunction)index_ancestors, METH_VARARGS,
2405 2407 "return the gca set of the given revs"},
2406 2408 {"commonancestorsheads", (PyCFunction)index_commonancestorsheads,
2407 2409 METH_VARARGS,
2408 2410 "return the heads of the common ancestors of the given revs"},
2409 2411 {"clearcaches", (PyCFunction)index_clearcaches, METH_NOARGS,
2410 2412 "clear the index caches"},
2411 2413 {"get", (PyCFunction)index_m_get, METH_VARARGS,
2412 2414 "get an index entry"},
2413 2415 {"computephasesmapsets", (PyCFunction)compute_phases_map_sets,
2414 2416 METH_VARARGS, "compute phases"},
2415 2417 {"reachableroots", (PyCFunction)reachableroots, METH_VARARGS,
2416 2418 "reachableroots"},
2417 2419 {"headrevs", (PyCFunction)index_headrevs, METH_VARARGS,
2418 2420 "get head revisions"}, /* Can do filtering since 3.2 */
2419 2421 {"headrevsfiltered", (PyCFunction)index_headrevs, METH_VARARGS,
2420 2422 "get filtered head revisions"}, /* Can always do filtering */
2421 2423 {"insert", (PyCFunction)index_insert, METH_VARARGS,
2422 2424 "insert an index entry"},
2423 2425 {"partialmatch", (PyCFunction)index_partialmatch, METH_VARARGS,
2424 2426 "match a potentially ambiguous node ID"},
2425 2427 {"stats", (PyCFunction)index_stats, METH_NOARGS,
2426 2428 "stats for the index"},
2427 2429 {NULL} /* Sentinel */
2428 2430 };
2429 2431
2430 2432 static PyGetSetDef index_getset[] = {
2431 2433 {"nodemap", (getter)index_nodemap, NULL, "nodemap", NULL},
2432 2434 {NULL} /* Sentinel */
2433 2435 };
2434 2436
2435 2437 static PyTypeObject indexType = {
2436 2438 PyObject_HEAD_INIT(NULL)
2437 2439 0, /* ob_size */
2438 2440 "parsers.index", /* tp_name */
2439 2441 sizeof(indexObject), /* tp_basicsize */
2440 2442 0, /* tp_itemsize */
2441 2443 (destructor)index_dealloc, /* tp_dealloc */
2442 2444 0, /* tp_print */
2443 2445 0, /* tp_getattr */
2444 2446 0, /* tp_setattr */
2445 2447 0, /* tp_compare */
2446 2448 0, /* tp_repr */
2447 2449 0, /* tp_as_number */
2448 2450 &index_sequence_methods, /* tp_as_sequence */
2449 2451 &index_mapping_methods, /* tp_as_mapping */
2450 2452 0, /* tp_hash */
2451 2453 0, /* tp_call */
2452 2454 0, /* tp_str */
2453 2455 0, /* tp_getattro */
2454 2456 0, /* tp_setattro */
2455 2457 0, /* tp_as_buffer */
2456 2458 Py_TPFLAGS_DEFAULT, /* tp_flags */
2457 2459 "revlog index", /* tp_doc */
2458 2460 0, /* tp_traverse */
2459 2461 0, /* tp_clear */
2460 2462 0, /* tp_richcompare */
2461 2463 0, /* tp_weaklistoffset */
2462 2464 0, /* tp_iter */
2463 2465 0, /* tp_iternext */
2464 2466 index_methods, /* tp_methods */
2465 2467 0, /* tp_members */
2466 2468 index_getset, /* tp_getset */
2467 2469 0, /* tp_base */
2468 2470 0, /* tp_dict */
2469 2471 0, /* tp_descr_get */
2470 2472 0, /* tp_descr_set */
2471 2473 0, /* tp_dictoffset */
2472 2474 (initproc)index_init, /* tp_init */
2473 2475 0, /* tp_alloc */
2474 2476 };
2475 2477
2476 2478 /*
2477 2479 * returns a tuple of the form (index, index, cache) with elements as
2478 2480 * follows:
2479 2481 *
2480 2482 * index: an index object that lazily parses RevlogNG records
2481 2483 * cache: if data is inlined, a tuple (index_file_content, 0), else None
2482 2484 *
2483 2485 * added complications are for backwards compatibility
2484 2486 */
2485 2487 static PyObject *parse_index2(PyObject *self, PyObject *args)
2486 2488 {
2487 2489 PyObject *tuple = NULL, *cache = NULL;
2488 2490 indexObject *idx;
2489 2491 int ret;
2490 2492
2491 2493 idx = PyObject_New(indexObject, &indexType);
2492 2494 if (idx == NULL)
2493 2495 goto bail;
2494 2496
2495 2497 ret = index_init(idx, args);
2496 2498 if (ret == -1)
2497 2499 goto bail;
2498 2500
2499 2501 if (idx->inlined) {
2500 2502 cache = Py_BuildValue("iO", 0, idx->data);
2501 2503 if (cache == NULL)
2502 2504 goto bail;
2503 2505 } else {
2504 2506 cache = Py_None;
2505 2507 Py_INCREF(cache);
2506 2508 }
2507 2509
2508 2510 tuple = Py_BuildValue("NN", idx, cache);
2509 2511 if (!tuple)
2510 2512 goto bail;
2511 2513 return tuple;
2512 2514
2513 2515 bail:
2514 2516 Py_XDECREF(idx);
2515 2517 Py_XDECREF(cache);
2516 2518 Py_XDECREF(tuple);
2517 2519 return NULL;
2518 2520 }
2519 2521
2520 2522 #define BUMPED_FIX 1
2521 2523 #define USING_SHA_256 2
2522 2524
2523 2525 static PyObject *readshas(
2524 2526 const char *source, unsigned char num, Py_ssize_t hashwidth)
2525 2527 {
2526 2528 int i;
2527 2529 PyObject *list = PyTuple_New(num);
2528 2530 if (list == NULL) {
2529 2531 return NULL;
2530 2532 }
2531 2533 for (i = 0; i < num; i++) {
2532 2534 PyObject *hash = PyString_FromStringAndSize(source, hashwidth);
2533 2535 if (hash == NULL) {
2534 2536 Py_DECREF(list);
2535 2537 return NULL;
2536 2538 }
2537 2539 PyTuple_SetItem(list, i, hash);
2538 2540 source += hashwidth;
2539 2541 }
2540 2542 return list;
2541 2543 }
2542 2544
2543 2545 static PyObject *fm1readmarker(const char *data, uint32_t *msize)
2544 2546 {
2545 2547 const char *meta;
2546 2548
2547 2549 double mtime;
2548 2550 int16_t tz;
2549 2551 uint16_t flags;
2550 2552 unsigned char nsuccs, nparents, nmetadata;
2551 2553 Py_ssize_t hashwidth = 20;
2552 2554
2553 2555 PyObject *prec = NULL, *parents = NULL, *succs = NULL;
2554 2556 PyObject *metadata = NULL, *ret = NULL;
2555 2557 int i;
2556 2558
2557 2559 *msize = getbe32(data);
2558 2560 data += 4;
2559 2561 mtime = getbefloat64(data);
2560 2562 data += 8;
2561 2563 tz = getbeint16(data);
2562 2564 data += 2;
2563 2565 flags = getbeuint16(data);
2564 2566 data += 2;
2565 2567
2566 2568 if (flags & USING_SHA_256) {
2567 2569 hashwidth = 32;
2568 2570 }
2569 2571
2570 2572 nsuccs = (unsigned char)(*data++);
2571 2573 nparents = (unsigned char)(*data++);
2572 2574 nmetadata = (unsigned char)(*data++);
2573 2575
2574 2576 prec = PyString_FromStringAndSize(data, hashwidth);
2575 2577 data += hashwidth;
2576 2578 if (prec == NULL) {
2577 2579 goto bail;
2578 2580 }
2579 2581
2580 2582 succs = readshas(data, nsuccs, hashwidth);
2581 2583 if (succs == NULL) {
2582 2584 goto bail;
2583 2585 }
2584 2586 data += nsuccs * hashwidth;
2585 2587
2586 2588 if (nparents == 1 || nparents == 2) {
2587 2589 parents = readshas(data, nparents, hashwidth);
2588 2590 if (parents == NULL) {
2589 2591 goto bail;
2590 2592 }
2591 2593 data += nparents * hashwidth;
2592 2594 } else {
2593 2595 parents = Py_None;
2594 2596 }
2595 2597
2596 2598 meta = data + (2 * nmetadata);
2597 2599 metadata = PyTuple_New(nmetadata);
2598 2600 if (metadata == NULL) {
2599 2601 goto bail;
2600 2602 }
2601 2603 for (i = 0; i < nmetadata; i++) {
2602 2604 PyObject *tmp, *left = NULL, *right = NULL;
2603 2605 Py_ssize_t metasize = (unsigned char)(*data++);
2604 2606 left = PyString_FromStringAndSize(meta, metasize);
2605 2607 meta += metasize;
2606 2608 metasize = (unsigned char)(*data++);
2607 2609 right = PyString_FromStringAndSize(meta, metasize);
2608 2610 meta += metasize;
2609 2611 if (!left || !right) {
2610 2612 Py_XDECREF(left);
2611 2613 Py_XDECREF(right);
2612 2614 goto bail;
2613 2615 }
2614 2616 tmp = PyTuple_Pack(2, left, right);
2615 2617 Py_DECREF(left);
2616 2618 Py_DECREF(right);
2617 2619 if (!tmp) {
2618 2620 goto bail;
2619 2621 }
2620 2622 PyTuple_SetItem(metadata, i, tmp);
2621 2623 }
2622 2624 ret = Py_BuildValue("(OOHO(di)O)", prec, succs, flags,
2623 2625 metadata, mtime, (int)tz * 60, parents);
2624 2626 bail:
2625 2627 Py_XDECREF(prec);
2626 2628 Py_XDECREF(succs);
2627 2629 Py_XDECREF(metadata);
2628 2630 if (parents != Py_None)
2629 2631 Py_XDECREF(parents);
2630 2632 return ret;
2631 2633 }
2632 2634
2633 2635
2634 2636 static PyObject *fm1readmarkers(PyObject *self, PyObject *args) {
2635 2637 const char *data;
2636 2638 Py_ssize_t datalen;
2637 2639 /* only unsigned long because python 2.4, should be Py_ssize_t */
2638 2640 unsigned long offset, stop;
2639 2641 PyObject *markers = NULL;
2640 2642
2641 2643 /* replace kk with nn when we drop Python 2.4 */
2642 2644 if (!PyArg_ParseTuple(args, "s#kk", &data, &datalen, &offset, &stop)) {
2643 2645 return NULL;
2644 2646 }
2645 2647 data += offset;
2646 2648 markers = PyList_New(0);
2647 2649 if (!markers) {
2648 2650 return NULL;
2649 2651 }
2650 2652 while (offset < stop) {
2651 2653 uint32_t msize;
2652 2654 int error;
2653 2655 PyObject *record = fm1readmarker(data, &msize);
2654 2656 if (!record) {
2655 2657 goto bail;
2656 2658 }
2657 2659 error = PyList_Append(markers, record);
2658 2660 Py_DECREF(record);
2659 2661 if (error) {
2660 2662 goto bail;
2661 2663 }
2662 2664 data += msize;
2663 2665 offset += msize;
2664 2666 }
2665 2667 return markers;
2666 2668 bail:
2667 2669 Py_DECREF(markers);
2668 2670 return NULL;
2669 2671 }
2670 2672
2671 2673 static char parsers_doc[] = "Efficient content parsing.";
2672 2674
2673 2675 PyObject *encodedir(PyObject *self, PyObject *args);
2674 2676 PyObject *pathencode(PyObject *self, PyObject *args);
2675 2677 PyObject *lowerencode(PyObject *self, PyObject *args);
2676 2678
2677 2679 static PyMethodDef methods[] = {
2678 2680 {"pack_dirstate", pack_dirstate, METH_VARARGS, "pack a dirstate\n"},
2679 2681 {"parse_manifest", parse_manifest, METH_VARARGS, "parse a manifest\n"},
2680 2682 {"parse_dirstate", parse_dirstate, METH_VARARGS, "parse a dirstate\n"},
2681 2683 {"parse_index2", parse_index2, METH_VARARGS, "parse a revlog index\n"},
2682 2684 {"asciilower", asciilower, METH_VARARGS, "lowercase an ASCII string\n"},
2683 2685 {"asciiupper", asciiupper, METH_VARARGS, "uppercase an ASCII string\n"},
2684 2686 {"dict_new_presized", dict_new_presized, METH_VARARGS,
2685 2687 "construct a dict with an expected size\n"},
2686 2688 {"make_file_foldmap", make_file_foldmap, METH_VARARGS,
2687 2689 "make file foldmap\n"},
2688 2690 {"encodedir", encodedir, METH_VARARGS, "encodedir a path\n"},
2689 2691 {"pathencode", pathencode, METH_VARARGS, "fncache-encode a path\n"},
2690 2692 {"lowerencode", lowerencode, METH_VARARGS, "lower-encode a path\n"},
2691 2693 {"fm1readmarkers", fm1readmarkers, METH_VARARGS,
2692 2694 "parse v1 obsolete markers\n"},
2693 2695 {NULL, NULL}
2694 2696 };
2695 2697
2696 2698 void dirs_module_init(PyObject *mod);
2697 2699 void manifest_module_init(PyObject *mod);
2698 2700
2699 2701 static void module_init(PyObject *mod)
2700 2702 {
2701 2703 /* This module constant has two purposes. First, it lets us unit test
2702 2704 * the ImportError raised without hard-coding any error text. This
2703 2705 * means we can change the text in the future without breaking tests,
2704 2706 * even across changesets without a recompile. Second, its presence
2705 2707 * can be used to determine whether the version-checking logic is
2706 2708 * present, which also helps in testing across changesets without a
2707 2709 * recompile. Note that this means the pure-Python version of parsers
2708 2710 * should not have this module constant. */
2709 2711 PyModule_AddStringConstant(mod, "versionerrortext", versionerrortext);
2710 2712
2711 2713 dirs_module_init(mod);
2712 2714 manifest_module_init(mod);
2713 2715
2714 2716 indexType.tp_new = PyType_GenericNew;
2715 2717 if (PyType_Ready(&indexType) < 0 ||
2716 2718 PyType_Ready(&dirstateTupleType) < 0)
2717 2719 return;
2718 2720 Py_INCREF(&indexType);
2719 2721 PyModule_AddObject(mod, "index", (PyObject *)&indexType);
2720 2722 Py_INCREF(&dirstateTupleType);
2721 2723 PyModule_AddObject(mod, "dirstatetuple",
2722 2724 (PyObject *)&dirstateTupleType);
2723 2725
2724 2726 nullentry = Py_BuildValue("iiiiiiis#", 0, 0, 0,
2725 2727 -1, -1, -1, -1, nullid, 20);
2726 2728 if (nullentry)
2727 2729 PyObject_GC_UnTrack(nullentry);
2728 2730 }
2729 2731
2730 2732 static int check_python_version(void)
2731 2733 {
2732 2734 PyObject *sys = PyImport_ImportModule("sys"), *ver;
2733 2735 long hexversion;
2734 2736 if (!sys)
2735 2737 return -1;
2736 2738 ver = PyObject_GetAttrString(sys, "hexversion");
2737 2739 Py_DECREF(sys);
2738 2740 if (!ver)
2739 2741 return -1;
2740 2742 hexversion = PyInt_AsLong(ver);
2741 2743 Py_DECREF(ver);
2742 2744 /* sys.hexversion is a 32-bit number by default, so the -1 case
2743 2745 * should only occur in unusual circumstances (e.g. if sys.hexversion
2744 2746 * is manually set to an invalid value). */
2745 2747 if ((hexversion == -1) || (hexversion >> 16 != PY_VERSION_HEX >> 16)) {
2746 2748 PyErr_Format(PyExc_ImportError, "%s: The Mercurial extension "
2747 2749 "modules were compiled with Python " PY_VERSION ", but "
2748 2750 "Mercurial is currently using Python with sys.hexversion=%ld: "
2749 2751 "Python %s\n at: %s", versionerrortext, hexversion,
2750 2752 Py_GetVersion(), Py_GetProgramFullPath());
2751 2753 return -1;
2752 2754 }
2753 2755 return 0;
2754 2756 }
2755 2757
2756 2758 #ifdef IS_PY3K
2757 2759 static struct PyModuleDef parsers_module = {
2758 2760 PyModuleDef_HEAD_INIT,
2759 2761 "parsers",
2760 2762 parsers_doc,
2761 2763 -1,
2762 2764 methods
2763 2765 };
2764 2766
2765 2767 PyMODINIT_FUNC PyInit_parsers(void)
2766 2768 {
2767 2769 PyObject *mod;
2768 2770
2769 2771 if (check_python_version() == -1)
2770 2772 return;
2771 2773 mod = PyModule_Create(&parsers_module);
2772 2774 module_init(mod);
2773 2775 return mod;
2774 2776 }
2775 2777 #else
2776 2778 PyMODINIT_FUNC initparsers(void)
2777 2779 {
2778 2780 PyObject *mod;
2779 2781
2780 2782 if (check_python_version() == -1)
2781 2783 return;
2782 2784 mod = Py_InitModule3("parsers", methods, parsers_doc);
2783 2785 module_init(mod);
2784 2786 }
2785 2787 #endif
General Comments 0
You need to be logged in to leave comments. Login now