xref: /openbsd/usr.bin/vi/ex/ex_source.c (revision 771fbea0)
1 /*	$OpenBSD: ex_source.c,v 1.10 2015/12/07 20:39:19 mmcc Exp $	*/
2 
3 /*-
4  * Copyright (c) 1992, 1993, 1994
5  *	The Regents of the University of California.  All rights reserved.
6  * Copyright (c) 1992, 1993, 1994, 1995, 1996
7  *	Keith Bostic.  All rights reserved.
8  *
9  * See the LICENSE file for redistribution information.
10  */
11 
12 #include "config.h"
13 
14 #include <sys/types.h>
15 #include <sys/queue.h>
16 #include <sys/stat.h>
17 
18 #include <bitstring.h>
19 #include <errno.h>
20 #include <fcntl.h>
21 #include <limits.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26 
27 #include "../common/common.h"
28 
29 /*
30  * ex_sourcefd -- :source already opened file
31  *	Execute ex commands from the given file descriptor
32  *
33  * PUBLIC: int ex_sourcefd(SCR *, EXCMD *, int);
34  */
35 int
36 ex_sourcefd(SCR *sp, EXCMD *cmdp, int fd)
37 {
38 	struct stat sb;
39 	int len;
40 	char *bp, *name;
41 
42 	name = cmdp->argv[0]->bp;
43 	if (fstat(fd, &sb))
44 		goto err;
45 
46 	/*
47 	 * XXX
48 	 * I'd like to test to see if the file is too large to malloc.  Since
49 	 * we don't know what size or type off_t's or size_t's are, what the
50 	 * largest unsigned integral type is, or what random insanity the local
51 	 * C compiler will perpetrate, doing the comparison in a portable way
52 	 * is flatly impossible.  So, put an fairly unreasonable limit on it,
53 	 * I don't want to be dropping core here.
54 	 */
55 #define	MEGABYTE	1048576
56 	if (sb.st_size > MEGABYTE) {
57 		errno = ENOMEM;
58 		goto err;
59 	}
60 
61 	MALLOC(sp, bp, (size_t)sb.st_size + 1);
62 	if (bp == NULL) {
63 		(void)close(fd);
64 		return (1);
65 	}
66 	bp[sb.st_size] = '\0';
67 
68 	/* Read the file into memory. */
69 	len = read(fd, bp, (int)sb.st_size);
70 	(void)close(fd);
71 	if (len == -1 || len != sb.st_size) {
72 		if (len != sb.st_size)
73 			errno = EIO;
74 		free(bp);
75 err:		msgq_str(sp, M_SYSERR, name, "%s");
76 		return (1);
77 	}
78 
79 	/* Put it on the ex queue. */
80 	return (ex_run_str(sp, name, bp, (size_t)sb.st_size, 1, 1));
81 }
82 
83 /*
84  * ex_source -- :source file
85  *	Execute ex commands from a file.
86  *
87  * PUBLIC: int ex_source(SCR *, EXCMD *);
88  */
89 int
90 ex_source(SCR *sp, EXCMD *cmdp)
91 {
92 	char *name;
93 	int fd;
94 
95 	name = cmdp->argv[0]->bp;
96 	if ((fd = open(name, O_RDONLY, 0)) >= 0)
97 		return (ex_sourcefd(sp, cmdp, fd));
98 
99 	msgq_str(sp, M_SYSERR, name, "%s");
100 	return (1);
101 }
102