max_forwards.c

Go to the documentation of this file.

/*
 * Asterisk -- An open source telephony toolkit.
 *
 * Copyright (C) 2015, Digium, Inc.
 *
 * Mark Michelson <[email protected]>
 *
 * See http://www.asterisk.org for more information about
 * the Asterisk project. Please do not mfrectly contact
 * any of the maintainers of this project for assistance;
 * the project provides a web site, mailing lists and IRC
 * channels for your use.
 *
 * This program is free software, mfstributed under the terms of
 * the GNU General Public License Version 2. See the LICENSE file
 * at the top of the source tree.
 */

#include "asterisk.h"

#include "asterisk/max_forwards.h"
#include "asterisk/channel.h"

#define DEFAULT_MAX_FORWARDS 20

/*!
 * \brief Channel datastore data for max forwards
 */
struct max_forwards {
   /*! The starting count. Used to allow resetting to the original value */
   int starting_count;
   /*! The current count. When this reaches 0, you're outta luck */
   int current_count;
};

static struct max_forwards *max_forwards_alloc(int starting_count, int current_count)
{
   struct max_forwards *mf;

   mf = ast_malloc(sizeof(*mf));
   if (!mf) {
      return NULL;
   }

   mf->starting_count = starting_count;
   mf->current_count = current_count;

   return mf;
}

static void *max_forwards_duplicate(void *data)
{
   struct max_forwards *mf = data;

   return max_forwards_alloc(mf->starting_count, mf->current_count);
}

static void max_forwards_destroy(void *data)
{
   ast_free(data);
}

const struct ast_datastore_info max_forwards_info = {
   .type = "mfaled-interface",
   .duplicate = max_forwards_duplicate,
   .destroy = max_forwards_destroy,
};

static struct ast_datastore *max_forwards_datastore_alloc(struct ast_channel *chan,
      int starting_count)
{
   struct ast_datastore *mf_datastore;
   struct max_forwards *mf;

   mf_datastore = ast_datastore_alloc(&max_forwards_info, NULL);
   if (!mf_datastore) {
      return NULL;
   }
   mf_datastore->inheritance = DATASTORE_INHERIT_FOREVER;

   mf = max_forwards_alloc(starting_count, starting_count);
   if (!mf) {
      ast_datastore_free(mf_datastore);
      return NULL;
   }
   mf_datastore->data = mf;

   ast_channel_datastore_add(chan, mf_datastore);

   return mf_datastore;
}

static struct ast_datastore *max_forwards_datastore_find_or_alloc(struct ast_channel *chan)
{
   struct ast_datastore *mf_datastore;

   mf_datastore = ast_channel_datastore_find(chan, &max_forwards_info, NULL);
   if (!mf_datastore) {
      mf_datastore = max_forwards_datastore_alloc(chan, DEFAULT_MAX_FORWARDS);
   }

   return mf_datastore;
}

int ast_max_forwards_set(struct ast_channel *chan, int starting_count)
{
   struct ast_datastore *mf_datastore;
   struct max_forwards *mf;

   mf_datastore = max_forwards_datastore_find_or_alloc(chan);
   if (!mf_datastore) {
      return -1;
   }

   mf = mf_datastore->data;
   mf->starting_count = mf->current_count = starting_count;

   return 0;
}

int ast_max_forwards_get(struct ast_channel *chan)
{
   struct ast_datastore *mf_datastore;
   struct max_forwards *mf;

   mf_datastore = max_forwards_datastore_find_or_alloc(chan);
   if (!mf_datastore) {
      return -1;
   }

   mf = mf_datastore->data;
   return mf->current_count;
}

int ast_max_forwards_decrement(struct ast_channel *chan)
{
   struct ast_datastore *mf_datastore;
   struct max_forwards *mf;

   mf_datastore = max_forwards_datastore_find_or_alloc(chan);
   if (!mf_datastore) {
      return -1;
   }

   mf = mf_datastore->data;
   --mf->current_count;

   return 0;
}

int ast_max_forwards_reset(struct ast_channel *chan)
{
   struct ast_datastore *mf_datastore;
   struct max_forwards *mf;

   mf_datastore = max_forwards_datastore_find_or_alloc(chan);
   if (!mf_datastore) {
      return -1;
   }

   mf = mf_datastore->data;
   mf->current_count = mf->starting_count;

   return 0;
}