88 lines
3.2 KiB
C++
88 lines
3.2 KiB
C++
// +------------------------------------------------------------------+
|
|
// | ____ _ _ __ __ _ __ |
|
|
// | / ___| |__ ___ ___| | __ | \/ | |/ / |
|
|
// | | | | '_ \ / _ \/ __| |/ / | |\/| | ' / |
|
|
// | | |___| | | | __/ (__| < | | | | . \ |
|
|
// | \____|_| |_|\___|\___|_|\_\___|_| |_|_|\_\ |
|
|
// | |
|
|
// | Copyright Mathias Kettner 2014 mk@mathias-kettner.de |
|
|
// +------------------------------------------------------------------+
|
|
//
|
|
// This file is part of Check_MK.
|
|
// The official homepage is at http://mathias-kettner.de/check_mk.
|
|
//
|
|
// check_mk 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 in version 2. check_mk is distributed
|
|
// in the hope that it will be useful, but WITHOUT ANY WARRANTY; with-
|
|
// out even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
// PARTICULAR PURPOSE. See the GNU General Public License for more de-
|
|
// ails. You should have received a copy of the GNU General Public
|
|
// License along with GNU Make; see the file COPYING. If not, write
|
|
// to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
// Boston, MA 02110-1301 USA.
|
|
|
|
#include "Livestatus.h"
|
|
#include <fcntl.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <sys/un.h>
|
|
#include <unistd.h>
|
|
|
|
void Livestatus::connectUNIX(const char *socket_path) {
|
|
_connection = socket(PF_LOCAL, SOCK_STREAM, 0);
|
|
struct sockaddr_un sockaddr;
|
|
sockaddr.sun_family = AF_UNIX;
|
|
strncpy(sockaddr.sun_path, socket_path, sizeof(sockaddr.sun_path) - 1);
|
|
sockaddr.sun_path[sizeof(sockaddr.sun_path) - 1] = '\0';
|
|
if (0 > connect(_connection, (const struct sockaddr *)&sockaddr,
|
|
sizeof(sockaddr))) {
|
|
close(_connection);
|
|
_connection = -1;
|
|
} else
|
|
_file = fdopen(_connection, "r");
|
|
}
|
|
|
|
Livestatus::~Livestatus() { disconnect(); }
|
|
|
|
void Livestatus::disconnect() {
|
|
if (isConnected()) {
|
|
if (_file)
|
|
fclose(_file);
|
|
else
|
|
close(_connection);
|
|
}
|
|
_connection = -1;
|
|
_file = 0;
|
|
}
|
|
|
|
void Livestatus::sendQuery(const char *query) {
|
|
write(_connection, query, strlen(query));
|
|
std::string separators = "Separators: 10 1 2 3\n";
|
|
write(_connection, separators.c_str(), separators.size());
|
|
shutdown(_connection, SHUT_WR);
|
|
}
|
|
|
|
std::vector<std::string> *Livestatus::nextRow() {
|
|
char line[65536];
|
|
if (0 != fgets(line, sizeof(line), _file)) {
|
|
// strip trailing linefeed
|
|
char *end = strlen(line) + line;
|
|
if (end > line && *(end - 1) == '\n') {
|
|
*(end - 1) = 0;
|
|
--end;
|
|
}
|
|
std::vector<std::string> *row = new std::vector<std::string>;
|
|
char *scan = line;
|
|
while (scan < end) {
|
|
char *zero = scan;
|
|
while (zero < end && *zero != '\001') zero++;
|
|
*zero = 0;
|
|
row->push_back(std::string(scan));
|
|
scan = zero + 1;
|
|
}
|
|
return row;
|
|
} else
|
|
return 0;
|
|
}
|