apt-pkg/tagfile.cc

   1 // -*- mode: cpp; mode: fold -*-
   2 // Description                                                          /*{{{*/
   3 // $Id: tagfile.cc,v 1.37.2.2 2003/12/31 16:02:30 mdz Exp $
   4 /* ######################################################################
   5
   6    Fast scanner for RFC-822 type header information
   7
   8    This uses a rotating buffer to load the package information into.
   9    The scanner runs over it and isolates and indexes a single section.
  10
  11    ##################################################################### */
  12                                                                         /*}}}*/
  13 // Include Files                                                        /*{{{*/
  14 #include<config.h>
  15
  16 #include <apt-pkg/tagfile.h>
  17 #include <apt-pkg/error.h>
  18 #include <apt-pkg/strutl.h>
  19 #include <apt-pkg/fileutl.h>
  20
  21 #include <string>
  22 #include <stdio.h>
  23 #include <ctype.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include <apti18n.h>
  28                                                                         /*}}}*/
  29
  30 using std::string;
  31
  32 class pkgTagFilePrivate
  33 {
  34 public:
  35    void Reset(FileFd * const pFd, unsigned long long const pSize)
  36    {
  37       if (Buffer != NULL)
  38          free(Buffer);
  39       Buffer = NULL;
  40       Fd = pFd;
  41       Start = NULL;
  42       End = NULL;
  43       Done = false;
  44       iOffset = 0;
  45       Size = pSize;
  46    }
  47
  48    pkgTagFilePrivate(FileFd * const pFd, unsigned long long const Size) : Buffer(NULL)
  49    {
  50       Reset(pFd, Size);
  51    }
  52    FileFd * Fd;
  53    char *Buffer;
  54    char *Start;
  55    char *End;
  56    bool Done;
  57    unsigned long long iOffset;
  58    unsigned long long Size;
  59
  60    ~pkgTagFilePrivate()
  61    {
  62       if (Buffer != NULL)
  63          free(Buffer);
  64    }
  65 };
  66
  67 class pkgTagSectionPrivate
  68 {
  69 public:
  70    pkgTagSectionPrivate()
  71    {
  72    }
  73    struct TagData {
  74       unsigned int StartTag;
  75       unsigned int EndTag;
  76       unsigned int StartValue;
  77       unsigned int NextInBucket;
  78
  79       explicit TagData(unsigned int const StartTag) : StartTag(StartTag), EndTag(0), StartValue(0), NextInBucket(0) {}
  80    };
  81    std::vector<TagData> Tags;
  82 };
  83
  84 static unsigned long AlphaHash(const char *Text, size_t Length)         /*{{{*/
  85 {
  86    /* This very simple hash function for the last 8 letters gives
  87       very good performance on the debian package files */
  88    if (Length > 8)
  89    {
  90     Text += (Length - 8);
  91     Length = 8;
  92    }
  93    unsigned long Res = 0;
  94    for (size_t i = 0; i < Length; ++i)
  95       Res = ((unsigned long)(Text[i]) & 0xDF) ^ (Res << 1);
  96    return Res & 0xFF;
  97 }
  98                                                                         /*}}}*/
  99
 100 // TagFile::pkgTagFile - Constructor                                    /*{{{*/
 101 // ---------------------------------------------------------------------
 102 /* */
 103 pkgTagFile::pkgTagFile(FileFd * const pFd,unsigned long long const Size)
 104    : d(new pkgTagFilePrivate(pFd, Size + 4))
 105 {
 106    Init(pFd, Size);
 107 }
 108 void pkgTagFile::Init(FileFd * const pFd,unsigned long long Size)
 109 {
 110    /* The size is increased by 4 because if we start with the Size of the
 111       filename we need to try to read 1 char more to see an EOF faster, 1
 112       char the end-pointer can be on and maybe 2 newlines need to be added
 113       to the end of the file -> 4 extra chars */
 114    Size += 4;
 115    d->Reset(pFd, Size);
 116
 117    if (d->Fd->IsOpen() == false)
 118       d->Start = d->End = d->Buffer = 0;
 119    else
 120       d->Buffer = (char*)malloc(sizeof(char) * Size);
 121
 122    if (d->Buffer == NULL)
 123       d->Done = true;
 124    else
 125       d->Done = false;
 126
 127    d->Start = d->End = d->Buffer;
 128    d->iOffset = 0;
 129    if (d->Done == false)
 130       Fill();
 131 }
 132                                                                         /*}}}*/
 133 // TagFile::~pkgTagFile - Destructor                                    /*{{{*/
 134 // ---------------------------------------------------------------------
 135 /* */
 136 pkgTagFile::~pkgTagFile()
 137 {
 138    delete d;
 139 }
 140                                                                         /*}}}*/
 141 // TagFile::Offset - Return the current offset in the buffer            /*{{{*/
 142 APT_PURE unsigned long pkgTagFile::Offset()
 143 {
 144    return d->iOffset;
 145 }
 146                                                                         /*}}}*/
 147 // TagFile::Resize - Resize the internal buffer                         /*{{{*/
 148 // ---------------------------------------------------------------------
 149 /* Resize the internal buffer (double it in size). Fail if a maximum size
 150  * size is reached.
 151  */
 152 bool pkgTagFile::Resize()
 153 {
 154    // fail is the buffer grows too big
 155    if(d->Size > 1024*1024+1)
 156       return false;
 157
 158    return Resize(d->Size * 2);
 159 }
 160 bool pkgTagFile::Resize(unsigned long long const newSize)
 161 {
 162    unsigned long long const EndSize = d->End - d->Start;
 163
 164    // get new buffer and use it
 165    char* newBuffer = (char*)realloc(d->Buffer, sizeof(char) * newSize);
 166    if (newBuffer == NULL)
 167       return false;
 168    d->Buffer = newBuffer;
 169    d->Size = newSize;
 170
 171    // update the start/end pointers to the new buffer
 172    d->Start = d->Buffer;
 173    d->End = d->Start + EndSize;
 174    return true;
 175 }
 176                                                                         /*}}}*/
 177 // TagFile::Step - Advance to the next section                          /*{{{*/
 178 // ---------------------------------------------------------------------
 179 /* If the Section Scanner fails we refill the buffer and try again.
 180  * If that fails too, double the buffer size and try again until a
 181  * maximum buffer is reached.
 182  */
 183 bool pkgTagFile::Step(pkgTagSection &Tag)
 184 {
 185    if(Tag.Scan(d->Start,d->End - d->Start) == false)
 186    {
 187       do
 188       {
 189          if (Fill() == false)
 190             return false;
 191
 192          if(Tag.Scan(d->Start,d->End - d->Start, false))
 193             break;
 194
 195          if (Resize() == false)
 196             return _error->Error(_("Unable to parse package file %s (%d)"),
 197                   d->Fd->Name().c_str(), 1);
 198
 199       } while (Tag.Scan(d->Start,d->End - d->Start, false) == false);
 200    }
 201
 202    d->Start += Tag.size();
 203    d->iOffset += Tag.size();
 204
 205    Tag.Trim();
 206    return true;
 207 }
 208                                                                         /*}}}*/
 209 // TagFile::Fill - Top up the buffer                                    /*{{{*/
 210 // ---------------------------------------------------------------------
 211 /* This takes the bit at the end of the buffer and puts it at the start
 212    then fills the rest from the file */
 213 bool pkgTagFile::Fill()
 214 {
 215    unsigned long long EndSize = d->End - d->Start;
 216    unsigned long long Actual = 0;
 217
 218    memmove(d->Buffer,d->Start,EndSize);
 219    d->Start = d->Buffer;
 220    d->End = d->Buffer + EndSize;
 221
 222    if (d->Done == false)
 223    {
 224       // See if only a bit of the file is left
 225       unsigned long long const dataSize = d->Size - ((d->End - d->Buffer) + 1);
 226       if (d->Fd->Read(d->End, dataSize, &Actual) == false)
 227          return false;
 228       if (Actual != dataSize)
 229          d->Done = true;
 230       d->End += Actual;
 231    }
 232
 233    if (d->Done == true)
 234    {
 235       if (EndSize <= 3 && Actual == 0)
 236          return false;
 237       if (d->Size - (d->End - d->Buffer) < 4)
 238          return true;
 239
 240       // Append a double new line if one does not exist
 241       unsigned int LineCount = 0;
 242       for (const char *E = d->End - 1; E - d->End < 6 && (*E == '\n' || *E == '\r'); E--)
 243          if (*E == '\n')
 244             LineCount++;
 245       if (LineCount < 2)
 246       {
 247          if ((unsigned)(d->End - d->Buffer) >= d->Size)
 248             Resize(d->Size + 3);
 249          for (; LineCount < 2; LineCount++)
 250             *d->End++ = '\n';
 251       }
 252
 253       return true;
 254    }
 255
 256    return true;
 257 }
 258                                                                         /*}}}*/
 259 // TagFile::Jump - Jump to a pre-recorded location in the file          /*{{{*/
 260 // ---------------------------------------------------------------------
 261 /* This jumps to a pre-recorded file location and reads the record
 262    that is there */
 263 bool pkgTagFile::Jump(pkgTagSection &Tag,unsigned long long Offset)
 264 {
 265    // We are within a buffer space of the next hit..
 266    if (Offset >= d->iOffset && d->iOffset + (d->End - d->Start) > Offset)
 267    {
 268       unsigned long long Dist = Offset - d->iOffset;
 269       d->Start += Dist;
 270       d->iOffset += Dist;
 271       // if we have seen the end, don't ask for more
 272       if (d->Done == true)
 273          return Tag.Scan(d->Start, d->End - d->Start);
 274       else
 275          return Step(Tag);
 276    }
 277
 278    // Reposition and reload..
 279    d->iOffset = Offset;
 280    d->Done = false;
 281    if (d->Fd->Seek(Offset) == false)
 282       return false;
 283    d->End = d->Start = d->Buffer;
 284
 285    if (Fill() == false)
 286       return false;
 287
 288    if (Tag.Scan(d->Start, d->End - d->Start) == true)
 289       return true;
 290
 291    // This appends a double new line (for the real eof handling)
 292    if (Fill() == false)
 293       return false;
 294
 295    if (Tag.Scan(d->Start, d->End - d->Start, false) == false)
 296       return _error->Error(_("Unable to parse package file %s (%d)"),d->Fd->Name().c_str(), 2);
 297
 298    return true;
 299 }
 300                                                                         /*}}}*/
 301 // pkgTagSection::pkgTagSection - Constructor                           /*{{{*/
 302 // ---------------------------------------------------------------------
 303 /* */
 304 APT_IGNORE_DEPRECATED_PUSH
 305 pkgTagSection::pkgTagSection()
 306    : Section(0), d(new pkgTagSectionPrivate()), Stop(0)
 307 {
 308    memset(&AlphaIndexes, 0, sizeof(AlphaIndexes));
 309 }
 310 APT_IGNORE_DEPRECATED_POP
 311                                                                         /*}}}*/
 312 // TagSection::Scan - Scan for the end of the header information        /*{{{*/
 313 bool pkgTagSection::Scan(const char *Start,unsigned long MaxLength, bool const Restart)
 314 {
 315    Section = Start;
 316    const char *End = Start + MaxLength;
 317
 318    if (Restart == false && d->Tags.empty() == false)
 319    {
 320       Stop = Section + d->Tags.back().StartTag;
 321       if (End <= Stop)
 322          return false;
 323       Stop = (const char *)memchr(Stop,'\n',End - Stop);
 324       if (Stop == NULL)
 325          return false;
 326       ++Stop;
 327    }
 328    else
 329    {
 330       Stop = Section;
 331       if (d->Tags.empty() == false)
 332       {
 333          memset(&AlphaIndexes, 0, sizeof(AlphaIndexes));
 334          d->Tags.clear();
 335       }
 336       d->Tags.reserve(0x100);
 337    }
 338    unsigned int TagCount = d->Tags.size();
 339
 340    if (Stop == 0)
 341       return false;
 342
 343    pkgTagSectionPrivate::TagData lastTagData(0);
 344    lastTagData.EndTag = 0;
 345    unsigned long lastTagHash = 0;
 346    while (Stop < End)
 347    {
 348       TrimRecord(true,End);
 349
 350       // this can happen when TrimRecord trims away the entire Record
 351       // (e.g. because it just contains comments)
 352       if(Stop == End)
 353          return true;
 354
 355       // Start a new index and add it to the hash
 356       if (isspace_ascii(Stop[0]) == 0)
 357       {
 358          // store the last found tag
 359          if (lastTagData.EndTag != 0)
 360          {
 361             if (AlphaIndexes[lastTagHash] != 0)
 362                lastTagData.NextInBucket = AlphaIndexes[lastTagHash];
 363             APT_IGNORE_DEPRECATED_PUSH
 364             AlphaIndexes[lastTagHash] = TagCount;
 365             APT_IGNORE_DEPRECATED_POP
 366             d->Tags.push_back(lastTagData);
 367          }
 368
 369          APT_IGNORE_DEPRECATED(++TagCount;)
 370          lastTagData = pkgTagSectionPrivate::TagData(Stop - Section);
 371          // find the colon separating tag and value
 372          char const * Colon = (char const *) memchr(Stop, ':', End - Stop);
 373          if (Colon == NULL)
 374             return false;
 375          // find the end of the tag (which might or might not be the colon)
 376          char const * EndTag = Colon;
 377          --EndTag;
 378          for (; EndTag > Stop && isspace_ascii(*EndTag) != 0; --EndTag)
 379             ;
 380          ++EndTag;
 381          lastTagData.EndTag = EndTag - Section;
 382          lastTagHash = AlphaHash(Stop, EndTag - Stop);
 383          // find the beginning of the value
 384          Stop = Colon + 1;
 385          for (; isspace_ascii(*Stop) != 0; ++Stop);
 386          if (Stop >= End)
 387             return false;
 388          lastTagData.StartValue = Stop - Section;
 389       }
 390
 391       Stop = (const char *)memchr(Stop,'\n',End - Stop);
 392
 393       if (Stop == 0)
 394          return false;
 395
 396       for (; Stop+1 < End && Stop[1] == '\r'; Stop++)
 397          /* nothing */
 398          ;
 399
 400       // Double newline marks the end of the record
 401       if (Stop+1 < End && Stop[1] == '\n')
 402       {
 403          if (lastTagData.EndTag != 0)
 404          {
 405             if (AlphaIndexes[lastTagHash] != 0)
 406                lastTagData.NextInBucket = AlphaIndexes[lastTagHash];
 407             APT_IGNORE_DEPRECATED(AlphaIndexes[lastTagHash] = TagCount;)
 408             d->Tags.push_back(lastTagData);
 409          }
 410
 411          pkgTagSectionPrivate::TagData const td(Stop - Section);
 412          d->Tags.push_back(td);
 413          TrimRecord(false,End);
 414          return true;
 415       }
 416
 417       Stop++;
 418    }
 419
 420    return false;
 421 }
 422                                                                         /*}}}*/
 423 // TagSection::TrimRecord - Trim off any garbage before/after a record  /*{{{*/
 424 // ---------------------------------------------------------------------
 425 /* There should be exactly 2 newline at the end of the record, no more. */
 426 void pkgTagSection::TrimRecord(bool BeforeRecord, const char*& End)
 427 {
 428    if (BeforeRecord == true)
 429       return;
 430    for (; Stop < End && (Stop[0] == '\n' || Stop[0] == '\r'); Stop++);
 431 }
 432                                                                         /*}}}*/
 433 // TagSection::Trim - Trim off any trailing garbage                     /*{{{*/
 434 // ---------------------------------------------------------------------
 435 /* There should be exactly 1 newline at the end of the buffer, no more. */
 436 void pkgTagSection::Trim()
 437 {
 438    for (; Stop > Section + 2 && (Stop[-2] == '\n' || Stop[-2] == '\r'); Stop--);
 439 }
 440                                                                         /*}}}*/
 441 // TagSection::Exists - return True if a tag exists                     /*{{{*/
 442 bool pkgTagSection::Exists(const char* const Tag) const
 443 {
 444    unsigned int tmp;
 445    return Find(Tag, tmp);
 446 }
 447                                                                         /*}}}*/
 448 // TagSection::Find - Locate a tag                                      /*{{{*/
 449 // ---------------------------------------------------------------------
 450 /* This searches the section for a tag that matches the given string. */
 451 bool pkgTagSection::Find(const char *Tag,unsigned int &Pos) const
 452 {
 453    size_t const Length = strlen(Tag);
 454    unsigned int Bucket = AlphaIndexes[AlphaHash(Tag, Length)];
 455    if (Bucket == 0)
 456       return false;
 457
 458    for (; Bucket != 0; Bucket = d->Tags[Bucket - 1].NextInBucket)
 459    {
 460       if ((d->Tags[Bucket - 1].EndTag - d->Tags[Bucket - 1].StartTag) != Length)
 461          continue;
 462
 463       char const * const St = Section + d->Tags[Bucket - 1].StartTag;
 464       if (strncasecmp(Tag,St,Length) != 0)
 465          continue;
 466
 467       Pos = Bucket - 1;
 468       return true;
 469    }
 470
 471    Pos = 0;
 472    return false;
 473 }
 474 bool pkgTagSection::Find(const char *Tag,const char *&Start,
 475                          const char *&End) const
 476 {
 477    unsigned int Pos;
 478    if (Find(Tag, Pos) == false)
 479       return false;
 480
 481    Start = Section + d->Tags[Pos].StartValue;
 482    // Strip off the gunk from the end
 483    End = Section + d->Tags[Pos + 1].StartTag;
 484    if (unlikely(Start > End))
 485       return _error->Error("Internal parsing error");
 486
 487    for (; isspace_ascii(End[-1]) != 0 && End > Start; --End);
 488
 489    return true;
 490 }
 491                                                                         /*}}}*/
 492 // TagSection::FindS - Find a string                                    /*{{{*/
 493 string pkgTagSection::FindS(const char *Tag) const
 494 {
 495    const char *Start;
 496    const char *End;
 497    if (Find(Tag,Start,End) == false)
 498       return string();
 499    return string(Start,End);
 500 }
 501                                                                         /*}}}*/
 502 // TagSection::FindRawS - Find a string                                 /*{{{*/
 503 string pkgTagSection::FindRawS(const char *Tag) const
 504 {
 505    unsigned int Pos;
 506    if (Find(Tag, Pos) == false)
 507       return "";
 508
 509    char const *Start = (char const *) memchr(Section + d->Tags[Pos].EndTag, ':', d->Tags[Pos].StartValue - d->Tags[Pos].EndTag);
 510    ++Start;
 511    char const *End = Section + d->Tags[Pos + 1].StartTag;
 512    if (unlikely(Start > End))
 513       return "";
 514
 515    for (; isspace_ascii(End[-1]) != 0 && End > Start; --End);
 516
 517    return std::string(Start, End - Start);
 518 }
 519                                                                         /*}}}*/
 520 // TagSection::FindI - Find an integer                                  /*{{{*/
 521 // ---------------------------------------------------------------------
 522 /* */
 523 signed int pkgTagSection::FindI(const char *Tag,signed long Default) const
 524 {
 525    const char *Start;
 526    const char *Stop;
 527    if (Find(Tag,Start,Stop) == false)
 528       return Default;
 529
 530    // Copy it into a temp buffer so we can use strtol
 531    char S[300];
 532    if ((unsigned)(Stop - Start) >= sizeof(S))
 533       return Default;
 534    strncpy(S,Start,Stop-Start);
 535    S[Stop - Start] = 0;
 536
 537    errno = 0;
 538    char *End;
 539    signed long Result = strtol(S,&End,10);
 540    if (errno == ERANGE ||
 541        Result < std::numeric_limits<int>::min() || Result > std::numeric_limits<int>::max()) {
 542       errno = ERANGE;
 543       _error->Error(_("Cannot convert %s to integer: out of range"), S);
 544    }
 545    if (S == End)
 546       return Default;
 547    return Result;
 548 }
 549                                                                         /*}}}*/
 550 // TagSection::FindULL - Find an unsigned long long integer             /*{{{*/
 551 // ---------------------------------------------------------------------
 552 /* */
 553 unsigned long long pkgTagSection::FindULL(const char *Tag, unsigned long long const &Default) const
 554 {
 555    const char *Start;
 556    const char *Stop;
 557    if (Find(Tag,Start,Stop) == false)
 558       return Default;
 559
 560    // Copy it into a temp buffer so we can use strtoull
 561    char S[100];
 562    if ((unsigned)(Stop - Start) >= sizeof(S))
 563       return Default;
 564    strncpy(S,Start,Stop-Start);
 565    S[Stop - Start] = 0;
 566
 567    char *End;
 568    unsigned long long Result = strtoull(S,&End,10);
 569    if (S == End)
 570       return Default;
 571    return Result;
 572 }
 573                                                                         /*}}}*/
 574 // TagSection::FindB - Find boolean value                               /*{{{*/
 575 // ---------------------------------------------------------------------
 576 /* */
 577 bool pkgTagSection::FindB(const char *Tag, bool const &Default) const
 578 {
 579    const char *Start, *Stop;
 580    if (Find(Tag, Start, Stop) == false)
 581       return Default;
 582    return StringToBool(string(Start, Stop));
 583 }
 584                                                                         /*}}}*/
 585 // TagSection::FindFlag - Locate a yes/no type flag                     /*{{{*/
 586 // ---------------------------------------------------------------------
 587 /* The bits marked in Flag are masked on/off in Flags */
 588 bool pkgTagSection::FindFlag(const char * const Tag, uint8_t &Flags,
 589                              uint8_t const Flag) const
 590 {
 591    const char *Start;
 592    const char *Stop;
 593    if (Find(Tag,Start,Stop) == false)
 594       return true;
 595    return FindFlag(Flags, Flag, Start, Stop);
 596 }
 597 bool pkgTagSection::FindFlag(uint8_t &Flags, uint8_t const Flag,
 598                                         char const* const Start, char const* const Stop)
 599 {
 600    switch (StringToBool(string(Start, Stop)))
 601    {
 602       case 0:
 603       Flags &= ~Flag;
 604       return true;
 605
 606       case 1:
 607       Flags |= Flag;
 608       return true;
 609
 610       default:
 611       _error->Warning("Unknown flag value: %s",string(Start,Stop).c_str());
 612       return true;
 613    }
 614    return true;
 615 }
 616 bool pkgTagSection::FindFlag(const char *Tag,unsigned long &Flags,
 617                              unsigned long Flag) const
 618 {
 619    const char *Start;
 620    const char *Stop;
 621    if (Find(Tag,Start,Stop) == false)
 622       return true;
 623    return FindFlag(Flags, Flag, Start, Stop);
 624 }
 625 bool pkgTagSection::FindFlag(unsigned long &Flags, unsigned long Flag,
 626                                         char const* Start, char const* Stop)
 627 {
 628    switch (StringToBool(string(Start, Stop)))
 629    {
 630       case 0:
 631       Flags &= ~Flag;
 632       return true;
 633
 634       case 1:
 635       Flags |= Flag;
 636       return true;
 637
 638       default:
 639       _error->Warning("Unknown flag value: %s",string(Start,Stop).c_str());
 640       return true;
 641    }
 642    return true;
 643 }
 644                                                                         /*}}}*/
 645 void pkgTagSection::Get(const char *&Start,const char *&Stop,unsigned int I) const
 646 {
 647    Start = Section + d->Tags[I].StartTag;
 648    Stop = Section + d->Tags[I+1].StartTag;
 649 }
 650 APT_PURE unsigned int pkgTagSection::Count() const {                    /*{{{*/
 651    if (d->Tags.empty() == true)
 652       return 0;
 653    // the last element is just marking the end and isn't a real one
 654    return d->Tags.size() - 1;
 655 }
 656                                                                         /*}}}*/
 657 // TagSection::Write - Ordered (re)writing of fields                    /*{{{*/
 658 pkgTagSection::Tag pkgTagSection::Tag::Remove(std::string const &Name)
 659 {
 660    return Tag(REMOVE, Name, "");
 661 }
 662 pkgTagSection::Tag pkgTagSection::Tag::Rename(std::string const &OldName, std::string const &NewName)
 663 {
 664    return Tag(RENAME, OldName, NewName);
 665 }
 666 pkgTagSection::Tag pkgTagSection::Tag::Rewrite(std::string const &Name, std::string const &Data)
 667 {
 668    if (Data.empty() == true)
 669       return Tag(REMOVE, Name, "");
 670    else
 671       return Tag(REWRITE, Name, Data);
 672 }
 673 static bool WriteTag(FileFd &File, std::string Tag, std::string const &Value)
 674 {
 675    if (Value.empty() || isspace_ascii(Value[0]) != 0)
 676       Tag.append(":");
 677    else
 678       Tag.append(": ");
 679    Tag.append(Value);
 680    Tag.append("\n");
 681    return File.Write(Tag.c_str(), Tag.length());
 682 }
 683 static bool RewriteTags(FileFd &File, pkgTagSection const * const This, char const * const Tag,
 684       std::vector<pkgTagSection::Tag>::const_iterator &R,
 685       std::vector<pkgTagSection::Tag>::const_iterator const &REnd)
 686 {
 687    size_t const TagLen = strlen(Tag);
 688    for (; R != REnd; ++R)
 689    {
 690       std::string data;
 691       if (R->Name.length() == TagLen && strncasecmp(R->Name.c_str(), Tag, R->Name.length()) == 0)
 692       {
 693          if (R->Action != pkgTagSection::Tag::REWRITE)
 694             break;
 695          data = R->Data;
 696       }
 697       else if(R->Action == pkgTagSection::Tag::RENAME && R->Data.length() == TagLen &&
 698             strncasecmp(R->Data.c_str(), Tag, R->Data.length()) == 0)
 699          data = This->FindRawS(R->Name.c_str());
 700       else
 701          continue;
 702
 703       return WriteTag(File, Tag, data);
 704    }
 705    return true;
 706 }
 707 bool pkgTagSection::Write(FileFd &File, char const * const * const Order, std::vector<Tag> const &Rewrite) const
 708 {
 709    // first pass: Write everything we have an order for
 710    if (Order != NULL)
 711    {
 712       for (unsigned int I = 0; Order[I] != 0; ++I)
 713       {
 714          std::vector<Tag>::const_iterator R = Rewrite.begin();
 715          if (RewriteTags(File, this, Order[I], R, Rewrite.end()) == false)
 716             return false;
 717          if (R != Rewrite.end())
 718             continue;
 719
 720          if (Exists(Order[I]) == false)
 721             continue;
 722
 723          if (WriteTag(File, Order[I], FindRawS(Order[I])) == false)
 724             return false;
 725       }
 726    }
 727    // second pass: See if we have tags which aren't ordered
 728    if (d->Tags.empty() == false)
 729    {
 730       for (std::vector<pkgTagSectionPrivate::TagData>::const_iterator T = d->Tags.begin(); T != d->Tags.end() - 1; ++T)
 731       {
 732          char const * const fieldname = Section + T->StartTag;
 733          size_t fieldnamelen = T->EndTag - T->StartTag;
 734          if (Order != NULL)
 735          {
 736             unsigned int I = 0;
 737             for (; Order[I] != 0; ++I)
 738             {
 739                if (fieldnamelen == strlen(Order[I]) && strncasecmp(fieldname, Order[I], fieldnamelen) == 0)
 740                   break;
 741             }
 742             if (Order[I] != 0)
 743                continue;
 744          }
 745
 746          std::string const name(fieldname, fieldnamelen);
 747          std::vector<Tag>::const_iterator R = Rewrite.begin();
 748          if (RewriteTags(File, this, name.c_str(), R, Rewrite.end()) == false)
 749             return false;
 750          if (R != Rewrite.end())
 751             continue;
 752
 753          if (WriteTag(File, name, FindRawS(name.c_str())) == false)
 754             return false;
 755       }
 756    }
 757    // last pass: see if there are any rewrites remaining we haven't done yet
 758    for (std::vector<Tag>::const_iterator R = Rewrite.begin(); R != Rewrite.end(); ++R)
 759    {
 760       if (R->Action == Tag::REMOVE)
 761          continue;
 762       std::string const name = ((R->Action == Tag::RENAME) ? R->Data : R->Name);
 763       if (Exists(name.c_str()))
 764          continue;
 765       if (Order != NULL)
 766       {
 767          unsigned int I = 0;
 768          for (; Order[I] != 0; ++I)
 769          {
 770             if (strncasecmp(name.c_str(), Order[I], name.length()) == 0 && name.length() == strlen(Order[I]))
 771                break;
 772          }
 773          if (Order[I] != 0)
 774             continue;
 775       }
 776
 777       if (WriteTag(File, name, ((R->Action == Tag::RENAME) ? FindRawS(R->Name.c_str()) : R->Data)) == false)
 778          return false;
 779    }
 780    return true;
 781 }
 782                                                                         /*}}}*/
 783
 784 void pkgUserTagSection::TrimRecord(bool /*BeforeRecord*/, const char* &End)/*{{{*/
 785 {
 786    for (; Stop < End && (Stop[0] == '\n' || Stop[0] == '\r' || Stop[0] == '#'); Stop++)
 787       if (Stop[0] == '#')
 788          Stop = (const char*) memchr(Stop,'\n',End-Stop);
 789 }
 790                                                                         /*}}}*/
 791
 792 #include "tagfile-order.c"
 793
 794 // TFRewrite - Rewrite a control record                                 /*{{{*/
 795 // ---------------------------------------------------------------------
 796 /* This writes the control record to stdout rewriting it as necessary. The
 797    override map item specificies the rewriting rules to follow. This also
 798    takes the time to sort the feild list. */
 799 APT_IGNORE_DEPRECATED_PUSH
 800 bool TFRewrite(FILE *Output,pkgTagSection const &Tags,const char *Order[],
 801                TFRewriteData *Rewrite)
 802 {
 803    unsigned char Visited[256];   // Bit 1 is Order, Bit 2 is Rewrite
 804    for (unsigned I = 0; I != 256; I++)
 805       Visited[I] = 0;
 806
 807    // Set new tag up as necessary.
 808    for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
 809    {
 810       if (Rewrite[J].NewTag == 0)
 811          Rewrite[J].NewTag = Rewrite[J].Tag;
 812    }
 813
 814    // Write all all of the tags, in order.
 815    if (Order != NULL)
 816    {
 817       for (unsigned int I = 0; Order[I] != 0; I++)
 818       {
 819          bool Rewritten = false;
 820
 821          // See if this is a field that needs to be rewritten
 822          for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
 823          {
 824             if (strcasecmp(Rewrite[J].Tag,Order[I]) == 0)
 825             {
 826                Visited[J] |= 2;
 827                if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
 828                {
 829                   if (isspace_ascii(Rewrite[J].Rewrite[0]))
 830                      fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 831                   else
 832                      fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 833                }
 834                Rewritten = true;
 835                break;
 836             }
 837          }
 838
 839          // See if it is in the fragment
 840          unsigned Pos;
 841          if (Tags.Find(Order[I],Pos) == false)
 842             continue;
 843          Visited[Pos] |= 1;
 844
 845          if (Rewritten == true)
 846             continue;
 847
 848          /* Write out this element, taking a moment to rewrite the tag
 849             in case of changes of case. */
 850          const char *Start;
 851          const char *Stop;
 852          Tags.Get(Start,Stop,Pos);
 853
 854          if (fputs(Order[I],Output) < 0)
 855             return _error->Errno("fputs","IO Error to output");
 856          Start += strlen(Order[I]);
 857          if (fwrite(Start,Stop - Start,1,Output) != 1)
 858             return _error->Errno("fwrite","IO Error to output");
 859          if (Stop[-1] != '\n')
 860             fprintf(Output,"\n");
 861       }
 862    }
 863
 864    // Now write all the old tags that were missed.
 865    for (unsigned int I = 0; I != Tags.Count(); I++)
 866    {
 867       if ((Visited[I] & 1) == 1)
 868          continue;
 869
 870       const char *Start;
 871       const char *Stop;
 872       Tags.Get(Start,Stop,I);
 873       const char *End = Start;
 874       for (; End < Stop && *End != ':'; End++);
 875
 876       // See if this is a field that needs to be rewritten
 877       bool Rewritten = false;
 878       for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
 879       {
 880          if (stringcasecmp(Start,End,Rewrite[J].Tag) == 0)
 881          {
 882             Visited[J] |= 2;
 883             if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
 884             {
 885                if (isspace_ascii(Rewrite[J].Rewrite[0]))
 886                   fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 887                else
 888                   fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 889             }
 890
 891             Rewritten = true;
 892             break;
 893          }
 894       }
 895
 896       if (Rewritten == true)
 897          continue;
 898
 899       // Write out this element
 900       if (fwrite(Start,Stop - Start,1,Output) != 1)
 901          return _error->Errno("fwrite","IO Error to output");
 902       if (Stop[-1] != '\n')
 903          fprintf(Output,"\n");
 904    }
 905
 906    // Now write all the rewrites that were missed
 907    for (unsigned int J = 0; Rewrite != 0 && Rewrite[J].Tag != 0; J++)
 908    {
 909       if ((Visited[J] & 2) == 2)
 910          continue;
 911
 912       if (Rewrite[J].Rewrite != 0 && Rewrite[J].Rewrite[0] != 0)
 913       {
 914          if (isspace_ascii(Rewrite[J].Rewrite[0]))
 915             fprintf(Output,"%s:%s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 916          else
 917             fprintf(Output,"%s: %s\n",Rewrite[J].NewTag,Rewrite[J].Rewrite);
 918       }
 919    }
 920
 921    return true;
 922 }
 923 APT_IGNORE_DEPRECATED_POP
 924                                                                         /*}}}*/
 925
 926 pkgTagSection::~pkgTagSection() { delete d; }