src/connector.cc

   1 ///
   2 /// \file       connector.cc
   3 ///             Base class interface for handling Mode connections to device
   4 ///
   5
   6 /*
   7     Copyright (C) 2011-2012, Net Direct Inc. (http://www.netdirect.ca/)
   8
   9     This program is free software; you can redistribute it and/or modify
  10     it under the terms of the GNU General Public License as published by
  11     the Free Software Foundation; either version 2 of the License, or
  12     (at your option) any later version.
  13
  14     This program is distributed in the hope that it will be useful,
  15     but WITHOUT ANY WARRANTY; without even the implied warranty of
  16     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  17
  18     See the GNU General Public License in the COPYING file at the
  19     root directory of this project for more details.
  20 */
  21
  22 #include "connector.h"
  23 #include "router.h"
  24 #include "controller.h"
  25 #include "m_desktop.h"
  26 #include "debug.h"
  27
  28 using namespace std;
  29 using namespace Barry;
  30
  31 namespace Barry {
  32
  33 //////////////////////////////////////////////////////////////////////////////
  34 // Connector base class
  35
  36 // we use a const char *password here because we don't want the
  37 // responsibility of clear its memory... that's the application's job
  38 Connector::Connector(const char *password,
  39                      const std::string &locale,
  40                      Barry::Pin pin)
  41         : m_password(password)
  42         , m_needs_reconnect(false)
  43         , m_ic(locale.c_str())
  44         , m_probe_result(FindDevice(pin))
  45         , m_connect_count(0)
  46         , m_last_disconnect(0)
  47         , m_bpcopy("", 0, 0)
  48 {
  49 }
  50
  51 Connector::Connector(const char *password,
  52                      const std::string &locale,
  53                      const Barry::ProbeResult &result)
  54         : m_password(password)
  55         , m_needs_reconnect(false)
  56         , m_ic(locale.c_str())
  57         , m_probe_result(result)
  58         , m_connect_count(0)
  59         , m_last_disconnect(0)
  60         , m_bpcopy("", 0, 0)
  61 {
  62 }
  63
  64 Connector::~Connector()
  65 {
  66 }
  67
  68 Barry::ProbeResult Connector::FindDevice(Barry::Pin pin)
  69 {
  70         Barry::Probe probe;
  71         int i = probe.FindActive(pin);
  72         if( i != -1 )
  73                 return probe.Get(i);
  74         else
  75                 throw Barry::PinNotFound(pin, probe.GetCount());
  76 }
  77
  78 void Connector::ClearPassword()
  79 {
  80         // blank the memory first
  81         size_t len = m_password.size();
  82         while( len ) {
  83                 len--;
  84                 m_password[len] = '0';
  85         }
  86
  87         // free it
  88         m_password.clear();
  89 }
  90
  91 void Connector::SetPassword(const char *password)
  92 {
  93         ClearPassword();
  94         m_password = password;
  95 }
  96
  97 bool Connector::Connect()
  98 {
  99         Disconnect();
 100
 101         bool started = false;
 102         for(;;) {
 103
 104                 try {
 105                         if( !started ) {
 106                                 started = true;
 107                                 StartConnect(m_password.c_str());
 108                         }
 109                         else {
 110                                 RetryPassword(m_password.c_str());
 111                         }
 112
 113                         FinishConnect();
 114                         m_connect_count++;
 115                         return true;
 116
 117                 }
 118                 catch( BadPassword &bp ) {
 119                         if( bp.out_of_tries() ) {
 120                                 throw;
 121                         }
 122
 123                         m_bpcopy = bp;
 124
 125                         // fall through to password prompt
 126                 }
 127
 128                 // ask user for device password
 129                 ClearPassword();
 130                 if( !PasswordPrompt(m_bpcopy, m_password) ) {
 131                         // user wants out
 132                         return false;
 133                 }
 134         }
 135 }
 136
 137 void Connector::Disconnect()
 138 {
 139         m_needs_reconnect = false;
 140         if( !IsDisconnected() ) {
 141                 DoDisconnect();
 142                 m_last_disconnect = time(NULL);
 143         }
 144 }
 145
 146 bool Connector::Reconnect(int total_tries)
 147 {
 148         int tries = 0;
 149
 150         while(1) try {
 151
 152                 tries++;
 153
 154                 Disconnect();
 155
 156                 if( m_connect_count ) {
 157                         // let the device settle... this seems to help prevent
 158                         // the firmware hang, and therefore ultimately speeds
 159                         // up the sync
 160                         if( (time(NULL) - m_last_disconnect) < 2 ) {
 161                                 // don't bother sleeping if 2 seconds have
 162                                 // already passed
 163                                 sleep(1);
 164                         }
 165
 166                         // temporary fix for odd reconnect message...
 167                         // without this probe, the reconnect will often fail on
 168                         // newer Blackberries due to an unexpected close socket
 169                         // message.
 170                         //
 171                         // It is unclear if this is really a message from
 172                         // the device, but until then, we add this probe.
 173                         m_probe_result = FindDevice(m_probe_result.m_pin);
 174                 }
 175
 176                 return Connect();
 177         }
 178         catch( Usb::Timeout & ) {
 179                 if( tries >= total_tries ) {
 180                         throw;
 181                 }
 182                 else {
 183                         dout("Timeout in Connector::Reconnect()... trying again");
 184                 }
 185         }
 186 }
 187
 188 bool Connector::ReconnectForDirtyFlags()
 189 {
 190         if( m_needs_reconnect ) {
 191                 return Reconnect();
 192         }
 193         else {
 194                 return true;
 195         }
 196 }
 197
 198 void Connector::RequireDirtyReconnect()
 199 {
 200         m_needs_reconnect = true;
 201 }
 202
 203
 204 //////////////////////////////////////////////////////////////////////////////
 205 // DesktopConnector class
 206
 207 DesktopConnector::DesktopConnector(const char *password,
 208                                 const std::string &locale,
 209                                 Barry::Pin pin,
 210                                 Barry::SocketRoutingQueue *router,
 211                                 int connect_timeout)
 212         : Connector(password, locale, pin)
 213         , m_router(router)
 214         , m_connect_timeout(connect_timeout)
 215 {
 216 }
 217
 218 DesktopConnector::DesktopConnector(const char *password,
 219                                 const std::string &locale,
 220                                 const Barry::ProbeResult &result,
 221                                 Barry::SocketRoutingQueue *router,
 222                                 int connect_timeout)
 223         : Connector(password, locale, result)
 224         , m_router(router)
 225         , m_connect_timeout(connect_timeout)
 226 {
 227 }
 228
 229 void DesktopConnector::StartConnect(const char *password)
 230 {
 231         // Note that there is a firmware issue that causes the firmware
 232         // to sometimes hang during a connect and it fails to respond
 233         // to a Desktop::Open() call.  To work around this, set the
 234         // timeout to something lower than the usual 30 seconds.
 235         // The default in DesktopConnector is 10 seconds, which should
 236         // be fine.
 237         //
 238         // If this bug triggers, a Timeout exception will be thrown,
 239         // which will be caught by the Reconnect() method, and Reconnect()
 240         // will retry according to the total number of retries it is
 241         // set to do.
 242         //
 243         if( m_router ) {
 244                 m_con.reset( new Barry::Controller(m_probe_result,
 245                                                 *m_router, m_connect_timeout) );
 246         }
 247         else {
 248                 m_con.reset( new Barry::Controller(m_probe_result,
 249                                                 m_connect_timeout) );
 250         }
 251         m_desktop.reset( new Barry::Mode::Desktop(*m_con, m_ic) );
 252         m_desktop->Open(password);
 253 }
 254
 255 void DesktopConnector::RetryPassword(const char *password)
 256 {
 257         m_desktop->RetryPassword(password);
 258 }
 259
 260 void DesktopConnector::FinishConnect()
 261 {
 262 }
 263
 264 void DesktopConnector::DoDisconnect()
 265 {
 266         m_desktop.reset();
 267         m_con.reset();
 268 }
 269
 270 bool DesktopConnector::IsDisconnected()
 271 {
 272         // return true if DoDisconnect can safely be skipped
 273         return !m_con.get() && !m_desktop.get();
 274 }
 275
 276 bool DesktopConnector::IsConnected()
 277 {
 278         if( m_con.get() && m_desktop.get() )
 279                 return true;
 280         return false;
 281 }
 282
 283 }
 284