json_gssapi/src/cache/GSSContextCache.cpp

   1 /*
   2  * Copyright (c) 2014, 2015 JANET(UK)
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  *
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * 3. Neither the name of JANET(UK) nor the names of its contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  24  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  31  * OF THE POSSIBILITY OF SUCH DAMAGE.
  32  *
  33  */
  34
  35 #include <stdexcept>
  36 #include "utils/util_random.h"
  37 #include "utils/util_base64.h"
  38
  39 #include "GSSContextCache.h"
  40
  41 #define KEYLEN 128
  42
  43 GSSContextCache* GSSContextCache::_instance = 0;
  44
  45 GSSContextCache::GSSContextCache()
  46 {
  47
  48 }
  49
  50 GSSContextCache::~GSSContextCache()
  51 {
  52
  53 }
  54
  55 GSSContextCache* GSSContextCache::instance()
  56 {
  57     if (_instance == 0)
  58       _instance = new GSSContextCache;
  59
  60     return _instance;
  61 }
  62
  63
  64 std::string GSSContextCache::store ( GSSContext& data, std::string inKey )
  65 {
  66   /* Variables      */
  67   std::string key;
  68
  69   /* Error checking */
  70   /* Setup          */
  71   /* Main           */
  72   // Generate a key
  73   if ( inKey.length() > 0 )
  74   {
  75     key = inKey;
  76   }
  77   else if ( !generateKey(key) )
  78   {
  79     // Key generation failed.  Eeek!
  80     throw std::runtime_error("Could not generate random data for an ID");
  81   }
  82
  83   // Store the key/value pair in the map
  84   // Store the key in the context for convenience
  85   contexts[key] = data;
  86
  87   /* Cleanup        */
  88   /* Return         */
  89   // Return the key for future reference
  90   return(key);
  91 }
  92
  93 /*************************************
  94  * Generate random bytes, and base64 *
  95  * encode them to be JSONable keys   *
  96  *************************************/
  97 bool GSSContextCache::generateKey(std::string &key)
  98 {
  99   /* Variables      */
 100   unsigned char theKey[KEYLEN];
 101
 102   if (!randGenPseudoRandom(theKey, KEYLEN))
 103       return false;
 104
 105   // Encode the binary string
 106   base64EncodeStr(theKey, KEYLEN, key);
 107
 108   /* Cleanup        */
 109   /* Return         */
 110   return(true);
 111 }
 112
 113 GSSContext& GSSContextCache::retrieve ( std::string key )
 114 {
 115   /* Variables      */
 116   //GSSContext data;
 117
 118   /* Error checking */
 119   /* Setup          */
 120   /* Main           */
 121   // Maybe do something about data entries that are expired?
 122
 123   /* Cleanup        */
 124   /* Return         */
 125   return contexts[key];
 126 }
 127
 128   /* Variables      */
 129   /* Error checking */
 130   /* Setup          */
 131   /* Main           */
 132   /* Cleanup        */
 133   /* Return         */