Adding upstream version 2.6.1.upstream/2.6.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 208 insertions, 0 deletions

diff --git a/ftparchive/writer.h b/ftparchive/writer.h
new file mode 100644
index 0000000..9aa2168
--- /dev/null
+++ b/ftparchive/writer.h
@@ -0,0 +1,208 @@
+// -*- mode: cpp; mode: fold -*-
+// Description								/*{{{*/
+/* ######################################################################
+
+   Writer 
+   
+   The file writer classes. These write various types of output, sources,
+   packages and contents.
+   
+   ##################################################################### */
+									/*}}}*/
+#ifndef WRITER_H
+#define WRITER_H
+
+#include <apt-pkg/hashes.h>
+
+#include <iostream>
+#include <map>
+#include <set>
+#include <string>
+#include <utility>
+#include <vector>
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/types.h>
+
+#include "apt-ftparchive.h"
+#include "cachedb.h"
+#include "contents.h"
+#include "override.h"
+
+using std::string;
+using std::cout;
+using std::endl;
+using std::vector;
+using std::map;
+
+class FTWScanner
+{
+   protected:
+   vector<string> Patterns;
+   vector<std::pair<string, bool>> FilesToProcess;
+   string Arch;
+   bool IncludeArchAll;
+   const char *OriginalPath;
+   bool ErrorPrinted;
+
+   // Stuff for the delinker
+   bool NoLinkAct;
+
+   static FTWScanner *Owner;
+   static int ScannerFTW(const char *File,const struct stat *sb,int Flag);
+   static int ScannerFile(const char *const File, bool const ReadLink);
+   static int ProcessFile(const char *const File, bool const ReadLink);
+
+   bool Delink(string &FileName,const char *OriginalPath,
+	       unsigned long long &Bytes,unsigned long long const &FileSize);
+
+   inline void NewLine(unsigned const &Priority)
+   {
+      if (ErrorPrinted == false && Quiet <= Priority)
+      {
+	 c1out << endl;
+	 ErrorPrinted = true;
+      }
+   }
+
+   public:
+   FileFd *Output;
+   bool OwnsOutput;
+   unsigned int DoHashes;
+
+   unsigned long DeLinkLimit;
+   string InternalPrefix;
+
+   virtual bool DoPackage(string FileName) = 0;
+   bool RecursiveScan(string const &Dir);
+   bool LoadFileList(string const &BaseDir,string const &File);
+   void ClearPatterns() { Patterns.clear(); };
+   void AddPattern(string const &Pattern) { Patterns.push_back(Pattern); };
+   void AddPattern(char const *Pattern) { Patterns.push_back(Pattern); };
+   void AddPatterns(std::vector<std::string> const &patterns) { Patterns.insert(Patterns.end(), patterns.begin(), patterns.end()); };
+   bool SetExts(string const &Vals);
+
+   FTWScanner(FileFd * const Output, string const &Arch = string(), bool const IncludeArchAll = true);
+   virtual ~FTWScanner();
+};
+
+class MultiCompress;
+
+class TranslationWriter
+{
+   MultiCompress *Comp;
+   std::set<string> Included;
+   FileFd *Output;
+
+   public:
+   bool DoPackage(string const &Pkg, string const &Desc, string const &MD5);
+
+   TranslationWriter(string const &File, string const &TransCompress, mode_t const &Permissions);
+   ~TranslationWriter();
+};
+
+class PackagesWriter : public FTWScanner
+{
+   Override Over;
+   CacheDB Db;
+
+   public:
+
+   // Some flags
+   bool DoAlwaysStat;
+   bool NoOverride;
+   bool DoContents;
+   bool LongDescription;
+
+   // General options
+   string PathPrefix;
+   string DirStrip;
+   struct CacheDB::Stats &Stats;
+   TranslationWriter * const TransWriter;
+
+   inline bool ReadOverride(string const &File) {return Over.ReadOverride(File);};
+   inline bool ReadExtraOverride(string const &File) 
+      {return Over.ReadExtraOverride(File);};
+   virtual bool DoPackage(string FileName) APT_OVERRIDE;
+
+   PackagesWriter(FileFd * const Output, TranslationWriter * const TransWriter, string const &DB,
+                  string const &Overrides,
+                  string const &ExtOverrides = "",
+		  string const &Arch = "",
+		  bool const IncludeArchAll = true);
+   virtual ~PackagesWriter();
+};
+
+class ContentsWriter : public FTWScanner
+{
+   CacheDB Db;
+
+   GenContents Gen;
+
+   public:
+
+   // General options
+   struct CacheDB::Stats &Stats;
+   string Prefix;
+
+   bool DoPackage(string FileName,string Package);
+   virtual bool DoPackage(string FileName) APT_OVERRIDE 
+             {return DoPackage(FileName,string());};
+   bool ReadFromPkgs(string const &PkgFile,string const &PkgCompress);
+
+   void Finish() {Gen.Print(*Output);};
+   inline bool ReadyDB(string const &DB) {return Db.ReadyDB(DB);};
+
+   ContentsWriter(FileFd * const Output, string const &DB, string const &Arch = string(),
+	 bool const IncludeArchAll = true);
+   virtual ~ContentsWriter() {};
+};
+
+class SourcesWriter : public FTWScanner
+{
+   CacheDB Db;
+   Override BOver;
+   Override SOver;
+   char *Buffer;
+   unsigned long long BufSize;
+
+   public:
+
+   bool NoOverride;
+   bool DoAlwaysStat;
+
+   // General options
+   string PathPrefix;
+   string DirStrip;
+   struct CacheDB::Stats &Stats;
+
+   virtual bool DoPackage(string FileName) APT_OVERRIDE;
+
+   SourcesWriter(FileFd * const Output, string const &DB,string const &BOverrides,string const &SOverrides,
+		 string const &ExtOverrides=string());
+   virtual ~SourcesWriter() {free(Buffer);};
+};
+
+class ReleaseWriter : public FTWScanner
+{
+public:
+   ReleaseWriter(FileFd * const Output, string const &DB);
+   virtual bool DoPackage(string FileName) APT_OVERRIDE;
+   void Finish();
+
+   // General options
+   string PathPrefix;
+   string DirStrip;
+
+   struct CheckSum
+   {
+      HashStringList Hashes;
+      // Limited by FileFd::Size()
+      unsigned long long size;
+      ~CheckSum() {};
+   };
+protected:
+   map<string,struct CheckSum> CheckSums;
+};
+
+#endif