icinga2/lib/base/unixsocket.cpp

71 lines
2.6 KiB
C++

/******************************************************************************
* Icinga 2 *
* Copyright (C) 2012-2018 Icinga Development Team (https://www.icinga.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 St, Fifth Floor, Boston, MA 02110-1301, USA. *
******************************************************************************/
#include "base/unixsocket.hpp"
#include "base/exception.hpp"
#ifndef _WIN32
using namespace icinga;
UnixSocket::UnixSocket()
{
int fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (fd < 0) {
BOOST_THROW_EXCEPTION(posix_error()
<< boost::errinfo_api_function("socket")
<< boost::errinfo_errno(errno));
}
SetFD(fd);
}
void UnixSocket::Bind(const String& path)
{
unlink(path.CStr());
sockaddr_un s_un;
memset(&s_un, 0, sizeof(s_un));
s_un.sun_family = AF_UNIX;
strncpy(s_un.sun_path, path.CStr(), sizeof(s_un.sun_path));
s_un.sun_path[sizeof(s_un.sun_path) - 1] = '\0';
if (bind(GetFD(), (sockaddr *)&s_un, SUN_LEN(&s_un)) < 0) {
BOOST_THROW_EXCEPTION(posix_error()
<< boost::errinfo_api_function("bind")
<< boost::errinfo_errno(errno));
}
}
void UnixSocket::Connect(const String& path)
{
sockaddr_un s_un;
memset(&s_un, 0, sizeof(s_un));
s_un.sun_family = AF_UNIX;
strncpy(s_un.sun_path, path.CStr(), sizeof(s_un.sun_path));
s_un.sun_path[sizeof(s_un.sun_path) - 1] = '\0';
if (connect(GetFD(), (sockaddr *)&s_un, SUN_LEN(&s_un)) < 0 && errno != EINPROGRESS) {
BOOST_THROW_EXCEPTION(posix_error()
<< boost::errinfo_api_function("connect")
<< boost::errinfo_errno(errno));
}
}
#endif /* _WIN32 */