Have better error handling

inverse-inc · Jan 9, 2025 · 655fb1d · 655fb1d
1 parent d6f7e47
commit 655fb1d
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/pf/Authentication/Source/SAMLSource.pm b/lib/pf/Authentication/Source/SAMLSource.pm
@@ -180,8 +180,16 @@ sub handle_response {
         }
 
         my $assertion = $lassoLogin->get_assertion();
-        my @attribute_list = $assertion->AttributeStatement->Attribute;
-
+        if (!$assertion) {
+            return ($FALSE, "No assertion found");
+        }
+
+        my $attribute_statement = $assertion->AttributeStatement;
+        if (!$attribute_statement) {
+            return ($FALSE, "No assertion statement found");
+        }
+
+        my @attribute_list = $attribute_statement->Attribute;
         my $username;
         # For debug
         foreach my $attribute (@attribute_list){