From d8183d2bb3cdbada38d82afb947aea7dc16814da Mon Sep 17 00:00:00 2001
From: Evangelos Mamalakis <mamalakis@google.com>
Date: Mon, 2 Dec 2024 15:16:37 +0100
Subject: [PATCH] Add null check in ProcessToken constuctor/destructor

---
 Source/santad/ProcessTree/process_tree.cc | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/Source/santad/ProcessTree/process_tree.cc b/Source/santad/ProcessTree/process_tree.cc
index 0f4bb3843..b1eb2272b 100644
--- a/Source/santad/ProcessTree/process_tree.cc
+++ b/Source/santad/ProcessTree/process_tree.cc
@@ -308,9 +308,17 @@ Tokens
 ProcessToken::ProcessToken(std::shared_ptr<ProcessTree> tree,
                            std::vector<struct Pid> pids)
     : tree_(std::move(tree)), pids_(std::move(pids)) {
+  if (tree_ == nullptr) {
+    return;
+  }
   tree_->RetainProcess(pids);
 }
 
-ProcessToken::~ProcessToken() { tree_->ReleaseProcess(pids_); }
+ProcessToken::~ProcessToken() {
+  if (tree_ == nullptr) {
+    return;
+  }
+  tree_->ReleaseProcess(pids_);
+}
 
 }  // namespace santa::santad::process_tree