|
@@ -27,17 +27,17 @@
|
|
|
|
|
|
#include "Constants.hpp"
|
|
|
|
|
|
+#include <stdio.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include <string.h>
|
|
|
+
|
|
|
#ifdef __WINDOWS__
|
|
|
#include <WinSock2.h>
|
|
|
#include <Windows.h>
|
|
|
#include <winhttp.h>
|
|
|
#include <locale>
|
|
|
#include <codecvt>
|
|
|
-#endif
|
|
|
-
|
|
|
-#include <stdio.h>
|
|
|
-#include <stdlib.h>
|
|
|
-#include <string.h>
|
|
|
+#endif // __WINDOWS__
|
|
|
|
|
|
#ifdef __UNIX_LIKE__
|
|
|
#include <unistd.h>
|
|
@@ -48,7 +48,7 @@
|
|
|
#include <sys/stat.h>
|
|
|
#include <sys/socket.h>
|
|
|
#include <sys/wait.h>
|
|
|
-#endif
|
|
|
+#endif // __UNIX_LIKE__
|
|
|
|
|
|
#include <vector>
|
|
|
#include <utility>
|
|
@@ -62,8 +62,6 @@
|
|
|
|
|
|
namespace ZeroTier {
|
|
|
|
|
|
-const std::map<std::string,std::string> HttpClient::NO_HEADERS;
|
|
|
-
|
|
|
#ifdef __UNIX_LIKE__
|
|
|
|
|
|
// The *nix implementation calls 'curl' externally rather than linking to it.
|
|
@@ -76,27 +74,36 @@ const std::map<std::string,std::string> HttpClient::NO_HEADERS;
|
|
|
#endif
|
|
|
|
|
|
// Paths where "curl" may be found on the system
|
|
|
-#define NUM_CURL_PATHS 5
|
|
|
-static const char *CURL_PATHS[NUM_CURL_PATHS] = { "/usr/bin/curl","/bin/curl","/usr/local/bin/curl","/usr/sbin/curl","/sbin/curl" };
|
|
|
+#define NUM_CURL_PATHS 6
|
|
|
+static const char *CURL_PATHS[NUM_CURL_PATHS] = { "/usr/bin/curl","/bin/curl","/usr/local/bin/curl","/usr/sbin/curl","/sbin/curl","/usr/libexec/curl" };
|
|
|
|
|
|
// Maximum message length
|
|
|
#define CURL_MAX_MESSAGE_LENGTH (1024 * 1024 * 64)
|
|
|
|
|
|
// Internal private thread class that performs request, notifies handler,
|
|
|
// and then commits suicide by deleting itself.
|
|
|
-class P_Req : NonCopyable
|
|
|
+class HttpClient_Private_Request : NonCopyable
|
|
|
{
|
|
|
public:
|
|
|
- P_Req(const char *method,const std::string &url,const std::map<std::string,std::string> &headers,unsigned int timeout,void (*handler)(void *,int,const std::string &,bool,const std::string &),void *arg) :
|
|
|
+ HttpClient_Private_Request(HttpClient *parent,const char *method,const std::string &url,const std::map<std::string,std::string> &headers,unsigned int timeout,void (*handler)(void *,int,const std::string &,const std::string &),void *arg) :
|
|
|
_url(url),
|
|
|
_headers(headers),
|
|
|
_timeout(timeout),
|
|
|
_handler(handler),
|
|
|
- _arg(arg)
|
|
|
+ _arg(arg),
|
|
|
+ _parent(parent),
|
|
|
+ _pid(0),
|
|
|
+ _cancelled(false)
|
|
|
{
|
|
|
_myThread = Thread::start(this);
|
|
|
}
|
|
|
|
|
|
+ ~HttpClient_Private_Request()
|
|
|
+ {
|
|
|
+ Mutex::Lock _l(_parent->_requests_m);
|
|
|
+ _parent->_requests.erase((HttpClient::Request)this);
|
|
|
+ }
|
|
|
+
|
|
|
void threadMain()
|
|
|
{
|
|
|
char *curlArgs[1024];
|
|
@@ -111,14 +118,14 @@ public:
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
if (!curlPath.length()) {
|
|
|
- _handler(_arg,-1,_url,false,"unable to locate 'curl' binary in /usr/bin, /bin, /usr/local/bin, /usr/sbin, or /sbin");
|
|
|
+ _doH(_arg,-1,_url,"unable to locate 'curl' binary in /usr/bin, /bin, /usr/local/bin, /usr/sbin, or /sbin");
|
|
|
delete this;
|
|
|
return;
|
|
|
}
|
|
|
-
|
|
|
if (!_url.length()) {
|
|
|
- _handler(_arg,-1,_url,false,"cannot fetch empty URL");
|
|
|
+ _doH(_arg,-1,_url,"cannot fetch empty URL");
|
|
|
delete this;
|
|
|
return;
|
|
|
}
|
|
@@ -147,17 +154,17 @@ public:
|
|
|
::pipe(curlStdout);
|
|
|
::pipe(curlStderr);
|
|
|
|
|
|
- long pid = (long)vfork();
|
|
|
- if (pid < 0) {
|
|
|
+ _pid = (long)vfork();
|
|
|
+ if (_pid < 0) {
|
|
|
// fork() failed
|
|
|
::close(curlStdout[0]);
|
|
|
::close(curlStdout[1]);
|
|
|
::close(curlStderr[0]);
|
|
|
::close(curlStderr[1]);
|
|
|
- _handler(_arg,-1,_url,false,"unable to fork()");
|
|
|
+ _doH(_arg,-1,_url,"unable to fork()");
|
|
|
delete this;
|
|
|
return;
|
|
|
- } else if (pid > 0) {
|
|
|
+ } else if (_pid > 0) {
|
|
|
// fork() succeeded, in parent process
|
|
|
::close(curlStdout[1]);
|
|
|
::close(curlStderr[1]);
|
|
@@ -189,7 +196,7 @@ public:
|
|
|
} else if (n < 0)
|
|
|
break;
|
|
|
if (_body.length() > CURL_MAX_MESSAGE_LENGTH) {
|
|
|
- ::kill(pid,SIGKILL);
|
|
|
+ ::kill(_pid,SIGKILL);
|
|
|
tooLong = true;
|
|
|
break;
|
|
|
}
|
|
@@ -200,12 +207,12 @@ public:
|
|
|
break;
|
|
|
|
|
|
if (Utils::now() >= timesOutAt) {
|
|
|
- ::kill(pid,SIGKILL);
|
|
|
+ ::kill(_pid,SIGKILL);
|
|
|
timedOut = true;
|
|
|
break;
|
|
|
}
|
|
|
|
|
|
- if (waitpid(pid,&exitCode,WNOHANG) > 0) {
|
|
|
+ if (waitpid(_pid,&exitCode,WNOHANG) > 0) {
|
|
|
for(;;) {
|
|
|
// Drain output...
|
|
|
int n = (int)::read(curlStdout[0],buf,sizeof(buf));
|
|
@@ -219,23 +226,24 @@ public:
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
- pid = 0;
|
|
|
+ _pid = 0;
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- if (pid > 0)
|
|
|
- waitpid(pid,&exitCode,0);
|
|
|
+ if (_pid > 0)
|
|
|
+ waitpid(_pid,&exitCode,0);
|
|
|
+ _pid = 0;
|
|
|
|
|
|
::close(curlStdout[0]);
|
|
|
::close(curlStderr[0]);
|
|
|
|
|
|
if (timedOut)
|
|
|
- _handler(_arg,-1,_url,false,"connection timed out");
|
|
|
+ _doH(_arg,-1,_url,"connection timed out");
|
|
|
else if (tooLong)
|
|
|
- _handler(_arg,-1,_url,false,"response too long");
|
|
|
+ _doH(_arg,-1,_url,"response too long");
|
|
|
else if (exitCode)
|
|
|
- _handler(_arg,-1,_url,false,"connection failed (curl returned non-zero exit code)");
|
|
|
+ _doH(_arg,-1,_url,"connection failed (curl returned non-zero exit code)");
|
|
|
else {
|
|
|
unsigned long idx = 0;
|
|
|
|
|
@@ -254,7 +262,7 @@ public:
|
|
|
headers.back().push_back(c);
|
|
|
}
|
|
|
if (headers.empty()||(!headers.front().length())) {
|
|
|
- _handler(_arg,-1,_url,false,"HTTP response empty");
|
|
|
+ _doH(_arg,-1,_url,"HTTP response empty");
|
|
|
delete this;
|
|
|
return;
|
|
|
}
|
|
@@ -262,24 +270,24 @@ public:
|
|
|
// Parse first line -- HTTP status code and response
|
|
|
size_t scPos = headers.front().find(' ');
|
|
|
if (scPos == std::string::npos) {
|
|
|
- _handler(_arg,-1,_url,false,"invalid HTTP response (no status line)");
|
|
|
+ _doH(_arg,-1,_url,"invalid HTTP response (no status line)");
|
|
|
delete this;
|
|
|
return;
|
|
|
}
|
|
|
++scPos;
|
|
|
unsigned int rcode = Utils::strToUInt(headers.front().substr(scPos,3).c_str());
|
|
|
if ((!rcode)||(rcode > 999)) {
|
|
|
- _handler(_arg,-1,_url,false,"invalid HTTP response (invalid response code)");
|
|
|
+ _doH(_arg,-1,_url,"invalid HTTP response (invalid response code)");
|
|
|
delete this;
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
// Serve up the resulting data to the handler
|
|
|
if (rcode == 200)
|
|
|
- _handler(_arg,rcode,_url,false,_body.substr(idx));
|
|
|
+ _doH(_arg,rcode,_url,_body.substr(idx));
|
|
|
else if ((scPos + 4) < headers.front().length())
|
|
|
- _handler(_arg,rcode,_url,false,headers.front().substr(scPos+4));
|
|
|
- else _handler(_arg,rcode,_url,false,"(no status message from server)");
|
|
|
+ _doH(_arg,rcode,_url,headers.front().substr(scPos+4));
|
|
|
+ else _doH(_arg,rcode,_url,"(no status message from server)");
|
|
|
}
|
|
|
|
|
|
delete this;
|
|
@@ -295,27 +303,41 @@ public:
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ inline void cancel()
|
|
|
+ {
|
|
|
+ {
|
|
|
+ Mutex::Lock _l(_cancelled_m);
|
|
|
+ _cancelled = true;
|
|
|
+ if (_pid > 0)
|
|
|
+ ::kill(_pid,SIGKILL);
|
|
|
+ }
|
|
|
+ Thread::join(_myThread);
|
|
|
+ }
|
|
|
+
|
|
|
+private:
|
|
|
+ inline void _doH(void *arg,int code,const std::string &url,const std::string &body)
|
|
|
+ {
|
|
|
+ Mutex::Lock _l(_cancelled_m);
|
|
|
+ try {
|
|
|
+ if ((!_cancelled)&&(_handler))
|
|
|
+ _handler(arg,code,url,body);
|
|
|
+ } catch ( ... ) {}
|
|
|
+ }
|
|
|
+
|
|
|
const std::string _url;
|
|
|
std::string _body;
|
|
|
std::map<std::string,std::string> _headers;
|
|
|
unsigned int _timeout;
|
|
|
- void (*_handler)(void *,int,const std::string &,bool,const std::string &);
|
|
|
+ void (*_handler)(void *,int,const std::string &,const std::string &);
|
|
|
void *_arg;
|
|
|
+ HttpClient *_parent;
|
|
|
+ long _pid;
|
|
|
+ volatile bool _cancelled;
|
|
|
+ Mutex _cancelled_m;
|
|
|
Thread _myThread;
|
|
|
};
|
|
|
|
|
|
-HttpClient::Request HttpClient::_do(
|
|
|
- const char *method,
|
|
|
- const std::string &url,
|
|
|
- const std::map<std::string,std::string> &headers,
|
|
|
- unsigned int timeout,
|
|
|
- void (*handler)(void *,int,const std::string &,bool,const std::string &),
|
|
|
- void *arg)
|
|
|
-{
|
|
|
- return (HttpClient::Request)(new P_Req(method,url,headers,timeout,handler,arg));
|
|
|
-}
|
|
|
-
|
|
|
-#endif
|
|
|
+#endif // __UNIX_LIKE__
|
|
|
|
|
|
#ifdef __WINDOWS__
|
|
|
|
|
@@ -323,10 +345,10 @@ HttpClient::Request HttpClient::_do(
|
|
|
|
|
|
// Internal private thread class that performs request, notifies handler,
|
|
|
// and then commits suicide by deleting itself.
|
|
|
-class P_Req : NonCopyable
|
|
|
+class HttpClient_Private_Request : NonCopyable
|
|
|
{
|
|
|
public:
|
|
|
- P_Req(const char *method,const std::string &url,const std::map<std::string,std::string> &headers,unsigned int timeout,void (*handler)(void *,int,const std::string &,bool,const std::string &),void *arg) :
|
|
|
+ HttpClient_Private_Request(const char *method,const std::string &url,const std::map<std::string,std::string> &headers,unsigned int timeout,void (*handler)(void *,int,const std::string &,const std::string &),void *arg) :
|
|
|
_url(url),
|
|
|
_headers(headers),
|
|
|
_timeout(timeout),
|
|
@@ -449,22 +471,52 @@ closeAndReturnFromHttp:
|
|
|
std::string _body;
|
|
|
std::map<std::string,std::string> _headers;
|
|
|
unsigned int _timeout;
|
|
|
- void (*_handler)(void *,int,const std::string &,bool,const std::string &);
|
|
|
+ void (*_handler)(void *,int,const std::string &,const std::string &);
|
|
|
void *_arg;
|
|
|
Thread _myThread;
|
|
|
};
|
|
|
|
|
|
+#endif // __WINDOWS__
|
|
|
+
|
|
|
+const std::map<std::string,std::string> HttpClient::NO_HEADERS;
|
|
|
+
|
|
|
+HttpClient::HttpClient()
|
|
|
+{
|
|
|
+}
|
|
|
+
|
|
|
+HttpClient::~HttpClient()
|
|
|
+{
|
|
|
+ std::set<Request> reqs;
|
|
|
+ {
|
|
|
+ Mutex::Lock _l(_requests_m);
|
|
|
+ reqs = _requests;
|
|
|
+ }
|
|
|
+ for(std::set<Request>::iterator r(reqs.begin());r!=reqs.end();++r)
|
|
|
+ this->cancel(*r);
|
|
|
+}
|
|
|
+
|
|
|
+void HttpClient::cancel(HttpClient::Request req)
|
|
|
+{
|
|
|
+ {
|
|
|
+ Mutex::Lock _l(_requests_m);
|
|
|
+ if (_requests.count(req) == 0)
|
|
|
+ return;
|
|
|
+ }
|
|
|
+ ((HttpClient_Private_Request *)req)->cancel();
|
|
|
+}
|
|
|
+
|
|
|
HttpClient::Request HttpClient::_do(
|
|
|
const char *method,
|
|
|
const std::string &url,
|
|
|
const std::map<std::string,std::string> &headers,
|
|
|
unsigned int timeout,
|
|
|
- void (*handler)(void *,int,const std::string &,bool,const std::string &),
|
|
|
+ void (*handler)(void *,int,const std::string &,const std::string &),
|
|
|
void *arg)
|
|
|
{
|
|
|
- return (HttpClient::Request)(new P_Req(method,url,headers,timeout,handler,arg));
|
|
|
+ HttpClient::Request r = (HttpClient::Request)(new HttpClient_Private_Request(this,method,url,headers,timeout,handler,arg));
|
|
|
+ Mutex::Lock _l(_requests_m);
|
|
|
+ _requests.insert(r);
|
|
|
+ return r;
|
|
|
}
|
|
|
|
|
|
-#endif
|
|
|
-
|
|
|
} // namespace ZeroTier
|