You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
334 lines
8.5 KiB
334 lines
8.5 KiB
/***************************************************************************
|
|
* CT Cron Implementation *
|
|
* -------------------------------------------------------------------- *
|
|
* Copyright (C) 1999, Gary Meyer <gary@meyer.net> *
|
|
* -------------------------------------------------------------------- *
|
|
* 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. *
|
|
***************************************************************************/
|
|
|
|
// Do not introduce any TQt or KDE dependencies into the "CT"-prefixed classes.
|
|
// I want to be able to reuse these classes with another GUI toolkit. -GM 11/99
|
|
|
|
#include "ctcron.h"
|
|
|
|
#include "cti18n.h"
|
|
#include "cttask.h"
|
|
#include "ctvariable.h"
|
|
#include <unistd.h> // getuid(), unlink()
|
|
#include <pwd.h> // pwd, getpwnam(), getpwuid()
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#include <tqfile.h>
|
|
|
|
#include <kprocess.h>
|
|
#include <klocale.h>
|
|
#include <ktempfile.h>
|
|
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
|
|
CTCron::CTCron(bool _syscron, string _login) :
|
|
syscron(_syscron)
|
|
{
|
|
int uid(getuid());
|
|
|
|
KTempFile tmp;
|
|
tmp.setAutoDelete(true);
|
|
tmp.close();
|
|
tmpFileName = tmp.name();
|
|
|
|
TQString readCommand;
|
|
|
|
if (uid == 0)
|
|
// root, so provide requested crontab
|
|
{
|
|
if (syscron)
|
|
{
|
|
readCommand = "cat /etc/crontab > " + KProcess::quote(tmpFileName);
|
|
writeCommand = "cat " + KProcess::quote(tmpFileName) + " > /etc/crontab";
|
|
login = (const char *)i18n("(System Crontab)").local8Bit();
|
|
name = "";
|
|
}
|
|
else
|
|
{
|
|
readCommand = TQString("crontab -u ") + _login.c_str() + " -l > " + KProcess::quote(tmpFileName);
|
|
writeCommand = TQString("crontab -u ") + _login.c_str() + " " + KProcess::quote(tmpFileName);
|
|
if (!initFromPasswd(getpwnam(_login.c_str())))
|
|
{
|
|
error = i18n("No password entry found for user '%1'").arg(_login.c_str());
|
|
}
|
|
}
|
|
}
|
|
else
|
|
// regular user, so provide user's own crontab
|
|
{
|
|
readCommand = "crontab -l > " + KProcess::quote(tmpFileName);
|
|
writeCommand = "crontab " + KProcess::quote(tmpFileName);
|
|
if (!initFromPasswd(getpwuid(uid)))
|
|
{
|
|
error = i18n("No password entry found for uid '%1'").arg(uid);
|
|
}
|
|
}
|
|
|
|
if (name.empty())
|
|
name = login;
|
|
|
|
initialTaskCount = 0;
|
|
initialVariableCount = 0;
|
|
|
|
if (isError())
|
|
return;
|
|
|
|
// Don't set error if it can't be read, it means the user
|
|
// doesn't have a crontab.
|
|
if (system(TQFile::encodeName(readCommand)) == 0)
|
|
{
|
|
ifstream cronfile(TQFile::encodeName(tmpFileName));
|
|
cronfile >> *this;
|
|
}
|
|
|
|
initialTaskCount = task.size();
|
|
initialVariableCount = variable.size();
|
|
}
|
|
|
|
CTCron::CTCron(const struct passwd *pwd) :
|
|
syscron(false)
|
|
{
|
|
Q_ASSERT(pwd != 0L);
|
|
|
|
KTempFile tmp;
|
|
tmp.setAutoDelete(true);
|
|
tmp.close();
|
|
tmpFileName = tmp.name();
|
|
|
|
TQString readCommand = TQString("crontab -u ") + TQString(pwd->pw_name) + " -l > " + KProcess::quote(tmpFileName);
|
|
writeCommand = TQString("crontab -u ") + TQString(pwd->pw_name) + " " + KProcess::quote(tmpFileName);
|
|
|
|
initFromPasswd(pwd);
|
|
|
|
initialTaskCount = 0;
|
|
initialVariableCount = 0;
|
|
|
|
if (isError())
|
|
return;
|
|
|
|
// Don't set error if it can't be read, it means the user
|
|
// doesn't have a crontab.
|
|
if (system(TQFile::encodeName(readCommand)) == 0)
|
|
{
|
|
ifstream cronfile(TQFile::encodeName(tmpFileName));
|
|
cronfile >> *this;
|
|
}
|
|
|
|
initialTaskCount = task.size();
|
|
initialVariableCount = variable.size();
|
|
}
|
|
|
|
bool CTCron::initFromPasswd(const struct passwd *pwd)
|
|
{
|
|
if (pwd == 0)
|
|
{
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
login = pwd->pw_name;
|
|
name = pwd->pw_gecos;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
void CTCron::operator = (const CTCron& source)
|
|
{
|
|
assert(!source.syscron);
|
|
|
|
for (CTVariableIterator i = const_cast<CTCron&>(source).variable.begin();
|
|
i != source.variable.end(); ++i)
|
|
{
|
|
CTVariable* tmp = new CTVariable(**i);
|
|
variable.push_back(tmp);
|
|
}
|
|
|
|
for (CTTaskIterator i = const_cast<CTCron&>(source).task.begin();
|
|
i != source.task.end(); ++i)
|
|
{
|
|
CTTask* tmp = new CTTask(**i);
|
|
task.push_back(tmp);
|
|
}
|
|
}
|
|
|
|
istream& operator >> (istream& inputStream, CTCron& cron)
|
|
{
|
|
const int MAX = 1024;
|
|
char buffer[MAX];
|
|
string line("");
|
|
string comment("");
|
|
|
|
while (inputStream)
|
|
{
|
|
inputStream.getline(buffer, MAX);
|
|
line = buffer;
|
|
|
|
// search for comments "#" but not disabled tasks "#\"
|
|
if ((line.find("#") == 0) && (line.find("\\") != 1))
|
|
{
|
|
// If the first 10 characters don't contain a character, it's probably a disabled entry.
|
|
int first_text = line.find_first_of("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ");
|
|
if (first_text < 0)
|
|
continue;
|
|
|
|
if (first_text < 10)
|
|
{
|
|
// remove leading pound sign
|
|
line = line.substr(1,line.length()-1);
|
|
// remove leading whitespace
|
|
while (line.find_first_of(" \t") == 0)
|
|
line = line.substr(1,line.length()-1);
|
|
comment = line;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// else
|
|
{
|
|
// either a task or a variable
|
|
int firstWhiteSpace(line.find_first_of(" \t"));
|
|
int firstEquals(line.find("="));
|
|
|
|
// if there is an equals sign and either there is no
|
|
// whitespace or the first whitespace is after the equals
|
|
// sign, it must be a variable
|
|
if ((firstEquals > 0) && ((firstWhiteSpace == -1) ||
|
|
firstWhiteSpace > firstEquals))
|
|
{
|
|
// create variable
|
|
CTVariable *tmp = new CTVariable(line, comment);
|
|
cron.variable.push_back(tmp);
|
|
comment = "";
|
|
}
|
|
else
|
|
// must be a task, either enabled or disabled
|
|
{
|
|
if (firstWhiteSpace > 0)
|
|
{
|
|
CTTask *tmp = new CTTask(line, comment, cron.syscron);
|
|
cron.task.push_back(tmp);
|
|
comment = "";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return inputStream;
|
|
}
|
|
|
|
ostream& operator << (ostream& outputStream, const CTCron& cron)
|
|
{
|
|
int itemCount(0);
|
|
|
|
for (CTVariableIterator i = const_cast<CTCron&>(cron).variable.begin();
|
|
i != cron.variable.end(); ++i)
|
|
{
|
|
outputStream << **i;
|
|
itemCount++;
|
|
}
|
|
|
|
for (CTTaskIterator i = const_cast<CTCron&>(cron).task.begin();
|
|
i != cron.task.end(); ++i)
|
|
{
|
|
outputStream << **i;
|
|
itemCount++;
|
|
}
|
|
|
|
if (itemCount > 0)
|
|
{
|
|
outputStream << "# This file was written by KCron. Copyright (c) 1999, Gary Meyer\n";
|
|
outputStream << "# Although KCron supports most crontab formats, use care when editing.\n";
|
|
outputStream << "# Note: Lines beginning with \"#\\\" indicates a disabled task.\n";
|
|
}
|
|
|
|
return outputStream;
|
|
}
|
|
|
|
CTCron::~CTCron()
|
|
{
|
|
for (CTTaskIterator i = task.begin(); i != task.end(); ++i)
|
|
delete *i;
|
|
for (CTVariableIterator i = variable.begin(); i != variable.end(); ++i)
|
|
delete *i;
|
|
}
|
|
|
|
void CTCron::apply()
|
|
{
|
|
// write to temp file
|
|
ofstream cronfile(TQFile::encodeName(tmpFileName));
|
|
cronfile << *this << flush;
|
|
|
|
// install temp file into crontab
|
|
if (system(TQFile::encodeName(writeCommand)) != 0)
|
|
{
|
|
error = i18n("An error occurred while updating crontab.");
|
|
}
|
|
|
|
// remove the temp file
|
|
(void) unlink(TQFile::encodeName(tmpFileName));
|
|
|
|
if (isError())
|
|
return;
|
|
|
|
// mark as applied
|
|
for (CTTaskIterator i = task.begin(); i != task.end(); ++i)
|
|
(*i)->apply();
|
|
|
|
for (CTVariableIterator i = variable.begin(); i != variable.end(); ++i)
|
|
(*i)->apply();
|
|
|
|
initialTaskCount = task.size();
|
|
initialVariableCount = variable.size();
|
|
}
|
|
|
|
void CTCron::cancel()
|
|
{
|
|
for (CTTaskIterator i = task.begin(); i != task.end(); ++i)
|
|
(*i)->cancel();
|
|
|
|
for (CTVariableIterator i = variable.begin(); i != variable.end(); ++i)
|
|
(*i)->cancel();
|
|
}
|
|
|
|
bool CTCron::dirty()
|
|
{
|
|
bool isDirty(false);
|
|
|
|
if (initialTaskCount != task.size()) isDirty = true;
|
|
|
|
if (initialVariableCount != variable.size()) isDirty = true;
|
|
|
|
for (CTTaskIterator i = task.begin(); i != task.end(); ++i)
|
|
if ((*i)->dirty()) isDirty = true;
|
|
|
|
for (CTVariableIterator i = variable.begin(); i != variable.end(); ++i)
|
|
if ((*i)->dirty()) isDirty = true;
|
|
|
|
return isDirty;
|
|
}
|
|
|
|
string CTCron::path() const
|
|
{
|
|
string path;
|
|
|
|
for (CTVariableIterator var = const_cast<CTCron*>(this)->variable.begin();
|
|
var != variable.end(); var++)
|
|
{
|
|
if ((*var)->variable == "PATH")
|
|
{
|
|
path = (*var)->value;
|
|
}
|
|
}
|
|
return path;
|
|
}
|