githubmirror/dbus-broker

src/util/user.h
96 строк2 KB
Tom Gundersen
broker: max per-UID limits configurable
19 июл 2017, 13:16
19 июл 2017, 13:160e60d64
#pragma once

/*
 * User Accounting
 */

#include <c-macro.h>
#include <c-rbtree.h>
#include <c-ref.h>
#include <stdlib.h>
#include <sys/types.h>

typedef struct UserCharge UserCharge;
typedef struct UserUsage UserUsage;
typedef struct User User;
typedef struct UserRegistry UserRegistry;

/* XXX: move this to some global broker header file */
enum {
        USER_SLOT_BYTES,
        USER_SLOT_FDS,
        USER_SLOT_MATCHES,
        USER_SLOT_OBJECTS,
        _USER_SLOT_N,
};

enum {
        _USER_E_SUCCESS,

        USER_E_QUOTA,
};

/* charge */

struct UserCharge {
        UserUsage *usage;
        size_t slot;
        unsigned int charge;
};

#define USER_CHARGE_INIT {}

void user_charge_init(UserCharge *charge);
void user_charge_deinit(UserCharge *charge);

/* user */

struct User {
        _Atomic unsigned long n_refs;
        UserRegistry *registry;
        uid_t uid;
        CRBNode registry_node;

        CRBTree usage_tree;
        unsigned int n_usages;

        struct {
                unsigned int n;
                unsigned int max;
        } slots[];
};

void user_free(_Atomic unsigned long *n_refs, void *userdata);
int user_charge(User *user, UserCharge *charge, User *actor, size_t slot, unsigned int amount);

/* registry */

struct UserRegistry {
        CRBTree user_tree;
        size_t n_slots;
        unsigned int *maxima;
};

#define USER_REGISTRY_NULL {                                                    \
                .user_tree = C_RBTREE_INIT,                                     \
        }

int user_registry_init(UserRegistry *registry, size_t n_slots, const unsigned int *maxima);
void user_registry_deinit(UserRegistry *registry);
int user_registry_ref_user(UserRegistry *registry, User **userp, uid_t uid);

/* inline helpers */

static inline User *user_ref(User *user) {
        if (user)
                c_ref_inc(&user->n_refs);
        return user;
}

static inline User *user_unref(User *user) {
        if (user)
                c_ref_dec(&user->n_refs, user_free, NULL);
        return NULL;
}

C_DEFINE_CLEANUP(User *, user_unref);