json_gssapi/src/cache/GSSCredentialCache.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 "GSSCredentialCache.h"
  37 #include "utils/util_base64.h"
  38 #include "utils/util_random.h"
  39
  40 #define KEYLEN 128
  41
  42 GSSCredentialCache* GSSCredentialCache::_instance = 0;
  43
  44 GSSCredentialCache::GSSCredentialCache()
  45 {
  46
  47 }
  48
  49 GSSCredentialCache::GSSCredentialCache ( const GSSCredentialCache& other )
  50 {
  51   credentials = other.credentials;
  52 }
  53
  54 GSSCredentialCache::~GSSCredentialCache()
  55 {
  56
  57 }
  58
  59 GSSCredentialCache& GSSCredentialCache::operator= ( const GSSCredentialCache& other )
  60 {
  61   credentials = other.credentials;
  62
  63   return *this;
  64 }
  65
  66
  67 std::string GSSCredentialCache::store ( GSSCredential& data, std::string inKey )
  68 {
  69   /* Variables      */
  70   std::string key;
  71
  72   /* Error checking */
  73   /* Setup          */
  74   /* Main           */
  75   // Generate a key
  76   if ( inKey.length() > 0 )
  77   {
  78     key = inKey;
  79   }
  80   else if ( !generateKey(key) )
  81   {
  82     // Key generation failed.  Eeek!
  83     throw std::runtime_error("Could not generate random data for an ID");
  84   }
  85
  86   // Store the key/value pair in the map
  87   credentials[key] = &data;
  88
  89   /* Cleanup        */
  90   /* Return         */
  91   // Return the key for future reference
  92   return(key);
  93 }
  94
  95 /*************************************
  96  * Generate random bytes, and base64 *
  97  * encode them to be JSONable keys   *
  98  *************************************/
  99 bool GSSCredentialCache::generateKey(std::string &key)
 100 {
 101   /* Variables      */
 102   unsigned char theKey[KEYLEN];
 103
 104   /* Setup          */
 105   /* Main           */
 106   // Generate random byte string
 107   if (!randGenPseudoRandom(theKey, KEYLEN))
 108     return(false);
 109
 110   // Encode the binary string
 111   base64EncodeStr(theKey, KEYLEN, key);
 112   /* Cleanup        */
 113   /* Return         */
 114   return(true);
 115 }
 116
 117 GSSCredential& GSSCredentialCache::retrieve ( std::string key )
 118 {
 119   /* Variables      */
 120   //GSSCredential data;
 121
 122   /* Error checking */
 123   /* Setup          */
 124   /* Main           */
 125
 126   /* Cleanup        */
 127   /* Return         */
 128   return *(credentials[key]);
 129 }
 130
 131 GSSCredentialCache* GSSCredentialCache::instance()
 132 {
 133     if (_instance == 0)
 134       _instance = new GSSCredentialCache;
 135
 136     return _instance;
 137 }
 138