Subversion Repositories or1k

/*
 *  linux/fs/binfmt_script.c
 *
 *  Copyright (C) 1996  Martin von Löwis
 *  original #!-checking implemented by tytso.
 *
 */
 
/*
 * uClinux revisions for NO_MM
 * Copyright (C) 1998  Kenneth Albanowski <kjahds@kjahds.com>,
 *                     The Silver Hammer Group, Ltd.
 *
 *  FIXME: Modifications are untested!
 */
 
#include <linux/module.h>
#include <linux/string.h>
#include <linux/stat.h>
#include <linux/malloc.h>
#include <linux/binfmts.h>
 
static int do_load_script(struct linux_binprm *bprm,struct pt_regs *regs)
{
	char *cp, *interp, *i_name, *i_arg;
	int retval;
	if ((bprm->buf[0] != '#') || (bprm->buf[1] != '!') || (bprm->sh_bang)) 
		return -ENOEXEC;
	/*
	 * This section does the #! interpretation.
	 * Sorta complicated, but hopefully it will work.  -TYT
	 */
 
	bprm->sh_bang++;
	iput(bprm->inode);
	bprm->dont_iput=1;
 
	bprm->buf[127] = '\0';
	if ((cp = strchr(bprm->buf, '\n')) == NULL)
		cp = bprm->buf+127;
	*cp = '\0';
	while (cp > bprm->buf) {
		cp--;
		if ((*cp == ' ') || (*cp == '\t'))
			*cp = '\0';
		else
			break;
	}
	for (cp = bprm->buf+2; (*cp == ' ') || (*cp == '\t'); cp++);
	if (!cp || *cp == '\0') 
		return -ENOEXEC; /* No interpreter name found */
	interp = i_name = cp;
	i_arg = 0;
	for ( ; *cp && (*cp != ' ') && (*cp != '\t'); cp++) {
 		if (*cp == '/')
			i_name = cp+1;
	}
	while ((*cp == ' ') || (*cp == '\t'))
		*cp++ = '\0';
	if (*cp)
		i_arg = cp;
	/*
	 * OK, we've parsed out the interpreter name and
	 * (optional) argument.
	 * Splice in (1) the interpreter's name for argv[0]
	 *           (2) (optional) argument to interpreter
	 *           (3) filename of shell script (replace argv[0])
	 *
	 * This is done in reverse order, because of how the
	 * user environment and arguments are stored.
	 */
#ifndef NO_MM
	remove_arg_zero(bprm);
	bprm->p = copy_strings(1, &bprm->filename, bprm->page, bprm->p, 2);
	bprm->argc++;
	if (i_arg) {
		bprm->p = copy_strings(1, &i_arg, bprm->page, bprm->p, 2);
		bprm->argc++;
	}
	bprm->p = copy_strings(1, &i_name, bprm->page, bprm->p, 2);
	bprm->argc++;
	if (!bprm->p) 
		return -E2BIG;
	/*
	 * OK, now restart the process with the interpreter's inode.
	 * Note that we use open_namei() as the name is now in kernel
	 * space, and we don't need to copy it.
	 */
	retval = open_namei(interp, 0, 0, &bprm->inode, NULL);
	if (retval)
		return retval;
	bprm->dont_iput=0;
	retval=prepare_binprm(bprm);
	if(retval<0)
		return retval;
	return search_binary_handler(bprm,regs);
#else /* !NO_MM */
	{
		char ** oldargv = bprm->argv;
		int oldargc = bprm->argc;
		int i;
 
		bprm->argc = oldargc + (i_arg ? 2 : 1);
		bprm->argv = kmalloc(sizeof(char*) * bprm->argc, GFP_KERNEL);
 
		if (!bprm->argv) {
			bprm->argc = oldargc;
			bprm->argv = oldargv;
			return -E2BIG;
		}
 
		for(i=1;i<=oldargc;i++)
			bprm->argv[bprm->argc-i] = oldargv[oldargc-i];
 
		bprm->argv[0] = i_name;
		i=1;
		if (i_arg)
			bprm->argv[i++] = i_arg;
		bprm->argv[i++] = bprm->filename;
 
 
		/*
		 * OK, now restart the process with the interpreter's inode.
		 * Note that we use open_namei() as the name is now in kernel
		 * space, and we don't need to copy it.
		 */
		retval = open_namei(interp, 0, 0, &bprm->inode, NULL);
		if (!retval) {
			bprm->dont_iput=0;
			retval=prepare_binprm(bprm);
		}
		if (!retval) {
			retval =  search_binary_handler(bprm,regs);
		}
 
		kfree(bprm->argv);
		bprm->argv = oldargv;
		bprm->argc = oldargc;
 
		return retval;
	}
#endif /* !NO_MM */
}
 
static int load_script(struct linux_binprm *bprm,struct pt_regs *regs)
{
	int retval;
	MOD_INC_USE_COUNT;
	retval = do_load_script(bprm,regs);
	MOD_DEC_USE_COUNT;
	return retval;
}
 
struct linux_binfmt script_format = {
#ifndef MODULE
	NULL, 0, load_script, NULL, NULL
#else
	NULL, &mod_use_count_, load_script, NULL, NULL
#endif
};
 
int init_script_binfmt(void) {
	return register_binfmt(&script_format);
}
 
#ifdef MODULE
int init_module(void)
{
	return init_script_binfmt();
}
 
void cleanup_module( void) {
	unregister_binfmt(&script_format);
}
#endif