Alf Gaida
9 years ago
parent
d286267068
commit
7aff9fade1
@ -0,0 +1 @@
|
|||||||
|
lxqt-sudo.1
|
||||||
@ -0,0 +1 @@
|
|||||||
|
lxqt-sudo.1
|
||||||
@ -1,193 +0,0 @@
|
|||||||
/* BEGIN_COMMON_COPYRIGHT_HEADER
|
|
||||||
* (c)LGPL2+
|
|
||||||
*
|
|
||||||
* LXQt - a lightweight, Qt based, desktop toolset
|
|
||||||
* http://lxqt.org
|
|
||||||
*
|
|
||||||
* Copyright: 2015 LXQt team
|
|
||||||
* Authors:
|
|
||||||
* Palo Kisa <palo.kisa@gmail.com>
|
|
||||||
*
|
|
||||||
* This program or library is free software; you can redistribute it
|
|
||||||
* and/or modify it under the terms of the GNU Lesser General Public
|
|
||||||
* License as published by the Free Software Foundation; either
|
|
||||||
* version 2.1 of the License, or (at your option) any later version.
|
|
||||||
*
|
|
||||||
* This library 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
|
|
||||||
* Lesser General Public License for more details.
|
|
||||||
|
|
||||||
* You should have received a copy of the GNU Lesser General
|
|
||||||
* Public License along with this library; if not, write to the
|
|
||||||
* Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
||||||
* Boston, MA 02110-1301 USA
|
|
||||||
*
|
|
||||||
* END_COMMON_COPYRIGHT_HEADER */
|
|
||||||
|
|
||||||
#include <LXQt/Application>
|
|
||||||
#include "passworddialog.h"
|
|
||||||
#include <QMessageBox>
|
|
||||||
#include <QSocketNotifier>
|
|
||||||
#include <QDebug>
|
|
||||||
#include <QThread>
|
|
||||||
#include <pty.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <memory>
|
|
||||||
#include <csignal>
|
|
||||||
#include <sys/wait.h>
|
|
||||||
#include <fcntl.h>
|
|
||||||
|
|
||||||
extern const QString app_master;
|
|
||||||
extern const QString su_prog;
|
|
||||||
const QString su_prog{QStringLiteral(LXQTSUDO_SU)};
|
|
||||||
static const QString su_pwd_prompt_end{QStringLiteral(": ")};
|
|
||||||
static const QChar nl{QLatin1Char('\n')};
|
|
||||||
|
|
||||||
static void child(QString const & arg, PasswordDialog & dlg)
|
|
||||||
{
|
|
||||||
std::string s_prog = su_prog.toStdString()
|
|
||||||
, s_arg = arg.toStdString();
|
|
||||||
|
|
||||||
char const * params[] = {
|
|
||||||
s_prog.c_str()
|
|
||||||
, "-c"
|
|
||||||
, s_arg.c_str()
|
|
||||||
, nullptr
|
|
||||||
};
|
|
||||||
|
|
||||||
setsid(); //session leader
|
|
||||||
execvp(params[0], const_cast<char **>(params));
|
|
||||||
|
|
||||||
//exec never returns in case of success
|
|
||||||
QTextStream{stderr, QIODevice::WriteOnly} << QObject::tr("%1: Failed to exec '%2': %3\n").arg(app_master).arg(su_prog).arg(strerror(errno));
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void stopChild(int & childPid
|
|
||||||
, int & pwdFd
|
|
||||||
, int & ret)
|
|
||||||
{
|
|
||||||
close(pwdFd);
|
|
||||||
kill(childPid, SIGINT);
|
|
||||||
int res, status;
|
|
||||||
for (int cnt = 10; 0 == (res = waitpid(childPid, &status, WNOHANG)) && 0 < cnt; --cnt)
|
|
||||||
QThread::msleep(100);
|
|
||||||
|
|
||||||
if (0 == res)
|
|
||||||
{
|
|
||||||
kill(childPid, SIGKILL);
|
|
||||||
ret = 1;
|
|
||||||
} else
|
|
||||||
{
|
|
||||||
ret = WIFEXITED(status) ? WEXITSTATUS(status) : 1;
|
|
||||||
}
|
|
||||||
childPid = -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
static int parent(int childPid, int pwdFd, PasswordDialog & dlg)
|
|
||||||
{
|
|
||||||
//set the FD as non-blocking
|
|
||||||
if (0 != fcntl(pwdFd, F_SETFL, O_NONBLOCK))
|
|
||||||
{
|
|
||||||
QMessageBox(QMessageBox::Critical, dlg.windowTitle()
|
|
||||||
, QObject::tr("Failed to set non-block: %1").arg(strerror(errno)), QMessageBox::Ok).exec();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
FILE * pwd_f = fdopen(pwdFd, "r+");
|
|
||||||
if (nullptr == pwd_f)
|
|
||||||
{
|
|
||||||
QMessageBox(QMessageBox::Critical, dlg.windowTitle()
|
|
||||||
, QObject::tr("Failed to fdopen: %1").arg(strerror(errno)), QMessageBox::Ok).exec();
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
QTextStream child_str{pwd_f};
|
|
||||||
|
|
||||||
int ret;
|
|
||||||
QObject::connect(&dlg, &QDialog::finished, [&] (int result)
|
|
||||||
{
|
|
||||||
if (QDialog::Accepted == result)
|
|
||||||
{
|
|
||||||
child_str << dlg.password().append(nl);
|
|
||||||
child_str.flush();
|
|
||||||
} else
|
|
||||||
{
|
|
||||||
stopChild(childPid, pwdFd, ret);
|
|
||||||
lxqtApp->quit();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
bool check_pwd = true;
|
|
||||||
QString last_line;
|
|
||||||
QScopedPointer<QSocketNotifier> pwd_watcher{new QSocketNotifier{pwdFd, QSocketNotifier::Read}};
|
|
||||||
QObject::connect(pwd_watcher.data(), &QSocketNotifier::activated, [&]
|
|
||||||
{
|
|
||||||
QString line = child_str.readAll();
|
|
||||||
if (line.isEmpty())
|
|
||||||
{
|
|
||||||
pwd_watcher.reset(nullptr);
|
|
||||||
if (last_line.startsWith(QStringLiteral("%1:").arg(su_prog)))
|
|
||||||
{
|
|
||||||
pwd_watcher.reset(nullptr); //stop the notifications events
|
|
||||||
stopChild(childPid, pwdFd, ret);
|
|
||||||
QMessageBox(QMessageBox::Critical, dlg.windowTitle()
|
|
||||||
, QObject::tr("Child '%1' process failed!\n%2").arg(su_prog).arg(last_line), QMessageBox::Ok).exec();
|
|
||||||
}
|
|
||||||
lxqtApp->quit();
|
|
||||||
} else
|
|
||||||
{
|
|
||||||
if (check_pwd)
|
|
||||||
{
|
|
||||||
//check only first output of child(su)
|
|
||||||
check_pwd = false;
|
|
||||||
if (!line.contains(nl) && line.endsWith(su_pwd_prompt_end))
|
|
||||||
{
|
|
||||||
//if now echo is turned off, su requests password struct termios tios;
|
|
||||||
struct termios tios;
|
|
||||||
Q_ASSERT(0 == tcgetattr(pwdFd, &tios));
|
|
||||||
if (!(ECHO & tios.c_lflag))
|
|
||||||
{
|
|
||||||
dlg.show();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
QTextStream{stderr, QIODevice::WriteOnly} << line;
|
|
||||||
//assuming text oriented output
|
|
||||||
QStringList lines = line.split(nl, QString::SkipEmptyParts);
|
|
||||||
last_line = lines.isEmpty() ? QString() : lines.back();
|
|
||||||
}
|
|
||||||
|
|
||||||
});
|
|
||||||
|
|
||||||
lxqtApp->exec();
|
|
||||||
|
|
||||||
if (0 < childPid)
|
|
||||||
{
|
|
||||||
int res, status;
|
|
||||||
res = waitpid(childPid, &status, 0);
|
|
||||||
ret = (childPid == res && WIFEXITED(status)) ? WEXITSTATUS(status) : 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
int su(QStringList const & args, PasswordDialog & dlg)
|
|
||||||
{
|
|
||||||
int pid, fd;
|
|
||||||
if (1 != args.size())
|
|
||||||
QMessageBox(QMessageBox::Critical, dlg.windowTitle()
|
|
||||||
, QObject::tr("With %1 backend only one argument/command is supported!").arg(su_prog), QMessageBox::Ok).exec();
|
|
||||||
else if (0 == (pid = forkpty(&fd, nullptr, nullptr, nullptr)))
|
|
||||||
child(args[0], dlg); //never returns
|
|
||||||
else if (-1 == pid)
|
|
||||||
QMessageBox(QMessageBox::Critical, dlg.windowTitle()
|
|
||||||
, QObject::tr("Failed to fork: %1").arg(strerror(errno)), QMessageBox::Ok).exec();
|
|
||||||
else
|
|
||||||
return parent(pid, fd, dlg);
|
|
||||||
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
@ -0,0 +1,71 @@
|
|||||||
|
/* BEGIN_COMMON_COPYRIGHT_HEADER
|
||||||
|
* (c)LGPL2+
|
||||||
|
*
|
||||||
|
* LXQt - a lightweight, Qt based, desktop toolset
|
||||||
|
* http://lxqt.org
|
||||||
|
*
|
||||||
|
* Copyright: 2015 LXQt team
|
||||||
|
* Authors:
|
||||||
|
* Palo Kisa <palo.kisa@gmail.com>
|
||||||
|
*
|
||||||
|
* This program or library is free software; you can redistribute it
|
||||||
|
* and/or modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation; either
|
||||||
|
* version 2.1 of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library 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
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
|
||||||
|
* You should have received a copy of the GNU Lesser General
|
||||||
|
* Public License along with this library; if not, write to the
|
||||||
|
* Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
||||||
|
* Boston, MA 02110-1301 USA
|
||||||
|
*
|
||||||
|
* END_COMMON_COPYRIGHT_HEADER */
|
||||||
|
|
||||||
|
#ifndef SUDO_H
|
||||||
|
#define SUDO_H
|
||||||
|
|
||||||
|
#include <QObject>
|
||||||
|
#include <QScopedPointer>
|
||||||
|
#include <QStringList>
|
||||||
|
|
||||||
|
class PasswordDialog;
|
||||||
|
|
||||||
|
class Sudo : public QObject
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
enum backend_t
|
||||||
|
{
|
||||||
|
BACK_NONE
|
||||||
|
, BACK_SUDO
|
||||||
|
, BACK_SU
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
Sudo();
|
||||||
|
~Sudo();
|
||||||
|
int main();
|
||||||
|
private:
|
||||||
|
//parent methods
|
||||||
|
int parent();
|
||||||
|
void stopChild();
|
||||||
|
|
||||||
|
//child methods
|
||||||
|
void child();
|
||||||
|
|
||||||
|
private:
|
||||||
|
QScopedPointer<PasswordDialog> mDlg;
|
||||||
|
QStringList mArgs;
|
||||||
|
backend_t mBackend;
|
||||||
|
|
||||||
|
int mChildPid;
|
||||||
|
int mPwdFd;
|
||||||
|
int mRet;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif //SUDO_H
|
||||||
Loading…
Reference in new issue