X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fsocket.cpp;h=2bfdecc5b07794976ddaab3d5d528081c492d9a7;hb=6489612c788dd57721fc181ff26a981b35668d1e;hp=decdbeb9e2fd663c05a7e07819a30bdd0ca6e4bb;hpb=f59eded7c3e162bbdfc6db424c9badc730017698;p=libs%2Fnet.git diff --git a/source/socket.cpp b/source/socket.cpp index decdbeb..2bfdecc 100644 --- a/source/socket.cpp +++ b/source/socket.cpp @@ -1,19 +1,17 @@ -/* $Id$ - -This file is part of libmspnet -Copyright © 2008 Mikkosoft Productions, Mikko Rasa -Distributed under the LGPL -*/ - #ifndef WIN32 -#include +#include #include #include #endif #include -#include +#include +#include +#include +#include #include +#include "sockaddr_private.h" #include "socket.h" +#include "socket_private.h" namespace { @@ -24,7 +22,7 @@ public: WinSockHelper() { WSADATA wsa_data; - int err=WSAStartup(0x0002, &wsa_data); + int err = WSAStartup(0x0002, &wsa_data); if(err) std::cerr<<"Failed to initialize WinSock: "<handle = p.handle; + + SockAddr::SysAddr sa; + getsockname(priv->handle, reinterpret_cast(&sa.addr), &sa.size); + local_addr = SockAddr::from_sys(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), + local_addr(0) { + priv->handle = socket(family_to_sys(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 Exception(format("Unable to bind: %s", strerror(errno))); +#ifdef WIN32 + closesocket(priv->handle); + CloseHandle(*priv->event); +#else + ::close(priv->handle); +#endif delete local_addr; - local_addr=addr.copy(); + 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(); - - delete local_addr; - local_addr=0; - delete peer_addr; - peer_addr=0; } -const SockAddr &Socket::get_local_address() const +const IO::Handle &Socket::get_event_handle() { - if(local_addr==0) - throw InvalidState("Local address not set"); - return *local_addr; + return priv->event; } -const SockAddr &Socket::get_peer_address() const -{ - if(peer_addr==0) - throw InvalidState("Peer address not set"); - return *peer_addr; -} -Socket::~Socket() +void Socket::bind(const SockAddr &addr) { - close(); -} + SockAddr::SysAddr sa = addr.to_sys(); -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); + int err = ::bind(priv->handle, reinterpret_cast(&sa.addr), sa.size); + if(err==-1) + throw system_error("bind"); -#ifdef WIN32 - event=CreateEvent(0, false, false, 0); -#endif + delete local_addr; + local_addr = addr.copy(); } -Socket::Socket(Family af, int type, int proto): - connected(false), - local_addr(0), - peer_addr(0) +const SockAddr &Socket::get_local_address() const { - handle=socket(af, type, proto); - -#ifdef WIN32 - event=CreateEvent(0, false, false, 0); -#endif + if(local_addr==0) + throw bad_socket_state("not bound"); + return *local_addr; } -void Socket::check_state(bool conn) const +void Socket::set_timeout(const Time::TimeDelta &timeout) { - if(handle==MSP_NET_INVALID_SOCKET_HANDLE) - throw Exception("Socket is closed"); - if(conn && !connected) - throw Exception("Socket is not connected"); +#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(timeout/Time::msec); + set_option(SOL_SOCKET, SO_RCVTIMEO, &msecs, sizeof(DWORD)); + set_option(SOL_SOCKET, SO_SNDTIMEO, &msecs, sizeof(DWORD)); +#endif } -int Socket::get_option(int level, int optname, void *optval, socklen_t *optlen) +int Socket::set_option(int level, int optname, const void *optval, socklen_t optlen) { #ifdef WIN32 - return getsockopt(handle, level, optname, reinterpret_cast(optval), optlen); + return setsockopt(priv->handle, level, optname, reinterpret_cast(optval), optlen); #else - return getsockopt(handle, level, optname, optval, optlen); + return setsockopt(priv->handle, level, optname, optval, optlen); #endif } -unsigned Socket::do_write(const char *buf, unsigned size) +int Socket::get_option(int level, int optname, void *optval, socklen_t *optlen) const { - check_state(true); - - if(size==0) - return 0; - - int ret=::send(handle, buf, size, 0); - if(ret<0) - { - if(errno==EAGAIN) - return 0; - else - throw Exception(format("Writing to socket failed: %s", strerror(errno))); - } - - return ret; -} - -unsigned Socket::do_read(char *buf, unsigned size) -{ - check_state(true); - - if(size==0) - return 0; - - int ret=::recv(handle, buf, size, 0); - if(ret<0) - { - if(errno==EAGAIN) - return 0; - else - throw Exception(format("Reading from socket failed: %s", strerror(errno))); - } - else if(ret==0 && !eof_flag) - { - eof_flag=true; - signal_end_of_file.emit(); - set_events(IO::P_NONE); - } - - return ret; +#ifdef WIN32 + return getsockopt(priv->handle, level, optname, reinterpret_cast(optval), optlen); +#else + return getsockopt(priv->handle, level, optname, optval, optlen); +#endif } } // namespace Net