util.h
上传用户:awang829
上传日期:2019-07-14
资源大小:2356k
文件大小:12k
源码类别:

网络

开发平台:

Unix_Linux

  1. /* Copyright (c) 2003-2004, Roger Dingledine
  2.  * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
  3.  * Copyright (c) 2007-2009, The Tor Project, Inc. */
  4. /* See LICENSE for licensing information */
  5. /**
  6.  * file util.h
  7.  * brief Headers for util.c
  8.  **/
  9. #ifndef _TOR_UTIL_H
  10. #define _TOR_UTIL_H
  11. #include "orconfig.h"
  12. #include "torint.h"
  13. #include "compat.h"
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #ifndef O_BINARY
  17. #define O_BINARY 0
  18. #endif
  19. #ifndef O_TEXT
  20. #define O_TEXT 0
  21. #endif
  22. /* Replace assert() with a variant that sends failures to the log before
  23.  * calling assert() normally.
  24.  */
  25. #ifdef NDEBUG
  26. /* Nobody should ever want to build with NDEBUG set.  99% of our asserts will
  27.  * be outside the critical path anyway, so it's silly to disable bug-checking
  28.  * throughout the entire program just because a few asserts are slowing you
  29.  * down.  Profile, optimize the critical path, and keep debugging on.
  30.  *
  31.  * And I'm not just saying that because some of our asserts check
  32.  * security-critical properties.
  33.  */
  34. #error "Sorry; we don't support building with NDEBUG."
  35. #endif
  36. /** Like assert(3), but send assertion failures to the log as well as to
  37.  * stderr. */
  38. #define tor_assert(expr) STMT_BEGIN                                     
  39.     if (PREDICT_UNLIKELY(!(expr))) {                                    
  40.       log(LOG_ERR, LD_BUG, "%s:%d: %s: Assertion %s failed; aborting.", 
  41.           _SHORT_FILE_, __LINE__, __func__, #expr);                     
  42.       fprintf(stderr,"%s:%d %s: Assertion %s failed; aborting.n",      
  43.               _SHORT_FILE_, __LINE__, __func__, #expr);                 
  44.       abort();                                                          
  45.     } STMT_END
  46. /* If we're building with dmalloc, we want all of our memory allocation
  47.  * functions to take an extra file/line pair of arguments.  If not, not.
  48.  * We define DMALLOC_PARAMS to the extra parameters to insert in the
  49.  * function prototypes, and DMALLOC_ARGS to the extra arguments to add
  50.  * to calls. */
  51. #ifdef USE_DMALLOC
  52. #define DMALLOC_PARAMS , const char *file, const int line
  53. #define DMALLOC_ARGS , _SHORT_FILE_, __LINE__
  54. #else
  55. #define DMALLOC_PARAMS
  56. #define DMALLOC_ARGS
  57. #endif
  58. /** Define this if you want Tor to crash when any problem comes up,
  59.  * so you can get a coredump and track things down. */
  60. // #define tor_fragile_assert() tor_assert(0)
  61. #define tor_fragile_assert()
  62. /* Memory management */
  63. void *_tor_malloc(size_t size DMALLOC_PARAMS) ATTR_MALLOC;
  64. void *_tor_malloc_zero(size_t size DMALLOC_PARAMS) ATTR_MALLOC;
  65. void *_tor_malloc_roundup(size_t *size DMALLOC_PARAMS) ATTR_MALLOC;
  66. void *_tor_realloc(void *ptr, size_t size DMALLOC_PARAMS);
  67. char *_tor_strdup(const char *s DMALLOC_PARAMS) ATTR_MALLOC ATTR_NONNULL((1));
  68. char *_tor_strndup(const char *s, size_t n DMALLOC_PARAMS)
  69.   ATTR_MALLOC ATTR_NONNULL((1));
  70. void *_tor_memdup(const void *mem, size_t len DMALLOC_PARAMS)
  71.   ATTR_MALLOC ATTR_NONNULL((1));
  72. void _tor_free(void *mem);
  73. #ifdef USE_DMALLOC
  74. extern int dmalloc_free(const char *file, const int line, void *pnt,
  75.                         const int func_id);
  76. #define tor_free(p) STMT_BEGIN 
  77.     if (PREDICT_LIKELY((p)!=NULL)) {                
  78.       dmalloc_free(_SHORT_FILE_, __LINE__, (p), 0); 
  79.       (p)=NULL;                                     
  80.     }                                               
  81.   STMT_END
  82. #else
  83. /** Release memory allocated by tor_malloc, tor_realloc, tor_strdup, etc.
  84.  * Unlike the free() function, tor_free() will still work on NULL pointers,
  85.  * and it sets the pointer value to NULL after freeing it.
  86.  *
  87.  * This is a macro.  If you need a function pointer to release memory from
  88.  * tor_malloc(), use _tor_free().
  89.  */
  90. #define tor_free(p) STMT_BEGIN                                 
  91.     if (PREDICT_LIKELY((p)!=NULL)) {                           
  92.       free(p);                                                 
  93.       (p)=NULL;                                                
  94.     }                                                          
  95.   STMT_END
  96. #endif
  97. #define tor_malloc(size)       _tor_malloc(size DMALLOC_ARGS)
  98. #define tor_malloc_zero(size)  _tor_malloc_zero(size DMALLOC_ARGS)
  99. #define tor_malloc_roundup(szp) _tor_malloc_roundup(szp DMALLOC_ARGS)
  100. #define tor_realloc(ptr, size) _tor_realloc(ptr, size DMALLOC_ARGS)
  101. #define tor_strdup(s)          _tor_strdup(s DMALLOC_ARGS)
  102. #define tor_strndup(s, n)      _tor_strndup(s, n DMALLOC_ARGS)
  103. #define tor_memdup(s, n)       _tor_memdup(s, n DMALLOC_ARGS)
  104. void tor_log_mallinfo(int severity);
  105. /** Return the offset of <b>member</b> within the type <b>tp</b>, in bytes */
  106. #if defined(__GNUC__) && __GNUC__ > 3
  107. #define STRUCT_OFFSET(tp, member) __builtin_offsetof(tp, member)
  108. #else
  109.  #define STRUCT_OFFSET(tp, member) 
  110.    ((off_t) (((char*)&((tp*)0)->member)-(char*)0))
  111. #endif
  112. /** Macro: yield a pointer to the field at position <b>off</b> within the
  113.  * structure <b>st</b>.  Example:
  114.  * <pre>
  115.  *   struct a { int foo; int bar; } x;
  116.  *   off_t bar_offset = STRUCT_OFFSET(struct a, bar);
  117.  *   int *bar_p = STRUCT_VAR_P(&x, bar_offset);
  118.  *   *bar_p = 3;
  119.  * </pre>
  120.  */
  121. #define STRUCT_VAR_P(st, off) ((void*) ( ((char*)(st)) + (off) ) )
  122. /** Macro: yield a pointer to an enclosing structure given a pointer to
  123.  * a substructure at offset <b>off</b>. Example:
  124.  * <pre>
  125.  *   struct base { ... };
  126.  *   struct subtype { int x; struct base b; } x;
  127.  *   struct base *bp = &x.base;
  128.  *   struct *sp = SUBTYPE_P(bp, struct subtype, b);
  129.  * </pre>
  130.  */
  131. #define SUBTYPE_P(p, subtype, basemember) 
  132.   ((void*) ( ((char*)(p)) - STRUCT_OFFSET(subtype, basemember) ))
  133. /* Logic */
  134. /** Macro: true if two values have the same boolean value. */
  135. #define bool_eq(a,b) (!(a)==!(b))
  136. /** Macro: true if two values have different boolean values. */
  137. #define bool_neq(a,b) (!(a)!=!(b))
  138. /* Math functions */
  139. int tor_log2(uint64_t u64) ATTR_CONST;
  140. uint64_t round_to_power_of_2(uint64_t u64);
  141. /* String manipulation */
  142. /** Allowable characters in a hexadecimal string. */
  143. #define HEX_CHARACTERS "0123456789ABCDEFabcdef"
  144. void tor_strlower(char *s) ATTR_NONNULL((1));
  145. void tor_strupper(char *s) ATTR_NONNULL((1));
  146. int tor_strisprint(const char *s) ATTR_PURE ATTR_NONNULL((1));
  147. int tor_strisnonupper(const char *s) ATTR_PURE ATTR_NONNULL((1));
  148. int strcmpstart(const char *s1, const char *s2) ATTR_PURE ATTR_NONNULL((1,2));
  149. int strcmp_len(const char *s1, const char *s2, size_t len)
  150.   ATTR_PURE ATTR_NONNULL((1,2));
  151. int strcasecmpstart(const char *s1, const char *s2)
  152.   ATTR_PURE ATTR_NONNULL((1,2));
  153. int strcmpend(const char *s1, const char *s2) ATTR_PURE ATTR_NONNULL((1,2));
  154. int strcasecmpend(const char *s1, const char *s2)
  155.   ATTR_PURE ATTR_NONNULL((1,2));
  156. int memcmpstart(const void *mem, size_t memlen,
  157.                 const char *prefix) ATTR_PURE;
  158. void tor_strstrip(char *s, const char *strip) ATTR_NONNULL((1,2));
  159. long tor_parse_long(const char *s, int base, long min,
  160.                     long max, int *ok, char **next);
  161. unsigned long tor_parse_ulong(const char *s, int base, unsigned long min,
  162.                               unsigned long max, int *ok, char **next);
  163. uint64_t tor_parse_uint64(const char *s, int base, uint64_t min,
  164.                          uint64_t max, int *ok, char **next);
  165. const char *hex_str(const char *from, size_t fromlen) ATTR_NONNULL((1));
  166. const char *eat_whitespace(const char *s) ATTR_PURE;
  167. const char *eat_whitespace_eos(const char *s, const char *eos) ATTR_PURE;
  168. const char *eat_whitespace_no_nl(const char *s) ATTR_PURE;
  169. const char *eat_whitespace_eos_no_nl(const char *s, const char *eos) ATTR_PURE;
  170. const char *find_whitespace(const char *s) ATTR_PURE;
  171. const char *find_whitespace_eos(const char *s, const char *eos) ATTR_PURE;
  172. int tor_mem_is_zero(const char *mem, size_t len) ATTR_PURE;
  173. int tor_digest_is_zero(const char *digest) ATTR_PURE;
  174. char *esc_for_log(const char *string) ATTR_MALLOC;
  175. const char *escaped(const char *string);
  176. struct smartlist_t;
  177. void wrap_string(struct smartlist_t *out, const char *string, size_t width,
  178.                  const char *prefix0, const char *prefixRest);
  179. int tor_vsscanf(const char *buf, const char *pattern, va_list ap);
  180. int tor_sscanf(const char *buf, const char *pattern, ...)
  181. #ifdef __GNUC__
  182.   __attribute__((format(scanf, 2, 3)))
  183. #endif
  184.   ;
  185. int hex_decode_digit(char c);
  186. void base16_encode(char *dest, size_t destlen, const char *src, size_t srclen);
  187. int base16_decode(char *dest, size_t destlen, const char *src, size_t srclen);
  188. /* Time helpers */
  189. long tv_udiff(const struct timeval *start, const struct timeval *end);
  190. time_t tor_timegm(struct tm *tm);
  191. #define RFC1123_TIME_LEN 29
  192. void format_rfc1123_time(char *buf, time_t t);
  193. int parse_rfc1123_time(const char *buf, time_t *t);
  194. #define ISO_TIME_LEN 19
  195. void format_local_iso_time(char *buf, time_t t);
  196. void format_iso_time(char *buf, time_t t);
  197. int parse_iso_time(const char *buf, time_t *t);
  198. int parse_http_time(const char *buf, struct tm *tm);
  199. int format_time_interval(char *out, size_t out_len, long interval);
  200. /* Cached time */
  201. #ifdef TIME_IS_FAST
  202. #define approx_time() time(NULL)
  203. #define update_approx_time(t) STMT_NIL
  204. #else
  205. time_t approx_time(void);
  206. void update_approx_time(time_t now);
  207. #endif
  208. /* Fuzzy time. */
  209. void ftime_set_maximum_sloppiness(int seconds);
  210. void ftime_set_estimated_skew(int seconds);
  211. /* typedef struct ftime_t { time_t earliest; time_t latest; } ftime_t; */
  212. /* void ftime_get_window(time_t now, ftime_t *ft_out); */
  213. int ftime_maybe_after(time_t now, time_t when);
  214. int ftime_maybe_before(time_t now, time_t when);
  215. int ftime_definitely_after(time_t now, time_t when);
  216. int ftime_definitely_before(time_t now, time_t when);
  217. /* File helpers */
  218. ssize_t write_all(int fd, const char *buf, size_t count, int isSocket);
  219. ssize_t read_all(int fd, char *buf, size_t count, int isSocket);
  220. /** Return values from file_status(); see that function's documentation
  221.  * for details. */
  222. typedef enum { FN_ERROR, FN_NOENT, FN_FILE, FN_DIR } file_status_t;
  223. file_status_t file_status(const char *filename);
  224. /** Possible behaviors for check_private_dir() on encountering a nonexistent
  225.  * directory; see that function's documentation for details. */
  226. typedef enum { CPD_NONE, CPD_CREATE, CPD_CHECK } cpd_check_t;
  227. int check_private_dir(const char *dirname, cpd_check_t check);
  228. #define OPEN_FLAGS_REPLACE (O_WRONLY|O_CREAT|O_TRUNC)
  229. #define OPEN_FLAGS_APPEND (O_WRONLY|O_CREAT|O_APPEND)
  230. typedef struct open_file_t open_file_t;
  231. int start_writing_to_file(const char *fname, int open_flags, int mode,
  232.                           open_file_t **data_out);
  233. FILE *start_writing_to_stdio_file(const char *fname, int open_flags, int mode,
  234.                                   open_file_t **data_out);
  235. FILE *fdopen_file(open_file_t *file_data);
  236. int finish_writing_to_file(open_file_t *file_data);
  237. int abort_writing_to_file(open_file_t *file_data);
  238. int write_str_to_file(const char *fname, const char *str, int bin);
  239. int write_bytes_to_file(const char *fname, const char *str, size_t len,
  240.                         int bin);
  241. /** An ad-hoc type to hold a string of characters and a count; used by
  242.  * write_chunks_to_file. */
  243. typedef struct sized_chunk_t {
  244.   const char *bytes;
  245.   size_t len;
  246. } sized_chunk_t;
  247. int write_chunks_to_file(const char *fname, const struct smartlist_t *chunks,
  248.                          int bin);
  249. int append_bytes_to_file(const char *fname, const char *str, size_t len,
  250.                          int bin);
  251. /** Flag for read_file_to_str: open the file in binary mode. */
  252. #define RFTS_BIN            1
  253. /** Flag for read_file_to_str: it's okay if the file doesn't exist. */
  254. #define RFTS_IGNORE_MISSING 2
  255. struct stat;
  256. char *read_file_to_str(const char *filename, int flags, struct stat *stat_out)
  257.   ATTR_MALLOC;
  258. const char *parse_config_line_from_str(const char *line,
  259.                                        char **key_out, char **value_out);
  260. char *expand_filename(const char *filename);
  261. struct smartlist_t *tor_listdir(const char *dirname);
  262. int path_is_relative(const char *filename) ATTR_PURE;
  263. /* Process helpers */
  264. void start_daemon(void);
  265. void finish_daemon(const char *desired_cwd);
  266. void write_pidfile(char *filename);
  267. #endif