2 common system utilities
4 Copyright (C) Amitay Isaacs 2014
5 Copyright (C) Martin Schwenke 2014
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "system/filesys.h"
23 #include "system/shmem.h"
24 #include "system/network.h"
29 #include "lib/util/debug.h"
31 #include "protocol/protocol.h"
33 #include "common/logging.h"
34 #include "common/system.h"
44 #include "lib/util/mkdir_p.h"
47 if possible, make this task real time
49 bool set_scheduler(void)
52 #if HAVE_THREAD_SETSCHED
53 struct thrdentry64 te;
57 if (getthrds64(getpid(), &te, sizeof(te), &ti, 1) != 1) {
58 DEBUG(DEBUG_ERR, ("Unable to get thread information\n"));
62 if (thread_setsched(te.ti_tid, 0, SCHED_RR) == -1) {
63 DEBUG(DEBUG_ERR, ("Unable to set scheduler to SCHED_RR (%s)\n",
71 #if HAVE_SCHED_SETSCHEDULER
76 if (sched_setscheduler(0, SCHED_FIFO, &p) == -1) {
77 DEBUG(DEBUG_CRIT,("Unable to set scheduler to SCHED_FIFO (%s)\n",
85 DEBUG(DEBUG_CRIT,("No way to set real-time priority.\n"));
90 reset scheduler from real-time to normal scheduling
92 void reset_scheduler(void)
95 #if HAVE_THREAD_SETSCHED
96 struct thrdentry64 te;
100 if (getthrds64(getpid(), &te, sizeof(te), &ti, 1) != 1) {
101 DEBUG(DEBUG_ERR, ("Unable to get thread information\n"));
103 if (thread_setsched(te.ti_tid, 0, SCHED_OTHER) == -1) {
104 DEBUG(DEBUG_ERR, ("Unable to set scheduler to SCHED_OTHER\n"));
108 #if HAVE_SCHED_SETSCHEDULER
109 struct sched_param p;
111 p.sched_priority = 0;
112 if (sched_setscheduler(0, SCHED_OTHER, &p) == -1) {
113 DEBUG(DEBUG_ERR, ("Unable to set scheduler to SCHED_OTHER\n"));
119 /* we don't lock future pages here; it would increase the chance that
120 * we'd fail to mmap later on. */
121 void lockdown_memory(bool valgrinding)
123 #if defined(HAVE_MLOCKALL) && !defined(_AIX_)
124 /* Extra stack, please! */
126 memset(dummy, 0, sizeof(dummy));
132 /* Ignore when running in local daemons mode */
137 /* Avoid compiler optimizing out dummy. */
138 mlock(dummy, sizeof(dummy));
139 if (mlockall(MCL_CURRENT) != 0) {
140 DEBUG(DEBUG_WARNING,("Failed to lockdown memory: %s'\n",
146 void mkdir_p_or_die(const char *dir, int mode)
150 ret = mkdir_p(dir, mode);
153 ("ctdb exiting with error: "
154 "failed to create directory \"%s\" (%s)\n",
155 dir, strerror(errno)));
160 void ctdb_wait_for_process_to_exit(pid_t pid)
162 while (kill(pid, 0) == 0 || errno != ESRCH) {
167 #ifdef HAVE_AF_PACKET
169 bool ctdb_sys_check_iface_exists(const char *iface)
174 s = socket(AF_PACKET, SOCK_RAW, 0);
176 /* We don't know if the interface exists, so assume yes */
177 DBG_ERR("Failed to open raw socket\n");
181 strlcpy(ifr.ifr_name, iface, sizeof(ifr.ifr_name));
182 if (ioctl(s, SIOCGIFINDEX, &ifr) < 0 && errno == ENODEV) {
183 DBG_ERR("Interface '%s' not found\n", iface);
192 #else /* HAVE_AF_PACKET */
194 bool ctdb_sys_check_iface_exists(const char *iface)
196 /* Not implemented: Interface always considered present */
200 #endif /* HAVE_AF_PACKET */
204 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
207 socklen_t crl = sizeof(struct ucred);
210 ret = getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &cr, &crl);
219 #else /* HAVE_PEERCRED */
223 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
225 struct peercred_struct cr;
226 socklen_t crl = sizeof(struct peercred_struct);
229 ret = getsockopt(fd, SOL_SOCKET, SO_PEERID, &cr, &crl);
240 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
242 /* Not implemented */
249 #endif /* HAVE_PEERCRED */