gtk2_ardour/pingback.cc

   1 /*
   2     Copyright (C) 2012 Paul Davis
   3     Inspired by code from Ben Loftis @ Harrison Consoles
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program; if not, write to the Free Software
  17     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18
  19 */
  20
  21 #include <string>
  22 #include <cstring>
  23
  24 #ifdef PLATFORM_WINDOWS
  25 #include <windows.h>
  26 #include <glibmm.h>
  27 #else
  28 #include <sys/utsname.h>
  29 #endif
  30
  31 #include <curl/curl.h>
  32
  33 #include "pbd/gstdio_compat.h"
  34 #include <glibmm/miscutils.h>
  35
  36 #include "pbd/compose.h"
  37 #include "pbd/pthread_utils.h"
  38
  39 #include "ardour/filesystem_paths.h"
  40 #include "ardour/rc_configuration.h"
  41
  42 #include "pingback.h"
  43 #include "utils.h"
  44
  45 using std::string;
  46 using namespace ARDOUR;
  47
  48 static size_t
  49 curl_write_data (char *bufptr, size_t size, size_t nitems, void *ptr)
  50 {
  51         /* we know its a string */
  52
  53         string* sptr = (string*) ptr;
  54
  55         for (size_t i = 0; i < nitems; ++i) {
  56                 for (size_t n = 0; n < size; ++n) {
  57                         if (*bufptr == '\n') {
  58                                 break;
  59                         }
  60
  61                         (*sptr) += *bufptr++;
  62                 }
  63         }
  64
  65         return size * nitems;
  66 }
  67
  68 struct ping_call {
  69     std::string version;
  70     std::string announce_path;
  71
  72     ping_call (const std::string& v, const std::string& a)
  73             : version (v), announce_path (a) {}
  74 };
  75
  76 #ifdef PLATFORM_WINDOWS
  77 static bool
  78 _query_registry (const char *regkey, const char *regval, std::string &rv) {
  79         HKEY key;
  80         DWORD size = PATH_MAX;
  81         char tmp[PATH_MAX+1];
  82
  83         if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ, &key))
  84             && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
  85                  )
  86         {
  87                 rv = Glib::locale_to_utf8 (tmp);
  88                 return true;
  89         }
  90
  91         if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ | KEY_WOW64_32KEY, &key))
  92             && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
  93                  )
  94         {
  95                 rv = Glib::locale_to_utf8 (tmp);
  96                 return true;
  97         }
  98
  99         return false;
 100 }
 101 #endif
 102
 103
 104 static void*
 105 _pingback (void *arg)
 106 {
 107         ping_call* cm = static_cast<ping_call*> (arg);
 108         CURL* c;
 109         string return_str;
 110         //initialize curl
 111
 112         curl_global_init (CURL_GLOBAL_NOTHING);
 113         c = curl_easy_init ();
 114
 115         curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, curl_write_data);
 116         curl_easy_setopt (c, CURLOPT_WRITEDATA, &return_str);
 117         char errbuf[CURL_ERROR_SIZE];
 118         curl_easy_setopt (c, CURLOPT_ERRORBUFFER, errbuf);
 119
 120         string url;
 121
 122 #ifdef __APPLE__
 123         url = Config->get_osx_pingback_url ();
 124 #elif defined PLATFORM_WINDOWS
 125         url = Config->get_windows_pingback_url ();
 126 #else
 127         url = Config->get_linux_pingback_url ();
 128 #endif
 129
 130         if (url.compare (0, 4, "http") != 0) {
 131                 delete cm;
 132                 return 0;
 133         }
 134
 135         char* v = curl_easy_escape (c, cm->version.c_str(), cm->version.length());
 136         url += v;
 137         url += '?';
 138         free (v);
 139
 140 #ifndef PLATFORM_WINDOWS
 141         struct utsname utb;
 142
 143         if (uname (&utb)) {
 144                 delete cm;
 145                 return 0;
 146         }
 147
 148         //string uts = string_compose ("%1 %2 %3 %4", utb.sysname, utb.release, utb.version, utb.machine);
 149         string s;
 150         char* query;
 151
 152         query = curl_easy_escape (c, utb.sysname, strlen (utb.sysname));
 153         s = string_compose ("s=%1", query);
 154         url += s;
 155         url += '&';
 156         free (query);
 157
 158         query = curl_easy_escape (c, utb.release, strlen (utb.release));
 159         s = string_compose ("r=%1", query);
 160         url += s;
 161         url += '&';
 162         free (query);
 163
 164         query = curl_easy_escape (c, utb.machine, strlen (utb.machine));
 165         s = string_compose ("m=%1", query);
 166         url += s;
 167         free (query);
 168 #else
 169         std::string val;
 170         if (_query_registry("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion", "ProductName", val)) {
 171                 char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
 172                 url += "r=";
 173                 url += query;
 174                 url += '&';
 175                 free (query);
 176         } else {
 177                 url += "r=&";
 178         }
 179
 180         if (_query_registry("Hardware\\Description\\System\\CentralProcessor\\0", "Identifier", val)) {
 181                 // remove "Family X Model YY Stepping Z" tail
 182                 size_t cut = val.find (" Family ");
 183                 if (string::npos != cut) {
 184                         val = val.substr (0, cut);
 185                 }
 186                 char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
 187                 url += "m=";
 188                 url += query;
 189                 url += '&';
 190                 free (query);
 191         } else {
 192                 url += "m=&";
 193         }
 194
 195 # if ( defined(__x86_64__) || defined(_M_X64) )
 196         url += "s=Windows64";
 197 # else
 198         url += "s=Windows32";
 199 # endif
 200
 201 #endif /* PLATFORM_WINDOWS */
 202
 203         curl_easy_setopt (c, CURLOPT_URL, url.c_str());
 204
 205         return_str = "";
 206
 207         if (curl_easy_perform (c) == 0) {
 208                 long http_status;
 209
 210                 curl_easy_getinfo (c, CURLINFO_RESPONSE_CODE, &http_status);
 211
 212                 if (http_status != 200) {
 213                         std::cerr << "Bad HTTP status" << std::endl;
 214                         return 0;
 215                 }
 216
 217                 if ( return_str.length() > 140 ) { // like a tweet :)
 218                         std::cerr << "Announcement string is too long (probably behind a proxy)." << std::endl;
 219                 } else {
 220                         std::cerr << "Announcement is: " << return_str << std::endl;
 221
 222                         //write announcements to local file, even if the
 223                         //announcement is empty
 224
 225                         FILE* fout = g_fopen (cm->announce_path.c_str(), "wb");
 226
 227                         if (fout) {
 228                                 fwrite (return_str.c_str(), sizeof(char), return_str.length (), fout);
 229                                 fclose (fout);
 230                         }
 231                 }
 232         } else {
 233                 std::cerr << "curl failed: " << errbuf << std::endl;
 234         }
 235
 236         curl_easy_cleanup (c);
 237         delete cm;
 238         return 0;
 239 }
 240
 241 namespace ARDOUR {
 242
 243 void pingback (const string& version, const string& announce_path)
 244 {
 245         if (ARDOUR_UI_UTILS::running_from_source_tree ()) {
 246                 /* we don't ping under these conditions, because the user is
 247                    probably just paul or robin :)
 248                 */
 249                 return;
 250         }
 251
 252         ping_call* cm = new ping_call (version, announce_path);
 253         pthread_t thread;
 254
 255         pthread_create_and_store ("pingback", &thread, _pingback, cm);
 256 }
 257
 258 }