our current experimental Neutrino branch

git-svn-id: file:///home/bas/coolstream_public_svn/THIRDPARTY/applications/neutrino-experimental@27 e54a6e83-5905-42d5-8d5c-058d10e6a962

src/nhttpd/yhttpd_core/ysocket.h

@@ -0,0 +1,98 @@
+//=============================================================================
+// YHTTPD
+// Socket Class : Basic Socket Operations 
+//-----------------------------------------------------------------------------
+// Socket-Handler
+//=============================================================================
+#ifndef __yhttpd_ysocket_h__
+#define __yhttpd_ysocket_h__
+
+// system
+#include <netinet/in.h>
+#include <sys/types.h>
+#include <netinet/tcp.h>
+#include <sys/socket.h>
+
+#include "yconfig.h"
+#include "ytypes_globals.h"
+
+#ifdef Y_CONFIG_USE_OPEN_SSL
+#include <openssl/ssl.h>
+#endif
+//-----------------------------------------------------------------------------
+// Some common socket definitions
+#ifndef SOCKET
+#define SOCKET int
+#endif
+#ifndef INVALID_SOCKET
+#define INVALID_SOCKET (SOCKET)(~0)
+#define SOCKET_ERROR	(-1)
+#endif
+
+//-----------------------------------------------------------------------------
+#define NON_BLOCKING_MAX_RETRIES 10	// Retries, if receive buffer is empty
+#define NON_BLOCKING_TRY_BYTES 	512	// try to catch min bytes .. else sleep
+#define RECEIVE_BLOCK_LEN 	1024	// receiving block length
+#define MAX_LINE_BUFFER 	1024*2	// Max length of one line
+
+//-----------------------------------------------------------------------------
+class CySocket
+{
+public:
+	// constructor & destructor
+			CySocket();
+			CySocket(SOCKET s):sock(s){init();};
+	virtual 	~CySocket(void);
+	void 		init(void);					// some extra initialization
+#ifdef Y_CONFIG_USE_OPEN_SSL
+	bool 			initAsSSL(void);			// initialize this socket as a SSL-socket
+	static bool 		initSSL(void);				// global initialize of SSL-Library, CTX and cerificate/key-file
+	static std::string	SSL_pemfile;				// "server.pem" file with SSL-Certificate (now: only one certificate at all)
+	static std::string	SSL_CA_file;				// CA Certificate (optional)
+#endif
+	// Socket handling
+	bool		handling;					// true: Socket yet is handled by a Connection Thread
+	bool		isOpened;					// is this socket open?
+	bool		isValid;					// false on Socket Errors. Must close.
+	struct timeval 	tv_start_waiting;				// Put keep-alive Socket to Wait-Queue
+	
+	void 		close(void);					// Close Socket
+	void 		shutdown(void);					// Shutdown Socket
+	bool 		listen(int port, int max_connections);		// Listen on Port for max Slave Connections
+	CySocket* 	accept();					// Wait for Connection. Returns "slave" Socket
+	void 		setAddr(sockaddr_in _clientaddr);
+	std::string 	get_client_ip(void);				// Get IP from Client
+	SOCKET 		get_socket(){return sock;}			// Return "C" Socket-ID
+	int		get_accept_port(void);				// Get Port for accepted connection
+
+	// send & receive (basic)
+	int 		Read(char *buffer, unsigned int length);	// Read a buffer (normal or SSL)
+	int 		Send(char const *buffer, unsigned int length);	// Send a buffer (normal or SSL)
+	bool 		CheckSocketOpen();				// check if socket was closed by client
+	
+	// send & receive
+	int 		SendFile(int filed);				// Send a File
+	std::string 	ReceiveBlock();					// receive a Block. Look at length
+	unsigned int 	ReceiveFileGivenLength(int filed, unsigned int _length); // Receive File of given length
+	std::string 	ReceiveLine();					// receive until "\n"
+
+protected:
+	long 		BytesSend;					// Bytes send over Socket
+	bool 		set_option(int typ, int option);		// Set Socket Options
+	void 		set_reuse_port();				// Set Reuse Port Option for Socket
+	void 		set_reuse_addr();				// Set Reuse Address Option for Socket
+	void		set_keep_alive();				// Set Keep-Alive Option for Socket
+	void		set_tcp_nodelay();
+
+#ifdef Y_CONFIG_USE_OPEN_SSL
+	bool		isSSLSocket;					// This is a SSL based Socket
+	static SSL_CTX 	*SSL_ctx;					// Global SSL ctx object
+	SSL		*ssl;						// ssl habdler for this socket
+#endif
+private:
+	std::string	receive_buffer;					// buffer for receive from socket 
+	sockaddr_in	addr;						// "slave" Client Socket Data
+	socklen_t	addr_len;					// Length of addr struct
+	SOCKET		sock;						// "C" Socket-ID
+};
+#endif // __yhttpd_ysocket_h__