GODT-2039: bridge monitors bridge-gui via its PID (port from v2.4)

internal/app/app.go

@@ -69,15 +69,16 @@ const (
 
 // Hidden flags.
 const (
-	flagLauncher = "launcher"
-	flagNoWindow = "no-window"
+	flagLauncher  = "launcher"
+	flagNoWindow  = "no-window"
+	flagParentPID = "parent-pid"
 )
 
 const (
 	appUsage = "Proton Mail IMAP and SMTP Bridge"
 )
 
-func New() *cli.App {
+func New() *cli.App { //nolint:funlen
 	app := cli.NewApp()
 
 	app.Name = constants.FullAppName
@@ -133,6 +134,11 @@ func New() *cli.App {
 			Usage:  "The launcher used to start the app",
 			Hidden: true,
 		},
+		&cli.IntFlag{
+			Name:   flagParentPID,
+			Usage:  "Process ID of the parent",
+			Hidden: true,
+		},
 	}
 
 	app.Action = run
@@ -203,9 +209,14 @@ func run(c *cli.Context) error { //nolint:funlen
 												b.PushError(bridge.ErrVaultCorrupt)
 											}
 
+											parentPID := -1
+											if pid := c.Int(flagParentPID); pid != 0 {
+												parentPID = pid
+											}
+
 											// Run the frontend.
 											return runFrontend(c, crashHandler, restarter,
-												locations, b, eventCh,
+												locations, b, eventCh, parentPID,
 											)
 										},
 									)

internal/app/frontend.go

