1 #pragma once
2
3 #include <inttypes.h>
4 #include <limits.h>
5 #include <stdbool.h>
6 #include <stdlib.h>
7 #include <stdio.h>
8 #include <sys/types.h>
9 #include <sys/stat.h>
10
11 #include <shared/macro.h>
12
13 /* string handling functions and memory allocations */
14 /* ************************************************************************ */
15 #define streq(a, b) (strcmp((a), (b)) == 0)
16 #define strstartswith(a, b) (strncmp(a, b, strlen(b)) == 0)
17 char *strchr_replace(char *s, char c, char r);
18 void *memdup(const void *p, size_t n) __attribute__((nonnull(1)));
19
20 /* module-related functions */
21 /* ************************************************************************ */
22 #define KMOD_EXTENSION_UNCOMPRESSED ".ko"
23
24 int alias_normalize(const char *alias, char buf[static PATH_MAX], size_t *len) _must_check_ __attribute__((nonnull(1,2)));
25 int underscores(char *s) _must_check_;
26 char *modname_normalize(const char *modname, char buf[static PATH_MAX], size_t *len) __attribute__((nonnull(1, 2)));
27 char *path_to_modname(const char *path, char buf[static PATH_MAX], size_t *len) __attribute__((nonnull(2)));
28 bool path_ends_with_kmod_ext(const char *path, size_t len) __attribute__((nonnull(1)));
29
30 /* read-like and fread-like functions */
31 /* ************************************************************************ */
32 ssize_t read_str_safe(int fd, char *buf, size_t buflen) _must_check_ __attribute__((nonnull(2)));
33 ssize_t write_str_safe(int fd, const char *buf, size_t buflen) __attribute__((nonnull(2)));
34 int read_str_long(int fd, long *value, int base) _must_check_ __attribute__((nonnull(2)));
35 int read_str_ulong(int fd, unsigned long *value, int base) _must_check_ __attribute__((nonnull(2)));
36 char *freadline_wrapped(FILE *fp, unsigned int *linenum) __attribute__((nonnull(1)));
37
38 /* path handling functions */
39 /* ************************************************************************ */
40 bool path_is_absolute(const char *p) _must_check_ __attribute__((nonnull(1)));
41 char *path_make_absolute_cwd(const char *p) _must_check_ __attribute__((nonnull(1)));
42 int mkdir_p(const char *path, int len, mode_t mode);
43 int mkdir_parents(const char *path, mode_t mode);
44 unsigned long long stat_mstamp(const struct stat *st);
45 unsigned long long ts_usec(const struct timespec *ts);
46
47 /* endianess and alignments */
48 /* ************************************************************************ */
49 #define get_unaligned(ptr) \
50 ({ \
51 struct __attribute__((packed)) { \
52 typeof(*(ptr)) __v; \
53 } *__p = (typeof(__p)) (ptr); \
54 __p->__v; \
55 })
56
57 #define put_unaligned(val, ptr) \
58 do { \
59 struct __attribute__((packed)) { \
60 typeof(*(ptr)) __v; \
61 } *__p = (typeof(__p)) (ptr); \
62 __p->__v = (val); \
63 } while(0)
64
ALIGN_POWER2(unsigned int u)65 static _always_inline_ unsigned int ALIGN_POWER2(unsigned int u)
66 {
67 return 1 << ((sizeof(u) * 8) - __builtin_clz(u - 1));
68 }
69
70 /* misc */
71 /* ************************************************************************ */
freep(void * p)72 static inline void freep(void *p) {
73 free(*(void**) p);
74 }
75 #define _cleanup_free_ _cleanup_(freep)
76
addu64_overflow(uint64_t a,uint64_t b,uint64_t * res)77 static inline bool addu64_overflow(uint64_t a, uint64_t b, uint64_t *res)
78 {
79 #if (HAVE___BUILTIN_UADDL_OVERFLOW && HAVE___BUILTIN_UADDLL_OVERFLOW)
80 #if __SIZEOF_LONG__ == 8
81 return __builtin_uaddl_overflow(a, b, res);
82 #elif __SIZEOF_LONG_LONG__ == 8
83 return __builtin_uaddll_overflow(a, b, res);
84 #else
85 #error "sizeof(long long) != 8"
86 #endif
87 #endif
88 *res = a + b;
89 return UINT64_MAX - a < b;
90 }
91