comparison mercurial/cext/revlog.c @ 48769:1bb62821f080

revlog: register changelogv2 C implementation in parsers

This allows Python code to make use of the C implementation of the changelogv2
base operations when the C extensions are enabled.

The `format_version` values are now shared between the C and Python sides,
avoiding an additional translation for the selection of the format version to
use.

Differential Revision: https://phab.mercurial-scm.org/D12179

48768:7dd5a2c0116a

	int inlined;
	long entry_size; /* size of index headers. Differs in v1 v.s. v2 format
	                  */
	long rust_ext_compat; /* compatibility with being used in rust
	                         extensions */
	char format_version;  /* size of index headers. Differs in v1 v.s. v2
	                         format */
};

static Py_ssize_t index_length(const indexObject *self)
{
	return self->length + self->new_length;

static const long v1_entry_size = 64;

/* A Revlogv2 index entry is 96 bytes long. */
static const long v2_entry_size = 96;

static const long format_v1 = 1;  /* Internal only, could be any number */
static const long format_v2 = 2;  /* Internal only, could be any number */
static const long format_cl2 = 3; /* Internal only, could be any number */

static const long entry_v1_offset_high = 0;
static const long entry_v1_offset_offset_flags = 4;
static const long entry_v1_offset_comp_len = 8;
static const long entry_v1_offset_uncomp_len = 12;

	return len;
}

static int index_init(indexObject *self, PyObject *args, PyObject *kwargs)
{
	PyObject *data_obj, *inlined_obj, *revlogv2;
	Py_ssize_t size;

	static char *kwlist[] = {"data", "inlined", "revlogv2", NULL};

	/* Initialize before argument-checking to avoid index_dealloc() crash.
	 */
	self->added = NULL;
	self->new_length = 0;

	self->ntinitialized = 0;
	self->offsets = NULL;
	self->nodelen = 20;
	self->nullentry = NULL;
	self->rust_ext_compat = 1;

	revlogv2 = NULL;
	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|O", kwlist,
	                                 &data_obj, &inlined_obj, &revlogv2))
		return -1;
	if (!PyObject_CheckBuffer(data_obj)) {
		PyErr_SetString(PyExc_TypeError,
		                "data does not support buffer interface");
		return -1;

	if (self->nodelen < 20 || self->nodelen > (Py_ssize_t)sizeof(nullid)) {
		PyErr_SetString(PyExc_RuntimeError, "unsupported node size");
		return -1;
	}

	if (revlogv2 && PyObject_IsTrue(revlogv2)) {
		self->format_version = format_v2;
		self->entry_size = v2_entry_size;
	} else {
		self->format_version = format_v1;
		self->entry_size = v1_entry_size;
	}

	self->nullentry =
	    Py_BuildValue(PY23("iiiiiiis#iiBBi", "iiiiiiiy#iiBBi"), 0, 0, 0, -1,
	                  -1, -1, -1, nullid, self->nodelen, 0, 0,

48769:1bb62821f080

	int inlined;
	long entry_size; /* size of index headers. Differs in v1 v.s. v2 format
	                  */
	long rust_ext_compat; /* compatibility with being used in rust
	                         extensions */
	long format_version;  /* format version selector (format_*) */
};

static Py_ssize_t index_length(const indexObject *self)
{
	return self->length + self->new_length;

static const long v1_entry_size = 64;

/* A Revlogv2 index entry is 96 bytes long. */
static const long v2_entry_size = 96;

/* A Changelogv2 index entry is 96 bytes long. */
static const long cl2_entry_size = 96;

/* Internal format version.
 * Must match their counterparts in revlogutils/constants.py */
static const long format_v1 = 1;       /* constants.py: REVLOGV1 */
static const long format_v2 = 0xDEAD;  /* constants.py: REVLOGV2 */
static const long format_cl2 = 0xD34D; /* constants.py: CHANGELOGV2 */

static const long entry_v1_offset_high = 0;
static const long entry_v1_offset_offset_flags = 4;
static const long entry_v1_offset_comp_len = 8;
static const long entry_v1_offset_uncomp_len = 12;

	return len;
}

static int index_init(indexObject *self, PyObject *args, PyObject *kwargs)
{
	PyObject *data_obj, *inlined_obj;
	Py_ssize_t size;

	static char *kwlist[] = {"data", "inlined", "format", NULL};

	/* Initialize before argument-checking to avoid index_dealloc() crash.
	 */
	self->added = NULL;
	self->new_length = 0;

	self->ntinitialized = 0;
	self->offsets = NULL;
	self->nodelen = 20;
	self->nullentry = NULL;
	self->rust_ext_compat = 1;
	self->format_version = format_v1;

	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "OO|l", kwlist,
	                                 &data_obj, &inlined_obj,
	                                 &(self->format_version)))
		return -1;
	if (!PyObject_CheckBuffer(data_obj)) {
		PyErr_SetString(PyExc_TypeError,
		                "data does not support buffer interface");
		return -1;

	if (self->nodelen < 20 || self->nodelen > (Py_ssize_t)sizeof(nullid)) {
		PyErr_SetString(PyExc_RuntimeError, "unsupported node size");
		return -1;
	}

	if (self->format_version == format_v1) {
		self->entry_size = v1_entry_size;
	} else if (self->format_version == format_v2) {
		self->entry_size = v2_entry_size;
	} else if (self->format_version == format_cl2) {
		self->entry_size = cl2_entry_size;
	}

	self->nullentry =
	    Py_BuildValue(PY23("iiiiiiis#iiBBi", "iiiiiiiy#iiBBi"), 0, 0, 0, -1,
	                  -1, -1, -1, nullid, self->nodelen, 0, 0,