/* Provides autoreleasing of malloc'ed pointers
   Copyright (C) 1995 Free Software Foundation, Inc.
   
   Written by:  R. Andrew McCallum <mccallum@gnu.ai.mit.edu>
   Date: January 1995
   
   This file is part of the GNU Objective C Class Library.

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 of the License, or (at your option) any later version.
   
   This library 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
   Library General Public License for more details.
   
   You should have received a copy of the GNU Library General Public
   License along with this library; if not, write to the Free
   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
   */ 

#include <objects/MallocAddress.h>
#include <objects/Dictionary.h>

static Dictionary* mallocAddresses;

@implementation MallocAddress

+ (void) initialize
{
  if (self == [MallocAddress class])
    {
      mallocAddresses = [[Dictionary alloc] initWithType:@encode(id)
			 keyType:@encode(void*)];
    }
}

+ objectForAddress: (void*)addr
{
  elt ret_nil(arglist_t a)
    {
      return (elt) nil;
    }
  return [mallocAddresses elementAtKey:addr ifAbsentCall:ret_nil].id_u;
}

+ autoreleaseMallocAddresss: (void*)addr
{
  id n = [[self alloc] initWithAddress:addr];
  [mallocAddresses putElement:n atKey:addr];
  return [n autorelease];
}

- initWithAddress: (void*)addr
{
  [super init];
  address = addr;
  return self;
}

- (void) dealloc
{
  if ([[self class] objectForAddress:address])
    [mallocAddresses removeElementAtKey:address];
  OBJC_FREE(address);
  [super dealloc];
}

@end