xref: /freebsd/bin/ed/buf.c (revision 417ed37975261df51f61d13e179ad04d8f4839c7)
1 /* buf.c: This file contains the scratch-file buffer rountines for the
2    ed line editor. */
3 /*-
4  * Copyright (c) 1993 Andrew Moore, Talke Studio.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  *
28  *	$Id$
29  */
30 #ifndef lint
31 static char *rcsid = "@(#)buf.c,v 1.4 1994/02/01 00:34:35 alm Exp";
32 #endif /* not lint */
33 
34 #include <sys/file.h>
35 #include <sys/stat.h>
36 
37 #include "ed.h"
38 
39 
40 FILE *sfp;				/* scratch file pointer */
41 off_t sfseek;				/* scratch file position */
42 int seek_write;				/* seek before writing */
43 line_t buffer_head;			/* incore buffer */
44 
45 /* get_sbuf_line: get a line of text from the scratch file; return pointer
46    to the text */
47 char *
48 get_sbuf_line(lp)
49 	line_t *lp;
50 {
51 	static char *sfbuf = NULL;	/* buffer */
52 	static int sfbufsz = 0;		/* buffer size */
53 
54 	int len, ct;
55 
56 	if (lp == &buffer_head)
57 		return NULL;
58 	seek_write = 1;				/* force seek on write */
59 	/* out of position */
60 	if (sfseek != lp->seek) {
61 		sfseek = lp->seek;
62 		if (fseek(sfp, sfseek, SEEK_SET) < 0) {
63 			fprintf(stderr, "%s\n", strerror(errno));
64 			sprintf(errmsg, "cannot seek temp file");
65 			return NULL;
66 		}
67 	}
68 	len = lp->len;
69 	REALLOC(sfbuf, sfbufsz, len + 1, NULL);
70 	if ((ct = fread(sfbuf, sizeof(char), len, sfp)) <  0 || ct != len) {
71 		fprintf(stderr, "%s\n", strerror(errno));
72 		sprintf(errmsg, "cannot read temp file");
73 		return NULL;
74 	}
75 	sfseek += len;				/* update file position */
76 	sfbuf[len] = '\0';
77 	return sfbuf;
78 }
79 
80 
81 /* put_sbuf_line: write a line of text to the scratch file and add a line node
82    to the editor buffer;  return a pointer to the end of the text */
83 char *
84 put_sbuf_line(cs)
85 	char *cs;
86 {
87 	line_t *lp;
88 	int len, ct;
89 	char *s;
90 
91 	if ((lp = (line_t *) malloc(sizeof(line_t))) == NULL) {
92 		fprintf(stderr, "%s\n", strerror(errno));
93 		sprintf(errmsg, "out of memory");
94 		return NULL;
95 	}
96 	/* assert: cs is '\n' terminated */
97 	for (s = cs; *s != '\n'; s++)
98 		;
99 	if (s - cs >= LINECHARS) {
100 		sprintf(errmsg, "line too long");
101 		return NULL;
102 	}
103 	len = s - cs;
104 	/* out of position */
105 	if (seek_write) {
106 		if (fseek(sfp, 0L, SEEK_END) < 0) {
107 			fprintf(stderr, "%s\n", strerror(errno));
108 			sprintf(errmsg, "cannot seek temp file");
109 			return NULL;
110 		}
111 		sfseek = ftell(sfp);
112 		seek_write = 0;
113 	}
114 	/* assert: SPL1() */
115 	if ((ct = fwrite(cs, sizeof(char), len, sfp)) < 0 || ct != len) {
116 		sfseek = -1;
117 		fprintf(stderr, "%s\n", strerror(errno));
118 		sprintf(errmsg, "cannot write temp file");
119 		return NULL;
120 	}
121 	lp->len = len;
122 	lp->seek  = sfseek;
123 	add_line_node(lp);
124 	sfseek += len;			/* update file position */
125 	return ++s;
126 }
127 
128 
129 /* add_line_node: add a line node in the editor buffer after the current line */
130 void
131 add_line_node(lp)
132 	line_t *lp;
133 {
134 	line_t *cp;
135 
136 	cp = get_addressed_line_node(current_addr);				/* this get_addressed_line_node last! */
137 	INSQUE(lp, cp);
138 	addr_last++;
139 	current_addr++;
140 }
141 
142 
143 /* get_line_node_addr: return line number of pointer */
144 long
145 get_line_node_addr(lp)
146 	line_t *lp;
147 {
148 	line_t *cp = &buffer_head;
149 	long n = 0;
150 
151 	while (cp != lp && (cp = cp->q_forw) != &buffer_head)
152 		n++;
153 	if (n && cp == &buffer_head) {
154 		sprintf(errmsg, "invalid address");
155 		return ERR;
156 	 }
157 	 return n;
158 }
159 
160 
161 /* get_addressed_line_node: return pointer to a line node in the editor buffer */
162 line_t *
163 get_addressed_line_node(n)
164 	long n;
165 {
166 	static line_t *lp = &buffer_head;
167 	static long on = 0;
168 
169 	SPL1();
170 	if (n > on)
171 		if (n <= (on + addr_last) >> 1)
172 			for (; on < n; on++)
173 				lp = lp->q_forw;
174 		else {
175 			lp = buffer_head.q_back;
176 			for (on = addr_last; on > n; on--)
177 				lp = lp->q_back;
178 		}
179 	else
180 		if (n >= on >> 1)
181 			for (; on > n; on--)
182 				lp = lp->q_back;
183 		else {
184 			lp = &buffer_head;
185 			for (on = 0; on < n; on++)
186 				lp = lp->q_forw;
187 		}
188 	SPL0();
189 	return lp;
190 }
191 
192 
193 extern int newline_added;
194 
195 char sfn[15] = "";				/* scratch file name */
196 
197 /* open_sbuf: open scratch file */
198 int
199 open_sbuf()
200 {
201 	int u;
202 
203 	isbinary = newline_added = 0;
204 	u = umask(077);
205 	strcpy(sfn, "/tmp/ed.XXXXXX");
206 	if (mktemp(sfn) == NULL || (sfp = fopen(sfn, "w+")) == NULL) {
207 		fprintf(stderr, "%s: %s\n", sfn, strerror(errno));
208 		sprintf(errmsg, "cannot open temp file");
209 		umask(u);
210 		return ERR;
211 	}
212 	umask(u);
213 	return 0;
214 }
215 
216 
217 /* close_sbuf: close scratch file */
218 int
219 close_sbuf()
220 {
221 	if (sfp) {
222 		if (fclose(sfp) < 0) {
223 			fprintf(stderr, "%s: %s\n", sfn, strerror(errno));
224 			sprintf(errmsg, "cannot close temp file");
225 			return ERR;
226 		}
227 		sfp = NULL;
228 		unlink(sfn);
229 	}
230 	sfseek = seek_write = 0;
231 	return 0;
232 }
233 
234 
235 /* quit: remove_lines scratch file and exit */
236 void
237 quit(n)
238 	int n;
239 {
240 	if (sfp) {
241 		fclose(sfp);
242 		unlink(sfn);
243 	}
244 	exit(n);
245 }
246 
247 
248 unsigned char ctab[256];		/* character translation table */
249 
250 /* init_buffers: open scratch buffer; initialize line queue */
251 void
252 init_buffers()
253 {
254 	int i = 0;
255 
256 	/* Read stdin one character at a time to avoid i/o contention
257 	   with shell escapes invoked by nonterminal input, e.g.,
258 	   ed - <<EOF
259 	   !cat
260 	   hello, world
261 	   EOF */
262 	setbuffer(stdin, stdinbuf, 1);
263 	if (open_sbuf() < 0)
264 		quit(2);
265 	REQUE(&buffer_head, &buffer_head);
266 	for (i = 0; i < 256; i++)
267 		ctab[i] = i;
268 }
269 
270 
271 /* translit_text: translate characters in a string */
272 char *
273 translit_text(s, len, from, to)
274 	char *s;
275 	int len;
276 	int from;
277 	int to;
278 {
279 	static int i = 0;
280 
281 	unsigned char *us;
282 
283 	ctab[i] = i;			/* restore table to initial state */
284 	ctab[i = from] = to;
285 	for (us = (unsigned char *) s; len-- > 0; us++)
286 		*us = ctab[*us];
287 	return s;
288 }
289