util.h

Go to the documentation of this file.

/*
 * util.h
 *
 * Part of librfn (a general utility library from redfelineninja.org.uk)
 *
 * Copyright (C) 2012-2014 Daniel Thompson <daniel@redfelineninja.org.uk>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 */

#ifndef RF_UTIL_H_
#define RF_UTIL_H_

#include <assert.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdlib.h>
#include <stdint.h>

#define containerof(ptr, type, member) \
        ((type *) (((char *) ptr) - offsetof(type, member)))

#define lengthof(x) ((sizeof(x) / sizeof(*(x))))

#ifdef VERBOSE
#define verify(s) \
        do { \
                printf("Checking %s ... ", #s); \
                assert(s); \
                printf("OK\n"); \
        } while (0)

#else
#define verify(x) assert(x)
#endif

int32_t cyclecmp32(uint32_t a, uint32_t b);

typedef struct {
        uint32_t time;
        uint32_t count;
} ratelimit_state_t;

bool ratelimit_check(ratelimit_state_t *rs, uint32_t n, uint32_t window);

#define RATELIMIT_TO(n, window, fn)                                            \
        {                                                                      \
                static ratelimit_state_t rs_;                                  \
                if (ratelimit_check(&rs_, n, window))                          \
                        (fn);                                                  \
        }

#define RATELIMIT(fn) RATELIMIT_TO(3, 10, fn)

void rf_internal_out_of_memory(void);
void *xmalloc(size_t sz);

#endif // RF_UTIL_H_