2018-05-01 23:45:06 +00:00
// +build linux
package main
import (
"fmt"
"os"
"path/filepath"
2020-07-14 18:21:44 +00:00
"github.com/cloudflare/cloudflared/cmd/cloudflared/cliutil"
2020-11-23 21:36:16 +00:00
"github.com/cloudflare/cloudflared/cmd/cloudflared/tunnel"
2021-03-08 16:46:23 +00:00
"github.com/cloudflare/cloudflared/config"
2020-04-29 20:51:32 +00:00
"github.com/cloudflare/cloudflared/logger"
2020-11-25 06:55:13 +00:00
"github.com/rs/zerolog"
"github.com/urfave/cli/v2"
2018-05-01 23:45:06 +00:00
)
2021-01-25 21:51:58 +00:00
func runApp ( app * cli . App , graceShutdownC chan struct { } ) {
2018-05-01 23:45:06 +00:00
app . Commands = append ( app . Commands , & cli . Command {
Name : "service" ,
Usage : "Manages the Argo Tunnel system service" ,
Subcommands : [ ] * cli . Command {
2020-11-25 06:55:13 +00:00
{
2018-05-01 23:45:06 +00:00
Name : "install" ,
Usage : "Install Argo Tunnel as a system service" ,
2020-07-14 18:21:44 +00:00
Action : cliutil . ErrorHandler ( installLinuxService ) ,
2020-10-19 12:30:25 +00:00
Flags : [ ] cli . Flag {
& cli . BoolFlag {
Name : "legacy" ,
Usage : "Generate service file for non-named tunnels" ,
} ,
} ,
2018-05-01 23:45:06 +00:00
} ,
2020-11-25 06:55:13 +00:00
{
2018-05-01 23:45:06 +00:00
Name : "uninstall" ,
Usage : "Uninstall the Argo Tunnel service" ,
2020-07-14 18:21:44 +00:00
Action : cliutil . ErrorHandler ( uninstallLinuxService ) ,
2018-05-01 23:45:06 +00:00
} ,
} ,
} )
app . Run ( os . Args )
}
2018-10-08 19:20:28 +00:00
// The directory and files that are used by the service.
// These are hard-coded in the templates below.
const (
serviceConfigDir = "/etc/cloudflared"
serviceConfigFile = "config.yml"
serviceCredentialFile = "cert.pem"
2020-10-19 12:30:25 +00:00
serviceConfigPath = serviceConfigDir + "/" + serviceConfigFile
2018-10-08 19:20:28 +00:00
)
2018-05-01 23:45:06 +00:00
var systemdTemplates = [ ] ServiceTemplate {
{
Path : "/etc/systemd/system/cloudflared.service" ,
Content : ` [ Unit ]
Description = Argo Tunnel
After = network . target
[ Service ]
TimeoutStartSec = 0
Type = notify
2020-10-19 12:30:25 +00:00
ExecStart = { { . Path } } -- config / etc / cloudflared / config . yml -- no - autoupdate { { range . ExtraArgs } } { { . } } { { end } }
2018-05-01 23:45:06 +00:00
Restart = on - failure
RestartSec = 5 s
[ Install ]
WantedBy = multi - user . target
` ,
} ,
{
Path : "/etc/systemd/system/cloudflared-update.service" ,
Content : ` [ Unit ]
Description = Update Argo Tunnel
After = network . target
[ Service ]
2020-09-17 21:52:21 +00:00
ExecStart = / bin / bash - c ' { { . Path } } update ; code = $ ? ; if [ $ code - eq 11 ] ; then systemctl restart cloudflared ; exit 0 ; fi ; exit $ code '
2018-05-01 23:45:06 +00:00
` ,
} ,
{
Path : "/etc/systemd/system/cloudflared-update.timer" ,
Content : ` [ Unit ]
Description = Update Argo Tunnel
[ Timer ]
2019-12-19 16:53:06 +00:00
OnCalendar = daily
2018-05-01 23:45:06 +00:00
[ Install ]
WantedBy = timers . target
` ,
} ,
}
var sysvTemplate = ServiceTemplate {
Path : "/etc/init.d/cloudflared" ,
FileMode : 0755 ,
2019-01-30 21:10:47 +00:00
Content : ` # ! / bin / sh
# For RedHat and cousins :
2018-05-01 23:45:06 +00:00
# chkconfig : 2345 99 01
# description : Argo Tunnel agent
# processname : { { . Path } }
# # # BEGIN INIT INFO
# Provides : { { . Path } }
# Required - Start :
# Required - Stop :
# Default - Start : 2 3 4 5
# Default - Stop : 0 1 6
# Short - Description : Argo Tunnel
# Description : Argo Tunnel agent
# # # END INIT INFO
name = $ ( basename $ ( readlink - f $ 0 ) )
2020-10-19 12:30:25 +00:00
cmd = "{{.Path}} --config /etc/cloudflared/config.yml --pidfile /var/run/$name.pid --autoupdate-freq 24h0m0s{{ range .ExtraArgs }} {{ . }}{{ end }}"
2018-05-01 23:45:06 +00:00
pid_file = "/var/run/$name.pid"
stdout_log = "/var/log/$name.log"
stderr_log = "/var/log/$name.err"
[ - e / etc / sysconfig / $ name ] && . / etc / sysconfig / $ name
get_pid ( ) {
cat "$pid_file"
}
is_running ( ) {
[ - f "$pid_file" ] && ps $ ( get_pid ) > / dev / null 2 > & 1
}
case "$1" in
start )
if is_running ; then
echo "Already started"
else
echo "Starting $name"
$ cmd >> "$stdout_log" 2 >> "$stderr_log" &
echo $ ! > "$pid_file"
fi
; ;
stop )
if is_running ; then
echo - n "Stopping $name.."
kill $ ( get_pid )
for i in { 1. .10 }
do
if ! is_running ; then
break
fi
echo - n "."
sleep 1
done
echo
if is_running ; then
echo "Not stopped; may still be shutting down or shutdown may have failed"
exit 1
else
echo "Stopped"
if [ - f "$pid_file" ] ; then
rm "$pid_file"
fi
fi
else
echo "Not running"
fi
; ;
restart )
$ 0 stop
if is_running ; then
echo "Unable to stop, will not attempt to start"
exit 1
fi
$ 0 start
; ;
status )
if is_running ; then
echo "Running"
else
echo "Stopped"
exit 1
fi
; ;
* )
echo "Usage: $0 {start|stop|restart|status}"
exit 1
; ;
esac
exit 0
` ,
}
func isSystemd ( ) bool {
if _ , err := os . Stat ( "/run/systemd/system" ) ; err == nil {
return true
}
return false
}
2020-11-25 06:55:13 +00:00
func copyUserConfiguration ( userConfigDir , userConfigFile , userCredentialFile string , log * zerolog . Logger ) error {
2018-10-08 19:20:28 +00:00
srcCredentialPath := filepath . Join ( userConfigDir , userCredentialFile )
destCredentialPath := filepath . Join ( serviceConfigDir , serviceCredentialFile )
2020-06-05 19:52:30 +00:00
if srcCredentialPath != destCredentialPath {
if err := copyCredential ( srcCredentialPath , destCredentialPath ) ; err != nil {
return err
}
2018-10-08 19:20:28 +00:00
}
2020-06-05 19:52:30 +00:00
2018-10-08 19:20:28 +00:00
srcConfigPath := filepath . Join ( userConfigDir , userConfigFile )
destConfigPath := filepath . Join ( serviceConfigDir , serviceConfigFile )
2020-06-05 19:52:30 +00:00
if srcConfigPath != destConfigPath {
if err := copyConfig ( srcConfigPath , destConfigPath ) ; err != nil {
return err
}
2020-11-25 06:55:13 +00:00
log . Info ( ) . Msgf ( "Copied %s to %s" , srcConfigPath , destConfigPath )
2018-10-08 19:20:28 +00:00
}
2020-06-05 19:52:30 +00:00
2018-10-08 19:20:28 +00:00
return nil
}
2018-05-01 23:45:06 +00:00
func installLinuxService ( c * cli . Context ) error {
2020-11-25 06:55:13 +00:00
log := logger . CreateLoggerFromContext ( c , logger . EnableTerminalLog )
2020-04-29 20:51:32 +00:00
2018-05-01 23:45:06 +00:00
etPath , err := os . Executable ( )
if err != nil {
return fmt . Errorf ( "error determining executable path: %v" , err )
}
2020-10-19 12:30:25 +00:00
templateArgs := ServiceTemplateArgs {
Path : etPath ,
}
2018-05-01 23:45:06 +00:00
2020-10-19 12:30:25 +00:00
if err := ensureConfigDirExists ( serviceConfigDir ) ; err != nil {
2018-05-01 23:45:06 +00:00
return err
}
2020-10-19 12:30:25 +00:00
if c . Bool ( "legacy" ) {
userConfigDir := filepath . Dir ( c . String ( "config" ) )
userConfigFile := filepath . Base ( c . String ( "config" ) )
userCredentialFile := config . DefaultCredentialFile
2020-11-25 06:55:13 +00:00
if err = copyUserConfiguration ( userConfigDir , userConfigFile , userCredentialFile , log ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msgf ( "Failed to copy user configuration. Before running the service, ensure that %s contains two files, %s and %s" ,
2020-10-19 12:30:25 +00:00
serviceConfigDir , serviceCredentialFile , serviceConfigFile )
return err
}
templateArgs . ExtraArgs = [ ] string {
"--origincert" , serviceConfigDir + "/" + serviceCredentialFile ,
}
} else {
2020-11-25 06:55:13 +00:00
src , err := config . ReadConfigFile ( c , log )
2020-10-19 12:30:25 +00:00
if err != nil {
return err
}
// can't use context because this command doesn't define "credentials-file" flag
configPresent := func ( s string ) bool {
val , err := src . String ( s )
return err == nil && val != ""
}
2020-11-23 21:36:16 +00:00
if src . TunnelID == "" || ! configPresent ( tunnel . CredFileFlag ) {
2020-10-19 12:30:25 +00:00
return fmt . Errorf ( ` Configuration file % s must contain entries for the tunnel to run and its associated credentials :
tunnel : TUNNEL - UUID
credentials - file : CREDENTIALS - FILE
` , src . Source ( ) )
}
if src . Source ( ) != serviceConfigPath {
if exists , err := config . FileExists ( serviceConfigPath ) ; err != nil || exists {
return fmt . Errorf ( "Possible conflicting configuration in %[1]s and %[2]s. Either remove %[2]s or run `cloudflared --config %[2]s service install`" , src . Source ( ) , serviceConfigPath )
}
if err := copyFile ( src . Source ( ) , serviceConfigPath ) ; err != nil {
return fmt . Errorf ( "failed to copy %s to %s: %w" , src . Source ( ) , serviceConfigPath , err )
}
}
templateArgs . ExtraArgs = [ ] string {
"tunnel" , "run" ,
}
}
2018-05-01 23:45:06 +00:00
switch {
case isSystemd ( ) :
2020-11-25 06:55:13 +00:00
log . Info ( ) . Msgf ( "Using Systemd" )
return installSystemd ( & templateArgs , log )
2018-05-01 23:45:06 +00:00
default :
2020-11-25 06:55:13 +00:00
log . Info ( ) . Msgf ( "Using SysV" )
return installSysv ( & templateArgs , log )
2018-05-01 23:45:06 +00:00
}
}
2020-11-25 06:55:13 +00:00
func installSystemd ( templateArgs * ServiceTemplateArgs , log * zerolog . Logger ) error {
2018-05-01 23:45:06 +00:00
for _ , serviceTemplate := range systemdTemplates {
err := serviceTemplate . Generate ( templateArgs )
if err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "error generating service template" )
2018-05-01 23:45:06 +00:00
return err
}
}
if err := runCommand ( "systemctl" , "enable" , "cloudflared.service" ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "systemctl enable cloudflared.service error" )
2018-05-01 23:45:06 +00:00
return err
}
if err := runCommand ( "systemctl" , "start" , "cloudflared-update.timer" ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "systemctl start cloudflared-update.timer error" )
2018-05-01 23:45:06 +00:00
return err
}
2020-12-28 18:10:01 +00:00
log . Info ( ) . Msg ( "systemctl daemon-reload" )
2018-05-01 23:45:06 +00:00
return runCommand ( "systemctl" , "daemon-reload" )
}
2020-11-25 06:55:13 +00:00
func installSysv ( templateArgs * ServiceTemplateArgs , log * zerolog . Logger ) error {
2018-05-01 23:45:06 +00:00
confPath , err := sysvTemplate . ResolvePath ( )
if err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "error resolving system path" )
2018-05-01 23:45:06 +00:00
return err
}
if err := sysvTemplate . Generate ( templateArgs ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "error generating system template" )
2018-05-01 23:45:06 +00:00
return err
}
for _ , i := range [ ... ] string { "2" , "3" , "4" , "5" } {
if err := os . Symlink ( confPath , "/etc/rc" + i + ".d/S50et" ) ; err != nil {
continue
}
}
for _ , i := range [ ... ] string { "0" , "1" , "6" } {
if err := os . Symlink ( confPath , "/etc/rc" + i + ".d/K02et" ) ; err != nil {
continue
}
}
return nil
}
func uninstallLinuxService ( c * cli . Context ) error {
2020-11-25 06:55:13 +00:00
log := logger . CreateLoggerFromContext ( c , logger . EnableTerminalLog )
2020-04-29 20:51:32 +00:00
2018-05-01 23:45:06 +00:00
switch {
case isSystemd ( ) :
2020-12-28 18:10:01 +00:00
log . Info ( ) . Msg ( "Using Systemd" )
2020-11-25 06:55:13 +00:00
return uninstallSystemd ( log )
2018-05-01 23:45:06 +00:00
default :
2020-12-28 18:10:01 +00:00
log . Info ( ) . Msg ( "Using SysV" )
2020-11-25 06:55:13 +00:00
return uninstallSysv ( log )
2018-05-01 23:45:06 +00:00
}
}
2020-11-25 06:55:13 +00:00
func uninstallSystemd ( log * zerolog . Logger ) error {
2018-05-01 23:45:06 +00:00
if err := runCommand ( "systemctl" , "disable" , "cloudflared.service" ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "systemctl disable cloudflared.service error" )
2018-05-01 23:45:06 +00:00
return err
}
if err := runCommand ( "systemctl" , "stop" , "cloudflared-update.timer" ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "systemctl stop cloudflared-update.timer error" )
2018-05-01 23:45:06 +00:00
return err
}
for _ , serviceTemplate := range systemdTemplates {
if err := serviceTemplate . Remove ( ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "error removing service template" )
2018-05-01 23:45:06 +00:00
return err
}
}
2020-12-28 18:10:01 +00:00
log . Info ( ) . Msgf ( "Successfully uninstalled cloudflared service from systemd" )
2018-05-01 23:45:06 +00:00
return nil
}
2020-11-25 06:55:13 +00:00
func uninstallSysv ( log * zerolog . Logger ) error {
2018-05-01 23:45:06 +00:00
if err := sysvTemplate . Remove ( ) ; err != nil {
2020-12-28 18:10:01 +00:00
log . Err ( err ) . Msg ( "error removing service template" )
2018-05-01 23:45:06 +00:00
return err
}
for _ , i := range [ ... ] string { "2" , "3" , "4" , "5" } {
if err := os . Remove ( "/etc/rc" + i + ".d/S50et" ) ; err != nil {
continue
}
}
for _ , i := range [ ... ] string { "0" , "1" , "6" } {
if err := os . Remove ( "/etc/rc" + i + ".d/K02et" ) ; err != nil {
continue
}
}
2020-12-28 18:10:01 +00:00
log . Info ( ) . Msgf ( "Successfully uninstalled cloudflared service from sysv" )
2018-05-01 23:45:06 +00:00
return nil
}