src/qt/qtipcserver.cpp

   1 // Copyright (c) 2009-2012 The Bitcoin developers
   2 // Distributed under the MIT/X11 software license, see the accompanying
   3 // file license.txt or http://www.opensource.org/licenses/mit-license.php.
   4
   5 #include <boost/algorithm/string.hpp>
   6 #include <boost/interprocess/ipc/message_queue.hpp>
   7 #include <boost/tokenizer.hpp>
   8 #include <boost/date_time/posix_time/posix_time.hpp>
   9
  10 #include "headers.h"
  11
  12 using namespace boost::interprocess;
  13 using namespace boost::posix_time;
  14 using namespace boost;
  15 using namespace std;
  16
  17 void ipcShutdown()
  18 {
  19     message_queue::remove("BitcoinURL");
  20 }
  21
  22 void ipcThread(void* parg)
  23 {
  24     message_queue* mq = (message_queue*)parg;
  25     char strBuf[257];
  26     size_t nSize;
  27     unsigned int nPriority;
  28     loop
  29     {
  30         ptime d = boost::posix_time::microsec_clock::universal_time() + millisec(100);
  31         if(mq->timed_receive(&strBuf, sizeof(strBuf), nSize, nPriority, d))
  32         {
  33             ThreadSafeHandleURL(std::string(strBuf, nSize));
  34             Sleep(1000);
  35         }
  36         if (fShutdown)
  37         {
  38             ipcShutdown();
  39             break;
  40         }
  41     }
  42     ipcShutdown();
  43 }
  44
  45 void ipcInit()
  46 {
  47 #ifdef MAC_OSX
  48     // TODO: implement bitcoin: URI handling the Mac Way
  49     return;
  50 #endif
  51
  52     message_queue* mq;
  53     char strBuf[257];
  54     size_t nSize;
  55     unsigned int nPriority;
  56     try {
  57         mq = new message_queue(open_or_create, "BitcoinURL", 2, 256);
  58
  59         // Make sure we don't lose any bitcoin: URIs
  60         for (int i = 0; i < 2; i++)
  61         {
  62             ptime d = boost::posix_time::microsec_clock::universal_time() + millisec(1);
  63             if(mq->timed_receive(&strBuf, sizeof(strBuf), nSize, nPriority, d))
  64             {
  65                 ThreadSafeHandleURL(std::string(strBuf, nSize));
  66             }
  67             else
  68                 break;
  69         }
  70
  71         // Make sure only one bitcoin instance is listening
  72         message_queue::remove("BitcoinURL");
  73         mq = new message_queue(open_or_create, "BitcoinURL", 2, 256);
  74     }
  75     catch (interprocess_exception &ex) {
  76         return;
  77     }
  78     if (!CreateThread(ipcThread, mq))
  79     {
  80         delete mq;
  81     }
  82 }