You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
143 lines
4.4 KiB
143 lines
4.4 KiB
/* Copyright (c) 2018, 2019, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
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, version 2.0, 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 "sql/sd_notify.h"
|
|
#include "my_config.h"
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
#include <sys/types.h> // connect
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_SOCKET_H
|
|
#include <sys/socket.h> // socket
|
|
#endif
|
|
|
|
#ifdef HAVE_SYS_UN_H
|
|
#include <sys/un.h> // AF_UNIX
|
|
#endif
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
#include <unistd.h> // write
|
|
#endif
|
|
|
|
#include <errno.h> // errno
|
|
#include <string.h> // strcpy
|
|
#include <iostream> // std::cout
|
|
|
|
#include <mysqld_error.h> // error logging
|
|
#include "my_sys.h" // my_strerror
|
|
#include "mysql/components/services/log_builtins.h" // error logging
|
|
#include "scope_guard.h" // scope_guard
|
|
#include "sql/log.h" // error logging
|
|
|
|
namespace sysd {
|
|
/** File descriptor for the systemd notification socket file. */
|
|
int NotifyGlobals::socket = -1;
|
|
|
|
/** Stringstream for formatting notification messages. */
|
|
std::stringstream NotifyGlobals::fmt;
|
|
|
|
/**
|
|
Looks for the name of the socket file in the environment variable
|
|
NOTIFY_SOCKET. Connects NotifyGlobals::socket to it if present.
|
|
*/
|
|
void notify_connect() {
|
|
#ifndef _WIN32
|
|
const char *sockstr = getenv("NOTIFY_SOCKET");
|
|
if (sockstr == nullptr) {
|
|
return;
|
|
}
|
|
size_t sockstrlen = strlen(sockstr);
|
|
size_t sunpathlen = sizeof(sockaddr_un::sun_path) - 1;
|
|
if (sockstrlen > sunpathlen) {
|
|
std::cerr << "NOTIFY_SOCKET too long" << std::endl;
|
|
LogErr(SYSTEM_LEVEL, ER_SYSTEMD_NOTIFY_PATH_TOO_LONG, sockstr, sockstrlen,
|
|
sunpathlen);
|
|
return;
|
|
}
|
|
NotifyGlobals::socket = socket(AF_UNIX, SOCK_DGRAM, 0);
|
|
|
|
sockaddr_un addr;
|
|
memset(&addr, 0, sizeof(sockaddr_un));
|
|
addr.sun_family = AF_UNIX;
|
|
strcpy(addr.sun_path, sockstr);
|
|
int ret = -1;
|
|
do {
|
|
ret =
|
|
connect(NotifyGlobals::socket,
|
|
reinterpret_cast<const sockaddr *>(&addr), sizeof(sockaddr_un));
|
|
} while (ret == -1 && errno == EINTR);
|
|
if (ret == -1) {
|
|
char errbuf[512];
|
|
LogErr(WARNING_LEVEL, ER_SYSTEMD_NOTIFY_CONNECT_FAILED, sockstr,
|
|
my_strerror(errbuf, sizeof(errbuf) - 1, errno));
|
|
NotifyGlobals::socket = -1;
|
|
}
|
|
#endif /* not defined _WIN32 */
|
|
}
|
|
|
|
#ifndef _WIN32
|
|
/**
|
|
Recursion terminator overload for varargs template function. Creates
|
|
a string from the current content of NotifyGlobals::fmt and sends
|
|
string to notification socket.
|
|
*/
|
|
void notify() {
|
|
std::string note = NotifyGlobals::fmt.str();
|
|
const char *src = note.c_str();
|
|
const char *end = src + note.size();
|
|
ssize_t status = -1;
|
|
|
|
auto sg = create_scope_guard([&]() {
|
|
NotifyGlobals::fmt.str(""); // clear the fmt buffer for new notification
|
|
});
|
|
|
|
#ifdef SYSD_DBUG
|
|
std::cout << "Send to systemd notify socket:\n" << note << std::endl;
|
|
if (NotifyGlobals::socket == -1) {
|
|
return;
|
|
}
|
|
#endif /* SYSD_DBUG */
|
|
|
|
while (true) {
|
|
size_t remaining = end - src;
|
|
status = write(NotifyGlobals::socket, src, remaining);
|
|
if (status == -1) {
|
|
if (errno == EINTR) {
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
size_t written = status;
|
|
if (written == remaining) {
|
|
break;
|
|
}
|
|
src += written;
|
|
}
|
|
if (status == -1) {
|
|
char errbuf[512];
|
|
LogErr(WARNING_LEVEL, ER_SYSTEMD_NOTIFY_WRITE_FAILED,
|
|
my_strerror(errbuf, sizeof(errbuf) - 1, errno));
|
|
}
|
|
}
|
|
#endif /* not defined _WIN32 */
|
|
} // namespace sysd
|
|
|