apt-extracttemplates stuff from debconf

Author: tausq
Date: 2001-02-23 04:28:49 GMT
apt-extracttemplates stuff from debconf

diff --git a/cmdline/apt-extracttemplates.cc b/cmdline/apt-extracttemplates.cc
new file mode 100644 (file)
index 0000000..b7af030
--- /dev/null
+++ b/cmdline/apt-extracttemplates.cc
@@ -0,0 +1,142 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#define _GNU_SOURCE
+#include <getopt.h>
+#include <wait.h>
+#include <fstream.h>
+
+#include <apt-pkg/init.h>
+#if APT_PKG_MAJOR >= 3
+#define APT_COMPATIBILITY 986
+#include <apt-pkg/debversion.h>
+#endif
+#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"
+       
+#define TMPDIR "/var/lib/debconf/"
+#define STR(x) (x ? x : "")
+//#define TMPDIR "tmp/"
+
+void help(void)
+{
+       fprintf(stderr, "apt-extracttemplates deb [deb]\n");
+       exit(0);
+}
+
+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++);
+
+       if (data == NULL) data = "";
+
+       ofstream ofs(fn);
+       if (!ofs) return NULL;
+       ofs << data;
+       ofs.close();
+       return strdup(fn);
+}
+
+void writeconfig(const DebFile &file)
+{
+       char *templatefile = writefile("template", file.Template);
+       char *configscript = writefile("config", file.Config);
+
+       if (templatefile == 0 || configscript == 0)
+       {
+               fprintf(stderr, "Cannot write config script or templates\n");
+               return;
+       }
+       printf("%s %s %s %s\n",
+               STR(file.Package),      // Package
+               STR(file.Version),      // Version
+               templatefile,   // Template
+               configscript    // Config
+       );
+}
+
+void init(MMap *&Map, pkgCache *&Cache)
+{
+       // Initialize the apt cache
+       if (pkgInitConfig(*_config) == false || pkgInitSystem(*_config, _system) == false)
+       {
+               fprintf(stderr, "Cannot initialize apt cache\n");
+               return;
+       }
+       pkgSourceList List;
+       List.ReadMainList();
+       OpProgress Prog;
+       pkgMakeStatusCache(List,Prog,&Map,true);
+       Cache = new pkgCache(Map);
+}
+
+int main(int argc, char **argv, char **env)
+{
+       int idx = 0;
+       char **debs = 0;
+       int numdebs = 0;
+       MMap *Map = 0;
+       const char *debconfver = NULL;
+
+       init(Map, DebFile::Cache);
+       if (Map == 0 || DebFile::Cache == 0)
+       {
+               fprintf(stderr, "Cannot initialize APT cache\n");
+               return 1;
+       }
+
+       debconfver = DebFile::GetInstalledVer("debconf");
+       
+       if (debconfver == NULL) 
+       {
+               fprintf(stderr, "Cannot get debconf version. Is debconf installed?\n");
+               return 1;
+       }
+
+       numdebs = argc - 1;
+       debs = new char *[numdebs];
+       memcpy(debs, &argv[1], sizeof(char *) * numdebs);
+
+       if (numdebs < 1) 
+       {
+               fprintf(stderr, "apt-extracttemplates foo.deb [...]\n");
+               return 0;
+       }
+
+       for (idx = 0; idx < numdebs; idx++)
+       {
+               DebFile file(debs[idx]);
+               if (file.Go() == false) 
+               {
+                       fprintf(stderr, "Cannot read %s\n", debs[idx]);
+                       continue;
+               }
+               if (file.Template != 0 && file.ParseInfo() == true)
+               {
+                       if (file.DepVer != 0 && *file.DepVer != 0 &&
+                           pkgCheckDep(file.DepVer, 
+                                       debconfver, file.DepOp) == false) 
+                               continue;
+                       if (file.PreDepVer != 0 && *file.PreDepVer != 0 &&
+                           pkgCheckDep(file.PreDepVer, 
+                                       debconfver, file.PreDepOp) == false) 
+                               continue;
+
+                       writeconfig(file);
+               }
+       }
+       
+
+       delete Map;
+       delete DebFile::Cache;
+
+       return 0;
+}

