blob: c50670cdf508e8bda8ce1d20379f56760629459b [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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/*
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +000011 * diff.c: code for diff'ing two, three or four buffers.
Bram Moolenaare828b762018-09-10 17:51:58 +020012 *
13 * There are three ways to diff:
14 * - Shell out to an external diff program, using files.
15 * - Use the compiled-in xdiff library.
16 * - Let 'diffexpr' do the work, using files.
Bram Moolenaar071d4272004-06-13 20:20:40 +000017 */
18
19#include "vim.h"
Bram Moolenaare828b762018-09-10 17:51:58 +020020#include "xdiff/xdiff.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000021
22#if defined(FEAT_DIFF) || defined(PROTO)
23
Bram Moolenaard2b58c02018-09-16 18:10:48 +020024static int diff_busy = FALSE; // using diff structs, don't change them
25static int diff_need_update = FALSE; // ex_diffupdate needs to be called
Bram Moolenaar071d4272004-06-13 20:20:40 +000026
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010027// flags obtained from the 'diffopt' option
Bram Moolenaar785fc652018-09-15 19:17:38 +020028#define DIFF_FILLER 0x001 // display filler lines
29#define DIFF_IBLANK 0x002 // ignore empty lines
30#define DIFF_ICASE 0x004 // ignore case
31#define DIFF_IWHITE 0x008 // ignore change in white space
32#define DIFF_IWHITEALL 0x010 // ignore all white space changes
33#define DIFF_IWHITEEOL 0x020 // ignore change in white space at EOL
34#define DIFF_HORIZONTAL 0x040 // horizontal splits
35#define DIFF_VERTICAL 0x080 // vertical splits
36#define DIFF_HIDDEN_OFF 0x100 // diffoff when hidden
37#define DIFF_INTERNAL 0x200 // use internal xdiff algorithm
Bram Moolenaarc8234772019-11-10 21:00:27 +010038#define DIFF_CLOSE_OFF 0x400 // diffoff when closing window
Bram Moolenaar4223d432021-02-10 13:18:17 +010039#define DIFF_FOLLOWWRAP 0x800 // follow the wrap option
Bram Moolenaar785fc652018-09-15 19:17:38 +020040#define ALL_WHITE_DIFF (DIFF_IWHITE | DIFF_IWHITEALL | DIFF_IWHITEEOL)
Bram Moolenaarc8234772019-11-10 21:00:27 +010041static int diff_flags = DIFF_INTERNAL | DIFF_FILLER | DIFF_CLOSE_OFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +000042
Bram Moolenaare828b762018-09-10 17:51:58 +020043static long diff_algorithm = 0;
44
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010045#define LBUFLEN 50 // length of line in diff file
Bram Moolenaar071d4272004-06-13 20:20:40 +000046
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010047static int diff_a_works = MAYBE; // TRUE when "diff -a" works, FALSE when it
48 // doesn't work, MAYBE when not checked yet
Bram Moolenaar48e330a2016-02-23 14:53:34 +010049#if defined(MSWIN)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +010050static int diff_bin_works = MAYBE; // TRUE when "diff --binary" works, FALSE
51 // when it doesn't work, MAYBE when not
52 // checked yet
Bram Moolenaar071d4272004-06-13 20:20:40 +000053#endif
54
Bram Moolenaare828b762018-09-10 17:51:58 +020055// used for diff input
56typedef struct {
57 char_u *din_fname; // used for external diff
58 mmfile_t din_mmfile; // used for internal diff
59} diffin_T;
60
61// used for diff result
62typedef struct {
63 char_u *dout_fname; // used for external diff
64 garray_T dout_ga; // used for internal diff
65} diffout_T;
66
Lewis Russelld9da86e2021-12-28 13:54:41 +000067// used for recording hunks from xdiff
68typedef struct {
69 linenr_T lnum_orig;
70 long count_orig;
71 linenr_T lnum_new;
72 long count_new;
73} diffhunk_T;
74
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +010075typedef enum {
76 DIO_OUTPUT_INDICES = 0, // default
77 DIO_OUTPUT_UNIFIED = 1 // unified diff format
78} dio_outfmt_T;
79
Bram Moolenaare828b762018-09-10 17:51:58 +020080// two diff inputs and one result
81typedef struct {
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +010082 diffin_T dio_orig; // original file input
83 diffin_T dio_new; // new file input
84 diffout_T dio_diff; // diff result
85 int dio_internal; // using internal diff
86 dio_outfmt_T dio_outfmt; // internal diff output format
87 int dio_ctxlen; // unified diff context length
Bram Moolenaare828b762018-09-10 17:51:58 +020088} diffio_T;
89
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010090static int diff_buf_idx(buf_T *buf);
91static int diff_buf_idx_tp(buf_T *buf, tabpage_T *tp);
92static void diff_mark_adjust_tp(tabpage_T *tp, int idx, linenr_T line1, linenr_T line2, long amount, long amount_after);
93static void diff_check_unchanged(tabpage_T *tp, diff_T *dp);
94static int diff_check_sanity(tabpage_T *tp, diff_T *dp);
Bram Moolenaare828b762018-09-10 17:51:58 +020095static int check_external_diff(diffio_T *diffio);
96static int diff_file(diffio_T *diffio);
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010097static int diff_equal_entry(diff_T *dp, int idx1, int idx2);
98static int diff_cmp(char_u *s1, char_u *s2);
Bram Moolenaar071d4272004-06-13 20:20:40 +000099#ifdef FEAT_FOLDING
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100100static void diff_fold_update(diff_T *dp, int skip_idx);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000101#endif
Lewis Russelld9da86e2021-12-28 13:54:41 +0000102static void diff_read(int idx_orig, int idx_new, diffio_T *dio);
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100103static void diff_copy_entry(diff_T *dprev, diff_T *dp, int idx_orig, int idx_new);
104static diff_T *diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp);
Lewis Russelld9da86e2021-12-28 13:54:41 +0000105static int parse_diff_ed(char_u *line, diffhunk_T *hunk);
106static int parse_diff_unified(char_u *line, diffhunk_T *hunk);
Yegappan Lakshmananfa378352024-02-01 22:05:27 +0100107static int xdiff_out_indices(long start_a, long count_a, long start_b, long count_b, void *priv);
108static int xdiff_out_unified(void *priv, mmbuffer_t *mb, int nbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
Bram Moolenaaraeea7212020-04-02 18:50:46 +0200110#define FOR_ALL_DIFFBLOCKS_IN_TAB(tp, dp) \
111 for ((dp) = (tp)->tp_first_diff; (dp) != NULL; (dp) = (dp)->df_next)
112
Bram Moolenaar071d4272004-06-13 20:20:40 +0000113/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000114 * Called when deleting or unloading a buffer: No longer make a diff with it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000115 */
116 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100117diff_buf_delete(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000118{
119 int i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000120 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000121
Bram Moolenaar29323592016-07-24 22:04:11 +0200122 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000123 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000124 i = diff_buf_idx_tp(buf, tp);
125 if (i != DB_COUNT)
126 {
127 tp->tp_diffbuf[i] = NULL;
128 tp->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000129 if (tp == curtab)
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100130 {
131 // don't redraw right away, more might change or buffer state
132 // is invalid right now
133 need_diff_redraw = TRUE;
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100134 redraw_later(UPD_VALID);
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100135 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000136 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000137 }
138}
139
140/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000141 * Check if the current buffer should be added to or removed from the list of
142 * diff buffers.
143 */
144 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100145diff_buf_adjust(win_T *win)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000146{
147 win_T *wp;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000148 int i;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000149
150 if (!win->w_p_diff)
151 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100152 // When there is no window showing a diff for this buffer, remove
153 // it from the diffs.
Bram Moolenaar29323592016-07-24 22:04:11 +0200154 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000155 if (wp->w_buffer == win->w_buffer && wp->w_p_diff)
156 break;
157 if (wp == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000158 {
159 i = diff_buf_idx(win->w_buffer);
160 if (i != DB_COUNT)
161 {
162 curtab->tp_diffbuf[i] = NULL;
163 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000164 diff_redraw(TRUE);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000165 }
166 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +0000167 }
168 else
169 diff_buf_add(win->w_buffer);
170}
171
172/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000173 * Add a buffer to make diffs for.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000174 * Call this when a new buffer is being edited in the current window where
175 * 'diff' is set.
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +0000176 * Marks the current buffer as being part of the diff and requiring updating.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000177 * This must be done before any autocmd, because a command may use info
178 * about the screen contents.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 */
180 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100181diff_buf_add(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000182{
183 int i;
184
185 if (diff_buf_idx(buf) != DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100186 return; // It's already there.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187
188 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000189 if (curtab->tp_diffbuf[i] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000190 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000191 curtab->tp_diffbuf[i] = buf;
192 curtab->tp_diff_invalid = TRUE;
Bram Moolenaar5d55c0f2008-11-30 14:16:57 +0000193 diff_redraw(TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000194 return;
195 }
196
Bram Moolenaare1242042021-12-16 20:56:57 +0000197 semsg(_(e_cannot_diff_more_than_nr_buffers), DB_COUNT);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198}
199
200/*
Bram Moolenaar25ea0542017-02-03 23:16:28 +0100201 * Remove all buffers to make diffs for.
202 */
203 static void
204diff_buf_clear(void)
205{
206 int i;
207
208 for (i = 0; i < DB_COUNT; ++i)
209 if (curtab->tp_diffbuf[i] != NULL)
210 {
211 curtab->tp_diffbuf[i] = NULL;
212 curtab->tp_diff_invalid = TRUE;
213 diff_redraw(TRUE);
214 }
215}
216
217/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000218 * Find buffer "buf" in the list of diff buffers for the current tab page.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219 * Return its index or DB_COUNT if not found.
220 */
221 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100222diff_buf_idx(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000223{
224 int idx;
225
226 for (idx = 0; idx < DB_COUNT; ++idx)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000227 if (curtab->tp_diffbuf[idx] == buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000228 break;
229 return idx;
230}
231
232/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000233 * Find buffer "buf" in the list of diff buffers for tab page "tp".
234 * Return its index or DB_COUNT if not found.
235 */
236 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100237diff_buf_idx_tp(buf_T *buf, tabpage_T *tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000238{
239 int idx;
240
241 for (idx = 0; idx < DB_COUNT; ++idx)
242 if (tp->tp_diffbuf[idx] == buf)
243 break;
244 return idx;
245}
246
247/*
248 * Mark the diff info involving buffer "buf" as invalid, it will be updated
249 * when info is requested.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000250 */
251 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100252diff_invalidate(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000253{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000254 tabpage_T *tp;
255 int i;
256
Bram Moolenaar29323592016-07-24 22:04:11 +0200257 FOR_ALL_TABPAGES(tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000258 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000259 i = diff_buf_idx_tp(buf, tp);
260 if (i != DB_COUNT)
261 {
262 tp->tp_diff_invalid = TRUE;
263 if (tp == curtab)
264 diff_redraw(TRUE);
265 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000266 }
267}
268
269/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000270 * Called by mark_adjust(): update line numbers in "curbuf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000271 */
272 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100273diff_mark_adjust(
274 linenr_T line1,
275 linenr_T line2,
276 long amount,
277 long amount_after)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000278{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000279 int idx;
280 tabpage_T *tp;
281
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100282 // Handle all tab pages that use the current buffer in a diff.
Bram Moolenaar29323592016-07-24 22:04:11 +0200283 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000284 {
285 idx = diff_buf_idx_tp(curbuf, tp);
286 if (idx != DB_COUNT)
287 diff_mark_adjust_tp(tp, idx, line1, line2, amount, amount_after);
288 }
289}
290
291/*
292 * Update line numbers in tab page "tp" for "curbuf" with index "idx".
293 * This attempts to update the changes as much as possible:
294 * When inserting/deleting lines outside of existing change blocks, create a
295 * new change block and update the line numbers in following blocks.
296 * When inserting/deleting lines in existing change blocks, update them.
297 */
298 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100299diff_mark_adjust_tp(
300 tabpage_T *tp,
301 int idx,
302 linenr_T line1,
303 linenr_T line2,
304 long amount,
305 long amount_after)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000306{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000307 diff_T *dp;
308 diff_T *dprev;
309 diff_T *dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 int i;
311 int inserted, deleted;
312 int n, off;
313 linenr_T last;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100314 linenr_T lnum_deleted = line1; // lnum of remaining deletion
Bram Moolenaar071d4272004-06-13 20:20:40 +0000315 int check_unchanged;
316
Bram Moolenaare3521d92018-09-16 14:10:31 +0200317 if (diff_internal())
318 {
Bram Moolenaar198fa062018-09-18 21:20:26 +0200319 // Will update diffs before redrawing. Set _invalid to update the
Bram Moolenaare3521d92018-09-16 14:10:31 +0200320 // diffs themselves, set _update to also update folds properly just
321 // before redrawing.
Bram Moolenaar5f57bdc2018-10-25 17:52:23 +0200322 // Do update marks here, it is needed for :%diffput.
Bram Moolenaare3521d92018-09-16 14:10:31 +0200323 tp->tp_diff_invalid = TRUE;
324 tp->tp_diff_update = TRUE;
Bram Moolenaare3521d92018-09-16 14:10:31 +0200325 }
326
Bram Moolenaar071d4272004-06-13 20:20:40 +0000327 if (line2 == MAXLNUM)
328 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100329 // mark_adjust(99, MAXLNUM, 9, 0): insert lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000330 inserted = amount;
331 deleted = 0;
332 }
333 else if (amount_after > 0)
334 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100335 // mark_adjust(99, 98, MAXLNUM, 9): a change that inserts lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000336 inserted = amount_after;
337 deleted = 0;
338 }
339 else
340 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100341 // mark_adjust(98, 99, MAXLNUM, -2): delete lines
Bram Moolenaar071d4272004-06-13 20:20:40 +0000342 inserted = 0;
343 deleted = -amount_after;
344 }
345
346 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000347 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000348 for (;;)
349 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100350 // If the change is after the previous diff block and before the next
351 // diff block, thus not touching an existing change, create a new diff
352 // block. Don't do this when ex_diffgetput() is busy.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000353 if ((dp == NULL || dp->df_lnum[idx] - 1 > line2
354 || (line2 == MAXLNUM && dp->df_lnum[idx] > line1))
355 && (dprev == NULL
356 || dprev->df_lnum[idx] + dprev->df_count[idx] < line1)
357 && !diff_busy)
358 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000359 dnext = diff_alloc_new(tp, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000360 if (dnext == NULL)
361 return;
362
363 dnext->df_lnum[idx] = line1;
364 dnext->df_count[idx] = inserted;
365 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000366 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367 {
368 if (dprev == NULL)
369 dnext->df_lnum[i] = line1;
370 else
371 dnext->df_lnum[i] = line1
372 + (dprev->df_lnum[i] + dprev->df_count[i])
373 - (dprev->df_lnum[idx] + dprev->df_count[idx]);
374 dnext->df_count[i] = deleted;
375 }
376 }
377
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100378 // if at end of the list, quit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000379 if (dp == NULL)
380 break;
381
382 /*
383 * Check for these situations:
384 * 1 2 3
385 * 1 2 3
386 * line1 2 3 4 5
387 * 2 3 4 5
388 * 2 3 4 5
389 * line2 2 3 4 5
390 * 3 5 6
391 * 3 5 6
392 */
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100393 // compute last line of this change
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394 last = dp->df_lnum[idx] + dp->df_count[idx] - 1;
395
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100396 // 1. change completely above line1: nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +0000397 if (last >= line1 - 1)
398 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100399 // 6. change below line2: only adjust for amount_after; also when
400 // "deleted" became zero when deleted all lines between two diffs
Bram Moolenaar071d4272004-06-13 20:20:40 +0000401 if (dp->df_lnum[idx] - (deleted + inserted != 0) > line2)
402 {
403 if (amount_after == 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100404 break; // nothing left to change
Bram Moolenaar071d4272004-06-13 20:20:40 +0000405 dp->df_lnum[idx] += amount_after;
406 }
407 else
408 {
409 check_unchanged = FALSE;
410
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100411 // 2. 3. 4. 5.: inserted/deleted lines touching this diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412 if (deleted > 0)
413 {
Bram Moolenaarc101abf2022-06-26 16:53:34 +0100414 off = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000415 if (dp->df_lnum[idx] >= line1)
416 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000417 if (last <= line2)
418 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100419 // 4. delete all lines of diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420 if (dp->df_next != NULL
421 && dp->df_next->df_lnum[idx] - 1 <= line2)
422 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100423 // delete continues in next diff, only do
424 // lines until that one
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425 n = dp->df_next->df_lnum[idx] - lnum_deleted;
426 deleted -= n;
427 n -= dp->df_count[idx];
428 lnum_deleted = dp->df_next->df_lnum[idx];
429 }
430 else
431 n = deleted - dp->df_count[idx];
432 dp->df_count[idx] = 0;
433 }
434 else
435 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100436 // 5. delete lines at or just before top of diff
Bram Moolenaarc101abf2022-06-26 16:53:34 +0100437 off = dp->df_lnum[idx] - lnum_deleted;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000438 n = off;
439 dp->df_count[idx] -= line2 - dp->df_lnum[idx] + 1;
440 check_unchanged = TRUE;
441 }
442 dp->df_lnum[idx] = line1;
443 }
444 else
445 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000446 if (last < line2)
447 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100448 // 2. delete at end of diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000449 dp->df_count[idx] -= last - lnum_deleted + 1;
450 if (dp->df_next != NULL
451 && dp->df_next->df_lnum[idx] - 1 <= line2)
452 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100453 // delete continues in next diff, only do
454 // lines until that one
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455 n = dp->df_next->df_lnum[idx] - 1 - last;
456 deleted -= dp->df_next->df_lnum[idx]
457 - lnum_deleted;
458 lnum_deleted = dp->df_next->df_lnum[idx];
459 }
460 else
461 n = line2 - last;
462 check_unchanged = TRUE;
463 }
464 else
465 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100466 // 3. delete lines inside the diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000467 n = 0;
468 dp->df_count[idx] -= deleted;
469 }
470 }
471
472 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000473 if (tp->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000474 {
Bram Moolenaar4e677b92022-07-28 18:44:27 +0100475 if (dp->df_lnum[i] > off)
476 dp->df_lnum[i] -= off;
477 else
478 dp->df_lnum[i] = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000479 dp->df_count[i] += n;
480 }
481 }
482 else
483 {
484 if (dp->df_lnum[idx] <= line1)
485 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100486 // inserted lines somewhere in this diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000487 dp->df_count[idx] += inserted;
488 check_unchanged = TRUE;
489 }
490 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100491 // inserted lines somewhere above this diff
Bram Moolenaar071d4272004-06-13 20:20:40 +0000492 dp->df_lnum[idx] += inserted;
493 }
494
495 if (check_unchanged)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100496 // Check if inserted lines are equal, may reduce the
497 // size of the diff. TODO: also check for equal lines
498 // in the middle and perhaps split the block.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000499 diff_check_unchanged(tp, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500 }
501 }
502
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100503 // check if this block touches the previous one, may merge them.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000504 if (dprev != NULL && dprev->df_lnum[idx] + dprev->df_count[idx]
505 == dp->df_lnum[idx])
506 {
507 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000508 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000509 dprev->df_count[i] += dp->df_count[i];
510 dprev->df_next = dp->df_next;
511 vim_free(dp);
512 dp = dprev->df_next;
513 }
514 else
515 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100516 // Advance to next entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517 dprev = dp;
518 dp = dp->df_next;
519 }
520 }
521
522 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000523 dp = tp->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000524 while (dp != NULL)
525 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100526 // All counts are zero, remove this entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000527 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000528 if (tp->tp_diffbuf[i] != NULL && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 break;
530 if (i == DB_COUNT)
531 {
532 dnext = dp->df_next;
533 vim_free(dp);
534 dp = dnext;
535 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000536 tp->tp_first_diff = dnext;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000537 else
538 dprev->df_next = dnext;
539 }
540 else
541 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100542 // Advance to next entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000543 dprev = dp;
544 dp = dp->df_next;
545 }
546
547 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000548
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000549 if (tp == curtab)
550 {
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200551 // Don't redraw right away, this updates the diffs, which can be slow.
552 need_diff_redraw = TRUE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000553
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100554 // Need to recompute the scroll binding, may remove or add filler
555 // lines (e.g., when adding lines above w_topline). But it's slow when
556 // making many changes, postpone until redrawing.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000557 diff_need_scrollbind = TRUE;
558 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559}
560
561/*
562 * Allocate a new diff block and link it between "dprev" and "dp".
563 */
564 static diff_T *
Bram Moolenaar7454a062016-01-30 15:14:10 +0100565diff_alloc_new(tabpage_T *tp, diff_T *dprev, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566{
567 diff_T *dnew;
568
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200569 dnew = ALLOC_ONE(diff_T);
Yegappan Lakshmanan1cfb14a2023-01-09 19:04:23 +0000570 if (dnew == NULL)
571 return NULL;
572
573 dnew->df_next = dp;
574 if (dprev == NULL)
575 tp->tp_first_diff = dnew;
576 else
577 dprev->df_next = dnew;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000578 return dnew;
579}
580
581/*
582 * Check if the diff block "dp" can be made smaller for lines at the start and
583 * end that are equal. Called after inserting lines.
584 * This may result in a change where all buffers have zero lines, the caller
585 * must take care of removing it.
586 */
587 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100588diff_check_unchanged(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000589{
590 int i_org;
591 int i_new;
592 int off_org, off_new;
593 char_u *line_org;
594 int dir = FORWARD;
595
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100596 // Find the first buffers, use it as the original, compare the other
597 // buffer lines against this one.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000598 for (i_org = 0; i_org < DB_COUNT; ++i_org)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000599 if (tp->tp_diffbuf[i_org] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100601 if (i_org == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +0000602 return;
603
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000604 if (diff_check_sanity(tp, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 return;
606
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100607 // First check lines at the top, then at the bottom.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608 off_org = 0;
609 off_new = 0;
610 for (;;)
611 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100612 // Repeat until a line is found which is different or the number of
613 // lines has become zero.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614 while (dp->df_count[i_org] > 0)
615 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100616 // Copy the line, the next ml_get() will invalidate it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 if (dir == BACKWARD)
618 off_org = dp->df_count[i_org] - 1;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000619 line_org = vim_strsave(ml_get_buf(tp->tp_diffbuf[i_org],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000620 dp->df_lnum[i_org] + off_org, FALSE));
621 if (line_org == NULL)
622 return;
623 for (i_new = i_org + 1; i_new < DB_COUNT; ++i_new)
624 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000625 if (tp->tp_diffbuf[i_new] == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626 continue;
627 if (dir == BACKWARD)
628 off_new = dp->df_count[i_new] - 1;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100629 // if other buffer doesn't have this line, it was inserted
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 if (off_new < 0 || off_new >= dp->df_count[i_new])
631 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000632 if (diff_cmp(line_org, ml_get_buf(tp->tp_diffbuf[i_new],
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 dp->df_lnum[i_new] + off_new, FALSE)) != 0)
634 break;
635 }
636 vim_free(line_org);
637
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100638 // Stop when a line isn't equal in all diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000639 if (i_new != DB_COUNT)
640 break;
641
Bram Moolenaar5d18efe2019-12-01 21:11:22 +0100642 // Line matched in all buffers, remove it from the diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000643 for (i_new = i_org; i_new < DB_COUNT; ++i_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000644 if (tp->tp_diffbuf[i_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645 {
646 if (dir == FORWARD)
647 ++dp->df_lnum[i_new];
648 --dp->df_count[i_new];
649 }
650 }
651 if (dir == BACKWARD)
652 break;
653 dir = BACKWARD;
654 }
655}
656
657/*
658 * Check if a diff block doesn't contain invalid line numbers.
659 * This can happen when the diff program returns invalid results.
660 */
661 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +0100662diff_check_sanity(tabpage_T *tp, diff_T *dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000663{
664 int i;
665
666 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000667 if (tp->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 if (dp->df_lnum[i] + dp->df_count[i] - 1
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000669 > tp->tp_diffbuf[i]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 return FAIL;
671 return OK;
672}
673
674/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000675 * Mark all diff buffers in the current tab page for redraw.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000676 */
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200677 void
Bram Moolenaar7454a062016-01-30 15:14:10 +0100678diff_redraw(
Bram Moolenaare3521d92018-09-16 14:10:31 +0200679 int dofold) // also recompute the folds
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680{
681 win_T *wp;
Bram Moolenaar04626c22021-09-01 16:02:07 +0200682 win_T *wp_other = NULL;
Bram Moolenaar841c2252021-10-22 20:56:55 +0100683 int used_max_fill_other = FALSE;
684 int used_max_fill_curwin = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000685 int n;
686
Bram Moolenaar4f57eef2019-08-24 20:54:19 +0200687 need_diff_redraw = FALSE;
Bram Moolenaar29323592016-07-24 22:04:11 +0200688 FOR_ALL_WINDOWS(wp)
zeertzjq101d57b2022-07-31 18:34:32 +0100689 {
Bram Moolenaarcd38bb42022-06-26 14:04:07 +0100690 // when closing windows or wiping buffers skip invalid window
zeertzjq101d57b2022-07-31 18:34:32 +0100691 if (!wp->w_p_diff || !buf_valid(wp->w_buffer))
692 continue;
693
Bram Moolenaara4d158b2022-08-14 14:17:45 +0100694 redraw_win_later(wp, UPD_SOME_VALID);
zeertzjq101d57b2022-07-31 18:34:32 +0100695 if (wp != curwin)
696 wp_other = wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000697#ifdef FEAT_FOLDING
zeertzjq101d57b2022-07-31 18:34:32 +0100698 if (dofold && foldmethodIsDiff(wp))
699 foldUpdateAll(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000700#endif
zeertzjq101d57b2022-07-31 18:34:32 +0100701 // A change may have made filler lines invalid, need to take care of
702 // that for other windows.
703 n = diff_check(wp, wp->w_topline);
704 if ((wp != curwin && wp->w_topfill > 0) || n > 0)
705 {
706 if (wp->w_topfill > n)
707 wp->w_topfill = (n < 0 ? 0 : n);
708 else if (n > 0 && n > wp->w_topfill)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000709 {
zeertzjq101d57b2022-07-31 18:34:32 +0100710 wp->w_topfill = n;
711 if (wp == curwin)
712 used_max_fill_curwin = TRUE;
713 else if (wp_other != NULL)
714 used_max_fill_other = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715 }
zeertzjq101d57b2022-07-31 18:34:32 +0100716 check_topfill(wp, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000717 }
zeertzjq101d57b2022-07-31 18:34:32 +0100718 }
Bram Moolenaar04626c22021-09-01 16:02:07 +0200719
Bram Moolenaar841c2252021-10-22 20:56:55 +0100720 if (wp_other != NULL && curwin->w_p_scb)
721 {
722 if (used_max_fill_curwin)
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000723 // The current window was set to use the maximum number of filler
Bram Moolenaar841c2252021-10-22 20:56:55 +0100724 // lines, may need to reduce them.
725 diff_set_topline(wp_other, curwin);
726 else if (used_max_fill_other)
Dominique Pelleaf4a61a2021-12-27 17:21:41 +0000727 // The other window was set to use the maximum number of filler
Bram Moolenaar841c2252021-10-22 20:56:55 +0100728 // lines, may need to reduce them.
729 diff_set_topline(curwin, wp_other);
730 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000731}
732
Bram Moolenaare828b762018-09-10 17:51:58 +0200733 static void
734clear_diffin(diffin_T *din)
735{
736 if (din->din_fname == NULL)
Yegappan Lakshmanan960dcbd2023-03-07 17:45:11 +0000737 VIM_CLEAR(din->din_mmfile.ptr);
Bram Moolenaare828b762018-09-10 17:51:58 +0200738 else
739 mch_remove(din->din_fname);
740}
741
742 static void
743clear_diffout(diffout_T *dout)
744{
745 if (dout->dout_fname == NULL)
746 ga_clear_strings(&dout->dout_ga);
747 else
748 mch_remove(dout->dout_fname);
749}
750
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751/*
Bram Moolenaare828b762018-09-10 17:51:58 +0200752 * Write buffer "buf" to a memory buffer.
753 * Return FAIL for failure.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754 */
755 static int
Bram Moolenaare828b762018-09-10 17:51:58 +0200756diff_write_buffer(buf_T *buf, diffin_T *din)
757{
758 linenr_T lnum;
759 char_u *s;
760 long len = 0;
761 char_u *ptr;
762
763 // xdiff requires one big block of memory with all the text.
764 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
zeertzjq94b7c322024-03-12 21:50:32 +0100765 len += ml_get_buf_len(buf, lnum) + 1;
Bram Moolenaar18a4ba22019-05-24 19:39:03 +0200766 ptr = alloc(len);
Bram Moolenaare828b762018-09-10 17:51:58 +0200767 if (ptr == NULL)
768 {
769 // Allocating memory failed. This can happen, because we try to read
770 // the whole buffer text into memory. Set the failed flag, the diff
771 // will be retried with external diff. The flag is never reset.
772 buf->b_diff_failed = TRUE;
773 if (p_verbose > 0)
774 {
775 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100776 smsg(_("Not enough memory to use internal diff for buffer \"%s\""),
Bram Moolenaare828b762018-09-10 17:51:58 +0200777 buf->b_fname);
778 verbose_leave();
779 }
780 return FAIL;
781 }
782 din->din_mmfile.ptr = (char *)ptr;
783 din->din_mmfile.size = len;
784
785 len = 0;
786 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
787 {
788 for (s = ml_get_buf(buf, lnum, FALSE); *s != NUL; )
789 {
790 if (diff_flags & DIFF_ICASE)
791 {
792 int c;
Bram Moolenaare828b762018-09-10 17:51:58 +0200793 int orig_len;
794 char_u cbuf[MB_MAXBYTES + 1];
795
Bram Moolenaar06f60952021-12-28 18:30:05 +0000796 if (*s == NL)
797 c = NUL;
798 else
799 {
800 // xdiff doesn't support ignoring case, fold-case the text.
801 c = PTR2CHAR(s);
802 c = MB_CASEFOLD(c);
803 }
Bram Moolenaar1614a142019-10-06 22:00:13 +0200804 orig_len = mb_ptr2len(s);
Bram Moolenaare828b762018-09-10 17:51:58 +0200805 if (mb_char2bytes(c, cbuf) != orig_len)
806 // TODO: handle byte length difference
807 mch_memmove(ptr + len, s, orig_len);
808 else
809 mch_memmove(ptr + len, cbuf, orig_len);
810
811 s += orig_len;
812 len += orig_len;
Bram Moolenaare828b762018-09-10 17:51:58 +0200813 }
814 else
Bram Moolenaar06f60952021-12-28 18:30:05 +0000815 {
816 ptr[len++] = *s == NL ? NUL : *s;
817 s++;
818 }
Bram Moolenaare828b762018-09-10 17:51:58 +0200819 }
820 ptr[len++] = NL;
821 }
822 return OK;
823}
824
825/*
826 * Write buffer "buf" to file or memory buffer.
827 * Return FAIL for failure.
828 */
829 static int
830diff_write(buf_T *buf, diffin_T *din)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831{
832 int r;
833 char_u *save_ff;
Bram Moolenaare1004402020-10-24 20:49:43 +0200834 int save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835
Bram Moolenaare828b762018-09-10 17:51:58 +0200836 if (din->din_fname == NULL)
837 return diff_write_buffer(buf, din);
838
839 // Always use 'fileformat' set to "unix".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000840 save_ff = buf->b_p_ff;
841 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
Bram Moolenaare1004402020-10-24 20:49:43 +0200842 save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100843 // Writing the buffer is an implementation detail of performing the diff,
844 // so it shouldn't update the '[ and '] marks.
Bram Moolenaare1004402020-10-24 20:49:43 +0200845 cmdmod.cmod_flags |= CMOD_LOCKMARKS;
Bram Moolenaare828b762018-09-10 17:51:58 +0200846 r = buf_write(buf, din->din_fname, NULL,
847 (linenr_T)1, buf->b_ml.ml_line_count,
848 NULL, FALSE, FALSE, FALSE, TRUE);
Bram Moolenaare1004402020-10-24 20:49:43 +0200849 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850 free_string_option(buf->b_p_ff);
851 buf->b_p_ff = save_ff;
852 return r;
853}
854
855/*
Bram Moolenaare828b762018-09-10 17:51:58 +0200856 * Update the diffs for all buffers involved.
857 */
858 static void
859diff_try_update(
860 diffio_T *dio,
861 int idx_orig,
862 exarg_T *eap) // "eap" can be NULL
863{
864 buf_T *buf;
865 int idx_new;
866
867 if (dio->dio_internal)
868 {
869 ga_init2(&dio->dio_diff.dout_ga, sizeof(char *), 1000);
870 }
871 else
872 {
873 // We need three temp file names.
874 dio->dio_orig.din_fname = vim_tempname('o', TRUE);
875 dio->dio_new.din_fname = vim_tempname('n', TRUE);
876 dio->dio_diff.dout_fname = vim_tempname('d', TRUE);
877 if (dio->dio_orig.din_fname == NULL
878 || dio->dio_new.din_fname == NULL
879 || dio->dio_diff.dout_fname == NULL)
880 goto theend;
881 }
882
883 // Check external diff is actually working.
884 if (!dio->dio_internal && check_external_diff(dio) == FAIL)
885 goto theend;
886
887 // :diffupdate!
888 if (eap != NULL && eap->forceit)
889 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
890 {
891 buf = curtab->tp_diffbuf[idx_new];
892 if (buf_valid(buf))
893 buf_check_timestamp(buf, FALSE);
894 }
895
896 // Write the first buffer to a tempfile or mmfile_t.
897 buf = curtab->tp_diffbuf[idx_orig];
898 if (diff_write(buf, &dio->dio_orig) == FAIL)
899 goto theend;
900
901 // Make a difference between the first buffer and every other.
902 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
903 {
904 buf = curtab->tp_diffbuf[idx_new];
905 if (buf == NULL || buf->b_ml.ml_mfp == NULL)
906 continue; // skip buffer that isn't loaded
907
908 // Write the other buffer and diff with the first one.
909 if (diff_write(buf, &dio->dio_new) == FAIL)
910 continue;
911 if (diff_file(dio) == FAIL)
912 continue;
913
914 // Read the diff output and add each entry to the diff list.
Lewis Russelld9da86e2021-12-28 13:54:41 +0000915 diff_read(idx_orig, idx_new, dio);
Bram Moolenaare828b762018-09-10 17:51:58 +0200916
917 clear_diffin(&dio->dio_new);
918 clear_diffout(&dio->dio_diff);
919 }
920 clear_diffin(&dio->dio_orig);
921
922theend:
923 vim_free(dio->dio_orig.din_fname);
924 vim_free(dio->dio_new.din_fname);
925 vim_free(dio->dio_diff.dout_fname);
926}
927
928/*
929 * Return TRUE if the options are set to use the internal diff library.
930 * Note that if the internal diff failed for one of the buffers, the external
931 * diff will be used anyway.
932 */
Bram Moolenaare3521d92018-09-16 14:10:31 +0200933 int
Bram Moolenaare828b762018-09-10 17:51:58 +0200934diff_internal(void)
935{
Bram Moolenaar975880b2019-03-03 14:42:11 +0100936 return (diff_flags & DIFF_INTERNAL) != 0
937#ifdef FEAT_EVAL
938 && *p_dex == NUL
939#endif
940 ;
Bram Moolenaare828b762018-09-10 17:51:58 +0200941}
942
943/*
944 * Return TRUE if the internal diff failed for one of the diff buffers.
945 */
946 static int
947diff_internal_failed(void)
948{
949 int idx;
950
951 // Only need to do something when there is another buffer.
952 for (idx = 0; idx < DB_COUNT; ++idx)
953 if (curtab->tp_diffbuf[idx] != NULL
954 && curtab->tp_diffbuf[idx]->b_diff_failed)
955 return TRUE;
956 return FALSE;
957}
958
959/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960 * Completely update the diffs for the buffers involved.
Bram Moolenaare3521d92018-09-16 14:10:31 +0200961 * When using the external "diff" command the buffers are written to a file,
962 * also for unmodified buffers (the file could have been produced by
963 * autocommands, e.g. the netrw plugin).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000964 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000965 void
Bram Moolenaare828b762018-09-10 17:51:58 +0200966ex_diffupdate(exarg_T *eap) // "eap" can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000968 int idx_orig;
969 int idx_new;
Bram Moolenaare828b762018-09-10 17:51:58 +0200970 diffio_T diffio;
Bram Moolenaar198fa062018-09-18 21:20:26 +0200971 int had_diffs = curtab->tp_first_diff != NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000972
Bram Moolenaard2b58c02018-09-16 18:10:48 +0200973 if (diff_busy)
974 {
975 diff_need_update = TRUE;
976 return;
977 }
978
Bram Moolenaare828b762018-09-10 17:51:58 +0200979 // Delete all diffblocks.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000980 diff_clear(curtab);
981 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000982
Bram Moolenaare828b762018-09-10 17:51:58 +0200983 // Use the first buffer as the original text.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000984 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000985 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000986 break;
987 if (idx_orig == DB_COUNT)
Bram Moolenaar198fa062018-09-18 21:20:26 +0200988 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989
Bram Moolenaare828b762018-09-10 17:51:58 +0200990 // Only need to do something when there is another buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000991 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000992 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000993 break;
994 if (idx_new == DB_COUNT)
Bram Moolenaar198fa062018-09-18 21:20:26 +0200995 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000996
Bram Moolenaare828b762018-09-10 17:51:58 +0200997 // Only use the internal method if it did not fail for one of the buffers.
Bram Moolenaara80faa82020-04-12 19:37:17 +0200998 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +0200999 diffio.dio_internal = diff_internal() && !diff_internal_failed();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001000
Bram Moolenaare828b762018-09-10 17:51:58 +02001001 diff_try_update(&diffio, idx_orig, eap);
1002 if (diffio.dio_internal && diff_internal_failed())
1003 {
1004 // Internal diff failed, use external diff instead.
Bram Moolenaara80faa82020-04-12 19:37:17 +02001005 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +02001006 diff_try_update(&diffio, idx_orig, eap);
1007 }
1008
1009 // force updating cursor position on screen
1010 curwin->w_valid_cursor.lnum = 0;
1011
Bram Moolenaar198fa062018-09-18 21:20:26 +02001012theend:
1013 // A redraw is needed if there were diffs and they were cleared, or there
1014 // are diffs now, which means they got updated.
1015 if (had_diffs || curtab->tp_first_diff != NULL)
1016 {
1017 diff_redraw(TRUE);
1018 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
1019 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001020}
1021
1022/*
1023 * Do a quick test if "diff" really works. Otherwise it looks like there
1024 * are no differences. Can't use the return value, it's non-zero when
1025 * there are differences.
1026 */
1027 static int
1028check_external_diff(diffio_T *diffio)
1029{
1030 FILE *fd;
1031 int ok;
1032 int io_error = FALSE;
1033
1034 // May try twice, first with "-a" and then without.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001035 for (;;)
1036 {
1037 ok = FALSE;
Bram Moolenaare828b762018-09-10 17:51:58 +02001038 fd = mch_fopen((char *)diffio->dio_orig.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001039 if (fd == NULL)
1040 io_error = TRUE;
1041 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001042 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001043 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
1044 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001046 fd = mch_fopen((char *)diffio->dio_new.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001047 if (fd == NULL)
1048 io_error = TRUE;
1049 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001050 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001051 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
1052 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001053 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001054 fd = NULL;
1055 if (diff_file(diffio) == OK)
1056 fd = mch_fopen((char *)diffio->dio_diff.dout_fname, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001057 if (fd == NULL)
1058 io_error = TRUE;
1059 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060 {
1061 char_u linebuf[LBUFLEN];
1062
1063 for (;;)
1064 {
glacambread5c1782021-05-24 14:20:53 +02001065 // For normal diff there must be a line that contains
1066 // "1c1". For unified diff "@@ -1 +1 @@".
Bram Moolenaar00590742019-02-15 21:06:09 +01001067 if (vim_fgets(linebuf, LBUFLEN, fd))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001068 break;
glacambread5c1782021-05-24 14:20:53 +02001069 if (STRNCMP(linebuf, "1c1", 3) == 0
1070 || STRNCMP(linebuf, "@@ -1 +1 @@", 11) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001071 ok = TRUE;
1072 }
1073 fclose(fd);
1074 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001075 mch_remove(diffio->dio_diff.dout_fname);
1076 mch_remove(diffio->dio_new.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001077 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001078 mch_remove(diffio->dio_orig.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001079 }
1080
1081#ifdef FEAT_EVAL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001082 // When using 'diffexpr' break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001083 if (*p_dex != NUL)
1084 break;
1085#endif
1086
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001087#if defined(MSWIN)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001088 // If the "-a" argument works, also check if "--binary" works.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001089 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
1090 {
1091 diff_a_works = TRUE;
1092 diff_bin_works = TRUE;
1093 continue;
1094 }
1095 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
1096 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001097 // Tried --binary, but it failed. "-a" works though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001098 diff_bin_works = FALSE;
1099 ok = TRUE;
1100 }
1101#endif
1102
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001103 // If we checked if "-a" works already, break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001104 if (diff_a_works != MAYBE)
1105 break;
1106 diff_a_works = ok;
1107
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001108 // If "-a" works break here, otherwise retry without "-a".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001109 if (ok)
1110 break;
1111 }
1112 if (!ok)
1113 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001114 if (io_error)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001115 emsg(_(e_cannot_read_or_write_temp_files));
Bram Moolenaare1242042021-12-16 20:56:57 +00001116 emsg(_(e_cannot_create_diffs));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001117 diff_a_works = MAYBE;
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001118#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001119 diff_bin_works = MAYBE;
1120#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001121 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001123 return OK;
1124}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125
Bram Moolenaare828b762018-09-10 17:51:58 +02001126/*
1127 * Invoke the xdiff function.
1128 */
1129 static int
1130diff_file_internal(diffio_T *diffio)
1131{
1132 xpparam_t param;
1133 xdemitconf_t emit_cfg;
1134 xdemitcb_t emit_cb;
Bram Moolenaarbd1d5602012-05-18 18:47:17 +02001135
Bram Moolenaara80faa82020-04-12 19:37:17 +02001136 CLEAR_FIELD(param);
1137 CLEAR_FIELD(emit_cfg);
1138 CLEAR_FIELD(emit_cb);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001139
Bram Moolenaare828b762018-09-10 17:51:58 +02001140 param.flags = diff_algorithm;
1141
1142 if (diff_flags & DIFF_IWHITE)
1143 param.flags |= XDF_IGNORE_WHITESPACE_CHANGE;
Bram Moolenaar785fc652018-09-15 19:17:38 +02001144 if (diff_flags & DIFF_IWHITEALL)
1145 param.flags |= XDF_IGNORE_WHITESPACE;
1146 if (diff_flags & DIFF_IWHITEEOL)
1147 param.flags |= XDF_IGNORE_WHITESPACE_AT_EOL;
1148 if (diff_flags & DIFF_IBLANK)
1149 param.flags |= XDF_IGNORE_BLANK_LINES;
Bram Moolenaare828b762018-09-10 17:51:58 +02001150
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001151 emit_cfg.ctxlen = diffio->dio_ctxlen;
Bram Moolenaare828b762018-09-10 17:51:58 +02001152 emit_cb.priv = &diffio->dio_diff;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001153 if (diffio->dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01001154 emit_cfg.hunk_func = xdiff_out_indices;
1155 else
1156 emit_cb.out_line = xdiff_out_unified;
Bram Moolenaare828b762018-09-10 17:51:58 +02001157 if (xdl_diff(&diffio->dio_orig.din_mmfile,
1158 &diffio->dio_new.din_mmfile,
1159 &param, &emit_cfg, &emit_cb) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001160 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001161 emsg(_(e_problem_creating_internal_diff));
Bram Moolenaare828b762018-09-10 17:51:58 +02001162 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001163 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001164 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001165}
1166
1167/*
1168 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
Bram Moolenaare828b762018-09-10 17:51:58 +02001169 * return OK or FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001170 */
Bram Moolenaare828b762018-09-10 17:51:58 +02001171 static int
1172diff_file(diffio_T *dio)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001173{
1174 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001175 size_t len;
Bram Moolenaare828b762018-09-10 17:51:58 +02001176 char_u *tmp_orig = dio->dio_orig.din_fname;
1177 char_u *tmp_new = dio->dio_new.din_fname;
1178 char_u *tmp_diff = dio->dio_diff.dout_fname;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001179
1180#ifdef FEAT_EVAL
1181 if (*p_dex != NUL)
Bram Moolenaare828b762018-09-10 17:51:58 +02001182 {
1183 // Use 'diffexpr' to generate the diff file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001184 eval_diff(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaare828b762018-09-10 17:51:58 +02001185 return OK;
1186 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001187 else
1188#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001189 // Use xdiff for generating the diff.
1190 if (dio->dio_internal)
Bram Moolenaare828b762018-09-10 17:51:58 +02001191 return diff_file_internal(dio);
Bram Moolenaar95fb60a2005-03-08 22:29:20 +00001192
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001193 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
1194 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
1195 cmd = alloc(len);
1196 if (cmd == NULL)
1197 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02001198
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001199 // We don't want $DIFF_OPTIONS to get in the way.
1200 if (getenv("DIFF_OPTIONS"))
1201 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
1202
1203 // Build the diff command and execute it. Always use -a, binary
1204 // differences are of no use. Ignore errors, diff returns
1205 // non-zero when differences have been found.
1206 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s%s%s%s %s",
1207 diff_a_works == FALSE ? "" : "-a ",
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001208#if defined(MSWIN)
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001209 diff_bin_works == TRUE ? "--binary " : "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210#else
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001211 "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001212#endif
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001213 (diff_flags & DIFF_IWHITE) ? "-b " : "",
1214 (diff_flags & DIFF_IWHITEALL) ? "-w " : "",
1215 (diff_flags & DIFF_IWHITEEOL) ? "-Z " : "",
1216 (diff_flags & DIFF_IBLANK) ? "-B " : "",
1217 (diff_flags & DIFF_ICASE) ? "-i " : "",
1218 tmp_orig, tmp_new);
1219 append_redir(cmd, (int)len, p_srr, tmp_diff);
1220 block_autocmds(); // avoid ShellCmdPost stuff
1221 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
1222 unblock_autocmds();
1223 vim_free(cmd);
1224 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001225}
1226
1227/*
1228 * Create a new version of a file from the current buffer and a diff file.
1229 * The buffer is written to a file, also for unmodified buffers (the file
1230 * could have been produced by autocommands, e.g. the netrw plugin).
1231 */
1232 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001233ex_diffpatch(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001234{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001235 char_u *tmp_orig; // name of original temp file
1236 char_u *tmp_new; // name of patched temp file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001237 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001238 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001239 win_T *old_curwin = curwin;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001240 char_u *newname = NULL; // name of patched file buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241#ifdef UNIX
1242 char_u dirbuf[MAXPATHL];
1243 char_u *fullname = NULL;
1244#endif
1245#ifdef FEAT_BROWSE
1246 char_u *browseFile = NULL;
Bram Moolenaare1004402020-10-24 20:49:43 +02001247 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02001249 stat_T st;
Bram Moolenaara95ab322017-03-11 19:21:53 +01001250 char_u *esc_name = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001251
1252#ifdef FEAT_BROWSE
Bram Moolenaare1004402020-10-24 20:49:43 +02001253 if (cmdmod.cmod_flags & CMOD_BROWSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001254 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +00001255 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaarc36651b2018-04-29 12:22:56 +02001256 eap->arg, NULL, NULL,
1257 (char_u *)_(BROWSE_FILTER_ALL_FILES), NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001258 if (browseFile == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001259 return; // operation cancelled
Bram Moolenaar071d4272004-06-13 20:20:40 +00001260 eap->arg = browseFile;
Bram Moolenaare1004402020-10-24 20:49:43 +02001261 cmdmod.cmod_flags &= ~CMOD_BROWSE; // don't let do_ecmd() browse again
Bram Moolenaar071d4272004-06-13 20:20:40 +00001262 }
1263#endif
1264
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001265 // We need two temp file names.
Bram Moolenaare5c421c2015-03-31 13:33:08 +02001266 tmp_orig = vim_tempname('o', FALSE);
1267 tmp_new = vim_tempname('n', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001268 if (tmp_orig == NULL || tmp_new == NULL)
1269 goto theend;
1270
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001271 // Write the current buffer to "tmp_orig".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001272 if (buf_write(curbuf, tmp_orig, NULL,
1273 (linenr_T)1, curbuf->b_ml.ml_line_count,
1274 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
1275 goto theend;
1276
1277#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001278 // Get the absolute path of the patchfile, changing directory below.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001279 fullname = FullName_save(eap->arg, FALSE);
1280#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001281 esc_name = vim_strsave_shellescape(
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282# ifdef UNIX
Bram Moolenaara95ab322017-03-11 19:21:53 +01001283 fullname != NULL ? fullname :
Bram Moolenaar071d4272004-06-13 20:20:40 +00001284# endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001285 eap->arg, TRUE, TRUE);
1286 if (esc_name == NULL)
1287 goto theend;
1288 buflen = STRLEN(tmp_orig) + STRLEN(esc_name) + STRLEN(tmp_new) + 16;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001289 buf = alloc(buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001290 if (buf == NULL)
1291 goto theend;
1292
1293#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001294 // Temporarily chdir to /tmp, to avoid patching files in the current
1295 // directory when the patch file contains more than one patch. When we
1296 // have our own temp dir use that instead, it will be cleaned up when we
1297 // exit (any .rej files created). Don't change directory if we can't
1298 // return to the current.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001299 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
1300 dirbuf[0] = NUL;
1301 else
1302 {
1303# ifdef TEMPDIRNAMES
1304 if (vim_tempdir != NULL)
Bram Moolenaar42335f52018-09-13 15:33:43 +02001305 vim_ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001306 else
1307# endif
Bram Moolenaar42335f52018-09-13 15:33:43 +02001308 vim_ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001309 shorten_fnames(TRUE);
1310 }
1311#endif
1312
1313#ifdef FEAT_EVAL
1314 if (*p_pex != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001315 // Use 'patchexpr' to generate the new file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001316 eval_patch(tmp_orig,
1317# ifdef UNIX
1318 fullname != NULL ? fullname :
1319# endif
1320 eap->arg, tmp_new);
1321 else
1322#endif
1323 {
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001324 if (check_restricted())
1325 goto theend;
1326
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001327 // Build the patch command and execute it. Ignore errors. Switch to
1328 // cooked mode to allow the user to respond to prompts.
Bram Moolenaara95ab322017-03-11 19:21:53 +01001329 vim_snprintf((char *)buf, buflen, "patch -o %s %s < %s",
1330 tmp_new, tmp_orig, esc_name);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001331 block_autocmds(); // Avoid ShellCmdPost stuff
Bram Moolenaar071d4272004-06-13 20:20:40 +00001332 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar78ab3312007-09-29 12:16:41 +00001333 unblock_autocmds();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001334 }
1335
1336#ifdef UNIX
1337 if (dirbuf[0] != NUL)
1338 {
1339 if (mch_chdir((char *)dirbuf) != 0)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001340 emsg(_(e_cannot_go_back_to_previous_directory));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001341 shorten_fnames(TRUE);
1342 }
1343#endif
1344
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001345 // patch probably has written over the screen
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001346 redraw_later(UPD_CLEAR);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001347
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001348 // Delete any .orig or .rej file created.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001349 STRCPY(buf, tmp_new);
1350 STRCAT(buf, ".orig");
1351 mch_remove(buf);
1352 STRCPY(buf, tmp_new);
1353 STRCAT(buf, ".rej");
1354 mch_remove(buf);
1355
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001356 // Only continue if the output file was created.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001357 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001358 emsg(_(e_cannot_read_patch_output));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001359 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001360 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001361 if (curbuf->b_fname != NULL)
1362 {
1363 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaardf44a272020-06-07 20:49:05 +02001364 STRLEN(curbuf->b_fname) + 4);
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001365 if (newname != NULL)
1366 STRCAT(newname, ".new");
1367 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001368
1369#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001370 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001371#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001372 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001373 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001374
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001375 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001376 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001377 // Pretend it was a ":split fname" command
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001378 eap->cmdidx = CMD_split;
1379 eap->arg = tmp_new;
1380 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001381
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001382 // check that split worked and editing tmp_new
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001383 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001384 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001385 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001386 diff_win_options(curwin, TRUE);
1387 diff_win_options(old_curwin, TRUE);
1388
1389 if (newname != NULL)
1390 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001391 // do a ":file filename.new" on the patched buffer
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001392 eap->arg = newname;
1393 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001394
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001395 // Do filetype detection with the new name.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001396 if (au_has_group((char_u *)"filetypedetect"))
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001397 do_cmdline_cmd(
1398 (char_u *)":doau filetypedetect BufRead");
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001399 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001400 }
1401 }
1402 }
1403
1404theend:
1405 if (tmp_orig != NULL)
1406 mch_remove(tmp_orig);
1407 vim_free(tmp_orig);
1408 if (tmp_new != NULL)
1409 mch_remove(tmp_new);
1410 vim_free(tmp_new);
1411 vim_free(newname);
1412 vim_free(buf);
1413#ifdef UNIX
1414 vim_free(fullname);
1415#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001416 vim_free(esc_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001417#ifdef FEAT_BROWSE
1418 vim_free(browseFile);
Bram Moolenaare1004402020-10-24 20:49:43 +02001419 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001420#endif
1421}
1422
1423/*
1424 * Split the window and edit another file, setting options to show the diffs.
1425 */
1426 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001427ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001428{
1429 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001430 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001431
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001432 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001433#ifdef FEAT_GUI
1434 need_mouse_correct = TRUE;
1435#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001436 // Need to compute w_fraction when no redraw happened yet.
Bram Moolenaar46328f92016-08-28 15:39:57 +02001437 validate_cursor();
1438 set_fraction(curwin);
1439
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001440 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001441 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001442
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001443 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) == FAIL)
1444 return;
1445
1446 // Pretend it was a ":split fname" command
1447 eap->cmdidx = CMD_split;
1448 curwin->w_p_diff = TRUE;
1449 do_exedit(eap, old_curwin);
1450
1451 if (curwin == old_curwin) // split didn't work
1452 return;
1453
1454 // Set 'diff', 'scrollbind' on and 'wrap' off.
1455 diff_win_options(curwin, TRUE);
1456 if (win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001457 {
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001458 diff_win_options(old_curwin, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001459
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001460 if (bufref_valid(&old_curbuf))
1461 // Move the cursor position to that of the old window.
1462 curwin->w_cursor.lnum = diff_get_corresponding_line(
1463 old_curbuf.br_buf, old_curwin->w_cursor.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001464 }
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001465 // Now that lines are folded scroll to show the cursor at the same
1466 // relative position.
1467 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001468}
1469
1470/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001471 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001472 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001473 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001474ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001476 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001477 diff_win_options(curwin, TRUE);
1478}
1479
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001480 static void
1481set_diff_option(win_T *wp, int value)
1482{
1483 win_T *old_curwin = curwin;
1484
1485 curwin = wp;
1486 curbuf = curwin->w_buffer;
1487 ++curbuf_lock;
Bram Moolenaar31e5c602022-04-15 13:53:33 +01001488 set_option_value_give_err((char_u *)"diff", (long)value, NULL, OPT_LOCAL);
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001489 --curbuf_lock;
1490 curwin = old_curwin;
1491 curbuf = curwin->w_buffer;
1492}
1493
Bram Moolenaar071d4272004-06-13 20:20:40 +00001494/*
1495 * Set options in window "wp" for diff mode.
1496 */
1497 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001498diff_win_options(
1499 win_T *wp,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001500 int addbuf) // Add buffer to diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001501{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001502# ifdef FEAT_FOLDING
1503 win_T *old_curwin = curwin;
1504
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001505 // close the manually opened folds
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001506 curwin = wp;
1507 newFoldLevel();
1508 curwin = old_curwin;
1509# endif
1510
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001511 // Use 'scrollbind' and 'cursorbind' when available
Bram Moolenaar43929962015-07-03 15:06:56 +02001512 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001513 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001514 wp->w_p_scb = TRUE;
Bram Moolenaar43929962015-07-03 15:06:56 +02001515 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001516 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001517 wp->w_p_crb = TRUE;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001518 if (!(diff_flags & DIFF_FOLLOWWRAP))
1519 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001520 if (!wp->w_p_diff)
Bram Moolenaar4223d432021-02-10 13:18:17 +01001521 wp->w_p_wrap_save = wp->w_p_wrap;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001522 wp->w_p_wrap = FALSE;
zeertzjq9e7f1fc2024-03-16 09:40:22 +01001523 wp->w_skipcol = 0;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001524 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525# ifdef FEAT_FOLDING
Bram Moolenaar43929962015-07-03 15:06:56 +02001526 if (!wp->w_p_diff)
1527 {
1528 if (wp->w_p_diff_saved)
1529 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001530 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001531 }
Bram Moolenaar20c023a2019-05-26 21:03:24 +02001532 set_string_option_direct_in_win(wp, (char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001533 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar43929962015-07-03 15:06:56 +02001534 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001535 {
1536 wp->w_p_fdc_save = wp->w_p_fdc;
1537 wp->w_p_fen_save = wp->w_p_fen;
1538 wp->w_p_fdl_save = wp->w_p_fdl;
1539 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001540 wp->w_p_fdc = diff_foldcolumn;
1541 wp->w_p_fen = TRUE;
1542 wp->w_p_fdl = 0;
1543 foldUpdateAll(wp);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001544 // make sure topline is not halfway a fold
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001545 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001546# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001547 if (vim_strchr(p_sbo, 'h') == NULL)
1548 do_cmdline_cmd((char_u *)"set sbo+=hor");
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001549 // Save the current values, to be restored in ex_diffoff().
Bram Moolenaara87aa802013-07-03 15:47:03 +02001550 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001552 set_diff_option(wp, TRUE);
Bram Moolenaar43929962015-07-03 15:06:56 +02001553
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554 if (addbuf)
1555 diff_buf_add(wp->w_buffer);
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001556 redraw_win_later(wp, UPD_NOT_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001557}
1558
1559/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001560 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001561 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001562 */
1563 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001564ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001565{
1566 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001567 int diffwin = FALSE;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001568
Bram Moolenaar29323592016-07-24 22:04:11 +02001569 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001570 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001571 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001572 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001573 // Set 'diff' off. If option values were saved in
1574 // diff_win_options(), restore the ones whose settings seem to have
1575 // been left over from diff mode.
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001576 set_diff_option(wp, FALSE);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001577
Bram Moolenaara87aa802013-07-03 15:47:03 +02001578 if (wp->w_p_diff_saved)
1579 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001580
Bram Moolenaar43929962015-07-03 15:06:56 +02001581 if (wp->w_p_scb)
1582 wp->w_p_scb = wp->w_p_scb_save;
Bram Moolenaar43929962015-07-03 15:06:56 +02001583 if (wp->w_p_crb)
1584 wp->w_p_crb = wp->w_p_crb_save;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001585 if (!(diff_flags & DIFF_FOLLOWWRAP))
1586 {
zeertzjq9e7f1fc2024-03-16 09:40:22 +01001587 if (!wp->w_p_wrap && wp->w_p_wrap_save)
1588 {
1589 wp->w_p_wrap = TRUE;
1590 wp->w_leftcol = 0;
1591 }
Bram Moolenaar4223d432021-02-10 13:18:17 +01001592 }
Bram Moolenaar43929962015-07-03 15:06:56 +02001593#ifdef FEAT_FOLDING
1594 free_string_option(wp->w_p_fdm);
Bram Moolenaar79a213d2017-05-16 13:15:18 +02001595 wp->w_p_fdm = vim_strsave(
1596 *wp->w_p_fdm_save ? wp->w_p_fdm_save : (char_u*)"manual");
Bram Moolenaar43929962015-07-03 15:06:56 +02001597
1598 if (wp->w_p_fdc == diff_foldcolumn)
1599 wp->w_p_fdc = wp->w_p_fdc_save;
1600 if (wp->w_p_fdl == 0)
1601 wp->w_p_fdl = wp->w_p_fdl_save;
1602
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001603 // Only restore 'foldenable' when 'foldmethod' is not
1604 // "manual", otherwise we continue to show the diff folds.
Bram Moolenaar43929962015-07-03 15:06:56 +02001605 if (wp->w_p_fen)
1606 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1607 : wp->w_p_fen_save;
1608
1609 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001610#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001611 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001612 // remove filler lines
Bram Moolenaare67d5462016-08-27 22:40:42 +02001613 wp->w_topfill = 0;
1614
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001615 // make sure topline is not halfway a fold and cursor is
1616 // invalidated
Bram Moolenaare67d5462016-08-27 22:40:42 +02001617 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001618
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001619 // Note: 'sbo' is not restored, it's a global option.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001620 diff_buf_adjust(wp);
1621 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001622 diffwin |= wp->w_p_diff;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001623 }
1624
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001625 // Also remove hidden buffers from the list.
Bram Moolenaar25ea0542017-02-03 23:16:28 +01001626 if (eap->forceit)
1627 diff_buf_clear();
1628
Bram Moolenaarc8234772019-11-10 21:00:27 +01001629 if (!diffwin)
1630 {
1631 diff_need_update = FALSE;
1632 curtab->tp_diff_invalid = FALSE;
1633 curtab->tp_diff_update = FALSE;
1634 diff_clear(curtab);
1635 }
1636
Dominique Pelleaf4a61a2021-12-27 17:21:41 +00001637 // Remove "hor" from 'scrollopt' if there are no diff windows left.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001638 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1639 do_cmdline_cmd((char_u *)"set sbo-=hor");
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001640}
1641
1642/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001643 * Read the diff output and add each entry to the diff list.
1644 */
1645 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001646diff_read(
Bram Moolenaare828b762018-09-10 17:51:58 +02001647 int idx_orig, // idx of original file
1648 int idx_new, // idx of new file
Lewis Russelld9da86e2021-12-28 13:54:41 +00001649 diffio_T *dio) // diff output
Bram Moolenaar071d4272004-06-13 20:20:40 +00001650{
Bram Moolenaare828b762018-09-10 17:51:58 +02001651 FILE *fd = NULL;
1652 int line_idx = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001653 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001654 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001655 diff_T *dn, *dpl;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001656 diffout_T *dout = &dio->dio_diff;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001657 char_u linebuf[LBUFLEN]; // only need to hold the diff line
Bram Moolenaare828b762018-09-10 17:51:58 +02001658 char_u *line;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001659 long off;
1660 int i;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001661 int notset = TRUE; // block "*dp" not set yet
Bram Moolenaar49166972021-12-30 10:51:45 +00001662 diffhunk_T *hunk = NULL; // init to avoid gcc warning
Lewis Russelld9da86e2021-12-28 13:54:41 +00001663
Bram Moolenaare828b762018-09-10 17:51:58 +02001664 enum {
1665 DIFF_ED,
1666 DIFF_UNIFIED,
1667 DIFF_NONE
1668 } diffstyle = DIFF_NONE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001669
Bram Moolenaare828b762018-09-10 17:51:58 +02001670 if (dout->dout_fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001671 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001672 diffstyle = DIFF_UNIFIED;
1673 }
1674 else
1675 {
1676 fd = mch_fopen((char *)dout->dout_fname, "r");
1677 if (fd == NULL)
1678 {
Bram Moolenaare1242042021-12-16 20:56:57 +00001679 emsg(_(e_cannot_read_diff_output));
Bram Moolenaare828b762018-09-10 17:51:58 +02001680 return;
1681 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001682 }
1683
Lewis Russelld9da86e2021-12-28 13:54:41 +00001684 if (!dio->dio_internal)
1685 {
1686 hunk = ALLOC_ONE(diffhunk_T);
1687 if (hunk == NULL)
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001688 {
1689 if (fd != NULL)
1690 fclose(fd);
Lewis Russelld9da86e2021-12-28 13:54:41 +00001691 return;
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001692 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00001693 }
1694
Bram Moolenaar071d4272004-06-13 20:20:40 +00001695 for (;;)
1696 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001697 if (dio->dio_internal)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001698 {
Bram Moolenaarebfec1c2023-01-22 21:14:53 +00001699 if (line_idx >= dout->dout_ga.ga_len)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001700 break; // did last line
Lewis Russelld9da86e2021-12-28 13:54:41 +00001701 hunk = ((diffhunk_T **)dout->dout_ga.ga_data)[line_idx++];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702 }
1703 else
1704 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001705 if (fd == NULL)
1706 {
1707 if (line_idx >= dout->dout_ga.ga_len)
1708 break; // did last line
1709 line = ((char_u **)dout->dout_ga.ga_data)[line_idx++];
1710 }
Bram Moolenaar3b8defd2018-09-13 13:03:11 +02001711 else
Lewis Russelld9da86e2021-12-28 13:54:41 +00001712 {
1713 if (vim_fgets(linebuf, LBUFLEN, fd))
1714 break; // end of file
1715 line = linebuf;
1716 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001717
Lewis Russelld9da86e2021-12-28 13:54:41 +00001718 if (diffstyle == DIFF_NONE)
1719 {
1720 // Determine diff style.
1721 // ed like diff looks like this:
1722 // {first}[,{last}]c{first}[,{last}]
1723 // {first}a{first}[,{last}]
1724 // {first}[,{last}]d{first}
1725 //
1726 // unified diff looks like this:
1727 // --- file1 2018-03-20 13:23:35.783153140 +0100
1728 // +++ file2 2018-03-20 13:23:41.183156066 +0100
1729 // @@ -1,3 +1,5 @@
Keith Thompson184f71c2024-01-04 21:19:04 +01001730 if (SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001731 diffstyle = DIFF_ED;
1732 else if ((STRNCMP(line, "@@ ", 3) == 0))
1733 diffstyle = DIFF_UNIFIED;
1734 else if ((STRNCMP(line, "--- ", 4) == 0)
1735 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1736 && (STRNCMP(line, "+++ ", 4) == 0)
1737 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1738 && (STRNCMP(line, "@@ ", 3) == 0))
1739 diffstyle = DIFF_UNIFIED;
1740 else
1741 // Format not recognized yet, skip over this line. Cygwin
1742 // diff may put a warning at the start of the file.
1743 continue;
1744 }
1745
1746 if (diffstyle == DIFF_ED)
1747 {
Keith Thompson184f71c2024-01-04 21:19:04 +01001748 if (!SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001749 continue; // not the start of a diff block
1750 if (parse_diff_ed(line, hunk) == FAIL)
1751 continue;
1752 }
1753 else if (diffstyle == DIFF_UNIFIED)
1754 {
1755 if (STRNCMP(line, "@@ ", 3) != 0)
1756 continue; // not the start of a diff block
1757 if (parse_diff_unified(line, hunk) == FAIL)
1758 continue;
1759 }
1760 else
1761 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001762 emsg(_(e_invalid_diff_format));
Lewis Russelld9da86e2021-12-28 13:54:41 +00001763 break;
1764 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001765 }
1766
1767 // Go over blocks before the change, for which orig and new are equal.
1768 // Copy blocks from orig to new.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001769 while (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001770 && hunk->lnum_orig > dp->df_lnum[idx_orig]
1771 + dp->df_count[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001772 {
1773 if (notset)
1774 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1775 dprev = dp;
1776 dp = dp->df_next;
1777 notset = TRUE;
1778 }
1779
1780 if (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001781 && hunk->lnum_orig <= dp->df_lnum[idx_orig]
1782 + dp->df_count[idx_orig]
1783 && hunk->lnum_orig + hunk->count_orig >= dp->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001784 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001785 // New block overlaps with existing block(s).
1786 // First find last block that overlaps.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001787 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001788 if (hunk->lnum_orig + hunk->count_orig
1789 < dpl->df_next->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 break;
1791
Bram Moolenaare828b762018-09-10 17:51:58 +02001792 // If the newly found block starts before the old one, set the
1793 // start back a number of lines.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001794 off = dp->df_lnum[idx_orig] - hunk->lnum_orig;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 if (off > 0)
1796 {
1797 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001798 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001799 dp->df_lnum[i] -= off;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001800 dp->df_lnum[idx_new] = hunk->lnum_new;
1801 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802 }
1803 else if (notset)
1804 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001805 // new block inside existing one, adjust new block
Lewis Russelld9da86e2021-12-28 13:54:41 +00001806 dp->df_lnum[idx_new] = hunk->lnum_new + off;
1807 dp->df_count[idx_new] = hunk->count_new - off;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001808 }
1809 else
Bram Moolenaare828b762018-09-10 17:51:58 +02001810 // second overlap of new block with existing block
Lewis Russelld9da86e2021-12-28 13:54:41 +00001811 dp->df_count[idx_new] += hunk->count_new - hunk->count_orig
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00001812 + dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]
1813 - (dp->df_lnum[idx_orig] + dp->df_count[idx_orig]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814
Bram Moolenaare828b762018-09-10 17:51:58 +02001815 // Adjust the size of the block to include all the lines to the
1816 // end of the existing block or the new diff, whatever ends last.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001817 off = (hunk->lnum_orig + hunk->count_orig)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001818 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1819 if (off < 0)
1820 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001821 // new change ends in existing block, adjust the end if not
1822 // done already
Bram Moolenaar071d4272004-06-13 20:20:40 +00001823 if (notset)
1824 dp->df_count[idx_new] += -off;
1825 off = 0;
1826 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001827 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001828 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001829 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1830 - dp->df_lnum[i] + off;
1831
Bram Moolenaare828b762018-09-10 17:51:58 +02001832 // Delete the diff blocks that have been merged into one.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 dn = dp->df_next;
1834 dp->df_next = dpl->df_next;
1835 while (dn != dp->df_next)
1836 {
1837 dpl = dn->df_next;
1838 vim_free(dn);
1839 dn = dpl;
1840 }
1841 }
1842 else
1843 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001844 // Allocate a new diffblock.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001845 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001846 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001847 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848
Lewis Russelld9da86e2021-12-28 13:54:41 +00001849 dp->df_lnum[idx_orig] = hunk->lnum_orig;
1850 dp->df_count[idx_orig] = hunk->count_orig;
1851 dp->df_lnum[idx_new] = hunk->lnum_new;
1852 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001853
Bram Moolenaare828b762018-09-10 17:51:58 +02001854 // Set values for other buffers, these must be equal to the
1855 // original buffer, otherwise there would have been a change
1856 // already.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001858 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859 diff_copy_entry(dprev, dp, idx_orig, i);
1860 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001861 notset = FALSE; // "*dp" has been set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001862 }
1863
Bram Moolenaare828b762018-09-10 17:51:58 +02001864 // for remaining diff blocks orig and new are equal
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865 while (dp != NULL)
1866 {
1867 if (notset)
1868 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1869 dprev = dp;
1870 dp = dp->df_next;
1871 notset = TRUE;
1872 }
1873
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001874done:
Lewis Russelld9da86e2021-12-28 13:54:41 +00001875 if (!dio->dio_internal)
1876 vim_free(hunk);
1877
Bram Moolenaare828b762018-09-10 17:51:58 +02001878 if (fd != NULL)
1879 fclose(fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001880}
1881
1882/*
1883 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1884 */
1885 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001886diff_copy_entry(
1887 diff_T *dprev,
1888 diff_T *dp,
1889 int idx_orig,
1890 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891{
1892 long off;
1893
1894 if (dprev == NULL)
1895 off = 0;
1896 else
1897 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1898 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1899 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1900 dp->df_count[idx_new] = dp->df_count[idx_orig];
1901}
1902
1903/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001904 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001905 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001906 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001907diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908{
1909 diff_T *p, *next_p;
1910
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001911 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001912 {
1913 next_p = p->df_next;
1914 vim_free(p);
1915 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001916 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001917}
1918
1919/*
1920 * Check diff status for line "lnum" in buffer "buf":
1921 * Returns 0 for nothing special
1922 * Returns -1 for a line that should be highlighted as changed.
1923 * Returns -2 for a line that should be highlighted as added/deleted.
1924 * Returns > 0 for inserting that many filler lines above it (never happens
1925 * when 'diffopt' doesn't contain "filler").
1926 * This should only be used for windows where 'diff' is set.
1927 */
1928 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001929diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001930{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001931 int idx; // index in tp_diffbuf[] for this buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 diff_T *dp;
1933 int maxcount;
1934 int i;
1935 buf_T *buf = wp->w_buffer;
1936 int cmp;
1937
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001938 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001939 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00001940
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001941 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) // no diffs at all
Bram Moolenaar071d4272004-06-13 20:20:40 +00001942 return 0;
1943
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001944 // safety check: "lnum" must be a buffer line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001945 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1946 return 0;
1947
1948 idx = diff_buf_idx(buf);
1949 if (idx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001950 return 0; // no diffs for buffer "buf"
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951
1952#ifdef FEAT_FOLDING
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001953 // A closed fold never has filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001954 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1955 return 0;
1956#endif
1957
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001958 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02001959 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1961 break;
1962 if (dp == NULL || lnum < dp->df_lnum[idx])
1963 return 0;
1964
1965 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1966 {
1967 int zero = FALSE;
1968
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001969 // Changed or inserted line. If the other buffers have a count of
1970 // zero, the lines were inserted. If the other buffers have the same
1971 // count, check if the lines are identical.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001972 cmp = FALSE;
1973 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001974 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001975 {
1976 if (dp->df_count[i] == 0)
1977 zero = TRUE;
1978 else
1979 {
1980 if (dp->df_count[i] != dp->df_count[idx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001981 return -1; // nr of lines changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001982 cmp = TRUE;
1983 }
1984 }
1985 if (cmp)
1986 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001987 // Compare all lines. If they are equal the lines were inserted
1988 // in some buffers, deleted in others, but not changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001989 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001990 if (i != idx && curtab->tp_diffbuf[i] != NULL
1991 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 if (!diff_equal_entry(dp, idx, i))
1993 return -1;
1994 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001995 // If there is no buffer with zero lines then there is no difference
1996 // any longer. Happens when making a change (or undo) that removes
1997 // the difference. Can't remove the entry here, we might be halfway
1998 // updating the window. Just report the text as unchanged. Other
1999 // windows might still show the change though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002000 if (zero == FALSE)
2001 return 0;
2002 return -2;
2003 }
2004
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002005 // If 'diffopt' doesn't contain "filler", return 0.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002006 if (!(diff_flags & DIFF_FILLER))
2007 return 0;
2008
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002009 // Insert filler lines above the line just below the change. Will return
2010 // 0 when this buf had the max count.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 maxcount = 0;
2012 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002013 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002014 maxcount = dp->df_count[i];
2015 return maxcount - dp->df_count[idx];
2016}
2017
2018/*
2019 * Compare two entries in diff "*dp" and return TRUE if they are equal.
2020 */
2021 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002022diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002023{
2024 int i;
2025 char_u *line;
2026 int cmp;
2027
2028 if (dp->df_count[idx1] != dp->df_count[idx2])
2029 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002030 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002031 return FALSE;
2032 for (i = 0; i < dp->df_count[idx1]; ++i)
2033 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002034 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 dp->df_lnum[idx1] + i, FALSE));
2036 if (line == NULL)
2037 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002038 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039 dp->df_lnum[idx2] + i, FALSE));
2040 vim_free(line);
2041 if (cmp != 0)
2042 return FALSE;
2043 }
2044 return TRUE;
2045}
2046
2047/*
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002048 * Compare the characters at "p1" and "p2". If they are equal (possibly
2049 * ignoring case) return TRUE and set "len" to the number of bytes.
2050 */
2051 static int
2052diff_equal_char(char_u *p1, char_u *p2, int *len)
2053{
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002054 int l = (*mb_ptr2len)(p1);
2055
2056 if (l != (*mb_ptr2len)(p2))
2057 return FALSE;
2058 if (l > 1)
2059 {
2060 if (STRNCMP(p1, p2, l) != 0
2061 && (!enc_utf8
2062 || !(diff_flags & DIFF_ICASE)
2063 || utf_fold(utf_ptr2char(p1))
2064 != utf_fold(utf_ptr2char(p2))))
2065 return FALSE;
2066 *len = l;
2067 }
2068 else
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002069 {
2070 if ((*p1 != *p2)
2071 && (!(diff_flags & DIFF_ICASE)
2072 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
2073 return FALSE;
2074 *len = 1;
2075 }
2076 return TRUE;
2077}
2078
2079/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002080 * Compare strings "s1" and "s2" according to 'diffopt'.
2081 * Return non-zero when they are different.
2082 */
2083 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002084diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002085{
2086 char_u *p1, *p2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002087 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002088
Bram Moolenaar785fc652018-09-15 19:17:38 +02002089 if ((diff_flags & DIFF_IBLANK)
2090 && (*skipwhite(s1) == NUL || *skipwhite(s2) == NUL))
2091 return 0;
2092
2093 if ((diff_flags & (DIFF_ICASE | ALL_WHITE_DIFF)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002094 return STRCMP(s1, s2);
Bram Moolenaar785fc652018-09-15 19:17:38 +02002095 if ((diff_flags & DIFF_ICASE) && !(diff_flags & ALL_WHITE_DIFF))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 return MB_STRICMP(s1, s2);
2097
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098 p1 = s1;
2099 p2 = s2;
Bram Moolenaar785fc652018-09-15 19:17:38 +02002100
2101 // Ignore white space changes and possibly ignore case.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 while (*p1 != NUL && *p2 != NUL)
2103 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002104 if (((diff_flags & DIFF_IWHITE)
2105 && VIM_ISWHITE(*p1) && VIM_ISWHITE(*p2))
2106 || ((diff_flags & DIFF_IWHITEALL)
2107 && (VIM_ISWHITE(*p1) || VIM_ISWHITE(*p2))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108 {
2109 p1 = skipwhite(p1);
2110 p2 = skipwhite(p2);
2111 }
2112 else
2113 {
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002114 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002115 break;
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002116 p1 += l;
2117 p2 += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002118 }
2119 }
2120
Bram Moolenaar785fc652018-09-15 19:17:38 +02002121 // Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002122 p1 = skipwhite(p1);
2123 p2 = skipwhite(p2);
2124 if (*p1 != NUL || *p2 != NUL)
2125 return 1;
2126 return 0;
2127}
2128
2129/*
2130 * Return the number of filler lines above "lnum".
2131 */
2132 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002133diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002134{
2135 int n;
2136
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002137 // be quick when there are no filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002138 if (!(diff_flags & DIFF_FILLER))
2139 return 0;
2140 n = diff_check(wp, lnum);
2141 if (n <= 0)
2142 return 0;
2143 return n;
2144}
2145
2146/*
2147 * Set the topline of "towin" to match the position in "fromwin", so that they
2148 * show the same diff'ed lines.
2149 */
2150 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002151diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002152{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002153 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002154 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002155 int fromidx;
2156 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002157 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002158 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002159 int i;
2160
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002161 fromidx = diff_buf_idx(frombuf);
2162 if (fromidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002163 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002164
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002165 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002166 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002167
2168 towin->w_topfill = 0;
2169
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002170 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002171 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002172 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173 break;
2174 if (dp == NULL)
2175 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002176 // After last change, compute topline relative to end of file; no
2177 // filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002178 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002179 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002180 }
2181 else
2182 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002183 // Find index for "towin".
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002184 toidx = diff_buf_idx(towin->w_buffer);
2185 if (toidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002186 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002187
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002188 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
2189 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002190 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002191 // Inside a change: compute filler lines. With three or more
2192 // buffers we need to know the largest count.
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002193 max_count = 0;
2194 for (i = 0; i < DB_COUNT; ++i)
2195 if (curtab->tp_diffbuf[i] != NULL
2196 && max_count < dp->df_count[i])
2197 max_count = dp->df_count[i];
2198
2199 if (dp->df_count[toidx] == dp->df_count[fromidx])
2200 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002201 // same number of lines: use same filler count
Bram Moolenaar071d4272004-06-13 20:20:40 +00002202 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002203 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002204 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002206 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002207 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002208 // more lines in towin and fromwin doesn't show diff
2209 // lines, only filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002210 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
2211 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002212 // towin also only shows filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002213 towin->w_topline = dp->df_lnum[toidx]
2214 + dp->df_count[toidx];
2215 towin->w_topfill = fromwin->w_topfill;
2216 }
2217 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002218 // towin still has some diff lines to show
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002219 towin->w_topline = dp->df_lnum[toidx]
2220 + max_count - fromwin->w_topfill;
2221 }
2222 }
2223 else if (towin->w_topline >= dp->df_lnum[toidx]
2224 + dp->df_count[toidx])
2225 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002226 // less lines in towin and no diff lines to show: compute
2227 // filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002228 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
2229 if (diff_flags & DIFF_FILLER)
2230 {
2231 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002232 // fromwin is also out of diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002233 towin->w_topfill = fromwin->w_topfill;
2234 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002235 // fromwin has some diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002236 towin->w_topfill = dp->df_lnum[fromidx]
2237 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002238 }
2239 }
2240 }
2241 }
2242
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002243 // safety check (if diff info gets outdated strange things may happen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244 towin->w_botfill = FALSE;
2245 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
2246 {
2247 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
2248 towin->w_botfill = TRUE;
2249 }
2250 if (towin->w_topline < 1)
2251 {
2252 towin->w_topline = 1;
2253 towin->w_topfill = 0;
2254 }
2255
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002256 // When w_topline changes need to recompute w_botline and cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00002257 invalidate_botline_win(towin);
2258 changed_line_abv_curs_win(towin);
2259
2260 check_topfill(towin, FALSE);
2261#ifdef FEAT_FOLDING
2262 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
2263 NULL, TRUE, NULL);
2264#endif
2265}
2266
2267/*
2268 * This is called when 'diffopt' is changed.
2269 */
2270 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002271diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002272{
2273 char_u *p;
2274 int diff_context_new = 6;
2275 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002276 int diff_foldcolumn_new = 2;
Bram Moolenaare828b762018-09-10 17:51:58 +02002277 long diff_algorithm_new = 0;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002278 long diff_indent_heuristic = 0;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002279 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280
2281 p = p_dip;
2282 while (*p != NUL)
2283 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002284 // Note: Keep this in sync with p_dip_values
Bram Moolenaar071d4272004-06-13 20:20:40 +00002285 if (STRNCMP(p, "filler", 6) == 0)
2286 {
2287 p += 6;
2288 diff_flags_new |= DIFF_FILLER;
2289 }
2290 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
2291 {
2292 p += 8;
2293 diff_context_new = getdigits(&p);
2294 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002295 else if (STRNCMP(p, "iblank", 6) == 0)
2296 {
2297 p += 6;
2298 diff_flags_new |= DIFF_IBLANK;
2299 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002300 else if (STRNCMP(p, "icase", 5) == 0)
2301 {
2302 p += 5;
2303 diff_flags_new |= DIFF_ICASE;
2304 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002305 else if (STRNCMP(p, "iwhiteall", 9) == 0)
2306 {
2307 p += 9;
2308 diff_flags_new |= DIFF_IWHITEALL;
2309 }
2310 else if (STRNCMP(p, "iwhiteeol", 9) == 0)
2311 {
2312 p += 9;
2313 diff_flags_new |= DIFF_IWHITEEOL;
2314 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002315 else if (STRNCMP(p, "iwhite", 6) == 0)
2316 {
2317 p += 6;
2318 diff_flags_new |= DIFF_IWHITE;
2319 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002320 else if (STRNCMP(p, "horizontal", 10) == 0)
2321 {
2322 p += 10;
2323 diff_flags_new |= DIFF_HORIZONTAL;
2324 }
2325 else if (STRNCMP(p, "vertical", 8) == 0)
2326 {
2327 p += 8;
2328 diff_flags_new |= DIFF_VERTICAL;
2329 }
2330 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
2331 {
2332 p += 11;
2333 diff_foldcolumn_new = getdigits(&p);
2334 }
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002335 else if (STRNCMP(p, "hiddenoff", 9) == 0)
2336 {
2337 p += 9;
2338 diff_flags_new |= DIFF_HIDDEN_OFF;
2339 }
Bram Moolenaarc8234772019-11-10 21:00:27 +01002340 else if (STRNCMP(p, "closeoff", 8) == 0)
2341 {
2342 p += 8;
2343 diff_flags_new |= DIFF_CLOSE_OFF;
2344 }
Bram Moolenaar4223d432021-02-10 13:18:17 +01002345 else if (STRNCMP(p, "followwrap", 10) == 0)
2346 {
2347 p += 10;
2348 diff_flags_new |= DIFF_FOLLOWWRAP;
2349 }
Bram Moolenaare828b762018-09-10 17:51:58 +02002350 else if (STRNCMP(p, "indent-heuristic", 16) == 0)
2351 {
2352 p += 16;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002353 diff_indent_heuristic = XDF_INDENT_HEURISTIC;
Bram Moolenaare828b762018-09-10 17:51:58 +02002354 }
2355 else if (STRNCMP(p, "internal", 8) == 0)
2356 {
2357 p += 8;
2358 diff_flags_new |= DIFF_INTERNAL;
2359 }
2360 else if (STRNCMP(p, "algorithm:", 10) == 0)
2361 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002362 // Note: Keep this in sync with p_dip_algorithm_values.
Bram Moolenaare828b762018-09-10 17:51:58 +02002363 p += 10;
2364 if (STRNCMP(p, "myers", 5) == 0)
2365 {
2366 p += 5;
2367 diff_algorithm_new = 0;
2368 }
2369 else if (STRNCMP(p, "minimal", 7) == 0)
2370 {
2371 p += 7;
2372 diff_algorithm_new = XDF_NEED_MINIMAL;
2373 }
2374 else if (STRNCMP(p, "patience", 8) == 0)
2375 {
2376 p += 8;
2377 diff_algorithm_new = XDF_PATIENCE_DIFF;
2378 }
2379 else if (STRNCMP(p, "histogram", 9) == 0)
2380 {
2381 p += 9;
2382 diff_algorithm_new = XDF_HISTOGRAM_DIFF;
2383 }
Bram Moolenaard0721052018-11-05 21:21:33 +01002384 else
2385 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02002386 }
2387
Bram Moolenaar071d4272004-06-13 20:20:40 +00002388 if (*p != ',' && *p != NUL)
2389 return FAIL;
2390 if (*p == ',')
2391 ++p;
2392 }
2393
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002394 diff_algorithm_new |= diff_indent_heuristic;
2395
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002396 // Can't have both "horizontal" and "vertical".
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002397 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
2398 return FAIL;
2399
Bram Moolenaar198fa062018-09-18 21:20:26 +02002400 // If flags were added or removed, or the algorithm was changed, need to
2401 // update the diff.
Bram Moolenaare828b762018-09-10 17:51:58 +02002402 if (diff_flags != diff_flags_new || diff_algorithm != diff_algorithm_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02002403 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002404 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002405
2406 diff_flags = diff_flags_new;
Bram Moolenaarb9ddda62019-02-19 23:00:50 +01002407 diff_context = diff_context_new == 0 ? 1 : diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002408 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaare828b762018-09-10 17:51:58 +02002409 diff_algorithm = diff_algorithm_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002410
2411 diff_redraw(TRUE);
2412
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002413 // recompute the scroll binding with the new option value, may
2414 // remove or add filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002415 check_scrollbind((linenr_T)0, 0L);
2416
2417 return OK;
2418}
2419
2420/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002421 * Return TRUE if 'diffopt' contains "horizontal".
2422 */
2423 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002424diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002425{
2426 return (diff_flags & DIFF_HORIZONTAL) != 0;
2427}
2428
2429/*
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002430 * Return TRUE if 'diffopt' contains "hiddenoff".
2431 */
2432 int
2433diffopt_hiddenoff(void)
2434{
2435 return (diff_flags & DIFF_HIDDEN_OFF) != 0;
2436}
2437
2438/*
Bram Moolenaarc8234772019-11-10 21:00:27 +01002439 * Return TRUE if 'diffopt' contains "closeoff".
2440 */
2441 int
2442diffopt_closeoff(void)
2443{
2444 return (diff_flags & DIFF_CLOSE_OFF) != 0;
2445}
2446
2447/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448 * Find the difference within a changed line.
2449 * Returns TRUE if the line was added, no other buffer has it.
2450 */
2451 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002452diff_find_change(
2453 win_T *wp,
2454 linenr_T lnum,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002455 int *startp, // first char of the change
2456 int *endp) // last char of the change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002457{
2458 char_u *line_org;
2459 char_u *line_new;
2460 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002461 int si_org, si_new;
2462 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002463 diff_T *dp;
2464 int idx;
2465 int off;
2466 int added = TRUE;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002467 char_u *p1, *p2;
2468 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002470 // Make a copy of the line, the next ml_get() will invalidate it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002471 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
2472 if (line_org == NULL)
2473 return FALSE;
2474
2475 idx = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002476 if (idx == DB_COUNT) // cannot happen
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002477 {
2478 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002480 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002481
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002482 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002483 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2485 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002486 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002487 {
2488 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002490 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491
2492 off = lnum - dp->df_lnum[idx];
2493
2494 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002495 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002496 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002497 // Skip lines that are not in the other change (filler lines).
Bram Moolenaar071d4272004-06-13 20:20:40 +00002498 if (off >= dp->df_count[i])
2499 continue;
2500 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002501 line_new = ml_get_buf(curtab->tp_diffbuf[i],
2502 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002503
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002504 // Search for start of difference
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002505 si_org = si_new = 0;
2506 while (line_org[si_org] != NUL)
2507 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002508 if (((diff_flags & DIFF_IWHITE)
2509 && VIM_ISWHITE(line_org[si_org])
2510 && VIM_ISWHITE(line_new[si_new]))
2511 || ((diff_flags & DIFF_IWHITEALL)
2512 && (VIM_ISWHITE(line_org[si_org])
2513 || VIM_ISWHITE(line_new[si_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002514 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002515 si_org = (int)(skipwhite(line_org + si_org) - line_org);
2516 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002517 }
2518 else
2519 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002520 if (!diff_equal_char(line_org + si_org, line_new + si_new,
2521 &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002522 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002523 si_org += l;
2524 si_new += l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002525 }
2526 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002527 if (has_mbyte)
2528 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002529 // Move back to first byte of character in both lines (may
2530 // have "nn^" in line_org and "n^ in line_new).
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002531 si_org -= (*mb_head_off)(line_org, line_org + si_org);
2532 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002533 }
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002534 if (*startp > si_org)
2535 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002536
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002537 // Search for end of difference, if any.
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002538 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002539 {
2540 ei_org = (int)STRLEN(line_org);
2541 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002542 while (ei_org >= *startp && ei_new >= si_new
2543 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002544 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002545 if (((diff_flags & DIFF_IWHITE)
2546 && VIM_ISWHITE(line_org[ei_org])
2547 && VIM_ISWHITE(line_new[ei_new]))
2548 || ((diff_flags & DIFF_IWHITEALL)
2549 && (VIM_ISWHITE(line_org[ei_org])
2550 || VIM_ISWHITE(line_new[ei_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002551 {
2552 while (ei_org >= *startp
Bram Moolenaar1c465442017-03-12 20:10:05 +01002553 && VIM_ISWHITE(line_org[ei_org]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002554 --ei_org;
2555 while (ei_new >= si_new
Bram Moolenaar1c465442017-03-12 20:10:05 +01002556 && VIM_ISWHITE(line_new[ei_new]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002557 --ei_new;
2558 }
2559 else
2560 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002561 p1 = line_org + ei_org;
2562 p2 = line_new + ei_new;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002563 p1 -= (*mb_head_off)(line_org, p1);
2564 p2 -= (*mb_head_off)(line_new, p2);
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002565 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002566 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002567 ei_org -= l;
2568 ei_new -= l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002569 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002570 }
2571 if (*endp < ei_org)
2572 *endp = ei_org;
2573 }
2574 }
2575
2576 vim_free(line_org);
2577 return added;
2578}
2579
2580#if defined(FEAT_FOLDING) || defined(PROTO)
2581/*
2582 * Return TRUE if line "lnum" is not close to a diff block, this line should
2583 * be in a fold.
2584 * Return FALSE if there are no diff blocks at all in this window.
2585 */
2586 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002587diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002588{
2589 int i;
2590 int idx = -1;
2591 int other = FALSE;
2592 diff_T *dp;
2593
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002594 // Return if 'diff' isn't set.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002595 if (!wp->w_p_diff)
2596 return FALSE;
2597
2598 for (i = 0; i < DB_COUNT; ++i)
2599 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002600 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002601 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002602 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002603 other = TRUE;
2604 }
2605
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002606 // return here if there are no diffs in the window
Bram Moolenaar071d4272004-06-13 20:20:40 +00002607 if (idx == -1 || !other)
2608 return FALSE;
2609
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002610 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002611 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002612
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002613 // Return if there are no diff blocks. All lines will be folded.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002614 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002615 return TRUE;
2616
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002617 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002618 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002619 // If this change is below the line there can't be any further match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002620 if (dp->df_lnum[idx] - diff_context > lnum)
2621 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002622 // If this change ends before the line we have a match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002623 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2624 return FALSE;
2625 }
2626 return TRUE;
2627}
2628#endif
2629
2630/*
2631 * "dp" and "do" commands.
2632 */
2633 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002634nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002635{
2636 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002637 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002638
Bram Moolenaarf2732452018-06-03 14:47:35 +02002639#ifdef FEAT_JOB_CHANNEL
2640 if (bt_prompt(curbuf))
2641 {
2642 vim_beep(BO_OPER);
2643 return;
2644 }
2645#endif
Bram Moolenaar6a643652014-10-31 13:54:25 +01002646 if (count == 0)
2647 ea.arg = (char_u *)"";
2648 else
2649 {
2650 vim_snprintf((char *)buf, 30, "%ld", count);
2651 ea.arg = buf;
2652 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002653 if (put)
2654 ea.cmdidx = CMD_diffput;
2655 else
2656 ea.cmdidx = CMD_diffget;
2657 ea.addr_count = 0;
2658 ea.line1 = curwin->w_cursor.lnum;
2659 ea.line2 = curwin->w_cursor.lnum;
2660 ex_diffgetput(&ea);
2661}
2662
2663/*
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002664 * Return TRUE if "diff" appears in the list of diff blocks of the current tab.
2665 */
2666 static int
2667valid_diff(diff_T *diff)
2668{
2669 diff_T *dp;
2670
Yegappan Lakshmanan14113fd2023-03-07 17:13:51 +00002671 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002672 if (dp == diff)
2673 return TRUE;
2674 return FALSE;
2675}
2676
2677/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002678 * ":diffget"
2679 * ":diffput"
2680 */
2681 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002682ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002683{
2684 linenr_T lnum;
2685 int count;
2686 linenr_T off = 0;
2687 diff_T *dp;
2688 diff_T *dprev;
2689 diff_T *dfree;
2690 int idx_cur;
2691 int idx_other;
2692 int idx_from;
2693 int idx_to;
2694 int i;
2695 int added;
2696 char_u *p;
2697 aco_save_T aco;
2698 buf_T *buf;
2699 int start_skip, end_skip;
2700 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002701 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002702 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002703
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002704 // Find the current buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705 idx_cur = diff_buf_idx(curbuf);
2706 if (idx_cur == DB_COUNT)
2707 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002708 emsg(_(e_current_buffer_is_not_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709 return;
2710 }
2711
2712 if (*eap->arg == NUL)
2713 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002714 // No argument: Find the other buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002715 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002716 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002717 && curtab->tp_diffbuf[idx_other] != NULL)
2718 {
2719 if (eap->cmdidx != CMD_diffput
2720 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2721 break;
2722 found_not_ma = TRUE;
2723 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002724 if (idx_other == DB_COUNT)
2725 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002726 if (found_not_ma)
Bram Moolenaar677658a2022-01-05 16:09:06 +00002727 emsg(_(e_no_other_buffer_in_diff_mode_is_modifiable));
Bram Moolenaar602eb742007-02-20 03:43:38 +00002728 else
Bram Moolenaare1242042021-12-16 20:56:57 +00002729 emsg(_(e_no_other_buffer_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 return;
2731 }
2732
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002733 // Check that there isn't a third buffer in the list
Bram Moolenaar071d4272004-06-13 20:20:40 +00002734 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002735 if (curtab->tp_diffbuf[i] != curbuf
2736 && curtab->tp_diffbuf[i] != NULL
2737 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002738 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002739 emsg(_(e_more_than_two_buffers_in_diff_mode_dont_know_which_one_to_use));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002740 return;
2741 }
2742 }
2743 else
2744 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002745 // Buffer number or pattern given. Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002746 p = eap->arg + STRLEN(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01002747 while (p > eap->arg && VIM_ISWHITE(p[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002748 --p;
2749 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2750 ;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002751 if (eap->arg + i == p) // digits only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002752 i = atol((char *)eap->arg);
2753 else
2754 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002755 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002756 if (i < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002757 return; // error message already given
Bram Moolenaar071d4272004-06-13 20:20:40 +00002758 }
2759 buf = buflist_findnr(i);
2760 if (buf == NULL)
2761 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002762 semsg(_(e_cant_find_buffer_str), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002763 return;
2764 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002765 if (buf == curbuf)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002766 return; // nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +00002767 idx_other = diff_buf_idx(buf);
2768 if (idx_other == DB_COUNT)
2769 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002770 semsg(_(e_buffer_str_is_not_in_diff_mode), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002771 return;
2772 }
2773 }
2774
2775 diff_busy = TRUE;
2776
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002777 // When no range given include the line above or below the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778 if (eap->addr_count == 0)
2779 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002780 // Make it possible that ":diffget" on the last line gets line below
2781 // the cursor line when there is no difference above the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002782 if (eap->cmdidx == CMD_diffget
2783 && eap->line1 == curbuf->b_ml.ml_line_count
2784 && diff_check(curwin, eap->line1) == 0
2785 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2786 ++eap->line2;
2787 else if (eap->line1 > 0)
2788 --eap->line1;
2789 }
2790
2791 if (eap->cmdidx == CMD_diffget)
2792 {
2793 idx_from = idx_other;
2794 idx_to = idx_cur;
2795 }
2796 else
2797 {
2798 idx_from = idx_cur;
2799 idx_to = idx_other;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002800 // Need to make the other buffer the current buffer to be able to make
2801 // changes in it.
Bram Moolenaare76062c2022-11-28 18:51:43 +00002802 // Set curwin/curbuf to buf and save a few things.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002803 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaare76062c2022-11-28 18:51:43 +00002804 if (curbuf != curtab->tp_diffbuf[idx_other])
2805 // Could not find a window for this buffer, the rest is likely to
2806 // fail.
2807 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002808 }
2809
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002810 // May give the warning for a changed buffer here, which can trigger the
2811 // FileChangedRO autocommand, which may do nasty things and mess
2812 // everything up.
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002813 if (!curbuf->b_changed)
2814 {
2815 change_warning(0);
2816 if (diff_buf_idx(curbuf) != idx_to)
2817 {
Bram Moolenaar677658a2022-01-05 16:09:06 +00002818 emsg(_(e_buffer_changed_unexpectedly));
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002819 goto theend;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002820 }
2821 }
2822
Bram Moolenaar071d4272004-06-13 20:20:40 +00002823 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002824 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002825 {
2826 if (dp->df_lnum[idx_cur] > eap->line2 + off)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002827 break; // past the range that was specified
Bram Moolenaar071d4272004-06-13 20:20:40 +00002828
2829 dfree = NULL;
2830 lnum = dp->df_lnum[idx_to];
2831 count = dp->df_count[idx_to];
2832 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2833 && u_save(lnum - 1, lnum + count) != FAIL)
2834 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002835 // Inside the specified range and saving for undo worked.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836 start_skip = 0;
2837 end_skip = 0;
2838 if (eap->addr_count > 0)
2839 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002840 // A range was specified: check if lines need to be skipped.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002841 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2842 if (start_skip > 0)
2843 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002844 // range starts below start of current diff block
Bram Moolenaar071d4272004-06-13 20:20:40 +00002845 if (start_skip > count)
2846 {
2847 lnum += count;
2848 count = 0;
2849 }
2850 else
2851 {
2852 count -= start_skip;
2853 lnum += start_skip;
2854 }
2855 }
2856 else
2857 start_skip = 0;
2858
2859 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2860 - (eap->line2 + off);
2861 if (end_skip > 0)
2862 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002863 // range ends above end of current/from diff block
2864 if (idx_cur == idx_from) // :diffput
Bram Moolenaar071d4272004-06-13 20:20:40 +00002865 {
2866 i = dp->df_count[idx_cur] - start_skip - end_skip;
2867 if (count > i)
2868 count = i;
2869 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002870 else // :diffget
Bram Moolenaar071d4272004-06-13 20:20:40 +00002871 {
2872 count -= end_skip;
2873 end_skip = dp->df_count[idx_from] - start_skip - count;
2874 if (end_skip < 0)
2875 end_skip = 0;
2876 }
2877 }
2878 else
2879 end_skip = 0;
2880 }
2881
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002882 buf_empty = BUFEMPTY();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002883 added = 0;
2884 for (i = 0; i < count; ++i)
2885 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002886 // remember deleting the last line of the buffer
Bram Moolenaar280f1262006-01-30 00:14:18 +00002887 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar4e677b92022-07-28 18:44:27 +01002888 if (ml_delete(lnum) == OK)
2889 --added;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002890 }
2891 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2892 {
2893 linenr_T nr;
2894
2895 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002896 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002897 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002898 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2899 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 if (p != NULL)
2901 {
2902 ml_append(lnum + i - 1, p, 0, FALSE);
2903 vim_free(p);
2904 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002905 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2906 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002907 // Added the first line into an empty buffer, need to
2908 // delete the dummy empty line.
Bram Moolenaar280f1262006-01-30 00:14:18 +00002909 buf_empty = FALSE;
Bram Moolenaarca70c072020-05-30 20:30:46 +02002910 ml_delete((linenr_T)2);
Bram Moolenaar280f1262006-01-30 00:14:18 +00002911 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002912 }
2913 }
2914 new_count = dp->df_count[idx_to] + added;
2915 dp->df_count[idx_to] = new_count;
2916
2917 if (start_skip == 0 && end_skip == 0)
2918 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002919 // Check if there are any other buffers and if the diff is
2920 // equal in them.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002921 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002922 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2923 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002924 && !diff_equal_entry(dp, idx_from, i))
2925 break;
2926 if (i == DB_COUNT)
2927 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002928 // delete the diff entry, the buffers are now equal here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002929 dfree = dp;
2930 dp = dp->df_next;
2931 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002932 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002933 else
2934 dprev->df_next = dp;
2935 }
2936 }
2937
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938 if (added != 0)
2939 {
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002940 // Adjust marks. This will change the following entries!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002941 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2942 if (curwin->w_cursor.lnum >= lnum)
2943 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002944 // Adjust the cursor position if it's in/after the changed
2945 // lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002946 if (curwin->w_cursor.lnum >= lnum + count)
2947 curwin->w_cursor.lnum += added;
2948 else if (added < 0)
2949 curwin->w_cursor.lnum = lnum;
2950 }
2951 }
2952 changed_lines(lnum, 0, lnum + count, (long)added);
2953
2954 if (dfree != NULL)
2955 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002956 // Diff is deleted, update folds in other windows.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002957#ifdef FEAT_FOLDING
2958 diff_fold_update(dfree, idx_to);
2959#endif
2960 vim_free(dfree);
2961 }
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002962
2963 // mark_adjust() may have made "dp" invalid. We don't know where
2964 // to continue then, bail out.
2965 if (added != 0 && !valid_diff(dp))
2966 break;
2967
2968 if (dfree == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002969 // mark_adjust() may have changed the count in a wrong way
Bram Moolenaar071d4272004-06-13 20:20:40 +00002970 dp->df_count[idx_to] = new_count;
2971
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002972 // When changing the current buffer, keep track of line numbers
Bram Moolenaar071d4272004-06-13 20:20:40 +00002973 if (idx_cur == idx_to)
2974 off += added;
2975 }
2976
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002977 // If before the range or not deleted, go to next diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002978 if (dfree == NULL)
2979 {
2980 dprev = dp;
2981 dp = dp->df_next;
2982 }
2983 }
2984
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002985 // restore curwin/curbuf and a few other things
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002986 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002988 // Syncing undo only works for the current buffer, but we change
2989 // another buffer. Sync undo if the command was typed. This isn't
2990 // 100% right when ":diffput" is used in a function or mapping.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002991 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002992 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002993 aucmd_restbuf(&aco);
2994 }
2995
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002996theend:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002997 diff_busy = FALSE;
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002998 if (diff_need_update)
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002999 ex_diffupdate(NULL);
Bram Moolenaardf77cef2018-10-07 17:46:42 +02003000
Bram Moolenaar5f57bdc2018-10-25 17:52:23 +02003001 // Check that the cursor is on a valid character and update its
Bram Moolenaardf77cef2018-10-07 17:46:42 +02003002 // position. When there were filler lines the topline has become
3003 // invalid.
3004 check_cursor();
3005 changed_line_abv_curs();
3006
3007 if (diff_need_update)
3008 // redraw already done by ex_diffupdate()
3009 diff_need_update = FALSE;
Bram Moolenaar198fa062018-09-18 21:20:26 +02003010 else
3011 {
Bram Moolenaar198fa062018-09-18 21:20:26 +02003012 // Also need to redraw the other buffers.
3013 diff_redraw(FALSE);
3014 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
3015 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003016}
3017
3018#ifdef FEAT_FOLDING
3019/*
3020 * Update folds for all diff buffers for entry "dp".
3021 * Skip buffer with index "skip_idx".
3022 * When there are no diffs, all folds are removed.
3023 */
3024 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01003025diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003026{
3027 int i;
3028 win_T *wp;
3029
Bram Moolenaar29323592016-07-24 22:04:11 +02003030 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003031 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003032 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003033 foldUpdate(wp, dp->df_lnum[i],
3034 dp->df_lnum[i] + dp->df_count[i]);
3035}
3036#endif
3037
3038/*
3039 * Return TRUE if buffer "buf" is in diff-mode.
3040 */
3041 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003042diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003043{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003044 tabpage_T *tp;
3045
Bram Moolenaar29323592016-07-24 22:04:11 +02003046 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003047 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
3048 return TRUE;
3049 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003050}
3051
3052/*
3053 * Move "count" times in direction "dir" to the next diff block.
3054 * Return FAIL if there isn't such a diff block.
3055 */
3056 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003057diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058{
3059 int idx;
3060 linenr_T lnum = curwin->w_cursor.lnum;
3061 diff_T *dp;
3062
3063 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003064 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003065 return FAIL;
3066
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003067 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003068 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003069
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003070 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar071d4272004-06-13 20:20:40 +00003071 return FAIL;
3072
3073 while (--count >= 0)
3074 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003075 // Check if already before first diff.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003076 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077 break;
3078
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003079 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003080 {
3081 if (dp == NULL)
3082 break;
3083 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
3084 || (dir == BACKWARD
3085 && (dp->df_next == NULL
3086 || lnum <= dp->df_next->df_lnum[idx])))
3087 {
3088 lnum = dp->df_lnum[idx];
3089 break;
3090 }
3091 }
3092 }
3093
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003094 // don't end up past the end of the file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003095 if (lnum > curbuf->b_ml.ml_line_count)
3096 lnum = curbuf->b_ml.ml_line_count;
3097
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003098 // When the cursor didn't move at all we fail.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003099 if (lnum == curwin->w_cursor.lnum)
3100 return FAIL;
3101
3102 setpcmark();
3103 curwin->w_cursor.lnum = lnum;
3104 curwin->w_cursor.col = 0;
3105
3106 return OK;
3107}
3108
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003109/*
3110 * Return the line number in the current window that is closest to "lnum1" in
3111 * "buf1" in diff mode.
3112 */
3113 static linenr_T
3114diff_get_corresponding_line_int(
Bram Moolenaar7454a062016-01-30 15:14:10 +01003115 buf_T *buf1,
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003116 linenr_T lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003117{
3118 int idx1;
3119 int idx2;
3120 diff_T *dp;
3121 int baseline = 0;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003122
3123 idx1 = diff_buf_idx(buf1);
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003124 idx2 = diff_buf_idx(curbuf);
Bram Moolenaar860cae12010-06-05 23:22:07 +02003125 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
3126 return lnum1;
3127
3128 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003129 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar860cae12010-06-05 23:22:07 +02003130
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003131 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar860cae12010-06-05 23:22:07 +02003132 return lnum1;
3133
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003134 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003135 {
3136 if (dp->df_lnum[idx1] > lnum1)
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003137 return lnum1 - baseline;
3138 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003139 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003140 // Inside the diffblock
Bram Moolenaar860cae12010-06-05 23:22:07 +02003141 baseline = lnum1 - dp->df_lnum[idx1];
3142 if (baseline > dp->df_count[idx2])
3143 baseline = dp->df_count[idx2];
3144
3145 return dp->df_lnum[idx2] + baseline;
3146 }
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003147 if ( (dp->df_lnum[idx1] == lnum1)
3148 && (dp->df_count[idx1] == 0)
3149 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum)
3150 && ((dp->df_lnum[idx2] + dp->df_count[idx2])
3151 > curwin->w_cursor.lnum))
Bram Moolenaar860cae12010-06-05 23:22:07 +02003152 /*
3153 * Special case: if the cursor is just after a zero-count
3154 * block (i.e. all filler) and the target cursor is already
3155 * inside the corresponding block, leave the target cursor
3156 * unmoved. This makes repeated CTRL-W W operations work
3157 * as expected.
3158 */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003159 return curwin->w_cursor.lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003160 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
3161 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
3162 }
3163
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003164 // If we get here then the cursor is after the last diff
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003165 return lnum1 - baseline;
3166}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003167
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003168/*
3169 * Return the line number in the current window that is closest to "lnum1" in
3170 * "buf1" in diff mode. Checks the line number to be valid.
3171 */
3172 linenr_T
3173diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1)
3174{
3175 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1);
3176
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003177 // don't end up past the end of the file
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003178 if (lnum > curbuf->b_ml.ml_line_count)
3179 return curbuf->b_ml.ml_line_count;
3180 return lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003181}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003182
Bram Moolenaar071d4272004-06-13 20:20:40 +00003183/*
3184 * For line "lnum" in the current window find the equivalent lnum in window
3185 * "wp", compensating for inserted/deleted lines.
3186 */
3187 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01003188diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003189{
3190 diff_T *dp;
3191 int idx;
3192 int i;
3193 linenr_T n;
3194
3195 idx = diff_buf_idx(curbuf);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003196 if (idx == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197 return (linenr_T)0;
3198
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003199 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003200 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003201
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003202 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003203 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
3205 break;
3206
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003207 // When after the last change, compute relative to the last line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003208 if (dp == NULL)
3209 return wp->w_buffer->b_ml.ml_line_count
3210 - (curbuf->b_ml.ml_line_count - lnum);
3211
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003212 // Find index for "wp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003213 i = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003214 if (i == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003215 return (linenr_T)0;
3216
3217 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
3218 if (n > dp->df_lnum[i] + dp->df_count[i])
3219 n = dp->df_lnum[i] + dp->df_count[i];
3220 return n;
3221}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003222
Bram Moolenaare828b762018-09-10 17:51:58 +02003223/*
3224 * Handle an ED style diff line.
3225 * Return FAIL if the line does not contain diff info.
3226 */
3227 static int
3228parse_diff_ed(
3229 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003230 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003231{
3232 char_u *p;
3233 long f1, l1, f2, l2;
3234 int difftype;
3235
3236 // The line must be one of three formats:
3237 // change: {first}[,{last}]c{first}[,{last}]
3238 // append: {first}a{first}[,{last}]
3239 // delete: {first}[,{last}]d{first}
3240 p = line;
3241 f1 = getdigits(&p);
3242 if (*p == ',')
3243 {
3244 ++p;
3245 l1 = getdigits(&p);
3246 }
3247 else
3248 l1 = f1;
3249 if (*p != 'a' && *p != 'c' && *p != 'd')
3250 return FAIL; // invalid diff format
3251 difftype = *p++;
3252 f2 = getdigits(&p);
3253 if (*p == ',')
3254 {
3255 ++p;
3256 l2 = getdigits(&p);
3257 }
3258 else
3259 l2 = f2;
3260 if (l1 < f1 || l2 < f2)
3261 return FAIL;
3262
3263 if (difftype == 'a')
3264 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003265 hunk->lnum_orig = f1 + 1;
3266 hunk->count_orig = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003267 }
3268 else
3269 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003270 hunk->lnum_orig = f1;
3271 hunk->count_orig = l1 - f1 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003272 }
3273 if (difftype == 'd')
3274 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003275 hunk->lnum_new = f2 + 1;
3276 hunk->count_new = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003277 }
3278 else
3279 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003280 hunk->lnum_new = f2;
3281 hunk->count_new = l2 - f2 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003282 }
3283 return OK;
3284}
3285
3286/*
3287 * Parses unified diff with zero(!) context lines.
3288 * Return FAIL if there is no diff information in "line".
3289 */
3290 static int
3291parse_diff_unified(
3292 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003293 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003294{
3295 char_u *p;
3296 long oldline, oldcount, newline, newcount;
3297
3298 // Parse unified diff hunk header:
3299 // @@ -oldline,oldcount +newline,newcount @@
3300 p = line;
3301 if (*p++ == '@' && *p++ == '@' && *p++ == ' ' && *p++ == '-')
3302 {
3303 oldline = getdigits(&p);
3304 if (*p == ',')
3305 {
3306 ++p;
3307 oldcount = getdigits(&p);
3308 }
3309 else
3310 oldcount = 1;
3311 if (*p++ == ' ' && *p++ == '+')
3312 {
3313 newline = getdigits(&p);
3314 if (*p == ',')
3315 {
3316 ++p;
3317 newcount = getdigits(&p);
3318 }
3319 else
3320 newcount = 1;
3321 }
3322 else
3323 return FAIL; // invalid diff format
3324
3325 if (oldcount == 0)
3326 oldline += 1;
3327 if (newcount == 0)
3328 newline += 1;
3329 if (newline == 0)
3330 newline = 1;
3331
Lewis Russelld9da86e2021-12-28 13:54:41 +00003332 hunk->lnum_orig = oldline;
3333 hunk->count_orig = oldcount;
3334 hunk->lnum_new = newline;
3335 hunk->count_new = newcount;
Bram Moolenaare828b762018-09-10 17:51:58 +02003336
3337 return OK;
3338 }
3339
3340 return FAIL;
3341}
3342
3343/*
3344 * Callback function for the xdl_diff() function.
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003345 * Stores the diff output (indices) in a grow array.
Bram Moolenaare828b762018-09-10 17:51:58 +02003346 */
3347 static int
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003348xdiff_out_indices(
Lewis Russelld9da86e2021-12-28 13:54:41 +00003349 long start_a,
3350 long count_a,
3351 long start_b,
3352 long count_b,
3353 void *priv)
Bram Moolenaare828b762018-09-10 17:51:58 +02003354{
3355 diffout_T *dout = (diffout_T *)priv;
Lewis Russelld9da86e2021-12-28 13:54:41 +00003356 diffhunk_T *p = ALLOC_ONE(diffhunk_T);
Bram Moolenaare828b762018-09-10 17:51:58 +02003357
Lewis Russelld9da86e2021-12-28 13:54:41 +00003358 if (p == NULL)
3359 return -1;
Bram Moolenaarf080d702018-10-31 22:57:26 +01003360
3361 if (ga_grow(&dout->dout_ga, 1) == FAIL)
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003362 {
3363 vim_free(p);
Bram Moolenaarf080d702018-10-31 22:57:26 +01003364 return -1;
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003365 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00003366
3367 p->lnum_orig = start_a + 1;
3368 p->count_orig = count_a;
3369 p->lnum_new = start_b + 1;
3370 p->count_new = count_b;
3371 ((diffhunk_T **)dout->dout_ga.ga_data)[dout->dout_ga.ga_len++] = p;
Bram Moolenaare828b762018-09-10 17:51:58 +02003372 return 0;
3373}
3374
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003375/*
3376 * Callback function for the xdl_diff() function.
3377 * Stores the unified diff output in a grow array.
3378 */
3379 static int
3380xdiff_out_unified(
3381 void *priv,
3382 mmbuffer_t *mb,
3383 int nbuf)
3384{
3385 diffout_T *dout = (diffout_T *)priv;
3386 int i;
3387
3388 for (i = 0; i < nbuf; i++)
3389 ga_concat_len(&dout->dout_ga, (char_u *)mb[i].ptr, mb[i].size);
3390
3391 return 0;
3392}
3393
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003394#endif // FEAT_DIFF
3395
3396#if defined(FEAT_EVAL) || defined(PROTO)
3397
3398/*
3399 * "diff_filler()" function
3400 */
3401 void
3402f_diff_filler(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3403{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003404# ifdef FEAT_DIFF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02003405 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3406 return;
3407
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003408 rettv->vval.v_number = diff_check_fill(curwin, tv_get_lnum(argvars));
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003409# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003410}
3411
3412/*
3413 * "diff_hlID()" function
3414 */
3415 void
3416f_diff_hlID(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3417{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003418# ifdef FEAT_DIFF
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003419 linenr_T lnum;
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003420 static linenr_T prev_lnum = 0;
3421 static varnumber_T changedtick = 0;
3422 static int fnum = 0;
3423 static int change_start = 0;
3424 static int change_end = 0;
3425 static hlf_T hlID = (hlf_T)0;
3426 int filler_lines;
3427 int col;
3428
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003429 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +02003430 && (check_for_lnum_arg(argvars,0) == FAIL
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003431 || check_for_number_arg(argvars, 1) == FAIL))
3432 return;
3433
3434 lnum = tv_get_lnum(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003435 if (lnum < 0) // ignore type error in {lnum} arg
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003436 lnum = 0;
3437 if (lnum != prev_lnum
3438 || changedtick != CHANGEDTICK(curbuf)
3439 || fnum != curbuf->b_fnum)
3440 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003441 // New line, buffer, change: need to get the values.
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003442 filler_lines = diff_check(curwin, lnum);
3443 if (filler_lines < 0)
3444 {
3445 if (filler_lines == -1)
3446 {
3447 change_start = MAXCOL;
3448 change_end = -1;
3449 if (diff_find_change(curwin, lnum, &change_start, &change_end))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003450 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003451 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003452 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003453 }
3454 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003455 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003456 }
3457 else
3458 hlID = (hlf_T)0;
3459 prev_lnum = lnum;
3460 changedtick = CHANGEDTICK(curbuf);
3461 fnum = curbuf->b_fnum;
3462 }
3463
3464 if (hlID == HLF_CHD || hlID == HLF_TXD)
3465 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003466 col = tv_get_number(&argvars[1]) - 1; // ignore type error in {col}
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003467 if (col >= change_start && col <= change_end)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003468 hlID = HLF_TXD; // changed text
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003469 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003470 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003471 }
3472 rettv->vval.v_number = hlID == (hlf_T)0 ? 0 : (int)hlID;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003473# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003474}
3475
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003476# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003477/*
3478 * Parse the diff options passed in "optarg" to the diff() function and return
3479 * the options in "diffopts" and the diff algorithm in "diffalgo".
3480 */
3481 static int
3482parse_diff_optarg(
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003483 typval_T *opts,
3484 int *diffopts,
3485 long *diffalgo,
3486 dio_outfmt_T *diff_output_fmt,
3487 int *diff_ctxlen)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003488{
3489 dict_T *d = opts->vval.v_dict;
3490
3491 char_u *algo = dict_get_string(d, "algorithm", FALSE);
3492 if (algo != NULL)
3493 {
3494 if (STRNCMP(algo, "myers", 5) == 0)
3495 *diffalgo = 0;
3496 else if (STRNCMP(algo, "minimal", 7) == 0)
3497 *diffalgo = XDF_NEED_MINIMAL;
3498 else if (STRNCMP(algo, "patience", 8) == 0)
3499 *diffalgo = XDF_PATIENCE_DIFF;
3500 else if (STRNCMP(algo, "histogram", 9) == 0)
3501 *diffalgo = XDF_HISTOGRAM_DIFF;
3502 }
3503
3504 char_u *output_fmt = dict_get_string(d, "output", FALSE);
3505 if (output_fmt != NULL)
3506 {
3507 if (STRNCMP(output_fmt, "unified", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003508 *diff_output_fmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003509 else if (STRNCMP(output_fmt, "indices", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003510 *diff_output_fmt = DIO_OUTPUT_INDICES;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003511 else
3512 {
3513 semsg(_(e_unsupported_diff_output_format_str), output_fmt);
3514 return FAIL;
3515 }
3516 }
3517
Yegappan Lakshmanana0010a12024-02-12 20:21:26 +01003518 *diff_ctxlen = dict_get_number_def(d, "context", 0);
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003519 if (*diff_ctxlen < 0)
Yegappan Lakshmanana0010a12024-02-12 20:21:26 +01003520 *diff_ctxlen = 0;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003521
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003522 if (dict_get_bool(d, "iblank", FALSE))
3523 *diffopts |= DIFF_IBLANK;
3524 if (dict_get_bool(d, "icase", FALSE))
3525 *diffopts |= DIFF_ICASE;
3526 if (dict_get_bool(d, "iwhite", FALSE))
3527 *diffopts |= DIFF_IWHITE;
3528 if (dict_get_bool(d, "iwhiteall", FALSE))
3529 *diffopts |= DIFF_IWHITEALL;
3530 if (dict_get_bool(d, "iwhiteeol", FALSE))
3531 *diffopts |= DIFF_IWHITEEOL;
3532 if (dict_get_bool(d, "indent-heuristic", FALSE))
3533 *diffalgo |= XDF_INDENT_HEURISTIC;
3534
3535 return OK;
3536}
3537
3538/*
3539 * Concatenate the List of strings in "l" and store the result in
3540 * "din->din_mmfile.ptr" and the length in "din->din_mmfile.size".
3541 */
3542 static void
3543list_to_diffin(list_T *l, diffin_T *din, int icase)
3544{
3545 garray_T ga;
3546 listitem_T *li;
3547 char_u *str;
3548
3549 ga_init2(&ga, 512, 4);
3550
3551 FOR_ALL_LIST_ITEMS(l, li)
3552 {
3553 str = tv_get_string(&li->li_tv);
3554 if (icase)
3555 {
3556 str = strlow_save(str);
3557 if (str == NULL)
3558 continue;
3559 }
3560 ga_concat(&ga, str);
3561 ga_concat(&ga, (char_u *)NL_STR);
3562 if (icase)
3563 vim_free(str);
3564 }
3565 if (ga.ga_len > 0)
3566 ((char *)ga.ga_data)[ga.ga_len] = NUL;
3567
3568 din->din_mmfile.ptr = (char *)ga.ga_data;
3569 din->din_mmfile.size = ga.ga_len;
3570}
3571
3572/*
3573 * Get the start and end indices from the diff "hunk".
3574 */
3575 static dict_T *
3576get_diff_hunk_indices(diffhunk_T *hunk)
3577{
3578 dict_T *hunk_dict;
3579
3580 hunk_dict = dict_alloc();
3581 if (hunk_dict == NULL)
3582 return NULL;
3583
3584 dict_add_number(hunk_dict, "from_idx", hunk->lnum_orig - 1);
3585 dict_add_number(hunk_dict, "from_count", hunk->count_orig);
3586 dict_add_number(hunk_dict, "to_idx", hunk->lnum_new - 1);
3587 dict_add_number(hunk_dict, "to_count", hunk->count_new);
3588
3589 return hunk_dict;
3590}
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003591# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003592
3593/*
3594 * "diff()" function
3595 */
3596 void
3597f_diff(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3598{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003599# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003600 diffio_T dio;
3601
3602 if (check_for_nonnull_list_arg(argvars, 0) == FAIL
3603 || check_for_nonnull_list_arg(argvars, 1) == FAIL
3604 || check_for_opt_nonnull_dict_arg(argvars, 2) == FAIL)
3605 return;
3606
3607 CLEAR_FIELD(dio);
3608 dio.dio_internal = TRUE;
3609 ga_init2(&dio.dio_diff.dout_ga, sizeof(char *), 1000);
3610
3611 list_T *orig_list = argvars[0].vval.v_list;
3612 list_T *new_list = argvars[1].vval.v_list;
3613
3614 // Save the 'diffopt' option value and restore it after getting the diff.
3615 int save_diff_flags = diff_flags;
3616 long save_diff_algorithm = diff_algorithm;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003617 diff_flags = DIFF_INTERNAL;
3618 diff_algorithm = 0;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003619 dio.dio_outfmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003620 if (argvars[2].v_type != VAR_UNKNOWN)
3621 if (parse_diff_optarg(&argvars[2], &diff_flags, &diff_algorithm,
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003622 &dio.dio_outfmt, &dio.dio_ctxlen) == FAIL)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003623 return;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003624
3625 // Concatenate the List of strings into a single string using newline
3626 // separator. Internal diff library expects a single string.
3627 list_to_diffin(orig_list, &dio.dio_orig, diff_flags & DIFF_ICASE);
3628 list_to_diffin(new_list, &dio.dio_new, diff_flags & DIFF_ICASE);
3629
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003630 // If 'diffexpr' is set, then the internal diff is not used. Set
3631 // 'diffexpr' to an empty string temporarily.
3632 int restore_diffexpr = FALSE;
3633 char_u cc = *p_dex;
3634 if (*p_dex != NUL)
3635 {
3636 restore_diffexpr = TRUE;
3637 *p_dex = NUL;
3638 }
3639
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003640 // Compute the diff
3641 int diff_status = diff_file(&dio);
3642
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003643 // restore 'diffexpr'
3644 if (restore_diffexpr)
3645 *p_dex = cc;
3646
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003647 if (diff_status == FAIL)
3648 goto done;
3649
3650 int hunk_idx = 0;
3651 dict_T *hunk_dict;
3652
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003653 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003654 {
3655 if (rettv_list_alloc(rettv) != OK)
3656 goto done;
3657 list_T *l = rettv->vval.v_list;
3658
3659 // Process each diff hunk
3660 diffhunk_T *hunk = NULL;
3661 while (hunk_idx < dio.dio_diff.dout_ga.ga_len)
3662 {
3663 hunk = ((diffhunk_T **)dio.dio_diff.dout_ga.ga_data)[hunk_idx++];
3664
3665 hunk_dict = get_diff_hunk_indices(hunk);
3666 if (hunk_dict == NULL)
3667 goto done;
3668
3669 list_append_dict(l, hunk_dict);
3670 }
3671 }
3672 else
3673 {
3674 ga_append(&dio.dio_diff.dout_ga, NUL);
3675 rettv->v_type = VAR_STRING;
3676 rettv->vval.v_string =
3677 vim_strsave((char_u *)dio.dio_diff.dout_ga.ga_data);
3678 }
3679
3680done:
3681 clear_diffin(&dio.dio_new);
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003682 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003683 clear_diffout(&dio.dio_diff);
3684 else
3685 ga_clear(&dio.dio_diff.dout_ga);
3686 clear_diffin(&dio.dio_orig);
3687 // Restore the 'diffopt' option value.
3688 diff_flags = save_diff_flags;
3689 diff_algorithm = save_diff_algorithm;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003690# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003691}
3692
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003693#endif