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"
40 #ifdef HAVE_PROCINFO_H
44 #include "lib/util/mkdir_p.h"
47 if possible, make this task real time
49 bool set_scheduler(void)
52 #ifdef 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 #ifdef 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 #ifdef 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 #ifdef 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 ctdb_wait_for_process_to_exit(pid_t pid)
148 while (kill(pid, 0) == 0 || errno != ESRCH) {
153 #ifdef HAVE_AF_PACKET
155 bool ctdb_sys_check_iface_exists(const char *iface)
160 s = socket(AF_PACKET, SOCK_RAW, 0);
162 /* We don't know if the interface exists, so assume yes */
163 DBG_ERR("Failed to open raw socket\n");
167 strlcpy(ifr.ifr_name, iface, sizeof(ifr.ifr_name));
168 if (ioctl(s, SIOCGIFINDEX, &ifr) < 0 && errno == ENODEV) {
169 DBG_ERR("Interface '%s' not found\n", iface);
178 #else /* HAVE_AF_PACKET */
180 bool ctdb_sys_check_iface_exists(const char *iface)
182 /* Not implemented: Interface always considered present */
186 #endif /* HAVE_AF_PACKET */
190 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
193 socklen_t crl = sizeof(struct ucred);
196 ret = getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &cr, &crl);
205 #else /* HAVE_PEERCRED */
209 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
211 struct peercred_struct cr;
212 socklen_t crl = sizeof(struct peercred_struct);
215 ret = getsockopt(fd, SOL_SOCKET, SO_PEERID, &cr, &crl);
226 int ctdb_get_peer_pid(const int fd, pid_t *peer_pid)
228 /* Not implemented */
235 #endif /* HAVE_PEERCRED */