Fix empty line on login

mobile-shell · Feb 3, 2023 · e51f81f · e51f81f
1 parent c16108f
commit e51f81f
Showing 1 changed file with 31 additions and 7 deletions.
diff --git a/src/frontend/mosh-server.cc b/src/frontend/mosh-server.cc
@@ -98,6 +98,7 @@
 typedef Network::Transport< Terminal::Complete, Network::UserStream > ServerConnection;
 
 static void serve( int host_fd,
+                   int pipe_fd,
 		   Terminal::Complete &terminal,
 		   ServerConnection &network,
 		   long network_timeout,
@@ -508,6 +509,12 @@ static int run_server( const char *desired_ip, const char *desired_port,
   snprintf( utmp_entry, 64, "mosh [%ld]", static_cast<long int>( getpid() ) );
 
   /* Fork child process */
+  int pipes[2];
+  int success = pipe(pipes);
+  if (success == -1) {
+    perror( "pipe" );
+    exit( 1 );
+  }
   pid_t child = forkpty( &master, NULL, NULL, &window_size );
 
   if ( child == -1 ) {
@@ -517,6 +524,10 @@ static int run_server( const char *desired_ip, const char *desired_port,
 
   if ( child == 0 ) {
     /* child */
+    if ( close( pipes[1] ) < 0 ) {
+      perror( "child write pipe close" );
+      exit( 1 );
+    }
 
     /* reenable signals */
     struct sigaction sa;
@@ -592,8 +603,12 @@ static int run_server( const char *desired_ip, const char *desired_port,
 
     /* Wait for parent to release us. */
     char linebuf[81];
-    if (fgets(linebuf, sizeof linebuf, stdin) == NULL) {
-      err( 1, "parent signal" );
+    // -1, errno == EINTR -- retry
+    // otherwise, give up
+    while ( read( pipes[0], linebuf, sizeof( linebuf ) ) == -1 ) {
+        if ( errno != EINTR ) {
+            err( 1, "parent signal" );
+        }
     }
 
     Crypto::reenable_dumping_core();
@@ -603,8 +618,17 @@ static int run_server( const char *desired_ip, const char *desired_port,
       sleep( 3 );
       exit( 1 );
     }
+
+    if ( close( pipes[0] ) < 0 ) {
+        perror( "child read pipe close" );
+        exit( 1 );
+    }
   } else {
-    /* parent */
+      /* parent */
+      if ( close(pipes[0]) < 0 ) {
+          perror( "parent read pipe close" );
+          exit( 1 );
+      }
 
     /* Drop unnecessary privileges */
 #ifdef HAVE_PLEDGE
@@ -621,7 +645,7 @@ static int run_server( const char *desired_ip, const char *desired_port,
 #endif
 
     try {
-      serve( master, terminal, *network, network_timeout, network_signaled_timeout );
+      serve( master, pipes[1], terminal, *network, network_timeout, network_signaled_timeout );
     } catch ( const Network::NetworkException &e ) {
       fprintf( stderr, "Network exception: %s\n",
 	       e.what() );
@@ -645,7 +669,7 @@ static int run_server( const char *desired_ip, const char *desired_port,
   return 0;
 }
 
-static void serve( int host_fd, Terminal::Complete &terminal, ServerConnection &network, long network_timeout, long network_signaled_timeout )
+static void serve( int host_fd, int pipe_fd, Terminal::Complete &terminal, ServerConnection &network, long network_timeout, long network_signaled_timeout )
 {
   /* scale timeouts */
   const uint64_t network_timeout_ms = static_cast<uint64_t>( network_timeout ) * 1000;
@@ -821,8 +845,8 @@ static void serve( int host_fd, Terminal::Complete &terminal, ServerConnection &
 
 	  /* Tell child to start login session. */
 	  if ( !child_released ) {
-	    if ( swrite( host_fd, "\n", 1 ) < 0) {
-	      err( 1, "child release" );
+	    if ( close( pipe_fd ) < 0 ) {
+	      err( 1, "child release" ); 
 	    }
 	    child_released = true;
 	  }