Authentication in different process
Patch status: needinfo
Patch by Philippe Virouleau
Long description:
Add the possibility to buffer keystrokes when a process is running authentication Fix #1090.
To apply this patch, use:
curl http://cr.i3wm.org/patch/298/raw.patch | git am
b/i3lock.c
15 |
@@ -63,6 +63,18 @@ cairo_surface_t *img = NULL; |
16 |
bool tile = false; |
17 |
bool ignore_empty_password = false; |
18 |
|
19 |
+typedef struct ringbuffer_t { |
20 |
+ int head; |
21 |
+ int tail; |
22 |
+ int size; |
23 |
+ xcb_generic_event_t *buf[512]; |
24 |
+} ringbuffer_t; |
25 |
+/* Size of the ringbuffer */ |
26 |
+const int ringbuffer_max = 512; |
27 |
+ |
28 |
+/* Ring buffer where keystrokes are stored during authentication */ |
29 |
+static ringbuffer_t buffered_events = { 0, 0, 0 }; |
30 |
+ |
31 |
/* isutf, u8_dec © 2005 Jeff Bezanson, public domain */ |
32 |
#define isutf(c) (((c) & 0xC0) != 0x80) |
33 |
|
34 |
@@ -75,6 +87,39 @@ void u8_dec(char *s, int *i) { |
35 |
} |
36 |
|
37 |
/* |
38 |
+ * Add an event to the ring buffer. |
39 |
+ * |
40 |
+ * Return 0 if the event was added, 1 if there was an error (no space available |
41 |
+ * or null buffer). |
42 |
+ * |
43 |
+ */ |
44 |
+static int ringbuffer_add(ringbuffer_t *bufring, xcb_generic_event_t *event) { |
45 |
+ if (bufring && bufring->size < ringbuffer_max) { |
46 |
+ bufring->buf[bufring->head++] = event; |
47 |
+ bufring->head %= ringbuffer_max; |
48 |
+ bufring->size++; |
49 |
+ return 0; |
50 |
+ } |
51 |
+ return 1; |
52 |
+} |
53 |
+ |
54 |
+/* |
55 |
+ * Remove an event from the ring buffer. |
56 |
+ * |
57 |
+ * Return null if there was an error (empty or null buffer). |
58 |
+ * |
59 |
+ */ |
60 |
+static xcb_generic_event_t *ringbuffer_remove(ringbuffer_t *bufring) { |
61 |
+ xcb_generic_event_t *ret_val = NULL; |
62 |
+ if (bufring && bufring->size > 0) { |
63 |
+ ret_val = bufring->buf[bufring->tail++]; |
64 |
+ bufring->tail %= ringbuffer_max; |
65 |
+ bufring->size--; |
66 |
+ } |
67 |
+ return ret_val; |
68 |
+} |
69 |
+ |
70 |
+/* |
71 |
* Loads the XKB keymap from the X11 server and feeds it to xkbcommon. |
72 |
* Necessary so that we can properly let xkbcommon track the keyboard state and |
73 |
* translate keypresses to utf-8. |
74 |
@@ -187,10 +232,14 @@ static void clear_pam_wrong(EV_P_ ev_timer *w, int revents) { |
75 |
clear_pam_wrong_timeout = NULL; |
76 |
} |
77 |
|
78 |
-static void clear_input(void) { |
79 |
+static void reset_input(void) { |
80 |
input_position = 0; |
81 |
clear_password_memory(); |
82 |
password[input_position] = '\0'; |
83 |
+} |
84 |
+ |
85 |
+static void clear_input(void) { |
86 |
+ reset_input(); |
87 |
|
88 |
/* Hide the unlock indicator after a bit if the password buffer is |
89 |
* empty. */ |
90 |
@@ -205,7 +254,6 @@ static void auth_failed(void) { |
91 |
fprintf(stderr, "Authentication failure\n"); |
92 |
|
93 |
pam_state = STATE_PAM_WRONG; |
94 |
- clear_input(); |
95 |
redraw_screen(); |
96 |
|
97 |
/* Clear this state after 2 seconds (unless the user enters another |
98 |
@@ -227,6 +275,9 @@ static void auth_failed(void) { |
99 |
} |
100 |
} |
101 |
|
102 |
+static void handle_key_press(xcb_key_press_event_t *event); |
103 |
+static void handle_key_release(xcb_key_release_event_t *event); |
104 |
+ |
105 |
static void child_cb(EV_P_ ev_child *child_watcher, int revents) { |
106 |
if (child_watcher->rstatus != 0) { |
107 |
DEBUG("Authentication successfull\n"); |
108 |
@@ -235,6 +286,17 @@ static void child_cb(EV_P_ ev_child *child_watcher, int revents) { |
109 |
exit(0); |
110 |
} else { |
111 |
auth_failed(); |
112 |
+ xcb_generic_event_t *event; |
113 |
+ while (pam_state != STATE_PAM_VERIFY |
114 |
+ && (event = ringbuffer_remove(&buffered_events))) { |
115 |
+ /* Strip off the highest bit (set if the event is generated) */ |
116 |
+ int type = (event->response_type & 0x7F); |
117 |
+ if (type == XCB_KEY_PRESS) |
118 |
+ handle_key_press((xcb_key_press_event_t*)event); |
119 |
+ else if (type == XCB_KEY_RELEASE) |
120 |
+ handle_key_release((xcb_key_release_event_t*)event); |
121 |
+ free(event); |
122 |
+ } |
123 |
} |
124 |
ev_child_stop(main_loop, child_watcher); |
125 |
free(child_watcher); |
126 |
@@ -259,7 +321,8 @@ static void input_done(void) { |
127 |
if (cpid == 0) { |
128 |
exit(pam_authenticate(pam_handle, 0) == PAM_SUCCESS); |
129 |
} else if (cpid > 0) { |
130 |
- struct ev_child *child_watcher = calloc(sizeof(struct ev_io), 1); |
131 |
+ reset_input(); |
132 |
+ struct ev_child *child_watcher = calloc(sizeof(struct ev_child), 1); |
133 |
ev_child_init(child_watcher, child_cb, cpid, 0); |
134 |
ev_child_set(child_watcher, cpid, 0); |
135 |
ev_child_start(EV_DEFAULT_ child_watcher); |
136 |
@@ -281,6 +344,11 @@ static void input_done(void) { |
137 |
*/ |
138 |
static void handle_key_release(xcb_key_release_event_t *event) { |
139 |
xkb_state_update_key(xkb_state, event->detail, XKB_KEY_UP); |
140 |
+ |
141 |
+ /* If this was the backspace or escape key we are back at an |
142 |
+ * empty input, so turn off the screen if DPMS is enabled */ |
143 |
+ if (dpms && input_position == 0) |
144 |
+ dpms_turn_off_screen(conn); |
145 |
} |
146 |
|
147 |
static void redraw_timeout(EV_P_ ev_timer *w, int revents) { |
148 |
@@ -518,6 +586,14 @@ static void xcb_check_cb(EV_P_ ev_check *w, int revents) { |
149 |
|
150 |
/* Strip off the highest bit (set if the event is generated) */ |
151 |
int type = (event->response_type & 0x7F); |
152 |
+ if (pam_state == STATE_PAM_VERIFY && |
153 |
+ (type == XCB_KEY_PRESS || type == XCB_KEY_RELEASE)) { |
154 |
+ /* Don’t process KeyPress/KeyRelease events when a verification is |
155 |
+ * in progress. Try to buffer them or drop them if the buffer is full. */ |
156 |
+ if (ringbuffer_add(&buffered_events, event)) |
157 |
+ free(event); |
158 |
+ continue; |
159 |
+ } |
160 |
switch (type) { |
161 |
case XCB_KEY_PRESS: |
162 |
handle_key_press((xcb_key_press_event_t*)event); |
163 |
@@ -525,12 +601,6 @@ static void xcb_check_cb(EV_P_ ev_check *w, int revents) { |
164 |
|
165 |
case XCB_KEY_RELEASE: |
166 |
handle_key_release((xcb_key_release_event_t*)event); |
167 |
- |
168 |
- /* If this was the backspace or escape key we are back at an |
169 |
- * empty input, so turn off the screen if DPMS is enabled */ |
170 |
- if (dpms && input_position == 0) |
171 |
- dpms_turn_off_screen(conn); |
172 |
- |
173 |
break; |
174 |
|
175 |
case XCB_VISIBILITY_NOTIFY: |