-#include <stdio.h>
-#include <string.h>
-#include <stdlib.h>
-#include <unistd.h>
-#define _GNU_SOURCE
-#include <getopt.h>
-#include <wait.h>
-#include <fstream.h>
+// -*- mode: cpp; mode: fold -*-
+// Description /*{{{*/
+// $Id: apt-extracttemplates.cc,v 1.15 2003/07/26 00:00:11 mdz Exp $
+/* ######################################################################
+
+ APT Extract Templates - Program to extract debconf config and template
+ files
+ This is a simple program to extract config and template information
+ from Debian packages. It can be used to speed up the preconfiguration
+ process for debconf-enabled packages
+
+ ##################################################################### */
+ /*}}}*/
+// Include Files /*{{{*/
#include <apt-pkg/init.h>
-#if APT_PKG_MAJOR >= 3
-#define APT_COMPATIBILITY 986
-#include <apt-pkg/debversion.h>
-#endif
+#include <apt-pkg/cmndline.h>
#include <apt-pkg/pkgcache.h>
#include <apt-pkg/configuration.h>
#include <apt-pkg/progress.h>
#include <apt-pkg/sourcelist.h>
#include <apt-pkg/pkgcachegen.h>
#include <apt-pkg/version.h>
-#include "debfile.h"
+#include <apt-pkg/tagfile.h>
+#include <apt-pkg/extracttar.h>
+#include <apt-pkg/arfile.h>
+#include <apt-pkg/deblistparser.h>
+#include <apt-pkg/error.h>
+#include <apt-pkg/strutl.h>
+#include <apt-pkg/fileutl.h>
-#define TMPDIR "/var/lib/debconf/"
-#define STR(x) (x ? x : "")
-//#define TMPDIR "tmp/"
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <fstream>
-void help(void)
-{
- fprintf(stderr, "apt-extracttemplates deb [deb]\n");
- exit(0);
-}
+#include <locale.h>
+#include <config.h>
+#include <apti18n.h>
+#include "apt-extracttemplates.h"
+ /*}}}*/
-char *writefile(const char *prefix, const char *data)
-{
- char fn[512];
- static int i;
- snprintf(fn, sizeof(fn), "%s%s.%u%d", TMPDIR, prefix, getpid(), i++);
+using namespace std;
- if (data == NULL) data = "";
+#define TMPDIR "/tmp"
- ofstream ofs(fn);
- if (!ofs) return NULL;
- ofs << data;
- ofs.close();
- return strdup(fn);
-}
+pkgCache *DebFile::Cache = 0;
-void writeconfig(const DebFile &file)
+// DebFile::DebFile - Construct the DebFile object /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+DebFile::DebFile(const char *debfile)
+ : File(debfile, FileFd::ReadOnly), Control(0), DepOp(0),
+ PreDepOp(0), Config(0), Template(0), Which(None)
{
- char *templatefile = writefile("template", file.Template);
- char *configscript = writefile("config", file.Config);
+}
+ /*}}}*/
+// DebFile::~DebFile - Destruct the DebFile object /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+DebFile::~DebFile()
+{
+ delete [] Control;
+ delete [] Config;
+ delete [] Template;
+}
+ /*}}}*/
+// DebFile::GetInstalledVer - Find out the installed version of a pkg /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+string DebFile::GetInstalledVer(const string &package)
+{
+ pkgCache::PkgIterator Pkg = Cache->FindPkg(package);
+ if (Pkg.end() == false)
+ {
+ pkgCache::VerIterator V = Pkg.CurrentVer();
+ if (V.end() == false)
+ {
+ return V.VerStr();
+ }
+ }
- if (templatefile == 0 || configscript == 0)
+ return string();
+}
+ /*}}}*/
+// DebFile::Go - Start extracting a debian package /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool DebFile::Go()
+{
+ ARArchive AR(File);
+ if (_error->PendingError() == true)
+ return false;
+
+ const ARArchive::Member *Member = AR.FindMember("control.tar.gz");
+ if (Member == 0)
+ return _error->Error(_("%s not a valid DEB package."),File.Name().c_str());
+
+ if (File.Seek(Member->Start) == false)
+ return false;
+ ExtractTar Tar(File, Member->Size,"gzip");
+ return Tar.Go(*this);
+}
+ /*}}}*/
+// DebFile::DoItem examine element in package and mark /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool DebFile::DoItem(Item &I, int &Fd)
+{
+ if (strcmp(I.Name, "control") == 0)
{
- fprintf(stderr, "Cannot write config script or templates\n");
- return;
+ delete [] Control;
+ Control = new char[I.Size+1];
+ Control[I.Size] = 0;
+ Which = IsControl;
+ ControlLen = I.Size;
+ // make it call the Process method below. this is so evil
+ Fd = -2;
}
- printf("%s %s %s %s\n",
- STR(file.Package), // Package
- STR(file.Version), // Version
- templatefile, // Template
- configscript // Config
- );
+ else if (strcmp(I.Name, "config") == 0)
+ {
+ delete [] Config;
+ Config = new char[I.Size+1];
+ Config[I.Size] = 0;
+ Which = IsConfig;
+ Fd = -2;
+ }
+ else if (strcmp(I.Name, "templates") == 0)
+ {
+ delete [] Template;
+ Template = new char[I.Size+1];
+ Template[I.Size] = 0;
+ Which = IsTemplate;
+ Fd = -2;
+ }
+ else
+ {
+ // Ignore it
+ Fd = -1;
+ }
+ return true;
}
-
-void init(MMap *&Map, pkgCache *&Cache)
+ /*}}}*/
+// DebFile::Process examine element in package and copy /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool DebFile::Process(Item &I, const unsigned char *data,
+ unsigned long size, unsigned long pos)
{
- // Initialize the apt cache
- if (pkgInitConfig(*_config) == false || pkgInitSystem(*_config, _system) == false)
+ switch (Which)
{
- fprintf(stderr, "Cannot initialize apt cache\n");
- return;
+ case IsControl:
+ memcpy(Control + pos, data, size);
+ break;
+ case IsConfig:
+ memcpy(Config + pos, data, size);
+ break;
+ case IsTemplate:
+ memcpy(Template + pos, data, size);
+ break;
+ default: /* throw it away */ ;
}
- pkgSourceList List;
- List.ReadMainList();
- OpProgress Prog;
- pkgMakeStatusCache(List,Prog,&Map,true);
- Cache = new pkgCache(Map);
+ return true;
}
-
-int main(int argc, char **argv, char **env)
+ /*}}}*/
+// DebFile::ParseInfo - Parse control file for dependency info /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool DebFile::ParseInfo()
{
- int idx = 0;
- char **debs = 0;
- int numdebs = 0;
- MMap *Map = 0;
- const char *debconfver = NULL;
+ if (Control == NULL) return false;
+
+ pkgTagSection Section;
+ Section.Scan(Control, ControlLen);
- init(Map, DebFile::Cache);
- if (Map == 0 || DebFile::Cache == 0)
+ Package = Section.FindS("Package");
+ Version = GetInstalledVer(Package);
+
+ const char *Start, *Stop;
+ if (Section.Find("Depends", Start, Stop) == true)
{
- fprintf(stderr, "Cannot initialize APT cache\n");
- return 1;
+ while (1)
+ {
+ string P, V;
+ unsigned int Op;
+ Start = debListParser::ParseDepends(Start, Stop, P, V, Op);
+ if (Start == 0) return false;
+ if (P == "debconf")
+ {
+ DepVer = V;
+ DepOp = Op;
+ break;
+ }
+ if (Start == Stop) break;
+ }
}
-
- debconfver = DebFile::GetInstalledVer("debconf");
- if (debconfver == NULL)
+ if (Section.Find("Pre-Depends", Start, Stop) == true)
{
- fprintf(stderr, "Cannot get debconf version. Is debconf installed?\n");
- return 1;
+ while (1)
+ {
+ string P, V;
+ unsigned int Op;
+ Start = debListParser::ParseDepends(Start, Stop, P, V, Op);
+ if (Start == 0) return false;
+ if (P == "debconf")
+ {
+ PreDepVer = V;
+ PreDepOp = Op;
+ break;
+ }
+ if (Start == Stop) break;
+ }
}
+
+ return true;
+}
+ /*}}}*/
+// ShowHelp - show a short help text /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+int ShowHelp(void)
+{
+ ioprintf(cout,_("%s %s for %s %s compiled on %s %s\n"),PACKAGE,VERSION,
+ COMMON_OS,COMMON_CPU,__DATE__,__TIME__);
+
+ if (_config->FindB("version") == true)
+ return 0;
+
+ cout <<
+ _("Usage: apt-extracttemplates file1 [file2 ...]\n"
+ "\n"
+ "apt-extracttemplates is a tool to extract config and template info\n"
+ "from debian packages\n"
+ "\n"
+ "Options:\n"
+ " -h This help text\n"
+ " -t Set the temp dir\n"
+ " -c=? Read this configuration file\n"
+ " -o=? Set an arbitrary configuration option, eg -o dir::cache=/tmp\n");
+ return 0;
+}
+ /*}}}*/
+// WriteFile - write the contents of the passed string to a file /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+string WriteFile(const char *package, const char *prefix, const char *data)
+{
+ char fn[512];
+ static int i;
+ char *tempdir = NULL;
+
+ tempdir = getenv("TMPDIR");
+ if (tempdir == NULL)
+ tempdir = TMPDIR;
- numdebs = argc - 1;
- debs = new char *[numdebs];
- memcpy(debs, &argv[1], sizeof(char *) * numdebs);
+ snprintf(fn, sizeof(fn), "%s/%s.%s.%u%d",
+ _config->Find("APT::ExtractTemplates::TempDir", tempdir).c_str(),
+ package, prefix, getpid(), i++);
+ FileFd f;
+ if (data == NULL)
+ data = "";
- if (numdebs < 1)
+ if (!f.Open(fn, FileFd::WriteTemp, 0600))
{
- fprintf(stderr, "apt-extracttemplates foo.deb [...]\n");
- return 0;
+ _error->Errno("ofstream::ofstream",_("Unable to write to %s"),fn);
+ return string();
}
- for (idx = 0; idx < numdebs; idx++)
+ f.Write(data, strlen(data));
+ f.Close();
+ return fn;
+}
+ /*}}}*/
+// WriteConfig - write out the config data from a debian package file /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+void WriteConfig(const DebFile &file)
+{
+ string templatefile = WriteFile(file.Package.c_str(), "template", file.Template);
+ string configscript = WriteFile(file.Package.c_str(), "config", file.Config);
+
+ if (templatefile.empty() == true || configscript.empty() == true)
+ return;
+ cout << file.Package << " " << file.Version << " "
+ << templatefile << " " << configscript << endl;
+}
+ /*}}}*/
+// InitCache - initialize the package cache /*{{{*/
+// ---------------------------------------------------------------------
+/* */
+bool Go(CommandLine &CmdL)
+{
+ // Initialize the apt cache
+ MMap *Map = 0;
+ pkgSourceList List;
+ List.ReadMainList();
+ OpProgress Prog;
+ pkgMakeStatusCache(List,Prog,&Map,true);
+ if (Map == 0)
+ return false;
+ DebFile::Cache = new pkgCache(Map);
+ if (_error->PendingError() == true)
+ return false;
+
+ // Find out what version of debconf is currently installed
+ string debconfver = DebFile::GetInstalledVer("debconf");
+ if (debconfver.empty() == true)
+ return _error->Error( _("Cannot get debconf version. Is debconf installed?"));
+
+ // Process each package passsed in
+ for (unsigned int I = 0; I != CmdL.FileSize(); I++)
{
- DebFile file(debs[idx]);
- if (file.Go() == false)
+ // Will pick up the errors later..
+ DebFile file(CmdL.FileList[I]);
+ if (file.Go() == false)
{
- fprintf(stderr, "Cannot read %s\n", debs[idx]);
+ _error->Error("Prior errors apply to %s",CmdL.FileList[I]);
continue;
}
+
+ // Does the package have templates?
if (file.Template != 0 && file.ParseInfo() == true)
{
- if (file.DepVer != 0 && *file.DepVer != 0 &&
- pkgCheckDep(file.DepVer,
- debconfver, file.DepOp) == false)
+ // Check to make sure debconf dependencies are
+ // satisfied
+ // cout << "Check " << file.DepVer << ',' << debconfver << endl;
+ if (file.DepVer != "" &&
+ DebFile::Cache->VS->CheckDep(debconfver.c_str(),
+ file.DepOp,file.DepVer.c_str()
+ ) == false)
continue;
- if (file.PreDepVer != 0 && *file.PreDepVer != 0 &&
- pkgCheckDep(file.PreDepVer,
- debconfver, file.PreDepOp) == false)
+ if (file.PreDepVer != "" &&
+ DebFile::Cache->VS->CheckDep(debconfver.c_str(),
+ file.PreDepOp,file.PreDepVer.c_str()
+ ) == false)
continue;
- writeconfig(file);
+ WriteConfig(file);
}
}
delete Map;
delete DebFile::Cache;
+
+ return !_error->PendingError();
+}
+ /*}}}*/
+int main(int argc, const char **argv)
+{
+ CommandLine::Args Args[] = {
+ {'h',"help","help",0},
+ {'v',"version","version",0},
+ {'t',"tempdir","APT::ExtractTemplates::TempDir",CommandLine::HasArg},
+ {'c',"config-file",0,CommandLine::ConfigFile},
+ {'o',"option",0,CommandLine::ArbItem},
+ {0,0,0,0}};
+
+ // Set up gettext support
+ setlocale(LC_ALL,"");
+ textdomain(PACKAGE);
+
+ // Parse the command line and initialize the package library
+ CommandLine CmdL(Args,_config);
+ if (pkgInitConfig(*_config) == false ||
+ CmdL.Parse(argc,argv) == false ||
+ pkgInitSystem(*_config,_system) == false)
+ {
+ _error->DumpErrors();
+ return 100;
+ }
+
+ // See if the help should be shown
+ if (_config->FindB("help") == true ||
+ CmdL.FileSize() == 0)
+ return ShowHelp();
+
+ Go(CmdL);
+
+ // Print any errors or warnings found during operation
+ if (_error->empty() == false)
+ {
+ // This goes to stderr..
+ bool Errors = _error->PendingError();
+ _error->DumpErrors();
+ return Errors == true?100:0;
+ }
+
return 0;
}