From 5863107cd3a37585272ee2186a0103f94932b063 Mon Sep 17 00:00:00 2001
From: Stefan Metzmacher <metze@samba.org>
Date: Fri, 20 Jul 2012 09:30:05 +0200
Subject: libcli/smb: prepare [en|de]cryption_key for SMB3

metze
---
 libcli/smb/smbXcli_base.c | 65 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

(limited to 'libcli')

diff --git a/libcli/smb/smbXcli_base.c b/libcli/smb/smbXcli_base.c
index 826e3bc4de..3c14f32893 100644
--- a/libcli/smb/smbXcli_base.c
+++ b/libcli/smb/smbXcli_base.c
@@ -137,6 +137,11 @@ struct smbXcli_session {
 		DATA_BLOB application_key;
 		DATA_BLOB signing_key;
 		bool should_sign;
+		bool should_encrypt;
+		DATA_BLOB encryption_key;
+		DATA_BLOB decryption_key;
+		uint64_t channel_nonce;
+		uint64_t channel_next;
 		DATA_BLOB channel_signing_key;
 	} smb2;
 };
@@ -4167,6 +4172,40 @@ NTSTATUS smb2cli_session_set_session_key(struct smbXcli_session *session,
 				    session->smb2.signing_key.data);
 	}
 
+	session->smb2.encryption_key = data_blob_dup_talloc(session,
+						session->smb2.signing_key);
+	if (session->smb2.encryption_key.data == NULL) {
+		ZERO_STRUCT(session_key);
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	if (conn->protocol >= PROTOCOL_SMB2_24) {
+		const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
+		const DATA_BLOB context = data_blob_string_const_null("ServerIn ");
+
+		smb2_key_derivation(session_key, sizeof(session_key),
+				    label.data, label.length,
+				    context.data, context.length,
+				    session->smb2.encryption_key.data);
+	}
+
+	session->smb2.decryption_key = data_blob_dup_talloc(session,
+						session->smb2.signing_key);
+	if (session->smb2.decryption_key.data == NULL) {
+		ZERO_STRUCT(session_key);
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	if (conn->protocol >= PROTOCOL_SMB2_24) {
+		const DATA_BLOB label = data_blob_string_const_null("SMB2AESCCM");
+		const DATA_BLOB context = data_blob_string_const_null("ServerOut");
+
+		smb2_key_derivation(session_key, sizeof(session_key),
+				    label.data, label.length,
+				    context.data, context.length,
+				    session->smb2.decryption_key.data);
+	}
+
 	session->smb2.application_key = data_blob_dup_talloc(session,
 						session->smb2.signing_key);
 	if (session->smb2.application_key.data == NULL) {
@@ -4199,6 +4238,7 @@ NTSTATUS smb2cli_session_set_session_key(struct smbXcli_session *session,
 	}
 
 	session->smb2.should_sign = false;
+	session->smb2.should_encrypt = false;
 
 	if (conn->desire_signing) {
 		session->smb2.should_sign = true;
@@ -4208,6 +4248,10 @@ NTSTATUS smb2cli_session_set_session_key(struct smbXcli_session *session,
 		session->smb2.should_sign = true;
 	}
 
+	generate_random_buffer((uint8_t *)&session->smb2.channel_nonce,
+			       sizeof(session->smb2.channel_nonce));
+	session->smb2.channel_next = 1;
+
 	return NT_STATUS_OK;
 }
 
@@ -4222,6 +4266,10 @@ NTSTATUS smb2cli_session_create_channel(TALLOC_CTX *mem_ctx,
 		return NT_STATUS_INVALID_PARAMETER_MIX;
 	}
 
+	if (session1->smb2.channel_next == 0) {
+		return NT_STATUS_INVALID_PARAMETER_MIX;
+	}
+
 	if (conn == NULL) {
 		return NT_STATUS_INVALID_PARAMETER_MIX;
 	}
@@ -4246,6 +4294,23 @@ NTSTATUS smb2cli_session_create_channel(TALLOC_CTX *mem_ctx,
 	}
 
 	session2->smb2.should_sign = session1->smb2.should_sign;
+	session2->smb2.should_encrypt = session1->smb2.should_encrypt;
+
+	session2->smb2.channel_nonce = session1->smb2.channel_nonce;
+	session2->smb2.channel_nonce += session1->smb2.channel_next;
+	session1->smb2.channel_next++;
+
+	session2->smb2.encryption_key = data_blob_dup_talloc(session2,
+						session1->smb2.encryption_key);
+	if (session2->smb2.encryption_key.data == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	session2->smb2.decryption_key = data_blob_dup_talloc(session2,
+						session1->smb2.decryption_key);
+	if (session2->smb2.decryption_key.data == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
 
 	talloc_set_destructor(session2, smbXcli_session_destructor);
 	DLIST_ADD_END(conn->sessions, session2, struct smbXcli_session *);
-- 
cgit