<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<title>Barry: brecsum.cc Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<link href="doxygen.css" rel="stylesheet" type="text/css"/>
</head>
<body>
<!-- Generated by Doxygen 1.6.2-20100208 -->
<div class="navigation" id="top">
<div class="tabs">
<ul>
<li><a href="index.html"><span>Main Page</span></a></li>
<li><a href="modules.html"><span>Modules</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>
<li><a href="dirs.html"><span>Directories</span></a></li>
</ul>
</div>
<div class="tabs">
<ul>
<li><a href="files.html"><span>File List</span></a></li>
</ul>
</div>
<div class="navpath"><a class="el" href="dir_cce6b820c61664c85c51a5136106e856.html">tools</a>
</div>
</div>
<div class="contents">
<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>
<a name="l00002"></a>00002 <span class="comment"></span><span class="comment">/// \file brecsum.cc</span>
<a name="l00003"></a>00003 <span class="comment"></span><span class="comment">/// Generate SHA1 sums of raw Blackberry database records.</span>
<a name="l00004"></a>00004 <span class="comment"></span><span class="comment">/// This is mostly useful for data verification during testing.</span>
<a name="l00005"></a>00005 <span class="comment"></span><span class="comment">///</span>
<a name="l00006"></a>00006 <span class="comment"></span>
<a name="l00007"></a>00007 <span class="comment">/*</span>
<a name="l00008"></a>00008 <span class="comment"> Copyright (C) 2008-2010, Net Direct Inc. (http://www.netdirect.ca/)</span>
<a name="l00009"></a>00009 <span class="comment"></span>
<a name="l00010"></a>00010 <span class="comment"> This program is free software; you can redistribute it and/or modify</span>
<a name="l00011"></a>00011 <span class="comment"> it under the terms of the GNU General Public License as published by</span>
<a name="l00012"></a>00012 <span class="comment"> the Free Software Foundation; either version 2 of the License, or</span>
<a name="l00013"></a>00013 <span class="comment"> (at your option) any later version.</span>
<a name="l00014"></a>00014 <span class="comment"></span>
<a name="l00015"></a>00015 <span class="comment"> This program is distributed in the hope that it will be useful,</span>
<a name="l00016"></a>00016 <span class="comment"> but WITHOUT ANY WARRANTY; without even the implied warranty of</span>
<a name="l00017"></a>00017 <span class="comment"> MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.</span>
<a name="l00018"></a>00018 <span class="comment"></span>
<a name="l00019"></a>00019 <span class="comment"> See the GNU General Public License in the COPYING file at the</span>
<a name="l00020"></a>00020 <span class="comment"> root directory of this project for more details.</span>
<a name="l00021"></a>00021 <span class="comment">*/</span>
<a name="l00022"></a>00022
<a name="l00023"></a>00023 <span class="preprocessor">#include <barry/barry.h></span>
<a name="l00024"></a>00024 <span class="preprocessor">#include <iomanip></span>
<a name="l00025"></a>00025 <span class="preprocessor">#include <iostream></span>
<a name="l00026"></a>00026 <span class="preprocessor">#include <vector></span>
<a name="l00027"></a>00027 <span class="preprocessor">#include <string></span>
<a name="l00028"></a>00028 <span class="preprocessor">#include <getopt.h></span>
<a name="l00029"></a>00029 <span class="preprocessor">#include "<a class="code" href="i18n_8h.html" title="Common internationalization defines, via gettext.">i18n.h</a>"</span>
<a name="l00030"></a>00030
<a name="l00031"></a>00031 <span class="keyword">using namespace </span>std;
<a name="l00032"></a>00032 <span class="keyword">using namespace </span>Barry;
<a name="l00033"></a>00033
<a name="l00034"></a>00034 <span class="keywordtype">void</span> Usage()
<a name="l00035"></a>00035 {
<a name="l00036"></a>00036 <span class="keywordtype">int</span> major, minor;
<a name="l00037"></a>00037 <span class="keyword">const</span> <span class="keywordtype">char</span> *<a class="code" href="namespaceBarry.html#a06ec709ed2313dd84291b2d295a87d26" title="Fills major and minor with integer version numbers, and returns a string containing...">Version</a> = <a class="code" href="namespaceBarry.html#a06ec709ed2313dd84291b2d295a87d26" title="Fills major and minor with integer version numbers, and returns a string containing...">Barry::Version</a>(major, minor);
<a name="l00038"></a>00038
<a name="l00039"></a>00039 cerr
<a name="l00040"></a>00040 << <span class="stringliteral">"brecsum - Generate SHA1 sums of raw Blackberry database records.\n"</span>
<a name="l00041"></a>00041 << <span class="stringliteral">" Copyright 2008-2010, Net Direct Inc. (http://www.netdirect.ca/)\n"</span>
<a name="l00042"></a>00042 << <span class="stringliteral">" Using: "</span> << Version << <span class="stringliteral">"\n"</span>
<a name="l00043"></a>00043 << <span class="stringliteral">"\n"</span>
<a name="l00044"></a>00044 << <span class="stringliteral">" -d db Read database 'db' and sum all its records.\n"</span>
<a name="l00045"></a>00045 << <span class="stringliteral">" Can be used multiple times to fetch more than one DB\n"</span>
<a name="l00046"></a>00046 << <span class="stringliteral">" -h This help\n"</span>
<a name="l00047"></a>00047 << <span class="stringliteral">" -i Include Type and Unique record IDs in the checksums\n"</span>
<a name="l00048"></a>00048 << <span class="stringliteral">" -p pin PIN of device to talk with\n"</span>
<a name="l00049"></a>00049 << <span class="stringliteral">" If only one device is plugged in, this flag is optional\n"</span>
<a name="l00050"></a>00050 << <span class="stringliteral">" -P pass Simplistic method to specify device password\n"</span>
<a name="l00051"></a>00051 << <span class="stringliteral">" -v Dump protocol data during operation\n"</span>
<a name="l00052"></a>00052 << endl;
<a name="l00053"></a>00053 }
<a name="l00054"></a>00054
<a name="l00055"></a><a class="code" href="classChecksumParser.html">00055</a> <span class="keyword">class </span><a class="code" href="classChecksumParser.html">ChecksumParser</a> : <span class="keyword">public</span> Barry::<a class="code" href="classBarry_1_1Parser.html" title="Base class for the parser hierarchy.">Parser</a>
<a name="l00056"></a>00056 {
<a name="l00057"></a>00057 <span class="keywordtype">bool</span> m_IncludeIds;
<a name="l00058"></a>00058 <a class="code" href="structBarry_1_1SHA__CTX.html">SHA_CTX</a> m_ctx;
<a name="l00059"></a>00059
<a name="l00060"></a>00060 <span class="keyword">public</span>:
<a name="l00061"></a>00061 <span class="keyword">explicit</span> <a class="code" href="classChecksumParser.html">ChecksumParser</a>(<span class="keywordtype">bool</span> IncludeIds)
<a name="l00062"></a>00062 : m_IncludeIds(IncludeIds)
<a name="l00063"></a>00063 {}
<a name="l00064"></a>00064
<a name="l00065"></a><a class="code" href="classChecksumParser.html#ae557918eae2b18aad4d2b9d211f637aa">00065</a> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classChecksumParser.html#ae557918eae2b18aad4d2b9d211f637aa" title="Reset and prepare for a new raw data packet.">Clear</a>()
<a name="l00066"></a>00066 {
<a name="l00067"></a>00067 SHA1_Init(&m_ctx);
<a name="l00068"></a>00068 }
<a name="l00069"></a>00069
<a name="l00070"></a><a class="code" href="classChecksumParser.html#ab3b20a2418a4cb1427cdf50524ad911a">00070</a> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classChecksumParser.html#ab3b20a2418a4cb1427cdf50524ad911a" title="Stores the IDs.">SetIds</a>(uint8_t RecType, uint32_t UniqueId)
<a name="l00071"></a>00071 {
<a name="l00072"></a>00072 <span class="keywordflow">if</span>( m_IncludeIds ) {
<a name="l00073"></a>00073 SHA1_Update(&m_ctx, &RecType, <span class="keyword">sizeof</span>(RecType));
<a name="l00074"></a>00074 SHA1_Update(&m_ctx, &UniqueId, <span class="keyword">sizeof</span>(UniqueId));
<a name="l00075"></a>00075 }
<a name="l00076"></a>00076 }
<a name="l00077"></a>00077
<a name="l00078"></a><a class="code" href="classChecksumParser.html#a3a2ac40a6d48218e17f43937a81668a7">00078</a> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classChecksumParser.html#a3a2ac40a6d48218e17f43937a81668a7" title="Called to parse the header portion of the raw data packet.">ParseHeader</a>(<span class="keyword">const</span> <a class="code" href="classBarry_1_1Data.html">Barry::Data</a> &, <span class="keywordtype">size_t</span> &)
<a name="l00079"></a>00079 {
<a name="l00080"></a>00080 <span class="comment">// do nothing here, parse it all at once in ParseFields</span>
<a name="l00081"></a>00081 }
<a name="l00082"></a>00082
<a name="l00083"></a><a class="code" href="classChecksumParser.html#aa8381c46cf4d2f1eca7d13ca0218c98b">00083</a> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classChecksumParser.html#aa8381c46cf4d2f1eca7d13ca0218c98b" title="Called to parse sub fields in the raw data packet.">ParseFields</a>(<span class="keyword">const</span> <a class="code" href="classBarry_1_1Data.html">Barry::Data</a> &data, <span class="keywordtype">size_t</span> &offset,
<a name="l00084"></a>00084 <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)
<a name="l00085"></a>00085 {
<a name="l00086"></a>00086 <span class="keywordtype">int</span> len = data.GetSize() - offset;
<a name="l00087"></a>00087 SHA1_Update(&m_ctx, data.GetData() + offset, len);
<a name="l00088"></a>00088 offset += len;
<a name="l00089"></a>00089 }
<a name="l00090"></a>00090
<a name="l00091"></a><a class="code" href="classChecksumParser.html#adac9dc038ae4436aeef29c04a9b5f9de">00091</a> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classChecksumParser.html#adac9dc038ae4436aeef29c04a9b5f9de" title="Called at the very end of record parsing, and used to store the final packet somewhere...">Store</a>()
<a name="l00092"></a>00092 {
<a name="l00093"></a>00093 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> sha1[SHA_DIGEST_LENGTH];
<a name="l00094"></a>00094 SHA1_Final(sha1, &m_ctx);
<a name="l00095"></a>00095
<a name="l00096"></a>00096 <span class="keywordflow">for</span>( <span class="keywordtype">int</span> i = 0; i < SHA_DIGEST_LENGTH; i++ ) {
<a name="l00097"></a>00097 cout << hex << setfill(<span class="charliteral">'0'</span>) << setw(2)
<a name="l00098"></a>00098 << (<span class="keywordtype">unsigned</span> int) sha1[i];
<a name="l00099"></a>00099 }
<a name="l00100"></a>00100 cout << endl;
<a name="l00101"></a>00101 }
<a name="l00102"></a>00102 };
<a name="l00103"></a>00103
<a name="l00104"></a>00104 <span class="keywordtype">int</span> main(<span class="keywordtype">int</span> argc, <span class="keywordtype">char</span> *argv[])
<a name="l00105"></a>00105 {
<a name="l00106"></a>00106 INIT_I18N(PACKAGE);
<a name="l00107"></a>00107
<a name="l00108"></a>00108 cout.sync_with_stdio(<span class="keyword">true</span>); <span class="comment">// leave this on, since libusb uses</span>
<a name="l00109"></a>00109 <span class="comment">// stdio for debug messages</span>
<a name="l00110"></a>00110
<a name="l00111"></a>00111 <span class="keywordflow">try</span> {
<a name="l00112"></a>00112
<a name="l00113"></a>00113 uint32_t pin = 0;
<a name="l00114"></a>00114 <span class="keywordtype">bool</span>
<a name="l00115"></a>00115 data_dump = <span class="keyword">false</span>,
<a name="l00116"></a>00116 include_ids = <span class="keyword">false</span>;
<a name="l00117"></a>00117 <span class="keywordtype">string</span> password;
<a name="l00118"></a>00118 vector<string> dbNames;
<a name="l00119"></a>00119
<a name="l00120"></a>00120 <span class="comment">// process command line options</span>
<a name="l00121"></a>00121 <span class="keywordflow">for</span>(;;) {
<a name="l00122"></a>00122 <span class="keywordtype">int</span> cmd = getopt(argc, argv, <span class="stringliteral">"d:hip:P:v"</span>);
<a name="l00123"></a>00123 <span class="keywordflow">if</span>( cmd == -1 )
<a name="l00124"></a>00124 <span class="keywordflow">break</span>;
<a name="l00125"></a>00125
<a name="l00126"></a>00126 <span class="keywordflow">switch</span>( cmd )
<a name="l00127"></a>00127 {
<a name="l00128"></a>00128 <span class="keywordflow">case</span> <span class="charliteral">'d'</span>: <span class="comment">// show dbname</span>
<a name="l00129"></a>00129 dbNames.push_back(<span class="keywordtype">string</span>(optarg));
<a name="l00130"></a>00130 <span class="keywordflow">break</span>;
<a name="l00131"></a>00131
<a name="l00132"></a>00132 <span class="keywordflow">case</span> <span class="charliteral">'i'</span>: <span class="comment">// Include IDs</span>
<a name="l00133"></a>00133 include_ids = <span class="keyword">true</span>;
<a name="l00134"></a>00134 <span class="keywordflow">break</span>;
<a name="l00135"></a>00135
<a name="l00136"></a>00136 <span class="keywordflow">case</span> <span class="charliteral">'p'</span>: <span class="comment">// Blackberry PIN</span>
<a name="l00137"></a>00137 pin = strtoul(optarg, NULL, 16);
<a name="l00138"></a>00138 <span class="keywordflow">break</span>;
<a name="l00139"></a>00139
<a name="l00140"></a>00140 <span class="keywordflow">case</span> <span class="charliteral">'P'</span>: <span class="comment">// Device password</span>
<a name="l00141"></a>00141 password = optarg;
<a name="l00142"></a>00142 <span class="keywordflow">break</span>;
<a name="l00143"></a>00143
<a name="l00144"></a>00144 <span class="keywordflow">case</span> <span class="charliteral">'v'</span>: <span class="comment">// data dump on</span>
<a name="l00145"></a>00145 data_dump = <span class="keyword">true</span>;
<a name="l00146"></a>00146 <span class="keywordflow">break</span>;
<a name="l00147"></a>00147
<a name="l00148"></a>00148 <span class="keywordflow">case</span> <span class="charliteral">'h'</span>: <span class="comment">// help</span>
<a name="l00149"></a>00149 <span class="keywordflow">default</span>:
<a name="l00150"></a>00150 Usage();
<a name="l00151"></a>00151 <span class="keywordflow">return</span> 0;
<a name="l00152"></a>00152 }
<a name="l00153"></a>00153 }
<a name="l00154"></a>00154
<a name="l00155"></a>00155 <span class="comment">// Display usage info if user appears confused</span>
<a name="l00156"></a>00156 <span class="keywordflow">if</span>( !dbNames.size() ) {
<a name="l00157"></a>00157 Usage();
<a name="l00158"></a>00158 <span class="keywordflow">return</span> 0;
<a name="l00159"></a>00159 }
<a name="l00160"></a>00160
<a name="l00161"></a>00161 <span class="comment">// Initialize the barry library. Must be called before</span>
<a name="l00162"></a>00162 <span class="comment">// anything else.</span>
<a name="l00163"></a>00163 <a class="code" href="namespaceBarry.html#a2e28bd59aebe0bece5a46f02ef7dbd3d" title="Barry library initializer.">Barry::Init</a>(data_dump);
<a name="l00164"></a>00164
<a name="l00165"></a>00165 <span class="comment">// Probe the USB bus for Blackberry devices and display.</span>
<a name="l00166"></a>00166 <a class="code" href="classBarry_1_1Probe.html">Barry::Probe</a> probe;
<a name="l00167"></a>00167 <span class="keywordtype">int</span> activeDevice = probe.FindActive(pin);
<a name="l00168"></a>00168 <span class="keywordflow">if</span>( activeDevice == -1 ) {
<a name="l00169"></a>00169 cerr << <span class="stringliteral">"No device selected, or PIN not found"</span> << endl;
<a name="l00170"></a>00170 <span class="keywordflow">return</span> 1;
<a name="l00171"></a>00171 }
<a name="l00172"></a>00172
<a name="l00173"></a>00173 <span class="comment">// Create our controller object</span>
<a name="l00174"></a>00174 <a class="code" href="classBarry_1_1Controller.html" title="The main interface class.">Barry::Controller</a> con(probe.Get(activeDevice));
<a name="l00175"></a>00175 <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);
<a name="l00176"></a>00176
<a name="l00177"></a>00177 <span class="comment">// Sum all specified databases</span>
<a name="l00178"></a>00178 <span class="keywordflow">if</span>( dbNames.size() ) {
<a name="l00179"></a>00179 vector<string>::iterator b = dbNames.begin();
<a name="l00180"></a>00180 <a class="code" href="classChecksumParser.html">ChecksumParser</a> parser(include_ids);
<a name="l00181"></a>00181
<a name="l00182"></a>00182 desktop.Open(password.c_str());
<a name="l00183"></a>00183 <span class="keywordflow">for</span>( ; b != dbNames.end(); b++ ) {
<a name="l00184"></a>00184 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <span class="keywordtype">id</span> = desktop.GetDBID(*b);
<a name="l00185"></a>00185 desktop.LoadDatabase(<span class="keywordtype">id</span>, parser);
<a name="l00186"></a>00186 }
<a name="l00187"></a>00187 }
<a name="l00188"></a>00188
<a name="l00189"></a>00189 }
<a name="l00190"></a>00190 <span class="keywordflow">catch</span>( std::exception &e ) {
<a name="l00191"></a>00191 std::cerr << e.what() << endl;
<a name="l00192"></a>00192 <span class="keywordflow">return</span> 1;
<a name="l00193"></a>00193 }
<a name="l00194"></a>00194
<a name="l00195"></a>00195 <span class="keywordflow">return</span> 0;
<a name="l00196"></a>00196 }
<a name="l00197"></a>00197
</pre></div></div>
<hr class="footer"/><address style="text-align: right;"><small>Generated by
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.6.2-20100208 </small></address>
</body>
</html>
