Add null check in ProcessToken constuctor/destructor

google · Dec 2, 2024 · d8183d2 · d8183d2
1 parent 737525b
commit d8183d2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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