src/ufsdump.cc

   1 /*
   2  * Copyright (C) 1996-2017 The Squid Software Foundation and contributors
   3  *
   4  * Squid software is distributed under GPLv2+ license and includes
   5  * contributions from numerous individuals and organizations.
   6  * Please see the COPYING and CONTRIBUTORS files for details.
   7  */
   8
   9 /* DEBUG: section 00    UFS Store Dump Tool */
  10
  11 #include "squid.h"
  12 #include "Generic.h"
  13 #include "md5.h"
  14 #include "mgr/Registration.h"
  15 #include "Store.h"
  16 #include "store_key_md5.h"
  17 #include "StoreMeta.h"
  18 #include "StoreMetaUnpacker.h"
  19
  20 #undef malloc
  21 #undef free
  22
  23 #include <cassert>
  24 #include <iostream>
  25 #include <stdexcept>
  26
  27 /* stub functions for parts of squid not factored to be dynamic yet */
  28 void
  29 eventAdd(const char *name, EVH * func, void *arg, double when, int, bool cbdata)
  30 {}
  31
  32 // required by storeKeyPublicByRequest*
  33 // XXX: what pulls in storeKeyPublicByRequest?
  34 const char *urlCanonical(HttpRequest *) { assert(false); return NULL; }
  35
  36 void
  37 storeAppendPrintf(StoreEntry * e, const char *fmt,...)
  38 {
  39     va_list args;
  40     va_start(args, fmt);
  41
  42     assert(false);
  43
  44     va_end(args);
  45 }
  46
  47 void
  48 Mgr::RegisterAction(char const * action, char const * desc, OBJH * handler, int pw_req_flag, int atomic) {}
  49
  50 /* MinGW needs also a stub of death() */
  51 void
  52 death(int sig)
  53 {
  54     std::cout << "Fatal: Signal " <<  sig;
  55     exit(1);
  56 }
  57
  58 void
  59 fatal(const char *message)
  60 {
  61     fprintf(stderr, "FATAL: %s\n", message);
  62     exit(1);
  63 }
  64
  65 /* end stub functions */
  66
  67 struct MetaStd {
  68     time_t timestamp;
  69     time_t lastref;
  70     time_t expires;
  71     time_t lastmod;
  72     size_t swap_file_sz;
  73     uint16_t refcount;
  74     uint16_t flags;
  75 };
  76
  77 struct MetaStdLfs {
  78     time_t timestamp;
  79     time_t lastref;
  80     time_t expires;
  81     time_t lastmod;
  82     uint64_t swap_file_sz;
  83     uint16_t refcount;
  84     uint16_t flags;
  85 };
  86
  87 struct DumpStoreMeta : public unary_function<StoreMeta, void> {
  88     DumpStoreMeta() {}
  89
  90     void operator()(StoreMeta const &x) {
  91         switch (x.getType()) {
  92
  93         case STORE_META_KEY:
  94             std::cout << "MD5: " << storeKeyText((const cache_key *)x.value) << std::endl;
  95             break;
  96
  97         case STORE_META_STD:
  98             std::cout << "STD, Size:" << ((struct MetaStd*)x.value)->swap_file_sz <<
  99                       " Flags: 0x" << std::hex << ((struct MetaStd*)x.value)->flags << std::dec <<
 100                       " Refcount: " << ((struct MetaStd*)x.value)->refcount <<
 101                       std::endl;
 102             break;
 103
 104         case STORE_META_STD_LFS:
 105             std::cout << "STD_LFS, Size: " << ((struct MetaStdLfs*)x.value)->swap_file_sz <<
 106                       " Flags: 0x" << std::hex << ((struct MetaStdLfs*)x.value)->flags << std::dec <<
 107                       " Refcount: " << ((struct MetaStdLfs*)x.value)->refcount <<
 108                       std::endl;
 109             break;
 110
 111         case STORE_META_URL:
 112             assert (((char *)x.value)[x.length - 1] == 0);
 113             std::cout << "URL: " << (char *)x.value << std::endl;
 114             break;
 115
 116         default:
 117             std::cout << "Unknown store meta type: " << (int)x.getType() <<
 118                       " of length " << x.length << std::endl;
 119             break;
 120         }
 121     }
 122 };
 123
 124 int
 125 main(int argc, char *argv[])
 126 {
 127     int fd = -1;
 128     StoreMeta *metadata = NULL;
 129
 130     try {
 131         if (argc != 2)
 132             throw std::runtime_error("No filename provided");
 133
 134         fd = open (argv[1], O_RDONLY | O_BINARY);
 135
 136         if (fd < 0)
 137             throw std::runtime_error("Could not open file.");
 138
 139         char tempbuf[SM_PAGE_SIZE];
 140
 141         int len = read(fd, tempbuf, SM_PAGE_SIZE);
 142
 143         if (len < 0)
 144             throw std::runtime_error("Could not read header into memory.");
 145
 146         close (fd);
 147
 148         fd = -1;
 149
 150         int hdr_len;
 151
 152         StoreMetaUnpacker aBuilder(tempbuf, len, &hdr_len);
 153
 154         metadata = aBuilder.createStoreMeta ();
 155
 156         cache_key key[SQUID_MD5_DIGEST_LENGTH];
 157
 158         memset(key, '\0', SQUID_MD5_DIGEST_LENGTH);
 159
 160         DumpStoreMeta dumper;
 161
 162         for_each(*metadata, dumper);
 163
 164         return 0;
 165     } catch (std::runtime_error error) {
 166         std::cout << "Failed : " << error.what() << std::endl;
 167
 168         if (fd >= 0)
 169             close(fd);
 170
 171         if (metadata)
 172             StoreMeta::FreeList(&metadata);
 173
 174         return 1;
 175     }
 176 }
 177