blob: ebe409e5f35550163152ef60040c5c0c535a0f53 [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
Yukihiro Nakadairaf1694b42024-09-22 11:26:13 +0200763 if (buf->b_ml.ml_flags & ML_EMPTY)
764 {
765 din->din_mmfile.ptr = NULL;
766 din->din_mmfile.size = 0;
767 return OK;
768 }
769
Bram Moolenaare828b762018-09-10 17:51:58 +0200770 // xdiff requires one big block of memory with all the text.
771 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
zeertzjq94b7c322024-03-12 21:50:32 +0100772 len += ml_get_buf_len(buf, lnum) + 1;
Bram Moolenaar18a4ba22019-05-24 19:39:03 +0200773 ptr = alloc(len);
Bram Moolenaare828b762018-09-10 17:51:58 +0200774 if (ptr == NULL)
775 {
776 // Allocating memory failed. This can happen, because we try to read
777 // the whole buffer text into memory. Set the failed flag, the diff
778 // will be retried with external diff. The flag is never reset.
779 buf->b_diff_failed = TRUE;
780 if (p_verbose > 0)
781 {
782 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100783 smsg(_("Not enough memory to use internal diff for buffer \"%s\""),
Bram Moolenaare828b762018-09-10 17:51:58 +0200784 buf->b_fname);
785 verbose_leave();
786 }
787 return FAIL;
788 }
789 din->din_mmfile.ptr = (char *)ptr;
790 din->din_mmfile.size = len;
791
792 len = 0;
793 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
794 {
795 for (s = ml_get_buf(buf, lnum, FALSE); *s != NUL; )
796 {
797 if (diff_flags & DIFF_ICASE)
798 {
799 int c;
Bram Moolenaare828b762018-09-10 17:51:58 +0200800 int orig_len;
801 char_u cbuf[MB_MAXBYTES + 1];
802
Bram Moolenaar06f60952021-12-28 18:30:05 +0000803 if (*s == NL)
804 c = NUL;
805 else
806 {
807 // xdiff doesn't support ignoring case, fold-case the text.
808 c = PTR2CHAR(s);
809 c = MB_CASEFOLD(c);
810 }
Bram Moolenaar1614a142019-10-06 22:00:13 +0200811 orig_len = mb_ptr2len(s);
Bram Moolenaare828b762018-09-10 17:51:58 +0200812 if (mb_char2bytes(c, cbuf) != orig_len)
813 // TODO: handle byte length difference
814 mch_memmove(ptr + len, s, orig_len);
815 else
816 mch_memmove(ptr + len, cbuf, orig_len);
817
818 s += orig_len;
819 len += orig_len;
Bram Moolenaare828b762018-09-10 17:51:58 +0200820 }
821 else
Bram Moolenaar06f60952021-12-28 18:30:05 +0000822 {
823 ptr[len++] = *s == NL ? NUL : *s;
824 s++;
825 }
Bram Moolenaare828b762018-09-10 17:51:58 +0200826 }
827 ptr[len++] = NL;
828 }
829 return OK;
830}
831
832/*
833 * Write buffer "buf" to file or memory buffer.
834 * Return FAIL for failure.
835 */
836 static int
837diff_write(buf_T *buf, diffin_T *din)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000838{
839 int r;
840 char_u *save_ff;
Bram Moolenaare1004402020-10-24 20:49:43 +0200841 int save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000842
Bram Moolenaare828b762018-09-10 17:51:58 +0200843 if (din->din_fname == NULL)
844 return diff_write_buffer(buf, din);
845
846 // Always use 'fileformat' set to "unix".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000847 save_ff = buf->b_p_ff;
848 buf->b_p_ff = vim_strsave((char_u *)FF_UNIX);
Bram Moolenaare1004402020-10-24 20:49:43 +0200849 save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100850 // Writing the buffer is an implementation detail of performing the diff,
851 // so it shouldn't update the '[ and '] marks.
Bram Moolenaare1004402020-10-24 20:49:43 +0200852 cmdmod.cmod_flags |= CMOD_LOCKMARKS;
Bram Moolenaare828b762018-09-10 17:51:58 +0200853 r = buf_write(buf, din->din_fname, NULL,
854 (linenr_T)1, buf->b_ml.ml_line_count,
855 NULL, FALSE, FALSE, FALSE, TRUE);
Bram Moolenaare1004402020-10-24 20:49:43 +0200856 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000857 free_string_option(buf->b_p_ff);
858 buf->b_p_ff = save_ff;
859 return r;
860}
861
862/*
Bram Moolenaare828b762018-09-10 17:51:58 +0200863 * Update the diffs for all buffers involved.
864 */
865 static void
866diff_try_update(
867 diffio_T *dio,
868 int idx_orig,
869 exarg_T *eap) // "eap" can be NULL
870{
871 buf_T *buf;
872 int idx_new;
873
874 if (dio->dio_internal)
875 {
876 ga_init2(&dio->dio_diff.dout_ga, sizeof(char *), 1000);
877 }
878 else
879 {
880 // We need three temp file names.
881 dio->dio_orig.din_fname = vim_tempname('o', TRUE);
882 dio->dio_new.din_fname = vim_tempname('n', TRUE);
883 dio->dio_diff.dout_fname = vim_tempname('d', TRUE);
884 if (dio->dio_orig.din_fname == NULL
885 || dio->dio_new.din_fname == NULL
886 || dio->dio_diff.dout_fname == NULL)
887 goto theend;
888 }
889
890 // Check external diff is actually working.
891 if (!dio->dio_internal && check_external_diff(dio) == FAIL)
892 goto theend;
893
894 // :diffupdate!
895 if (eap != NULL && eap->forceit)
896 for (idx_new = idx_orig; idx_new < DB_COUNT; ++idx_new)
897 {
898 buf = curtab->tp_diffbuf[idx_new];
899 if (buf_valid(buf))
900 buf_check_timestamp(buf, FALSE);
901 }
902
903 // Write the first buffer to a tempfile or mmfile_t.
904 buf = curtab->tp_diffbuf[idx_orig];
905 if (diff_write(buf, &dio->dio_orig) == FAIL)
906 goto theend;
907
908 // Make a difference between the first buffer and every other.
909 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
910 {
911 buf = curtab->tp_diffbuf[idx_new];
912 if (buf == NULL || buf->b_ml.ml_mfp == NULL)
913 continue; // skip buffer that isn't loaded
914
915 // Write the other buffer and diff with the first one.
916 if (diff_write(buf, &dio->dio_new) == FAIL)
917 continue;
918 if (diff_file(dio) == FAIL)
919 continue;
920
921 // Read the diff output and add each entry to the diff list.
Lewis Russelld9da86e2021-12-28 13:54:41 +0000922 diff_read(idx_orig, idx_new, dio);
Bram Moolenaare828b762018-09-10 17:51:58 +0200923
924 clear_diffin(&dio->dio_new);
925 clear_diffout(&dio->dio_diff);
926 }
927 clear_diffin(&dio->dio_orig);
928
929theend:
930 vim_free(dio->dio_orig.din_fname);
931 vim_free(dio->dio_new.din_fname);
932 vim_free(dio->dio_diff.dout_fname);
933}
934
935/*
936 * Return TRUE if the options are set to use the internal diff library.
937 * Note that if the internal diff failed for one of the buffers, the external
938 * diff will be used anyway.
939 */
Bram Moolenaare3521d92018-09-16 14:10:31 +0200940 int
Bram Moolenaare828b762018-09-10 17:51:58 +0200941diff_internal(void)
942{
Bram Moolenaar975880b2019-03-03 14:42:11 +0100943 return (diff_flags & DIFF_INTERNAL) != 0
944#ifdef FEAT_EVAL
945 && *p_dex == NUL
946#endif
947 ;
Bram Moolenaare828b762018-09-10 17:51:58 +0200948}
949
950/*
951 * Return TRUE if the internal diff failed for one of the diff buffers.
952 */
953 static int
954diff_internal_failed(void)
955{
956 int idx;
957
958 // Only need to do something when there is another buffer.
959 for (idx = 0; idx < DB_COUNT; ++idx)
960 if (curtab->tp_diffbuf[idx] != NULL
961 && curtab->tp_diffbuf[idx]->b_diff_failed)
962 return TRUE;
963 return FALSE;
964}
965
966/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967 * Completely update the diffs for the buffers involved.
Bram Moolenaare3521d92018-09-16 14:10:31 +0200968 * When using the external "diff" command the buffers are written to a file,
969 * also for unmodified buffers (the file could have been produced by
970 * autocommands, e.g. the netrw plugin).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000971 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000972 void
Bram Moolenaare828b762018-09-10 17:51:58 +0200973ex_diffupdate(exarg_T *eap) // "eap" can be NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000974{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000975 int idx_orig;
976 int idx_new;
Bram Moolenaare828b762018-09-10 17:51:58 +0200977 diffio_T diffio;
Bram Moolenaar198fa062018-09-18 21:20:26 +0200978 int had_diffs = curtab->tp_first_diff != NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000979
Bram Moolenaard2b58c02018-09-16 18:10:48 +0200980 if (diff_busy)
981 {
982 diff_need_update = TRUE;
983 return;
984 }
985
Bram Moolenaare828b762018-09-10 17:51:58 +0200986 // Delete all diffblocks.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000987 diff_clear(curtab);
988 curtab->tp_diff_invalid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989
Bram Moolenaare828b762018-09-10 17:51:58 +0200990 // Use the first buffer as the original text.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000991 for (idx_orig = 0; idx_orig < DB_COUNT; ++idx_orig)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000992 if (curtab->tp_diffbuf[idx_orig] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000993 break;
994 if (idx_orig == 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 need to do something when there is another buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000998 for (idx_new = idx_orig + 1; idx_new < DB_COUNT; ++idx_new)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +0000999 if (curtab->tp_diffbuf[idx_new] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001000 break;
1001 if (idx_new == DB_COUNT)
Bram Moolenaar198fa062018-09-18 21:20:26 +02001002 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001003
Bram Moolenaare828b762018-09-10 17:51:58 +02001004 // Only use the internal method if it did not fail for one of the buffers.
Bram Moolenaara80faa82020-04-12 19:37:17 +02001005 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +02001006 diffio.dio_internal = diff_internal() && !diff_internal_failed();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001007
Bram Moolenaare828b762018-09-10 17:51:58 +02001008 diff_try_update(&diffio, idx_orig, eap);
1009 if (diffio.dio_internal && diff_internal_failed())
1010 {
1011 // Internal diff failed, use external diff instead.
Bram Moolenaara80faa82020-04-12 19:37:17 +02001012 CLEAR_FIELD(diffio);
Bram Moolenaare828b762018-09-10 17:51:58 +02001013 diff_try_update(&diffio, idx_orig, eap);
1014 }
1015
1016 // force updating cursor position on screen
1017 curwin->w_valid_cursor.lnum = 0;
1018
Bram Moolenaar198fa062018-09-18 21:20:26 +02001019theend:
1020 // A redraw is needed if there were diffs and they were cleared, or there
1021 // are diffs now, which means they got updated.
1022 if (had_diffs || curtab->tp_first_diff != NULL)
1023 {
1024 diff_redraw(TRUE);
1025 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
1026 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001027}
1028
1029/*
1030 * Do a quick test if "diff" really works. Otherwise it looks like there
1031 * are no differences. Can't use the return value, it's non-zero when
1032 * there are differences.
1033 */
1034 static int
1035check_external_diff(diffio_T *diffio)
1036{
1037 FILE *fd;
1038 int ok;
1039 int io_error = FALSE;
1040
1041 // May try twice, first with "-a" and then without.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001042 for (;;)
1043 {
1044 ok = FALSE;
Bram Moolenaare828b762018-09-10 17:51:58 +02001045 fd = mch_fopen((char *)diffio->dio_orig.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001046 if (fd == NULL)
1047 io_error = TRUE;
1048 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001049 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001050 if (fwrite("line1\n", (size_t)6, (size_t)1, fd) != 1)
1051 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001052 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001053 fd = mch_fopen((char *)diffio->dio_new.din_fname, "w");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001054 if (fd == NULL)
1055 io_error = TRUE;
1056 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001058 if (fwrite("line2\n", (size_t)6, (size_t)1, fd) != 1)
1059 io_error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060 fclose(fd);
Bram Moolenaare828b762018-09-10 17:51:58 +02001061 fd = NULL;
1062 if (diff_file(diffio) == OK)
1063 fd = mch_fopen((char *)diffio->dio_diff.dout_fname, "r");
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001064 if (fd == NULL)
1065 io_error = TRUE;
1066 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001067 {
1068 char_u linebuf[LBUFLEN];
1069
1070 for (;;)
1071 {
glacambread5c1782021-05-24 14:20:53 +02001072 // For normal diff there must be a line that contains
1073 // "1c1". For unified diff "@@ -1 +1 @@".
Bram Moolenaar00590742019-02-15 21:06:09 +01001074 if (vim_fgets(linebuf, LBUFLEN, fd))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001075 break;
glacambread5c1782021-05-24 14:20:53 +02001076 if (STRNCMP(linebuf, "1c1", 3) == 0
1077 || STRNCMP(linebuf, "@@ -1 +1 @@", 11) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001078 ok = TRUE;
1079 }
1080 fclose(fd);
1081 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001082 mch_remove(diffio->dio_diff.dout_fname);
1083 mch_remove(diffio->dio_new.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001084 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001085 mch_remove(diffio->dio_orig.din_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001086 }
1087
1088#ifdef FEAT_EVAL
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001089 // When using 'diffexpr' break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001090 if (*p_dex != NUL)
1091 break;
1092#endif
1093
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001094#if defined(MSWIN)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001095 // If the "-a" argument works, also check if "--binary" works.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001096 if (ok && diff_a_works == MAYBE && diff_bin_works == MAYBE)
1097 {
1098 diff_a_works = TRUE;
1099 diff_bin_works = TRUE;
1100 continue;
1101 }
1102 if (!ok && diff_a_works == TRUE && diff_bin_works == TRUE)
1103 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001104 // Tried --binary, but it failed. "-a" works though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001105 diff_bin_works = FALSE;
1106 ok = TRUE;
1107 }
1108#endif
1109
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001110 // If we checked if "-a" works already, break here.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001111 if (diff_a_works != MAYBE)
1112 break;
1113 diff_a_works = ok;
1114
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001115 // If "-a" works break here, otherwise retry without "-a".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116 if (ok)
1117 break;
1118 }
1119 if (!ok)
1120 {
Bram Moolenaarfe86f2d2008-11-28 20:29:07 +00001121 if (io_error)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001122 emsg(_(e_cannot_read_or_write_temp_files));
Bram Moolenaare1242042021-12-16 20:56:57 +00001123 emsg(_(e_cannot_create_diffs));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001124 diff_a_works = MAYBE;
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001125#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001126 diff_bin_works = MAYBE;
1127#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001128 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001129 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001130 return OK;
1131}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132
Bram Moolenaare828b762018-09-10 17:51:58 +02001133/*
1134 * Invoke the xdiff function.
1135 */
1136 static int
1137diff_file_internal(diffio_T *diffio)
1138{
1139 xpparam_t param;
1140 xdemitconf_t emit_cfg;
1141 xdemitcb_t emit_cb;
Bram Moolenaarbd1d5602012-05-18 18:47:17 +02001142
Bram Moolenaara80faa82020-04-12 19:37:17 +02001143 CLEAR_FIELD(param);
1144 CLEAR_FIELD(emit_cfg);
1145 CLEAR_FIELD(emit_cb);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001146
Bram Moolenaare828b762018-09-10 17:51:58 +02001147 param.flags = diff_algorithm;
1148
1149 if (diff_flags & DIFF_IWHITE)
1150 param.flags |= XDF_IGNORE_WHITESPACE_CHANGE;
Bram Moolenaar785fc652018-09-15 19:17:38 +02001151 if (diff_flags & DIFF_IWHITEALL)
1152 param.flags |= XDF_IGNORE_WHITESPACE;
1153 if (diff_flags & DIFF_IWHITEEOL)
1154 param.flags |= XDF_IGNORE_WHITESPACE_AT_EOL;
1155 if (diff_flags & DIFF_IBLANK)
1156 param.flags |= XDF_IGNORE_BLANK_LINES;
Bram Moolenaare828b762018-09-10 17:51:58 +02001157
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001158 emit_cfg.ctxlen = diffio->dio_ctxlen;
Bram Moolenaare828b762018-09-10 17:51:58 +02001159 emit_cb.priv = &diffio->dio_diff;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01001160 if (diffio->dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01001161 emit_cfg.hunk_func = xdiff_out_indices;
1162 else
1163 emit_cb.out_line = xdiff_out_unified;
Bram Moolenaare828b762018-09-10 17:51:58 +02001164 if (xdl_diff(&diffio->dio_orig.din_mmfile,
1165 &diffio->dio_new.din_mmfile,
1166 &param, &emit_cfg, &emit_cb) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001167 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001168 emsg(_(e_problem_creating_internal_diff));
Bram Moolenaare828b762018-09-10 17:51:58 +02001169 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001170 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001171 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001172}
1173
1174/*
1175 * Make a diff between files "tmp_orig" and "tmp_new", results in "tmp_diff".
Bram Moolenaare828b762018-09-10 17:51:58 +02001176 * return OK or FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001177 */
Bram Moolenaare828b762018-09-10 17:51:58 +02001178 static int
1179diff_file(diffio_T *dio)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001180{
1181 char_u *cmd;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001182 size_t len;
Bram Moolenaare828b762018-09-10 17:51:58 +02001183 char_u *tmp_orig = dio->dio_orig.din_fname;
1184 char_u *tmp_new = dio->dio_new.din_fname;
1185 char_u *tmp_diff = dio->dio_diff.dout_fname;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001186
1187#ifdef FEAT_EVAL
1188 if (*p_dex != NUL)
Bram Moolenaare828b762018-09-10 17:51:58 +02001189 {
1190 // Use 'diffexpr' to generate the diff file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001191 eval_diff(tmp_orig, tmp_new, tmp_diff);
Bram Moolenaare828b762018-09-10 17:51:58 +02001192 return OK;
1193 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001194 else
1195#endif
Bram Moolenaare828b762018-09-10 17:51:58 +02001196 // Use xdiff for generating the diff.
1197 if (dio->dio_internal)
Bram Moolenaare828b762018-09-10 17:51:58 +02001198 return diff_file_internal(dio);
Bram Moolenaar95fb60a2005-03-08 22:29:20 +00001199
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001200 len = STRLEN(tmp_orig) + STRLEN(tmp_new)
1201 + STRLEN(tmp_diff) + STRLEN(p_srr) + 27;
1202 cmd = alloc(len);
1203 if (cmd == NULL)
1204 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02001205
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001206 // We don't want $DIFF_OPTIONS to get in the way.
1207 if (getenv("DIFF_OPTIONS"))
1208 vim_setenv((char_u *)"DIFF_OPTIONS", (char_u *)"");
1209
1210 // Build the diff command and execute it. Always use -a, binary
1211 // differences are of no use. Ignore errors, diff returns
1212 // non-zero when differences have been found.
1213 vim_snprintf((char *)cmd, len, "diff %s%s%s%s%s%s%s%s %s",
1214 diff_a_works == FALSE ? "" : "-a ",
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001215#if defined(MSWIN)
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001216 diff_bin_works == TRUE ? "--binary " : "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001217#else
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001218 "",
Bram Moolenaar071d4272004-06-13 20:20:40 +00001219#endif
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001220 (diff_flags & DIFF_IWHITE) ? "-b " : "",
1221 (diff_flags & DIFF_IWHITEALL) ? "-w " : "",
1222 (diff_flags & DIFF_IWHITEEOL) ? "-Z " : "",
1223 (diff_flags & DIFF_IBLANK) ? "-B " : "",
1224 (diff_flags & DIFF_ICASE) ? "-i " : "",
1225 tmp_orig, tmp_new);
1226 append_redir(cmd, (int)len, p_srr, tmp_diff);
1227 block_autocmds(); // avoid ShellCmdPost stuff
1228 (void)call_shell(cmd, SHELL_FILTER|SHELL_SILENT|SHELL_DOOUT);
1229 unblock_autocmds();
1230 vim_free(cmd);
1231 return OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001232}
1233
1234/*
1235 * Create a new version of a file from the current buffer and a diff file.
1236 * The buffer is written to a file, also for unmodified buffers (the file
1237 * could have been produced by autocommands, e.g. the netrw plugin).
1238 */
1239 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001240ex_diffpatch(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001242 char_u *tmp_orig; // name of original temp file
1243 char_u *tmp_new; // name of patched temp file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001244 char_u *buf = NULL;
Bram Moolenaar5fd0ca72009-05-13 16:56:33 +00001245 size_t buflen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001246 win_T *old_curwin = curwin;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001247 char_u *newname = NULL; // name of patched file buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248#ifdef UNIX
1249 char_u dirbuf[MAXPATHL];
1250 char_u *fullname = NULL;
1251#endif
1252#ifdef FEAT_BROWSE
1253 char_u *browseFile = NULL;
Bram Moolenaare1004402020-10-24 20:49:43 +02001254 int save_cmod_flags = cmdmod.cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001255#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02001256 stat_T st;
Bram Moolenaara95ab322017-03-11 19:21:53 +01001257 char_u *esc_name = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001258
1259#ifdef FEAT_BROWSE
Bram Moolenaare1004402020-10-24 20:49:43 +02001260 if (cmdmod.cmod_flags & CMOD_BROWSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001261 {
Bram Moolenaar7171abe2004-10-11 10:06:20 +00001262 browseFile = do_browse(0, (char_u *)_("Patch file"),
Bram Moolenaarc36651b2018-04-29 12:22:56 +02001263 eap->arg, NULL, NULL,
1264 (char_u *)_(BROWSE_FILTER_ALL_FILES), NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001265 if (browseFile == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001266 return; // operation cancelled
Bram Moolenaar071d4272004-06-13 20:20:40 +00001267 eap->arg = browseFile;
Bram Moolenaare1004402020-10-24 20:49:43 +02001268 cmdmod.cmod_flags &= ~CMOD_BROWSE; // don't let do_ecmd() browse again
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269 }
1270#endif
1271
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001272 // We need two temp file names.
Bram Moolenaare5c421c2015-03-31 13:33:08 +02001273 tmp_orig = vim_tempname('o', FALSE);
1274 tmp_new = vim_tempname('n', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001275 if (tmp_orig == NULL || tmp_new == NULL)
1276 goto theend;
1277
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001278 // Write the current buffer to "tmp_orig".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001279 if (buf_write(curbuf, tmp_orig, NULL,
1280 (linenr_T)1, curbuf->b_ml.ml_line_count,
1281 NULL, FALSE, FALSE, FALSE, TRUE) == FAIL)
1282 goto theend;
1283
1284#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001285 // Get the absolute path of the patchfile, changing directory below.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001286 fullname = FullName_save(eap->arg, FALSE);
1287#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001288 esc_name = vim_strsave_shellescape(
Bram Moolenaar071d4272004-06-13 20:20:40 +00001289# ifdef UNIX
Bram Moolenaara95ab322017-03-11 19:21:53 +01001290 fullname != NULL ? fullname :
Bram Moolenaar071d4272004-06-13 20:20:40 +00001291# endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001292 eap->arg, TRUE, TRUE);
1293 if (esc_name == NULL)
1294 goto theend;
1295 buflen = STRLEN(tmp_orig) + STRLEN(esc_name) + STRLEN(tmp_new) + 16;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001296 buf = alloc(buflen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001297 if (buf == NULL)
1298 goto theend;
1299
1300#ifdef UNIX
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001301 // Temporarily chdir to /tmp, to avoid patching files in the current
1302 // directory when the patch file contains more than one patch. When we
1303 // have our own temp dir use that instead, it will be cleaned up when we
1304 // exit (any .rej files created). Don't change directory if we can't
1305 // return to the current.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001306 if (mch_dirname(dirbuf, MAXPATHL) != OK || mch_chdir((char *)dirbuf) != 0)
1307 dirbuf[0] = NUL;
1308 else
1309 {
1310# ifdef TEMPDIRNAMES
1311 if (vim_tempdir != NULL)
Bram Moolenaar42335f52018-09-13 15:33:43 +02001312 vim_ignored = mch_chdir((char *)vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001313 else
1314# endif
Bram Moolenaar42335f52018-09-13 15:33:43 +02001315 vim_ignored = mch_chdir("/tmp");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001316 shorten_fnames(TRUE);
1317 }
1318#endif
1319
1320#ifdef FEAT_EVAL
1321 if (*p_pex != NUL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001322 // Use 'patchexpr' to generate the new file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001323 eval_patch(tmp_orig,
1324# ifdef UNIX
1325 fullname != NULL ? fullname :
1326# endif
1327 eap->arg, tmp_new);
1328 else
1329#endif
1330 {
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001331 if (check_restricted())
1332 goto theend;
1333
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001334 // Build the patch command and execute it. Ignore errors. Switch to
1335 // cooked mode to allow the user to respond to prompts.
Bram Moolenaara95ab322017-03-11 19:21:53 +01001336 vim_snprintf((char *)buf, buflen, "patch -o %s %s < %s",
1337 tmp_new, tmp_orig, esc_name);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001338 block_autocmds(); // Avoid ShellCmdPost stuff
Bram Moolenaar071d4272004-06-13 20:20:40 +00001339 (void)call_shell(buf, SHELL_FILTER | SHELL_COOKED);
Bram Moolenaar78ab3312007-09-29 12:16:41 +00001340 unblock_autocmds();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001341 }
1342
1343#ifdef UNIX
1344 if (dirbuf[0] != NUL)
1345 {
1346 if (mch_chdir((char *)dirbuf) != 0)
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00001347 emsg(_(e_cannot_go_back_to_previous_directory));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001348 shorten_fnames(TRUE);
1349 }
1350#endif
1351
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001352 // patch probably has written over the screen
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001353 redraw_later(UPD_CLEAR);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001354
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001355 // Delete any .orig or .rej file created.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001356 STRCPY(buf, tmp_new);
1357 STRCAT(buf, ".orig");
1358 mch_remove(buf);
1359 STRCPY(buf, tmp_new);
1360 STRCAT(buf, ".rej");
1361 mch_remove(buf);
1362
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001363 // Only continue if the output file was created.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001364 if (mch_stat((char *)tmp_new, &st) < 0 || st.st_size == 0)
Bram Moolenaar9d00e4a2022-01-05 17:49:15 +00001365 emsg(_(e_cannot_read_patch_output));
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001366 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001367 {
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001368 if (curbuf->b_fname != NULL)
1369 {
1370 newname = vim_strnsave(curbuf->b_fname,
Bram Moolenaardf44a272020-06-07 20:49:05 +02001371 STRLEN(curbuf->b_fname) + 4);
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001372 if (newname != NULL)
1373 STRCAT(newname, ".new");
1374 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001375
1376#ifdef FEAT_GUI
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001377 need_mouse_correct = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001378#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001379 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001380 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001381
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001382 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) != FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001383 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001384 // Pretend it was a ":split fname" command
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001385 eap->cmdidx = CMD_split;
1386 eap->arg = tmp_new;
1387 do_exedit(eap, old_curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001388
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001389 // check that split worked and editing tmp_new
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001390 if (curwin != old_curwin && win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001391 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001392 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001393 diff_win_options(curwin, TRUE);
1394 diff_win_options(old_curwin, TRUE);
1395
1396 if (newname != NULL)
1397 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001398 // do a ":file filename.new" on the patched buffer
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001399 eap->arg = newname;
1400 ex_file(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001401
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001402 // Do filetype detection with the new name.
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001403 if (au_has_group((char_u *)"filetypedetect"))
Bram Moolenaar23a971d2023-04-04 22:04:53 +01001404 do_cmdline_cmd(
1405 (char_u *)":doau filetypedetect BufRead");
Bram Moolenaar6ec0a6c2009-07-22 14:23:13 +00001406 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001407 }
1408 }
1409 }
1410
1411theend:
1412 if (tmp_orig != NULL)
1413 mch_remove(tmp_orig);
1414 vim_free(tmp_orig);
1415 if (tmp_new != NULL)
1416 mch_remove(tmp_new);
1417 vim_free(tmp_new);
1418 vim_free(newname);
1419 vim_free(buf);
1420#ifdef UNIX
1421 vim_free(fullname);
1422#endif
Bram Moolenaara95ab322017-03-11 19:21:53 +01001423 vim_free(esc_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001424#ifdef FEAT_BROWSE
1425 vim_free(browseFile);
Bram Moolenaare1004402020-10-24 20:49:43 +02001426 cmdmod.cmod_flags = save_cmod_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001427#endif
1428}
1429
1430/*
1431 * Split the window and edit another file, setting options to show the diffs.
1432 */
1433 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001434ex_diffsplit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001435{
1436 win_T *old_curwin = curwin;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001437 bufref_T old_curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001438
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001439 set_bufref(&old_curbuf, curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001440#ifdef FEAT_GUI
1441 need_mouse_correct = TRUE;
1442#endif
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001443 // Need to compute w_fraction when no redraw happened yet.
Bram Moolenaar46328f92016-08-28 15:39:57 +02001444 validate_cursor();
1445 set_fraction(curwin);
1446
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001447 // don't use a new tab page, each tab page has its own diffs
Bram Moolenaare1004402020-10-24 20:49:43 +02001448 cmdmod.cmod_tab = 0;
Bram Moolenaar80a94a52006-02-23 21:26:58 +00001449
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001450 if (win_split(0, (diff_flags & DIFF_VERTICAL) ? WSP_VERT : 0) == FAIL)
1451 return;
1452
1453 // Pretend it was a ":split fname" command
1454 eap->cmdidx = CMD_split;
1455 curwin->w_p_diff = TRUE;
1456 do_exedit(eap, old_curwin);
1457
1458 if (curwin == old_curwin) // split didn't work
1459 return;
1460
1461 // Set 'diff', 'scrollbind' on and 'wrap' off.
1462 diff_win_options(curwin, TRUE);
1463 if (win_valid(old_curwin))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001464 {
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001465 diff_win_options(old_curwin, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001466
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001467 if (bufref_valid(&old_curbuf))
1468 // Move the cursor position to that of the old window.
1469 curwin->w_cursor.lnum = diff_get_corresponding_line(
1470 old_curbuf.br_buf, old_curwin->w_cursor.lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001471 }
Yegappan Lakshmanan465de3a2022-12-26 12:50:04 +00001472 // Now that lines are folded scroll to show the cursor at the same
1473 // relative position.
1474 scroll_to_fraction(curwin, curwin->w_height);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475}
1476
1477/*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001478 * Set options to show diffs for the current window.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001479 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001480 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001481ex_diffthis(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001482{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001483 // Set 'diff', 'scrollbind' on and 'wrap' off.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001484 diff_win_options(curwin, TRUE);
1485}
1486
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001487 static void
1488set_diff_option(win_T *wp, int value)
1489{
1490 win_T *old_curwin = curwin;
1491
1492 curwin = wp;
1493 curbuf = curwin->w_buffer;
1494 ++curbuf_lock;
Bram Moolenaar31e5c602022-04-15 13:53:33 +01001495 set_option_value_give_err((char_u *)"diff", (long)value, NULL, OPT_LOCAL);
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001496 --curbuf_lock;
1497 curwin = old_curwin;
1498 curbuf = curwin->w_buffer;
1499}
1500
Bram Moolenaar071d4272004-06-13 20:20:40 +00001501/*
1502 * Set options in window "wp" for diff mode.
1503 */
1504 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001505diff_win_options(
1506 win_T *wp,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001507 int addbuf) // Add buffer to diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001508{
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001509# ifdef FEAT_FOLDING
1510 win_T *old_curwin = curwin;
1511
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001512 // close the manually opened folds
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001513 curwin = wp;
1514 newFoldLevel();
1515 curwin = old_curwin;
1516# endif
1517
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001518 // Use 'scrollbind' and 'cursorbind' when available
Bram Moolenaar43929962015-07-03 15:06:56 +02001519 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001520 wp->w_p_scb_save = wp->w_p_scb;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02001521 wp->w_p_scb = TRUE;
Bram Moolenaar43929962015-07-03 15:06:56 +02001522 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001523 wp->w_p_crb_save = wp->w_p_crb;
Bram Moolenaar860cae12010-06-05 23:22:07 +02001524 wp->w_p_crb = TRUE;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001525 if (!(diff_flags & DIFF_FOLLOWWRAP))
1526 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001527 if (!wp->w_p_diff)
Bram Moolenaar4223d432021-02-10 13:18:17 +01001528 wp->w_p_wrap_save = wp->w_p_wrap;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001529 wp->w_p_wrap = FALSE;
zeertzjq9e7f1fc2024-03-16 09:40:22 +01001530 wp->w_skipcol = 0;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001531 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001532# ifdef FEAT_FOLDING
Bram Moolenaar43929962015-07-03 15:06:56 +02001533 if (!wp->w_p_diff)
1534 {
1535 if (wp->w_p_diff_saved)
1536 free_string_option(wp->w_p_fdm_save);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001537 wp->w_p_fdm_save = vim_strsave(wp->w_p_fdm);
Bram Moolenaar43929962015-07-03 15:06:56 +02001538 }
Bram Moolenaar20c023a2019-05-26 21:03:24 +02001539 set_string_option_direct_in_win(wp, (char_u *)"fdm", -1, (char_u *)"diff",
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00001540 OPT_LOCAL|OPT_FREE, 0);
Bram Moolenaar43929962015-07-03 15:06:56 +02001541 if (!wp->w_p_diff)
Bram Moolenaara87aa802013-07-03 15:47:03 +02001542 {
1543 wp->w_p_fdc_save = wp->w_p_fdc;
1544 wp->w_p_fen_save = wp->w_p_fen;
1545 wp->w_p_fdl_save = wp->w_p_fdl;
1546 }
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001547 wp->w_p_fdc = diff_foldcolumn;
1548 wp->w_p_fen = TRUE;
1549 wp->w_p_fdl = 0;
1550 foldUpdateAll(wp);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001551 // make sure topline is not halfway a fold
Bram Moolenaarf4d7f942010-02-24 14:34:19 +01001552 changed_window_setting_win(wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001553# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554 if (vim_strchr(p_sbo, 'h') == NULL)
1555 do_cmdline_cmd((char_u *)"set sbo+=hor");
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001556 // Save the current values, to be restored in ex_diffoff().
Bram Moolenaara87aa802013-07-03 15:47:03 +02001557 wp->w_p_diff_saved = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001558
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001559 set_diff_option(wp, TRUE);
Bram Moolenaar43929962015-07-03 15:06:56 +02001560
Bram Moolenaar071d4272004-06-13 20:20:40 +00001561 if (addbuf)
1562 diff_buf_add(wp->w_buffer);
Bram Moolenaara4d158b2022-08-14 14:17:45 +01001563 redraw_win_later(wp, UPD_NOT_VALID);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001564}
1565
1566/*
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001567 * Set options not to show diffs. For the current window or all windows.
Bram Moolenaarf740b292006-02-16 22:11:02 +00001568 * Only in the current tab page.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001569 */
1570 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001571ex_diffoff(exarg_T *eap)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001572{
1573 win_T *wp;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001574 int diffwin = FALSE;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001575
Bram Moolenaar29323592016-07-24 22:04:11 +02001576 FOR_ALL_WINDOWS(wp)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001577 {
Bram Moolenaar00462ff2013-09-20 20:13:53 +02001578 if (eap->forceit ? wp->w_p_diff : wp == curwin)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001579 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001580 // Set 'diff' off. If option values were saved in
1581 // diff_win_options(), restore the ones whose settings seem to have
1582 // been left over from diff mode.
Bram Moolenaar04f62f82017-07-19 18:18:39 +02001583 set_diff_option(wp, FALSE);
Bram Moolenaara87aa802013-07-03 15:47:03 +02001584
Bram Moolenaara87aa802013-07-03 15:47:03 +02001585 if (wp->w_p_diff_saved)
1586 {
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001587
Bram Moolenaar43929962015-07-03 15:06:56 +02001588 if (wp->w_p_scb)
1589 wp->w_p_scb = wp->w_p_scb_save;
Bram Moolenaar43929962015-07-03 15:06:56 +02001590 if (wp->w_p_crb)
1591 wp->w_p_crb = wp->w_p_crb_save;
Bram Moolenaar4223d432021-02-10 13:18:17 +01001592 if (!(diff_flags & DIFF_FOLLOWWRAP))
1593 {
zeertzjq9e7f1fc2024-03-16 09:40:22 +01001594 if (!wp->w_p_wrap && wp->w_p_wrap_save)
1595 {
1596 wp->w_p_wrap = TRUE;
1597 wp->w_leftcol = 0;
1598 }
Bram Moolenaar4223d432021-02-10 13:18:17 +01001599 }
Bram Moolenaar43929962015-07-03 15:06:56 +02001600#ifdef FEAT_FOLDING
1601 free_string_option(wp->w_p_fdm);
Bram Moolenaar79a213d2017-05-16 13:15:18 +02001602 wp->w_p_fdm = vim_strsave(
1603 *wp->w_p_fdm_save ? wp->w_p_fdm_save : (char_u*)"manual");
Bram Moolenaar43929962015-07-03 15:06:56 +02001604
1605 if (wp->w_p_fdc == diff_foldcolumn)
1606 wp->w_p_fdc = wp->w_p_fdc_save;
1607 if (wp->w_p_fdl == 0)
1608 wp->w_p_fdl = wp->w_p_fdl_save;
1609
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001610 // Only restore 'foldenable' when 'foldmethod' is not
1611 // "manual", otherwise we continue to show the diff folds.
Bram Moolenaar43929962015-07-03 15:06:56 +02001612 if (wp->w_p_fen)
1613 wp->w_p_fen = foldmethodIsManual(wp) ? FALSE
1614 : wp->w_p_fen_save;
1615
1616 foldUpdateAll(wp);
Bram Moolenaar43929962015-07-03 15:06:56 +02001617#endif
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001618 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001619 // remove filler lines
Bram Moolenaare67d5462016-08-27 22:40:42 +02001620 wp->w_topfill = 0;
1621
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001622 // make sure topline is not halfway a fold and cursor is
1623 // invalidated
Bram Moolenaare67d5462016-08-27 22:40:42 +02001624 changed_window_setting_win(wp);
Bram Moolenaar33ca6bf2013-07-17 13:43:39 +02001625
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001626 // Note: 'sbo' is not restored, it's a global option.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001627 diff_buf_adjust(wp);
1628 }
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001629 diffwin |= wp->w_p_diff;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001630 }
1631
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001632 // Also remove hidden buffers from the list.
Bram Moolenaar25ea0542017-02-03 23:16:28 +01001633 if (eap->forceit)
1634 diff_buf_clear();
1635
Bram Moolenaarc8234772019-11-10 21:00:27 +01001636 if (!diffwin)
1637 {
1638 diff_need_update = FALSE;
1639 curtab->tp_diff_invalid = FALSE;
1640 curtab->tp_diff_update = FALSE;
1641 diff_clear(curtab);
1642 }
1643
Dominique Pelleaf4a61a2021-12-27 17:21:41 +00001644 // Remove "hor" from 'scrollopt' if there are no diff windows left.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001645 if (!diffwin && vim_strchr(p_sbo, 'h') != NULL)
1646 do_cmdline_cmd((char_u *)"set sbo-=hor");
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001647}
1648
1649/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001650 * Read the diff output and add each entry to the diff list.
1651 */
1652 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001653diff_read(
Bram Moolenaare828b762018-09-10 17:51:58 +02001654 int idx_orig, // idx of original file
1655 int idx_new, // idx of new file
Lewis Russelld9da86e2021-12-28 13:54:41 +00001656 diffio_T *dio) // diff output
Bram Moolenaar071d4272004-06-13 20:20:40 +00001657{
Bram Moolenaare828b762018-09-10 17:51:58 +02001658 FILE *fd = NULL;
1659 int line_idx = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001660 diff_T *dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001661 diff_T *dp = curtab->tp_first_diff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001662 diff_T *dn, *dpl;
Lewis Russelld9da86e2021-12-28 13:54:41 +00001663 diffout_T *dout = &dio->dio_diff;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001664 char_u linebuf[LBUFLEN]; // only need to hold the diff line
Bram Moolenaare828b762018-09-10 17:51:58 +02001665 char_u *line;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001666 long off;
1667 int i;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001668 int notset = TRUE; // block "*dp" not set yet
Bram Moolenaar49166972021-12-30 10:51:45 +00001669 diffhunk_T *hunk = NULL; // init to avoid gcc warning
Lewis Russelld9da86e2021-12-28 13:54:41 +00001670
Bram Moolenaare828b762018-09-10 17:51:58 +02001671 enum {
1672 DIFF_ED,
1673 DIFF_UNIFIED,
1674 DIFF_NONE
1675 } diffstyle = DIFF_NONE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001676
Bram Moolenaare828b762018-09-10 17:51:58 +02001677 if (dout->dout_fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001679 diffstyle = DIFF_UNIFIED;
1680 }
1681 else
1682 {
1683 fd = mch_fopen((char *)dout->dout_fname, "r");
1684 if (fd == NULL)
1685 {
Bram Moolenaare1242042021-12-16 20:56:57 +00001686 emsg(_(e_cannot_read_diff_output));
Bram Moolenaare828b762018-09-10 17:51:58 +02001687 return;
1688 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001689 }
1690
Lewis Russelld9da86e2021-12-28 13:54:41 +00001691 if (!dio->dio_internal)
1692 {
1693 hunk = ALLOC_ONE(diffhunk_T);
1694 if (hunk == NULL)
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001695 {
1696 if (fd != NULL)
1697 fclose(fd);
Lewis Russelld9da86e2021-12-28 13:54:41 +00001698 return;
Bram Moolenaar5d46dcf2022-03-25 14:46:47 +00001699 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00001700 }
1701
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702 for (;;)
1703 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001704 if (dio->dio_internal)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001705 {
Bram Moolenaarebfec1c2023-01-22 21:14:53 +00001706 if (line_idx >= dout->dout_ga.ga_len)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001707 break; // did last line
Lewis Russelld9da86e2021-12-28 13:54:41 +00001708 hunk = ((diffhunk_T **)dout->dout_ga.ga_data)[line_idx++];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001709 }
1710 else
1711 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00001712 if (fd == NULL)
1713 {
1714 if (line_idx >= dout->dout_ga.ga_len)
1715 break; // did last line
1716 line = ((char_u **)dout->dout_ga.ga_data)[line_idx++];
1717 }
Bram Moolenaar3b8defd2018-09-13 13:03:11 +02001718 else
Lewis Russelld9da86e2021-12-28 13:54:41 +00001719 {
1720 if (vim_fgets(linebuf, LBUFLEN, fd))
1721 break; // end of file
1722 line = linebuf;
1723 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001724
Lewis Russelld9da86e2021-12-28 13:54:41 +00001725 if (diffstyle == DIFF_NONE)
1726 {
1727 // Determine diff style.
1728 // ed like diff looks like this:
1729 // {first}[,{last}]c{first}[,{last}]
1730 // {first}a{first}[,{last}]
1731 // {first}[,{last}]d{first}
1732 //
1733 // unified diff looks like this:
1734 // --- file1 2018-03-20 13:23:35.783153140 +0100
1735 // +++ file2 2018-03-20 13:23:41.183156066 +0100
1736 // @@ -1,3 +1,5 @@
Keith Thompson184f71c2024-01-04 21:19:04 +01001737 if (SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001738 diffstyle = DIFF_ED;
1739 else if ((STRNCMP(line, "@@ ", 3) == 0))
1740 diffstyle = DIFF_UNIFIED;
1741 else if ((STRNCMP(line, "--- ", 4) == 0)
1742 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1743 && (STRNCMP(line, "+++ ", 4) == 0)
1744 && (vim_fgets(linebuf, LBUFLEN, fd) == 0)
1745 && (STRNCMP(line, "@@ ", 3) == 0))
1746 diffstyle = DIFF_UNIFIED;
1747 else
1748 // Format not recognized yet, skip over this line. Cygwin
1749 // diff may put a warning at the start of the file.
1750 continue;
1751 }
1752
1753 if (diffstyle == DIFF_ED)
1754 {
Keith Thompson184f71c2024-01-04 21:19:04 +01001755 if (!SAFE_isdigit(*line))
Lewis Russelld9da86e2021-12-28 13:54:41 +00001756 continue; // not the start of a diff block
1757 if (parse_diff_ed(line, hunk) == FAIL)
1758 continue;
1759 }
1760 else if (diffstyle == DIFF_UNIFIED)
1761 {
1762 if (STRNCMP(line, "@@ ", 3) != 0)
1763 continue; // not the start of a diff block
1764 if (parse_diff_unified(line, hunk) == FAIL)
1765 continue;
1766 }
1767 else
1768 {
Bram Moolenaard82a47d2022-01-05 20:24:39 +00001769 emsg(_(e_invalid_diff_format));
Lewis Russelld9da86e2021-12-28 13:54:41 +00001770 break;
1771 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001772 }
1773
1774 // Go over blocks before the change, for which orig and new are equal.
1775 // Copy blocks from orig to new.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001776 while (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001777 && hunk->lnum_orig > dp->df_lnum[idx_orig]
1778 + dp->df_count[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001779 {
1780 if (notset)
1781 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1782 dprev = dp;
1783 dp = dp->df_next;
1784 notset = TRUE;
1785 }
1786
1787 if (dp != NULL
Lewis Russelld9da86e2021-12-28 13:54:41 +00001788 && hunk->lnum_orig <= dp->df_lnum[idx_orig]
1789 + dp->df_count[idx_orig]
1790 && hunk->lnum_orig + hunk->count_orig >= dp->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001791 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001792 // New block overlaps with existing block(s).
1793 // First find last block that overlaps.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001794 for (dpl = dp; dpl->df_next != NULL; dpl = dpl->df_next)
Lewis Russelld9da86e2021-12-28 13:54:41 +00001795 if (hunk->lnum_orig + hunk->count_orig
1796 < dpl->df_next->df_lnum[idx_orig])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797 break;
1798
Bram Moolenaare828b762018-09-10 17:51:58 +02001799 // If the newly found block starts before the old one, set the
1800 // start back a number of lines.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001801 off = dp->df_lnum[idx_orig] - hunk->lnum_orig;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802 if (off > 0)
1803 {
1804 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001805 if (curtab->tp_diffbuf[i] != NULL)
Yukihiro Nakadaira06fe70c2024-09-26 16:19:42 +02001806 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001807 dp->df_lnum[i] -= off;
Yukihiro Nakadaira06fe70c2024-09-26 16:19:42 +02001808 dp->df_count[i] += off;
1809 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00001810 dp->df_lnum[idx_new] = hunk->lnum_new;
1811 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001812 }
1813 else if (notset)
1814 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001815 // new block inside existing one, adjust new block
Lewis Russelld9da86e2021-12-28 13:54:41 +00001816 dp->df_lnum[idx_new] = hunk->lnum_new + off;
1817 dp->df_count[idx_new] = hunk->count_new - off;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001818 }
1819 else
Bram Moolenaare828b762018-09-10 17:51:58 +02001820 // second overlap of new block with existing block
Yukihiro Nakadaira06fe70c2024-09-26 16:19:42 +02001821 dp->df_count[idx_new] += hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001822
Bram Moolenaare828b762018-09-10 17:51:58 +02001823 // Adjust the size of the block to include all the lines to the
1824 // end of the existing block or the new diff, whatever ends last.
Lewis Russelld9da86e2021-12-28 13:54:41 +00001825 off = (hunk->lnum_orig + hunk->count_orig)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 - (dpl->df_lnum[idx_orig] + dpl->df_count[idx_orig]);
1827 if (off < 0)
1828 {
Yukihiro Nakadaira06fe70c2024-09-26 16:19:42 +02001829 // new change ends in existing block, adjust the end
1830 dp->df_count[idx_new] += -off;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001831 off = 0;
1832 }
Bram Moolenaard4b96bc2007-10-19 15:33:39 +00001833 for (i = idx_orig; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001834 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001835 dp->df_count[i] = dpl->df_lnum[i] + dpl->df_count[i]
1836 - dp->df_lnum[i] + off;
1837
Bram Moolenaare828b762018-09-10 17:51:58 +02001838 // Delete the diff blocks that have been merged into one.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839 dn = dp->df_next;
1840 dp->df_next = dpl->df_next;
1841 while (dn != dp->df_next)
1842 {
1843 dpl = dn->df_next;
1844 vim_free(dn);
1845 dn = dpl;
1846 }
1847 }
1848 else
1849 {
Bram Moolenaare828b762018-09-10 17:51:58 +02001850 // Allocate a new diffblock.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001851 dp = diff_alloc_new(curtab, dprev, dp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001852 if (dp == NULL)
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001853 goto done;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854
Lewis Russelld9da86e2021-12-28 13:54:41 +00001855 dp->df_lnum[idx_orig] = hunk->lnum_orig;
1856 dp->df_count[idx_orig] = hunk->count_orig;
1857 dp->df_lnum[idx_new] = hunk->lnum_new;
1858 dp->df_count[idx_new] = hunk->count_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859
Bram Moolenaare828b762018-09-10 17:51:58 +02001860 // Set values for other buffers, these must be equal to the
1861 // original buffer, otherwise there would have been a change
1862 // already.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 for (i = idx_orig + 1; i < idx_new; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001864 if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865 diff_copy_entry(dprev, dp, idx_orig, i);
1866 }
Bram Moolenaare828b762018-09-10 17:51:58 +02001867 notset = FALSE; // "*dp" has been set
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868 }
1869
Bram Moolenaare828b762018-09-10 17:51:58 +02001870 // for remaining diff blocks orig and new are equal
Bram Moolenaar071d4272004-06-13 20:20:40 +00001871 while (dp != NULL)
1872 {
1873 if (notset)
1874 diff_copy_entry(dprev, dp, idx_orig, idx_new);
1875 dprev = dp;
1876 dp = dp->df_next;
1877 notset = TRUE;
1878 }
1879
Bram Moolenaareb3593b2006-04-22 22:33:57 +00001880done:
Lewis Russelld9da86e2021-12-28 13:54:41 +00001881 if (!dio->dio_internal)
1882 vim_free(hunk);
1883
Bram Moolenaare828b762018-09-10 17:51:58 +02001884 if (fd != NULL)
1885 fclose(fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886}
1887
1888/*
1889 * Copy an entry at "dp" from "idx_orig" to "idx_new".
1890 */
1891 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001892diff_copy_entry(
1893 diff_T *dprev,
1894 diff_T *dp,
1895 int idx_orig,
1896 int idx_new)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001897{
1898 long off;
1899
1900 if (dprev == NULL)
1901 off = 0;
1902 else
1903 off = (dprev->df_lnum[idx_orig] + dprev->df_count[idx_orig])
1904 - (dprev->df_lnum[idx_new] + dprev->df_count[idx_new]);
1905 dp->df_lnum[idx_new] = dp->df_lnum[idx_orig] - off;
1906 dp->df_count[idx_new] = dp->df_count[idx_orig];
1907}
1908
1909/*
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001910 * Clear the list of diffblocks for tab page "tp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911 */
Bram Moolenaarea408852005-06-25 22:49:46 +00001912 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01001913diff_clear(tabpage_T *tp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001914{
1915 diff_T *p, *next_p;
1916
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001917 for (p = tp->tp_first_diff; p != NULL; p = next_p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001918 {
1919 next_p = p->df_next;
1920 vim_free(p);
1921 }
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001922 tp->tp_first_diff = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001923}
1924
1925/*
1926 * Check diff status for line "lnum" in buffer "buf":
1927 * Returns 0 for nothing special
1928 * Returns -1 for a line that should be highlighted as changed.
1929 * Returns -2 for a line that should be highlighted as added/deleted.
1930 * Returns > 0 for inserting that many filler lines above it (never happens
1931 * when 'diffopt' doesn't contain "filler").
1932 * This should only be used for windows where 'diff' is set.
1933 */
1934 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01001935diff_check(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001936{
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001937 int idx; // index in tp_diffbuf[] for this buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00001938 diff_T *dp;
1939 int maxcount;
1940 int i;
1941 buf_T *buf = wp->w_buffer;
1942 int cmp;
1943
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001944 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001945 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00001946
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001947 if (curtab->tp_first_diff == NULL || !wp->w_p_diff) // no diffs at all
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 return 0;
1949
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001950 // safety check: "lnum" must be a buffer line
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951 if (lnum < 1 || lnum > buf->b_ml.ml_line_count + 1)
1952 return 0;
1953
1954 idx = diff_buf_idx(buf);
1955 if (idx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001956 return 0; // no diffs for buffer "buf"
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957
1958#ifdef FEAT_FOLDING
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001959 // A closed fold never has filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960 if (hasFoldingWin(wp, lnum, NULL, NULL, TRUE, NULL))
1961 return 0;
1962#endif
1963
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001964 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02001965 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001966 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
1967 break;
1968 if (dp == NULL || lnum < dp->df_lnum[idx])
1969 return 0;
1970
1971 if (lnum < dp->df_lnum[idx] + dp->df_count[idx])
1972 {
1973 int zero = FALSE;
1974
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001975 // Changed or inserted line. If the other buffers have a count of
1976 // zero, the lines were inserted. If the other buffers have the same
1977 // count, check if the lines are identical.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978 cmp = FALSE;
1979 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00001980 if (i != idx && curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001981 {
1982 if (dp->df_count[i] == 0)
1983 zero = TRUE;
1984 else
1985 {
1986 if (dp->df_count[i] != dp->df_count[idx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001987 return -1; // nr of lines changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001988 cmp = TRUE;
1989 }
1990 }
1991 if (cmp)
1992 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01001993 // Compare all lines. If they are equal the lines were inserted
1994 // in some buffers, deleted in others, but not changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001995 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001996 if (i != idx && curtab->tp_diffbuf[i] != NULL
1997 && dp->df_count[i] != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001998 if (!diff_equal_entry(dp, idx, i))
1999 return -1;
2000 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002001 // If there is no buffer with zero lines then there is no difference
2002 // any longer. Happens when making a change (or undo) that removes
2003 // the difference. Can't remove the entry here, we might be halfway
2004 // updating the window. Just report the text as unchanged. Other
2005 // windows might still show the change though.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002006 if (zero == FALSE)
2007 return 0;
2008 return -2;
2009 }
2010
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002011 // If 'diffopt' doesn't contain "filler", return 0.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002012 if (!(diff_flags & DIFF_FILLER))
2013 return 0;
2014
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002015 // Insert filler lines above the line just below the change. Will return
2016 // 0 when this buf had the max count.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 maxcount = 0;
2018 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002019 if (curtab->tp_diffbuf[i] != NULL && dp->df_count[i] > maxcount)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002020 maxcount = dp->df_count[i];
2021 return maxcount - dp->df_count[idx];
2022}
2023
2024/*
2025 * Compare two entries in diff "*dp" and return TRUE if they are equal.
2026 */
2027 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002028diff_equal_entry(diff_T *dp, int idx1, int idx2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002029{
2030 int i;
2031 char_u *line;
2032 int cmp;
2033
2034 if (dp->df_count[idx1] != dp->df_count[idx2])
2035 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002036 if (diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037 return FALSE;
2038 for (i = 0; i < dp->df_count[idx1]; ++i)
2039 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002040 line = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx1],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002041 dp->df_lnum[idx1] + i, FALSE));
2042 if (line == NULL)
2043 return FALSE;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002044 cmp = diff_cmp(line, ml_get_buf(curtab->tp_diffbuf[idx2],
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045 dp->df_lnum[idx2] + i, FALSE));
2046 vim_free(line);
2047 if (cmp != 0)
2048 return FALSE;
2049 }
2050 return TRUE;
2051}
2052
2053/*
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002054 * Compare the characters at "p1" and "p2". If they are equal (possibly
2055 * ignoring case) return TRUE and set "len" to the number of bytes.
2056 */
2057 static int
2058diff_equal_char(char_u *p1, char_u *p2, int *len)
2059{
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002060 int l = (*mb_ptr2len)(p1);
2061
2062 if (l != (*mb_ptr2len)(p2))
2063 return FALSE;
2064 if (l > 1)
2065 {
2066 if (STRNCMP(p1, p2, l) != 0
2067 && (!enc_utf8
2068 || !(diff_flags & DIFF_ICASE)
2069 || utf_fold(utf_ptr2char(p1))
2070 != utf_fold(utf_ptr2char(p2))))
2071 return FALSE;
2072 *len = l;
2073 }
2074 else
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002075 {
2076 if ((*p1 != *p2)
2077 && (!(diff_flags & DIFF_ICASE)
2078 || TOLOWER_LOC(*p1) != TOLOWER_LOC(*p2)))
2079 return FALSE;
2080 *len = 1;
2081 }
2082 return TRUE;
2083}
2084
2085/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002086 * Compare strings "s1" and "s2" according to 'diffopt'.
2087 * Return non-zero when they are different.
2088 */
2089 static int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002090diff_cmp(char_u *s1, char_u *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002091{
2092 char_u *p1, *p2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002093 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002094
Bram Moolenaar785fc652018-09-15 19:17:38 +02002095 if ((diff_flags & DIFF_IBLANK)
2096 && (*skipwhite(s1) == NUL || *skipwhite(s2) == NUL))
2097 return 0;
2098
2099 if ((diff_flags & (DIFF_ICASE | ALL_WHITE_DIFF)) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002100 return STRCMP(s1, s2);
Bram Moolenaar785fc652018-09-15 19:17:38 +02002101 if ((diff_flags & DIFF_ICASE) && !(diff_flags & ALL_WHITE_DIFF))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 return MB_STRICMP(s1, s2);
2103
Bram Moolenaar071d4272004-06-13 20:20:40 +00002104 p1 = s1;
2105 p2 = s2;
Bram Moolenaar785fc652018-09-15 19:17:38 +02002106
2107 // Ignore white space changes and possibly ignore case.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108 while (*p1 != NUL && *p2 != NUL)
2109 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002110 if (((diff_flags & DIFF_IWHITE)
2111 && VIM_ISWHITE(*p1) && VIM_ISWHITE(*p2))
2112 || ((diff_flags & DIFF_IWHITEALL)
2113 && (VIM_ISWHITE(*p1) || VIM_ISWHITE(*p2))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114 {
2115 p1 = skipwhite(p1);
2116 p2 = skipwhite(p2);
2117 }
2118 else
2119 {
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002120 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121 break;
Bram Moolenaarae96b8d2017-09-03 15:04:21 +02002122 p1 += l;
2123 p2 += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002124 }
2125 }
2126
Bram Moolenaar785fc652018-09-15 19:17:38 +02002127 // Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002128 p1 = skipwhite(p1);
2129 p2 = skipwhite(p2);
2130 if (*p1 != NUL || *p2 != NUL)
2131 return 1;
2132 return 0;
2133}
2134
2135/*
2136 * Return the number of filler lines above "lnum".
2137 */
2138 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002139diff_check_fill(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002140{
2141 int n;
2142
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002143 // be quick when there are no filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002144 if (!(diff_flags & DIFF_FILLER))
2145 return 0;
2146 n = diff_check(wp, lnum);
2147 if (n <= 0)
2148 return 0;
2149 return n;
2150}
2151
2152/*
2153 * Set the topline of "towin" to match the position in "fromwin", so that they
2154 * show the same diff'ed lines.
2155 */
2156 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002157diff_set_topline(win_T *fromwin, win_T *towin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002158{
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002159 buf_T *frombuf = fromwin->w_buffer;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160 linenr_T lnum = fromwin->w_topline;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002161 int fromidx;
2162 int toidx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163 diff_T *dp;
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002164 int max_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002165 int i;
2166
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002167 fromidx = diff_buf_idx(frombuf);
2168 if (fromidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002169 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002170
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002171 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002172 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173
2174 towin->w_topfill = 0;
2175
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002176 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002177 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002178 if (lnum <= dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002179 break;
2180 if (dp == NULL)
2181 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002182 // After last change, compute topline relative to end of file; no
2183 // filler lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002184 towin->w_topline = towin->w_buffer->b_ml.ml_line_count
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002185 - (frombuf->b_ml.ml_line_count - lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002186 }
2187 else
2188 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002189 // Find index for "towin".
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002190 toidx = diff_buf_idx(towin->w_buffer);
2191 if (toidx == DB_COUNT)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002192 return; // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00002193
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002194 towin->w_topline = lnum + (dp->df_lnum[toidx] - dp->df_lnum[fromidx]);
2195 if (lnum >= dp->df_lnum[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002196 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002197 // Inside a change: compute filler lines. With three or more
2198 // buffers we need to know the largest count.
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002199 max_count = 0;
2200 for (i = 0; i < DB_COUNT; ++i)
2201 if (curtab->tp_diffbuf[i] != NULL
2202 && max_count < dp->df_count[i])
2203 max_count = dp->df_count[i];
2204
2205 if (dp->df_count[toidx] == dp->df_count[fromidx])
2206 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002207 // same number of lines: use same filler count
Bram Moolenaar071d4272004-06-13 20:20:40 +00002208 towin->w_topfill = fromwin->w_topfill;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002209 }
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002210 else if (dp->df_count[toidx] > dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002211 {
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002212 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00002213 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002214 // more lines in towin and fromwin doesn't show diff
2215 // lines, only filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002216 if (max_count - fromwin->w_topfill >= dp->df_count[toidx])
2217 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002218 // towin also only shows filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002219 towin->w_topline = dp->df_lnum[toidx]
2220 + dp->df_count[toidx];
2221 towin->w_topfill = fromwin->w_topfill;
2222 }
2223 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002224 // towin still has some diff lines to show
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002225 towin->w_topline = dp->df_lnum[toidx]
2226 + max_count - fromwin->w_topfill;
2227 }
2228 }
2229 else if (towin->w_topline >= dp->df_lnum[toidx]
2230 + dp->df_count[toidx])
2231 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002232 // less lines in towin and no diff lines to show: compute
2233 // filler lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002234 towin->w_topline = dp->df_lnum[toidx] + dp->df_count[toidx];
2235 if (diff_flags & DIFF_FILLER)
2236 {
2237 if (lnum == dp->df_lnum[fromidx] + dp->df_count[fromidx])
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002238 // fromwin is also out of diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002239 towin->w_topfill = fromwin->w_topfill;
2240 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002241 // fromwin has some diff lines
Bram Moolenaarbb8f88b2008-01-18 16:40:00 +00002242 towin->w_topfill = dp->df_lnum[fromidx]
2243 + max_count - lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244 }
2245 }
2246 }
2247 }
2248
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002249 // safety check (if diff info gets outdated strange things may happen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002250 towin->w_botfill = FALSE;
2251 if (towin->w_topline > towin->w_buffer->b_ml.ml_line_count)
2252 {
2253 towin->w_topline = towin->w_buffer->b_ml.ml_line_count;
2254 towin->w_botfill = TRUE;
2255 }
2256 if (towin->w_topline < 1)
2257 {
2258 towin->w_topline = 1;
2259 towin->w_topfill = 0;
2260 }
2261
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002262 // When w_topline changes need to recompute w_botline and cursor position
Bram Moolenaar071d4272004-06-13 20:20:40 +00002263 invalidate_botline_win(towin);
2264 changed_line_abv_curs_win(towin);
2265
2266 check_topfill(towin, FALSE);
2267#ifdef FEAT_FOLDING
2268 (void)hasFoldingWin(towin, towin->w_topline, &towin->w_topline,
2269 NULL, TRUE, NULL);
2270#endif
2271}
2272
2273/*
2274 * This is called when 'diffopt' is changed.
2275 */
2276 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002277diffopt_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002278{
2279 char_u *p;
2280 int diff_context_new = 6;
2281 int diff_flags_new = 0;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002282 int diff_foldcolumn_new = 2;
Bram Moolenaare828b762018-09-10 17:51:58 +02002283 long diff_algorithm_new = 0;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002284 long diff_indent_heuristic = 0;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002285 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002286
2287 p = p_dip;
2288 while (*p != NUL)
2289 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002290 // Note: Keep this in sync with p_dip_values
Bram Moolenaar071d4272004-06-13 20:20:40 +00002291 if (STRNCMP(p, "filler", 6) == 0)
2292 {
2293 p += 6;
2294 diff_flags_new |= DIFF_FILLER;
2295 }
2296 else if (STRNCMP(p, "context:", 8) == 0 && VIM_ISDIGIT(p[8]))
2297 {
2298 p += 8;
2299 diff_context_new = getdigits(&p);
2300 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002301 else if (STRNCMP(p, "iblank", 6) == 0)
2302 {
2303 p += 6;
2304 diff_flags_new |= DIFF_IBLANK;
2305 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002306 else if (STRNCMP(p, "icase", 5) == 0)
2307 {
2308 p += 5;
2309 diff_flags_new |= DIFF_ICASE;
2310 }
Bram Moolenaar785fc652018-09-15 19:17:38 +02002311 else if (STRNCMP(p, "iwhiteall", 9) == 0)
2312 {
2313 p += 9;
2314 diff_flags_new |= DIFF_IWHITEALL;
2315 }
2316 else if (STRNCMP(p, "iwhiteeol", 9) == 0)
2317 {
2318 p += 9;
2319 diff_flags_new |= DIFF_IWHITEEOL;
2320 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321 else if (STRNCMP(p, "iwhite", 6) == 0)
2322 {
2323 p += 6;
2324 diff_flags_new |= DIFF_IWHITE;
2325 }
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002326 else if (STRNCMP(p, "horizontal", 10) == 0)
2327 {
2328 p += 10;
2329 diff_flags_new |= DIFF_HORIZONTAL;
2330 }
2331 else if (STRNCMP(p, "vertical", 8) == 0)
2332 {
2333 p += 8;
2334 diff_flags_new |= DIFF_VERTICAL;
2335 }
2336 else if (STRNCMP(p, "foldcolumn:", 11) == 0 && VIM_ISDIGIT(p[11]))
2337 {
2338 p += 11;
2339 diff_foldcolumn_new = getdigits(&p);
2340 }
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002341 else if (STRNCMP(p, "hiddenoff", 9) == 0)
2342 {
2343 p += 9;
2344 diff_flags_new |= DIFF_HIDDEN_OFF;
2345 }
Bram Moolenaarc8234772019-11-10 21:00:27 +01002346 else if (STRNCMP(p, "closeoff", 8) == 0)
2347 {
2348 p += 8;
2349 diff_flags_new |= DIFF_CLOSE_OFF;
2350 }
Bram Moolenaar4223d432021-02-10 13:18:17 +01002351 else if (STRNCMP(p, "followwrap", 10) == 0)
2352 {
2353 p += 10;
2354 diff_flags_new |= DIFF_FOLLOWWRAP;
2355 }
Bram Moolenaare828b762018-09-10 17:51:58 +02002356 else if (STRNCMP(p, "indent-heuristic", 16) == 0)
2357 {
2358 p += 16;
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002359 diff_indent_heuristic = XDF_INDENT_HEURISTIC;
Bram Moolenaare828b762018-09-10 17:51:58 +02002360 }
2361 else if (STRNCMP(p, "internal", 8) == 0)
2362 {
2363 p += 8;
2364 diff_flags_new |= DIFF_INTERNAL;
2365 }
2366 else if (STRNCMP(p, "algorithm:", 10) == 0)
2367 {
Yee Cheng Chin900894b2023-09-29 20:42:32 +02002368 // Note: Keep this in sync with p_dip_algorithm_values.
Bram Moolenaare828b762018-09-10 17:51:58 +02002369 p += 10;
2370 if (STRNCMP(p, "myers", 5) == 0)
2371 {
2372 p += 5;
2373 diff_algorithm_new = 0;
2374 }
2375 else if (STRNCMP(p, "minimal", 7) == 0)
2376 {
2377 p += 7;
2378 diff_algorithm_new = XDF_NEED_MINIMAL;
2379 }
2380 else if (STRNCMP(p, "patience", 8) == 0)
2381 {
2382 p += 8;
2383 diff_algorithm_new = XDF_PATIENCE_DIFF;
2384 }
2385 else if (STRNCMP(p, "histogram", 9) == 0)
2386 {
2387 p += 9;
2388 diff_algorithm_new = XDF_HISTOGRAM_DIFF;
2389 }
Bram Moolenaard0721052018-11-05 21:21:33 +01002390 else
2391 return FAIL;
Bram Moolenaare828b762018-09-10 17:51:58 +02002392 }
2393
Bram Moolenaar071d4272004-06-13 20:20:40 +00002394 if (*p != ',' && *p != NUL)
2395 return FAIL;
2396 if (*p == ',')
2397 ++p;
2398 }
2399
Bram Moolenaarb6fc7282018-12-04 22:24:16 +01002400 diff_algorithm_new |= diff_indent_heuristic;
2401
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002402 // Can't have both "horizontal" and "vertical".
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002403 if ((diff_flags_new & DIFF_HORIZONTAL) && (diff_flags_new & DIFF_VERTICAL))
2404 return FAIL;
2405
Bram Moolenaar198fa062018-09-18 21:20:26 +02002406 // If flags were added or removed, or the algorithm was changed, need to
2407 // update the diff.
Bram Moolenaare828b762018-09-10 17:51:58 +02002408 if (diff_flags != diff_flags_new || diff_algorithm != diff_algorithm_new)
Bram Moolenaar29323592016-07-24 22:04:11 +02002409 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002410 tp->tp_diff_invalid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411
2412 diff_flags = diff_flags_new;
Bram Moolenaarb9ddda62019-02-19 23:00:50 +01002413 diff_context = diff_context_new == 0 ? 1 : diff_context_new;
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002414 diff_foldcolumn = diff_foldcolumn_new;
Bram Moolenaare828b762018-09-10 17:51:58 +02002415 diff_algorithm = diff_algorithm_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416
2417 diff_redraw(TRUE);
2418
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002419 // recompute the scroll binding with the new option value, may
2420 // remove or add filler lines
Bram Moolenaar071d4272004-06-13 20:20:40 +00002421 check_scrollbind((linenr_T)0, 0L);
2422
2423 return OK;
2424}
2425
2426/*
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002427 * Return TRUE if 'diffopt' contains "horizontal".
2428 */
2429 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002430diffopt_horizontal(void)
Bram Moolenaarc4675a12006-03-15 22:50:30 +00002431{
2432 return (diff_flags & DIFF_HORIZONTAL) != 0;
2433}
2434
2435/*
Bram Moolenaar97ce4192017-12-01 20:35:58 +01002436 * Return TRUE if 'diffopt' contains "hiddenoff".
2437 */
2438 int
2439diffopt_hiddenoff(void)
2440{
2441 return (diff_flags & DIFF_HIDDEN_OFF) != 0;
2442}
2443
2444/*
Bram Moolenaarc8234772019-11-10 21:00:27 +01002445 * Return TRUE if 'diffopt' contains "closeoff".
2446 */
2447 int
2448diffopt_closeoff(void)
2449{
2450 return (diff_flags & DIFF_CLOSE_OFF) != 0;
2451}
2452
2453/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002454 * Find the difference within a changed line.
2455 * Returns TRUE if the line was added, no other buffer has it.
2456 */
2457 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002458diff_find_change(
2459 win_T *wp,
2460 linenr_T lnum,
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002461 int *startp, // first char of the change
2462 int *endp) // last char of the change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002463{
2464 char_u *line_org;
2465 char_u *line_new;
2466 int i;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002467 int si_org, si_new;
2468 int ei_org, ei_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469 diff_T *dp;
2470 int idx;
2471 int off;
2472 int added = TRUE;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002473 char_u *p1, *p2;
2474 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002475
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002476 // Make a copy of the line, the next ml_get() will invalidate it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002477 line_org = vim_strsave(ml_get_buf(wp->w_buffer, lnum, FALSE));
2478 if (line_org == NULL)
2479 return FALSE;
2480
2481 idx = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002482 if (idx == DB_COUNT) // cannot happen
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002483 {
2484 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002486 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002488 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002489 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002490 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
2491 break;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002492 if (dp == NULL || diff_check_sanity(curtab, dp) == FAIL)
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002493 {
2494 vim_free(line_org);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002495 return FALSE;
Bram Moolenaarfa3491a2007-02-20 02:49:19 +00002496 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002497
2498 off = lnum - dp->df_lnum[idx];
2499
2500 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002501 if (curtab->tp_diffbuf[i] != NULL && i != idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002502 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002503 // Skip lines that are not in the other change (filler lines).
Bram Moolenaar071d4272004-06-13 20:20:40 +00002504 if (off >= dp->df_count[i])
2505 continue;
2506 added = FALSE;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002507 line_new = ml_get_buf(curtab->tp_diffbuf[i],
2508 dp->df_lnum[i] + off, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002509
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002510 // Search for start of difference
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002511 si_org = si_new = 0;
2512 while (line_org[si_org] != NUL)
2513 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002514 if (((diff_flags & DIFF_IWHITE)
2515 && VIM_ISWHITE(line_org[si_org])
2516 && VIM_ISWHITE(line_new[si_new]))
2517 || ((diff_flags & DIFF_IWHITEALL)
2518 && (VIM_ISWHITE(line_org[si_org])
2519 || VIM_ISWHITE(line_new[si_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002520 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002521 si_org = (int)(skipwhite(line_org + si_org) - line_org);
2522 si_new = (int)(skipwhite(line_new + si_new) - line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002523 }
2524 else
2525 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002526 if (!diff_equal_char(line_org + si_org, line_new + si_new,
2527 &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002528 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002529 si_org += l;
2530 si_new += l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002531 }
2532 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002533 if (has_mbyte)
2534 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002535 // Move back to first byte of character in both lines (may
2536 // have "nn^" in line_org and "n^ in line_new).
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002537 si_org -= (*mb_head_off)(line_org, line_org + si_org);
2538 si_new -= (*mb_head_off)(line_new, line_new + si_new);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002539 }
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002540 if (*startp > si_org)
2541 *startp = si_org;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002542
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002543 // Search for end of difference, if any.
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002544 if (line_org[si_org] != NUL || line_new[si_new] != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002545 {
2546 ei_org = (int)STRLEN(line_org);
2547 ei_new = (int)STRLEN(line_new);
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002548 while (ei_org >= *startp && ei_new >= si_new
2549 && ei_org >= 0 && ei_new >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002550 {
Bram Moolenaar785fc652018-09-15 19:17:38 +02002551 if (((diff_flags & DIFF_IWHITE)
2552 && VIM_ISWHITE(line_org[ei_org])
2553 && VIM_ISWHITE(line_new[ei_new]))
2554 || ((diff_flags & DIFF_IWHITEALL)
2555 && (VIM_ISWHITE(line_org[ei_org])
2556 || VIM_ISWHITE(line_new[ei_new]))))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002557 {
2558 while (ei_org >= *startp
Bram Moolenaar1c465442017-03-12 20:10:05 +01002559 && VIM_ISWHITE(line_org[ei_org]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002560 --ei_org;
2561 while (ei_new >= si_new
Bram Moolenaar1c465442017-03-12 20:10:05 +01002562 && VIM_ISWHITE(line_new[ei_new]))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002563 --ei_new;
2564 }
2565 else
2566 {
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002567 p1 = line_org + ei_org;
2568 p2 = line_new + ei_new;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002569 p1 -= (*mb_head_off)(line_org, p1);
2570 p2 -= (*mb_head_off)(line_new, p2);
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002571 if (!diff_equal_char(p1, p2, &l))
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002572 break;
Bram Moolenaarda22b8c2017-09-02 18:01:50 +02002573 ei_org -= l;
2574 ei_new -= l;
Bram Moolenaar9e54a0e2006-04-14 20:42:25 +00002575 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002576 }
2577 if (*endp < ei_org)
2578 *endp = ei_org;
2579 }
2580 }
2581
2582 vim_free(line_org);
2583 return added;
2584}
2585
2586#if defined(FEAT_FOLDING) || defined(PROTO)
2587/*
2588 * Return TRUE if line "lnum" is not close to a diff block, this line should
2589 * be in a fold.
2590 * Return FALSE if there are no diff blocks at all in this window.
2591 */
2592 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01002593diff_infold(win_T *wp, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002594{
2595 int i;
2596 int idx = -1;
2597 int other = FALSE;
2598 diff_T *dp;
2599
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002600 // Return if 'diff' isn't set.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002601 if (!wp->w_p_diff)
2602 return FALSE;
2603
2604 for (i = 0; i < DB_COUNT; ++i)
2605 {
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002606 if (curtab->tp_diffbuf[i] == wp->w_buffer)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002607 idx = i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002608 else if (curtab->tp_diffbuf[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002609 other = TRUE;
2610 }
2611
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002612 // return here if there are no diffs in the window
Bram Moolenaar071d4272004-06-13 20:20:40 +00002613 if (idx == -1 || !other)
2614 return FALSE;
2615
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002616 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002617 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00002618
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002619 // Return if there are no diff blocks. All lines will be folded.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002620 if (curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002621 return TRUE;
2622
Bram Moolenaaraeea7212020-04-02 18:50:46 +02002623 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002624 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002625 // If this change is below the line there can't be any further match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002626 if (dp->df_lnum[idx] - diff_context > lnum)
2627 break;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002628 // If this change ends before the line we have a match.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002629 if (dp->df_lnum[idx] + dp->df_count[idx] + diff_context > lnum)
2630 return FALSE;
2631 }
2632 return TRUE;
2633}
2634#endif
2635
2636/*
2637 * "dp" and "do" commands.
2638 */
2639 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002640nv_diffgetput(int put, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002641{
2642 exarg_T ea;
Bram Moolenaar6a643652014-10-31 13:54:25 +01002643 char_u buf[30];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002644
Bram Moolenaarf2732452018-06-03 14:47:35 +02002645#ifdef FEAT_JOB_CHANNEL
2646 if (bt_prompt(curbuf))
2647 {
2648 vim_beep(BO_OPER);
2649 return;
2650 }
2651#endif
Bram Moolenaar6a643652014-10-31 13:54:25 +01002652 if (count == 0)
2653 ea.arg = (char_u *)"";
2654 else
2655 {
2656 vim_snprintf((char *)buf, 30, "%ld", count);
2657 ea.arg = buf;
2658 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002659 if (put)
2660 ea.cmdidx = CMD_diffput;
2661 else
2662 ea.cmdidx = CMD_diffget;
2663 ea.addr_count = 0;
2664 ea.line1 = curwin->w_cursor.lnum;
2665 ea.line2 = curwin->w_cursor.lnum;
2666 ex_diffgetput(&ea);
2667}
2668
2669/*
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002670 * Return TRUE if "diff" appears in the list of diff blocks of the current tab.
2671 */
2672 static int
2673valid_diff(diff_T *diff)
2674{
2675 diff_T *dp;
2676
Yegappan Lakshmanan14113fd2023-03-07 17:13:51 +00002677 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002678 if (dp == diff)
2679 return TRUE;
2680 return FALSE;
2681}
2682
2683/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002684 * ":diffget"
2685 * ":diffput"
2686 */
2687 void
Bram Moolenaar7454a062016-01-30 15:14:10 +01002688ex_diffgetput(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002689{
2690 linenr_T lnum;
2691 int count;
2692 linenr_T off = 0;
2693 diff_T *dp;
2694 diff_T *dprev;
2695 diff_T *dfree;
2696 int idx_cur;
2697 int idx_other;
2698 int idx_from;
2699 int idx_to;
2700 int i;
2701 int added;
2702 char_u *p;
2703 aco_save_T aco;
2704 buf_T *buf;
2705 int start_skip, end_skip;
2706 int new_count;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002707 int buf_empty;
Bram Moolenaar602eb742007-02-20 03:43:38 +00002708 int found_not_ma = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002710 // Find the current buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002711 idx_cur = diff_buf_idx(curbuf);
2712 if (idx_cur == DB_COUNT)
2713 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002714 emsg(_(e_current_buffer_is_not_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002715 return;
2716 }
2717
2718 if (*eap->arg == NUL)
2719 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002720 // No argument: Find the other buffer in the list of diff buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002721 for (idx_other = 0; idx_other < DB_COUNT; ++idx_other)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002722 if (curtab->tp_diffbuf[idx_other] != curbuf
Bram Moolenaar602eb742007-02-20 03:43:38 +00002723 && curtab->tp_diffbuf[idx_other] != NULL)
2724 {
2725 if (eap->cmdidx != CMD_diffput
2726 || curtab->tp_diffbuf[idx_other]->b_p_ma)
2727 break;
2728 found_not_ma = TRUE;
2729 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 if (idx_other == DB_COUNT)
2731 {
Bram Moolenaar602eb742007-02-20 03:43:38 +00002732 if (found_not_ma)
Bram Moolenaar677658a2022-01-05 16:09:06 +00002733 emsg(_(e_no_other_buffer_in_diff_mode_is_modifiable));
Bram Moolenaar602eb742007-02-20 03:43:38 +00002734 else
Bram Moolenaare1242042021-12-16 20:56:57 +00002735 emsg(_(e_no_other_buffer_in_diff_mode));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736 return;
2737 }
2738
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002739 // Check that there isn't a third buffer in the list
Bram Moolenaar071d4272004-06-13 20:20:40 +00002740 for (i = idx_other + 1; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002741 if (curtab->tp_diffbuf[i] != curbuf
2742 && curtab->tp_diffbuf[i] != NULL
2743 && (eap->cmdidx != CMD_diffput || curtab->tp_diffbuf[i]->b_p_ma))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002744 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002745 emsg(_(e_more_than_two_buffers_in_diff_mode_dont_know_which_one_to_use));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002746 return;
2747 }
2748 }
2749 else
2750 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002751 // Buffer number or pattern given. Ignore trailing white space.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002752 p = eap->arg + STRLEN(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01002753 while (p > eap->arg && VIM_ISWHITE(p[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002754 --p;
2755 for (i = 0; vim_isdigit(eap->arg[i]) && eap->arg + i < p; ++i)
2756 ;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002757 if (eap->arg + i == p) // digits only
Bram Moolenaar071d4272004-06-13 20:20:40 +00002758 i = atol((char *)eap->arg);
2759 else
2760 {
Bram Moolenaar0c279bb2013-03-19 14:25:54 +01002761 i = buflist_findpat(eap->arg, p, FALSE, TRUE, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002762 if (i < 0)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002763 return; // error message already given
Bram Moolenaar071d4272004-06-13 20:20:40 +00002764 }
2765 buf = buflist_findnr(i);
2766 if (buf == NULL)
2767 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002768 semsg(_(e_cant_find_buffer_str), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002769 return;
2770 }
Bram Moolenaar5cc6a6e2009-01-22 19:48:55 +00002771 if (buf == curbuf)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002772 return; // nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773 idx_other = diff_buf_idx(buf);
2774 if (idx_other == DB_COUNT)
2775 {
Bram Moolenaare1242042021-12-16 20:56:57 +00002776 semsg(_(e_buffer_str_is_not_in_diff_mode), eap->arg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002777 return;
2778 }
2779 }
2780
2781 diff_busy = TRUE;
2782
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002783 // When no range given include the line above or below the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002784 if (eap->addr_count == 0)
2785 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002786 // Make it possible that ":diffget" on the last line gets line below
2787 // the cursor line when there is no difference above the cursor.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002788 if (eap->cmdidx == CMD_diffget
2789 && eap->line1 == curbuf->b_ml.ml_line_count
2790 && diff_check(curwin, eap->line1) == 0
2791 && (eap->line1 == 1 || diff_check(curwin, eap->line1 - 1) == 0))
2792 ++eap->line2;
2793 else if (eap->line1 > 0)
2794 --eap->line1;
2795 }
2796
2797 if (eap->cmdidx == CMD_diffget)
2798 {
2799 idx_from = idx_other;
2800 idx_to = idx_cur;
2801 }
2802 else
2803 {
2804 idx_from = idx_cur;
2805 idx_to = idx_other;
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002806 // Need to make the other buffer the current buffer to be able to make
2807 // changes in it.
Bram Moolenaare76062c2022-11-28 18:51:43 +00002808 // Set curwin/curbuf to buf and save a few things.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002809 aucmd_prepbuf(&aco, curtab->tp_diffbuf[idx_other]);
Bram Moolenaare76062c2022-11-28 18:51:43 +00002810 if (curbuf != curtab->tp_diffbuf[idx_other])
2811 // Could not find a window for this buffer, the rest is likely to
2812 // fail.
2813 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814 }
2815
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002816 // May give the warning for a changed buffer here, which can trigger the
2817 // FileChangedRO autocommand, which may do nasty things and mess
2818 // everything up.
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002819 if (!curbuf->b_changed)
2820 {
2821 change_warning(0);
2822 if (diff_buf_idx(curbuf) != idx_to)
2823 {
Bram Moolenaar677658a2022-01-05 16:09:06 +00002824 emsg(_(e_buffer_changed_unexpectedly));
Bram Moolenaard2b58c02018-09-16 18:10:48 +02002825 goto theend;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002826 }
2827 }
2828
Bram Moolenaar071d4272004-06-13 20:20:40 +00002829 dprev = NULL;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002830 for (dp = curtab->tp_first_diff; dp != NULL; )
Bram Moolenaar071d4272004-06-13 20:20:40 +00002831 {
2832 if (dp->df_lnum[idx_cur] > eap->line2 + off)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002833 break; // past the range that was specified
Bram Moolenaar071d4272004-06-13 20:20:40 +00002834
2835 dfree = NULL;
2836 lnum = dp->df_lnum[idx_to];
2837 count = dp->df_count[idx_to];
2838 if (dp->df_lnum[idx_cur] + dp->df_count[idx_cur] > eap->line1 + off
2839 && u_save(lnum - 1, lnum + count) != FAIL)
2840 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002841 // Inside the specified range and saving for undo worked.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002842 start_skip = 0;
2843 end_skip = 0;
2844 if (eap->addr_count > 0)
2845 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002846 // A range was specified: check if lines need to be skipped.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002847 start_skip = eap->line1 + off - dp->df_lnum[idx_cur];
2848 if (start_skip > 0)
2849 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002850 // range starts below start of current diff block
Bram Moolenaar071d4272004-06-13 20:20:40 +00002851 if (start_skip > count)
2852 {
2853 lnum += count;
2854 count = 0;
2855 }
2856 else
2857 {
2858 count -= start_skip;
2859 lnum += start_skip;
2860 }
2861 }
2862 else
2863 start_skip = 0;
2864
2865 end_skip = dp->df_lnum[idx_cur] + dp->df_count[idx_cur] - 1
2866 - (eap->line2 + off);
2867 if (end_skip > 0)
2868 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002869 // range ends above end of current/from diff block
2870 if (idx_cur == idx_from) // :diffput
Bram Moolenaar071d4272004-06-13 20:20:40 +00002871 {
2872 i = dp->df_count[idx_cur] - start_skip - end_skip;
2873 if (count > i)
2874 count = i;
2875 }
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002876 else // :diffget
Bram Moolenaar071d4272004-06-13 20:20:40 +00002877 {
2878 count -= end_skip;
2879 end_skip = dp->df_count[idx_from] - start_skip - count;
2880 if (end_skip < 0)
2881 end_skip = 0;
2882 }
2883 }
2884 else
2885 end_skip = 0;
2886 }
2887
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01002888 buf_empty = BUFEMPTY();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002889 added = 0;
2890 for (i = 0; i < count; ++i)
2891 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002892 // remember deleting the last line of the buffer
Bram Moolenaar280f1262006-01-30 00:14:18 +00002893 buf_empty = curbuf->b_ml.ml_line_count == 1;
Bram Moolenaar4e677b92022-07-28 18:44:27 +01002894 if (ml_delete(lnum) == OK)
2895 --added;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002896 }
2897 for (i = 0; i < dp->df_count[idx_from] - start_skip - end_skip; ++i)
2898 {
2899 linenr_T nr;
2900
2901 nr = dp->df_lnum[idx_from] + start_skip + i;
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002902 if (nr > curtab->tp_diffbuf[idx_from]->b_ml.ml_line_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002903 break;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002904 p = vim_strsave(ml_get_buf(curtab->tp_diffbuf[idx_from],
2905 nr, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906 if (p != NULL)
2907 {
2908 ml_append(lnum + i - 1, p, 0, FALSE);
2909 vim_free(p);
2910 ++added;
Bram Moolenaar280f1262006-01-30 00:14:18 +00002911 if (buf_empty && curbuf->b_ml.ml_line_count == 2)
2912 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002913 // Added the first line into an empty buffer, need to
2914 // delete the dummy empty line.
Bram Moolenaar280f1262006-01-30 00:14:18 +00002915 buf_empty = FALSE;
Bram Moolenaarca70c072020-05-30 20:30:46 +02002916 ml_delete((linenr_T)2);
Bram Moolenaar280f1262006-01-30 00:14:18 +00002917 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002918 }
2919 }
2920 new_count = dp->df_count[idx_to] + added;
2921 dp->df_count[idx_to] = new_count;
2922
2923 if (start_skip == 0 && end_skip == 0)
2924 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002925 // Check if there are any other buffers and if the diff is
2926 // equal in them.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002927 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002928 if (curtab->tp_diffbuf[i] != NULL && i != idx_from
2929 && i != idx_to
Bram Moolenaar071d4272004-06-13 20:20:40 +00002930 && !diff_equal_entry(dp, idx_from, i))
2931 break;
2932 if (i == DB_COUNT)
2933 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002934 // delete the diff entry, the buffers are now equal here
Bram Moolenaar071d4272004-06-13 20:20:40 +00002935 dfree = dp;
2936 dp = dp->df_next;
2937 if (dprev == NULL)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00002938 curtab->tp_first_diff = dp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002939 else
2940 dprev->df_next = dp;
2941 }
2942 }
2943
Bram Moolenaar071d4272004-06-13 20:20:40 +00002944 if (added != 0)
2945 {
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002946 // Adjust marks. This will change the following entries!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002947 mark_adjust(lnum, lnum + count - 1, (long)MAXLNUM, (long)added);
2948 if (curwin->w_cursor.lnum >= lnum)
2949 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002950 // Adjust the cursor position if it's in/after the changed
2951 // lines.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952 if (curwin->w_cursor.lnum >= lnum + count)
2953 curwin->w_cursor.lnum += added;
2954 else if (added < 0)
2955 curwin->w_cursor.lnum = lnum;
2956 }
2957 }
2958 changed_lines(lnum, 0, lnum + count, (long)added);
2959
2960 if (dfree != NULL)
2961 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002962 // Diff is deleted, update folds in other windows.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002963#ifdef FEAT_FOLDING
2964 diff_fold_update(dfree, idx_to);
2965#endif
2966 vim_free(dfree);
2967 }
Bram Moolenaarc5274dd2022-07-02 15:10:00 +01002968
2969 // mark_adjust() may have made "dp" invalid. We don't know where
2970 // to continue then, bail out.
2971 if (added != 0 && !valid_diff(dp))
2972 break;
2973
2974 if (dfree == NULL)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002975 // mark_adjust() may have changed the count in a wrong way
Bram Moolenaar071d4272004-06-13 20:20:40 +00002976 dp->df_count[idx_to] = new_count;
2977
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002978 // When changing the current buffer, keep track of line numbers
Bram Moolenaar071d4272004-06-13 20:20:40 +00002979 if (idx_cur == idx_to)
2980 off += added;
2981 }
2982
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002983 // If before the range or not deleted, go to next diff.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002984 if (dfree == NULL)
2985 {
2986 dprev = dp;
2987 dp = dp->df_next;
2988 }
2989 }
2990
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002991 // restore curwin/curbuf and a few other things
Bram Moolenaara9d52e32010-07-31 16:44:19 +02002992 if (eap->cmdidx != CMD_diffget)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002993 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01002994 // Syncing undo only works for the current buffer, but we change
2995 // another buffer. Sync undo if the command was typed. This isn't
2996 // 100% right when ":diffput" is used in a function or mapping.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002997 if (KeyTyped)
Bram Moolenaar779b74b2006-04-10 14:55:34 +00002998 u_sync(FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002999 aucmd_restbuf(&aco);
3000 }
3001
Bram Moolenaard2b58c02018-09-16 18:10:48 +02003002theend:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003003 diff_busy = FALSE;
Bram Moolenaard2b58c02018-09-16 18:10:48 +02003004 if (diff_need_update)
Bram Moolenaard2b58c02018-09-16 18:10:48 +02003005 ex_diffupdate(NULL);
Bram Moolenaardf77cef2018-10-07 17:46:42 +02003006
Bram Moolenaar5f57bdc2018-10-25 17:52:23 +02003007 // Check that the cursor is on a valid character and update its
Bram Moolenaardf77cef2018-10-07 17:46:42 +02003008 // position. When there were filler lines the topline has become
3009 // invalid.
3010 check_cursor();
3011 changed_line_abv_curs();
3012
3013 if (diff_need_update)
3014 // redraw already done by ex_diffupdate()
3015 diff_need_update = FALSE;
Bram Moolenaar198fa062018-09-18 21:20:26 +02003016 else
3017 {
Bram Moolenaar198fa062018-09-18 21:20:26 +02003018 // Also need to redraw the other buffers.
3019 diff_redraw(FALSE);
3020 apply_autocmds(EVENT_DIFFUPDATED, NULL, NULL, FALSE, curbuf);
3021 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003022}
3023
3024#ifdef FEAT_FOLDING
3025/*
3026 * Update folds for all diff buffers for entry "dp".
3027 * Skip buffer with index "skip_idx".
3028 * When there are no diffs, all folds are removed.
3029 */
3030 static void
Bram Moolenaar7454a062016-01-30 15:14:10 +01003031diff_fold_update(diff_T *dp, int skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003032{
3033 int i;
3034 win_T *wp;
3035
Bram Moolenaar29323592016-07-24 22:04:11 +02003036 FOR_ALL_WINDOWS(wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037 for (i = 0; i < DB_COUNT; ++i)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003038 if (curtab->tp_diffbuf[i] == wp->w_buffer && i != skip_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003039 foldUpdate(wp, dp->df_lnum[i],
3040 dp->df_lnum[i] + dp->df_count[i]);
3041}
3042#endif
3043
3044/*
3045 * Return TRUE if buffer "buf" is in diff-mode.
3046 */
3047 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003048diff_mode_buf(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049{
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003050 tabpage_T *tp;
3051
Bram Moolenaar29323592016-07-24 22:04:11 +02003052 FOR_ALL_TABPAGES(tp)
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003053 if (diff_buf_idx_tp(buf, tp) != DB_COUNT)
3054 return TRUE;
3055 return FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003056}
3057
3058/*
3059 * Move "count" times in direction "dir" to the next diff block.
3060 * Return FAIL if there isn't such a diff block.
3061 */
3062 int
Bram Moolenaar7454a062016-01-30 15:14:10 +01003063diff_move_to(int dir, long count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003064{
3065 int idx;
3066 linenr_T lnum = curwin->w_cursor.lnum;
3067 diff_T *dp;
3068
3069 idx = diff_buf_idx(curbuf);
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003070 if (idx == DB_COUNT || curtab->tp_first_diff == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003071 return FAIL;
3072
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003073 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003074 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003076 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077 return FAIL;
3078
3079 while (--count >= 0)
3080 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003081 // Check if already before first diff.
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003082 if (dir == BACKWARD && lnum <= curtab->tp_first_diff->df_lnum[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +00003083 break;
3084
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003085 for (dp = curtab->tp_first_diff; ; dp = dp->df_next)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003086 {
3087 if (dp == NULL)
3088 break;
3089 if ((dir == FORWARD && lnum < dp->df_lnum[idx])
3090 || (dir == BACKWARD
3091 && (dp->df_next == NULL
3092 || lnum <= dp->df_next->df_lnum[idx])))
3093 {
3094 lnum = dp->df_lnum[idx];
3095 break;
3096 }
3097 }
3098 }
3099
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003100 // don't end up past the end of the file
Bram Moolenaar071d4272004-06-13 20:20:40 +00003101 if (lnum > curbuf->b_ml.ml_line_count)
3102 lnum = curbuf->b_ml.ml_line_count;
3103
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003104 // When the cursor didn't move at all we fail.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003105 if (lnum == curwin->w_cursor.lnum)
3106 return FAIL;
3107
3108 setpcmark();
3109 curwin->w_cursor.lnum = lnum;
3110 curwin->w_cursor.col = 0;
3111
3112 return OK;
3113}
3114
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003115/*
3116 * Return the line number in the current window that is closest to "lnum1" in
3117 * "buf1" in diff mode.
3118 */
3119 static linenr_T
3120diff_get_corresponding_line_int(
Bram Moolenaar7454a062016-01-30 15:14:10 +01003121 buf_T *buf1,
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003122 linenr_T lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003123{
3124 int idx1;
3125 int idx2;
3126 diff_T *dp;
3127 int baseline = 0;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003128
3129 idx1 = diff_buf_idx(buf1);
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003130 idx2 = diff_buf_idx(curbuf);
Bram Moolenaar860cae12010-06-05 23:22:07 +02003131 if (idx1 == DB_COUNT || idx2 == DB_COUNT || curtab->tp_first_diff == NULL)
3132 return lnum1;
3133
3134 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003135 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar860cae12010-06-05 23:22:07 +02003136
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003137 if (curtab->tp_first_diff == NULL) // no diffs today
Bram Moolenaar860cae12010-06-05 23:22:07 +02003138 return lnum1;
3139
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003140 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003141 {
3142 if (dp->df_lnum[idx1] > lnum1)
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003143 return lnum1 - baseline;
3144 if ((dp->df_lnum[idx1] + dp->df_count[idx1]) > lnum1)
Bram Moolenaar860cae12010-06-05 23:22:07 +02003145 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003146 // Inside the diffblock
Bram Moolenaar860cae12010-06-05 23:22:07 +02003147 baseline = lnum1 - dp->df_lnum[idx1];
3148 if (baseline > dp->df_count[idx2])
3149 baseline = dp->df_count[idx2];
3150
3151 return dp->df_lnum[idx2] + baseline;
3152 }
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003153 if ( (dp->df_lnum[idx1] == lnum1)
3154 && (dp->df_count[idx1] == 0)
3155 && (dp->df_lnum[idx2] <= curwin->w_cursor.lnum)
3156 && ((dp->df_lnum[idx2] + dp->df_count[idx2])
3157 > curwin->w_cursor.lnum))
Bram Moolenaar860cae12010-06-05 23:22:07 +02003158 /*
3159 * Special case: if the cursor is just after a zero-count
3160 * block (i.e. all filler) and the target cursor is already
3161 * inside the corresponding block, leave the target cursor
3162 * unmoved. This makes repeated CTRL-W W operations work
3163 * as expected.
3164 */
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003165 return curwin->w_cursor.lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003166 baseline = (dp->df_lnum[idx1] + dp->df_count[idx1])
3167 - (dp->df_lnum[idx2] + dp->df_count[idx2]);
3168 }
3169
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003170 // If we get here then the cursor is after the last diff
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003171 return lnum1 - baseline;
3172}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003173
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003174/*
3175 * Return the line number in the current window that is closest to "lnum1" in
3176 * "buf1" in diff mode. Checks the line number to be valid.
3177 */
3178 linenr_T
3179diff_get_corresponding_line(buf_T *buf1, linenr_T lnum1)
3180{
3181 linenr_T lnum = diff_get_corresponding_line_int(buf1, lnum1);
3182
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003183 // don't end up past the end of the file
Bram Moolenaar025e3e02016-10-18 14:50:18 +02003184 if (lnum > curbuf->b_ml.ml_line_count)
3185 return curbuf->b_ml.ml_line_count;
3186 return lnum;
Bram Moolenaar860cae12010-06-05 23:22:07 +02003187}
Bram Moolenaar860cae12010-06-05 23:22:07 +02003188
Bram Moolenaar071d4272004-06-13 20:20:40 +00003189/*
3190 * For line "lnum" in the current window find the equivalent lnum in window
3191 * "wp", compensating for inserted/deleted lines.
3192 */
3193 linenr_T
Bram Moolenaar7454a062016-01-30 15:14:10 +01003194diff_lnum_win(linenr_T lnum, win_T *wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003195{
3196 diff_T *dp;
3197 int idx;
3198 int i;
3199 linenr_T n;
3200
3201 idx = diff_buf_idx(curbuf);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003202 if (idx == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003203 return (linenr_T)0;
3204
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00003205 if (curtab->tp_diff_invalid)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003206 ex_diffupdate(NULL); // update after a big change
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003208 // search for a change that includes "lnum" in the list of diffblocks.
Bram Moolenaaraeea7212020-04-02 18:50:46 +02003209 FOR_ALL_DIFFBLOCKS_IN_TAB(curtab, dp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003210 if (lnum <= dp->df_lnum[idx] + dp->df_count[idx])
3211 break;
3212
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003213 // When after the last change, compute relative to the last line number.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003214 if (dp == NULL)
3215 return wp->w_buffer->b_ml.ml_line_count
3216 - (curbuf->b_ml.ml_line_count - lnum);
3217
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003218 // Find index for "wp".
Bram Moolenaar071d4272004-06-13 20:20:40 +00003219 i = diff_buf_idx(wp->w_buffer);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003220 if (i == DB_COUNT) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +00003221 return (linenr_T)0;
3222
3223 n = lnum + (dp->df_lnum[i] - dp->df_lnum[idx]);
3224 if (n > dp->df_lnum[i] + dp->df_count[i])
3225 n = dp->df_lnum[i] + dp->df_count[i];
3226 return n;
3227}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003228
Bram Moolenaare828b762018-09-10 17:51:58 +02003229/*
3230 * Handle an ED style diff line.
3231 * Return FAIL if the line does not contain diff info.
3232 */
3233 static int
3234parse_diff_ed(
3235 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003236 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003237{
3238 char_u *p;
3239 long f1, l1, f2, l2;
3240 int difftype;
3241
3242 // The line must be one of three formats:
3243 // change: {first}[,{last}]c{first}[,{last}]
3244 // append: {first}a{first}[,{last}]
3245 // delete: {first}[,{last}]d{first}
3246 p = line;
3247 f1 = getdigits(&p);
3248 if (*p == ',')
3249 {
3250 ++p;
3251 l1 = getdigits(&p);
3252 }
3253 else
3254 l1 = f1;
3255 if (*p != 'a' && *p != 'c' && *p != 'd')
3256 return FAIL; // invalid diff format
3257 difftype = *p++;
3258 f2 = getdigits(&p);
3259 if (*p == ',')
3260 {
3261 ++p;
3262 l2 = getdigits(&p);
3263 }
3264 else
3265 l2 = f2;
3266 if (l1 < f1 || l2 < f2)
3267 return FAIL;
3268
3269 if (difftype == 'a')
3270 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003271 hunk->lnum_orig = f1 + 1;
3272 hunk->count_orig = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003273 }
3274 else
3275 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003276 hunk->lnum_orig = f1;
3277 hunk->count_orig = l1 - f1 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003278 }
3279 if (difftype == 'd')
3280 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003281 hunk->lnum_new = f2 + 1;
3282 hunk->count_new = 0;
Bram Moolenaare828b762018-09-10 17:51:58 +02003283 }
3284 else
3285 {
Lewis Russelld9da86e2021-12-28 13:54:41 +00003286 hunk->lnum_new = f2;
3287 hunk->count_new = l2 - f2 + 1;
Bram Moolenaare828b762018-09-10 17:51:58 +02003288 }
3289 return OK;
3290}
3291
3292/*
3293 * Parses unified diff with zero(!) context lines.
3294 * Return FAIL if there is no diff information in "line".
3295 */
3296 static int
3297parse_diff_unified(
3298 char_u *line,
Lewis Russelld9da86e2021-12-28 13:54:41 +00003299 diffhunk_T *hunk)
Bram Moolenaare828b762018-09-10 17:51:58 +02003300{
3301 char_u *p;
3302 long oldline, oldcount, newline, newcount;
3303
3304 // Parse unified diff hunk header:
3305 // @@ -oldline,oldcount +newline,newcount @@
3306 p = line;
3307 if (*p++ == '@' && *p++ == '@' && *p++ == ' ' && *p++ == '-')
3308 {
3309 oldline = getdigits(&p);
3310 if (*p == ',')
3311 {
3312 ++p;
3313 oldcount = getdigits(&p);
3314 }
3315 else
3316 oldcount = 1;
3317 if (*p++ == ' ' && *p++ == '+')
3318 {
3319 newline = getdigits(&p);
3320 if (*p == ',')
3321 {
3322 ++p;
3323 newcount = getdigits(&p);
3324 }
3325 else
3326 newcount = 1;
3327 }
3328 else
3329 return FAIL; // invalid diff format
3330
3331 if (oldcount == 0)
3332 oldline += 1;
3333 if (newcount == 0)
3334 newline += 1;
3335 if (newline == 0)
3336 newline = 1;
3337
Lewis Russelld9da86e2021-12-28 13:54:41 +00003338 hunk->lnum_orig = oldline;
3339 hunk->count_orig = oldcount;
3340 hunk->lnum_new = newline;
3341 hunk->count_new = newcount;
Bram Moolenaare828b762018-09-10 17:51:58 +02003342
3343 return OK;
3344 }
3345
3346 return FAIL;
3347}
3348
3349/*
3350 * Callback function for the xdl_diff() function.
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003351 * Stores the diff output (indices) in a grow array.
Bram Moolenaare828b762018-09-10 17:51:58 +02003352 */
3353 static int
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003354xdiff_out_indices(
Lewis Russelld9da86e2021-12-28 13:54:41 +00003355 long start_a,
3356 long count_a,
3357 long start_b,
3358 long count_b,
3359 void *priv)
Bram Moolenaare828b762018-09-10 17:51:58 +02003360{
3361 diffout_T *dout = (diffout_T *)priv;
Lewis Russelld9da86e2021-12-28 13:54:41 +00003362 diffhunk_T *p = ALLOC_ONE(diffhunk_T);
Bram Moolenaare828b762018-09-10 17:51:58 +02003363
Lewis Russelld9da86e2021-12-28 13:54:41 +00003364 if (p == NULL)
3365 return -1;
Bram Moolenaarf080d702018-10-31 22:57:26 +01003366
3367 if (ga_grow(&dout->dout_ga, 1) == FAIL)
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003368 {
3369 vim_free(p);
Bram Moolenaarf080d702018-10-31 22:57:26 +01003370 return -1;
Bram Moolenaarfebb78f2021-12-29 11:59:53 +00003371 }
Lewis Russelld9da86e2021-12-28 13:54:41 +00003372
3373 p->lnum_orig = start_a + 1;
3374 p->count_orig = count_a;
3375 p->lnum_new = start_b + 1;
3376 p->count_new = count_b;
3377 ((diffhunk_T **)dout->dout_ga.ga_data)[dout->dout_ga.ga_len++] = p;
Bram Moolenaare828b762018-09-10 17:51:58 +02003378 return 0;
3379}
3380
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003381/*
3382 * Callback function for the xdl_diff() function.
3383 * Stores the unified diff output in a grow array.
3384 */
3385 static int
3386xdiff_out_unified(
3387 void *priv,
3388 mmbuffer_t *mb,
3389 int nbuf)
3390{
3391 diffout_T *dout = (diffout_T *)priv;
3392 int i;
3393
3394 for (i = 0; i < nbuf; i++)
3395 ga_concat_len(&dout->dout_ga, (char_u *)mb[i].ptr, mb[i].size);
3396
3397 return 0;
3398}
3399
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003400#endif // FEAT_DIFF
3401
3402#if defined(FEAT_EVAL) || defined(PROTO)
3403
3404/*
3405 * "diff_filler()" function
3406 */
3407 void
3408f_diff_filler(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3409{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003410# ifdef FEAT_DIFF
Yegappan Lakshmanan4490ec42021-07-27 22:00:44 +02003411 if (in_vim9script() && check_for_lnum_arg(argvars, 0) == FAIL)
3412 return;
3413
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003414 rettv->vval.v_number = diff_check_fill(curwin, tv_get_lnum(argvars));
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003415# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003416}
3417
3418/*
3419 * "diff_hlID()" function
3420 */
3421 void
3422f_diff_hlID(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3423{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003424# ifdef FEAT_DIFF
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003425 linenr_T lnum;
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003426 static linenr_T prev_lnum = 0;
3427 static varnumber_T changedtick = 0;
3428 static int fnum = 0;
3429 static int change_start = 0;
3430 static int change_end = 0;
3431 static hlf_T hlID = (hlf_T)0;
3432 int filler_lines;
3433 int col;
3434
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003435 if (in_vim9script()
Yegappan Lakshmanancd917202021-07-21 19:09:09 +02003436 && (check_for_lnum_arg(argvars,0) == FAIL
Yegappan Lakshmanana9a7c0c2021-07-17 19:11:07 +02003437 || check_for_number_arg(argvars, 1) == FAIL))
3438 return;
3439
3440 lnum = tv_get_lnum(argvars);
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003441 if (lnum < 0) // ignore type error in {lnum} arg
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003442 lnum = 0;
3443 if (lnum != prev_lnum
3444 || changedtick != CHANGEDTICK(curbuf)
3445 || fnum != curbuf->b_fnum)
3446 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003447 // New line, buffer, change: need to get the values.
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003448 filler_lines = diff_check(curwin, lnum);
3449 if (filler_lines < 0)
3450 {
3451 if (filler_lines == -1)
3452 {
3453 change_start = MAXCOL;
3454 change_end = -1;
3455 if (diff_find_change(curwin, lnum, &change_start, &change_end))
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003456 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003457 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003458 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003459 }
3460 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003461 hlID = HLF_ADD; // added line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003462 }
3463 else
3464 hlID = (hlf_T)0;
3465 prev_lnum = lnum;
3466 changedtick = CHANGEDTICK(curbuf);
3467 fnum = curbuf->b_fnum;
3468 }
3469
3470 if (hlID == HLF_CHD || hlID == HLF_TXD)
3471 {
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003472 col = tv_get_number(&argvars[1]) - 1; // ignore type error in {col}
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003473 if (col >= change_start && col <= change_end)
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003474 hlID = HLF_TXD; // changed text
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003475 else
Bram Moolenaar5d18efe2019-12-01 21:11:22 +01003476 hlID = HLF_CHD; // changed line
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003477 }
3478 rettv->vval.v_number = hlID == (hlf_T)0 ? 0 : (int)hlID;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003479# endif
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003480}
3481
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003482# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003483/*
3484 * Parse the diff options passed in "optarg" to the diff() function and return
3485 * the options in "diffopts" and the diff algorithm in "diffalgo".
3486 */
3487 static int
3488parse_diff_optarg(
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003489 typval_T *opts,
3490 int *diffopts,
3491 long *diffalgo,
3492 dio_outfmt_T *diff_output_fmt,
3493 int *diff_ctxlen)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003494{
3495 dict_T *d = opts->vval.v_dict;
3496
3497 char_u *algo = dict_get_string(d, "algorithm", FALSE);
3498 if (algo != NULL)
3499 {
3500 if (STRNCMP(algo, "myers", 5) == 0)
3501 *diffalgo = 0;
3502 else if (STRNCMP(algo, "minimal", 7) == 0)
3503 *diffalgo = XDF_NEED_MINIMAL;
3504 else if (STRNCMP(algo, "patience", 8) == 0)
3505 *diffalgo = XDF_PATIENCE_DIFF;
3506 else if (STRNCMP(algo, "histogram", 9) == 0)
3507 *diffalgo = XDF_HISTOGRAM_DIFF;
3508 }
3509
3510 char_u *output_fmt = dict_get_string(d, "output", FALSE);
3511 if (output_fmt != NULL)
3512 {
3513 if (STRNCMP(output_fmt, "unified", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003514 *diff_output_fmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003515 else if (STRNCMP(output_fmt, "indices", 7) == 0)
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003516 *diff_output_fmt = DIO_OUTPUT_INDICES;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003517 else
3518 {
3519 semsg(_(e_unsupported_diff_output_format_str), output_fmt);
3520 return FAIL;
3521 }
3522 }
3523
Yegappan Lakshmanana0010a12024-02-12 20:21:26 +01003524 *diff_ctxlen = dict_get_number_def(d, "context", 0);
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003525 if (*diff_ctxlen < 0)
Yegappan Lakshmanana0010a12024-02-12 20:21:26 +01003526 *diff_ctxlen = 0;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003527
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003528 if (dict_get_bool(d, "iblank", FALSE))
3529 *diffopts |= DIFF_IBLANK;
3530 if (dict_get_bool(d, "icase", FALSE))
3531 *diffopts |= DIFF_ICASE;
3532 if (dict_get_bool(d, "iwhite", FALSE))
3533 *diffopts |= DIFF_IWHITE;
3534 if (dict_get_bool(d, "iwhiteall", FALSE))
3535 *diffopts |= DIFF_IWHITEALL;
3536 if (dict_get_bool(d, "iwhiteeol", FALSE))
3537 *diffopts |= DIFF_IWHITEEOL;
3538 if (dict_get_bool(d, "indent-heuristic", FALSE))
3539 *diffalgo |= XDF_INDENT_HEURISTIC;
3540
3541 return OK;
3542}
3543
3544/*
3545 * Concatenate the List of strings in "l" and store the result in
3546 * "din->din_mmfile.ptr" and the length in "din->din_mmfile.size".
3547 */
3548 static void
3549list_to_diffin(list_T *l, diffin_T *din, int icase)
3550{
3551 garray_T ga;
3552 listitem_T *li;
3553 char_u *str;
3554
3555 ga_init2(&ga, 512, 4);
3556
3557 FOR_ALL_LIST_ITEMS(l, li)
3558 {
3559 str = tv_get_string(&li->li_tv);
3560 if (icase)
3561 {
3562 str = strlow_save(str);
3563 if (str == NULL)
3564 continue;
3565 }
3566 ga_concat(&ga, str);
3567 ga_concat(&ga, (char_u *)NL_STR);
3568 if (icase)
3569 vim_free(str);
3570 }
3571 if (ga.ga_len > 0)
3572 ((char *)ga.ga_data)[ga.ga_len] = NUL;
3573
3574 din->din_mmfile.ptr = (char *)ga.ga_data;
3575 din->din_mmfile.size = ga.ga_len;
3576}
3577
3578/*
3579 * Get the start and end indices from the diff "hunk".
3580 */
3581 static dict_T *
3582get_diff_hunk_indices(diffhunk_T *hunk)
3583{
3584 dict_T *hunk_dict;
3585
3586 hunk_dict = dict_alloc();
3587 if (hunk_dict == NULL)
3588 return NULL;
3589
3590 dict_add_number(hunk_dict, "from_idx", hunk->lnum_orig - 1);
3591 dict_add_number(hunk_dict, "from_count", hunk->count_orig);
3592 dict_add_number(hunk_dict, "to_idx", hunk->lnum_new - 1);
3593 dict_add_number(hunk_dict, "to_count", hunk->count_new);
3594
3595 return hunk_dict;
3596}
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003597# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003598
3599/*
3600 * "diff()" function
3601 */
3602 void
3603f_diff(typval_T *argvars UNUSED, typval_T *rettv UNUSED)
3604{
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003605# ifdef FEAT_DIFF
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003606 diffio_T dio;
3607
3608 if (check_for_nonnull_list_arg(argvars, 0) == FAIL
3609 || check_for_nonnull_list_arg(argvars, 1) == FAIL
3610 || check_for_opt_nonnull_dict_arg(argvars, 2) == FAIL)
3611 return;
3612
3613 CLEAR_FIELD(dio);
3614 dio.dio_internal = TRUE;
3615 ga_init2(&dio.dio_diff.dout_ga, sizeof(char *), 1000);
3616
3617 list_T *orig_list = argvars[0].vval.v_list;
3618 list_T *new_list = argvars[1].vval.v_list;
3619
3620 // Save the 'diffopt' option value and restore it after getting the diff.
3621 int save_diff_flags = diff_flags;
3622 long save_diff_algorithm = diff_algorithm;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003623 diff_flags = DIFF_INTERNAL;
3624 diff_algorithm = 0;
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003625 dio.dio_outfmt = DIO_OUTPUT_UNIFIED;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003626 if (argvars[2].v_type != VAR_UNKNOWN)
3627 if (parse_diff_optarg(&argvars[2], &diff_flags, &diff_algorithm,
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003628 &dio.dio_outfmt, &dio.dio_ctxlen) == FAIL)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003629 return;
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003630
3631 // Concatenate the List of strings into a single string using newline
3632 // separator. Internal diff library expects a single string.
3633 list_to_diffin(orig_list, &dio.dio_orig, diff_flags & DIFF_ICASE);
3634 list_to_diffin(new_list, &dio.dio_new, diff_flags & DIFF_ICASE);
3635
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003636 // If 'diffexpr' is set, then the internal diff is not used. Set
3637 // 'diffexpr' to an empty string temporarily.
3638 int restore_diffexpr = FALSE;
3639 char_u cc = *p_dex;
3640 if (*p_dex != NUL)
3641 {
3642 restore_diffexpr = TRUE;
3643 *p_dex = NUL;
3644 }
3645
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003646 // Compute the diff
3647 int diff_status = diff_file(&dio);
3648
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003649 // restore 'diffexpr'
3650 if (restore_diffexpr)
3651 *p_dex = cc;
3652
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003653 if (diff_status == FAIL)
3654 goto done;
3655
3656 int hunk_idx = 0;
3657 dict_T *hunk_dict;
3658
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003659 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003660 {
3661 if (rettv_list_alloc(rettv) != OK)
3662 goto done;
3663 list_T *l = rettv->vval.v_list;
3664
3665 // Process each diff hunk
3666 diffhunk_T *hunk = NULL;
3667 while (hunk_idx < dio.dio_diff.dout_ga.ga_len)
3668 {
3669 hunk = ((diffhunk_T **)dio.dio_diff.dout_ga.ga_data)[hunk_idx++];
3670
3671 hunk_dict = get_diff_hunk_indices(hunk);
3672 if (hunk_dict == NULL)
3673 goto done;
3674
3675 list_append_dict(l, hunk_dict);
3676 }
3677 }
3678 else
3679 {
3680 ga_append(&dio.dio_diff.dout_ga, NUL);
3681 rettv->v_type = VAR_STRING;
3682 rettv->vval.v_string =
3683 vim_strsave((char_u *)dio.dio_diff.dout_ga.ga_data);
3684 }
3685
3686done:
3687 clear_diffin(&dio.dio_new);
Yegappan Lakshmananbe156a32024-02-11 17:08:29 +01003688 if (dio.dio_outfmt == DIO_OUTPUT_INDICES)
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003689 clear_diffout(&dio.dio_diff);
3690 else
3691 ga_clear(&dio.dio_diff.dout_ga);
3692 clear_diffin(&dio.dio_orig);
3693 // Restore the 'diffopt' option value.
3694 diff_flags = save_diff_flags;
3695 diff_algorithm = save_diff_algorithm;
Yegappan Lakshmanan60937032024-02-03 17:41:54 +01003696# endif
Yegappan Lakshmananfa378352024-02-01 22:05:27 +01003697}
3698
Bram Moolenaaraf7645d2019-09-05 22:33:28 +02003699#endif