ZeroTierOne/installer.cpp
2013-11-08 17:37:47 -05:00

257 lines
7.0 KiB
C++

/*
* ZeroTier One - Global Peer to Peer Ethernet
* Copyright (C) 2012-2013 ZeroTier Networks LLC
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
* --
*
* ZeroTier may be used and distributed under the terms of the GPLv3, which
* are available at: http://www.gnu.org/licenses/gpl-3.0.html
*
* If you would like to embed ZeroTier into a commercial application or
* redistribute it in a modified binary form, please contact ZeroTier Networks
* LLC. Start here: http://www.zerotier.com/
*/
/*
* This can be run to install ZT1 for the first time or to update it. It
* carries all payloads internally as LZ4 compressed blobs.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include "node/Constants.hpp"
#include "version.h"
#ifdef __WINDOWS__
#include <Windows.h>
#include <tchar.h>
#include <wchar.h>
#else
#include <unistd.h>
#include <pwd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <signal.h>
#endif
#include "ext/lz4/lz4.h"
#include "ext/lz4/lz4hc.h"
// Include Lz4 comrpessed blobs -----------------------------------------------
// zerotier-one binary (or zerotier-one.exe for Windows)
#include "installer-build/zerotier_one.h"
// Unix uninstall script, installed in home for user to remove
#ifdef __UNIX_LIKE__
#include "installer-build/uninstall_sh.h"
#endif
// Linux init.d script
#ifdef __LINUX__
#include "installer-build/linux__init_d__zerotier_one.h"
#endif
// Apple Tap device driver
#ifdef __APPLE__
#include "installer-build/tap_mac__Info_plist.h"
#include "installer-build/tap_mac__tap.h"
#endif
// Windows Tap device drivers for x86 and x64 (installer will be x86)
#ifdef __WINDOWS__
#include "installer-build/tap_windows__x64__ztTap100_sys.h"
#include "installer-build/tap_windows__x64__ztTap100_inf.h"
#include "installer-build/tap_windows__x86__ztTap100_sys.h"
#include "installer-build/tap_windows__x86__ztTap100_inf.h"
#include "installer-build/tap_windows__devcon32_exe.h"
#include "installer-build/tap_windows__devcon64_exe.h"
#endif
// ----------------------------------------------------------------------------
static unsigned char *_unlz4(const void *lz4,int decompressedLen)
{
unsigned char *buf = new unsigned char[decompressedLen];
if (LZ4_decompress_fast((const char *)lz4,(char *)buf,decompressedLen) <= 0) {
delete [] buf;
return (unsigned char *)0;
}
return buf;
}
static bool _putBlob(const void *lz4,int decompressedLen,const char *path)
{
unsigned char *data = _unlz4(lz4,decompressedLen);
if (!data)
return false;
#ifdef __WINDOWS__
DeleteFileA(path);
#else
unlink(path);
#endif
FILE *f = fopen(path,"w");
if (!f) {
delete [] data;
return false;
}
if (fwrite(data,decompressedLen,1,f) != 1) {
fclose(f);
delete [] data;
#ifdef __WINDOWS__
DeleteFileA(path);
#else
unlink(path);
#endif
return false;
}
fclose(f);
delete [] data;
return true;
}
#define putBlob(name,path) _putBlob(name,name##_UNCOMPRESSED_LEN,path)
// ----------------------------------------------------------------------------
#ifdef __WINDOWS__
int _tmain(int argc, _TCHAR* argv[])
#else
int main(int argc,char **argv)
#endif
{
#ifdef __UNIX_LIKE__ // -------------------------------------------------------
char buf[4096];
if (getuid() != 0) {
printf("! ZeroTier One installer must be run as root.\n");
return 2;
}
printf("# ZeroTier One installer/updater starting...\n");
const char *zthome;
#ifdef __APPLE__
mkdir("/Library/Application Support/ZeroTier",0755);
chmod("/Library/Application Support/ZeroTier",0755);
chown("/Library/Application Support/ZeroTier",0,0);
printf("mkdir /Library/Application Support/ZeroTier\n");
mkdir(zthome = "/Library/Application Support/ZeroTier/One",0755);
#else
mkdir("/var/lib",0755);
printf("mkdir /var/lib\n");
mkdir(zthome = "/var/lib/zerotier-one",0755);
#endif
chmod(zthome,0755);
chown(zthome,0,0);
printf("mkdir %s\n",zthome);
sprintf(buf,"%s/zerotier-one",zthome);
if (!putBlob(zerotier_one,buf)) {
printf("! unable to write %s\n",buf);
return 1;
}
chmod(buf,0755);
chown(buf,0,0);
printf("write %s\n",buf);
unlink("/usr/bin/zerotier-cli");
symlink(buf,"/usr/bin/zerotier-cli");
printf("link %s /usr/bin/zerotier-cli\n",buf);
sprintf(buf,"%s/uninstall.sh",zthome);
if (!putBlob(uninstall_sh,buf)) {
printf("! unable to write %s\n",buf);
return 1;
}
chmod(buf,0755);
chown(buf,0,0);
printf("write %s\n",buf);
#ifdef __APPLE__
sprintf(buf,"%s/tap.kext");
mkdir(buf,0755);
chmod(buf,0755);
chown(buf,0,0);
printf("mkdir %s\n",buf);
sprintf(buf,"%s/tap.kext/Contents");
mkdir(buf,0755);
chmod(buf,0755);
chown(buf,0,0);
printf("mkdir %s\n",buf);
sprintf(buf,"%s/tap.kext/Contents/MacOS");
mkdir(buf,0755);
chmod(buf,0755);
chown(buf,0,0);
printf("mkdir %s\n",buf);
sprintf(buf,"%s/tap.kext/Contents/Info.plist",zthome);
if (!putBlob(tap_mac__Info_plist,buf)) {
printf("! unable to write %s\n",buf);
return 1;
}
chmod(buf,0644);
chown(buf,0,0);
printf("write %s\n",buf);
sprintf(buf,"%s/tap.kext/Contents/MacOS/tap",zthome);
if (!putBlob(tap_mac__tap,buf)) {
printf("! unable to write %s\n",buf);
return 1;
}
chmod(buf,0755);
chown(buf,0,0);
printf("write %s\n",buf);
#endif
#ifdef __LINUX__
sprintf(buf,"/etc/init.d/zerotier-one");
if (!putBlob(linux__init_d__zerotier_one,buf)) {
printf("! unable to write %s\n",buf);
return 1;
}
chown(buf,0,0);
chmod(buf,0755);
printf("write %s\n",buf);
symlink("/etc/init.d/zerotier-one","/etc/rc0.d/K89zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc0.d/K89zerotier-one\n");
symlink("/etc/init.d/zerotier-one","/etc/rc2.d/S11zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc2.d/S11zerotier-one\n");
symlink("/etc/init.d/zerotier-one","/etc/rc3.d/S11zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc3.d/S11zerotier-one\n");
symlink("/etc/init.d/zerotier-one","/etc/rc4.d/S11zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc4.d/S11zerotier-one\n");
symlink("/etc/init.d/zerotier-one","/etc/rc5.d/S11zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc5.d/S11zerotier-one\n");
symlink("/etc/init.d/zerotier-one","/etc/rc6.d/S11zerotier-one");
printf("link /etc/init.d/zerotier-one /etc/rc6.d/S11zerotier-one\n");
#endif
printf("# Done!\n");
#endif // __UNIX_LIKE__ -------------------------------------------------------
#ifdef __WINDOWS__ // ---------------------------------------------------------
#endif // __WINDOWS__ ---------------------------------------------------------
return 0;
}