Close server-related file descriptors before executing child process

Closes #170 github issue.
This commit is contained in:
Keith Winstein
2012-04-13 13:48:56 -04:00
parent 0eec0b60f0
commit 7be90ae244
3 changed files with 15 additions and 22 deletions
+8 -3
View File
@@ -227,11 +227,11 @@ int run_server( const char *desired_ip, const char *desired_port,
/* open network */ /* open network */
Network::UserStream blank; Network::UserStream blank;
ServerConnection network( terminal, blank, desired_ip, desired_port ); ServerConnection *network = new ServerConnection( terminal, blank, desired_ip, desired_port );
/* network.set_verbose(); */ /* network.set_verbose(); */
printf( "\nMOSH CONNECT %d %s\n", network.port(), network.get_key().c_str() ); printf( "\nMOSH CONNECT %d %s\n", network->port(), network->get_key().c_str() );
fflush( stdout ); fflush( stdout );
/* don't let signals kill us */ /* don't let signals kill us */
@@ -294,6 +294,9 @@ int run_server( const char *desired_ip, const char *desired_port,
fatal_assert( sigemptyset( &signals_to_block ) == 0 ); fatal_assert( sigemptyset( &signals_to_block ) == 0 );
fatal_assert( sigprocmask( SIG_SETMASK, &signals_to_block, NULL ) == 0 ); fatal_assert( sigprocmask( SIG_SETMASK, &signals_to_block, NULL ) == 0 );
/* close server-related file descriptors */
delete network;
/* set TERM */ /* set TERM */
const char default_term[] = "xterm"; const char default_term[] = "xterm";
const char color_term[] = "xterm-256color"; const char color_term[] = "xterm-256color";
@@ -337,7 +340,7 @@ int run_server( const char *desired_ip, const char *desired_port,
#endif #endif
try { try {
serve( master, terminal, network ); serve( master, terminal, *network );
} catch ( Network::NetworkException e ) { } catch ( Network::NetworkException e ) {
fprintf( stderr, "Network exception: %s: %s\n", fprintf( stderr, "Network exception: %s: %s\n",
e.function.c_str(), strerror( e.the_errno ) ); e.function.c_str(), strerror( e.the_errno ) );
@@ -351,6 +354,8 @@ int run_server( const char *desired_ip, const char *desired_port,
exit( 1 ); exit( 1 );
} }
delete network;
#ifdef HAVE_UTEMPTER #ifdef HAVE_UTEMPTER
utempter_remove_added_record(); utempter_remove_added_record();
#endif #endif
+5 -18
View File
@@ -427,22 +427,9 @@ uint64_t Connection::timeout( void ) const
return RTO; return RTO;
} }
class Socket { Connection::~Connection()
public: {
int fd; if ( close( sock ) < 0 ) {
throw NetworkException( "close", errno );
Socket( int domain, int type, int protocol )
: fd( socket( domain, type, protocol ) )
{
if ( fd < 0 ) {
throw NetworkException( "socket", errno );
}
} }
}
~Socket()
{
if ( close( fd ) < 0 ) {
throw NetworkException( "close", errno );
}
}
};
+1
View File
@@ -107,6 +107,7 @@ namespace Network {
public: public:
Connection( const char *desired_ip, const char *desired_port ); /* server */ Connection( const char *desired_ip, const char *desired_port ); /* server */
Connection( const char *key_str, const char *ip, int port ); /* client */ Connection( const char *key_str, const char *ip, int port ); /* client */
~Connection();
void send( string s ); void send( string s );
string recv( void ); string recv( void );