?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Fix a use-after-free when cancelling copy mode, or trying to repeat cancel.
[tmux-openbsd.git] / window-copy.c
blobc27b182121ea39f04185af83f33456df3f938802
1 /* $OpenBSD$ */
2
3 /*
4 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
5 *
6 * Permission to use, copy, modify, and distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15 * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16 * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include <sys/types.h>
20
21 #include <stdlib.h>
22 #include <string.h>
23
24 #include "tmux.h"
25
26 struct screen *window_copy_init(struct window_pane *);
27 void window_copy_free(struct window_pane *);
28 void window_copy_resize(struct window_pane *, u_int, u_int);
29 void window_copy_key(struct window_pane *, struct client *, int);
30 int window_copy_key_input(struct window_pane *, int);
31 int window_copy_key_numeric_prefix(struct window_pane *, int);
32 void window_copy_mouse(
33 struct window_pane *, struct client *, struct mouse_event *);
34
35 void window_copy_redraw_lines(struct window_pane *, u_int, u_int);
36 void window_copy_redraw_screen(struct window_pane *);
37 void window_copy_write_line(
38 struct window_pane *, struct screen_write_ctx *, u_int);
39 void window_copy_write_lines(
40 struct window_pane *, struct screen_write_ctx *, u_int, u_int);
41
42 void window_copy_scroll_to(struct window_pane *, u_int, u_int);
43 int window_copy_search_compare(
44 struct grid *, u_int, u_int, struct grid *, u_int);
45 int window_copy_search_lr(
46 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
47 int window_copy_search_rl(
48 struct grid *, struct grid *, u_int *, u_int, u_int, u_int);
49 void window_copy_search_up(struct window_pane *, const char *);
50 void window_copy_search_down(struct window_pane *, const char *);
51 void window_copy_goto_line(struct window_pane *, const char *);
52 void window_copy_update_cursor(struct window_pane *, u_int, u_int);
53 void window_copy_start_selection(struct window_pane *);
54 int window_copy_update_selection(struct window_pane *);
55 void window_copy_copy_selection(struct window_pane *, struct client *);
56 void window_copy_clear_selection(struct window_pane *);
57 void window_copy_copy_line(
58 struct window_pane *, char **, size_t *, u_int, u_int, u_int);
59 int window_copy_in_set(struct window_pane *, u_int, u_int, const char *);
60 u_int window_copy_find_length(struct window_pane *, u_int);
61 void window_copy_cursor_start_of_line(struct window_pane *);
62 void window_copy_cursor_back_to_indentation(struct window_pane *);
63 void window_copy_cursor_end_of_line(struct window_pane *);
64 void window_copy_cursor_left(struct window_pane *);
65 void window_copy_cursor_right(struct window_pane *);
66 void window_copy_cursor_up(struct window_pane *, int);
67 void window_copy_cursor_down(struct window_pane *, int);
68 void window_copy_cursor_next_word(struct window_pane *, const char *);
69 void window_copy_cursor_next_word_end(struct window_pane *, const char *);
70 void window_copy_cursor_previous_word(struct window_pane *, const char *);
71 void window_copy_scroll_up(struct window_pane *, u_int);
72 void window_copy_scroll_down(struct window_pane *, u_int);
73 void window_copy_rectangle_toggle(struct window_pane *);
74
75 const struct window_mode window_copy_mode = {
76 window_copy_init,
77 window_copy_free,
78 window_copy_resize,
79 window_copy_key,
80 window_copy_mouse,
81 NULL,
82 };
83
84 enum window_copy_input_type {
85 WINDOW_COPY_OFF,
86 WINDOW_COPY_NUMERICPREFIX,
87 WINDOW_COPY_SEARCHUP,
88 WINDOW_COPY_SEARCHDOWN,
89 WINDOW_COPY_GOTOLINE,
90 };
91
92 struct window_copy_mode_data {
93 struct screen screen;
94
95 struct mode_key_data mdata;
96
97 u_int oy;
98
99 u_int selx;
100 u_int sely;
101
102 u_int rectflag; /* are we in rectangle copy mode? */
103
104 u_int cx;
105 u_int cy;
106
107 u_int lastcx; /* position in last line with content */
108 u_int lastsx; /* size of last line with content */
109
110 enum window_copy_input_type inputtype;
111 const char *inputprompt;
112 char *inputstr;
113
114 u_int numprefix;
115
116 enum window_copy_input_type searchtype;
117 char *searchstr;
118 };
119
120 struct screen *
121 window_copy_init(struct window_pane *wp)
122 {
123 struct window_copy_mode_data *data;
124 struct screen *s;
125 struct screen_write_ctx ctx;
126 u_int i;
127 int keys;
128
129 wp->modedata = data = xmalloc(sizeof *data);
130 data->oy = 0;
131 data->cx = wp->base.cx;
132 data->cy = wp->base.cy;
133
134 data->lastcx = 0;
135 data->lastsx = 0;
136
137 data->rectflag = 0;
138
139 data->inputtype = WINDOW_COPY_OFF;
140 data->inputprompt = NULL;
141 data->inputstr = xstrdup("");
142 data->numprefix = 0;
143
144 data->searchtype = WINDOW_COPY_OFF;
145 data->searchstr = NULL;
146
147 wp->flags |= PANE_FREEZE;
148 bufferevent_disable(wp->event, EV_READ|EV_WRITE);
149
150 s = &data->screen;
151 screen_init(s, screen_size_x(&wp->base), screen_size_y(&wp->base), 0);
152 if (options_get_number(&wp->window->options, "mode-mouse"))
153 s->mode |= MODE_MOUSE;
154
155 keys = options_get_number(&wp->window->options, "mode-keys");
156 if (keys == MODEKEY_EMACS)
157 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
158 else
159 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
160
161 s->cx = data->cx;
162 s->cy = data->cy;
163
164 screen_write_start(&ctx, NULL, s);
165 for (i = 0; i < screen_size_y(s); i++)
166 window_copy_write_line(wp, &ctx, i);
167 screen_write_cursormove(&ctx, data->cx, data->cy);
168 screen_write_stop(&ctx);
169
170 return (s);
171 }
172
173 void
174 window_copy_free(struct window_pane *wp)
175 {
176 struct window_copy_mode_data *data = wp->modedata;
177
178 wp->flags &= ~PANE_FREEZE;
179 bufferevent_enable(wp->event, EV_READ|EV_WRITE);
180
181 if (data->searchstr != NULL)
182 xfree(data->searchstr);
183 xfree(data->inputstr);
184
185 screen_free(&data->screen);
186
187 xfree(data);
188 }
189
190 void
191 window_copy_pageup(struct window_pane *wp)
192 {
193 struct window_copy_mode_data *data = wp->modedata;
194 struct screen *s = &data->screen;
195 u_int n;
196
197 n = 1;
198 if (screen_size_y(s) > 2)
199 n = screen_size_y(s) - 2;
200 if (data->oy + n > screen_hsize(&wp->base))
201 data->oy = screen_hsize(&wp->base);
202 else
203 data->oy += n;
204 window_copy_update_selection(wp);
205 window_copy_redraw_screen(wp);
206 }
207
208 void
209 window_copy_resize(struct window_pane *wp, u_int sx, u_int sy)
210 {
211 struct window_copy_mode_data *data = wp->modedata;
212 struct screen *s = &data->screen;
213 struct screen_write_ctx ctx;
214
215 screen_resize(s, sx, sy);
216
217 if (data->cy > sy - 1)
218 data->cy = sy - 1;
219 if (data->cx > sx)
220 data->cx = sx;
221
222 window_copy_clear_selection(wp);
223
224 screen_write_start(&ctx, NULL, s);
225 window_copy_write_lines(wp, &ctx, 0, screen_size_y(s) - 1);
226 screen_write_stop(&ctx);
227
228 window_copy_redraw_screen(wp);
229 }
230
231 void
232 window_copy_key(struct window_pane *wp, struct client *c, int key)
233 {
234 const char *word_separators;
235 struct window_copy_mode_data *data = wp->modedata;
236 struct screen *s = &data->screen;
237 u_int n, np;
238 int keys;
239 enum mode_key_cmd cmd;
240
241 np = data->numprefix;
242 if (np == 0)
243 np = 1;
244
245 if (data->inputtype == WINDOW_COPY_NUMERICPREFIX) {
246 if (window_copy_key_numeric_prefix(wp, key) == 0)
247 return;
248 data->inputtype = WINDOW_COPY_OFF;
249 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
250 } else if (data->inputtype != WINDOW_COPY_OFF) {
251 if (window_copy_key_input(wp, key) != 0)
252 goto input_off;
253 return;
254 }
255
256 cmd = mode_key_lookup(&data->mdata, key);
257 switch (cmd) {
258 case MODEKEYCOPY_CANCEL:
259 window_pane_reset_mode(wp);
260 return;
261 case MODEKEYCOPY_LEFT:
262 for (; np != 0; np--)
263 window_copy_cursor_left(wp);
264 break;
265 case MODEKEYCOPY_RIGHT:
266 for (; np != 0; np--)
267 window_copy_cursor_right(wp);
268 break;
269 case MODEKEYCOPY_UP:
270 for (; np != 0; np--)
271 window_copy_cursor_up(wp, 0);
272 break;
273 case MODEKEYCOPY_DOWN:
274 for (; np != 0; np--)
275 window_copy_cursor_down(wp, 0);
276 break;
277 case MODEKEYCOPY_SCROLLUP:
278 for (; np != 0; np--)
279 window_copy_cursor_up(wp, 1);
280 break;
281 case MODEKEYCOPY_SCROLLDOWN:
282 for (; np != 0; np--)
283 window_copy_cursor_down(wp, 1);
284 break;
285 case MODEKEYCOPY_PREVIOUSPAGE:
286 for (; np != 0; np--)
287 window_copy_pageup(wp);
288 break;
289 case MODEKEYCOPY_NEXTPAGE:
290 n = 1;
291 if (screen_size_y(s) > 2)
292 n = screen_size_y(s) - 2;
293 for (; np != 0; np--) {
294 if (data->oy < n)
295 data->oy = 0;
296 else
297 data->oy -= n;
298 }
299 window_copy_update_selection(wp);
300 window_copy_redraw_screen(wp);
301 break;
302 case MODEKEYCOPY_HALFPAGEUP:
303 n = screen_size_y(s) / 2;
304 for (; np != 0; np--) {
305 if (data->oy + n > screen_hsize(&wp->base))
306 data->oy = screen_hsize(&wp->base);
307 else
308 data->oy += n;
309 }
310 window_copy_update_selection(wp);
311 window_copy_redraw_screen(wp);
312 break;
313 case MODEKEYCOPY_HALFPAGEDOWN:
314 n = screen_size_y(s) / 2;
315 for (; np != 0; np--) {
316 if (data->oy < n)
317 data->oy = 0;
318 else
319 data->oy -= n;
320 }
321 window_copy_update_selection(wp);
322 window_copy_redraw_screen(wp);
323 break;
324 case MODEKEYCOPY_TOPLINE:
325 data->cx = 0;
326 data->cy = 0;
327 window_copy_update_selection(wp);
328 window_copy_redraw_screen(wp);
329 break;
330 case MODEKEYCOPY_MIDDLELINE:
331 data->cx = 0;
332 data->cy = (screen_size_y(s) - 1) / 2;
333 window_copy_update_selection(wp);
334 window_copy_redraw_screen(wp);
335 break;
336 case MODEKEYCOPY_BOTTOMLINE:
337 data->cx = 0;
338 data->cy = screen_size_y(s) - 1;
339 window_copy_update_selection(wp);
340 window_copy_redraw_screen(wp);
341 break;
342 case MODEKEYCOPY_HISTORYTOP:
343 data->cx = 0;
344 data->cy = 0;
345 data->oy = screen_hsize(&wp->base);
346 window_copy_update_selection(wp);
347 window_copy_redraw_screen(wp);
348 break;
349 case MODEKEYCOPY_HISTORYBOTTOM:
350 data->cx = 0;
351 data->cy = screen_size_y(s) - 1;
352 data->oy = 0;
353 window_copy_update_selection(wp);
354 window_copy_redraw_screen(wp);
355 break;
356 case MODEKEYCOPY_STARTSELECTION:
357 window_copy_start_selection(wp);
358 window_copy_redraw_screen(wp);
359 break;
360 case MODEKEYCOPY_CLEARSELECTION:
361 window_copy_clear_selection(wp);
362 window_copy_redraw_screen(wp);
363 break;
364 case MODEKEYCOPY_COPYSELECTION:
365 if (c != NULL && c->session != NULL) {
366 window_copy_copy_selection(wp, c);
367 window_pane_reset_mode(wp);
368 return;
369 }
370 break;
371 case MODEKEYCOPY_STARTOFLINE:
372 window_copy_cursor_start_of_line(wp);
373 break;
374 case MODEKEYCOPY_BACKTOINDENTATION:
375 window_copy_cursor_back_to_indentation(wp);
376 break;
377 case MODEKEYCOPY_ENDOFLINE:
378 window_copy_cursor_end_of_line(wp);
379 break;
380 case MODEKEYCOPY_NEXTSPACE:
381 for (; np != 0; np--)
382 window_copy_cursor_next_word(wp, " ");
383 break;
384 case MODEKEYCOPY_NEXTSPACEEND:
385 for (; np != 0; np--)
386 window_copy_cursor_next_word_end(wp, " ");
387 break;
388 case MODEKEYCOPY_NEXTWORD:
389 word_separators =
390 options_get_string(&wp->window->options, "word-separators");
391 for (; np != 0; np--)
392 window_copy_cursor_next_word(wp, word_separators);
393 break;
394 case MODEKEYCOPY_NEXTWORDEND:
395 word_separators =
396 options_get_string(&wp->window->options, "word-separators");
397 for (; np != 0; np--)
398 window_copy_cursor_next_word_end(wp, word_separators);
399 break;
400 case MODEKEYCOPY_PREVIOUSSPACE:
401 for (; np != 0; np--)
402 window_copy_cursor_previous_word(wp, " ");
403 break;
404 case MODEKEYCOPY_PREVIOUSWORD:
405 word_separators =
406 options_get_string(&wp->window->options, "word-separators");
407 for (; np != 0; np--)
408 window_copy_cursor_previous_word(wp, word_separators);
409 break;
410 case MODEKEYCOPY_SEARCHUP:
411 data->inputtype = WINDOW_COPY_SEARCHUP;
412 data->inputprompt = "Search Up";
413 goto input_on;
414 case MODEKEYCOPY_SEARCHDOWN:
415 data->inputtype = WINDOW_COPY_SEARCHDOWN;
416 data->inputprompt = "Search Down";
417 goto input_on;
418 case MODEKEYCOPY_SEARCHAGAIN:
419 case MODEKEYCOPY_SEARCHREVERSE:
420 switch (data->searchtype) {
421 case WINDOW_COPY_OFF:
422 case WINDOW_COPY_GOTOLINE:
423 case WINDOW_COPY_NUMERICPREFIX:
424 break;
425 case WINDOW_COPY_SEARCHUP:
426 if (cmd == MODEKEYCOPY_SEARCHAGAIN) {
427 for (; np != 0; np--) {
428 window_copy_search_up(
429 wp, data->searchstr);
430 }
431 } else {
432 for (; np != 0; np--) {
433 window_copy_search_down(
434 wp, data->searchstr);
435 }
436 }
437 break;
438 case WINDOW_COPY_SEARCHDOWN:
439 if (cmd == MODEKEYCOPY_SEARCHAGAIN) {
440 for (; np != 0; np--) {
441 window_copy_search_down(
442 wp, data->searchstr);
443 }
444 } else {
445 for (; np != 0; np--) {
446 window_copy_search_up(
447 wp, data->searchstr);
448 }
449 }
450 break;
451 }
452 break;
453 case MODEKEYCOPY_GOTOLINE:
454 data->inputtype = WINDOW_COPY_GOTOLINE;
455 data->inputprompt = "Goto Line";
456 *data->inputstr = '\0';
457 goto input_on;
458 case MODEKEYCOPY_STARTNUMBERPREFIX:
459 key &= 0xff;
460 if (key >= '0' && key <= '9') {
461 data->inputtype = WINDOW_COPY_NUMERICPREFIX;
462 data->numprefix = 0;
463 window_copy_key_numeric_prefix(wp, key);
464 return;
465 }
466 break;
467 case MODEKEYCOPY_RECTANGLETOGGLE:
468 window_copy_rectangle_toggle(wp);
469 break;
470 default:
471 break;
472 }
473
474 data->numprefix = 0;
475 return;
476
477 input_on:
478 keys = options_get_number(&wp->window->options, "mode-keys");
479 if (keys == MODEKEY_EMACS)
480 mode_key_init(&data->mdata, &mode_key_tree_emacs_edit);
481 else
482 mode_key_init(&data->mdata, &mode_key_tree_vi_edit);
483
484 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
485 return;
486
487 input_off:
488 keys = options_get_number(&wp->window->options, "mode-keys");
489 if (keys == MODEKEY_EMACS)
490 mode_key_init(&data->mdata, &mode_key_tree_emacs_copy);
491 else
492 mode_key_init(&data->mdata, &mode_key_tree_vi_copy);
493
494 data->inputtype = WINDOW_COPY_OFF;
495 data->inputprompt = NULL;
496
497 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
498 }
499
500 int
501 window_copy_key_input(struct window_pane *wp, int key)
502 {
503 struct window_copy_mode_data *data = wp->modedata;
504 struct screen *s = &data->screen;
505 size_t inputlen;
506 u_int np;
507
508 switch (mode_key_lookup(&data->mdata, key)) {
509 case MODEKEYEDIT_CANCEL:
510 data->numprefix = 0;
511 return (-1);
512 case MODEKEYEDIT_BACKSPACE:
513 inputlen = strlen(data->inputstr);
514 if (inputlen > 0)
515 data->inputstr[inputlen - 1] = '\0';
516 break;
517 case MODEKEYEDIT_DELETELINE:
518 *data->inputstr = '\0';
519 break;
520 case MODEKEYEDIT_ENTER:
521 np = data->numprefix;
522 if (np == 0)
523 np = 1;
524
525 switch (data->inputtype) {
526 case WINDOW_COPY_OFF:
527 case WINDOW_COPY_NUMERICPREFIX:
528 break;
529 case WINDOW_COPY_SEARCHUP:
530 for (; np != 0; np--)
531 window_copy_search_up(wp, data->inputstr);
532 data->searchtype = data->inputtype;
533 data->searchstr = xstrdup(data->inputstr);
534 break;
535 case WINDOW_COPY_SEARCHDOWN:
536 for (; np != 0; np--)
537 window_copy_search_down(wp, data->inputstr);
538 data->searchtype = data->inputtype;
539 data->searchstr = xstrdup(data->inputstr);
540 break;
541 case WINDOW_COPY_GOTOLINE:
542 window_copy_goto_line(wp, data->inputstr);
543 *data->inputstr = '\0';
544 break;
545 }
546 data->numprefix = 0;
547 return (1);
548 case MODEKEY_OTHER:
549 if (key < 32 || key > 126)
550 break;
551 inputlen = strlen(data->inputstr) + 2;
552
553 data->inputstr = xrealloc(data->inputstr, 1, inputlen);
554 data->inputstr[inputlen - 2] = key;
555 data->inputstr[inputlen - 1] = '\0';
556 break;
557 default:
558 break;
559 }
560
561 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
562 return (0);
563 }
564
565 int
566 window_copy_key_numeric_prefix(struct window_pane *wp, int key)
567 {
568 struct window_copy_mode_data *data = wp->modedata;
569 struct screen *s = &data->screen;
570
571 key &= 0xff;
572 if (key < '0' || key > '9')
573 return 1;
574
575 if (data->numprefix >= 100) /* no more than three digits */
576 return 0;
577 data->numprefix = data->numprefix * 10 + key - '0';
578
579 window_copy_redraw_lines(wp, screen_size_y(s) - 1, 1);
580 return 0;
581 }
582
583 /* ARGSUSED */
584 void
585 window_copy_mouse(
586 struct window_pane *wp, unused struct client *c, struct mouse_event *m)
587 {
588 struct window_copy_mode_data *data = wp->modedata;
589 struct screen *s = &data->screen;
590
591 if ((m->b & 3) == 3)
592 return;
593 if (m->x >= screen_size_x(s))
594 return;
595 if (m->y >= screen_size_y(s))
596 return;
597
598 window_copy_update_cursor(wp, m->x, m->y);
599 if (window_copy_update_selection(wp))
600 window_copy_redraw_screen(wp);
601 }
602
603 void
604 window_copy_scroll_to(struct window_pane *wp, u_int px, u_int py)
605 {
606 struct window_copy_mode_data *data = wp->modedata;
607 struct screen *s = &wp->base;
608 struct grid *gd = s->grid;
609 u_int offset, gap;
610
611 data->cx = px;
612
613 gap = gd->sy / 4;
614 if (py < gd->sy) {
615 offset = 0;
616 data->cy = py;
617 } else if (py > gd->hsize + gd->sy - gap) {
618 offset = gd->hsize;
619 data->cy = py - gd->hsize;
620 } else {
621 offset = py + gap - gd->sy;
622 data->cy = py - offset;
623 }
624 data->oy = gd->hsize - offset;
625
626 window_copy_update_selection(wp);
627 window_copy_redraw_screen(wp);
628 }
629
630 int
631 window_copy_search_compare(
632 struct grid *gd, u_int px, u_int py, struct grid *sgd, u_int spx)
633 {
634 const struct grid_cell *gc, *sgc;
635 const struct grid_utf8 *gu, *sgu;
636
637 gc = grid_peek_cell(gd, px, py);
638 sgc = grid_peek_cell(sgd, spx, 0);
639
640 if ((gc->flags & GRID_FLAG_UTF8) != (sgc->flags & GRID_FLAG_UTF8))
641 return (0);
642
643 if (gc->flags & GRID_FLAG_UTF8) {
644 gu = grid_peek_utf8(gd, px, py);
645 sgu = grid_peek_utf8(sgd, spx, 0);
646 if (grid_utf8_compare(gu, sgu))
647 return (1);
648 } else {
649 if (gc->data == sgc->data)
650 return (1);
651 }
652 return (0);
653 }
654
655 int
656 window_copy_search_lr(struct grid *gd,
657 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
658 {
659 u_int ax, bx, px;
660
661 for (ax = first; ax < last; ax++) {
662 if (ax + sgd->sx >= gd->sx)
663 break;
664 for (bx = 0; bx < sgd->sx; bx++) {
665 px = ax + bx;
666 if (!window_copy_search_compare(gd, px, py, sgd, bx))
667 break;
668 }
669 if (bx == sgd->sx) {
670 *ppx = ax;
671 return (1);
672 }
673 }
674 return (0);
675 }
676
677 int
678 window_copy_search_rl(struct grid *gd,
679 struct grid *sgd, u_int *ppx, u_int py, u_int first, u_int last)
680 {
681 u_int ax, bx, px;
682
683 for (ax = last + 1; ax > first; ax--) {
684 if (gd->sx - (ax - 1) < sgd->sx)
685 continue;
686 for (bx = 0; bx < sgd->sx; bx++) {
687 px = ax - 1 + bx;
688 if (!window_copy_search_compare(gd, px, py, sgd, bx))
689 break;
690 }
691 if (bx == sgd->sx) {
692 *ppx = ax - 1;
693 return (1);
694 }
695 }
696 return (0);
697 }
698
699 void
700 window_copy_search_up(struct window_pane *wp, const char *searchstr)
701 {
702 struct window_copy_mode_data *data = wp->modedata;
703 struct screen *s = &wp->base, ss;
704 struct screen_write_ctx ctx;
705 struct grid *gd = s->grid, *sgd;
706 struct grid_cell gc;
707 size_t searchlen;
708 u_int i, last, fx, fy, px;
709 int utf8flag, n, wrapped;
710
711 if (*searchstr == '\0')
712 return;
713 utf8flag = options_get_number(&wp->window->options, "utf8");
714 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
715
716 screen_init(&ss, searchlen, 1, 0);
717 screen_write_start(&ctx, NULL, &ss);
718 memcpy(&gc, &grid_default_cell, sizeof gc);
719 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
720 screen_write_stop(&ctx);
721
722 fx = data->cx;
723 fy = gd->hsize - data->oy + data->cy;
724
725 if (fx == 0) {
726 if (fy == 0)
727 return;
728 fx = gd->sx - 1;
729 fy--;
730 } else
731 fx--;
732 n = wrapped = 0;
733
734 retry:
735 sgd = ss.grid;
736 for (i = fy + 1; i > 0; i--) {
737 last = screen_size_x(s);
738 if (i == fy + 1)
739 last = fx;
740 n = window_copy_search_rl(gd, sgd, &px, i - 1, 0, last);
741 if (n) {
742 window_copy_scroll_to(wp, px, i - 1);
743 break;
744 }
745 }
746 if (!n && !wrapped) {
747 fx = gd->sx - 1;
748 fy = gd->hsize + gd->sy - 1;
749 wrapped = 1;
750 goto retry;
751 }
752
753 screen_free(&ss);
754 }
755
756 void
757 window_copy_search_down(struct window_pane *wp, const char *searchstr)
758 {
759 struct window_copy_mode_data *data = wp->modedata;
760 struct screen *s = &wp->base, ss;
761 struct screen_write_ctx ctx;
762 struct grid *gd = s->grid, *sgd;
763 struct grid_cell gc;
764 size_t searchlen;
765 u_int i, first, fx, fy, px;
766 int utf8flag, n, wrapped;
767
768 if (*searchstr == '\0')
769 return;
770 utf8flag = options_get_number(&wp->window->options, "utf8");
771 searchlen = screen_write_strlen(utf8flag, "%s", searchstr);
772
773 screen_init(&ss, searchlen, 1, 0);
774 screen_write_start(&ctx, NULL, &ss);
775 memcpy(&gc, &grid_default_cell, sizeof gc);
776 screen_write_nputs(&ctx, -1, &gc, utf8flag, "%s", searchstr);
777 screen_write_stop(&ctx);
778
779 fx = data->cx;
780 fy = gd->hsize - data->oy + data->cy;
781
782 if (fx == gd->sx - 1) {
783 if (fy == gd->hsize + gd->sy)
784 return;
785 fx = 0;
786 fy++;
787 } else
788 fx++;
789 n = wrapped = 0;
790
791 retry:
792 sgd = ss.grid;
793 for (i = fy + 1; i < gd->hsize + gd->sy; i++) {
794 first = 0;
795 if (i == fy + 1)
796 first = fx;
797 n = window_copy_search_lr(gd, sgd, &px, i - 1, first, gd->sx);
798 if (n) {
799 window_copy_scroll_to(wp, px, i - 1);
800 break;
801 }
802 }
803 if (!n && !wrapped) {
804 fx = 0;
805 fy = 0;
806 wrapped = 1;
807 goto retry;
808 }
809
810 screen_free(&ss);
811 }
812
813 void
814 window_copy_goto_line(struct window_pane *wp, const char *linestr)
815 {
816 struct window_copy_mode_data *data = wp->modedata;
817 const char *errstr;
818 u_int lineno;
819
820 lineno = strtonum(linestr, 0, screen_hsize(&wp->base), &errstr);
821 if (errstr != NULL)
822 return;
823
824 data->oy = lineno;
825 window_copy_update_selection(wp);
826 window_copy_redraw_screen(wp);
827 }
828
829 void
830 window_copy_write_line(
831 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py)
832 {
833 struct window_copy_mode_data *data = wp->modedata;
834 struct screen *s = &data->screen;
835 struct options *oo = &wp->window->options;
836 struct grid_cell gc;
837 char hdr[32];
838 size_t last, xoff = 0, size = 0;
839
840 memcpy(&gc, &grid_default_cell, sizeof gc);
841 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
842 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
843 gc.attr |= options_get_number(oo, "mode-attr");
844
845 last = screen_size_y(s) - 1;
846 if (py == 0) {
847 size = xsnprintf(hdr, sizeof hdr,
848 "[%u/%u]", data->oy, screen_hsize(&wp->base));
849 screen_write_cursormove(ctx, screen_size_x(s) - size, 0);
850 screen_write_puts(ctx, &gc, "%s", hdr);
851 } else if (py == last && data->inputtype != WINDOW_COPY_OFF) {
852 if (data->inputtype == WINDOW_COPY_NUMERICPREFIX) {
853 xoff = size = xsnprintf(hdr, sizeof hdr,
854 "Repeat: %u", data->numprefix);
855 } else {
856 xoff = size = xsnprintf(hdr, sizeof hdr,
857 "%s: %s", data->inputprompt, data->inputstr);
858 }
859 screen_write_cursormove(ctx, 0, last);
860 screen_write_puts(ctx, &gc, "%s", hdr);
861 } else
862 size = 0;
863
864 screen_write_cursormove(ctx, xoff, py);
865 screen_write_copy(ctx, &wp->base, xoff, (screen_hsize(&wp->base) -
866 data->oy) + py, screen_size_x(s) - size, 1);
867
868 if (py == data->cy && data->cx == screen_size_x(s)) {
869 memcpy(&gc, &grid_default_cell, sizeof gc);
870 screen_write_cursormove(ctx, screen_size_x(s) - 1, py);
871 screen_write_putc(ctx, &gc, '$');
872 }
873 }
874
875 void
876 window_copy_write_lines(
877 struct window_pane *wp, struct screen_write_ctx *ctx, u_int py, u_int ny)
878 {
879 u_int yy;
880
881 for (yy = py; yy < py + ny; yy++)
882 window_copy_write_line(wp, ctx, py);
883 }
884
885 void
886 window_copy_redraw_lines(struct window_pane *wp, u_int py, u_int ny)
887 {
888 struct window_copy_mode_data *data = wp->modedata;
889 struct screen_write_ctx ctx;
890 u_int i;
891
892 screen_write_start(&ctx, wp, NULL);
893 for (i = py; i < py + ny; i++)
894 window_copy_write_line(wp, &ctx, i);
895 screen_write_cursormove(&ctx, data->cx, data->cy);
896 screen_write_stop(&ctx);
897 }
898
899 void
900 window_copy_redraw_screen(struct window_pane *wp)
901 {
902 struct window_copy_mode_data *data = wp->modedata;
903
904 window_copy_redraw_lines(wp, 0, screen_size_y(&data->screen));
905 }
906
907 void
908 window_copy_update_cursor(struct window_pane *wp, u_int cx, u_int cy)
909 {
910 struct window_copy_mode_data *data = wp->modedata;
911 struct screen *s = &data->screen;
912 struct screen_write_ctx ctx;
913 u_int old_cx, old_cy;
914
915 old_cx = data->cx; old_cy = data->cy;
916 data->cx = cx; data->cy = cy;
917 if (old_cx == screen_size_x(s))
918 window_copy_redraw_lines(wp, old_cy, 1);
919 if (data->cx == screen_size_x(s))
920 window_copy_redraw_lines(wp, data->cy, 1);
921 else {
922 screen_write_start(&ctx, wp, NULL);
923 screen_write_cursormove(&ctx, data->cx, data->cy);
924 screen_write_stop(&ctx);
925 }
926 }
927
928 void
929 window_copy_start_selection(struct window_pane *wp)
930 {
931 struct window_copy_mode_data *data = wp->modedata;
932 struct screen *s = &data->screen;
933
934 data->selx = data->cx;
935 data->sely = screen_hsize(&wp->base) + data->cy - data->oy;
936
937 s->sel.flag = 1;
938 window_copy_update_selection(wp);
939 }
940
941 int
942 window_copy_update_selection(struct window_pane *wp)
943 {
944 struct window_copy_mode_data *data = wp->modedata;
945 struct screen *s = &data->screen;
946 struct options *oo = &wp->window->options;
947 struct grid_cell gc;
948 u_int sx, sy, ty, cy;
949
950 if (!s->sel.flag)
951 return (0);
952
953 /* Set colours. */
954 memcpy(&gc, &grid_default_cell, sizeof gc);
955 colour_set_fg(&gc, options_get_number(oo, "mode-fg"));
956 colour_set_bg(&gc, options_get_number(oo, "mode-bg"));
957 gc.attr |= options_get_number(oo, "mode-attr");
958
959 /* Find top of screen. */
960 ty = screen_hsize(&wp->base) - data->oy;
961
962 /* Adjust the selection. */
963 sx = data->selx;
964 sy = data->sely;
965 if (sy < ty) { /* above screen */
966 if (!data->rectflag)
967 sx = 0;
968 sy = 0;
969 } else if (sy > ty + screen_size_y(s) - 1) { /* below screen */
970 if (!data->rectflag)
971 sx = screen_size_x(s) - 1;
972 sy = screen_size_y(s) - 1;
973 } else
974 sy -= ty;
975 sy = screen_hsize(s) + sy;
976
977 screen_set_selection(s,
978 sx, sy, data->cx, screen_hsize(s) + data->cy, data->rectflag, &gc);
979
980 if (data->rectflag) {
981 /*
982 * Can't rely on the caller to redraw the right lines for
983 * rectangle selection - find the highest line and the number
984 * of lines, and redraw just past that in both directions
985 */
986 cy = data->cy;
987 if (sy < cy)
988 window_copy_redraw_lines(wp, sy, cy - sy + 1);
989 else
990 window_copy_redraw_lines(wp, cy, sy - cy + 1);
991 }
992
993 return (1);
994 }
995
996 void
997 window_copy_copy_selection(struct window_pane *wp, struct client *c)
998 {
999 struct window_copy_mode_data *data = wp->modedata;
1000 struct screen *s = &data->screen;
1001 char *buf;
1002 size_t off;
1003 u_int i, xx, yy, sx, sy, ex, ey, limit;
1004 u_int firstsx, lastex, restex, restsx;
1005
1006 if (!s->sel.flag)
1007 return;
1008
1009 buf = xmalloc(1);
1010 off = 0;
1011
1012 *buf = '\0';
1013
1014 /*
1015 * The selection extends from selx,sely to (adjusted) cx,cy on
1016 * the base screen.
1017 */
1018
1019 /* Find start and end. */
1020 xx = data->cx;
1021 yy = screen_hsize(&wp->base) + data->cy - data->oy;
1022 if (yy < data->sely || (yy == data->sely && xx < data->selx)) {
1023 sx = xx; sy = yy;
1024 ex = data->selx; ey = data->sely;
1025 } else {
1026 sx = data->selx; sy = data->sely;
1027 ex = xx; ey = yy;
1028 }
1029
1030 /* Trim ex to end of line. */
1031 xx = window_copy_find_length(wp, ey);
1032 if (ex > xx)
1033 ex = xx;
1034
1035 /*
1036 * Deal with rectangle-copy if necessary; four situations: start of
1037 * first line (firstsx), end of last line (lastex), start (restsx) and
1038 * end (restex) of all other lines.
1039 */
1040 xx = screen_size_x(s);
1041 if (data->rectflag) {
1042 /*
1043 * Need to ignore the column with the cursor in it, which for
1044 * rectangular copy means knowing which side the cursor is on.
1045 */
1046 if (data->selx < data->cx) {
1047 /* Selection start is on the left. */
1048 lastex = data->cx;
1049 restex = data->cx;
1050 firstsx = data->selx;
1051 restsx = data->selx;
1052 } else {
1053 /* Cursor is on the left. */
1054 lastex = data->selx + 1;
1055 restex = data->selx + 1;
1056 firstsx = data->cx + 1;
1057 restsx = data->cx + 1;
1058 }
1059 } else {
1060 /*
1061 * Like emacs, keep the top-left-most character, and drop the
1062 * bottom-right-most, regardless of copy direction.
1063 */
1064 lastex = ex;
1065 restex = xx;
1066 firstsx = sx;
1067 restsx = 0;
1068 }
1069
1070 /* Copy the lines. */
1071 if (sy == ey)
1072 window_copy_copy_line(wp, &buf, &off, sy, firstsx, lastex);
1073 else {
1074 window_copy_copy_line(wp, &buf, &off, sy, firstsx, restex);
1075 if (ey - sy > 1) {
1076 for (i = sy + 1; i < ey; i++) {
1077 window_copy_copy_line(
1078 wp, &buf, &off, i, restsx, restex);
1079 }
1080 }
1081 window_copy_copy_line(wp, &buf, &off, ey, restsx, lastex);
1082 }
1083
1084 /* Don't bother if no data. */
1085 if (off == 0) {
1086 xfree(buf);
1087 return;
1088 }
1089 off--; /* remove final \n */
1090
1091 /* Add the buffer to the stack. */
1092 limit = options_get_number(&c->session->options, "buffer-limit");
1093 paste_add(&c->session->buffers, buf, off, limit);
1094 }
1095
1096 void
1097 window_copy_copy_line(struct window_pane *wp,
1098 char **buf, size_t *off, u_int sy, u_int sx, u_int ex)
1099 {
1100 struct grid *gd = wp->base.grid;
1101 const struct grid_cell *gc;
1102 const struct grid_utf8 *gu;
1103 struct grid_line *gl;
1104 u_int i, xx, wrapped = 0;
1105 size_t size;
1106
1107 if (sx > ex)
1108 return;
1109
1110 /*
1111 * Work out if the line was wrapped at the screen edge and all of it is
1112 * on screen.
1113 */
1114 gl = &gd->linedata[sy];
1115 if (gl->flags & GRID_LINE_WRAPPED && gl->cellsize <= gd->sx)
1116 wrapped = 1;
1117
1118 /* If the line was wrapped, don't strip spaces (use the full length). */
1119 if (wrapped)
1120 xx = gl->cellsize;
1121 else
1122 xx = window_copy_find_length(wp, sy);
1123 if (ex > xx)
1124 ex = xx;
1125 if (sx > xx)
1126 sx = xx;
1127
1128 if (sx < ex) {
1129 for (i = sx; i < ex; i++) {
1130 gc = grid_peek_cell(gd, i, sy);
1131 if (gc->flags & GRID_FLAG_PADDING)
1132 continue;
1133 if (!(gc->flags & GRID_FLAG_UTF8)) {
1134 *buf = xrealloc(*buf, 1, (*off) + 1);
1135 (*buf)[(*off)++] = gc->data;
1136 } else {
1137 gu = grid_peek_utf8(gd, i, sy);
1138 size = grid_utf8_size(gu);
1139 *buf = xrealloc(*buf, 1, (*off) + size);
1140 *off += grid_utf8_copy(gu, *buf + *off, size);
1141 }
1142 }
1143 }
1144
1145 /* Only add a newline if the line wasn't wrapped. */
1146 if (!wrapped || ex != xx) {
1147 *buf = xrealloc(*buf, 1, (*off) + 1);
1148 (*buf)[(*off)++] = '\n';
1149 }
1150 }
1151
1152 void
1153 window_copy_clear_selection(struct window_pane *wp)
1154 {
1155 struct window_copy_mode_data *data = wp->modedata;
1156 u_int px, py;
1157
1158 screen_clear_selection(&data->screen);
1159
1160 py = screen_hsize(&wp->base) + data->cy - data->oy;
1161 px = window_copy_find_length(wp, py);
1162 if (data->cx > px)
1163 window_copy_update_cursor(wp, px, data->cy);
1164 }
1165
1166 int
1167 window_copy_in_set(struct window_pane *wp, u_int px, u_int py, const char *set)
1168 {
1169 const struct grid_cell *gc;
1170
1171 gc = grid_peek_cell(wp->base.grid, px, py);
1172 if (gc->flags & (GRID_FLAG_PADDING|GRID_FLAG_UTF8))
1173 return (0);
1174 if (gc->data == 0x00 || gc->data == 0x7f)
1175 return (0);
1176 return (strchr(set, gc->data) != NULL);
1177 }
1178
1179 u_int
1180 window_copy_find_length(struct window_pane *wp, u_int py)
1181 {
1182 const struct grid_cell *gc;
1183 u_int px;
1184
1185 /*
1186 * If the pane has been resized, its grid can contain old overlong
1187 * lines. grid_peek_cell does not allow accessing cells beyond the
1188 * width of the grid, and screen_write_copy treats them as spaces, so
1189 * ignore them here too.
1190 */
1191 px = wp->base.grid->linedata[py].cellsize;
1192 if (px > screen_size_x(&wp->base))
1193 px = screen_size_x(&wp->base);
1194 while (px > 0) {
1195 gc = grid_peek_cell(wp->base.grid, px - 1, py);
1196 if (gc->flags & GRID_FLAG_UTF8)
1197 break;
1198 if (gc->data != ' ')
1199 break;
1200 px--;
1201 }
1202 return (px);
1203 }
1204
1205 void
1206 window_copy_cursor_start_of_line(struct window_pane *wp)
1207 {
1208 struct window_copy_mode_data *data = wp->modedata;
1209
1210 window_copy_update_cursor(wp, 0, data->cy);
1211 if (window_copy_update_selection(wp))
1212 window_copy_redraw_lines(wp, data->cy, 1);
1213 }
1214
1215 void
1216 window_copy_cursor_back_to_indentation(struct window_pane *wp)
1217 {
1218 struct window_copy_mode_data *data = wp->modedata;
1219 u_int px, py, xx;
1220 const struct grid_cell *gc;
1221
1222 px = 0;
1223 py = screen_hsize(&wp->base) + data->cy - data->oy;
1224 xx = window_copy_find_length(wp, py);
1225
1226 while (px < xx) {
1227 gc = grid_peek_cell(wp->base.grid, px, py);
1228 if (gc->flags & GRID_FLAG_UTF8)
1229 break;
1230 if (gc->data != ' ')
1231 break;
1232 px++;
1233 }
1234
1235 window_copy_update_cursor(wp, px, data->cy);
1236 if (window_copy_update_selection(wp))
1237 window_copy_redraw_lines(wp, data->cy, 1);
1238 }
1239
1240 void
1241 window_copy_cursor_end_of_line(struct window_pane *wp)
1242 {
1243 struct window_copy_mode_data *data = wp->modedata;
1244 struct screen *base_s = &wp->base;
1245 struct grid *gd = base_s->grid;
1246 u_int px, py;
1247
1248 py = screen_hsize(base_s) + data->cy - data->oy;
1249 px = window_copy_find_length(wp, py);
1250
1251 if (data->cx == px) {
1252 if (data->screen.sel.flag && data->rectflag)
1253 px = screen_size_x(&wp->base);
1254 if (gd->linedata[py].flags & GRID_LINE_WRAPPED) {
1255 while (py < gd->sy + gd->hsize &&
1256 gd->linedata[py].flags & GRID_LINE_WRAPPED) {
1257 window_copy_cursor_down(wp, 0);
1258 py = screen_hsize(base_s) + data->cy - data->oy;
1259 }
1260 px = window_copy_find_length(wp, py);
1261 }
1262 }
1263 window_copy_update_cursor(wp, px, data->cy);
1264
1265 if (window_copy_update_selection(wp))
1266 window_copy_redraw_lines(wp, data->cy, 1);
1267 }
1268
1269 void
1270 window_copy_cursor_left(struct window_pane *wp)
1271 {
1272 struct window_copy_mode_data *data = wp->modedata;
1273
1274 if (data->cx == 0) {
1275 window_copy_cursor_up(wp, 0);
1276 window_copy_cursor_end_of_line(wp);
1277 } else {
1278 window_copy_update_cursor(wp, data->cx - 1, data->cy);
1279 if (window_copy_update_selection(wp))
1280 window_copy_redraw_lines(wp, data->cy, 1);
1281 }
1282 }
1283
1284 void
1285 window_copy_cursor_right(struct window_pane *wp)
1286 {
1287 struct window_copy_mode_data *data = wp->modedata;
1288 u_int px, py;
1289
1290 if (data->screen.sel.flag && data->rectflag)
1291 px = screen_size_x(&data->screen);
1292 else {
1293 py = screen_hsize(&wp->base) + data->cy - data->oy;
1294 px = window_copy_find_length(wp, py);
1295 }
1296
1297 if (data->cx >= px) {
1298 window_copy_cursor_start_of_line(wp);
1299 window_copy_cursor_down(wp, 0);
1300 } else {
1301 window_copy_update_cursor(wp, data->cx + 1, data->cy);
1302 if (window_copy_update_selection(wp))
1303 window_copy_redraw_lines(wp, data->cy, 1);
1304 }
1305 }
1306
1307 void
1308 window_copy_cursor_up(struct window_pane *wp, int scroll_only)
1309 {
1310 struct window_copy_mode_data *data = wp->modedata;
1311 struct screen *s = &data->screen;
1312 u_int ox, oy, px, py;
1313
1314 oy = screen_hsize(&wp->base) + data->cy - data->oy;
1315 ox = window_copy_find_length(wp, oy);
1316 if (ox != 0) {
1317 data->lastcx = data->cx;
1318 data->lastsx = ox;
1319 }
1320
1321 data->cx = data->lastcx;
1322 if (scroll_only || data->cy == 0) {
1323 window_copy_scroll_down(wp, 1);
1324 if (scroll_only) {
1325 if (data->cy == screen_size_y(s) - 1)
1326 window_copy_redraw_lines(wp, data->cy, 1);
1327 else
1328 window_copy_redraw_lines(wp, data->cy, 2);
1329 }
1330 } else {
1331 window_copy_update_cursor(wp, data->cx, data->cy - 1);
1332 if (window_copy_update_selection(wp)) {
1333 if (data->cy == screen_size_y(s) - 1)
1334 window_copy_redraw_lines(wp, data->cy, 1);
1335 else
1336 window_copy_redraw_lines(wp, data->cy, 2);
1337 }
1338 }
1339
1340 if (!data->screen.sel.flag || !data->rectflag) {
1341 py = screen_hsize(&wp->base) + data->cy - data->oy;
1342 px = window_copy_find_length(wp, py);
1343 if ((data->cx >= data->lastsx && data->cx != px) ||
1344 data->cx > px)
1345 window_copy_cursor_end_of_line(wp);
1346 }
1347 }
1348
1349 void
1350 window_copy_cursor_down(struct window_pane *wp, int scroll_only)
1351 {
1352 struct window_copy_mode_data *data = wp->modedata;
1353 struct screen *s = &data->screen;
1354 u_int ox, oy, px, py;
1355
1356 oy = screen_hsize(&wp->base) + data->cy - data->oy;
1357 ox = window_copy_find_length(wp, oy);
1358 if (ox != 0) {
1359 data->lastcx = data->cx;
1360 data->lastsx = ox;
1361 }
1362
1363 data->cx = data->lastcx;
1364 if (scroll_only || data->cy == screen_size_y(s) - 1) {
1365 window_copy_scroll_up(wp, 1);
1366 if (scroll_only && data->cy > 0)
1367 window_copy_redraw_lines(wp, data->cy - 1, 2);
1368 } else {
1369 window_copy_update_cursor(wp, data->cx, data->cy + 1);
1370 if (window_copy_update_selection(wp))
1371 window_copy_redraw_lines(wp, data->cy - 1, 2);
1372 }
1373
1374 if (!data->screen.sel.flag || !data->rectflag) {
1375 py = screen_hsize(&wp->base) + data->cy - data->oy;
1376 px = window_copy_find_length(wp, py);
1377 if ((data->cx >= data->lastsx && data->cx != px) ||
1378 data->cx > px)
1379 window_copy_cursor_end_of_line(wp);
1380 }
1381 }
1382
1383 void
1384 window_copy_cursor_next_word(struct window_pane *wp, const char *separators)
1385 {
1386 struct window_copy_mode_data *data = wp->modedata;
1387 struct screen *base_s = &wp->base;
1388 u_int px, py, xx, yy;
1389 int expected = 0;
1390
1391 px = data->cx;
1392 py = screen_hsize(base_s) + data->cy - data->oy;
1393 xx = window_copy_find_length(wp, py);
1394 yy = screen_hsize(base_s) + screen_size_y(base_s) - 1;
1395
1396 /*
1397 * First skip past any nonword characters and then any word characters.
1398 *
1399 * expected is initially set to 0 for the former and then 1 for the
1400 * latter.
1401 */
1402 do {
1403 while (px > xx ||
1404 window_copy_in_set(wp, px, py, separators) == expected) {
1405 /* Move down if we're past the end of the line. */
1406 if (px > xx) {
1407 if (py == yy)
1408 return;
1409 window_copy_cursor_down(wp, 0);
1410 px = 0;
1411
1412 py = screen_hsize(base_s) + data->cy - data->oy;
1413 xx = window_copy_find_length(wp, py);
1414 } else
1415 px++;
1416 }
1417 expected = !expected;
1418 } while (expected == 1);
1419
1420 window_copy_update_cursor(wp, px, data->cy);
1421 if (window_copy_update_selection(wp))
1422 window_copy_redraw_lines(wp, data->cy, 1);
1423 }
1424
1425 void
1426 window_copy_cursor_next_word_end(struct window_pane *wp, const char *separators)
1427 {
1428 struct window_copy_mode_data *data = wp->modedata;
1429 struct screen *base_s = &wp->base;
1430 u_int px, py, xx, yy;
1431 int expected = 1;
1432
1433 px = data->cx;
1434 py = screen_hsize(base_s) + data->cy - data->oy;
1435 xx = window_copy_find_length(wp, py);
1436 yy = screen_hsize(base_s) + screen_size_y(base_s) - 1;
1437
1438 /*
1439 * First skip past any word characters, then any nonword characters.
1440 *
1441 * expected is initially set to 1 for the former and then 0 for the
1442 * latter.
1443 */
1444 do {
1445 while (px > xx ||
1446 window_copy_in_set(wp, px, py, separators) == expected) {
1447 /* Move down if we're past the end of the line. */
1448 if (px > xx) {
1449 if (py == yy)
1450 return;
1451 window_copy_cursor_down(wp, 0);
1452 px = 0;
1453
1454 py = screen_hsize(base_s) + data->cy - data->oy;
1455 xx = window_copy_find_length(wp, py);
1456 } else
1457 px++;
1458 }
1459 expected = !expected;
1460 } while (expected == 0);
1461
1462 window_copy_update_cursor(wp, px, data->cy);
1463 if (window_copy_update_selection(wp))
1464 window_copy_redraw_lines(wp, data->cy, 1);
1465 }
1466
1467 /* Move to the previous place where a word begins. */
1468 void
1469 window_copy_cursor_previous_word(struct window_pane *wp, const char *separators)
1470 {
1471 struct window_copy_mode_data *data = wp->modedata;
1472 u_int px, py;
1473
1474 px = data->cx;
1475 py = screen_hsize(&wp->base) + data->cy - data->oy;
1476
1477 /* Move back to the previous word character. */
1478 for (;;) {
1479 if (px > 0) {
1480 px--;
1481 if (!window_copy_in_set(wp, px, py, separators))
1482 break;
1483 } else {
1484 if (data->cy == 0 &&
1485 (screen_hsize(&wp->base) == 0 ||
1486 data->oy >= screen_hsize(&wp->base) - 1))
1487 goto out;
1488 window_copy_cursor_up(wp, 0);
1489
1490 py = screen_hsize(&wp->base) + data->cy - data->oy;
1491 px = window_copy_find_length(wp, py);
1492 }
1493 }
1494
1495 /* Move back to the beginning of this word. */
1496 while (px > 0 && !window_copy_in_set(wp, px - 1, py, separators))
1497 px--;
1498
1499 out:
1500 window_copy_update_cursor(wp, px, data->cy);
1501 if (window_copy_update_selection(wp))
1502 window_copy_redraw_lines(wp, data->cy, 1);
1503 }
1504
1505 void
1506 window_copy_scroll_up(struct window_pane *wp, u_int ny)
1507 {
1508 struct window_copy_mode_data *data = wp->modedata;
1509 struct screen *s = &data->screen;
1510 struct screen_write_ctx ctx;
1511
1512 if (data->oy < ny)
1513 ny = data->oy;
1514 if (ny == 0)
1515 return;
1516 data->oy -= ny;
1517
1518 screen_write_start(&ctx, wp, NULL);
1519 screen_write_cursormove(&ctx, 0, 0);
1520 screen_write_deleteline(&ctx, ny);
1521 window_copy_write_lines(wp, &ctx, screen_size_y(s) - ny, ny);
1522 window_copy_write_line(wp, &ctx, 0);
1523 if (screen_size_y(s) > 1)
1524 window_copy_write_line(wp, &ctx, 1);
1525 if (screen_size_y(s) > 3)
1526 window_copy_write_line(wp, &ctx, screen_size_y(s) - 2);
1527 if (s->sel.flag && screen_size_y(s) > ny) {
1528 window_copy_update_selection(wp);
1529 window_copy_write_line(wp, &ctx, screen_size_y(s) - ny - 1);
1530 }
1531 screen_write_cursormove(&ctx, data->cx, data->cy);
1532 window_copy_update_selection(wp);
1533 screen_write_stop(&ctx);
1534 }
1535
1536 void
1537 window_copy_scroll_down(struct window_pane *wp, u_int ny)
1538 {
1539 struct window_copy_mode_data *data = wp->modedata;
1540 struct screen *s = &data->screen;
1541 struct screen_write_ctx ctx;
1542
1543 if (ny > screen_hsize(&wp->base))
1544 return;
1545
1546 if (data->oy > screen_hsize(&wp->base) - ny)
1547 ny = screen_hsize(&wp->base) - data->oy;
1548 if (ny == 0)
1549 return;
1550 data->oy += ny;
1551
1552 screen_write_start(&ctx, wp, NULL);
1553 screen_write_cursormove(&ctx, 0, 0);
1554 screen_write_insertline(&ctx, ny);
1555 window_copy_write_lines(wp, &ctx, 0, ny);
1556 if (s->sel.flag && screen_size_y(s) > ny) {
1557 window_copy_update_selection(wp);
1558 window_copy_write_line(wp, &ctx, ny);
1559 } else if (ny == 1) /* nuke position */
1560 window_copy_write_line(wp, &ctx, 1);
1561 screen_write_cursormove(&ctx, data->cx, data->cy);
1562 window_copy_update_selection(wp);
1563 screen_write_stop(&ctx);
1564 }
1565
1566 void
1567 window_copy_rectangle_toggle(struct window_pane *wp)
1568 {
1569 struct window_copy_mode_data *data = wp->modedata;
1570 u_int px, py;
1571
1572 data->rectflag = !data->rectflag;
1573
1574 py = screen_hsize(&wp->base) + data->cy - data->oy;
1575 px = window_copy_find_length(wp, py);
1576 if (data->cx > px)
1577 window_copy_update_cursor(wp, px, data->cy);
1578
1579 window_copy_update_selection(wp);
1580 window_copy_redraw_screen(wp);
1581 }