mirror of https://github.com/jetkvm/kvm.git
99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package kvm
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/gwatts/rootcerts"
|
|
)
|
|
|
|
var appCtx context.Context
|
|
|
|
func Main() {
|
|
var cancel context.CancelFunc
|
|
appCtx, cancel = context.WithCancel(context.Background())
|
|
defer cancel()
|
|
logger.Info().Msg("Starting JetKvm")
|
|
go runWatchdog()
|
|
go confirmCurrentSystem()
|
|
|
|
http.DefaultClient.Timeout = 1 * time.Minute
|
|
LoadConfig()
|
|
logger.Debug().Msg("config loaded")
|
|
|
|
err := rootcerts.UpdateDefaultTransport()
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("failed to load CA certs")
|
|
}
|
|
|
|
go TimeSyncLoop()
|
|
|
|
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)
|
|
}
|