20#include <sys/sysinfo.h>
28#include <sys/sysctl.h>
32#include <QReadWriteLock>
33#include <QNetworkProxy>
41#include <QHostAddress>
42#include <QRegularExpression>
43#include <QRegularExpressionMatchIterator>
61 struct sysinfo sinfo {};
62 if (sysinfo(&sinfo) == -1)
64 LOG(VB_GENERAL, LOG_ERR,
"sysinfo() error");
67 uptime = std::chrono::seconds(sinfo.uptime);
69#elif defined(Q_OS_BSD4)
70 std::array<int,2> mib { CTL_KERN, KERN_BOOTTIME };
71 struct timeval bootTime;
76 len =
sizeof(bootTime);
77 if (sysctl(mib.data(), 2, &bootTime, &len,
nullptr, 0) == -1)
79 LOG(VB_GENERAL, LOG_ERR,
"sysctl() error");
82 uptime = std::chrono::seconds(time(
nullptr) - bootTime.tv_sec);
84 uptime = std::chrono::seconds(::GetTickCount() / 1000);
87 LOG(VB_GENERAL, LOG_NOTICE,
"Unknown platform. How do I get the uptime?");
102 [[maybe_unused]]
int &freeMB,
103 [[maybe_unused]]
int &totalVM,
104 [[maybe_unused]]
int &freeVM)
107 static constexpr size_t MB { 1024LL * 1024 };
108 struct sysinfo sinfo {};
109 if (sysinfo(&sinfo) == -1)
111 LOG(VB_GENERAL, LOG_ERR,
112 "getMemStats(): Error, sysinfo() call failed.");
116 totalMB = (int)((sinfo.totalram * sinfo.mem_unit)/MB);
117 freeMB = (int)((sinfo.freeram * sinfo.mem_unit)/MB);
118 totalVM = (int)((sinfo.totalswap * sinfo.mem_unit)/MB);
119 freeVM = (int)((sinfo.freeswap * sinfo.mem_unit)/MB);
121#elif defined(Q_OS_DARWIN)
123 mach_msg_type_number_t count;
125 vm_statistics_data_t s;
127 mp = mach_host_self();
130 if (host_page_size(mp, &pageSize) != KERN_SUCCESS)
133 count = HOST_VM_INFO_COUNT;
134 if (host_statistics(mp, HOST_VM_INFO,
135 (host_info_t)&s, &count) != KERN_SUCCESS)
137 LOG(VB_GENERAL, LOG_ERR,
"getMemStats(): Error, "
138 "failed to get virtual memory statistics.");
143 totalMB = (s.active_count + s.inactive_count +
144 s.wire_count + s.free_count) * pageSize / 1024;
145 freeMB = s.free_count * pageSize / 1024;
153 totalVM = (int)(fsInfo.getTotalSpace() >> 10);
154 freeVM = (int)(fsInfo.getFreeSpace() >> 10);
170#if !defined(_WIN32) && !defined(Q_OS_ANDROID)
172 if (
getloadavg(loads.data(), loads.size()) != -1)
180 QStringList strlist(QString(
"QUERY_LOAD"));
184 load[0] = strlist[0].toDouble();
185 load[1] = strlist[1].toDouble();
186 load[2] = strlist[2].toDouble();
195 QStringList strlist(QString(
"QUERY_UPTIME"));
200 if (strlist[0].isEmpty() || !strlist[0].at(0).isNumber())
203 if (
sizeof(std::chrono::seconds::rep) ==
sizeof(
long long))
204 uptime = std::chrono::seconds(strlist[0].toLongLong());
205 else if (
sizeof(std::chrono::seconds::rep) ==
sizeof(
long))
206 uptime = std::chrono::seconds(strlist[0].toLong());
207 else if (
sizeof(std::chrono::seconds::rep) ==
sizeof(
int))
208 uptime = std::chrono::seconds(strlist[0].toInt());
215 QStringList strlist(QString(
"QUERY_MEMSTATS"));
219 totalMB = strlist[0].toInt();
220 freeMB = strlist[1].toInt();
221 totalVM = strlist[2].toInt();
222 freeVM = strlist[3].toInt();
246 QString cmd = QString(
"%systemroot%\\system32\\ping.exe -w %1 -n 1 %2>NUL")
247 .arg(
timeout.count()) .arg(host);
254 QHostAddress addr = QHostAddress(addrstr);
255#if defined(__FreeBSD__) || defined(Q_OS_DARWIN)
256 QString timeoutparam(
"-t");
259 QString timeoutparam(
"-w");
262 addr.protocol() == QAbstractSocket::IPv6Protocol ?
"ping6" :
"ping";
263 QString cmd = QString(
"%1 %2 %3 -c 1 %4 >/dev/null 2>&1")
264 .arg(pingcmd, timeoutparam,
265 QString::number(duration_cast<std::chrono::seconds>(
timeout).count()),
276bool telnet(
const QString &host,
int port)
280 bool connected = s->ConnectToHost(host, port);
310 char *buf =
new char[buflen];
317 if (!dst.isWritable() && !dst.isOpen())
319 odst = dst.open(QIODevice::Unbuffered |
320 QIODevice::WriteOnly |
321 QIODevice::Truncate);
324 if (!src.isReadable() && !src.isOpen())
325 osrc = src.open(QIODevice::Unbuffered|QIODevice::ReadOnly);
327 bool ok = dst.isWritable() && src.isReadable();
328 long long total_bytes = 0LL;
332 long long rlen = src.read(buf, buflen);
335 LOG(VB_GENERAL, LOG_ERR,
"read error");
344 while ((rlen-off>0) && ok)
346 long long wlen = dst.write(buf + off, rlen - off);
351 LOG(VB_GENERAL, LOG_ERR,
"write error");
364 return (ok) ? total_bytes : -1LL;
373 char tempfilename[MAX_PATH] =
"";
376 if (GetTempFileNameA(
temppath,
"mth", 0, tempfilename))
381 unlink(tempfilename);
382 ret = mkdir(tempfilename);
385 ret = open(tempfilename, O_CREAT | O_RDWR, S_IREAD | S_IWRITE);
387 QString tmpFileName(tempfilename);
389 QByteArray safe_name_template = name_template.toLatin1();
390 const char *
tmp = safe_name_template.constData();
391 char *ctemplate = strdup(
tmp);
395 ret = (mkdtemp(ctemplate)) ? 0 : -1;
399 mode_t cur_umask = umask(S_IRWXO | S_IRWXG);
400 ret = mkstemp(ctemplate);
404 QString tmpFileName(ctemplate);
410 LOG(VB_GENERAL, LOG_ERR, QString(
"createTempFile(%1), Error ")
411 .arg(name_template) +
ENO);
412 return name_template;
415 if (!dir && (ret >= 0))
441 QByteArray fname =
filename.toLatin1();
442 int ret = chmod(fname.constData(), 0666);
445 LOG(VB_GENERAL, LOG_ERR, QString(
"Unable to change permissions on file. (%1)").arg(
filename));
456 QByteArray
tmp = query.toLocal8Bit();
457 std::cout <<
tmp.constData();
459 tmp = def.toLocal8Bit();
461 std::cout <<
" [" <<
tmp.constData() <<
"] ";
465 if (!isatty(fileno(stdin)) || !isatty(fileno(
stdout)))
467 std::cout << std::endl <<
"[console is not interactive, using default '"
468 <<
tmp.constData() <<
"']" << std::endl;
472 QTextStream stream(stdin);
473 QString qresponse = stream.readLine();
475 if (qresponse.isEmpty())
486 QString str_resp =
getResponse(query, QString(
"%1").arg(def));
487 if (str_resp.isEmpty())
490 int resp = str_resp.toInt(&ok);
491 return (ok ? resp : def);
496 QStringList *intermediaries,
500 LOG(VB_GENERAL, LOG_DEBUG,
501 QString(
"getSymlinkTarget('%1', 0x%2, %3)")
502 .arg(start_file).arg((uint64_t)intermediaries,0,16)
507 QString cur_file = start_file;
508 QFileInfo fi(cur_file);
512 intermediaries->clear();
513 intermediaries->push_back(start_file);
516 for (
uint i = 0; (i <= maxLinks) && fi.isSymLink() &&
517 !(link = fi.symLinkTarget()).isEmpty(); i++)
519 cur_file = (link[0] ==
'/') ?
521 fi.absoluteDir().absolutePath() +
"/" + link;
523 if (intermediaries && !intermediaries->contains(cur_file))
524 intermediaries->push_back(cur_file);
526 fi = QFileInfo(cur_file);
532 for (
uint i = 0; i < intermediaries->size(); i++)
534 LOG(VB_GENERAL, LOG_DEBUG, QString(
" inter%1: %2")
535 .arg(i).arg((*intermediaries)[i]));
539 LOG(VB_GENERAL, LOG_DEBUG,
540 QString(
"getSymlinkTarget() -> '%1'")
541 .arg((!fi.isSymLink()) ? cur_file : QString()));
544 return (!fi.isSymLink()) ? cur_file : QString();
549 QStringList tokens = MAC.split(
':');
550 if (tokens.size() != 6)
552 LOG(VB_NETWORK, LOG_ERR,
553 QString(
"IsMACAddress(%1) = false, doesn't have 6 parts").arg(MAC));
557 for (
int y = 0; y < 6; y++)
559 if (tokens[y].isEmpty())
561 LOG(VB_NETWORK, LOG_ERR,
562 QString(
"IsMACAddress(%1) = false, part #%2 is empty.")
568 int value = tokens[y].toInt(&ok, 16);
571 LOG(VB_NETWORK, LOG_ERR,
572 QString(
"IsMACAddress(%1) = false, unable to "
573 "convert part '%2' to integer.")
574 .arg(MAC, tokens[y]));
580 LOG(VB_NETWORK, LOG_ERR,
581 QString(
"IsMACAddress(%1) = false, part #%2 "
582 "evaluates to %3 which is higher than 255.")
583 .arg(MAC).arg(y).arg(value));
588 LOG(VB_NETWORK, LOG_DEBUG, QString(
"IsMACAddress(%1) = true").arg(MAC));
595 QFileInfo fileinfo(
file);
596 qint64 initialsize = fileinfo.size();
599 if (initialsize == 0)
602 if (
file.open(QIODevice::ReadOnly))
606 LOG(VB_GENERAL, LOG_ERR,
607 "Error: Unable to open selected file, missing read permissions?");
612 QDataStream stream(&
file);
613 stream.setByteOrder(QDataStream::LittleEndian);
614 for (quint64
tmp = 0, i = 0; i < 65536/
sizeof(
tmp); i++)
620 file.seek(initialsize - 65536);
621 for (quint64
tmp = 0, i = 0; i < 65536/
sizeof(
tmp); i++)
629 QString
output = QString(
"%1").arg(hash, 0, 16);
635 std::vector<char> msg(6,
static_cast<char>(0xFF));
636 std::array<char,6> macaddr {};
637 QStringList tokens = MAC.split(
':');
639 if (tokens.size() != 6)
641 LOG(VB_GENERAL, LOG_ERR,
642 QString(
"WakeOnLan(%1): Incorrect MAC length").arg(MAC));
646 for (
int y = 0; y < 6; y++)
649 macaddr[y] = tokens[y].toInt(&ok, 16);
653 LOG(VB_GENERAL, LOG_ERR,
654 QString(
"WakeOnLan(%1): Invalid MAC address").arg(MAC));
660 for (
int x = 0; x < 16; x++)
661 msg.insert(msg.end(), macaddr.cbegin(), macaddr.cend());
663 LOG(VB_NETWORK, LOG_INFO,
664 QString(
"WakeOnLan(): Sending WOL packet to %1").arg(MAC));
666 QUdpSocket udp_socket;
667 qlonglong msglen = msg.size();
668 return udp_socket.writeDatagram(
669 msg.data(), msglen, QHostAddress::Broadcast, 32767) == msglen;
688#if defined(Q_OS_DARWIN) || defined(__FreeBSD__) || defined(__OpenBSD__)
689 const char *command =
"ps -ax | grep -i pulseaudio | grep -v grep > /dev/null";
691 const char *command =
"ps ch -C pulseaudio -o pid > /dev/null";
705 if ((-1 == ret) && (0 != errno) && (val >= 0))
707 LOG(VB_GENERAL, LOG_ERR,
"Failed to nice process" +
ENO);
716#ifdef _POSIX_PRIORITY_SCHEDULING
741#if defined(__linux__) && ( defined(__i386__) || defined(__ppc__) || \
742 defined(__x86_64__) || defined(__ia64__) )
746#include <sys/ptrace.h>
747#include <sys/syscall.h>
748#if __has_include(<linux/ioprio.h>)
753#include <linux/ioprio.h>
756static constexpr int8_t IOPRIO_BITS { 16 };
757static constexpr int8_t IOPRIO_CLASS_SHIFT { 13 };
758static constexpr int IOPRIO_PRIO_MASK { (1UL << IOPRIO_CLASS_SHIFT) - 1 };
759static constexpr int IOPRIO_PRIO_CLASS(
int mask)
760 {
return mask >> IOPRIO_CLASS_SHIFT; };
761static constexpr int IOPRIO_PRIO_DATA(
int mask)
762 {
return mask & IOPRIO_PRIO_MASK; };
763static constexpr int IOPRIO_PRIO_VALUE(
int pclass,
int data)
764 {
return (pclass << IOPRIO_CLASS_SHIFT) | data; };
766enum { IOPRIO_CLASS_NONE,IOPRIO_CLASS_RT,IOPRIO_CLASS_BE,IOPRIO_CLASS_IDLE, };
767enum { IOPRIO_WHO_PROCESS = 1, IOPRIO_WHO_PGRP, IOPRIO_WHO_USER, };
772 int new_ioclass {IOPRIO_CLASS_BE};
774 new_ioclass = IOPRIO_CLASS_RT;
776 new_ioclass = IOPRIO_CLASS_IDLE;
777 int new_iodata = (new_ioclass == IOPRIO_CLASS_BE) ? val : 0;
778 int new_ioprio = IOPRIO_PRIO_VALUE(new_ioclass, new_iodata);
781 int old_ioprio = syscall(SYS_ioprio_get, IOPRIO_WHO_PROCESS, pid);
782 if (old_ioprio == new_ioprio)
785 int ret = syscall(SYS_ioprio_set, IOPRIO_WHO_PROCESS, pid, new_ioprio);
787 if (-1 == ret && EPERM == errno && IOPRIO_CLASS_BE != new_ioclass)
789 new_iodata = (new_ioclass == IOPRIO_CLASS_RT) ? 0 : 7;
790 new_ioprio = IOPRIO_PRIO_VALUE(IOPRIO_CLASS_BE, new_iodata);
791 ret = syscall(SYS_ioprio_set, IOPRIO_WHO_PROCESS, pid, new_ioprio);
808 QFileInfoList entries = aDir.entryInfoList(QDir::NoDotAndDotDot |
809 QDir::Dirs | QDir::Files);
810 int count = entries.size();
811 bool has_err =
false;
813 for (
int idx = 0; idx < count && !has_err; idx++)
815 const QFileInfo& entryInfo = entries[idx];
816 QString path = entryInfo.absoluteFilePath();
817 if (entryInfo.isDir())
830 if (!has_err && !aDir.rmdir(aDir.absolutePath()))
849 QString
LOC =
"setHttpProxy() - ";
852 QString var(qEnvironmentVariable(
"http_proxy"));
854 var = qEnvironmentVariable(
"HTTP_PROXY");
857 if (!var.startsWith(
"http://"))
858 var.prepend(
"http://");
860 QUrl url = QUrl(var, QUrl::TolerantMode);
861 QString host = url.host();
862 int port = url.port();
875 LOG(VB_NETWORK, LOG_INFO,
LOC +
876 QString(
"assuming port %1 on host %2") .arg(port).arg(host));
879 else if (!
ping(host, 1s))
881 LOG(VB_GENERAL, LOG_ERR,
LOC +
882 QString(
"cannot locate host %1").arg(host) +
883 "\n\t\t\tPlease check HTTP_PROXY environment variable!");
885 else if (!
telnet(host,port))
887 LOG(VB_GENERAL, LOG_ERR,
LOC +
888 QString(
"%1:%2 - cannot connect!").arg(host).arg(port) +
889 "\n\t\t\tPlease check HTTP_PROXY environment variable!");
893 LOG(VB_NETWORK, LOG_DEBUG,
LOC + QString(
"using http://%1:%2@%3:%4")
894 .arg(url.userName()).arg(url.password())
895 .arg(host).arg(port));
898 QNetworkProxy(QNetworkProxy::HttpCachingProxy,
899 host, port, url.userName(), url.password());
900 QNetworkProxy::setApplicationProxy(
p);
904 LOG(VB_NETWORK, LOG_DEBUG,
LOC +
"no HTTP_PROXY environment var.");
909 QNetworkProxyQuery query(QUrl(
"http://www.mythtv.org"));
911 proxies = QNetworkProxyFactory::systemProxyForQuery(query);
913 for (
const auto&
p : std::as_const(
proxies))
915 QString host =
p.hostName();
918 if (
p.type() == QNetworkProxy::NoProxy)
923 LOG(VB_NETWORK, LOG_ERR,
LOC +
924 "failed to contact proxy host " + host);
928 LOG(VB_NETWORK, LOG_INFO,
LOC + QString(
"using proxy host %1:%2")
929 .arg(host).arg(port));
930 QNetworkProxy::setApplicationProxy(
p);
936 if (!
p.user().isEmpty())
938 url =
"http://%1:%2@%3:%4",
939 url = url.arg(
p.user(),
p.password());
943 url =
"http://%1:%2";
946 url = url.arg(
p.hostName()).arg(
p.port());
947 setenv(
"HTTP_PROXY", url.toLatin1(), 1);
948 setenv(
"http_proxy", url.toLatin1(), 0);
953 LOG(VB_NETWORK, LOG_ERR,
LOC +
"failed to find a network proxy");
static QString resolveAddress(const QString &host, ResolveType type=ResolveAny, bool keepscope=false)
if host is an IP address, it will be returned or resolved otherwise.
bool SendReceiveStringList(QStringList &strlist, bool quickTimeout=false, bool block=true)
Send a message to the backend and wait for a response.
Class for communcating between myth backends and frontends.
@ GENERIC_EXIT_OK
Exited with no error.
MythCoreContext * gCoreContext
This global variable contains the MythCoreContext instance for the app.
#define ENO
This can be appended to the LOG args with "+".
#define LOG(_MASK_, _LEVEL_, _QSTRING_)
bool RemoteGetUptime(std::chrono::seconds &uptime)
bool getMemStats(int &totalMB, int &freeMB, int &totalVM, int &freeVM)
Returns memory statistics in megabytes.
loadArray getLoadAvgs(void)
Returns the system load averages.
bool RemoteGetLoad(loadArray &load)
bool MythWakeup(const QString &wakeUpCommand, uint flags, std::chrono::seconds timeout)
bool MythRemoveDirectory(QDir &aDir)
bool IsMACAddress(const QString &MAC)
int intResponse(const QString &query, int def)
In an interactive shell, prompt the user to input a number.
bool getUptime(std::chrono::seconds &uptime)
Returns uptime statistics.
QString getResponse(const QString &query, const QString &def)
In an interactive shell, prompt the user to input a string.
bool IsPulseAudioRunning(void)
Is A/V Sync destruction daemon is running on this host?
QString getSymlinkTarget(const QString &start_file, QStringList *intermediaries, unsigned maxLinks)
void setHttpProxy(void)
Get network proxy settings from OS, and use for [Q]Http[Comms].
bool WakeOnLAN(const QString &MAC)
QString FileHash(const QString &filename)
bool telnet(const QString &host, int port)
Can we talk to port on host?
bool RemoteGetMemStats(int &totalMB, int &freeMB, int &totalVM, int &freeVM)
bool ping(const QString &host, std::chrono::milliseconds timeout)
Can we ping host within timeout seconds?
bool myth_ioprio(int)
Allows setting the I/O priority of the current process/thread.
bool makeFileAccessible(const QString &filename)
QString createTempFile(QString name_template, bool dir)
std::array< double, 3 > loadArray
@ kMSDontBlockInputDevs
avoid blocking LIRC & Joystick Menu
@ kMSProcessEvents
process events while waiting
@ kMSDontDisableDrawing
avoid disabling UI drawing
uint myth_system(const QString &command, uint flags, std::chrono::seconds timeout)
MBASE_PUBLIC long long copy(QFile &dst, QFile &src, uint block_size=0)
Copies src file to dst file.