793962c5b8
Users now have to call Close() explicitly. This simplifies using the class, as most users have automatic socket management already, and Steal() had to be used often.
133 lines
2.4 KiB
C++
133 lines
2.4 KiB
C++
/*
|
|
* Copyright (C) 2003-2013 The Music Player Daemon Project
|
|
* http://www.musicpd.org
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include "SocketMonitor.hxx"
|
|
#include "Loop.hxx"
|
|
#include "system/fd_util.h"
|
|
#include "Compiler.h"
|
|
|
|
#include <assert.h>
|
|
|
|
#ifdef WIN32
|
|
#include <winsock2.h>
|
|
#else
|
|
#include <sys/socket.h>
|
|
#endif
|
|
|
|
void
|
|
SocketMonitor::Dispatch(unsigned flags)
|
|
{
|
|
flags &= GetScheduledFlags();
|
|
|
|
if (flags != 0 && !OnSocketReady(flags) && IsDefined())
|
|
Cancel();
|
|
}
|
|
|
|
SocketMonitor::~SocketMonitor()
|
|
{
|
|
if (IsDefined())
|
|
Cancel();
|
|
}
|
|
|
|
void
|
|
SocketMonitor::Open(int _fd)
|
|
{
|
|
assert(fd < 0);
|
|
assert(_fd >= 0);
|
|
|
|
fd = _fd;
|
|
}
|
|
|
|
int
|
|
SocketMonitor::Steal()
|
|
{
|
|
assert(IsDefined());
|
|
|
|
Cancel();
|
|
|
|
int result = fd;
|
|
fd = -1;
|
|
|
|
return result;
|
|
}
|
|
|
|
void
|
|
SocketMonitor::Abandon()
|
|
{
|
|
assert(IsDefined());
|
|
|
|
int old_fd = fd;
|
|
fd = -1;
|
|
loop.Abandon(old_fd, *this);
|
|
}
|
|
|
|
void
|
|
SocketMonitor::Close()
|
|
{
|
|
close_socket(Steal());
|
|
}
|
|
|
|
void
|
|
SocketMonitor::Schedule(unsigned flags)
|
|
{
|
|
assert(IsDefined());
|
|
|
|
if (flags == GetScheduledFlags())
|
|
return;
|
|
|
|
if (scheduled_flags == 0)
|
|
loop.AddFD(fd, flags, *this);
|
|
else if (flags == 0)
|
|
loop.RemoveFD(fd, *this);
|
|
else
|
|
loop.ModifyFD(fd, flags, *this);
|
|
|
|
scheduled_flags = flags;
|
|
}
|
|
|
|
SocketMonitor::ssize_t
|
|
SocketMonitor::Read(void *data, size_t length)
|
|
{
|
|
assert(IsDefined());
|
|
|
|
int flags = 0;
|
|
#ifdef MSG_DONTWAIT
|
|
flags |= MSG_DONTWAIT;
|
|
#endif
|
|
|
|
return recv(Get(), (char *)data, length, flags);
|
|
}
|
|
|
|
SocketMonitor::ssize_t
|
|
SocketMonitor::Write(const void *data, size_t length)
|
|
{
|
|
assert(IsDefined());
|
|
|
|
int flags = 0;
|
|
#ifdef MSG_NOSIGNAL
|
|
flags |= MSG_NOSIGNAL;
|
|
#endif
|
|
#ifdef MSG_DONTWAIT
|
|
flags |= MSG_DONTWAIT;
|
|
#endif
|
|
|
|
return send(Get(), (const char *)data, length, flags);
|
|
}
|