src/StoreMetaUnpacker.cc

   1
   2 /*
   3  * $Id$
   4  *
   5  * DEBUG: section 20    Storage Manager Swapfile Unpacker
   6  * AUTHOR: Robert Collins
   7  *
   8  * SQUID Web Proxy Cache          http://www.squid-cache.org/
   9  * ----------------------------------------------------------
  10  *
  11  *  Squid is the result of efforts by numerous individuals from
  12  *  the Internet community; see the CONTRIBUTORS file for full
  13  *  details.   Many organizations have provided support for Squid's
  14  *  development; see the SPONSORS file for full details.  Squid is
  15  *  Copyrighted (C) 2001 by the Regents of the University of
  16  *  California; see the COPYRIGHT file for full details.  Squid
  17  *  incorporates software developed and/or copyrighted by other
  18  *  sources; see the CREDITS file for full details.
  19  *
  20  *  This program is free software; you can redistribute it and/or modify
  21  *  it under the terms of the GNU General Public License as published by
  22  *  the Free Software Foundation; either version 2 of the License, or
  23  *  (at your option) any later version.
  24  *
  25  *  This program is distributed in the hope that it will be useful,
  26  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  27  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  28  *  GNU General Public License for more details.
  29  *
  30  *  You should have received a copy of the GNU General Public License
  31  *  along with this program; if not, write to the Free Software
  32  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
  33  *
  34  */
  35
  36 #include "squid-old.h"
  37 #include "StoreMetaUnpacker.h"
  38 #include "StoreMeta.h"
  39
  40 int const StoreMetaUnpacker::MinimumBufferLength = sizeof(char) + sizeof(int);
  41
  42 /// useful for meta stored in pre-initialized (with zeros) db files
  43 bool
  44 StoreMetaUnpacker::isBufferZero()
  45 {
  46     // We could memcmp the entire buffer, but it is probably safe enough
  47     // to test a few bytes because if we do not detect a corrupted entry
  48     // it is not a big deal. Empty entries are not isBufferSane anyway.
  49     const int depth = 10;
  50     if (buflen < depth)
  51         return false; // cannot be sure enough
  52
  53     for (int i = 0; i < depth; ++i) {
  54         if (buf[i])
  55             return false;
  56     }
  57     return true;
  58 }
  59
  60 bool
  61 StoreMetaUnpacker::isBufferSane()
  62 {
  63     if (buf[0] != (char) STORE_META_OK)
  64         return false;
  65
  66     /*
  67      * sanity check on 'buflen' value.  It should be at least big
  68      * enough to hold one type and one length.
  69      */
  70     getBufferLength();
  71
  72     if (*hdr_len < MinimumBufferLength)
  73         return false;
  74
  75     if (*hdr_len > buflen)
  76         return false;
  77
  78     return true;
  79 }
  80
  81 void
  82 StoreMetaUnpacker::getBufferLength()
  83 {
  84     memcpy(hdr_len, &buf[1], sizeof(int));
  85 }
  86
  87 StoreMetaUnpacker::StoreMetaUnpacker (char const *aBuffer, ssize_t aLen, int *anInt) : buf (aBuffer), buflen(aLen), hdr_len(anInt), position(1 + sizeof(int))
  88 {
  89     assert (aBuffer != NULL);
  90 }
  91
  92 void
  93 StoreMetaUnpacker::getType()
  94 {
  95     type = buf[position++];
  96 }
  97
  98 void
  99 StoreMetaUnpacker::getLength()
 100 {
 101     memcpy(&length, &buf[position], sizeof(int));
 102     position += sizeof(int);
 103 }
 104
 105 bool
 106 StoreMetaUnpacker::doOneEntry()
 107 {
 108     getType();
 109     getLength();
 110
 111     if (position + length > *hdr_len) {
 112         debugs(20, 0, "storeSwapMetaUnpack: overflow!");
 113         debugs(20, 0, "\ttype=" << type << ", length=" << length << ", *hdr_len=" << *hdr_len << ", offset=" << position);
 114         return false;
 115     }
 116
 117     StoreMeta *newNode = StoreMeta::Factory(type, length, &buf[position]);
 118
 119     if (newNode)
 120         tail = StoreMeta::Add (tail, newNode);
 121
 122     position += length;
 123
 124     return true;
 125 }
 126
 127 bool
 128 StoreMetaUnpacker::moreToProcess() const
 129 {
 130     return *hdr_len - position - MinimumBufferLength >= 0;
 131 }
 132
 133 StoreMeta *
 134 StoreMetaUnpacker::createStoreMeta ()
 135 {
 136     tlv *TLV = NULL;
 137     tail = &TLV;
 138     assert(hdr_len != NULL);
 139
 140     if (!isBufferSane())
 141         return NULL;
 142
 143     getBufferLength();
 144
 145     assert (position == 1 + sizeof(int));
 146
 147     while (moreToProcess()) {
 148         if (!doOneEntry())
 149             break;
 150     }
 151
 152     return TLV;
 153 }