badclient.c

来自「一个很好的unix网络编程框架」· C语言 代码 · 共 51 行

#include "etcp.h"
int main( int argc, char **argv )
{
	SOCKET s;
	fd_set readmask;
	fd_set allreads;
	int rc;
	int len;
	char lin[ 1024 ];
	char lout[ 1024 ];

	INIT();
	s = tcp_client( argv[ optind ], argv[ optind + 1 ] );
	FD_ZERO( &allreads );
	FD_SET( 0, &allreads );
	FD_SET( s, &allreads );
	for ( ;; )
	{
		readmask = allreads;
		rc = select( s + 1, &readmask, NULL, NULL, NULL );
		if ( rc <= 0 )
			error( 1, errno, "bad select return (%d)", rc );
		if ( FD_ISSET( s, &readmask ) )
		{
			rc = recv( s, lin, sizeof( lin ) - 1, 0 );
			if ( rc < 0 )
				error( 1, errno, "recv error" );
			if ( rc == 0 )
				error( 1, 0, "server disconnected\n" );
			lin[ rc ] = '\0';
			if ( fputs( lin, stdout ) )
				error( 1, errno, "fputs failed" );
		}
		if ( FD_ISSET( 0, &readmask ) )
		{
			if ( fgets( lout, sizeof( lout ), stdin ) == NULL )
			{
				if ( shutdown( s, 1 ) )
					error( 1, errno, "shutdown failed" );
			}
			else
			{
				len = strlen( lout );
				rc = send( s, lout, len, 0 );
				if ( rc < 0 )
					error( 1, errno, "send error" );
			}
		}
	}
}