// **********************************************************************
//
// Copyright (c) 2003
// ZeroC, Inc.
// Billerica, MA, USA
//
// All Rights Reserved.
//
// Ice is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License version 2 as published by
// the Free Software Foundation.
//
// **********************************************************************

#include <Freeze/Util.h>
#include <Freeze/Exception.h>

using namespace Freeze;
using namespace Ice;
using namespace std;

void
Freeze::handleMemoryException(const DbMemoryException& dx, Key& key, Dbt& dbKey)
{
    if(dbKey.get_size() > dbKey.get_ulen())
    {
	//
	// Keep the old key size in case it's used as input
	//
	size_t oldKeySize = key.size();

	key.resize(dbKey.get_size());
	initializeOutDbt(key, dbKey);
	dbKey.set_size(oldKeySize);
    }
    else
    {
	//
	// Real problem
	//
	DatabaseException ex(__FILE__, __LINE__);
	ex.message = dx.what();
	throw ex;
    }
}

void
Freeze::handleMemoryException(const DbMemoryException& dx, Key& key, Dbt& dbKey, 
			      Value& value, Dbt& dbValue)
{
    bool resized = false;
    if(dbKey.get_size() > dbKey.get_ulen())
    {
	size_t oldKeySize = key.size();
	key.resize(dbKey.get_size());
	initializeOutDbt(key, dbKey);
	dbKey.set_size(oldKeySize);
	resized = true;
    }
    
    if(dbValue.get_size() > dbValue.get_ulen())
    {
	value.resize(dbValue.get_size());
	initializeOutDbt(value, dbValue);
	resized = true;
    }
    
    if(!resized)
    {
	//
	// Real problem
	//
	DatabaseException ex(__FILE__, __LINE__);
	ex.message = dx.what();
	throw ex;
    }
}