123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913 |
- // -*- mode: cpp; mode: fold -*-
- // Description /*{{{*/
- // $Id: tagfile.cc,v 1.37.2.2 2003/12/31 16:02:30 mdz Exp $
- /* ######################################################################
- Fast scanner for RFC-822 type header information
-
- This uses a rotating buffer to load the package information into.
- The scanner runs over it and isolates and indexes a single section.
-
- ##################################################################### */
- /*}}}*/
- // Include Files /*{{{*/
- #include<config.h>
- #include <apt-pkg/mmap.h>
- #include <apt-pkg/tagfile.h>
- #include <apt-pkg/tagfile-keys.h>
- #include <apt-pkg/error.h>
- #include <apt-pkg/strutl.h>
- #include <apt-pkg/fileutl.h>
- #include <apt-pkg/string_view.h>
- #include <list>
- #include <string>
- #include <stdio.h>
- #include <ctype.h>
- #include <stdlib.h>
- #include <string.h>
- #include <apti18n.h>
- /*}}}*/
- using std::string;
- using APT::StringView;
- class APT_HIDDEN pkgTagFilePrivate /*{{{*/
- {
- public:
- void Reset(FileFd * const pFd, pkgTagFile::Flags const pFlags)
- {
- if (Map != NULL)
- delete Map;
- Map = NULL;
- Buffer = NULL;
- Fd = pFd;
- Flags = pFlags;
- Start = NULL;
- End = NULL;
- Done = false;
- iOffset = 0;
- }
- pkgTagFilePrivate(FileFd * const pFd, pkgTagFile::Flags const pFlags) : Map(NULL)
- {
- Reset(pFd, pFlags);
- }
- FileFd * Fd;
- pkgTagFile::Flags Flags;
- char *Buffer;
- char *Start;
- char *End;
- bool Done;
- unsigned long long iOffset;
- MMap *Map;
- ~pkgTagFilePrivate()
- {
- if (Map != NULL)
- delete Map;
- }
- };
- /*}}}*/
- class APT_HIDDEN pkgTagSectionPrivate /*{{{*/
- {
- public:
- pkgTagSectionPrivate()
- {
- }
- struct TagData {
- unsigned int StartTag;
- unsigned int EndTag;
- unsigned int StartValue;
- unsigned int NextInBucket;
- explicit TagData(unsigned int const StartTag) : StartTag(StartTag), EndTag(0), StartValue(0), NextInBucket(0) {}
- };
- std::vector<TagData> Tags;
- };
- /*}}}*/
- static unsigned long BetaHash(const char *Text, size_t Length) /*{{{*/
- {
- /* This very simple hash function for the last 8 letters gives
- very good performance on the debian package files */
- if (Length > 8)
- {
- Text += (Length - 8);
- Length = 8;
- }
- unsigned long Res = 0;
- for (size_t i = 0; i < Length; ++i)
- Res = ((unsigned long)(Text[i]) & 0xDF) ^ (Res << 1);
- return Res & 0x7F;
- }
- /*}}}*/
- // TagFile::pkgTagFile - Constructor /*{{{*/
- pkgTagFile::pkgTagFile(FileFd * const pFd,pkgTagFile::Flags const pFlags)
- : d(new pkgTagFilePrivate(pFd, pFlags))
- {
- Init(pFd, pFlags);
- }
- pkgTagFile::pkgTagFile(FileFd * const pFd)
- : pkgTagFile(pFd, pkgTagFile::STRICT)
- {
- }
- void pkgTagFile::Init(FileFd * const pFd, pkgTagFile::Flags const pFlags)
- {
- d->Reset(pFd, pFlags);
- if (d->Fd->IsOpen() == false || d->Fd->Size() == 0)
- _error->Discard();
- else {
- d->Map = new MMap(*d->Fd, MMap::ReadOnly);
- d->Buffer = static_cast<char *>(d->Map->Data());
- }
- if (d->Buffer == NULL)
- d->Done = true;
- else {
- d->Done = false;
- d->End = d->Buffer + d->Map->Size();
- }
- d->Start = d->Buffer;
- }
- void pkgTagFile::Init(FileFd * const pFd)
- {
- Init(pFd, pkgTagFile::STRICT);
- }
- /*}}}*/
- // TagFile::~pkgTagFile - Destructor /*{{{*/
- pkgTagFile::~pkgTagFile()
- {
- delete d;
- }
- /*}}}*/
- // TagFile::Offset - Return the current offset in the buffer /*{{{*/
- APT_PURE unsigned long pkgTagFile::Offset()
- {
- return d->iOffset;
- }
- /*}}}*/
- // TagFile::Step - Advance to the next section /*{{{*/
- // ---------------------------------------------------------------------
- /* If the Section Scanner fails we refill the buffer and try again.
- * If that fails too, double the buffer size and try again until a
- * maximum buffer is reached.
- */
- bool pkgTagFile::Step(pkgTagSection &Tag)
- {
- if(Tag.Scan(d->Start,d->End - d->Start,(d->Flags & SUPPORT_COMMENTS) != 0) == false)
- {
- if (d->Start == d->End)
- return false;
- else
- return _error->Warning(_("Unable to parse package file %s (%d)"),
- d->Fd->Name().c_str(), 1);
- }
- size_t tagSize = Tag.size();
- d->Start += tagSize;
- d->iOffset += tagSize;
- return true;
- }
- /*}}}*/
- // TagFile::Jump - Jump to a pre-recorded location in the file /*{{{*/
- // ---------------------------------------------------------------------
- /* This jumps to a pre-recorded file location and reads the record
- that is there */
- bool pkgTagFile::Jump(pkgTagSection &Tag,unsigned long long Offset)
- {
- unsigned int Size(d->Map->Size());
- if (Offset >= Size)
- return false;
- // Reposition and reload..
- d->iOffset = Offset;
- d->Done = false;
- d->Start = d->Buffer + d->iOffset;
- return Step(Tag);
- }
- /*}}}*/
- // pkgTagSection::pkgTagSection - Constructor /*{{{*/
- // ---------------------------------------------------------------------
- /* */
- APT_IGNORE_DEPRECATED_PUSH
- pkgTagSection::pkgTagSection()
- : Section(0), d(new pkgTagSectionPrivate()), Stop(0)
- {
- memset(&AlphaIndexes, 0, sizeof(AlphaIndexes));
- memset(&BetaIndexes, 0, sizeof(BetaIndexes));
- }
- APT_IGNORE_DEPRECATED_POP
- /*}}}*/
- // TagSection::Scan - Scan for the end of the header information /*{{{*/
- bool pkgTagSection::Scan(const char *Start,unsigned long MaxLength,bool const SupportComments)
- {
- Section = Start;
- const char *End = Start + MaxLength;
- Stop = Section;
- if (d->Tags.empty() == false)
- {
- memset(&AlphaIndexes, 0, sizeof(AlphaIndexes));
- memset(&BetaIndexes, 0, sizeof(BetaIndexes));
- d->Tags.clear();
- }
- d->Tags.reserve(0x100);
- unsigned int TagCount = d->Tags.size();
- if (Stop == 0)
- return false;
- pkgTagSectionPrivate::TagData lastTagData(0);
- lastTagData.EndTag = 0;
- Key lastTagKey = Key::Unknown;
- unsigned int lastTagHash = 0;
- while (Stop < End)
- {
- TrimRecord(true,End,SupportComments);
- // this can happen when TrimRecord trims away the entire Record
- // (e.g. because it just contains comments)
- if(Stop == End)
- goto end;
- // Start a new index and add it to the hash
- if (isspace_ascii(Stop[0]) == 0)
- {
- // store the last found tag
- if (lastTagData.EndTag != 0)
- {
- if (lastTagKey != Key::Unknown) {
- AlphaIndexes[static_cast<size_t>(lastTagKey)] = TagCount;
- } else {
- if (BetaIndexes[lastTagHash] != 0)
- lastTagData.NextInBucket = BetaIndexes[lastTagHash];
- APT_IGNORE_DEPRECATED_PUSH
- BetaIndexes[lastTagHash] = TagCount;
- APT_IGNORE_DEPRECATED_POP
- }
- d->Tags.push_back(lastTagData);
- }
- APT_IGNORE_DEPRECATED(++TagCount;)
- lastTagData = pkgTagSectionPrivate::TagData(Stop - Section);
- // find the colon separating tag and value
- char const * Colon = (char const *) memchr(Stop, ':', End - Stop);
- if (Colon == NULL)
- return false;
- // find the end of the tag (which might or might not be the colon)
- char const * EndTag = Colon;
- --EndTag;
- for (; EndTag > Stop && isspace_ascii(*EndTag) != 0; --EndTag)
- ;
- ++EndTag;
- lastTagData.EndTag = EndTag - Section;
- lastTagKey = pkgTagHash(Stop, EndTag - Stop);
- if (lastTagKey == Key::Unknown)
- lastTagHash = BetaHash(Stop, EndTag - Stop);
- // find the beginning of the value
- Stop = Colon + 1;
- for (; Stop < End && isspace_ascii(*Stop) != 0; ++Stop)
- if (*Stop == '\n' && (Stop+1 == End || Stop[1] != ' '))
- break;
- lastTagData.StartValue = Stop - Section;
- }
- Stop = (const char *)memchr(Stop,'\n',End - Stop);
- if (Stop == 0) {
- Stop = End;
- goto end;
- }
- for (; Stop+1 < End && Stop[1] == '\r'; Stop++)
- /* nothing */
- ;
- // Double newline marks the end of the record
- if (Stop+1 < End && Stop[1] == '\n')
- end: {
- if (lastTagData.EndTag != 0)
- {
- if (lastTagKey != Key::Unknown) {
- AlphaIndexes[static_cast<size_t>(lastTagKey)] = TagCount;
- } else {
- if (BetaIndexes[lastTagHash] != 0)
- lastTagData.NextInBucket = BetaIndexes[lastTagHash];
- APT_IGNORE_DEPRECATED(BetaIndexes[lastTagHash] = TagCount;)
- }
- d->Tags.push_back(lastTagData);
- }
- if (d->Tags.empty())
- return false;
- pkgTagSectionPrivate::TagData const td(Stop - Section);
- d->Tags.push_back(td);
- TrimRecord(false,End,SupportComments);
- return true;
- }
-
- Stop++;
- }
- goto end;
- }
- /*}}}*/
- // TagSection::TrimRecord - Trim off any garbage before/after a record /*{{{*/
- // ---------------------------------------------------------------------
- /* There should be exactly 2 newline at the end of the record, no more. */
- void pkgTagSection::TrimRecord(bool BeforeRecord, const char*& End, bool SupportComments)
- { trim:
- if (BeforeRecord == false)
- for (; Stop < End && (Stop[0] == '\n' || Stop[0] == '\r'); Stop++);
- if (SupportComments && Stop < End && Stop[0] == '#') {
- Stop = (const char*) memchr(Stop,'\n',End-Stop) ?: End;
- goto trim;
- }
- }
- /*}}}*/
- // TagSection::Exists - return True if a tag exists /*{{{*/
- bool pkgTagSection::Exists(StringView Tag) const
- {
- unsigned int tmp;
- return Find(Tag, tmp);
- }
- /*}}}*/
- // TagSection::Find - Locate a tag /*{{{*/
- // ---------------------------------------------------------------------
- /* This searches the section for a tag that matches the given string. */
- bool pkgTagSection::Find(Key key,unsigned int &Pos) const
- {
- auto Bucket = AlphaIndexes[static_cast<size_t>(key)];
- Pos = Bucket - 1;
- return Bucket != 0;
- }
- bool pkgTagSection::Find(StringView TagView,unsigned int &Pos) const
- {
- const char * const Tag = TagView.data();
- size_t const Length = TagView.length();
- auto key = pkgTagHash(Tag, Length);
- if (key != Key::Unknown)
- return Find(key, Pos);
- unsigned int Bucket = BetaIndexes[BetaHash(Tag, Length)];
- if (Bucket == 0)
- return false;
- for (; Bucket != 0; Bucket = d->Tags[Bucket - 1].NextInBucket)
- {
- if ((d->Tags[Bucket - 1].EndTag - d->Tags[Bucket - 1].StartTag) != Length)
- continue;
- char const * const St = Section + d->Tags[Bucket - 1].StartTag;
- if (strncasecmp(Tag,St,Length) != 0)
- continue;
- Pos = Bucket - 1;
- return true;
- }
- Pos = 0;
- return false;
- }
- bool pkgTagSection::FindInternal(unsigned int Pos, const char *&Start,
- const char *&End) const
- {
- Start = Section + d->Tags[Pos].StartValue;
- // Strip off the gunk from the end
- End = Section + d->Tags[Pos + 1].StartTag;
- if (unlikely(Start > End))
- return _error->Error("Internal parsing error");
- for (; End > Start && isspace_ascii(End[-1]) != 0; --End);
- return true;
- }
- bool pkgTagSection::Find(StringView Tag,const char *&Start,
- const char *&End) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) && FindInternal(Pos, Start, End);
- }
- bool pkgTagSection::Find(Key key,const char *&Start,
- const char *&End) const
- {
- unsigned int Pos;
- return Find(key, Pos) && FindInternal(Pos, Start, End);
- }
- /*}}}*/
- // TagSection::FindS - Find a string /*{{{*/
- StringView pkgTagSection::Find(StringView Tag) const
- {
- const char *Start;
- const char *End;
- if (Find(Tag,Start,End) == false)
- return StringView();
- return StringView(Start, End - Start);
- }
- StringView pkgTagSection::Find(Key key) const
- {
- const char *Start;
- const char *End;
- if (Find(key,Start,End) == false)
- return StringView();
- return StringView(Start, End - Start);
- }
- /*}}}*/
- // TagSection::FindRawS - Find a string /*{{{*/
- StringView pkgTagSection::FindRawInternal(unsigned int Pos) const
- {
- char const *Start = (char const *) memchr(Section + d->Tags[Pos].EndTag, ':', d->Tags[Pos].StartValue - d->Tags[Pos].EndTag);
- ++Start;
- char const *End = Section + d->Tags[Pos + 1].StartTag;
- if (unlikely(Start > End))
- return "";
- for (; isspace_ascii(End[-1]) != 0 && End > Start; --End);
- return StringView(Start, End - Start);
- }
- StringView pkgTagSection::FindRaw(StringView Tag) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) ? FindRawInternal(Pos) : "";
- }
- StringView pkgTagSection::FindRaw(Key key) const
- {
- unsigned int Pos;
- return Find(key, Pos) ? FindRawInternal(Pos) : "";
- }
- /*}}}*/
- // TagSection::FindI - Find an integer /*{{{*/
- // ---------------------------------------------------------------------
- /* */
- signed int pkgTagSection::FindIInternal(unsigned int Pos,signed long Default) const
- {
- const char *Start;
- const char *Stop;
- if (FindInternal(Pos,Start,Stop) == false)
- return Default;
- // Copy it into a temp buffer so we can use strtol
- char S[300];
- if ((unsigned)(Stop - Start) >= sizeof(S))
- return Default;
- strncpy(S,Start,Stop-Start);
- S[Stop - Start] = 0;
- errno = 0;
- char *End;
- signed long Result = strtol(S,&End,10);
- if (errno == ERANGE ||
- Result < std::numeric_limits<int>::min() || Result > std::numeric_limits<int>::max()) {
- errno = ERANGE;
- _error->Error(_("Cannot convert %s to integer: out of range"), S);
- }
- if (S == End)
- return Default;
- return Result;
- }
- signed int pkgTagSection::FindI(Key key,signed long Default) const
- {
- unsigned int Pos;
- return Find(key, Pos) ? FindIInternal(Pos) : Default;
- }
- signed int pkgTagSection::FindI(StringView Tag,signed long Default) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) ? FindIInternal(Pos, Default) : Default;
- }
- /*}}}*/
- // TagSection::FindULL - Find an unsigned long long integer /*{{{*/
- // ---------------------------------------------------------------------
- /* */
- unsigned long long pkgTagSection::FindULLInternal(unsigned int Pos, unsigned long long const &Default) const
- {
- const char *Start;
- const char *Stop;
- if (FindInternal(Pos,Start,Stop) == false)
- return Default;
- // Copy it into a temp buffer so we can use strtoull
- char S[100];
- if ((unsigned)(Stop - Start) >= sizeof(S))
- return Default;
- strncpy(S,Start,Stop-Start);
- S[Stop - Start] = 0;
-
- char *End;
- unsigned long long Result = strtoull(S,&End,10);
- if (S == End)
- return Default;
- return Result;
- }
- unsigned long long pkgTagSection::FindULL(Key key, unsigned long long const &Default) const
- {
- unsigned int Pos;
- return Find(key, Pos) ? FindULLInternal(Pos, Default) : Default;
- }
- unsigned long long pkgTagSection::FindULL(StringView Tag, unsigned long long const &Default) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) ? FindULLInternal(Pos, Default) : Default;
- }
- /*}}}*/
- // TagSection::FindB - Find boolean value /*{{{*/
- // ---------------------------------------------------------------------
- /* */
- bool pkgTagSection::FindBInternal(unsigned int Pos, bool Default) const
- {
- const char *Start, *Stop;
- if (FindInternal(Pos, Start, Stop) == false)
- return Default;
- return StringToBool(string(Start, Stop));
- }
- bool pkgTagSection::FindB(Key key, bool Default) const
- {
- unsigned int Pos;
- return Find(key, Pos) ? FindBInternal(Pos, Default): Default;
- }
- bool pkgTagSection::FindB(StringView Tag, bool Default) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) ? FindBInternal(Pos, Default) : Default;
- }
- /*}}}*/
- // TagSection::FindFlag - Locate a yes/no type flag /*{{{*/
- // ---------------------------------------------------------------------
- /* The bits marked in Flag are masked on/off in Flags */
- bool pkgTagSection::FindFlagInternal(unsigned int Pos, uint8_t &Flags,
- uint8_t const Flag) const
- {
- const char *Start;
- const char *Stop;
- if (FindInternal(Pos,Start,Stop) == false)
- return true;
- return FindFlag(Flags, Flag, Start, Stop);
- }
- bool pkgTagSection::FindFlag(Key key, uint8_t &Flags,
- uint8_t const Flag) const
- {
- unsigned int Pos;
- if (Find(key,Pos) == false)
- return true;
- return FindFlagInternal(Pos, Flags, Flag);
- }
- bool pkgTagSection::FindFlag(StringView Tag, uint8_t &Flags,
- uint8_t const Flag) const
- {
- unsigned int Pos;
- if (Find(Tag,Pos) == false)
- return true;
- return FindFlagInternal(Pos, Flags, Flag);
- }
- bool pkgTagSection::FindFlag(uint8_t &Flags, uint8_t const Flag,
- char const* const Start, char const* const Stop)
- {
- switch (StringToBool(string(Start, Stop)))
- {
- case 0:
- Flags &= ~Flag;
- return true;
- case 1:
- Flags |= Flag;
- return true;
- default:
- _error->Warning("Unknown flag value: %s",string(Start,Stop).c_str());
- return true;
- }
- return true;
- }
- bool pkgTagSection::FindFlagInternal(unsigned int Pos,unsigned long &Flags,
- unsigned long Flag) const
- {
- const char *Start;
- const char *Stop;
- if (FindInternal(Pos,Start,Stop) == false)
- return true;
- return FindFlag(Flags, Flag, Start, Stop);
- }
- bool pkgTagSection::FindFlag(Key key,unsigned long &Flags,
- unsigned long Flag) const
- {
- unsigned int Pos;
- return Find(key, Pos) ? FindFlagInternal(Pos, Flags, Flag) : true;
- }
- bool pkgTagSection::FindFlag(StringView Tag,unsigned long &Flags,
- unsigned long Flag) const
- {
- unsigned int Pos;
- return Find(Tag, Pos) ? FindFlagInternal(Pos, Flags, Flag) : true;
- }
- bool pkgTagSection::FindFlag(unsigned long &Flags, unsigned long Flag,
- char const* Start, char const* Stop)
- {
- switch (StringToBool(string(Start, Stop)))
- {
- case 0:
- Flags &= ~Flag;
- return true;
- case 1:
- Flags |= Flag;
- return true;
- default:
- _error->Warning("Unknown flag value: %s",string(Start,Stop).c_str());
- return true;
- }
- return true;
- }
- /*}}}*/
- void pkgTagSection::Get(const char *&Start,const char *&Stop,unsigned int I) const/*{{{*/
- {
- Start = Section + d->Tags[I].StartTag;
- Stop = Section + d->Tags[I+1].StartTag;
- }
- /*}}}*/
- APT_PURE unsigned int pkgTagSection::Count() const { /*{{{*/
- if (d->Tags.empty() == true)
- return 0;
- // the last element is just marking the end and isn't a real one
- return d->Tags.size() - 1;
- }
- /*}}}*/
- // TagSection::Write - Ordered (re)writing of fields /*{{{*/
- pkgTagSection::Tag pkgTagSection::Tag::Remove(std::string const &Name)
- {
- return Tag(REMOVE, Name, "");
- }
- pkgTagSection::Tag pkgTagSection::Tag::Rename(std::string const &OldName, std::string const &NewName)
- {
- return Tag(RENAME, OldName, NewName);
- }
- pkgTagSection::Tag pkgTagSection::Tag::Rewrite(std::string const &Name, std::string const &Data)
- {
- if (Data.empty() == true)
- return Tag(REMOVE, Name, "");
- else
- return Tag(REWRITE, Name, Data);
- }
- static bool WriteTag(FileFd &File, std::string Tag, StringView Value)
- {
- if (Value.empty() || isspace_ascii(Value[0]) != 0)
- Tag.append(":");
- else
- Tag.append(": ");
- Tag.append(Value.data(), Value.length());
- Tag.append("\n");
- return File.Write(Tag.c_str(), Tag.length());
- }
- static bool RewriteTags(FileFd &File, pkgTagSection const * const This, char const * const Tag,
- std::vector<pkgTagSection::Tag>::const_iterator &R,
- std::vector<pkgTagSection::Tag>::const_iterator const &REnd)
- {
- size_t const TagLen = strlen(Tag);
- for (; R != REnd; ++R)
- {
- std::string data;
- if (R->Name.length() == TagLen && strncasecmp(R->Name.c_str(), Tag, R->Name.length()) == 0)
- {
- if (R->Action != pkgTagSection::Tag::REWRITE)
- break;
- data = R->Data;
- }
- else if(R->Action == pkgTagSection::Tag::RENAME && R->Data.length() == TagLen &&
- strncasecmp(R->Data.c_str(), Tag, R->Data.length()) == 0)
- data = This->FindRaw(R->Name.c_str()).to_string();
- else
- continue;
- return WriteTag(File, Tag, data);
- }
- return true;
- }
- bool pkgTagSection::Write(FileFd &File, char const * const * const Order, std::vector<Tag> const &Rewrite) const
- {
- // first pass: Write everything we have an order for
- if (Order != NULL)
- {
- for (unsigned int I = 0; Order[I] != 0; ++I)
- {
- std::vector<Tag>::const_iterator R = Rewrite.begin();
- if (RewriteTags(File, this, Order[I], R, Rewrite.end()) == false)
- return false;
- if (R != Rewrite.end())
- continue;
- if (Exists(Order[I]) == false)
- continue;
- if (WriteTag(File, Order[I], FindRaw(Order[I])) == false)
- return false;
- }
- }
- // second pass: See if we have tags which aren't ordered
- if (d->Tags.empty() == false)
- {
- for (std::vector<pkgTagSectionPrivate::TagData>::const_iterator T = d->Tags.begin(); T != d->Tags.end() - 1; ++T)
- {
- char const * const fieldname = Section + T->StartTag;
- size_t fieldnamelen = T->EndTag - T->StartTag;
- if (Order != NULL)
- {
- unsigned int I = 0;
- for (; Order[I] != 0; ++I)
- {
- if (fieldnamelen == strlen(Order[I]) && strncasecmp(fieldname, Order[I], fieldnamelen) == 0)
- break;
- }
- if (Order[I] != 0)
- continue;
- }
- std::string const name(fieldname, fieldnamelen);
- std::vector<Tag>::const_iterator R = Rewrite.begin();
- if (RewriteTags(File, this, name.c_str(), R, Rewrite.end()) == false)
- return false;
- if (R != Rewrite.end())
- continue;
- if (WriteTag(File, name, FindRaw(name)) == false)
- return false;
- }
- }
- // last pass: see if there are any rewrites remaining we haven't done yet
- for (std::vector<Tag>::const_iterator R = Rewrite.begin(); R != Rewrite.end(); ++R)
- {
- if (R->Action == Tag::REMOVE)
- continue;
- std::string const name = ((R->Action == Tag::RENAME) ? R->Data : R->Name);
- if (Exists(name.c_str()))
- continue;
- if (Order != NULL)
- {
- unsigned int I = 0;
- for (; Order[I] != 0; ++I)
- {
- if (strncasecmp(name.c_str(), Order[I], name.length()) == 0 && name.length() == strlen(Order[I]))
- break;
- }
- if (Order[I] != 0)
- continue;
- }
- if (WriteTag(File, name, ((R->Action == Tag::RENAME) ? FindRaw(R->Name) : R->Data)) == false)
- return false;
- }
- return true;
- }
- /*}}}*/
- #include "tagfile-order.c"
- // TFRewrite - Rewrite a control record /*{{{*/
- // ---------------------------------------------------------------------
- /* This writes the control record to stdout rewriting it as necessary. The
- override map item specificies the rewriting rules to follow. This also
- takes the time to sort the feild list. */
- APT_IGNORE_DEPRECATED_PUSH
- bool TFRewrite(FILE *Output,pkgTagSection const &Tags,const char *Order[],
- TFRewriteData *Rewrite)
- {
- unsigned char Visited[256]; // Bit 1 is Order, Bit 2 is Rewrite
- for (unsigned I = 0; I != 256; I++)
- Visited[I] = 0;
- // Set new tag up as necessary.
- for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
- {
- if (Rewrite[J].NewTag == 0)
- Rewrite[J].NewTag = Rewrite[J].Tag;
- }
-
- // Write all all of the tags, in order.
- if (Order != NULL)
- {
- for (unsigned int I = 0; Order[I] != 0; I++)
- {
- bool Rewritten = false;
-
- // See if this is a field that needs to be rewritten
- for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
- {
- if (strcasecmp(Rewrite[J].Tag,Order[I]) == 0)
- {
- Visited[J] |= 2;
- if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
- {
- if (isspace_ascii(Rewrite[J].Rewrite[0]))
- fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- else
- fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- }
- Rewritten = true;
- break;
- }
- }
-
- // See if it is in the fragment
- unsigned Pos;
- if (Tags.Find(StringView(Order[I]),Pos) == false)
- continue;
- Visited[Pos] |= 1;
- if (Rewritten == true)
- continue;
-
- /* Write out this element, taking a moment to rewrite the tag
- in case of changes of case. */
- const char *Start;
- const char *Stop;
- Tags.Get(Start,Stop,Pos);
-
- if (fputs(Order[I],Output) < 0)
- return _error->Errno("fputs","IO Error to output");
- Start += strlen(Order[I]);
- if (fwrite(Start,Stop - Start,1,Output) != 1)
- return _error->Errno("fwrite","IO Error to output");
- if (Stop[-1] != '\n')
- fprintf(Output,"\n");
- }
- }
- // Now write all the old tags that were missed.
- for (unsigned int I = 0; I != Tags.Count(); I++)
- {
- if ((Visited[I] & 1) == 1)
- continue;
- const char *Start;
- const char *Stop;
- Tags.Get(Start,Stop,I);
- const char *End = Start;
- for (; End < Stop && *End != ':'; End++);
- // See if this is a field that needs to be rewritten
- bool Rewritten = false;
- for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
- {
- if (stringcasecmp(Start,End,Rewrite[J].Tag) == 0)
- {
- Visited[J] |= 2;
- if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
- {
- if (isspace_ascii(Rewrite[J].Rewrite[0]))
- fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- else
- fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- }
-
- Rewritten = true;
- break;
- }
- }
-
- if (Rewritten == true)
- continue;
-
- // Write out this element
- if (fwrite(Start,Stop - Start,1,Output) != 1)
- return _error->Errno("fwrite","IO Error to output");
- if (Stop[-1] != '\n')
- fprintf(Output,"\n");
- }
-
- // Now write all the rewrites that were missed
- for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
- {
- if ((Visited[J] & 2) == 2)
- continue;
-
- if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
- {
- if (isspace_ascii(Rewrite[J].Rewrite[0]))
- fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- else
- fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
- }
- }
-
- return true;
- }
- APT_IGNORE_DEPRECATED_POP
- /*}}}*/
- pkgTagSection::~pkgTagSection() { delete d; }
|