diff --git a/cmdline/debfile.cc b/cmdline/debfile.cc
new file mode 100644 (file)
index 0000000..0aa2462
--- /dev/null
+++ b/cmdline/debfile.cc
@@ -0,0 +1,304 @@
+#include <stdio.h>
+#include <apt-pkg/tagfile.h>
+#include <apt-pkg/extracttar.h>
+#include <apt-pkg/arfile.h>
+#include <apt-pkg/pkgcache.h>
+
+#include "debfile.h"
+
+pkgCache *DebFile::Cache = 0;
+
+DebFile::DebFile(const char *debfile)
+       : File(debfile, FileFd::ReadOnly), Control(0), Package(0), Version(0), DepVer(0), PreDepVer(0), DepOp(0), PreDepOp(0), Config(0), Template(0), Which(None)
+{
+}
+
+DebFile::~DebFile()
+{
+       delete [] Control;
+       delete [] Config;
+       delete [] Template;
+}
+
+char *DebFile::GetInstalledVer(const char *package)
+{
+       char *ver = 0;
+
+       pkgCache::PkgIterator Pkg = Cache->FindPkg(package);
+       if (Pkg.end() == false) 
+       {
+               pkgCache::VerIterator V = Pkg.CurrentVer();
+               if (V.end() == false) 
+               {
+                       ver = strdup(V.VerStr());
+               }
+       }
+
+       return ver;
+}
+
+bool DebFile::Go()
+{
+       ARArchive AR(File);
+       const ARArchive::Member *Member = AR.FindMember("control.tar.gz");
+       if (Member == 0)
+       {
+               fprintf(stderr, "This is not a valid DEB archive.\n");
+               return false;
+       }
+       
+       if (File.Seek(Member->Start) == false)
+       {
+               return false;
+       }
+
+       ExtractTar Tar(File, Member->Size);
+       return Tar.Go(*this);
+}
+
+bool DebFile::DoItem(Item &I, int &Fd)
+{
+       if (strcmp(I.Name, "control") == 0)
+       {
+               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;
+       }
+       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 
+       {
+               Fd = -1;
+       }
+       return true;
+}
+
+bool DebFile::Process(Item &I, const unsigned char *data, 
+               unsigned long size, unsigned long pos)
+{
+       switch (Which)
+       {
+       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 */ ;
+       }
+       return true;
+}
+
+bool DebFile::ParseInfo()
+{
+       if (Control == NULL) return false;
+       pkgTagSection Section;
+       Section.Scan(Control, ControlLen);
+
+       const char *pkgtmp = Section.FindS("Package").c_str();
+       Package = CopyString(pkgtmp, strlen(pkgtmp));
+       Version = GetInstalledVer(Package);
+
+       const char *Start, *Stop;
+       if (Section.Find("Depends", Start, Stop) == true)
+       {
+               while (1)
+               {
+                       char *P = 0, *V = 0;
+                       unsigned int Op;
+                       Start = ParseDepends(Start, Stop, P, V, Op);
+                       if (Start == 0) return false;
+                       if (strcmp(P, "debconf") == 0)
+                       {
+                               DepVer = V;
+                               DepOp = Op;
+                               delete[] P;
+                               break;
+                       }
+                       else
+                       {
+                               delete[] P;
+                               delete[] V;
+                       }
+                       if (Start == Stop) break;
+               }
+       }
+       
+       if (Section.Find("Pre-Depends", Start, Stop) == true)
+       {
+               while (1)
+               {
+                       char *P = 0, *V = 0;
+                       unsigned int Op;
+                       Start = ParseDepends(Start, Stop, P, V, Op);
+                       if (Start == 0) return false;
+                       if (strcmp(P, "debconf") == 0)
+                       {
+                               PreDepVer = V;
+                               PreDepOp = Op;
+                               delete[] P;
+                               break;
+                       }
+                       else
+                       {
+                               delete[] P;
+                               delete[] V;
+                       }
+                       if (Start == Stop) break;
+               }
+       }
+       
+       return true;
+}
+
+char *DebFile::CopyString(const char *start, unsigned int len)
+{
+       char *s = new char[len + 1];
+       s[len] = 0;
+       memcpy(s, start, len);
+       return s;
+}
+
+const char *DebFile::ParseDepends(const char *Start,const char *Stop,
+                               char *&Package, char *&Ver,
+                               unsigned int &Op)
+{
+   // Strip off leading space
+   for (;Start != Stop && isspace(*Start) != 0; Start++);
+   
+   // Parse off the package name
+   const char *I = Start;
+   for (;I != Stop && isspace(*I) == 0 && *I != '(' && *I != ')' &&
+       *I != ',' && *I != '|'; I++);
+   
+   // Malformed, no '('
+   if (I != Stop && *I == ')')
+      return 0;
+
+   if (I == Start)
+      return 0;
+   
+   // Stash the package name
+   Package = CopyString(Start, I - Start);
+   
+   // Skip white space to the '('
+   for (;I != Stop && isspace(*I) != 0 ; I++);
+   
+   // Parse a version
+   if (I != Stop && *I == '(')
+   {
+      // Skip the '('
+      for (I++; I != Stop && isspace(*I) != 0 ; I++);
+      if (I + 3 >= Stop)
+        return 0;
+      
+      // Determine the operator
+      switch (*I)
+      {
+        case '<':
+        I++;
+        if (*I == '=')
+        {
+           I++;
+           Op = pkgCache::Dep::LessEq;
+           break;
+        }
+        
+        if (*I == '<')
+        {
+           I++;
+           Op = pkgCache::Dep::Less;
+           break;
+        }
+        
+        // < is the same as <= and << is really Cs < for some reason
+        Op = pkgCache::Dep::LessEq;
+        break;
+        
+        case '>':
+        I++;
+        if (*I == '=')
+        {
+           I++;
+           Op = pkgCache::Dep::GreaterEq;
+           break;
+        }
+        
+        if (*I == '>')
+        {
+           I++;
+           Op = pkgCache::Dep::Greater;
+           break;
+        }
+        
+        // > is the same as >= and >> is really Cs > for some reason
+        Op = pkgCache::Dep::GreaterEq;
+        break;
+        
+        case '=':
+        Op = pkgCache::Dep::Equals;
+        I++;
+        break;
+        
+        // HACK around bad package definitions
+        default:
+        Op = pkgCache::Dep::Equals;
+        break;
+      }
+      
+      // Skip whitespace
+      for (;I != Stop && isspace(*I) != 0; I++);
+      Start = I;
+      for (;I != Stop && *I != ')'; I++);
+      if (I == Stop || Start == I)
+        return 0;     
+      
+      // Skip trailing whitespace
+      const char *End = I;
+      for (; End > Start && isspace(End[-1]); End--);
+      
+      Ver = CopyString(Start, End - Start);
+      I++;
+   }
+   else
+   {
+      Ver = CopyString("", 0);
+      Op = pkgCache::Dep::NoOp;
+   }
+   
+   // Skip whitespace
+   for (;I != Stop && isspace(*I) != 0; I++);
+
+   if (I != Stop && *I == '|')
+      Op |= pkgCache::Dep::Or;
+   
+   if (I == Stop || *I == ',' || *I == '|')
+   {
+      if (I != Stop)
+        for (I++; I != Stop && isspace(*I) != 0; I++);
+      return I;
+   }
+   
+   return 0;
+}

