Initial import
[samba] / 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 2 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, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24 #ifndef _SMB_MACROS_H
25 #define _SMB_MACROS_H
26
27 /* no ops to help reduce the diff between the current 3.0 and release branch */
28
29 #define toupper_ascii(x)        toupper(x)
30 #define tolower_ascii(x)        tolower(x)
31 #define isupper_ascii(x)        isupper(x)
32 #define islower_ascii(x)        islower(x)
33
34
35 /* Misc bit macros */
36 #define BOOLSTR(b) ((b) ? "Yes" : "No")
37 #define BITSETW(ptr,bit) ((SVAL(ptr,0) & (1<<(bit)))!=0)
38
39 /* for readability... */
40 #define IS_DOS_READONLY(test_mode) (((test_mode) & aRONLY) != 0)
41 #define IS_DOS_DIR(test_mode)      (((test_mode) & aDIR) != 0)
42 #define IS_DOS_ARCHIVE(test_mode)  (((test_mode) & aARCH) != 0)
43 #define IS_DOS_SYSTEM(test_mode)   (((test_mode) & aSYSTEM) != 0)
44 #define IS_DOS_HIDDEN(test_mode)   (((test_mode) & aHIDDEN) != 0)
45
46 #ifndef SAFE_FREE /* Oh no this is also defined in tdb.h */
47
48 /**
49  * Free memory if the pointer and zero the pointer.
50  *
51  * @note You are explicitly allowed to pass NULL pointers -- they will
52  * always be ignored.
53  **/
54 #define SAFE_FREE(x) do { if ((x) != NULL) {free(x); x=NULL;} } while(0)
55 #endif
56
57 /* zero a structure */
58 #define ZERO_STRUCT(x) memset((char *)&(x), 0, sizeof(x))
59
60 /* zero a structure given a pointer to the structure */
61 #define ZERO_STRUCTP(x) do { if ((x) != NULL) memset((char *)(x), 0, sizeof(*(x))); } while(0)
62
63 /* zero a structure given a pointer to the structure - no zero check */
64 #define ZERO_STRUCTPN(x) memset((char *)(x), 0, sizeof(*(x)))
65
66 /* zero an array - note that sizeof(array) must work - ie. it must not be a 
67    pointer */
68 #define ZERO_ARRAY(x) memset((char *)(x), 0, sizeof(x))
69
70 /* pointer difference macro */
71 #define PTR_DIFF(p1,p2) ((ptrdiff_t)(((const char *)(p1)) - (const char *)(p2)))
72
73 /* work out how many elements there are in a static array */
74 #define ARRAY_SIZE(a) (sizeof(a)/sizeof(a[0]))
75
76 /* assert macros */
77 #ifdef DEVELOPER
78 #define SMB_ASSERT(b) ( (b) ? (void)0 : \
79         (DEBUG(0,("PANIC: assert failed at %s(%d)\n", \
80                  __FILE__, __LINE__)), smb_panic("assert failed")))
81 #else
82 /* redefine the assert macro for non-developer builds */
83 #define SMB_ASSERT(b) ( (b) ? (void)0 : \
84         (DEBUG(0,("PANIC: assert failed at %s(%d)\n", __FILE__, __LINE__))))
85 #endif
86
87 #define SMB_ASSERT_ARRAY(a,n) SMB_ASSERT((sizeof(a)/sizeof((a)[0])) >= (n))
88
89 /* these are useful macros for checking validity of handles */
90 #define OPEN_FSP(fsp)    ((fsp) && !(fsp)->is_directory)
91 #define OPEN_CONN(conn)    ((conn) && (conn)->open)
92 #define IS_IPC(conn)       ((conn) && (conn)->ipc)
93 #define IS_PRINT(conn)       ((conn) && (conn)->printer)
94 /* you must add the following extern declaration to files using this macro
95  * extern struct current_user current_user;
96  */
97 #define FSP_BELONGS_CONN(fsp,conn) do {\
98                         extern struct current_user current_user;\
99                         if (!((fsp) && (conn) && ((conn)==(fsp)->conn) && (current_user.vuid==(fsp)->vuid))) \
100                                 return(ERROR_DOS(ERRDOS,ERRbadfid));\
101                         } while(0)
102
103 #define FNUM_OK(fsp,c) (OPEN_FSP(fsp) && (c)==(fsp)->conn && current_user.vuid==(fsp)->vuid)
104
105 /* you must add the following extern declaration to files using this macro
106  * extern struct current_user current_user;
107  */
108 #define CHECK_FSP(fsp,conn) do {\
109                         extern struct current_user current_user;\
110                         if (!FNUM_OK(fsp,conn)) \
111                                 return(ERROR_DOS(ERRDOS,ERRbadfid)); \
112                         else if((fsp)->fh->fd == -1) \
113                                 return(ERROR_DOS(ERRDOS,ERRbadaccess));\
114                         (fsp)->num_smb_operations++;\
115                         } while(0)
116
117 #define CHECK_READ(fsp,inbuf) (((fsp)->fh->fd != -1) && ((fsp)->can_read || \
118                         ((SVAL((inbuf),smb_flg2) & FLAGS2_READ_PERMIT_EXECUTE) && \
119                          (fsp->access_mask & FILE_EXECUTE))))
120
121 #define CHECK_WRITE(fsp) ((fsp)->can_write && ((fsp)->fh->fd != -1))
122
123 #define ERROR_WAS_LOCK_DENIED(status) (NT_STATUS_EQUAL((status), NT_STATUS_LOCK_NOT_GRANTED) || \
124                                 NT_STATUS_EQUAL((status), NT_STATUS_FILE_LOCK_CONFLICT) )
125
126 /* the service number for the [globals] defaults */ 
127 #define GLOBAL_SECTION_SNUM     (-1)
128 /* translates a connection number into a service number */
129 #define SNUM(conn)              ((conn)?(conn)->service:GLOBAL_SECTION_SNUM)
130
131
132 /* access various service details */
133 #define SERVICE(snum)      (lp_servicename(snum))
134 #define PRINTERNAME(snum)  (lp_printername(snum))
135 #define CAN_WRITE(conn)    (!conn->read_only)
136 #define VALID_SNUM(snum)   (lp_snum_ok(snum))
137 #define GUEST_OK(snum)     (VALID_SNUM(snum) && lp_guest_ok(snum))
138 #define GUEST_ONLY(snum)   (VALID_SNUM(snum) && lp_guest_only(snum))
139 #define CAN_SETDIR(snum)   (!lp_no_set_dir(snum))
140 #define CAN_PRINT(conn)    ((conn) && lp_print_ok((conn)->service))
141 #define MAP_HIDDEN(conn)   ((conn) && lp_map_hidden((conn)->service))
142 #define MAP_SYSTEM(conn)   ((conn) && lp_map_system((conn)->service))
143 #define MAP_ARCHIVE(conn)   ((conn) && lp_map_archive((conn)->service))
144 #define IS_HIDDEN_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->hide_list,(conn)->case_sensitive))
145 #define IS_VETO_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->veto_list,(conn)->case_sensitive))
146 #define IS_VETO_OPLOCK_PATH(conn,path)  ((conn) && is_in_path((path),(conn)->veto_oplock_list,(conn)->case_sensitive))
147
148 /* 
149  * Used by the stat cache code to check if a returned
150  * stat structure is valid.
151  */
152
153 #define VALID_STAT(st) ((st).st_nlink != 0)  
154 #define VALID_STAT_OF_DIR(st) (VALID_STAT(st) && S_ISDIR((st).st_mode))
155 #define SET_STAT_INVALID(st) ((st).st_nlink = 0)
156
157 #ifndef MIN
158 #define MIN(a,b) ((a)<(b)?(a):(b))
159 #endif
160 #ifndef MAX
161 #define MAX(a,b) ((a)>(b)?(a):(b))
162 #endif
163
164 #ifndef ABS
165 #define ABS(a) ((a)>0?(a):(-(a)))
166 #endif
167
168 /* Macros to get at offsets within smb_lkrng and smb_unlkrng
169    structures. We cannot define these as actual structures
170    due to possible differences in structure packing
171    on different machines/compilers. */
172
173 #define SMB_LPID_OFFSET(indx) (10 * (indx))
174 #define SMB_LKOFF_OFFSET(indx) ( 2 + (10 * (indx)))
175 #define SMB_LKLEN_OFFSET(indx) ( 6 + (10 * (indx)))
176 #define SMB_LARGE_LPID_OFFSET(indx) (20 * (indx))
177 #define SMB_LARGE_LKOFF_OFFSET_HIGH(indx) (4 + (20 * (indx)))
178 #define SMB_LARGE_LKOFF_OFFSET_LOW(indx) (8 + (20 * (indx)))
179 #define SMB_LARGE_LKLEN_OFFSET_HIGH(indx) (12 + (20 * (indx)))
180 #define SMB_LARGE_LKLEN_OFFSET_LOW(indx) (16 + (20 * (indx)))
181
182 /* Macro to test if an error has been cached for this fnum */
183 #define HAS_CACHED_ERROR(fsp) ((fsp)->wbmpx_ptr && \
184                 (fsp)->wbmpx_ptr->wr_discard)
185 /* Macro to turn the cached error into an error packet */
186 #define CACHED_ERROR(fsp) cached_error_packet(outbuf,fsp,__LINE__,__FILE__)
187
188 #define ERROR_DOS(class,code) error_packet(outbuf,class,code,NT_STATUS_OK,__LINE__,__FILE__)
189 #define ERROR_FORCE_DOS(class,code) error_packet(outbuf,class,code,NT_STATUS_INVALID,__LINE__,__FILE__)
190 #define ERROR_NT(status) error_packet(outbuf,0,0,status,__LINE__,__FILE__)
191 #define ERROR_FORCE_NT(status) error_packet(outbuf,-1,-1,status,__LINE__,__FILE__)
192 #define ERROR_BOTH(status,class,code) error_packet(outbuf,class,code,status,__LINE__,__FILE__)
193
194 /* this is how errors are generated */
195 #define UNIXERROR(defclass,deferror) unix_error_packet(outbuf,defclass,deferror,NT_STATUS_OK,__LINE__,__FILE__)
196
197 /* these are the datagram types */
198 #define DGRAM_DIRECT_UNIQUE 0x10
199
200 #define SMB_ROUNDUP(x,r) ( ((x)%(r)) ? ( (((x)+(r))/(r))*(r) ) : (x))
201
202 /* Extra macros added by Ying Chen at IBM - speed increase by inlining. */
203 #define smb_buf(buf) (((char *)(buf)) + smb_size + CVAL(buf,smb_wct)*2)
204 #define smb_buflen(buf) (SVAL(buf,smb_vwv0 + (int)CVAL(buf, smb_wct)*2))
205
206 /* the remaining number of bytes in smb buffer 'buf' from pointer 'p'. */
207 #define smb_bufrem(buf, p) (smb_buflen(buf)-PTR_DIFF(p, smb_buf(buf)))
208
209 /* Note that chain_size must be available as an extern int to this macro. */
210 #define smb_offset(p,buf) (PTR_DIFF(p,buf+4) + chain_size)
211
212 #define smb_len(buf) (PVAL(buf,3)|(PVAL(buf,2)<<8)|((PVAL(buf,1)&1)<<16))
213 #define _smb_setlen(buf,len) do { buf[0] = 0; buf[1] = (len&0x10000)>>16; \
214         buf[2] = (len&0xFF00)>>8; buf[3] = len&0xFF; } while (0)
215
216 /*******************************************************************
217 find the difference in milliseconds between two struct timeval
218 values
219 ********************************************************************/
220
221 #define TvalDiff(tvalold,tvalnew) \
222   (((tvalnew)->tv_sec - (tvalold)->tv_sec)*1000 +  \
223          ((int)(tvalnew)->tv_usec - (int)(tvalold)->tv_usec)/1000)
224
225 /****************************************************************************
226 true if two IP addresses are equal
227 ****************************************************************************/
228
229 #define ip_equal(ip1,ip2) ((ip1).s_addr == (ip2).s_addr)
230 #define ip_service_equal(ip1,ip2) ( ((ip1).ip.s_addr == (ip2).ip.s_addr) && ((ip1).port == (ip2).port) )
231
232 /*****************************************************************
233  splits out the last subkey of a key
234  *****************************************************************/  
235
236 #define reg_get_subkey(full_keyname, key_name, subkey_name) \
237         split_at_last_component(full_keyname, key_name, '\\', subkey_name)
238
239 /****************************************************************************
240  Return True if the offset is at zero.
241 ****************************************************************************/
242
243 #define dptr_zero(buf) (IVAL(buf,1) == 0)
244
245 /*******************************************************************
246 copy an IP address from one buffer to another
247 ********************************************************************/
248
249 #define putip(dest,src) memcpy(dest,src,4)
250
251 /*******************************************************************
252  Return True if a server has CIFS UNIX capabilities.
253 ********************************************************************/
254
255 #define SERVER_HAS_UNIX_CIFS(c) ((c)->capabilities & CAP_UNIX)
256
257 /****************************************************************************
258  Make a filename into unix format.
259 ****************************************************************************/
260
261 #define IS_DIRECTORY_SEP(c) ((c) == '\\' || (c) == '/')
262 #define unix_format(fname) string_replace(fname,'\\','/')
263 #define unix_format_w(fname) string_replace_w(fname, UCS2_CHAR('\\'), UCS2_CHAR('/'))
264
265 /****************************************************************************
266  Make a file into DOS format.
267 ****************************************************************************/
268
269 #define dos_format(fname) string_replace(fname,'/','\\')
270
271 /*****************************************************************************
272  Check to see if we are a DC for this domain
273 *****************************************************************************/
274
275 #define IS_DC  (lp_server_role()==ROLE_DOMAIN_PDC || lp_server_role()==ROLE_DOMAIN_BDC) 
276
277 /*****************************************************************************
278  Safe allocation macros.
279 *****************************************************************************/
280
281 #define SMB_MALLOC_ARRAY(type,count) (type *)malloc_array(sizeof(type),(count))
282 #define SMB_REALLOC(p,s) Realloc((p),(s))
283 #define SMB_REALLOC_ARRAY(p,type,count) (type *)realloc_array((p),sizeof(type),(count))
284 #define SMB_CALLOC_ARRAY(type,count) (type *)calloc_array(sizeof(type),(count))
285 #define SMB_XMALLOC_P(type) (type *)smb_xmalloc_array(sizeof(type),1)
286 #define SMB_XMALLOC_ARRAY(type,count) (type *)smb_xmalloc_array(sizeof(type),(count))
287
288 /* limiting size of ipc replies */
289 #define SMB_REALLOC_LIMIT(ptr,size) SMB_REALLOC(ptr,MAX((size),4*1024))
290
291 /* The new talloc is paranoid malloc checker safe. */
292
293 #define TALLOC(ctx, size) talloc_named_const(ctx, size, __location__)
294 #define TALLOC_P(ctx, type) (type *)talloc_named_const(ctx, sizeof(type), #type)
295 #define TALLOC_ARRAY(ctx, type, count) (type *)_talloc_array(ctx, sizeof(type), count, #type)
296 #define TALLOC_MEMDUP(ctx, ptr, size) _talloc_memdup(ctx, ptr, size, __location__)
297 #define TALLOC_ZERO(ctx, size) _talloc_zero(ctx, size, __location__)
298 #define TALLOC_ZERO_P(ctx, type) (type *)_talloc_zero(ctx, sizeof(type), #type)
299 #define TALLOC_ZERO_ARRAY(ctx, type, count) (type *)_talloc_zero_array(ctx, sizeof(type), count, #type)
300 #define TALLOC_REALLOC(ctx, ptr, count) _talloc_realloc(ctx, ptr, count, __location__)
301 #define TALLOC_REALLOC_ARRAY(ctx, ptr, type, count) (type *)_talloc_realloc_array(ctx, ptr, sizeof(type), count, #type)
302 #define talloc_destroy(ctx) talloc_free(ctx)
303 #define TALLOC_FREE(ctx) do { if ((ctx) != NULL) {talloc_free(ctx); ctx=NULL;} } while(0)
304
305 /* only define PARANOID_MALLOC_CHECKER with --enable-developer and not compiling
306    the smbmount utils */
307
308 #if defined(DEVELOPER) && !defined(SMBMOUNT_MALLOC)
309 #  define PARANOID_MALLOC_CHECKER 1
310 #endif
311
312 #if defined(PARANOID_MALLOC_CHECKER)
313
314 #define PRS_ALLOC_MEM(ps, type, count) (type *)prs_alloc_mem_((ps),sizeof(type),(count))
315 #define PRS_ALLOC_MEM_VOID(ps, size) prs_alloc_mem_((ps),(size),1)
316
317 /* Get medieval on our ass about malloc.... */
318
319 /* Restrictions on malloc/realloc/calloc. */
320 #ifdef malloc
321 #undef malloc
322 #endif
323 #define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
324
325 #ifdef realloc
326 #undef realloc
327 #endif
328 #define realloc(p,s) __ERROR_DONT_USE_REALLOC_DIRECTLY
329
330 #ifdef calloc
331 #undef calloc
332 #endif
333 #define calloc(n,s) __ERROR_DONT_USE_CALLOC_DIRECTLY
334
335 #ifdef strndup
336 #undef strndup
337 #endif
338 #define strndup(s,n) __ERROR_DONT_USE_STRNDUP_DIRECTLY
339
340 #ifdef strdup
341 #undef strdup
342 #endif
343 #define strdup(s) __ERROR_DONT_USE_STRDUP_DIRECTLY
344
345 #define SMB_MALLOC(s) malloc_(s)
346 #define SMB_MALLOC_P(type) (type *)malloc_(sizeof(type))
347
348 #define SMB_STRDUP(s) smb_xstrdup(s)
349 #define SMB_STRNDUP(s,n) smb_xstrndup(s,n)
350
351 #else
352
353 #define _STRING_LINE_(s)    #s
354 #define _STRING_LINE2_(s)   _STRING_LINE_(s)
355 #define __LINESTR__       _STRING_LINE2_(__LINE__)
356 #define __location__ __FILE__ ":" __LINESTR__
357
358 #define PRS_ALLOC_MEM(ps, type, count) (type *)prs_alloc_mem((ps),sizeof(type),(count))
359 #define PRS_ALLOC_MEM_VOID(ps, size) prs_alloc_mem((ps),(size),1)
360
361 /* Regular malloc code. */
362
363 #define SMB_MALLOC(s) malloc(s)
364 #define SMB_MALLOC_P(type) (type *)malloc(sizeof(type))
365
366 #define SMB_STRDUP(s) strdup(s)
367 #define SMB_STRNDUP(s,n) strndup(s,n)
368
369 #endif
370
371 #define ADD_TO_ARRAY(mem_ctx, type, elem, array, num) \
372 do { \
373         *(array) = ((mem_ctx) != NULL) ? \
374                 TALLOC_REALLOC_ARRAY(mem_ctx, (*(array)), type, (*(num))+1) : \
375                 SMB_REALLOC_ARRAY((*(array)), type, (*(num))+1); \
376         SMB_ASSERT((*(array)) != NULL); \
377         (*(array))[*(num)] = (elem); \
378         (*(num)) += 1; \
379 } while (0)
380
381 #define ADD_TO_LARGE_ARRAY(mem_ctx, type, elem, array, num, size) \
382         add_to_large_array((mem_ctx), sizeof(type), &(elem), (void **)(array), (num), (size));
383
384 #endif /* _SMB_MACROS_H */