diffhelpers: harden testhunk

mercurial/diffhelpers.c

@@ -1,190 +1,190 b''
 /*
  * diffhelpers.c - helper routines for mpatch
  *
  * Copyright 2007 Chris Mason <[e-mail]>
  *
  * This software may be used and distributed according to the terms
  * of the GNU General Public License v2, incorporated herein by reference.
  */
 
 #include <Python.h>
 #include <stdlib.h>
 #include <string.h>
 
 #include "util.h"
 
 static char diffhelpers_doc[] = "Efficient diff parsing";
 static PyObject *diffhelpers_Error;
 
 
 /* fixup the last lines of a and b when the patch has no newline at eof */
 static void _fix_newline(PyObject *hunk, PyObject *a, PyObject *b)
 {
 	int hunksz = PyList_Size(hunk);
 	PyObject *s = PyList_GET_ITEM(hunk, hunksz-1);
 	char *l = PyBytes_AsString(s);
 	int alen = PyList_Size(a);
 	int blen = PyList_Size(b);
 	char c = l[0];
 	PyObject *hline;
 	int sz = PyBytes_GET_SIZE(s);
 
 	if (sz > 1 && l[sz-2] == '\r')
 		/* tolerate CRLF in last line */
 		sz -= 1;
 
 	hline = PyBytes_FromStringAndSize(l, sz-1);
 
 	if (c == ' ' || c == '+') {
 		PyObject *rline = PyBytes_FromStringAndSize(l + 1, sz - 2);
 		PyList_SetItem(b, blen-1, rline);
 	}
 	if (c == ' ' || c == '-') {
 		Py_INCREF(hline);
 		PyList_SetItem(a, alen-1, hline);
 	}
 	PyList_SetItem(hunk, hunksz-1, hline);
 }
 
 /* python callable form of _fix_newline */
 static PyObject *
 fix_newline(PyObject *self, PyObject *args)
 {
 	PyObject *hunk, *a, *b;
 	if (!PyArg_ParseTuple(args, "OOO", &hunk, &a, &b))
 		return NULL;
 	_fix_newline(hunk, a, b);
 	return Py_BuildValue("l", 0);
 }
 
 /*
  * read lines from fp into the hunk.  The hunk is parsed into two arrays
  * a and b.  a gets the old state of the text, b gets the new state
  * The control char from the hunk is saved when inserting into a, but not b
  * (for performance while deleting files)
  */
 static PyObject *
 addlines(PyObject *self, PyObject *args)
 {
 
 	PyObject *fp, *hunk, *a, *b, *x;
 	int i;
 	int lena, lenb;
 	int num;
 	int todoa, todob;
 	char *s, c;
 	PyObject *l;
 	if (!PyArg_ParseTuple(args, "OOiiOO", &fp, &hunk, &lena, &lenb, &a, &b))
 		return NULL;
 
 	while (1) {
 		todoa = lena - PyList_Size(a);
 		todob = lenb - PyList_Size(b);
 		num = todoa > todob ? todoa : todob;
 		if (num == 0)
 		    break;
 		for (i = 0; i < num; i++) {
 			x = PyFile_GetLine(fp, 0);
 			s = PyBytes_AsString(x);
 			c = *s;
 			if (strcmp(s, "\\ No newline at end of file\n") == 0) {
 				_fix_newline(hunk, a, b);
 				continue;
 			}
 			if (c == '\n') {
 				/* Some patches may be missing the control char
 				 * on empty lines. Supply a leading space. */
 				Py_DECREF(x);
 				x = PyBytes_FromString(" \n");
 			}
 			PyList_Append(hunk, x);
 			if (c == '+') {
 				l = PyBytes_FromString(s + 1);
 				PyList_Append(b, l);
 				Py_DECREF(l);
 			} else if (c == '-') {
 				PyList_Append(a, x);
 			} else {
 				l = PyBytes_FromString(s + 1);
 				PyList_Append(b, l);
 				Py_DECREF(l);
 				PyList_Append(a, x);
 			}
 			Py_DECREF(x);
 		}
 	}
 	return Py_BuildValue("l", 0);
 }
 
 /*
  * compare the lines in a with the lines in b.  a is assumed to have
  * a control char at the start of each line, this char is ignored in the
  * compare
  */
 static PyObject *
 testhunk(PyObject *self, PyObject *args)
 {
 
 	PyObject *a, *b;
 	long bstart;
 	int alen, blen;
 	int i;
 	char *sa, *sb;
 
 	if (!PyArg_ParseTuple(args, "OOl", &a, &b, &bstart))
 		return NULL;
 	alen = PyList_Size(a);
 	blen = PyList_Size(b);
-	if (alen > blen - bstart) {
+	if (alen > blen - bstart || bstart < 0) {
 		return Py_BuildValue("l", -1);
 	}
 	for (i = 0; i < alen; i++) {
 		sa = PyBytes_AsString(PyList_GET_ITEM(a, i));
 		sb = PyBytes_AsString(PyList_GET_ITEM(b, i + bstart));
 		if (strcmp(sa + 1, sb) != 0)
 			return Py_BuildValue("l", -1);
 	}
 	return Py_BuildValue("l", 0);
 }
 
 static PyMethodDef methods[] = {
 	{"addlines", addlines, METH_VARARGS, "add lines to a hunk\n"},
 	{"fix_newline", fix_newline, METH_VARARGS, "fixup newline counters\n"},
 	{"testhunk", testhunk, METH_VARARGS, "test lines in a hunk\n"},
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef diffhelpers_module = {
 	PyModuleDef_HEAD_INIT,
 	"diffhelpers",
 	diffhelpers_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_diffhelpers(void)
 {
 	PyObject *m;
 
 	m = PyModule_Create(&diffhelpers_module);
 	if (m == NULL)
 		return NULL;
 
 	diffhelpers_Error = PyErr_NewException("diffhelpers.diffhelpersError",
 											NULL, NULL);
 	Py_INCREF(diffhelpers_Error);
 	PyModule_AddObject(m, "diffhelpersError", diffhelpers_Error);
 
 	return m;
 }
 #else
 PyMODINIT_FUNC
 initdiffhelpers(void)
 {
 	Py_InitModule3("diffhelpers", methods, diffhelpers_doc);
 	diffhelpers_Error = PyErr_NewException("diffhelpers.diffhelpersError",
 	                                        NULL, NULL);
 }
 #endif