1 files changed, 140 insertions, 0 deletions

diff --git a/src/hdb/hdbfix.cc b/src/hdb/hdbfix.cc
new file mode 100644
index 0000000..b82f151
--- /dev/null
+++ b/src/hdb/hdbfix.cc
@@ -0,0 +1,140 @@
+/*************************************************************************
+ *
+ * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <hdb/db.h>
+
+static fp_stream mout(stdout),merr(stderr);
+
+main(int argc,char* argv[]) try
+{
+  if(argc<2) {
+    mout<<"Usage: "<<argv[0]<<" [-s] <path>\n";
+    return -1;
+  }
+  mstring path; int silent;
+  if(argc>2) { silent=1; path=argv[2]; }
+  else { silent=0; path=argv[1]; }
+  mstring map; if(map.load(path)<0) THROW("hdbfix: could not open "<<path);
+  filemap::head_t head;
+  mstring version=First(sref(head.ident,sizeof(head.ident)));
+  memcpy(&head,map.data(),sizeof(head));
+  mstring cver=First(sref(head.ident,sizeof(head.ident)));
+  mstring answer;
+  int save=0;
+  if(version!=cver) {
+    merr<<path<<": Not a "<<version<<" file\n";
+    return -1;
+  }
+  else {
+    int top=head.frame.get(),last=0;
+    while(top<head.filetop.get()&&top<map.size()) {
+      filemap::rec_t* rec=(filemap::rec_t*)(map.data()+top);
+      if(rec->mem.get()<=0) {
+        if(!silent&&rec->mem.get()<0)
+          mout<<"ending mem: "<<rec->mem.get()<<"\n";
+        break;
+      }
+      last=top;
+      top+=rec->mem.get();
+    }
+    if(top==head.filetop.get()&&head.filesize.get()==map.size()) {
+      if(!silent) mout<<path<<": OK\n";
+    }
+    else {
+      mout<<path<<": CORRUPTED\n";
+      if(silent) return 0;
+      mout<<"\n";
+      mout<<"Head info:\n";
+      mout<<"Size:     "<<head.filesize.get()<<"\n";
+      mout<<"Frame:    "<<head.frame.get()<<"\n";
+      mout<<"Top:      "<<head.filetop.get()<<"\n";
+      mout<<"\n";
+
+      mout<<"Actual sizes:\n";
+      mout<<"Size:     "<<map.size()<<"\n";
+      mout<<"Last:     "<<last<<"\n";
+      mout<<"End:      "<<top<<"\n";
+      mout<<"\n";
+
+      if(map.size()!=head.filesize.get()) {
+        mout<<"Filesizes differ - do you want to set the filesize to "
+            <<head.filesize.get()<<"?\n";
+        getline(answer,stdin); answer.tolower();
+        if(*answer=='y') {
+          map.resize(head.filesize.get());
+          save=1;
+          mout<<"Done.\n";
+        }
+        else {
+          mout<<"No change - quitting.\n";
+          return 0;
+        }
+      }
+
+      int nend=0,end=head.filetop.get();
+      for(int p=head.filetop.get()-4;p>=last;p-=4){
+        int val=ntohl(*(long*)(map.data()+p));
+        if(p+val==end) { nend++; end=p; }
+      }
+      int gap=end-top;
+
+      mout<<"GAP DATA\n";
+      mout<<"==============================\n";
+      for(int i=top;i<end;i++) {
+        if(cset_printable.test(map[i])) mout.put(map[i]);
+        else mout<<"["<<(int)map[i]<<"]";
+      }
+      mout<<"\n==============================\n";
+      mout<<"\n";
+
+      mout<<"At-end sequence:\n";
+      mout<<"Start:   "<<end<<"\n";
+      mout<<"Gap:     "<<gap<<"\n";
+      mout<<"\n";
+
+      if(gap>0) {
+        mout<<"Do you want to mark the gap as deleted?\n";
+        getline(answer,stdin); answer.tolower();
+        if(*answer=='y') {
+          filemap::rec_t* rec=(filemap::rec_t*)(map.data()+top);
+          rec->mem.set(gap);
+          rec->caps.set(0);
+          save=1;
+          mout<<"Done.\n";
+        }
+        else {
+          mout<<"No change - quitting.\n";
+          return 0;
+        }
+      }
+    }
+  }
+  
+  if(save) map.save(path);
+
+  return 0;
+}
+catch(const merror_t& e) {
+  merr<<e.desc<<"\n";
+  return -1;
+}
+catch(...) {
+  merr<<"FATAL ERROR\n";
+  return -1;
+}