-/* $Id$
-
-This file is part of libmspnet
-Copyright © 2008 Mikkosoft Productions, Mikko Rasa
-Distributed under the LGPL
-*/
-
#ifndef WIN32
-#include <errno.h>
+#include <cerrno>
#include <fcntl.h>
#include <sys/socket.h>
#endif
#include <iostream>
-#include <msp/strings/formatter.h>
+#include <msp/core/systemerror.h>
+#include <msp/io/handle_private.h>
+#include <msp/strings/format.h>
+#include <msp/time/rawtime_private.h>
#include <msp/time/units.h>
#include "socket.h"
+#include "socket_private.h"
namespace {
WinSockHelper()
{
WSADATA wsa_data;
- int err=WSAStartup(0x0002, &wsa_data);
+ int err = WSAStartup(0x0002, &wsa_data);
if(err)
std::cerr<<"Failed to initialize WinSock: "<<err<<'\n';
}
namespace Msp {
namespace Net {
-void Socket::set_block(bool b)
+Socket::Socket(const Private &p, const SockAddr &paddr):
+ priv(new Private),
+ connected(true),
+ local_addr(0),
+ peer_addr(paddr.copy())
{
- mode=(mode&~IO::M_NONBLOCK);
- if(b)
- mode=(mode|IO::M_NONBLOCK);
+ priv->handle = p.handle;
+
+ sockaddr_storage sa;
+ socklen_t size = sizeof(sockaddr_storage);
+ getsockname(priv->handle, reinterpret_cast<sockaddr *>(&sa), &size);
+ local_addr = SockAddr::create(sa);
#ifdef WIN32
- u_long flag=!b;
- ioctlsocket(handle, FIONBIO, &flag);
+ *priv->event = CreateEvent(0, false, false, 0);
#else
- int flags=fcntl(handle, F_GETFL);
- fcntl(handle, F_SETFL, (flags&O_NONBLOCK)|(b?0:O_NONBLOCK));
+ *priv->event = priv->handle;
#endif
}
-IO::Handle Socket::get_event_handle()
+Socket::Socket(Family af, int type, int proto):
+ priv(new Private),
+ connected(false),
+ local_addr(0),
+ peer_addr(0)
{
+ priv->handle = socket(af, type, proto);
+
#ifdef WIN32
- return event;
+ *priv->event = CreateEvent(0, false, false, 0);
#else
- return handle;
+ *priv->event = priv->handle;
#endif
}
-
-void Socket::bind(const SockAddr &addr)
+Socket::~Socket()
{
- check_state(false);
-
- sockaddr sa;
- unsigned size=addr.fill_sockaddr(sa);
-
- int err=::bind(handle, &sa, size);
- if(err==-1)
- throw SystemError("Unable to bind", errno);
+ signal_flush_required.emit();
+#ifdef WIN32
+ closesocket(priv->handle);
+ CloseHandle(*priv->event);
+#else
+ ::close(priv->handle);
+#endif
delete local_addr;
- local_addr=addr.copy();
+ delete peer_addr;
+ delete priv;
}
-/**
-Closes the socket. Most operations on the socket will return an error after
-this.
-*/
-void Socket::close()
+void Socket::set_block(bool b)
{
- if(handle==MSP_NET_INVALID_SOCKET_HANDLE)
- return;
-
- set_events(IO::P_NONE);
+ mode = (mode&~IO::M_NONBLOCK);
+ if(b)
+ mode = (mode|IO::M_NONBLOCK);
- signal_flush_required.emit();
#ifdef WIN32
- closesocket(handle);
- CloseHandle(event);
+ u_long flag = !b;
+ ioctlsocket(priv->handle, FIONBIO, &flag);
#else
- ::close(handle);
+ int flags = fcntl(priv->handle, F_GETFL);
+ fcntl(priv->handle, F_SETFL, (flags&O_NONBLOCK)|(b?0:O_NONBLOCK));
#endif
- handle=MSP_NET_INVALID_SOCKET_HANDLE;
- signal_closed.emit();
+}
+
+const IO::Handle &Socket::get_event_handle()
+{
+ return priv->event;
+}
+
+
+void Socket::bind(const SockAddr &addr)
+{
+ sockaddr_storage sa;
+ unsigned size = addr.fill_sockaddr(sa);
+
+ int err = ::bind(priv->handle, reinterpret_cast<sockaddr *>(&sa), size);
+ if(err==-1)
+ throw system_error("bind");
delete local_addr;
- local_addr=0;
- delete peer_addr;
- peer_addr=0;
+ local_addr = addr.copy();
+}
+
+void Socket::set_timeout(const Time::TimeDelta &timeout)
+{
+#ifndef WIN32
+ timeval tv = Time::rawtime_to_timeval(timeout.raw());
+ set_option(SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(timeval));
+ set_option(SOL_SOCKET, SO_SNDTIMEO, &tv, sizeof(timeval));
+#else
+ DWORD msecs = static_cast<DWORD>(timeout/Time::msec);
+ set_option(SOL_SOCKET, SO_RCVTIMEO, &msecs, sizeof(DWORD));
+ set_option(SOL_SOCKET, SO_SNDTIMEO, &msecs, sizeof(DWORD));
+#endif
}
const SockAddr &Socket::get_local_address() const
{
if(local_addr==0)
- throw InvalidState("Local address not set");
+ throw bad_socket_state("not bound");
return *local_addr;
}
const SockAddr &Socket::get_peer_address() const
{
if(peer_addr==0)
- throw InvalidState("Peer address not set");
+ throw bad_socket_state("not connected");
return *peer_addr;
}
-Socket::~Socket()
-{
- close();
-}
-
-Socket::Socket(SocketHandle h, const SockAddr &paddr):
- handle(h),
- connected(true),
- local_addr(0),
- peer_addr(paddr.copy())
-{
- sockaddr sa;
- socklen_t size=sizeof(sockaddr);
- getsockname(handle, &sa, &size);
- local_addr=SockAddr::create(sa);
-
-#ifdef WIN32
- event=CreateEvent(0, false, false, 0);
-#endif
-}
-
-Socket::Socket(Family af, int type, int proto):
- connected(false),
- local_addr(0),
- peer_addr(0)
+int Socket::set_option(int level, int optname, const void *optval, socklen_t optlen)
{
- handle=socket(af, type, proto);
-
#ifdef WIN32
- event=CreateEvent(0, false, false, 0);
+ return setsockopt(priv->handle, level, optname, reinterpret_cast<const char *>(optval), optlen);
+#else
+ return setsockopt(priv->handle, level, optname, optval, optlen);
#endif
}
-void Socket::check_state(bool conn) const
-{
- if(handle==MSP_NET_INVALID_SOCKET_HANDLE)
- throw Exception("Socket is closed");
- if(conn && !connected)
- throw Exception("Socket is not connected");
-}
-
-int Socket::get_option(int level, int optname, void *optval, socklen_t *optlen)
+int Socket::get_option(int level, int optname, void *optval, socklen_t *optlen) const
{
#ifdef WIN32
- return getsockopt(handle, level, optname, reinterpret_cast<char *>(optval), optlen);
+ return getsockopt(priv->handle, level, optname, reinterpret_cast<char *>(optval), optlen);
#else
- return getsockopt(handle, level, optname, optval, optlen);
+ return getsockopt(priv->handle, level, optname, optval, optlen);
#endif
}
unsigned Socket::do_write(const char *buf, unsigned size)
{
- check_state(true);
+ if(!connected)
+ throw bad_socket_state("not connected");
if(size==0)
return 0;
- int ret=::send(handle, buf, size, 0);
+ int ret = ::send(priv->handle, buf, size, 0);
if(ret<0)
{
if(errno==EAGAIN)
return 0;
else
- throw SystemError("Writing to socket failed", errno);
+ throw system_error("send");
}
return ret;
unsigned Socket::do_read(char *buf, unsigned size)
{
- check_state(true);
+ if(!connected)
+ throw bad_socket_state("not connected");
+
if(size==0)
return 0;
- int ret=::recv(handle, buf, size, 0);
+ int ret = ::recv(priv->handle, buf, size, 0);
if(ret<0)
{
if(errno==EAGAIN)
return 0;
else
- throw SystemError("Reading from socket failed", errno);
+ throw system_error("recv");
}
else if(ret==0 && !eof_flag)
{
- eof_flag=true;
+ eof_flag = true;
signal_end_of_file.emit();
set_events(IO::P_NONE);
}
projects / libs / net.git / blobdiff