spnksocket.hpp

来自「linux 下的程序源代码」· HPP 代码 · 共 152 行

/*
 * Copyright 2008 Stephen Liu
 * For license terms, see the file COPYING along with this library.
 */

#ifndef __spnksocket_hpp__
#define __spnksocket_hpp__

#include <sys/socket.h>

class SP_NKSocket {
public:
	static const int DEFAULT_SOCKET_TIMEOUT  = 600;
	static const int DEFAULT_CONNECT_TIMEOUT = 60;

	/**
	 * return > 0 : how many events
	 * return 0 : timeout,
	 * return -1 : error, and errno is set appropriately
	 */
	static int poll( int fd, int events, int * revents, int timeout );

	static void setLogSocketDefault( int logSocket );

private:
	static int mLogSocketDefault;

public:

	virtual ~SP_NKSocket();

	void setSocketTimeout( int socketTimeout );

	void setLogSocket( int logSocket );

	// get the underlying socket file descriptor
	int getSocketFd();

	const char * getPeerHost();

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : recv bytes, include trailing '\n' or '\r\n'
	 * @return 0 : connection close by peer
	 * @param len : readline does not write more than len bytes
	 *       (including the trailing '\0')
	 * @param buffer : excluding the trailing '\n' or '\r\n'
	 */
	int readline( char * buffer, size_t len );

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : recv bytes
	 * @return 0 : connection close by peer
	 */
	int read( void * buffer, size_t len );

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : recv bytes
	 * @return 0 : connection close by peer
	 */
	int readn( void * buffer, size_t len );

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : send bytes
	 */
	int printf( const char * format, ... );

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : send bytes
	 */
	int writen( const void * buffer, size_t len );

	/**
	 * @return -1 : error, and errno is set appropriately
	 * @return > 0 : recv bytes
	 * @return 0 : connection close by peer
	 */
	int probe( void * buffer, size_t len );

	// return internal pending buffer and length
	size_t peek( char ** const buffer = NULL );

	int close();

protected:

	SP_NKSocket();

	void init( int socketFd, int toBeOwner );

	virtual int realRecv( int fd, void * buffer, size_t len ) = 0;
	virtual int realSend( int fd, const void * buffer, size_t len ) = 0;

private:

	int mSocketFd;
	int mToBeOwner;
	char mPeerName[ 64 ];

	char mBuffer[ 8192 ];
	size_t mBufferLen;

	int mSocketTimeout;

	int mLogSocket;
};

class SP_NKTcpSocket : public SP_NKSocket {
public:
	static int openSocket( const char * ip, int port,
		int connectTimeout, const char * bindAddr );

public:
	SP_NKTcpSocket( int socketFd );

	SP_NKTcpSocket( const char * ip, int port,
			int connectTimeout = 0, const char * bindAddr = 0 );

	virtual ~SP_NKTcpSocket();

private:

	static int connectNonblock( int sockFd, struct sockaddr * addr,
		socklen_t addrLen, int connectTimeout );

	virtual int realRecv( int fd, void * buffer, size_t len );
	virtual int realSend( int fd, const void * buffer, size_t len );
};

class SP_NKUdpSocket : public SP_NKSocket {
public:
	SP_NKUdpSocket( int socketFd, struct sockaddr * addr, socklen_t len );

	SP_NKUdpSocket( const char * ip, int port );

	virtual ~SP_NKUdpSocket();

private:
	struct sockaddr * mAddr;
	socklen_t mLen;

	virtual int realRecv( int fd, void * buffer, size_t len );
	virtual int realSend( int fd, const void * buffer, size_t len );
};

#endif