1  /* SPDX-License-Identifier: GPL-2.0-only */
2  /*
3   * include/linux/sunrpc/cache.h
4   *
5   * Generic code for various authentication-related caches
6   * used by sunrpc clients and servers.
7   *
8   * Copyright (C) 2002 Neil Brown <neilb@cse.unsw.edu.au>
9   */
10  
11  #ifndef _LINUX_SUNRPC_CACHE_H_
12  #define _LINUX_SUNRPC_CACHE_H_
13  
14  #include <linux/kref.h>
15  #include <linux/slab.h>
16  #include <linux/atomic.h>
17  #include <linux/kstrtox.h>
18  #include <linux/proc_fs.h>
19  
20  /*
21   * Each cache requires:
22   *  - A 'struct cache_detail' which contains information specific to the cache
23   *    for common code to use.
24   *  - An item structure that must contain a "struct cache_head"
25   *  - A lookup function defined using DefineCacheLookup
26   *  - A 'put' function that can release a cache item. It will only
27   *    be called after cache_put has succeed, so there are guarantee
28   *    to be no references.
29   *  - A function to calculate a hash of an item's key.
30   *
31   * as well as assorted code fragments (e.g. compare keys) and numbers
32   * (e.g. hash size, goal_age, etc).
33   *
34   * Each cache must be registered so that it can be cleaned regularly.
35   * When the cache is unregistered, it is flushed completely.
36   *
37   * Entries have a ref count and a 'hashed' flag which counts the existence
38   * in the hash table.
39   * We only expire entries when refcount is zero.
40   * Existence in the cache is counted  the refcount.
41   */
42  
43  /* Every cache item has a common header that is used
44   * for expiring and refreshing entries.
45   *
46   */
47  struct cache_head {
48  	struct hlist_node	cache_list;
49  	time64_t	expiry_time;	/* After time expiry_time, don't use
50  					 * the data */
51  	time64_t	last_refresh;   /* If CACHE_PENDING, this is when upcall was
52  					 * sent, else this is when update was
53  					 * received, though it is alway set to
54  					 * be *after* ->flush_time.
55  					 */
56  	struct kref	ref;
57  	unsigned long	flags;
58  };
59  
60  /* cache_head.flags */
61  enum {
62  	CACHE_VALID,		/* Entry contains valid data */
63  	CACHE_NEGATIVE,		/* Negative entry - there is no match for the key */
64  	CACHE_PENDING,		/* An upcall has been sent but no reply received yet*/
65  	CACHE_CLEANED,		/* Entry has been cleaned from cache */
66  };
67  
68  #define	CACHE_NEW_EXPIRY 120	/* keep new things pending confirmation for 120 seconds */
69  
70  struct cache_detail {
71  	struct module *		owner;
72  	int			hash_size;
73  	struct hlist_head *	hash_table;
74  	spinlock_t		hash_lock;
75  
76  	char			*name;
77  	void			(*cache_put)(struct kref *);
78  
79  	int			(*cache_upcall)(struct cache_detail *,
80  						struct cache_head *);
81  
82  	void			(*cache_request)(struct cache_detail *cd,
83  						 struct cache_head *ch,
84  						 char **bpp, int *blen);
85  
86  	int			(*cache_parse)(struct cache_detail *,
87  					       char *buf, int len);
88  
89  	int			(*cache_show)(struct seq_file *m,
90  					      struct cache_detail *cd,
91  					      struct cache_head *h);
92  	void			(*warn_no_listener)(struct cache_detail *cd,
93  					      int has_died);
94  
95  	struct cache_head *	(*alloc)(void);
96  	void			(*flush)(void);
97  	int			(*match)(struct cache_head *orig, struct cache_head *new);
98  	void			(*init)(struct cache_head *orig, struct cache_head *new);
99  	void			(*update)(struct cache_head *orig, struct cache_head *new);
100  
101  	/* fields below this comment are for internal use
102  	 * and should not be touched by cache owners
103  	 */
104  	time64_t		flush_time;		/* flush all cache items with
105  							 * last_refresh at or earlier
106  							 * than this.  last_refresh
107  							 * is never set at or earlier
108  							 * than this.
109  							 */
110  	struct list_head	others;
111  	time64_t		nextcheck;
112  	int			entries;
113  
114  	/* fields for communication over channel */
115  	struct list_head	queue;
116  
117  	atomic_t		writers;		/* how many time is /channel open */
118  	time64_t		last_close;		/* if no writers, when did last close */
119  	time64_t		last_warn;		/* when we last warned about no writers */
120  
121  	union {
122  		struct proc_dir_entry	*procfs;
123  		struct dentry		*pipefs;
124  	};
125  	struct net		*net;
126  };
127  
128  /* this must be embedded in any request structure that
129   * identifies an object that will want a callback on
130   * a cache fill
131   */
132  struct cache_req {
133  	struct cache_deferred_req *(*defer)(struct cache_req *req);
134  	unsigned long	thread_wait;	/* How long (jiffies) we can block the
135  					 * current thread to wait for updates.
136  					 */
137  };
138  
139  /* this must be embedded in a deferred_request that is being
140   * delayed awaiting cache-fill
141   */
142  struct cache_deferred_req {
143  	struct hlist_node	hash;	/* on hash chain */
144  	struct list_head	recent; /* on fifo */
145  	struct cache_head	*item;  /* cache item we wait on */
146  	void			*owner; /* we might need to discard all defered requests
147  					 * owned by someone */
148  	void			(*revisit)(struct cache_deferred_req *req,
149  					   int too_many);
150  };
151  
152  /*
153   * timestamps kept in the cache are expressed in seconds
154   * since boot.  This is the best for measuring differences in
155   * real time.
156   * This reimplemnts ktime_get_boottime_seconds() in a slightly
157   * faster but less accurate way. When we end up converting
158   * back to wallclock (CLOCK_REALTIME), that error often
159   * cancels out during the reverse operation.
160   */
seconds_since_boot(void)161  static inline time64_t seconds_since_boot(void)
162  {
163  	struct timespec64 boot;
164  	getboottime64(&boot);
165  	return ktime_get_real_seconds() - boot.tv_sec;
166  }
167  
convert_to_wallclock(time64_t sinceboot)168  static inline time64_t convert_to_wallclock(time64_t sinceboot)
169  {
170  	struct timespec64 boot;
171  	getboottime64(&boot);
172  	return boot.tv_sec + sinceboot;
173  }
174  
175  extern const struct file_operations cache_file_operations_pipefs;
176  extern const struct file_operations content_file_operations_pipefs;
177  extern const struct file_operations cache_flush_operations_pipefs;
178  
179  extern struct cache_head *
180  sunrpc_cache_lookup_rcu(struct cache_detail *detail,
181  			struct cache_head *key, int hash);
182  extern struct cache_head *
183  sunrpc_cache_update(struct cache_detail *detail,
184  		    struct cache_head *new, struct cache_head *old, int hash);
185  
186  extern int
187  sunrpc_cache_pipe_upcall(struct cache_detail *detail, struct cache_head *h);
188  extern int
189  sunrpc_cache_pipe_upcall_timeout(struct cache_detail *detail,
190  				 struct cache_head *h);
191  
192  
193  extern void cache_clean_deferred(void *owner);
194  
cache_get(struct cache_head * h)195  static inline struct cache_head  *cache_get(struct cache_head *h)
196  {
197  	kref_get(&h->ref);
198  	return h;
199  }
200  
cache_get_rcu(struct cache_head * h)201  static inline struct cache_head  *cache_get_rcu(struct cache_head *h)
202  {
203  	if (kref_get_unless_zero(&h->ref))
204  		return h;
205  	return NULL;
206  }
207  
cache_put(struct cache_head * h,struct cache_detail * cd)208  static inline void cache_put(struct cache_head *h, struct cache_detail *cd)
209  {
210  	if (kref_read(&h->ref) <= 2 &&
211  	    h->expiry_time < cd->nextcheck)
212  		cd->nextcheck = h->expiry_time;
213  	kref_put(&h->ref, cd->cache_put);
214  }
215  
cache_is_expired(struct cache_detail * detail,struct cache_head * h)216  static inline bool cache_is_expired(struct cache_detail *detail, struct cache_head *h)
217  {
218  	if (h->expiry_time < seconds_since_boot())
219  		return true;
220  	if (!test_bit(CACHE_VALID, &h->flags))
221  		return false;
222  	return detail->flush_time >= h->last_refresh;
223  }
224  
225  extern int cache_check(struct cache_detail *detail,
226  		       struct cache_head *h, struct cache_req *rqstp);
227  extern void cache_flush(void);
228  extern void cache_purge(struct cache_detail *detail);
229  #define NEVER (0x7FFFFFFF)
230  extern void __init cache_initialize(void);
231  extern int cache_register_net(struct cache_detail *cd, struct net *net);
232  extern void cache_unregister_net(struct cache_detail *cd, struct net *net);
233  
234  extern struct cache_detail *cache_create_net(const struct cache_detail *tmpl, struct net *net);
235  extern void cache_destroy_net(struct cache_detail *cd, struct net *net);
236  
237  extern void sunrpc_init_cache_detail(struct cache_detail *cd);
238  extern void sunrpc_destroy_cache_detail(struct cache_detail *cd);
239  extern int sunrpc_cache_register_pipefs(struct dentry *parent, const char *,
240  					umode_t, struct cache_detail *);
241  extern void sunrpc_cache_unregister_pipefs(struct cache_detail *);
242  extern void sunrpc_cache_unhash(struct cache_detail *, struct cache_head *);
243  
244  /* Must store cache_detail in seq_file->private if using next three functions */
245  extern void *cache_seq_start_rcu(struct seq_file *file, loff_t *pos);
246  extern void *cache_seq_next_rcu(struct seq_file *file, void *p, loff_t *pos);
247  extern void cache_seq_stop_rcu(struct seq_file *file, void *p);
248  
249  extern void qword_add(char **bpp, int *lp, char *str);
250  extern void qword_addhex(char **bpp, int *lp, char *buf, int blen);
251  extern int qword_get(char **bpp, char *dest, int bufsize);
252  
get_int(char ** bpp,int * anint)253  static inline int get_int(char **bpp, int *anint)
254  {
255  	char buf[50];
256  	char *ep;
257  	int rv;
258  	int len = qword_get(bpp, buf, sizeof(buf));
259  
260  	if (len < 0)
261  		return -EINVAL;
262  	if (len == 0)
263  		return -ENOENT;
264  
265  	rv = simple_strtol(buf, &ep, 0);
266  	if (*ep)
267  		return -EINVAL;
268  
269  	*anint = rv;
270  	return 0;
271  }
272  
get_uint(char ** bpp,unsigned int * anint)273  static inline int get_uint(char **bpp, unsigned int *anint)
274  {
275  	char buf[50];
276  	int len = qword_get(bpp, buf, sizeof(buf));
277  
278  	if (len < 0)
279  		return -EINVAL;
280  	if (len == 0)
281  		return -ENOENT;
282  
283  	if (kstrtouint(buf, 0, anint))
284  		return -EINVAL;
285  
286  	return 0;
287  }
288  
get_time(char ** bpp,time64_t * time)289  static inline int get_time(char **bpp, time64_t *time)
290  {
291  	char buf[50];
292  	long long ll;
293  	int len = qword_get(bpp, buf, sizeof(buf));
294  
295  	if (len < 0)
296  		return -EINVAL;
297  	if (len == 0)
298  		return -ENOENT;
299  
300  	if (kstrtoll(buf, 0, &ll))
301  		return -EINVAL;
302  
303  	*time = ll;
304  	return 0;
305  }
306  
get_expiry(char ** bpp,time64_t * rvp)307  static inline int get_expiry(char **bpp, time64_t *rvp)
308  {
309  	int error;
310  	struct timespec64 boot;
311  
312  	error = get_time(bpp, rvp);
313  	if (error)
314  		return error;
315  
316  	getboottime64(&boot);
317  	(*rvp) -= boot.tv_sec;
318  	return 0;
319  }
320  
321  #endif /*  _LINUX_SUNRPC_CACHE_H_ */
322