swaylock: Grow the password buffer

1 files changed, 9 insertions, 2 deletions

diff --git a/swaylock/main.c b/swaylock/main.c
index b20883af..f2015a05 100644
--- a/swaylock/main.c
+++ b/swaylock/main.c
@@ -36,6 +36,7 @@ void sway_terminate(int exit_code) {
 }
 
 char *password;
+int password_size;
 
 int function_conversation(int num_msg, const struct pam_message **msg,
 		struct pam_response **resp, void *appdata_ptr) {
@@ -101,12 +102,17 @@ void notify_key(enum wl_keyboard_key_state state, xkb_keysym_t sym, uint32_t cod
 			if (verify_password()) {
 				exit(0);
 			}
-			password = malloc(1024); // TODO: Let this grow
+			password_size = 1024;
+			password = malloc(password_size);
 			password[0] = '\0';
 			break;
 		default:
 			{
 				int i = strlen(password);
+				if (i + 1 == password_size) {
+					password_size += 1024;
+					password = realloc(password, password_size);
+				}
 				password[i] = (char)codepoint;
 				password[i + 1] = '\0';
 				break;
@@ -280,7 +286,8 @@ int main(int argc, char **argv) {
 		sway_abort("Unsupported scaling mode: %s", scaling_mode_str);
 	}
 
-	password = malloc(1024); // TODO: Let this grow
+	password_size = 1024;
+	password = malloc(password_size);
 	password[0] = '\0';
 	surfaces = create_list();
 	registry = registry_poll();