add a scope guard sort of thing to allow restoring terminal state

1 files changed, 23 insertions, 0 deletions

diff --git a/src/termios_wrapper.c b/src/termios_wrapper.c
index b23ae2d..77d589b 100644
--- a/src/termios_wrapper.c
+++ b/src/termios_wrapper.c
@@ -1,3 +1,4 @@
+#include <stdlib.h>
 #include <termios.h>
 
 /* very simplistic, ignores a lot of the settings that i don't understand,
@@ -63,3 +64,25 @@ int echo(int enabled)
 
     return tcsetattr(0, TCSANOW, &t) == 0;
 }
+
+struct termios *get()
+{
+    struct termios *t;
+
+    t = malloc(sizeof(struct termios));
+    if (tcgetattr(0, t) == -1) {
+        return NULL;
+    }
+
+    return t;
+}
+
+void set(struct termios *t)
+{
+    if (t == NULL) {
+        return;
+    }
+
+    tcsetattr(0, TCSANOW, t);
+    free(t);
+}