3 /// Generate SHA1 sums of raw Blackberry database records.
4 /// This is mostly useful for data verification during testing.
8 Copyright (C) 2008, Net Direct Inc. (http://www.netdirect.ca/)
10 This program is free software; you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation; either version 2 of the License, or
13 (at your option) any later version.
15 This program is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
19 See the GNU General Public License in the COPYING file at the
20 root directory of this project for more details.
23 #include <barry/barry.h>
31 using namespace Barry
;
36 const char *Version
= Barry::Version(major
, minor
);
39 << "brecsum - Generate SHA1 sums of raw Blackberry database records.\n"
40 << " Copyright 2008, Net Direct Inc. (http://www.netdirect.ca/)\n"
41 << " Using: " << Version
<< "\n"
43 << " -d db Read database 'db' and sum all its records.\n"
44 << " Can be used multiple times to fetch more than one DB\n"
46 << " -i Include Type and Unique record IDs in the checksums\n"
47 << " -p pin PIN of device to talk with\n"
48 << " If only one device is plugged in, this flag is optional\n"
49 << " -P pass Simplistic method to specify device password\n"
50 << " -v Dump protocol data during operation\n"
54 class ChecksumParser
: public Barry::Parser
60 explicit ChecksumParser(bool IncludeIds
)
61 : m_IncludeIds(IncludeIds
)
69 virtual void SetIds(uint8_t RecType
, uint32_t UniqueId
)
72 SHA1_Update(&m_ctx
, &RecType
, sizeof(RecType
));
73 SHA1_Update(&m_ctx
, &UniqueId
, sizeof(UniqueId
));
77 virtual void ParseHeader(const Barry::Data
&, size_t &)
79 // do nothing here, parse it all at once in ParseFields
82 virtual void ParseFields(const Barry::Data
&data
, size_t &offset
)
84 int len
= data
.GetSize() - offset
;
85 SHA1_Update(&m_ctx
, data
.GetData() + offset
, len
);
91 unsigned char sha1
[SHA_DIGEST_LENGTH
];
92 SHA1_Final(sha1
, &m_ctx
);
94 for( int i
= 0; i
< SHA_DIGEST_LENGTH
; i
++ ) {
95 cout
<< hex
<< setfill('0') << setw(2)
96 << (unsigned int) sha1
[i
];
102 int main(int argc
, char *argv
[])
104 cout
.sync_with_stdio(true); // leave this on, since libusb uses
105 // stdio for debug messages
114 vector
<string
> dbNames
;
116 // process command line options
118 int cmd
= getopt(argc
, argv
, "d:hip:P:v");
124 case 'd': // show dbname
125 dbNames
.push_back(string(optarg
));
128 case 'i': // Include IDs
132 case 'p': // Blackberry PIN
133 pin
= strtoul(optarg
, NULL
, 16);
136 case 'P': // Device password
140 case 'v': // data dump on
151 // Display usage info if user appears confused
152 if( !dbNames
.size() ) {
157 // Initialize the barry library. Must be called before
159 Barry::Init(data_dump
);
161 // Probe the USB bus for Blackberry devices and display.
163 int activeDevice
= probe
.FindActive(pin
);
164 if( activeDevice
== -1 ) {
165 cerr
<< "No device selected, or PIN not found" << endl
;
169 // Create our controller object
170 Barry::Controller
con(probe
.Get(activeDevice
));
171 Barry::Mode::Desktop
desktop(con
);
173 // Sum all specified databases
174 if( dbNames
.size() ) {
175 vector
<string
>::iterator b
= dbNames
.begin();
176 ChecksumParser
parser(include_ids
);
178 desktop
.Open(password
.c_str());
179 for( ; b
!= dbNames
.end(); b
++ ) {
180 unsigned int id
= desktop
.GetDBID(*b
);
181 desktop
.LoadDatabase(id
, parser
);
186 catch( std::exception
&e
) {
187 std::cerr
<< e
.what() << endl
;