Add copyright
[unix-history] / usr / src / sys / kern / kern_exec.c
/*
* Copyright (c) 1982 Regents of the University of California.
* All rights reserved. The Berkeley software License Agreement
* specifies the terms and conditions for redistribution.
*
* @(#)kern_exec.c 6.13 (Berkeley) %G%
*/
#include "../machine/reg.h"
#include "../machine/pte.h"
#include "../machine/psl.h"
#include "param.h"
#include "systm.h"
#include "map.h"
#include "dir.h"
#include "user.h"
#include "kernel.h"
#include "proc.h"
#include "buf.h"
#include "inode.h"
#include "seg.h"
#include "vm.h"
#include "text.h"
#include "file.h"
#include "uio.h"
#include "acct.h"
#include "exec.h"
#ifdef vax
#include "../vax/mtpr.h"
#endif
/*
* exec system call, with and without environments.
*/
struct execa {
char *fname;
char **argp;
char **envp;
};
execv()
{
((struct execa *)u.u_ap)->envp = NULL;
execve();
}
execve()
{
register nc;
register char *cp;
register struct buf *bp;
register struct execa *uap;
int na, ne, ucp, ap, len, cc;
int indir, uid, gid;
char *sharg;
struct inode *ip;
swblk_t bno;
char cfname[MAXCOMLEN + 1];
#define SHSIZE 32
char cfarg[SHSIZE];
union {
char ex_shell[SHSIZE]; /* #! and name of interpreter */
struct exec ex_exec;
} exdata;
register struct nameidata *ndp = &u.u_nd;
int resid, error;
ndp->ni_nameiop = LOOKUP | FOLLOW;
ndp->ni_segflg = UIO_USERSPACE;
ndp->ni_dirp = ((struct execa *)u.u_ap)->fname;
if ((ip = namei(ndp)) == NULL)
return;
bno = 0;
bp = 0;
indir = 0;
uid = u.u_uid;
gid = u.u_gid;
if (ip->i_mode & ISUID)
uid = ip->i_uid;
if (ip->i_mode & ISGID)
gid = ip->i_gid;
again:
if (access(ip, IEXEC))
goto bad;
if ((u.u_procp->p_flag&STRC) && access(ip, IREAD))
goto bad;
if ((ip->i_mode & IFMT) != IFREG ||
(ip->i_mode & (IEXEC|(IEXEC>>3)|(IEXEC>>6))) == 0) {
u.u_error = EACCES;
goto bad;
}
/*
* Read in first few bytes of file for segment sizes, magic number:
* 407 = plain executable
* 410 = RO text
* 413 = demand paged RO text
* Also an ASCII line beginning with #! is
* the file name of a ``shell'' and arguments may be prepended
* to the argument list if given here.
*
* SHELL NAMES ARE LIMITED IN LENGTH.
*
* ONLY ONE ARGUMENT MAY BE PASSED TO THE SHELL FROM
* THE ASCII LINE.
*/
exdata.ex_shell[0] = '\0'; /* for zero length files */
u.u_error = rdwri(UIO_READ, ip, (caddr_t)&exdata, sizeof (exdata),
0, 1, &resid);
if (u.u_error)
goto bad;
#ifndef lint
if (resid > sizeof(exdata) - sizeof(exdata.ex_exec) &&
exdata.ex_shell[0] != '#') {
u.u_error = ENOEXEC;
goto bad;
}
#endif
switch (exdata.ex_exec.a_magic) {
case 0407:
exdata.ex_exec.a_data += exdata.ex_exec.a_text;
exdata.ex_exec.a_text = 0;
break;
case 0413:
case 0410:
if (exdata.ex_exec.a_text == 0) {
u.u_error = ENOEXEC;
goto bad;
}
break;
default:
if (exdata.ex_shell[0] != '#' ||
exdata.ex_shell[1] != '!' ||
indir) {
u.u_error = ENOEXEC;
goto bad;
}
cp = &exdata.ex_shell[2]; /* skip "#!" */
while (cp < &exdata.ex_shell[SHSIZE]) {
if (*cp == '\t')
*cp = ' ';
else if (*cp == '\n') {
*cp = '\0';
break;
}
cp++;
}
if (*cp != '\0') {
u.u_error = ENOEXEC;
goto bad;
}
cp = &exdata.ex_shell[2];
while (*cp == ' ')
cp++;
ndp->ni_dirp = cp;
while (*cp && *cp != ' ')
cp++;
cfarg[0] = '\0';
if (*cp) {
*cp++ = '\0';
while (*cp == ' ')
cp++;
if (*cp)
bcopy((caddr_t)cp, (caddr_t)cfarg, SHSIZE);
}
indir = 1;
iput(ip);
ndp->ni_nameiop = LOOKUP | FOLLOW;
ndp->ni_segflg = UIO_SYSSPACE;
ip = namei(ndp);
if (ip == NULL)
return;
bcopy((caddr_t)ndp->ni_dent.d_name, (caddr_t)cfname,
MAXCOMLEN);
cfname[MAXCOMLEN] = '\0';
goto again;
}
/*
* Collect arguments on "file" in swap space.
*/
na = 0;
ne = 0;
nc = 0;
cc = 0;
uap = (struct execa *)u.u_ap;
bno = rmalloc(argmap, (long)ctod(clrnd((int)btoc(NCARGS))));
if (bno == 0) {
swkill(u.u_procp, "exec: no swap space");
goto bad;
}
if (bno % CLSIZE)
panic("execa rmalloc");
/*
* Copy arguments into file in argdev area.
*/
if (uap->argp) for (;;) {
ap = NULL;
sharg = NULL;
if (indir && na == 0) {
sharg = cfname;
ap = (int)sharg;
uap->argp++; /* ignore argv[0] */
} else if (indir && (na == 1 && cfarg[0])) {
sharg = cfarg;
ap = (int)sharg;
} else if (indir && (na == 1 || na == 2 && cfarg[0]))
ap = (int)uap->fname;
else if (uap->argp) {
ap = fuword((caddr_t)uap->argp);
uap->argp++;
}
if (ap == NULL && uap->envp) {
uap->argp = NULL;
if ((ap = fuword((caddr_t)uap->envp)) != NULL)
uap->envp++, ne++;
}
if (ap == NULL)
break;
na++;
if (ap == -1) {
error = EFAULT;
break;
}
do {
if (cc <= 0) {
/*
* We depend on NCARGS being a multiple of
* CLSIZE*NBPG. This way we need only check
* overflow before each buffer allocation.
*/
if (nc >= NCARGS-1) {
error = E2BIG;
break;
}
if (bp)
bdwrite(bp);
cc = CLSIZE*NBPG;
bp = getblk(argdev, bno + ctod(nc/NBPG), cc);
cp = bp->b_un.b_addr;
}
if (sharg) {
error = copystr(sharg, cp, cc, &len);
sharg += len;
} else {
error = copyinstr((caddr_t)ap, cp, cc, &len);
ap += len;
}
cp += len;
nc += len;
cc -= len;
} while (error == ENOENT);
if (error) {
u.u_error = error;
if (bp)
brelse(bp);
bp = 0;
goto badarg;
}
}
if (bp)
bdwrite(bp);
bp = 0;
nc = (nc + NBPW-1) & ~(NBPW-1);
getxfile(ip, &exdata.ex_exec, nc + (na+4)*NBPW, uid, gid);
if (u.u_error) {
badarg:
for (cc = 0; cc < nc; cc += CLSIZE*NBPG) {
bp = baddr(argdev, bno + ctod(cc/NBPG), CLSIZE*NBPG);
if (bp) {
bp->b_flags |= B_AGE; /* throw away */
bp->b_flags &= ~B_DELWRI; /* cancel io */
brelse(bp);
bp = 0;
}
}
goto bad;
}
iput(ip);
ip = NULL;
/*
* Copy back arglist.
*/
ucp = USRSTACK - nc - NBPW;
ap = ucp - na*NBPW - 3*NBPW;
u.u_ar0[SP] = ap;
(void) suword((caddr_t)ap, na-ne);
nc = 0;
cc = 0;
for (;;) {
ap += NBPW;
if (na == ne) {
(void) suword((caddr_t)ap, 0);
ap += NBPW;
}
if (--na < 0)
break;
(void) suword((caddr_t)ap, ucp);
do {
if (cc <= 0) {
if (bp)
brelse(bp);
cc = CLSIZE*NBPG;
bp = bread(argdev, bno + ctod(nc / NBPG), cc);
bp->b_flags |= B_AGE; /* throw away */
bp->b_flags &= ~B_DELWRI; /* cancel io */
cp = bp->b_un.b_addr;
}
error = copyoutstr(cp, (caddr_t)ucp, cc, &len);
ucp += len;
cp += len;
nc += len;
cc -= len;
} while (error == ENOENT);
if (error == EFAULT)
panic("exec: EFAULT");
}
(void) suword((caddr_t)ap, 0);
/*
* Reset caught signals. Held signals
* remain held through p_sigmask.
*/
while (u.u_procp->p_sigcatch) {
nc = ffs(u.u_procp->p_sigcatch);
u.u_procp->p_sigcatch &= ~sigmask(nc);
u.u_signal[nc] = SIG_DFL;
}
/*
* Reset stack state to the user stack.
* Clear set of signals caught on the signal stack.
*/
u.u_onstack = 0;
u.u_sigsp = 0;
u.u_sigonstack = 0;
for (nc = u.u_lastfile; nc >= 0; --nc) {
if (u.u_pofile[nc] & UF_EXCLOSE) {
closef(u.u_ofile[nc]);
u.u_ofile[nc] = NULL;
u.u_pofile[nc] = 0;
}
u.u_pofile[nc] &= ~UF_MAPPED;
}
while (u.u_lastfile >= 0 && u.u_ofile[u.u_lastfile] == NULL)
u.u_lastfile--;
setregs(exdata.ex_exec.a_entry);
/*
* Remember file name for accounting.
*/
u.u_acflag &= ~AFORK;
if (indir)
bcopy((caddr_t)cfname, (caddr_t)u.u_comm, MAXCOMLEN);
else {
if (ndp->ni_dent.d_namlen > MAXCOMLEN)
ndp->ni_dent.d_namlen = MAXCOMLEN;
bcopy((caddr_t)ndp->ni_dent.d_name, (caddr_t)u.u_comm,
(unsigned)(ndp->ni_dent.d_namlen + 1));
}
bad:
if (bp)
brelse(bp);
if (bno)
rmfree(argmap, (long)ctod(clrnd((int) btoc(NCARGS))), bno);
if (ip)
iput(ip);
}
/*
* Read in and set up memory for executed file.
*/
getxfile(ip, ep, nargc, uid, gid)
register struct inode *ip;
register struct exec *ep;
int nargc, uid, gid;
{
size_t ts, ds, ids, uds, ss;
int pagi;
if (ep->a_magic == 0413)
pagi = SPAGI;
else
pagi = 0;
if (ip->i_flag & IXMOD) { /* XXX */
u.u_error = ETXTBSY;
goto bad;
}
if (ep->a_text != 0 && (ip->i_flag&ITEXT) == 0 &&
ip->i_count != 1) {
register struct file *fp;
for (fp = file; fp < fileNFILE; fp++) {
if (fp->f_type == DTYPE_INODE &&
fp->f_count > 0 &&
(struct inode *)fp->f_data == ip &&
(fp->f_flag&FWRITE)) {
u.u_error = ETXTBSY;
goto bad;
}
}
}
/*
* Compute text and data sizes and make sure not too large.
* NB - Check data and bss separately as they may overflow
* when summed together.
*/
ts = clrnd(btoc(ep->a_text));
ids = clrnd(btoc(ep->a_data));
uds = clrnd(btoc(ep->a_bss));
ds = clrnd(btoc(ep->a_data + ep->a_bss));
ss = clrnd(SSIZE + btoc(nargc));
if (chksize((unsigned)ts, (unsigned)ids, (unsigned)uds, (unsigned)ss))
goto bad;
/*
* Make sure enough space to start process.
*/
u.u_cdmap = zdmap;
u.u_csmap = zdmap;
if (swpexpand(ds, ss, &u.u_cdmap, &u.u_csmap) == NULL)
goto bad;
/*
* At this point, committed to the new image!
* Release virtual memory resources of old process, and
* initialize the virtual memory of the new process.
* If we resulted from vfork(), instead wakeup our
* parent who will set SVFDONE when he has taken back
* our resources.
*/
if ((u.u_procp->p_flag & SVFORK) == 0)
vrelvm();
else {
u.u_procp->p_flag &= ~SVFORK;
u.u_procp->p_flag |= SKEEP;
wakeup((caddr_t)u.u_procp);
while ((u.u_procp->p_flag & SVFDONE) == 0)
sleep((caddr_t)u.u_procp, PZERO - 1);
u.u_procp->p_flag &= ~(SVFDONE|SKEEP);
}
u.u_procp->p_flag &= ~(SPAGI|SSEQL|SUANOM|SOUSIG);
u.u_procp->p_flag |= pagi;
u.u_dmap = u.u_cdmap;
u.u_smap = u.u_csmap;
vgetvm(ts, ds, ss);
if (pagi == 0)
u.u_error =
rdwri(UIO_READ, ip,
(char *)ctob(dptov(u.u_procp, 0)),
(int)ep->a_data,
(int)(sizeof (struct exec) + ep->a_text),
0, (int *)0);
xalloc(ip, ep, pagi);
if (pagi && u.u_procp->p_textp)
vinifod((struct fpte *)dptopte(u.u_procp, 0),
PG_FTEXT, u.u_procp->p_textp->x_iptr,
(long)(1 + ts/CLSIZE), (int)btoc(ep->a_data));
#ifdef vax
/* THIS SHOULD BE DONE AT A LOWER LEVEL, IF AT ALL */
mtpr(TBIA, 0);
#endif
if (u.u_error)
swkill(u.u_procp, "exec: I/O error mapping pages");
/*
* set SUID/SGID protections, if no tracing
*/
if ((u.u_procp->p_flag&STRC)==0) {
u.u_uid = uid;
u.u_procp->p_uid = uid;
u.u_gid = gid;
} else
psignal(u.u_procp, SIGTRAP);
u.u_tsize = ts;
u.u_dsize = ds;
u.u_ssize = ss;
u.u_prof.pr_scale = 0;
bad:
return;
}