From e03b312b12785dc226c8f63e5d728e158fad38fd Mon Sep 17 00:00:00 2001
From: Stefan Metzmacher <metze@samba.org>
Date: Thu, 20 Oct 2011 09:42:10 +0200
Subject: libcli/smb: add smb_setlen_[nbt|tcp] macros

metze
---
 libcli/smb/smb_constants.h | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'libcli')

diff --git a/libcli/smb/smb_constants.h b/libcli/smb/smb_constants.h
index 886e35884f..3bedf2593b 100644
--- a/libcli/smb/smb_constants.h
+++ b/libcli/smb/smb_constants.h
@@ -61,9 +61,17 @@
 
 #define smb_len_nbt(buf) (RIVAL(buf, 0) & 0x1FFFF)
 #define _smb_setlen_nbt(buf,len) RSIVAL(buf, 0, (len) & 0x1FFFF)
+#define smb_setlen_nbt(buf, len) do { \
+	_smb_setlen_nbt(buf, len); \
+	SIVAL(buf, 4, SMB_MAGIC); \
+} while (0)
 
 #define smb_len_tcp(buf) (RIVAL(buf, 0) & 0xFFFFFF)
 #define _smb_setlen_tcp(buf,len) RSIVAL(buf, 0, (len) & 0xFFFFFF)
+#define smb_setlen_tcp(buf, len) do { \
+	_smb_setlen_tcp(buf, len); \
+	SIVAL(buf, 4, SMB_MAGIC); \
+} while (0)
 
 /* protocol types. It assumes that higher protocols include lower protocols
    as subsets. */
-- 
cgit