--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/usr/src/lib/libkmsagent/common/KMSAgentDataUnitCache.cpp	Mon Jun 28 16:04:11 2010 -0700
@@ -0,0 +1,202 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+ * or http://www.opensolaris.org/os/licensing.
+ * See the License for the specific language governing permissions
+ * and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
+ */
+
+/**
+ * \file KMSAgentDataUnitCache.cpp
+ */
+
+#include <stdio.h>
+#include "SYSCommon.h"
+#include "KMSClientProfile.h"
+#include "KMSAgentDataUnitCache.h"
+
+CDataUnitCache::CDataUnitCache(int i_iMaxSize)
+{
+    m_iSize = 0;
+    m_iIndex = 0;
+    m_iMaxSize = i_iMaxSize;
+    m_pCache = 0;
+
+    K_CreateMutex(&m_Lock);
+}
+
+CDataUnitCache::~CDataUnitCache()
+{
+    delete[] m_pCache;
+    K_DestroyMutex(m_Lock);
+}
+
+bool CDataUnitCache::Insert(
+                const unsigned char* const i_pDataUnitID,
+                int i_iDataUnitIDMaxLen, 
+                const unsigned char* const i_pDataUnitKeyID ,
+                int i_iDataUnitKeyIDMaxLen,
+                const utf8char* const i_wsApplianceNetworkAddress )
+{
+    FATAL_ASSERT( (i_pDataUnitID && i_iDataUnitIDMaxLen == KMS_DATA_UNIT_ID_SIZE) ||
+                  (i_pDataUnitKeyID && i_iDataUnitKeyIDMaxLen == KMS_KEY_ID_SIZE));
+    FATAL_ASSERT( i_wsApplianceNetworkAddress && 
+                  strlen( i_wsApplianceNetworkAddress ) < KMS_MAX_NETWORK_ADDRESS );
+
+    Lock();
+
+    if ( m_pCache == 0 )
+    {
+        m_pCache = new DataUnitCacheEntry[m_iMaxSize];
+
+        if ( !m_pCache )
+        {
+            // no error logged on out of memory
+            Unlock();
+#if defined(DEBUG) && defined(METAWARE)
+            log_printf("CDataUnitCache::Insert new DataUnitCacheEntry alloc failure\n");
+#endif
+            return false;
+        }
+    }
+
+    if( m_iSize >= m_iMaxSize )
+    {
+        // the cache is full, so reuse an old slot
+
+        m_iIndex  = (m_iIndex + 1) % m_iMaxSize;
+    }
+    else
+    {
+        m_iIndex = m_iSize;
+
+        m_iSize++;
+    }
+
+    strncpy( m_pCache[m_iIndex].m_wsApplianceNetworkAddress, 
+             i_wsApplianceNetworkAddress, 
+             sizeof(m_pCache[m_iIndex].m_wsApplianceNetworkAddress) );
+    m_pCache[m_iIndex].m_wsApplianceNetworkAddress[sizeof(m_pCache[m_iIndex].m_wsApplianceNetworkAddress)-1] = '\0';
+
+    if ( i_pDataUnitID )
+    {
+        memcpy( m_pCache[m_iIndex].m_aDataUnitID, 
+                i_pDataUnitID, 
+                i_iDataUnitIDMaxLen );
+    }
+    else
+    {
+        memset( m_pCache[m_iIndex].m_aDataUnitID,0,KMS_DATA_UNIT_ID_SIZE);
+    }
+
+    if ( i_pDataUnitKeyID )
+    {
+        memcpy( m_pCache[m_iIndex].m_aDataUnitKeyID, 
+            i_pDataUnitKeyID, 
+            i_iDataUnitKeyIDMaxLen );
+    }
+    else
+    {
+        memset(m_pCache[m_iIndex].m_aDataUnitKeyID,0,KMS_KEY_ID_SIZE);
+    }
+
+    Unlock();
+
+    return true;
+}
+
+bool CDataUnitCache::GetApplianceByDataUnitID(
+                const unsigned char* const i_pDataUnitID,
+                int i_iDataUnitIDMaxLen,
+                utf8char* const o_wsApplianceNetworkAddress,
+                int i_iMaxApplianceNetworkAddressLen )
+{
+    FATAL_ASSERT( i_pDataUnitID );
+    FATAL_ASSERT( i_iDataUnitIDMaxLen == KMS_DATA_UNIT_ID_SIZE );
+    FATAL_ASSERT( i_iMaxApplianceNetworkAddressLen <= KMS_MAX_NETWORK_ADDRESS );
+
+    // assumes o_wsApplianceNetworkAddress points to at least KMS_MAX_NETWORK_ADDRESS
+
+    Lock();
+
+    int i;
+    for( i = 0; i < m_iSize; i++ )
+    {
+        if( memcmp(m_pCache[i].m_aDataUnitID, i_pDataUnitID, KMS_DATA_UNIT_ID_SIZE) == 0 )
+        {
+            strncpy( o_wsApplianceNetworkAddress, 
+                m_pCache[i].m_wsApplianceNetworkAddress, 
+                i_iMaxApplianceNetworkAddressLen );
+            o_wsApplianceNetworkAddress[i_iMaxApplianceNetworkAddressLen-1] = '\0';
+            Unlock();
+            return true;
+        }
+    }
+
+    Unlock();
+
+    return false;
+}
+
+bool CDataUnitCache::GetApplianceByDataUnitKeyID(
+                const unsigned char* const i_pDataUnitKeyID,
+                int i_iDataUnitKeyIDMaxLen,
+                utf8char* const o_wsApplianceNetworkAddress,
+                int i_iMaxApplianceNetworkAddressLen )
+{
+    FATAL_ASSERT( i_pDataUnitKeyID );
+    FATAL_ASSERT( i_iDataUnitKeyIDMaxLen == KMS_KEY_ID_SIZE );
+    FATAL_ASSERT( i_iMaxApplianceNetworkAddressLen <= KMS_MAX_NETWORK_ADDRESS );
+
+    // assumes o_wsApplianceNetworkAddress points to at least KMS_MAX_NETWORK_ADDRESS
+
+    Lock();
+
+    int i;
+    for( i = 0; i < m_iSize; i++ )
+    {
+        if( memcmp(m_pCache[i].m_aDataUnitKeyID,
+                i_pDataUnitKeyID, KMS_KEY_ID_SIZE) == 0 )
+        {
+            strncpy( o_wsApplianceNetworkAddress, 
+                m_pCache[i].m_wsApplianceNetworkAddress, 
+                i_iMaxApplianceNetworkAddressLen );
+            o_wsApplianceNetworkAddress[i_iMaxApplianceNetworkAddressLen-1] = '\0';
+
+            Unlock();
+
+            return true;
+        }
+    }
+
+    Unlock();
+
+    return false;
+}
+
+void CDataUnitCache::Lock()
+{
+    K_LockMutex(m_Lock);
+}
+
+void CDataUnitCache::Unlock()
+{
+    K_UnlockMutex(m_Lock);
+}