1 files changed, 83 insertions, 0 deletions
diff --git a/libktorrent/net/socket.h b/libktorrent/net/socket.h
new file mode 100644
index 0000000..db8953b
--- /dev/null
+++ b/libktorrent/net/socket.h
@@ -0,0 +1,83 @@
+/***************************************************************************
+ *   Copyright (C) 2005 by Joris Guisson                                   *
+ *   joris.guisson@gmail.com                                               *
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ *   This program is distributed in the hope that it will be useful,       *
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+ *   GNU General Public License for more details.                          *
+ *                                                                         *
+ *   You should have received a copy of the GNU General Public License     *
+ *   along with this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.          *
+ ***************************************************************************/
+#ifndef NETSOCKET_H
+#define NETSOCKET_H
+
+#include <util/constants.h>
+#include "address.h"
+
+namespace net
+{
+
+	/**
+		@author Joris Guisson <joris.guisson@gmail.com>
+	*/
+	class Socket
+	{
+	public:
+		enum State
+		{
+			IDLE,
+			CONNECTING,
+			CONNECTED,
+			BOUND,
+			CLOSED
+		};
+		
+		Socket(int fd);
+		Socket(bool tcp);
+		virtual ~Socket();
+		
+		void setNonBlocking();
+		bool connectTo(const Address & addr);
+		/// See if a connectTo was succesfull in non blocking mode
+		bool connectSuccesFull();
+		bool bind(Uint16 port,bool also_listen);
+		int send(const bt::Uint8* buf,int len);
+		int recv(bt::Uint8* buf,int max_len);
+		int sendTo(const bt::Uint8* buf,int size,const Address & addr);
+		int recvFrom(bt::Uint8* buf,int max_size,Address & addr);
+		int accept(Address & a);
+		bool ok() const {return m_fd >= 0;}
+		int fd() const {return m_fd;}
+		bool setTOS(unsigned char type_of_service);
+		const Address & getPeerName() const {return addr;}
+		void close();
+		State state() const {return m_state;}
+		
+		/**
+		 * Set the size of the TCP read buffer.
+		 * @param rbs 
+		 */
+//		void setReadBufferSize(Uint32 rbs);
+		
+		Uint32 bytesAvailable() const;
+	private:
+		void cacheAddress();
+		
+	private:
+		int m_fd;
+		State m_state;
+		Address addr;
+	};
+
+}
+
+#endif