rlm_sometimes.c

Go to the documentation of this file.

/*
 *   This program is is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or (at
 *   your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */
 
/**
 * $Id: e4d7b0f1d3b89592f128bade78433aa0f60f9d6d $
 * @file rlm_sometimes.c
 * @brief Switches between retuning different return codes.
 *
 * @copyright 2012 The FreeRADIUS server project
 */
RCSID("$Id: e4d7b0f1d3b89592f128bade78433aa0f60f9d6d $")
 
#include <freeradius-devel/server/base.h>
#include <freeradius-devel/server/module_rlm.h>
#include <freeradius-devel/util/debug.h>
 
/*
 *      The instance data for rlm_sometimes is the list of fake values we are
 *      going to return.
 */
typedef struct {
        char const      *rcode_str;
        rlm_rcode_t     rcode;
        float           percentage;
        tmpl_t  *key;
} rlm_sometimes_t;
 
static const conf_parser_t module_config[] = {
        { FR_CONF_OFFSET("rcode", rlm_sometimes_t, rcode_str), .dflt = "fail" },
        { FR_CONF_OFFSET_FLAGS("key", CONF_FLAG_ATTRIBUTE, rlm_sometimes_t, key), .dflt = "User-Name", .quote = T_BARE_WORD },
        { FR_CONF_OFFSET("percentage", rlm_sometimes_t, percentage), .dflt = "0" },
        CONF_PARSER_TERMINATOR
};
 
static int mod_instantiate(module_inst_ctx_t const *mctx)
{
        rlm_sometimes_t *inst = talloc_get_type_abort(mctx->mi->data, rlm_sometimes_t);
        CONF_SECTION    *conf = mctx->mi->conf;
 
        /*
         *      Convert the rcode string to an int, and get rid of it
         */
        inst->rcode = fr_table_value_by_str(rcode_table, inst->rcode_str, RLM_MODULE_NOT_SET);
        if (inst->rcode == RLM_MODULE_NOT_SET) {
                cf_log_err(conf, "Unknown module return code '%s'", inst->rcode_str);
                return -1;
        }
 
        if ((inst->percentage < 0) || (inst->percentage > 100)) {
                cf_log_err(conf, "Invalid value for 'percentage'.  It must be 0..100 inclusive");
                return -1;
        }
 
        return 0;
}
 
/*
 *      A lie!  It always returns!
 */
static unlang_action_t sometimes_return(unlang_result_t *p_result, module_ctx_t const *mctx, request_t *request,
                                        fr_packet_t *packet, fr_packet_t *reply)
{
        rlm_sometimes_t const   *inst = talloc_get_type_abort_const(mctx->mi->data, rlm_sometimes_t);
        uint32_t                hash;
        fr_pair_t               *vp;
        float                   value;
 
        /*
         *      Set it to NOOP and the module will always do nothing
         */
        if (inst->rcode == RLM_MODULE_NOOP) RETURN_UNLANG_RCODE(inst->rcode);
 
        /*
         *      Hash based on the given key.  Usually User-Name.
         */
        if (tmpl_find_vp(&vp, request, inst->key) < 0) RETURN_UNLANG_NOOP;
 
        switch (vp->vp_type) {
        case FR_TYPE_OCTETS:
        case FR_TYPE_STRING:
                hash = fr_hash(vp->data.datum.ptr, vp->vp_length);
                break;
 
        case FR_TYPE_STRUCTURAL:
                RETURN_UNLANG_FAIL;
 
        default:
                hash = fr_hash(&vp->data.datum, fr_value_box_field_sizes[vp->vp_type]);
                break;
        }
 
        hash &= 0xffff;         /* all we need are 2^16 bits of precision */
        value = hash;
        value /= (1 << 16);
        value *= 100;
 
        if (value > inst->percentage) RETURN_UNLANG_NOOP;
 
        /*
         *      If we're returning "handled", then set the packet
         *      code in the reply, so that the server responds.
         *
         *      @todo - MULTI_PROTOCOL - make this protocol agnostic
         */
        if ((inst->rcode == RLM_MODULE_HANDLED) && reply) {
                switch (packet->code) {
                case FR_RADIUS_CODE_ACCESS_REQUEST:
                        reply->code = FR_RADIUS_CODE_ACCESS_ACCEPT;
                        break;
 
                case FR_RADIUS_CODE_ACCOUNTING_REQUEST:
                        reply->code = FR_RADIUS_CODE_ACCOUNTING_RESPONSE;
                        break;
 
                case FR_RADIUS_CODE_COA_REQUEST:
                        reply->code = FR_RADIUS_CODE_COA_ACK;
                        break;
 
                case FR_RADIUS_CODE_DISCONNECT_REQUEST:
                        reply->code = FR_RADIUS_CODE_DISCONNECT_ACK;
                        break;
 
                default:
                        break;
                }
        }
 
        RETURN_UNLANG_RCODE(inst->rcode);
}
 
static unlang_action_t CC_HINT(nonnull) mod_sometimes_packet(unlang_result_t *p_result, module_ctx_t const *mctx, request_t *request)
{
        return sometimes_return(p_result, mctx, request, request->packet, request->reply);
}
 
static unlang_action_t CC_HINT(nonnull) mod_sometimes_reply(unlang_result_t *p_result, module_ctx_t const *mctx, request_t *request)
{
        return sometimes_return(p_result, mctx, request, request->reply, NULL);
}
 
extern module_rlm_t rlm_sometimes;
module_rlm_t rlm_sometimes = {
        .common = {
                .magic          = MODULE_MAGIC_INIT,
                .name           = "sometimes",
                .inst_size      = sizeof(rlm_sometimes_t),
                .config         = module_config,
                .instantiate    = mod_instantiate
        },
        .method_group = {
                .bindings = (module_method_binding_t[]){
                        { .section = SECTION_NAME("send", CF_IDENT_ANY), .method = mod_sometimes_reply },
                        { .section = SECTION_NAME(CF_IDENT_ANY, CF_IDENT_ANY), .method = mod_sometimes_packet },
                        MODULE_BINDING_TERMINATOR
                }
        }
};