Close client fd using thread local storage

iCodeIN · May 8, 2017 · 28fe87c · 28fe87c
1 parent 5a41e88
commit 28fe87c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/su_client.c b/su_client.c
@@ -26,6 +26,7 @@
 #define ATTY_ERR    4
 
 struct ucred su_credentials;
+static __thread int client_fd;
 
 static void sighandler(int sig) {
 	restore_stdin();
@@ -52,6 +53,8 @@ static void sighandler(int sig) {
 
 static void sigpipe_handler(int sig) {
 	LOGD("su: Client killed unexpectedly\n");
+	close(client_fd);
+	pthread_exit(NULL);
 }
 
 void su_daemon_receiver(int client) {
@@ -69,6 +72,9 @@ void su_daemon_receiver(int client) {
 		return;
 	} else if (child != 0) {
 
+		// For sighandler to close it
+		client_fd = client;
+
 		// Wait result
 		LOGD("su: wait_result waiting for %d\n", child);
 		int status, code;