X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fnet%2Fclientsocket.cpp;fp=source%2Fnet%2Fclientsocket.cpp;h=d28d0e90b930487b4a573cb751356baf10575666;hb=30f362b6a1257aa0a01a926de8b774fe90d4be39;hp=80303de7ea9a136aa61ce54337effb80e049d934;hpb=5e168bbd5b545732e10513aaed59412e934efa5b;p=libs%2Fnet.git

diff --git a/source/net/clientsocket.cpp b/source/net/clientsocket.cpp
index 80303de..d28d0e9 100644
--- a/source/net/clientsocket.cpp
+++ b/source/net/clientsocket.cpp
@@ -32,6 +32,32 @@ ClientSocket::~ClientSocket()
 	delete peer_addr;
 }
 
+void ClientSocket::shutdown(IO::Mode m)
+{
+	int how;
+	m = m&IO::M_RDWR;
+#ifdef WIN32
+	if(m==IO::M_READ)
+		how = SD_RECEIVE;
+	else if(m==IO::M_WRITE)
+		how = SD_SEND;
+	else if(m==IO::M_RDWR)
+		how = SD_BOTH;
+#else
+	if(m==IO::M_READ)
+		how = SHUT_RD;
+	else if(m==IO::M_WRITE)
+		how = SHUT_WR;
+	else if(m==IO::M_RDWR)
+		how = SHUT_RDWR;
+#endif
+	else
+		return;
+
+	::shutdown(priv->handle, how);
+	mode = mode&~m;
+}
+
 const SockAddr &ClientSocket::get_peer_address() const
 {
 	if(peer_addr==0)
@@ -41,6 +67,7 @@ const SockAddr &ClientSocket::get_peer_address() const
 
 unsigned ClientSocket::do_write(const char *buf, unsigned size)
 {
+	check_access(IO::M_WRITE);
 	if(!connected)
 		throw bad_socket_state("not connected");
 
@@ -61,6 +88,7 @@ unsigned ClientSocket::do_write(const char *buf, unsigned size)
 
 unsigned ClientSocket::do_read(char *buf, unsigned size)
 {
+	check_access(IO::M_READ);
 	if(!connected)
 		throw bad_socket_state("not connected");