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.
tdelibs/tdeprint/lpd/kmlpdmanager.cpp

652 lines
18 KiB

/*
* This file is part of the KDE libraries
* Copyright (c) 2001 Michael Goffioul <tdeprint@swing.be>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License version 2 as published by the Free Software Foundation.
*
* 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
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public License
* along with this library; see the file COPYING.LIB. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
**/
#include "kmlpdmanager.h"
#include "kmprinter.h"
#include "kmdbentry.h"
#include "driver.h"
#include "kmfactory.h"
#include "lpdtools.h"
#include "gschecker.h"
#include "kpipeprocess.h"
#include <tqfile.h>
#include <tqfileinfo.h>
#include <textstream.h>
#include <tqmap.h>
#include <tqregexp.h>
#include <klocale.h>
#include <kstandarddirs.h>
#include <kconfig.h>
#include <kprocess.h>
#include <pwd.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/stat.h>
// only there to allow testing on my system. Should be removed
// when everything has proven to be working and stable
QString lpdprefix = "";
TQString ptPrinterType(KMPrinter*);
//************************************************************************************************
KMLpdManager::KMLpdManager(TQObject *parent, const char *name)
: KMManager(parent,name)
{
m_entries.setAutoDelete(true);
m_ptentries.setAutoDelete(true);
setHasManagement(getuid() == 0);
setPrinterOperationMask(KMManager::PrinterCreation|KMManager::PrinterConfigure|KMManager::PrinterRemoval|KMManager::PrinterEnabling);
m_gschecker = new GsChecker(this,"GsChecker");
}
KMLpdManager::~KMLpdManager()
{
}
TQString KMLpdManager::driverDbCreationProgram()
{
return TQString::fromLatin1("make_driver_db_lpd");
}
TQString KMLpdManager::driverDirectory()
{
return TQString::fromLatin1("/usr/lib/rhs/rhs-printfilters");
}
bool KMLpdManager::completePrinter(KMPrinter *printer)
{
return completePrinterShort(printer);
}
bool KMLpdManager::completePrinterShort(KMPrinter *printer)
{
PrintcapEntry *entry = m_entries.find(printer->name());
if (entry)
{
QString type(entry->comment(2)), driver(entry->comment(7)), lp(entry->arg("lp"));
printer->setDescription(i18n("Local printer queue (%1)").arg(type.isEmpty() ? i18n("Unknown type of local printer queue", "Unknown") : type));
printer->setLocation(i18n("<Not available>"));
printer->setDriverInfo(driver.isEmpty() ? i18n("Unknown Driver", "Unknown") : driver);
// device
KURL url;
if (!entry->arg("rm").isEmpty())
{
url = TQString::fromLatin1("lpd://%1/%2").arg(entry->arg("rm")).arg(entry->arg("rp"));
printer->setDescription(i18n("Remote LPD queue %1@%2").arg(entry->arg("rp")).arg(entry->arg("rm")));
}
else if (!lp.isEmpty() && lp != "/dev/null")
url = TQString::fromLatin1("parallel:%1").arg(lp);
else if (TQFile::exists(entry->arg("sd")+"/.config"))
{
TQMap<TQString,TQString> map = loadPrinttoolCfgFile(entry->arg("sd")+"/.config");
if (type == "SMB")
{
QStringList l = TQStringList::split('\\',map["share"],false);
if (map["workgroup"].isEmpty())
url = TQString::fromLatin1("smb://%1/%2").arg(l[0]).arg(l[1]);
else
url = TQString::fromLatin1("smb://%1/%2/%3").arg(map["workgroup"]).arg(l[0]).arg(l[1]);
url.setUser(map["user"]);
url.setPass(map["password"]);
}
else if (type == "DIRECT")
url = TQString::fromLatin1("socket://%1:%2").arg(map["printer_ip"]).arg(map["port"]);
else if (type == "NCP")
{
url = TQString::fromLatin1("ncp://%1/%2").arg(map["server"]).arg(map["queue"]);
url.setUser(map["user"]);
url.setPass(map["password"]);
}
}
printer->setDevice(url);
return true;
}
else return false;
}
bool KMLpdManager::createPrinter(KMPrinter *printer)
{
// 1) create the printcap entry
PrintcapEntry *ent = findPrintcapEntry(printer->printerName());
if (!ent)
{
ent = new PrintcapEntry();
ent->m_name = printer->printerName();
}
else
{
if (!printer->driver() && printer->option("kde-driver") != "raw")
printer->setDriver(loadPrinterDriver(printer,true));
// remove it from current entries
ent = m_entries.take(ent->m_name);
ent->m_args.clear();
}
// Standard options
if (printer->device().protocol() == "lpd")
{
// remote lpd queue
ent->m_args["rm"] = printer->device().host();
ent->m_args["rp"] = printer->device().path().replace("/",TQString::fromLatin1(""));
ent->m_args["lpd_bounce"] = "true";
ent->m_comment = TQString::fromLatin1("##PRINTTOOL3## REMOTE");
}
ent->m_args["mx"] = (printer->option("mx").isEmpty() ? "#0" : printer->option("mx"));
ent->m_args["sh"] = TQString::null;
// create spool directory (if necessary) and update PrintcapEntry object
if (!createSpooldir(ent))
{
setErrorMsg(i18n("Unable to create spool directory %1 for printer %2.").arg(ent->arg("sd")).arg(ent->m_name));
delete ent;
return false;
}
if (!printer->driver() || printer->driver()->get("drtype") == "printtool")
if (!createPrinttoolEntry(printer,ent))
{
setErrorMsg(i18n("Unable to save information for printer <b>%1</b>.").arg(printer->printerName()));
delete ent;
return false;
}
// 2) write the printcap file
m_entries.insert(ent->m_name,ent);
if (!writePrinters())
return false;
// 3) save the printer driver (if any)
if (printer->driver())
{
if (!savePrinterDriver(printer,printer->driver()))
{
m_entries.remove(ent->m_name);
writePrinters();
return false;
}
}
// 4) change permissions of spool directory
TQCString cmd = "chmod -R o-rwx,g+rwX ";
cmd += TQFile::encodeName(KProcess::quote(ent->arg("sd")));
cmd += "&& chown -R lp.lp ";
cmd += TQFile::encodeName(KProcess::quote(ent->arg("sd")));
if (system(cmd.data()) != 0)
{
setErrorMsg(i18n("Unable to set correct permissions on spool directory %1 for printer <b>%2</b>.").arg(ent->arg("sd")).arg(ent->m_name));
return false;
}
return true;
}
bool KMLpdManager::removePrinter(KMPrinter *printer)
{
PrintcapEntry *ent = findPrintcapEntry(printer->printerName());
if (ent)
{
ent = m_entries.take(printer->printerName());
if (!writePrinters())
{
m_entries.insert(ent->m_name,ent);
return false;
}
TQCString cmd = "rm -rf ";
cmd += TQFile::encodeName(KProcess::quote(ent->arg("sd")));
system(cmd.data());
delete ent;
return true;
}
else
return false;
}
bool KMLpdManager::enablePrinter(KMPrinter *printer, bool state)
{
KPipeProcess proc;
QString cmd = programName(0);
cmd += " ";
cmd += state ? "up" : "down";
cmd += " ";
cmd += KProcess::quote(printer->printerName());
if (proc.open(cmd))
{
QTextStream t(&proc);
QString buffer;
while (!t.eof())
buffer.append(t.readLine());
if (buffer.startsWith("?Privilege"))
{
setErrorMsg(i18n("Permission denied: you must be root."));
return false;
}
return true;
}
else
{
setErrorMsg(i18n("Unable to execute command \"%1\".").arg(cmd));
return false;
}
}
bool KMLpdManager::enablePrinter(KMPrinter *printer)
{
return enablePrinter(printer,true);
}
bool KMLpdManager::disablePrinter(KMPrinter *printer)
{
return enablePrinter(printer,false);
}
void KMLpdManager::listPrinters()
{
m_entries.clear();
loadPrintcapFile(TQString::fromLatin1("%1/etc/printcap").arg(lpdprefix));
TQDictIterator<PrintcapEntry> it(m_entries);
for (;it.current();++it)
{
KMPrinter *printer = it.current()->createPrinter();
addPrinter(printer);
}
checkStatus();
}
TQString KMLpdManager::programName(int f)
{
KConfig *conf = KMFactory::self()->printConfig();
conf->setGroup("LPD");
switch (f)
{
case 0: return conf->readPathEntry("LpdCommand","/usr/sbin/lpc");
case 1: return conf->readPathEntry("LpdQueue","lpq");
case 2: return conf->readPathEntry("LpdRemove","lprm");
}
return TQString::null;
}
void KMLpdManager::checkStatus()
{
KPipeProcess proc;
QString cmd = programName(0) + " status all";
if (proc.open(cmd))
{
QTextStream t(&proc);
QString line;
KMPrinter *printer(0);
int p(-1);
while (!t.eof())
{
line = t.readLine().stripWhiteSpace();
if (line.isEmpty())
continue;
if ((p=line.find(':')) != -1)
printer = findPrinter(line.left(p));
else if (line.startsWith("printing") && printer)
printer->setState(line.find("enabled") != -1 ? KMPrinter::Idle : KMPrinter::Stopped);
else if (line.find("entries") != -1 && printer)
if (!line.startsWith("no") && printer->state() == KMPrinter::Idle)
printer->setState(KMPrinter::Processing);
}
}
}
bool KMLpdManager::writePrinters()
{
if (!writePrintcapFile(TQString::fromLatin1("%1/etc/printcap").arg(lpdprefix)))
{
setErrorMsg(i18n("Unable to write printcap file."));
return false;
}
return true;
}
void KMLpdManager::loadPrintcapFile(const TQString& filename)
{
QFile f(filename);
if (f.exists() && f.open(IO_ReadOnly))
{
QTextStream t(&f);
QString line, comment;
PrintcapEntry *entry;
while (!t.eof())
{
line = getPrintcapLine(t,&comment);
if (line.isEmpty())
continue;
entry = new PrintcapEntry;
if (entry->readLine(line))
{
m_entries.insert(entry->m_name,entry);
entry->m_comment = comment;
}
else
{
delete entry;
break;
}
}
}
}
bool KMLpdManager::writePrintcapFile(const TQString& filename)
{
QFile f(filename);
if (f.open(IO_WriteOnly))
{
QTextStream t(&f);
t << "# File generated by KDE print (LPD plugin).\n#Don't edit by hand." << endl << endl;
TQDictIterator<PrintcapEntry> it(m_entries);
for (;it.current();++it)
it.current()->writeEntry(t);
return true;
}
return false;
}
PrinttoolEntry* KMLpdManager::findPrinttoolEntry(const TQString& name)
{
if (m_ptentries.count() == 0)
loadPrinttoolDb(driverDirectory()+"/printerdb");
PrinttoolEntry *ent = m_ptentries.find(name);
if (!ent)
setErrorMsg(i18n("Couldn't find driver <b>%1</b> in printtool database.").arg(name));
return ent;
}
void KMLpdManager::loadPrinttoolDb(const TQString& filename)
{
QFile f(filename);
if (f.exists() && f.open(IO_ReadOnly))
{
QTextStream t(&f);
PrinttoolEntry *entry = new PrinttoolEntry;
while (entry->readEntry(t))
{
m_ptentries.insert(entry->m_name,entry);
entry = new PrinttoolEntry;
}
delete entry;
}
}
DrMain* KMLpdManager::loadDbDriver(KMDBEntry *entry)
{
QString ptdbfilename = driverDirectory() + "/printerdb";
if (entry->file == ptdbfilename)
{
PrinttoolEntry *ptentry = findPrinttoolEntry(entry->modelname);
if (ptentry)
{
DrMain *dr = ptentry->createDriver();
return dr;
}
}
return NULL;
}
PrintcapEntry* KMLpdManager::findPrintcapEntry(const TQString& name)
{
PrintcapEntry *ent = m_entries.find(name);
if (!ent)
setErrorMsg(i18n("Couldn't find printer <b>%1</b> in printcap file.").arg(name));
return ent;
}
DrMain* KMLpdManager::loadPrinterDriver(KMPrinter *printer, bool config)
{
PrintcapEntry *entry = findPrintcapEntry(printer->name());
if (!entry)
return NULL;
// check for printtool driver (only for configuration)
QString sd = entry->arg("sd"), dr(entry->comment(7));
if (TQFile::exists(sd+"/postscript.cfg") && config && !dr.isEmpty())
{
TQMap<TQString,TQString> map = loadPrinttoolCfgFile(sd+"/postscript.cfg");
PrinttoolEntry *ptentry = findPrinttoolEntry(dr);
if (!ptentry)
return NULL;
DrMain *dr = ptentry->createDriver();
dr->setOptions(map);
map = loadPrinttoolCfgFile(sd+"/general.cfg");
dr->setOptions(map);
map = loadPrinttoolCfgFile(sd+"/textonly.cfg");
dr->setOptions(map);
return dr;
}
// default
if (entry->m_comment.startsWith("##PRINTTOOL3##"))
setErrorMsg(i18n("No driver found (raw printer)"));
else
setErrorMsg(i18n("Printer type not recognized."));
return NULL;
}
bool KMLpdManager::checkGsDriver(const TQString& gsdriver)
{
if (gsdriver == "ppa" || gsdriver == "POSTSCRIPT" || gsdriver == "TEXT")
return true;
else if (!m_gschecker->checkGsDriver(gsdriver))
{
setErrorMsg(i18n("The driver device <b>%1</b> is not compiled in your GhostScript distribution. Check your installation or use another driver.").arg(gsdriver));
return false;
}
return true;
}
TQMap<TQString,TQString> KMLpdManager::loadPrinttoolCfgFile(const TQString& filename)
{
QFile f(filename);
TQMap<TQString,TQString> map;
if (f.exists() && f.open(IO_ReadOnly))
{
QTextStream t(&f);
QString line, name, val;
int p(-1);
while (!t.eof())
{
line = getPrintcapLine(t);
if (line.isEmpty())
break;
if (line.startsWith("export "))
line.replace(0,7,"");
if ((p=line.find('=')) != -1)
{
name = line.left(p);
val = line.right(line.length()-p-1);
val.replace("\"","");
val.replace("'","");
if (!name.isEmpty() && !val.isEmpty())
map[name] = val;
}
}
}
return map;
}
bool KMLpdManager::savePrinttoolCfgFile(const TQString& templatefile, const TQString& dirname, const TQMap<TQString,TQString>& options)
{
// defines input and output file
QString fname = TQFileInfo(templatefile).fileName();
fname.replace(TQRegExp("\\.in$"),TQString::fromLatin1(""));
QFile fin(templatefile);
QFile fout(dirname + "/" + fname);
if (fin.exists() && fin.open(IO_ReadOnly) && fout.open(IO_WriteOnly))
{
QTextStream tin(&fin), tout(&fout);
QString line, name;
int p(-1);
while (!tin.eof())
{
line = tin.readLine().stripWhiteSpace();
if (line.isEmpty() || line[0] == '#')
{
tout << line << endl;
continue;
}
if (line.startsWith("export "))
{
tout << "export ";
line.replace(0,7,TQString::fromLatin1(""));
}
if ((p=line.find('=')) != -1)
{
name = line.left(p);
tout << name << '=' << options[name] << endl;
}
}
return true;
}
else return false;
}
bool KMLpdManager::savePrinterDriver(KMPrinter *printer, DrMain *driver)
{
// To be able to save a printer driver, a printcap entry MUST exist.
// We can then retrieve the spool directory from it.
QString spooldir;
PrintcapEntry *ent = findPrintcapEntry(printer->printerName());
if (!ent)
return false;
spooldir = ent->arg("sd");
if (driver->get("drtype") == "printtool" && !spooldir.isEmpty())
{
TQMap<TQString,TQString> options;
driver->getOptions(options,true);
// add some standard options
options["DESIRED_TO"] = "ps";
options["PRINTER_TYPE"] = ent->comment(2); // get type from printcap entry (works in anycases)
options["PS_SEND_EOF"] = "NO";
if (!checkGsDriver(options["GSDEVICE"]))
return false;
QString resol(options["RESOLUTION"]), color(options["COLOR"]);
// update entry comment to make printtool happy and save printcap file
ent->m_comment = TQString::fromLatin1("##PRINTTOOL3## %1 %2 %3 %4 {} {%5} %6 {}").arg(options["PRINTER_TYPE"]).arg(options["GSDEVICE"]).arg((resol.isEmpty() ? TQString::fromLatin1("NAxNA") : resol)).arg(options["PAPERSIZE"]).arg(driver->name()).arg((color.isEmpty() ? TQString::fromLatin1("Default") : color.right(color.length()-15)));
ent->m_args["if"] = spooldir+TQString::fromLatin1("/filter");
if (!writePrinters())
return false;
// write various driver files using templates
TQCString cmd = "cp ";
cmd += TQFile::encodeName(KProcess::quote(driverDirectory()+"/master-filter"));
cmd += " ";
cmd += TQFile::encodeName(KProcess::quote(spooldir + "/filter"));
if (system(cmd.data()) == 0 &&
savePrinttoolCfgFile(driverDirectory()+"/general.cfg.in",spooldir,options) &&
savePrinttoolCfgFile(driverDirectory()+"/postscript.cfg.in",spooldir,options) &&
savePrinttoolCfgFile(driverDirectory()+"/textonly.cfg.in",spooldir,options))
return true;
setErrorMsg(i18n("Unable to write driver associated files in spool directory."));
}
return false;
}
bool KMLpdManager::createPrinttoolEntry(KMPrinter *printer, PrintcapEntry *entry)
{
KURL dev(printer->device());
QString prot = dev.protocol(), sd(entry->arg("sd"));
entry->m_comment = TQString::fromLatin1("##PRINTTOOL3## %1").arg(ptPrinterType(printer));
if (prot == "smb" || prot == "ncp" || prot == "socket")
{
entry->m_args["af"] = sd+TQString::fromLatin1("/acct");
QFile f(sd+TQString::fromLatin1("/.config"));
if (f.open(IO_WriteOnly))
{
QTextStream t(&f);
if (prot == "socket")
{
t << "printer_ip=" << dev.host() << endl;
t << "port=" << dev.port() << endl;
entry->m_args["if"] = driverDirectory()+TQString::fromLatin1("/directprint");
}
else if (prot == "smb")
{
QStringList l = TQStringList::split('/',dev.path(),false);
if (l.count() == 2)
{
t << "share='\\\\" << l[0] << '\\' << l[1] << '\'' << endl;
}
else if (l.count() == 1)
{
t << "share='\\\\" << dev.host() << '\\' << l[0] << '\'' << endl;
}
t << "hostip=" << endl;
t << "user='" << dev.user() << '\'' << endl;
t << "password='" << dev.pass() << '\'' << endl;
t << "workgroup='" << (l.count() == 2 ? dev.host() : TQString::fromLatin1("")) << '\'' << endl;
entry->m_args["if"] = driverDirectory()+TQString::fromLatin1("/smbprint");
}
else if (prot == "ncp")
{
t << "server=" << dev.host() << endl;
t << "queue=" << dev.path().replace("/",TQString::fromLatin1("")) << endl;
t << "user=" << dev.user() << endl;
t << "password=" << dev.pass() << endl;
entry->m_args["if"] = driverDirectory()+TQString::fromLatin1("/ncpprint");
}
}
else return false;
entry->m_args["lp"] = TQString::fromLatin1("/dev/null");
}
else if (prot != "lpd")
entry->m_args["lp"] = dev.path();
return true;
}
bool KMLpdManager::createSpooldir(PrintcapEntry *entry)
{
// first check if it has a "sd" defined
if (entry->arg("sd").isEmpty())
entry->m_args["sd"] = TQString::fromLatin1("/var/spool/lpd/")+entry->m_name;
QString sd = entry->arg("sd");
if (!KStandardDirs::exists(sd))
{
if (!KStandardDirs::makeDir(sd,0750))
return false;
struct passwd *lp_pw = getpwnam("lp");
if (lp_pw && chown(TQFile::encodeName(sd),lp_pw->pw_uid,lp_pw->pw_gid) != 0)
return false;
}
return true;
}
bool KMLpdManager::validateDbDriver(KMDBEntry *entry)
{
PrinttoolEntry *ptentry = findPrinttoolEntry(entry->modelname);
return (ptentry && checkGsDriver(ptentry->m_gsdriver));
}
//************************************************************************************************
TQString ptPrinterType(KMPrinter *p)
{
QString type, prot = p->device().protocol();
if (prot == "lpd") type = "REMOTE";
else if (prot == "smb") type = "SMB";
else if (prot == "ncp") type = "NCP";
else if (prot == "socket") type = "DIRECT";
else type = "LOCAL";
return type;
}