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
rcu_string_strdup(const char * src,gfp_t mask)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
rcu_string_free(struct rcu_string * str)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