ref: 0663fb8286dfe7d76770dc0d27e7d41e0ab0670c
dir: /undo.c/
/* vi:set ts=8 sts=4 sw=4: * * VIM - Vi IMproved by Bram Moolenaar * * Do ":help uganda" in Vim to read copying and usage conditions. * Do ":help credits" in Vim to see a list of people who contributed. * See README.txt for an overview of the Vim source code. */ /* * undo.c: multi level undo facility * * The saved lines are stored in a list of lists (one for each buffer): * * b_u_oldhead------------------------------------------------+ * | * V * +--------------+ +--------------+ +--------------+ * b_u_newhead--->| u_header | | u_header | | u_header | * | uh_next------>| uh_next------>| uh_next---->NULL * NULL<--------uh_prev |<---------uh_prev |<---------uh_prev | * | uh_entry | | uh_entry | | uh_entry | * +--------|-----+ +--------|-----+ +--------|-----+ * | | | * V V V * +--------------+ +--------------+ +--------------+ * | u_entry | | u_entry | | u_entry | * | ue_next | | ue_next | | ue_next | * +--------|-----+ +--------|-----+ +--------|-----+ * | | | * V V V * +--------------+ NULL NULL * | u_entry | * | ue_next | * +--------|-----+ * | * V * etc. * * Each u_entry list contains the information for one undo or redo. * curbuf->b_u_curhead points to the header of the last undo (the next redo), * or is NULL if nothing has been undone (end of the branch). * * For keeping alternate undo/redo branches the uh_alt field is used. Thus at * each point in the list a branch may appear for an alternate to redo. The * uh_seq field is numbered sequentially to be able to find a newer or older * branch. * * +---------------+ +---------------+ * b_u_oldhead --->| u_header | | u_header | * | uh_alt_next ---->| uh_alt_next ----> NULL * NULL <----- uh_alt_prev |<------ uh_alt_prev | * | uh_prev | | uh_prev | * +-----|---------+ +-----|---------+ * | | * V V * +---------------+ +---------------+ * | u_header | | u_header | * | uh_alt_next | | uh_alt_next | * b_u_newhead --->| uh_alt_prev | | uh_alt_prev | * | uh_prev | | uh_prev | * +-----|---------+ +-----|---------+ * | | * V V * NULL +---------------+ +---------------+ * | u_header | | u_header | * | uh_alt_next ---->| uh_alt_next | * | uh_alt_prev |<------ uh_alt_prev | * | uh_prev | | uh_prev | * +-----|---------+ +-----|---------+ * | | * etc. etc. * * * All data is allocated with U_ALLOC_LINE(), it will be freed as soon as the * buffer is unloaded. */ #include "vim.h" /* See below: use malloc()/free() for memory management. */ #define U_USE_MALLOC 1 static void u_unch_branch __ARGS((u_header_T *uhp)); static u_entry_T *u_get_headentry __ARGS((void)); static void u_getbot __ARGS((void)); static int u_savecommon __ARGS((linenr_T, linenr_T, linenr_T)); static void u_doit __ARGS((int count)); static void u_undoredo __ARGS((int undo)); static void u_undo_end __ARGS((int did_undo, int absolute)); static void u_add_time __ARGS((char_u *buf, size_t buflen, time_t tt)); static void u_freeheader __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp)); static void u_freebranch __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp)); static void u_freeentries __ARGS((buf_T *buf, u_header_T *uhp, u_header_T **uhpp)); static void u_freeentry __ARGS((u_entry_T *, long)); #ifdef U_USE_MALLOC # define U_FREE_LINE(ptr) vim_free(ptr) # define U_ALLOC_LINE(size) lalloc((long_u)((size) + 1), FALSE) #else static void u_free_line __ARGS((char_u *ptr, int keep)); static char_u *u_alloc_line __ARGS((unsigned size)); # define U_FREE_LINE(ptr) u_free_line((ptr), FALSE) # define U_ALLOC_LINE(size) u_alloc_line(size) #endif static char_u *u_save_line __ARGS((linenr_T)); static long u_newcount, u_oldcount; /* * When 'u' flag included in 'cpoptions', we behave like vi. Need to remember * the action that "u" should do. */ static int undo_undoes = FALSE; /* * Save the current line for both the "u" and "U" command. * Returns OK or FAIL. */ int u_save_cursor() { return (u_save((linenr_T)(curwin->w_cursor.lnum - 1), (linenr_T)(curwin->w_cursor.lnum + 1))); } /* * Save the lines between "top" and "bot" for both the "u" and "U" command. * "top" may be 0 and bot may be curbuf->b_ml.ml_line_count + 1. * Returns FAIL when lines could not be saved, OK otherwise. */ int u_save(top, bot) linenr_T top, bot; { if (undo_off) return OK; if (top > curbuf->b_ml.ml_line_count || top >= bot || bot > curbuf->b_ml.ml_line_count + 1) return FALSE; /* rely on caller to do error messages */ if (top + 2 == bot) u_saveline((linenr_T)(top + 1)); return (u_savecommon(top, bot, (linenr_T)0)); } /* * save the line "lnum" (used by ":s" and "~" command) * The line is replaced, so the new bottom line is lnum + 1. */ int u_savesub(lnum) linenr_T lnum; { if (undo_off) return OK; return (u_savecommon(lnum - 1, lnum + 1, lnum + 1)); } /* * a new line is inserted before line "lnum" (used by :s command) * The line is inserted, so the new bottom line is lnum + 1. */ int u_inssub(lnum) linenr_T lnum; { if (undo_off) return OK; return (u_savecommon(lnum - 1, lnum, lnum + 1)); } /* * save the lines "lnum" - "lnum" + nlines (used by delete command) * The lines are deleted, so the new bottom line is lnum, unless the buffer * becomes empty. */ int u_savedel(lnum, nlines) linenr_T lnum; long nlines; { if (undo_off) return OK; return (u_savecommon(lnum - 1, lnum + nlines, nlines == curbuf->b_ml.ml_line_count ? 2 : lnum)); } /* * Return TRUE when undo is allowed. Otherwise give an error message and * return FALSE. */ int undo_allowed() { /* Don't allow changes when 'modifiable' is off. */ if (!curbuf->b_p_ma) { EMSG(_(e_modifiable)); return FALSE; } #ifdef HAVE_SANDBOX /* In the sandbox it's not allowed to change the text. */ if (sandbox != 0) { EMSG(_(e_sandbox)); return FALSE; } #endif /* Don't allow changes in the buffer while editing the cmdline. The * caller of getcmdline() may get confused. */ if (textlock != 0) { EMSG(_(e_secure)); return FALSE; } return TRUE; } static int u_savecommon(top, bot, newbot) linenr_T top, bot; linenr_T newbot; { linenr_T lnum; long i; u_header_T *uhp; u_header_T *old_curhead; u_entry_T *uep; u_entry_T *prev_uep; long size; /* When making changes is not allowed return FAIL. It's a crude way to * make all change commands fail. */ if (!undo_allowed()) return FAIL; #ifdef FEAT_NETBEANS_INTG /* * Netbeans defines areas that cannot be modified. Bail out here when * trying to change text in a guarded area. */ if (usingNetbeans) { if (netbeans_is_guarded(top, bot)) { EMSG(_(e_guarded)); return FAIL; } if (curbuf->b_p_ro) { EMSG(_(e_nbreadonly)); return FAIL; } } #endif #ifdef FEAT_AUTOCMD /* * Saving text for undo means we are going to make a change. Give a * warning for a read-only file before making the change, so that the * FileChangedRO event can replace the buffer with a read-write version * (e.g., obtained from a source control system). */ change_warning(0); #endif size = bot - top - 1; /* * if curbuf->b_u_synced == TRUE make a new header */ if (curbuf->b_u_synced) { #ifdef FEAT_JUMPLIST /* Need to create new entry in b_changelist. */ curbuf->b_new_change = TRUE; #endif if (p_ul >= 0) { /* * Make a new header entry. Do this first so that we don't mess * up the undo info when out of memory. */ uhp = (u_header_T *)U_ALLOC_LINE((unsigned)sizeof(u_header_T)); if (uhp == NULL) goto nomem; } else uhp = NULL; /* * If we undid more than we redid, move the entry lists before and * including curbuf->b_u_curhead to an alternate branch. */ old_curhead = curbuf->b_u_curhead; if (old_curhead != NULL) { curbuf->b_u_newhead = old_curhead->uh_next; curbuf->b_u_curhead = NULL; } /* * free headers to keep the size right */ while (curbuf->b_u_numhead > p_ul && curbuf->b_u_oldhead != NULL) { u_header_T *uhfree = curbuf->b_u_oldhead; /* If there is no branch only free one header. */ if (uhfree->uh_alt_next == NULL) u_freeheader(curbuf, uhfree, &old_curhead); else { /* Free the oldest alternate branch as a whole. */ while (uhfree->uh_alt_next != NULL) uhfree = uhfree->uh_alt_next; u_freebranch(curbuf, uhfree, &old_curhead); } } if (uhp == NULL) /* no undo at all */ { if (old_curhead != NULL) u_freebranch(curbuf, old_curhead, NULL); curbuf->b_u_synced = FALSE; return OK; } uhp->uh_prev = NULL; uhp->uh_next = curbuf->b_u_newhead; uhp->uh_alt_next = old_curhead; if (old_curhead != NULL) { uhp->uh_alt_prev = old_curhead->uh_alt_prev; if (uhp->uh_alt_prev != NULL) uhp->uh_alt_prev->uh_alt_next = uhp; old_curhead->uh_alt_prev = uhp; if (curbuf->b_u_oldhead == old_curhead) curbuf->b_u_oldhead = uhp; } else uhp->uh_alt_prev = NULL; if (curbuf->b_u_newhead != NULL) curbuf->b_u_newhead->uh_prev = uhp; uhp->uh_seq = ++curbuf->b_u_seq_last; curbuf->b_u_seq_cur = uhp->uh_seq; uhp->uh_time = time(NULL); curbuf->b_u_seq_time = uhp->uh_time + 1; uhp->uh_walk = 0; uhp->uh_entry = NULL; uhp->uh_getbot_entry = NULL; uhp->uh_cursor = curwin->w_cursor; /* save cursor pos. for undo */ #ifdef FEAT_VIRTUALEDIT if (virtual_active() && curwin->w_cursor.coladd > 0) uhp->uh_cursor_vcol = getviscol(); else uhp->uh_cursor_vcol = -1; #endif /* save changed and buffer empty flag for undo */ uhp->uh_flags = (curbuf->b_changed ? UH_CHANGED : 0) + ((curbuf->b_ml.ml_flags & ML_EMPTY) ? UH_EMPTYBUF : 0); /* save named marks and Visual marks for undo */ mch_memmove(uhp->uh_namedm, curbuf->b_namedm, sizeof(pos_T) * NMARKS); #ifdef FEAT_VISUAL uhp->uh_visual = curbuf->b_visual; #endif curbuf->b_u_newhead = uhp; if (curbuf->b_u_oldhead == NULL) curbuf->b_u_oldhead = uhp; ++curbuf->b_u_numhead; } else { if (p_ul < 0) /* no undo at all */ return OK; /* * When saving a single line, and it has been saved just before, it * doesn't make sense saving it again. Saves a lot of memory when * making lots of changes inside the same line. * This is only possible if the previous change didn't increase or * decrease the number of lines. * Check the ten last changes. More doesn't make sense and takes too * long. */ if (size == 1) { uep = u_get_headentry(); prev_uep = NULL; for (i = 0; i < 10; ++i) { if (uep == NULL) break; /* If lines have been inserted/deleted we give up. * Also when the line was included in a multi-line save. */ if ((curbuf->b_u_newhead->uh_getbot_entry != uep ? (uep->ue_top + uep->ue_size + 1 != (uep->ue_bot == 0 ? curbuf->b_ml.ml_line_count + 1 : uep->ue_bot)) : uep->ue_lcount != curbuf->b_ml.ml_line_count) || (uep->ue_size > 1 && top >= uep->ue_top && top + 2 <= uep->ue_top + uep->ue_size + 1)) break; /* If it's the same line we can skip saving it again. */ if (uep->ue_size == 1 && uep->ue_top == top) { if (i > 0) { /* It's not the last entry: get ue_bot for the last * entry now. Following deleted/inserted lines go to * the re-used entry. */ u_getbot(); curbuf->b_u_synced = FALSE; /* Move the found entry to become the last entry. The * order of undo/redo doesn't matter for the entries * we move it over, since they don't change the line * count and don't include this line. It does matter * for the found entry if the line count is changed by * the executed command. */ prev_uep->ue_next = uep->ue_next; uep->ue_next = curbuf->b_u_newhead->uh_entry; curbuf->b_u_newhead->uh_entry = uep; } /* The executed command may change the line count. */ if (newbot != 0) uep->ue_bot = newbot; else if (bot > curbuf->b_ml.ml_line_count) uep->ue_bot = 0; else { uep->ue_lcount = curbuf->b_ml.ml_line_count; curbuf->b_u_newhead->uh_getbot_entry = uep; } return OK; } prev_uep = uep; uep = uep->ue_next; } } /* find line number for ue_bot for previous u_save() */ u_getbot(); } #if !defined(UNIX) && !defined(DJGPP) && !defined(WIN32) && !defined(__EMX__) /* * With Amiga and MSDOS 16 bit we can't handle big undo's, because * then u_alloc_line would have to allocate a block larger than 32K */ if (size >= 8000) goto nomem; #endif /* * add lines in front of entry list */ uep = (u_entry_T *)U_ALLOC_LINE((unsigned)sizeof(u_entry_T)); if (uep == NULL) goto nomem; uep->ue_size = size; uep->ue_top = top; if (newbot != 0) uep->ue_bot = newbot; /* * Use 0 for ue_bot if bot is below last line. * Otherwise we have to compute ue_bot later. */ else if (bot > curbuf->b_ml.ml_line_count) uep->ue_bot = 0; else { uep->ue_lcount = curbuf->b_ml.ml_line_count; curbuf->b_u_newhead->uh_getbot_entry = uep; } if (size > 0) { if ((uep->ue_array = (char_u **)U_ALLOC_LINE( (unsigned)(sizeof(char_u *) * size))) == NULL) { u_freeentry(uep, 0L); goto nomem; } for (i = 0, lnum = top + 1; i < size; ++i) { fast_breakcheck(); if (got_int) { u_freeentry(uep, i); return FAIL; } if ((uep->ue_array[i] = u_save_line(lnum++)) == NULL) { u_freeentry(uep, i); goto nomem; } } } else uep->ue_array = NULL; uep->ue_next = curbuf->b_u_newhead->uh_entry; curbuf->b_u_newhead->uh_entry = uep; curbuf->b_u_synced = FALSE; undo_undoes = FALSE; return OK; nomem: msg_silent = 0; /* must display the prompt */ if (ask_yesno((char_u *)_("No undo possible; continue anyway"), TRUE) == 'y') { undo_off = TRUE; /* will be reset when character typed */ return OK; } do_outofmem_msg((long_u)0); return FAIL; } /* * If 'cpoptions' contains 'u': Undo the previous undo or redo (vi compatible). * If 'cpoptions' does not contain 'u': Always undo. */ void u_undo(count) int count; { /* * If we get an undo command while executing a macro, we behave like the * original vi. If this happens twice in one macro the result will not * be compatible. */ if (curbuf->b_u_synced == FALSE) { u_sync(TRUE); count = 1; } if (vim_strchr(p_cpo, CPO_UNDO) == NULL) undo_undoes = TRUE; else undo_undoes = !undo_undoes; u_doit(count); } /* * If 'cpoptions' contains 'u': Repeat the previous undo or redo. * If 'cpoptions' does not contain 'u': Always redo. */ void u_redo(count) int count; { if (vim_strchr(p_cpo, CPO_UNDO) == NULL) undo_undoes = FALSE; u_doit(count); } /* * Undo or redo, depending on 'undo_undoes', 'count' times. */ static void u_doit(startcount) int startcount; { int count = startcount; if (!undo_allowed()) return; u_newcount = 0; u_oldcount = 0; if (curbuf->b_ml.ml_flags & ML_EMPTY) u_oldcount = -1; while (count--) { if (undo_undoes) { if (curbuf->b_u_curhead == NULL) /* first undo */ curbuf->b_u_curhead = curbuf->b_u_newhead; else if (p_ul > 0) /* multi level undo */ /* get next undo */ curbuf->b_u_curhead = curbuf->b_u_curhead->uh_next; /* nothing to undo */ if (curbuf->b_u_numhead == 0 || curbuf->b_u_curhead == NULL) { /* stick curbuf->b_u_curhead at end */ curbuf->b_u_curhead = curbuf->b_u_oldhead; beep_flush(); if (count == startcount - 1) { MSG(_("Already at oldest change")); return; } break; } u_undoredo(TRUE); } else { if (curbuf->b_u_curhead == NULL || p_ul <= 0) { beep_flush(); /* nothing to redo */ if (count == startcount - 1) { MSG(_("Already at newest change")); return; } break; } u_undoredo(FALSE); /* Advance for next redo. Set "newhead" when at the end of the * redoable changes. */ if (curbuf->b_u_curhead->uh_prev == NULL) curbuf->b_u_newhead = curbuf->b_u_curhead; curbuf->b_u_curhead = curbuf->b_u_curhead->uh_prev; } } u_undo_end(undo_undoes, FALSE); } static int lastmark = 0; /* * Undo or redo over the timeline. * When "step" is negative go back in time, otherwise goes forward in time. * When "sec" is FALSE make "step" steps, when "sec" is TRUE use "step" as * seconds. * When "absolute" is TRUE use "step" as the sequence number to jump to. * "sec" must be FALSE then. */ void undo_time(step, sec, absolute) long step; int sec; int absolute; { long target; long closest; long closest_start; long closest_seq = 0; long val; u_header_T *uhp; u_header_T *last; int mark; int nomark; int round; int dosec = sec; int above = FALSE; int did_undo = TRUE; /* First make sure the current undoable change is synced. */ if (curbuf->b_u_synced == FALSE) u_sync(TRUE); u_newcount = 0; u_oldcount = 0; if (curbuf->b_ml.ml_flags & ML_EMPTY) u_oldcount = -1; /* "target" is the node below which we want to be. * Init "closest" to a value we can't reach. */ if (absolute) { target = step; closest = -1; } else { /* When doing computations with time_t subtract starttime, because * time_t converted to a long may result in a wrong number. */ if (sec) target = (long)(curbuf->b_u_seq_time - starttime) + step; else target = curbuf->b_u_seq_cur + step; if (step < 0) { if (target < 0) target = 0; closest = -1; } else { if (sec) closest = (long)(time(NULL) - starttime + 1); else closest = curbuf->b_u_seq_last + 2; if (target >= closest) target = closest - 1; } } closest_start = closest; closest_seq = curbuf->b_u_seq_cur; /* * May do this twice: * 1. Search for "target", update "closest" to the best match found. * 2. If "target" not found search for "closest". * * When using the closest time we use the sequence number in the second * round, because there may be several entries with the same time. */ for (round = 1; round <= 2; ++round) { /* Find the path from the current state to where we want to go. The * desired state can be anywhere in the undo tree, need to go all over * it. We put "nomark" in uh_walk where we have been without success, * "mark" where it could possibly be. */ mark = ++lastmark; nomark = ++lastmark; if (curbuf->b_u_curhead == NULL) /* at leaf of the tree */ uhp = curbuf->b_u_newhead; else uhp = curbuf->b_u_curhead; while (uhp != NULL) { uhp->uh_walk = mark; val = (long)(dosec ? (uhp->uh_time - starttime) : uhp->uh_seq); if (round == 1) { /* Remember the header that is closest to the target. * It must be at least in the right direction (checked with * "b_u_seq_cur"). When the timestamp is equal find the * highest/lowest sequence number. */ if ((step < 0 ? uhp->uh_seq <= curbuf->b_u_seq_cur : uhp->uh_seq > curbuf->b_u_seq_cur) && ((dosec && val == closest) ? (step < 0 ? uhp->uh_seq < closest_seq : uhp->uh_seq > closest_seq) : closest == closest_start || (val > target ? (closest > target ? val - target <= closest - target : val - target <= target - closest) : (closest > target ? target - val <= closest - target : target - val <= target - closest)))) { closest = val; closest_seq = uhp->uh_seq; } } /* Quit searching when we found a match. But when searching for a * time we need to continue looking for the best uh_seq. */ if (target == val && !dosec) break; /* go down in the tree if we haven't been there */ if (uhp->uh_prev != NULL && uhp->uh_prev->uh_walk != nomark && uhp->uh_prev->uh_walk != mark) uhp = uhp->uh_prev; /* go to alternate branch if we haven't been there */ else if (uhp->uh_alt_next != NULL && uhp->uh_alt_next->uh_walk != nomark && uhp->uh_alt_next->uh_walk != mark) uhp = uhp->uh_alt_next; /* go up in the tree if we haven't been there and we are at the * start of alternate branches */ else if (uhp->uh_next != NULL && uhp->uh_alt_prev == NULL && uhp->uh_next->uh_walk != nomark && uhp->uh_next->uh_walk != mark) { /* If still at the start we don't go through this change. */ if (uhp == curbuf->b_u_curhead) uhp->uh_walk = nomark; uhp = uhp->uh_next; } else { /* need to backtrack; mark this node as useless */ uhp->uh_walk = nomark; if (uhp->uh_alt_prev != NULL) uhp = uhp->uh_alt_prev; else uhp = uhp->uh_next; } } if (uhp != NULL) /* found it */ break; if (absolute) { EMSGN(_("Undo number %ld not found"), step); return; } if (closest == closest_start) { if (step < 0) MSG(_("Already at oldest change")); else MSG(_("Already at newest change")); return; } target = closest_seq; dosec = FALSE; if (step < 0) above = TRUE; /* stop above the header */ } /* If we found it: Follow the path to go to where we want to be. */ if (uhp != NULL) { /* * First go up the tree as much as needed. */ for (;;) { uhp = curbuf->b_u_curhead; if (uhp == NULL) uhp = curbuf->b_u_newhead; else uhp = uhp->uh_next; if (uhp == NULL || uhp->uh_walk != mark || (uhp->uh_seq == target && !above)) break; curbuf->b_u_curhead = uhp; u_undoredo(TRUE); uhp->uh_walk = nomark; /* don't go back down here */ } /* * And now go down the tree (redo), branching off where needed. */ uhp = curbuf->b_u_curhead; while (uhp != NULL) { /* Go back to the first branch with a mark. */ while (uhp->uh_alt_prev != NULL && uhp->uh_alt_prev->uh_walk == mark) uhp = uhp->uh_alt_prev; /* Find the last branch with a mark, that's the one. */ last = uhp; while (last->uh_alt_next != NULL && last->uh_alt_next->uh_walk == mark) last = last->uh_alt_next; if (last != uhp) { /* Make the used branch the first entry in the list of * alternatives to make "u" and CTRL-R take this branch. */ while (uhp->uh_alt_prev != NULL) uhp = uhp->uh_alt_prev; if (last->uh_alt_next != NULL) last->uh_alt_next->uh_alt_prev = last->uh_alt_prev; last->uh_alt_prev->uh_alt_next = last->uh_alt_next; last->uh_alt_prev = NULL; last->uh_alt_next = uhp; uhp->uh_alt_prev = last; uhp = last; if (uhp->uh_next != NULL) uhp->uh_next->uh_prev = uhp; } curbuf->b_u_curhead = uhp; if (uhp->uh_walk != mark) break; /* must have reached the target */ /* Stop when going backwards in time and didn't find the exact * header we were looking for. */ if (uhp->uh_seq == target && above) { curbuf->b_u_seq_cur = target - 1; break; } u_undoredo(FALSE); /* Advance "curhead" to below the header we last used. If it * becomes NULL then we need to set "newhead" to this leaf. */ if (uhp->uh_prev == NULL) curbuf->b_u_newhead = uhp; curbuf->b_u_curhead = uhp->uh_prev; did_undo = FALSE; if (uhp->uh_seq == target) /* found it! */ break; uhp = uhp->uh_prev; if (uhp == NULL || uhp->uh_walk != mark) { /* Need to redo more but can't find it... */ EMSG2(_(e_intern2), "undo_time()"); break; } } } u_undo_end(did_undo, absolute); } /* * u_undoredo: common code for undo and redo * * The lines in the file are replaced by the lines in the entry list at * curbuf->b_u_curhead. The replaced lines in the file are saved in the entry * list for the next undo/redo. * * When "undo" is TRUE we go up in the tree, when FALSE we go down. */ static void u_undoredo(undo) int undo; { char_u **newarray = NULL; linenr_T oldsize; linenr_T newsize; linenr_T top, bot; linenr_T lnum; linenr_T newlnum = MAXLNUM; long i; u_entry_T *uep, *nuep; u_entry_T *newlist = NULL; int old_flags; int new_flags; pos_T namedm[NMARKS]; #ifdef FEAT_VISUAL visualinfo_T visualinfo; #endif int empty_buffer; /* buffer became empty */ u_header_T *curhead = curbuf->b_u_curhead; old_flags = curhead->uh_flags; new_flags = (curbuf->b_changed ? UH_CHANGED : 0) + ((curbuf->b_ml.ml_flags & ML_EMPTY) ? UH_EMPTYBUF : 0); setpcmark(); /* * save marks before undo/redo */ mch_memmove(namedm, curbuf->b_namedm, sizeof(pos_T) * NMARKS); #ifdef FEAT_VISUAL visualinfo = curbuf->b_visual; #endif curbuf->b_op_start.lnum = curbuf->b_ml.ml_line_count; curbuf->b_op_start.col = 0; curbuf->b_op_end.lnum = 0; curbuf->b_op_end.col = 0; for (uep = curhead->uh_entry; uep != NULL; uep = nuep) { top = uep->ue_top; bot = uep->ue_bot; if (bot == 0) bot = curbuf->b_ml.ml_line_count + 1; if (top > curbuf->b_ml.ml_line_count || top >= bot || bot > curbuf->b_ml.ml_line_count + 1) { EMSG(_("E438: u_undo: line numbers wrong")); changed(); /* don't want UNCHANGED now */ return; } oldsize = bot - top - 1; /* number of lines before undo */ newsize = uep->ue_size; /* number of lines after undo */ if (top < newlnum) { /* If the saved cursor is somewhere in this undo block, move it to * the remembered position. Makes "gwap" put the cursor back * where it was. */ lnum = curhead->uh_cursor.lnum; if (lnum >= top && lnum <= top + newsize + 1) { curwin->w_cursor = curhead->uh_cursor; newlnum = curwin->w_cursor.lnum - 1; } else { /* Use the first line that actually changed. Avoids that * undoing auto-formatting puts the cursor in the previous * line. */ for (i = 0; i < newsize && i < oldsize; ++i) if (STRCMP(uep->ue_array[i], ml_get(top + 1 + i)) != 0) break; if (i == newsize && newlnum == MAXLNUM && uep->ue_next == NULL) { newlnum = top; curwin->w_cursor.lnum = newlnum + 1; } else if (i < newsize) { newlnum = top + i; curwin->w_cursor.lnum = newlnum + 1; } } } empty_buffer = FALSE; /* delete the lines between top and bot and save them in newarray */ if (oldsize > 0) { if ((newarray = (char_u **)U_ALLOC_LINE( (unsigned)(sizeof(char_u *) * oldsize))) == NULL) { do_outofmem_msg((long_u)(sizeof(char_u *) * oldsize)); /* * We have messed up the entry list, repair is impossible. * we have to free the rest of the list. */ while (uep != NULL) { nuep = uep->ue_next; u_freeentry(uep, uep->ue_size); uep = nuep; } break; } /* delete backwards, it goes faster in most cases */ for (lnum = bot - 1, i = oldsize; --i >= 0; --lnum) { /* what can we do when we run out of memory? */ if ((newarray[i] = u_save_line(lnum)) == NULL) do_outofmem_msg((long_u)0); /* remember we deleted the last line in the buffer, and a * dummy empty line will be inserted */ if (curbuf->b_ml.ml_line_count == 1) empty_buffer = TRUE; ml_delete(lnum, FALSE); } } else newarray = NULL; /* insert the lines in u_array between top and bot */ if (newsize) { for (lnum = top, i = 0; i < newsize; ++i, ++lnum) { /* * If the file is empty, there is an empty line 1 that we * should get rid of, by replacing it with the new line */ if (empty_buffer && lnum == 0) ml_replace((linenr_T)1, uep->ue_array[i], TRUE); else ml_append(lnum, uep->ue_array[i], (colnr_T)0, FALSE); U_FREE_LINE(uep->ue_array[i]); } U_FREE_LINE((char_u *)uep->ue_array); } /* adjust marks */ if (oldsize != newsize) { mark_adjust(top + 1, top + oldsize, (long)MAXLNUM, (long)newsize - (long)oldsize); if (curbuf->b_op_start.lnum > top + oldsize) curbuf->b_op_start.lnum += newsize - oldsize; if (curbuf->b_op_end.lnum > top + oldsize) curbuf->b_op_end.lnum += newsize - oldsize; } changed_lines(top + 1, 0, bot, newsize - oldsize); /* set '[ and '] mark */ if (top + 1 < curbuf->b_op_start.lnum) curbuf->b_op_start.lnum = top + 1; if (newsize == 0 && top + 1 > curbuf->b_op_end.lnum) curbuf->b_op_end.lnum = top + 1; else if (top + newsize > curbuf->b_op_end.lnum) curbuf->b_op_end.lnum = top + newsize; u_newcount += newsize; u_oldcount += oldsize; uep->ue_size = oldsize; uep->ue_array = newarray; uep->ue_bot = top + newsize + 1; /* * insert this entry in front of the new entry list */ nuep = uep->ue_next; uep->ue_next = newlist; newlist = uep; } curhead->uh_entry = newlist; curhead->uh_flags = new_flags; if ((old_flags & UH_EMPTYBUF) && bufempty()) curbuf->b_ml.ml_flags |= ML_EMPTY; if (old_flags & UH_CHANGED) changed(); else #ifdef FEAT_NETBEANS_INTG /* per netbeans undo rules, keep it as modified */ if (!isNetbeansModified(curbuf)) #endif unchanged(curbuf, FALSE); /* * restore marks from before undo/redo */ for (i = 0; i < NMARKS; ++i) if (curhead->uh_namedm[i].lnum != 0) { curbuf->b_namedm[i] = curhead->uh_namedm[i]; curhead->uh_namedm[i] = namedm[i]; } #ifdef FEAT_VISUAL if (curhead->uh_visual.vi_start.lnum != 0) { curbuf->b_visual = curhead->uh_visual; curhead->uh_visual = visualinfo; } #endif /* * If the cursor is only off by one line, put it at the same position as * before starting the change (for the "o" command). * Otherwise the cursor should go to the first undone line. */ if (curhead->uh_cursor.lnum + 1 == curwin->w_cursor.lnum && curwin->w_cursor.lnum > 1) --curwin->w_cursor.lnum; if (curhead->uh_cursor.lnum == curwin->w_cursor.lnum) { curwin->w_cursor.col = curhead->uh_cursor.col; #ifdef FEAT_VIRTUALEDIT if (virtual_active() && curhead->uh_cursor_vcol >= 0) coladvance((colnr_T)curhead->uh_cursor_vcol); else curwin->w_cursor.coladd = 0; #endif } else if (curwin->w_cursor.lnum <= curbuf->b_ml.ml_line_count) beginline(BL_SOL | BL_FIX); else { /* We get here with the current cursor line being past the end (eg * after adding lines at the end of the file, and then undoing it). * check_cursor() will move the cursor to the last line. Move it to * the first column here. */ curwin->w_cursor.col = 0; #ifdef FEAT_VIRTUALEDIT curwin->w_cursor.coladd = 0; #endif } /* Make sure the cursor is on an existing line and column. */ check_cursor(); /* Remember where we are for "g-" and ":earlier 10s". */ curbuf->b_u_seq_cur = curhead->uh_seq; if (undo) /* We are below the previous undo. However, to make ":earlier 1s" * work we compute this as being just above the just undone change. */ --curbuf->b_u_seq_cur; /* The timestamp can be the same for multiple changes, just use the one of * the undone/redone change. */ curbuf->b_u_seq_time = curhead->uh_time; } /* * If we deleted or added lines, report the number of less/more lines. * Otherwise, report the number of changes (this may be incorrect * in some cases, but it's better than nothing). */ static void u_undo_end(did_undo, absolute) int did_undo; /* just did an undo */ int absolute; /* used ":undo N" */ { char *msgstr; u_header_T *uhp; char_u msgbuf[80]; #ifdef FEAT_FOLDING if ((fdo_flags & FDO_UNDO) && KeyTyped) foldOpenCursor(); #endif if (global_busy /* no messages now, wait until global is finished */ || !messaging()) /* 'lazyredraw' set, don't do messages now */ return; if (curbuf->b_ml.ml_flags & ML_EMPTY) --u_newcount; u_oldcount -= u_newcount; if (u_oldcount == -1) msgstr = N_("more line"); else if (u_oldcount < 0) msgstr = N_("more lines"); else if (u_oldcount == 1) msgstr = N_("line less"); else if (u_oldcount > 1) msgstr = N_("fewer lines"); else { u_oldcount = u_newcount; if (u_newcount == 1) msgstr = N_("change"); else msgstr = N_("changes"); } if (curbuf->b_u_curhead != NULL) { /* For ":undo N" we prefer a "after #N" message. */ if (absolute && curbuf->b_u_curhead->uh_next != NULL) { uhp = curbuf->b_u_curhead->uh_next; did_undo = FALSE; } else if (did_undo) uhp = curbuf->b_u_curhead; else uhp = curbuf->b_u_curhead->uh_next; } else uhp = curbuf->b_u_newhead; if (uhp == NULL) *msgbuf = NUL; else u_add_time(msgbuf, sizeof(msgbuf), uhp->uh_time); smsg((char_u *)_("%ld %s; %s #%ld %s"), u_oldcount < 0 ? -u_oldcount : u_oldcount, _(msgstr), did_undo ? _("before") : _("after"), uhp == NULL ? 0L : uhp->uh_seq, msgbuf); } /* * u_sync: stop adding to the current entry list */ void u_sync(force) int force; /* Also sync when no_u_sync is set. */ { /* Skip it when already synced or syncing is disabled. */ if (curbuf->b_u_synced || (!force && no_u_sync > 0)) return; #if defined(FEAT_XIM) && defined(FEAT_GUI_GTK) if (im_is_preediting()) return; /* XIM is busy, don't break an undo sequence */ #endif if (p_ul < 0) curbuf->b_u_synced = TRUE; /* no entries, nothing to do */ else { u_getbot(); /* compute ue_bot of previous u_save */ curbuf->b_u_curhead = NULL; } } /* * ":undolist": List the leafs of the undo tree */ /*ARGSUSED*/ void ex_undolist(eap) exarg_T *eap; { garray_T ga; u_header_T *uhp; int mark; int nomark; int changes = 1; int i; /* * 1: walk the tree to find all leafs, put the info in "ga". * 2: sort the lines * 3: display the list */ mark = ++lastmark; nomark = ++lastmark; ga_init2(&ga, (int)sizeof(char *), 20); uhp = curbuf->b_u_oldhead; while (uhp != NULL) { if (uhp->uh_prev == NULL && uhp->uh_walk != nomark && uhp->uh_walk != mark) { if (ga_grow(&ga, 1) == FAIL) break; vim_snprintf((char *)IObuff, IOSIZE, "%6ld %7ld ", uhp->uh_seq, changes); u_add_time(IObuff + STRLEN(IObuff), IOSIZE - STRLEN(IObuff), uhp->uh_time); ((char_u **)(ga.ga_data))[ga.ga_len++] = vim_strsave(IObuff); } uhp->uh_walk = mark; /* go down in the tree if we haven't been there */ if (uhp->uh_prev != NULL && uhp->uh_prev->uh_walk != nomark && uhp->uh_prev->uh_walk != mark) { uhp = uhp->uh_prev; ++changes; } /* go to alternate branch if we haven't been there */ else if (uhp->uh_alt_next != NULL && uhp->uh_alt_next->uh_walk != nomark && uhp->uh_alt_next->uh_walk != mark) uhp = uhp->uh_alt_next; /* go up in the tree if we haven't been there and we are at the * start of alternate branches */ else if (uhp->uh_next != NULL && uhp->uh_alt_prev == NULL && uhp->uh_next->uh_walk != nomark && uhp->uh_next->uh_walk != mark) { uhp = uhp->uh_next; --changes; } else { /* need to backtrack; mark this node as done */ uhp->uh_walk = nomark; if (uhp->uh_alt_prev != NULL) uhp = uhp->uh_alt_prev; else { uhp = uhp->uh_next; --changes; } } } if (ga.ga_len == 0) MSG(_("Nothing to undo")); else { sort_strings((char_u **)ga.ga_data, ga.ga_len); msg_start(); msg_puts_attr((char_u *)_("number changes time"), hl_attr(HLF_T)); for (i = 0; i < ga.ga_len && !got_int; ++i) { msg_putchar('\n'); if (got_int) break; msg_puts(((char_u **)ga.ga_data)[i]); } msg_end(); ga_clear_strings(&ga); } } /* * Put the timestamp of an undo header in "buf[buflen]" in a nice format. */ static void u_add_time(buf, buflen, tt) char_u *buf; size_t buflen; time_t tt; { #ifdef HAVE_STRFTIME struct tm *curtime; if (time(NULL) - tt >= 100) { curtime = localtime(&tt); (void)strftime((char *)buf, buflen, "%H:%M:%S", curtime); } else #endif vim_snprintf((char *)buf, buflen, _("%ld seconds ago"), (long)(time(NULL) - tt)); } /* * ":undojoin": continue adding to the last entry list */ /*ARGSUSED*/ void ex_undojoin(eap) exarg_T *eap; { if (curbuf->b_u_newhead == NULL) return; /* nothing changed before */ if (curbuf->b_u_curhead != NULL) { EMSG(_("E790: undojoin is not allowed after undo")); return; } if (!curbuf->b_u_synced) return; /* already unsynced */ if (p_ul < 0) return; /* no entries, nothing to do */ else { /* Go back to the last entry */ curbuf->b_u_curhead = curbuf->b_u_newhead; curbuf->b_u_synced = FALSE; /* no entries, nothing to do */ } } /* * Called after writing the file and setting b_changed to FALSE. * Now an undo means that the buffer is modified. */ void u_unchanged(buf) buf_T *buf; { u_unch_branch(buf->b_u_oldhead); buf->b_did_warn = FALSE; } static void u_unch_branch(uhp) u_header_T *uhp; { u_header_T *uh; for (uh = uhp; uh != NULL; uh = uh->uh_prev) { uh->uh_flags |= UH_CHANGED; if (uh->uh_alt_next != NULL) u_unch_branch(uh->uh_alt_next); /* recursive */ } } /* * Get pointer to last added entry. * If it's not valid, give an error message and return NULL. */ static u_entry_T * u_get_headentry() { if (curbuf->b_u_newhead == NULL || curbuf->b_u_newhead->uh_entry == NULL) { EMSG(_("E439: undo list corrupt")); return NULL; } return curbuf->b_u_newhead->uh_entry; } /* * u_getbot(): compute the line number of the previous u_save * It is called only when b_u_synced is FALSE. */ static void u_getbot() { u_entry_T *uep; linenr_T extra; uep = u_get_headentry(); /* check for corrupt undo list */ if (uep == NULL) return; uep = curbuf->b_u_newhead->uh_getbot_entry; if (uep != NULL) { /* * the new ue_bot is computed from the number of lines that has been * inserted (0 - deleted) since calling u_save. This is equal to the * old line count subtracted from the current line count. */ extra = curbuf->b_ml.ml_line_count - uep->ue_lcount; uep->ue_bot = uep->ue_top + uep->ue_size + 1 + extra; if (uep->ue_bot < 1 || uep->ue_bot > curbuf->b_ml.ml_line_count) { EMSG(_("E440: undo line missing")); uep->ue_bot = uep->ue_top + 1; /* assume all lines deleted, will * get all the old lines back * without deleting the current * ones */ } curbuf->b_u_newhead->uh_getbot_entry = NULL; } curbuf->b_u_synced = TRUE; } /* * Free one header and its entry list and adjust the pointers. */ static void u_freeheader(buf, uhp, uhpp) buf_T *buf; u_header_T *uhp; u_header_T **uhpp; /* if not NULL reset when freeing this header */ { /* When there is an alternate redo list free that branch completely, * because we can never go there. */ if (uhp->uh_alt_next != NULL) u_freebranch(buf, uhp->uh_alt_next, uhpp); if (uhp->uh_alt_prev != NULL) uhp->uh_alt_prev->uh_alt_next = NULL; /* Update the links in the list to remove the header. */ if (uhp->uh_next == NULL) buf->b_u_oldhead = uhp->uh_prev; else uhp->uh_next->uh_prev = uhp->uh_prev; if (uhp->uh_prev == NULL) buf->b_u_newhead = uhp->uh_next; else uhp->uh_prev->uh_next = uhp->uh_next; u_freeentries(buf, uhp, uhpp); } /* * Free an alternate branch and any following alternate branches. */ static void u_freebranch(buf, uhp, uhpp) buf_T *buf; u_header_T *uhp; u_header_T **uhpp; /* if not NULL reset when freeing this header */ { u_header_T *tofree, *next; if (uhp->uh_alt_prev != NULL) uhp->uh_alt_prev->uh_alt_next = NULL; next = uhp; while (next != NULL) { tofree = next; if (tofree->uh_alt_next != NULL) u_freebranch(buf, tofree->uh_alt_next, uhpp); /* recursive */ next = tofree->uh_prev; u_freeentries(buf, tofree, uhpp); } } /* * Free all the undo entries for one header and the header itself. * This means that "uhp" is invalid when returning. */ static void u_freeentries(buf, uhp, uhpp) buf_T *buf; u_header_T *uhp; u_header_T **uhpp; /* if not NULL reset when freeing this header */ { u_entry_T *uep, *nuep; /* Check for pointers to the header that become invalid now. */ if (buf->b_u_curhead == uhp) buf->b_u_curhead = NULL; if (uhpp != NULL && uhp == *uhpp) *uhpp = NULL; for (uep = uhp->uh_entry; uep != NULL; uep = nuep) { nuep = uep->ue_next; u_freeentry(uep, uep->ue_size); } U_FREE_LINE((char_u *)uhp); --buf->b_u_numhead; } /* * free entry 'uep' and 'n' lines in uep->ue_array[] */ static void u_freeentry(uep, n) u_entry_T *uep; long n; { while (n > 0) U_FREE_LINE(uep->ue_array[--n]); U_FREE_LINE((char_u *)uep->ue_array); U_FREE_LINE((char_u *)uep); } /* * invalidate the undo buffer; called when storage has already been released */ void u_clearall(buf) buf_T *buf; { buf->b_u_newhead = buf->b_u_oldhead = buf->b_u_curhead = NULL; buf->b_u_synced = TRUE; buf->b_u_numhead = 0; buf->b_u_line_ptr = NULL; buf->b_u_line_lnum = 0; } /* * save the line "lnum" for the "U" command */ void u_saveline(lnum) linenr_T lnum; { if (lnum == curbuf->b_u_line_lnum) /* line is already saved */ return; if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count) /* should never happen */ return; u_clearline(); curbuf->b_u_line_lnum = lnum; if (curwin->w_cursor.lnum == lnum) curbuf->b_u_line_colnr = curwin->w_cursor.col; else curbuf->b_u_line_colnr = 0; if ((curbuf->b_u_line_ptr = u_save_line(lnum)) == NULL) do_outofmem_msg((long_u)0); } /* * clear the line saved for the "U" command * (this is used externally for crossing a line while in insert mode) */ void u_clearline() { if (curbuf->b_u_line_ptr != NULL) { U_FREE_LINE(curbuf->b_u_line_ptr); curbuf->b_u_line_ptr = NULL; curbuf->b_u_line_lnum = 0; } } /* * Implementation of the "U" command. * Differentiation from vi: "U" can be undone with the next "U". * We also allow the cursor to be in another line. */ void u_undoline() { colnr_T t; char_u *oldp; if (undo_off) return; if (curbuf->b_u_line_ptr == NULL || curbuf->b_u_line_lnum > curbuf->b_ml.ml_line_count) { beep_flush(); return; } /* first save the line for the 'u' command */ if (u_savecommon(curbuf->b_u_line_lnum - 1, curbuf->b_u_line_lnum + 1, (linenr_T)0) == FAIL) return; oldp = u_save_line(curbuf->b_u_line_lnum); if (oldp == NULL) { do_outofmem_msg((long_u)0); return; } ml_replace(curbuf->b_u_line_lnum, curbuf->b_u_line_ptr, TRUE); changed_bytes(curbuf->b_u_line_lnum, 0); U_FREE_LINE(curbuf->b_u_line_ptr); curbuf->b_u_line_ptr = oldp; t = curbuf->b_u_line_colnr; if (curwin->w_cursor.lnum == curbuf->b_u_line_lnum) curbuf->b_u_line_colnr = curwin->w_cursor.col; curwin->w_cursor.col = t; curwin->w_cursor.lnum = curbuf->b_u_line_lnum; } /* * There are two implementations of the memory management for undo: * 1. Use the standard malloc()/free() functions. * This should be fast for allocating memory, but when a buffer is * abandoned every single allocated chunk must be freed, which may be slow. * 2. Allocate larger blocks of memory and keep track of chunks ourselves. * This is fast for abandoning, but the use of linked lists is slow for * finding a free chunk. Esp. when a lot of lines are changed or deleted. * A bit of profiling showed that the first method is faster, especially when * making a large number of changes, under the condition that malloc()/free() * is implemented efficiently. */ #ifdef U_USE_MALLOC /* * Version of undo memory allocation using malloc()/free() * * U_FREE_LINE() and U_ALLOC_LINE() are macros that invoke vim_free() and * lalloc() directly. */ /* * Free all allocated memory blocks for the buffer 'buf'. */ void u_blockfree(buf) buf_T *buf; { while (buf->b_u_oldhead != NULL) u_freeheader(buf, buf->b_u_oldhead, NULL); U_FREE_LINE(buf->b_u_line_ptr); } #else /* * Storage allocation for the undo lines and blocks of the current file. * Version where Vim keeps track of the available memory. */ /* * Memory is allocated in relatively large blocks. These blocks are linked * in the allocated block list, headed by curbuf->b_block_head. They are all * freed when abandoning a file, so we don't have to free every single line. * The list is kept sorted on memory address. * block_alloc() allocates a block. * m_blockfree() frees all blocks. * * The available chunks of memory are kept in free chunk lists. There is * one free list for each block of allocated memory. The list is kept sorted * on memory address. * u_alloc_line() gets a chunk from the free lists. * u_free_line() returns a chunk to the free lists. * curbuf->b_m_search points to the chunk before the chunk that was * freed/allocated the last time. * curbuf->b_mb_current points to the b_head where curbuf->b_m_search * points into the free list. * * * b_block_head /---> block #1 /---> block #2 * mb_next ---/ mb_next ---/ mb_next ---> NULL * mb_info mb_info mb_info * | | | * V V V * NULL free chunk #1.1 free chunk #2.1 * | | * V V * free chunk #1.2 NULL * | * V * NULL * * When a single free chunk list would have been used, it could take a lot * of time in u_free_line() to find the correct place to insert a chunk in the * free list. The single free list would become very long when many lines are * changed (e.g. with :%s/^M$//). */ /* * this blocksize is used when allocating new lines */ #define MEMBLOCKSIZE 2044 /* * The size field contains the size of the chunk, including the size field * itself. * * When the chunk is not in-use it is preceded with the m_info structure. * The m_next field links it in one of the free chunk lists. * * On most unix systems structures have to be longword (32 or 64 bit) aligned. * On most other systems they are short (16 bit) aligned. */ /* the structure definitions are now in structs.h */ #ifdef ALIGN_LONG /* size of m_size */ # define M_OFFSET (sizeof(long_u)) #else /* size of m_size */ # define M_OFFSET (sizeof(short_u)) #endif static char_u *u_blockalloc __ARGS((long_u)); /* * Allocate a block of memory and link it in the allocated block list. */ static char_u * u_blockalloc(size) long_u size; { mblock_T *p; mblock_T *mp, *next; p = (mblock_T *)lalloc(size + sizeof(mblock_T), FALSE); if (p != NULL) { /* Insert the block into the allocated block list, keeping it sorted on address. */ for (mp = &curbuf->b_block_head; (next = mp->mb_next) != NULL && next < p; mp = next) ; p->mb_next = next; /* link in block list */ p->mb_size = size; p->mb_maxsize = 0; /* nothing free yet */ mp->mb_next = p; p->mb_info.m_next = NULL; /* clear free list */ p->mb_info.m_size = 0; curbuf->b_mb_current = p; /* remember current block */ curbuf->b_m_search = NULL; p++; /* return usable memory */ } return (char_u *)p; } /* * free all allocated memory blocks for the buffer 'buf' */ void u_blockfree(buf) buf_T *buf; { mblock_T *p, *np; for (p = buf->b_block_head.mb_next; p != NULL; p = np) { np = p->mb_next; vim_free(p); } buf->b_block_head.mb_next = NULL; buf->b_m_search = NULL; buf->b_mb_current = NULL; } /* * Free a chunk of memory for the current buffer. * Insert the chunk into the correct free list, keeping it sorted on address. */ static void u_free_line(ptr, keep) char_u *ptr; int keep; /* don't free the block when it's empty */ { minfo_T *next; minfo_T *prev, *curr; minfo_T *mp; mblock_T *nextb; mblock_T *prevb; long_u maxsize; if (ptr == NULL || ptr == IObuff) return; /* illegal address can happen in out-of-memory situations */ mp = (minfo_T *)(ptr - M_OFFSET); /* find block where chunk could be a part off */ /* if we change curbuf->b_mb_current, curbuf->b_m_search is set to NULL */ if (curbuf->b_mb_current == NULL || mp < (minfo_T *)curbuf->b_mb_current) { curbuf->b_mb_current = curbuf->b_block_head.mb_next; curbuf->b_m_search = NULL; } if ((nextb = curbuf->b_mb_current->mb_next) != NULL && (minfo_T *)nextb < mp) { curbuf->b_mb_current = nextb; curbuf->b_m_search = NULL; } while ((nextb = curbuf->b_mb_current->mb_next) != NULL && (minfo_T *)nextb < mp) curbuf->b_mb_current = nextb; curr = NULL; /* * If mp is smaller than curbuf->b_m_search->m_next go to the start of * the free list */ if (curbuf->b_m_search == NULL || mp < (curbuf->b_m_search->m_next)) next = &(curbuf->b_mb_current->mb_info); else next = curbuf->b_m_search; /* * The following loop is executed very often. * Therefore it has been optimized at the cost of readability. * Keep it fast! */ #ifdef SLOW_BUT_EASY_TO_READ do { prev = curr; curr = next; next = next->m_next; } while (mp > next && next != NULL); #else do /* first, middle, last */ { prev = next->m_next; /* curr, next, prev */ if (prev == NULL || mp <= prev) { prev = curr; curr = next; next = next->m_next; break; } curr = prev->m_next; /* next, prev, curr */ if (curr == NULL || mp <= curr) { prev = next; curr = prev->m_next; next = curr->m_next; break; } next = curr->m_next; /* prev, curr, next */ } while (mp > next && next != NULL); #endif /* if *mp and *next are concatenated, join them into one chunk */ if ((char_u *)mp + mp->m_size == (char_u *)next) { mp->m_size += next->m_size; mp->m_next = next->m_next; } else mp->m_next = next; maxsize = mp->m_size; /* if *curr and *mp are concatenated, join them */ if (prev != NULL && (char_u *)curr + curr->m_size == (char_u *)mp) { curr->m_size += mp->m_size; maxsize = curr->m_size; curr->m_next = mp->m_next; curbuf->b_m_search = prev; } else { curr->m_next = mp; curbuf->b_m_search = curr; /* put curbuf->b_m_search before freed chunk */ } /* * If the block only contains free memory now, release it. */ if (!keep && curbuf->b_mb_current->mb_size == curbuf->b_mb_current->mb_info.m_next->m_size) { /* Find the block before the current one to be able to unlink it from * the list of blocks. */ prevb = &curbuf->b_block_head; for (nextb = prevb->mb_next; nextb != curbuf->b_mb_current; nextb = nextb->mb_next) prevb = nextb; prevb->mb_next = nextb->mb_next; vim_free(nextb); curbuf->b_mb_current = NULL; curbuf->b_m_search = NULL; } else if (curbuf->b_mb_current->mb_maxsize < maxsize) curbuf->b_mb_current->mb_maxsize = maxsize; } /* * Allocate and initialize a new line structure with room for at least * 'size' characters plus a terminating NUL. */ static char_u * u_alloc_line(size) unsigned size; { minfo_T *mp, *mprev, *mp2; mblock_T *mbp; int size_align; /* * Add room for size field and trailing NUL byte. * Adjust for minimal size (must be able to store minfo_T * plus a trailing NUL, so the chunk can be released again) */ size += M_OFFSET + 1; if (size < sizeof(minfo_T) + 1) size = sizeof(minfo_T) + 1; /* * round size up for alignment */ size_align = (size + ALIGN_MASK) & ~ALIGN_MASK; /* * If curbuf->b_m_search is NULL (uninitialized free list) start at * curbuf->b_block_head */ if (curbuf->b_mb_current == NULL || curbuf->b_m_search == NULL) { curbuf->b_mb_current = &curbuf->b_block_head; curbuf->b_m_search = &(curbuf->b_block_head.mb_info); } /* Search for a block with enough space. */ mbp = curbuf->b_mb_current; while (mbp->mb_maxsize < size_align) { if (mbp->mb_next != NULL) mbp = mbp->mb_next; else mbp = &curbuf->b_block_head; if (mbp == curbuf->b_mb_current) { int n = (size_align > (MEMBLOCKSIZE / 4) ? size_align : MEMBLOCKSIZE); /* Back where we started in block list: need to add a new block * with enough space. */ mp = (minfo_T *)u_blockalloc((long_u)n); if (mp == NULL) return (NULL); mp->m_size = n; u_free_line((char_u *)mp + M_OFFSET, TRUE); mbp = curbuf->b_mb_current; break; } } if (mbp != curbuf->b_mb_current) curbuf->b_m_search = &(mbp->mb_info); /* In this block find a chunk with enough space. */ mprev = curbuf->b_m_search; mp = curbuf->b_m_search->m_next; for (;;) { if (mp == NULL) /* at end of the list */ mp = &(mbp->mb_info); /* wrap around to begin */ if (mp->m_size >= size) break; if (mp == curbuf->b_m_search) { /* back where we started in free chunk list: "cannot happen" */ EMSG2(_(e_intern2), "u_alloc_line()"); return NULL; } mprev = mp; mp = mp->m_next; } /* when using the largest chunk adjust mb_maxsize */ if (mp->m_size >= mbp->mb_maxsize) mbp->mb_maxsize = 0; /* if the chunk we found is large enough, split it up in two */ if ((long)mp->m_size - size_align >= (long)(sizeof(minfo_T) + 1)) { mp2 = (minfo_T *)((char_u *)mp + size_align); mp2->m_size = mp->m_size - size_align; mp2->m_next = mp->m_next; mprev->m_next = mp2; mp->m_size = size_align; } else /* remove *mp from the free list */ { mprev->m_next = mp->m_next; } curbuf->b_m_search = mprev; curbuf->b_mb_current = mbp; /* If using the largest chunk need to find the new largest chunk */ if (mbp->mb_maxsize == 0) for (mp2 = &(mbp->mb_info); mp2 != NULL; mp2 = mp2->m_next) if (mbp->mb_maxsize < mp2->m_size) mbp->mb_maxsize = mp2->m_size; mp = (minfo_T *)((char_u *)mp + M_OFFSET); *(char_u *)mp = NUL; /* set the first byte to NUL */ return ((char_u *)mp); } #endif /* * u_save_line(): allocate memory with u_alloc_line() and copy line 'lnum' * into it. */ static char_u * u_save_line(lnum) linenr_T lnum; { char_u *src; char_u *dst; unsigned len; src = ml_get(lnum); len = (unsigned)STRLEN(src); if ((dst = U_ALLOC_LINE(len)) != NULL) mch_memmove(dst, src, (size_t)(len + 1)); return (dst); } /* * Check if the 'modified' flag is set, or 'ff' has changed (only need to * check the first character, because it can only be "dos", "unix" or "mac"). * "nofile" and "scratch" type buffers are considered to always be unchanged. */ int bufIsChanged(buf) buf_T *buf; { return #ifdef FEAT_QUICKFIX !bt_dontwrite(buf) && #endif (buf->b_changed || file_ff_differs(buf)); } int curbufIsChanged() { return #ifdef FEAT_QUICKFIX !bt_dontwrite(curbuf) && #endif (curbuf->b_changed || file_ff_differs(curbuf)); }