mosh: Fix C++ mosh wrapper color counting

[termux-packages] / packages / mosh / mosh.cc

diff --git a/packages/mosh/mosh.cc b/packages/mosh/mosh.cc
index 9f0352e..3772d0d 100644 (file)
--- a/packages/mosh/mosh.cc
+++ b/packages/mosh/mosh.cc
@@ -315,9 +315,7 @@ int main( int argc, char *argv[] )
     freeaddrinfo( servinfo ); // all done with this structure
 
     int pid = fork();
-    if ( pid == -1 ) {
-      die( "%s: fork: %d", argv[0], errno );
-    }
+    if ( pid == -1 ) die( "%s: fork: %d", argv[0], errno );
     if ( pid == 0 ) {
       cat( sockfd, 1 );
       shutdown( sockfd, 0 );
@@ -340,15 +338,15 @@ int main( int argc, char *argv[] )
 
   string color_invocation = client + " -c";
   FILE *color_file = popen( color_invocation.c_str(), "r" );
-  if ( !color_file ) {
-    die( "%s: popen: %d", argv[0], errno );
-  }
+  if ( !color_file ) die( "%s: popen: %d", argv[0], errno );
   char *buf = NULL;
   size_t buf_sz = 0;
   ssize_t n;
   if ( ( n = getline( &buf, &buf_sz, color_file ) ) < 0 ) {
     die( "%s: Can't count colors: %d", argv[0], errno );
   }
+  // Chomp the trailing newline:
+  if ( n > 0 && buf[n - 1] == '\n' ) n--;
   string colors = string( buf, n );
   pclose( color_file );
 
@@ -389,6 +387,19 @@ int main( int argc, char *argv[] )
       server_args.push_back( "-p" );
       server_args.push_back( port_request );
     }
+
+    for (char const* env_name : {
+        "LANG", "LANGUAGE", "LC_CTYPE", "LC_NUMERIC",
+        "LC_TIME", "LC_COLLATE", "LC_MONETARY", "LC_MESSAGES", "LC_PAPER",
+        "LC_NAME", "LC_ADDRESS", "LC_TELEPHONE", "LC_MEASUREMENT",
+        "LC_IDENTIFICATION", "LC_ALL" }) {
+      char* env_value = getenv(env_name);
+      if (env_value) {
+        server_args.push_back("-l");
+        server_args.push_back(string(env_name) + "=" + env_value);
+      }
+    }
+
     if ( commands ) {
       server_args.insert( server_args.end(), command, command + commands );
     }