tools/leaky/dict.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is mozilla.org code.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Netscape Communications Corporation.
  18  * Portions created by the Initial Developer are Copyright (C) 1999
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *   Kipp E.B. Hickman <kipp@netscape.com> (original author)
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either the GNU General Public License Version 2 or later (the "GPL"), or
  26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 #include <malloc.h>
  39 #include "dict.h"
  40
  41 #ifdef __QNXNTO__
  42 /* Need definition of NULL */
  43 #include <stdio.h>
  44 #endif
  45
  46 MallocDict::MallocDict(int nb)
  47 {
  48     numBuckets = nb;
  49     buckets = (MallocDictEntry**) calloc(numBuckets, sizeof(MallocDictEntry*));
  50     rewind();
  51 }
  52
  53 void MallocDict::rewind(void)
  54 {
  55     iterNextBucket = -1;
  56     iterNextEntry = 0;
  57 }
  58
  59 malloc_log_entry* MallocDict::next(void)
  60 {
  61     if (iterNextEntry) {
  62         iterNextEntry = iterNextEntry->next;
  63     }
  64     while (!iterNextEntry) {
  65         iterNextBucket++;
  66         if (iterNextBucket >= numBuckets) {
  67             return 0;
  68         }
  69         iterNextEntry = buckets[iterNextBucket];
  70     }
  71     return iterNextEntry->logEntry;
  72 }
  73
  74 malloc_log_entry** MallocDict::find(u_long addr)
  75 {
  76     u_long hash = addr % numBuckets;
  77     MallocDictEntry* mde = buckets[hash];
  78     while (mde) {
  79         if (mde->addr == addr) {
  80             return &mde->logEntry;
  81         }
  82         mde = mde->next;
  83     }
  84     return 0;
  85 }
  86
  87 void MallocDict::add(u_long addr, malloc_log_entry *lep)
  88 {
  89     u_long hash = addr % numBuckets;
  90     MallocDictEntry** mdep = &buckets[hash];
  91     MallocDictEntry* mde = new MallocDictEntry;
  92     mde->addr = addr;
  93     mde->logEntry = lep;
  94     mde->next = *mdep;
  95     *mdep = mde;
  96 }
  97
  98 void MallocDict::remove(u_long addr)
  99 {
 100     u_long hash = addr % numBuckets;
 101     MallocDictEntry** mdep = &buckets[hash];
 102     MallocDictEntry* mde;
 103
 104     while (NULL != (mde = *mdep)) {
 105         if (mde->addr == addr) {
 106             *mdep = mde->next;
 107 /*XXX       delete mde; */
 108             return;
 109         }
 110         mdep = &mde->next;
 111     }
 112 }