2014-01-10 23:27:44 +00:00
/*
* 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/
*/
2014-01-04 06:14:30 +00:00
# include "installdialog.h"
# include "mainwindow.h"
# include "ui_installdialog.h"
2014-01-06 04:59:41 +00:00
# include "../node/Constants.hpp"
2014-01-04 06:14:30 +00:00
# include "../node/Defaults.hpp"
2014-01-06 00:24:12 +00:00
# include "../node/SoftwareUpdater.hpp"
2014-01-04 06:14:30 +00:00
2014-01-06 04:59:41 +00:00
# ifdef __UNIX_LIKE__
# include <stdio.h>
# include <stdlib.h>
# include <unistd.h>
# include <sys/stat.h>
2014-01-16 01:00:53 +00:00
# include <sys/types.h>
2014-01-06 04:59:41 +00:00
# include <fcntl.h>
# endif
2014-01-16 01:00:53 +00:00
# ifdef __APPLE__
# include "mac_doprivileged.h"
# endif
2014-01-04 06:14:30 +00:00
# include <QMainWindow>
# include <QMessageBox>
# include <QByteArray>
# include <QSslSocket>
2014-01-06 04:59:41 +00:00
# include <QFile>
# include <QDir>
# include <QProcess>
2014-01-04 06:14:30 +00:00
InstallDialog : : InstallDialog ( QWidget * parent ) :
QDialog ( parent ) ,
ui ( new Ui : : InstallDialog ) ,
2014-01-06 00:24:12 +00:00
nam ( new QNetworkAccessManager ( this ) ) ,
phase ( FETCHING_NFO )
2014-01-04 06:14:30 +00:00
{
ui - > setupUi ( this ) ;
QObject : : connect ( nam , SIGNAL ( finished ( QNetworkReply * ) ) , this , SLOT ( on_networkReply ( QNetworkReply * ) ) ) ;
const char * nfoUrl = ZeroTier : : ZT_DEFAULTS . updateLatestNfoURL . c_str ( ) ;
if ( ! * nfoUrl ) {
QMessageBox : : critical ( this , " Download Failed " , " Download failed: internal error: no update URL configured in build! " , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
return ;
}
QNetworkReply * reply = nam - > get ( QNetworkRequest ( QUrl ( nfoUrl ) ) ) ;
QObject : : connect ( reply , SIGNAL ( downloadProgress ( qint64 , qint64 ) ) , this , SLOT ( on_downloadProgress ( qint64 , qint64 ) ) ) ;
}
InstallDialog : : ~ InstallDialog ( )
{
delete ui ;
}
void InstallDialog : : on_networkReply ( QNetworkReply * reply )
{
reply - > deleteLater ( ) ;
if ( reply - > error ( ) ! = QNetworkReply : : NoError ) {
2014-01-06 04:59:41 +00:00
QMessageBox : : critical ( this , " Download Failed " , QString ( " Download failed: " ) + reply - > errorString ( ) + " \n \n Are you connected to the Internet? " , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
2014-01-04 06:14:30 +00:00
QApplication : : exit ( 1 ) ;
} else {
if ( reply - > attribute ( QNetworkRequest : : HttpStatusCodeAttribute ) = = 200 ) {
QByteArray installerData ( reply - > readAll ( ) ) ;
2014-01-06 00:24:12 +00:00
switch ( phase ) {
case FETCHING_NFO : {
unsigned int vMajor = 0 , vMinor = 0 , vRevision = 0 ;
installerData . append ( ( char ) 0 ) ;
const char * err = ZeroTier : : SoftwareUpdater : : parseNfo ( installerData . data ( ) , vMajor , vMinor , vRevision , signedBy , signature , url ) ;
if ( err ) {
QMessageBox : : critical ( this , " Download Failed " , " Download failed: there is a problem with the software update web site. \n Try agian later. (invalid .nfo file) " , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
return ;
}
phase = FETCHING_INSTALLER ;
reply = nam - > get ( QNetworkRequest ( QUrl ( url . c_str ( ) ) ) ) ;
QObject : : connect ( reply , SIGNAL ( downloadProgress ( qint64 , qint64 ) ) , this , SLOT ( on_downloadProgress ( qint64 , qint64 ) ) ) ;
} break ;
case FETCHING_INSTALLER : {
if ( ! ZeroTier : : SoftwareUpdater : : validateUpdate ( installerData . data ( ) , installerData . length ( ) , signedBy , signature ) ) {
2014-01-16 01:00:53 +00:00
QMessageBox : : critical ( this , " Download Failed " , " Download failed: there is a problem with the software update web site. Try agian later. (downloaded data failed signature check) " , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
2014-01-06 00:24:12 +00:00
QApplication : : exit ( 1 ) ;
return ;
}
2014-01-06 04:59:41 +00:00
# ifdef __APPLE__
2014-01-16 01:00:53 +00:00
{
std : : string homePath ( QDir : : homePath ( ) . toStdString ( ) ) ;
QString zt1Caches ( QDir : : homePath ( ) + " /Library/Caches/ZeroTier/One " ) ;
QDir : : root ( ) . mkpath ( zt1Caches ) ;
std : : string instPath ( ( zt1Caches + " /ZeroTierOneInstaller " ) . toStdString ( ) ) ;
std : : string tmpPath ( ( zt1Caches + " /inst.sh " ) . toStdString ( ) ) ;
int outfd = : : open ( instPath . c_str ( ) , O_CREAT | O_TRUNC | O_WRONLY , 0755 ) ;
if ( outfd < = 0 ) {
QMessageBox : : critical ( this , " Download Failed " , QString ( " Installation failed: unable to write to " ) + instPath . c_str ( ) , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
return ;
}
if ( : : write ( outfd , installerData . data ( ) , installerData . length ( ) ) ! = installerData . length ( ) ) {
QMessageBox : : critical ( this , " Installation Failed " , QString ( " Installation failed: unable to write to " ) + instPath . c_str ( ) , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
return ;
}
: : close ( outfd ) ;
chmod ( instPath . c_str ( ) , 0755 ) ;
FILE * scr = fopen ( tmpPath . c_str ( ) , " w " ) ;
if ( ! scr ) {
QMessageBox : : critical ( this , " Installation Failed " , " Cannot write script to temporary Library/Caches/ZeroTier/One folder. " , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
return ;
}
fprintf ( scr , " #!/bin/bash \n " ) ;
fprintf ( scr , " export PATH= \" /bin:/usr/bin:/sbin:/usr/sbin \" \n " ) ;
fprintf ( scr , " '%s' \n " , instPath . c_str ( ) ) ;
fprintf ( scr , " if [ -f '/Library/Application Support/ZeroTier/One/authtoken.secret' ]; then \n " ) ;
fprintf ( scr , " cp -f '/Library/Application Support/ZeroTier/One/authtoken.secret' '%s/Library/Application Support/ZeroTier/One/authtoken.secret' \n " , homePath . c_str ( ) ) ;
fprintf ( scr , " chown %d '%s/Library/Application Support/ZeroTier/One/authtoken.secret' \n " , ( int ) getuid ( ) , homePath . c_str ( ) ) ;
fprintf ( scr , " chgrp %d '%s/Library/Application Support/ZeroTier/One/authtoken.secret' \n " , ( int ) getgid ( ) , homePath . c_str ( ) ) ;
fprintf ( scr , " chmod 0600 '%s/Library/Application Support/ZeroTier/One/authtoken.secret' \n " , homePath . c_str ( ) ) ;
fprintf ( scr , " fi \n " ) ;
fprintf ( scr , " exit 0 \n " ) ;
fclose ( scr ) ;
chmod ( tmpPath . c_str ( ) , 0755 ) ;
macExecutePrivilegedShellCommand ( ( std : : string ( " ' " ) + tmpPath + " ' >>/dev/null 2>&1 " ) . c_str ( ) ) ;
unlink ( tmpPath . c_str ( ) ) ;
unlink ( instPath . c_str ( ) ) ;
2014-01-06 04:59:41 +00:00
return ;
}
# endif
2014-01-06 00:24:12 +00:00
} break ;
}
ui - > progressBar - > setMinimum ( 0 ) ;
ui - > progressBar - > setMaximum ( 100 ) ;
ui - > progressBar - > setValue ( 0 ) ;
2014-01-04 06:14:30 +00:00
} else {
QMessageBox : : critical ( this , " Download Failed " , QString ( " Download failed: HTTP status code " ) + reply - > attribute ( QNetworkRequest : : HttpStatusCodeAttribute ) . toString ( ) , QMessageBox : : Ok , QMessageBox : : NoButton ) ;
QApplication : : exit ( 1 ) ;
}
}
}
void InstallDialog : : on_InstallDialog_rejected ( )
{
QApplication : : exit ( ) ;
}
void InstallDialog : : on_cancelButton_clicked ( )
{
QApplication : : exit ( ) ;
}
void InstallDialog : : on_downloadProgress ( qint64 bytesReceived , qint64 bytesTotal )
{
if ( bytesTotal < = 0 ) {
ui - > progressBar - > setValue ( 0 ) ;
ui - > progressBar - > setMinimum ( 0 ) ;
ui - > progressBar - > setMaximum ( 0 ) ;
} else {
double pct = ( ( double ) bytesReceived / ( double ) bytesTotal ) * 100.0 ;
if ( pct > 100.0 )
pct = 100.0 ;
ui - > progressBar - > setMinimum ( 0 ) ;
ui - > progressBar - > setMaximum ( 100 ) ;
ui - > progressBar - > setValue ( ( int ) pct ) ;
}
}