xref: /openbsd-src/usr.bin/vi/common/cut.c (revision 8500990981f885cbe5e6a4958549cacc238b5ae6)
1 /*	$OpenBSD: cut.c,v 1.7 2002/02/16 21:27:56 millert 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 #ifndef lint
15 static const char sccsid[] = "@(#)cut.c	10.10 (Berkeley) 9/15/96";
16 #endif /* not lint */
17 
18 #include <sys/types.h>
19 #include <sys/queue.h>
20 
21 #include <bitstring.h>
22 #include <ctype.h>
23 #include <errno.h>
24 #include <fcntl.h>
25 #include <limits.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <string.h>
29 
30 #include "common.h"
31 
32 static void	cb_rotate(SCR *);
33 
34 /*
35  * cut --
36  *	Put a range of lines/columns into a TEXT buffer.
37  *
38  * There are two buffer areas, both found in the global structure.  The first
39  * is the linked list of all the buffers the user has named, the second is the
40  * unnamed buffer storage.  There is a pointer, too, which is the current
41  * default buffer, i.e. it may point to the unnamed buffer or a named buffer
42  * depending on into what buffer the last text was cut.  Logically, in both
43  * delete and yank operations, if the user names a buffer, the text is cut
44  * into it.  If it's a delete of information on more than a single line, the
45  * contents of the numbered buffers are rotated up one, the contents of the
46  * buffer named '9' are discarded, and the text is cut into the buffer named
47  * '1'.  The text is always cut into the unnamed buffer.
48  *
49  * In all cases, upper-case buffer names are the same as lower-case names,
50  * with the exception that they cause the buffer to be appended to instead
51  * of replaced.  Note, however, that if text is appended to a buffer, the
52  * default buffer only contains the appended text, not the entire contents
53  * of the buffer.
54  *
55  * !!!
56  * The contents of the default buffer would disappear after most operations
57  * in historic vi.  It's unclear that this is useful, so we don't bother.
58  *
59  * When users explicitly cut text into the numeric buffers, historic vi became
60  * genuinely strange.  I've never been able to figure out what was supposed to
61  * happen.  It behaved differently if you deleted text than if you yanked text,
62  * and, in the latter case, the text was appended to the buffer instead of
63  * replacing the contents.  Hopefully it's not worth getting right, and here
64  * we just treat the numeric buffers like any other named buffer.
65  *
66  * PUBLIC: int cut(SCR *, CHAR_T *, MARK *, MARK *, int);
67  */
68 int
69 cut(sp, namep, fm, tm, flags)
70 	SCR *sp;
71 	CHAR_T *namep;
72 	MARK *fm, *tm;
73 	int flags;
74 {
75 	CB *cbp;
76 	CHAR_T name;
77 	recno_t lno;
78 	int append, copy_one, copy_def;
79 
80 	/*
81 	 * If the user specified a buffer, put it there.  (This may require
82 	 * a copy into the numeric buffers.  We do the copy so that we don't
83 	 * have to reference count and so we don't have to deal with things
84 	 * like appends to buffers that are used multiple times.)
85 	 *
86 	 * Otherwise, if it's supposed to be put in a numeric buffer (usually
87 	 * a delete) put it there.  The rules for putting things in numeric
88 	 * buffers were historically a little strange.  There were three cases.
89 	 *
90 	 *	1: Some motions are always line mode motions, which means
91 	 *	   that the cut always goes into the numeric buffers.
92 	 *	2: Some motions aren't line mode motions, e.g. d10w, but
93 	 *	   can cross line boundaries.  For these commands, if the
94 	 *	   cut crosses a line boundary, it goes into the numeric
95 	 *	   buffers.  This includes most of the commands.
96 	 *	3: Some motions aren't line mode motions, e.g. d`<char>,
97 	 *	   but always go into the numeric buffers, regardless.  This
98 	 *	   was the commands: % ` / ? ( ) N n { } -- and nvi adds ^A.
99 	 *
100 	 * Otherwise, put it in the unnamed buffer.
101 	 */
102 	append = copy_one = copy_def = 0;
103 	if (namep != NULL) {
104 		name = *namep;
105 		if (LF_ISSET(CUT_NUMREQ) || LF_ISSET(CUT_NUMOPT) &&
106 		    (LF_ISSET(CUT_LINEMODE) || fm->lno != tm->lno)) {
107 			copy_one = 1;
108 			cb_rotate(sp);
109 		}
110 		if ((append = isupper(name)) == 1) {
111 			if (!copy_one)
112 				copy_def = 1;
113 			name = tolower(name);
114 		}
115 namecb:		CBNAME(sp, cbp, name);
116 	} else if (LF_ISSET(CUT_NUMREQ) || LF_ISSET(CUT_NUMOPT) &&
117 	    (LF_ISSET(CUT_LINEMODE) || fm->lno != tm->lno)) {
118 		name = '1';
119 		cb_rotate(sp);
120 		goto namecb;
121 	} else
122 		cbp = &sp->gp->dcb_store;
123 
124 copyloop:
125 	/*
126 	 * If this is a new buffer, create it and add it into the list.
127 	 * Otherwise, if it's not an append, free its current contents.
128 	 */
129 	if (cbp == NULL) {
130 		CALLOC_RET(sp, cbp, CB *, 1, sizeof(CB));
131 		cbp->name = name;
132 		CIRCLEQ_INIT(&cbp->textq);
133 		LIST_INSERT_HEAD(&sp->gp->cutq, cbp, q);
134 	} else if (!append) {
135 		text_lfree(&cbp->textq);
136 		cbp->len = 0;
137 		cbp->flags = 0;
138 	}
139 
140 
141 	/* In line mode, it's pretty easy, just cut the lines. */
142 	if (LF_ISSET(CUT_LINEMODE)) {
143 		cbp->flags |= CB_LMODE;
144 		for (lno = fm->lno; lno <= tm->lno; ++lno)
145 			if (cut_line(sp, lno, 0, CUT_LINE_TO_EOL, cbp))
146 				goto cut_line_err;
147 	} else {
148 		/*
149 		 * Get the first line.  A length of CUT_LINE_TO_EOL causes
150 		 * cut_line() to cut from the MARK to the end of the line.
151 		 */
152 		if (cut_line(sp, fm->lno, fm->cno, fm->lno != tm->lno ?
153 		    CUT_LINE_TO_EOL : (tm->cno - fm->cno) + 1, cbp))
154 			goto cut_line_err;
155 
156 		/* Get the intermediate lines. */
157 		for (lno = fm->lno; ++lno < tm->lno;)
158 			if (cut_line(sp, lno, 0, CUT_LINE_TO_EOL, cbp))
159 				goto cut_line_err;
160 
161 		/* Get the last line. */
162 		if (tm->lno != fm->lno &&
163 		    cut_line(sp, lno, 0, tm->cno + 1, cbp))
164 			goto cut_line_err;
165 	}
166 
167 	append = 0;		/* Only append to the named buffer. */
168 	sp->gp->dcbp = cbp;	/* Repoint the default buffer on each pass. */
169 
170 	if (copy_one) {		/* Copy into numeric buffer 1. */
171 		name = '1';
172 		CBNAME(sp, cbp, name);
173 		copy_one = 0;
174 		goto copyloop;
175 	}
176 	if (copy_def) {		/* Copy into the default buffer. */
177 		cbp = &sp->gp->dcb_store;
178 		copy_def = 0;
179 		goto copyloop;
180 	}
181 	return (0);
182 
183 cut_line_err:
184 	text_lfree(&cbp->textq);
185 	cbp->len = 0;
186 	cbp->flags = 0;
187 	return (1);
188 }
189 
190 /*
191  * cb_rotate --
192  *	Rotate the numbered buffers up one.
193  */
194 static void
195 cb_rotate(sp)
196 	SCR *sp;
197 {
198 	CB *cbp, *del_cbp;
199 
200 	del_cbp = NULL;
201 	for (cbp = sp->gp->cutq.lh_first; cbp != NULL; cbp = cbp->q.le_next)
202 		switch(cbp->name) {
203 		case '1':
204 			cbp->name = '2';
205 			break;
206 		case '2':
207 			cbp->name = '3';
208 			break;
209 		case '3':
210 			cbp->name = '4';
211 			break;
212 		case '4':
213 			cbp->name = '5';
214 			break;
215 		case '5':
216 			cbp->name = '6';
217 			break;
218 		case '6':
219 			cbp->name = '7';
220 			break;
221 		case '7':
222 			cbp->name = '8';
223 			break;
224 		case '8':
225 			cbp->name = '9';
226 			break;
227 		case '9':
228 			del_cbp = cbp;
229 			break;
230 		}
231 	if (del_cbp != NULL) {
232 		LIST_REMOVE(del_cbp, q);
233 		text_lfree(&del_cbp->textq);
234 		free(del_cbp);
235 	}
236 }
237 
238 /*
239  * cut_line --
240  *	Cut a portion of a single line.
241  *
242  * PUBLIC: int cut_line(SCR *, recno_t, size_t, size_t, CB *);
243  */
244 int
245 cut_line(sp, lno, fcno, clen, cbp)
246 	SCR *sp;
247 	recno_t lno;
248 	size_t fcno, clen;
249 	CB *cbp;
250 {
251 	TEXT *tp;
252 	size_t len;
253 	char *p;
254 
255 	/* Get the line. */
256 	if (db_get(sp, lno, DBG_FATAL, &p, &len))
257 		return (1);
258 
259 	/* Create a TEXT structure that can hold the entire line. */
260 	if ((tp = text_init(sp, NULL, 0, len)) == NULL)
261 		return (1);
262 
263 	/*
264 	 * If the line isn't empty and it's not the entire line,
265 	 * copy the portion we want, and reset the TEXT length.
266 	 */
267 	if (len != 0) {
268 		if (clen == CUT_LINE_TO_EOL)
269 			clen = len - fcno;
270 		memcpy(tp->lb, p + fcno, clen);
271 		tp->len = clen;
272 	}
273 
274 	/* Append to the end of the cut buffer. */
275 	CIRCLEQ_INSERT_TAIL(&cbp->textq, tp, q);
276 	cbp->len += tp->len;
277 
278 	return (0);
279 }
280 
281 /*
282  * cut_close --
283  *	Discard all cut buffers.
284  *
285  * PUBLIC: void cut_close(GS *);
286  */
287 void
288 cut_close(gp)
289 	GS *gp;
290 {
291 	CB *cbp;
292 
293 	/* Free cut buffer list. */
294 	while ((cbp = gp->cutq.lh_first) != NULL) {
295 		if (cbp->textq.cqh_first != (void *)&cbp->textq)
296 			text_lfree(&cbp->textq);
297 		LIST_REMOVE(cbp, q);
298 		free(cbp);
299 	}
300 
301 	/* Free default cut storage. */
302 	cbp = &gp->dcb_store;
303 	if (cbp->textq.cqh_first != (void *)&cbp->textq)
304 		text_lfree(&cbp->textq);
305 }
306 
307 /*
308  * text_init --
309  *	Allocate a new TEXT structure.
310  *
311  * PUBLIC: TEXT *text_init(SCR *, const char *, size_t, size_t);
312  */
313 TEXT *
314 text_init(sp, p, len, total_len)
315 	SCR *sp;
316 	const char *p;
317 	size_t len, total_len;
318 {
319 	TEXT *tp;
320 
321 	CALLOC(sp, tp, TEXT *, 1, sizeof(TEXT));
322 	if (tp == NULL)
323 		return (NULL);
324 	/* ANSI C doesn't define a call to malloc(3) for 0 bytes. */
325 	if ((tp->lb_len = total_len) != 0) {
326 		MALLOC(sp, tp->lb, CHAR_T *, tp->lb_len);
327 		if (tp->lb == NULL) {
328 			free(tp);
329 			return (NULL);
330 		}
331 		if (p != NULL && len != 0)
332 			memcpy(tp->lb, p, len);
333 	}
334 	tp->len = len;
335 	return (tp);
336 }
337 
338 /*
339  * text_lfree --
340  *	Free a chain of text structures.
341  *
342  * PUBLIC: void text_lfree(TEXTH *);
343  */
344 void
345 text_lfree(headp)
346 	TEXTH *headp;
347 {
348 	TEXT *tp;
349 
350 	while ((tp = headp->cqh_first) != (void *)headp) {
351 		CIRCLEQ_REMOVE(headp, tp, q);
352 		text_free(tp);
353 	}
354 }
355 
356 /*
357  * text_free --
358  *	Free a text structure.
359  *
360  * PUBLIC: void text_free(TEXT *);
361  */
362 void
363 text_free(tp)
364 	TEXT *tp;
365 {
366 	if (tp->lb != NULL)
367 		free(tp->lb);
368 	free(tp);
369 }
370