io.c revision 1.1 1 1.1 alm /* io.c: This file contains the i/o routines for the ed line editor */
2 1.1 alm /*-
3 1.1 alm * Copyright (c) 1993 Andrew Moore, Talke Studio.
4 1.1 alm * All rights reserved.
5 1.1 alm *
6 1.1 alm * Redistribution and use in source and binary forms, with or without
7 1.1 alm * modification, are permitted provided that the following conditions
8 1.1 alm * are met:
9 1.1 alm * 1. Redistributions of source code must retain the above copyright
10 1.1 alm * notice, this list of conditions and the following disclaimer.
11 1.1 alm * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 alm * notice, this list of conditions and the following disclaimer in the
13 1.1 alm * documentation and/or other materials provided with the distribution.
14 1.1 alm *
15 1.1 alm * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 1.1 alm * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 1.1 alm * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 1.1 alm * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 alm * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 alm * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 1.1 alm * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 alm * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 alm * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 alm * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 alm * SUCH DAMAGE.
26 1.1 alm */
27 1.1 alm
28 1.1 alm #ifndef lint
29 1.1 alm static char *rcsid = "@(#)io.c,v 1.1 1994/02/01 00:34:41 alm Exp";
30 1.1 alm #endif /* not lint */
31 1.1 alm
32 1.1 alm #include "ed.h"
33 1.1 alm
34 1.1 alm
35 1.1 alm extern int scripted;
36 1.1 alm
37 1.1 alm /* read_file: read a named file/pipe into the buffer; return line count */
38 1.1 alm long
39 1.1 alm read_file(fn, n)
40 1.1 alm char *fn;
41 1.1 alm long n;
42 1.1 alm {
43 1.1 alm FILE *fp;
44 1.1 alm long size;
45 1.1 alm
46 1.1 alm
47 1.1 alm fp = (*fn == '!') ? popen(fn + 1, "r") : fopen(strip_escapes(fn), "r");
48 1.1 alm if (fp == NULL) {
49 1.1 alm fprintf(stderr, "%s: %s\n", fn, strerror(errno));
50 1.1 alm sprintf(errmsg, "cannot open input file");
51 1.1 alm return ERR;
52 1.1 alm } else if ((size = read_stream(fp, n)) < 0)
53 1.1 alm return ERR;
54 1.1 alm else if (((*fn == '!') ? pclose(fp) : fclose(fp)) < 0) {
55 1.1 alm fprintf(stderr, "%s: %s\n", fn, strerror(errno));
56 1.1 alm sprintf(errmsg, "cannot close input file");
57 1.1 alm return ERR;
58 1.1 alm }
59 1.1 alm fprintf(stderr, !scripted ? "%lu\n" : "", size);
60 1.1 alm return current_addr - n;
61 1.1 alm }
62 1.1 alm
63 1.1 alm
64 1.1 alm extern int des;
65 1.1 alm
66 1.1 alm char *sbuf; /* file i/o buffer */
67 1.1 alm int sbufsz; /* file i/o buffer size */
68 1.1 alm int newline_added; /* if set, newline appended to input file */
69 1.1 alm
70 1.1 alm /* read_stream: read a stream into the editor buffer; return status */
71 1.1 alm long
72 1.1 alm read_stream(fp, n)
73 1.1 alm FILE *fp;
74 1.1 alm long n;
75 1.1 alm {
76 1.1 alm line_t *lp = get_addressed_line_node(n);
77 1.1 alm undo_t *up = NULL;
78 1.1 alm unsigned long size = 0;
79 1.1 alm int o_newline_added = newline_added;
80 1.1 alm int o_isbinary = isbinary;
81 1.1 alm int appended = (n == addr_last);
82 1.1 alm int len;
83 1.1 alm
84 1.1 alm isbinary = newline_added = 0;
85 1.1 alm if (des)
86 1.1 alm init_des_cipher();
87 1.1 alm for (current_addr = n; (len = get_stream_line(fp)) > 0; size += len) {
88 1.1 alm SPL1();
89 1.1 alm if (put_sbuf_line(sbuf) == NULL) {
90 1.1 alm SPL0();
91 1.1 alm return ERR;
92 1.1 alm }
93 1.1 alm lp = lp->q_forw;
94 1.1 alm if (up)
95 1.1 alm up->t = lp;
96 1.1 alm else if ((up = push_undo_stack(UADD, current_addr,
97 1.1 alm current_addr)) == NULL) {
98 1.1 alm SPL0();
99 1.1 alm return ERR;
100 1.1 alm }
101 1.1 alm SPL0();
102 1.1 alm }
103 1.1 alm if (len < 0)
104 1.1 alm return ERR;
105 1.1 alm if (appended && size && o_isbinary && o_newline_added)
106 1.1 alm fputs("newline inserted\n", stderr);
107 1.1 alm else if (newline_added && (!appended || !isbinary && !o_isbinary))
108 1.1 alm fputs("newline appended\n", stderr);
109 1.1 alm if (isbinary && newline_added && !appended)
110 1.1 alm size += 1;
111 1.1 alm if (!size)
112 1.1 alm newline_added = 1;
113 1.1 alm newline_added = appended ? newline_added : o_newline_added;
114 1.1 alm isbinary = isbinary | o_isbinary;
115 1.1 alm if (des)
116 1.1 alm size += 8 - size % 8; /* adjust DES size */
117 1.1 alm return size;
118 1.1 alm }
119 1.1 alm
120 1.1 alm
121 1.1 alm /* get_stream_line: read a line of text from a stream; return line length */
122 1.1 alm int
123 1.1 alm get_stream_line(fp)
124 1.1 alm FILE *fp;
125 1.1 alm {
126 1.1 alm register int c;
127 1.1 alm register int i = 0;
128 1.1 alm
129 1.1 alm while (((c = des ? get_des_char(fp) : getc(fp)) != EOF || !feof(fp) &&
130 1.1 alm !ferror(fp)) && c != '\n') {
131 1.1 alm REALLOC(sbuf, sbufsz, i + 1, ERR);
132 1.1 alm if (!(sbuf[i++] = c))
133 1.1 alm isbinary = 1;
134 1.1 alm }
135 1.1 alm REALLOC(sbuf, sbufsz, i + 2, ERR);
136 1.1 alm if (c == '\n')
137 1.1 alm sbuf[i++] = c;
138 1.1 alm else if (ferror(fp)) {
139 1.1 alm fprintf(stderr, "%s\n", strerror(errno));
140 1.1 alm sprintf(errmsg, "cannot read input file");
141 1.1 alm return ERR;
142 1.1 alm } else if (i) {
143 1.1 alm sbuf[i++] = '\n';
144 1.1 alm newline_added = 1;
145 1.1 alm }
146 1.1 alm sbuf[i] = '\0';
147 1.1 alm return (isbinary && newline_added && i) ? --i : i;
148 1.1 alm }
149 1.1 alm
150 1.1 alm
151 1.1 alm /* write_file: write a range of lines to a named file/pipe; return line count */
152 1.1 alm long
153 1.1 alm write_file(fn, mode, n, m)
154 1.1 alm char *fn;
155 1.1 alm char *mode;
156 1.1 alm long n;
157 1.1 alm long m;
158 1.1 alm {
159 1.1 alm FILE *fp;
160 1.1 alm long size;
161 1.1 alm
162 1.1 alm fp = (*fn == '!') ? popen(fn+1, "w") : fopen(strip_escapes(fn), mode);
163 1.1 alm if (fp == NULL) {
164 1.1 alm fprintf(stderr, "%s: %s\n", fn, strerror(errno));
165 1.1 alm sprintf(errmsg, "cannot open output file");
166 1.1 alm return ERR;
167 1.1 alm } else if ((size = write_stream(fp, n, m)) < 0)
168 1.1 alm return ERR;
169 1.1 alm else if (((*fn == '!') ? pclose(fp) : fclose(fp)) < 0) {
170 1.1 alm fprintf(stderr, "%s: %s\n", fn, strerror(errno));
171 1.1 alm sprintf(errmsg, "cannot close output file");
172 1.1 alm return ERR;
173 1.1 alm }
174 1.1 alm fprintf(stderr, !scripted ? "%lu\n" : "", size);
175 1.1 alm return n ? m - n + 1 : 0;
176 1.1 alm }
177 1.1 alm
178 1.1 alm
179 1.1 alm /* write_stream: write a range of lines to a stream; return status */
180 1.1 alm long
181 1.1 alm write_stream(fp, n, m)
182 1.1 alm FILE *fp;
183 1.1 alm long n;
184 1.1 alm long m;
185 1.1 alm {
186 1.1 alm line_t *lp = get_addressed_line_node(n);
187 1.1 alm unsigned long size = 0;
188 1.1 alm char *s;
189 1.1 alm int len;
190 1.1 alm
191 1.1 alm if (des)
192 1.1 alm init_des_cipher();
193 1.1 alm for (; n && n <= m; n++, lp = lp->q_forw) {
194 1.1 alm if ((s = get_sbuf_line(lp)) == NULL)
195 1.1 alm return ERR;
196 1.1 alm len = lp->len;
197 1.1 alm if (n != addr_last || !isbinary || !newline_added)
198 1.1 alm s[len++] = '\n';
199 1.1 alm if (put_stream_line(fp, s, len) < 0)
200 1.1 alm return ERR;
201 1.1 alm size += len;
202 1.1 alm }
203 1.1 alm if (des) {
204 1.1 alm flush_des_file(fp); /* flush buffer */
205 1.1 alm size += 8 - size % 8; /* adjust DES size */
206 1.1 alm }
207 1.1 alm return size;
208 1.1 alm }
209 1.1 alm
210 1.1 alm
211 1.1 alm /* put_stream_line: write a line of text to a stream; return status */
212 1.1 alm int
213 1.1 alm put_stream_line(fp, s, len)
214 1.1 alm FILE *fp;
215 1.1 alm char *s;
216 1.1 alm int len;
217 1.1 alm {
218 1.1 alm while (len--)
219 1.1 alm if ((des ? put_des_char(*s++, fp) : fputc(*s++, fp)) < 0) {
220 1.1 alm fprintf(stderr, "%s\n", strerror(errno));
221 1.1 alm sprintf(errmsg, "cannot write file");
222 1.1 alm return ERR;
223 1.1 alm }
224 1.1 alm return 0;
225 1.1 alm }
226 1.1 alm
227 1.1 alm /* get_extended_line: get a an extended line from stdin */
228 1.1 alm char *
229 1.1 alm get_extended_line(sizep, nonl)
230 1.1 alm int *sizep;
231 1.1 alm int nonl;
232 1.1 alm {
233 1.1 alm static char *cvbuf = NULL; /* buffer */
234 1.1 alm static int cvbufsz = 0; /* buffer size */
235 1.1 alm
236 1.1 alm int l, n;
237 1.1 alm char *t = ibufp;
238 1.1 alm
239 1.1 alm while (*t++ != '\n')
240 1.1 alm ;
241 1.1 alm if ((l = t - ibufp) < 2 || !has_trailing_escape(ibufp, ibufp + l - 1)) {
242 1.1 alm *sizep = l;
243 1.1 alm return ibufp;
244 1.1 alm }
245 1.1 alm *sizep = -1;
246 1.1 alm REALLOC(cvbuf, cvbufsz, l, NULL);
247 1.1 alm memcpy(cvbuf, ibufp, l);
248 1.1 alm *(cvbuf + --l - 1) = '\n'; /* strip trailing esc */
249 1.1 alm if (nonl) l--; /* strip newline */
250 1.1 alm for (;;) {
251 1.1 alm if ((n = get_tty_line()) < 0)
252 1.1 alm return NULL;
253 1.1 alm else if (n == 0 || ibuf[n - 1] != '\n') {
254 1.1 alm sprintf(errmsg, "unexpected end-of-file");
255 1.1 alm return NULL;
256 1.1 alm }
257 1.1 alm REALLOC(cvbuf, cvbufsz, l + n, NULL);
258 1.1 alm memcpy(cvbuf + l, ibuf, n);
259 1.1 alm l += n;
260 1.1 alm if (n < 2 || !has_trailing_escape(cvbuf, cvbuf + l - 1))
261 1.1 alm break;
262 1.1 alm *(cvbuf + --l - 1) = '\n'; /* strip trailing esc */
263 1.1 alm if (nonl) l--; /* strip newline */
264 1.1 alm }
265 1.1 alm REALLOC(cvbuf, cvbufsz, l + 1, NULL);
266 1.1 alm cvbuf[l] = '\0';
267 1.1 alm *sizep = l;
268 1.1 alm return cvbuf;
269 1.1 alm }
270 1.1 alm
271 1.1 alm
272 1.1 alm /* get_tty_line: read a line of text from stdin; return line length */
273 1.1 alm int
274 1.1 alm get_tty_line()
275 1.1 alm {
276 1.1 alm register int oi = 0;
277 1.1 alm register int i = 0;
278 1.1 alm int c;
279 1.1 alm
280 1.1 alm for (;;)
281 1.1 alm switch (c = getchar()) {
282 1.1 alm default:
283 1.1 alm oi = 0;
284 1.1 alm REALLOC(ibuf, ibufsz, i + 2, ERR);
285 1.1 alm if (!(ibuf[i++] = c)) isbinary = 1;
286 1.1 alm if (c != '\n')
287 1.1 alm continue;
288 1.1 alm lineno++;
289 1.1 alm ibuf[i] = '\0';
290 1.1 alm ibufp = ibuf;
291 1.1 alm return i;
292 1.1 alm case EOF:
293 1.1 alm if (ferror(stdin)) {
294 1.1 alm fprintf(stderr, "stdin: %s\n", strerror(errno));
295 1.1 alm sprintf(errmsg, "cannot read stdin");
296 1.1 alm clearerr(stdin);
297 1.1 alm ibufp = NULL;
298 1.1 alm return ERR;
299 1.1 alm } else {
300 1.1 alm clearerr(stdin);
301 1.1 alm if (i != oi) {
302 1.1 alm oi = i;
303 1.1 alm continue;
304 1.1 alm } else if (i)
305 1.1 alm ibuf[i] = '\0';
306 1.1 alm ibufp = ibuf;
307 1.1 alm return i;
308 1.1 alm }
309 1.1 alm }
310 1.1 alm }
311 1.1 alm
312 1.1 alm
313 1.1 alm
314 1.1 alm #define ESCAPES "\a\b\f\n\r\t\v\\"
315 1.1 alm #define ESCCHARS "abfnrtv\\"
316 1.1 alm
317 1.1 alm extern int rows;
318 1.1 alm extern int cols;
319 1.1 alm
320 1.1 alm /* put_tty_line: print text to stdout */
321 1.1 alm int
322 1.1 alm put_tty_line(s, l, n, gflag)
323 1.1 alm char *s;
324 1.1 alm int l;
325 1.1 alm long n;
326 1.1 alm int gflag;
327 1.1 alm {
328 1.1 alm int col = 0;
329 1.1 alm int lc = 0;
330 1.1 alm char *cp;
331 1.1 alm
332 1.1 alm if (gflag & GNP) {
333 1.1 alm printf("%ld\t", n);
334 1.1 alm col = 8;
335 1.1 alm }
336 1.1 alm for (; l--; s++) {
337 1.1 alm if ((gflag & GLS) && ++col > cols) {
338 1.1 alm fputs("\\\n", stdout);
339 1.1 alm col = 1;
340 1.1 alm #ifndef BACKWARDS
341 1.1 alm if (!scripted && !isglobal && ++lc > rows) {
342 1.1 alm lc = 0;
343 1.1 alm fputs("Press <RETURN> to continue... ", stdout);
344 1.1 alm fflush(stdout);
345 1.1 alm if (get_tty_line() < 0)
346 1.1 alm return ERR;
347 1.1 alm }
348 1.1 alm #endif
349 1.1 alm }
350 1.1 alm if (gflag & GLS) {
351 1.1 alm if (31 < *s && *s < 127 && *s != '\\')
352 1.1 alm putchar(*s);
353 1.1 alm else {
354 1.1 alm putchar('\\');
355 1.1 alm col++;
356 1.1 alm if (*s && (cp = strchr(ESCAPES, *s)) != NULL)
357 1.1 alm putchar(ESCCHARS[cp - ESCAPES]);
358 1.1 alm else {
359 1.1 alm putchar((((unsigned char) *s & 0300) >> 6) + '0');
360 1.1 alm putchar((((unsigned char) *s & 070) >> 3) + '0');
361 1.1 alm putchar(((unsigned char) *s & 07) + '0');
362 1.1 alm col += 2;
363 1.1 alm }
364 1.1 alm }
365 1.1 alm
366 1.1 alm } else
367 1.1 alm putchar(*s);
368 1.1 alm }
369 1.1 alm #ifndef BACKWARDS
370 1.1 alm if (gflag & GLS)
371 1.1 alm putchar('$');
372 1.1 alm #endif
373 1.1 alm putchar('\n');
374 1.1 alm return 0;
375 1.1 alm }
376