Home | History | Annotate | Download | only in shared 
      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 
     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 /* ************************************************************************ */
     72 static inline void freep(void *p) {
     73         free(*(void**) p);
     74 }
     75 #define _cleanup_free_ _cleanup_(freep)
     76 
     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