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
 9struct rcu_string {
10	struct rcu_head rcu;
11	char str[0];
12};
13
14static 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	strncpy(ret->str, src, len);
 
 
 
 
22	return ret;
23}
24
25static inline void rcu_string_free(struct rcu_string *str)
26{
27	if (str)
28		kfree_rcu(str, rcu);
29}
30
31#define printk_in_rcu(fmt, ...) do {	\
32	rcu_read_lock();		\
33	printk(fmt, __VA_ARGS__);	\
34	rcu_read_unlock();		\
35} while (0)
36
37#define printk_ratelimited_in_rcu(fmt, ...) do {	\
38	rcu_read_lock();				\
39	printk_ratelimited(fmt, __VA_ARGS__);		\
40	rcu_read_unlock();				\
41} while (0)
42
43#define rcu_str_deref(rcu_str) ({				\
44	struct rcu_string *__str = rcu_dereference(rcu_str);	\
45	__str->str;						\
46})
47
48#endif

 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
 9struct rcu_string {
10	struct rcu_head rcu;
11	char str[];
12};
13
14static 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
29static 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