JasonWoof Got questions, comments, patches, etc.? Contact Jason Woofenden
fix segfault when selecting big buffers
[st.git] / st.c
diff --git a/st.c b/st.c
index fb1172a..c2f294a 100644 (file)
--- a/st.c
+++ b/st.c
@@ -125,6 +125,14 @@ typedef struct {
        GC gc;
 } DC;
 
+typedef struct {
+       int mode;
+       int bx, by;
+       int ex, ey;
+       int b[2], e[2];
+       char *clip;
+} Selection;
+
 #include "config.h"
 
 static void die(const char *errstr, ...);
@@ -177,6 +185,9 @@ static char* kmap(KeySym);
 static void kpress(XEvent *);
 static void resize(XEvent *);
 static void focus(XEvent *);
+static void brelease(XEvent *);
+static void bpress(XEvent *);
+static void bmotion(XEvent *);
 
 
 static void (*handler[LASTEvent])(XEvent *) = {
@@ -185,6 +196,9 @@ static void (*handler[LASTEvent])(XEvent *) = {
        [ConfigureNotify] = resize,
        [FocusIn] = focus,
        [FocusOut] = focus,
+       [MotionNotify] = bmotion,
+       [ButtonPress] = bpress,
+       [ButtonRelease] = brelease,
 };
 
 /* Globals */
@@ -194,6 +208,94 @@ static Term term;
 static CSIEscape escseq;
 static int cmdfd;
 static pid_t pid;
+static Selection sel;
+
+void
+selinit(void) {
+       sel.mode = 0;
+       sel.bx = -1;
+       sel.clip = NULL;
+}
+
+static inline int selected(int x, int y) {
+       if ((sel.ey==y && sel.by==y)) {
+               int bx = MIN(sel.bx, sel.ex);
+               int ex = MAX(sel.bx, sel.ex);
+               return (x>=bx && x<=ex);
+       }
+       return (((y>sel.b[1] && y<sel.e[1]) || (y==sel.e[1] && x<=sel.e[0])) || \
+               (y==sel.b[1] && x>=sel.b[0] && (x<=sel.e[0] || sel.b[1]!=sel.e[1])));
+}
+
+static void getbuttoninfo(XEvent *e, int *b, int *x, int *y) {
+       if(b) *b = e->xbutton.state,
+               *b=*b==4096?5:*b==2048?4:*b==1024?3:*b==512?2:*b==256?1:-1;
+       *x = e->xbutton.x/xw.cw;
+       *y = e->xbutton.y/xw.ch;
+       sel.b[0] = sel.by<sel.ey?sel.bx:sel.ex;
+       sel.b[1] = MIN(sel.by, sel.ey);
+       sel.e[0] = sel.by<sel.ey?sel.ex:sel.bx;
+       sel.e[1] = MAX(sel.by, sel.ey);
+}
+
+static void bpress(XEvent *e) {
+       sel.mode = 1;
+       sel.ex = sel.bx = e->xbutton.x/xw.cw;
+       sel.ey = sel.by = e->xbutton.y/xw.ch;
+}
+
+static char *getseltext() {
+       char *str, *ptr;
+       int ls, x, y, sz;
+       if(sel.bx==-1)
+               return NULL;
+       sz = ((term.col+1) * (sel.e[1]-sel.b[1]+1));
+       ptr = str = malloc (sz);
+       for(y = 0; y < term.row; y++) {
+               for(x = 0; x < term.col; x++) {
+                       if(term.line[y][x].state & GLYPH_SET && (ls=selected(x, y)))
+                               *ptr = term.line[y][x].c, ptr++;
+               }
+               if (ls)
+                       *ptr = '\n', ptr++;
+       }
+       *ptr = 0;
+       return str;
+}
+
+/* TODO: use X11 clipboard */
+static void selcopy(char *str) {
+       free(sel.clip);
+       sel.clip = str;
+}
+
+static void selpaste() {
+       if(sel.clip)
+               ttywrite(sel.clip, strlen(sel.clip));
+}
+
+/* TODO: doubleclick to select word */
+static void brelease(XEvent *e) {
+       int b;
+       sel.mode = 0;
+       getbuttoninfo(e, &b, &sel.ex, &sel.ey);
+       if(sel.bx==sel.ex && sel.by==sel.ey) {
+               sel.bx = -1;
+               if(b==2)
+                       selpaste();
+       } else {
+               if(b==1)
+                       selcopy(getseltext());
+       }
+       draw(1);
+}
+
+static void bmotion(XEvent *e) {
+       if (sel.mode) {
+               getbuttoninfo(e, NULL, &sel.ex, &sel.ey);
+               draw(1);
+       }
+}
 
 #ifdef DEBUG
 void
@@ -228,7 +330,7 @@ die(const char *errstr, ...) {
 void
 execsh(void) {
        char *args[3] = {getenv("SHELL"), "-i", NULL};
-       DEFAULT(args[0], "/bin/sh"); /* if getenv() failed */
+       DEFAULT(args[0], SHELL); /* if getenv() failed */
        putenv("TERM=" TNAME);
        execvp(args[0], args);
 }
@@ -681,7 +783,6 @@ csihandle(void) {
                case 2: /* all */
                        tclearregion(0, 0, term.col-1, term.row-1);
                        break;
-               case 3: /* XXX: erase saved lines (xterm) */
                default:
                        goto unknown;
                }
@@ -725,22 +826,17 @@ csihandle(void) {
                        case 25:
                                term.c.state |= CURSOR_HIDE;
                                break;
+                       case 1049: /* = 1047 and 1048 */
                        case 1047:
                                if(IS_SET(MODE_ALTSCREEN)) {
                                        tclearregion(0, 0, term.col-1, term.row-1);
                                        tswapscreen();
                                }
-                               break;
+                               if(escseq.arg[0] == 1047)
+                                       break;
                        case 1048:
                                tcursor(CURSOR_LOAD);
                                break;
-                       case 1049:
-                               tcursor(CURSOR_LOAD);
-                               if(IS_SET(MODE_ALTSCREEN)) {
-                                       tclearregion(0, 0, term.col-1, term.row-1);
-                                       tswapscreen();
-                               }
-                               break;
                        default:
                                goto unknown;
                        }
@@ -785,22 +881,17 @@ csihandle(void) {
                        case 25:
                                term.c.state &= ~CURSOR_HIDE;
                                break;
+                       case 1049: /* = 1047 and 1048 */
                        case 1047:
                                if(IS_SET(MODE_ALTSCREEN))
                                        tclearregion(0, 0, term.col-1, term.row-1);
                                else
                                        tswapscreen();
-                               break;                          
+                               if(escseq.arg[0] == 1047)
+                                       break;
                        case 1048:
                                tcursor(CURSOR_SAVE);
                                break;
-                       case 1049:
-                               tcursor(CURSOR_SAVE);
-                               if(IS_SET(MODE_ALTSCREEN))
-                                       tclearregion(0, 0, term.col-1, term.row-1);
-                               else
-                                       tswapscreen();
-                               break;
                        default: goto unknown;
                        }
                } else {
@@ -932,7 +1023,7 @@ tputc(char c) {
                                term.mode &= ~MODE_APPKEYPAD;
                                term.esc = 0;
                                break;
-                       case '7': /* DECSC -- Save Cursor*/
+                       case '7': /* DECSC -- Save Cursor */
                                tcursor(CURSOR_SAVE);
                                term.esc = 0;
                                break;
@@ -1003,7 +1094,7 @@ tresize(int col, int row) {
 
        /* resize to new height */
        term.line = realloc(term.line, row * sizeof(Line));
-       term.alt = realloc(term.alt,  row * sizeof(Line));
+       term.alt  = realloc(term.alt,  row * sizeof(Line));
 
        /* resize each row to new width, zero-pad if needed */
        for(i = 0; i < minrow; i++) {
@@ -1121,6 +1212,12 @@ xinit(void) {
        xw.buf = XCreatePixmap(xw.dis, xw.win, xw.bufw, xw.bufh, XDefaultDepth(xw.dis, xw.scr));
        /* gc */
        dc.gc = XCreateGC(xw.dis, xw.win, 0, NULL);
+       
+       /* event mask */
+       XSelectInput(xw.dis, xw.win, ExposureMask | KeyPressMask
+               | StructureNotifyMask | FocusChangeMask | PointerMotionMask
+               | ButtonPressMask | ButtonReleaseMask);
+       
        XMapWindow(xw.dis, xw.win);
        xhints();
        XStoreName(xw.dis, xw.win, "st");
@@ -1210,7 +1307,7 @@ draw(int redraw_all) {
        int i, x, y, ox;
        Glyph base, new;
        char buf[DRAW_BUF_SIZ];
-       
+
        XSetForeground(xw.dis, dc.gc, dc.col[DefaultBG]);
        XFillRectangle(xw.dis, xw.buf, dc.gc, 0, 0, xw.bufw, xw.bufh);
        for(y = 0; y < term.row; y++) {
@@ -1218,8 +1315,10 @@ draw(int redraw_all) {
                i = ox = 0;
                for(x = 0; x < term.col; x++) {
                        new = term.line[y][x];
+                       if(sel.bx!=-1 && new.c && selected(x, y))
+                               new.mode ^= ATTR_REVERSE;
                        if(i > 0 && (!(new.state & GLYPH_SET) || ATTRCMP(base, new) ||
-                                               i >= DRAW_BUF_SIZ)) {
+                                       i >= DRAW_BUF_SIZ)) {
                                xdraws(buf, base, ox, y, i);
                                i = 0;
                        }
@@ -1302,7 +1401,7 @@ kpress(XEvent *ev) {
                        break;
                case XK_Insert:
                        if(shift)
-                               draw(1), puts("draw!")/* XXX: paste X clipboard */;
+                               selpaste(), draw(1);
                        break;
                default:
                        fprintf(stderr, "errkey: %d\n", (int)ksym);
@@ -1325,6 +1424,8 @@ resize(XEvent *e) {
        row = xw.bufh / xw.ch;
        tresize(col, row);
        ttyresize(col, row);
+       xw.bufh = MAX(1, xw.bufh);
+       xw.bufw = MAX(1, xw.bufw);
        XFreePixmap(xw.dis, xw.buf);
        xw.buf = XCreatePixmap(xw.dis, xw.win, xw.bufw, xw.bufh, XDefaultDepth(xw.dis, xw.scr));
        draw(SCREEN_REDRAW);
@@ -1335,12 +1436,8 @@ run(void) {
        XEvent ev;
        fd_set rfd;
        int xfd = XConnectionNumber(xw.dis);
-       long mask = ExposureMask | KeyPressMask | StructureNotifyMask | FocusChangeMask;
-
-       XSelectInput(xw.dis, xw.win, mask);
-       XResizeWindow(xw.dis, xw.win, xw.w, xw.h); /* XXX: fix resize bug in wmii (?) */
 
-       while(1) {
+       for(;;) {
                FD_ZERO(&rfd);
                FD_SET(cmdfd, &rfd);
                FD_SET(xfd, &rfd);
@@ -1371,6 +1468,7 @@ main(int argc, char *argv[]) {
        tnew(80, 24);
        ttynew();
        xinit();
+       selinit();
        run();
        return 0;
 }