specify C indention style using Emacs file local variables

contrib/hgsh/hgsh.c

@@ -1,440 +1,445 b''
 /*
  * hgsh.c - restricted login shell for mercurial
  *
  * Copyright 2006 Vadim Gelfer <[e-mail]>
  *
  * This software may be used and distributed according to the terms of the
  * GNU General Public License, incorporated herein by reference.
  *
  * this program is login shell for dedicated mercurial user account. it
  * only allows few actions:
  *
  * 1. run hg in server mode on specific repository. no other hg commands
  * are allowed. we try to verify that repo to be accessed exists under
  * given top-level directory.
  *
  * 2. (optional) forward ssh connection from firewall/gateway machine to
  * "real" mercurial host, to let users outside intranet pull and push
  * changes through firewall.
  *
  * 3. (optional) run normal shell, to allow to "su" to mercurial user, use
  * "sudo" to run programs as that user, or run cron jobs as that user.
  *
  * only tested on linux yet. patches for non-linux systems welcome.
  */
 
 #ifndef _GNU_SOURCE
 #define _GNU_SOURCE /* for asprintf */
 #endif
 
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <sysexits.h>
 #include <unistd.h>
 
 /*
  * user config.
  *
  * if you see a hostname below, just use first part of hostname. example,
  * if you have host named foo.bar.com, use "foo".
  */
 
 /*
  * HG_GATEWAY: hostname of gateway/firewall machine that people outside your
  * intranet ssh into if they need to ssh to other machines. if you do not
  * have such machine, set to NULL.
  */
 #ifndef HG_GATEWAY
 #define HG_GATEWAY     "gateway"
 #endif
 
 /*
  * HG_HOST: hostname of mercurial server. if any machine is allowed, set to
  * NULL.
  */
 #ifndef HG_HOST
 #define HG_HOST         "mercurial"
 #endif
 
 /*
  * HG_USER: username to log in from HG_GATEWAY to HG_HOST. if gateway and
  * host username are same, set to NULL.
  */
 #ifndef HG_USER
 #define HG_USER         "hg"
 #endif
 
 /*
  * HG_ROOT: root of tree full of mercurial repos. if you do not want to
  * validate location of repo when someone is try to access, set to NULL.
  */
 #ifndef HG_ROOT
 #define HG_ROOT         "/home/hg/repos"
 #endif
 
 /*
  * HG: path to the mercurial executable to run.
  */
 #ifndef HG
 #define HG              "/home/hg/bin/hg"
 #endif
 
 /*
  * HG_SHELL: shell to use for actions like "sudo" and "su" access to
  * mercurial user, and cron jobs. if you want to make these things
  * impossible, set to NULL.
  */
 #ifndef HG_SHELL
 #define HG_SHELL        NULL
 /* #define HG_SHELL        "/bin/bash" */
 #endif
 
 /*
  * HG_HELP: some way for users to get support if they have problem. if they
  * should not get helpful message, set to NULL.
  */
 #ifndef HG_HELP
 #define HG_HELP         "please contact [e-mail] for help."
 #endif
 
 /*
  * SSH: path to ssh executable to run, if forwarding from HG_GATEWAY to
  * HG_HOST. if you want to use rsh instead (why?), you need to modify
  * arguments it is called with. see forward_through_gateway.
  */
 #ifndef SSH
 #define SSH             "/usr/bin/ssh"
 #endif
 
 /*
  * tell whether to print command that is to be executed. useful for
  * debugging. should not interfere with mercurial operation, since
  * mercurial only cares about stdin and stdout, and this prints to stderr.
  */
 static const int debug = 0;
 
 static void print_cmdline(int argc, char **argv)
 {
 	FILE *fp = stderr;
 	int i;
 
 	fputs("command: ", fp);
 
 	for (i = 0; i < argc; i++) {
 		char *spc = strpbrk(argv[i], " \t\r\n");
 		if (spc) {
 			fputc('\'', fp);
 		}
 		fputs(argv[i], fp);
 		if (spc) {
 			fputc('\'', fp);
 		}
 		if (i < argc - 1) {
 			fputc(' ', fp);
 		}
 	}
 	fputc('\n', fp);
 	fflush(fp);
 }
 
 static void usage(const char *reason, int exitcode)
 {
 	char *hg_help = HG_HELP;
 
 	if (reason) {
 		fprintf(stderr, "*** Error: %s.\n", reason);
 	}
 	fprintf(stderr, "*** This program has been invoked incorrectly.\n");
 	if (hg_help) {
 		fprintf(stderr, "*** %s\n", hg_help);
 	}
 	exit(exitcode ? exitcode : EX_USAGE);
 }
 
 /*
  * run on gateway host to make another ssh connection, to "real" mercurial
  * server. it sends its command line unmodified to far end.
  *
  * never called if HG_GATEWAY is NULL.
  */
 static void forward_through_gateway(int argc, char **argv)
 {
 	char *ssh = SSH;
 	char *hg_host = HG_HOST;
 	char *hg_user = HG_USER;
 	char **nargv = alloca((10 + argc) * sizeof(char *));
 	int i = 0, j;
 
 	nargv[i++] = ssh;
 	nargv[i++] = "-q";
 	nargv[i++] = "-T";
 	nargv[i++] = "-x";
 	if (hg_user) {
 		nargv[i++] = "-l";
 		nargv[i++] = hg_user;
 	}
 	nargv[i++] = hg_host;
 
 	/*
 	 * sshd called us with added "-c", because it thinks we are a shell.
 	 * drop it if we find it.
 	 */
 	j = 1;
 	if (j < argc && strcmp(argv[j], "-c") == 0) {
 		j++;
 	}
 
 	for (; j < argc; i++, j++) {
 		nargv[i] = argv[j];
 	}
 	nargv[i] = NULL;
 
 	if (debug) {
 		print_cmdline(i, nargv);
 	}
 
 	execv(ssh, nargv);
 	perror(ssh);
 	exit(EX_UNAVAILABLE);
 }
 
 /*
  * run shell. let administrator "su" to mercurial user's account to do
  * administrative works.
  *
  * never called if HG_SHELL is NULL.
  */
 static void run_shell(int argc, char **argv)
 {
 	char *hg_shell = HG_SHELL;
 	char **nargv;
 	char *c;
 	int i;
 
 	nargv = alloca((argc + 3) * sizeof(char *));
 	c = strrchr(hg_shell, '/');
 
 	/* tell "real" shell it is login shell, if needed. */
 
 	if (argv[0][0] == '-' && c) {
 		nargv[0] = strdup(c);
 		if (nargv[0] == NULL) {
 			perror("malloc");
 			exit(EX_OSERR);
 		}
 		nargv[0][0] = '-';
 	} else {
 		nargv[0] = hg_shell;
 	}
 
 	for (i = 1; i < argc; i++) {
 		nargv[i] = argv[i];
 	}
 	nargv[i] = NULL;
 
 	if (debug) {
 		print_cmdline(i, nargv);
 	}
 
 	execv(hg_shell, nargv);
 	perror(hg_shell);
 	exit(EX_OSFILE);
 }
 
 enum cmdline {
 	hg_init,
 	hg_serve,
 };
 
 
 /*
  * attempt to verify that a directory is really a hg repo, by testing
  * for the existence of a subdirectory.
  */
 static int validate_repo(const char *repo_root, const char *subdir)
 {
 	char *abs_path;
 	struct stat st;
 	int ret;
 
 	if (asprintf(&abs_path, "%s.hg/%s", repo_root, subdir) == -1) {
 		ret = -1;
 		goto bail;
 	}
 
 	/* verify that we really are looking at valid repo. */
 
 	if (stat(abs_path, &st) == -1) {
 		ret = 0;
 	} else {
 		ret = 1;
 	}
 
 bail:
 	return ret;
 }
 
 /*
  * paranoid wrapper, runs hg executable in server mode.
  */
 static void serve_data(int argc, char **argv)
 {
 	char *hg_root = HG_ROOT;
 	char *repo, *repo_root;
 	enum cmdline cmd;
 	char *nargv[6];
 	size_t repolen;
 	int i;
 
 	/*
 	 * check argv for looking okay. we should be invoked with argv
 	 * resembling like this:
 	 *
 	 *   hgsh
 	 *   -c
 	 *   hg -R some/path serve --stdio
 	 *
 	 * the "-c" is added by sshd, because it thinks we are login shell.
 	 */
 
 	if (argc != 3) {
 		goto badargs;
 	}
 
 	if (strcmp(argv[1], "-c") != 0) {
 		goto badargs;
 	}
 
 	if (sscanf(argv[2], "hg init %as", &repo) == 1) {
 		cmd = hg_init;
 	}
 	else if (sscanf(argv[2], "hg -R %as serve --stdio", &repo) == 1) {
 		cmd = hg_serve;
 	} else {
 		goto badargs;
 	}
 
 	repolen = repo ? strlen(repo) : 0;
 
 	if (repolen == 0) {
 		goto badargs;
 	}
 
 	if (hg_root) {
 		if (asprintf(&repo_root, "%s/%s/", hg_root, repo) == -1) {
 			goto badargs;
 		}
 
 		/*
 		 * attempt to stop break out from inside the
 		 * repository tree. could do something more clever
 		 * here, because e.g. we could traverse a symlink that
 		 * looks safe, but really breaks us out of tree.
 		 */
 
 		if (strstr(repo_root, "/../") != NULL) {
 			goto badargs;
 		}
 
 		/* only hg init expects no repo. */
 
 		if (cmd != hg_init) {
 			int valid;
 
 			valid = validate_repo(repo_root, "data");
 
 			if (valid == -1) {
 				goto badargs;
 			}
 
 			if (valid == 0) {
 				valid = validate_repo(repo_root, "store");
 
 				if (valid == -1) {
 					goto badargs;
 				}
 			}
 
 			if (valid == 0) {
 				perror(repo);
 				exit(EX_DATAERR);
 			}
 		}
 
 		if (chdir(hg_root) == -1) {
 			perror(hg_root);
 			exit(EX_SOFTWARE);
 		}
 	}
 
 	i = 0;
 
 	switch (cmd) {
 	case hg_serve:
 		nargv[i++] = HG;
 		nargv[i++] = "-R";
 		nargv[i++] = repo;
 		nargv[i++] = "serve";
 		nargv[i++] = "--stdio";
 		break;
 	case hg_init:
 		nargv[i++] = HG;
 		nargv[i++] = "init";
 		nargv[i++] = repo;
 		break;
 	}
 
 	nargv[i] = NULL;
 
 	if (debug) {
 		print_cmdline(i, nargv);
 	}
 
 	execv(HG, nargv);
 	perror(HG);
 	exit(EX_UNAVAILABLE);
 
 badargs:
 	/* print useless error message. */
 
 	usage("invalid arguments", EX_DATAERR);
 }
 
 int main(int argc, char **argv)
 {
 	char host[1024];
 	char *c;
 
 	if (gethostname(host, sizeof(host)) == -1) {
 		perror("gethostname");
 		exit(EX_OSERR);
 	}
 
 	if ((c = strchr(host, '.')) != NULL) {
 		*c = '\0';
 	}
 
 	if (getenv("SSH_CLIENT")) {
 		char *hg_gateway = HG_GATEWAY;
 		char *hg_host = HG_HOST;
 
 		if (hg_gateway && strcmp(host, hg_gateway) == 0) {
 			forward_through_gateway(argc, argv);
 		}
 
 		if (hg_host && strcmp(host, hg_host) != 0) {
 			usage("invoked on unexpected host", EX_USAGE);
 		}
 
 		serve_data(argc, argv);
 	} else if (HG_SHELL) {
 		run_shell(argc, argv);
 	} else {
 		usage("invalid arguments", EX_DATAERR);
 	}
 
 	return 0;
 }
+
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

hgext/inotify/linux/_inotify.c

