smatch_function_hooks.c

   1 /*
   2  * Copyright (C) 2009 Dan Carpenter.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, see http://www.gnu.org/copyleft/gpl.txt
  16  */
  17
  18 /*
  19  * There are several types of function hooks:
  20  * add_function_hook()        - For any time a function is called.
  21  * add_function_assign_hook() - foo = the_function().
  22  * add_implied_return_hook()  - Calculates the implied return value.
  23  * add_macro_assign_hook()    - foo = the_macro().
  24  * return_implies_state()     - For when a return value of 1 implies locked
  25  *                              and 0 implies unlocked. etc. etc.
  26  *
  27  */
  28
  29 #include <stdlib.h>
  30 #include <stdio.h>
  31 #include "smatch.h"
  32 #include "smatch_slist.h"
  33 #include "smatch_extra.h"
  34 #include "smatch_function_hashtable.h"
  35
  36 struct fcall_back {
  37         int type;
  38         struct data_range *range;
  39         union {
  40                 func_hook *call_back;
  41                 implication_hook *ranged;
  42                 implied_return_hook *implied_return;
  43         } u;
  44         void *info;
  45 };
  46
  47 ALLOCATOR(fcall_back, "call backs");
  48 DECLARE_PTR_LIST(call_back_list, struct fcall_back);
  49
  50 DEFINE_FUNCTION_HASHTABLE_STATIC(callback, struct fcall_back, struct call_back_list);
  51 static struct hashtable *func_hash;
  52
  53 #define REGULAR_CALL       0
  54 #define RANGED_CALL        1
  55 #define ASSIGN_CALL        2
  56 #define IMPLIED_RETURN     3
  57 #define MACRO_ASSIGN       4
  58 #define MACRO_ASSIGN_EXTRA 5
  59
  60 struct return_implies_callback {
  61         int type;
  62         return_implies_hook *callback;
  63 };
  64 ALLOCATOR(return_implies_callback, "return_implies callbacks");
  65 DECLARE_PTR_LIST(db_implies_list, struct return_implies_callback);
  66 static struct db_implies_list *db_return_states_list;
  67
  68 typedef void (void_fn)(void);
  69 DECLARE_PTR_LIST(void_fn_list, void_fn *);
  70 static struct void_fn_list *return_states_before;
  71 static struct void_fn_list *return_states_after;
  72
  73 static struct fcall_back *alloc_fcall_back(int type, void *call_back,
  74                                            void *info)
  75 {
  76         struct fcall_back *cb;
  77
  78         cb = __alloc_fcall_back(0);
  79         cb->type = type;
  80         cb->u.call_back = call_back;
  81         cb->info = info;
  82         return cb;
  83 }
  84
  85 void add_function_hook(const char *look_for, func_hook *call_back, void *info)
  86 {
  87         struct fcall_back *cb;
  88
  89         cb = alloc_fcall_back(REGULAR_CALL, call_back, info);
  90         add_callback(func_hash, look_for, cb);
  91 }
  92
  93 void add_function_assign_hook(const char *look_for, func_hook *call_back,
  94                               void *info)
  95 {
  96         struct fcall_back *cb;
  97
  98         cb = alloc_fcall_back(ASSIGN_CALL, call_back, info);
  99         add_callback(func_hash, look_for, cb);
 100 }
 101
 102 void add_implied_return_hook(const char *look_for,
 103                              implied_return_hook *call_back,
 104                              void *info)
 105 {
 106         struct fcall_back *cb;
 107
 108         cb = alloc_fcall_back(IMPLIED_RETURN, call_back, info);
 109         add_callback(func_hash, look_for, cb);
 110 }
 111
 112 void add_macro_assign_hook(const char *look_for, func_hook *call_back,
 113                         void *info)
 114 {
 115         struct fcall_back *cb;
 116
 117         cb = alloc_fcall_back(MACRO_ASSIGN, call_back, info);
 118         add_callback(func_hash, look_for, cb);
 119 }
 120
 121 void add_macro_assign_hook_extra(const char *look_for, func_hook *call_back,
 122                         void *info)
 123 {
 124         struct fcall_back *cb;
 125
 126         cb = alloc_fcall_back(MACRO_ASSIGN_EXTRA, call_back, info);
 127         add_callback(func_hash, look_for, cb);
 128 }
 129
 130 void return_implies_state(const char *look_for, long long start, long long end,
 131                          implication_hook *call_back, void *info)
 132 {
 133         struct fcall_back *cb;
 134
 135         cb = alloc_fcall_back(RANGED_CALL, call_back, info);
 136         cb->range = alloc_range_perm(ll_to_sval(start), ll_to_sval(end));
 137         add_callback(func_hash, look_for, cb);
 138 }
 139
 140 void select_return_states_hook(int type, return_implies_hook *callback)
 141 {
 142         struct return_implies_callback *cb = __alloc_return_implies_callback(0);
 143
 144         cb->type = type;
 145         cb->callback = callback;
 146         add_ptr_list(&db_return_states_list, cb);
 147 }
 148
 149 void select_return_states_before(void_fn *fn)
 150 {
 151         void_fn **p = malloc(sizeof(void_fn *));
 152         *p = fn;
 153         add_ptr_list(&return_states_before, p);
 154 }
 155
 156 void select_return_states_after(void_fn *fn)
 157 {
 158         void_fn **p = malloc(sizeof(void_fn *));
 159         *p = fn;
 160         add_ptr_list(&return_states_after, p);
 161 }
 162
 163 static void call_return_states_before_hooks(void)
 164 {
 165         void_fn **fn;
 166
 167         FOR_EACH_PTR(return_states_before, fn) {
 168                 (*fn)();
 169         } END_FOR_EACH_PTR(fn);
 170 }
 171
 172 static void call_return_states_after_hooks(void)
 173 {
 174         void_fn **fn;
 175
 176         FOR_EACH_PTR(return_states_after, fn) {
 177                 (*fn)();
 178         } END_FOR_EACH_PTR(fn);
 179 }
 180
 181 static int call_call_backs(struct call_back_list *list, int type,
 182                             const char *fn, struct expression *expr)
 183 {
 184         struct fcall_back *tmp;
 185         int handled = 0;
 186
 187         FOR_EACH_PTR(list, tmp) {
 188                 if (tmp->type == type) {
 189                         (tmp->u.call_back)(fn, expr, tmp->info);
 190                         handled = 1;
 191                 }
 192         } END_FOR_EACH_PTR(tmp);
 193
 194         return handled;
 195 }
 196
 197 static void call_ranged_call_backs(struct call_back_list *list,
 198                                 const char *fn, struct expression *call_expr,
 199                                 struct expression *assign_expr)
 200 {
 201         struct fcall_back *tmp;
 202
 203         FOR_EACH_PTR(list, tmp) {
 204                 (tmp->u.ranged)(fn, call_expr, assign_expr, tmp->info);
 205         } END_FOR_EACH_PTR(tmp);
 206 }
 207
 208 static struct call_back_list *get_same_ranged_call_backs(struct call_back_list *list,
 209                                                 struct data_range *drange)
 210 {
 211         struct call_back_list *ret = NULL;
 212         struct fcall_back *tmp;
 213
 214         FOR_EACH_PTR(list, tmp) {
 215                 if (tmp->type != RANGED_CALL)
 216                         continue;
 217                 if (ranges_equiv(tmp->range, drange))
 218                         add_ptr_list(&ret, tmp);
 219         } END_FOR_EACH_PTR(tmp);
 220         return ret;
 221 }
 222
 223 static int in_list_exact_sval(struct range_list *list, struct data_range *drange)
 224 {
 225         struct data_range *tmp;
 226
 227         FOR_EACH_PTR(list, tmp) {
 228                 if (ranges_equiv(tmp, drange))
 229                         return 1;
 230         } END_FOR_EACH_PTR(tmp);
 231         return 0;
 232 }
 233
 234 static int assign_ranged_funcs(const char *fn, struct expression *expr,
 235                                  struct call_back_list *call_backs)
 236 {
 237         struct fcall_back *tmp;
 238         struct sm_state *sm;
 239         char *var_name;
 240         struct symbol *sym;
 241         struct smatch_state *estate;
 242         struct stree *tmp_stree;
 243         struct stree *final_states = NULL;
 244         struct range_list *handled_ranges = NULL;
 245         struct call_back_list *same_range_call_backs = NULL;
 246         int handled = 0;
 247
 248         if (!call_backs)
 249                 return 0;
 250
 251         var_name = expr_to_var_sym(expr->left, &sym);
 252         if (!var_name || !sym)
 253                 goto free;
 254
 255         FOR_EACH_PTR(call_backs, tmp) {
 256                 if (tmp->type != RANGED_CALL)
 257                         continue;
 258
 259                 if (in_list_exact_sval(handled_ranges, tmp->range))
 260                         continue;
 261                 __push_fake_cur_stree();
 262                 tack_on(&handled_ranges, tmp->range);
 263
 264                 same_range_call_backs = get_same_ranged_call_backs(call_backs, tmp->range);
 265                 call_ranged_call_backs(same_range_call_backs, fn, expr->right, expr);
 266                 __free_ptr_list((struct ptr_list **)&same_range_call_backs);
 267
 268                 estate = alloc_estate_range(tmp->range->min, tmp->range->max);
 269                 set_extra_mod(var_name, sym, estate);
 270
 271                 tmp_stree = __pop_fake_cur_stree();
 272                 merge_fake_stree(&final_states, tmp_stree);
 273                 free_stree(&tmp_stree);
 274                 handled = 1;
 275         } END_FOR_EACH_PTR(tmp);
 276
 277         FOR_EACH_SM(final_states, sm) {
 278                 __set_sm(sm);
 279         } END_FOR_EACH_SM(sm);
 280
 281         free_stree(&final_states);
 282 free:
 283         free_string(var_name);
 284         return handled;
 285 }
 286
 287 static int call_implies_callbacks(int comparison, struct expression *expr, sval_t sval, int left)
 288 {
 289         struct call_back_list *call_backs;
 290         struct fcall_back *tmp;
 291         const char *fn;
 292         struct data_range *value_range;
 293         struct stree *true_states = NULL;
 294         struct stree *false_states = NULL;
 295         struct stree *tmp_stree;
 296         struct sm_state *sm;
 297
 298         if (expr->fn->type != EXPR_SYMBOL || !expr->fn->symbol)
 299                 return 0;
 300         fn = expr->fn->symbol->ident->name;
 301         call_backs = search_callback(func_hash, (char *)expr->fn->symbol->ident->name);
 302         if (!call_backs)
 303                 return 0;
 304         value_range = alloc_range(sval, sval);
 305
 306         /* set true states */
 307         __push_fake_cur_stree();
 308         FOR_EACH_PTR(call_backs, tmp) {
 309                 if (tmp->type != RANGED_CALL)
 310                         continue;
 311                 if (!true_comparison_range_LR(comparison, tmp->range, value_range, left))
 312                         continue;
 313                 (tmp->u.ranged)(fn, expr, NULL, tmp->info);
 314         } END_FOR_EACH_PTR(tmp);
 315         tmp_stree = __pop_fake_cur_stree();
 316         merge_fake_stree(&true_states, tmp_stree);
 317         free_stree(&tmp_stree);
 318
 319         /* set false states */
 320         __push_fake_cur_stree();
 321         FOR_EACH_PTR(call_backs, tmp) {
 322                 if (tmp->type != RANGED_CALL)
 323                         continue;
 324                 if (!false_comparison_range_LR(comparison, tmp->range, value_range, left))
 325                         continue;
 326                 (tmp->u.ranged)(fn, expr, NULL, tmp->info);
 327         } END_FOR_EACH_PTR(tmp);
 328         tmp_stree = __pop_fake_cur_stree();
 329         merge_fake_stree(&false_states, tmp_stree);
 330         free_stree(&tmp_stree);
 331
 332         FOR_EACH_SM(true_states, sm) {
 333                 __set_true_false_sm(sm, NULL);
 334         } END_FOR_EACH_SM(sm);
 335         FOR_EACH_SM(false_states, sm) {
 336                 __set_true_false_sm(NULL, sm);
 337         } END_FOR_EACH_SM(sm);
 338
 339         free_stree(&true_states);
 340         free_stree(&false_states);
 341         return 1;
 342 }
 343
 344 struct db_callback_info {
 345         int true_side;
 346         int comparison;
 347         struct expression *expr;
 348         struct range_list *rl;
 349         int left;
 350         struct stree *stree;
 351         struct db_implies_list *callbacks;
 352 };
 353 static struct db_callback_info db_info;
 354 static int prev_return_id;
 355
 356 static int db_compare_callback(void *unused, int argc, char **argv, char **azColName)
 357 {
 358         struct range_list *ret_range;
 359         int type, param;
 360         char *key, *value;
 361         struct return_implies_callback *tmp;
 362         struct stree *stree;
 363         int return_id;
 364
 365         if (argc != 6)
 366                 return 0;
 367
 368         return_id = atoi(argv[0]);
 369         call_results_to_rl(db_info.expr, get_type(strip_expr(db_info.expr)), argv[1], &ret_range);
 370         ret_range = cast_rl(get_type(db_info.expr), ret_range);
 371         type = atoi(argv[2]);
 372         param = atoi(argv[3]);
 373         key = argv[4];
 374         value = argv[5];
 375
 376         if (db_info.true_side) {
 377                 if (!possibly_true_rl_LR(db_info.comparison,
 378                                                   ret_range, db_info.rl,
 379                                                   db_info.left))
 380                         return 0;
 381         } else {
 382                 if (!possibly_false_rl_LR(db_info.comparison,
 383                                                   ret_range, db_info.rl,
 384                                                   db_info.left))
 385                         return 0;
 386         }
 387
 388         if (prev_return_id != -1 && return_id != prev_return_id) {
 389                 stree = __pop_fake_cur_stree();
 390                 merge_fake_stree(&db_info.stree, stree);
 391                 free_stree(&stree);
 392                 __push_fake_cur_stree();
 393         }
 394         prev_return_id = return_id;
 395
 396         FOR_EACH_PTR(db_info.callbacks, tmp) {
 397                 if (tmp->type == type)
 398                         tmp->callback(db_info.expr, param, key, value);
 399         } END_FOR_EACH_PTR(tmp);
 400         return 0;
 401 }
 402
 403 void compare_db_return_states_callbacks(int comparison, struct expression *expr, sval_t sval, int left)
 404 {
 405         struct stree *stree;
 406         struct stree *true_states;
 407         struct stree *false_states;
 408         struct sm_state *sm;
 409
 410         db_info.comparison = comparison;
 411         db_info.expr = expr;
 412         db_info.rl = alloc_rl(sval, sval);
 413         db_info.left = left;
 414         db_info.callbacks = db_return_states_list;
 415
 416         call_return_states_before_hooks();
 417
 418         db_info.true_side = 1;
 419         db_info.stree = NULL;
 420         prev_return_id = -1;
 421         __push_fake_cur_stree();
 422         sql_select_return_states("return_id, return, type, parameter, key, value", expr,
 423                         db_compare_callback);
 424         stree = __pop_fake_cur_stree();
 425         merge_fake_stree(&db_info.stree, stree);
 426         free_stree(&stree);
 427         true_states = db_info.stree;
 428
 429         db_info.true_side = 0;
 430         db_info.stree = NULL;
 431         prev_return_id = -1;
 432         __push_fake_cur_stree();
 433         sql_select_return_states("return_id, return, type, parameter, key, value", expr,
 434                         db_compare_callback);
 435         stree = __pop_fake_cur_stree();
 436         merge_fake_stree(&db_info.stree, stree);
 437         free_stree(&stree);
 438         false_states = db_info.stree;
 439
 440         FOR_EACH_SM(true_states, sm) {
 441                 __set_true_false_sm(sm, NULL);
 442         } END_FOR_EACH_SM(sm);
 443         FOR_EACH_SM(false_states, sm) {
 444                 __set_true_false_sm(NULL, sm);
 445         } END_FOR_EACH_SM(sm);
 446
 447         call_return_states_after_hooks();
 448
 449         free_stree(&true_states);
 450         free_stree(&false_states);
 451 }
 452
 453 void function_comparison(int comparison, struct expression *expr, sval_t sval, int left)
 454 {
 455         if (call_implies_callbacks(comparison, expr, sval, left))
 456                 return;
 457         compare_db_return_states_callbacks(comparison, expr, sval, left);
 458 }
 459
 460 static int db_assign_return_states_callback(void *unused, int argc, char **argv, char **azColName)
 461 {
 462         struct range_list *ret_range;
 463         int type, param;
 464         char *key, *value;
 465         struct return_implies_callback *tmp;
 466         struct stree *stree;
 467         int return_id;
 468
 469         if (argc != 6)
 470                 return 0;
 471
 472         return_id = atoi(argv[0]);
 473         call_results_to_rl(db_info.expr->right, get_type(strip_expr(db_info.expr->right)), argv[1], &ret_range);
 474         __add_comparison_info(db_info.expr->left, strip_expr(db_info.expr->right), argv[1]);
 475         if (!ret_range)
 476                 ret_range = alloc_whole_rl(get_type(strip_expr(db_info.expr->right)));
 477         ret_range = cast_rl(get_type(db_info.expr->right), ret_range);
 478         type = atoi(argv[2]);
 479         param = atoi(argv[3]);
 480         key = argv[4];
 481         value = argv[5];
 482
 483         if (prev_return_id != -1 && return_id != prev_return_id) {
 484                 stree = __pop_fake_cur_stree();
 485                 merge_fake_stree(&db_info.stree, stree);
 486                 free_stree(&stree);
 487                 __push_fake_cur_stree();
 488         }
 489         prev_return_id = return_id;
 490
 491         FOR_EACH_PTR(db_return_states_list, tmp) {
 492                 if (tmp->type == type)
 493                         tmp->callback(db_info.expr, param, key, value);
 494         } END_FOR_EACH_PTR(tmp);
 495         ret_range = cast_rl(get_type(db_info.expr->left), ret_range);
 496         set_extra_expr_mod(db_info.expr->left, alloc_estate_rl(ret_range));
 497
 498         return 0;
 499 }
 500
 501 static int db_return_states_assign(struct expression *expr)
 502 {
 503         struct expression *right;
 504         struct sm_state *sm;
 505         struct stree *stree;
 506         int handled = 0;
 507
 508         right = strip_expr(expr->right);
 509
 510         prev_return_id = -1;
 511         db_info.expr = expr;
 512         db_info.stree = NULL;
 513
 514         call_return_states_before_hooks();
 515
 516         __push_fake_cur_stree();
 517         sql_select_return_states("return_id, return, type, parameter, key, value",
 518                         right, db_assign_return_states_callback);
 519         stree = __pop_fake_cur_stree();
 520         merge_fake_stree(&db_info.stree, stree);
 521         free_stree(&stree);
 522
 523         FOR_EACH_SM(db_info.stree, sm) {
 524                 __set_sm(sm);
 525                 handled = 1;
 526         } END_FOR_EACH_SM(sm);
 527
 528         free_stree(&db_info.stree);
 529         call_return_states_after_hooks();
 530
 531         return handled;
 532 }
 533
 534 static int handle_implied_return(struct expression *expr)
 535 {
 536         struct range_list *rl;
 537
 538         if (!get_implied_return(expr->right, &rl))
 539                 return 0;
 540         rl = cast_rl(get_type(expr->left), rl);
 541         set_extra_expr_mod(expr->left, alloc_estate_rl(rl));
 542         return 1;
 543 }
 544
 545 static void match_assign_call(struct expression *expr)
 546 {
 547         struct call_back_list *call_backs;
 548         const char *fn;
 549         struct expression *right;
 550         int handled = 0;
 551         struct range_list *rl;
 552
 553         if (expr->op != '=')
 554                 return;
 555
 556         right = strip_expr(expr->right);
 557         if (right->fn->type != EXPR_SYMBOL || !right->fn->symbol) {
 558                 handled |= db_return_states_assign(expr);
 559                 if (!handled)
 560                         goto assigned_unknown;
 561                 return;
 562         }
 563         if (is_fake_call(right)) {
 564                 if (!get_state_expr(SMATCH_EXTRA, expr->left))
 565                         return;
 566                 set_extra_expr_mod(expr->left, alloc_estate_whole(get_type(expr->left)));
 567                 return;
 568         }
 569
 570         fn = right->fn->symbol->ident->name;
 571
 572         /*
 573          * some of these conflict (they try to set smatch extra twice), so we
 574          * call them in order from least important to most important.
 575          */
 576
 577         call_backs = search_callback(func_hash, (char *)fn);
 578         call_call_backs(call_backs, ASSIGN_CALL, fn, expr);
 579
 580         handled |= db_return_states_assign(expr);
 581         handled |= assign_ranged_funcs(fn, expr, call_backs);
 582         handled |= handle_implied_return(expr);
 583
 584         if (handled)
 585                 return;
 586
 587 assigned_unknown:
 588         get_absolute_rl(expr->right, &rl);
 589         rl = cast_rl(get_type(expr->left), rl);
 590         set_extra_expr_mod(expr->left, alloc_estate_rl(rl));
 591 }
 592
 593 static int db_return_states_callback(void *unused, int argc, char **argv, char **azColName)
 594 {
 595         struct range_list *ret_range;
 596         int type, param;
 597         char *key, *value;
 598         struct return_implies_callback *tmp;
 599         struct stree *stree;
 600         int return_id;
 601
 602         if (argc != 6)
 603                 return 0;
 604
 605         return_id = atoi(argv[0]);
 606         call_results_to_rl(db_info.expr, get_type(strip_expr(db_info.expr)), argv[1], &ret_range);
 607         ret_range = cast_rl(get_type(db_info.expr), ret_range);
 608         type = atoi(argv[2]);
 609         param = atoi(argv[3]);
 610         key = argv[4];
 611         value = argv[5];
 612
 613         if (prev_return_id != -1 && return_id != prev_return_id) {
 614                 stree = __pop_fake_cur_stree();
 615                 merge_fake_stree(&db_info.stree, stree);
 616                 free_stree(&stree);
 617                 __push_fake_cur_stree();
 618                 __unnullify_path();
 619         }
 620         prev_return_id = return_id;
 621
 622         FOR_EACH_PTR(db_return_states_list, tmp) {
 623                 if (tmp->type == type)
 624                         tmp->callback(db_info.expr, param, key, value);
 625         } END_FOR_EACH_PTR(tmp);
 626
 627         return 0;
 628 }
 629
 630 static void db_return_states(struct expression *expr)
 631 {
 632         struct sm_state *sm;
 633         struct stree *stree;
 634
 635         if (!__get_cur_stree())  /* no return functions */
 636                 return;
 637
 638         prev_return_id = -1;
 639         db_info.expr = expr;
 640         db_info.stree = NULL;
 641
 642         call_return_states_before_hooks();
 643
 644         __push_fake_cur_stree();
 645         __unnullify_path();
 646         sql_select_return_states("return_id, return, type, parameter, key, value",
 647                         expr, db_return_states_callback);
 648         stree = __pop_fake_cur_stree();
 649         merge_fake_stree(&db_info.stree, stree);
 650         free_stree(&stree);
 651
 652         FOR_EACH_SM(db_info.stree, sm) {
 653                 __set_sm(sm);
 654         } END_FOR_EACH_SM(sm);
 655
 656         free_stree(&db_info.stree);
 657         call_return_states_after_hooks();
 658 }
 659
 660 static int is_assigned_call(struct expression *expr)
 661 {
 662         struct expression *tmp;
 663
 664         FOR_EACH_PTR_REVERSE(big_expression_stack, tmp) {
 665                 if (tmp->type == EXPR_ASSIGNMENT && strip_expr(tmp->right) == expr)
 666                         return 1;
 667                 if (tmp->pos.line < expr->pos.line)
 668                         return 0;
 669         } END_FOR_EACH_PTR_REVERSE(tmp);
 670         return 0;
 671 }
 672
 673 static void db_return_states_call(struct expression *expr)
 674 {
 675         if (is_assigned_call(expr))
 676                 return;
 677         db_return_states(expr);
 678 }
 679
 680 static void match_function_call(struct expression *expr)
 681 {
 682         struct call_back_list *call_backs;
 683
 684         if (expr->fn->type == EXPR_SYMBOL && expr->fn->symbol) {
 685                 call_backs = search_callback(func_hash, (char *)expr->fn->symbol->ident->name);
 686                 if (call_backs)
 687                         call_call_backs(call_backs, REGULAR_CALL,
 688                                         expr->fn->symbol->ident->name, expr);
 689         }
 690         db_return_states_call(expr);
 691 }
 692
 693 static void match_macro_assign(struct expression *expr)
 694 {
 695         struct call_back_list *call_backs;
 696         const char *macro;
 697         struct expression *right;
 698
 699         right = strip_expr(expr->right);
 700         macro = get_macro_name(right->pos);
 701         call_backs = search_callback(func_hash, (char *)macro);
 702         if (!call_backs)
 703                 return;
 704         call_call_backs(call_backs, MACRO_ASSIGN, macro, expr);
 705         call_call_backs(call_backs, MACRO_ASSIGN_EXTRA, macro, expr);
 706 }
 707
 708 int get_implied_return(struct expression *expr, struct range_list **rl)
 709 {
 710         struct call_back_list *call_backs;
 711         struct fcall_back *tmp;
 712         int handled = 0;
 713         char *fn;
 714
 715         *rl = NULL;
 716
 717         expr = strip_expr(expr);
 718         fn = expr_to_var(expr->fn);
 719         if (!fn)
 720                 goto out;
 721
 722         call_backs = search_callback(func_hash, fn);
 723
 724         FOR_EACH_PTR(call_backs, tmp) {
 725                 if (tmp->type == IMPLIED_RETURN) {
 726                         (tmp->u.implied_return)(expr, tmp->info, rl);
 727                         handled = 1;
 728                 }
 729         } END_FOR_EACH_PTR(tmp);
 730
 731 out:
 732         free_string(fn);
 733         return handled;
 734 }
 735
 736 void create_function_hook_hash(void)
 737 {
 738         func_hash = create_function_hashtable(5000);
 739 }
 740
 741 void register_function_hooks(int id)
 742 {
 743         add_hook(&match_function_call, CALL_HOOK_AFTER_INLINE);
 744         add_hook(&match_assign_call, CALL_ASSIGNMENT_HOOK);
 745         add_hook(&match_macro_assign, MACRO_ASSIGNMENT_HOOK);
 746 }