Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2016  David Lamparter, for NetDEF, Inc.
       3             :  *
       4             :  * Permission to use, copy, modify, and distribute this software for any
       5             :  * purpose with or without fee is hereby granted, provided that the above
       6             :  * copyright notice and this permission notice appear in all copies.
       7             :  *
       8             :  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
       9             :  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
      10             :  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
      11             :  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
      12             :  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
      13             :  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
      14             :  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
      15             :  */
      16             : 
      17             : #ifdef HAVE_CONFIG_H
      18             : #include "config.h"
      19             : #endif
      20             : 
      21             : #include <string.h>
      22             : 
      23             : #include "memory.h"
      24             : #include "hook.h"
      25             : 
      26         670 : DEFINE_MTYPE_STATIC(LIB, HOOK_ENTRY, "Hook entry");
      27             : 
      28        4590 : void _hook_register(struct hook *hook, struct hookent *stackent, void *funcptr,
      29             :                     void *arg, bool has_arg, struct frrmod_runtime *module,
      30             :                     const char *funcname, int priority)
      31             : {
      32        4590 :         struct hookent *he, **pos;
      33             : 
      34        4590 :         if (!stackent->ent_used)
      35             :                 he = stackent;
      36             :         else {
      37           0 :                 he = XCALLOC(MTYPE_HOOK_ENTRY, sizeof(*he));
      38           0 :                 he->ent_on_heap = true;
      39             :         }
      40        4590 :         he->ent_used = true;
      41        4590 :         he->hookfn = funcptr;
      42        4590 :         he->hookarg = arg;
      43        4590 :         he->has_arg = has_arg;
      44        4590 :         he->module = module;
      45        4590 :         he->fnname = funcname;
      46        4590 :         he->priority = priority;
      47             : 
      48        8878 :         for (pos = &hook->entries; *pos; pos = &(*pos)->next)
      49        4740 :                 if (hook->reverse ? (*pos)->priority < priority
      50         226 :                                   : (*pos)->priority >= priority)
      51             :                         break;
      52             : 
      53        4590 :         he->next = *pos;
      54        4590 :         *pos = he;
      55        4590 : }
      56             : 
      57         625 : void _hook_unregister(struct hook *hook, void *funcptr, void *arg, bool has_arg)
      58             : {
      59         625 :         struct hookent *he, **prev;
      60             : 
      61         904 :         for (prev = &hook->entries; (he = *prev) != NULL; prev = &he->next)
      62         904 :                 if (he->hookfn == funcptr && he->hookarg == arg
      63         625 :                     && he->has_arg == has_arg) {
      64         625 :                         *prev = he->next;
      65         625 :                         if (he->ent_on_heap)
      66           0 :                                 XFREE(MTYPE_HOOK_ENTRY, he);
      67             :                         else
      68         625 :                                 memset(he, 0, sizeof(*he));
      69             :                         break;
      70             :                 }
      71         625 : }