/* python-crack - Python bindings for cracklib
 * Copyright (C) 2003 Domenico Andreoli
 *
 * This file is part of python-crack.
 *
 * python-crack is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * python-crack is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Prua; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <Python.h>
#include <packer.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>

#define FILENAME_MAXLEN 512

static char const DEFAULT_DICTPATH[] = "@DEFAULT_DICTPATH@";

static char _crack_FascistCheck_doc [] =
	"arguments: passwd, dictpath (optional)\n"
	"\n"
	"  passwd - password to be checked for weakness\n"
	"  dictpath - full path name to the cracklib dictionary database\n"
	"\n"
	"if dictpath is not specified value of CRACKLIB_DICTPATH will be used.\n"
	"\n"
	"return value: the same password passed as first argument.\n"
	"\n"
	"if password is weak, exception ValueError is raised with argument\n"
	"set to the reason of weakness.\n"
;

static PyObject* _crack_FascistCheck(PyObject* self, PyObject* args)
{
	char* passwd;
	char const* dictpath;
	char const* reason;

	int i, fd;
	char filename[FILENAME_MAXLEN];
	char const* ext[] = { "hwm", "pwd", "pwi" };

	dictpath = DEFAULT_DICTPATH;

	if(!PyArg_ParseTuple(args, "s|s:FascistCheck", &passwd, &dictpath)) 
		return NULL;

	/* we need to check if files are readable, FascistCheck doesn't */
	for(i = 0; i < 3; i++) {
		snprintf(filename, FILENAME_MAXLEN - 1, "%s.%s", dictpath, ext[i]);
		fd = open(filename, O_RDONLY);
		if(fd == -1) {
			PyErr_SetFromErrnoWithFilename(PyExc_IOError, filename);
			return NULL;
		} else {
			close(fd);
		}
	}

	if((reason = FascistCheck(passwd, dictpath)) != NULL) {
		PyErr_SetString(PyExc_ValueError, reason);
		return NULL;
	}

	return Py_BuildValue("s", passwd);
}

static PyMethodDef _crack_methods[] = {
	{ "FascistCheck", _crack_FascistCheck, METH_VARARGS, _crack_FascistCheck_doc },
	{ NULL } /* Sentinel */
};

static char _crack_doc[] =
	"Python bindings for cracklib. Version @VERSION@.\n"
	"\n"
	"This module anables the use of cracklib features from within a Python\n"
	"program or interpreter.\n"
;

DL_EXPORT(void) init_crack(void)
{
	PyObject *m, *d, *dictpath;
	m = Py_InitModule3("_crack", _crack_methods, _crack_doc);
	d = PyModule_GetDict(m);
	dictpath = PyString_FromString(DEFAULT_DICTPATH);
	if (!d || !dictpath || PyDict_SetItemString(d, "default_dictpath", dictpath) < 0)
		return;

	Py_DECREF(dictpath);
}