]> git.ipfire.org Git - people/ms/strongswan.git/blob - src/libstrongswan/utils/utils.h
c42f88168700670f819aec2befadb5deeb0e3d97
[people/ms/strongswan.git] / src / libstrongswan / utils / utils.h
1 /*
2 * Copyright (C) 2008-2014 Tobias Brunner
3 * Copyright (C) 2008 Martin Willi
4 * Hochschule fuer Technik Rapperswil
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; either version 2 of the License, or (at your
9 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
10 *
11 * This program is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
13 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * for more details.
15 */
16
17 /**
18 * @defgroup utils_i utils
19 * @{ @ingroup utils
20 */
21
22 #ifndef UTILS_H_
23 #define UTILS_H_
24
25 #include <sys/types.h>
26 #include <stdlib.h>
27 #include <stddef.h>
28 #include <sys/time.h>
29 #include <string.h>
30
31 #ifdef WIN32
32 # include "compat/windows.h"
33 #else
34 # define _GNU_SOURCE
35 # include <arpa/inet.h>
36 # include <sys/socket.h>
37 # include <netdb.h>
38 # include <netinet/in.h>
39 # include <sched.h>
40 # include <poll.h>
41 #endif
42
43 /**
44 * strongSwan program return codes
45 */
46 #define SS_RC_LIBSTRONGSWAN_INTEGRITY 64
47 #define SS_RC_DAEMON_INTEGRITY 65
48 #define SS_RC_INITIALIZATION_FAILED 66
49
50 #define SS_RC_FIRST SS_RC_LIBSTRONGSWAN_INTEGRITY
51 #define SS_RC_LAST SS_RC_INITIALIZATION_FAILED
52
53 /**
54 * Number of bits in a byte
55 */
56 #define BITS_PER_BYTE 8
57
58 /**
59 * Default length for various auxiliary text buffers
60 */
61 #define BUF_LEN 512
62
63 /**
64 * Build assertion macro for integer expressions, evaluates to 0
65 */
66 #define BUILD_ASSERT(x) (sizeof(char[(x) ? 0 : -1]))
67
68 /**
69 * Build time check to assert a is an array, evaluates to 0
70 *
71 * The address of an array element has a pointer type, which is not compatible
72 * to the array type.
73 */
74 #define BUILD_ASSERT_ARRAY(a) \
75 BUILD_ASSERT(!__builtin_types_compatible_p(typeof(a), typeof(&(a)[0])))
76
77 /**
78 * General purpose boolean type.
79 */
80 #ifdef HAVE_STDBOOL_H
81 # include <stdbool.h>
82 #else
83 # ifndef HAVE__BOOL
84 # define _Bool signed char
85 # endif /* HAVE__BOOL */
86 # define bool _Bool
87 # define false 0
88 # define true 1
89 # define __bool_true_false_are_defined 1
90 #endif /* HAVE_STDBOOL_H */
91 #ifndef FALSE
92 # define FALSE false
93 #endif /* FALSE */
94 #ifndef TRUE
95 # define TRUE true
96 #endif /* TRUE */
97
98 #include "enum.h"
99 #include "utils/strerror.h"
100 #ifdef __APPLE__
101 # include "compat/apple.h"
102 #endif
103
104 /**
105 * Directory separator character in paths on this platform
106 */
107 #ifdef WIN32
108 # define DIRECTORY_SEPARATOR "\\"
109 #else
110 # define DIRECTORY_SEPARATOR "/"
111 #endif
112
113 /**
114 * Initialize utility functions
115 */
116 void utils_init();
117
118 /**
119 * Deinitialize utility functions
120 */
121 void utils_deinit();
122
123 /**
124 * Helper function that compares two strings for equality
125 */
126 static inline bool streq(const char *x, const char *y)
127 {
128 return strcmp(x, y) == 0;
129 }
130
131 /**
132 * Helper function that compares two strings for equality, length limited
133 */
134 static inline bool strneq(const char *x, const char *y, size_t len)
135 {
136 return strncmp(x, y, len) == 0;
137 }
138
139 /**
140 * Helper function that checks if a string starts with a given prefix
141 */
142 static inline bool strpfx(const char *x, const char *prefix)
143 {
144 return strneq(x, prefix, strlen(prefix));
145 }
146
147 /**
148 * Helper function that compares two strings for equality ignoring case
149 */
150 static inline bool strcaseeq(const char *x, const char *y)
151 {
152 return strcasecmp(x, y) == 0;
153 }
154
155 /**
156 * Helper function that compares two strings for equality ignoring case, length limited
157 */
158 static inline bool strncaseeq(const char *x, const char *y, size_t len)
159 {
160 return strncasecmp(x, y, len) == 0;
161 }
162
163 /**
164 * Helper function that checks if a string starts with a given prefix
165 */
166 static inline bool strcasepfx(const char *x, const char *prefix)
167 {
168 return strncaseeq(x, prefix, strlen(prefix));
169 }
170
171 /**
172 * NULL-safe strdup variant
173 */
174 static inline char *strdupnull(const char *s)
175 {
176 return s ? strdup(s) : NULL;
177 }
178
179 /**
180 * Helper function that compares two binary blobs for equality
181 */
182 static inline bool memeq(const void *x, const void *y, size_t len)
183 {
184 return memcmp(x, y, len) == 0;
185 }
186
187 /**
188 * Same as memeq(), but with a constant runtime, safe for cryptographic use.
189 */
190 bool memeq_const(const void *x, const void *y, size_t len);
191
192 /**
193 * Calling memcpy() with NULL pointers, even with n == 0, results in undefined
194 * behavior according to the C standard. This version is guaranteed to not
195 * access the pointers if n is 0.
196 */
197 static inline void *memcpy_noop(void *dst, const void *src, size_t n)
198 {
199 return n ? memcpy(dst, src, n) : dst;
200 }
201 #ifdef memcpy
202 # undef memcpy
203 #endif
204 #define memcpy(d,s,n) memcpy_noop(d,s,n)
205
206 /**
207 * Calling memmove() with NULL pointers, even with n == 0, results in undefined
208 * behavior according to the C standard. This version is guaranteed to not
209 * access the pointers if n is 0.
210 */
211 static inline void *memmove_noop(void *dst, const void *src, size_t n)
212 {
213 return n ? memmove(dst, src, n) : dst;
214 }
215 #ifdef memmove
216 # undef memmove
217 #endif
218 #define memmove(d,s,n) memmove_noop(d,s,n)
219
220 /**
221 * Calling memset() with a NULL pointer, even with n == 0, results in undefined
222 * behavior according to the C standard. This version is guaranteed to not
223 * access the pointer if n is 0.
224 */
225 static inline void *memset_noop(void *s, int c, size_t n)
226 {
227 return n ? memset(s, c, n) : s;
228 }
229 #ifdef memset
230 # undef memset
231 #endif
232 #define memset(s,c,n) memset_noop(s,c,n)
233
234 /**
235 * Macro gives back larger of two values.
236 */
237 #define max(x,y) ({ \
238 typeof(x) _x = (x); \
239 typeof(y) _y = (y); \
240 _x > _y ? _x : _y; })
241
242 /**
243 * Macro gives back smaller of two values.
244 */
245 #define min(x,y) ({ \
246 typeof(x) _x = (x); \
247 typeof(y) _y = (y); \
248 _x < _y ? _x : _y; })
249
250 /**
251 * Call destructor of an object, if object != NULL
252 */
253 #define DESTROY_IF(obj) if (obj) (obj)->destroy(obj)
254
255 /**
256 * Call offset destructor of an object, if object != NULL
257 */
258 #define DESTROY_OFFSET_IF(obj, offset) if (obj) obj->destroy_offset(obj, offset);
259
260 /**
261 * Call function destructor of an object, if object != NULL
262 */
263 #define DESTROY_FUNCTION_IF(obj, fn) if (obj) obj->destroy_function(obj, fn);
264
265 /**
266 * Debug macro to follow control flow
267 */
268 #define POS printf("%s, line %d\n", __FILE__, __LINE__)
269
270 /**
271 * Object allocation/initialization macro, using designated initializer.
272 */
273 #define INIT(this, ...) { (this) = malloc(sizeof(*(this))); \
274 *(this) = (typeof(*(this))){ __VA_ARGS__ }; }
275
276 /**
277 * Aligning version of INIT().
278 *
279 * The returned pointer must be freed using free_align(), not free().
280 *
281 * @param this object to allocate/initialize
282 * @param align alignment for allocation, in bytes
283 * @param ... initializer
284 */
285 #define INIT_ALIGN(this, align, ...) { \
286 (this) = malloc_align(sizeof(*(this)), align); \
287 *(this) = (typeof(*(this))){ __VA_ARGS__ }; }
288
289 /**
290 * Object allocation/initialization macro, with extra allocated bytes at tail.
291 *
292 * The extra space gets zero-initialized.
293 *
294 * @param this pointer to object to allocate memory for
295 * @param extra number of bytes to allocate at end of this
296 * @param ... initializer
297 */
298 #define INIT_EXTRA(this, extra, ...) { \
299 typeof(extra) _extra = (extra); \
300 (this) = malloc(sizeof(*(this)) + _extra); \
301 *(this) = (typeof(*(this))){ __VA_ARGS__ }; \
302 memset((this) + 1, 0, _extra); }
303
304 /**
305 * Aligning version of INIT_EXTRA().
306 *
307 * The returned pointer must be freed using free_align(), not free().
308 *
309 * @param this object to allocate/initialize
310 * @param extra number of bytes to allocate at end of this
311 * @param align alignment for allocation, in bytes
312 * @param ... initializer
313 */
314 #define INIT_EXTRA_ALIGN(this, extra, align, ...) { \
315 typeof(extra) _extra = (extra); \
316 (this) = malloc_align(sizeof(*(this)) + _extra, align); \
317 *(this) = (typeof(*(this))){ __VA_ARGS__ }; \
318 memset((this) + 1, 0, _extra); }
319
320 /**
321 * Method declaration/definition macro, providing private and public interface.
322 *
323 * Defines a method name with this as first parameter and a return value ret,
324 * and an alias for this method with a _ prefix, having the this argument
325 * safely casted to the public interface iface.
326 * _name is provided a function pointer, but will get optimized out by GCC.
327 */
328 #define METHOD(iface, name, ret, this, ...) \
329 static ret name(union {iface *_public; this;} \
330 __attribute__((transparent_union)), ##__VA_ARGS__); \
331 static typeof(name) *_##name = (typeof(name)*)name; \
332 static ret name(this, ##__VA_ARGS__)
333
334 /**
335 * Same as METHOD(), but is defined for two public interfaces.
336 */
337 #define METHOD2(iface1, iface2, name, ret, this, ...) \
338 static ret name(union {iface1 *_public1; iface2 *_public2; this;} \
339 __attribute__((transparent_union)), ##__VA_ARGS__); \
340 static typeof(name) *_##name = (typeof(name)*)name; \
341 static ret name(this, ##__VA_ARGS__)
342
343 /**
344 * Callback declaration/definition macro, allowing casted first parameter.
345 *
346 * This is very similar to METHOD, but instead of casting the first parameter
347 * to a public interface, it uses a void*. This allows type safe definition
348 * of a callback function, while using the real type for the first parameter.
349 */
350 #define CALLBACK(name, ret, param1, ...) \
351 static ret _cb_##name(union {void *_generic; param1;} \
352 __attribute__((transparent_union)), ##__VA_ARGS__); \
353 static typeof(_cb_##name) *name = (typeof(_cb_##name)*)_cb_##name; \
354 static ret _cb_##name(param1, ##__VA_ARGS__)
355
356 /**
357 * This macro allows counting the number of arguments passed to a macro.
358 * Combined with the VA_ARGS_DISPATCH() macro this can be used to implement
359 * macro overloading based on the number of arguments.
360 * 0 to 10 arguments are currently supported.
361 */
362 #define VA_ARGS_NUM(...) _VA_ARGS_NUM(0,##__VA_ARGS__,10,9,8,7,6,5,4,3,2,1,0)
363 #define _VA_ARGS_NUM(_0,_1,_2,_3,_4,_5,_6,_7,_8,_9,_10,NUM,...) NUM
364
365 /**
366 * This macro can be used to dispatch a macro call based on the number of given
367 * arguments, for instance:
368 *
369 * @code
370 * #define MY_MACRO(...) VA_ARGS_DISPATCH(MY_MACRO, __VA_ARGS__)(__VA_ARGS__)
371 * #define MY_MACRO1(arg) one_arg(arg)
372 * #define MY_MACRO2(arg1,arg2) two_args(arg1,arg2)
373 * @endcode
374 *
375 * MY_MACRO() can now be called with either one or two arguments, which will
376 * resolve to one_arg(arg) or two_args(arg1,arg2), respectively.
377 */
378 #define VA_ARGS_DISPATCH(func, ...) _VA_ARGS_DISPATCH(func, VA_ARGS_NUM(__VA_ARGS__))
379 #define _VA_ARGS_DISPATCH(func, num) __VA_ARGS_DISPATCH(func, num)
380 #define __VA_ARGS_DISPATCH(func, num) func ## num
381
382 /**
383 * Architecture independent bitfield definition helpers (at least with GCC).
384 *
385 * Defines a bitfield with a type t and a fixed size of bitfield members, e.g.:
386 * BITFIELD2(u_int8_t,
387 * low: 4,
388 * high: 4,
389 * ) flags;
390 * The member defined first placed at bit 0.
391 */
392 #if BYTE_ORDER == LITTLE_ENDIAN
393 #define BITFIELD2(t, a, b,...) struct { t a; t b; __VA_ARGS__}
394 #define BITFIELD3(t, a, b, c,...) struct { t a; t b; t c; __VA_ARGS__}
395 #define BITFIELD4(t, a, b, c, d,...) struct { t a; t b; t c; t d; __VA_ARGS__}
396 #define BITFIELD5(t, a, b, c, d, e,...) struct { t a; t b; t c; t d; t e; __VA_ARGS__}
397 #elif BYTE_ORDER == BIG_ENDIAN
398 #define BITFIELD2(t, a, b,...) struct { t b; t a; __VA_ARGS__}
399 #define BITFIELD3(t, a, b, c,...) struct { t c; t b; t a; __VA_ARGS__}
400 #define BITFIELD4(t, a, b, c, d,...) struct { t d; t c; t b; t a; __VA_ARGS__}
401 #define BITFIELD5(t, a, b, c, d, e,...) struct { t e; t d; t c; t b; t a; __VA_ARGS__}
402 #endif
403
404 /**
405 * Macro to allocate a sized type.
406 */
407 #define malloc_thing(thing) ((thing*)malloc(sizeof(thing)))
408
409 /**
410 * Get the number of elements in an array
411 */
412 #define countof(array) (sizeof(array)/sizeof((array)[0]) \
413 + BUILD_ASSERT_ARRAY(array))
414
415 /**
416 * Ignore result of functions tagged with warn_unused_result attributes
417 */
418 #define ignore_result(call) { if(call){}; }
419
420 /**
421 * Assign a function as a class method
422 */
423 #define ASSIGN(method, function) (method = (typeof(method))function)
424
425 /**
426 * time_t not defined
427 */
428 #define UNDEFINED_TIME 0
429
430 /**
431 * Maximum time since epoch causing wrap-around on Jan 19 03:14:07 UTC 2038
432 */
433 #define TIME_32_BIT_SIGNED_MAX 0x7fffffff
434
435 /**
436 * define some missing fixed width int types on OpenSolaris.
437 * TODO: since the uintXX_t types are defined by the C99 standard we should
438 * probably use those anyway
439 */
440 #if defined __sun || defined WIN32
441 #include <stdint.h>
442 typedef uint8_t u_int8_t;
443 typedef uint16_t u_int16_t;
444 typedef uint32_t u_int32_t;
445 typedef uint64_t u_int64_t;
446 #endif
447
448 #ifdef HAVE_INT128
449 /**
450 * 128 bit wide signed integer, if supported
451 */
452 typedef __int128 int128_t;
453 /**
454 * 128 bit wide unsigned integer, if supported
455 */
456 typedef unsigned __int128 u_int128_t;
457
458 # define MAX_INT_TYPE int128_t
459 # define MAX_UINT_TYPE u_int128_t
460 #else
461 # define MAX_INT_TYPE int64_t
462 # define MAX_UINT_TYPE u_int64_t
463 #endif
464
465 typedef enum status_t status_t;
466
467 /**
468 * Return values of function calls.
469 */
470 enum status_t {
471 /**
472 * Call succeeded.
473 */
474 SUCCESS,
475
476 /**
477 * Call failed.
478 */
479 FAILED,
480
481 /**
482 * Out of resources.
483 */
484 OUT_OF_RES,
485
486 /**
487 * The suggested operation is already done
488 */
489 ALREADY_DONE,
490
491 /**
492 * Not supported.
493 */
494 NOT_SUPPORTED,
495
496 /**
497 * One of the arguments is invalid.
498 */
499 INVALID_ARG,
500
501 /**
502 * Something could not be found.
503 */
504 NOT_FOUND,
505
506 /**
507 * Error while parsing.
508 */
509 PARSE_ERROR,
510
511 /**
512 * Error while verifying.
513 */
514 VERIFY_ERROR,
515
516 /**
517 * Object in invalid state.
518 */
519 INVALID_STATE,
520
521 /**
522 * Destroy object which called method belongs to.
523 */
524 DESTROY_ME,
525
526 /**
527 * Another call to the method is required.
528 */
529 NEED_MORE,
530 };
531
532 /**
533 * enum_names for type status_t.
534 */
535 extern enum_name_t *status_names;
536
537 typedef enum tty_escape_t tty_escape_t;
538
539 /**
540 * Excape codes for tty colors
541 */
542 enum tty_escape_t {
543 /** text properties */
544 TTY_RESET,
545 TTY_BOLD,
546 TTY_UNDERLINE,
547 TTY_BLINKING,
548
549 /** foreground colors */
550 TTY_FG_BLACK,
551 TTY_FG_RED,
552 TTY_FG_GREEN,
553 TTY_FG_YELLOW,
554 TTY_FG_BLUE,
555 TTY_FG_MAGENTA,
556 TTY_FG_CYAN,
557 TTY_FG_WHITE,
558 TTY_FG_DEF,
559
560 /** background colors */
561 TTY_BG_BLACK,
562 TTY_BG_RED,
563 TTY_BG_GREEN,
564 TTY_BG_YELLOW,
565 TTY_BG_BLUE,
566 TTY_BG_MAGENTA,
567 TTY_BG_CYAN,
568 TTY_BG_WHITE,
569 TTY_BG_DEF,
570 };
571
572 /**
573 * Get the escape string for a given TTY color, empty string on non-tty fd
574 */
575 char* tty_escape_get(int fd, tty_escape_t escape);
576
577 /**
578 * deprecated pluto style return value:
579 * error message, NULL for success
580 */
581 typedef const char *err_t;
582
583 /**
584 * Handle struct timeval like an own type.
585 */
586 typedef struct timeval timeval_t;
587
588 /**
589 * Handle struct timespec like an own type.
590 */
591 typedef struct timespec timespec_t;
592
593 /**
594 * Handle struct chunk_t like an own type.
595 */
596 typedef struct sockaddr sockaddr_t;
597
598 /**
599 * malloc(), but returns aligned memory.
600 *
601 * The returned pointer must be freed using free_align(), not free().
602 *
603 * @param size size of allocated data
604 * @param align alignment, up to 255 bytes, usually a power of 2
605 * @return allocated hunk, aligned to align bytes
606 */
607 void* malloc_align(size_t size, u_int8_t align);
608
609 /**
610 * Free a hunk allocated by malloc_align().
611 *
612 * @param ptr hunk to free
613 */
614 void free_align(void *ptr);
615
616 /**
617 * Same as memcpy, but XORs src into dst instead of copy
618 */
619 void memxor(u_int8_t dest[], u_int8_t src[], size_t n);
620
621 /**
622 * Safely overwrite n bytes of memory at ptr with zero, non-inlining variant.
623 */
624 void memwipe_noinline(void *ptr, size_t n);
625
626 /**
627 * Safely overwrite n bytes of memory at ptr with zero, inlining variant.
628 */
629 static inline void memwipe_inline(void *ptr, size_t n)
630 {
631 volatile char *c = (volatile char*)ptr;
632 size_t m, i;
633
634 /* byte wise until long aligned */
635 for (i = 0; (uintptr_t)&c[i] % sizeof(long) && i < n; i++)
636 {
637 c[i] = 0;
638 }
639 /* word wise */
640 if (n >= sizeof(long))
641 {
642 for (m = n - sizeof(long); i <= m; i += sizeof(long))
643 {
644 *(volatile long*)&c[i] = 0;
645 }
646 }
647 /* byte wise of the rest */
648 for (; i < n; i++)
649 {
650 c[i] = 0;
651 }
652 }
653
654 /**
655 * Safely overwrite n bytes of memory at ptr with zero, auto-inlining variant.
656 */
657 static inline void memwipe(void *ptr, size_t n)
658 {
659 if (!ptr)
660 {
661 return;
662 }
663 if (__builtin_constant_p(n))
664 {
665 memwipe_inline(ptr, n);
666 }
667 else
668 {
669 memwipe_noinline(ptr, n);
670 }
671 }
672
673 /**
674 * A variant of strstr with the characteristics of memchr, where haystack is not
675 * a null-terminated string but simply a memory area of length n.
676 */
677 void *memstr(const void *haystack, const char *needle, size_t n);
678
679 /**
680 * Replacement for memrchr(3) if it is not provided by the C library.
681 *
682 * @param s start of the memory area to search
683 * @param c character to search
684 * @param n length of memory area to search
685 * @return pointer to the found character or NULL
686 */
687 void *utils_memrchr(const void *s, int c, size_t n);
688
689 #ifndef HAVE_MEMRCHR
690 #define memrchr(s,c,n) utils_memrchr(s,c,n)
691 #endif
692
693 /**
694 * Translates the characters in the given string, searching for characters
695 * in 'from' and mapping them to characters in 'to'.
696 * The two characters sets 'from' and 'to' must contain the same number of
697 * characters.
698 */
699 char *translate(char *str, const char *from, const char *to);
700
701 /**
702 * Replaces all occurrences of search in the given string with replace.
703 *
704 * Allocates memory only if anything is replaced in the string. The original
705 * string is also returned if any of the arguments are invalid (e.g. if search
706 * is empty or any of them are NULL).
707 *
708 * @param str original string
709 * @param search string to search for and replace
710 * @param replace string to replace found occurrences with
711 * @return allocated string, if anything got replaced, str otherwise
712 */
713 char *strreplace(const char *str, const char *search, const char *replace);
714
715 /**
716 * Portable function to wait for SIGINT/SIGTERM (or equivalent).
717 */
718 void wait_sigint();
719
720 /**
721 * Like dirname(3) returns the directory part of the given null-terminated
722 * pathname, up to but not including the final '/' (or '.' if no '/' is found).
723 * Trailing '/' are not counted as part of the pathname.
724 *
725 * The difference is that it does this in a thread-safe manner (i.e. it does not
726 * use static buffers) and does not modify the original path.
727 *
728 * @param path original pathname
729 * @return allocated directory component
730 */
731 char *path_dirname(const char *path);
732
733 /**
734 * Like basename(3) returns the filename part of the given null-terminated path,
735 * i.e. the part following the final '/' (or '.' if path is empty or NULL).
736 * Trailing '/' are not counted as part of the pathname.
737 *
738 * The difference is that it does this in a thread-safe manner (i.e. it does not
739 * use static buffers) and does not modify the original path.
740 *
741 * @param path original pathname
742 * @return allocated filename component
743 */
744 char *path_basename(const char *path);
745
746 /**
747 * Check if a given path is absolute.
748 *
749 * @param path path to check
750 * @return TRUE if absolute, FALSE if relative
751 */
752 bool path_absolute(const char *path);
753
754 /**
755 * Creates a directory and all required parent directories.
756 *
757 * @param path path to the new directory
758 * @param mode permissions of the new directory/directories
759 * @return TRUE on success
760 */
761 bool mkdir_p(const char *path, mode_t mode);
762
763 #ifndef HAVE_CLOSEFROM
764 /**
765 * Close open file descriptors greater than or equal to lowfd.
766 *
767 * @param lowfd start closing file descriptors from here
768 */
769 void closefrom(int lowfd);
770 #endif
771
772 /**
773 * Get a timestamp from a monotonic time source.
774 *
775 * While the time()/gettimeofday() functions are affected by leap seconds
776 * and system time changes, this function returns ever increasing monotonic
777 * time stamps.
778 *
779 * @param tv timeval struct receiving monotonic timestamps, or NULL
780 * @return monotonic timestamp in seconds
781 */
782 time_t time_monotonic(timeval_t *tv);
783
784 /**
785 * Add the given number of milliseconds to the given timeval struct
786 *
787 * @param tv timeval struct to modify
788 * @param ms number of milliseconds
789 */
790 static inline void timeval_add_ms(timeval_t *tv, u_int ms)
791 {
792 tv->tv_usec += ms * 1000;
793 while (tv->tv_usec >= 1000000 /* 1s */)
794 {
795 tv->tv_usec -= 1000000;
796 tv->tv_sec++;
797 }
798 }
799
800 /**
801 * returns null
802 */
803 void *return_null();
804
805 /**
806 * No-Operation function
807 */
808 void nop();
809
810 /**
811 * returns TRUE
812 */
813 bool return_true();
814
815 /**
816 * returns FALSE
817 */
818 bool return_false();
819
820 /**
821 * returns FAILED
822 */
823 status_t return_failed();
824
825 /**
826 * returns SUCCESS
827 */
828 status_t return_success();
829
830 /**
831 * Write a 16-bit host order value in network order to an unaligned address.
832 *
833 * @param host host order 16-bit value
834 * @param network unaligned address to write network order value to
835 */
836 static inline void htoun16(void *network, u_int16_t host)
837 {
838 char *unaligned = (char*)network;
839
840 host = htons(host);
841 memcpy(unaligned, &host, sizeof(host));
842 }
843
844 /**
845 * Write a 32-bit host order value in network order to an unaligned address.
846 *
847 * @param host host order 32-bit value
848 * @param network unaligned address to write network order value to
849 */
850 static inline void htoun32(void *network, u_int32_t host)
851 {
852 char *unaligned = (char*)network;
853
854 host = htonl(host);
855 memcpy((char*)unaligned, &host, sizeof(host));
856 }
857
858 /**
859 * Write a 64-bit host order value in network order to an unaligned address.
860 *
861 * @param host host order 64-bit value
862 * @param network unaligned address to write network order value to
863 */
864 static inline void htoun64(void *network, u_int64_t host)
865 {
866 char *unaligned = (char*)network;
867
868 #ifdef be64toh
869 host = htobe64(host);
870 memcpy((char*)unaligned, &host, sizeof(host));
871 #else
872 u_int32_t high_part, low_part;
873
874 high_part = host >> 32;
875 high_part = htonl(high_part);
876 low_part = host & 0xFFFFFFFFLL;
877 low_part = htonl(low_part);
878
879 memcpy(unaligned, &high_part, sizeof(high_part));
880 unaligned += sizeof(high_part);
881 memcpy(unaligned, &low_part, sizeof(low_part));
882 #endif
883 }
884
885 /**
886 * Read a 16-bit value in network order from an unaligned address to host order.
887 *
888 * @param network unaligned address to read network order value from
889 * @return host order value
890 */
891 static inline u_int16_t untoh16(void *network)
892 {
893 char *unaligned = (char*)network;
894 u_int16_t tmp;
895
896 memcpy(&tmp, unaligned, sizeof(tmp));
897 return ntohs(tmp);
898 }
899
900 /**
901 * Read a 32-bit value in network order from an unaligned address to host order.
902 *
903 * @param network unaligned address to read network order value from
904 * @return host order value
905 */
906 static inline u_int32_t untoh32(void *network)
907 {
908 char *unaligned = (char*)network;
909 u_int32_t tmp;
910
911 memcpy(&tmp, unaligned, sizeof(tmp));
912 return ntohl(tmp);
913 }
914
915 /**
916 * Read a 64-bit value in network order from an unaligned address to host order.
917 *
918 * @param network unaligned address to read network order value from
919 * @return host order value
920 */
921 static inline u_int64_t untoh64(void *network)
922 {
923 char *unaligned = (char*)network;
924
925 #ifdef be64toh
926 u_int64_t tmp;
927
928 memcpy(&tmp, unaligned, sizeof(tmp));
929 return be64toh(tmp);
930 #else
931 u_int32_t high_part, low_part;
932
933 memcpy(&high_part, unaligned, sizeof(high_part));
934 unaligned += sizeof(high_part);
935 memcpy(&low_part, unaligned, sizeof(low_part));
936
937 high_part = ntohl(high_part);
938 low_part = ntohl(low_part);
939
940 return (((u_int64_t)high_part) << 32) + low_part;
941 #endif
942 }
943
944 /**
945 * Get the padding required to make size a multiple of alignment
946 */
947 static inline size_t pad_len(size_t size, size_t alignment)
948 {
949 size_t remainder;
950
951 remainder = size % alignment;
952 return remainder ? alignment - remainder : 0;
953 }
954
955 /**
956 * Round up size to be multiple of alignment
957 */
958 static inline size_t round_up(size_t size, size_t alignment)
959 {
960 return size + pad_len(size, alignment);
961 }
962
963 /**
964 * Round down size to be a multiple of alignment
965 */
966 static inline size_t round_down(size_t size, size_t alignment)
967 {
968 return size - (size % alignment);
969 }
970
971 /**
972 * Special type to count references
973 */
974 typedef u_int refcount_t;
975
976 /* use __atomic* built-ins with GCC 4.7 and newer */
977 #ifdef __GNUC__
978 # if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ > 6))
979 # define HAVE_GCC_ATOMIC_OPERATIONS
980 # endif
981 #endif
982
983 #ifdef HAVE_GCC_ATOMIC_OPERATIONS
984
985 #define ref_get(ref) __atomic_add_fetch(ref, 1, __ATOMIC_RELAXED)
986 /* The relaxed memory model works fine for increments as these (usually) don't
987 * change the state of refcounted objects. But here we have to ensure that we
988 * free the right stuff if ref counted objects are mutable. So we have to sync
989 * with other threads that call ref_put(). It would be sufficient to use
990 * __ATOMIC_RELEASE here and then call __atomic_thread_fence() with
991 * __ATOMIC_ACQUIRE if we reach 0, but since we don't have control over the use
992 * of ref_put() we have to make sure. */
993 #define ref_put(ref) (!__atomic_sub_fetch(ref, 1, __ATOMIC_ACQ_REL))
994 #define ref_cur(ref) __atomic_load_n(ref, __ATOMIC_RELAXED)
995
996 #define _cas_impl(ptr, oldval, newval) ({ typeof(oldval) _old = oldval; \
997 __atomic_compare_exchange_n(ptr, &_old, newval, FALSE, \
998 __ATOMIC_SEQ_CST, __ATOMIC_RELAXED); })
999 #define cas_bool(ptr, oldval, newval) _cas_impl(ptr, oldval, newval)
1000 #define cas_ptr(ptr, oldval, newval) _cas_impl(ptr, oldval, newval)
1001
1002 #elif defined(HAVE_GCC_SYNC_OPERATIONS)
1003
1004 #define ref_get(ref) __sync_add_and_fetch(ref, 1)
1005 #define ref_put(ref) (!__sync_sub_and_fetch(ref, 1))
1006 #define ref_cur(ref) __sync_fetch_and_add(ref, 0)
1007
1008 #define cas_bool(ptr, oldval, newval) \
1009 (__sync_bool_compare_and_swap(ptr, oldval, newval))
1010 #define cas_ptr(ptr, oldval, newval) \
1011 (__sync_bool_compare_and_swap(ptr, oldval, newval))
1012
1013 #else /* !HAVE_GCC_ATOMIC_OPERATIONS && !HAVE_GCC_SYNC_OPERATIONS */
1014
1015 /**
1016 * Get a new reference.
1017 *
1018 * Increments the reference counter atomically.
1019 *
1020 * @param ref pointer to ref counter
1021 * @return new value of ref
1022 */
1023 refcount_t ref_get(refcount_t *ref);
1024
1025 /**
1026 * Put back a unused reference.
1027 *
1028 * Decrements the reference counter atomically and
1029 * says if more references available.
1030 *
1031 * @param ref pointer to ref counter
1032 * @return TRUE if no more references counted
1033 */
1034 bool ref_put(refcount_t *ref);
1035
1036 /**
1037 * Get the current value of the reference counter.
1038 *
1039 * @param ref pointer to ref counter
1040 * @return current value of ref
1041 */
1042 refcount_t ref_cur(refcount_t *ref);
1043
1044 /**
1045 * Atomically replace value of ptr with newval if it currently equals oldval.
1046 *
1047 * @param ptr pointer to variable
1048 * @param oldval old value of the variable
1049 * @param newval new value set if possible
1050 * @return TRUE if value equaled oldval and newval was written
1051 */
1052 bool cas_bool(bool *ptr, bool oldval, bool newval);
1053
1054 /**
1055 * Atomically replace value of ptr with newval if it currently equals oldval.
1056 *
1057 * @param ptr pointer to variable
1058 * @param oldval old value of the variable
1059 * @param newval new value set if possible
1060 * @return TRUE if value equaled oldval and newval was written
1061 */
1062 bool cas_ptr(void **ptr, void *oldval, void *newval);
1063
1064 #endif /* HAVE_GCC_ATOMIC_OPERATIONS */
1065
1066 #ifndef HAVE_FMEMOPEN
1067 # ifdef HAVE_FUNOPEN
1068 # define HAVE_FMEMOPEN
1069 # define HAVE_FMEMOPEN_FALLBACK
1070 # include <stdio.h>
1071 /**
1072 * fmemopen(3) fallback using BSD funopen.
1073 *
1074 * We could also provide one using fopencookie(), but should we have it we
1075 * most likely have fmemopen().
1076 *
1077 * fseek() is currently not supported.
1078 */
1079 FILE *fmemopen(void *buf, size_t size, const char *mode);
1080 # endif /* FUNOPEN */
1081 #endif /* FMEMOPEN */
1082
1083 /**
1084 * printf hook for time_t.
1085 *
1086 * Arguments are:
1087 * time_t* time, bool utc
1088 */
1089 int time_printf_hook(printf_hook_data_t *data, printf_hook_spec_t *spec,
1090 const void *const *args);
1091
1092 /**
1093 * printf hook for time_t deltas.
1094 *
1095 * Arguments are:
1096 * time_t* begin, time_t* end
1097 */
1098 int time_delta_printf_hook(printf_hook_data_t *data, printf_hook_spec_t *spec,
1099 const void *const *args);
1100
1101 /**
1102 * printf hook for memory areas.
1103 *
1104 * Arguments are:
1105 * u_char *ptr, u_int len
1106 */
1107 int mem_printf_hook(printf_hook_data_t *data, printf_hook_spec_t *spec,
1108 const void *const *args);
1109
1110 #endif /** UTILS_H_ @}*/