blob: 787f62590ee769b69be7868537d1fa46f175d648 [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * undo.c: multi level undo facility
12 *
13 * The saved lines are stored in a list of lists (one for each buffer):
14 *
15 * b_u_oldhead------------------------------------------------+
16 * |
17 * V
18 * +--------------+ +--------------+ +--------------+
19 * b_u_newhead--->| u_header | | u_header | | u_header |
20 * | uh_next------>| uh_next------>| uh_next---->NULL
21 * NULL<--------uh_prev |<---------uh_prev |<---------uh_prev |
22 * | uh_entry | | uh_entry | | uh_entry |
23 * +--------|-----+ +--------|-----+ +--------|-----+
24 * | | |
25 * V V V
26 * +--------------+ +--------------+ +--------------+
27 * | u_entry | | u_entry | | u_entry |
28 * | ue_next | | ue_next | | ue_next |
29 * +--------|-----+ +--------|-----+ +--------|-----+
30 * | | |
31 * V V V
32 * +--------------+ NULL NULL
33 * | u_entry |
34 * | ue_next |
35 * +--------|-----+
36 * |
37 * V
38 * etc.
39 *
40 * Each u_entry list contains the information for one undo or redo.
41 * curbuf->b_u_curhead points to the header of the last undo (the next redo),
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +000042 * or is NULL if nothing has been undone (end of the branch).
Bram Moolenaar071d4272004-06-13 20:20:40 +000043 *
Bram Moolenaar1e607892006-03-13 22:15:53 +000044 * For keeping alternate undo/redo branches the uh_alt field is used. Thus at
45 * each point in the list a branch may appear for an alternate to redo. The
46 * uh_seq field is numbered sequentially to be able to find a newer or older
47 * branch.
48 *
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +000049 * +---------------+ +---------------+
50 * b_u_oldhead --->| u_header | | u_header |
51 * | uh_alt_next ---->| uh_alt_next ----> NULL
52 * NULL <----- uh_alt_prev |<------ uh_alt_prev |
53 * | uh_prev | | uh_prev |
54 * +-----|---------+ +-----|---------+
55 * | |
56 * V V
57 * +---------------+ +---------------+
58 * | u_header | | u_header |
59 * | uh_alt_next | | uh_alt_next |
60 * b_u_newhead --->| uh_alt_prev | | uh_alt_prev |
61 * | uh_prev | | uh_prev |
62 * +-----|---------+ +-----|---------+
63 * | |
64 * V V
65 * NULL +---------------+ +---------------+
66 * | u_header | | u_header |
67 * | uh_alt_next ---->| uh_alt_next |
68 * | uh_alt_prev |<------ uh_alt_prev |
69 * | uh_prev | | uh_prev |
70 * +-----|---------+ +-----|---------+
71 * | |
72 * etc. etc.
73 *
74 *
Bram Moolenaar26a60b42005-02-22 08:49:11 +000075 * All data is allocated with U_ALLOC_LINE(), it will be freed as soon as the
76 * buffer is unloaded.
Bram Moolenaar071d4272004-06-13 20:20:40 +000077 */
78
Bram Moolenaarfecb6602007-10-01 20:54:15 +000079/* Uncomment the next line for including the u_check() function. This warns
80 * for errors in the debug information. */
81/* #define U_DEBUG 1 */
82#define UH_MAGIC 0x18dade /* value for uh_magic when in use */
83#define UE_MAGIC 0xabc123 /* value for ue_magic when in use */
84
Bram Moolenaar442b4222010-05-24 21:34:22 +020085#if defined(MSDOS) || defined(WIN16) || defined(WIN32) || defined(_WIN64)
86# include "vimio.h" /* for vim_read(), must be before vim.h */
87#endif
88
Bram Moolenaar071d4272004-06-13 20:20:40 +000089#include "vim.h"
90
Bram Moolenaar26a60b42005-02-22 08:49:11 +000091/* See below: use malloc()/free() for memory management. */
92#define U_USE_MALLOC 1
93
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +000094static void u_unch_branch __ARGS((u_header_T *uhp));
Bram Moolenaar071d4272004-06-13 20:20:40 +000095static u_entry_T *u_get_headentry __ARGS((void));
96static void u_getbot __ARGS((void));
97static int u_savecommon __ARGS((linenr_T, linenr_T, linenr_T));
98static void u_doit __ARGS((int count));
Bram Moolenaarca003e12006-03-17 23:19:38 +000099static void u_undoredo __ARGS((int undo));
Bram Moolenaardb552d602006-03-23 22:59:57 +0000100static void u_undo_end __ARGS((int did_undo, int absolute));
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000101static void u_add_time __ARGS((char_u *buf, size_t buflen, time_t tt));
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000102static void u_freeheader __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp));
Bram Moolenaar1e607892006-03-13 22:15:53 +0000103static void u_freebranch __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp));
104static void u_freeentries __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000105static void u_freeentry __ARGS((u_entry_T *, long));
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200106#ifdef FEAT_PERSISTENT_UNDO
107static void unserialize_pos __ARGS((pos_T *pos, FILE *fp));
108static void unserialize_visualinfo __ARGS((visualinfo_T *info, FILE *fp));
109static char_u *u_get_undo_file_name __ARGS((char_u *, int reading));
110static int serialize_uep __ARGS((u_entry_T *uep, FILE *fp));
111static void serialize_pos __ARGS((pos_T pos, FILE *fp));
112static void serialize_visualinfo __ARGS((visualinfo_T info, FILE *fp));
113#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000114
Bram Moolenaar26a60b42005-02-22 08:49:11 +0000115#ifdef U_USE_MALLOC
116# define U_FREE_LINE(ptr) vim_free(ptr)
117# define U_ALLOC_LINE(size) lalloc((long_u)((size) + 1), FALSE)
118#else
119static void u_free_line __ARGS((char_u *ptr, int keep));
120static char_u *u_alloc_line __ARGS((unsigned size));
121# define U_FREE_LINE(ptr) u_free_line((ptr), FALSE)
122# define U_ALLOC_LINE(size) u_alloc_line(size)
123#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000124static char_u *u_save_line __ARGS((linenr_T));
125
126static long u_newcount, u_oldcount;
127
128/*
129 * When 'u' flag included in 'cpoptions', we behave like vi. Need to remember
130 * the action that "u" should do.
131 */
132static int undo_undoes = FALSE;
133
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200134static int lastmark = 0;
135
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000136#ifdef U_DEBUG
137/*
138 * Check the undo structures for being valid. Print a warning when something
139 * looks wrong.
140 */
141static int seen_b_u_curhead;
142static int seen_b_u_newhead;
143static int header_count;
144
145 static void
146u_check_tree(u_header_T *uhp,
147 u_header_T *exp_uh_next,
148 u_header_T *exp_uh_alt_prev)
149{
150 u_entry_T *uep;
151
152 if (uhp == NULL)
153 return;
154 ++header_count;
155 if (uhp == curbuf->b_u_curhead && ++seen_b_u_curhead > 1)
156 {
157 EMSG("b_u_curhead found twice (looping?)");
158 return;
159 }
160 if (uhp == curbuf->b_u_newhead && ++seen_b_u_newhead > 1)
161 {
162 EMSG("b_u_newhead found twice (looping?)");
163 return;
164 }
165
166 if (uhp->uh_magic != UH_MAGIC)
167 EMSG("uh_magic wrong (may be using freed memory)");
168 else
169 {
170 /* Check pointers back are correct. */
171 if (uhp->uh_next != exp_uh_next)
172 {
173 EMSG("uh_next wrong");
174 smsg((char_u *)"expected: 0x%x, actual: 0x%x",
175 exp_uh_next, uhp->uh_next);
176 }
177 if (uhp->uh_alt_prev != exp_uh_alt_prev)
178 {
179 EMSG("uh_alt_prev wrong");
180 smsg((char_u *)"expected: 0x%x, actual: 0x%x",
181 exp_uh_alt_prev, uhp->uh_alt_prev);
182 }
183
184 /* Check the undo tree at this header. */
185 for (uep = uhp->uh_entry; uep != NULL; uep = uep->ue_next)
186 {
187 if (uep->ue_magic != UE_MAGIC)
188 {
189 EMSG("ue_magic wrong (may be using freed memory)");
190 break;
191 }
192 }
193
194 /* Check the next alt tree. */
195 u_check_tree(uhp->uh_alt_next, uhp->uh_next, uhp);
196
197 /* Check the next header in this branch. */
198 u_check_tree(uhp->uh_prev, uhp, NULL);
199 }
200}
201
202 void
203u_check(int newhead_may_be_NULL)
204{
205 seen_b_u_newhead = 0;
206 seen_b_u_curhead = 0;
207 header_count = 0;
208
209 u_check_tree(curbuf->b_u_oldhead, NULL, NULL);
210
211 if (seen_b_u_newhead == 0 && curbuf->b_u_oldhead != NULL
212 && !(newhead_may_be_NULL && curbuf->b_u_newhead == NULL))
213 EMSGN("b_u_newhead invalid: 0x%x", curbuf->b_u_newhead);
214 if (curbuf->b_u_curhead != NULL && seen_b_u_curhead == 0)
215 EMSGN("b_u_curhead invalid: 0x%x", curbuf->b_u_curhead);
216 if (header_count != curbuf->b_u_numhead)
217 {
218 EMSG("b_u_numhead invalid");
219 smsg((char_u *)"expected: %ld, actual: %ld",
220 (long)header_count, (long)curbuf->b_u_numhead);
221 }
222}
223#endif
224
Bram Moolenaar071d4272004-06-13 20:20:40 +0000225/*
Bram Moolenaard857f0e2005-06-21 22:37:39 +0000226 * Save the current line for both the "u" and "U" command.
227 * Returns OK or FAIL.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000228 */
229 int
230u_save_cursor()
231{
232 return (u_save((linenr_T)(curwin->w_cursor.lnum - 1),
233 (linenr_T)(curwin->w_cursor.lnum + 1)));
234}
235
236/*
237 * Save the lines between "top" and "bot" for both the "u" and "U" command.
238 * "top" may be 0 and bot may be curbuf->b_ml.ml_line_count + 1.
239 * Returns FAIL when lines could not be saved, OK otherwise.
240 */
241 int
242u_save(top, bot)
243 linenr_T top, bot;
244{
245 if (undo_off)
246 return OK;
247
248 if (top > curbuf->b_ml.ml_line_count ||
249 top >= bot || bot > curbuf->b_ml.ml_line_count + 1)
250 return FALSE; /* rely on caller to do error messages */
251
252 if (top + 2 == bot)
253 u_saveline((linenr_T)(top + 1));
254
255 return (u_savecommon(top, bot, (linenr_T)0));
256}
257
258/*
Bram Moolenaarfff2bee2010-05-15 13:56:02 +0200259 * Save the line "lnum" (used by ":s" and "~" command).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000260 * The line is replaced, so the new bottom line is lnum + 1.
261 */
262 int
263u_savesub(lnum)
264 linenr_T lnum;
265{
266 if (undo_off)
267 return OK;
268
269 return (u_savecommon(lnum - 1, lnum + 1, lnum + 1));
270}
271
272/*
Bram Moolenaarfff2bee2010-05-15 13:56:02 +0200273 * A new line is inserted before line "lnum" (used by :s command).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000274 * The line is inserted, so the new bottom line is lnum + 1.
275 */
276 int
277u_inssub(lnum)
278 linenr_T lnum;
279{
280 if (undo_off)
281 return OK;
282
283 return (u_savecommon(lnum - 1, lnum, lnum + 1));
284}
285
286/*
Bram Moolenaarfff2bee2010-05-15 13:56:02 +0200287 * Save the lines "lnum" - "lnum" + nlines (used by delete command).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000288 * The lines are deleted, so the new bottom line is lnum, unless the buffer
289 * becomes empty.
290 */
291 int
292u_savedel(lnum, nlines)
293 linenr_T lnum;
294 long nlines;
295{
296 if (undo_off)
297 return OK;
298
299 return (u_savecommon(lnum - 1, lnum + nlines,
300 nlines == curbuf->b_ml.ml_line_count ? 2 : lnum));
301}
302
Bram Moolenaar8ada17c2006-01-19 22:16:24 +0000303/*
304 * Return TRUE when undo is allowed. Otherwise give an error message and
305 * return FALSE.
306 */
Bram Moolenaarce6ef252006-07-12 19:49:41 +0000307 int
Bram Moolenaar8ada17c2006-01-19 22:16:24 +0000308undo_allowed()
309{
310 /* Don't allow changes when 'modifiable' is off. */
311 if (!curbuf->b_p_ma)
312 {
313 EMSG(_(e_modifiable));
314 return FALSE;
315 }
316
317#ifdef HAVE_SANDBOX
318 /* In the sandbox it's not allowed to change the text. */
319 if (sandbox != 0)
320 {
321 EMSG(_(e_sandbox));
322 return FALSE;
323 }
324#endif
325
326 /* Don't allow changes in the buffer while editing the cmdline. The
327 * caller of getcmdline() may get confused. */
Bram Moolenaarb71eaae2006-01-20 23:10:18 +0000328 if (textlock != 0)
Bram Moolenaar8ada17c2006-01-19 22:16:24 +0000329 {
330 EMSG(_(e_secure));
331 return FALSE;
332 }
333
334 return TRUE;
335}
336
Bram Moolenaar071d4272004-06-13 20:20:40 +0000337 static int
338u_savecommon(top, bot, newbot)
339 linenr_T top, bot;
340 linenr_T newbot;
341{
Bram Moolenaar1e607892006-03-13 22:15:53 +0000342 linenr_T lnum;
343 long i;
344 u_header_T *uhp;
345 u_header_T *old_curhead;
346 u_entry_T *uep;
347 u_entry_T *prev_uep;
348 long size;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000349
Bram Moolenaar8ada17c2006-01-19 22:16:24 +0000350 /* When making changes is not allowed return FAIL. It's a crude way to
351 * make all change commands fail. */
352 if (!undo_allowed())
Bram Moolenaar071d4272004-06-13 20:20:40 +0000353 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000354
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000355#ifdef U_DEBUG
356 u_check(FALSE);
357#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000358#ifdef FEAT_NETBEANS_INTG
359 /*
360 * Netbeans defines areas that cannot be modified. Bail out here when
361 * trying to change text in a guarded area.
362 */
Bram Moolenaarb26e6322010-05-22 21:34:09 +0200363 if (netbeans_active())
Bram Moolenaar071d4272004-06-13 20:20:40 +0000364 {
Bram Moolenaar009b2592004-10-24 19:18:58 +0000365 if (netbeans_is_guarded(top, bot))
366 {
367 EMSG(_(e_guarded));
368 return FAIL;
369 }
370 if (curbuf->b_p_ro)
371 {
372 EMSG(_(e_nbreadonly));
373 return FAIL;
374 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000375 }
376#endif
377
378#ifdef FEAT_AUTOCMD
379 /*
380 * Saving text for undo means we are going to make a change. Give a
381 * warning for a read-only file before making the change, so that the
382 * FileChangedRO event can replace the buffer with a read-write version
383 * (e.g., obtained from a source control system).
384 */
385 change_warning(0);
386#endif
387
388 size = bot - top - 1;
389
390 /*
391 * if curbuf->b_u_synced == TRUE make a new header
392 */
393 if (curbuf->b_u_synced)
394 {
395#ifdef FEAT_JUMPLIST
396 /* Need to create new entry in b_changelist. */
397 curbuf->b_new_change = TRUE;
398#endif
399
Bram Moolenaar1e607892006-03-13 22:15:53 +0000400 if (p_ul >= 0)
401 {
402 /*
403 * Make a new header entry. Do this first so that we don't mess
404 * up the undo info when out of memory.
405 */
406 uhp = (u_header_T *)U_ALLOC_LINE((unsigned)sizeof(u_header_T));
407 if (uhp == NULL)
408 goto nomem;
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000409#ifdef U_DEBUG
410 uhp->uh_magic = UH_MAGIC;
411#endif
Bram Moolenaar1e607892006-03-13 22:15:53 +0000412 }
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +0000413 else
414 uhp = NULL;
Bram Moolenaar1e607892006-03-13 22:15:53 +0000415
Bram Moolenaar071d4272004-06-13 20:20:40 +0000416 /*
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000417 * If we undid more than we redid, move the entry lists before and
418 * including curbuf->b_u_curhead to an alternate branch.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000419 */
Bram Moolenaar1e607892006-03-13 22:15:53 +0000420 old_curhead = curbuf->b_u_curhead;
421 if (old_curhead != NULL)
422 {
423 curbuf->b_u_newhead = old_curhead->uh_next;
424 curbuf->b_u_curhead = NULL;
425 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000426
427 /*
428 * free headers to keep the size right
429 */
430 while (curbuf->b_u_numhead > p_ul && curbuf->b_u_oldhead != NULL)
Bram Moolenaar1e607892006-03-13 22:15:53 +0000431 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000432 u_header_T *uhfree = curbuf->b_u_oldhead;
Bram Moolenaar1e607892006-03-13 22:15:53 +0000433
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000434 if (uhfree == old_curhead)
435 /* Can't reconnect the branch, delete all of it. */
436 u_freebranch(curbuf, uhfree, &old_curhead);
437 else if (uhfree->uh_alt_next == NULL)
438 /* There is no branch, only free one header. */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000439 u_freeheader(curbuf, uhfree, &old_curhead);
Bram Moolenaar1e607892006-03-13 22:15:53 +0000440 else
441 {
442 /* Free the oldest alternate branch as a whole. */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000443 while (uhfree->uh_alt_next != NULL)
444 uhfree = uhfree->uh_alt_next;
445 u_freebranch(curbuf, uhfree, &old_curhead);
Bram Moolenaar1e607892006-03-13 22:15:53 +0000446 }
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000447#ifdef U_DEBUG
448 u_check(TRUE);
449#endif
Bram Moolenaar1e607892006-03-13 22:15:53 +0000450 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000451
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +0000452 if (uhp == NULL) /* no undo at all */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453 {
Bram Moolenaar1e607892006-03-13 22:15:53 +0000454 if (old_curhead != NULL)
455 u_freebranch(curbuf, old_curhead, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000456 curbuf->b_u_synced = FALSE;
457 return OK;
458 }
459
Bram Moolenaar071d4272004-06-13 20:20:40 +0000460 uhp->uh_prev = NULL;
461 uhp->uh_next = curbuf->b_u_newhead;
Bram Moolenaar1e607892006-03-13 22:15:53 +0000462 uhp->uh_alt_next = old_curhead;
463 if (old_curhead != NULL)
464 {
Bram Moolenaar89ed3df2007-01-09 19:23:12 +0000465 uhp->uh_alt_prev = old_curhead->uh_alt_prev;
466 if (uhp->uh_alt_prev != NULL)
467 uhp->uh_alt_prev->uh_alt_next = uhp;
Bram Moolenaar1e607892006-03-13 22:15:53 +0000468 old_curhead->uh_alt_prev = uhp;
469 if (curbuf->b_u_oldhead == old_curhead)
470 curbuf->b_u_oldhead = uhp;
471 }
Bram Moolenaar89ed3df2007-01-09 19:23:12 +0000472 else
473 uhp->uh_alt_prev = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000474 if (curbuf->b_u_newhead != NULL)
475 curbuf->b_u_newhead->uh_prev = uhp;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000476
Bram Moolenaarca003e12006-03-17 23:19:38 +0000477 uhp->uh_seq = ++curbuf->b_u_seq_last;
478 curbuf->b_u_seq_cur = uhp->uh_seq;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +0000479 uhp->uh_time = time(NULL);
480 curbuf->b_u_seq_time = uhp->uh_time + 1;
481
Bram Moolenaar1e607892006-03-13 22:15:53 +0000482 uhp->uh_walk = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000483 uhp->uh_entry = NULL;
484 uhp->uh_getbot_entry = NULL;
485 uhp->uh_cursor = curwin->w_cursor; /* save cursor pos. for undo */
486#ifdef FEAT_VIRTUALEDIT
487 if (virtual_active() && curwin->w_cursor.coladd > 0)
488 uhp->uh_cursor_vcol = getviscol();
489 else
490 uhp->uh_cursor_vcol = -1;
491#endif
492
493 /* save changed and buffer empty flag for undo */
494 uhp->uh_flags = (curbuf->b_changed ? UH_CHANGED : 0) +
495 ((curbuf->b_ml.ml_flags & ML_EMPTY) ? UH_EMPTYBUF : 0);
496
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000497 /* save named marks and Visual marks for undo */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000498 mch_memmove(uhp->uh_namedm, curbuf->b_namedm, sizeof(pos_T) * NMARKS);
Bram Moolenaara23ccb82006-02-27 00:08:02 +0000499#ifdef FEAT_VISUAL
500 uhp->uh_visual = curbuf->b_visual;
501#endif
502
Bram Moolenaar071d4272004-06-13 20:20:40 +0000503 curbuf->b_u_newhead = uhp;
504 if (curbuf->b_u_oldhead == NULL)
505 curbuf->b_u_oldhead = uhp;
506 ++curbuf->b_u_numhead;
507 }
508 else
509 {
510 if (p_ul < 0) /* no undo at all */
511 return OK;
512
513 /*
514 * When saving a single line, and it has been saved just before, it
515 * doesn't make sense saving it again. Saves a lot of memory when
516 * making lots of changes inside the same line.
517 * This is only possible if the previous change didn't increase or
518 * decrease the number of lines.
519 * Check the ten last changes. More doesn't make sense and takes too
520 * long.
521 */
522 if (size == 1)
523 {
524 uep = u_get_headentry();
525 prev_uep = NULL;
526 for (i = 0; i < 10; ++i)
527 {
528 if (uep == NULL)
529 break;
530
531 /* If lines have been inserted/deleted we give up.
532 * Also when the line was included in a multi-line save. */
533 if ((curbuf->b_u_newhead->uh_getbot_entry != uep
534 ? (uep->ue_top + uep->ue_size + 1
535 != (uep->ue_bot == 0
536 ? curbuf->b_ml.ml_line_count + 1
537 : uep->ue_bot))
538 : uep->ue_lcount != curbuf->b_ml.ml_line_count)
539 || (uep->ue_size > 1
540 && top >= uep->ue_top
541 && top + 2 <= uep->ue_top + uep->ue_size + 1))
542 break;
543
544 /* If it's the same line we can skip saving it again. */
545 if (uep->ue_size == 1 && uep->ue_top == top)
546 {
547 if (i > 0)
548 {
549 /* It's not the last entry: get ue_bot for the last
550 * entry now. Following deleted/inserted lines go to
551 * the re-used entry. */
552 u_getbot();
553 curbuf->b_u_synced = FALSE;
554
555 /* Move the found entry to become the last entry. The
556 * order of undo/redo doesn't matter for the entries
557 * we move it over, since they don't change the line
558 * count and don't include this line. It does matter
559 * for the found entry if the line count is changed by
560 * the executed command. */
561 prev_uep->ue_next = uep->ue_next;
562 uep->ue_next = curbuf->b_u_newhead->uh_entry;
563 curbuf->b_u_newhead->uh_entry = uep;
564 }
565
566 /* The executed command may change the line count. */
567 if (newbot != 0)
568 uep->ue_bot = newbot;
569 else if (bot > curbuf->b_ml.ml_line_count)
570 uep->ue_bot = 0;
571 else
572 {
573 uep->ue_lcount = curbuf->b_ml.ml_line_count;
574 curbuf->b_u_newhead->uh_getbot_entry = uep;
575 }
576 return OK;
577 }
578 prev_uep = uep;
579 uep = uep->ue_next;
580 }
581 }
582
583 /* find line number for ue_bot for previous u_save() */
584 u_getbot();
585 }
586
587#if !defined(UNIX) && !defined(DJGPP) && !defined(WIN32) && !defined(__EMX__)
588 /*
589 * With Amiga and MSDOS 16 bit we can't handle big undo's, because
590 * then u_alloc_line would have to allocate a block larger than 32K
591 */
592 if (size >= 8000)
593 goto nomem;
594#endif
595
596 /*
597 * add lines in front of entry list
598 */
Bram Moolenaar26a60b42005-02-22 08:49:11 +0000599 uep = (u_entry_T *)U_ALLOC_LINE((unsigned)sizeof(u_entry_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 if (uep == NULL)
601 goto nomem;
Bram Moolenaar7db5fc82010-05-24 11:59:29 +0200602 vim_memset(uep, 0, sizeof(u_entry_T));
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000603#ifdef U_DEBUG
604 uep->ue_magic = UE_MAGIC;
605#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000606
607 uep->ue_size = size;
608 uep->ue_top = top;
609 if (newbot != 0)
610 uep->ue_bot = newbot;
611 /*
612 * Use 0 for ue_bot if bot is below last line.
613 * Otherwise we have to compute ue_bot later.
614 */
615 else if (bot > curbuf->b_ml.ml_line_count)
616 uep->ue_bot = 0;
617 else
618 {
619 uep->ue_lcount = curbuf->b_ml.ml_line_count;
620 curbuf->b_u_newhead->uh_getbot_entry = uep;
621 }
622
Bram Moolenaar26a60b42005-02-22 08:49:11 +0000623 if (size > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000624 {
Bram Moolenaar26a60b42005-02-22 08:49:11 +0000625 if ((uep->ue_array = (char_u **)U_ALLOC_LINE(
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626 (unsigned)(sizeof(char_u *) * size))) == NULL)
627 {
628 u_freeentry(uep, 0L);
629 goto nomem;
630 }
631 for (i = 0, lnum = top + 1; i < size; ++i)
632 {
Bram Moolenaarae5bce12005-08-15 21:41:48 +0000633 fast_breakcheck();
634 if (got_int)
635 {
636 u_freeentry(uep, i);
637 return FAIL;
638 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000639 if ((uep->ue_array[i] = u_save_line(lnum++)) == NULL)
640 {
641 u_freeentry(uep, i);
642 goto nomem;
643 }
644 }
645 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000646 else
647 uep->ue_array = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000648 uep->ue_next = curbuf->b_u_newhead->uh_entry;
649 curbuf->b_u_newhead->uh_entry = uep;
650 curbuf->b_u_synced = FALSE;
651 undo_undoes = FALSE;
652
Bram Moolenaarfecb6602007-10-01 20:54:15 +0000653#ifdef U_DEBUG
654 u_check(FALSE);
655#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656 return OK;
657
658nomem:
659 msg_silent = 0; /* must display the prompt */
660 if (ask_yesno((char_u *)_("No undo possible; continue anyway"), TRUE)
661 == 'y')
662 {
663 undo_off = TRUE; /* will be reset when character typed */
664 return OK;
665 }
666 do_outofmem_msg((long_u)0);
667 return FAIL;
668}
669
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200670#ifdef FEAT_PERSISTENT_UNDO
671
672# define UF_START_MAGIC 0xfeac /* magic at start of undofile */
673# define UF_HEADER_MAGIC 0x5fd0 /* magic at start of header */
674# define UF_END_MAGIC 0xe7aa /* magic after last header */
675# define UF_VERSION 1 /* 2-byte undofile version number */
676
677/*
678 * Compute the hash for the current buffer text into hash[UNDO_HASH_SIZE].
679 */
680 void
681u_compute_hash(hash)
682 char_u *hash;
683{
684 context_sha256_T ctx;
685 linenr_T lnum;
686 char_u *p;
687
688 sha256_start(&ctx);
689 for (lnum = 1; lnum < curbuf->b_ml.ml_line_count; ++lnum)
690 {
691 p = ml_get(lnum);
Bram Moolenaar442b4222010-05-24 21:34:22 +0200692 sha256_update(&ctx, p, (UINT32_T)(STRLEN(p) + 1));
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200693 }
694 sha256_finish(&ctx, hash);
695}
696
697/*
698 * Unserialize the pos_T at the current position in fp.
699 */
700 static void
701unserialize_pos(pos, fp)
702 pos_T *pos;
703 FILE *fp;
704{
705 pos->lnum = get4c(fp);
706 pos->col = get4c(fp);
707#ifdef FEAT_VIRTUALEDIT
708 pos->coladd = get4c(fp);
709#else
710 (void)get4c(fp);
711#endif
712}
713
714/*
715 * Unserialize the visualinfo_T at the current position in fp.
716 */
717 static void
718unserialize_visualinfo(info, fp)
719 visualinfo_T *info;
720 FILE *fp;
721{
722 unserialize_pos(&info->vi_start, fp);
723 unserialize_pos(&info->vi_end, fp);
724 info->vi_mode = get4c(fp);
725 info->vi_curswant = get4c(fp);
726}
727
728/*
729 * Return an allocated string of the full path of the target undofile.
730 * When "reading" is TRUE find the file to read, go over all directories in
731 * 'undodir'.
732 * When "reading" is FALSE use the first name where the directory exists.
733 */
734 static char_u *
735u_get_undo_file_name(buf_ffname, reading)
736 char_u *buf_ffname;
737 int reading;
738{
739 char_u *dirp;
740 char_u dir_name[IOSIZE + 1];
741 char_u *munged_name = NULL;
742 char_u *undo_file_name = NULL;
743 int dir_len;
744 char_u *p;
745 struct stat st;
746 char_u *ffname = buf_ffname;
747#ifdef HAVE_READLINK
748 char_u fname_buf[MAXPATHL];
749#endif
750
751 if (ffname == NULL)
752 return NULL;
753
754#ifdef HAVE_READLINK
755 /* Expand symlink in the file name, so that we put the undo file with the
756 * actual file instead of with the symlink. */
757 if (resolve_symlink(ffname, fname_buf) == OK)
758 ffname = fname_buf;
759#endif
760
761 /* Loop over 'undodir'. When reading find the first file that exists.
762 * When not reading use the first directory that exists or ".". */
763 dirp = p_udir;
764 while (*dirp != NUL)
765 {
766 dir_len = copy_option_part(&dirp, dir_name, IOSIZE, ",");
767 if (dir_len == 1 && dir_name[0] == '.')
768 {
769 /* Use same directory as the ffname,
770 * "dir/name" -> "dir/.name.un~" */
Bram Moolenaar442b4222010-05-24 21:34:22 +0200771 undo_file_name = vim_strnsave(ffname, (int)(STRLEN(ffname) + 5));
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200772 if (undo_file_name == NULL)
773 break;
774 p = gettail(undo_file_name);
775 mch_memmove(p + 1, p, STRLEN(p) + 1);
776 *p = '.';
777 STRCAT(p, ".un~");
778 }
779 else
780 {
781 dir_name[dir_len] = NUL;
782 if (mch_isdir(dir_name))
783 {
784 if (munged_name == NULL)
785 {
786 munged_name = vim_strsave(ffname);
787 if (munged_name == NULL)
788 return NULL;
789 for (p = munged_name; *p != NUL; mb_ptr_adv(p))
790 if (vim_ispathsep(*p))
791 *p = '%';
792 }
793 undo_file_name = concat_fnames(dir_name, munged_name, TRUE);
794 }
795 }
796
797 /* When reading check if the file exists. */
798 if (undo_file_name != NULL && (!reading
799 || mch_stat((char *)undo_file_name, &st) >= 0))
800 break;
801 vim_free(undo_file_name);
802 undo_file_name = NULL;
803 }
804
805 vim_free(munged_name);
806 return undo_file_name;
807}
808
809/*
810 * Load the undo tree from an undo file.
811 * If "name" is not NULL use it as the undo file name. This also means being
812 * a bit more verbose.
813 * Otherwise use curbuf->b_ffname to generate the undo file name.
814 * "hash[UNDO_HASH_SIZE]" must be the hash value of the buffer text.
815 */
816 void
817u_read_undo(name, hash)
818 char_u *name;
819 char_u *hash;
820{
821 char_u *file_name;
822 FILE *fp;
823 long magic, version, str_len;
824 char_u *line_ptr = NULL;
825 linenr_T line_lnum;
826 colnr_T line_colnr;
827 linenr_T line_count;
828 int uep_len;
829 int line_len;
Bram Moolenaar442b4222010-05-24 21:34:22 +0200830 int num_head = 0;
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200831 long old_header_seq, new_header_seq, cur_header_seq;
832 long seq_last, seq_cur;
833 short old_idx = -1, new_idx = -1, cur_idx = -1;
834 long num_read_uhps = 0;
835 time_t seq_time;
836 int i, j;
837 int c;
838 short found_first_uep = 0;
839 char_u **array;
840 char_u *line;
841 u_entry_T *uep, *last_uep, *nuep;
842 u_header_T *uhp;
843 u_header_T **uhp_table = NULL;
844 char_u read_hash[UNDO_HASH_SIZE];
845
846 if (name == NULL)
847 {
848 file_name = u_get_undo_file_name(curbuf->b_ffname, TRUE);
849 if (file_name == NULL)
850 return;
851 }
852 else
853 file_name = name;
854
855 if (p_verbose > 0)
856 smsg((char_u *)_("Reading undo file: %s"), file_name);
857 fp = mch_fopen((char *)file_name, "r");
858 if (fp == NULL)
859 {
860 if (name != NULL || p_verbose > 0)
861 EMSG2(_("E822: Cannot open undo file for reading: %s"), file_name);
862 goto error;
863 }
864
865 /* Begin overall file information */
866 magic = get2c(fp);
867 if (magic != UF_START_MAGIC)
868 {
869 EMSG2(_("E823: Corrupted undo file: %s"), file_name);
870 goto error;
871 }
872 version = get2c(fp);
873 if (version != UF_VERSION)
874 {
875 EMSG2(_("E824: Incompatible undo file: %s"), file_name);
876 goto error;
877 }
878
879 fread(read_hash, UNDO_HASH_SIZE, 1, fp);
880 line_count = (linenr_T)get4c(fp);
881 if (memcmp(hash, read_hash, UNDO_HASH_SIZE) != 0
882 || line_count != curbuf->b_ml.ml_line_count)
883 {
884 if (p_verbose > 0 || name != NULL)
885 {
886 verbose_enter();
Bram Moolenaar7db5fc82010-05-24 11:59:29 +0200887 give_warning((char_u *)_("File contents changed, cannot use undo info"), TRUE);
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200888 verbose_leave();
889 }
890 goto error;
891 }
892
893 /* Begin undo data for U */
894 str_len = get4c(fp);
895 if (str_len < 0)
896 goto error;
897 else if (str_len > 0)
898 {
899 if ((line_ptr = U_ALLOC_LINE(str_len)) == NULL)
900 goto error;
901 for (i = 0; i < str_len; i++)
902 line_ptr[i] = (char_u)getc(fp);
903 line_ptr[i] = NUL;
904 }
905 line_lnum = (linenr_T)get4c(fp);
906 line_colnr = (colnr_T)get4c(fp);
907
908 /* Begin general undo data */
909 old_header_seq = get4c(fp);
910 new_header_seq = get4c(fp);
911 cur_header_seq = get4c(fp);
912 num_head = get4c(fp);
913 seq_last = get4c(fp);
914 seq_cur = get4c(fp);
915 seq_time = get4c(fp);
916
917 /* uhp_table will store the freshly created undo headers we allocate
918 * until we insert them into curbuf. The table remains sorted by the
919 * sequence numbers of the headers. */
920 uhp_table = (u_header_T **)U_ALLOC_LINE(num_head * sizeof(u_header_T *));
921 if (uhp_table == NULL)
922 goto error;
923 vim_memset(uhp_table, 0, num_head * sizeof(u_header_T *));
924
925 c = get2c(fp);
926 while (c == UF_HEADER_MAGIC)
927 {
928 found_first_uep = 0;
929 uhp = (u_header_T *)U_ALLOC_LINE((unsigned)sizeof(u_header_T));
930 if (uhp == NULL)
931 goto error;
932 vim_memset(uhp, 0, sizeof(u_header_T));
933 /* We're not actually trying to store pointers here. We're just storing
934 * IDs so we can swizzle them into pointers later - hence the type
935 * cast. */
Bram Moolenaar442b4222010-05-24 21:34:22 +0200936 uhp->uh_next = (u_header_T *)get4c(fp);
937 uhp->uh_prev = (u_header_T *)get4c(fp);
938 uhp->uh_alt_next = (u_header_T *)get4c(fp);
939 uhp->uh_alt_prev = (u_header_T *)get4c(fp);
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200940 uhp->uh_seq = get4c(fp);
941 if (uhp->uh_seq <= 0)
942 {
943 EMSG2(_("E825: Undo file corruption: invalid uh_seq.: %s"),
944 file_name);
945 U_FREE_LINE(uhp);
946 goto error;
947 }
948 uhp->uh_walk = 0;
949 unserialize_pos(&uhp->uh_cursor, fp);
950#ifdef FEAT_VIRTUALEDIT
951 uhp->uh_cursor_vcol = get4c(fp);
952#else
953 (void)get4c(fp);
954#endif
955 uhp->uh_flags = get2c(fp);
956 for (i = 0; i < NMARKS; ++i)
957 unserialize_pos(&uhp->uh_namedm[i], fp);
958#ifdef FEAT_VISUAL
959 unserialize_visualinfo(&uhp->uh_visual, fp);
960#else
961 {
962 visualinfo_T info;
963 unserialize_visualinfo(&info, fp);
964 }
965#endif
966 uhp->uh_time = get4c(fp);
967
968 /* Unserialize uep list. The first 4 bytes is the length of the
969 * entire uep in bytes minus the length of the strings within.
970 * -1 is a sentinel value meaning no more ueps.*/
971 last_uep = NULL;
972 while ((uep_len = get4c(fp)) != -1)
973 {
974 uep = (u_entry_T *)U_ALLOC_LINE((unsigned)sizeof(u_entry_T));
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200975 if (uep == NULL)
976 goto error;
Bram Moolenaar7db5fc82010-05-24 11:59:29 +0200977 vim_memset(uep, 0, sizeof(u_entry_T));
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200978 uep->ue_top = get4c(fp);
979 uep->ue_bot = get4c(fp);
980 uep->ue_lcount = get4c(fp);
981 uep->ue_size = get4c(fp);
982 uep->ue_next = NULL;
983 array = (char_u **)U_ALLOC_LINE(
984 (unsigned)(sizeof(char_u *) * uep->ue_size));
985 for (i = 0; i < uep->ue_size; i++)
986 {
987 line_len = get4c(fp);
988 /* U_ALLOC_LINE provides an extra byte for the NUL terminator.*/
989 line = (char_u *)U_ALLOC_LINE(
990 (unsigned) (sizeof(char_u) * line_len));
991 if (line == NULL)
992 goto error;
993 for (j = 0; j < line_len; j++)
994 {
995 line[j] = getc(fp);
996 }
997 line[j] = '\0';
998 array[i] = line;
999 }
1000 uep->ue_array = array;
1001 if (found_first_uep == 0)
1002 {
1003 uhp->uh_entry = uep;
1004 found_first_uep = 1;
1005 }
1006 else
1007 {
1008 last_uep->ue_next = uep;
1009 }
1010 last_uep = uep;
1011 }
1012
1013 /* Insertion sort the uhp into the table by its uh_seq. This is
1014 * required because, while the number of uhps is limited to
1015 * num_heads, and the uh_seq order is monotonic with respect to
1016 * creation time, the starting uh_seq can be > 0 if any undolevel
1017 * culling was done at undofile write time, and there can be uh_seq
1018 * gaps in the uhps.
1019 */
1020 for (i = num_read_uhps - 1; i >= -1; i--)
1021 {
1022 /* if i == -1, we've hit the leftmost side of the table, so insert
1023 * at uhp_table[0]. */
1024 if (i == -1 || uhp->uh_seq > uhp_table[i]->uh_seq)
1025 {
1026 /* If we've had to move from the rightmost side of the table,
1027 * we have to shift everything to the right by one spot. */
Bram Moolenaar9d728072010-05-24 22:06:04 +02001028 if (num_read_uhps - i - 1 > 0)
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001029 {
1030 memmove(uhp_table + i + 2, uhp_table + i + 1,
Bram Moolenaar9d728072010-05-24 22:06:04 +02001031 (num_read_uhps - i - 1) * sizeof(u_header_T *));
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001032 }
1033 uhp_table[i + 1] = uhp;
1034 break;
1035 }
1036 else if (uhp->uh_seq == uhp_table[i]->uh_seq)
1037 {
1038 EMSG2(_("E826 Undo file corruption: duplicate uh_seq: %s"),
1039 file_name);
1040 goto error;
1041 }
1042 }
1043 num_read_uhps++;
1044 c = get2c(fp);
1045 }
1046
1047 if (c != UF_END_MAGIC)
1048 {
1049 EMSG2(_("E827: Undo file corruption; no end marker: %s"), file_name);
1050 goto error;
1051 }
1052
1053 /* We've organized all of the uhps into a table sorted by uh_seq. Now we
1054 * iterate through the table and swizzle each sequence number we've
1055 * stored in uh_foo into a pointer corresponding to the header with that
1056 * sequence number. Then free curbuf's old undo structure, give curbuf
1057 * the updated {old,new,cur}head pointers, and then free the table. */
1058 for (i = 0; i < num_head; i++)
1059 {
1060 uhp = uhp_table[i];
1061 if (uhp == NULL)
1062 continue;
1063 for (j = 0; j < num_head; j++)
1064 {
1065 if (uhp_table[j] == NULL)
1066 continue;
1067 if (uhp_table[j]->uh_seq == (long)uhp->uh_next)
1068 uhp->uh_next = uhp_table[j];
1069 if (uhp_table[j]->uh_seq == (long)uhp->uh_prev)
1070 uhp->uh_prev = uhp_table[j];
1071 if (uhp_table[j]->uh_seq == (long)uhp->uh_alt_next)
1072 uhp->uh_alt_next = uhp_table[j];
1073 if (uhp_table[j]->uh_seq == (long)uhp->uh_alt_prev)
1074 uhp->uh_alt_prev = uhp_table[j];
1075 }
1076 if (old_header_seq > 0 && old_idx < 0 && uhp->uh_seq == old_header_seq)
1077 old_idx = i;
1078 if (new_header_seq > 0 && new_idx < 0 && uhp->uh_seq == new_header_seq)
1079 new_idx = i;
1080 if (cur_header_seq > 0 && cur_idx < 0 && uhp->uh_seq == cur_header_seq)
1081 cur_idx = i;
1082 }
1083 u_blockfree(curbuf);
1084 curbuf->b_u_oldhead = old_idx < 0 ? 0 : uhp_table[old_idx];
1085 curbuf->b_u_newhead = new_idx < 0 ? 0 : uhp_table[new_idx];
1086 curbuf->b_u_curhead = cur_idx < 0 ? 0 : uhp_table[cur_idx];
1087 curbuf->b_u_line_ptr = line_ptr;
1088 curbuf->b_u_line_lnum = line_lnum;
1089 curbuf->b_u_line_colnr = line_colnr;
1090 curbuf->b_u_numhead = num_head;
1091 curbuf->b_u_seq_last = seq_last;
1092 curbuf->b_u_seq_cur = seq_cur;
1093 curbuf->b_u_seq_time = seq_time;
1094 U_FREE_LINE(uhp_table);
1095#ifdef U_DEBUG
1096 u_check(TRUE);
1097#endif
1098 if (name != NULL)
1099 smsg((char_u *)_("Finished reading undo file %s"), file_name);
1100 goto theend;
1101
1102error:
1103 if (line_ptr != NULL)
1104 U_FREE_LINE(line_ptr);
1105 if (uhp_table != NULL)
1106 {
1107 for (i = 0; i < num_head; i++)
1108 {
1109 if (uhp_table[i] != NULL)
1110 {
1111 uep = uhp_table[i]->uh_entry;
1112 while (uep != NULL)
1113 {
1114 nuep = uep->ue_next;
1115 u_freeentry(uep, uep->ue_size);
1116 uep = nuep;
1117 }
1118 U_FREE_LINE(uhp_table[i]);
1119 }
1120 }
1121 U_FREE_LINE(uhp_table);
1122 }
1123
1124theend:
1125 if (fp != NULL)
1126 fclose(fp);
1127 if (file_name != name)
1128 vim_free(file_name);
1129 return;
1130}
1131
1132/*
1133 * Serialize "uep" to "fp".
1134 */
1135 static int
1136serialize_uep(uep, fp)
1137 u_entry_T *uep;
1138 FILE *fp;
1139{
1140 int i;
1141 int uep_len;
1142 int *entry_lens;
1143
1144 if (uep->ue_size > 0)
1145 entry_lens = (int *)alloc(uep->ue_size * sizeof(int));
Bram Moolenaar442b4222010-05-24 21:34:22 +02001146 else
1147 entry_lens = NULL;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001148
1149 /* Define uep_len to be the size of the entire uep minus the size of its
1150 * component strings, in bytes. The sizes of the component strings
1151 * are written before each individual string.
1152 * We have 4 entries each of 4 bytes, plus ue_size * 4 bytes
1153 * of string size information. */
1154
1155 uep_len = uep->ue_size * 4;
1156 /* Collect sizing information for later serialization. */
1157 for (i = 0; i < uep->ue_size; i++)
1158 {
1159 entry_lens[i] = (int)STRLEN(uep->ue_array[i]);
1160 uep_len += entry_lens[i];
1161 }
1162 put_bytes(fp, (long_u)uep_len, 4);
1163 put_bytes(fp, (long_u)uep->ue_top, 4);
1164 put_bytes(fp, (long_u)uep->ue_bot, 4);
1165 put_bytes(fp, (long_u)uep->ue_lcount, 4);
1166 put_bytes(fp, (long_u)uep->ue_size, 4);
1167 for (i = 0; i < uep->ue_size; i++)
1168 {
1169 if (put_bytes(fp, (long_u)entry_lens[i], 4) == FAIL)
1170 return FAIL;
1171 fprintf(fp, "%s", uep->ue_array[i]);
1172 }
1173 if (uep->ue_size > 0)
1174 vim_free(entry_lens);
1175 return OK;
1176}
1177
1178/*
1179 * Serialize "pos" to "fp".
1180 */
1181 static void
1182serialize_pos(pos, fp)
1183 pos_T pos;
1184 FILE *fp;
1185{
1186 put_bytes(fp, (long_u)pos.lnum, 4);
1187 put_bytes(fp, (long_u)pos.col, 4);
1188#ifdef FEAT_VIRTUALEDIT
1189 put_bytes(fp, (long_u)pos.coladd, 4);
1190#else
1191 put_bytes(fp, (long_u)0, 4);
1192#endif
1193}
1194
1195/*
1196 * Serialize "info" to "fp".
1197 */
1198 static void
1199serialize_visualinfo(info, fp)
1200 visualinfo_T info;
1201 FILE *fp;
1202{
1203 serialize_pos(info.vi_start, fp);
1204 serialize_pos(info.vi_end, fp);
1205 put_bytes(fp, (long_u)info.vi_mode, 4);
1206 put_bytes(fp, (long_u)info.vi_curswant, 4);
1207}
1208
1209static char_u e_not_open[] = N_("E828: Cannot open undo file for writing: %s");
1210
1211/*
1212 * Write the undo tree in an undo file.
1213 * When "name" is not NULL, use it as the name of the undo file.
1214 * Otherwise use buf->b_ffname to generate the undo file name.
1215 * "buf" must never be null, buf->b_ffname is used to obtain the original file
1216 * permissions.
1217 * "forceit" is TRUE for ":wundo!", FALSE otherwise.
1218 * "hash[UNDO_HASH_SIZE]" must be the hash value of the buffer text.
1219 */
1220 void
1221u_write_undo(name, forceit, buf, hash)
1222 char_u *name;
1223 int forceit;
1224 buf_T *buf;
1225 char_u *hash;
1226{
1227 u_header_T *uhp;
1228 u_entry_T *uep;
1229 char_u *file_name;
1230 int str_len, i, uep_len, mark;
1231 int fd;
1232 FILE *fp = NULL;
1233 int perm;
1234 int write_ok = FALSE;
1235#ifdef UNIX
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001236 int st_old_valid = FALSE;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001237 struct stat st_old;
1238 struct stat st_new;
1239#endif
1240
1241 if (name == NULL)
1242 {
1243 file_name = u_get_undo_file_name(buf->b_ffname, FALSE);
1244 if (file_name == NULL)
1245 return;
1246 }
1247 else
1248 file_name = name;
1249
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001250 if (buf->b_ffname == NULL)
1251 perm = 0600;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001252 else
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001253 {
1254#ifdef UNIX
1255 if (mch_stat((char *)buf->b_ffname, &st_old) >= 0)
1256 {
1257 perm = st_old.st_mode;
1258 st_old_valid = TRUE;
1259 }
1260 else
1261 perm = 0600;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001262#else
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001263 perm = mch_getperm(buf->b_ffname);
1264 if (perm < 0)
1265 perm = 0600;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001266#endif
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001267 }
1268
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001269 /* set file protection same as original file, but strip s-bit */
1270 perm = perm & 0777;
1271
1272 /* If the undo file exists, verify that it actually is an undo file, and
1273 * delete it. */
1274 if (mch_getperm(file_name) >= 0)
1275 {
1276 if (name == NULL || !forceit)
1277 {
1278 /* Check we can read it and it's an undo file. */
1279 fd = mch_open((char *)file_name, O_RDONLY|O_EXTRA, 0);
1280 if (fd < 0)
1281 {
1282 if (name != NULL || p_verbose > 0)
1283 smsg((char_u *)_("Will not overwrite with undo file, cannot read: %s"),
1284 file_name);
1285 goto theend;
1286 }
1287 else
1288 {
Bram Moolenaar83ad0142010-05-25 22:09:21 +02001289 char_u buf[2];
1290 int len;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001291
Bram Moolenaar83ad0142010-05-25 22:09:21 +02001292 len = vim_read(fd, buf, 2);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001293 close(fd);
Bram Moolenaar83ad0142010-05-25 22:09:21 +02001294 if (len < 2 || (buf[0] << 8) + buf[1] != UF_START_MAGIC)
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001295 {
1296 if (name != NULL || p_verbose > 0)
1297 smsg((char_u *)_("Will not overwrite, this is not an undo file: %s"),
1298 file_name);
1299 goto theend;
1300 }
1301 }
1302 }
1303 mch_remove(file_name);
1304 }
1305
1306 fd = mch_open((char *)file_name,
1307 O_CREAT|O_EXTRA|O_WRONLY|O_EXCL|O_NOFOLLOW, perm);
1308 (void)mch_setperm(file_name, perm);
1309 if (fd < 0)
1310 {
1311 EMSG2(_(e_not_open), file_name);
1312 goto theend;
1313 }
1314 if (p_verbose > 0)
1315 smsg((char_u *)_("Writing undo file: %s"), file_name);
1316
1317#ifdef UNIX
1318 /*
1319 * Try to set the group of the undo file same as the original file. If
1320 * this fails, set the protection bits for the group same as the
1321 * protection bits for others.
1322 */
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001323 if (st_old_valid && (mch_stat((char *)file_name, &st_new) >= 0
1324 && st_new.st_gid != st_old.st_gid
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001325# ifdef HAVE_FCHOWN /* sequent-ptx lacks fchown() */
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001326 && fchown(fd, (uid_t)-1, st_old.st_gid) != 0)
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001327# endif
1328 )
1329 mch_setperm(file_name, (perm & 0707) | ((perm & 07) << 3));
1330# ifdef HAVE_SELINUX
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001331 if (buf->b_ffname != NULL)
1332 mch_copy_sec(buf->b_ffname, file_name);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001333# endif
1334#endif
1335
1336 fp = fdopen(fd, "w");
1337 if (fp == NULL)
1338 {
1339 EMSG2(_(e_not_open), file_name);
1340 close(fd);
1341 mch_remove(file_name);
1342 goto theend;
1343 }
1344
1345 /* Start writing, first overall file information */
1346 put_bytes(fp, (long_u)UF_START_MAGIC, 2);
1347 put_bytes(fp, (long_u)UF_VERSION, 2);
1348
1349 /* Write a hash of the buffer text, so that we can verify it is still the
1350 * same when reading the buffer text. */
1351 if (fwrite(hash, (size_t)UNDO_HASH_SIZE, (size_t)1, fp) != 1)
1352 goto write_error;
1353 put_bytes(fp, (long_u)buf->b_ml.ml_line_count, 4);
1354
1355 /* Begin undo data for U */
Bram Moolenaar442b4222010-05-24 21:34:22 +02001356 str_len = buf->b_u_line_ptr != NULL ? (int)STRLEN(buf->b_u_line_ptr) : 0;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001357 put_bytes(fp, (long_u)str_len, 4);
1358 if (str_len > 0 && fwrite(buf->b_u_line_ptr, (size_t)str_len,
1359 (size_t)1, fp) != 1)
1360 goto write_error;
1361
1362 put_bytes(fp, (long_u)buf->b_u_line_lnum, 4);
1363 put_bytes(fp, (long_u)buf->b_u_line_colnr, 4);
1364
1365 /* Begin general undo data */
1366 uhp = buf->b_u_oldhead;
1367 put_bytes(fp, (long_u)(uhp != NULL ? uhp->uh_seq : 0), 4);
1368
1369 uhp = buf->b_u_newhead;
1370 put_bytes(fp, (long_u)(uhp != NULL ? uhp->uh_seq : 0), 4);
1371
1372 uhp = buf->b_u_curhead;
1373 put_bytes(fp, (long_u)(uhp != NULL ? uhp->uh_seq : 0), 4);
1374
1375 put_bytes(fp, (long_u)buf->b_u_numhead, 4);
1376 put_bytes(fp, (long_u)buf->b_u_seq_last, 4);
1377 put_bytes(fp, (long_u)buf->b_u_seq_cur, 4);
1378 put_bytes(fp, (long_u)buf->b_u_seq_time, 4);
1379
1380 /* Iteratively serialize UHPs and their UEPs from the top down. */
1381 mark = ++lastmark;
1382 uhp = buf->b_u_oldhead;
1383 while (uhp != NULL)
1384 {
1385 /* Serialize current UHP if we haven't seen it */
1386 if (uhp->uh_walk != mark)
1387 {
1388 if (put_bytes(fp, (long_u)UF_HEADER_MAGIC, 2) == FAIL)
1389 goto write_error;
1390
1391 put_bytes(fp, (long_u)((uhp->uh_next != NULL)
1392 ? uhp->uh_next->uh_seq : 0), 4);
1393 put_bytes(fp, (long_u)((uhp->uh_prev != NULL)
1394 ? uhp->uh_prev->uh_seq : 0), 4);
1395 put_bytes(fp, (long_u)((uhp->uh_alt_next != NULL)
1396 ? uhp->uh_alt_next->uh_seq : 0), 4);
1397 put_bytes(fp, (long_u)((uhp->uh_alt_prev != NULL)
1398 ? uhp->uh_alt_prev->uh_seq : 0), 4);
1399 put_bytes(fp, uhp->uh_seq, 4);
1400 serialize_pos(uhp->uh_cursor, fp);
1401#ifdef FEAT_VIRTUALEDIT
1402 put_bytes(fp, (long_u)uhp->uh_cursor_vcol, 4);
1403#else
1404 put_bytes(fp, (long_u)0, 4);
1405#endif
1406 put_bytes(fp, (long_u)uhp->uh_flags, 2);
1407 /* Assume NMARKS will stay the same. */
1408 for (i = 0; i < NMARKS; ++i)
1409 {
1410 serialize_pos(uhp->uh_namedm[i], fp);
1411 }
1412#ifdef FEAT_VISUAL
1413 serialize_visualinfo(uhp->uh_visual, fp);
1414#endif
1415 put_bytes(fp, (long_u)uhp->uh_time, 4);
1416
1417 uep = uhp->uh_entry;
1418 while (uep != NULL)
1419 {
1420 if (serialize_uep(uep, fp) == FAIL)
1421 goto write_error;
1422 uep = uep->ue_next;
1423 }
1424 /* Sentinel value: no more ueps */
1425 uep_len = -1;
1426 put_bytes(fp, (long_u)uep_len, 4);
1427 uhp->uh_walk = mark;
1428 }
1429
1430 /* Now walk through the tree - algorithm from undo_time */
1431 if (uhp->uh_prev != NULL && uhp->uh_prev->uh_walk != mark)
1432 uhp = uhp->uh_prev;
1433 else if (uhp->uh_alt_next != NULL && uhp->uh_alt_next->uh_walk != mark)
1434 uhp = uhp->uh_alt_next;
1435 else if (uhp->uh_next != NULL && uhp->uh_alt_prev == NULL
1436 && uhp->uh_next->uh_walk != mark)
1437 uhp = uhp->uh_next;
1438 else if (uhp->uh_alt_prev != NULL)
1439 uhp = uhp->uh_alt_prev;
1440 else
1441 uhp = uhp->uh_next;
1442 }
1443
1444 if (put_bytes(fp, (long_u)UF_END_MAGIC, 2) == OK)
1445 write_ok = TRUE;
1446
1447write_error:
1448 fclose(fp);
1449 if (!write_ok)
1450 EMSG2(_("E829: write error in undo file: %s"), file_name);
1451
1452#if defined(MACOS_CLASSIC) || defined(WIN3264)
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001453 if (buf->b_ffname != NULL)
1454 (void)mch_copy_file_attribute(buf->b_ffname, file_name);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001455#endif
1456#ifdef HAVE_ACL
Bram Moolenaar6a244fe2010-05-24 22:02:24 +02001457 if (buf->b_ffname != NULL)
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001458 {
1459 vim_acl_T acl;
1460
1461 /* For systems that support ACL: get the ACL from the original file. */
1462 acl = mch_get_acl(buf->b_ffname);
1463 mch_set_acl(file_name, acl);
1464 }
1465#endif
1466
1467theend:
1468 if (file_name != name)
1469 vim_free(file_name);
1470}
1471
1472#endif /* FEAT_PERSISTENT_UNDO */
1473
1474
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475/*
1476 * If 'cpoptions' contains 'u': Undo the previous undo or redo (vi compatible).
1477 * If 'cpoptions' does not contain 'u': Always undo.
1478 */
1479 void
1480u_undo(count)
1481 int count;
1482{
1483 /*
1484 * If we get an undo command while executing a macro, we behave like the
1485 * original vi. If this happens twice in one macro the result will not
1486 * be compatible.
1487 */
1488 if (curbuf->b_u_synced == FALSE)
1489 {
Bram Moolenaar779b74b2006-04-10 14:55:34 +00001490 u_sync(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001491 count = 1;
1492 }
1493
1494 if (vim_strchr(p_cpo, CPO_UNDO) == NULL)
1495 undo_undoes = TRUE;
1496 else
1497 undo_undoes = !undo_undoes;
1498 u_doit(count);
1499}
1500
1501/*
1502 * If 'cpoptions' contains 'u': Repeat the previous undo or redo.
1503 * If 'cpoptions' does not contain 'u': Always redo.
1504 */
1505 void
1506u_redo(count)
1507 int count;
1508{
1509 if (vim_strchr(p_cpo, CPO_UNDO) == NULL)
1510 undo_undoes = FALSE;
1511 u_doit(count);
1512}
1513
1514/*
1515 * Undo or redo, depending on 'undo_undoes', 'count' times.
1516 */
1517 static void
Bram Moolenaarca003e12006-03-17 23:19:38 +00001518u_doit(startcount)
1519 int startcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001520{
Bram Moolenaarca003e12006-03-17 23:19:38 +00001521 int count = startcount;
1522
Bram Moolenaar8ada17c2006-01-19 22:16:24 +00001523 if (!undo_allowed())
Bram Moolenaar071d4272004-06-13 20:20:40 +00001524 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525
1526 u_newcount = 0;
1527 u_oldcount = 0;
Bram Moolenaar19a09a12005-03-04 23:39:37 +00001528 if (curbuf->b_ml.ml_flags & ML_EMPTY)
1529 u_oldcount = -1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001530 while (count--)
1531 {
1532 if (undo_undoes)
1533 {
1534 if (curbuf->b_u_curhead == NULL) /* first undo */
1535 curbuf->b_u_curhead = curbuf->b_u_newhead;
1536 else if (p_ul > 0) /* multi level undo */
1537 /* get next undo */
1538 curbuf->b_u_curhead = curbuf->b_u_curhead->uh_next;
1539 /* nothing to undo */
1540 if (curbuf->b_u_numhead == 0 || curbuf->b_u_curhead == NULL)
1541 {
1542 /* stick curbuf->b_u_curhead at end */
1543 curbuf->b_u_curhead = curbuf->b_u_oldhead;
1544 beep_flush();
Bram Moolenaarca003e12006-03-17 23:19:38 +00001545 if (count == startcount - 1)
1546 {
1547 MSG(_("Already at oldest change"));
1548 return;
1549 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001550 break;
1551 }
1552
Bram Moolenaarca003e12006-03-17 23:19:38 +00001553 u_undoredo(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554 }
1555 else
1556 {
1557 if (curbuf->b_u_curhead == NULL || p_ul <= 0)
1558 {
1559 beep_flush(); /* nothing to redo */
Bram Moolenaarca003e12006-03-17 23:19:38 +00001560 if (count == startcount - 1)
1561 {
1562 MSG(_("Already at newest change"));
1563 return;
1564 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001565 break;
1566 }
1567
Bram Moolenaarca003e12006-03-17 23:19:38 +00001568 u_undoredo(FALSE);
Bram Moolenaar1e607892006-03-13 22:15:53 +00001569
1570 /* Advance for next redo. Set "newhead" when at the end of the
1571 * redoable changes. */
1572 if (curbuf->b_u_curhead->uh_prev == NULL)
1573 curbuf->b_u_newhead = curbuf->b_u_curhead;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574 curbuf->b_u_curhead = curbuf->b_u_curhead->uh_prev;
1575 }
1576 }
Bram Moolenaardb552d602006-03-23 22:59:57 +00001577 u_undo_end(undo_undoes, FALSE);
Bram Moolenaar1e607892006-03-13 22:15:53 +00001578}
1579
Bram Moolenaar1e607892006-03-13 22:15:53 +00001580/*
1581 * Undo or redo over the timeline.
1582 * When "step" is negative go back in time, otherwise goes forward in time.
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001583 * When "sec" is FALSE make "step" steps, when "sec" is TRUE use "step" as
1584 * seconds.
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001585 * When "absolute" is TRUE use "step" as the sequence number to jump to.
1586 * "sec" must be FALSE then.
Bram Moolenaar1e607892006-03-13 22:15:53 +00001587 */
1588 void
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001589undo_time(step, sec, absolute)
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001590 long step;
1591 int sec;
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001592 int absolute;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001593{
1594 long target;
1595 long closest;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001596 long closest_start;
1597 long closest_seq = 0;
1598 long val;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001599 u_header_T *uhp;
1600 u_header_T *last;
1601 int mark;
1602 int nomark;
1603 int round;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001604 int dosec = sec;
1605 int above = FALSE;
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001606 int did_undo = TRUE;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001607
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001608 /* First make sure the current undoable change is synced. */
1609 if (curbuf->b_u_synced == FALSE)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00001610 u_sync(TRUE);
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001611
Bram Moolenaar1e607892006-03-13 22:15:53 +00001612 u_newcount = 0;
1613 u_oldcount = 0;
Bram Moolenaar19a09a12005-03-04 23:39:37 +00001614 if (curbuf->b_ml.ml_flags & ML_EMPTY)
Bram Moolenaar1e607892006-03-13 22:15:53 +00001615 u_oldcount = -1;
1616
Bram Moolenaarca003e12006-03-17 23:19:38 +00001617 /* "target" is the node below which we want to be.
1618 * Init "closest" to a value we can't reach. */
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001619 if (absolute)
1620 {
1621 target = step;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001622 closest = -1;
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001623 }
Bram Moolenaarca003e12006-03-17 23:19:38 +00001624 else
Bram Moolenaar1e607892006-03-13 22:15:53 +00001625 {
Bram Moolenaarca003e12006-03-17 23:19:38 +00001626 /* When doing computations with time_t subtract starttime, because
1627 * time_t converted to a long may result in a wrong number. */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001628 if (sec)
Bram Moolenaarca003e12006-03-17 23:19:38 +00001629 target = (long)(curbuf->b_u_seq_time - starttime) + step;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001630 else
1631 target = curbuf->b_u_seq_cur + step;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001632 if (step < 0)
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001633 {
Bram Moolenaarca003e12006-03-17 23:19:38 +00001634 if (target < 0)
1635 target = 0;
1636 closest = -1;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001637 }
1638 else
1639 {
Bram Moolenaarca003e12006-03-17 23:19:38 +00001640 if (sec)
Bram Moolenaardb552d602006-03-23 22:59:57 +00001641 closest = (long)(time(NULL) - starttime + 1);
Bram Moolenaarca003e12006-03-17 23:19:38 +00001642 else
1643 closest = curbuf->b_u_seq_last + 2;
1644 if (target >= closest)
1645 target = closest - 1;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001646 }
Bram Moolenaar1e607892006-03-13 22:15:53 +00001647 }
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001648 closest_start = closest;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001649 closest_seq = curbuf->b_u_seq_cur;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001650
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001651 /*
1652 * May do this twice:
Bram Moolenaar1e607892006-03-13 22:15:53 +00001653 * 1. Search for "target", update "closest" to the best match found.
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001654 * 2. If "target" not found search for "closest".
1655 *
1656 * When using the closest time we use the sequence number in the second
1657 * round, because there may be several entries with the same time.
1658 */
Bram Moolenaar1e607892006-03-13 22:15:53 +00001659 for (round = 1; round <= 2; ++round)
1660 {
1661 /* Find the path from the current state to where we want to go. The
1662 * desired state can be anywhere in the undo tree, need to go all over
1663 * it. We put "nomark" in uh_walk where we have been without success,
1664 * "mark" where it could possibly be. */
1665 mark = ++lastmark;
1666 nomark = ++lastmark;
1667
1668 if (curbuf->b_u_curhead == NULL) /* at leaf of the tree */
1669 uhp = curbuf->b_u_newhead;
1670 else
1671 uhp = curbuf->b_u_curhead;
1672
1673 while (uhp != NULL)
1674 {
1675 uhp->uh_walk = mark;
Bram Moolenaardb552d602006-03-23 22:59:57 +00001676 val = (long)(dosec ? (uhp->uh_time - starttime) : uhp->uh_seq);
Bram Moolenaar1e607892006-03-13 22:15:53 +00001677
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001678 if (round == 1)
1679 {
1680 /* Remember the header that is closest to the target.
1681 * It must be at least in the right direction (checked with
Bram Moolenaarca003e12006-03-17 23:19:38 +00001682 * "b_u_seq_cur"). When the timestamp is equal find the
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001683 * highest/lowest sequence number. */
Bram Moolenaarca003e12006-03-17 23:19:38 +00001684 if ((step < 0 ? uhp->uh_seq <= curbuf->b_u_seq_cur
1685 : uhp->uh_seq > curbuf->b_u_seq_cur)
1686 && ((dosec && val == closest)
1687 ? (step < 0
1688 ? uhp->uh_seq < closest_seq
1689 : uhp->uh_seq > closest_seq)
1690 : closest == closest_start
1691 || (val > target
1692 ? (closest > target
1693 ? val - target <= closest - target
1694 : val - target <= target - closest)
1695 : (closest > target
1696 ? target - val <= closest - target
1697 : target - val <= target - closest))))
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001698 {
1699 closest = val;
1700 closest_seq = uhp->uh_seq;
1701 }
1702 }
1703
1704 /* Quit searching when we found a match. But when searching for a
1705 * time we need to continue looking for the best uh_seq. */
1706 if (target == val && !dosec)
1707 break;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001708
1709 /* go down in the tree if we haven't been there */
1710 if (uhp->uh_prev != NULL && uhp->uh_prev->uh_walk != nomark
1711 && uhp->uh_prev->uh_walk != mark)
1712 uhp = uhp->uh_prev;
1713
1714 /* go to alternate branch if we haven't been there */
1715 else if (uhp->uh_alt_next != NULL
1716 && uhp->uh_alt_next->uh_walk != nomark
1717 && uhp->uh_alt_next->uh_walk != mark)
1718 uhp = uhp->uh_alt_next;
1719
1720 /* go up in the tree if we haven't been there and we are at the
1721 * start of alternate branches */
1722 else if (uhp->uh_next != NULL && uhp->uh_alt_prev == NULL
1723 && uhp->uh_next->uh_walk != nomark
1724 && uhp->uh_next->uh_walk != mark)
Bram Moolenaardb552d602006-03-23 22:59:57 +00001725 {
1726 /* If still at the start we don't go through this change. */
1727 if (uhp == curbuf->b_u_curhead)
1728 uhp->uh_walk = nomark;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001729 uhp = uhp->uh_next;
Bram Moolenaardb552d602006-03-23 22:59:57 +00001730 }
Bram Moolenaar1e607892006-03-13 22:15:53 +00001731
1732 else
1733 {
1734 /* need to backtrack; mark this node as useless */
1735 uhp->uh_walk = nomark;
1736 if (uhp->uh_alt_prev != NULL)
1737 uhp = uhp->uh_alt_prev;
1738 else
1739 uhp = uhp->uh_next;
1740 }
1741 }
1742
1743 if (uhp != NULL) /* found it */
1744 break;
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001745
1746 if (absolute)
1747 {
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001748 EMSGN(_("E830: Undo number %ld not found"), step);
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00001749 return;
1750 }
1751
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001752 if (closest == closest_start)
Bram Moolenaar1e607892006-03-13 22:15:53 +00001753 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001754 if (step < 0)
1755 MSG(_("Already at oldest change"));
1756 else
1757 MSG(_("Already at newest change"));
Bram Moolenaar1e607892006-03-13 22:15:53 +00001758 return;
1759 }
1760
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001761 target = closest_seq;
1762 dosec = FALSE;
1763 if (step < 0)
1764 above = TRUE; /* stop above the header */
Bram Moolenaar1e607892006-03-13 22:15:53 +00001765 }
1766
1767 /* If we found it: Follow the path to go to where we want to be. */
1768 if (uhp != NULL)
1769 {
1770 /*
1771 * First go up the tree as much as needed.
1772 */
1773 for (;;)
1774 {
1775 uhp = curbuf->b_u_curhead;
1776 if (uhp == NULL)
1777 uhp = curbuf->b_u_newhead;
1778 else
Bram Moolenaar1e607892006-03-13 22:15:53 +00001779 uhp = uhp->uh_next;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001780 if (uhp == NULL || uhp->uh_walk != mark
1781 || (uhp->uh_seq == target && !above))
Bram Moolenaar1e607892006-03-13 22:15:53 +00001782 break;
1783 curbuf->b_u_curhead = uhp;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001784 u_undoredo(TRUE);
Bram Moolenaar1e607892006-03-13 22:15:53 +00001785 uhp->uh_walk = nomark; /* don't go back down here */
Bram Moolenaar1e607892006-03-13 22:15:53 +00001786 }
1787
1788 /*
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001789 * And now go down the tree (redo), branching off where needed.
Bram Moolenaar1e607892006-03-13 22:15:53 +00001790 */
1791 uhp = curbuf->b_u_curhead;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001792 while (uhp != NULL)
Bram Moolenaar1e607892006-03-13 22:15:53 +00001793 {
Bram Moolenaar89ed3df2007-01-09 19:23:12 +00001794 /* Go back to the first branch with a mark. */
1795 while (uhp->uh_alt_prev != NULL
1796 && uhp->uh_alt_prev->uh_walk == mark)
1797 uhp = uhp->uh_alt_prev;
1798
Bram Moolenaar1e607892006-03-13 22:15:53 +00001799 /* Find the last branch with a mark, that's the one. */
1800 last = uhp;
1801 while (last->uh_alt_next != NULL
1802 && last->uh_alt_next->uh_walk == mark)
1803 last = last->uh_alt_next;
1804 if (last != uhp)
1805 {
1806 /* Make the used branch the first entry in the list of
1807 * alternatives to make "u" and CTRL-R take this branch. */
Bram Moolenaar89ed3df2007-01-09 19:23:12 +00001808 while (uhp->uh_alt_prev != NULL)
1809 uhp = uhp->uh_alt_prev;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001810 if (last->uh_alt_next != NULL)
1811 last->uh_alt_next->uh_alt_prev = last->uh_alt_prev;
1812 last->uh_alt_prev->uh_alt_next = last->uh_alt_next;
1813 last->uh_alt_prev = NULL;
1814 last->uh_alt_next = uhp;
1815 uhp->uh_alt_prev = last;
1816
1817 uhp = last;
Bram Moolenaarca003e12006-03-17 23:19:38 +00001818 if (uhp->uh_next != NULL)
1819 uhp->uh_next->uh_prev = uhp;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001820 }
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001821 curbuf->b_u_curhead = uhp;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001822
1823 if (uhp->uh_walk != mark)
1824 break; /* must have reached the target */
1825
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001826 /* Stop when going backwards in time and didn't find the exact
1827 * header we were looking for. */
1828 if (uhp->uh_seq == target && above)
Bram Moolenaardb552d602006-03-23 22:59:57 +00001829 {
1830 curbuf->b_u_seq_cur = target - 1;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001831 break;
Bram Moolenaardb552d602006-03-23 22:59:57 +00001832 }
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001833
Bram Moolenaarca003e12006-03-17 23:19:38 +00001834 u_undoredo(FALSE);
Bram Moolenaar1e607892006-03-13 22:15:53 +00001835
1836 /* Advance "curhead" to below the header we last used. If it
1837 * becomes NULL then we need to set "newhead" to this leaf. */
1838 if (uhp->uh_prev == NULL)
1839 curbuf->b_u_newhead = uhp;
1840 curbuf->b_u_curhead = uhp->uh_prev;
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001841 did_undo = FALSE;
Bram Moolenaar1e607892006-03-13 22:15:53 +00001842
1843 if (uhp->uh_seq == target) /* found it! */
1844 break;
1845
1846 uhp = uhp->uh_prev;
1847 if (uhp == NULL || uhp->uh_walk != mark)
1848 {
Bram Moolenaarca003e12006-03-17 23:19:38 +00001849 /* Need to redo more but can't find it... */
Bram Moolenaar1e607892006-03-13 22:15:53 +00001850 EMSG2(_(e_intern2), "undo_time()");
1851 break;
1852 }
1853 }
1854 }
Bram Moolenaardb552d602006-03-23 22:59:57 +00001855 u_undo_end(did_undo, absolute);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001856}
1857
1858/*
1859 * u_undoredo: common code for undo and redo
1860 *
1861 * The lines in the file are replaced by the lines in the entry list at
1862 * curbuf->b_u_curhead. The replaced lines in the file are saved in the entry
1863 * list for the next undo/redo.
Bram Moolenaarca003e12006-03-17 23:19:38 +00001864 *
1865 * When "undo" is TRUE we go up in the tree, when FALSE we go down.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001866 */
1867 static void
Bram Moolenaarca003e12006-03-17 23:19:38 +00001868u_undoredo(undo)
1869 int undo;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001870{
1871 char_u **newarray = NULL;
1872 linenr_T oldsize;
1873 linenr_T newsize;
1874 linenr_T top, bot;
1875 linenr_T lnum;
1876 linenr_T newlnum = MAXLNUM;
1877 long i;
1878 u_entry_T *uep, *nuep;
1879 u_entry_T *newlist = NULL;
1880 int old_flags;
1881 int new_flags;
1882 pos_T namedm[NMARKS];
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001883#ifdef FEAT_VISUAL
1884 visualinfo_T visualinfo;
1885#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886 int empty_buffer; /* buffer became empty */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001887 u_header_T *curhead = curbuf->b_u_curhead;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001888
Bram Moolenaarfecb6602007-10-01 20:54:15 +00001889#ifdef U_DEBUG
1890 u_check(FALSE);
1891#endif
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001892 old_flags = curhead->uh_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001893 new_flags = (curbuf->b_changed ? UH_CHANGED : 0) +
1894 ((curbuf->b_ml.ml_flags & ML_EMPTY) ? UH_EMPTYBUF : 0);
1895 setpcmark();
1896
1897 /*
1898 * save marks before undo/redo
1899 */
1900 mch_memmove(namedm, curbuf->b_namedm, sizeof(pos_T) * NMARKS);
Bram Moolenaara23ccb82006-02-27 00:08:02 +00001901#ifdef FEAT_VISUAL
1902 visualinfo = curbuf->b_visual;
1903#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904 curbuf->b_op_start.lnum = curbuf->b_ml.ml_line_count;
1905 curbuf->b_op_start.col = 0;
1906 curbuf->b_op_end.lnum = 0;
1907 curbuf->b_op_end.col = 0;
1908
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001909 for (uep = curhead->uh_entry; uep != NULL; uep = nuep)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910 {
1911 top = uep->ue_top;
1912 bot = uep->ue_bot;
1913 if (bot == 0)
1914 bot = curbuf->b_ml.ml_line_count + 1;
1915 if (top > curbuf->b_ml.ml_line_count || top >= bot
1916 || bot > curbuf->b_ml.ml_line_count + 1)
1917 {
1918 EMSG(_("E438: u_undo: line numbers wrong"));
1919 changed(); /* don't want UNCHANGED now */
1920 return;
1921 }
1922
1923 oldsize = bot - top - 1; /* number of lines before undo */
1924 newsize = uep->ue_size; /* number of lines after undo */
1925
1926 if (top < newlnum)
1927 {
1928 /* If the saved cursor is somewhere in this undo block, move it to
1929 * the remembered position. Makes "gwap" put the cursor back
1930 * where it was. */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001931 lnum = curhead->uh_cursor.lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 if (lnum >= top && lnum <= top + newsize + 1)
1933 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00001934 curwin->w_cursor = curhead->uh_cursor;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001935 newlnum = curwin->w_cursor.lnum - 1;
1936 }
1937 else
1938 {
1939 /* Use the first line that actually changed. Avoids that
1940 * undoing auto-formatting puts the cursor in the previous
1941 * line. */
1942 for (i = 0; i < newsize && i < oldsize; ++i)
1943 if (STRCMP(uep->ue_array[i], ml_get(top + 1 + i)) != 0)
1944 break;
1945 if (i == newsize && newlnum == MAXLNUM && uep->ue_next == NULL)
1946 {
1947 newlnum = top;
1948 curwin->w_cursor.lnum = newlnum + 1;
1949 }
1950 else if (i < newsize)
1951 {
1952 newlnum = top + i;
1953 curwin->w_cursor.lnum = newlnum + 1;
1954 }
1955 }
1956 }
1957
1958 empty_buffer = FALSE;
1959
1960 /* delete the lines between top and bot and save them in newarray */
Bram Moolenaar26a60b42005-02-22 08:49:11 +00001961 if (oldsize > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001962 {
Bram Moolenaar26a60b42005-02-22 08:49:11 +00001963 if ((newarray = (char_u **)U_ALLOC_LINE(
Bram Moolenaar071d4272004-06-13 20:20:40 +00001964 (unsigned)(sizeof(char_u *) * oldsize))) == NULL)
1965 {
1966 do_outofmem_msg((long_u)(sizeof(char_u *) * oldsize));
1967 /*
1968 * We have messed up the entry list, repair is impossible.
1969 * we have to free the rest of the list.
1970 */
1971 while (uep != NULL)
1972 {
1973 nuep = uep->ue_next;
1974 u_freeentry(uep, uep->ue_size);
1975 uep = nuep;
1976 }
1977 break;
1978 }
1979 /* delete backwards, it goes faster in most cases */
1980 for (lnum = bot - 1, i = oldsize; --i >= 0; --lnum)
1981 {
1982 /* what can we do when we run out of memory? */
1983 if ((newarray[i] = u_save_line(lnum)) == NULL)
1984 do_outofmem_msg((long_u)0);
1985 /* remember we deleted the last line in the buffer, and a
1986 * dummy empty line will be inserted */
1987 if (curbuf->b_ml.ml_line_count == 1)
1988 empty_buffer = TRUE;
1989 ml_delete(lnum, FALSE);
1990 }
1991 }
Bram Moolenaar8d343302005-07-12 22:46:17 +00001992 else
1993 newarray = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001994
1995 /* insert the lines in u_array between top and bot */
1996 if (newsize)
1997 {
1998 for (lnum = top, i = 0; i < newsize; ++i, ++lnum)
1999 {
2000 /*
2001 * If the file is empty, there is an empty line 1 that we
2002 * should get rid of, by replacing it with the new line
2003 */
2004 if (empty_buffer && lnum == 0)
2005 ml_replace((linenr_T)1, uep->ue_array[i], TRUE);
2006 else
2007 ml_append(lnum, uep->ue_array[i], (colnr_T)0, FALSE);
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002008 U_FREE_LINE(uep->ue_array[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002009 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002010 U_FREE_LINE((char_u *)uep->ue_array);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 }
2012
2013 /* adjust marks */
2014 if (oldsize != newsize)
2015 {
2016 mark_adjust(top + 1, top + oldsize, (long)MAXLNUM,
2017 (long)newsize - (long)oldsize);
2018 if (curbuf->b_op_start.lnum > top + oldsize)
2019 curbuf->b_op_start.lnum += newsize - oldsize;
2020 if (curbuf->b_op_end.lnum > top + oldsize)
2021 curbuf->b_op_end.lnum += newsize - oldsize;
2022 }
2023
2024 changed_lines(top + 1, 0, bot, newsize - oldsize);
2025
2026 /* set '[ and '] mark */
2027 if (top + 1 < curbuf->b_op_start.lnum)
2028 curbuf->b_op_start.lnum = top + 1;
2029 if (newsize == 0 && top + 1 > curbuf->b_op_end.lnum)
2030 curbuf->b_op_end.lnum = top + 1;
2031 else if (top + newsize > curbuf->b_op_end.lnum)
2032 curbuf->b_op_end.lnum = top + newsize;
2033
2034 u_newcount += newsize;
2035 u_oldcount += oldsize;
2036 uep->ue_size = oldsize;
2037 uep->ue_array = newarray;
2038 uep->ue_bot = top + newsize + 1;
2039
2040 /*
2041 * insert this entry in front of the new entry list
2042 */
2043 nuep = uep->ue_next;
2044 uep->ue_next = newlist;
2045 newlist = uep;
2046 }
2047
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002048 curhead->uh_entry = newlist;
2049 curhead->uh_flags = new_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002050 if ((old_flags & UH_EMPTYBUF) && bufempty())
2051 curbuf->b_ml.ml_flags |= ML_EMPTY;
2052 if (old_flags & UH_CHANGED)
2053 changed();
2054 else
Bram Moolenaar009b2592004-10-24 19:18:58 +00002055#ifdef FEAT_NETBEANS_INTG
2056 /* per netbeans undo rules, keep it as modified */
2057 if (!isNetbeansModified(curbuf))
2058#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 unchanged(curbuf, FALSE);
2060
2061 /*
2062 * restore marks from before undo/redo
2063 */
2064 for (i = 0; i < NMARKS; ++i)
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002065 if (curhead->uh_namedm[i].lnum != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002066 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002067 curbuf->b_namedm[i] = curhead->uh_namedm[i];
2068 curhead->uh_namedm[i] = namedm[i];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002069 }
Bram Moolenaara23ccb82006-02-27 00:08:02 +00002070#ifdef FEAT_VISUAL
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002071 if (curhead->uh_visual.vi_start.lnum != 0)
Bram Moolenaara23ccb82006-02-27 00:08:02 +00002072 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002073 curbuf->b_visual = curhead->uh_visual;
2074 curhead->uh_visual = visualinfo;
Bram Moolenaara23ccb82006-02-27 00:08:02 +00002075 }
2076#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002077
2078 /*
2079 * If the cursor is only off by one line, put it at the same position as
2080 * before starting the change (for the "o" command).
2081 * Otherwise the cursor should go to the first undone line.
2082 */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002083 if (curhead->uh_cursor.lnum + 1 == curwin->w_cursor.lnum
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 && curwin->w_cursor.lnum > 1)
2085 --curwin->w_cursor.lnum;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002086 if (curhead->uh_cursor.lnum == curwin->w_cursor.lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002087 {
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002088 curwin->w_cursor.col = curhead->uh_cursor.col;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089#ifdef FEAT_VIRTUALEDIT
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002090 if (virtual_active() && curhead->uh_cursor_vcol >= 0)
2091 coladvance((colnr_T)curhead->uh_cursor_vcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002092 else
2093 curwin->w_cursor.coladd = 0;
2094#endif
2095 }
2096 else if (curwin->w_cursor.lnum <= curbuf->b_ml.ml_line_count)
2097 beginline(BL_SOL | BL_FIX);
2098 else
2099 {
2100 /* We get here with the current cursor line being past the end (eg
2101 * after adding lines at the end of the file, and then undoing it).
2102 * check_cursor() will move the cursor to the last line. Move it to
2103 * the first column here. */
2104 curwin->w_cursor.col = 0;
2105#ifdef FEAT_VIRTUALEDIT
2106 curwin->w_cursor.coladd = 0;
2107#endif
2108 }
2109
2110 /* Make sure the cursor is on an existing line and column. */
2111 check_cursor();
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002112
2113 /* Remember where we are for "g-" and ":earlier 10s". */
2114 curbuf->b_u_seq_cur = curhead->uh_seq;
Bram Moolenaarca003e12006-03-17 23:19:38 +00002115 if (undo)
2116 /* We are below the previous undo. However, to make ":earlier 1s"
2117 * work we compute this as being just above the just undone change. */
2118 --curbuf->b_u_seq_cur;
2119
2120 /* The timestamp can be the same for multiple changes, just use the one of
2121 * the undone/redone change. */
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002122 curbuf->b_u_seq_time = curhead->uh_time;
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002123#ifdef U_DEBUG
2124 u_check(FALSE);
2125#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126}
2127
2128/*
2129 * If we deleted or added lines, report the number of less/more lines.
2130 * Otherwise, report the number of changes (this may be incorrect
2131 * in some cases, but it's better than nothing).
2132 */
2133 static void
Bram Moolenaardb552d602006-03-23 22:59:57 +00002134u_undo_end(did_undo, absolute)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00002135 int did_undo; /* just did an undo */
Bram Moolenaardb552d602006-03-23 22:59:57 +00002136 int absolute; /* used ":undo N" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002137{
Bram Moolenaar89d40322006-08-29 15:30:07 +00002138 char *msgstr;
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002139 u_header_T *uhp;
2140 char_u msgbuf[80];
Bram Moolenaar1e607892006-03-13 22:15:53 +00002141
Bram Moolenaar071d4272004-06-13 20:20:40 +00002142#ifdef FEAT_FOLDING
2143 if ((fdo_flags & FDO_UNDO) && KeyTyped)
2144 foldOpenCursor();
2145#endif
Bram Moolenaar1e607892006-03-13 22:15:53 +00002146
2147 if (global_busy /* no messages now, wait until global is finished */
2148 || !messaging()) /* 'lazyredraw' set, don't do messages now */
2149 return;
2150
2151 if (curbuf->b_ml.ml_flags & ML_EMPTY)
2152 --u_newcount;
2153
2154 u_oldcount -= u_newcount;
2155 if (u_oldcount == -1)
Bram Moolenaar89d40322006-08-29 15:30:07 +00002156 msgstr = N_("more line");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002157 else if (u_oldcount < 0)
Bram Moolenaar89d40322006-08-29 15:30:07 +00002158 msgstr = N_("more lines");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002159 else if (u_oldcount == 1)
Bram Moolenaar89d40322006-08-29 15:30:07 +00002160 msgstr = N_("line less");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002161 else if (u_oldcount > 1)
Bram Moolenaar89d40322006-08-29 15:30:07 +00002162 msgstr = N_("fewer lines");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002163 else
2164 {
2165 u_oldcount = u_newcount;
2166 if (u_newcount == 1)
Bram Moolenaar89d40322006-08-29 15:30:07 +00002167 msgstr = N_("change");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002168 else
Bram Moolenaar89d40322006-08-29 15:30:07 +00002169 msgstr = N_("changes");
Bram Moolenaar1e607892006-03-13 22:15:53 +00002170 }
2171
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002172 if (curbuf->b_u_curhead != NULL)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00002173 {
Bram Moolenaardb552d602006-03-23 22:59:57 +00002174 /* For ":undo N" we prefer a "after #N" message. */
2175 if (absolute && curbuf->b_u_curhead->uh_next != NULL)
2176 {
2177 uhp = curbuf->b_u_curhead->uh_next;
2178 did_undo = FALSE;
2179 }
2180 else if (did_undo)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00002181 uhp = curbuf->b_u_curhead;
2182 else
2183 uhp = curbuf->b_u_curhead->uh_next;
2184 }
Bram Moolenaar1e607892006-03-13 22:15:53 +00002185 else
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002186 uhp = curbuf->b_u_newhead;
Bram Moolenaar1e607892006-03-13 22:15:53 +00002187
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002188 if (uhp == NULL)
2189 *msgbuf = NUL;
2190 else
2191 u_add_time(msgbuf, sizeof(msgbuf), uhp->uh_time);
2192
Bram Moolenaar433f7c82006-03-21 21:29:36 +00002193 smsg((char_u *)_("%ld %s; %s #%ld %s"),
Bram Moolenaarca003e12006-03-17 23:19:38 +00002194 u_oldcount < 0 ? -u_oldcount : u_oldcount,
Bram Moolenaar89d40322006-08-29 15:30:07 +00002195 _(msgstr),
Bram Moolenaar433f7c82006-03-21 21:29:36 +00002196 did_undo ? _("before") : _("after"),
2197 uhp == NULL ? 0L : uhp->uh_seq,
2198 msgbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002199}
2200
2201/*
2202 * u_sync: stop adding to the current entry list
2203 */
2204 void
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002205u_sync(force)
2206 int force; /* Also sync when no_u_sync is set. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002207{
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002208 /* Skip it when already synced or syncing is disabled. */
2209 if (curbuf->b_u_synced || (!force && no_u_sync > 0))
2210 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002211#if defined(FEAT_XIM) && defined(FEAT_GUI_GTK)
2212 if (im_is_preediting())
2213 return; /* XIM is busy, don't break an undo sequence */
2214#endif
2215 if (p_ul < 0)
2216 curbuf->b_u_synced = TRUE; /* no entries, nothing to do */
2217 else
2218 {
2219 u_getbot(); /* compute ue_bot of previous u_save */
2220 curbuf->b_u_curhead = NULL;
2221 }
2222}
2223
2224/*
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002225 * ":undolist": List the leafs of the undo tree
2226 */
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002227 void
2228ex_undolist(eap)
Bram Moolenaarfff2bee2010-05-15 13:56:02 +02002229 exarg_T *eap UNUSED;
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002230{
2231 garray_T ga;
2232 u_header_T *uhp;
2233 int mark;
2234 int nomark;
2235 int changes = 1;
2236 int i;
2237
2238 /*
2239 * 1: walk the tree to find all leafs, put the info in "ga".
2240 * 2: sort the lines
2241 * 3: display the list
2242 */
2243 mark = ++lastmark;
2244 nomark = ++lastmark;
2245 ga_init2(&ga, (int)sizeof(char *), 20);
2246
2247 uhp = curbuf->b_u_oldhead;
2248 while (uhp != NULL)
2249 {
Bram Moolenaarca003e12006-03-17 23:19:38 +00002250 if (uhp->uh_prev == NULL && uhp->uh_walk != nomark
2251 && uhp->uh_walk != mark)
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002252 {
2253 if (ga_grow(&ga, 1) == FAIL)
2254 break;
2255 vim_snprintf((char *)IObuff, IOSIZE, "%6ld %7ld ",
2256 uhp->uh_seq, changes);
2257 u_add_time(IObuff + STRLEN(IObuff), IOSIZE - STRLEN(IObuff),
2258 uhp->uh_time);
2259 ((char_u **)(ga.ga_data))[ga.ga_len++] = vim_strsave(IObuff);
2260 }
2261
2262 uhp->uh_walk = mark;
2263
2264 /* go down in the tree if we haven't been there */
2265 if (uhp->uh_prev != NULL && uhp->uh_prev->uh_walk != nomark
2266 && uhp->uh_prev->uh_walk != mark)
2267 {
2268 uhp = uhp->uh_prev;
2269 ++changes;
2270 }
2271
2272 /* go to alternate branch if we haven't been there */
2273 else if (uhp->uh_alt_next != NULL
2274 && uhp->uh_alt_next->uh_walk != nomark
2275 && uhp->uh_alt_next->uh_walk != mark)
2276 uhp = uhp->uh_alt_next;
2277
2278 /* go up in the tree if we haven't been there and we are at the
2279 * start of alternate branches */
2280 else if (uhp->uh_next != NULL && uhp->uh_alt_prev == NULL
2281 && uhp->uh_next->uh_walk != nomark
2282 && uhp->uh_next->uh_walk != mark)
2283 {
2284 uhp = uhp->uh_next;
2285 --changes;
2286 }
2287
2288 else
2289 {
2290 /* need to backtrack; mark this node as done */
2291 uhp->uh_walk = nomark;
2292 if (uhp->uh_alt_prev != NULL)
2293 uhp = uhp->uh_alt_prev;
2294 else
2295 {
2296 uhp = uhp->uh_next;
2297 --changes;
2298 }
2299 }
2300 }
2301
2302 if (ga.ga_len == 0)
2303 MSG(_("Nothing to undo"));
2304 else
2305 {
2306 sort_strings((char_u **)ga.ga_data, ga.ga_len);
2307
2308 msg_start();
2309 msg_puts_attr((char_u *)_("number changes time"), hl_attr(HLF_T));
2310 for (i = 0; i < ga.ga_len && !got_int; ++i)
2311 {
2312 msg_putchar('\n');
2313 if (got_int)
2314 break;
2315 msg_puts(((char_u **)ga.ga_data)[i]);
2316 }
2317 msg_end();
2318
2319 ga_clear_strings(&ga);
2320 }
2321}
2322
2323/*
2324 * Put the timestamp of an undo header in "buf[buflen]" in a nice format.
2325 */
2326 static void
2327u_add_time(buf, buflen, tt)
2328 char_u *buf;
2329 size_t buflen;
2330 time_t tt;
2331{
2332#ifdef HAVE_STRFTIME
2333 struct tm *curtime;
2334
2335 if (time(NULL) - tt >= 100)
2336 {
2337 curtime = localtime(&tt);
Bram Moolenaar3991dab2006-03-27 17:01:56 +00002338 (void)strftime((char *)buf, buflen, "%H:%M:%S", curtime);
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002339 }
2340 else
2341#endif
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002342 vim_snprintf((char *)buf, buflen, _("%ld seconds ago"),
Bram Moolenaarefd2bf12006-03-16 21:41:35 +00002343 (long)(time(NULL) - tt));
2344}
2345
2346/*
Bram Moolenaare224ffa2006-03-01 00:01:28 +00002347 * ":undojoin": continue adding to the last entry list
2348 */
Bram Moolenaare224ffa2006-03-01 00:01:28 +00002349 void
2350ex_undojoin(eap)
Bram Moolenaarfff2bee2010-05-15 13:56:02 +02002351 exarg_T *eap UNUSED;
Bram Moolenaare224ffa2006-03-01 00:01:28 +00002352{
Bram Moolenaare224ffa2006-03-01 00:01:28 +00002353 if (curbuf->b_u_newhead == NULL)
2354 return; /* nothing changed before */
Bram Moolenaar57657d82006-04-21 22:12:41 +00002355 if (curbuf->b_u_curhead != NULL)
2356 {
2357 EMSG(_("E790: undojoin is not allowed after undo"));
2358 return;
2359 }
2360 if (!curbuf->b_u_synced)
2361 return; /* already unsynced */
Bram Moolenaare224ffa2006-03-01 00:01:28 +00002362 if (p_ul < 0)
2363 return; /* no entries, nothing to do */
2364 else
2365 {
2366 /* Go back to the last entry */
2367 curbuf->b_u_curhead = curbuf->b_u_newhead;
2368 curbuf->b_u_synced = FALSE; /* no entries, nothing to do */
2369 }
2370}
2371
2372/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002373 * Called after writing the file and setting b_changed to FALSE.
2374 * Now an undo means that the buffer is modified.
2375 */
2376 void
2377u_unchanged(buf)
2378 buf_T *buf;
2379{
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002380 u_unch_branch(buf->b_u_oldhead);
2381 buf->b_did_warn = FALSE;
2382}
2383
2384 static void
2385u_unch_branch(uhp)
2386 u_header_T *uhp;
2387{
Bram Moolenaar1e607892006-03-13 22:15:53 +00002388 u_header_T *uh;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002389
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002390 for (uh = uhp; uh != NULL; uh = uh->uh_prev)
2391 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002392 uh->uh_flags |= UH_CHANGED;
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002393 if (uh->uh_alt_next != NULL)
2394 u_unch_branch(uh->uh_alt_next); /* recursive */
2395 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002396}
2397
2398/*
2399 * Get pointer to last added entry.
2400 * If it's not valid, give an error message and return NULL.
2401 */
2402 static u_entry_T *
2403u_get_headentry()
2404{
2405 if (curbuf->b_u_newhead == NULL || curbuf->b_u_newhead->uh_entry == NULL)
2406 {
2407 EMSG(_("E439: undo list corrupt"));
2408 return NULL;
2409 }
2410 return curbuf->b_u_newhead->uh_entry;
2411}
2412
2413/*
2414 * u_getbot(): compute the line number of the previous u_save
2415 * It is called only when b_u_synced is FALSE.
2416 */
2417 static void
2418u_getbot()
2419{
2420 u_entry_T *uep;
2421 linenr_T extra;
2422
2423 uep = u_get_headentry(); /* check for corrupt undo list */
2424 if (uep == NULL)
2425 return;
2426
2427 uep = curbuf->b_u_newhead->uh_getbot_entry;
2428 if (uep != NULL)
2429 {
2430 /*
2431 * the new ue_bot is computed from the number of lines that has been
2432 * inserted (0 - deleted) since calling u_save. This is equal to the
2433 * old line count subtracted from the current line count.
2434 */
2435 extra = curbuf->b_ml.ml_line_count - uep->ue_lcount;
2436 uep->ue_bot = uep->ue_top + uep->ue_size + 1 + extra;
2437 if (uep->ue_bot < 1 || uep->ue_bot > curbuf->b_ml.ml_line_count)
2438 {
2439 EMSG(_("E440: undo line missing"));
2440 uep->ue_bot = uep->ue_top + 1; /* assume all lines deleted, will
2441 * get all the old lines back
2442 * without deleting the current
2443 * ones */
2444 }
2445
2446 curbuf->b_u_newhead->uh_getbot_entry = NULL;
2447 }
2448
2449 curbuf->b_u_synced = TRUE;
2450}
2451
2452/*
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002453 * Free one header "uhp" and its entry list and adjust the pointers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002454 */
2455 static void
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002456u_freeheader(buf, uhp, uhpp)
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002457 buf_T *buf;
Bram Moolenaar1e607892006-03-13 22:15:53 +00002458 u_header_T *uhp;
2459 u_header_T **uhpp; /* if not NULL reset when freeing this header */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002460{
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002461 u_header_T *uhap;
2462
Bram Moolenaar1e607892006-03-13 22:15:53 +00002463 /* When there is an alternate redo list free that branch completely,
2464 * because we can never go there. */
2465 if (uhp->uh_alt_next != NULL)
2466 u_freebranch(buf, uhp->uh_alt_next, uhpp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002467
Bram Moolenaar1e607892006-03-13 22:15:53 +00002468 if (uhp->uh_alt_prev != NULL)
2469 uhp->uh_alt_prev->uh_alt_next = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002470
Bram Moolenaar1e607892006-03-13 22:15:53 +00002471 /* Update the links in the list to remove the header. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002472 if (uhp->uh_next == NULL)
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002473 buf->b_u_oldhead = uhp->uh_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002474 else
2475 uhp->uh_next->uh_prev = uhp->uh_prev;
2476
2477 if (uhp->uh_prev == NULL)
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002478 buf->b_u_newhead = uhp->uh_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 else
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002480 for (uhap = uhp->uh_prev; uhap != NULL; uhap = uhap->uh_alt_next)
2481 uhap->uh_next = uhp->uh_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002482
Bram Moolenaar1e607892006-03-13 22:15:53 +00002483 u_freeentries(buf, uhp, uhpp);
2484}
2485
2486/*
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002487 * Free an alternate branch and any following alternate branches.
Bram Moolenaar1e607892006-03-13 22:15:53 +00002488 */
2489 static void
2490u_freebranch(buf, uhp, uhpp)
2491 buf_T *buf;
2492 u_header_T *uhp;
2493 u_header_T **uhpp; /* if not NULL reset when freeing this header */
2494{
2495 u_header_T *tofree, *next;
2496
Bram Moolenaar07d06772007-11-10 21:51:15 +00002497 /* If this is the top branch we may need to use u_freeheader() to update
2498 * all the pointers. */
2499 if (uhp == buf->b_u_oldhead)
2500 {
2501 u_freeheader(buf, uhp, uhpp);
2502 return;
2503 }
2504
Bram Moolenaar1e607892006-03-13 22:15:53 +00002505 if (uhp->uh_alt_prev != NULL)
2506 uhp->uh_alt_prev->uh_alt_next = NULL;
2507
2508 next = uhp;
2509 while (next != NULL)
2510 {
2511 tofree = next;
2512 if (tofree->uh_alt_next != NULL)
2513 u_freebranch(buf, tofree->uh_alt_next, uhpp); /* recursive */
2514 next = tofree->uh_prev;
2515 u_freeentries(buf, tofree, uhpp);
2516 }
2517}
2518
2519/*
2520 * Free all the undo entries for one header and the header itself.
2521 * This means that "uhp" is invalid when returning.
2522 */
2523 static void
2524u_freeentries(buf, uhp, uhpp)
2525 buf_T *buf;
2526 u_header_T *uhp;
2527 u_header_T **uhpp; /* if not NULL reset when freeing this header */
2528{
2529 u_entry_T *uep, *nuep;
2530
2531 /* Check for pointers to the header that become invalid now. */
2532 if (buf->b_u_curhead == uhp)
2533 buf->b_u_curhead = NULL;
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002534 if (buf->b_u_newhead == uhp)
2535 buf->b_u_newhead = NULL; /* freeing the newest entry */
Bram Moolenaar1e607892006-03-13 22:15:53 +00002536 if (uhpp != NULL && uhp == *uhpp)
2537 *uhpp = NULL;
2538
2539 for (uep = uhp->uh_entry; uep != NULL; uep = nuep)
2540 {
2541 nuep = uep->ue_next;
2542 u_freeentry(uep, uep->ue_size);
2543 }
2544
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002545#ifdef U_DEBUG
2546 uhp->uh_magic = 0;
2547#endif
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002548 U_FREE_LINE((char_u *)uhp);
2549 --buf->b_u_numhead;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002550}
2551
2552/*
2553 * free entry 'uep' and 'n' lines in uep->ue_array[]
2554 */
2555 static void
2556u_freeentry(uep, n)
2557 u_entry_T *uep;
2558 long n;
2559{
Bram Moolenaar8d343302005-07-12 22:46:17 +00002560 while (n > 0)
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002561 U_FREE_LINE(uep->ue_array[--n]);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00002562 U_FREE_LINE((char_u *)uep->ue_array);
Bram Moolenaarfecb6602007-10-01 20:54:15 +00002563#ifdef U_DEBUG
2564 uep->ue_magic = 0;
2565#endif
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002566 U_FREE_LINE((char_u *)uep);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002567}
2568
2569/*
2570 * invalidate the undo buffer; called when storage has already been released
2571 */
2572 void
2573u_clearall(buf)
2574 buf_T *buf;
2575{
2576 buf->b_u_newhead = buf->b_u_oldhead = buf->b_u_curhead = NULL;
2577 buf->b_u_synced = TRUE;
2578 buf->b_u_numhead = 0;
2579 buf->b_u_line_ptr = NULL;
2580 buf->b_u_line_lnum = 0;
2581}
2582
2583/*
2584 * save the line "lnum" for the "U" command
2585 */
2586 void
2587u_saveline(lnum)
2588 linenr_T lnum;
2589{
2590 if (lnum == curbuf->b_u_line_lnum) /* line is already saved */
2591 return;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00002592 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count) /* should never happen */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002593 return;
2594 u_clearline();
2595 curbuf->b_u_line_lnum = lnum;
2596 if (curwin->w_cursor.lnum == lnum)
2597 curbuf->b_u_line_colnr = curwin->w_cursor.col;
2598 else
2599 curbuf->b_u_line_colnr = 0;
2600 if ((curbuf->b_u_line_ptr = u_save_line(lnum)) == NULL)
2601 do_outofmem_msg((long_u)0);
2602}
2603
2604/*
2605 * clear the line saved for the "U" command
2606 * (this is used externally for crossing a line while in insert mode)
2607 */
2608 void
2609u_clearline()
2610{
2611 if (curbuf->b_u_line_ptr != NULL)
2612 {
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002613 U_FREE_LINE(curbuf->b_u_line_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 curbuf->b_u_line_ptr = NULL;
2615 curbuf->b_u_line_lnum = 0;
2616 }
2617}
2618
2619/*
2620 * Implementation of the "U" command.
2621 * Differentiation from vi: "U" can be undone with the next "U".
2622 * We also allow the cursor to be in another line.
2623 */
2624 void
2625u_undoline()
2626{
2627 colnr_T t;
2628 char_u *oldp;
2629
2630 if (undo_off)
2631 return;
2632
Bram Moolenaare3300c82008-02-13 14:21:38 +00002633 if (curbuf->b_u_line_ptr == NULL
2634 || curbuf->b_u_line_lnum > curbuf->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002635 {
2636 beep_flush();
2637 return;
2638 }
Bram Moolenaare3300c82008-02-13 14:21:38 +00002639
2640 /* first save the line for the 'u' command */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002641 if (u_savecommon(curbuf->b_u_line_lnum - 1,
2642 curbuf->b_u_line_lnum + 1, (linenr_T)0) == FAIL)
2643 return;
2644 oldp = u_save_line(curbuf->b_u_line_lnum);
2645 if (oldp == NULL)
2646 {
2647 do_outofmem_msg((long_u)0);
2648 return;
2649 }
2650 ml_replace(curbuf->b_u_line_lnum, curbuf->b_u_line_ptr, TRUE);
2651 changed_bytes(curbuf->b_u_line_lnum, 0);
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002652 U_FREE_LINE(curbuf->b_u_line_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002653 curbuf->b_u_line_ptr = oldp;
2654
2655 t = curbuf->b_u_line_colnr;
2656 if (curwin->w_cursor.lnum == curbuf->b_u_line_lnum)
2657 curbuf->b_u_line_colnr = curwin->w_cursor.col;
2658 curwin->w_cursor.col = t;
2659 curwin->w_cursor.lnum = curbuf->b_u_line_lnum;
Bram Moolenaare3300c82008-02-13 14:21:38 +00002660 check_cursor_col();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002661}
2662
2663/*
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002664 * There are two implementations of the memory management for undo:
2665 * 1. Use the standard malloc()/free() functions.
2666 * This should be fast for allocating memory, but when a buffer is
2667 * abandoned every single allocated chunk must be freed, which may be slow.
2668 * 2. Allocate larger blocks of memory and keep track of chunks ourselves.
2669 * This is fast for abandoning, but the use of linked lists is slow for
2670 * finding a free chunk. Esp. when a lot of lines are changed or deleted.
2671 * A bit of profiling showed that the first method is faster, especially when
2672 * making a large number of changes, under the condition that malloc()/free()
2673 * is implemented efficiently.
2674 */
2675#ifdef U_USE_MALLOC
2676/*
2677 * Version of undo memory allocation using malloc()/free()
2678 *
2679 * U_FREE_LINE() and U_ALLOC_LINE() are macros that invoke vim_free() and
2680 * lalloc() directly.
2681 */
2682
2683/*
2684 * Free all allocated memory blocks for the buffer 'buf'.
2685 */
2686 void
2687u_blockfree(buf)
2688 buf_T *buf;
2689{
Bram Moolenaar1e607892006-03-13 22:15:53 +00002690 while (buf->b_u_oldhead != NULL)
Bram Moolenaar1f4d4de2006-03-14 23:00:46 +00002691 u_freeheader(buf, buf->b_u_oldhead, NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00002692 U_FREE_LINE(buf->b_u_line_ptr);
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002693}
2694
2695#else
2696/*
2697 * Storage allocation for the undo lines and blocks of the current file.
2698 * Version where Vim keeps track of the available memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699 */
2700
2701/*
2702 * Memory is allocated in relatively large blocks. These blocks are linked
2703 * in the allocated block list, headed by curbuf->b_block_head. They are all
2704 * freed when abandoning a file, so we don't have to free every single line.
2705 * The list is kept sorted on memory address.
2706 * block_alloc() allocates a block.
2707 * m_blockfree() frees all blocks.
2708 *
2709 * The available chunks of memory are kept in free chunk lists. There is
2710 * one free list for each block of allocated memory. The list is kept sorted
2711 * on memory address.
2712 * u_alloc_line() gets a chunk from the free lists.
2713 * u_free_line() returns a chunk to the free lists.
2714 * curbuf->b_m_search points to the chunk before the chunk that was
2715 * freed/allocated the last time.
2716 * curbuf->b_mb_current points to the b_head where curbuf->b_m_search
2717 * points into the free list.
2718 *
2719 *
2720 * b_block_head /---> block #1 /---> block #2
2721 * mb_next ---/ mb_next ---/ mb_next ---> NULL
2722 * mb_info mb_info mb_info
2723 * | | |
2724 * V V V
2725 * NULL free chunk #1.1 free chunk #2.1
2726 * | |
2727 * V V
2728 * free chunk #1.2 NULL
2729 * |
2730 * V
2731 * NULL
2732 *
2733 * When a single free chunk list would have been used, it could take a lot
2734 * of time in u_free_line() to find the correct place to insert a chunk in the
2735 * free list. The single free list would become very long when many lines are
2736 * changed (e.g. with :%s/^M$//).
2737 */
2738
2739 /*
2740 * this blocksize is used when allocating new lines
2741 */
2742#define MEMBLOCKSIZE 2044
2743
2744/*
2745 * The size field contains the size of the chunk, including the size field
2746 * itself.
2747 *
2748 * When the chunk is not in-use it is preceded with the m_info structure.
2749 * The m_next field links it in one of the free chunk lists.
2750 *
2751 * On most unix systems structures have to be longword (32 or 64 bit) aligned.
2752 * On most other systems they are short (16 bit) aligned.
2753 */
2754
2755/* the structure definitions are now in structs.h */
2756
2757#ifdef ALIGN_LONG
2758 /* size of m_size */
2759# define M_OFFSET (sizeof(long_u))
2760#else
2761 /* size of m_size */
2762# define M_OFFSET (sizeof(short_u))
2763#endif
2764
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002765static char_u *u_blockalloc __ARGS((long_u));
2766
Bram Moolenaar071d4272004-06-13 20:20:40 +00002767/*
2768 * Allocate a block of memory and link it in the allocated block list.
2769 */
2770 static char_u *
2771u_blockalloc(size)
2772 long_u size;
2773{
2774 mblock_T *p;
2775 mblock_T *mp, *next;
2776
2777 p = (mblock_T *)lalloc(size + sizeof(mblock_T), FALSE);
2778 if (p != NULL)
2779 {
2780 /* Insert the block into the allocated block list, keeping it
2781 sorted on address. */
2782 for (mp = &curbuf->b_block_head;
2783 (next = mp->mb_next) != NULL && next < p;
2784 mp = next)
2785 ;
2786 p->mb_next = next; /* link in block list */
2787 p->mb_size = size;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002788 p->mb_maxsize = 0; /* nothing free yet */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002789 mp->mb_next = p;
2790 p->mb_info.m_next = NULL; /* clear free list */
2791 p->mb_info.m_size = 0;
2792 curbuf->b_mb_current = p; /* remember current block */
2793 curbuf->b_m_search = NULL;
2794 p++; /* return usable memory */
2795 }
2796 return (char_u *)p;
2797}
2798
2799/*
2800 * free all allocated memory blocks for the buffer 'buf'
2801 */
2802 void
2803u_blockfree(buf)
2804 buf_T *buf;
2805{
2806 mblock_T *p, *np;
2807
2808 for (p = buf->b_block_head.mb_next; p != NULL; p = np)
2809 {
2810 np = p->mb_next;
2811 vim_free(p);
2812 }
2813 buf->b_block_head.mb_next = NULL;
2814 buf->b_m_search = NULL;
2815 buf->b_mb_current = NULL;
2816}
2817
2818/*
2819 * Free a chunk of memory for the current buffer.
2820 * Insert the chunk into the correct free list, keeping it sorted on address.
2821 */
2822 static void
2823u_free_line(ptr, keep)
2824 char_u *ptr;
2825 int keep; /* don't free the block when it's empty */
2826{
2827 minfo_T *next;
2828 minfo_T *prev, *curr;
2829 minfo_T *mp;
2830 mblock_T *nextb;
2831 mblock_T *prevb;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002832 long_u maxsize;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002833
2834 if (ptr == NULL || ptr == IObuff)
2835 return; /* illegal address can happen in out-of-memory situations */
2836
2837 mp = (minfo_T *)(ptr - M_OFFSET);
2838
2839 /* find block where chunk could be a part off */
2840 /* if we change curbuf->b_mb_current, curbuf->b_m_search is set to NULL */
2841 if (curbuf->b_mb_current == NULL || mp < (minfo_T *)curbuf->b_mb_current)
2842 {
2843 curbuf->b_mb_current = curbuf->b_block_head.mb_next;
2844 curbuf->b_m_search = NULL;
2845 }
2846 if ((nextb = curbuf->b_mb_current->mb_next) != NULL
2847 && (minfo_T *)nextb < mp)
2848 {
2849 curbuf->b_mb_current = nextb;
2850 curbuf->b_m_search = NULL;
2851 }
2852 while ((nextb = curbuf->b_mb_current->mb_next) != NULL
2853 && (minfo_T *)nextb < mp)
2854 curbuf->b_mb_current = nextb;
2855
2856 curr = NULL;
2857 /*
2858 * If mp is smaller than curbuf->b_m_search->m_next go to the start of
2859 * the free list
2860 */
2861 if (curbuf->b_m_search == NULL || mp < (curbuf->b_m_search->m_next))
2862 next = &(curbuf->b_mb_current->mb_info);
2863 else
2864 next = curbuf->b_m_search;
2865 /*
2866 * The following loop is executed very often.
2867 * Therefore it has been optimized at the cost of readability.
2868 * Keep it fast!
2869 */
2870#ifdef SLOW_BUT_EASY_TO_READ
2871 do
2872 {
2873 prev = curr;
2874 curr = next;
2875 next = next->m_next;
2876 }
2877 while (mp > next && next != NULL);
2878#else
2879 do /* first, middle, last */
2880 {
2881 prev = next->m_next; /* curr, next, prev */
2882 if (prev == NULL || mp <= prev)
2883 {
2884 prev = curr;
2885 curr = next;
2886 next = next->m_next;
2887 break;
2888 }
2889 curr = prev->m_next; /* next, prev, curr */
2890 if (curr == NULL || mp <= curr)
2891 {
2892 prev = next;
2893 curr = prev->m_next;
2894 next = curr->m_next;
2895 break;
2896 }
2897 next = curr->m_next; /* prev, curr, next */
2898 }
2899 while (mp > next && next != NULL);
2900#endif
2901
2902 /* if *mp and *next are concatenated, join them into one chunk */
2903 if ((char_u *)mp + mp->m_size == (char_u *)next)
2904 {
2905 mp->m_size += next->m_size;
2906 mp->m_next = next->m_next;
2907 }
2908 else
2909 mp->m_next = next;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002910 maxsize = mp->m_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002911
2912 /* if *curr and *mp are concatenated, join them */
2913 if (prev != NULL && (char_u *)curr + curr->m_size == (char_u *)mp)
2914 {
2915 curr->m_size += mp->m_size;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002916 maxsize = curr->m_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002917 curr->m_next = mp->m_next;
2918 curbuf->b_m_search = prev;
2919 }
2920 else
2921 {
2922 curr->m_next = mp;
2923 curbuf->b_m_search = curr; /* put curbuf->b_m_search before freed
2924 chunk */
2925 }
2926
2927 /*
Bram Moolenaar035db9f2007-05-10 18:02:27 +00002928 * If the block only contains free memory now, release it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002929 */
2930 if (!keep && curbuf->b_mb_current->mb_size
2931 == curbuf->b_mb_current->mb_info.m_next->m_size)
2932 {
2933 /* Find the block before the current one to be able to unlink it from
2934 * the list of blocks. */
2935 prevb = &curbuf->b_block_head;
2936 for (nextb = prevb->mb_next; nextb != curbuf->b_mb_current;
2937 nextb = nextb->mb_next)
2938 prevb = nextb;
2939 prevb->mb_next = nextb->mb_next;
2940 vim_free(nextb);
2941 curbuf->b_mb_current = NULL;
2942 curbuf->b_m_search = NULL;
2943 }
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002944 else if (curbuf->b_mb_current->mb_maxsize < maxsize)
2945 curbuf->b_mb_current->mb_maxsize = maxsize;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002946}
2947
2948/*
2949 * Allocate and initialize a new line structure with room for at least
2950 * 'size' characters plus a terminating NUL.
2951 */
2952 static char_u *
2953u_alloc_line(size)
2954 unsigned size;
2955{
2956 minfo_T *mp, *mprev, *mp2;
2957 mblock_T *mbp;
2958 int size_align;
2959
2960 /*
2961 * Add room for size field and trailing NUL byte.
2962 * Adjust for minimal size (must be able to store minfo_T
2963 * plus a trailing NUL, so the chunk can be released again)
2964 */
2965 size += M_OFFSET + 1;
2966 if (size < sizeof(minfo_T) + 1)
2967 size = sizeof(minfo_T) + 1;
2968
2969 /*
2970 * round size up for alignment
2971 */
2972 size_align = (size + ALIGN_MASK) & ~ALIGN_MASK;
2973
2974 /*
2975 * If curbuf->b_m_search is NULL (uninitialized free list) start at
2976 * curbuf->b_block_head
2977 */
2978 if (curbuf->b_mb_current == NULL || curbuf->b_m_search == NULL)
2979 {
2980 curbuf->b_mb_current = &curbuf->b_block_head;
2981 curbuf->b_m_search = &(curbuf->b_block_head.mb_info);
2982 }
2983
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002984 /* Search for a block with enough space. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002985 mbp = curbuf->b_mb_current;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002986 while (mbp->mb_maxsize < size_align)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987 {
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002988 if (mbp->mb_next != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002989 mbp = mbp->mb_next;
2990 else
2991 mbp = &curbuf->b_block_head;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002992 if (mbp == curbuf->b_mb_current)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002993 {
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002994 int n = (size_align > (MEMBLOCKSIZE / 4)
2995 ? size_align : MEMBLOCKSIZE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002996
Bram Moolenaar26a60b42005-02-22 08:49:11 +00002997 /* Back where we started in block list: need to add a new block
2998 * with enough space. */
2999 mp = (minfo_T *)u_blockalloc((long_u)n);
3000 if (mp == NULL)
3001 return (NULL);
3002 mp->m_size = n;
3003 u_free_line((char_u *)mp + M_OFFSET, TRUE);
3004 mbp = curbuf->b_mb_current;
3005 break;
3006 }
3007 }
3008 if (mbp != curbuf->b_mb_current)
3009 curbuf->b_m_search = &(mbp->mb_info);
3010
3011 /* In this block find a chunk with enough space. */
3012 mprev = curbuf->b_m_search;
3013 mp = curbuf->b_m_search->m_next;
Bram Moolenaar35fdbb52005-07-09 21:08:57 +00003014 for (;;)
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003015 {
3016 if (mp == NULL) /* at end of the list */
3017 mp = &(mbp->mb_info); /* wrap around to begin */
3018 if (mp->m_size >= size)
3019 break;
3020 if (mp == curbuf->b_m_search)
3021 {
3022 /* back where we started in free chunk list: "cannot happen" */
3023 EMSG2(_(e_intern2), "u_alloc_line()");
3024 return NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003025 }
3026 mprev = mp;
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003027 mp = mp->m_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003028 }
3029
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003030 /* when using the largest chunk adjust mb_maxsize */
3031 if (mp->m_size >= mbp->mb_maxsize)
3032 mbp->mb_maxsize = 0;
3033
Bram Moolenaar071d4272004-06-13 20:20:40 +00003034 /* if the chunk we found is large enough, split it up in two */
3035 if ((long)mp->m_size - size_align >= (long)(sizeof(minfo_T) + 1))
3036 {
3037 mp2 = (minfo_T *)((char_u *)mp + size_align);
3038 mp2->m_size = mp->m_size - size_align;
3039 mp2->m_next = mp->m_next;
3040 mprev->m_next = mp2;
3041 mp->m_size = size_align;
3042 }
3043 else /* remove *mp from the free list */
3044 {
3045 mprev->m_next = mp->m_next;
3046 }
3047 curbuf->b_m_search = mprev;
3048 curbuf->b_mb_current = mbp;
3049
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003050 /* If using the largest chunk need to find the new largest chunk */
3051 if (mbp->mb_maxsize == 0)
3052 for (mp2 = &(mbp->mb_info); mp2 != NULL; mp2 = mp2->m_next)
3053 if (mbp->mb_maxsize < mp2->m_size)
3054 mbp->mb_maxsize = mp2->m_size;
3055
Bram Moolenaar071d4272004-06-13 20:20:40 +00003056 mp = (minfo_T *)((char_u *)mp + M_OFFSET);
3057 *(char_u *)mp = NUL; /* set the first byte to NUL */
3058
3059 return ((char_u *)mp);
3060}
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003061#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003062
3063/*
3064 * u_save_line(): allocate memory with u_alloc_line() and copy line 'lnum'
3065 * into it.
3066 */
3067 static char_u *
3068u_save_line(lnum)
3069 linenr_T lnum;
3070{
3071 char_u *src;
3072 char_u *dst;
3073 unsigned len;
3074
3075 src = ml_get(lnum);
3076 len = (unsigned)STRLEN(src);
Bram Moolenaar26a60b42005-02-22 08:49:11 +00003077 if ((dst = U_ALLOC_LINE(len)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003078 mch_memmove(dst, src, (size_t)(len + 1));
3079 return (dst);
3080}
3081
3082/*
3083 * Check if the 'modified' flag is set, or 'ff' has changed (only need to
3084 * check the first character, because it can only be "dos", "unix" or "mac").
3085 * "nofile" and "scratch" type buffers are considered to always be unchanged.
3086 */
3087 int
3088bufIsChanged(buf)
3089 buf_T *buf;
3090{
3091 return
3092#ifdef FEAT_QUICKFIX
3093 !bt_dontwrite(buf) &&
3094#endif
3095 (buf->b_changed || file_ff_differs(buf));
3096}
3097
3098 int
3099curbufIsChanged()
3100{
3101 return
3102#ifdef FEAT_QUICKFIX
3103 !bt_dontwrite(curbuf) &&
3104#endif
3105 (curbuf->b_changed || file_ff_differs(curbuf));
3106}