summaryrefslogtreecommitdiffstats
path: root/kdesudo/kdesudo.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'kdesudo/kdesudo.cpp')
-rw-r--r--kdesudo/kdesudo.cpp439
1 files changed, 439 insertions, 0 deletions
diff --git a/kdesudo/kdesudo.cpp b/kdesudo/kdesudo.cpp
new file mode 100644
index 0000000..e746703
--- /dev/null
+++ b/kdesudo/kdesudo.cpp
@@ -0,0 +1,439 @@
+/***************************************************************************
+ kdesudo.cpp - description
+ -------------------
+ begin : Sam Feb 15 15:42:12 CET 2003
+ copyright : (C) 2003 by Robert Gruber <rgruber@users.sourceforge.net>
+ (C) 2007 by Martin Böhm <martin.bohm@kubuntu.org>
+ Anthony Mercatante <tonio@kubuntu.org>
+ Canonical Ltd (Jonathan Riddell <jriddell@ubuntu.com>)
+
+ ***************************************************************************/
+
+/***************************************************************************
+ * *
+ * This program is free software; you can redistribute it and/or modify *
+ * it under the terms of the GNU General Public License as published by *
+ * the Free Software Foundation; either version 2 of the License, or *
+ * (at your option) any later version. *
+ * *
+ ***************************************************************************/
+
+#include "kdesudo.h"
+
+#include <qfile.h>
+#include <qdir.h>
+#include <qdatastream.h>
+#include <qstring.h>
+
+#include <kcmdlineargs.h>
+#include <klocale.h>
+#include <kmessagebox.h>
+#include <kpushbutton.h>
+#include <kpassdlg.h>
+#include <kstandarddirs.h>
+#include <kdesu/kcookie.h>
+#include <kdebug.h>
+#include <kshell.h>
+#include <ktempfile.h>
+
+#include <iostream>
+#include <cstdio>
+#include <cstdlib>
+#include <signal.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+KdeSudo::KdeSudo(QWidget *parent, const char *name,const QString& icon, const QString& generic, bool withIgnoreButton)
+ : KPasswordDialog(KPasswordDialog::Password, false, (withIgnoreButton ? User1: false), icon, parent, name)
+{
+ KCmdLineArgs *args = KCmdLineArgs::parsedArgs();
+
+ QString defaultComment = i18n("<b>%1</b> needs administrative privileges. Please enter your password for verification.");
+ p=NULL;
+ bError=false;
+
+ m_pCookie = new KCookie;
+
+ // Set vars
+ bool newDcop = args->isSet("newdcop");
+ bool realtime = args->isSet("r");
+ bool priority = args->isSet("p");
+ bool showCommand = (!args->isSet("d"));
+ bool changeUID = true;
+ bool noExec = false;
+ keepPwd = (!args->isSet("n"));
+ emptyPwd = args->isSet("s");
+ QString runas = args->getOption("u");
+ QString cmd;
+
+ if (!args->isSet("c") && !args->count() && (!args->isSet("s")))
+ {
+ KMessageBox::information(NULL, i18n("No command arguments supplied!\nUsage: kdesudo [-u <runas>] <command>\nKdeSudo will now exit..."));
+ noExec = true;
+ }
+
+ p = new KProcess;
+ p->clearArguments();
+
+ // Parsins args
+
+ /* Get the comment out of cli args */
+ QByteArray commentBytes = args->getOption("comment");
+ QTextCodec* tCodecConv = QTextCodec::codecForLocale();
+ QString comment = tCodecConv->toUnicode(commentBytes, commentBytes.size());
+
+ if (args->isSet("f"))
+ {
+ // If file is writeable, do not change uid
+ QString filename = QFile::decodeName(args->getOption("f"));
+ QString file = filename;
+ if (!file.isEmpty())
+ {
+ if (file.at(0) != '/')
+ {
+ KStandardDirs dirs;
+ dirs.addKDEDefaults();
+ file = dirs.findResource("config", file);
+ if (file.isEmpty())
+ {
+ kdError(1206) << "Config file not found: " << file << "\n";
+ exit(1);
+ }
+ }
+ QFileInfo fi(file);
+ if (!fi.exists())
+ {
+ kdError(1206) << "File does not exist: " << file << "\n";
+ exit(1);
+ }
+ if (fi.isWritable())
+ {
+ changeUID = false;
+ }
+ }
+ }
+
+ if (withIgnoreButton)
+ {
+ setButtonText(User1, i18n("&Ignore"));
+ }
+
+ // Apologies for the C code, taken from kdelibs/kdesu/kdesu_stub.c
+ // KControl and other places need to use the user's existing DCOP server
+ // For that we set DCOPSERVER. Create a file in /tmp and use iceauth to add magic cookies
+ // from the existing server and set ICEAUTHORITY to point to the file
+ if (!newDcop) {
+ dcopServer = m_pCookie->dcopServer();
+ QCString dcopAuth = m_pCookie->dcopAuth();
+ QCString iceAuth = m_pCookie->iceAuth();
+
+ FILE *fout;
+ char iceauthority[200];
+ char *host, *auth;
+ host = qstrdup(dcopServer);
+ auth = qstrdup(iceAuth);
+ int tempfile;
+ int oldumask = umask(077);
+
+ strcpy(iceauthority, "/tmp/iceauth.XXXXXXXXXX");
+ tempfile = mkstemp(iceauthority);
+ umask(oldumask);
+ if (tempfile == -1) {
+ kdError() << "error in kdesudo mkstemp" << endl;
+ exit(1);
+ } else {
+ // close(tempfile); //FIXME why does this make the connect() call later crash?
+ }
+ iceauthorityFile = iceauthority;
+ //FIXME we should change owner of iceauthority file, but don't have permissions
+ setenv("ICEAUTHORITY", iceauthorityFile, 1);
+
+ fout = popen("iceauth >/dev/null 2>&1", "w");
+ if (!fout) {
+ kdError() << "error in kdesudo running iceauth" << endl;
+ exit(1);
+ }
+ fprintf(fout, "add ICE \"\" %s %s\n", host, auth);
+ auth = qstrdup(dcopAuth);
+ //auth = xstrsep(params[P_DCOP_AUTH].value);
+ fprintf(fout, "add DCOP \"\" %s %s\n", host, auth);
+ unsetenv("ICEAUTHORITY");
+ pclose(fout);
+
+ // Exporting the user's sycoca
+ kdeSycoca = QFile::encodeName(locateLocal("cache", "ksycoca"));
+ }
+
+ connect( p, SIGNAL(receivedStdout(KProcess*, char*, int)), this, SLOT(receivedOut(KProcess*, char*, int)) );
+ connect( p, SIGNAL(receivedStderr(KProcess*, char*, int)), this, SLOT(receivedOut(KProcess*, char*, int)) );
+ connect( p, SIGNAL(processExited (KProcess *)), this, SLOT(procExited(KProcess*)));
+
+ QString xauthenv = QString(getenv("HOME")) + "/.Xauthority";
+ p->setEnvironment("XAUTHORITY", xauthenv);
+
+ // Generate the xauth cookie and put it in a tempfile
+ // set the environment variables to reflect that.
+ // Default cookie-timeout is 60 sec. .
+ // 'man xauth' for more info on xauth cookies.
+
+ KTempFile temp = KTempFile("/tmp/kdesudo-","-xauth");
+ m_tmpname = temp.name();
+
+ FILE *f;
+ char buf[1024];
+
+ QCString disp = m_pCookie->display();
+ // command: xauth -q -f m_tmpname generate $DISPLAy . trusted timeout 60
+ QString c = "/usr/bin/xauth -q -f " + m_tmpname + " generate "
+ + QString::fromLocal8Bit(disp) + " . trusted timeout 60";
+ blockSigChild(); // pclose uses waitpid()
+
+ if (!(f = popen(c, "r"))) {
+ kdWarning() << k_lineinfo << "Cannot run: " << c << "\n";
+ unblockSigChild();
+ return;
+ }
+
+ // non root users need to be able to read the xauth file.
+ // the xauth file is deleted when kdesudo exits. security?
+ QFile tf(m_tmpname);
+ if (!runas.isEmpty() && runas != "root" && tf.exists())
+ chmod(m_tmpname.ascii(),0644);
+
+ QCStringList output;
+ while (fgets(buf, 1024, f) > 0)
+ output += buf;
+ if (pclose(f) < 0) {
+ kdError() << k_lineinfo << "Could not run xauth.\n";
+ unblockSigChild();
+ return;
+ }
+ unblockSigChild();
+
+ p->setEnvironment("DISPLAY", disp);
+ p->setEnvironment("XAUTHORITY", m_tmpname);
+
+ if (emptyPwd)
+ *p << "sudo" << "-k";
+ else
+ {
+ if (changeUID)
+ {
+ *p << "sudo" << "-H" << "-S" << "-p" << "passprompt";
+
+ if (!runas.isEmpty())
+ *p << "-u" << runas;
+ }
+
+ if (!dcopServer.isEmpty())
+ *p << "DCOPSERVER=" + dcopServer;
+
+ if (!iceauthorityFile.isEmpty())
+ *p << "ICEAUTHORITY=" + iceauthorityFile;
+
+ if (!kdeSycoca.isEmpty())
+ *p << "KDESYCOCA=" + kdeSycoca;
+
+ if (realtime)
+ {
+ *p << "nice" << "-n" << "10";
+ addLine(i18n("Priority:"), i18n("realtime:") + QChar(' ') + QString("50/100"));
+ }
+ else if (priority)
+ {
+ QString n = args->getOption("p");
+ int intn = atoi(n);
+ intn = (intn * 40 / 100) - (20 + 0.5);
+
+ QString strn;
+ strn.sprintf("%d",intn);
+
+ *p << "nice" << "-n" << strn;
+ addLine(i18n("Priority:"), n + QString("/100"));
+ }
+
+ *p << "--";
+
+ if (args->isSet("c"))
+ {
+ QString command = args->getOption("c");
+ QStringList commandSplit = KShell::splitArgs(command);
+
+ for (int i = 0; i < commandSplit.count(); i++)
+ {
+ QString arg = validArg(commandSplit[i]);
+ *p << arg;
+ if (i == 0)
+ cmd += validArg(commandSplit[i]) + QChar(' ');
+ else
+ cmd += KProcess::quote(validArg(commandSplit[i])) + QChar(' ');
+ }
+ }
+
+ if (args->count())
+ {
+ for (int i = 0; i < args->count(); i++)
+ {
+ if ((!args->isSet("c")) && (i == 0))
+ {
+ QStringList argsSplit = KShell::splitArgs(args->arg(i));
+ for (int j = 0; j < argsSplit.count(); j++)
+ {
+ *p << validArg(argsSplit[j]);
+ if (j == 0)
+ cmd += validArg(argsSplit[j]) + QChar(' ');
+ else
+ cmd += KProcess::quote(validArg(argsSplit[j])) + QChar(' ');
+ }
+ }
+ else
+ {
+ *p << validArg(args->arg(i));
+ cmd += validArg(args->arg(i)) + QChar(' ');
+ }
+ }
+ }
+
+ if (showCommand && !cmd.isEmpty())
+ addLine(i18n("Command:"), cmd);
+ }
+
+ if (comment.isEmpty())
+ {
+ if (!generic.isEmpty())
+ setPrompt(defaultComment.arg(generic));
+ else
+ setPrompt(defaultComment.arg(cmd));
+ }
+ else
+ setPrompt(comment);
+
+ if (noExec)
+ exit(0);
+ else
+ p->start( KProcess::NotifyOnExit, KProcess::All );
+}
+
+KdeSudo::~KdeSudo()
+{
+}
+
+void KdeSudo::receivedOut(KProcess*, char*buffer, int buflen)
+{
+ char *pcTmp= new char[buflen+1];
+ strncpy(pcTmp,buffer,buflen);
+ pcTmp[buflen]='\0';
+ QString strOut(pcTmp);
+
+ std::cout << strOut << std::endl;
+
+ static int badpass = 0;
+
+ if (strOut.find("Sorry, try again")!=-1)
+ {
+ badpass++;
+ if (badpass>2)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Wrong password! Exiting..."));
+ kapp->quit();
+ }
+ }
+ if (strOut.find("command not found")!=-1)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Command not found!"));
+ kapp->quit();
+ }
+ if (strOut.find("is not in the sudoers file")!=-1)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Your username is unknown to sudo!"));
+ kapp->quit();
+ }
+ if (strOut.find("is not allowed to execute")!=-1)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Your user is not allowed to run the specified command!"));
+ kapp->quit();
+ }
+ if (strOut.find("is not allowed to run sudo on")!=-1)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Your user is not allowed to run sudo on this host!"));
+ kapp->quit();
+ }
+ if (strOut.find("may not run sudo on")!=-1)
+ {
+ bError=true;
+ KMessageBox::error(this, i18n("Your user is not allowed to run sudo on this host!"));
+ kapp->quit();
+ }
+ if ((strOut.find("passprompt")!=-1) || (strOut.find("PIN (CHV2)")!=-1))
+ {
+ this->clearPassword();
+ this->show();
+ }
+}
+
+void KdeSudo::procExited(KProcess*)
+{
+ if (!keepPwd && unCleaned)
+ {
+ unCleaned = false;
+ p->clearArguments();
+ *p << "sudo" << "-k";
+ p->start( KProcess::NotifyOnExit, KProcess::All );
+ }
+
+ if (!newDcop && !iceauthorityFile.isEmpty())
+ if (!iceauthorityFile.isEmpty())
+ QFile::remove(iceauthorityFile);
+
+ if (!bError) {
+ if (!m_tmpname.isEmpty())
+ QFile::remove(m_tmpname);
+ kapp->quit();
+ }
+}
+
+void KdeSudo::slotOk()
+{
+ QString strTmp(password());
+ strTmp+="\n";
+ p->writeStdin(strTmp.ascii(),(int)strTmp.length());
+ this->hide();
+}
+
+void KdeSudo::slotUser1()
+{
+ done(AsUser);
+}
+
+void KdeSudo::blockSigChild()
+{
+ sigset_t sset;
+ sigemptyset(&sset);
+ sigaddset(&sset, SIGCHLD);
+ sigprocmask(SIG_BLOCK, &sset, 0L);
+}
+void KdeSudo::unblockSigChild()
+{
+ sigset_t sset;
+ sigemptyset(&sset);
+ sigaddset(&sset, SIGCHLD);
+ sigprocmask(SIG_UNBLOCK, &sset, 0L);
+}
+
+QString KdeSudo::validArg(QString arg)
+{
+ QChar firstChar = arg.at(0);
+ QChar lastChar = arg.at(arg.length() - 1);
+
+ if ( (firstChar == '"' && lastChar == '"') || (firstChar == '\'' && lastChar == '\'') )
+ {
+ arg = arg.remove(0, 1);
+ arg = arg.remove(arg.length() - 1, 1);
+ }
+ return arg;
+}