socket.cc

来自「linux 下的源代码分析阅读器 red hat公司新版」· CC 代码 · 共 71 行

// $Header$

#include <stdio.h>
#include <osfcn.h>

#include "Socket.h"
#include "system.h"


Socket::Socket( int arg_is_local, int socket_fd )
	{
	is_local = arg_is_local;

	if ( socket_fd < 0 )
		fd = is_local ? get_local_socket() : get_tcp_socket();
	else
		fd = socket_fd;

	port = 0;
	}

Socket::~Socket()
	{
	close( fd );
	}

void Socket::Gripe( char* msg )
	{
	fprintf( stderr, "Socket error: %s\n", msg );
	perror( "perror value" );
	exit( 1 );
	}

AcceptSocket::AcceptSocket( int is_local, int port_hint, int is_a_hint )
: Socket( is_local )
	{
	int result;

	do
		{
		if ( is_local )
			result = bind_local_socket( fd );
		else
			result = bind_socket( fd, port_hint );

		if ( ! is_a_hint )
			break;
		}
	while ( result == 0 && ++port_hint < 65535 );

	if ( result < 0 )
		Gripe( "couldn't bind AcceptSocket to any port" );

	if ( result == 0 )
		port = 0;
	else
		port = port_hint;
	}


Socket* AcceptSocket::Accept()
	{
	int new_connection = is_local ?
		accept_local_connection( fd ) : accept_connection( fd );

	if ( new_connection < 0 )
		Gripe( "problems with Accept()" );

	return new Socket( is_local, new_connection );
	}