119304Speter/*-
219304Speter * Copyright (c) 1992, 1993, 1994
319304Speter *	The Regents of the University of California.  All rights reserved.
419304Speter * Copyright (c) 1992, 1993, 1994, 1995, 1996
519304Speter *	Keith Bostic.  All rights reserved.
619304Speter *
719304Speter * See the LICENSE file for redistribution information.
819304Speter */
919304Speter
1019304Speter#include "config.h"
1119304Speter
1219304Speter#ifndef lint
13254225Speterstatic const char sccsid[] = "$Id: ex_source.c,v 10.17 2011/12/19 16:17:06 zy Exp $";
1419304Speter#endif /* not lint */
1519304Speter
1619304Speter#include <sys/types.h>
1719304Speter#include <sys/queue.h>
1819304Speter#include <sys/stat.h>
1919304Speter
2019304Speter#include <bitstring.h>
2119304Speter#include <errno.h>
2219304Speter#include <fcntl.h>
2319304Speter#include <limits.h>
2419304Speter#include <stdio.h>
2519304Speter#include <stdlib.h>
2619304Speter#include <string.h>
2719304Speter#include <unistd.h>
2819304Speter
2919304Speter#include "../common/common.h"
3019304Speter
3119304Speter/*
3219304Speter * ex_source -- :source file
3319304Speter *	Execute ex commands from a file.
3419304Speter *
3519304Speter * PUBLIC: int ex_source __P((SCR *, EXCMD *));
3619304Speter */
3719304Speterint
38254225Speterex_source(SCR *sp, EXCMD *cmdp)
3919304Speter{
4019304Speter	struct stat sb;
4119304Speter	int fd, len;
42254225Speter	char *bp;
43254225Speter	char *name, *np;
44254225Speter	size_t nlen;
45254225Speter	CHAR_T *wp;
46254225Speter	size_t wlen;
47254225Speter	int rc;
4819304Speter
49254225Speter	INT2CHAR(sp, cmdp->argv[0]->bp, cmdp->argv[0]->len + 1, name, nlen);
5019304Speter	if ((fd = open(name, O_RDONLY, 0)) < 0 || fstat(fd, &sb))
5119304Speter		goto err;
5219304Speter
5319304Speter	/*
5419304Speter	 * XXX
5519304Speter	 * I'd like to test to see if the file is too large to malloc.  Since
5619304Speter	 * we don't know what size or type off_t's or size_t's are, what the
5719304Speter	 * largest unsigned integral type is, or what random insanity the local
5819304Speter	 * C compiler will perpetrate, doing the comparison in a portable way
5919304Speter	 * is flatly impossible.  So, put an fairly unreasonable limit on it,
6019304Speter	 * I don't want to be dropping core here.
6119304Speter	 */
6219304Speter#define	MEGABYTE	1048576
6319304Speter	if (sb.st_size > MEGABYTE) {
6419304Speter		errno = ENOMEM;
6519304Speter		goto err;
6619304Speter	}
6719304Speter
6819304Speter	MALLOC(sp, bp, char *, (size_t)sb.st_size + 1);
6919304Speter	if (bp == NULL) {
7019304Speter		(void)close(fd);
7119304Speter		return (1);
7219304Speter	}
7319304Speter	bp[sb.st_size] = '\0';
7419304Speter
7519304Speter	/* Read the file into memory. */
7619304Speter	len = read(fd, bp, (int)sb.st_size);
7719304Speter	(void)close(fd);
7819304Speter	if (len == -1 || len != sb.st_size) {
7919304Speter		if (len != sb.st_size)
8019304Speter			errno = EIO;
8119304Speter		free(bp);
8219304Spetererr:		msgq_str(sp, M_SYSERR, name, "%s");
8319304Speter		return (1);
8419304Speter	}
8519304Speter
86254225Speter	np = strdup(name);
87254225Speter	if (CHAR2INT(sp, bp, (size_t)sb.st_size + 1, wp, wlen))
88254225Speter		msgq(sp, M_ERR, "323|Invalid input. Truncated.");
8919304Speter	/* Put it on the ex queue. */
90254225Speter	rc = ex_run_str(sp, np, wp, wlen - 1, 1, 0);
91254225Speter	if (np != NULL)
92254225Speter		free(np);
93254225Speter	free(bp);
94254225Speter	return (rc);
9519304Speter}
96