1 /* SPDX-License-Identifier: GPL-2.0 */ 2 /* 3 * Copyright (C) 2012 Red Hat. All rights reserved. 4 */ 5 6 #ifndef BTRFS_RCU_STRING_H 7 #define BTRFS_RCU_STRING_H 8 9 struct rcu_string { 10 struct rcu_head rcu; 11 char str[]; 12 }; 13 14 static inline struct rcu_string *rcu_string_strdup(const char *src, gfp_t mask) 15 { 16 size_t len = strlen(src) + 1; 17 struct rcu_string *ret = kzalloc(sizeof(struct rcu_string) + 18 (len * sizeof(char)), mask); 19 if (!ret) 20 return ret; 21 /* Warn if the source got unexpectedly truncated. */ 22 if (WARN_ON(strscpy(ret->str, src, len) < 0)) { 23 kfree(ret); 24 return NULL; 25 } 26 return ret; 27 } 28 29 static inline void rcu_string_free(struct rcu_string *str) 30 { 31 if (str) 32 kfree_rcu(str, rcu); 33 } 34 35 #define printk_in_rcu(fmt, ...) do { \ 36 rcu_read_lock(); \ 37 printk(fmt, __VA_ARGS__); \ 38 rcu_read_unlock(); \ 39 } while (0) 40 41 #define printk_ratelimited_in_rcu(fmt, ...) do { \ 42 rcu_read_lock(); \ 43 printk_ratelimited(fmt, __VA_ARGS__); \ 44 rcu_read_unlock(); \ 45 } while (0) 46 47 #define rcu_str_deref(rcu_str) ({ \ 48 struct rcu_string *__str = rcu_dereference(rcu_str); \ 49 __str->str; \ 50 }) 51 52 #endif 53