xmltooling/util/ReplayCache.cpp

   1 /*
   2  *  Copyright 2001-2010 Internet2
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /**
  18  * ReplayCache.cpp
  19  *
  20  * Helper class on top of StorageService for detecting message replay.
  21  */
  22
  23 #include "internal.h"
  24 #include "util/ReplayCache.h"
  25 #include "util/StorageService.h"
  26
  27 using namespace xmltooling;
  28 using namespace std;
  29
  30 ReplayCache::ReplayCache(StorageService* storage) : m_owned(storage==nullptr), m_storage(storage)
  31 {
  32     if (!m_storage)
  33         m_storage = XMLToolingConfig::getConfig().StorageServiceManager.newPlugin(MEMORY_STORAGE_SERVICE, nullptr);
  34 }
  35
  36 ReplayCache::~ReplayCache()
  37 {
  38     if (m_owned)
  39         delete m_storage;
  40 }
  41
  42 bool ReplayCache::check(const char* context, const char* s, time_t expires)
  43 {
  44     // In storage already?
  45     if (m_storage->readString(context, s))
  46         return false;
  47     m_storage->createString(context, s, "x", expires);
  48     return true;
  49 }
  50
  51 bool ReplayCache::check(const char* context, const XMLCh* s, time_t expires)
  52 {
  53     auto_ptr_char temp(s);
  54     return check(context, temp.get(), expires);
  55 }