/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _LINUX_SCHED_USER_H
#define _LINUX_SCHED_USER_H

#include <linux/uidgid.h>
#include <linux/atomic.h>
#include <linux/percpu_counter.h>
#include <linux/refcount.h>
#include <linux/ratelimit.h>

/*
 * Some day this will be a full-fledged user tracking system..
 */
struct user_struct {
    refcount_t __count;    /* reference count */
#ifdef CONFIG_EPOLL
    struct percpu_counter epoll_watches; /* The number of file descriptors currently watched */
#endif
    unsigned long unix_inflight;    /* How many files in flight in unix sockets */
    atomic_long_t pipe_bufs;  /* how many pages are allocated in pipe buffers */

#ifdef CONFIG_GRKERNSEC_KERN_LOCKOUT
    unsigned char kernel_banned;
#endif
#ifdef CONFIG_GRKERNSEC_BRUTE
    unsigned char sugid_banned;
    unsigned long sugid_ban_expires;
#endif

    /* Hash table maintenance information */
    struct hlist_node uidhash_node;
    kuid_t uid;

#if defined(CONFIG_PERF_EVENTS) || defined(CONFIG_BPF_SYSCALL) || \
    defined(CONFIG_NET) || defined(CONFIG_IO_URING) || \
    defined(CONFIG_VFIO_PCI_ZDEV_KVM) || IS_ENABLED(CONFIG_IOMMUFD)
    atomic_long_t locked_vm;
#endif
#ifdef CONFIG_WATCH_QUEUE
    atomic_t nr_watches;    /* The number of watches this user currently has */
#endif

    /* Miscellaneous per-user rate limit */
    struct ratelimit_state ratelimit;
}  __randomize_layout;

extern int uids_sysfs_init(void);

extern struct user_struct *find_user(kuid_t);

extern struct user_struct root_user;
#define INIT_USER (&root_user)


/* per-UID process charging. */
extern struct user_struct * alloc_uid(kuid_t);
static inline struct user_struct *get_uid(struct user_struct *u)
{
    refcount_inc(&u->__count);
    return u;
}
extern void free_uid(struct user_struct *);

#endif /* _LINUX_SCHED_USER_H */