@@ -1,649 +1,654 b''
 /*
  * _inotify.c - Python extension interfacing to the Linux inotify subsystem
  *
  * Copyright 2006 Bryan O'Sullivan <[e-mail]>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of version 2.1 of the GNU Lesser General
  * Public License or any later version.
  */
 
 #include <Python.h>
 #include <alloca.h>
 #include <sys/inotify.h>
 #include <stdint.h>
 #include <sys/ioctl.h>
 #include <unistd.h>
 
 #include <util.h>
 
 /* Variables used in the event string representation */
 static PyObject *join;
 static PyObject *er_wm;
 static PyObject *er_wmc;
 static PyObject *er_wmn;
 static PyObject *er_wmcn;
 
 static PyObject *init(PyObject *self, PyObject *args)
 {
 	PyObject *ret = NULL;
 	int fd = -1;
 
 	if (!PyArg_ParseTuple(args, ":init"))
 		goto bail;
 
 	Py_BEGIN_ALLOW_THREADS;
 	fd = inotify_init();
 	Py_END_ALLOW_THREADS;
 
 	if (fd == -1) {
 		PyErr_SetFromErrno(PyExc_OSError);
 		goto bail;
 	}
 
 	ret = PyInt_FromLong(fd);
 	if (ret == NULL)
 		goto bail;
 
 	goto done;
 
 bail:
 	if (fd != -1)
 		close(fd);
 
 	Py_CLEAR(ret);
 
 done:
 	return ret;
 }
 
 PyDoc_STRVAR(
 	init_doc,
 	"init() -> fd\n"
 	"\n"
 	"Initialise an inotify instance.\n"
 	"Return a file descriptor associated with a new inotify event queue.");
 
 static PyObject *add_watch(PyObject *self, PyObject *args)
 {
 	PyObject *ret = NULL;
 	uint32_t mask;
 	int wd = -1;
 	char *path;
 	int fd;
 
 	if (!PyArg_ParseTuple(args, "isI:add_watch", &fd, &path, &mask))
 		goto bail;
 
 	Py_BEGIN_ALLOW_THREADS;
 	wd = inotify_add_watch(fd, path, mask);
 	Py_END_ALLOW_THREADS;
 
 	if (wd == -1) {
 		PyErr_SetFromErrnoWithFilename(PyExc_OSError, path);
 		goto bail;
 	}
 
 	ret = PyInt_FromLong(wd);
 	if (ret == NULL)
 		goto bail;
 
 	goto done;
 
 bail:
 	if (wd != -1)
 		inotify_rm_watch(fd, wd);
 
 	Py_CLEAR(ret);
 
 done:
 	return ret;
 }
 
 PyDoc_STRVAR(
 	add_watch_doc,
 	"add_watch(fd, path, mask) -> wd\n"
 	"\n"
 	"Add a watch to an inotify instance, or modify an existing watch.\n"
 	"\n"
 	"        fd: file descriptor returned by init()\n"
 	"        path: path to watch\n"
 	"        mask: mask of events to watch for\n"
 	"\n"
 	"Return a unique numeric watch descriptor for the inotify instance\n"
 	"mapped by the file descriptor.");
 
 static PyObject *remove_watch(PyObject *self, PyObject *args)
 {
 	uint32_t wd;
 	int fd;
 	int r;
 
 	if (!PyArg_ParseTuple(args, "iI:remove_watch", &fd, &wd))
 		return NULL;
 
 	Py_BEGIN_ALLOW_THREADS;
 	r = inotify_rm_watch(fd, wd);
 	Py_END_ALLOW_THREADS;
 
 	if (r == -1) {
 		PyErr_SetFromErrno(PyExc_OSError);
 		return NULL;
 	}
 
 	Py_INCREF(Py_None);
 	return Py_None;
 }
 
 PyDoc_STRVAR(
 	remove_watch_doc,
 	"remove_watch(fd, wd)\n"
 	"\n"
 	"        fd: file descriptor returned by init()\n"
 	"        wd: watch descriptor returned by add_watch()\n"
 	"\n"
 	"Remove a watch associated with the watch descriptor wd from the\n"
 	"inotify instance associated with the file descriptor fd.\n"
 	"\n"
 	"Removing a watch causes an IN_IGNORED event to be generated for this\n"
 	"watch descriptor.");
 
 #define bit_name(x) {x, #x}
 
 static struct {
 	int bit;
 	const char *name;
 	PyObject *pyname;
 } bit_names[] = {
 	bit_name(IN_ACCESS),
 	bit_name(IN_MODIFY),
 	bit_name(IN_ATTRIB),
 	bit_name(IN_CLOSE_WRITE),
 	bit_name(IN_CLOSE_NOWRITE),
 	bit_name(IN_OPEN),
 	bit_name(IN_MOVED_FROM),
 	bit_name(IN_MOVED_TO),
 	bit_name(IN_CREATE),
 	bit_name(IN_DELETE),
 	bit_name(IN_DELETE_SELF),
 	bit_name(IN_MOVE_SELF),
 	bit_name(IN_UNMOUNT),
 	bit_name(IN_Q_OVERFLOW),
 	bit_name(IN_IGNORED),
 	bit_name(IN_ONLYDIR),
 	bit_name(IN_DONT_FOLLOW),
 	bit_name(IN_MASK_ADD),
 	bit_name(IN_ISDIR),
 	bit_name(IN_ONESHOT),
 	{0}
 };
 
 static PyObject *decode_mask(int mask)
 {
 	PyObject *ret = PyList_New(0);
 	int i;
 
 	if (ret == NULL)
 		goto bail;
 
 	for (i = 0; bit_names[i].bit; i++) {
 		if (mask & bit_names[i].bit) {
 			if (bit_names[i].pyname == NULL) {
 				bit_names[i].pyname = PyString_FromString(bit_names[i].name);
 				if (bit_names[i].pyname == NULL)
 					goto bail;
 			}
 			Py_INCREF(bit_names[i].pyname);
 			if (PyList_Append(ret, bit_names[i].pyname) == -1)
 				goto bail;
 		}
 	}
 
 	goto done;
 
 bail:
 	Py_CLEAR(ret);
 
 done:
 	return ret;
 }
 
 static PyObject *pydecode_mask(PyObject *self, PyObject *args)
 {
 	int mask;
 
 	if (!PyArg_ParseTuple(args, "i:decode_mask", &mask))
 		return NULL;
 
 	return decode_mask(mask);
 }
 
 PyDoc_STRVAR(
 	decode_mask_doc,
 	"decode_mask(mask) -> list_of_strings\n"
 	"\n"
 	"Decode an inotify mask value into a list of strings that give the\n"
 	"name of each bit set in the mask.");
 
 static char doc[] = "Low-level inotify interface wrappers.";
 
 static void define_const(PyObject *dict, const char *name, uint32_t val)
 {
 	PyObject *pyval = PyInt_FromLong(val);
 	PyObject *pyname = PyString_FromString(name);
 
 	if (!pyname || !pyval)
 		goto bail;
 
 	PyDict_SetItem(dict, pyname, pyval);
 
 bail:
 	Py_XDECREF(pyname);
 	Py_XDECREF(pyval);
 }
 
 static void define_consts(PyObject *dict)
 {
 	define_const(dict, "IN_ACCESS", IN_ACCESS);
 	define_const(dict, "IN_MODIFY", IN_MODIFY);
 	define_const(dict, "IN_ATTRIB", IN_ATTRIB);
 	define_const(dict, "IN_CLOSE_WRITE", IN_CLOSE_WRITE);
 	define_const(dict, "IN_CLOSE_NOWRITE", IN_CLOSE_NOWRITE);
 	define_const(dict, "IN_OPEN", IN_OPEN);
 	define_const(dict, "IN_MOVED_FROM", IN_MOVED_FROM);
 	define_const(dict, "IN_MOVED_TO", IN_MOVED_TO);
 
 	define_const(dict, "IN_CLOSE", IN_CLOSE);
 	define_const(dict, "IN_MOVE", IN_MOVE);
 
 	define_const(dict, "IN_CREATE", IN_CREATE);
 	define_const(dict, "IN_DELETE", IN_DELETE);
 	define_const(dict, "IN_DELETE_SELF", IN_DELETE_SELF);
 	define_const(dict, "IN_MOVE_SELF", IN_MOVE_SELF);
 	define_const(dict, "IN_UNMOUNT", IN_UNMOUNT);
 	define_const(dict, "IN_Q_OVERFLOW", IN_Q_OVERFLOW);
 	define_const(dict, "IN_IGNORED", IN_IGNORED);
 
 	define_const(dict, "IN_ONLYDIR", IN_ONLYDIR);
 	define_const(dict, "IN_DONT_FOLLOW", IN_DONT_FOLLOW);
 	define_const(dict, "IN_MASK_ADD", IN_MASK_ADD);
 	define_const(dict, "IN_ISDIR", IN_ISDIR);
 	define_const(dict, "IN_ONESHOT", IN_ONESHOT);
 	define_const(dict, "IN_ALL_EVENTS", IN_ALL_EVENTS);
 }
 
 struct event {
 	PyObject_HEAD
 	PyObject *wd;
 	PyObject *mask;
 	PyObject *cookie;
 	PyObject *name;
 };
 
 static PyObject *event_wd(PyObject *self, void *x)
 {
 	struct event *evt = (struct event *)self;
 	Py_INCREF(evt->wd);
 	return evt->wd;
 }
 
 static PyObject *event_mask(PyObject *self, void *x)
 {
 	struct event *evt = (struct event *)self;
 	Py_INCREF(evt->mask);
 	return evt->mask;
 }
 
 static PyObject *event_cookie(PyObject *self, void *x)
 {
 	struct event *evt = (struct event *)self;
 	Py_INCREF(evt->cookie);
 	return evt->cookie;
 }
 
 static PyObject *event_name(PyObject *self, void *x)
 {
 	struct event *evt = (struct event *)self;
 	Py_INCREF(evt->name);
 	return evt->name;
 }
 
 static struct PyGetSetDef event_getsets[] = {
 	{"wd", event_wd, NULL,
 	 "watch descriptor"},
 	{"mask", event_mask, NULL,
 	 "event mask"},
 	{"cookie", event_cookie, NULL,
 	 "rename cookie, if rename-related event"},
 	{"name", event_name, NULL,
 	 "file name"},
 	{NULL}
 };
 
 PyDoc_STRVAR(
 	event_doc,
 	"event: Structure describing an inotify event.");
 
 static PyObject *event_new(PyTypeObject *t, PyObject *a, PyObject *k)
 {
 	return (*t->tp_alloc)(t, 0);
 }
 
 static void event_dealloc(struct event *evt)
 {
 	Py_XDECREF(evt->wd);
 	Py_XDECREF(evt->mask);
 	Py_XDECREF(evt->cookie);
 	Py_XDECREF(evt->name);
 
 	Py_TYPE(evt)->tp_free(evt);
 }
 
 static PyObject *event_repr(struct event *evt)
 {
 	int cookie = evt->cookie == Py_None ? -1 : PyInt_AsLong(evt->cookie);
 	PyObject *ret = NULL, *pymasks = NULL, *pymask = NULL;
 	PyObject *tuple = NULL, *formatstr = NULL;
 
 	pymasks = decode_mask(PyInt_AsLong(evt->mask));
 	if (pymasks == NULL)
 		goto bail;
 
 	pymask = _PyString_Join(join, pymasks);
 	if (pymask == NULL)
 		goto bail;
 
 	if (evt->name != Py_None) {
 		if (cookie == -1) {
 			formatstr = er_wmn;
 			tuple = PyTuple_Pack(3, evt->wd, pymask, evt->name);
 		}
 		else {
 			formatstr = er_wmcn;
 			tuple = PyTuple_Pack(4, evt->wd, pymask,
 					     evt->cookie, evt->name);
 		}
 	} else {
 		if (cookie == -1) {
 			formatstr = er_wm;
 			tuple = PyTuple_Pack(2, evt->wd, pymask);
 		}
 		else {
 			formatstr = er_wmc;
 			tuple = PyTuple_Pack(3, evt->wd, pymask, evt->cookie);
 		}
 	}
 
 	if (tuple == NULL)
 		goto bail;
 
 	ret = PyNumber_Remainder(formatstr, tuple);
 
 	if (ret == NULL)
 		goto bail;
 
 	goto done;
 bail:
 	Py_CLEAR(ret);
 
 done:
 	Py_XDECREF(pymask);
 	Py_XDECREF(pymasks);
 	Py_XDECREF(tuple);
 
 	return ret;
 }
 
 static PyTypeObject event_type = {
 	PyVarObject_HEAD_INIT(NULL, 0)
 	"_inotify.event",             /*tp_name*/
 	sizeof(struct event), /*tp_basicsize*/
 	0,                         /*tp_itemsize*/
 	(destructor)event_dealloc, /*tp_dealloc*/
 	0,                         /*tp_print*/
 	0,                         /*tp_getattr*/
 	0,                         /*tp_setattr*/
 	0,                         /*tp_compare*/
 	(reprfunc)event_repr,      /*tp_repr*/
 	0,                         /*tp_as_number*/
 	0,                         /*tp_as_sequence*/
 	0,                         /*tp_as_mapping*/
 	0,                         /*tp_hash */
 	0,                         /*tp_call*/
 	0,                         /*tp_str*/
 	0,                         /*tp_getattro*/
 	0,                         /*tp_setattro*/
 	0,                         /*tp_as_buffer*/
 	Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
 	event_doc,           /* tp_doc */
 	0,                         /* tp_traverse */
 	0,                         /* tp_clear */
 	0,                         /* tp_richcompare */
 	0,                         /* tp_weaklistoffset */
 	0,                         /* tp_iter */
 	0,                         /* tp_iternext */
 	0,                         /* tp_methods */
 	0,                         /* tp_members */
 	event_getsets,      /* tp_getset */
 	0,                         /* tp_base */
 	0,                         /* tp_dict */
 	0,                         /* tp_descr_get */
 	0,                         /* tp_descr_set */
 	0,                         /* tp_dictoffset */
 	0,                         /* tp_init */
 	0,                         /* tp_alloc */
 	event_new,          /* tp_new */
 };
 
 PyObject *read_events(PyObject *self, PyObject *args)
 {
 	PyObject *ctor_args = NULL;
 	PyObject *pybufsize = NULL;
 	PyObject *ret = NULL;
 	int bufsize = 65536;
 	char *buf = NULL;
 	int nread, pos;
 	int fd;
 
 	if (!PyArg_ParseTuple(args, "i|O:read", &fd, &pybufsize))
 		goto bail;
 
 	if (pybufsize && pybufsize != Py_None)
 		bufsize = PyInt_AsLong(pybufsize);
 
 	ret = PyList_New(0);
 	if (ret == NULL)
 		goto bail;
 
 	if (bufsize <= 0) {
 		int r;
 
 		Py_BEGIN_ALLOW_THREADS;
 		r = ioctl(fd, FIONREAD, &bufsize);
 		Py_END_ALLOW_THREADS;
 
 		if (r == -1) {
 			PyErr_SetFromErrno(PyExc_OSError);
 			goto bail;
 		}
 		if (bufsize == 0)
 			goto done;
 	}
 	else {
 		static long name_max;
 		static long name_fd = -1;
 		long min;
 
 		if (name_fd != fd) {
 			name_fd = fd;
 			Py_BEGIN_ALLOW_THREADS;
 			name_max = fpathconf(fd, _PC_NAME_MAX);
 			Py_END_ALLOW_THREADS;
 		}
 
 		min = sizeof(struct inotify_event) + name_max + 1;
 
 		if (bufsize < min) {
 			PyErr_Format(PyExc_ValueError,
 				     "bufsize must be at least %d", (int)min);
 			goto bail;
 		}
 	}
 
 	buf = alloca(bufsize);
 
 	Py_BEGIN_ALLOW_THREADS;
 	nread = read(fd, buf, bufsize);
 	Py_END_ALLOW_THREADS;
 
 	if (nread == -1) {
 		PyErr_SetFromErrno(PyExc_OSError);
 		goto bail;
 	}
 
 	ctor_args = PyTuple_New(0);
 
 	if (ctor_args == NULL)
 		goto bail;
 
 	pos = 0;
 
 	while (pos < nread) {
 		struct inotify_event *in = (struct inotify_event *)(buf + pos);
 		struct event *evt;
 		PyObject *obj;
 
 		obj = PyObject_CallObject((PyObject *)&event_type, ctor_args);
 
 		if (obj == NULL)
 			goto bail;
 
 		evt = (struct event *)obj;
 
 		evt->wd = PyInt_FromLong(in->wd);
 		evt->mask = PyInt_FromLong(in->mask);
 		if (in->mask & IN_MOVE)
 			evt->cookie = PyInt_FromLong(in->cookie);
 		else {
 			Py_INCREF(Py_None);
 			evt->cookie = Py_None;
 		}
 		if (in->len)
 			evt->name = PyString_FromString(in->name);
 		else {
 			Py_INCREF(Py_None);
 			evt->name = Py_None;
 		}
 
 		if (!evt->wd || !evt->mask || !evt->cookie || !evt->name)
 			goto mybail;
 
 		if (PyList_Append(ret, obj) == -1)
 			goto mybail;
 
 		pos += sizeof(struct inotify_event) + in->len;
 		continue;
 
 	mybail:
 		Py_CLEAR(evt->wd);
 		Py_CLEAR(evt->mask);
 		Py_CLEAR(evt->cookie);
 		Py_CLEAR(evt->name);
 		Py_DECREF(obj);
 
 		goto bail;
 	}
 
 	goto done;
 
 bail:
 	Py_CLEAR(ret);
 
 done:
 	Py_XDECREF(ctor_args);
 
 	return ret;
 }
 
 static int init_globals(void)
 {
 	join = PyString_FromString("|");
 	er_wm = PyString_FromString("event(wd=%d, mask=%s)");
 	er_wmn = PyString_FromString("event(wd=%d, mask=%s, name=%s)");
 	er_wmc = PyString_FromString("event(wd=%d, mask=%s, cookie=0x%x)");
 	er_wmcn = PyString_FromString("event(wd=%d, mask=%s, cookie=0x%x, name=%s)");
 
 	return join && er_wm && er_wmn && er_wmc && er_wmcn;
 }
 
 PyDoc_STRVAR(
 	read_doc,
 	"read(fd, bufsize[=65536]) -> list_of_events\n"
 	"\n"
 	"\nRead inotify events from a file descriptor.\n"
 	"\n"
 	"        fd: file descriptor returned by init()\n"
 	"        bufsize: size of buffer to read into, in bytes\n"
 	"\n"
 	"Return a list of event objects.\n"
 	"\n"
 	"If bufsize is > 0, block until events are available to be read.\n"
 	"Otherwise, immediately return all events that can be read without\n"
 	"blocking.");
 
 static PyMethodDef methods[] = {
 	{"init", init, METH_VARARGS, init_doc},
 	{"add_watch", add_watch, METH_VARARGS, add_watch_doc},
 	{"remove_watch", remove_watch, METH_VARARGS, remove_watch_doc},
 	{"read", read_events, METH_VARARGS, read_doc},
 	{"decode_mask", pydecode_mask, METH_VARARGS, decode_mask_doc},
 	{NULL},
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef _inotify_module = {
 	PyModuleDef_HEAD_INIT,
 	"_inotify",
 	doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit__inotify(void)
 {
 	PyObject *mod, *dict;
 
 	mod = PyModule_Create(&_inotify_module);
 
 	if (mod == NULL)
 		return NULL;
 
 	if (!init_globals())
 		return;
 
 	dict = PyModule_GetDict(mod);
 
 	if (dict)
 		define_consts(dict);
 
 	return mod;
 }
 #else
 void init_inotify(void)
 {
 	PyObject *mod, *dict;
 
 	if (PyType_Ready(&event_type) == -1)
 		return;
 
 	if (!init_globals())
 		return;
 
 	mod = Py_InitModule3("_inotify", methods, doc);
 
 	dict = PyModule_GetDict(mod);
 
 	if (dict)
 		define_consts(dict);
 }
 #endif
+
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/base85.c

@@ -1,180 +1,185 b''
 /*
  base85 codec
 
  Copyright 2006 Brendan Cully <[e-mail]>
 
  This software may be used and distributed according to the terms of
  the GNU General Public License, incorporated herein by reference.
 
  Largely based on git's implementation
 */
 
 #include <Python.h>
 
 #include "util.h"
 
 static const char b85chars[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
 	"abcdefghijklmnopqrstuvwxyz!#$%&()*+-;<=>?@^_`{|}~";
 static char b85dec[256];
 
 static void
 b85prep(void)
 {
 	int i;
 
 	memset(b85dec, 0, sizeof(b85dec));
 	for (i = 0; i < sizeof(b85chars); i++)
 		b85dec[(int)(b85chars[i])] = i + 1;
 }
 
 static PyObject *
 b85encode(PyObject *self, PyObject *args)
 {
 	const unsigned char *text;
 	PyObject *out;
 	char *dst;
 	int len, olen, i;
 	unsigned int acc, val, ch;
 	int pad = 0;
 
 	if (!PyArg_ParseTuple(args, "s#|i", &text, &len, &pad))
 		return NULL;
 
 	if (pad)
 		olen = ((len + 3) / 4 * 5) - 3;
 	else {
 		olen = len % 4;
 		if (olen)
 			olen++;
 		olen += len / 4 * 5;
 	}
 	if (!(out = PyBytes_FromStringAndSize(NULL, olen + 3)))
 		return NULL;
 
 	dst = PyBytes_AsString(out);
 
 	while (len) {
 		acc = 0;
 		for (i = 24; i >= 0; i -= 8) {
 			ch = *text++;
 			acc |= ch << i;
 			if (--len == 0)
 				break;
 		}
 		for (i = 4; i >= 0; i--) {
 			val = acc % 85;
 			acc /= 85;
 			dst[i] = b85chars[val];
 		}
 		dst += 5;
 	}
 
 	if (!pad)
 		_PyBytes_Resize(&out, olen);
 
 	return out;
 }
 
 static PyObject *
 b85decode(PyObject *self, PyObject *args)
 {
 	PyObject *out;
 	const char *text;
 	char *dst;
 	int len, i, j, olen, c, cap;
 	unsigned int acc;
 
 	if (!PyArg_ParseTuple(args, "s#", &text, &len))
 		return NULL;
 
 	olen = len / 5 * 4;
 	i = len % 5;
 	if (i)
 		olen += i - 1;
 	if (!(out = PyBytes_FromStringAndSize(NULL, olen)))
 		return NULL;
 
 	dst = PyBytes_AsString(out);
 
 	i = 0;
 	while (i < len)
 	{
 		acc = 0;
 		cap = len - i - 1;
 		if (cap > 4)
 			cap = 4;
 		for (j = 0; j < cap; i++, j++)
 		{
 			c = b85dec[(int)*text++] - 1;
 			if (c < 0)
 				return PyErr_Format(
 					PyExc_ValueError,
 					"Bad base85 character at position %d", i);
 			acc = acc * 85 + c;
 		}
 		if (i++ < len)
 		{
 			c = b85dec[(int)*text++] - 1;
 			if (c < 0)
 				return PyErr_Format(
 					PyExc_ValueError,
 					"Bad base85 character at position %d", i);
 			/* overflow detection: 0xffffffff == "|NsC0",
 			 * "|NsC" == 0x03030303 */
 			if (acc > 0x03030303 || (acc *= 85) > 0xffffffff - c)
 				return PyErr_Format(
 					PyExc_ValueError,
 					"Bad base85 sequence at position %d", i);
 			acc += c;
 		}
 
 		cap = olen < 4 ? olen : 4;
 		olen -= cap;
 		for (j = 0; j < 4 - cap; j++)
 			acc *= 85;
 		if (cap && cap < 4)
 			acc += 0xffffff >> (cap - 1) * 8;
 		for (j = 0; j < cap; j++)
 		{
 			acc = (acc << 8) | (acc >> 24);
 			*dst++ = acc;
 		}
 	}
 
 	return out;
 }
 
 static char base85_doc[] = "Base85 Data Encoding";
 
 static PyMethodDef methods[] = {
 	{"b85encode", b85encode, METH_VARARGS,
 	 "Encode text in base85.\n\n"
 	 "If the second parameter is true, pad the result to a multiple of "
 	 "five characters.\n"},
 	{"b85decode", b85decode, METH_VARARGS, "Decode base85 text.\n"},
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef base85_module = {
 	PyModuleDef_HEAD_INIT,
 	"base85",
 	base85_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_base85(void)
 {
 	b85prep();
 
 	return PyModule_Create(&base85_module);
 }
 #else
 PyMODINIT_FUNC initbase85(void)
 {
 	Py_InitModule3("base85", methods, base85_doc);
 
 	b85prep();
 }
 #endif
+
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/bdiff.c

@@ -1,453 +1,457 b''
 /*
  bdiff.c - efficient binary diff extension for Mercurial
 
  Copyright 2005, 2006 Matt Mackall <[e-mail]>
 
  This software may be used and distributed according to the terms of
  the GNU General Public License, incorporated herein by reference.
 
  Based roughly on Python difflib
 */
 
 #include <Python.h>
 #include <stdlib.h>
 #include <string.h>
 #include <limits.h>
 
 #if defined __hpux || defined __SUNPRO_C || defined _AIX
 #define inline
 #endif
 
 #ifdef __linux
 #define inline __inline
 #endif
 
 #ifdef _WIN32
 #ifdef _MSC_VER
 #define inline __inline
 typedef unsigned long uint32_t;
 #else
 #include <stdint.h>
 #endif
 static uint32_t htonl(uint32_t x)
 {
 	return ((x & 0x000000ffUL) << 24) |
 		((x & 0x0000ff00UL) <<  8) |
 		((x & 0x00ff0000UL) >>  8) |
 		((x & 0xff000000UL) >> 24);
 }
 #else
 #include <sys/types.h>
 #if defined __BEOS__ && !defined __HAIKU__
 #include <ByteOrder.h>
 #else
 #include <arpa/inet.h>
 #endif
 #include <inttypes.h>
 #endif
 
 #include "util.h"
 
 struct line {
 	int h, len, n, e;
 	const char *l;
 };
 
 struct pos {
 	int pos, len;
 };
 
 struct hunk;
 struct hunk {
 	int a1, a2, b1, b2;
 	struct hunk *next;
 };
 
 int splitlines(const char *a, int len, struct line **lr)
 {
 	int h, i;
 	const char *p, *b = a;
 	const char * const plast = a + len - 1;
 	struct line *l;
 
 	/* count the lines */
 	i = 1; /* extra line for sentinel */
 	for (p = a; p < a + len; p++)
 		if (*p == '\n' || p == plast)
 			i++;
 
 	*lr = l = (struct line *)malloc(sizeof(struct line) * i);
 	if (!l)
 		return -1;
 
 	/* build the line array and calculate hashes */
 	h = 0;
 	for (p = a; p < a + len; p++) {
 		/* Leonid Yuriev's hash */
 		h = (h * 1664525) + *p + 1013904223;
 
 		if (*p == '\n' || p == plast) {
 			l->h = h;
 			h = 0;
 			l->len = p - b + 1;
 			l->l = b;
 			l->n = INT_MAX;
 			l++;
 			b = p + 1;
 		}
 	}
 
 	/* set up a sentinel */
 	l->h = l->len = 0;
 	l->l = a + len;
 	return i - 1;
 }
 
 int inline cmp(struct line *a, struct line *b)
 {
 	return a->h != b->h || a->len != b->len || memcmp(a->l, b->l, a->len);
 }
 
 static int equatelines(struct line *a, int an, struct line *b, int bn)
 {
 	int i, j, buckets = 1, t, scale;
 	struct pos *h = NULL;
 
 	/* build a hash table of the next highest power of 2 */
 	while (buckets < bn + 1)
 		buckets *= 2;
 
 	/* try to allocate a large hash table to avoid collisions */
 	for (scale = 4; scale; scale /= 2) {
 		h = (struct pos *)malloc(scale * buckets * sizeof(struct pos));
 		if (h)
 			break;
 	}
 
 	if (!h)
 		return 0;
 
 	buckets = buckets * scale - 1;
 
 	/* clear the hash table */
 	for (i = 0; i <= buckets; i++) {
 		h[i].pos = INT_MAX;
 		h[i].len = 0;
 	}
 
 	/* add lines to the hash table chains */
 	for (i = bn - 1; i >= 0; i--) {
 		/* find the equivalence class */
 		for (j = b[i].h & buckets; h[j].pos != INT_MAX;
 		     j = (j + 1) & buckets)
 			if (!cmp(b + i, b + h[j].pos))
 				break;
 
 		/* add to the head of the equivalence class */
 		b[i].n = h[j].pos;
 		b[i].e = j;
 		h[j].pos = i;
 		h[j].len++; /* keep track of popularity */
 	}
 
 	/* compute popularity threshold */
 	t = (bn >= 31000) ? bn / 1000 : 1000000 / (bn + 1);
 
 	/* match items in a to their equivalence class in b */
 	for (i = 0; i < an; i++) {
 		/* find the equivalence class */
 		for (j = a[i].h & buckets; h[j].pos != INT_MAX;
 		     j = (j + 1) & buckets)
 			if (!cmp(a + i, b + h[j].pos))
 				break;
 
 		a[i].e = j; /* use equivalence class for quick compare */
 		if (h[j].len <= t)
 			a[i].n = h[j].pos; /* point to head of match list */
 		else
 			a[i].n = INT_MAX; /* too popular */
 	}
 
 	/* discard hash tables */
 	free(h);
 	return 1;
 }
 
 static int longest_match(struct line *a, struct line *b, struct pos *pos,
 			 int a1, int a2, int b1, int b2, int *omi, int *omj)
 {
 	int mi = a1, mj = b1, mk = 0, mb = 0, i, j, k;
 
 	for (i = a1; i < a2; i++) {
 		/* skip things before the current block */
 		for (j = a[i].n; j < b1; j = b[j].n)
 			;
 
 		/* loop through all lines match a[i] in b */
 		for (; j < b2; j = b[j].n) {
 			/* does this extend an earlier match? */
 			if (i > a1 && j > b1 && pos[j - 1].pos == i - 1)
 				k = pos[j - 1].len + 1;
 			else
 				k = 1;
 			pos[j].pos = i;
 			pos[j].len = k;
 
 			/* best match so far? */
 			if (k > mk) {
 				mi = i;
 				mj = j;
 				mk = k;
 			}
 		}
 	}
 
 	if (mk) {
 		mi = mi - mk + 1;
 		mj = mj - mk + 1;
 	}
 
 	/* expand match to include neighboring popular lines */
 	while (mi - mb > a1 && mj - mb > b1 &&
 	       a[mi - mb - 1].e == b[mj - mb - 1].e)
 		mb++;
 	while (mi + mk < a2 && mj + mk < b2 &&
 	       a[mi + mk].e == b[mj + mk].e)
 		mk++;
 
 	*omi = mi - mb;
 	*omj = mj - mb;
 
 	return mk + mb;
 }
 
 static struct hunk *recurse(struct line *a, struct line *b, struct pos *pos,
 			    int a1, int a2, int b1, int b2, struct hunk *l)
 {
 	int i, j, k;
 
 	while (1) {
 		/* find the longest match in this chunk */
 		k = longest_match(a, b, pos, a1, a2, b1, b2, &i, &j);
 		if (!k)
 			return l;
 
 		/* and recurse on the remaining chunks on either side */
 		l = recurse(a, b, pos, a1, i, b1, j, l);
 		if (!l)
 			return NULL;
 
 		l->next = (struct hunk *)malloc(sizeof(struct hunk));
 		if (!l->next)
 			return NULL;
 
 		l = l->next;
 		l->a1 = i;
 		l->a2 = i + k;
 		l->b1 = j;
 		l->b2 = j + k;
 		l->next = NULL;
 
 		/* tail-recursion didn't happen, so do equivalent iteration */
 		a1 = i + k;
 		b1 = j + k;
 	}
 }
 
 static int diff(struct line *a, int an, struct line *b, int bn,
 		 struct hunk *base)
 {
 	struct hunk *curr;
 	struct pos *pos;
 	int t, count = 0;
 
 	/* allocate and fill arrays */
 	t = equatelines(a, an, b, bn);
 	pos = (struct pos *)calloc(bn ? bn : 1, sizeof(struct pos));
 
 	if (pos && t) {
 		/* generate the matching block list */
 
 		curr = recurse(a, b, pos, 0, an, 0, bn, base);
 		if (!curr)
 			return -1;
 
 		/* sentinel end hunk */
 		curr->next = (struct hunk *)malloc(sizeof(struct hunk));
 		if (!curr->next)
 			return -1;
 		curr = curr->next;
 		curr->a1 = curr->a2 = an;
 		curr->b1 = curr->b2 = bn;
 		curr->next = NULL;
 	}
 
 	free(pos);
 
 	/* normalize the hunk list, try to push each hunk towards the end */
 	for (curr = base->next; curr; curr = curr->next) {
 		struct hunk *next = curr->next;
 		int shift = 0;
 
 		if (!next)
 			break;
 
 		if (curr->a2 == next->a1)
 			while (curr->a2 + shift < an && curr->b2 + shift < bn
 			       && !cmp(a + curr->a2 + shift,
 				       b + curr->b2 + shift))
 				shift++;
 		else if (curr->b2 == next->b1)
 			while (curr->b2 + shift < bn && curr->a2 + shift < an
 			       && !cmp(b + curr->b2 + shift,
 				       a + curr->a2 + shift))
 				shift++;
 		if (!shift)
 			continue;
 		curr->b2 += shift;
 		next->b1 += shift;
 		curr->a2 += shift;
 		next->a1 += shift;
 	}
 
 	for (curr = base->next; curr; curr = curr->next)
 		count++;
 	return count;
 }
 
 static void freehunks(struct hunk *l)
 {
 	struct hunk *n;
 	for (; l; l = n) {
 		n = l->next;
 		free(l);
 	}
 }
 
 static PyObject *blocks(PyObject *self, PyObject *args)
 {
 	PyObject *sa, *sb, *rl = NULL, *m;
 	struct line *a, *b;
 	struct hunk l, *h;
 	int an, bn, count, pos = 0;
 
 	if (!PyArg_ParseTuple(args, "SS:bdiff", &sa, &sb))
 		return NULL;
 
 	an = splitlines(PyBytes_AsString(sa), PyBytes_Size(sa), &a);
 	bn = splitlines(PyBytes_AsString(sb), PyBytes_Size(sb), &b);
 
 	if (!a || !b)
 		goto nomem;
 
 	l.next = NULL;
 	count = diff(a, an, b, bn, &l);
 	if (count < 0)
 		goto nomem;
 
 	rl = PyList_New(count);
 	if (!rl)
 		goto nomem;
 
 	for (h = l.next; h; h = h->next) {
 		m = Py_BuildValue("iiii", h->a1, h->a2, h->b1, h->b2);
 		PyList_SetItem(rl, pos, m);
 		pos++;
 	}
 
 nomem:
 	free(a);
 	free(b);
 	freehunks(l.next);
 	return rl ? rl : PyErr_NoMemory();
 }
 
 static PyObject *bdiff(PyObject *self, PyObject *args)
 {
 	char *sa, *sb;
 	PyObject *result = NULL;
 	struct line *al, *bl;
 	struct hunk l, *h;
 	char encode[12], *rb;
 	int an, bn, len = 0, la, lb, count;
 
 	if (!PyArg_ParseTuple(args, "s#s#:bdiff", &sa, &la, &sb, &lb))
 		return NULL;
 
 	an = splitlines(sa, la, &al);
 	bn = splitlines(sb, lb, &bl);
 	if (!al || !bl)
 		goto nomem;
 
 	l.next = NULL;
 	count = diff(al, an, bl, bn, &l);
 	if (count < 0)
 		goto nomem;
 
 	/* calculate length of output */
 	la = lb = 0;
 	for (h = l.next; h; h = h->next) {
 		if (h->a1 != la || h->b1 != lb)
 			len += 12 + bl[h->b1].l - bl[lb].l;
 		la = h->a2;
 		lb = h->b2;
 	}
 
 	result = PyBytes_FromStringAndSize(NULL, len);
 
 	if (!result)
 		goto nomem;
 
 	/* build binary patch */
 	rb = PyBytes_AsString(result);
 	la = lb = 0;
 
 	for (h = l.next; h; h = h->next) {
 		if (h->a1 != la || h->b1 != lb) {
 			len = bl[h->b1].l - bl[lb].l;
 			*(uint32_t *)(encode)     = htonl(al[la].l - al->l);
 			*(uint32_t *)(encode + 4) = htonl(al[h->a1].l - al->l);
 			*(uint32_t *)(encode + 8) = htonl(len);
 			memcpy(rb, encode, 12);
 			memcpy(rb + 12, bl[lb].l, len);
 			rb += 12 + len;
 		}
 		la = h->a2;
 		lb = h->b2;
 	}
 
 nomem:
 	free(al);
 	free(bl);
 	freehunks(l.next);
 	return result ? result : PyErr_NoMemory();
 }
 
 static char mdiff_doc[] = "Efficient binary diff.";
 
 static PyMethodDef methods[] = {
 	{"bdiff", bdiff, METH_VARARGS, "calculate a binary diff\n"},
 	{"blocks", blocks, METH_VARARGS, "find a list of matching lines\n"},
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef bdiff_module = {
 	PyModuleDef_HEAD_INIT,
 	"bdiff",
 	mdiff_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_bdiff(void)
 {
 	return PyModule_Create(&bdiff_module);
 }
 #else
 PyMODINIT_FUNC initbdiff(void)
 {
 	Py_InitModule3("bdiff", methods, mdiff_doc);
 }
 #endif
 
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/diffhelpers.c

@@ -1,190 +1,194 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) {
 		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
 
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/mpatch.c

@@ -1,470 +1,475 b''
 /*
  mpatch.c - efficient binary patching for Mercurial
 
  This implements a patch algorithm that's O(m + nlog n) where m is the
  size of the output and n is the number of patches.
 
  Given a list of binary patches, it unpacks each into a hunk list,
  then combines the hunk lists with a treewise recursion to form a
  single hunk list. This hunk list is then applied to the original
  text.
 
  The text (or binary) fragments are copied directly from their source
  Python objects into a preallocated output string to avoid the
  allocation of intermediate Python objects. Working memory is about 2x
  the total number of hunks.
 
  Copyright 2005, 2006 Matt Mackall <[e-mail]>
 
  This software may be used and distributed according to the terms
  of the GNU General Public License, incorporated herein by reference.
 */
 
 #include <Python.h>
 #include <stdlib.h>
 #include <string.h>
 
 #include "util.h"
 
 /* Definitions to get compatibility with python 2.4 and earlier which
    does not have Py_ssize_t. See also PEP 353.
    Note: msvc (8 or earlier) does not have ssize_t, so we use Py_ssize_t.
 */
 #if PY_VERSION_HEX < 0x02050000 && !defined(PY_SSIZE_T_MIN)
 typedef int Py_ssize_t;
 #define PY_SSIZE_T_MAX INT_MAX
 #define PY_SSIZE_T_MIN INT_MIN
 #endif
 
 #ifdef _WIN32
 #ifdef _MSC_VER
 /* msvc 6.0 has problems */
 #define inline __inline
 typedef unsigned long uint32_t;
 #else
 #include <stdint.h>
 #endif
 static uint32_t ntohl(uint32_t x)
 {
 	return ((x & 0x000000ffUL) << 24) |
 		((x & 0x0000ff00UL) <<  8) |
 		((x & 0x00ff0000UL) >>  8) |
 		((x & 0xff000000UL) >> 24);
 }
 #else
 /* not windows */
 #include <sys/types.h>
 #if defined __BEOS__ && !defined __HAIKU__
 #include <ByteOrder.h>
 #else
 #include <arpa/inet.h>
 #endif
 #include <inttypes.h>
 #endif
 
 static char mpatch_doc[] = "Efficient binary patching.";
 static PyObject *mpatch_Error;
 
 struct frag {
 	int start, end, len;
 	const char *data;
 };
 
 struct flist {
 	struct frag *base, *head, *tail;
 };
 
 static struct flist *lalloc(int size)
 {
 	struct flist *a = NULL;
 
 	if (size < 1)
 		size = 1;
 
 	a = (struct flist *)malloc(sizeof(struct flist));
 	if (a) {
 		a->base = (struct frag *)malloc(sizeof(struct frag) * size);
 		if (a->base) {
 			a->head = a->tail = a->base;
 			return a;
 		}
 		free(a);
 		a = NULL;
 	}
 	if (!PyErr_Occurred())
 		PyErr_NoMemory();
 	return NULL;
 }
 
 static void lfree(struct flist *a)
 {
 	if (a) {
 		free(a->base);
 		free(a);
 	}
 }
 
 static int lsize(struct flist *a)
 {
 	return a->tail - a->head;
 }
 
 /* move hunks in source that are less cut to dest, compensating
    for changes in offset. the last hunk may be split if necessary.
 */
 static int gather(struct flist *dest, struct flist *src, int cut, int offset)
 {
 	struct frag *d = dest->tail, *s = src->head;
 	int postend, c, l;
 
 	while (s != src->tail) {
 		if (s->start + offset >= cut)
 			break; /* we've gone far enough */
 
 		postend = offset + s->start + s->len;
 		if (postend <= cut) {
 			/* save this hunk */
 			offset += s->start + s->len - s->end;
 			*d++ = *s++;
 		}
 		else {
 			/* break up this hunk */
 			c = cut - offset;
 			if (s->end < c)
 				c = s->end;
 			l = cut - offset - s->start;
 			if (s->len < l)
 				l = s->len;
 
 			offset += s->start + l - c;
 
 			d->start = s->start;
 			d->end = c;
 			d->len = l;
 			d->data = s->data;
 			d++;
 			s->start = c;
 			s->len = s->len - l;
 			s->data = s->data + l;
 
 			break;
 		}
 	}
 
 	dest->tail = d;
 	src->head = s;
 	return offset;
 }
 
 /* like gather, but with no output list */
 static int discard(struct flist *src, int cut, int offset)
 {
 	struct frag *s = src->head;
 	int postend, c, l;
 
 	while (s != src->tail) {
 		if (s->start + offset >= cut)
 			break;
 
 		postend = offset + s->start + s->len;
 		if (postend <= cut) {
 			offset += s->start + s->len - s->end;
 			s++;
 		}
 		else {
 			c = cut - offset;
 			if (s->end < c)
 				c = s->end;
 			l = cut - offset - s->start;
 			if (s->len < l)
 				l = s->len;
 
 			offset += s->start + l - c;
 			s->start = c;
 			s->len = s->len - l;
 			s->data = s->data + l;
 
 			break;
 		}
 	}
 
 	src->head = s;
 	return offset;
 }
 
 /* combine hunk lists a and b, while adjusting b for offset changes in a/
    this deletes a and b and returns the resultant list. */
 static struct flist *combine(struct flist *a, struct flist *b)
 {
 	struct flist *c = NULL;
 	struct frag *bh, *ct;
 	int offset = 0, post;
 
 	if (a && b)
 		c = lalloc((lsize(a) + lsize(b)) * 2);
 
 	if (c) {
 
 		for (bh = b->head; bh != b->tail; bh++) {
 			/* save old hunks */
 			offset = gather(c, a, bh->start, offset);
 
 			/* discard replaced hunks */
 			post = discard(a, bh->end, offset);
 
 			/* insert new hunk */
 			ct = c->tail;
 			ct->start = bh->start - offset;
 			ct->end = bh->end - post;
 			ct->len = bh->len;
 			ct->data = bh->data;
 			c->tail++;
 			offset = post;
 		}
 
 		/* hold on to tail from a */
 		memcpy(c->tail, a->head, sizeof(struct frag) * lsize(a));
 		c->tail += lsize(a);
 	}
 
 	lfree(a);
 	lfree(b);
 	return c;
 }
 
 /* decode a binary patch into a hunk list */
 static struct flist *decode(const char *bin, int len)
 {
 	struct flist *l;
 	struct frag *lt;
 	const char *data = bin + 12, *end = bin + len;
 	char decode[12]; /* for dealing with alignment issues */
 
 	/* assume worst case size, we won't have many of these lists */
 	l = lalloc(len / 12);
 	if (!l)
 		return NULL;
 
 	lt = l->tail;
 
 	while (data <= end) {
 		memcpy(decode, bin, 12);
 		lt->start = ntohl(*(uint32_t *)decode);
 		lt->end = ntohl(*(uint32_t *)(decode + 4));
 		lt->len = ntohl(*(uint32_t *)(decode + 8));
 		if (lt->start > lt->end)
 			break; /* sanity check */
 		bin = data + lt->len;
 		if (bin < data)
 			break; /* big data + big (bogus) len can wrap around */
 		lt->data = data;
 		data = bin + 12;
 		lt++;
 	}
 
 	if (bin != end) {
 		if (!PyErr_Occurred())
 			PyErr_SetString(mpatch_Error, "patch cannot be decoded");
 		lfree(l);
 		return NULL;
 	}
 
 	l->tail = lt;
 	return l;
 }
 
 /* calculate the size of resultant text */
 static int calcsize(int len, struct flist *l)
 {
 	int outlen = 0, last = 0;
 	struct frag *f = l->head;
 
 	while (f != l->tail) {
 		if (f->start < last || f->end > len) {
 			if (!PyErr_Occurred())
 				PyErr_SetString(mpatch_Error,
 				                "invalid patch");
 			return -1;
 		}
 		outlen += f->start - last;
 		last = f->end;
 		outlen += f->len;
 		f++;
 	}
 
 	outlen += len - last;
 	return outlen;
 }
 
 static int apply(char *buf, const char *orig, int len, struct flist *l)
 {
 	struct frag *f = l->head;
 	int last = 0;
 	char *p = buf;
 
 	while (f != l->tail) {
 		if (f->start < last || f->end > len) {
 			if (!PyErr_Occurred())
 				PyErr_SetString(mpatch_Error,
 				                "invalid patch");
 			return 0;
 		}
 		memcpy(p, orig + last, f->start - last);
 		p += f->start - last;
 		memcpy(p, f->data, f->len);
 		last = f->end;
 		p += f->len;
 		f++;
 	}
 	memcpy(p, orig + last, len - last);
 	return 1;
 }
 
 /* recursively generate a patch of all bins between start and end */
 static struct flist *fold(PyObject *bins, int start, int end)
 {
 	int len;
 	Py_ssize_t blen;
 	const char *buffer;
 
 	if (start + 1 == end) {
 		/* trivial case, output a decoded list */
 		PyObject *tmp = PyList_GetItem(bins, start);
 		if (!tmp)
 			return NULL;
 		if (PyObject_AsCharBuffer(tmp, &buffer, &blen))
 			return NULL;
 		return decode(buffer, blen);
 	}
 
 	/* divide and conquer, memory management is elsewhere */
 	len = (end - start) / 2;
 	return combine(fold(bins, start, start + len),
 		       fold(bins, start + len, end));
 }
 
 static PyObject *
 patches(PyObject *self, PyObject *args)
 {
 	PyObject *text, *bins, *result;
 	struct flist *patch;
 	const char *in;
 	char *out;
 	int len, outlen;
 	Py_ssize_t inlen;
 
 	if (!PyArg_ParseTuple(args, "OO:mpatch", &text, &bins))
 		return NULL;
 
 	len = PyList_Size(bins);
 	if (!len) {
 		/* nothing to do */
 		Py_INCREF(text);
 		return text;
 	}
 
 	if (PyObject_AsCharBuffer(text, &in, &inlen))
 		return NULL;
 
 	patch = fold(bins, 0, len);
 	if (!patch)
 		return NULL;
 
 	outlen = calcsize(inlen, patch);
 	if (outlen < 0) {
 		result = NULL;
 		goto cleanup;
 	}
 	result = PyBytes_FromStringAndSize(NULL, outlen);
 	if (!result) {
 		result = NULL;
 		goto cleanup;
 	}
 	out = PyBytes_AsString(result);
 	if (!apply(out, in, inlen, patch)) {
 		Py_DECREF(result);
 		result = NULL;
 	}
 cleanup:
 	lfree(patch);
 	return result;
 }
 
 /* calculate size of a patched file directly */
 static PyObject *
 patchedsize(PyObject *self, PyObject *args)
 {
 	long orig, start, end, len, outlen = 0, last = 0;
 	int patchlen;
 	char *bin, *binend, *data;
 	char decode[12]; /* for dealing with alignment issues */
 
 	if (!PyArg_ParseTuple(args, "ls#", &orig, &bin, &patchlen))
 		return NULL;
 
 	binend = bin + patchlen;
 	data = bin + 12;
 
 	while (data <= binend) {
 		memcpy(decode, bin, 12);
 		start = ntohl(*(uint32_t *)decode);
 		end = ntohl(*(uint32_t *)(decode + 4));
 		len = ntohl(*(uint32_t *)(decode + 8));
 		if (start > end)
 			break; /* sanity check */
 		bin = data + len;
 		if (bin < data)
 			break; /* big data + big (bogus) len can wrap around */
 		data = bin + 12;
 		outlen += start - last;
 		last = end;
 		outlen += len;
 	}
 
 	if (bin != binend) {
 		if (!PyErr_Occurred())
 			PyErr_SetString(mpatch_Error, "patch cannot be decoded");
 		return NULL;
 	}
 
 	outlen += orig - last;
 	return Py_BuildValue("l", outlen);
 }
 
 static PyMethodDef methods[] = {
 	{"patches", patches, METH_VARARGS, "apply a series of patches\n"},
 	{"patchedsize", patchedsize, METH_VARARGS, "calculed patched size\n"},
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef mpatch_module = {
 	PyModuleDef_HEAD_INIT,
 	"mpatch",
 	mpatch_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_mpatch(void)
 {
 	PyObject *m;
 
 	m = PyModule_Create(&mpatch_module);
 	if (m == NULL)
 		return NULL;
 
 	mpatch_Error = PyErr_NewException("mpatch.mpatchError", NULL, NULL);
 	Py_INCREF(mpatch_Error);
 	PyModule_AddObject(m, "mpatchError", mpatch_Error);
 
 	return m;
 }
 #else
 PyMODINIT_FUNC
 initmpatch(void)
 {
 	Py_InitModule3("mpatch", methods, mpatch_doc);
 	mpatch_Error = PyErr_NewException("mpatch.mpatchError", NULL, NULL);
 }
 #endif
+
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/osutil.c

@@ -1,554 +1,559 b''
 /*
  osutil.c - native operating system services
 
  Copyright 2007 Matt Mackall and others
 
  This software may be used and distributed according to the terms of
  the GNU General Public License, incorporated herein by reference.
 */
 
 #define _ATFILE_SOURCE
 #include <Python.h>
 #include <fcntl.h>
 #include <stdio.h>
 #include <string.h>
 
 #ifdef _WIN32
 #include <windows.h>
 #include <io.h>
 #else
 #include <dirent.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 #include <unistd.h>
 #endif
 
 #include "util.h"
 
 /* some platforms lack the PATH_MAX definition (eg. GNU/Hurd) */
 #ifndef PATH_MAX
 #define PATH_MAX 4096
 #endif
 
 #ifdef _WIN32
 /*
 stat struct compatible with hg expectations
 Mercurial only uses st_mode, st_size and st_mtime
 the rest is kept to minimize changes between implementations
 */
 struct hg_stat {
 	int st_dev;
 	int st_mode;
 	int st_nlink;
 	__int64 st_size;
 	int st_mtime;
 	int st_ctime;
 };
 struct listdir_stat {
 	PyObject_HEAD
 	struct hg_stat st;
 };
 #else
 struct listdir_stat {
 	PyObject_HEAD
 	struct stat st;
 };
 #endif
 
 #define listdir_slot(name) \
 	static PyObject *listdir_stat_##name(PyObject *self, void *x) \
 	{ \
 		return PyInt_FromLong(((struct listdir_stat *)self)->st.name); \
 	}
 
 listdir_slot(st_dev)
 listdir_slot(st_mode)
 listdir_slot(st_nlink)
 #ifdef _WIN32
 static PyObject *listdir_stat_st_size(PyObject *self, void *x)
 {
 	return PyLong_FromLongLong(
 		(PY_LONG_LONG)((struct listdir_stat *)self)->st.st_size);
 }
 #else
 listdir_slot(st_size)
 #endif
 listdir_slot(st_mtime)
 listdir_slot(st_ctime)
 
 static struct PyGetSetDef listdir_stat_getsets[] = {
 	{"st_dev", listdir_stat_st_dev, 0, 0, 0},
 	{"st_mode", listdir_stat_st_mode, 0, 0, 0},
 	{"st_nlink", listdir_stat_st_nlink, 0, 0, 0},
 	{"st_size", listdir_stat_st_size, 0, 0, 0},
 	{"st_mtime", listdir_stat_st_mtime, 0, 0, 0},
 	{"st_ctime", listdir_stat_st_ctime, 0, 0, 0},
 	{0, 0, 0, 0, 0}
 };
 
 static PyObject *listdir_stat_new(PyTypeObject *t, PyObject *a, PyObject *k)
 {
 	return t->tp_alloc(t, 0);
 }
 
 static void listdir_stat_dealloc(PyObject *o)
 {
 	o->ob_type->tp_free(o);
 }
 
 static PyTypeObject listdir_stat_type = {
 	PyVarObject_HEAD_INIT(NULL, 0)
 	"osutil.stat",             /*tp_name*/
 	sizeof(struct listdir_stat), /*tp_basicsize*/
 	0,                         /*tp_itemsize*/
 	(destructor)listdir_stat_dealloc, /*tp_dealloc*/
 	0,                         /*tp_print*/
 	0,                         /*tp_getattr*/
 	0,                         /*tp_setattr*/
 	0,                         /*tp_compare*/
 	0,                         /*tp_repr*/
 	0,                         /*tp_as_number*/
 	0,                         /*tp_as_sequence*/
 	0,                         /*tp_as_mapping*/
 	0,                         /*tp_hash */
 	0,                         /*tp_call*/
 	0,                         /*tp_str*/
 	0,                         /*tp_getattro*/
 	0,                         /*tp_setattro*/
 	0,                         /*tp_as_buffer*/
 	Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
 	"stat objects",            /* tp_doc */
 	0,                         /* tp_traverse */
 	0,                         /* tp_clear */
 	0,                         /* tp_richcompare */
 	0,                         /* tp_weaklistoffset */
 	0,                         /* tp_iter */
 	0,                         /* tp_iternext */
 	0,                         /* tp_methods */
 	0,                         /* tp_members */
 	listdir_stat_getsets,      /* tp_getset */
 	0,                         /* tp_base */
 	0,                         /* tp_dict */
 	0,                         /* tp_descr_get */
 	0,                         /* tp_descr_set */
 	0,                         /* tp_dictoffset */
 	0,                         /* tp_init */
 	0,                         /* tp_alloc */
 	listdir_stat_new,          /* tp_new */
 };
 
 #ifdef _WIN32
 
 static int to_python_time(const FILETIME *tm)
 {
 	/* number of seconds between epoch and January 1 1601 */
 	const __int64 a0 = (__int64)134774L * (__int64)24L * (__int64)3600L;
 	/* conversion factor from 100ns to 1s */
 	const __int64 a1 = 10000000;
 	/* explicit (int) cast to suspend compiler warnings */
 	return (int)((((__int64)tm->dwHighDateTime << 32)
 			+ tm->dwLowDateTime) / a1 - a0);
 }
 
 static PyObject *make_item(const WIN32_FIND_DATAA *fd, int wantstat)
 {
 	PyObject *py_st;
 	struct hg_stat *stp;
 
 	int kind = (fd->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
 		? _S_IFDIR : _S_IFREG;
 
 	if (!wantstat)
 		return Py_BuildValue("si", fd->cFileName, kind);
 
 	py_st = PyObject_CallObject((PyObject *)&listdir_stat_type, NULL);
 	if (!py_st)
 		return NULL;
 
 	stp = &((struct listdir_stat *)py_st)->st;
 	/*
 	use kind as st_mode
 	rwx bits on Win32 are meaningless
 	and Hg does not use them anyway
 	*/
 	stp->st_mode  = kind;
 	stp->st_mtime = to_python_time(&fd->ftLastWriteTime);
 	stp->st_ctime = to_python_time(&fd->ftCreationTime);
 	if (kind == _S_IFREG)
 		stp->st_size = ((__int64)fd->nFileSizeHigh << 32)
 				+ fd->nFileSizeLow;
 	return Py_BuildValue("siN", fd->cFileName,
 		kind, py_st);
 }
 
 static PyObject *_listdir(char *path, int plen, int wantstat, char *skip)
 {
 	PyObject *rval = NULL; /* initialize - return value */
 	PyObject *list;
 	HANDLE fh;
 	WIN32_FIND_DATAA fd;
 	char *pattern;
 
 	/* build the path + \* pattern string */
 	pattern = malloc(plen + 3); /* path + \* + \0 */
 	if (!pattern) {
 		PyErr_NoMemory();
 		goto error_nomem;
 	}
 	strcpy(pattern, path);
 
 	if (plen > 0) {
 		char c = path[plen-1];
 		if (c != ':' && c != '/' && c != '\\')
 			pattern[plen++] = '\\';
 	}
 	strcpy(pattern + plen, "*");
 
 	fh = FindFirstFileA(pattern, &fd);
 	if (fh == INVALID_HANDLE_VALUE) {
 		PyErr_SetFromWindowsErrWithFilename(GetLastError(), path);
 		goto error_file;
 	}
 
 	list = PyList_New(0);
 	if (!list)
 		goto error_list;
 
 	do {
 		PyObject *item;
 
 		if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
 			if (!strcmp(fd.cFileName, ".")
 			|| !strcmp(fd.cFileName, ".."))
 				continue;
 
 			if (skip && !strcmp(fd.cFileName, skip)) {
 				rval = PyList_New(0);
 				goto error;
 			}
 		}
 
 		item = make_item(&fd, wantstat);
 		if (!item)
 			goto error;
 
 		if (PyList_Append(list, item)) {
 			Py_XDECREF(item);
 			goto error;
 		}
 
 		Py_XDECREF(item);
 	} while (FindNextFileA(fh, &fd));
 
 	if (GetLastError() != ERROR_NO_MORE_FILES) {
 		PyErr_SetFromWindowsErrWithFilename(GetLastError(), path);
 		goto error;
 	}
 
 	rval = list;
 	Py_XINCREF(rval);
 error:
 	Py_XDECREF(list);
 error_list:
 	FindClose(fh);
 error_file:
 	free(pattern);
 error_nomem:
 	return rval;
 }
 
 #else
 
 int entkind(struct dirent *ent)
 {
 #ifdef DT_REG
 	switch (ent->d_type) {
 	case DT_REG: return S_IFREG;
 	case DT_DIR: return S_IFDIR;
 	case DT_LNK: return S_IFLNK;
 	case DT_BLK: return S_IFBLK;
 	case DT_CHR: return S_IFCHR;
 	case DT_FIFO: return S_IFIFO;
 	case DT_SOCK: return S_IFSOCK;
 	}
 #endif
 	return -1;
 }
 
 static PyObject *_listdir(char *path, int pathlen, int keepstat, char *skip)
 {
 	PyObject *list, *elem, *stat, *ret = NULL;
 	char fullpath[PATH_MAX + 10];
 	int kind, err;
 	struct stat st;
 	struct dirent *ent;
 	DIR *dir;
 #ifdef AT_SYMLINK_NOFOLLOW
 	int dfd = -1;
 #endif
 
 	if (pathlen >= PATH_MAX) {
 		PyErr_SetString(PyExc_ValueError, "path too long");
 		goto error_value;
 	}
 	strncpy(fullpath, path, PATH_MAX);
 	fullpath[pathlen] = '/';
 
 #ifdef AT_SYMLINK_NOFOLLOW
 	dfd = open(path, O_RDONLY);
 	if (dfd == -1) {
 		PyErr_SetFromErrnoWithFilename(PyExc_OSError, path);
 		goto error_value;
 	}
 	dir = fdopendir(dfd);
 #else
 	dir = opendir(path);
 #endif
 	if (!dir) {
 		PyErr_SetFromErrnoWithFilename(PyExc_OSError, path);
 		goto error_dir;
  	}
 
 	list = PyList_New(0);
 	if (!list)
 		goto error_list;
 
 	while ((ent = readdir(dir))) {
 		if (!strcmp(ent->d_name, ".") || !strcmp(ent->d_name, ".."))
 			continue;
 
 		kind = entkind(ent);
 		if (kind == -1 || keepstat) {
 #ifdef AT_SYMLINK_NOFOLLOW
 			err = fstatat(dfd, ent->d_name, &st,
 				      AT_SYMLINK_NOFOLLOW);
 #else
 			strncpy(fullpath + pathlen + 1, ent->d_name,
 				PATH_MAX - pathlen);
 			fullpath[PATH_MAX] = 0;
 			err = lstat(fullpath, &st);
 #endif
 			if (err == -1) {
 				strncpy(fullpath + pathlen + 1, ent->d_name,
 					PATH_MAX - pathlen);
 				fullpath[PATH_MAX] = 0;
 				PyErr_SetFromErrnoWithFilename(PyExc_OSError,
 							       fullpath);
 				goto error;
 			}
 			kind = st.st_mode & S_IFMT;
 		}
 
 		/* quit early? */
 		if (skip && kind == S_IFDIR && !strcmp(ent->d_name, skip)) {
 			ret = PyList_New(0);
 			goto error;
 		}
 
 		if (keepstat) {
 			stat = PyObject_CallObject((PyObject *)&listdir_stat_type, NULL);
 			if (!stat)
 				goto error;
 			memcpy(&((struct listdir_stat *)stat)->st, &st, sizeof(st));
 			elem = Py_BuildValue("siN", ent->d_name, kind, stat);
 		} else
 			elem = Py_BuildValue("si", ent->d_name, kind);
 		if (!elem)
 			goto error;
 
 		PyList_Append(list, elem);
 		Py_DECREF(elem);
 	}
 
 	ret = list;
 	Py_INCREF(ret);
 
 error:
 	Py_DECREF(list);
 error_list:
 	closedir(dir);
 error_dir:
 #ifdef AT_SYMLINK_NOFOLLOW
 	close(dfd);
 #endif
 error_value:
 	return ret;
 }
 
 #endif /* ndef _WIN32 */
 
 static PyObject *listdir(PyObject *self, PyObject *args, PyObject *kwargs)
 {
 	PyObject *statobj = NULL; /* initialize - optional arg */
 	PyObject *skipobj = NULL; /* initialize - optional arg */
 	char *path, *skip = NULL;
 	int wantstat, plen;
 
 	static char *kwlist[] = {"path", "stat", "skip", NULL};
 
 	if (!PyArg_ParseTupleAndKeywords(args, kwargs, "s#|OO:listdir",
 			kwlist, &path, &plen, &statobj, &skipobj))
 		return NULL;
 
 	wantstat = statobj && PyObject_IsTrue(statobj);
 
 	if (skipobj && skipobj != Py_None) {
 		skip = PyBytes_AsString(skipobj);
 		if (!skip)
 			return NULL;
 	}
 
 	return _listdir(path, plen, wantstat, skip);
 }
 
 #ifdef _WIN32
 static PyObject *posixfile(PyObject *self, PyObject *args, PyObject *kwds)
 {
 	static char *kwlist[] = {"name", "mode", "buffering", NULL};
 	PyObject *file_obj = NULL;
 	char *name = NULL;
 	char *mode = "rb";
 	DWORD access = 0;
 	DWORD creation;
 	HANDLE handle;
 	int fd, flags = 0;
 	int bufsize = -1;
 	char m0, m1, m2;
 	char fpmode[4];
 	int fppos = 0;
 	int plus;
 	FILE *fp;
 
 	if (!PyArg_ParseTupleAndKeywords(args, kwds, "et|si:posixfile", kwlist,
 					 Py_FileSystemDefaultEncoding,
 					 &name, &mode, &bufsize))
 		return NULL;
 
 	m0 = mode[0];
 	m1 = m0 ? mode[1] : '\0';
 	m2 = m1 ? mode[2] : '\0';
 	plus = m1 == '+' || m2 == '+';
 
 	fpmode[fppos++] = m0;
 	if (m1 == 'b' || m2 == 'b') {
 		flags = _O_BINARY;
 		fpmode[fppos++] = 'b';
 	}
 	else
 		flags = _O_TEXT;
 	if (m0 == 'r' && !plus) {
 		flags |= _O_RDONLY;
 		access = GENERIC_READ;
 	} else {
 		/*
 		work around http://support.microsoft.com/kb/899149 and
 		set _O_RDWR for 'w' and 'a', even if mode has no '+'
 		*/
 		flags |= _O_RDWR;
 		access = GENERIC_READ | GENERIC_WRITE;
 		fpmode[fppos++] = '+';
 	}
 	fpmode[fppos++] = '\0';
 
 	switch (m0) {
 	case 'r':
 		creation = OPEN_EXISTING;
 		break;
 	case 'w':
 		creation = CREATE_ALWAYS;
 		break;
 	case 'a':
 		creation = OPEN_ALWAYS;
 		flags |= _O_APPEND;
 		break;
 	default:
 		PyErr_Format(PyExc_ValueError,
 			     "mode string must begin with one of 'r', 'w', "
 			     "or 'a', not '%c'", m0);
 		goto bail;
 	}
 
 	handle = CreateFile(name, access,
 			    FILE_SHARE_READ | FILE_SHARE_WRITE |
 			    FILE_SHARE_DELETE,
 			    NULL,
 			    creation,
 			    FILE_ATTRIBUTE_NORMAL,
 			    0);
 
 	if (handle == INVALID_HANDLE_VALUE) {
 		PyErr_SetFromWindowsErrWithFilename(GetLastError(), name);
 		goto bail;
 	}
 
 	fd = _open_osfhandle((intptr_t)handle, flags);
 
 	if (fd == -1) {
 		CloseHandle(handle);
 		PyErr_SetFromErrnoWithFilename(PyExc_IOError, name);
 		goto bail;
 	}
 #ifndef IS_PY3K
 	fp = _fdopen(fd, fpmode);
 	if (fp == NULL) {
 		_close(fd);
 		PyErr_SetFromErrnoWithFilename(PyExc_IOError, name);
 		goto bail;
 	}
 
 	file_obj = PyFile_FromFile(fp, name, mode, fclose);
 	if (file_obj == NULL) {
 		fclose(fp);
 		goto bail;
 	}
 
 	PyFile_SetBufSize(file_obj, bufsize);
 #else
 	file_obj = PyFile_FromFd(fd, name, mode, bufsize, NULL, NULL, NULL, 1);
 	if (file_obj == NULL)
 		goto bail;
 #endif
 bail:
 	PyMem_Free(name);
 	return file_obj;
 }
 #endif
 
 static char osutil_doc[] = "Native operating system services.";
 
 static PyMethodDef methods[] = {
 	{"listdir", (PyCFunction)listdir, METH_VARARGS | METH_KEYWORDS,
 	 "list a directory\n"},
 #ifdef _WIN32
 	{"posixfile", (PyCFunction)posixfile, METH_VARARGS | METH_KEYWORDS,
 	 "Open a file with POSIX-like semantics.\n"
 "On error, this function may raise either a WindowsError or an IOError."},
 #endif
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef osutil_module = {
 	PyModuleDef_HEAD_INIT,
 	"osutil",
 	osutil_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_osutil(void)
 {
 	if (PyType_Ready(&listdir_stat_type) < 0)
 		return NULL;
 
 	return PyModule_Create(&osutil_module);
 }
 #else
 PyMODINIT_FUNC initosutil(void)
 {
 	if (PyType_Ready(&listdir_stat_type) == -1)
 		return;
 
 	Py_InitModule3("osutil", methods, osutil_doc);
 }
 #endif
+
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/parsers.c

@@ -1,418 +1,422 b''
 /*
  parsers.c - efficient content parsing
 
  Copyright 2008 Matt Mackall <[e-mail]> and others
 
  This software may be used and distributed according to the terms of
  the GNU General Public License, incorporated herein by reference.
 */
 
 #include <Python.h>
 #include <ctype.h>
 #include <string.h>
 
 #include "util.h"
 
 static int hexdigit(char c)
 {
 	if (c >= '0' && c <= '9')
 		return c - '0';
 	if (c >= 'a' && c <= 'f')
 		return c - 'a' + 10;
 	if (c >= 'A' && c <= 'F')
 		return c - 'A' + 10;
 
 	PyErr_SetString(PyExc_ValueError, "input contains non-hex character");
 	return 0;
 }
 
 /*
  * Turn a hex-encoded string into binary.
  */
 static PyObject *unhexlify(const char *str, int len)
 {
 	PyObject *ret;
 	const char *c;
 	char *d;
 
 	ret = PyBytes_FromStringAndSize(NULL, len / 2);
 
 	if (!ret)
 		return NULL;
 
 	d = PyBytes_AsString(ret);
 
 	for (c = str; c < str + len;) {
 		int hi = hexdigit(*c++);
 		int lo = hexdigit(*c++);
 		*d++ = (hi << 4) | lo;
 	}
 
 	return ret;
 }
 
 /*
  * This code assumes that a manifest is stitched together with newline
  * ('\n') characters.
  */
 static PyObject *parse_manifest(PyObject *self, PyObject *args)
 {
 	PyObject *mfdict, *fdict;
 	char *str, *cur, *start, *zero;
 	int len;
 
 	if (!PyArg_ParseTuple(args, "O!O!s#:parse_manifest",
 			      &PyDict_Type, &mfdict,
 			      &PyDict_Type, &fdict,
 			      &str, &len))
 		goto quit;
 
 	for (start = cur = str, zero = NULL; cur < str + len; cur++) {
 		PyObject *file = NULL, *node = NULL;
 		PyObject *flags = NULL;
 		int nlen;
 
 		if (!*cur) {
 			zero = cur;
 			continue;
 		}
 		else if (*cur != '\n')
 			continue;
 
 		if (!zero) {
 			PyErr_SetString(PyExc_ValueError,
 					"manifest entry has no separator");
 			goto quit;
 		}
 
 		file = PyBytes_FromStringAndSize(start, zero - start);
 
 		if (!file)
 			goto bail;
 
 		nlen = cur - zero - 1;
 
 		node = unhexlify(zero + 1, nlen > 40 ? 40 : nlen);
 		if (!node)
 			goto bail;
 
 		if (nlen > 40) {
 			flags = PyBytes_FromStringAndSize(zero + 41,
 							   nlen - 40);
 			if (!flags)
 				goto bail;
 
 			if (PyDict_SetItem(fdict, file, flags) == -1)
 				goto bail;
 		}
 
 		if (PyDict_SetItem(mfdict, file, node) == -1)
 			goto bail;
 
 		start = cur + 1;
 		zero = NULL;
 
 		Py_XDECREF(flags);
 		Py_XDECREF(node);
 		Py_XDECREF(file);
 		continue;
 	bail:
 		Py_XDECREF(flags);
 		Py_XDECREF(node);
 		Py_XDECREF(file);
 		goto quit;
 	}
 
 	if (len > 0 && *(cur - 1) != '\n') {
 		PyErr_SetString(PyExc_ValueError,
 				"manifest contains trailing garbage");
 		goto quit;
 	}
 
 	Py_INCREF(Py_None);
 	return Py_None;
 quit:
 	return NULL;
 }
 
 #ifdef _WIN32
 #ifdef _MSC_VER
 /* msvc 6.0 has problems */
 #define inline __inline
 typedef unsigned long uint32_t;
 typedef unsigned __int64 uint64_t;
 #else
 #include <stdint.h>
 #endif
 static uint32_t ntohl(uint32_t x)
 {
 	return ((x & 0x000000ffUL) << 24) |
 	       ((x & 0x0000ff00UL) <<  8) |
 	       ((x & 0x00ff0000UL) >>  8) |
 	       ((x & 0xff000000UL) >> 24);
 }
 #else
 /* not windows */
 #include <sys/types.h>
 #if defined __BEOS__ && !defined __HAIKU__
 #include <ByteOrder.h>
 #else
 #include <arpa/inet.h>
 #endif
 #include <inttypes.h>
 #endif
 
 static PyObject *parse_dirstate(PyObject *self, PyObject *args)
 {
 	PyObject *dmap, *cmap, *parents = NULL, *ret = NULL;
 	PyObject *fname = NULL, *cname = NULL, *entry = NULL;
 	char *str, *cur, *end, *cpos;
 	int state, mode, size, mtime;
 	unsigned int flen;
 	int len;
 	char decode[16]; /* for alignment */
 
 	if (!PyArg_ParseTuple(args, "O!O!s#:parse_dirstate",
 			      &PyDict_Type, &dmap,
 			      &PyDict_Type, &cmap,
 			      &str, &len))
 		goto quit;
 
 	/* read parents */
 	if (len < 40)
 		goto quit;
 
 	parents = Py_BuildValue("s#s#", str, 20, str + 20, 20);
 	if (!parents)
 		goto quit;
 
 	/* read filenames */
 	cur = str + 40;
 	end = str + len;
 
 	while (cur < end - 17) {
 		/* unpack header */
 		state = *cur;
 		memcpy(decode, cur + 1, 16);
 		mode = ntohl(*(uint32_t *)(decode));
 		size = ntohl(*(uint32_t *)(decode + 4));
 		mtime = ntohl(*(uint32_t *)(decode + 8));
 		flen = ntohl(*(uint32_t *)(decode + 12));
 		cur += 17;
 		if (cur + flen > end || cur + flen < cur) {
 			PyErr_SetString(PyExc_ValueError, "overflow in dirstate");
 			goto quit;
 		}
 
 		entry = Py_BuildValue("ciii", state, mode, size, mtime);
 		if (!entry)
 			goto quit;
 		PyObject_GC_UnTrack(entry); /* don't waste time with this */
 
 		cpos = memchr(cur, 0, flen);
 		if (cpos) {
 			fname = PyBytes_FromStringAndSize(cur, cpos - cur);
 			cname = PyBytes_FromStringAndSize(cpos + 1,
 							   flen - (cpos - cur) - 1);
 			if (!fname || !cname ||
 			    PyDict_SetItem(cmap, fname, cname) == -1 ||
 			    PyDict_SetItem(dmap, fname, entry) == -1)
 				goto quit;
 			Py_DECREF(cname);
 		} else {
 			fname = PyBytes_FromStringAndSize(cur, flen);
 			if (!fname ||
 			    PyDict_SetItem(dmap, fname, entry) == -1)
 				goto quit;
 		}
 		cur += flen;
 		Py_DECREF(fname);
 		Py_DECREF(entry);
 		fname = cname = entry = NULL;
 	}
 
 	ret = parents;
 	Py_INCREF(ret);
 quit:
 	Py_XDECREF(fname);
 	Py_XDECREF(cname);
 	Py_XDECREF(entry);
 	Py_XDECREF(parents);
 	return ret;
 }
 
 const char nullid[20];
 const int nullrev = -1;
 
 /* RevlogNG format (all in big endian, data may be inlined):
  *    6 bytes: offset
  *    2 bytes: flags
  *    4 bytes: compressed length
  *    4 bytes: uncompressed length
  *    4 bytes: base revision
  *    4 bytes: link revision
  *    4 bytes: parent 1 revision
  *    4 bytes: parent 2 revision
  *   32 bytes: nodeid (only 20 bytes used)
  */
 static int _parse_index_ng(const char *data, int size, int inlined,
 			   PyObject *index)
 {
 	PyObject *entry;
 	int n = 0, err;
 	uint64_t offset_flags;
 	int comp_len, uncomp_len, base_rev, link_rev, parent_1, parent_2;
 	const char *c_node_id;
 	const char *end = data + size;
 	char decode[64]; /* to enforce alignment with inline data */
 
 	while (data < end) {
 		unsigned int step;
 
 		memcpy(decode, data, 64);
 		offset_flags = ntohl(*((uint32_t *) (decode + 4)));
 		if (n == 0) /* mask out version number for the first entry */
 			offset_flags &= 0xFFFF;
 		else {
 			uint32_t offset_high =  ntohl(*((uint32_t *)decode));
 			offset_flags |= ((uint64_t)offset_high) << 32;
 		}
 
 		comp_len = ntohl(*((uint32_t *)(decode + 8)));
 		uncomp_len = ntohl(*((uint32_t *)(decode + 12)));
 		base_rev = ntohl(*((uint32_t *)(decode + 16)));
 		link_rev = ntohl(*((uint32_t *)(decode + 20)));
 		parent_1 = ntohl(*((uint32_t *)(decode + 24)));
 		parent_2 = ntohl(*((uint32_t *)(decode + 28)));
 		c_node_id = decode + 32;
 
 		entry = Py_BuildValue("Liiiiiis#", offset_flags, comp_len,
 			      uncomp_len, base_rev, link_rev,
 			      parent_1, parent_2, c_node_id, 20);
 
 		if (!entry)
 			return 0;
 
 		PyObject_GC_UnTrack(entry); /* don't waste time with this */
 
 		if (inlined) {
 			err = PyList_Append(index, entry);
 			Py_DECREF(entry);
 			if (err)
 				return 0;
 		} else
 			PyList_SET_ITEM(index, n, entry); /* steals reference */
 
 		n++;
 		step = 64 + (inlined ? comp_len : 0);
 		if (data + step > end || data + step < data)
 			break;
 		data += step;
 	}
 	if (data != end) {
 		if (!PyErr_Occurred())
 			PyErr_SetString(PyExc_ValueError, "corrupt index file");
 		return 0;
 	}
 
 	/* create the magic nullid entry in the index at [-1] */
 	entry = Py_BuildValue("Liiiiiis#", (uint64_t)0, 0, 0, -1, -1, -1, -1, nullid, 20);
 
 	if (!entry)
 		return 0;
 
 	PyObject_GC_UnTrack(entry); /* don't waste time with this */
 
 	if (inlined) {
 		err = PyList_Append(index, entry);
 		Py_DECREF(entry);
 		if (err)
 			return 0;
 	} else
 		PyList_SET_ITEM(index, n, entry); /* steals reference */
 
 	return 1;
 }
 
 /* This function parses a index file and returns a Python tuple of the
  * following format: (index, cache)
  *
  * index: a list of tuples containing the RevlogNG records
  * cache: if data is inlined, a tuple (index_file_content, 0) else None
  */
 static PyObject *parse_index2(PyObject *self, PyObject *args)
 {
 	const char *data;
 	int size, inlined;
 	PyObject *rval = NULL, *index = NULL, *cache = NULL;
 	PyObject *data_obj = NULL, *inlined_obj;
 
 	if (!PyArg_ParseTuple(args, "s#O", &data, &size, &inlined_obj))
 		return NULL;
 	inlined = inlined_obj && PyObject_IsTrue(inlined_obj);
 
 	/* If no data is inlined, we know the size of the index list in
 	 * advance: size divided by the size of one revlog record (64 bytes)
 	 * plus one for nullid */
 	index = inlined ? PyList_New(0) : PyList_New(size / 64 + 1);
 	if (!index)
 		goto quit;
 
 	/* set up the cache return value */
 	if (inlined) {
 		/* Note that the reference to data_obj is only borrowed */
 		data_obj = PyTuple_GET_ITEM(args, 0);
 		cache = Py_BuildValue("iO", 0, data_obj);
 		if (!cache)
 			goto quit;
 	} else {
 		cache = Py_None;
 		Py_INCREF(Py_None);
 	}
 
 	/* actually populate the index with data */
 	if (!_parse_index_ng(data, size, inlined, index))
 		goto quit;
 
 	rval = Py_BuildValue("NN", index, cache);
 	if (!rval)
 		goto quit;
 	return rval;
 
 quit:
 	Py_XDECREF(index);
 	Py_XDECREF(cache);
 	Py_XDECREF(rval);
 	return NULL;
 }
 
 
 static char parsers_doc[] = "Efficient content parsing.";
 
 static PyMethodDef methods[] = {
 	{"parse_manifest", parse_manifest, METH_VARARGS, "parse a manifest\n"},
 	{"parse_dirstate", parse_dirstate, METH_VARARGS, "parse a dirstate\n"},
 	{"parse_index2", parse_index2, METH_VARARGS, "parse a revlog index\n"},
 	{NULL, NULL}
 };
 
 #ifdef IS_PY3K
 static struct PyModuleDef parsers_module = {
 	PyModuleDef_HEAD_INIT,
 	"parsers",
 	parsers_doc,
 	-1,
 	methods
 };
 
 PyMODINIT_FUNC PyInit_parsers(void)
 {
 	return PyModule_Create(&parsers_module);
 }
 #else
 PyMODINIT_FUNC initparsers(void)
 {
 	Py_InitModule3("parsers", methods, parsers_doc);
 }
 #endif
 
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */

mercurial/util.h

@@ -1,105 +1,109 b''
 /*
  util.h - utility functions for interfacing with the various python APIs.
 
  This software may be used and distributed according to the terms of
  the GNU General Public License, incorporated herein by reference.
 */
 
 #ifndef _HG_UTIL_H_
 #define _HG_UTIL_H_
 
 #if PY_MAJOR_VERSION >= 3
 
 #define IS_PY3K
 #define PyInt_FromLong PyLong_FromLong
 #define PyInt_AsLong PyLong_AsLong
 
 /*
  Mapping of some of the python < 2.x PyString* functions to py3k's PyUnicode.
 
  The commented names below represent those that are present in the PyBytes
  definitions for python < 2.6 (below in this file) that don't have a direct
  implementation.
 */
 
 #define PyStringObject PyUnicodeObject
 #define PyString_Type PyUnicode_Type
 
 #define PyString_Check PyUnicode_Check
 #define PyString_CheckExact PyUnicode_CheckExact
 #define PyString_CHECK_INTERNED PyUnicode_CHECK_INTERNED
 #define PyString_AS_STRING PyUnicode_AsLatin1String
 #define PyString_GET_SIZE PyUnicode_GET_SIZE
 
 #define PyString_FromStringAndSize PyUnicode_FromStringAndSize
 #define PyString_FromString PyUnicode_FromString
 #define PyString_FromFormatV PyUnicode_FromFormatV
 #define PyString_FromFormat PyUnicode_FromFormat
 /* #define PyString_Size PyUnicode_GET_SIZE */
 /* #define PyString_AsString */
 /* #define PyString_Repr */
 #define PyString_Concat PyUnicode_Concat
 #define PyString_ConcatAndDel PyUnicode_AppendAndDel
 #define _PyString_Resize PyUnicode_Resize
 /* #define _PyString_Eq */
 #define PyString_Format PyUnicode_Format
 /* #define _PyString_FormatLong */
 /* #define PyString_DecodeEscape */
 #define _PyString_Join PyUnicode_Join
 #define PyString_Decode PyUnicode_Decode
 #define PyString_Encode PyUnicode_Encode
 #define PyString_AsEncodedObject PyUnicode_AsEncodedObject
 #define PyString_AsEncodedString PyUnicode_AsEncodedString
 #define PyString_AsDecodedObject PyUnicode_AsDecodedObject
 #define PyString_AsDecodedString PyUnicode_AsDecodedUnicode
 /* #define PyString_AsStringAndSize */
 #define _PyString_InsertThousandsGrouping _PyUnicode_InsertThousandsGrouping
 
 #endif /* PY_MAJOR_VERSION */
 
 /* Backports from 2.6 */
 #if PY_VERSION_HEX < 0x02060000
 
 #define Py_TYPE(ob) (ob)->ob_type
 #define Py_SIZE(ob) (ob)->ob_size
 #define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
 
 /* Shamelessly stolen from bytesobject.h */
 #define PyBytesObject PyStringObject
 #define PyBytes_Type PyString_Type
 
 #define PyBytes_Check PyString_Check
 #define PyBytes_CheckExact PyString_CheckExact
 #define PyBytes_CHECK_INTERNED PyString_CHECK_INTERNED
 #define PyBytes_AS_STRING PyString_AS_STRING
 #define PyBytes_GET_SIZE PyString_GET_SIZE
 #define Py_TPFLAGS_BYTES_SUBCLASS Py_TPFLAGS_STRING_SUBCLASS
 
 #define PyBytes_FromStringAndSize PyString_FromStringAndSize
 #define PyBytes_FromString PyString_FromString
 #define PyBytes_FromFormatV PyString_FromFormatV
 #define PyBytes_FromFormat PyString_FromFormat
 #define PyBytes_Size PyString_Size
 #define PyBytes_AsString PyString_AsString
 #define PyBytes_Repr PyString_Repr
 #define PyBytes_Concat PyString_Concat
 #define PyBytes_ConcatAndDel PyString_ConcatAndDel
 #define _PyBytes_Resize _PyString_Resize
 #define _PyBytes_Eq _PyString_Eq
 #define PyBytes_Format PyString_Format
 #define _PyBytes_FormatLong _PyString_FormatLong
 #define PyBytes_DecodeEscape PyString_DecodeEscape
 #define _PyBytes_Join _PyString_Join
 #define PyBytes_Decode PyString_Decode
 #define PyBytes_Encode PyString_Encode
 #define PyBytes_AsEncodedObject PyString_AsEncodedObject
 #define PyBytes_AsEncodedString PyString_AsEncodedString
 #define PyBytes_AsDecodedObject PyString_AsDecodedObject
 #define PyBytes_AsDecodedString PyString_AsDecodedString
 #define PyBytes_AsStringAndSize PyString_AsStringAndSize
 #define _PyBytes_InsertThousandsGrouping _PyString_InsertThousandsGrouping
 
 #endif /* PY_VERSION_HEX */
 
 #endif /* _HG_UTIL_H_ */
 
+/* Local Variables:      */
+/* c-file-style: "linux" */
+/* indent-tabs-mode: t   */
+/* End:                  */