@@ -38,6 +38,7 @@ func runFrontend(
 	locations *locations.Locations,
 	bridge *bridge.Bridge,
 	eventCh <-chan events.Event,
+	parentPID int,
 ) error {
 	switch {
 	case c.Bool(flagCLI):
@@ -47,7 +48,7 @@ func runFrontend(
 		select {}
 
 	case c.Bool(flagGRPC):
-		service, err := grpc.NewService(crashHandler, restarter, locations, bridge, eventCh, !c.Bool(flagNoWindow))
+		service, err := grpc.NewService(crashHandler, restarter, locations, bridge, eventCh, !c.Bool(flagNoWindow), parentPID)
 		if err != nil {
 			return fmt.Errorf("could not create service: %w", err)
 		}

internal/bridge/bug_report.go

@@ -36,7 +36,7 @@ const (
 	MaxCompressedFilesCount = 6
 )
 
-func (bridge *Bridge) ReportBug(ctx context.Context, osType, osVersion, description, username, email, client string, attachLogs bool) error {
+func (bridge *Bridge) ReportBug(ctx context.Context, osType, osVersion, description, username, email, client string, attachLogs bool) error { //nolint:funlen
 	var account string
 
 	if info, err := bridge.QueryUserInfo(username); err == nil {
@@ -59,15 +59,15 @@ func (bridge *Bridge) ReportBug(ctx context.Context, osType, osVersion, descript
 			return err
 		}
 
-		guiLogs, err := getMatchingLogs(bridge.locator, func(filename string) bool {
-			return logging.MatchGUILogName(filename) && !logging.MatchStackTraceName(filename)
+		crashes, err := getMatchingLogs(bridge.locator, func(filename string) bool {
+			return logging.MatchLogName(filename) && logging.MatchStackTraceName(filename)
 		})
 		if err != nil {
 			return err
 		}
 
-		crashes, err := getMatchingLogs(bridge.locator, func(filename string) bool {
-			return logging.MatchLogName(filename) && logging.MatchStackTraceName(filename)
+		guiLogs, err := getMatchingLogs(bridge.locator, func(filename string) bool {
+			return logging.MatchGUILogName(filename) && !logging.MatchStackTraceName(filename)
 		})
 		if err != nil {
 			return err

internal/frontend/bridge-gui/bridge-gui/main.cpp

@@ -257,7 +257,10 @@ void launchBridge(QStringList const &args)
     else
         app().log().debug(QString("Bridge executable path: %1").arg(QDir::toNativeSeparators(bridgeExePath)));
 
-    overseer = std::make_unique<Overseer>(new ProcessMonitor(bridgeExePath, QStringList("--grpc") + args, nullptr), nullptr);
+    qint64 const pid = qApp->applicationPid();
+    QStringList const  params =  QStringList { "--grpc", "--parent-pid", QString::number(pid) } + args ;
+    app().log().info(QString("Launching bridge process with command \"%1\" %2").arg(bridgeExePath, params.join(" ")));
+    overseer = std::make_unique<Overseer>(new ProcessMonitor(bridgeExePath, params , nullptr), nullptr);
     overseer->startWorker(true);
 }
 

internal/frontend/grpc/service.go

@@ -34,6 +34,9 @@ import (
 	"github.com/ProtonMail/proton-bridge/v2/internal/events"
 	"github.com/ProtonMail/proton-bridge/v2/internal/safe"
 	"github.com/ProtonMail/proton-bridge/v2/internal/updater"
+	"github.com/bradenaw/juniper/xslices"
+	"github.com/elastic/go-sysinfo"
+	sysinfotypes "github.com/elastic/go-sysinfo/types"
 	"github.com/google/uuid"
 	"github.com/sirupsen/logrus"
 	"gitlab.protontech.ch/go/liteapi"
@@ -80,8 +83,9 @@ type Service struct { // nolint:structcheck
 	initializing       sync.WaitGroup
 	initializationDone sync.Once
 	firstTimeAutostart sync.Once
-
-	showOnStartup bool
+	parentPID          int
+	parentPIDDoneCh    chan struct{}
+	showOnStartup      bool
 }
 
 // NewService returns a new instance of the service.
@@ -94,6 +98,7 @@ func NewService(
 	bridge *bridge.Bridge,
 	eventCh <-chan events.Event,
 	showOnStartup bool,
+	parentPID int,
 ) (*Service, error) {
 	tlsConfig, certPEM, err := newTLSConfig()
 	if err != nil {
@@ -137,7 +142,9 @@ func NewService(
 		initializationDone: sync.Once{},
 		firstTimeAutostart: sync.Once{},
 
-		showOnStartup: showOnStartup,
+		parentPID:       parentPID,
+		parentPIDDoneCh: make(chan struct{}),
+		showOnStartup:   showOnStartup,
 	}
 
 	// Initializing.Done is only called sync.Once. Please keep the increment set to 1
@@ -170,6 +177,12 @@ func (s *Service) initAutostart() {
 }
 
 func (s *Service) Loop() error {
+	if s.parentPID < 0 {
+		s.log.Info("Not monitoring parent PID")
+	} else {
+		go s.monitorParentPID()
+	}
+
 	defer func() {
 		_ = s.bridge.SetFirstStartGUI(false)
 	}()
@@ -486,3 +499,41 @@ func newStreamTokenValidator(wantToken string) grpc.StreamServerInterceptor {
 		return handler(srv, stream)
 	}
 }
+
+// monitorParentPID check at regular intervals that the parent process is still alive, and if not shuts down the server
+// and the applications.
+func (s *Service) monitorParentPID() {
+	s.log.Infof("Starting to monitor parent PID %v", s.parentPID)
+	ticker := time.NewTicker(5 * time.Second)
+
+	for {
+		select {
+		case <-ticker.C:
+			if s.parentPID < 0 {
+				continue
+			}
+
+			processes, err := sysinfo.Processes() // sysinfo.Process(pid) does not seem to work on Windows.
+			if err != nil {
+				s.log.Debug("Could not retrieve process list")
+				continue
+			}
+
+			if !xslices.Any(processes, func(p sysinfotypes.Process) bool { return p != nil && p.PID() == s.parentPID }) {
+				s.log.Info("Parent process does not exist anymore. Initiating shutdown")
+				// quit will write to the parentPIDDoneCh, so we launch a goroutine.
+				go func() {
+					if err := s.quit(); err != nil {
+						logrus.WithError(err).Error("Error on quit")
+					}
+				}()
+			} else {
+				s.log.Tracef("Parent process %v is still alive", s.parentPID)
+			}
+
+		case <-s.parentPIDDoneCh:
+			s.log.Infof("Stopping process monitoring for PID %v", s.parentPID)
+			return
+		}
+	}
+}

internal/frontend/grpc/service_methods.go

@@ -104,6 +104,10 @@ func (s *Service) Quit(ctx context.Context, empty *emptypb.Empty) (*emptypb.Empt
 func (s *Service) quit() error {
 	// Windows is notably slow at Quitting. We do it in a goroutine to speed things up a bit.
 	go func() {
+		if s.parentPID >= 0 {
+			s.parentPIDDoneCh <- struct{}{}
+		}
+
 		var err error
 		if s.isStreamingEvents() {
 			if err = s.stopEventStream(); err != nil {