<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"> <html><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8"> <title>DTN Reference Implementation: KeyDB.cc Source File</title> <link href="doxygen.css" rel="stylesheet" type="text/css"> <link href="tabs.css" rel="stylesheet" type="text/css"> </head><body> <!-- Generated by Doxygen 1.5.6 --> <div class="navigation" id="top"> <div class="tabs"> <ul> <li><a href="main.html"><span>Main Page</span></a></li> <li><a href="namespaces.html"><span>Namespaces</span></a></li> <li><a href="annotated.html"><span>Classes</span></a></li> <li class="current"><a href="files.html"><span>Files</span></a></li> </ul> </div> <h1>KeyDB.cc</h1><a href="KeyDB_8cc.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*</span> <a name="l00002"></a>00002 <span class="comment"> * Copyright 2007 BBN Technologies Corporation</span> <a name="l00003"></a>00003 <span class="comment"> *</span> <a name="l00004"></a>00004 <span class="comment"> * Licensed under the Apache License, Version 2.0 (the "License"); you</span> <a name="l00005"></a>00005 <span class="comment"> * may not use this file except in compliance with the License. You</span> <a name="l00006"></a>00006 <span class="comment"> * may obtain a copy of the License at</span> <a name="l00007"></a>00007 <span class="comment"> *</span> <a name="l00008"></a>00008 <span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span> <a name="l00009"></a>00009 <span class="comment"> *</span> <a name="l00010"></a>00010 <span class="comment"> * Unless required by applicable law or agreed to in writing, software</span> <a name="l00011"></a>00011 <span class="comment"> * distributed under the License is distributed on an "AS IS" BASIS,</span> <a name="l00012"></a>00012 <span class="comment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or</span> <a name="l00013"></a>00013 <span class="comment"> * implied.</span> <a name="l00014"></a>00014 <span class="comment"> */</span> <a name="l00015"></a>00015 <a name="l00016"></a>00016 <span class="comment">/*</span> <a name="l00017"></a>00017 <span class="comment"> * $Id$</span> <a name="l00018"></a>00018 <span class="comment"> */</span> <a name="l00019"></a>00019 <a name="l00020"></a>00020 <span class="preprocessor">#ifdef HAVE_CONFIG_H</span> <a name="l00021"></a>00021 <span class="preprocessor"></span><span class="preprocessor"># include <<a class="code" href="dtn-config_8h.html">dtn-config.h</a>></span> <a name="l00022"></a>00022 <span class="preprocessor">#endif</span> <a name="l00023"></a>00023 <span class="preprocessor"></span> <a name="l00024"></a>00024 <span class="preprocessor">#ifdef BSP_ENABLED</span> <a name="l00025"></a>00025 <span class="preprocessor"></span> <a name="l00026"></a>00026 <span class="preprocessor">#include <cstring></span> <a name="l00027"></a>00027 <span class="preprocessor">#include <string></span> <a name="l00028"></a>00028 <span class="preprocessor">#include <oasys/compat/inttypes.h></span> <a name="l00029"></a>00029 <span class="preprocessor">#include <oasys/util/StringBuffer.h></span> <a name="l00030"></a>00030 <a name="l00031"></a>00031 <span class="preprocessor">#include "<a class="code" href="KeyDB_8h.html">KeyDB.h</a>"</span> <a name="l00032"></a>00032 <span class="preprocessor">#include "<a class="code" href="Ciphersuite_8h.html">Ciphersuite.h</a>"</span> <a name="l00033"></a>00033 <a name="l00034"></a>00034 <span class="keyword">namespace </span>dtn { <a name="l00035"></a>00035 <a name="l00036"></a>00036 <span class="keyword">template</span> <> <a name="l00037"></a>00037 KeyDB* oasys::Singleton<KeyDB, false>::instance_ = NULL; <a name="l00038"></a>00038 <a name="l00039"></a>00039 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">char</span> * <a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a> = <span class="stringliteral">"/dtn/bundle/security"</span>; <a name="l00040"></a>00040 <a name="l00041"></a>00041 KeyDB::KeyDB() <a name="l00042"></a>00042 { <a name="l00043"></a>00043 } <a name="l00044"></a>00044 <a name="l00045"></a>00045 KeyDB::~KeyDB() <a name="l00046"></a>00046 { <a name="l00047"></a>00047 } <a name="l00048"></a>00048 <a name="l00049"></a>00049 <span class="keywordtype">void</span> <a name="l00050"></a>00050 KeyDB::init() <a name="l00051"></a>00051 { <a name="l00052"></a>00052 <span class="keywordflow">if</span> (instance_ != NULL) <a name="l00053"></a>00053 { <a name="l00054"></a>00054 PANIC(<span class="stringliteral">"KeyDB already initialized"</span>); <a name="l00055"></a>00055 } <a name="l00056"></a>00056 <a name="l00057"></a>00057 instance_ = <span class="keyword">new</span> KeyDB(); <a name="l00058"></a>00058 log_debug_p(<a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a>, <span class="stringliteral">"KeyDB::init() done"</span>); <a name="l00059"></a>00059 } <a name="l00060"></a>00060 <a name="l00061"></a>00061 <span class="keywordtype">void</span> <a name="l00062"></a>00062 KeyDB::set_key(Entry& entry) <a name="l00063"></a>00063 { <a name="l00064"></a>00064 EntryList& keys = instance()->keys_; <a name="l00065"></a>00065 EntryList::iterator iter; <a name="l00066"></a>00066 <a name="l00067"></a>00067 <span class="keywordflow">for</span> (iter = keys.begin(); iter != keys.end(); iter++) <a name="l00068"></a>00068 { <a name="l00069"></a>00069 <span class="keywordflow">if</span> (iter->match(entry)) { <a name="l00070"></a>00070 *iter = entry; <a name="l00071"></a>00071 <span class="keywordflow">return</span>; <a name="l00072"></a>00072 } <a name="l00073"></a>00073 } <a name="l00074"></a>00074 <a name="l00075"></a>00075 <span class="comment">// if we get here then there was no existing matching entry;</span> <a name="l00076"></a>00076 <span class="comment">// insert the new entry just before any existing wildcard entries</span> <a name="l00077"></a>00077 <span class="keywordflow">for</span> (iter = keys.begin(); <a name="l00078"></a>00078 iter != keys.end() && iter->host() != std::string(<span class="stringliteral">"*"</span>); <a name="l00079"></a>00079 iter++) <a name="l00080"></a>00080 { <a name="l00081"></a>00081 } <a name="l00082"></a>00082 keys.insert(iter, entry); <a name="l00083"></a>00083 log_debug_p(<a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a>, <span class="stringliteral">"KeyDB::set_key() done"</span>); <a name="l00084"></a>00084 } <a name="l00085"></a>00085 <a name="l00086"></a>00086 <span class="keyword">const</span> KeyDB::Entry* <a name="l00087"></a>00087 KeyDB::find_key(<span class="keyword">const</span> <span class="keywordtype">char</span>* host, u_int16_t cs_num) <a name="l00088"></a>00088 { <a name="l00089"></a>00089 EntryList& keys = instance()->keys_; <a name="l00090"></a>00090 EntryList::iterator iter; <a name="l00091"></a>00091 log_debug_p(<a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a>, <span class="stringliteral">"KeyDB::find_key()"</span>); <a name="l00092"></a>00092 <a name="l00093"></a>00093 <span class="keywordflow">for</span> (iter = keys.begin(); iter != keys.end(); iter++) <a name="l00094"></a>00094 { <a name="l00095"></a>00095 <span class="keywordflow">if</span> (iter->match_wildcard(host, cs_num)) <a name="l00096"></a>00096 <span class="keywordflow">return</span> &(*iter); <a name="l00097"></a>00097 } <a name="l00098"></a>00098 <a name="l00099"></a>00099 <span class="comment">// not found</span> <a name="l00100"></a>00100 <span class="keywordflow">return</span> NULL; <a name="l00101"></a>00101 } <a name="l00102"></a>00102 <a name="l00103"></a>00103 <span class="keywordtype">void</span> <a name="l00104"></a>00104 KeyDB::del_key(<span class="keyword">const</span> <span class="keywordtype">char</span>* host, u_int16_t cs_num) <a name="l00105"></a>00105 { <a name="l00106"></a>00106 EntryList& keys = instance()->keys_; <a name="l00107"></a>00107 EntryList::iterator iter; <a name="l00108"></a>00108 log_debug_p(<a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a>, <span class="stringliteral">"KeyDB::del_key()"</span>); <a name="l00109"></a>00109 <a name="l00110"></a>00110 <span class="keywordflow">for</span> (iter = keys.begin(); iter != keys.end(); iter++) <a name="l00111"></a>00111 { <a name="l00112"></a>00112 <span class="keywordflow">if</span> (iter->match(host, cs_num)) { <a name="l00113"></a>00113 keys.erase(iter); <a name="l00114"></a>00114 <span class="keywordflow">return</span>; <a name="l00115"></a>00115 } <a name="l00116"></a>00116 } <a name="l00117"></a>00117 <a name="l00118"></a>00118 <span class="comment">// if not found, then nothing to do</span> <a name="l00119"></a>00119 } <a name="l00120"></a>00120 <a name="l00121"></a>00121 <span class="keywordtype">void</span> <a name="l00122"></a>00122 KeyDB::flush_keys() <a name="l00123"></a>00123 { <a name="l00124"></a>00124 log_debug_p(<a class="code" href="namespacedtn.html#68bf4270dd0a7e5e8390869b83889922">log</a>, <span class="stringliteral">"KeyDB::flush_keys()"</span>); <a name="l00125"></a>00125 instance()->keys_.clear(); <a name="l00126"></a>00126 } <a name="l00127"></a>00127 <a name="l00129"></a>00129 <span class="keywordtype">void</span> <a name="l00130"></a>00130 <a class="code" href="serialsource_8c.html#87b713d53023266d98d45b86312c26aa">KeyDB::dump</a>(oasys::StringBuffer* <a class="code" href="num2sdnv_8c.html#a81cdcc7ff6987bc85c073253e32715f">buf</a>) <a name="l00131"></a>00131 { <a name="l00132"></a>00132 EntryList& keys = instance()->keys_; <a name="l00133"></a>00133 EntryList::iterator iter; <a name="l00134"></a>00134 <a name="l00135"></a>00135 <span class="keywordflow">for</span> (iter = keys.begin(); iter != keys.end(); iter++) <a name="l00136"></a>00136 iter->dump(buf); <a name="l00137"></a>00137 } <a name="l00138"></a>00138 <a name="l00140"></a>00140 <span class="keywordtype">void</span> <a name="l00141"></a>00141 KeyDB::dump_header(oasys::StringBuffer* buf) <a name="l00142"></a>00142 { <a name="l00143"></a>00143 KeyDB::Entry::dump_header(buf); <a name="l00144"></a>00144 } <a name="l00145"></a>00145 <a name="l00148"></a>00148 <span class="keywordtype">bool</span> <a name="l00149"></a>00149 KeyDB::validate_cs_num(u_int16_t cs_num) <a name="l00150"></a>00150 { <a name="l00151"></a>00151 <span class="keywordflow">return</span> (Ciphersuite::find_suite(cs_num) != NULL); <a name="l00152"></a>00152 } <a name="l00153"></a>00153 <a name="l00154"></a>00154 <span class="keywordtype">bool</span> <a name="l00155"></a>00155 KeyDB::validate_key_len(u_int16_t cs_num, <span class="keywordtype">size_t</span>* key_len) <a name="l00156"></a>00156 { <a name="l00157"></a>00157 Ciphersuite* cs = <a name="l00158"></a>00158 <span class="keyword">dynamic_cast<</span>Ciphersuite*<span class="keyword">></span>(Ciphersuite::find_suite(cs_num)); <a name="l00159"></a>00159 <span class="keywordflow">if</span> (cs == NULL) <a name="l00160"></a>00160 <span class="keywordflow">return</span> <span class="keyword">false</span>; <a name="l00161"></a>00161 <span class="keywordflow">if</span> (*key_len != cs->result_len()) { <a name="l00162"></a>00162 *key_len = cs->result_len(); <a name="l00163"></a>00163 <span class="keywordflow">return</span> <span class="keyword">false</span>; <a name="l00164"></a>00164 } <a name="l00165"></a>00165 <a name="l00166"></a>00166 <span class="keywordflow">return</span> <span class="keyword">true</span>; <a name="l00167"></a>00167 } <a name="l00168"></a>00168 <a name="l00170"></a>00170 KeyDB::Entry::Entry(<span class="keyword">const</span> <span class="keywordtype">char</span>* host, u_int16_t cs_num, <span class="keyword">const</span> u_char* key, <a name="l00171"></a>00171 <span class="keywordtype">size_t</span> key_len) <a name="l00172"></a>00172 : host_(host), <a name="l00173"></a>00173 cs_num_(cs_num), <a name="l00174"></a>00174 key_len_(key_len) <a name="l00175"></a>00175 { <a name="l00176"></a>00176 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(key != NULL); <a name="l00177"></a>00177 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(key_len != 0); <a name="l00178"></a>00178 <a name="l00179"></a>00179 key_ = <span class="keyword">new</span> u_char[key_len]; <a name="l00180"></a>00180 memcpy(key_, key, key_len); <a name="l00181"></a>00181 } <a name="l00182"></a>00182 <a name="l00184"></a>00184 KeyDB::Entry::Entry() <a name="l00185"></a>00185 : host_(<span class="stringliteral">""</span>), <a name="l00186"></a>00186 cs_num_(0), <a name="l00187"></a>00187 key_(NULL), <a name="l00188"></a>00188 key_len_(0) <a name="l00189"></a>00189 { <a name="l00190"></a>00190 } <a name="l00191"></a>00191 <a name="l00193"></a>00193 KeyDB::Entry::Entry(<span class="keyword">const</span> Entry& other) <a name="l00194"></a>00194 : host_(other.host_), <a name="l00195"></a>00195 cs_num_(other.cs_num_), <a name="l00196"></a>00196 key_len_(other.key_len_) <a name="l00197"></a>00197 { <a name="l00198"></a>00198 <span class="keywordflow">if</span> (other.key_ == NULL) <a name="l00199"></a>00199 key_ = NULL; <a name="l00200"></a>00200 <span class="keywordflow">else</span> { <a name="l00201"></a>00201 key_ = <span class="keyword">new</span> u_char[other.key_len_]; <a name="l00202"></a>00202 memcpy(key_, other.key_, other.key_len_); <a name="l00203"></a>00203 } <a name="l00204"></a>00204 } <a name="l00205"></a>00205 <a name="l00207"></a>00207 KeyDB::Entry::~Entry() <a name="l00208"></a>00208 { <a name="l00209"></a>00209 <span class="keywordflow">if</span> (key_ != NULL) { <a name="l00210"></a>00210 memset(key_, 0, key_len_); <a name="l00211"></a>00211 <span class="keyword">delete</span> key_; <a name="l00212"></a>00212 key_len_ = 0; <a name="l00213"></a>00213 } <span class="keywordflow">else</span> <a name="l00214"></a>00214 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(key_len_ == 0); <a name="l00215"></a>00215 } <a name="l00216"></a>00216 <a name="l00218"></a>00218 <span class="keywordtype">void</span> <a name="l00219"></a>00219 KeyDB::Entry::operator=(<span class="keyword">const</span> Entry& other) <a name="l00220"></a>00220 { <a name="l00221"></a>00221 <span class="keywordflow">if</span> (key_ != NULL) { <a name="l00222"></a>00222 memset(key_, 0, key_len_); <a name="l00223"></a>00223 <span class="keyword">delete</span> key_; <a name="l00224"></a>00224 } <a name="l00225"></a>00225 <a name="l00226"></a>00226 host_ = other.host_; <a name="l00227"></a>00227 cs_num_ = other.cs_num_; <a name="l00228"></a>00228 <a name="l00229"></a>00229 <span class="keywordflow">if</span> (other.key_ == NULL) <a name="l00230"></a>00230 key_ = NULL; <a name="l00231"></a>00231 <span class="keywordflow">else</span> { <a name="l00232"></a>00232 key_ = <span class="keyword">new</span> u_char[other.key_len_]; <a name="l00233"></a>00233 memcpy(key_, other.key_, other.key_len_); <a name="l00234"></a>00234 } <a name="l00235"></a>00235 <a name="l00236"></a>00236 key_len_ = other.key_len_; <a name="l00237"></a>00237 } <a name="l00238"></a>00238 <a name="l00241"></a>00241 <span class="keywordtype">bool</span> <a name="l00242"></a>00242 KeyDB::Entry::match(<span class="keyword">const</span> Entry& other)<span class="keyword"> const</span> <a name="l00243"></a>00243 <span class="keyword"></span>{ <a name="l00244"></a>00244 <span class="keywordflow">return</span> (host_ == other.host_ && cs_num_ == other.cs_num_); <a name="l00245"></a>00245 } <a name="l00246"></a>00246 <a name="l00249"></a>00249 <span class="keywordtype">bool</span> <a name="l00250"></a>00250 KeyDB::Entry::match(<span class="keyword">const</span> <span class="keywordtype">char</span>* host, u_int16_t cs_num)<span class="keyword"> const</span> <a name="l00251"></a>00251 <span class="keyword"></span>{ <a name="l00252"></a>00252 <span class="keywordflow">return</span> (host_ == std::string(host) && cs_num_ == cs_num); <a name="l00253"></a>00253 } <a name="l00254"></a>00254 <a name="l00257"></a>00257 <span class="keywordtype">bool</span> <a name="l00258"></a>00258 KeyDB::Entry::match_wildcard(<span class="keyword">const</span> <span class="keywordtype">char</span>* host, u_int16_t cs_num)<span class="keyword"> const</span> <a name="l00259"></a>00259 <span class="keyword"></span>{ <a name="l00260"></a>00260 <span class="keywordflow">return</span> ((host_ == std::string(host) || host_ == std::string(<span class="stringliteral">"*"</span>)) <a name="l00261"></a>00261 && cs_num_ == cs_num); <a name="l00262"></a>00262 } <a name="l00263"></a>00263 <a name="l00265"></a>00265 <span class="keywordtype">void</span> <a name="l00266"></a>00266 <a class="code" href="serialsource_8c.html#87b713d53023266d98d45b86312c26aa">KeyDB::Entry::dump</a>(oasys::StringBuffer* buf)<span class="keyword"> const</span> <a name="l00267"></a>00267 <span class="keyword"></span>{ <a name="l00268"></a>00268 buf->appendf(<span class="stringliteral">"%15s 0x%03x "</span>, host_.c_str(), cs_num_); <a name="l00269"></a>00269 <span class="comment">//buf->appendf("(-- not shown --)");</span> <a name="l00270"></a>00270 <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < (int)key_len_; i++) <a name="l00271"></a>00271 buf->appendf(<span class="stringliteral">"%02x"</span>, (<span class="keywordtype">int</span>)key_[i]); <a name="l00272"></a>00272 buf->appendf(<span class="stringliteral">"\n"</span>); <a name="l00273"></a>00273 } <a name="l00274"></a>00274 <a name="l00276"></a>00276 <span class="keywordtype">void</span> <a name="l00277"></a>00277 KeyDB::Entry::dump_header(oasys::StringBuffer* buf) <a name="l00278"></a>00278 { <a name="l00279"></a>00279 buf->appendf(<span class="stringliteral">"host cs_num key\n"</span> ); <a name="l00280"></a>00280 buf->appendf(<span class="stringliteral">"--------------- ------ -------------------\n"</span>); <a name="l00281"></a>00281 } <a name="l00282"></a>00282 <a name="l00283"></a>00283 } <span class="comment">// namespace dtn</span> <a name="l00284"></a>00284 <a name="l00285"></a>00285 <span class="preprocessor">#endif </span><span class="comment">/* BSP_ENABLED */</span> </pre></div></div> <hr size="1"><address style="text-align: right;"><small>Generated on Mon Jul 21 14:09:46 2008 for DTN Reference Implementation by <a href="http://www.doxygen.org/index.html"> <img src="doxygen.png" alt="doxygen" align="middle" border="0"></a> 1.5.6 </small></address> </body> </html>