infra.h

Go to the documentation of this file.

/*
 * services/cache/infra.h - infrastructure cache, server rtt and capabilities
 *
 * Copyright (c) 2007, NLnet Labs. All rights reserved.
 *
 * This software is open source.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * Neither the name of the NLNET LABS nor the names of its contributors may
 * be used to endorse or promote products derived from this software without
 * specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

 
#ifndef SERVICES_CACHE_INFRA_H
#define SERVICES_CACHE_INFRA_H
#include "util/storage/lruhash.h"
#include "util/storage/dnstree.h"
#include "util/rtt.h"
#include "util/netevent.h"
#include "util/data/msgreply.h"
struct slabhash;
struct config_file;

struct infra_key {
    struct sockaddr_storage addr;
    socklen_t addrlen;
    uint8_t* zonename;
    size_t namelen;
    struct lruhash_entry entry;
};

struct infra_data {
    time_t ttl;

    time_t probedelay;
    struct rtt_info rtt;

    int edns_version;
    uint8_t edns_lame_known;

    uint8_t isdnsseclame;
    uint8_t rec_lame;
    uint8_t lame_type_A;
    uint8_t lame_other;

    uint8_t timeout_A;
    uint8_t timeout_AAAA;
    uint8_t timeout_other;
};

struct infra_cache {
    struct slabhash* hosts;
    int host_ttl;
    struct slabhash* domain_rates;
    rbtree_type domain_limits;
    struct slabhash* client_ip_rates;
};

extern int infra_dp_ratelimit;

struct domain_limit_data {
    struct name_tree_node node;
    int lim;
    int below;
};

struct rate_key {
    struct lruhash_entry entry;
    uint8_t* name;
    size_t namelen;
};

extern int infra_ip_ratelimit;

struct ip_rate_key {
    struct lruhash_entry entry;
    struct sockaddr_storage addr;
    socklen_t addrlen;
};

#define RATE_WINDOW 2

struct rate_data {
    int qps[RATE_WINDOW];
    time_t timestamp[RATE_WINDOW];
};
 
#define ip_rate_data rate_data

#define INFRA_HOST_STARTSIZE 32
#define INFRA_BYTES_NAME 14

struct infra_cache* infra_create(struct config_file* cfg);

void infra_delete(struct infra_cache* infra);

struct infra_cache* infra_adjust(struct infra_cache* infra, 
    struct config_file* cfg);

struct lruhash_entry* infra_lookup_nottl(struct infra_cache* infra,
    struct sockaddr_storage* addr, socklen_t addrlen, uint8_t* name,
    size_t namelen, int wr);

int infra_host(struct infra_cache* infra, struct sockaddr_storage* addr, 
    socklen_t addrlen, uint8_t* name, size_t namelen,
    time_t timenow, int* edns_vs, uint8_t* edns_lame_known, int* to);

int infra_set_lame(struct infra_cache* infra,
        struct sockaddr_storage* addr, socklen_t addrlen,
    uint8_t* name, size_t namelen, time_t timenow, int dnsseclame,
    int reclame, uint16_t qtype);

int infra_rtt_update(struct infra_cache* infra, struct sockaddr_storage* addr,
    socklen_t addrlen, uint8_t* name, size_t namelen, int qtype,
    int roundtrip, int orig_rtt, time_t timenow);

void infra_update_tcp_works(struct infra_cache* infra,
        struct sockaddr_storage* addr, socklen_t addrlen,
    uint8_t* name, size_t namelen);

int infra_edns_update(struct infra_cache* infra,
        struct sockaddr_storage* addr, socklen_t addrlen,
    uint8_t* name, size_t namelen, int edns_version, time_t timenow);

int infra_get_lame_rtt(struct infra_cache* infra,
        struct sockaddr_storage* addr, socklen_t addrlen, 
    uint8_t* name, size_t namelen, uint16_t qtype, 
    int* lame, int* dnsseclame, int* reclame, int* rtt, time_t timenow);

long long infra_get_host_rto(struct infra_cache* infra,
        struct sockaddr_storage* addr, socklen_t addrlen, uint8_t* name,
    size_t namelen, struct rtt_info* rtt, int* delay, time_t timenow,
    int* tA, int* tAAAA, int* tother);

int infra_ratelimit_inc(struct infra_cache* infra, uint8_t* name,
    size_t namelen, time_t timenow);

void infra_ratelimit_dec(struct infra_cache* infra, uint8_t* name,
    size_t namelen, time_t timenow);

int infra_ratelimit_exceeded(struct infra_cache* infra, uint8_t* name,
    size_t namelen, time_t timenow);

int infra_rate_max(void* data, time_t now);

int infra_find_ratelimit(struct infra_cache* infra, uint8_t* name,
    size_t namelen);

int infra_ip_ratelimit_inc(struct infra_cache* infra,
    struct comm_reply* repinfo, time_t timenow);

size_t infra_get_mem(struct infra_cache* infra);

size_t infra_sizefunc(void* k, void* d);

int infra_compfunc(void* key1, void* key2);

void infra_delkeyfunc(void* k, void* arg);

void infra_deldatafunc(void* d, void* arg);

size_t rate_sizefunc(void* k, void* d);

int rate_compfunc(void* key1, void* key2);

void rate_delkeyfunc(void* k, void* arg);

void rate_deldatafunc(void* d, void* arg);
 
/* calculate size for the client ip hashtable */
size_t ip_rate_sizefunc(void* k, void* d);
 
/* compare two addresses */
int ip_rate_compfunc(void* key1, void* key2);
 
/* delete key, and destroy the lock */
void ip_rate_delkeyfunc(void* d, void* arg);
 
/* delete data */
#define ip_rate_deldatafunc rate_deldatafunc
 
#endif /* SERVICES_CACHE_INFRA_H */