1 /* vi:set ts=8 sts=4 sw=4 noet: 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 * diff.c: code for diff'ing two, three or four buffers. 12 */ 13 14 #include "vim.h" 15 16 #if defined(FEAT_DIFF) || defined(PROTO) 17 18 static int diff_busy = FALSE; /* ex_diffgetput() is busy */ 19 20 /* flags obtained from the 'diffopt' option */ 21 #define DIFF_FILLER 1 /* display filler lines */ 22 #define DIFF_ICASE 2 /* ignore case */ 23 #define DIFF_IWHITE 4 /* ignore change in white space */ 24 #define DIFF_HORIZONTAL 8 /* horizontal splits */ 25 #define DIFF_VERTICAL 16 /* vertical splits */ 26 static int diff_flags = DIFF_FILLER; 27 28 #define LBUFLEN 50 /* length of line in diff file */ 29 30 static int diff_a_works = MAYBE; /* TRUE when "diff -a" works, FALSE when it 31 doesn't work, MAYBE when not checked yet */ 32 #if defined(MSWIN) 33 static int diff_bin_works = MAYBE; /* TRUE when "diff --binary" works, FALSE 34 when it doesn't work, MAYBE when not 35 checked yet */ 36 #endif 37 38 static int diff_buf_idx(buf_T *buf); 39 static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp); 40 static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, linenr_T line2, long amount, long amount_after); 41 static void diff_check_unchanged(tabpage_T *tp, diff_T *dp); 42 static int diff_check_sanity(tabpage_T *tp, diff_T *dp); 43 static void diff_redraw(int dofold); 44 static int diff_write(buf_T *buf, char_u *fname); 45 static void diff_file(char_u *tmp_orig, char_u *tmp_new, char_u *tmp_diff); 46 static int diff_equal_entry(diff_T *dp, int idx1, int idx2); 47 static int diff_cmp(char_u *s1, char_u *s2); 48 #ifdef FEAT_FOLDING 49 static void diff_fold_update(diff_T *dp, int skip_idx); 50 #endif 51 static void diff_read(int idx_orig, int idx_new, char_u *fname); 52 static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, int idx_new); 53 static diff_T *diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp); 54 55 #ifndef USE_CR 56 # define tag_fgets vim_fgets 57 #endif 58 59 /* 60 * Called when deleting or unloading a buffer: No longer make a diff with it. 61 */ 62 void 63 diff_buf_delete(buf_T *buf) 64 { 65 int i; 66 tabpage_T *tp; 67 68 FOR_ALL_TABPAGES(tp) 69 { 70 i = diff_buf_idx_tp(buf, tp); 71 if (i != DB_COUNT) 72 { 73 tp->tp_diffbuf[i] = NULL; 74 tp->tp_diff_invalid = TRUE; 75 if (tp == curtab) 76 diff_redraw(TRUE); 77 } 78 } 79 } 80 81 /* 82 * Check if the current buffer should be added to or removed from the list of 83 * diff buffers. 84 */ 85 void 86 diff_buf_adjust(win_T *win) 87 { 88 win_T *wp; 89 int i; 90 91 if (!win->w_p_diff) 92 { 93 /* When there is no window showing a diff for this buffer, remove 94 * it from the diffs. */ 95 FOR_ALL_WINDOWS(wp) 96 if (wp->w_buffer == win->w_buffer && wp->w_p_diff) 97 break; 98 if (wp == NULL) 99 { 100 i = diff_buf_idx(win->w_buffer); 101 if (i != DB_COUNT) 102 { 103 curtab->tp_diffbuf[i] = NULL; 104 curtab->tp_diff_invalid = TRUE; 105 diff_redraw(TRUE); 106 } 107 } 108 } 109 else 110 diff_buf_add(win->w_buffer); 111 } 112 113 /* 114 * Add a buffer to make diffs for. 115 * Call this when a new buffer is being edited in the current window where 116 * 'diff' is set. 117 * Marks the current buffer as being part of the diff and requiring updating. 118 * This must be done before any autocmd, because a command may use info 119 * about the screen contents. 120 */ 121 void 122 diff_buf_add(buf_T *buf) 123 { 124 int i; 125 126 if (diff_buf_idx(buf) != DB_COUNT) 127 return; /* It's already there. */ 128 129 for (i = 0; i < DB_COUNT; ++i) 130 if (curtab->tp_diffbuf[i] == NULL) 131 { 132 curtab->tp_diffbuf[i] = buf; 133 curtab->tp_diff_invalid = TRUE; 134 diff_redraw(TRUE); 135 return; 136 } 137 138 EMSGN(_("E96: Cannot diff more than %ld buffers"), DB_COUNT); 139 } 140 141 /* 142 * Find buffer "buf" in the list of diff buffers for the current tab page. 143 * Return its index or DB_COUNT if not found. 144 */ 145 static int 146 diff_buf_idx(buf_T *buf) 147 { 148 int idx; 149 150 for (idx = 0; idx < DB_COUNT; ++idx) 151 if (curtab->tp_diffbuf[idx] == buf) 152 break; 153 return idx; 154 } 155 156 /* 157 * Find buffer "buf" in the list of diff buffers for tab page "tp". 158 * Return its index or DB_COUNT if not found. 159 */ 160 static int 161 diff_buf_idx_tp(buf_T *buf, tabpage_T *tp) 162 { 163 int idx; 164 165 for (idx = 0; idx < DB_COUNT; ++idx) 166 if (tp->tp_diffbuf[idx] == buf) 167 break; 168 return idx; 169 } 170 171 /* 172 * Mark the diff info involving buffer "buf" as invalid, it will be updated 173 * when info is requested. 174 */ 175 void 176 diff_invalidate(buf_T *buf) 177 { 178 tabpage_T *tp; 179 int i; 180 181 FOR_ALL_TABPAGES(tp) 182 { 183 i = diff_buf_idx_tp(buf, tp); 184 if (i != DB_COUNT) 185 { 186 tp->tp_diff_invalid = TRUE; 187 if (tp == curtab) 188 diff_redraw(TRUE); 189 } 190 } 191 } 192 193 /* 194 * Called by mark_adjust(): update line numbers in "curbuf". 195 */ 196 void 197 diff_mark_adjust( 198 linenr_T line1, 199 linenr_T line2, 200 long amount, 201 long amount_after) 202 { 203 int idx; 204 tabpage_T *tp; 205 206 /* Handle all tab pages that use the current buffer in a diff. */ 207 FOR_ALL_TABPAGES(tp) 208 { 209 idx = diff_buf_idx_tp(curbuf, tp); 210 if (idx != DB_COUNT) 211 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after); 212 } 213 } 214 215 /* 216 * Update line numbers in tab page "tp" for "curbuf" with index "idx". 217 * This attempts to update the changes as much as possible: 218 * When inserting/deleting lines outside of existing change blocks, create a 219 * new change block and update the line numbers in following blocks. 220 * When inserting/deleting lines in existing change blocks, update them. 221 */ 222 static void 223 diff_mark_adjust_tp( 224 tabpage_T *tp, 225 int idx, 226 linenr_T line1, 227 linenr_T line2, 228 long amount, 229 long amount_after) 230 { 231 diff_T *dp; 232 diff_T *dprev; 233 diff_T *dnext; 234 int i; 235 int inserted, deleted; 236 int n, off; 237 linenr_T last; 238 linenr_T lnum_deleted = line1; /* lnum of remaining deletion */ 239 int check_unchanged; 240 241 if (line2 == MAXLNUM) 242 { 243 /* mark_adjust(99, MAXLNUM, 9, 0): insert lines */ 244 inserted = amount; 245 deleted = 0; 246 } 247 else if (amount_after > 0) 248 { 249 /* mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines*/ 250 inserted = amount_after; 251 deleted = 0; 252 } 253 else 254 { 255 /* mark_adjust(98, 99, MAXLNUM, -2): delete lines */ 256 inserted = 0; 257 deleted = -amount_after; 258 } 259 260 dprev = NULL; 261 dp = tp->tp_first_diff; 262 for (;;) 263 { 264 /* If the change is after the previous diff block and before the next 265 * diff block, thus not touching an existing change, create a new diff 266 * block. Don't do this when ex_diffgetput() is busy. */ 267 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2 268 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1)) 269 && (dprev == NULL 270 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1) 271 && !diff_busy) 272 { 273 dnext = diff_alloc_new(tp, dprev, dp); 274 if (dnext == NULL) 275 return; 276 277 dnext->df_lnum[idx] = line1; 278 dnext->df_count[idx] = inserted; 279 for (i = 0; i < DB_COUNT; ++i) 280 if (tp->tp_diffbuf[i] != NULL && i != idx) 281 { 282 if (dprev == NULL) 283 dnext->df_lnum[i] = line1; 284 else 285 dnext->df_lnum[i] = line1 286 + (dprev->df_lnum[i] + dprev->df_count[i]) 287 - (dprev->df_lnum[idx] + dprev->df_count[idx]); 288 dnext->df_count[i] = deleted; 289 } 290 } 291 292 /* if at end of the list, quit */ 293 if (dp == NULL) 294 break; 295 296 /* 297 * Check for these situations: 298 * 1 2 3 299 * 1 2 3 300 * line1 2 3 4 5 301 * 2 3 4 5 302 * 2 3 4 5 303 * line2 2 3 4 5 304 * 3 5 6 305 * 3 5 6 306 */ 307 /* compute last line of this change */ 308 last = dp->df_lnum[idx] + dp->df_count[idx] - 1; 309 310 /* 1. change completely above line1: nothing to do */ 311 if (last >= line1 - 1) 312 { 313 /* 6. change below line2: only adjust for amount_after; also when 314 * "deleted" became zero when deleted all lines between two diffs */ 315 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2) 316 { 317 if (amount_after == 0) 318 break; /* nothing left to change */ 319 dp->df_lnum[idx] += amount_after; 320 } 321 else 322 { 323 check_unchanged = FALSE; 324 325 /* 2. 3. 4. 5.: inserted/deleted lines touching this diff. */ 326 if (deleted > 0) 327 { 328 if (dp->df_lnum[idx] >= line1) 329 { 330 off = dp->df_lnum[idx] - lnum_deleted; 331 if (last <= line2) 332 { 333 /* 4. delete all lines of diff */ 334 if (dp->df_next != NULL 335 && dp->df_next->df_lnum[idx] - 1 <= line2) 336 { 337 /* delete continues in next diff, only do 338 * lines until that one */ 339 n = dp->df_next->df_lnum[idx] - lnum_deleted; 340 deleted -= n; 341 n -= dp->df_count[idx]; 342 lnum_deleted = dp->df_next->df_lnum[idx]; 343 } 344 else 345 n = deleted - dp->df_count[idx]; 346 dp->df_count[idx] = 0; 347 } 348 else 349 { 350 /* 5. delete lines at or just before top of diff */ 351 n = off; 352 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1; 353 check_unchanged = TRUE; 354 } 355 dp->df_lnum[idx] = line1; 356 } 357 else 358 { 359 off = 0; 360 if (last < line2) 361 { 362 /* 2. delete at end of of diff */ 363 dp->df_count[idx] -= last - lnum_deleted + 1; 364 if (dp->df_next != NULL 365 && dp->df_next->df_lnum[idx] - 1 <= line2) 366 { 367 /* delete continues in next diff, only do 368 * lines until that one */ 369 n = dp->df_next->df_lnum[idx] - 1 - last; 370 deleted -= dp->df_next->df_lnum[idx] 371 - lnum_deleted; 372 lnum_deleted = dp->df_next->df_lnum[idx]; 373 } 374 else 375 n = line2 - last; 376 check_unchanged = TRUE; 377 } 378 else 379 { 380 /* 3. delete lines inside the diff */ 381 n = 0; 382 dp->df_count[idx] -= deleted; 383 } 384 } 385 386 for (i = 0; i < DB_COUNT; ++i) 387 if (tp->tp_diffbuf[i] != NULL && i != idx) 388 { 389 dp->df_lnum[i] -= off; 390 dp->df_count[i] += n; 391 } 392 } 393 else 394 { 395 if (dp->df_lnum[idx] <= line1) 396 { 397 /* inserted lines somewhere in this diff */ 398 dp->df_count[idx] += inserted; 399 check_unchanged = TRUE; 400 } 401 else 402 /* inserted lines somewhere above this diff */ 403 dp->df_lnum[idx] += inserted; 404 } 405 406 if (check_unchanged) 407 /* Check if inserted lines are equal, may reduce the 408 * size of the diff. TODO: also check for equal lines 409 * in the middle and perhaps split the block. */ 410 diff_check_unchanged(tp, dp); 411 } 412 } 413 414 /* check if this block touches the previous one, may merge them. */ 415 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx] 416 == dp->df_lnum[idx]) 417 { 418 for (i = 0; i < DB_COUNT; ++i) 419 if (tp->tp_diffbuf[i] != NULL) 420 dprev->df_count[i] += dp->df_count[i]; 421 dprev->df_next = dp->df_next; 422 vim_free(dp); 423 dp = dprev->df_next; 424 } 425 else 426 { 427 /* Advance to next entry. */ 428 dprev = dp; 429 dp = dp->df_next; 430 } 431 } 432 433 dprev = NULL; 434 dp = tp->tp_first_diff; 435 while (dp != NULL) 436 { 437 /* All counts are zero, remove this entry. */ 438 for (i = 0; i < DB_COUNT; ++i) 439 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0) 440 break; 441 if (i == DB_COUNT) 442 { 443 dnext = dp->df_next; 444 vim_free(dp); 445 dp = dnext; 446 if (dprev == NULL) 447 tp->tp_first_diff = dnext; 448 else 449 dprev->df_next = dnext; 450 } 451 else 452 { 453 /* Advance to next entry. */ 454 dprev = dp; 455 dp = dp->df_next; 456 } 457 458 } 459 460 if (tp == curtab) 461 { 462 diff_redraw(TRUE); 463 464 /* Need to recompute the scroll binding, may remove or add filler 465 * lines (e.g., when adding lines above w_topline). But it's slow when 466 * making many changes, postpone until redrawing. */ 467 diff_need_scrollbind = TRUE; 468 } 469 } 470 471 /* 472 * Allocate a new diff block and link it between "dprev" and "dp". 473 */ 474 static diff_T * 475 diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp) 476 { 477 diff_T *dnew; 478 479 dnew = (diff_T *)alloc((unsigned)sizeof(diff_T)); 480 if (dnew != NULL) 481 { 482 dnew->df_next = dp; 483 if (dprev == NULL) 484 tp->tp_first_diff = dnew; 485 else 486 dprev->df_next = dnew; 487 } 488 return dnew; 489 } 490 491 /* 492 * Check if the diff block "dp" can be made smaller for lines at the start and 493 * end that are equal. Called after inserting lines. 494 * This may result in a change where all buffers have zero lines, the caller 495 * must take care of removing it. 496 */ 497 static void 498 diff_check_unchanged(tabpage_T *tp, diff_T *dp) 499 { 500 int i_org; 501 int i_new; 502 int off_org, off_new; 503 char_u *line_org; 504 int dir = FORWARD; 505 506 /* Find the first buffers, use it as the original, compare the other 507 * buffer lines against this one. */ 508 for (i_org = 0; i_org < DB_COUNT; ++i_org) 509 if (tp->tp_diffbuf[i_org] != NULL) 510 break; 511 if (i_org == DB_COUNT) /* safety check */ 512 return; 513 514 if (diff_check_sanity(tp, dp) == FAIL) 515 return; 516 517 /* First check lines at the top, then at the bottom. */ 518 off_org = 0; 519 off_new = 0; 520 for (;;) 521 { 522 /* Repeat until a line is found which is different or the number of 523 * lines has become zero. */ 524 while (dp->df_count[i_org] > 0) 525 { 526 /* Copy the line, the next ml_get() will invalidate it. */ 527 if (dir == BACKWARD) 528 off_org = dp->df_count[i_org] - 1; 529 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org], 530 dp->df_lnum[i_org] + off_org, FALSE)); 531 if (line_org == NULL) 532 return; 533 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new) 534 { 535 if (tp->tp_diffbuf[i_new] == NULL) 536 continue; 537 if (dir == BACKWARD) 538 off_new = dp->df_count[i_new] - 1; 539 /* if other buffer doesn't have this line, it was inserted */ 540 if (off_new < 0 || off_new >= dp->df_count[i_new]) 541 break; 542 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new], 543 dp->df_lnum[i_new] + off_new, FALSE)) != 0) 544 break; 545 } 546 vim_free(line_org); 547 548 /* Stop when a line isn't equal in all diff buffers. */ 549 if (i_new != DB_COUNT) 550 break; 551 552 /* Line matched in all buffers, remove it from the diff. */ 553 for (i_new = i_org; i_new < DB_COUNT; ++i_new) 554 if (tp->tp_diffbuf[i_new] != NULL) 555 { 556 if (dir == FORWARD) 557 ++dp->df_lnum[i_new]; 558 --dp->df_count[i_new]; 559 } 560 } 561 if (dir == BACKWARD) 562 break; 563 dir = BACKWARD; 564 } 565 } 566 567 /* 568 * Check if a diff block doesn't contain invalid line numbers. 569 * This can happen when the diff program returns invalid results. 570 */ 571 static int 572 diff_check_sanity(tabpage_T *tp, diff_T *dp) 573 { 574 int i; 575 576 for (i = 0; i < DB_COUNT; ++i) 577 if (tp->tp_diffbuf[i] != NULL) 578 if (dp->df_lnum[i] + dp->df_count[i] - 1 579 > tp->tp_diffbuf[i]->b_ml.ml_line_count) 580 return FAIL; 581 return OK; 582 } 583 584 /* 585 * Mark all diff buffers in the current tab page for redraw. 586 */ 587 static void 588 diff_redraw( 589 int dofold) /* also recompute the folds */ 590 { 591 win_T *wp; 592 int n; 593 594 FOR_ALL_WINDOWS(wp) 595 if (wp->w_p_diff) 596 { 597 redraw_win_later(wp, SOME_VALID); 598 #ifdef FEAT_FOLDING 599 if (dofold && foldmethodIsDiff(wp)) 600 foldUpdateAll(wp); 601 #endif 602 /* A change may have made filler lines invalid, need to take care 603 * of that for other windows. */ 604 n = diff_check(wp, wp->w_topline); 605 if ((wp != curwin && wp->w_topfill > 0) || n > 0) 606 { 607 if (wp->w_topfill > n) 608 wp->w_topfill = (n < 0 ? 0 : n); 609 else if (n > 0 && n > wp->w_topfill) 610 wp->w_topfill = n; 611 check_topfill(wp, FALSE); 612 } 613 } 614 } 615 616 /* 617 * Write buffer "buf" to file "name". 618 * Always use 'fileformat' set to "unix". 619 * Return FAIL for failure 620 */ 621 static int 622 diff_write(buf_T *buf, char_u *fname) 623 { 624 int r; 625 char_u *save_ff; 626 627 save_ff = buf->b_p_ff; 628 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX); 629 r = buf_write(buf, fname, NULL, (linenr_T)1, buf->b_ml.ml_line_count, 630 NULL, FALSE, FALSE, FALSE, TRUE); 631 free_string_option(buf->b_p_ff); 632 buf->b_p_ff = save_ff; 633 return r; 634 } 635 636 /* 637 * Completely update the diffs for the buffers involved. 638 * This uses the ordinary "diff" command. 639 * The buffers are written to a file, also for unmodified buffers (the file 640 * could have been produced by autocommands, e.g. the netrw plugin). 641 */ 642 void 643 ex_diffupdate( 644 exarg_T *eap) /* can be NULL */ 645 { 646 buf_T *buf; 647 int idx_orig; 648 int idx_new; 649 char_u *tmp_orig; 650 char_u *tmp_new; 651 char_u *tmp_diff; 652 FILE *fd; 653 int ok; 654 int io_error = FALSE; 655 656 /* Delete all diffblocks. */ 657 diff_clear(curtab); 658 curtab->tp_diff_invalid = FALSE; 659 660 /* Use the first buffer as the original text. */ 661 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig) 662 if (curtab->tp_diffbuf[idx_orig] != NULL) 663 break; 664 if (idx_orig == DB_COUNT) 665 return; 666 667 /* Only need to do something when there is another buffer. */ 668 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new) 669 if (curtab->tp_diffbuf[idx_new] != NULL) 670 break; 671 if (idx_new == DB_COUNT) 672 return; 673 674 /* We need three temp file names. */ 675 tmp_orig = vim_tempname('o', TRUE); 676 tmp_new = vim_tempname('n', TRUE); 677 tmp_diff = vim_tempname('d', TRUE); 678 if (tmp_orig == NULL || tmp_new == NULL || tmp_diff == NULL) 679 goto theend; 680 681 /* 682 * Do a quick test if "diff" really works. Otherwise it looks like there 683 * are no differences. Can't use the return value, it's non-zero when 684 * there are differences. 685 * May try twice, first with "-a" and then without. 686 */ 687 for (;;) 688 { 689 ok = FALSE; 690 fd = mch_fopen((char *)tmp_orig, "w"); 691 if (fd == NULL) 692 io_error = TRUE; 693 else 694 { 695 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1) 696 io_error = TRUE; 697 fclose(fd); 698 fd = mch_fopen((char *)tmp_new, "w"); 699 if (fd == NULL) 700 io_error = TRUE; 701 else 702 { 703 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1) 704 io_error = TRUE; 705 fclose(fd); 706 diff_file(tmp_orig, tmp_new, tmp_diff); 707 fd = mch_fopen((char *)tmp_diff, "r"); 708 if (fd == NULL) 709 io_error = TRUE; 710 else 711 { 712 char_u linebuf[LBUFLEN]; 713 714 for (;;) 715 { 716 /* There must be a line that contains "1c1". */ 717 if (tag_fgets(linebuf, LBUFLEN, fd)) 718 break; 719 if (STRNCMP(linebuf, "1c1", 3) == 0) 720 ok = TRUE; 721 } 722 fclose(fd); 723 } 724 mch_remove(tmp_diff); 725 mch_remove(tmp_new); 726 } 727 mch_remove(tmp_orig); 728 } 729 730 #ifdef FEAT_EVAL 731 /* When using 'diffexpr' break here. */ 732 if (*p_dex != NUL) 733 break; 734 #endif 735 736 #if defined(MSWIN) 737 /* If the "-a" argument works, also check if "--binary" works. */ 738 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE) 739 { 740 diff_a_works = TRUE; 741 diff_bin_works = TRUE; 742 continue; 743 } 744 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE) 745 { 746 /* Tried --binary, but it failed. "-a" works though. */ 747 diff_bin_works = FALSE; 748 ok = TRUE; 749 } 750 #endif 751 752 /* If we checked if "-a" works already, break here. */ 753 if (diff_a_works != MAYBE) 754 break; 755 diff_a_works = ok; 756 757 /* If "-a" works break here, otherwise retry without "-a". */ 758 if (ok) 759 break; 760 } 761 if (!ok) 762 { 763 if (io_error) 764 EMSG(_("E810: Cannot read or write temp files")); 765 EMSG(_("E97: Cannot create diffs")); 766 diff_a_works = MAYBE; 767 #if defined(MSWIN) 768 diff_bin_works = MAYBE; 769 #endif 770 goto theend; 771 } 772 773 /* :diffupdate! */ 774 if (eap != NULL && eap->forceit) 775 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new) 776 { 777 buf = curtab->tp_diffbuf[idx_new]; 778 if (buf_valid(buf)) 779 buf_check_timestamp(buf, FALSE); 780 } 781 782 /* Write the first buffer to a tempfile. */ 783 buf = curtab->tp_diffbuf[idx_orig]; 784 if (diff_write(buf, tmp_orig) == FAIL) 785 goto theend; 786 787 /* Make a difference between the first buffer and every other. */ 788 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new) 789 { 790 buf = curtab->tp_diffbuf[idx_new]; 791 if (buf == NULL || buf->b_ml.ml_mfp == NULL) 792 continue; /* skip buffer that isn't loaded */ 793 if (diff_write(buf, tmp_new) == FAIL) 794 continue; 795 diff_file(tmp_orig, tmp_new, tmp_diff); 796 797 /* Read the diff output and add each entry to the diff list. */ 798 diff_read(idx_orig, idx_new, tmp_diff); 799 mch_remove(tmp_diff); 800 mch_remove(tmp_new); 801 } 802 mch_remove(tmp_orig); 803 804 /* force updating cursor position on screen */ 805 curwin->w_valid_cursor.lnum = 0; 806 807 diff_redraw(TRUE); 808 809 theend: 810 vim_free(tmp_orig); 811 vim_free(tmp_new); 812 vim_free(tmp_diff); 813 } 814 815 /* 816 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff". 817 */ 818 static void 819 diff_file( 820 char_u *tmp_orig, 821 char_u *tmp_new, 822 char_u *tmp_diff) 823 { 824 char_u *cmd; 825 size_t len; 826 827 #ifdef FEAT_EVAL 828 if (*p_dex != NUL) 829 /* Use 'diffexpr' to generate the diff file. */ 830 eval_diff(tmp_orig, tmp_new, tmp_diff); 831 else 832 #endif 833 { 834 len = STRLEN(tmp_orig) + STRLEN(tmp_new) 835 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27; 836 cmd = alloc((unsigned)len); 837 if (cmd != NULL) 838 { 839 /* We don't want $DIFF_OPTIONS to get in the way. */ 840 if (getenv("DIFF_OPTIONS")) 841 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)""); 842 843 /* Build the diff command and execute it. Always use -a, binary 844 * differences are of no use. Ignore errors, diff returns 845 * non-zero when differences have been found. */ 846 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s %s", 847 diff_a_works == FALSE ? "" : "-a ", 848 #if defined(MSWIN) 849 diff_bin_works == TRUE ? "--binary " : "", 850 #else 851 "", 852 #endif 853 (diff_flags & DIFF_IWHITE) ? "-b " : "", 854 (diff_flags & DIFF_ICASE) ? "-i " : "", 855 tmp_orig, tmp_new); 856 append_redir(cmd, (int)len, p_srr, tmp_diff); 857 #ifdef FEAT_AUTOCMD 858 block_autocmds(); /* Avoid ShellCmdPost stuff */ 859 #endif 860 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT); 861 #ifdef FEAT_AUTOCMD 862 unblock_autocmds(); 863 #endif 864 vim_free(cmd); 865 } 866 } 867 } 868 869 /* 870 * Create a new version of a file from the current buffer and a diff file. 871 * The buffer is written to a file, also for unmodified buffers (the file 872 * could have been produced by autocommands, e.g. the netrw plugin). 873 */ 874 void 875 ex_diffpatch(exarg_T *eap) 876 { 877 char_u *tmp_orig; /* name of original temp file */ 878 char_u *tmp_new; /* name of patched temp file */ 879 char_u *buf = NULL; 880 size_t buflen; 881 win_T *old_curwin = curwin; 882 char_u *newname = NULL; /* name of patched file buffer */ 883 #ifdef UNIX 884 char_u dirbuf[MAXPATHL]; 885 char_u *fullname = NULL; 886 #endif 887 #ifdef FEAT_BROWSE 888 char_u *browseFile = NULL; 889 int browse_flag = cmdmod.browse; 890 #endif 891 stat_T st; 892 893 #ifdef FEAT_BROWSE 894 if (cmdmod.browse) 895 { 896 browseFile = do_browse(0, (char_u *)_("Patch file"), 897 eap->arg, NULL, NULL, BROWSE_FILTER_ALL_FILES, NULL); 898 if (browseFile == NULL) 899 return; /* operation cancelled */ 900 eap->arg = browseFile; 901 cmdmod.browse = FALSE; /* don't let do_ecmd() browse again */ 902 } 903 #endif 904 905 /* We need two temp file names. */ 906 tmp_orig = vim_tempname('o', FALSE); 907 tmp_new = vim_tempname('n', FALSE); 908 if (tmp_orig == NULL || tmp_new == NULL) 909 goto theend; 910 911 /* Write the current buffer to "tmp_orig". */ 912 if (buf_write(curbuf, tmp_orig, NULL, 913 (linenr_T)1, curbuf->b_ml.ml_line_count, 914 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL) 915 goto theend; 916 917 #ifdef UNIX 918 /* Get the absolute path of the patchfile, changing directory below. */ 919 fullname = FullName_save(eap->arg, FALSE); 920 #endif 921 buflen = STRLEN(tmp_orig) + ( 922 # ifdef UNIX 923 fullname != NULL ? STRLEN(fullname) : 924 # endif 925 STRLEN(eap->arg)) + STRLEN(tmp_new) + 16; 926 buf = alloc((unsigned)buflen); 927 if (buf == NULL) 928 goto theend; 929 930 #ifdef UNIX 931 /* Temporarily chdir to /tmp, to avoid patching files in the current 932 * directory when the patch file contains more than one patch. When we 933 * have our own temp dir use that instead, it will be cleaned up when we 934 * exit (any .rej files created). Don't change directory if we can't 935 * return to the current. */ 936 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0) 937 dirbuf[0] = NUL; 938 else 939 { 940 # ifdef TEMPDIRNAMES 941 if (vim_tempdir != NULL) 942 ignored = mch_chdir((char *)vim_tempdir); 943 else 944 # endif 945 ignored = mch_chdir("/tmp"); 946 shorten_fnames(TRUE); 947 } 948 #endif 949 950 #ifdef FEAT_EVAL 951 if (*p_pex != NUL) 952 /* Use 'patchexpr' to generate the new file. */ 953 eval_patch(tmp_orig, 954 # ifdef UNIX 955 fullname != NULL ? fullname : 956 # endif 957 eap->arg, tmp_new); 958 else 959 #endif 960 { 961 /* Build the patch command and execute it. Ignore errors. Switch to 962 * cooked mode to allow the user to respond to prompts. */ 963 vim_snprintf((char *)buf, buflen, "patch -o %s %s < \"%s\"", 964 tmp_new, tmp_orig, 965 # ifdef UNIX 966 fullname != NULL ? fullname : 967 # endif 968 eap->arg); 969 #ifdef FEAT_AUTOCMD 970 block_autocmds(); /* Avoid ShellCmdPost stuff */ 971 #endif 972 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED); 973 #ifdef FEAT_AUTOCMD 974 unblock_autocmds(); 975 #endif 976 } 977 978 #ifdef UNIX 979 if (dirbuf[0] != NUL) 980 { 981 if (mch_chdir((char *)dirbuf) != 0) 982 EMSG(_(e_prev_dir)); 983 shorten_fnames(TRUE); 984 } 985 #endif 986 987 /* patch probably has written over the screen */ 988 redraw_later(CLEAR); 989 990 /* Delete any .orig or .rej file created. */ 991 STRCPY(buf, tmp_new); 992 STRCAT(buf, ".orig"); 993 mch_remove(buf); 994 STRCPY(buf, tmp_new); 995 STRCAT(buf, ".rej"); 996 mch_remove(buf); 997 998 /* Only continue if the output file was created. */ 999 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0) 1000 EMSG(_("E816: Cannot read patch output")); 1001 else 1002 { 1003 if (curbuf->b_fname != NULL) 1004 { 1005 newname = vim_strnsave(curbuf->b_fname, 1006 (int)(STRLEN(curbuf->b_fname) + 4)); 1007 if (newname != NULL) 1008 STRCAT(newname, ".new"); 1009 } 1010 1011 #ifdef FEAT_GUI 1012 need_mouse_correct = TRUE; 1013 #endif 1014 /* don't use a new tab page, each tab page has its own diffs */ 1015 cmdmod.tab = 0; 1016 1017 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL) 1018 { 1019 /* Pretend it was a ":split fname" command */ 1020 eap->cmdidx = CMD_split; 1021 eap->arg = tmp_new; 1022 do_exedit(eap, old_curwin); 1023 1024 /* check that split worked and editing tmp_new */ 1025 if (curwin != old_curwin && win_valid(old_curwin)) 1026 { 1027 /* Set 'diff', 'scrollbind' on and 'wrap' off. */ 1028 diff_win_options(curwin, TRUE); 1029 diff_win_options(old_curwin, TRUE); 1030 1031 if (newname != NULL) 1032 { 1033 /* do a ":file filename.new" on the patched buffer */ 1034 eap->arg = newname; 1035 ex_file(eap); 1036 1037 #ifdef FEAT_AUTOCMD 1038 /* Do filetype detection with the new name. */ 1039 if (au_has_group((char_u *)"filetypedetect")) 1040 do_cmdline_cmd((char_u *)":doau filetypedetect BufRead"); 1041 #endif 1042 } 1043 } 1044 } 1045 } 1046 1047 theend: 1048 if (tmp_orig != NULL) 1049 mch_remove(tmp_orig); 1050 vim_free(tmp_orig); 1051 if (tmp_new != NULL) 1052 mch_remove(tmp_new); 1053 vim_free(tmp_new); 1054 vim_free(newname); 1055 vim_free(buf); 1056 #ifdef UNIX 1057 vim_free(fullname); 1058 #endif 1059 #ifdef FEAT_BROWSE 1060 vim_free(browseFile); 1061 cmdmod.browse = browse_flag; 1062 #endif 1063 } 1064 1065 /* 1066 * Split the window and edit another file, setting options to show the diffs. 1067 */ 1068 void 1069 ex_diffsplit(exarg_T *eap) 1070 { 1071 win_T *old_curwin = curwin; 1072 bufref_T old_curbuf; 1073 1074 set_bufref(&old_curbuf, curbuf); 1075 #ifdef FEAT_GUI 1076 need_mouse_correct = TRUE; 1077 #endif 1078 /* Need to compute w_fraction when no redraw happened yet. */ 1079 validate_cursor(); 1080 set_fraction(curwin); 1081 1082 /* don't use a new tab page, each tab page has its own diffs */ 1083 cmdmod.tab = 0; 1084 1085 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL) 1086 { 1087 /* Pretend it was a ":split fname" command */ 1088 eap->cmdidx = CMD_split; 1089 curwin->w_p_diff = TRUE; 1090 do_exedit(eap, old_curwin); 1091 1092 if (curwin != old_curwin) /* split must have worked */ 1093 { 1094 /* Set 'diff', 'scrollbind' on and 'wrap' off. */ 1095 diff_win_options(curwin, TRUE); 1096 if (win_valid(old_curwin)) 1097 { 1098 diff_win_options(old_curwin, TRUE); 1099 1100 if (bufref_valid(&old_curbuf)) 1101 /* Move the cursor position to that of the old window. */ 1102 curwin->w_cursor.lnum = diff_get_corresponding_line( 1103 old_curbuf.br_buf, old_curwin->w_cursor.lnum); 1104 } 1105 /* Now that lines are folded scroll to show the cursor at the same 1106 * relative position. */ 1107 scroll_to_fraction(curwin, curwin->w_height); 1108 } 1109 } 1110 } 1111 1112 /* 1113 * Set options to show diffs for the current window. 1114 */ 1115 void 1116 ex_diffthis(exarg_T *eap UNUSED) 1117 { 1118 /* Set 'diff', 'scrollbind' on and 'wrap' off. */ 1119 diff_win_options(curwin, TRUE); 1120 } 1121 1122 /* 1123 * Set options in window "wp" for diff mode. 1124 */ 1125 void 1126 diff_win_options( 1127 win_T *wp, 1128 int addbuf) /* Add buffer to diff. */ 1129 { 1130 # ifdef FEAT_FOLDING 1131 win_T *old_curwin = curwin; 1132 1133 /* close the manually opened folds */ 1134 curwin = wp; 1135 newFoldLevel(); 1136 curwin = old_curwin; 1137 # endif 1138 1139 /* Use 'scrollbind' and 'cursorbind' when available */ 1140 #ifdef FEAT_SCROLLBIND 1141 if (!wp->w_p_diff) 1142 wp->w_p_scb_save = wp->w_p_scb; 1143 wp->w_p_scb = TRUE; 1144 #endif 1145 #ifdef FEAT_CURSORBIND 1146 if (!wp->w_p_diff) 1147 wp->w_p_crb_save = wp->w_p_crb; 1148 wp->w_p_crb = TRUE; 1149 #endif 1150 if (!wp->w_p_diff) 1151 wp->w_p_wrap_save = wp->w_p_wrap; 1152 wp->w_p_wrap = FALSE; 1153 # ifdef FEAT_FOLDING 1154 curwin = wp; 1155 curbuf = curwin->w_buffer; 1156 if (!wp->w_p_diff) 1157 { 1158 if (wp->w_p_diff_saved) 1159 free_string_option(wp->w_p_fdm_save); 1160 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm); 1161 } 1162 set_string_option_direct((char_u *)"fdm", -1, (char_u *)"diff", 1163 OPT_LOCAL|OPT_FREE, 0); 1164 curwin = old_curwin; 1165 curbuf = curwin->w_buffer; 1166 if (!wp->w_p_diff) 1167 { 1168 wp->w_p_fdc_save = wp->w_p_fdc; 1169 wp->w_p_fen_save = wp->w_p_fen; 1170 wp->w_p_fdl_save = wp->w_p_fdl; 1171 } 1172 wp->w_p_fdc = diff_foldcolumn; 1173 wp->w_p_fen = TRUE; 1174 wp->w_p_fdl = 0; 1175 foldUpdateAll(wp); 1176 /* make sure topline is not halfway a fold */ 1177 changed_window_setting_win(wp); 1178 # endif 1179 #ifdef FEAT_SCROLLBIND 1180 if (vim_strchr(p_sbo, 'h') == NULL) 1181 do_cmdline_cmd((char_u *)"set sbo+=hor"); 1182 #endif 1183 /* Saved the current values, to be restored in ex_diffoff(). */ 1184 wp->w_p_diff_saved = TRUE; 1185 1186 wp->w_p_diff = TRUE; 1187 1188 if (addbuf) 1189 diff_buf_add(wp->w_buffer); 1190 redraw_win_later(wp, NOT_VALID); 1191 } 1192 1193 /* 1194 * Set options not to show diffs. For the current window or all windows. 1195 * Only in the current tab page. 1196 */ 1197 void 1198 ex_diffoff(exarg_T *eap) 1199 { 1200 win_T *wp; 1201 #ifdef FEAT_SCROLLBIND 1202 int diffwin = FALSE; 1203 #endif 1204 1205 FOR_ALL_WINDOWS(wp) 1206 { 1207 if (eap->forceit ? wp->w_p_diff : wp == curwin) 1208 { 1209 /* Set 'diff' off. If option values were saved in 1210 * diff_win_options(), restore the ones whose settings seem to have 1211 * been left over from diff mode. */ 1212 wp->w_p_diff = FALSE; 1213 1214 if (wp->w_p_diff_saved) 1215 { 1216 1217 #ifdef FEAT_SCROLLBIND 1218 if (wp->w_p_scb) 1219 wp->w_p_scb = wp->w_p_scb_save; 1220 #endif 1221 #ifdef FEAT_CURSORBIND 1222 if (wp->w_p_crb) 1223 wp->w_p_crb = wp->w_p_crb_save; 1224 #endif 1225 if (!wp->w_p_wrap) 1226 wp->w_p_wrap = wp->w_p_wrap_save; 1227 #ifdef FEAT_FOLDING 1228 free_string_option(wp->w_p_fdm); 1229 wp->w_p_fdm = vim_strsave(wp->w_p_fdm_save); 1230 1231 if (wp->w_p_fdc == diff_foldcolumn) 1232 wp->w_p_fdc = wp->w_p_fdc_save; 1233 if (wp->w_p_fdl == 0) 1234 wp->w_p_fdl = wp->w_p_fdl_save; 1235 1236 /* Only restore 'foldenable' when 'foldmethod' is not 1237 * "manual", otherwise we continue to show the diff folds. */ 1238 if (wp->w_p_fen) 1239 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE 1240 : wp->w_p_fen_save; 1241 1242 foldUpdateAll(wp); 1243 #endif 1244 } 1245 /* remove filler lines */ 1246 wp->w_topfill = 0; 1247 1248 /* make sure topline is not halfway a fold and cursor is 1249 * invalidated */ 1250 changed_window_setting_win(wp); 1251 1252 /* Note: 'sbo' is not restored, it's a global option. */ 1253 diff_buf_adjust(wp); 1254 } 1255 #ifdef FEAT_SCROLLBIND 1256 diffwin |= wp->w_p_diff; 1257 #endif 1258 } 1259 1260 #ifdef FEAT_SCROLLBIND 1261 /* Remove "hor" from from 'scrollopt' if there are no diff windows left. */ 1262 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL) 1263 do_cmdline_cmd((char_u *)"set sbo-=hor"); 1264 #endif 1265 } 1266 1267 /* 1268 * Read the diff output and add each entry to the diff list. 1269 */ 1270 static void 1271 diff_read( 1272 int idx_orig, /* idx of original file */ 1273 int idx_new, /* idx of new file */ 1274 char_u *fname) /* name of diff output file */ 1275 { 1276 FILE *fd; 1277 diff_T *dprev = NULL; 1278 diff_T *dp = curtab->tp_first_diff; 1279 diff_T *dn, *dpl; 1280 long f1, l1, f2, l2; 1281 char_u linebuf[LBUFLEN]; /* only need to hold the diff line */ 1282 int difftype; 1283 char_u *p; 1284 long off; 1285 int i; 1286 linenr_T lnum_orig, lnum_new; 1287 long count_orig, count_new; 1288 int notset = TRUE; /* block "*dp" not set yet */ 1289 1290 fd = mch_fopen((char *)fname, "r"); 1291 if (fd == NULL) 1292 { 1293 EMSG(_("E98: Cannot read diff output")); 1294 return; 1295 } 1296 1297 for (;;) 1298 { 1299 if (tag_fgets(linebuf, LBUFLEN, fd)) 1300 break; /* end of file */ 1301 if (!isdigit(*linebuf)) 1302 continue; /* not the start of a diff block */ 1303 1304 /* This line must be one of three formats: 1305 * {first}[,{last}]c{first}[,{last}] 1306 * {first}a{first}[,{last}] 1307 * {first}[,{last}]d{first} 1308 */ 1309 p = linebuf; 1310 f1 = getdigits(&p); 1311 if (*p == ',') 1312 { 1313 ++p; 1314 l1 = getdigits(&p); 1315 } 1316 else 1317 l1 = f1; 1318 if (*p != 'a' && *p != 'c' && *p != 'd') 1319 continue; /* invalid diff format */ 1320 difftype = *p++; 1321 f2 = getdigits(&p); 1322 if (*p == ',') 1323 { 1324 ++p; 1325 l2 = getdigits(&p); 1326 } 1327 else 1328 l2 = f2; 1329 if (l1 < f1 || l2 < f2) 1330 continue; /* invalid line range */ 1331 1332 if (difftype == 'a') 1333 { 1334 lnum_orig = f1 + 1; 1335 count_orig = 0; 1336 } 1337 else 1338 { 1339 lnum_orig = f1; 1340 count_orig = l1 - f1 + 1; 1341 } 1342 if (difftype == 'd') 1343 { 1344 lnum_new = f2 + 1; 1345 count_new = 0; 1346 } 1347 else 1348 { 1349 lnum_new = f2; 1350 count_new = l2 - f2 + 1; 1351 } 1352 1353 /* Go over blocks before the change, for which orig and new are equal. 1354 * Copy blocks from orig to new. */ 1355 while (dp != NULL 1356 && lnum_orig > dp->df_lnum[idx_orig] + dp->df_count[idx_orig]) 1357 { 1358 if (notset) 1359 diff_copy_entry(dprev, dp, idx_orig, idx_new); 1360 dprev = dp; 1361 dp = dp->df_next; 1362 notset = TRUE; 1363 } 1364 1365 if (dp != NULL 1366 && lnum_orig <= dp->df_lnum[idx_orig] + dp->df_count[idx_orig] 1367 && lnum_orig + count_orig >= dp->df_lnum[idx_orig]) 1368 { 1369 /* New block overlaps with existing block(s). 1370 * First find last block that overlaps. */ 1371 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next) 1372 if (lnum_orig + count_orig < dpl->df_next->df_lnum[idx_orig]) 1373 break; 1374 1375 /* If the newly found block starts before the old one, set the 1376 * start back a number of lines. */ 1377 off = dp->df_lnum[idx_orig] - lnum_orig; 1378 if (off > 0) 1379 { 1380 for (i = idx_orig; i < idx_new; ++i) 1381 if (curtab->tp_diffbuf[i] != NULL) 1382 dp->df_lnum[i] -= off; 1383 dp->df_lnum[idx_new] = lnum_new; 1384 dp->df_count[idx_new] = count_new; 1385 } 1386 else if (notset) 1387 { 1388 /* new block inside existing one, adjust new block */ 1389 dp->df_lnum[idx_new] = lnum_new + off; 1390 dp->df_count[idx_new] = count_new - off; 1391 } 1392 else 1393 /* second overlap of new block with existing block */ 1394 dp->df_count[idx_new] += count_new - count_orig 1395 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig] 1396 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]); 1397 1398 /* Adjust the size of the block to include all the lines to the 1399 * end of the existing block or the new diff, whatever ends last. */ 1400 off = (lnum_orig + count_orig) 1401 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]); 1402 if (off < 0) 1403 { 1404 /* new change ends in existing block, adjust the end if not 1405 * done already */ 1406 if (notset) 1407 dp->df_count[idx_new] += -off; 1408 off = 0; 1409 } 1410 for (i = idx_orig; i < idx_new; ++i) 1411 if (curtab->tp_diffbuf[i] != NULL) 1412 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i] 1413 - dp->df_lnum[i] + off; 1414 1415 /* Delete the diff blocks that have been merged into one. */ 1416 dn = dp->df_next; 1417 dp->df_next = dpl->df_next; 1418 while (dn != dp->df_next) 1419 { 1420 dpl = dn->df_next; 1421 vim_free(dn); 1422 dn = dpl; 1423 } 1424 } 1425 else 1426 { 1427 /* Allocate a new diffblock. */ 1428 dp = diff_alloc_new(curtab, dprev, dp); 1429 if (dp == NULL) 1430 goto done; 1431 1432 dp->df_lnum[idx_orig] = lnum_orig; 1433 dp->df_count[idx_orig] = count_orig; 1434 dp->df_lnum[idx_new] = lnum_new; 1435 dp->df_count[idx_new] = count_new; 1436 1437 /* Set values for other buffers, these must be equal to the 1438 * original buffer, otherwise there would have been a change 1439 * already. */ 1440 for (i = idx_orig + 1; i < idx_new; ++i) 1441 if (curtab->tp_diffbuf[i] != NULL) 1442 diff_copy_entry(dprev, dp, idx_orig, i); 1443 } 1444 notset = FALSE; /* "*dp" has been set */ 1445 } 1446 1447 /* for remaining diff blocks orig and new are equal */ 1448 while (dp != NULL) 1449 { 1450 if (notset) 1451 diff_copy_entry(dprev, dp, idx_orig, idx_new); 1452 dprev = dp; 1453 dp = dp->df_next; 1454 notset = TRUE; 1455 } 1456 1457 done: 1458 fclose(fd); 1459 } 1460 1461 /* 1462 * Copy an entry at "dp" from "idx_orig" to "idx_new". 1463 */ 1464 static void 1465 diff_copy_entry( 1466 diff_T *dprev, 1467 diff_T *dp, 1468 int idx_orig, 1469 int idx_new) 1470 { 1471 long off; 1472 1473 if (dprev == NULL) 1474 off = 0; 1475 else 1476 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig]) 1477 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]); 1478 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off; 1479 dp->df_count[idx_new] = dp->df_count[idx_orig]; 1480 } 1481 1482 /* 1483 * Clear the list of diffblocks for tab page "tp". 1484 */ 1485 void 1486 diff_clear(tabpage_T *tp) 1487 { 1488 diff_T *p, *next_p; 1489 1490 for (p = tp->tp_first_diff; p != NULL; p = next_p) 1491 { 1492 next_p = p->df_next; 1493 vim_free(p); 1494 } 1495 tp->tp_first_diff = NULL; 1496 } 1497 1498 /* 1499 * Check diff status for line "lnum" in buffer "buf": 1500 * Returns 0 for nothing special 1501 * Returns -1 for a line that should be highlighted as changed. 1502 * Returns -2 for a line that should be highlighted as added/deleted. 1503 * Returns > 0 for inserting that many filler lines above it (never happens 1504 * when 'diffopt' doesn't contain "filler"). 1505 * This should only be used for windows where 'diff' is set. 1506 */ 1507 int 1508 diff_check(win_T *wp, linenr_T lnum) 1509 { 1510 int idx; /* index in tp_diffbuf[] for this buffer */ 1511 diff_T *dp; 1512 int maxcount; 1513 int i; 1514 buf_T *buf = wp->w_buffer; 1515 int cmp; 1516 1517 if (curtab->tp_diff_invalid) 1518 ex_diffupdate(NULL); /* update after a big change */ 1519 1520 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) /* no diffs at all */ 1521 return 0; 1522 1523 /* safety check: "lnum" must be a buffer line */ 1524 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1) 1525 return 0; 1526 1527 idx = diff_buf_idx(buf); 1528 if (idx == DB_COUNT) 1529 return 0; /* no diffs for buffer "buf" */ 1530 1531 #ifdef FEAT_FOLDING 1532 /* A closed fold never has filler lines. */ 1533 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL)) 1534 return 0; 1535 #endif 1536 1537 /* search for a change that includes "lnum" in the list of diffblocks. */ 1538 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 1539 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) 1540 break; 1541 if (dp == NULL || lnum < dp->df_lnum[idx]) 1542 return 0; 1543 1544 if (lnum < dp->df_lnum[idx] + dp->df_count[idx]) 1545 { 1546 int zero = FALSE; 1547 1548 /* Changed or inserted line. If the other buffers have a count of 1549 * zero, the lines were inserted. If the other buffers have the same 1550 * count, check if the lines are identical. */ 1551 cmp = FALSE; 1552 for (i = 0; i < DB_COUNT; ++i) 1553 if (i != idx && curtab->tp_diffbuf[i] != NULL) 1554 { 1555 if (dp->df_count[i] == 0) 1556 zero = TRUE; 1557 else 1558 { 1559 if (dp->df_count[i] != dp->df_count[idx]) 1560 return -1; /* nr of lines changed. */ 1561 cmp = TRUE; 1562 } 1563 } 1564 if (cmp) 1565 { 1566 /* Compare all lines. If they are equal the lines were inserted 1567 * in some buffers, deleted in others, but not changed. */ 1568 for (i = 0; i < DB_COUNT; ++i) 1569 if (i != idx && curtab->tp_diffbuf[i] != NULL 1570 && dp->df_count[i] != 0) 1571 if (!diff_equal_entry(dp, idx, i)) 1572 return -1; 1573 } 1574 /* If there is no buffer with zero lines then there is no difference 1575 * any longer. Happens when making a change (or undo) that removes 1576 * the difference. Can't remove the entry here, we might be halfway 1577 * updating the window. Just report the text as unchanged. Other 1578 * windows might still show the change though. */ 1579 if (zero == FALSE) 1580 return 0; 1581 return -2; 1582 } 1583 1584 /* If 'diffopt' doesn't contain "filler", return 0. */ 1585 if (!(diff_flags & DIFF_FILLER)) 1586 return 0; 1587 1588 /* Insert filler lines above the line just below the change. Will return 1589 * 0 when this buf had the max count. */ 1590 maxcount = 0; 1591 for (i = 0; i < DB_COUNT; ++i) 1592 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount) 1593 maxcount = dp->df_count[i]; 1594 return maxcount - dp->df_count[idx]; 1595 } 1596 1597 /* 1598 * Compare two entries in diff "*dp" and return TRUE if they are equal. 1599 */ 1600 static int 1601 diff_equal_entry(diff_T *dp, int idx1, int idx2) 1602 { 1603 int i; 1604 char_u *line; 1605 int cmp; 1606 1607 if (dp->df_count[idx1] != dp->df_count[idx2]) 1608 return FALSE; 1609 if (diff_check_sanity(curtab, dp) == FAIL) 1610 return FALSE; 1611 for (i = 0; i < dp->df_count[idx1]; ++i) 1612 { 1613 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1], 1614 dp->df_lnum[idx1] + i, FALSE)); 1615 if (line == NULL) 1616 return FALSE; 1617 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2], 1618 dp->df_lnum[idx2] + i, FALSE)); 1619 vim_free(line); 1620 if (cmp != 0) 1621 return FALSE; 1622 } 1623 return TRUE; 1624 } 1625 1626 /* 1627 * Compare strings "s1" and "s2" according to 'diffopt'. 1628 * Return non-zero when they are different. 1629 */ 1630 static int 1631 diff_cmp(char_u *s1, char_u *s2) 1632 { 1633 char_u *p1, *p2; 1634 #ifdef FEAT_MBYTE 1635 int l; 1636 #endif 1637 1638 if ((diff_flags & (DIFF_ICASE | DIFF_IWHITE)) == 0) 1639 return STRCMP(s1, s2); 1640 if ((diff_flags & DIFF_ICASE) && !(diff_flags & DIFF_IWHITE)) 1641 return MB_STRICMP(s1, s2); 1642 1643 /* Ignore white space changes and possibly ignore case. */ 1644 p1 = s1; 1645 p2 = s2; 1646 while (*p1 != NUL && *p2 != NUL) 1647 { 1648 if (vim_iswhite(*p1) && vim_iswhite(*p2)) 1649 { 1650 p1 = skipwhite(p1); 1651 p2 = skipwhite(p2); 1652 } 1653 else 1654 { 1655 #ifdef FEAT_MBYTE 1656 l = (*mb_ptr2len)(p1); 1657 if (l != (*mb_ptr2len)(p2)) 1658 break; 1659 if (l > 1) 1660 { 1661 if (STRNCMP(p1, p2, l) != 0 1662 && (!enc_utf8 1663 || !(diff_flags & DIFF_ICASE) 1664 || utf_fold(utf_ptr2char(p1)) 1665 != utf_fold(utf_ptr2char(p2)))) 1666 break; 1667 p1 += l; 1668 p2 += l; 1669 } 1670 else 1671 #endif 1672 { 1673 if (*p1 != *p2 && (!(diff_flags & DIFF_ICASE) 1674 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2))) 1675 break; 1676 ++p1; 1677 ++p2; 1678 } 1679 } 1680 } 1681 1682 /* Ignore trailing white space. */ 1683 p1 = skipwhite(p1); 1684 p2 = skipwhite(p2); 1685 if (*p1 != NUL || *p2 != NUL) 1686 return 1; 1687 return 0; 1688 } 1689 1690 /* 1691 * Return the number of filler lines above "lnum". 1692 */ 1693 int 1694 diff_check_fill(win_T *wp, linenr_T lnum) 1695 { 1696 int n; 1697 1698 /* be quick when there are no filler lines */ 1699 if (!(diff_flags & DIFF_FILLER)) 1700 return 0; 1701 n = diff_check(wp, lnum); 1702 if (n <= 0) 1703 return 0; 1704 return n; 1705 } 1706 1707 /* 1708 * Set the topline of "towin" to match the position in "fromwin", so that they 1709 * show the same diff'ed lines. 1710 */ 1711 void 1712 diff_set_topline(win_T *fromwin, win_T *towin) 1713 { 1714 buf_T *frombuf = fromwin->w_buffer; 1715 linenr_T lnum = fromwin->w_topline; 1716 int fromidx; 1717 int toidx; 1718 diff_T *dp; 1719 int max_count; 1720 int i; 1721 1722 fromidx = diff_buf_idx(frombuf); 1723 if (fromidx == DB_COUNT) 1724 return; /* safety check */ 1725 1726 if (curtab->tp_diff_invalid) 1727 ex_diffupdate(NULL); /* update after a big change */ 1728 1729 towin->w_topfill = 0; 1730 1731 /* search for a change that includes "lnum" in the list of diffblocks. */ 1732 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 1733 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx]) 1734 break; 1735 if (dp == NULL) 1736 { 1737 /* After last change, compute topline relative to end of file; no 1738 * filler lines. */ 1739 towin->w_topline = towin->w_buffer->b_ml.ml_line_count 1740 - (frombuf->b_ml.ml_line_count - lnum); 1741 } 1742 else 1743 { 1744 /* Find index for "towin". */ 1745 toidx = diff_buf_idx(towin->w_buffer); 1746 if (toidx == DB_COUNT) 1747 return; /* safety check */ 1748 1749 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]); 1750 if (lnum >= dp->df_lnum[fromidx]) 1751 { 1752 /* Inside a change: compute filler lines. With three or more 1753 * buffers we need to know the largest count. */ 1754 max_count = 0; 1755 for (i = 0; i < DB_COUNT; ++i) 1756 if (curtab->tp_diffbuf[i] != NULL 1757 && max_count < dp->df_count[i]) 1758 max_count = dp->df_count[i]; 1759 1760 if (dp->df_count[toidx] == dp->df_count[fromidx]) 1761 { 1762 /* same number of lines: use same filler count */ 1763 towin->w_topfill = fromwin->w_topfill; 1764 } 1765 else if (dp->df_count[toidx] > dp->df_count[fromidx]) 1766 { 1767 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx]) 1768 { 1769 /* more lines in towin and fromwin doesn't show diff 1770 * lines, only filler lines */ 1771 if (max_count - fromwin->w_topfill >= dp->df_count[toidx]) 1772 { 1773 /* towin also only shows filler lines */ 1774 towin->w_topline = dp->df_lnum[toidx] 1775 + dp->df_count[toidx]; 1776 towin->w_topfill = fromwin->w_topfill; 1777 } 1778 else 1779 /* towin still has some diff lines to show */ 1780 towin->w_topline = dp->df_lnum[toidx] 1781 + max_count - fromwin->w_topfill; 1782 } 1783 } 1784 else if (towin->w_topline >= dp->df_lnum[toidx] 1785 + dp->df_count[toidx]) 1786 { 1787 /* less lines in towin and no diff lines to show: compute 1788 * filler lines */ 1789 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx]; 1790 if (diff_flags & DIFF_FILLER) 1791 { 1792 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx]) 1793 /* fromwin is also out of diff lines */ 1794 towin->w_topfill = fromwin->w_topfill; 1795 else 1796 /* fromwin has some diff lines */ 1797 towin->w_topfill = dp->df_lnum[fromidx] 1798 + max_count - lnum; 1799 } 1800 } 1801 } 1802 } 1803 1804 /* safety check (if diff info gets outdated strange things may happen) */ 1805 towin->w_botfill = FALSE; 1806 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count) 1807 { 1808 towin->w_topline = towin->w_buffer->b_ml.ml_line_count; 1809 towin->w_botfill = TRUE; 1810 } 1811 if (towin->w_topline < 1) 1812 { 1813 towin->w_topline = 1; 1814 towin->w_topfill = 0; 1815 } 1816 1817 /* When w_topline changes need to recompute w_botline and cursor position */ 1818 invalidate_botline_win(towin); 1819 changed_line_abv_curs_win(towin); 1820 1821 check_topfill(towin, FALSE); 1822 #ifdef FEAT_FOLDING 1823 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline, 1824 NULL, TRUE, NULL); 1825 #endif 1826 } 1827 1828 /* 1829 * This is called when 'diffopt' is changed. 1830 */ 1831 int 1832 diffopt_changed(void) 1833 { 1834 char_u *p; 1835 int diff_context_new = 6; 1836 int diff_flags_new = 0; 1837 int diff_foldcolumn_new = 2; 1838 tabpage_T *tp; 1839 1840 p = p_dip; 1841 while (*p != NUL) 1842 { 1843 if (STRNCMP(p, "filler", 6) == 0) 1844 { 1845 p += 6; 1846 diff_flags_new |= DIFF_FILLER; 1847 } 1848 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8])) 1849 { 1850 p += 8; 1851 diff_context_new = getdigits(&p); 1852 } 1853 else if (STRNCMP(p, "icase", 5) == 0) 1854 { 1855 p += 5; 1856 diff_flags_new |= DIFF_ICASE; 1857 } 1858 else if (STRNCMP(p, "iwhite", 6) == 0) 1859 { 1860 p += 6; 1861 diff_flags_new |= DIFF_IWHITE; 1862 } 1863 else if (STRNCMP(p, "horizontal", 10) == 0) 1864 { 1865 p += 10; 1866 diff_flags_new |= DIFF_HORIZONTAL; 1867 } 1868 else if (STRNCMP(p, "vertical", 8) == 0) 1869 { 1870 p += 8; 1871 diff_flags_new |= DIFF_VERTICAL; 1872 } 1873 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11])) 1874 { 1875 p += 11; 1876 diff_foldcolumn_new = getdigits(&p); 1877 } 1878 if (*p != ',' && *p != NUL) 1879 return FAIL; 1880 if (*p == ',') 1881 ++p; 1882 } 1883 1884 /* Can't have both "horizontal" and "vertical". */ 1885 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL)) 1886 return FAIL; 1887 1888 /* If "icase" or "iwhite" was added or removed, need to update the diff. */ 1889 if (diff_flags != diff_flags_new) 1890 FOR_ALL_TABPAGES(tp) 1891 tp->tp_diff_invalid = TRUE; 1892 1893 diff_flags = diff_flags_new; 1894 diff_context = diff_context_new; 1895 diff_foldcolumn = diff_foldcolumn_new; 1896 1897 diff_redraw(TRUE); 1898 1899 /* recompute the scroll binding with the new option value, may 1900 * remove or add filler lines */ 1901 check_scrollbind((linenr_T)0, 0L); 1902 1903 return OK; 1904 } 1905 1906 /* 1907 * Return TRUE if 'diffopt' contains "horizontal". 1908 */ 1909 int 1910 diffopt_horizontal(void) 1911 { 1912 return (diff_flags & DIFF_HORIZONTAL) != 0; 1913 } 1914 1915 /* 1916 * Find the difference within a changed line. 1917 * Returns TRUE if the line was added, no other buffer has it. 1918 */ 1919 int 1920 diff_find_change( 1921 win_T *wp, 1922 linenr_T lnum, 1923 int *startp, /* first char of the change */ 1924 int *endp) /* last char of the change */ 1925 { 1926 char_u *line_org; 1927 char_u *line_new; 1928 int i; 1929 int si_org, si_new; 1930 int ei_org, ei_new; 1931 diff_T *dp; 1932 int idx; 1933 int off; 1934 int added = TRUE; 1935 1936 /* Make a copy of the line, the next ml_get() will invalidate it. */ 1937 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE)); 1938 if (line_org == NULL) 1939 return FALSE; 1940 1941 idx = diff_buf_idx(wp->w_buffer); 1942 if (idx == DB_COUNT) /* cannot happen */ 1943 { 1944 vim_free(line_org); 1945 return FALSE; 1946 } 1947 1948 /* search for a change that includes "lnum" in the list of diffblocks. */ 1949 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 1950 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) 1951 break; 1952 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL) 1953 { 1954 vim_free(line_org); 1955 return FALSE; 1956 } 1957 1958 off = lnum - dp->df_lnum[idx]; 1959 1960 for (i = 0; i < DB_COUNT; ++i) 1961 if (curtab->tp_diffbuf[i] != NULL && i != idx) 1962 { 1963 /* Skip lines that are not in the other change (filler lines). */ 1964 if (off >= dp->df_count[i]) 1965 continue; 1966 added = FALSE; 1967 line_new = ml_get_buf(curtab->tp_diffbuf[i], 1968 dp->df_lnum[i] + off, FALSE); 1969 1970 /* Search for start of difference */ 1971 si_org = si_new = 0; 1972 while (line_org[si_org] != NUL) 1973 { 1974 if ((diff_flags & DIFF_IWHITE) 1975 && vim_iswhite(line_org[si_org]) 1976 && vim_iswhite(line_new[si_new])) 1977 { 1978 si_org = (int)(skipwhite(line_org + si_org) - line_org); 1979 si_new = (int)(skipwhite(line_new + si_new) - line_new); 1980 } 1981 else 1982 { 1983 if (line_org[si_org] != line_new[si_new]) 1984 break; 1985 ++si_org; 1986 ++si_new; 1987 } 1988 } 1989 #ifdef FEAT_MBYTE 1990 if (has_mbyte) 1991 { 1992 /* Move back to first byte of character in both lines (may 1993 * have "nn^" in line_org and "n^ in line_new). */ 1994 si_org -= (*mb_head_off)(line_org, line_org + si_org); 1995 si_new -= (*mb_head_off)(line_new, line_new + si_new); 1996 } 1997 #endif 1998 if (*startp > si_org) 1999 *startp = si_org; 2000 2001 /* Search for end of difference, if any. */ 2002 if (line_org[si_org] != NUL || line_new[si_new] != NUL) 2003 { 2004 ei_org = (int)STRLEN(line_org); 2005 ei_new = (int)STRLEN(line_new); 2006 while (ei_org >= *startp && ei_new >= si_new 2007 && ei_org >= 0 && ei_new >= 0) 2008 { 2009 if ((diff_flags & DIFF_IWHITE) 2010 && vim_iswhite(line_org[ei_org]) 2011 && vim_iswhite(line_new[ei_new])) 2012 { 2013 while (ei_org >= *startp 2014 && vim_iswhite(line_org[ei_org])) 2015 --ei_org; 2016 while (ei_new >= si_new 2017 && vim_iswhite(line_new[ei_new])) 2018 --ei_new; 2019 } 2020 else 2021 { 2022 if (line_org[ei_org] != line_new[ei_new]) 2023 break; 2024 --ei_org; 2025 --ei_new; 2026 } 2027 } 2028 if (*endp < ei_org) 2029 *endp = ei_org; 2030 } 2031 } 2032 2033 vim_free(line_org); 2034 return added; 2035 } 2036 2037 #if defined(FEAT_FOLDING) || defined(PROTO) 2038 /* 2039 * Return TRUE if line "lnum" is not close to a diff block, this line should 2040 * be in a fold. 2041 * Return FALSE if there are no diff blocks at all in this window. 2042 */ 2043 int 2044 diff_infold(win_T *wp, linenr_T lnum) 2045 { 2046 int i; 2047 int idx = -1; 2048 int other = FALSE; 2049 diff_T *dp; 2050 2051 /* Return if 'diff' isn't set. */ 2052 if (!wp->w_p_diff) 2053 return FALSE; 2054 2055 for (i = 0; i < DB_COUNT; ++i) 2056 { 2057 if (curtab->tp_diffbuf[i] == wp->w_buffer) 2058 idx = i; 2059 else if (curtab->tp_diffbuf[i] != NULL) 2060 other = TRUE; 2061 } 2062 2063 /* return here if there are no diffs in the window */ 2064 if (idx == -1 || !other) 2065 return FALSE; 2066 2067 if (curtab->tp_diff_invalid) 2068 ex_diffupdate(NULL); /* update after a big change */ 2069 2070 /* Return if there are no diff blocks. All lines will be folded. */ 2071 if (curtab->tp_first_diff == NULL) 2072 return TRUE; 2073 2074 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 2075 { 2076 /* If this change is below the line there can't be any further match. */ 2077 if (dp->df_lnum[idx] - diff_context > lnum) 2078 break; 2079 /* If this change ends before the line we have a match. */ 2080 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum) 2081 return FALSE; 2082 } 2083 return TRUE; 2084 } 2085 #endif 2086 2087 /* 2088 * "dp" and "do" commands. 2089 */ 2090 void 2091 nv_diffgetput(int put, long count) 2092 { 2093 exarg_T ea; 2094 char_u buf[30]; 2095 2096 if (count == 0) 2097 ea.arg = (char_u *)""; 2098 else 2099 { 2100 vim_snprintf((char *)buf, 30, "%ld", count); 2101 ea.arg = buf; 2102 } 2103 if (put) 2104 ea.cmdidx = CMD_diffput; 2105 else 2106 ea.cmdidx = CMD_diffget; 2107 ea.addr_count = 0; 2108 ea.line1 = curwin->w_cursor.lnum; 2109 ea.line2 = curwin->w_cursor.lnum; 2110 ex_diffgetput(&ea); 2111 } 2112 2113 /* 2114 * ":diffget" 2115 * ":diffput" 2116 */ 2117 void 2118 ex_diffgetput(exarg_T *eap) 2119 { 2120 linenr_T lnum; 2121 int count; 2122 linenr_T off = 0; 2123 diff_T *dp; 2124 diff_T *dprev; 2125 diff_T *dfree; 2126 int idx_cur; 2127 int idx_other; 2128 int idx_from; 2129 int idx_to; 2130 int i; 2131 int added; 2132 char_u *p; 2133 aco_save_T aco; 2134 buf_T *buf; 2135 int start_skip, end_skip; 2136 int new_count; 2137 int buf_empty; 2138 int found_not_ma = FALSE; 2139 2140 /* Find the current buffer in the list of diff buffers. */ 2141 idx_cur = diff_buf_idx(curbuf); 2142 if (idx_cur == DB_COUNT) 2143 { 2144 EMSG(_("E99: Current buffer is not in diff mode")); 2145 return; 2146 } 2147 2148 if (*eap->arg == NUL) 2149 { 2150 /* No argument: Find the other buffer in the list of diff buffers. */ 2151 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other) 2152 if (curtab->tp_diffbuf[idx_other] != curbuf 2153 && curtab->tp_diffbuf[idx_other] != NULL) 2154 { 2155 if (eap->cmdidx != CMD_diffput 2156 || curtab->tp_diffbuf[idx_other]->b_p_ma) 2157 break; 2158 found_not_ma = TRUE; 2159 } 2160 if (idx_other == DB_COUNT) 2161 { 2162 if (found_not_ma) 2163 EMSG(_("E793: No other buffer in diff mode is modifiable")); 2164 else 2165 EMSG(_("E100: No other buffer in diff mode")); 2166 return; 2167 } 2168 2169 /* Check that there isn't a third buffer in the list */ 2170 for (i = idx_other + 1; i < DB_COUNT; ++i) 2171 if (curtab->tp_diffbuf[i] != curbuf 2172 && curtab->tp_diffbuf[i] != NULL 2173 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma)) 2174 { 2175 EMSG(_("E101: More than two buffers in diff mode, don't know which one to use")); 2176 return; 2177 } 2178 } 2179 else 2180 { 2181 /* Buffer number or pattern given. Ignore trailing white space. */ 2182 p = eap->arg + STRLEN(eap->arg); 2183 while (p > eap->arg && vim_iswhite(p[-1])) 2184 --p; 2185 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i) 2186 ; 2187 if (eap->arg + i == p) /* digits only */ 2188 i = atol((char *)eap->arg); 2189 else 2190 { 2191 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE); 2192 if (i < 0) 2193 return; /* error message already given */ 2194 } 2195 buf = buflist_findnr(i); 2196 if (buf == NULL) 2197 { 2198 EMSG2(_("E102: Can't find buffer \"%s\""), eap->arg); 2199 return; 2200 } 2201 if (buf == curbuf) 2202 return; /* nothing to do */ 2203 idx_other = diff_buf_idx(buf); 2204 if (idx_other == DB_COUNT) 2205 { 2206 EMSG2(_("E103: Buffer \"%s\" is not in diff mode"), eap->arg); 2207 return; 2208 } 2209 } 2210 2211 diff_busy = TRUE; 2212 2213 /* When no range given include the line above or below the cursor. */ 2214 if (eap->addr_count == 0) 2215 { 2216 /* Make it possible that ":diffget" on the last line gets line below 2217 * the cursor line when there is no difference above the cursor. */ 2218 if (eap->cmdidx == CMD_diffget 2219 && eap->line1 == curbuf->b_ml.ml_line_count 2220 && diff_check(curwin, eap->line1) == 0 2221 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0)) 2222 ++eap->line2; 2223 else if (eap->line1 > 0) 2224 --eap->line1; 2225 } 2226 2227 if (eap->cmdidx == CMD_diffget) 2228 { 2229 idx_from = idx_other; 2230 idx_to = idx_cur; 2231 } 2232 else 2233 { 2234 idx_from = idx_cur; 2235 idx_to = idx_other; 2236 /* Need to make the other buffer the current buffer to be able to make 2237 * changes in it. */ 2238 /* set curwin/curbuf to buf and save a few things */ 2239 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]); 2240 } 2241 2242 /* May give the warning for a changed buffer here, which can trigger the 2243 * FileChangedRO autocommand, which may do nasty things and mess 2244 * everything up. */ 2245 if (!curbuf->b_changed) 2246 { 2247 change_warning(0); 2248 if (diff_buf_idx(curbuf) != idx_to) 2249 { 2250 EMSG(_("E787: Buffer changed unexpectedly")); 2251 return; 2252 } 2253 } 2254 2255 dprev = NULL; 2256 for (dp = curtab->tp_first_diff; dp != NULL; ) 2257 { 2258 if (dp->df_lnum[idx_cur] > eap->line2 + off) 2259 break; /* past the range that was specified */ 2260 2261 dfree = NULL; 2262 lnum = dp->df_lnum[idx_to]; 2263 count = dp->df_count[idx_to]; 2264 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off 2265 && u_save(lnum - 1, lnum + count) != FAIL) 2266 { 2267 /* Inside the specified range and saving for undo worked. */ 2268 start_skip = 0; 2269 end_skip = 0; 2270 if (eap->addr_count > 0) 2271 { 2272 /* A range was specified: check if lines need to be skipped. */ 2273 start_skip = eap->line1 + off - dp->df_lnum[idx_cur]; 2274 if (start_skip > 0) 2275 { 2276 /* range starts below start of current diff block */ 2277 if (start_skip > count) 2278 { 2279 lnum += count; 2280 count = 0; 2281 } 2282 else 2283 { 2284 count -= start_skip; 2285 lnum += start_skip; 2286 } 2287 } 2288 else 2289 start_skip = 0; 2290 2291 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1 2292 - (eap->line2 + off); 2293 if (end_skip > 0) 2294 { 2295 /* range ends above end of current/from diff block */ 2296 if (idx_cur == idx_from) /* :diffput */ 2297 { 2298 i = dp->df_count[idx_cur] - start_skip - end_skip; 2299 if (count > i) 2300 count = i; 2301 } 2302 else /* :diffget */ 2303 { 2304 count -= end_skip; 2305 end_skip = dp->df_count[idx_from] - start_skip - count; 2306 if (end_skip < 0) 2307 end_skip = 0; 2308 } 2309 } 2310 else 2311 end_skip = 0; 2312 } 2313 2314 buf_empty = bufempty(); 2315 added = 0; 2316 for (i = 0; i < count; ++i) 2317 { 2318 /* remember deleting the last line of the buffer */ 2319 buf_empty = curbuf->b_ml.ml_line_count == 1; 2320 ml_delete(lnum, FALSE); 2321 --added; 2322 } 2323 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i) 2324 { 2325 linenr_T nr; 2326 2327 nr = dp->df_lnum[idx_from] + start_skip + i; 2328 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count) 2329 break; 2330 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from], 2331 nr, FALSE)); 2332 if (p != NULL) 2333 { 2334 ml_append(lnum + i - 1, p, 0, FALSE); 2335 vim_free(p); 2336 ++added; 2337 if (buf_empty && curbuf->b_ml.ml_line_count == 2) 2338 { 2339 /* Added the first line into an empty buffer, need to 2340 * delete the dummy empty line. */ 2341 buf_empty = FALSE; 2342 ml_delete((linenr_T)2, FALSE); 2343 } 2344 } 2345 } 2346 new_count = dp->df_count[idx_to] + added; 2347 dp->df_count[idx_to] = new_count; 2348 2349 if (start_skip == 0 && end_skip == 0) 2350 { 2351 /* Check if there are any other buffers and if the diff is 2352 * equal in them. */ 2353 for (i = 0; i < DB_COUNT; ++i) 2354 if (curtab->tp_diffbuf[i] != NULL && i != idx_from 2355 && i != idx_to 2356 && !diff_equal_entry(dp, idx_from, i)) 2357 break; 2358 if (i == DB_COUNT) 2359 { 2360 /* delete the diff entry, the buffers are now equal here */ 2361 dfree = dp; 2362 dp = dp->df_next; 2363 if (dprev == NULL) 2364 curtab->tp_first_diff = dp; 2365 else 2366 dprev->df_next = dp; 2367 } 2368 } 2369 2370 /* Adjust marks. This will change the following entries! */ 2371 if (added != 0) 2372 { 2373 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added); 2374 if (curwin->w_cursor.lnum >= lnum) 2375 { 2376 /* Adjust the cursor position if it's in/after the changed 2377 * lines. */ 2378 if (curwin->w_cursor.lnum >= lnum + count) 2379 curwin->w_cursor.lnum += added; 2380 else if (added < 0) 2381 curwin->w_cursor.lnum = lnum; 2382 } 2383 } 2384 changed_lines(lnum, 0, lnum + count, (long)added); 2385 2386 if (dfree != NULL) 2387 { 2388 /* Diff is deleted, update folds in other windows. */ 2389 #ifdef FEAT_FOLDING 2390 diff_fold_update(dfree, idx_to); 2391 #endif 2392 vim_free(dfree); 2393 } 2394 else 2395 /* mark_adjust() may have changed the count in a wrong way */ 2396 dp->df_count[idx_to] = new_count; 2397 2398 /* When changing the current buffer, keep track of line numbers */ 2399 if (idx_cur == idx_to) 2400 off += added; 2401 } 2402 2403 /* If before the range or not deleted, go to next diff. */ 2404 if (dfree == NULL) 2405 { 2406 dprev = dp; 2407 dp = dp->df_next; 2408 } 2409 } 2410 2411 /* restore curwin/curbuf and a few other things */ 2412 if (eap->cmdidx != CMD_diffget) 2413 { 2414 /* Syncing undo only works for the current buffer, but we change 2415 * another buffer. Sync undo if the command was typed. This isn't 2416 * 100% right when ":diffput" is used in a function or mapping. */ 2417 if (KeyTyped) 2418 u_sync(FALSE); 2419 aucmd_restbuf(&aco); 2420 } 2421 2422 diff_busy = FALSE; 2423 2424 /* Check that the cursor is on a valid character and update it's position. 2425 * When there were filler lines the topline has become invalid. */ 2426 check_cursor(); 2427 changed_line_abv_curs(); 2428 2429 /* Also need to redraw the other buffers. */ 2430 diff_redraw(FALSE); 2431 } 2432 2433 #ifdef FEAT_FOLDING 2434 /* 2435 * Update folds for all diff buffers for entry "dp". 2436 * Skip buffer with index "skip_idx". 2437 * When there are no diffs, all folds are removed. 2438 */ 2439 static void 2440 diff_fold_update(diff_T *dp, int skip_idx) 2441 { 2442 int i; 2443 win_T *wp; 2444 2445 FOR_ALL_WINDOWS(wp) 2446 for (i = 0; i < DB_COUNT; ++i) 2447 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx) 2448 foldUpdate(wp, dp->df_lnum[i], 2449 dp->df_lnum[i] + dp->df_count[i]); 2450 } 2451 #endif 2452 2453 /* 2454 * Return TRUE if buffer "buf" is in diff-mode. 2455 */ 2456 int 2457 diff_mode_buf(buf_T *buf) 2458 { 2459 tabpage_T *tp; 2460 2461 FOR_ALL_TABPAGES(tp) 2462 if (diff_buf_idx_tp(buf, tp) != DB_COUNT) 2463 return TRUE; 2464 return FALSE; 2465 } 2466 2467 /* 2468 * Move "count" times in direction "dir" to the next diff block. 2469 * Return FAIL if there isn't such a diff block. 2470 */ 2471 int 2472 diff_move_to(int dir, long count) 2473 { 2474 int idx; 2475 linenr_T lnum = curwin->w_cursor.lnum; 2476 diff_T *dp; 2477 2478 idx = diff_buf_idx(curbuf); 2479 if (idx == DB_COUNT || curtab->tp_first_diff == NULL) 2480 return FAIL; 2481 2482 if (curtab->tp_diff_invalid) 2483 ex_diffupdate(NULL); /* update after a big change */ 2484 2485 if (curtab->tp_first_diff == NULL) /* no diffs today */ 2486 return FAIL; 2487 2488 while (--count >= 0) 2489 { 2490 /* Check if already before first diff. */ 2491 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx]) 2492 break; 2493 2494 for (dp = curtab->tp_first_diff; ; dp = dp->df_next) 2495 { 2496 if (dp == NULL) 2497 break; 2498 if ((dir == FORWARD && lnum < dp->df_lnum[idx]) 2499 || (dir == BACKWARD 2500 && (dp->df_next == NULL 2501 || lnum <= dp->df_next->df_lnum[idx]))) 2502 { 2503 lnum = dp->df_lnum[idx]; 2504 break; 2505 } 2506 } 2507 } 2508 2509 /* don't end up past the end of the file */ 2510 if (lnum > curbuf->b_ml.ml_line_count) 2511 lnum = curbuf->b_ml.ml_line_count; 2512 2513 /* When the cursor didn't move at all we fail. */ 2514 if (lnum == curwin->w_cursor.lnum) 2515 return FAIL; 2516 2517 setpcmark(); 2518 curwin->w_cursor.lnum = lnum; 2519 curwin->w_cursor.col = 0; 2520 2521 return OK; 2522 } 2523 2524 /* 2525 * Return the line number in the current window that is closest to "lnum1" in 2526 * "buf1" in diff mode. 2527 */ 2528 static linenr_T 2529 diff_get_corresponding_line_int( 2530 buf_T *buf1, 2531 linenr_T lnum1) 2532 { 2533 int idx1; 2534 int idx2; 2535 diff_T *dp; 2536 int baseline = 0; 2537 2538 idx1 = diff_buf_idx(buf1); 2539 idx2 = diff_buf_idx(curbuf); 2540 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL) 2541 return lnum1; 2542 2543 if (curtab->tp_diff_invalid) 2544 ex_diffupdate(NULL); /* update after a big change */ 2545 2546 if (curtab->tp_first_diff == NULL) /* no diffs today */ 2547 return lnum1; 2548 2549 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 2550 { 2551 if (dp->df_lnum[idx1] > lnum1) 2552 return lnum1 - baseline; 2553 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1) 2554 { 2555 /* Inside the diffblock */ 2556 baseline = lnum1 - dp->df_lnum[idx1]; 2557 if (baseline > dp->df_count[idx2]) 2558 baseline = dp->df_count[idx2]; 2559 2560 return dp->df_lnum[idx2] + baseline; 2561 } 2562 if ( (dp->df_lnum[idx1] == lnum1) 2563 && (dp->df_count[idx1] == 0) 2564 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum) 2565 && ((dp->df_lnum[idx2] + dp->df_count[idx2]) 2566 > curwin->w_cursor.lnum)) 2567 /* 2568 * Special case: if the cursor is just after a zero-count 2569 * block (i.e. all filler) and the target cursor is already 2570 * inside the corresponding block, leave the target cursor 2571 * unmoved. This makes repeated CTRL-W W operations work 2572 * as expected. 2573 */ 2574 return curwin->w_cursor.lnum; 2575 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1]) 2576 - (dp->df_lnum[idx2] + dp->df_count[idx2]); 2577 } 2578 2579 /* If we get here then the cursor is after the last diff */ 2580 return lnum1 - baseline; 2581 } 2582 2583 /* 2584 * Return the line number in the current window that is closest to "lnum1" in 2585 * "buf1" in diff mode. Checks the line number to be valid. 2586 */ 2587 linenr_T 2588 diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1) 2589 { 2590 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1); 2591 2592 /* don't end up past the end of the file */ 2593 if (lnum > curbuf->b_ml.ml_line_count) 2594 return curbuf->b_ml.ml_line_count; 2595 return lnum; 2596 } 2597 2598 #if defined(FEAT_FOLDING) || defined(PROTO) 2599 /* 2600 * For line "lnum" in the current window find the equivalent lnum in window 2601 * "wp", compensating for inserted/deleted lines. 2602 */ 2603 linenr_T 2604 diff_lnum_win(linenr_T lnum, win_T *wp) 2605 { 2606 diff_T *dp; 2607 int idx; 2608 int i; 2609 linenr_T n; 2610 2611 idx = diff_buf_idx(curbuf); 2612 if (idx == DB_COUNT) /* safety check */ 2613 return (linenr_T)0; 2614 2615 if (curtab->tp_diff_invalid) 2616 ex_diffupdate(NULL); /* update after a big change */ 2617 2618 /* search for a change that includes "lnum" in the list of diffblocks. */ 2619 for (dp = curtab->tp_first_diff; dp != NULL; dp = dp->df_next) 2620 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx]) 2621 break; 2622 2623 /* When after the last change, compute relative to the last line number. */ 2624 if (dp == NULL) 2625 return wp->w_buffer->b_ml.ml_line_count 2626 - (curbuf->b_ml.ml_line_count - lnum); 2627 2628 /* Find index for "wp". */ 2629 i = diff_buf_idx(wp->w_buffer); 2630 if (i == DB_COUNT) /* safety check */ 2631 return (linenr_T)0; 2632 2633 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]); 2634 if (n > dp->df_lnum[i] + dp->df_count[i]) 2635 n = dp->df_lnum[i] + dp->df_count[i]; 2636 return n; 2637 } 2638 #endif 2639 2640 #endif /* FEAT_DIFF */ 2641