Add extra NULL checks in smb2_set_[domain|workstation]

From bdodge <[email protected]> Signed-off-by: Ronnie Sahlberg <[email protected]>
sahlberg · Dec 27, 2024 · 1899df3 · 1899df3
1 parent 22c6086
commit 1899df3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/init.c b/lib/init.c
@@ -656,6 +656,10 @@ void smb2_set_domain(struct smb2_context *smb2, const char *domain)
 {
         if (smb2->domain) {
                 free(discard_const(smb2->domain));
+                smb2->domain = NULL;
+        }
+        if (domain == NULL) {
+                return;
         }
         smb2->domain = strdup(domain);
 }
@@ -672,6 +676,10 @@ void smb2_set_workstation(struct smb2_context *smb2, const char *workstation)
 {
         if (smb2->workstation) {
                 free(discard_const(smb2->workstation));
+                smb2->workstation = NULL;
+        }
+        if (workstation == NULL) {
+                return;
         }
         smb2->workstation = strdup(workstation);
 }