doc/www/doxygen/html/brecsum_8cc-source.html

   1 <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
   2 <html><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8">
   3 <title>Barry: brecsum.cc Source File</title>
   4 <link href="doxygen.css" rel="stylesheet" type="text/css">
   5 <link href="tabs.css" rel="stylesheet" type="text/css">
   6 </head><body>
   7 <!-- Generated by Doxygen 1.5.6 -->
   8 <div class="navigation" id="top">
   9   <div class="tabs">
  10     <ul>
  11       <li><a href="index.html"><span>Main&nbsp;Page</span></a></li>
  12       <li><a href="modules.html"><span>Modules</span></a></li>
  13       <li><a href="namespaces.html"><span>Namespaces</span></a></li>
  14       <li><a href="classes.html"><span>Classes</span></a></li>
  15       <li class="current"><a href="files.html"><span>Files</span></a></li>
  16       <li><a href="dirs.html"><span>Directories</span></a></li>
  17     </ul>
  18   </div>
  19   <div class="navpath"><a class="el" href="dir_ae818e64ea3dc686ac96c0f260a1301d.html">tools</a>
  20   </div>
  21 </div>
  22 <div class="contents">
  23 <h1>brecsum.cc</h1><a href="brecsum_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>
  24 <a name="l00002"></a>00002 <span class="comment"></span><span class="comment">/// \file       brecsum.cc</span>
  25 <a name="l00003"></a>00003 <span class="comment"></span><span class="comment">///             Generate SHA1 sums of raw Blackberry database records.</span>
  26 <a name="l00004"></a>00004 <span class="comment"></span><span class="comment">///             This is mostly useful for data verification during testing.</span>
  27 <a name="l00005"></a>00005 <span class="comment"></span><span class="comment">///</span>
  28 <a name="l00006"></a>00006 <span class="comment"></span>
  29 <a name="l00007"></a>00007 <span class="comment">/*</span>
  30 <a name="l00008"></a>00008 <span class="comment">    Copyright (C) 2008-2009, Net Direct Inc. (http://www.netdirect.ca/)</span>
  31 <a name="l00009"></a>00009 <span class="comment"></span>
  32 <a name="l00010"></a>00010 <span class="comment">    This program is free software; you can redistribute it and/or modify</span>
  33 <a name="l00011"></a>00011 <span class="comment">    it under the terms of the GNU General Public License as published by</span>
  34 <a name="l00012"></a>00012 <span class="comment">    the Free Software Foundation; either version 2 of the License, or</span>
  35 <a name="l00013"></a>00013 <span class="comment">    (at your option) any later version.</span>
  36 <a name="l00014"></a>00014 <span class="comment"></span>
  37 <a name="l00015"></a>00015 <span class="comment">    This program is distributed in the hope that it will be useful,</span>
  38 <a name="l00016"></a>00016 <span class="comment">    but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
  39 <a name="l00017"></a>00017 <span class="comment">    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.</span>
  40 <a name="l00018"></a>00018 <span class="comment"></span>
  41 <a name="l00019"></a>00019 <span class="comment">    See the GNU General Public License in the COPYING file at the</span>
  42 <a name="l00020"></a>00020 <span class="comment">    root directory of this project for more details.</span>
  43 <a name="l00021"></a>00021 <span class="comment">*/</span>
  44 <a name="l00022"></a>00022
  45 <a name="l00023"></a>00023 <span class="preprocessor">#include &lt;barry/barry.h&gt;</span>
  46 <a name="l00024"></a>00024 <span class="preprocessor">#include &lt;iomanip&gt;</span>
  47 <a name="l00025"></a>00025 <span class="preprocessor">#include &lt;iostream&gt;</span>
  48 <a name="l00026"></a>00026 <span class="preprocessor">#include &lt;vector&gt;</span>
  49 <a name="l00027"></a>00027 <span class="preprocessor">#include &lt;string&gt;</span>
  50 <a name="l00028"></a>00028 <span class="preprocessor">#include &lt;getopt.h&gt;</span>
  51 <a name="l00029"></a>00029
  52 <a name="l00030"></a>00030 <span class="keyword">using namespace </span>std;
  53 <a name="l00031"></a>00031 <span class="keyword">using namespace </span>Barry;
  54 <a name="l00032"></a>00032
  55 <a name="l00033"></a>00033 <span class="keywordtype">void</span> Usage()
  56 <a name="l00034"></a>00034 {
  57 <a name="l00035"></a>00035    <span class="keywordtype">int</span> major, minor;
  58 <a name="l00036"></a>00036    <span class="keyword">const</span> <span class="keywordtype">char</span> *<a class="code" href="namespaceBarry.html#06ec709ed2313dd84291b2d295a87d26" title="Fills major and minor with integer version numbers, and returns a string containing...">Version</a> = <a class="code" href="namespaceBarry.html#06ec709ed2313dd84291b2d295a87d26" title="Fills major and minor with integer version numbers, and returns a string containing...">Barry::Version</a>(major, minor);
  59 <a name="l00037"></a>00037
  60 <a name="l00038"></a>00038    cerr
  61 <a name="l00039"></a>00039    &lt;&lt; <span class="stringliteral">"brecsum - Generate SHA1 sums of raw Blackberry database records.\n"</span>
  62 <a name="l00040"></a>00040    &lt;&lt; <span class="stringliteral">"        Copyright 2008-2009, Net Direct Inc. (http://www.netdirect.ca/)\n"</span>
  63 <a name="l00041"></a>00041    &lt;&lt; <span class="stringliteral">"        Using: "</span> &lt;&lt; Version &lt;&lt; <span class="stringliteral">"\n"</span>
  64 <a name="l00042"></a>00042    &lt;&lt; <span class="stringliteral">"\n"</span>
  65 <a name="l00043"></a>00043    &lt;&lt; <span class="stringliteral">"   -d db     Read database 'db' and sum all its records.\n"</span>
  66 <a name="l00044"></a>00044    &lt;&lt; <span class="stringliteral">"             Can be used multiple times to fetch more than one DB\n"</span>
  67 <a name="l00045"></a>00045    &lt;&lt; <span class="stringliteral">"   -h        This help\n"</span>
  68 <a name="l00046"></a>00046    &lt;&lt; <span class="stringliteral">"   -i        Include Type and Unique record IDs in the checksums\n"</span>
  69 <a name="l00047"></a>00047    &lt;&lt; <span class="stringliteral">"   -p pin    PIN of device to talk with\n"</span>
  70 <a name="l00048"></a>00048    &lt;&lt; <span class="stringliteral">"             If only one device is plugged in, this flag is optional\n"</span>
  71 <a name="l00049"></a>00049    &lt;&lt; <span class="stringliteral">"   -P pass   Simplistic method to specify device password\n"</span>
  72 <a name="l00050"></a>00050    &lt;&lt; <span class="stringliteral">"   -v        Dump protocol data during operation\n"</span>
  73 <a name="l00051"></a>00051    &lt;&lt; endl;
  74 <a name="l00052"></a>00052 }
  75 <a name="l00053"></a>00053
  76 <a name="l00054"></a>00054 <span class="keyword">class </span>ChecksumParser : <span class="keyword">public</span> Barry::<a class="code" href="classBarry_1_1Parser.html" title="Base class for the parser hierarchy.">Parser</a>
  77 <a name="l00055"></a>00055 {
  78 <a name="l00056"></a>00056         <span class="keywordtype">bool</span> m_IncludeIds;
  79 <a name="l00057"></a>00057         SHA_CTX m_ctx;
  80 <a name="l00058"></a>00058
  81 <a name="l00059"></a>00059 <span class="keyword">public</span>:
  82 <a name="l00060"></a>00060         <span class="keyword">explicit</span> ChecksumParser(<span class="keywordtype">bool</span> IncludeIds)
  83 <a name="l00061"></a>00061                 : m_IncludeIds(IncludeIds)
  84 <a name="l00062"></a>00062         {}
  85 <a name="l00063"></a>00063
  86 <a name="l00064"></a>00064         <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classBarry_1_1Parser.html#15acb3544613d43d8f928cd63a19022b" title="Reset and prepare for a new raw data packet.">Clear</a>()
  87 <a name="l00065"></a>00065         {
  88 <a name="l00066"></a>00066                 SHA1_Init(&amp;m_ctx);
  89 <a name="l00067"></a>00067         }
  90 <a name="l00068"></a>00068
  91 <a name="l00069"></a>00069         <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classBarry_1_1Parser.html#95bfc4014e3e8ded94625f43d1d29800" title="Stores the IDs.">SetIds</a>(uint8_t RecType, uint32_t UniqueId)
  92 <a name="l00070"></a>00070         {
  93 <a name="l00071"></a>00071                 <span class="keywordflow">if</span>( m_IncludeIds ) {
  94 <a name="l00072"></a>00072                         SHA1_Update(&amp;m_ctx, &amp;RecType, <span class="keyword">sizeof</span>(RecType));
  95 <a name="l00073"></a>00073                         SHA1_Update(&amp;m_ctx, &amp;UniqueId, <span class="keyword">sizeof</span>(UniqueId));
  96 <a name="l00074"></a>00074                 }
  97 <a name="l00075"></a>00075         }
  98 <a name="l00076"></a>00076
  99 <a name="l00077"></a>00077         <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classBarry_1_1Parser.html#2087d07fb5d8d57c934c4e60278f41d8" title="Called to parse the header portion of the raw data packet.">ParseHeader</a>(<span class="keyword">const</span> Barry::Data &amp;, <span class="keywordtype">size_t</span> &amp;)
 100 <a name="l00078"></a>00078         {
 101 <a name="l00079"></a>00079                 <span class="comment">// do nothing here, parse it all at once in ParseFields</span>
 102 <a name="l00080"></a>00080         }
 103 <a name="l00081"></a>00081
 104 <a name="l00082"></a>00082         <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classBarry_1_1Parser.html#645718d87aaad3ee2c36664de05046d2" title="Called to parse sub fields in the raw data packet.">ParseFields</a>(<span class="keyword">const</span> Barry::Data &amp;data, <span class="keywordtype">size_t</span> &amp;offset,
 105 <a name="l00083"></a>00083                                 <span class="keyword">const</span> <a class="code" href="classBarry_1_1IConverter.html" title="Main charset conversion class, primarily focused on converting between the Blackberry...">Barry::IConverter</a> *ic)
 106 <a name="l00084"></a>00084         {
 107 <a name="l00085"></a>00085                 <span class="keywordtype">int</span> len = data.GetSize() - offset;
 108 <a name="l00086"></a>00086                 SHA1_Update(&amp;m_ctx, data.GetData() + offset, len);
 109 <a name="l00087"></a>00087                 offset += len;
 110 <a name="l00088"></a>00088         }
 111 <a name="l00089"></a>00089
 112 <a name="l00090"></a>00090         <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classBarry_1_1Parser.html#bf8ab1598be74268ee486af938c527d4" title="Called at the very end of record parsing, and used to store the final packet somewhere...">Store</a>()
 113 <a name="l00091"></a>00091         {
 114 <a name="l00092"></a>00092                 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> sha1[SHA_DIGEST_LENGTH];
 115 <a name="l00093"></a>00093                 SHA1_Final(sha1, &amp;m_ctx);
 116 <a name="l00094"></a>00094
 117 <a name="l00095"></a>00095                 <span class="keywordflow">for</span>( <span class="keywordtype">int</span> i = 0; i &lt; SHA_DIGEST_LENGTH; i++ ) {
 118 <a name="l00096"></a>00096                         cout &lt;&lt; hex &lt;&lt; setfill(<span class="charliteral">'0'</span>) &lt;&lt; setw(2)
 119 <a name="l00097"></a>00097                                 &lt;&lt; (<span class="keywordtype">unsigned</span> int) sha1[i];
 120 <a name="l00098"></a>00098                 }
 121 <a name="l00099"></a>00099                 cout &lt;&lt; endl;
 122 <a name="l00100"></a>00100         }
 123 <a name="l00101"></a>00101 };
 124 <a name="l00102"></a>00102
 125 <a name="l00103"></a>00103 <span class="keywordtype">int</span> main(<span class="keywordtype">int</span> argc, <span class="keywordtype">char</span> *argv[])
 126 <a name="l00104"></a>00104 {
 127 <a name="l00105"></a>00105         cout.sync_with_stdio(<span class="keyword">true</span>);     <span class="comment">// leave this on, since libusb uses</span>
 128 <a name="l00106"></a>00106                                         <span class="comment">// stdio for debug messages</span>
 129 <a name="l00107"></a>00107
 130 <a name="l00108"></a>00108         <span class="keywordflow">try</span> {
 131 <a name="l00109"></a>00109
 132 <a name="l00110"></a>00110                 uint32_t pin = 0;
 133 <a name="l00111"></a>00111                 <span class="keywordtype">bool</span>
 134 <a name="l00112"></a>00112                         data_dump = <span class="keyword">false</span>,
 135 <a name="l00113"></a>00113                         include_ids = <span class="keyword">false</span>;
 136 <a name="l00114"></a>00114                 <span class="keywordtype">string</span> password;
 137 <a name="l00115"></a>00115                 vector&lt;string&gt; dbNames;
 138 <a name="l00116"></a>00116
 139 <a name="l00117"></a>00117                 <span class="comment">// process command line options</span>
 140 <a name="l00118"></a>00118                 <span class="keywordflow">for</span>(;;) {
 141 <a name="l00119"></a>00119                         <span class="keywordtype">int</span> cmd = getopt(argc, argv, <span class="stringliteral">"d:hip:P:v"</span>);
 142 <a name="l00120"></a>00120                         <span class="keywordflow">if</span>( cmd == -1 )
 143 <a name="l00121"></a>00121                                 <span class="keywordflow">break</span>;
 144 <a name="l00122"></a>00122
 145 <a name="l00123"></a>00123                         <span class="keywordflow">switch</span>( cmd )
 146 <a name="l00124"></a>00124                         {
 147 <a name="l00125"></a>00125                         <span class="keywordflow">case</span> <span class="charliteral">'d'</span>:       <span class="comment">// show dbname</span>
 148 <a name="l00126"></a>00126                                 dbNames.push_back(<span class="keywordtype">string</span>(optarg));
 149 <a name="l00127"></a>00127                                 <span class="keywordflow">break</span>;
 150 <a name="l00128"></a>00128
 151 <a name="l00129"></a>00129                         <span class="keywordflow">case</span> <span class="charliteral">'i'</span>:       <span class="comment">// Include IDs</span>
 152 <a name="l00130"></a>00130                                 include_ids = <span class="keyword">true</span>;
 153 <a name="l00131"></a>00131                                 <span class="keywordflow">break</span>;
 154 <a name="l00132"></a>00132
 155 <a name="l00133"></a>00133                         <span class="keywordflow">case</span> <span class="charliteral">'p'</span>:       <span class="comment">// Blackberry PIN</span>
 156 <a name="l00134"></a>00134                                 pin = strtoul(optarg, NULL, 16);
 157 <a name="l00135"></a>00135                                 <span class="keywordflow">break</span>;
 158 <a name="l00136"></a>00136
 159 <a name="l00137"></a>00137                         <span class="keywordflow">case</span> <span class="charliteral">'P'</span>:       <span class="comment">// Device password</span>
 160 <a name="l00138"></a>00138                                 password = optarg;
 161 <a name="l00139"></a>00139                                 <span class="keywordflow">break</span>;
 162 <a name="l00140"></a>00140
 163 <a name="l00141"></a>00141                         <span class="keywordflow">case</span> <span class="charliteral">'v'</span>:       <span class="comment">// data dump on</span>
 164 <a name="l00142"></a>00142                                 data_dump = <span class="keyword">true</span>;
 165 <a name="l00143"></a>00143                                 <span class="keywordflow">break</span>;
 166 <a name="l00144"></a>00144
 167 <a name="l00145"></a>00145                         <span class="keywordflow">case</span> <span class="charliteral">'h'</span>:       <span class="comment">// help</span>
 168 <a name="l00146"></a>00146                         <span class="keywordflow">default</span>:
 169 <a name="l00147"></a>00147                                 Usage();
 170 <a name="l00148"></a>00148                                 <span class="keywordflow">return</span> 0;
 171 <a name="l00149"></a>00149                         }
 172 <a name="l00150"></a>00150                 }
 173 <a name="l00151"></a>00151
 174 <a name="l00152"></a>00152                 <span class="comment">// Display usage info if user appears confused</span>
 175 <a name="l00153"></a>00153                 <span class="keywordflow">if</span>( !dbNames.size() ) {
 176 <a name="l00154"></a>00154                         Usage();
 177 <a name="l00155"></a>00155                         <span class="keywordflow">return</span> 0;
 178 <a name="l00156"></a>00156                 }
 179 <a name="l00157"></a>00157
 180 <a name="l00158"></a>00158                 <span class="comment">// Initialize the barry library.  Must be called before</span>
 181 <a name="l00159"></a>00159                 <span class="comment">// anything else.</span>
 182 <a name="l00160"></a>00160                 <a class="code" href="namespaceBarry.html#2e28bd59aebe0bece5a46f02ef7dbd3d" title="Barry library initializer.">Barry::Init</a>(data_dump);
 183 <a name="l00161"></a>00161
 184 <a name="l00162"></a>00162                 <span class="comment">// Probe the USB bus for Blackberry devices and display.</span>
 185 <a name="l00163"></a>00163                 Barry::Probe probe;
 186 <a name="l00164"></a>00164                 <span class="keywordtype">int</span> activeDevice = probe.FindActive(pin);
 187 <a name="l00165"></a>00165                 <span class="keywordflow">if</span>( activeDevice == -1 ) {
 188 <a name="l00166"></a>00166                         cerr &lt;&lt; <span class="stringliteral">"No device selected, or PIN not found"</span> &lt;&lt; endl;
 189 <a name="l00167"></a>00167                         <span class="keywordflow">return</span> 1;
 190 <a name="l00168"></a>00168                 }
 191 <a name="l00169"></a>00169
 192 <a name="l00170"></a>00170                 <span class="comment">// Create our controller object</span>
 193 <a name="l00171"></a>00171                 <a class="code" href="classBarry_1_1Controller.html" title="The main interface class.">Barry::Controller</a> con(probe.Get(activeDevice));
 194 <a name="l00172"></a>00172                 <a class="code" href="classBarry_1_1Mode_1_1Desktop.html" title="The main interface class to the device databases.">Barry::Mode::Desktop</a> desktop(con);
 195 <a name="l00173"></a>00173
 196 <a name="l00174"></a>00174                 <span class="comment">// Sum all specified databases</span>
 197 <a name="l00175"></a>00175                 <span class="keywordflow">if</span>( dbNames.size() ) {
 198 <a name="l00176"></a>00176                         vector&lt;string&gt;::iterator b = dbNames.begin();
 199 <a name="l00177"></a>00177                         ChecksumParser parser(include_ids);
 200 <a name="l00178"></a>00178
 201 <a name="l00179"></a>00179                         desktop.Open(password.c_str());
 202 <a name="l00180"></a>00180                         <span class="keywordflow">for</span>( ; b != dbNames.end(); b++ ) {
 203 <a name="l00181"></a>00181                                 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <span class="keywordtype">id</span> = desktop.GetDBID(*b);
 204 <a name="l00182"></a>00182                                 desktop.LoadDatabase(<span class="keywordtype">id</span>, parser);
 205 <a name="l00183"></a>00183                         }
 206 <a name="l00184"></a>00184                 }
 207 <a name="l00185"></a>00185
 208 <a name="l00186"></a>00186         }
 209 <a name="l00187"></a>00187         <span class="keywordflow">catch</span>( std::exception &amp;e ) {
 210 <a name="l00188"></a>00188                 std::cerr &lt;&lt; e.what() &lt;&lt; endl;
 211 <a name="l00189"></a>00189                 <span class="keywordflow">return</span> 1;
 212 <a name="l00190"></a>00190         }
 213 <a name="l00191"></a>00191
 214 <a name="l00192"></a>00192         <span class="keywordflow">return</span> 0;
 215 <a name="l00193"></a>00193 }
 216 <a name="l00194"></a>00194
 217 </pre></div></div>
 218 <hr size="1"><address style="text-align: right;"><small>Generated on Sat Aug 29 00:01:52 2009 for Barry by&nbsp;
 219 <a href="http://www.doxygen.org/index.html">
 220 <img src="doxygen.png" alt="doxygen" align="middle" border="0"></a> 1.5.6 </small></address>
 221 </body>
 222 </html>