s3: Add an async smbsock_connect
[Samba.git] / lib / util / util.h
blobdefef127d93e4f162a8d2186a5a4ca6f40cf939b
1 /*
2 Unix SMB/CIFS implementation.
3 Utility functions for Samba
4 Copyright (C) Andrew Tridgell 1992-1999
5 Copyright (C) Jelmer Vernooij 2005
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #ifndef _SAMBA_UTIL_H_
22 #define _SAMBA_UTIL_H_
24 #include <netinet/in.h>
26 #if _SAMBA_BUILD_ == 4
27 #include "../lib/util/charset/charset.h"
28 #endif
29 #include "../lib/util/attr.h"
31 /* for TALLOC_CTX */
32 #include <talloc.h>
34 /**
35 * @file
36 * @brief Helpful macros
39 struct smbsrv_tcon;
41 extern const char *logfile;
42 extern const char *panic_action;
44 #include "../lib/util/time.h"
45 #include "../lib/util/data_blob.h"
46 #include "../lib/util/xfile.h"
47 #include "../lib/util/mutex.h"
48 #include "../lib/util/byteorder.h"
49 #include "../lib/util/talloc_stack.h"
51 /**
52 * assert macros
54 #ifdef DEVELOPER
55 #define SMB_ASSERT(b) do { if (!(b)) { \
56 DEBUG(0,("PANIC: assert failed at %s(%d): %s\n", \
57 __FILE__, __LINE__, #b)); smb_panic("assert failed: " #b); }} while(0)
58 #else
59 /* redefine the assert macro for non-developer builds */
60 #define SMB_ASSERT(b) do { if (!(b)) { \
61 DEBUG(0,("PANIC: assert failed at %s(%d): %s\n", \
62 __FILE__, __LINE__, #b)); }} while (0)
63 #endif
65 #if _SAMBA_BUILD_ == 4
66 #ifdef VALGRIND
67 #define strlen(x) valgrind_strlen(x)
68 size_t valgrind_strlen(const char *s);
69 #endif
70 #endif
72 #ifndef ABS
73 #define ABS(a) ((a)>0?(a):(-(a)))
74 #endif
76 #include "../lib/util/memory.h"
78 /**
79 * Write backtrace to debug log
81 _PUBLIC_ void call_backtrace(void);
83 /**
84 Something really nasty happened - panic !
85 **/
86 _PUBLIC_ _NORETURN_ void smb_panic(const char *why);
88 #if _SAMBA_BUILD_ == 4
89 /**
90 setup our fault handlers
91 **/
92 _PUBLIC_ void fault_setup(const char *pname);
93 #endif
95 /**
96 register a fault handler.
97 Should only be called once in the execution of smbd.
99 _PUBLIC_ bool register_fault_handler(const char *name, void (*fault_handler)(int sig));
101 /* The following definitions come from lib/util/signal.c */
105 Block sigs.
107 void BlockSignals(bool block, int signum);
110 Catch a signal. This should implement the following semantics:
112 1) The handler remains installed after being called.
113 2) The signal should be blocked during handler execution.
115 void (*CatchSignal(int signum,void (*handler)(int )))(int);
118 Ignore SIGCLD via whatever means is necessary for this OS.
120 void CatchChild(void);
123 Catch SIGCLD but leave the child around so it's status can be reaped.
125 void CatchChildLeaveStatus(void);
127 /* The following definitions come from lib/util/system.c */
129 /**************************************************************************
130 A wrapper for gethostbyname() that tries avoids looking up hostnames
131 in the root domain, which can cause dial-on-demand links to come up for no
132 apparent reason.
133 ****************************************************************************/
134 _PUBLIC_ struct hostent *sys_gethostbyname(const char *name);
135 _PUBLIC_ struct in_addr sys_inet_makeaddr(int net, int host);
138 * Wrapper for fork used to invalid pid cache.
140 _PUBLIC_ pid_t sys_fork(void);
143 * Wrapper for getpid. Ensures we only do a system call *once*.
145 _PUBLIC_ pid_t sys_getpid(void);
147 /* The following definitions come from lib/util/genrand.c */
150 Copy any user given reseed data.
152 _PUBLIC_ void set_rand_reseed_callback(void (*fn)(void *, int *), void *);
155 * Tell the random number generator it needs to reseed.
157 _PUBLIC_ void set_need_random_reseed(void);
160 Interface to the (hopefully) good crypto random number generator.
161 Will use our internal PRNG if more than 40 bytes of random generation
162 has been requested, otherwise tries to read from /dev/random
164 _PUBLIC_ void generate_random_buffer(uint8_t *out, int len);
167 Interface to the (hopefully) good crypto random number generator.
168 Will always use /dev/urandom if available.
170 _PUBLIC_ void generate_secret_buffer(uint8_t *out, int len);
173 generate a single random uint32_t
175 _PUBLIC_ uint32_t generate_random(void);
178 very basic password quality checker
180 _PUBLIC_ bool check_password_quality(const char *s);
183 Use the random number generator to generate a random string.
185 _PUBLIC_ char *generate_random_str_list(TALLOC_CTX *mem_ctx, size_t len, const char *list);
188 * Generate a random text string consisting of the specified length.
189 * The returned string will be allocated.
191 * Characters used are: ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+_-#.,
193 _PUBLIC_ char *generate_random_str(TALLOC_CTX *mem_ctx, size_t len);
195 /* The following definitions come from lib/util/dprintf.c */
196 #if _SAMBA_BUILD_ == 4
198 _PUBLIC_ void d_set_iconv(smb_iconv_t);
199 _PUBLIC_ int d_vfprintf(FILE *f, const char *format, va_list ap) PRINTF_ATTRIBUTE(2,0);
200 _PUBLIC_ int d_fprintf(FILE *f, const char *format, ...) PRINTF_ATTRIBUTE(2,3);
201 _PUBLIC_ int d_printf(const char *format, ...) PRINTF_ATTRIBUTE(1,2);
202 _PUBLIC_ void display_set_stderr(void);
203 #endif
205 /* The following definitions come from lib/util/util_str.c */
207 bool next_token_talloc(TALLOC_CTX *ctx,
208 const char **ptr,
209 char **pp_buff,
210 const char *sep);
213 * Get the next token from a string, return false if none found. Handles
214 * double-quotes. This version does not trim leading separator characters
215 * before looking for a token.
217 bool next_token_no_ltrim_talloc(TALLOC_CTX *ctx,
218 const char **ptr,
219 char **pp_buff,
220 const char *sep);
224 Trim the specified elements off the front and back of a string.
226 _PUBLIC_ bool trim_string(char *s, const char *front, const char *back);
229 Find the number of 'c' chars in a string
231 _PUBLIC_ _PURE_ size_t count_chars(const char *s, char c);
234 Safe string copy into a known length string. maxlength does not
235 include the terminating zero.
237 _PUBLIC_ char *safe_strcpy(char *dest,const char *src, size_t maxlength);
240 Safe string cat into a string. maxlength does not
241 include the terminating zero.
243 _PUBLIC_ char *safe_strcat(char *dest, const char *src, size_t maxlength);
246 Routine to get hex characters and turn them into a 16 byte array.
247 the array can be variable length, and any non-hex-numeric
248 characters are skipped. "0xnn" or "0Xnn" is specially catered
249 for.
251 valid examples: "0A5D15"; "0x15, 0x49, 0xa2"; "59\ta9\te3\n"
255 _PUBLIC_ size_t strhex_to_str(char *p, size_t p_len, const char *strhex, size_t strhex_len);
257 /**
258 * Parse a hex string and return a data blob.
260 _PUBLIC_ _PURE_ DATA_BLOB strhex_to_data_blob(TALLOC_CTX *mem_ctx, const char *strhex) ;
263 * Routine to print a buffer as HEX digits, into an allocated string.
265 _PUBLIC_ void hex_encode(const unsigned char *buff_in, size_t len, char **out_hex_buffer);
268 * talloc version of hex_encode()
270 _PUBLIC_ char *hex_encode_talloc(TALLOC_CTX *mem_ctx, const unsigned char *buff_in, size_t len);
273 Substitute a string for a pattern in another string. Make sure there is
274 enough room!
276 This routine looks for pattern in s and replaces it with
277 insert. It may do multiple replacements.
279 Any of " ; ' $ or ` in the insert string are replaced with _
280 if len==0 then the string cannot be extended. This is different from the old
281 use of len==0 which was for no length checks to be done.
283 _PUBLIC_ void string_sub(char *s,const char *pattern, const char *insert, size_t len);
286 _PUBLIC_ char *string_sub_talloc(TALLOC_CTX *mem_ctx, const char *s,
287 const char *pattern, const char *insert);
290 Similar to string_sub() but allows for any character to be substituted.
291 Use with caution!
292 if len==0 then the string cannot be extended. This is different from the old
293 use of len==0 which was for no length checks to be done.
295 _PUBLIC_ void all_string_sub(char *s,const char *pattern,const char *insert, size_t len);
298 Unescape a URL encoded string, in place.
300 _PUBLIC_ void rfc1738_unescape(char *buf);
303 format a string into length-prefixed dotted domain format, as used in NBT
304 and in some ADS structures
306 _PUBLIC_ const char *str_format_nbt_domain(TALLOC_CTX *mem_ctx, const char *s);
309 * Add a string to an array of strings.
311 * num should be a pointer to an integer that holds the current
312 * number of elements in strings. It will be updated by this function.
314 _PUBLIC_ bool add_string_to_array(TALLOC_CTX *mem_ctx,
315 const char *str, const char ***strings, int *num);
318 varient of strcmp() that handles NULL ptrs
320 _PUBLIC_ int strcmp_safe(const char *s1, const char *s2);
323 return the number of bytes occupied by a buffer in ASCII format
324 the result includes the null termination
325 limited by 'n' bytes
327 _PUBLIC_ size_t ascii_len_n(const char *src, size_t n);
330 Set a boolean variable from the text value stored in the passed string.
331 Returns true in success, false if the passed string does not correctly
332 represent a boolean.
334 _PUBLIC_ bool set_boolean(const char *boolean_string, bool *boolean);
337 * Parse a string containing a boolean value.
339 * val will be set to the read value.
341 * @retval true if a boolean value was parsed, false otherwise.
343 _PUBLIC_ bool conv_str_bool(const char * str, bool * val);
345 #if _SAMBA_BUILD_ == 4
347 * Convert a size specification like 16K into an integral number of bytes.
349 _PUBLIC_ bool conv_str_size(const char * str, uint64_t * val);
350 #endif
353 * Parse a uint64_t value from a string
355 * val will be set to the value read.
357 * @retval true if parsing was successful, false otherwise
359 _PUBLIC_ bool conv_str_u64(const char * str, uint64_t * val);
362 return the number of bytes occupied by a buffer in CH_UTF16 format
363 the result includes the null termination
365 _PUBLIC_ size_t utf16_len(const void *buf);
368 return the number of bytes occupied by a buffer in CH_UTF16 format
369 the result includes the null termination
370 limited by 'n' bytes
372 _PUBLIC_ size_t utf16_len_n(const void *src, size_t n);
373 _PUBLIC_ size_t ucs2_align(const void *base_ptr, const void *p, int flags);
376 Do a case-insensitive, whitespace-ignoring string compare.
378 _PUBLIC_ int strwicmp(const char *psz1, const char *psz2);
381 String replace.
383 _PUBLIC_ void string_replace(char *s, char oldc, char newc);
386 * Compare 2 strings.
388 * @note The comparison is case-insensitive.
390 _PUBLIC_ bool strequal(const char *s1, const char *s2);
392 /* The following definitions come from lib/util/util_strlist.c */
394 /* separators for lists */
395 #ifndef LIST_SEP
396 #define LIST_SEP " \t,\n\r"
397 #endif
400 build a null terminated list of strings from a input string and a
401 separator list. The separator list must contain characters less than
402 or equal to 0x2f for this to work correctly on multi-byte strings
404 _PUBLIC_ char **str_list_make(TALLOC_CTX *mem_ctx, const char *string, const char *sep);
407 * build a null terminated list of strings from an argv-like input string
408 * Entries are seperated by spaces and can be enclosed by quotes.
409 * Does NOT support escaping
411 _PUBLIC_ const char **str_list_make_shell(TALLOC_CTX *mem_ctx, const char *string, const char *sep);
414 * join a list back to one string
416 _PUBLIC_ char *str_list_join(TALLOC_CTX *mem_ctx, const char **list, char seperator);
418 /** join a list back to one (shell-like) string; entries
419 * seperated by spaces, using quotes where necessary */
420 _PUBLIC_ char *str_list_join_shell(TALLOC_CTX *mem_ctx, const char **list, char sep);
423 return the number of elements in a string list
425 _PUBLIC_ size_t str_list_length(const char * const *list);
428 copy a string list
430 _PUBLIC_ char **str_list_copy(TALLOC_CTX *mem_ctx, const char **list);
433 Return true if all the elements of the list match exactly.
435 _PUBLIC_ bool str_list_equal(const char **list1, const char **list2);
438 add an entry to a string list
440 _PUBLIC_ const char **str_list_add(const char **list, const char *s);
443 remove an entry from a string list
445 _PUBLIC_ void str_list_remove(const char **list, const char *s);
448 return true if a string is in a list
450 _PUBLIC_ bool str_list_check(const char **list, const char *s);
453 return true if a string is in a list, case insensitively
455 _PUBLIC_ bool str_list_check_ci(const char **list, const char *s);
457 /* The following definitions come from lib/util/util_file.c */
461 read a line from a file with possible \ continuation chars.
462 Blanks at the start or end of a line are stripped.
463 The string will be allocated if s2 is NULL
465 _PUBLIC_ char *fgets_slash(char *s2,int maxlen,XFILE *f);
468 * Read one line (data until next newline or eof) and allocate it
470 _PUBLIC_ char *afdgets(int fd, TALLOC_CTX *mem_ctx, size_t hint);
473 load a file into memory from a fd.
475 _PUBLIC_ char *fd_load(int fd, size_t *size, size_t maxsize, TALLOC_CTX *mem_ctx);
478 char **file_lines_parse(char *p, size_t size, int *numlines, TALLOC_CTX *mem_ctx);
481 load a file into memory
483 _PUBLIC_ char *file_load(const char *fname, size_t *size, size_t maxsize, TALLOC_CTX *mem_ctx);
486 mmap (if possible) or read a file
488 _PUBLIC_ void *map_file(const char *fname, size_t size);
491 load a file into memory and return an array of pointers to lines in the file
492 must be freed with talloc_free().
494 _PUBLIC_ char **file_lines_load(const char *fname, int *numlines, size_t maxsize, TALLOC_CTX *mem_ctx);
497 load a fd into memory and return an array of pointers to lines in the file
498 must be freed with talloc_free(). If convert is true calls unix_to_dos on
499 the list.
501 _PUBLIC_ char **fd_lines_load(int fd, int *numlines, size_t maxsize, TALLOC_CTX *mem_ctx);
504 take a list of lines and modify them to produce a list where \ continues
505 a line
507 _PUBLIC_ void file_lines_slashcont(char **lines);
510 save a lump of data into a file. Mostly used for debugging
512 _PUBLIC_ bool file_save(const char *fname, const void *packet, size_t length);
513 _PUBLIC_ int vfdprintf(int fd, const char *format, va_list ap) PRINTF_ATTRIBUTE(2,0);
514 _PUBLIC_ int fdprintf(int fd, const char *format, ...) PRINTF_ATTRIBUTE(2,3);
515 _PUBLIC_ bool large_file_support(const char *path);
517 /* The following definitions come from lib/util/util.c */
521 Find a suitable temporary directory. The result should be copied immediately
522 as it may be overwritten by a subsequent call.
524 _PUBLIC_ const char *tmpdir(void);
527 Check if a file exists - call vfs_file_exist for samba files.
529 _PUBLIC_ bool file_exist(const char *fname);
532 Check a files mod time.
534 _PUBLIC_ time_t file_modtime(const char *fname);
537 Check if a directory exists.
539 _PUBLIC_ bool directory_exist(const char *dname);
542 * Try to create the specified directory if it didn't exist.
544 * @retval true if the directory already existed and has the right permissions
545 * or was successfully created.
547 _PUBLIC_ bool directory_create_or_exist(const char *dname, uid_t uid,
548 mode_t dir_perms);
551 Set a fd into blocking/nonblocking mode. Uses POSIX O_NONBLOCK if available,
552 else
553 if SYSV use O_NDELAY
554 if BSD use FNDELAY
556 _PUBLIC_ int set_blocking(int fd, bool set);
559 Sleep for a specified number of milliseconds.
561 _PUBLIC_ void msleep(unsigned int t);
564 Get my own name, return in talloc'ed storage.
566 _PUBLIC_ char* get_myname(TALLOC_CTX *mem_ctx);
569 Return true if a string could be a pure IP address.
571 _PUBLIC_ bool is_ipaddress(const char *str);
574 Interpret an internet address or name into an IP address in 4 byte form.
576 _PUBLIC_ uint32_t interpret_addr(const char *str);
579 A convenient addition to interpret_addr().
581 _PUBLIC_ struct in_addr interpret_addr2(const char *str);
584 Check if an IP is the 0.0.0.0.
586 _PUBLIC_ bool is_zero_ip_v4(struct in_addr ip);
589 Are two IPs on the same subnet?
591 _PUBLIC_ bool same_net_v4(struct in_addr ip1,struct in_addr ip2,struct in_addr mask);
593 _PUBLIC_ bool is_ipaddress_v4(const char *str);
596 Check if a process exists. Does this work on all unixes?
598 _PUBLIC_ bool process_exists_by_pid(pid_t pid);
601 Simple routine to do POSIX file locking. Cruft in NFS and 64->32 bit mapping
602 is dealt with in posix.c
604 _PUBLIC_ bool fcntl_lock(int fd, int op, off_t offset, off_t count, int type);
607 * Write dump of binary data to the log file.
609 * The data is only written if the log level is at least level.
611 _PUBLIC_ void dump_data(int level, const uint8_t *buf,int len);
614 * Write dump of binary data to the log file.
616 * The data is only written if the log level is at least level.
617 * 16 zero bytes in a row are ommited
619 _PUBLIC_ void dump_data_skip_zeros(int level, const uint8_t *buf, int len);
622 malloc that aborts with smb_panic on fail or zero size.
624 _PUBLIC_ void *smb_xmalloc(size_t size);
627 Memdup with smb_panic on fail.
629 _PUBLIC_ void *smb_xmemdup(const void *p, size_t size);
632 strdup that aborts on malloc fail.
634 _PUBLIC_ char *smb_xstrdup(const char *s);
636 char *smb_xstrndup(const char *s, size_t n);
639 Like strdup but for memory.
641 _PUBLIC_ void *memdup(const void *p, size_t size);
644 * Write a password to the log file.
646 * @note Only actually does something if DEBUG_PASSWORD was defined during
647 * compile-time.
649 _PUBLIC_ void dump_data_pw(const char *msg, const uint8_t * data, size_t len);
652 * see if a range of memory is all zero. A NULL pointer is considered
653 * to be all zero
655 _PUBLIC_ bool all_zero(const uint8_t *ptr, size_t size);
658 realloc an array, checking for integer overflow in the array size
660 _PUBLIC_ void *realloc_array(void *ptr, size_t el_size, unsigned count, bool free_on_fail);
662 void *malloc_array(size_t el_size, unsigned int count);
664 /* The following definitions come from lib/util/fsusage.c */
668 * Retrieve amount of free disk space.
669 * this does all of the system specific guff to get the free disk space.
670 * It is derived from code in the GNU fileutils package, but has been
671 * considerably mangled for use here
673 * results are returned in *dfree and *dsize, in 512 byte units
675 _PUBLIC_ int sys_fsusage(const char *path, uint64_t *dfree, uint64_t *dsize);
677 /* The following definitions come from lib/util/ms_fnmatch.c */
681 * @file
682 * @brief MS-style Filename matching
685 #if _SAMBA_BUILD_ == 4
686 /* protocol types. It assumes that higher protocols include lower protocols
687 as subsets. FIXME: Move to one of the smb-specific headers */
688 enum protocol_types {
689 PROTOCOL_NONE,
690 PROTOCOL_CORE,
691 PROTOCOL_COREPLUS,
692 PROTOCOL_LANMAN1,
693 PROTOCOL_LANMAN2,
694 PROTOCOL_NT1,
695 PROTOCOL_SMB2
698 int ms_fnmatch(const char *pattern, const char *string, enum protocol_types protocol);
700 /** a generic fnmatch function - uses for non-CIFS pattern matching */
701 int gen_fnmatch(const char *pattern, const char *string);
702 #endif
704 /* The following definitions come from lib/util/mutex.c */
708 register a set of mutex/rwlock handlers.
709 Should only be called once in the execution of smbd.
711 _PUBLIC_ bool register_mutex_handlers(const char *name, struct mutex_ops *ops);
713 /* The following definitions come from lib/util/idtree.c */
717 initialise a idr tree. The context return value must be passed to
718 all subsequent idr calls. To destroy the idr tree use talloc_free()
719 on this context
721 _PUBLIC_ struct idr_context *idr_init(TALLOC_CTX *mem_ctx);
724 allocate the next available id, and assign 'ptr' into its slot.
725 you can retrieve later this pointer using idr_find()
727 _PUBLIC_ int idr_get_new(struct idr_context *idp, void *ptr, int limit);
730 allocate a new id, giving the first available value greater than or
731 equal to the given starting id
733 _PUBLIC_ int idr_get_new_above(struct idr_context *idp, void *ptr, int starting_id, int limit);
736 allocate a new id randomly in the given range
738 _PUBLIC_ int idr_get_new_random(struct idr_context *idp, void *ptr, int limit);
741 find a pointer value previously set with idr_get_new given an id
743 _PUBLIC_ void *idr_find(struct idr_context *idp, int id);
746 remove an id from the idr tree
748 _PUBLIC_ int idr_remove(struct idr_context *idp, int id);
750 /* The following definitions come from lib/util/become_daemon.c */
753 Close the low 3 fd's and open dev/null in their place
755 _PUBLIC_ void close_low_fds(bool stderr_too);
758 Become a daemon, discarding the controlling terminal.
760 _PUBLIC_ void become_daemon(bool do_fork, bool no_process_group);
763 * Load a ini-style file.
765 bool pm_process( const char *fileName,
766 bool (*sfunc)(const char *, void *),
767 bool (*pfunc)(const char *, const char *, void *),
768 void *userdata);
770 bool unmap_file(void *start, size_t size);
772 void print_asc(int level, const uint8_t *buf,int len);
774 #endif /* _SAMBA_UTIL_H_ */