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
  44 using std::string;
  45 using namespace ARDOUR;
  46
  47 static size_t
  48 curl_write_data (char *bufptr, size_t size, size_t nitems, void *ptr)
  49 {
  50         /* we know its a string */
  51
  52         string* sptr = (string*) ptr;
  53
  54         for (size_t i = 0; i < nitems; ++i) {
  55                 for (size_t n = 0; n < size; ++n) {
  56                         if (*bufptr == '\n') {
  57                                 break;
  58                         }
  59
  60                         (*sptr) += *bufptr++;
  61                 }
  62         }
  63
  64         return size * nitems;
  65 }
  66
  67 struct ping_call {
  68     std::string version;
  69     std::string announce_path;
  70
  71     ping_call (const std::string& v, const std::string& a)
  72             : version (v), announce_path (a) {}
  73 };
  74
  75 #ifdef PLATFORM_WINDOWS
  76 static bool
  77 _query_registry (const char *regkey, const char *regval, std::string &rv) {
  78         HKEY key;
  79         DWORD size = PATH_MAX;
  80         char tmp[PATH_MAX+1];
  81
  82         if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ, &key))
  83             && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
  84                  )
  85         {
  86                 rv = Glib::locale_to_utf8 (tmp);
  87                 return true;
  88         }
  89
  90         if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ | KEY_WOW64_32KEY, &key))
  91             && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
  92                  )
  93         {
  94                 rv = Glib::locale_to_utf8 (tmp);
  95                 return true;
  96         }
  97
  98         return false;
  99 }
 100 #endif
 101
 102
 103 static void*
 104 _pingback (void *arg)
 105 {
 106         ping_call* cm = static_cast<ping_call*> (arg);
 107         CURL* c;
 108         string return_str;
 109         //initialize curl
 110
 111         curl_global_init (CURL_GLOBAL_NOTHING);
 112         c = curl_easy_init ();
 113
 114         curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, curl_write_data);
 115         curl_easy_setopt (c, CURLOPT_WRITEDATA, &return_str);
 116         char errbuf[CURL_ERROR_SIZE];
 117         curl_easy_setopt (c, CURLOPT_ERRORBUFFER, errbuf);
 118
 119         string url;
 120
 121 #ifdef __APPLE__
 122         url = Config->get_osx_pingback_url ();
 123 #elif defined PLATFORM_WINDOWS
 124         url = Config->get_windows_pingback_url ();
 125 #else
 126         url = Config->get_linux_pingback_url ();
 127 #endif
 128
 129         if (url.compare (0, 4, "http") != 0) {
 130                 delete cm;
 131                 return 0;
 132         }
 133
 134         char* v = curl_easy_escape (c, cm->version.c_str(), cm->version.length());
 135         url += v;
 136         url += '?';
 137         free (v);
 138
 139 #ifndef PLATFORM_WINDOWS
 140         struct utsname utb;
 141
 142         if (uname (&utb)) {
 143                 delete cm;
 144                 return 0;
 145         }
 146
 147         //string uts = string_compose ("%1 %2 %3 %4", utb.sysname, utb.release, utb.version, utb.machine);
 148         string s;
 149         char* query;
 150
 151         query = curl_easy_escape (c, utb.sysname, strlen (utb.sysname));
 152         s = string_compose ("s=%1", query);
 153         url += s;
 154         url += '&';
 155         free (query);
 156
 157         query = curl_easy_escape (c, utb.release, strlen (utb.release));
 158         s = string_compose ("r=%1", query);
 159         url += s;
 160         url += '&';
 161         free (query);
 162
 163         query = curl_easy_escape (c, utb.machine, strlen (utb.machine));
 164         s = string_compose ("m=%1", query);
 165         url += s;
 166         free (query);
 167 #else
 168         std::string val;
 169         if (_query_registry("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion", "ProductName", val)) {
 170                 char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
 171                 url += "r=";
 172                 url += query;
 173                 url += '&';
 174                 free (query);
 175         } else {
 176                 url += "r=&";
 177         }
 178
 179         if (_query_registry("Hardware\\Description\\System\\CentralProcessor\\0", "Identifier", val)) {
 180                 // remove "Family X Model YY Stepping Z" tail
 181                 size_t cut = val.find (" Family ");
 182                 if (string::npos != cut) {
 183                         val = val.substr (0, cut);
 184                 }
 185                 char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
 186                 url += "m=";
 187                 url += query;
 188                 url += '&';
 189                 free (query);
 190         } else {
 191                 url += "m=&";
 192         }
 193
 194 # if ( defined(__x86_64__) || defined(_M_X64) )
 195         url += "s=Windows64";
 196 # else
 197         url += "s=Windows32";
 198 # endif
 199
 200 #endif /* PLATFORM_WINDOWS */
 201
 202         curl_easy_setopt (c, CURLOPT_URL, url.c_str());
 203
 204         return_str = "";
 205
 206         if (curl_easy_perform (c) == 0) {
 207                 long http_status;
 208
 209                 curl_easy_getinfo (c, CURLINFO_RESPONSE_CODE, &http_status);
 210
 211                 if (http_status != 200) {
 212                         std::cerr << "Bad HTTP status" << std::endl;
 213                         return 0;
 214                 }
 215
 216                 if ( return_str.length() > 140 ) { // like a tweet :)
 217                         std::cerr << "Announcement string is too long (probably behind a proxy)." << std::endl;
 218                 } else {
 219                         std::cerr << "Announcement is: " << return_str << std::endl;
 220
 221                         //write announcements to local file, even if the
 222                         //announcement is empty
 223
 224                         FILE* fout = g_fopen (cm->announce_path.c_str(), "wb");
 225
 226                         if (fout) {
 227                                 fwrite (return_str.c_str(), sizeof(char), return_str.length (), fout);
 228                                 fclose (fout);
 229                         }
 230                 }
 231         } else {
 232                 std::cerr << "curl failed: " << errbuf << std::endl;
 233         }
 234
 235         curl_easy_cleanup (c);
 236         delete cm;
 237         return 0;
 238 }
 239
 240 namespace ARDOUR {
 241
 242 void pingback (const string& version, const string& announce_path)
 243 {
 244         /* check this is not being run from ./ardev etc. */
 245         gchar const *x = g_getenv ("ARDOUR_THEMES_PATH");
 246
 247         if (x && string (x).find ("gtk2_ardour") != string::npos) {
 248                 std::cerr << "no ping\n";
 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 }