trickle.c 2.6 KB
/*
 * Trickle implementation
 *
 * Copyright (C) 2013, 2014  INRIA.
 *
 * This file is subject to the terms and conditions of the GNU Lesser
 * General Public License v2.1. See the file LICENSE in the top level
 * directory for more details.
 */

/**
 * @author  Eric Engel <eric.engel@fu-berlin.de>
 * @author  Cenk Gündoğan <cnkgndgn@gmail.com>
 */

#include <string.h>
#include <stdio.h>
#include <stdlib.h>

#include "inttypes.h"
#include "random.h"
#include "trickle.h"

#define ENABLE_DEBUG    (0)
#include "debug.h"

void trickle_callback(trickle_t *trickle)
{
    /* Handle k=0 like k=infinity (according to RFC6206, section 6.5) */
    if ((trickle->c < trickle->k) || (trickle->k == 0)) {
        (*trickle->callback.func)(trickle->callback.args);
    }
}

void trickle_interval(trickle_t *trickle)
{
    uint32_t max_interval;

    trickle->I = trickle->I * 2;
    max_interval = trickle->Imin << trickle->Imax;

    if ((trickle->I == 0) || (trickle->I > max_interval)) {
        trickle->I = max_interval;
    }

    DEBUG("trickle: I == %" PRIu32 "\n", trickle->I);

    trickle->c = 0;
    trickle->t = (trickle->I / 2) + random_uint32_range(0, (trickle->I / 2) + 1);

    trickle->msg_callback_time = trickle->t * SEC_IN_MS;
    xtimer_set_msg64(&trickle->msg_callback_timer, trickle->msg_callback_time,
                     &trickle->msg_callback, trickle->pid);

    trickle->msg_interval_time = trickle->I * SEC_IN_MS;
    xtimer_set_msg64(&trickle->msg_interval_timer, trickle->msg_interval_time,
                     &trickle->msg_interval, trickle->pid);
}

void trickle_reset_timer(trickle_t *trickle)
{
    trickle_stop(trickle);
    trickle_start(trickle->pid, trickle, trickle->msg_interval.type, trickle->msg_callback.type,
                  trickle->Imin, trickle->Imax, trickle->k);
}

void trickle_start(kernel_pid_t pid, trickle_t *trickle, uint16_t interval_msg_type,
                   uint16_t callback_msg_type, uint32_t Imin, uint8_t Imax, uint8_t k)
{
    trickle->pid = pid;

    trickle->c = 0;
    trickle->k = k;
    trickle->Imin = Imin;
    trickle->Imax = Imax;
    trickle->I = trickle->Imin + random_uint32_range(0, 4 * trickle->Imin);
    trickle->pid = pid;
    trickle->msg_interval.content.ptr = trickle;
    trickle->msg_interval.type = interval_msg_type;
    trickle->msg_callback.content.ptr = trickle;
    trickle->msg_callback.type = callback_msg_type;

    trickle_interval(trickle);
}

void trickle_stop(trickle_t *trickle)
{
    xtimer_remove(&trickle->msg_interval_timer);
    xtimer_remove(&trickle->msg_callback_timer);
}

void trickle_increment_counter(trickle_t *trickle)
{
    trickle->c++;
}