summaryrefslogtreecommitdiff
path: root/pinentry-dmenu.c
diff options
context:
space:
mode:
Diffstat (limited to 'pinentry-dmenu.c')
-rw-r--r--pinentry-dmenu.c423
1 files changed, 423 insertions, 0 deletions
diff --git a/pinentry-dmenu.c b/pinentry-dmenu.c
new file mode 100644
index 0000000..b14ae99
--- /dev/null
+++ b/pinentry-dmenu.c
@@ -0,0 +1,423 @@
+#include <signal.h>
+#include <time.h>
+#include <ctype.h>
+#include <locale.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <strings.h>
+#include <string.h>
+#include <unistd.h>
+#include <X11/Xlib.h>
+#include <X11/Xatom.h>
+#include <X11/Xutil.h>
+#include <X11/Xft/Xft.h>
+
+#include "drw.h"
+#include "util.h"
+
+#include "pinentry/pinentry.h"
+#include "pinentry/memory.h"
+
+/* macros */
+#define INTERSECT(x, y, w, h, r) \
+ (MAX(0, MIN((x)+(w),(r).x_org+(r).width) - MAX((x),(r).x_org)) \
+ && MAX(0, MIN((y)+(h),(r).y_org+(r).height) - MAX((y),(r).y_org)))
+#define LENGTH(X) (sizeof X / sizeof X[0])
+#define TEXTNW(X, N) (drw_font_getexts_width(drw->fonts[0], (X), (N)))
+#define TEXTW(X) (drw_text(drw, 0, 0, 0, 0, (X), 0) + drw->fonts[0]->h)
+
+const char *str_OK = "OK\n";
+const char *str_ERRUNPARS = "ERR1337 dunno what to do with it\n";
+const char *str_ERRNOTIMP = "ERR4100 not implemented yet\n";
+
+/* enums */
+enum { SchemeNorm, SchemeSel, SchemeLast }; /* color schemes */
+enum { WinPin, WinConfirm }; /* window modes */
+enum { Ok, NotOk, Cancel }; /* return status */
+
+static char text[2048] = "";
+static int bh, mw, mh;
+static int inputw, promptw;
+static size_t cursor = 0;
+static Atom clip, utf8;
+static Window win;
+static XIC xic;
+static int mon = -1;
+
+static ClrScheme scheme[SchemeLast];
+static Display *dpy;
+static int screen;
+static Window root;
+static Drw *drw;
+static int sw, sh;
+
+static int timed_out;
+
+static int confirmed;
+static int winmode;
+
+pinentry_t pinentry;
+
+#include "config.h"
+
+void
+grabkeyboard(void) {
+ int i;
+
+ /* try to grab keyboard,
+ * we may have to wait for another process to ungrab */
+ for (i = 0; i < 1000; i++) {
+ if (XGrabKeyboard(dpy, DefaultRootWindow(dpy), True, GrabModeAsync,
+ GrabModeAsync, CurrentTime) == GrabSuccess) {
+ return;
+ }
+ usleep(1000);
+ }
+ die("cannot grab keyboard\n");
+}
+
+size_t
+nextrune(int cursor, int inc) {
+ ssize_t n;
+
+ /* return location of next utf8 rune in the given direction (+1 or -1) */
+ for (n = cursor + inc; n + inc >= 0 && (text[n] & 0xc0) == 0x80; n += inc);
+
+ return n;
+}
+
+
+void
+insert(const char *str, ssize_t n) {
+ if (strlen(text) + n > sizeof text - 1) {
+ return;
+ }
+
+ /* move existing text out of the way, insert new text, and update cursor */
+ memmove(&text[cursor + n], &text[cursor], sizeof text - cursor - MAX(n, 0));
+
+ if (n > 0) {
+ memcpy(&text[cursor], str, n);
+ }
+ cursor += n;
+}
+
+void
+drawwin(void) {
+ int curpos;
+ int x = 0, y = 0, bh, w;
+ char *sectext;
+ int i;
+ int seccursor = 0;
+ ssize_t n;
+
+ drw_setscheme(drw, &scheme[SchemeNorm]);
+ drw_rect(drw, 0, 0, mw, mh, True, 1, 1);
+
+ if ((pinentry->description) && *(pinentry->description)) {
+ drw_setscheme(drw, &scheme[SchemeSel]);
+ drw_text(drw, 0, 0, mw, bh, pinentry->description, 0);
+ y += bh;
+ }
+
+ if ((pinentry->prompt) && *(pinentry->prompt)) {
+ drw_setscheme(drw, &scheme[SchemeSel]);
+ drw_text(drw, x, y, promptw, bh, pinentry->prompt, 0);
+ x += promptw;
+ }
+
+ w = inputw;
+ drw_setscheme(drw, &scheme[SchemeNorm]);
+
+ sectext = malloc (sizeof (char) * 2048);
+ sectext[0] = '\0';
+
+ for (i=0; text[i]!='\0'; i=nextrune(i, +1)) {
+ strcat(sectext, secstring);
+ if (i < cursor) {
+ for (n = seccursor + 1; n + 1 >= 0 && (sectext[n] & 0xc0) == 0x80;
+ n ++);
+ seccursor = n;
+ }
+ }
+
+ if (winmode == WinPin) {
+ drw_text(drw, x, y, mw, bh, sectext, 0);
+
+ if ((curpos = TEXTNW(sectext, seccursor) + bh/2 - 2) < w) {
+ drw_rect(drw, x + curpos + 2, y + 2, 1 , bh - 4 , True, 1, 0);
+ }
+ } else {
+ drw_text(drw, x, y, mw, bh, "(y/n)", 0);
+ }
+
+ drw_map(drw, win, 0, 0, mw, mh);
+}
+
+void
+setup(void){
+ int x,y;
+ XSetWindowAttributes swa;
+ XIM xim;
+ scheme[SchemeNorm].bg = drw_clr_create(drw, normbgcolor);
+ scheme[SchemeNorm].fg = drw_clr_create(drw, normfgcolor);
+ scheme[SchemeSel].bg = drw_clr_create(drw, selbgcolor);
+ scheme[SchemeSel].fg = drw_clr_create(drw, selfgcolor);
+ clip = XInternAtom(dpy, "CLIPBOARD", False);
+ utf8 = XInternAtom(dpy, "UTF8_STRING", False);
+ bh = drw->fonts[0]->h + 2;
+ mh = (pinentry->description && *(pinentry->description))? bh * 2 : bh;
+ x = 0;
+ y = topbar ? 0 : sh - mh;
+ mw = sw;
+ promptw = (pinentry->prompt && *(pinentry->prompt)) ? TEXTW(pinentry->prompt) : 0;
+ inputw = mw-promptw;
+ swa.override_redirect = True;
+ swa.background_pixel = scheme[SchemeNorm].bg->pix;
+ swa.event_mask = ExposureMask | KeyPressMask | VisibilityChangeMask;
+ win = XCreateWindow(dpy, root, x, y, mw, mh, 0,
+ DefaultDepth(dpy, screen), CopyFromParent,
+ DefaultVisual(dpy, screen),
+ CWOverrideRedirect | CWBackPixel | CWEventMask, &swa);
+
+ xim = XOpenIM(dpy, NULL, NULL, NULL);
+ xic = XCreateIC(xim, XNInputStyle, XIMPreeditNothing | XIMStatusNothing,
+ XNClientWindow, win, XNFocusWindow, win, NULL);
+
+ XMapRaised(dpy, win);
+ drw_resize(drw, mw, mh);
+
+ drawwin();
+}
+
+void
+cleanup(void) {
+ XUngrabKey(dpy, AnyKey, AnyModifier, root);
+ drw_clr_free(scheme[SchemeNorm].bg);
+ drw_clr_free(scheme[SchemeNorm].fg);
+ drw_clr_free(scheme[SchemeSel].fg);
+ drw_clr_free(scheme[SchemeSel].bg);
+ drw_free(drw);
+ XSync(dpy, False);
+ XCloseDisplay(dpy);
+}
+
+static int
+keypress(XKeyEvent *ev) {
+ char buf[32];
+ int len;
+
+ KeySym ksym = NoSymbol;
+ Status status;
+ len = XmbLookupString(xic, ev, buf, sizeof buf, &ksym, &status);
+
+ if (status == XBufferOverflow) {
+ return 0;
+ }
+
+ if (winmode == WinConfirm) {
+ switch(ksym){
+ case XK_KP_Enter:
+ case XK_Return:
+ case XK_y:
+ confirmed = 1;
+ return 1;
+ break;
+ case XK_n:
+ confirmed = 0;
+ return 1;
+ break;
+ case XK_Escape:
+ pinentry->canceled = 1;
+ confirmed = 0;
+ return 1;
+ break;
+ }
+ } else {
+ switch(ksym){
+ default:
+ if (!iscntrl(*buf)) {
+ insert(buf, len);
+ }
+ break;
+ case XK_Delete:
+ if (text[cursor] == '\0') {
+ return 0;
+ }
+ cursor = nextrune(cursor, +1);
+ /* fallthrough */
+ case XK_BackSpace:
+ if (cursor == 0) {
+ return 0;
+ }
+ insert(NULL, nextrune(cursor, -1) - cursor);
+ break;
+ case XK_Escape:
+ pinentry->canceled = 1;
+ return 1;
+ /*cleanup();
+ exit(1);*/
+ break;
+ case XK_Left:
+ if (cursor > 0) {
+ cursor = nextrune(cursor, -1);
+ }
+ break;
+ case XK_Right:
+ if (text[cursor]!='\0') {
+ cursor = nextrune(cursor, +1);
+ }
+ break;
+ case XK_Return:
+ case XK_KP_Enter:
+ return 1;
+ break;
+ }
+ }
+ drawwin();
+ return 0;
+}
+
+void
+paste(void) {
+ char *p, *q;
+ int di;
+ unsigned long dl;
+ Atom da;
+
+ /* we have been given the current selection, now insert it into input */
+ XGetWindowProperty(dpy, win, utf8, 0, (sizeof text / 4) + 1, False,
+ utf8, &da, &di, &dl, &dl, (unsigned char **)&p);
+ insert(p, (q = strchr(p, '\n')) ? q-p : (ssize_t)strlen(p));
+ XFree(p);
+ drawwin();
+}
+
+void
+run(void) {
+ XEvent ev;
+ while(!XNextEvent(dpy, &ev)) {
+ if (XFilterEvent(&ev, win)) {
+ continue; /* what is this I don't even */
+ }
+ switch(ev.type) {
+ case Expose:
+ if (ev.xexpose.count == 0) {
+ drw_map(drw, win, 0, 0, mw, mh);
+ }
+ break;
+ case KeyPress:
+ if (keypress(&ev.xkey)) {
+ return;
+ }
+ break;
+ case SelectionNotify:
+ if (ev.xselection.property == utf8) {
+ paste();
+ }
+ break;
+ case VisibilityNotify:
+ if (ev.xvisibility.state != VisibilityUnobscured) {
+ XRaiseWindow(dpy, win);
+ }
+ break;
+ }
+ }
+}
+
+void
+promptwin(void) {
+ grabkeyboard();
+ setup();
+ run();
+ cleanup();
+}
+
+static void
+catchsig(int sig) {
+ if (sig == SIGALRM) {
+ timed_out = 1;
+ }
+}
+
+static int
+password (void) {
+ char *buf;
+
+ winmode = WinPin;
+ promptwin();
+
+ if (pinentry->canceled) {
+ return -1;
+ }
+
+ buf = secmem_malloc(strlen(text));
+ strcpy(buf, text);
+ pinentry_setbuffer_use (pinentry, buf, 0);
+
+ return 1;
+}
+
+static int
+confirm(void) {
+ winmode = WinConfirm;
+ confirmed = 0;
+ promptwin();
+
+ return confirmed;
+}
+
+static int
+cmdhandler (pinentry_t recieved_pinentry) {
+ struct sigaction sa;
+
+ text[0]='\0';
+ cursor = 0;
+ pinentry = recieved_pinentry;
+
+ if (!setlocale(LC_CTYPE, "") || !XSupportsLocale()) {
+ fputs("warning: no locale support\n", stderr);
+ }
+ if (!(dpy = XOpenDisplay(pinentry->display))) { /* NULL was here */
+ die("dmenu: cannot open display\n");
+ }
+ screen = DefaultScreen(dpy);
+ root = RootWindow(dpy, screen);
+ sw = DisplayWidth(dpy, screen);
+ sh = DisplayHeight(dpy, screen);
+ drw = drw_create(dpy, screen, root, sw, sh);
+ drw_load_fonts(drw, fonts, LENGTH(fonts));
+ if (!drw->fontcount) {
+ die("No fonts could be loaded.\n");
+ }
+ drw_setscheme(drw, &scheme[SchemeNorm]);
+
+ if (pinentry->timeout) {
+ memset(&sa, 0, sizeof(sa));
+ sa.sa_handler = catchsig;
+ sigaction(SIGALRM, &sa, NULL);
+ alarm(pinentry->timeout);
+ }
+
+ if (pinentry->pin) {
+ return password();
+ } else {
+ return confirm();
+ }
+
+ return -1;
+}
+
+pinentry_cmd_handler_t pinentry_cmd_handler = cmdhandler;
+
+int
+main(int argc, char *argv[]) {
+ pinentry_init("pinentry-dmenu");
+ pinentry_parse_opts(argc, argv);
+
+ if (pinentry_loop()) {
+ return 1;
+ }
+
+ return 0;
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-22 04:37:12 +0000