mirror of https://github.com/jetkvm/kvm.git
115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package kvm
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/gwatts/rootcerts"
|
|
)
|
|
|
|
var appCtx context.Context
|
|
|
|
func Main() {
|
|
LoadConfig()
|
|
|
|
var cancel context.CancelFunc
|
|
appCtx, cancel = context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
systemVersionLocal, appVersionLocal, err := GetLocalVersion()
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("failed to get local version")
|
|
}
|
|
|
|
logger.Info().
|
|
Interface("system_version", systemVersionLocal).
|
|
Interface("app_version", appVersionLocal).
|
|
Msg("starting JetKVM")
|
|
|
|
go runWatchdog()
|
|
go confirmCurrentSystem()
|
|
|
|
http.DefaultClient.Timeout = 1 * time.Minute
|
|
|
|
err = rootcerts.UpdateDefaultTransport()
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("failed to load Root CA certificates")
|
|
}
|
|
logger.Info().
|
|
Int("ca_certs_loaded", len(rootcerts.Certs())).
|
|
Msg("loaded Root CA certificates")
|
|
|
|
initNetwork()
|
|
initTimeSync()
|
|
|
|
timeSync.Start()
|
|
|
|
StartNativeCtrlSocketServer()
|
|
StartNativeVideoSocketServer()
|
|
|
|
initPrometheus()
|
|
|
|
go func() {
|
|
err = ExtractAndRunNativeBin()
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("failed to extract and run native bin")
|
|
//TODO: prepare an error message screen buffer to show on kvm screen
|
|
}
|
|
}()
|
|
|
|
initUsbGadget()
|
|
|
|
go func() {
|
|
time.Sleep(15 * time.Minute)
|
|
for {
|
|
logger.Debug().Bool("auto_update_enabled", config.AutoUpdateEnabled).Msg("UPDATING")
|
|
if !config.AutoUpdateEnabled {
|
|
return
|
|
}
|
|
if currentSession != nil {
|
|
logger.Debug().Msg("skipping update since a session is active")
|
|
time.Sleep(1 * time.Minute)
|
|
continue
|
|
}
|
|
includePreRelease := config.IncludePreRelease
|
|
err = TryUpdate(context.Background(), GetDeviceID(), includePreRelease)
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("failed to auto update")
|
|
}
|
|
time.Sleep(1 * time.Hour)
|
|
}
|
|
}()
|
|
//go RunFuseServer()
|
|
go RunWebServer()
|
|
|
|
go RunWebSecureServer()
|
|
// Web secure server is started only if TLS mode is enabled
|
|
if config.TLSMode != "" {
|
|
startWebSecureServer()
|
|
}
|
|
|
|
// As websocket client already checks if the cloud token is set, we can start it here.
|
|
go RunWebsocketClient()
|
|
|
|
initSerialPort()
|
|
sigs := make(chan os.Signal, 1)
|
|
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
|
<-sigs
|
|
logger.Info().Msg("JetKVM Shutting Down")
|
|
//if fuseServer != nil {
|
|
// err := setMassStorageImage(" ")
|
|
// if err != nil {
|
|
// logger.Infof("Failed to unmount mass storage image: %v", err)
|
|
// }
|
|
// err = fuseServer.Unmount()
|
|
// if err != nil {
|
|
// logger.Infof("Failed to unmount fuse: %v", err)
|
|
// }
|
|
|
|
// os.Exit(0)
|
|
}
|