xmltooling/util/ReplayCache.cpp

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