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