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