diff --git a/cmdline/debfile.h b/cmdline/debfile.h
new file mode 100644 (file)
index 0000000..b2db3c8
--- /dev/null
+++ b/cmdline/debfile.h
@@ -0,0 +1,45 @@
+#ifndef _debfile_H
+#define _debfile_H
+
+#include <apt-pkg/fileutl.h>
+#include <apt-pkg/pkgcache.h>
+#include <apt-pkg/dirstream.h>
+
+class DebFile : public pkgDirStream
+{
+       const char *ParseDepends(const char *Start,const char *Stop,
+                               char *&Package, char *&Ver,
+                               unsigned int &Op);
+
+       char *CopyString(const char *start, unsigned int len);
+
+       FileFd File;
+       unsigned long Size;
+       char *Control;
+       unsigned long ControlLen;
+       
+public:
+       DebFile(const char *FileName);
+       ~DebFile();
+       bool DoItem(Item &I, int &fd);
+       bool Process(pkgDirStream::Item &I, const unsigned char *data, 
+               unsigned long size, unsigned long pos);
+
+       bool Go();
+       bool ParseInfo();
+
+       static char *GetInstalledVer(const char *package);
+
+       char *Package;
+       char *Version;
+       char *DepVer, *PreDepVer;
+       unsigned int DepOp, PreDepOp;
+
+       char *Config;
+       char *Template;
+
+       static pkgCache *Cache;
+       enum { None, IsControl, IsConfig, IsTemplate } Which;
+};
+
+#endif

diff --git a/cmdline/makefile b/cmdline/makefile
index c0647cead0021428a96391e712ee8f509fe666cc..0f19d8e596200a3432a66eef099473a562f8e15d 100644 (file)
--- a/cmdline/makefile
+++ b/cmdline/makefile
@@ -39,3 +39,10 @@ SLIBS = -lapt-pkg
 LIB_MAKES = apt-pkg/makefile
 SOURCE = apt-sortpkgs.cc
 include $(PROGRAM_H)
 LIB_MAKES = apt-pkg/makefile
 SOURCE = apt-sortpkgs.cc
 include $(PROGRAM_H)

+
+# The apt-extracttemplates program
+PROGRAM=apt-extracttemplates
+SLIBS = -lapt-pkg -lapt-inst
+LIB_MAKES = apt-pkg/makefile
+SOURCE = apt-extracttemplates.cc debfile.cc
+include $(PROGRAM_H)