diff options
Diffstat (limited to 'service/SoftwareUpdater.cpp')
| -rw-r--r-- | service/SoftwareUpdater.cpp | 250 | 
1 files changed, 39 insertions, 211 deletions
diff --git a/service/SoftwareUpdater.cpp b/service/SoftwareUpdater.cpp index d6542609..cb36dceb 100644 --- a/service/SoftwareUpdater.cpp +++ b/service/SoftwareUpdater.cpp @@ -21,6 +21,20 @@  #include <string.h>  #include <stdint.h> +#ifdef __WINDOWS__ +#include <WinSock2.h> +#include <Windows.h> +#include <ShlObj.h> +#include <netioapi.h> +#include <iphlpapi.h> +#else +#include <sys/types.h> +#include <sys/socket.h> +#include <sys/wait.h> +#include <unistd.h> +#include <ifaddrs.h> +#endif +  #include "SoftwareUpdater.hpp"  #include "../version.h" @@ -42,217 +56,6 @@  namespace ZeroTier { -#if 0 -#ifdef ZT_AUTO_UPDATE -#define ZT_AUTO_UPDATE_MAX_HTTP_RESPONSE_SIZE (1024 * 1024 * 64) -#define ZT_AUTO_UPDATE_CHECK_PERIOD 21600000 -class BackgroundSoftwareUpdateChecker -{ -public: -	bool isValidSigningIdentity(const Identity &id) -	{ -		return ( -			/* 0001 - 0004 : obsolete, used in old versions */ -		  /* 0005 */   (id == Identity("ba57ea350e:0:9d4be6d7f86c5660d5ee1951a3d759aa6e12a84fc0c0b74639500f1dbc1a8c566622e7d1c531967ebceb1e9d1761342f88324a8ba520c93c35f92f35080fa23f")) -		  /* 0006 */ ||(id == Identity("5067b21b83:0:8af477730f5055c48135b84bed6720a35bca4c0e34be4060a4c636288b1ec22217eb22709d610c66ed464c643130c51411bbb0294eef12fbe8ecc1a1e2c63a7a")) -		  /* 0007 */ ||(id == Identity("4f5e97a8f1:0:57880d056d7baeb04bbc057d6f16e6cb41388570e87f01492fce882485f65a798648595610a3ad49885604e7fb1db2dd3c2c534b75e42c3c0b110ad07b4bb138")) -		  /* 0008 */ ||(id == Identity("580bbb8e15:0:ad5ef31155bebc6bc413991992387e083fed26d699997ef76e7c947781edd47d1997161fa56ba337b1a2b44b129fd7c7197ce5185382f06011bc88d1363b4ddd")) -		); -	} - -	void doUpdateCheck() -	{ -		std::string url(OneService::autoUpdateUrl()); -		if ((url.length() <= 7)||(url.substr(0,7) != "http://")) -			return; - -		std::string httpHost; -		std::string httpPath; -		{ -			std::size_t slashIdx = url.substr(7).find_first_of('/'); -			if (slashIdx == std::string::npos) { -				httpHost = url.substr(7); -				httpPath = "/"; -			} else { -				httpHost = url.substr(7,slashIdx); -				httpPath = url.substr(slashIdx + 7); -			} -		} -		if (httpHost.length() == 0) -			return; - -		std::vector<InetAddress> ips(OSUtils::resolve(httpHost.c_str())); -		for(std::vector<InetAddress>::iterator ip(ips.begin());ip!=ips.end();++ip) { -			if (!ip->port()) -				ip->setPort(80); -			std::string nfoPath = httpPath + "LATEST.nfo"; -			std::map<std::string,std::string> requestHeaders,responseHeaders; -			std::string body; -			requestHeaders["Host"] = httpHost; -			unsigned int scode = Http::GET(ZT_AUTO_UPDATE_MAX_HTTP_RESPONSE_SIZE,60000,reinterpret_cast<const struct sockaddr *>(&(*ip)),nfoPath.c_str(),requestHeaders,responseHeaders,body); -			//fprintf(stderr,"UPDATE %s %s %u %lu\n",ip->toString().c_str(),nfoPath.c_str(),scode,body.length()); -			if ((scode == 200)&&(body.length() > 0)) { -				/* NFO fields: -				 * -				 * file=<filename> -				 * signedBy=<signing identity> -				 * ed25519=<ed25519 ECC signature of archive in hex> -				 * vMajor=<major version> -				 * vMinor=<minor version> -				 * vRevision=<revision> */ -				Dictionary<4096> nfo(body.c_str()); -				char tmp[2048]; - -				if (nfo.get("vMajor",tmp,sizeof(tmp)) <= 0) return; -				const unsigned int vMajor = Utils::strToUInt(tmp); -				if (nfo.get("vMinor",tmp,sizeof(tmp)) <= 0) return; -				const unsigned int vMinor = Utils::strToUInt(tmp); -				if (nfo.get("vRevision",tmp,sizeof(tmp)) <= 0) return; -				const unsigned int vRevision = Utils::strToUInt(tmp); -				if (Utils::compareVersion(vMajor,vMinor,vRevision,ZEROTIER_ONE_VERSION_MAJOR,ZEROTIER_ONE_VERSION_MINOR,ZEROTIER_ONE_VERSION_REVISION) <= 0) { -					//fprintf(stderr,"UPDATE %u.%u.%u is not newer than our version\n",vMajor,vMinor,vRevision); -					return; -				} - -				if (nfo.get("signedBy",tmp,sizeof(tmp)) <= 0) return; -				Identity signedBy; -				if ((!signedBy.fromString(tmp))||(!isValidSigningIdentity(signedBy))) { -					//fprintf(stderr,"UPDATE invalid signedBy or not authorized signing identity.\n"); -					return; -				} - -				if (nfo.get("file",tmp,sizeof(tmp)) <= 0) return; -				std::string filePath(tmp); -				if ((!filePath.length())||(filePath.find("..") != std::string::npos)) -					return; -				filePath = httpPath + filePath; - -				std::string fileData; -				if (Http::GET(ZT_AUTO_UPDATE_MAX_HTTP_RESPONSE_SIZE,60000,reinterpret_cast<const struct sockaddr *>(&(*ip)),filePath.c_str(),requestHeaders,responseHeaders,fileData) != 200) { -					//fprintf(stderr,"UPDATE GET %s failed\n",filePath.c_str()); -					return; -				} - -				if (nfo.get("ed25519",tmp,sizeof(tmp)) <= 0) return; -				std::string ed25519(Utils::unhex(tmp)); -				if ((ed25519.length() == 0)||(!signedBy.verify(fileData.data(),(unsigned int)fileData.length(),ed25519.data(),(unsigned int)ed25519.length()))) { -					//fprintf(stderr,"UPDATE %s failed signature check!\n",filePath.c_str()); -					return; -				} - -				/* --------------------------------------------------------------- */ -				/* We made it! Begin OS-specific installation code. */ - -#ifdef __APPLE__ -				/* OSX version is in the form of a MacOSX .pkg file, so we will -				 * launch installer (normally in /usr/sbin) to install it. It will -				 * then turn around and shut down the service, update files, and -				 * relaunch. */ -				{ -					char bashp[128],pkgp[128]; -					Utils::snprintf(bashp,sizeof(bashp),"/tmp/ZeroTierOne-update-%u.%u.%u.sh",vMajor,vMinor,vRevision); -					Utils::snprintf(pkgp,sizeof(pkgp),"/tmp/ZeroTierOne-update-%u.%u.%u.pkg",vMajor,vMinor,vRevision); -					FILE *pkg = fopen(pkgp,"w"); -					if ((!pkg)||(fwrite(fileData.data(),fileData.length(),1,pkg) != 1)) { -						fclose(pkg); -						unlink(bashp); -						unlink(pkgp); -						fprintf(stderr,"UPDATE error writing %s\n",pkgp); -						return; -					} -					fclose(pkg); -					FILE *bash = fopen(bashp,"w"); -					if (!bash) { -						fclose(pkg); -						unlink(bashp); -						unlink(pkgp); -						fprintf(stderr,"UPDATE error writing %s\n",bashp); -						return; -					} -					fprintf(bash, -						"#!/bin/bash\n" -						"export PATH=/bin:/usr/bin:/usr/sbin:/sbin:/usr/local/bin:/usr/local/sbin\n" -						"sleep 1\n" -						"installer -pkg \"%s\" -target /\n" -						"sleep 1\n" -						"rm -f \"%s\" \"%s\"\n" -						"exit 0\n", -						pkgp, -						pkgp, -						bashp); -					fclose(bash); -					long pid = (long)vfork(); -					if (pid == 0) { -						setsid(); // detach from parent so that shell isn't killed when parent is killed -						signal(SIGHUP,SIG_IGN); -						signal(SIGTERM,SIG_IGN); -						signal(SIGQUIT,SIG_IGN); -						execl("/bin/bash","/bin/bash",bashp,(char *)0); -						exit(0); -					} -				} -#endif // __APPLE__ - -#ifdef __WINDOWS__ -				/* Windows version comes in the form of .MSI package that -				 * takes care of everything. */ -				{ -					char tempp[512],batp[512],msip[512],cmdline[512]; -					if (GetTempPathA(sizeof(tempp),tempp) <= 0) -						return; -					CreateDirectoryA(tempp,(LPSECURITY_ATTRIBUTES)0); -					Utils::snprintf(batp,sizeof(batp),"%s\\ZeroTierOne-update-%u.%u.%u.bat",tempp,vMajor,vMinor,vRevision); -					Utils::snprintf(msip,sizeof(msip),"%s\\ZeroTierOne-update-%u.%u.%u.msi",tempp,vMajor,vMinor,vRevision); -					FILE *msi = fopen(msip,"wb"); -					if ((!msi)||(fwrite(fileData.data(),(size_t)fileData.length(),1,msi) != 1)) { -						fclose(msi); -						return; -					} -					fclose(msi); -					FILE *bat = fopen(batp,"wb"); -					if (!bat) -						return; -					fprintf(bat, -						"TIMEOUT.EXE /T 1 /NOBREAK\r\n" -						"NET.EXE STOP \"ZeroTierOneService\"\r\n" -						"TIMEOUT.EXE /T 1 /NOBREAK\r\n" -						"MSIEXEC.EXE /i \"%s\" /qn\r\n" -						"TIMEOUT.EXE /T 1 /NOBREAK\r\n" -						"NET.EXE START \"ZeroTierOneService\"\r\n" -						"DEL \"%s\"\r\n" -						"DEL \"%s\"\r\n", -						msip, -						msip, -						batp); -					fclose(bat); -					STARTUPINFOA si; -					PROCESS_INFORMATION pi; -					memset(&si,0,sizeof(si)); -					memset(&pi,0,sizeof(pi)); -					Utils::snprintf(cmdline,sizeof(cmdline),"CMD.EXE /c \"%s\"",batp); -					CreateProcessA(NULL,cmdline,NULL,NULL,FALSE,CREATE_NO_WINDOW|CREATE_NEW_PROCESS_GROUP,NULL,NULL,&si,&pi); -				} -#endif // __WINDOWS__ - -				/* --------------------------------------------------------------- */ - -				return; -			} // else try to fetch from next IP address -		} -	} - -	void threadMain() -		throw() -	{ -		try { -			this->doUpdateCheck(); -		} catch ( ... ) {} -	} -}; -static BackgroundSoftwareUpdateChecker backgroundSoftwareUpdateChecker; -#endif // ZT_AUTO_UPDATE -#endif -  SoftwareUpdater::SoftwareUpdater(Node &node,const std::string &homePath) :  	_node(node),  	_lastCheckTime(0), @@ -531,6 +334,31 @@ bool SoftwareUpdater::check(const uint64_t now)  void SoftwareUpdater::apply()  { +	std::string updatePath(_homePath + ZT_PATH_SEPARATOR_S ZT_SOFTWARE_UPDATE_BIN_FILENAME); +	if ((_latestMeta.is_object())&&(_latestValid)&&(OSUtils::fileExists(updatePath.c_str(),false))) { +#ifdef __WINDOWS__ +		std::string cmdArgs = " "; +		cmdArgs.append(OSUtils::jsonString(_latestMeta[ZT_SOFTWARE_UPDATE_JSON_UPDATE_EXEC_ARGS],"")); +		if (cmdArgs.length() == 1) cmdArgs = std::string(); +		STARTUPINFOA si; +		PROCESS_INFORMATION pi; +		memset(&si,0,sizeof(si)); +		memset(&pi,0,sizeof(pi)); +		CreateProcessA(NULL,(updatePath + cmdArgs).c_str(),NULL,NULL,FALSE,CREATE_NO_WINDOW|CREATE_NEW_PROCESS_GROUP,NULL,NULL,&si,&pi); +#else +		char *argv[256]; +		unsigned long ac = 0; +		argv[ac++] = const_cast<char *>(updatePath.c_str()); +		std::vector<std::string> argsSplit(OSUtils::split(OSUtils::jsonString(_latestMeta[ZT_SOFTWARE_UPDATE_JSON_UPDATE_EXEC_ARGS],"").c_str()," ","\\","\"")); +		for(std::vector<std::string>::iterator a(argsSplit.begin());a!=argsSplit.end();++a) { +			argv[ac] = const_cast<char *>(a->c_str()); +			if (++ac == 255) break; +		} +		argv[ac] = (char *)0; +		chmod(updatePath.c_str(),0700); +		execv(updatePath.c_str(),argv); +#endif +	}  }  } // namespace ZeroTier  | 
