nettcptransport.h #1

// -*- mode: C++; tab-width: 4; -*-
// vi:ts=8 sw=4 noexpandtab autoindent

/*
 * Copyright 1995, 1996 Perforce Software.  All rights reserved.
 *
 * This file is part of Perforce - the FAST SCM System.
 * - previously part of nettcp.h
 */

/*
 * nettcptransport.h - TCP driver for NetTransport
 *
 * Classes Defined:
 *
 *    NetTcpTransport - a TCP subclass of NetTransport
 */

class KeepAlive;
class NetTcpSelector;

#define NET_CLOSE_SOCKET(fd) \
    if ( fd >= 0 ) { \
	close(fd); \
	fd = -1; \
    }

class NetTcpTransport : public NetTransport {

    public:
			NetTcpTransport( int t, bool fromClient );
			~NetTcpTransport();

	void		SetupKeepAlives( int t );

	void		Send( const char *buffer, int length, Error *e );
	int		Receive( char *buffer, int length, Error *e );
	void		Close();
	virtual bool	IsAccepted()
			{
			    return isAccepted;
			}
	void		ClientMismatch( Error *e );
	int		SendOrReceive( NetIoPtrs &io, Error *se, Error *re );

	StrPtr	*	GetAddress( int raf_flags )
			{
			    GetAddress( t, raf_flags, myAddr );
			    return &myAddr;
			}

	StrPtr	*	GetPeerAddress( int raf_flags )
			{
			    GetPeerAddress( t, raf_flags, peerAddr );
			    return &peerAddr;
			}

	int		GetPortNum()
			{
			    return GetPortNum( t );
			}

	bool		IsSockIPv6()
			{
			    return IsSockIPv6( t );
			}

	static void	GetAddress( int t, int raf_flags, StrBuf &addr );
	static void	GetPeerAddress( int t, int raf_flags, StrBuf &addr );
	static int	GetPortNum( int t );
	static bool	IsSockIPv6( int t );


	int		IsAlive();

	void		SetBreak( KeepAlive *b ) { breakCallback = b; }

	int		GetSendBuffering();
	int		GetRecvBuffering();
	const NetPortParser &GetPortParser() const;
	void            SetPortParser(const NetPortParser &portParser);

	int		GetFd() { return t; }

    protected:
#ifdef OS_NT
	bool		SetWin32KeepAlives(
			    int		socket,
			    const SOCKOPT_T
			    		ka_idlesecs,
			    const int	ka_intvlsecs);
#endif // OS_NT
	void            CloseSocket();
	int 		Peek( int fd, char *buffer, int length );

	int		t;
	KeepAlive	*breakCallback;
	int		lastRead;    // to avoid server TIME_WAIT
	NetTcpSelector	*selector;
	bool             isAccepted;

    private:
	StrBuf		myAddr;
	StrBuf		peerAddr;
	NetPortParser   portParser;
} ;