apt-pkg/tagfile.h

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: tagfile.h,v 1.20 2003/05/19 17:13:57 doogie Exp $
   4 /* ######################################################################
   5
   6    Fast scanner for RFC-822 type header information
   7
   8    This parser handles Debian package files (and others). Their form is
   9    RFC-822 type header fields in groups separated by a blank line.
  10
  11    The parser reads the file and provides methods to step linearly
  12    over it or to jump to a pre-recorded start point and read that record.
  13
  14    A second class is used to perform pre-parsing of the record. It works
  15    by indexing the start of each header field and providing lookup
  16    functions for header fields.
  17
  18    ##################################################################### */
  19                                                                         /*}}}*/
  20 #ifndef PKGLIB_TAGFILE_H
  21 #define PKGLIB_TAGFILE_H
  22
  23 #include <apt-pkg/macros.h>
  24
  25 #include <stdio.h>
  26
  27 #include <string>
  28 #include <vector>
  29 #include <list>
  30
  31 #ifndef APT_8_CLEANER_HEADERS
  32 #include <apt-pkg/fileutl.h>
  33 #endif
  34
  35 class FileFd;
  36
  37 class pkgTagSection
  38 {
  39    const char *Section;
  40    struct TagData {
  41       unsigned int StartTag;
  42       unsigned int EndTag;
  43       unsigned int StartValue;
  44       unsigned int NextInBucket;
  45
  46       TagData(unsigned int const StartTag) : StartTag(StartTag), EndTag(0), StartValue(0), NextInBucket(0) {}
  47    };
  48    std::vector<TagData> Tags;
  49    unsigned int LookupTable[0x100];
  50
  51    // dpointer placeholder (for later in case we need it)
  52    void *d;
  53
  54    protected:
  55    const char *Stop;
  56
  57    public:
  58
  59    inline bool operator ==(const pkgTagSection &rhs) {return Section == rhs.Section;};
  60    inline bool operator !=(const pkgTagSection &rhs) {return Section != rhs.Section;};
  61
  62    bool Find(const char *Tag,const char *&Start, const char *&End) const;
  63    bool Find(const char *Tag,unsigned int &Pos) const;
  64    std::string FindS(const char *Tag) const;
  65    signed int FindI(const char *Tag,signed long Default = 0) const;
  66    bool FindB(const char *Tag, bool const &Default = false) const;
  67    unsigned long long FindULL(const char *Tag, unsigned long long const &Default = 0) const;
  68    bool FindFlag(const char *Tag,unsigned long &Flags,
  69                  unsigned long Flag) const;
  70    bool static FindFlag(unsigned long &Flags, unsigned long Flag,
  71                                 const char* Start, const char* Stop);
  72
  73    /** \brief searches the boundaries of the current section
  74     *
  75     * While parameter Start marks the beginning of the section, this method
  76     * will search for the first double newline in the data stream which marks
  77     * the end of the section. It also does a first pass over the content of
  78     * the section parsing it as encountered for processing later on by Find
  79     *
  80     * @param Start is the beginning of the section
  81     * @param MaxLength is the size of valid data in the stream pointed to by Start
  82     * @param Restart if enabled internal state will be cleared, otherwise it is
  83     *  assumed that now more data is available in the stream and the parsing will
  84     *  start were it encountered insufficent data the last time.
  85     *
  86     * @return \b true if section end was found, \b false otherwise.
  87     *  Beware that internal state will be inconsistent if \b false is returned!
  88     */
  89 #if APT_PKG_ABI >= 413
  90    APT_MUSTCHECK bool Scan(const char *Start, unsigned long MaxLength, bool const Restart = true);
  91 #else
  92    APT_MUSTCHECK bool Scan(const char *Start, unsigned long MaxLength, bool const Restart);
  93    APT_MUSTCHECK bool Scan(const char *Start, unsigned long MaxLength);
  94 #endif
  95
  96    inline unsigned long size() const {return Stop - Section;};
  97    void Trim();
  98    virtual void TrimRecord(bool BeforeRecord, const char* &End);
  99
 100    /** \brief amount of Tags in the current section
 101     *
 102     * Note: if a Tag is mentioned repeatly it will be counted multiple
 103     * times, but only the last occurrence is available via Find methods.
 104     */
 105    unsigned int Count() const;
 106 #if APT_PKG_ABI >= 413
 107    bool Exists(const char* const Tag) const;
 108 #else
 109    bool Exists(const char* const Tag);
 110 #endif
 111
 112    inline void Get(const char *&Start,const char *&Stop,unsigned int I) const
 113                    {Start = Section + Tags[I].StartTag; Stop = Section + Tags[I+1].StartTag;}
 114
 115    inline void GetSection(const char *&Start,const char *&Stop) const
 116    {
 117       Start = Section;
 118       Stop = this->Stop;
 119    };
 120
 121    pkgTagSection();
 122    virtual ~pkgTagSection();
 123 };
 124
 125 class pkgTagFilePrivate;
 126 class pkgTagFile
 127 {
 128    pkgTagFilePrivate *d;
 129
 130    APT_HIDDEN bool Fill();
 131    APT_HIDDEN bool Resize();
 132    APT_HIDDEN bool Resize(unsigned long long const newSize);
 133
 134    public:
 135
 136    bool Step(pkgTagSection &Section);
 137    unsigned long Offset();
 138    bool Jump(pkgTagSection &Tag,unsigned long long Offset);
 139
 140    void Init(FileFd *F,unsigned long long Size = 32*1024);
 141
 142    pkgTagFile(FileFd *F,unsigned long long Size = 32*1024);
 143    virtual ~pkgTagFile();
 144 };
 145
 146 /* This is the list of things to rewrite. The rewriter
 147    goes through and changes or adds each of these headers
 148    to suit. A zero forces the header to be erased, an empty string
 149    causes the old value to be used. (rewrite rule ignored) */
 150 struct TFRewriteData
 151 {
 152    const char *Tag;
 153    const char *Rewrite;
 154    const char *NewTag;
 155 };
 156 extern const char **TFRewritePackageOrder;
 157 extern const char **TFRewriteSourceOrder;
 158
 159 bool TFRewrite(FILE *Output,pkgTagSection const &Tags,const char *Order[],
 160                TFRewriteData *Rewrite);
 161
 162 #endif