portable c-code for reading pingback/announcement message

[ardour.git] / gtk2_ardour / pingback.cc

diff --git a/gtk2_ardour/pingback.cc b/gtk2_ardour/pingback.cc
index ccec81cba7a70164d93728cd5a36e21c7e97470f..9df81650fa122f4751bef9f5222a78834bd5533e 100644 (file)
--- a/gtk2_ardour/pingback.cc
+++ b/gtk2_ardour/pingback.cc
@@ -23,12 +23,16 @@
 #include <fstream>
 #include <cstring>
 
-#ifndef WIN32
+#ifdef PLATFORM_WINDOWS
+#include <windows.h>
+#include <glibmm.h>
+#else
 #include <sys/utsname.h>
 #endif
 
 #include <curl/curl.h>
 
+#include <glib/gstdio.h>
 #include <glibmm/miscutils.h>
 
 #include "pbd/compose.h"
@@ -70,19 +74,40 @@ struct ping_call {
            : version (v), announce_path (a) {}
 };
 
+#ifdef PLATFORM_WINDOWS
+static bool
+_query_registry (const char *regkey, const char *regval, std::string &rv) {
+       HKEY key;
+       DWORD size = PATH_MAX;
+       char tmp[PATH_MAX+1];
+
+       if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ, &key))
+           && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
+                )
+       {
+               rv = Glib::locale_to_utf8 (tmp);
+               return true;
+       }
+
+       if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, regkey, 0, KEY_READ | KEY_WOW64_32KEY, &key))
+           && (ERROR_SUCCESS == RegQueryValueExA (key, regval, 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
+                )
+       {
+               rv = Glib::locale_to_utf8 (tmp);
+               return true;
+       }
+
+       return false;
+}
+#endif
+
+
 static void*
 _pingback (void *arg)
 {
-#ifndef WIN32
        ping_call* cm = static_cast<ping_call*> (arg);
        CURL* c;
-       struct utsname utb;
        string return_str;
-
-       if (uname (&utb)) {
-               return 0;
-       }
-
        //initialize curl
 
        curl_global_init (CURL_GLOBAL_NOTHING);
@@ -97,16 +122,31 @@ _pingback (void *arg)
 
 #ifdef __APPLE__
        url = Config->get_osx_pingback_url ();
+#elif defined PLATFORM_WINDOWS
+       url = Config->get_windows_pingback_url ();
 #else
        url = Config->get_linux_pingback_url ();
 #endif
 
+       if (url.compare (0, 4, "http") != 0) {
+               delete cm;
+               return 0;
+       }
+
        char* v = curl_easy_escape (c, cm->version.c_str(), cm->version.length());
        url += v;
        url += '?';
        free (v);
 
-       string uts = string_compose ("%1 %2 %3 %4", utb.sysname, utb.release, utb.version, utb.machine);
+#ifndef PLATFORM_WINDOWS
+       struct utsname utb;
+
+       if (uname (&utb)) {
+               delete cm;
+               return 0;
+       }
+
+       //string uts = string_compose ("%1 %2 %3 %4", utb.sysname, utb.release, utb.version, utb.machine);
        string s;
        char* query;
 
@@ -126,6 +166,40 @@ _pingback (void *arg)
        s = string_compose ("m=%1", query);
        url += s;
        free (query);
+#else
+       std::string val;
+       if (_query_registry("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion", "ProductName", val)) {
+               char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
+               url += "r=";
+               url += query;
+               url += '&';
+               free (query);
+       } else {
+               url += "r=&";
+       }
+
+       if (_query_registry("Hardware\\Description\\System\\CentralProcessor\\0", "Identifier", val)) {
+               // remove "Family X Model YY Stepping Z" tail
+               size_t cut = val.find (" Family ");
+               if (string::npos != cut) {
+                       val = val.substr (0, cut);
+               }
+               char* query = curl_easy_escape (c, val.c_str(), strlen (val.c_str()));
+               url += "m=";
+               url += query;
+               url += '&';
+               free (query);
+       } else {
+               url += "m=&";
+       }
+
+# if ( defined(__x86_64__) || defined(_M_X64) )
+       url += "s=Windows64";
+# else
+       url += "s=Windows32";
+# endif
+
+#endif /* PLATFORM_WINDOWS */
 
        curl_easy_setopt (c, CURLOPT_URL, url.c_str());
 
@@ -148,11 +222,12 @@ _pingback (void *arg)
                        
                        //write announcements to local file, even if the
                        //announcement is empty
-                               
-                       std::ofstream annc_file (cm->announce_path.c_str());
-                       
-                       if (annc_file) {
-                               annc_file << return_str;
+
+                       FILE* fout = g_fopen (cm->announce_path.c_str(), "wb");
+
+                       if (fout) {
+                               fwrite (return_str.c_str(), sizeof(char), return_str.length (), fout);
+                               fclose (fout);
                        }
                }
        } else {
@@ -161,9 +236,6 @@ _pingback (void *arg)
 
        curl_easy_cleanup (c);
        delete cm;
-
-#endif /* WIN32 */
-
        return 0;
 }