Fix inspired by work done by David Disseldorp for bug #8040 - smbclient segfaults...

[bbaumbach/samba-autobuild/.git] / libcli / auth / ntlmssp_sign.c

diff --git a/libcli/auth/ntlmssp_sign.c b/libcli/auth/ntlmssp_sign.c
index 0e57c07a8d550be771b93cf08a8e998778cbcd07..42b459c6d47a0f0ee5b5d71b1d65e4e537f65794 100644 (file)
--- a/libcli/auth/ntlmssp_sign.c
+++ b/libcli/auth/ntlmssp_sign.c
@@ -130,17 +130,17 @@ static NTSTATUS ntlmssp_make_packet_signature(struct ntlmssp_state *ntlmssp_stat
                dump_data_pw("ntlmssp v2 sig ", sig->data, sig->length);
 
        } else {
-               bool ok;
+               NTSTATUS status;
                uint32_t crc;
 
                crc = crc32_calc_buffer(data, length);
 
-               ok = msrpc_gen(sig_mem_ctx,
+               status = msrpc_gen(sig_mem_ctx,
                               sig, "dddd",
                               NTLMSSP_SIGN_VERSION, 0, crc,
                               ntlmssp_state->crypt->ntlm.seq_num);
-               if (!ok) {
-                       return NT_STATUS_NO_MEMORY;
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
 
                ntlmssp_state->crypt->ntlm.seq_num++;
@@ -307,17 +307,17 @@ NTSTATUS ntlmssp_seal_packet(struct ntlmssp_state *ntlmssp_state,
                                           sig->data+4, 8);
                }
        } else {
-               bool ok;
+               NTSTATUS status;
                uint32_t crc;
 
                crc = crc32_calc_buffer(data, length);
 
-               ok = msrpc_gen(sig_mem_ctx,
+               status = msrpc_gen(sig_mem_ctx,
                               sig, "dddd",
                               NTLMSSP_SIGN_VERSION, 0, crc,
                               ntlmssp_state->crypt->ntlm.seq_num);
-               if (!ok) {
-                       return NT_STATUS_NO_MEMORY;
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
 
                /*