223497cbffcfcdb1522d72dfaf2fa750ef832b7b
[ardour.git] / 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 <iostream>
23 #include <fstream>
24 #include <cstring>
25
26 #ifdef PLATFORM_WINDOWS
27 #include <windows.h>
28 #include <glibmm.h>
29 #else
30 #include <sys/utsname.h>
31 #endif
32
33 #include <curl/curl.h>
34
35 #include <glibmm/miscutils.h>
36
37 #include "pbd/compose.h"
38 #include "pbd/pthread_utils.h"
39
40 #include "ardour/filesystem_paths.h"
41 #include "ardour/rc_configuration.h"
42
43 #include "pingback.h"
44
45 using std::string;
46 using namespace ARDOUR;
47
48 #ifndef PLATFORM_WINDOWS // no pingback for windows, so far
49 static size_t
50 curl_write_data (char *bufptr, size_t size, size_t nitems, void *ptr)
51 {
52         /* we know its a string */
53
54         string* sptr = (string*) ptr;
55
56         for (size_t i = 0; i < nitems; ++i) {
57                 for (size_t n = 0; n < size; ++n) {
58                         if (*bufptr == '\n') {
59                                 break;
60                         }
61
62                         (*sptr) += *bufptr++;
63                 }
64         }
65
66         return size * nitems;
67 }
68 #endif
69
70 struct ping_call {
71     std::string version;
72     std::string announce_path;
73
74     ping_call (const std::string& v, const std::string& a)
75             : version (v), announce_path (a) {}
76 };
77
78 static void*
79 _pingback (void *arg)
80 {
81         ping_call* cm = static_cast<ping_call*> (arg);
82         CURL* c;
83         string return_str;
84         //initialize curl
85
86         curl_global_init (CURL_GLOBAL_NOTHING);
87         c = curl_easy_init ();
88         
89         curl_easy_setopt (c, CURLOPT_WRITEFUNCTION, curl_write_data); 
90         curl_easy_setopt (c, CURLOPT_WRITEDATA, &return_str); 
91         char errbuf[CURL_ERROR_SIZE];
92         curl_easy_setopt (c, CURLOPT_ERRORBUFFER, errbuf); 
93
94         string url;
95
96 #ifdef __APPLE__
97         url = Config->get_osx_pingback_url ();
98 #elif defined PLATFORM_WINDOWS
99         url = Config->get_windows_pingback_url ();
100 #else
101         url = Config->get_linux_pingback_url ();
102 #endif
103
104         if (url.compare (0, 4, "http") != 0) {
105                 delete cm;
106                 return 0;
107         }
108
109         char* v = curl_easy_escape (c, cm->version.c_str(), cm->version.length());
110         url += v;
111         url += '?';
112         free (v);
113
114 #ifndef PLATFORM_WINDOWS
115         struct utsname utb;
116
117         if (uname (&utb)) {
118                 delete cm;
119                 return 0;
120         }
121
122         //string uts = string_compose ("%1 %2 %3 %4", utb.sysname, utb.release, utb.version, utb.machine);
123         string s;
124         char* query;
125
126         query = curl_easy_escape (c, utb.sysname, strlen (utb.sysname));
127         s = string_compose ("s=%1", query);
128         url += s;
129         url += '&';
130         free (query);
131
132         query = curl_easy_escape (c, utb.release, strlen (utb.release));
133         s = string_compose ("r=%1", query);
134         url += s;
135         url += '&';
136         free (query);
137
138         query = curl_easy_escape (c, utb.machine, strlen (utb.machine));
139         s = string_compose ("m=%1", query);
140         url += s;
141         free (query);
142 #else
143         // this is hilarious: https://msdn.microsoft.com/en-us/library/windows/desktop/ms724429%28v=vs.85%29.aspx
144         url += "r=&";
145
146         HKEY key;
147         DWORD size = PATH_MAX;
148         char tmp[PATH_MAX+1];
149         if (   (ERROR_SUCCESS == RegOpenKeyExA (HKEY_LOCAL_MACHINE, "Hardware\\Description\\System\\CentralProcessor\\0", 0, KEY_READ, &key))
150             && (ERROR_SUCCESS == RegQueryValueExA (key, "Identifier", 0, NULL, reinterpret_cast<LPBYTE>(tmp), &size))
151                         // or "ProcessorNameString"
152                  )
153         {
154                 string s = Glib::locale_to_utf8 (tmp);
155                 char* query = curl_easy_escape (c, s.c_str(), strlen (s.c_str()));
156                 s = string_compose ("m=%1", query);
157                 url += s;
158                 url += '&';
159                 free (query);
160         } else {
161                 url += "m=&";
162         }
163
164         url += "r=&";
165 # if ( defined(__x86_64__) || defined(_M_X64) )
166         url += "s=Windows64";
167 # else
168         url += "s=Windows32";
169 # endif
170
171 #ifndef NDEBUG
172         std::cerr << "Pingback: " << url << std::endl;
173 #endif
174
175 #endif /* PLATFORM_WINDOWS */
176
177         curl_easy_setopt (c, CURLOPT_URL, url.c_str());
178
179         return_str = "";
180
181         if (curl_easy_perform (c) == 0) {
182                 long http_status; 
183
184                 curl_easy_getinfo (c, CURLINFO_RESPONSE_CODE, &http_status);
185
186                 if (http_status != 200) {
187                         std::cerr << "Bad HTTP status" << std::endl;
188                         return 0;
189                 }
190
191                 if ( return_str.length() > 140 ) { // like a tweet :)
192                         std::cerr << "Announcement string is too long (probably behind a proxy)." << std::endl;
193                 } else {
194                         std::cerr << "Announcement is: " << return_str << std::endl;
195                         
196                         //write announcements to local file, even if the
197                         //announcement is empty
198                                 
199                         std::ofstream annc_file (cm->announce_path.c_str());
200                         
201                         if (annc_file) {
202                                 annc_file << return_str;
203                         }
204                 }
205         } else {
206                 std::cerr << "curl failed: " << errbuf << std::endl;
207         }
208
209         curl_easy_cleanup (c);
210         delete cm;
211         return 0;
212 }
213
214 namespace ARDOUR {
215
216 void pingback (const string& version, const string& announce_path) 
217 {
218         ping_call* cm = new ping_call (version, announce_path);
219         pthread_t thread;
220
221         pthread_create_and_store ("pingback", &thread, _pingback, cm);
222 }
223
224 }