source/include/smb_macros.h

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB parameters and setup
   4    Copyright (C) Andrew Tridgell 1992-1999
   5    Copyright (C) John H Terpstra 1996-1999
   6    Copyright (C) Luke Kenneth Casson Leighton 1996-1999
   7    Copyright (C) Paul Ashton 1998 - 1999
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #ifndef _SMB_MACROS_H
  24 #define _SMB_MACROS_H
  25
  26 /* Misc bit macros */
  27 #define BOOLSTR(b) ((b) ? "Yes" : "No")
  28 #define BITSETW(ptr,bit) ((SVAL(ptr,0) & (1<<(bit)))!=0)
  29
  30 /* for readability... */
  31 #define IS_DOS_READONLY(test_mode) (((test_mode) & aRONLY) != 0)
  32 #define IS_DOS_DIR(test_mode)      (((test_mode) & aDIR) != 0)
  33 #define IS_DOS_ARCHIVE(test_mode)  (((test_mode) & aARCH) != 0)
  34 #define IS_DOS_SYSTEM(test_mode)   (((test_mode) & aSYSTEM) != 0)
  35 #define IS_DOS_HIDDEN(test_mode)   (((test_mode) & aHIDDEN) != 0)
  36
  37 #ifndef SAFE_FREE /* Oh no this is also defined in tdb.h */
  38
  39 /**
  40  * Free memory if the pointer and zero the pointer.
  41  *
  42  * @note You are explicitly allowed to pass NULL pointers -- they will
  43  * always be ignored.
  44  **/
  45 #define SAFE_FREE(x) do { if ((x) != NULL) {free(x); x=NULL;} } while(0)
  46 #endif
  47
  48 /* assert macros */
  49 #ifdef DEVELOPER
  50 #define SMB_ASSERT(b) ( (b) ? (void)0 : \
  51         (DEBUG(0,("PANIC: assert failed at %s(%d): %s\n", \
  52                  __FILE__, __LINE__, #b)), smb_panic("assert failed: " #b)))
  53 #else
  54 /* redefine the assert macro for non-developer builds */
  55 #define SMB_ASSERT(b) ( (b) ? (void)0 : \
  56         (DEBUG(0,("PANIC: assert failed at %s(%d): %s\n", \
  57             __FILE__, __LINE__, #b))))
  58 #endif
  59
  60 #define SMB_WARN(condition, message) \
  61     ((condition) ? (void)0 : \
  62      DEBUG(0, ("WARNING: %s: %s\n", #condition, message)))
  63
  64 #define SMB_ASSERT_ARRAY(a,n) SMB_ASSERT((sizeof(a)/sizeof((a)[0])) >= (n))
  65
  66 /* these are useful macros for checking validity of handles */
  67 #define IS_IPC(conn)       ((conn) && (conn)->ipc)
  68 #define IS_PRINT(conn)       ((conn) && (conn)->printer)
  69 /* you must add the following extern declaration to files using this macro
  70  * (do not add it to the macro as that causes nested extern declaration warnings)
  71  * extern struct current_user current_user;
  72  */
  73 #define FSP_BELONGS_CONN(fsp,conn) do {\
  74                         if (!((fsp) && (conn) && ((conn)==(fsp)->conn) && (current_user.vuid==(fsp)->vuid))) \
  75                                 return ERROR_NT(NT_STATUS_INVALID_HANDLE); \
  76                         } while(0)
  77
  78 #define FNUM_OK(fsp,c) ((fsp) && !(fsp)->is_directory && (c)==(fsp)->conn && current_user.vuid==(fsp)->vuid)
  79
  80 /* you must add the following extern declaration to files using this macro
  81  * (do not add it to the macro as that causes nested extern declaration warnings)
  82  * extern struct current_user current_user;
  83  */
  84 #define CHECK_FSP(fsp,conn) do {\
  85                         if (!(fsp) || !(conn)) \
  86                                 return ERROR_NT(NT_STATUS_INVALID_HANDLE); \
  87                         else if (((conn) != (fsp)->conn) || current_user.vuid != (fsp)->vuid) \
  88                                 return ERROR_NT(NT_STATUS_INVALID_HANDLE); \
  89                         else if ((fsp)->is_directory) \
  90                                 return ERROR_NT(NT_STATUS_INVALID_DEVICE_REQUEST); \
  91                         else if ((fsp)->fh->fd == -1) \
  92                                 return ERROR_NT(NT_STATUS_ACCESS_DENIED); \
  93                         (fsp)->num_smb_operations++;\
  94                         } while(0)
  95
  96 #define CHECK_READ(fsp,inbuf) (((fsp)->fh->fd != -1) && ((fsp)->can_read || \
  97                         ((SVAL((inbuf),smb_flg2) & FLAGS2_READ_PERMIT_EXECUTE) && \
  98                          (fsp->access_mask & FILE_EXECUTE))))
  99
 100 #define CHECK_WRITE(fsp) ((fsp)->can_write && ((fsp)->fh->fd != -1))
 101
 102 #define ERROR_WAS_LOCK_DENIED(status) (NT_STATUS_EQUAL((status), NT_STATUS_LOCK_NOT_GRANTED) || \
 103                                 NT_STATUS_EQUAL((status), NT_STATUS_FILE_LOCK_CONFLICT) )
 104
 105 /* the service number for the [globals] defaults */
 106 #define GLOBAL_SECTION_SNUM     (-1)
 107 /* translates a connection number into a service number */
 108 #define SNUM(conn)              ((conn)?(conn)->params->service:GLOBAL_SECTION_SNUM)
 109
 110
 111 /* access various service details */
 112 #define SERVICE(snum)      (lp_servicename(snum))
 113 #define PRINTERNAME(snum)  (lp_printername(snum))
 114 #define CAN_WRITE(conn)    (!conn->read_only)
 115 #define VALID_SNUM(snum)   (lp_snum_ok(snum))
 116 #define GUEST_OK(snum)     (VALID_SNUM(snum) && lp_guest_ok(snum))
 117 #define GUEST_ONLY(snum)   (VALID_SNUM(snum) && lp_guest_only(snum))
 118 #define CAN_SETDIR(snum)   (!lp_no_set_dir(snum))
 119 #define CAN_PRINT(conn)    ((conn) && lp_print_ok(SNUM(conn)))
 120 #define MAP_HIDDEN(conn)   ((conn) && lp_map_hidden(SNUM(conn)))
 121 #define MAP_SYSTEM(conn)   ((conn) && lp_map_system(SNUM(conn)))
 122 #define MAP_ARCHIVE(conn)   ((conn) && lp_map_archive(SNUM(conn)))
 123 #define IS_HIDDEN_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->hide_list,(conn)->case_sensitive))
 124 #define IS_VETO_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->veto_list,(conn)->case_sensitive))
 125 #define IS_VETO_OPLOCK_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->veto_oplock_list,(conn)->case_sensitive))
 126
 127 /*
 128  * Used by the stat cache code to check if a returned
 129  * stat structure is valid.
 130  */
 131
 132 #define VALID_STAT(st) ((st).st_nlink != 0)
 133 #define VALID_STAT_OF_DIR(st) (VALID_STAT(st) && S_ISDIR((st).st_mode))
 134 #define SET_STAT_INVALID(st) ((st).st_nlink = 0)
 135
 136 #ifndef MIN
 137 #define MIN(a,b) ((a)<(b)?(a):(b))
 138 #endif
 139 #ifndef MAX
 140 #define MAX(a,b) ((a)>(b)?(a):(b))
 141 #endif
 142
 143 #ifndef ABS
 144 #define ABS(a) ((a)>0?(a):(-(a)))
 145 #endif
 146
 147 /* Macros to get at offsets within smb_lkrng and smb_unlkrng
 148    structures. We cannot define these as actual structures
 149    due to possible differences in structure packing
 150    on different machines/compilers. */
 151
 152 #define SMB_LPID_OFFSET(indx) (10 * (indx))
 153 #define SMB_LKOFF_OFFSET(indx) ( 2 + (10 * (indx)))
 154 #define SMB_LKLEN_OFFSET(indx) ( 6 + (10 * (indx)))
 155 #define SMB_LARGE_LPID_OFFSET(indx) (20 * (indx))
 156 #define SMB_LARGE_LKOFF_OFFSET_HIGH(indx) (4 + (20 * (indx)))
 157 #define SMB_LARGE_LKOFF_OFFSET_LOW(indx) (8 + (20 * (indx)))
 158 #define SMB_LARGE_LKLEN_OFFSET_HIGH(indx) (12 + (20 * (indx)))
 159 #define SMB_LARGE_LKLEN_OFFSET_LOW(indx) (16 + (20 * (indx)))
 160
 161 #define ERROR_DOS(class,code) error_packet(outbuf,class,code,NT_STATUS_OK,__LINE__,__FILE__)
 162 #define ERROR_NT(status) error_packet(outbuf,0,0,status,__LINE__,__FILE__)
 163 #define ERROR_FORCE_NT(status) error_packet(outbuf,-1,-1,status,__LINE__,__FILE__)
 164 #define ERROR_BOTH(status,class,code) error_packet(outbuf,class,code,status,__LINE__,__FILE__)
 165
 166 #define reply_nterror(req,status) reply_nt_error(req,status,__LINE__,__FILE__)
 167 #define reply_force_nterror(req,status) reply_force_nt_error(req,status,__LINE__,__FILE__)
 168 #define reply_doserror(req,eclass,ecode) reply_dos_error(req,eclass,ecode,__LINE__,__FILE__)
 169 #define reply_botherror(req,status,eclass,ecode) reply_both_error(req,eclass,ecode,status,__LINE__,__FILE__)
 170 #define reply_unixerror(req,defclass,deferror) reply_unix_error(req,defclass,deferror,NT_STATUS_OK,__LINE__,__FILE__)
 171
 172 /* these are the datagram types */
 173 #define DGRAM_DIRECT_UNIQUE 0x10
 174
 175 #define SMB_ROUNDUP(x,r) ( ((x)%(r)) ? ( (((x)+(r))/(r))*(r) ) : (x))
 176
 177 /* Extra macros added by Ying Chen at IBM - speed increase by inlining. */
 178 #define smb_buf(buf) (((char *)(buf)) + smb_size + CVAL(buf,smb_wct)*2)
 179 #define smb_buflen(buf) (SVAL(buf,smb_vwv0 + (int)CVAL(buf, smb_wct)*2))
 180
 181 /* the remaining number of bytes in smb buffer 'buf' from pointer 'p'. */
 182 #define smb_bufrem(buf, p) (smb_buflen(buf)-PTR_DIFF(p, smb_buf(buf)))
 183
 184 /* Note that chain_size must be available as an extern int to this macro. */
 185 #define smb_offset(p,buf) (PTR_DIFF(p,buf+4) + chain_size)
 186
 187 #define smb_len(buf) (PVAL(buf,3)|(PVAL(buf,2)<<8)|((PVAL(buf,1)&1)<<16))
 188 #define _smb_setlen(buf,len) do { buf[0] = 0; buf[1] = ((len)&0x10000)>>16; \
 189         buf[2] = ((len)&0xFF00)>>8; buf[3] = (len)&0xFF; } while (0)
 190
 191 #define smb_len_large(buf) (PVAL(buf,3)|(PVAL(buf,2)<<8)|(PVAL(buf,1)<<16))
 192 #define _smb_setlen_large(buf,len) do { buf[0] = 0; buf[1] = ((len)&0xFF0000)>>16; \
 193         buf[2] = ((len)&0xFF00)>>8; buf[3] = (len)&0xFF; } while (0)
 194
 195 #define ENCRYPTION_REQUIRED(conn) ((conn) ? ((conn)->encrypt_level == Required) : false)
 196 #define IS_CONN_ENCRYPTED(conn) ((conn) ? (conn)->encrypted_tid : false)
 197
 198 /*******************************************************************
 199 find the difference in milliseconds between two struct timeval
 200 values
 201 ********************************************************************/
 202
 203 #define TvalDiff(tvalold,tvalnew) \
 204   (((tvalnew)->tv_sec - (tvalold)->tv_sec)*1000 +  \
 205          ((int)(tvalnew)->tv_usec - (int)(tvalold)->tv_usec)/1000)
 206
 207 /****************************************************************************
 208 true if two IPv4 addresses are equal
 209 ****************************************************************************/
 210
 211 #define ip_equal_v4(ip1,ip2) ((ip1).s_addr == (ip2).s_addr)
 212
 213 /*****************************************************************
 214  splits out the last subkey of a key
 215  *****************************************************************/
 216
 217 #define reg_get_subkey(full_keyname, key_name, subkey_name) \
 218         split_at_last_component(full_keyname, key_name, '\\', subkey_name)
 219
 220 /****************************************************************************
 221  Return True if the offset is at zero.
 222 ****************************************************************************/
 223
 224 #define dptr_zero(buf) (IVAL(buf,1) == 0)
 225
 226 /*******************************************************************
 227 copy an IP address from one buffer to another
 228 ********************************************************************/
 229
 230 #define putip(dest,src) memcpy(dest,src,4)
 231
 232 /*******************************************************************
 233  Return True if a server has CIFS UNIX capabilities.
 234 ********************************************************************/
 235
 236 #define SERVER_HAS_UNIX_CIFS(c) ((c)->capabilities & CAP_UNIX)
 237
 238 /****************************************************************************
 239  Make a filename into unix format.
 240 ****************************************************************************/
 241
 242 #define IS_DIRECTORY_SEP(c) ((c) == '\\' || (c) == '/')
 243 #define unix_format(fname) string_replace(fname,'\\','/')
 244 #define unix_format_w(fname) string_replace_w(fname, UCS2_CHAR('\\'), UCS2_CHAR('/'))
 245
 246 /****************************************************************************
 247  Make a file into DOS format.
 248 ****************************************************************************/
 249
 250 #define dos_format(fname) string_replace(fname,'/','\\')
 251
 252 /*****************************************************************************
 253  Check to see if we are a DC for this domain
 254 *****************************************************************************/
 255
 256 #define IS_DC  (lp_server_role()==ROLE_DOMAIN_PDC || lp_server_role()==ROLE_DOMAIN_BDC)
 257
 258 /*****************************************************************************
 259  Safe allocation macros.
 260 *****************************************************************************/
 261
 262 #define SMB_MALLOC_ARRAY(type,count) (type *)malloc_array(sizeof(type),(count))
 263 #define SMB_MEMALIGN_ARRAY(type,align,count) (type *)memalign_array(sizeof(type),align,(count))
 264 #define SMB_REALLOC(p,s) Realloc((p),(s),True)  /* Always frees p on error or s == 0 */
 265 #define SMB_REALLOC_KEEP_OLD_ON_ERROR(p,s) Realloc((p),(s),False) /* Never frees p on error or s == 0 */
 266 #define SMB_REALLOC_ARRAY(p,type,count) (type *)realloc_array((p),sizeof(type),(count),True) /* Always frees p on error or s == 0 */
 267 #define SMB_REALLOC_ARRAY_KEEP_OLD_ON_ERROR(p,type,count) (type *)realloc_array((p),sizeof(type),(count),False) /* Never frees p on error or s == 0 */
 268 #define SMB_CALLOC_ARRAY(type,count) (type *)calloc_array(sizeof(type),(count))
 269 #define SMB_XMALLOC_P(type) (type *)smb_xmalloc_array(sizeof(type),1)
 270 #define SMB_XMALLOC_ARRAY(type,count) (type *)smb_xmalloc_array(sizeof(type),(count))
 271
 272 /* The new talloc is paranoid malloc checker safe. */
 273
 274 #if 0
 275
 276 Disable these now we have checked all code paths and ensured
 277 NULL returns on zero request. JRA.
 278
 279 #define TALLOC(ctx, size) talloc_zeronull(ctx, size, __location__)
 280 #define TALLOC_P(ctx, type) (type *)talloc_zeronull(ctx, sizeof(type), #type)
 281 #define TALLOC_ARRAY(ctx, type, count) (type *)_talloc_array_zeronull(ctx, sizeof(type), count, #type)
 282 #define TALLOC_MEMDUP(ctx, ptr, size) _talloc_memdup_zeronull(ctx, ptr, size, __location__)
 283 #define TALLOC_ZERO(ctx, size) _talloc_zero_zeronull(ctx, size, __location__)
 284 #define TALLOC_ZERO_P(ctx, type) (type *)_talloc_zero_zeronull(ctx, sizeof(type), #type)
 285 #define TALLOC_ZERO_ARRAY(ctx, type, count) (type *)_talloc_zero_array_zeronull(ctx, sizeof(type), count, #type)
 286 #define TALLOC_SIZE(ctx, size) talloc_zeronull(ctx, size, __location__)
 287 #define TALLOC_ZERO_SIZE(ctx, size) _talloc_zero_zeronull(ctx, size, __location__)
 288
 289 #else
 290
 291 #define TALLOC(ctx, size) talloc_named_const(ctx, size, __location__)
 292 #define TALLOC_P(ctx, type) (type *)talloc_named_const(ctx, sizeof(type), #type)
 293 #define TALLOC_ARRAY(ctx, type, count) (type *)_talloc_array(ctx, sizeof(type), count, #type)
 294 #define TALLOC_MEMDUP(ctx, ptr, size) _talloc_memdup(ctx, ptr, size, __location__)
 295 #define TALLOC_ZERO(ctx, size) _talloc_zero(ctx, size, __location__)
 296 #define TALLOC_ZERO_P(ctx, type) (type *)_talloc_zero(ctx, sizeof(type), #type)
 297 #define TALLOC_ZERO_ARRAY(ctx, type, count) (type *)_talloc_zero_array(ctx, sizeof(type), count, #type)
 298 #define TALLOC_SIZE(ctx, size) talloc_named_const(ctx, size, __location__)
 299 #define TALLOC_ZERO_SIZE(ctx, size) _talloc_zero(ctx, size, __location__)
 300
 301 #endif
 302
 303 #define TALLOC_REALLOC(ctx, ptr, count) _talloc_realloc(ctx, ptr, count, __location__)
 304 #define TALLOC_REALLOC_ARRAY(ctx, ptr, type, count) (type *)_talloc_realloc_array(ctx, ptr, sizeof(type), count, #type)
 305 #define talloc_destroy(ctx) talloc_free(ctx)
 306 #define TALLOC_FREE(ctx) do { if ((ctx) != NULL) {talloc_free(ctx); ctx=NULL;} } while(0)
 307
 308 /* only define PARANOID_MALLOC_CHECKER with --enable-developer */
 309
 310 #if defined(DEVELOPER)
 311 #  define PARANOID_MALLOC_CHECKER 1
 312 #endif
 313
 314 #if defined(PARANOID_MALLOC_CHECKER)
 315
 316 #define PRS_ALLOC_MEM(ps, type, count) (type *)prs_alloc_mem_((ps),sizeof(type),(count))
 317
 318 /* Get medieval on our ass about malloc.... */
 319
 320 /* Restrictions on malloc/realloc/calloc. */
 321 #ifdef malloc
 322 #undef malloc
 323 #endif
 324 #define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
 325
 326 #ifdef realloc
 327 #undef realloc
 328 #endif
 329 #define realloc(p,s) __ERROR_DONT_USE_REALLOC_DIRECTLY
 330
 331 #ifdef calloc
 332 #undef calloc
 333 #endif
 334 #define calloc(n,s) __ERROR_DONT_USE_CALLOC_DIRECTLY
 335
 336 #ifdef strndup
 337 #undef strndup
 338 #endif
 339 #define strndup(s,n) __ERROR_DONT_USE_STRNDUP_DIRECTLY
 340
 341 #ifdef strdup
 342 #undef strdup
 343 #endif
 344 #define strdup(s) __ERROR_DONT_USE_STRDUP_DIRECTLY
 345
 346 #define SMB_MALLOC(s) malloc_(s)
 347 #define SMB_MALLOC_P(type) (type *)malloc_(sizeof(type))
 348
 349 #define SMB_STRDUP(s) smb_xstrdup(s)
 350 #define SMB_STRNDUP(s,n) smb_xstrndup(s,n)
 351
 352 #else
 353
 354 #define PRS_ALLOC_MEM(ps, type, count) (type *)prs_alloc_mem((ps),sizeof(type),(count))
 355
 356 /* Regular malloc code. */
 357
 358 #define SMB_MALLOC(s) malloc(s)
 359 #define SMB_MALLOC_P(type) (type *)malloc(sizeof(type))
 360
 361 #define SMB_STRDUP(s) strdup(s)
 362 #define SMB_STRNDUP(s,n) strndup(s,n)
 363
 364 #endif
 365
 366 #define ADD_TO_ARRAY(mem_ctx, type, elem, array, num) \
 367 do { \
 368         *(array) = ((mem_ctx) != NULL) ? \
 369                 TALLOC_REALLOC_ARRAY(mem_ctx, (*(array)), type, (*(num))+1) : \
 370                 SMB_REALLOC_ARRAY((*(array)), type, (*(num))+1); \
 371         SMB_ASSERT((*(array)) != NULL); \
 372         (*(array))[*(num)] = (elem); \
 373         (*(num)) += 1; \
 374 } while (0)
 375
 376 #define ADD_TO_LARGE_ARRAY(mem_ctx, type, elem, array, num, size) \
 377         add_to_large_array((mem_ctx), sizeof(type), &(elem), (void *)(array), (num), (size));
 378
 379 #ifndef ISDOT
 380 #define ISDOT(p) (*(p) == '.' && *((p) + 1) == '\0')
 381 #endif /* ISDOT */
 382
 383 #ifndef ISDOTDOT
 384 #define ISDOTDOT(p) (*(p) == '.' && *((p) + 1) == '.' && *((p) + 2) == '\0')
 385 #endif /* ISDOTDOT */
 386
 387 #ifndef toupper_ascii_fast
 388 /* Warning - this must only be called with 0 <= c < 128. IT WILL
 389  * GIVE GARBAGE if c > 128 or c < 0. JRA.
 390  */
 391 extern char toupper_ascii_fast_table[];
 392 #define toupper_ascii_fast(c) toupper_ascii_fast_table[(unsigned int)(c)];
 393 #endif
 394
 395 #endif /* _SMB_MACROS_H */