smatch_buf_size.c

   1 /*
   2  * Copyright (C) 2010 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 #include <stdlib.h>
  19 #include <errno.h>
  20 #include "parse.h"
  21 #include "smatch.h"
  22 #include "smatch_slist.h"
  23 #include "smatch_extra.h"
  24 #include "smatch_function_hashtable.h"
  25
  26 #define UNKNOWN_SIZE -1
  27
  28 static int my_size_id;
  29
  30 static DEFINE_HASHTABLE_INSERT(insert_func, char, int);
  31 static DEFINE_HASHTABLE_SEARCH(search_func, char, int);
  32 static struct hashtable *allocation_funcs;
  33
  34 static char *get_fn_name(struct expression *expr)
  35 {
  36         if (expr->type != EXPR_CALL)
  37                 return NULL;
  38         if (expr->fn->type != EXPR_SYMBOL)
  39                 return NULL;
  40         return expr_to_var(expr->fn);
  41 }
  42
  43 static int is_allocation_function(struct expression *expr)
  44 {
  45         char *func;
  46         int ret = 0;
  47
  48         func = get_fn_name(expr);
  49         if (!func)
  50                 return 0;
  51         if (search_func(allocation_funcs, func))
  52                 ret = 1;
  53         free_string(func);
  54         return ret;
  55 }
  56
  57 static void add_allocation_function(const char *func, void *call_back, int param)
  58 {
  59         insert_func(allocation_funcs, (char *)func, (int *)1);
  60         add_function_assign_hook(func, call_back, INT_PTR(param));
  61 }
  62
  63 static struct smatch_state *size_to_estate(int size)
  64 {
  65         sval_t sval;
  66
  67         sval.type = &int_ctype;
  68         sval.value = size;
  69
  70         return alloc_estate_sval(sval);
  71 }
  72
  73 static struct range_list *size_to_rl(int size)
  74 {
  75         sval_t sval;
  76
  77         sval.type = &int_ctype;
  78         sval.value = size;
  79
  80         return alloc_rl(sval, sval);
  81 }
  82
  83 static struct smatch_state *unmatched_size_state(struct sm_state *sm)
  84 {
  85         return size_to_estate(UNKNOWN_SIZE);
  86 }
  87
  88 static void set_size_undefined(struct sm_state *sm, struct expression *mod_expr)
  89 {
  90         set_state(sm->owner, sm->name, sm->sym, size_to_estate(UNKNOWN_SIZE));
  91 }
  92
  93 static struct smatch_state *merge_size_func(struct smatch_state *s1, struct smatch_state *s2)
  94 {
  95         return merge_estates(s1, s2);
  96 }
  97
  98 void set_param_buf_size(const char *name, struct symbol *sym, char *key, char *value)
  99 {
 100         struct range_list *rl = NULL;
 101         struct smatch_state *state;
 102         char fullname[256];
 103
 104         if (strncmp(key, "$", 1) != 0)
 105                 return;
 106
 107         snprintf(fullname, 256, "%s%s", name, key + 1);
 108
 109         str_to_rl(&int_ctype, value, &rl);
 110         if (!rl || is_whole_rl(rl))
 111                 return;
 112         state = alloc_estate_rl(rl);
 113         set_state(my_size_id, fullname, sym, state);
 114 }
 115
 116 static int bytes_per_element(struct expression *expr)
 117 {
 118         struct symbol *type;
 119
 120         if (!expr)
 121                 return 0;
 122         if (expr->type == EXPR_STRING)
 123                 return expr->wide + 1;
 124         if (expr->type == EXPR_PREOP && expr->op == '&') {
 125                 type = get_type(expr->unop);
 126                 if (type && type->type == SYM_ARRAY)
 127                         expr = expr->unop;
 128         }
 129         type = get_type(expr);
 130         if (!type)
 131                 return 0;
 132
 133         if (type->type != SYM_PTR && type->type != SYM_ARRAY)
 134                 return 0;
 135
 136         type = get_base_type(type);
 137         return type_bytes(type);
 138 }
 139
 140 int bytes_to_elements(struct expression *expr, int bytes)
 141 {
 142         int bpe;
 143
 144         bpe = bytes_per_element(expr);
 145         if (bpe == 0)
 146                 return 0;
 147         return bytes / bpe;
 148 }
 149
 150 static int elements_to_bytes(struct expression *expr, int elements)
 151 {
 152         int bpe;
 153
 154         bpe = bytes_per_element(expr);
 155         return elements * bpe;
 156 }
 157
 158 static int get_initializer_size(struct expression *expr)
 159 {
 160         switch (expr->type) {
 161         case EXPR_STRING:
 162                 return expr->string->length * (expr->wide + 1);
 163         case EXPR_INITIALIZER: {
 164                 struct expression *tmp;
 165                 int i = 0;
 166
 167                 FOR_EACH_PTR(expr->expr_list, tmp) {
 168                         if (tmp->type == EXPR_INDEX) {
 169                                 if (tmp->idx_to >= i)
 170                                         i = tmp->idx_to;
 171                                 else
 172                                         continue;
 173                         }
 174
 175                         i++;
 176                 } END_FOR_EACH_PTR(tmp);
 177                 return i;
 178         }
 179         case EXPR_SYMBOL:
 180                 return get_array_size(expr);
 181         }
 182         return 0;
 183 }
 184
 185 static struct range_list *db_size_rl;
 186 static int db_size_callback(void *unused, int argc, char **argv, char **azColName)
 187 {
 188         struct range_list *tmp = NULL;
 189
 190         if (!db_size_rl) {
 191                 str_to_rl(&int_ctype, argv[0], &db_size_rl);
 192         } else {
 193                 str_to_rl(&int_ctype, argv[0], &tmp);
 194                 db_size_rl = rl_union(db_size_rl, tmp);
 195         }
 196         return 0;
 197 }
 198
 199 static struct expression *cached_type_expr, *cached_sym_expr;
 200 static struct range_list *cached_type_rl, *cached_sym_rl;
 201
 202 static void match_clear_cache(struct symbol *sym)
 203 {
 204         cached_type_expr = NULL;
 205         cached_type_rl = NULL;
 206         cached_sym_expr = NULL;
 207         cached_sym_rl = NULL;
 208 }
 209
 210 static char *get_stored_buffer_name(struct expression *buffer, bool *add_static)
 211 {
 212         struct expression *array;
 213         char buf[64];
 214         char *name;
 215
 216         *add_static = false;
 217
 218         name = get_member_name(buffer);
 219         if (name)
 220                 return name;
 221
 222         if (is_static(buffer)) {
 223                 name = expr_to_var(buffer);
 224                 if (!name)
 225                         return NULL;
 226                 *add_static = 1;
 227                 return name;
 228         }
 229
 230         array = get_array_base(buffer);
 231         if (array) {
 232                 name = get_member_name(array);
 233                 if (!name)
 234                         return NULL;
 235                 snprintf(buf, sizeof(buf), "%s[]", name);
 236                 free_string(name);
 237                 return alloc_string(buf);
 238         }
 239
 240         return NULL;
 241 }
 242
 243 static struct range_list *size_from_db_type(struct expression *expr)
 244 {
 245         bool this_file_only;
 246         char *name;
 247
 248         name = get_stored_buffer_name(expr, &this_file_only);
 249         if (!name)
 250                 return NULL;
 251
 252         if (expr == cached_type_expr)
 253                 return clone_rl(cached_type_rl);
 254         cached_type_expr = expr;
 255         cached_type_rl = NULL;
 256
 257         if (this_file_only) {
 258                 db_size_rl = NULL;
 259                 run_sql(db_size_callback, NULL,
 260                         "select size from function_type_size where type = '%s' and file = %d;",
 261                         name, get_file_id());
 262                 cached_type_rl = clone_rl(db_size_rl);
 263                 return db_size_rl;
 264         }
 265
 266         db_size_rl = NULL;
 267         run_sql(db_size_callback, NULL,
 268                 "select size from type_size where type = '%s';",
 269                 name);
 270         cached_type_rl = clone_rl(db_size_rl);
 271         return db_size_rl;
 272 }
 273
 274 static struct range_list *size_from_db_symbol(struct expression *expr)
 275 {
 276         struct symbol *sym;
 277
 278         if (expr->type != EXPR_SYMBOL)
 279                 return NULL;
 280         sym = expr->symbol;
 281         if (!sym || !sym->ident ||
 282             !(sym->ctype.modifiers & MOD_TOPLEVEL) ||
 283             sym->ctype.modifiers & MOD_STATIC)
 284                 return NULL;
 285
 286         if (expr == cached_sym_expr)
 287                 return clone_rl(cached_sym_rl);
 288         cached_sym_expr = expr;
 289         cached_sym_rl = NULL;
 290
 291         db_size_rl = NULL;
 292         run_sql(db_size_callback, NULL,
 293                 "select value from data_info where file = 0 and data = '%s' and type = %d;",
 294                 sym->ident->name, BUF_SIZE);
 295         cached_sym_rl = clone_rl(db_size_rl);
 296         return db_size_rl;
 297 }
 298
 299 static struct range_list *size_from_db(struct expression *expr)
 300 {
 301         struct range_list *rl;
 302
 303         rl = size_from_db_symbol(expr);
 304         if (rl)
 305                 return rl;
 306         return size_from_db_type(expr);
 307 }
 308
 309 static void db_returns_buf_size(struct expression *expr, int param, char *unused, char *math)
 310 {
 311         struct expression *call;
 312         struct range_list *rl;
 313         sval_t sval;
 314
 315         if (expr->type != EXPR_ASSIGNMENT)
 316                 return;
 317         call = strip_expr(expr->right);
 318
 319         call_results_to_rl(call, &int_ctype, math, &rl);
 320         rl = cast_rl(&int_ctype, rl);
 321         if (rl_to_sval(rl, &sval) && sval.value == 0)
 322                 return;
 323         set_state_expr(my_size_id, expr->left, alloc_estate_rl(rl));
 324 }
 325
 326 static int get_real_array_size_from_type(struct symbol *type)
 327 {
 328         sval_t sval;
 329
 330         if (!type)
 331                 return 0;
 332         if (!type || type->type != SYM_ARRAY)
 333                 return 0;
 334
 335         if (!get_implied_value(type->array_size, &sval))
 336                 return 0;
 337
 338         return sval.value;
 339 }
 340
 341 int get_real_array_size(struct expression *expr)
 342 {
 343         if (!expr)
 344                 return 0;
 345         if (expr->type == EXPR_PREOP && expr->op == '&')
 346                 expr = expr->unop;
 347         if (expr->type == EXPR_BINOP) /* array elements foo[5] */
 348                 return 0;
 349         return get_real_array_size_from_type(get_type(expr));
 350 }
 351
 352 static int get_size_from_initializer(struct expression *expr)
 353 {
 354         if (expr->type != EXPR_SYMBOL || !expr->symbol || !expr->symbol->initializer)
 355                 return 0;
 356         if (expr->symbol->initializer == expr) /* int a = a; */
 357                 return 0;
 358         return get_initializer_size(expr->symbol->initializer);
 359 }
 360
 361 static struct range_list *get_stored_size_bytes(struct expression *expr)
 362 {
 363         struct smatch_state *state;
 364
 365         state = get_state_expr(my_size_id, expr);
 366         if (!state)
 367                 return NULL;
 368         return estate_rl(state);
 369 }
 370
 371 static int get_bytes_from_address(struct expression *expr)
 372 {
 373         struct symbol *type;
 374         int ret;
 375
 376         if (expr->type != EXPR_PREOP || expr->op != '&')
 377                 return 0;
 378         type = get_type(expr);
 379         if (!type)
 380                 return 0;
 381
 382         if (type->type == SYM_PTR)
 383                 type = get_base_type(type);
 384
 385         ret = type_bytes(type);
 386         if (ret == 1)
 387                 return 0;  /* ignore char pointers */
 388
 389         return ret;
 390 }
 391
 392 static struct expression *remove_addr_fluff(struct expression *expr)
 393 {
 394         struct expression *tmp;
 395         sval_t sval;
 396
 397         expr = strip_expr(expr);
 398
 399         /* remove '&' and '*' operations that cancel */
 400         while (expr && expr->type == EXPR_PREOP && expr->op == '&') {
 401                 tmp = strip_expr(expr->unop);
 402                 if (tmp->type != EXPR_PREOP)
 403                         break;
 404                 if (tmp->op != '*')
 405                         break;
 406                 expr = strip_expr(tmp->unop);
 407         }
 408
 409         if (!expr)
 410                 return NULL;
 411
 412         /* "foo + 0" is just "foo" */
 413         if (expr->type == EXPR_BINOP && expr->op == '+' &&
 414             get_value(expr->right, &sval) && sval.value == 0)
 415                 return expr->left;
 416
 417         return expr;
 418 }
 419
 420 static int is_last_member_of_struct(struct symbol *sym, struct ident *member)
 421 {
 422         struct symbol *tmp;
 423         int i;
 424
 425         if (sym->type != SYM_STRUCT)
 426                 return false;
 427
 428         i = 0;
 429         FOR_EACH_PTR_REVERSE(sym->symbol_list, tmp) {
 430                 if (i++ || !tmp->ident)
 431                         return 0;
 432                 if (tmp->ident == member)
 433                         return 1;
 434                 return 0;
 435         } END_FOR_EACH_PTR_REVERSE(tmp);
 436
 437         return 0;
 438 }
 439
 440 int last_member_is_resizable(struct symbol *sym)
 441 {
 442         struct symbol *last_member;
 443         struct symbol *type;
 444         sval_t sval;
 445
 446         if (!sym || sym->type != SYM_STRUCT)
 447                 return 0;
 448
 449         last_member = last_ptr_list((struct ptr_list *)sym->symbol_list);
 450         if (!last_member || !last_member->ident)
 451                 return 0;
 452
 453         type = get_real_base_type(last_member);
 454         if (type->type == SYM_STRUCT)
 455                 return last_member_is_resizable(type);
 456         if (type->type != SYM_ARRAY)
 457                 return 0;
 458
 459         if (!type->array_size)
 460                 return 1;
 461
 462         if (!get_implied_value(type->array_size, &sval))
 463                 return 0;
 464
 465         if (sval.value != 0 && sval.value != 1)
 466                 return 0;
 467
 468         return 1;
 469 }
 470
 471 static struct range_list *get_stored_size_end_struct_bytes(struct expression *expr)
 472 {
 473         struct symbol *sym;
 474         struct symbol *base_sym;
 475         struct smatch_state *state;
 476         sval_t base_size = {
 477                 .type = &int_ctype,
 478         };
 479
 480         if (expr->type == EXPR_BINOP) /* array elements foo[5] */
 481                 return NULL;
 482
 483         if (expr->type == EXPR_PREOP && expr->op == '&')
 484                 expr = strip_parens(expr->unop);
 485
 486         sym = expr_to_sym(expr);
 487         if (!sym || !sym->ident)
 488                 return NULL;
 489         if (!type_bytes(sym))
 490                 return NULL;
 491         if (sym->type != SYM_NODE)
 492                 return NULL;
 493
 494         base_sym = get_real_base_type(sym);
 495         if (!base_sym || base_sym->type != SYM_PTR)
 496                 return NULL;
 497         base_sym = get_real_base_type(base_sym);
 498         if (!base_sym || base_sym->type != SYM_STRUCT)
 499                 return NULL;
 500
 501         if (!is_last_member_of_struct(base_sym, expr->member))
 502                 return NULL;
 503         if (!last_member_is_resizable(base_sym))
 504                 return NULL;
 505
 506         state = get_state(my_size_id, sym->ident->name, sym);
 507         if (!estate_rl(state))
 508                 return NULL;
 509         base_size.value = type_bytes(base_sym) - type_bytes(get_type(expr));
 510         if (sval_cmp(estate_min(state), base_size) < 0)
 511                 return NULL;
 512
 513         return rl_binop(estate_rl(state), '-', alloc_rl(base_size, base_size));
 514 }
 515
 516 static int get_last_element_bytes(struct expression *expr)
 517 {
 518         struct symbol *type, *member, *member_type;
 519
 520         type = get_type(expr);
 521         if (!type || type->type != SYM_STRUCT)
 522                 return 0;
 523         member = last_ptr_list((struct ptr_list *)type->symbol_list);
 524         if (!member)
 525                 return 0;
 526         member_type = get_real_base_type(member);
 527         if (!member_type)
 528                 return 0;
 529         return type_bytes(member_type);
 530 }
 531
 532 static struct range_list *get_outer_end_struct_bytes(struct expression *expr)
 533 {
 534         struct expression *outer;
 535         int last_element_size;
 536         struct symbol *type;
 537         sval_t bytes = {
 538                 .type = &int_ctype,
 539         };
 540
 541         /*
 542          * What we're checking here is for when "u.bar.baz" is a zero element
 543          * array and "u" has a buffer to determine the space for baz.  Like
 544          * this:
 545          * struct { struct bar bar; char buf[256]; } u;
 546          *
 547          */
 548
 549         if (expr->type == EXPR_PREOP && expr->op == '&')
 550                 expr = strip_parens(expr->unop);
 551         if (expr->type != EXPR_DEREF)
 552                 return NULL;
 553         outer = expr->deref;
 554         if (outer->type != EXPR_DEREF)
 555                 return NULL;
 556         type = get_type(outer);
 557         if (!type)
 558                 return NULL;
 559
 560         if (!is_last_member_of_struct(type, expr->member))
 561                 return NULL;
 562         if (!last_member_is_resizable(type))
 563                 return NULL;
 564
 565         last_element_size = get_last_element_bytes(outer->deref);
 566         if (last_element_size == 0)
 567                 return NULL;
 568         if (type_bytes(get_type(outer)) + last_element_size !=
 569             type_bytes(get_type(outer->deref)))
 570                 return NULL;
 571
 572         bytes.value = last_element_size;
 573         return alloc_rl(bytes, bytes);
 574 }
 575
 576 static struct range_list *alloc_int_rl(int value)
 577 {
 578         sval_t sval = {
 579                 .type = &int_ctype,
 580                 {.value = value},
 581         };
 582
 583         return alloc_rl(sval, sval);
 584 }
 585
 586 struct range_list *filter_size_rl(struct range_list *rl)
 587 {
 588         sval_t minus_one = {
 589                 .type = &int_ctype,
 590                 { .value = -1 },
 591         };
 592
 593         sval_t zero = {
 594                 .type = &int_ctype,
 595                 { .value = 0 },
 596         };
 597
 598         return remove_range(rl, minus_one, zero);
 599 }
 600
 601 struct range_list *get_array_size_bytes_rl(struct expression *expr)
 602 {
 603         struct range_list *ret = NULL;
 604         sval_t sval;
 605         int size;
 606
 607         if (is_fake_call(expr))
 608                 return NULL;
 609
 610         expr = remove_addr_fluff(expr);
 611         if (!expr)
 612                 return NULL;
 613
 614         /* "BAR" */
 615         if (expr->type == EXPR_STRING)
 616                 return alloc_int_rl(expr->string->length * (expr->wide + 1));
 617
 618         if (expr->type == EXPR_BINOP && expr->op == '+') {
 619                 sval_t offset;
 620                 struct symbol *type;
 621                 int bytes;
 622
 623                 if (!get_implied_value(expr->right, &offset))
 624                         return NULL;
 625                 type = get_type(expr->left);
 626                 if (!type)
 627                         return NULL;
 628                 if (type->type != SYM_ARRAY && type->type != SYM_PTR)
 629                         return NULL;
 630                 type = get_real_base_type(type);
 631                 bytes = type_bytes(type);
 632                 if (bytes == 0)
 633                         return NULL;
 634                 offset.value *= bytes;
 635                 size = get_array_size_bytes(expr->left);
 636                 if (size <= 0)
 637                         return NULL;
 638                 return alloc_int_rl(size - offset.value);
 639         }
 640
 641         /* buf = malloc(1024); */
 642         ret = get_stored_size_bytes(expr);
 643         if (ret)
 644                 return ret;
 645
 646         ret = get_stored_size_end_struct_bytes(expr);
 647         if (ret)
 648                 return ret;
 649
 650         ret = get_outer_end_struct_bytes(expr);
 651         if (ret)
 652                 return ret;
 653
 654         /* buf[4] */
 655         size = get_real_array_size(expr);
 656         if (size)
 657                 return alloc_int_rl(elements_to_bytes(expr, size));
 658
 659         /* char *foo = "BAR" */
 660         size = get_size_from_initializer(expr);
 661         if (size)
 662                 return alloc_int_rl(elements_to_bytes(expr, size));
 663
 664         size = get_bytes_from_address(expr);
 665         if (size)
 666                 return alloc_int_rl(size);
 667
 668         ret = size_from_db(expr);
 669         return filter_size_rl(ret);
 670
 671         if (rl_to_sval(ret, &sval) && sval.value == -1)
 672                 return NULL;
 673         if (ret)
 674                 return ret;
 675
 676         return NULL;
 677 }
 678
 679 int get_array_size_bytes(struct expression *expr)
 680 {
 681         struct range_list *rl;
 682         sval_t sval;
 683
 684         rl = get_array_size_bytes_rl(expr);
 685         if (!rl_to_sval(rl, &sval))
 686                 return 0;
 687         if (sval.uvalue >= INT_MAX)
 688                 return 0;
 689         return sval.value;
 690 }
 691
 692 int get_array_size_bytes_max(struct expression *expr)
 693 {
 694         struct range_list *rl;
 695         sval_t bytes;
 696
 697         rl = get_array_size_bytes_rl(expr);
 698         if (!rl)
 699                 return 0;
 700         bytes = rl_min(rl);
 701         if (bytes.value < 0)
 702                 return 0;
 703         bytes = rl_max(rl);
 704         if (bytes.uvalue >= INT_MAX)
 705                 return 0;
 706         return bytes.value;
 707 }
 708
 709 int get_array_size_bytes_min(struct expression *expr)
 710 {
 711         struct range_list *rl;
 712         struct data_range *range;
 713
 714         rl = get_array_size_bytes_rl(expr);
 715         if (!rl)
 716                 return 0;
 717
 718         FOR_EACH_PTR(rl, range) {
 719                 if (range->min.value <= 0)
 720                         return 0;
 721                 if (range->max.value <= 0)
 722                         return 0;
 723                 if (range->min.uvalue >= INT_MAX)
 724                         return 0;
 725                 return range->min.value;
 726         } END_FOR_EACH_PTR(range);
 727
 728         return 0;
 729 }
 730
 731 int get_array_size(struct expression *expr)
 732 {
 733         if (!expr)
 734                 return 0;
 735         return bytes_to_elements(expr, get_array_size_bytes_max(expr));
 736 }
 737
 738 static struct expression *strip_ampersands(struct expression *expr)
 739 {
 740         struct symbol *type;
 741
 742         if (expr->type != EXPR_PREOP)
 743                 return expr;
 744         if (expr->op != '&')
 745                 return expr;
 746         type = get_type(expr->unop);
 747         if (!type || type->type != SYM_ARRAY)
 748                 return expr;
 749         return expr->unop;
 750 }
 751
 752 static void info_record_alloction(struct expression *buffer, struct range_list *rl)
 753 {
 754         bool add_static;
 755         char *name;
 756
 757         if (!option_info)
 758                 return;
 759
 760         name = get_stored_buffer_name(buffer, &add_static);
 761         if (!name)
 762                 return;
 763         if (rl && !is_whole_rl(rl))
 764                 sql_insert_function_type_size(name, show_rl(rl));
 765         else
 766                 sql_insert_function_type_size(name, "(-1)");
 767
 768         free_string(name);
 769 }
 770
 771 static void store_alloc(struct expression *expr, struct range_list *rl)
 772 {
 773         struct symbol *type;
 774
 775         rl = clone_rl(rl); // FIXME!!!
 776         if (!rl)
 777                 rl = size_to_rl(UNKNOWN_SIZE);
 778
 779         if (rl_min(rl).value != UNKNOWN_SIZE ||
 780             rl_max(rl).value != UNKNOWN_SIZE ||
 781             get_state_expr(my_size_id, expr))
 782                 set_state_expr(my_size_id, expr, alloc_estate_rl(rl));
 783
 784         type = get_type(expr);
 785         if (!type)
 786                 return;
 787         if (type->type != SYM_PTR)
 788                 return;
 789         type = get_real_base_type(type);
 790         if (!type)
 791                 return;
 792 // Why not store the size for void pointers?
 793         if (type == &void_ctype)
 794                 return;
 795 // Because of 4 files that produce 30 warnings like this:
 796 // arch/x86/crypto/des3_ede_glue.c:145 __cbc_encrypt() warn: potential memory corrupting cast 8 vs 1 bytes
 797         if (type->type != SYM_BASETYPE && type->type != SYM_PTR)
 798                 return;
 799
 800         info_record_alloction(expr, rl);
 801 }
 802
 803 static bool is_array_base(struct expression *expr)
 804 {
 805         struct symbol *type;
 806
 807         type = get_type(expr);
 808         if (type && type->type == SYM_ARRAY)
 809                 return true;
 810         return false;
 811 }
 812
 813 static void match_array_assignment(struct expression *expr)
 814 {
 815         struct expression *left;
 816         struct expression *right;
 817         char *left_member, *right_member;
 818         struct range_list *rl;
 819         sval_t sval;
 820
 821         if (expr->op != '=')
 822                 return;
 823
 824         left = strip_expr(expr->left);
 825         right = strip_expr(expr->right);
 826         right = strip_ampersands(right);
 827
 828         if (!is_pointer(left) && !get_state_expr(my_size_id, expr))
 829                 return;
 830
 831         /* char buf[24] = "str"; */
 832         if (is_array_base(left))
 833                 return;
 834         if (is_allocation_function(right))
 835                 return;
 836
 837         left_member = get_member_name(left);
 838         right_member = get_member_name(right);
 839         if (left_member && right_member && strcmp(left_member, right_member) == 0) {
 840                 free_string(left_member);
 841                 free_string(right_member);
 842                 return;
 843         }
 844         free_string(left_member);
 845         free_string(right_member);
 846
 847         if (get_implied_value(right, &sval) && sval.value == 0) {
 848                 rl = alloc_int_rl(0);
 849                 goto store;
 850         }
 851
 852         rl = get_array_size_bytes_rl(right);
 853         if (!rl && __in_fake_assign)
 854                 return;
 855
 856 store:
 857         store_alloc(left, rl);
 858 }
 859
 860 static void match_alloc(const char *fn, struct expression *expr, void *_size_arg)
 861 {
 862         int size_arg = PTR_INT(_size_arg);
 863         struct expression *right;
 864         struct expression *arg;
 865         struct range_list *rl;
 866
 867         right = strip_expr(expr->right);
 868         arg = get_argument_from_call_expr(right->args, size_arg);
 869         get_absolute_rl(arg, &rl);
 870         rl = cast_rl(&ulong_ctype, rl);
 871         store_alloc(expr->left, rl);
 872 }
 873
 874 static void match_calloc(const char *fn, struct expression *expr, void *_param)
 875 {
 876         struct expression *right;
 877         struct expression *size, *nr, *mult;
 878         struct range_list *rl;
 879         int param = PTR_INT(_param);
 880
 881         right = strip_expr(expr->right);
 882         nr = get_argument_from_call_expr(right->args, param);
 883         size = get_argument_from_call_expr(right->args, param + 1);
 884         mult = binop_expression(nr, '*', size);
 885         if (get_implied_rl(mult, &rl))
 886                 store_alloc(expr->left, rl);
 887         else
 888                 store_alloc(expr->left, size_to_rl(UNKNOWN_SIZE));
 889 }
 890
 891 static void match_page(const char *fn, struct expression *expr, void *_unused)
 892 {
 893         sval_t page_size = {
 894                 .type = &int_ctype,
 895                 {.value = 4096},
 896         };
 897
 898         store_alloc(expr->left, alloc_rl(page_size, page_size));
 899 }
 900
 901 static void match_bitmap_alloc(const char *fn, struct expression *expr, void *_size_arg)
 902 {
 903         int size_arg = PTR_INT(_size_arg);
 904         struct expression *right;
 905         struct expression *arg;
 906         struct range_list *rl;
 907         sval_t int_8 = {
 908                 .type = &int_ctype,
 909                 .value = 8,
 910         };
 911
 912         right = strip_expr(expr->right);
 913         arg = get_argument_from_call_expr(right->args, size_arg);
 914         get_absolute_rl(arg, &rl);
 915         if (rl_max(rl).uvalue <= SHRT_MAX && rl_max(rl).uvalue % 8) {
 916                 sval_t max = rl_max(rl);
 917                 /* round up */
 918                 max.uvalue += 7;
 919                 rl = alloc_rl(rl_min(rl), max);
 920         }
 921         rl = rl_binop(rl, '/', alloc_rl(int_8, int_8));
 922         rl = cast_rl(&ulong_ctype, rl);
 923         store_alloc(expr->left, rl);
 924 }
 925
 926 static void match_strndup(const char *fn, struct expression *expr, void *unused)
 927 {
 928         struct expression *fn_expr;
 929         struct expression *size_expr;
 930         sval_t size;
 931
 932         fn_expr = strip_expr(expr->right);
 933         size_expr = get_argument_from_call_expr(fn_expr->args, 1);
 934         if (get_implied_max(size_expr, &size)) {
 935                 size.value++;
 936                 store_alloc(expr->left, size_to_rl(size.value));
 937         } else {
 938                 store_alloc(expr->left, size_to_rl(UNKNOWN_SIZE));
 939         }
 940 }
 941
 942 static void match_alloc_pages(const char *fn, struct expression *expr, void *_order_arg)
 943 {
 944         int order_arg = PTR_INT(_order_arg);
 945         struct expression *right;
 946         struct expression *arg;
 947         sval_t sval;
 948
 949         right = strip_expr(expr->right);
 950         arg = get_argument_from_call_expr(right->args, order_arg);
 951         if (!get_implied_value(arg, &sval))
 952                 return;
 953         if (sval.value < 0 || sval.value > 10)
 954                 return;
 955
 956         sval.type = &int_ctype;
 957         sval.value = (1 << sval.value) * 4096;
 958
 959         store_alloc(expr->left, alloc_rl(sval, sval));
 960 }
 961
 962 static int is_type_bytes(struct range_list *rl, struct expression *call, int nr)
 963 {
 964         struct symbol *type;
 965         sval_t sval;
 966
 967         if (!rl_to_sval(rl, &sval))
 968                 return 0;
 969
 970         type = get_arg_type(call->fn, nr);
 971         if (!type)
 972                 return 0;
 973         if (type->type != SYM_PTR)
 974                 return 0;
 975         type = get_real_base_type(type);
 976         if (sval.value != type_bytes(type))
 977                 return 0;
 978         return 1;
 979 }
 980
 981 static void match_call(struct expression *expr)
 982 {
 983         struct expression *arg;
 984         struct symbol *type;
 985         struct range_list *rl;
 986         int i;
 987
 988         i = -1;
 989         FOR_EACH_PTR(expr->args, arg) {
 990                 i++;
 991                 type = get_type(arg);
 992                 if (!type || (type->type != SYM_PTR && type->type != SYM_ARRAY))
 993                         continue;
 994                 rl = get_array_size_bytes_rl(arg);
 995                 if (!rl)
 996                         continue;
 997                 if (rl_min(rl).value == UNKNOWN_SIZE &&
 998                     rl_max(rl).value == UNKNOWN_SIZE)
 999                         continue;
1000                 if (is_whole_rl(rl))
1001                         continue;
1002                 if (is_type_bytes(rl, expr, i))
1003                         continue;
1004                 sql_insert_caller_info(expr, BUF_SIZE, i, "$", show_rl(rl));
1005         } END_FOR_EACH_PTR(arg);
1006 }
1007
1008 static void struct_member_callback(struct expression *call, int param, char *printed_name, struct sm_state *sm)
1009 {
1010         sval_t sval;
1011
1012         if (!estate_rl(sm->state) ||
1013             (estate_get_single_value(sm->state, &sval) &&
1014              (sval.value == -1 || sval.value == 0)))
1015                 return;
1016
1017         sql_insert_caller_info(call, BUF_SIZE, param, printed_name, sm->state->name);
1018 }
1019
1020 /*
1021  * This is slightly (very) weird because half of this stuff is handled in
1022  * smatch_parse_call_math.c which is poorly named.  But anyway, add some buf
1023  * sizes here.
1024  *
1025  */
1026 static void print_returned_allocations(int return_id, char *return_ranges, struct expression *expr)
1027 {
1028         const char *param_math;
1029         struct range_list *rl;
1030         char buf[64];
1031         sval_t sval;
1032
1033         rl = get_array_size_bytes_rl(expr);
1034         param_math = get_allocation_math(expr);
1035         if (!rl && !param_math)
1036                 return;
1037
1038         if (!param_math &&
1039             rl_to_sval(rl, &sval) &&
1040             (sval.value == -1 || sval.value == 0))
1041                 return;
1042
1043         if (param_math)
1044                 snprintf(buf, sizeof(buf), "%s[%s]", show_rl(rl), param_math);
1045         else
1046                 snprintf(buf, sizeof(buf), "%s", show_rl(rl));
1047
1048         // FIXME: don't store if you can guess the size from the type
1049         // FIXME: return if we allocate a parameter $0->bar
1050         sql_insert_return_states(return_id, return_ranges, BUF_SIZE, -1, "", buf);
1051 }
1052
1053 static void record_global_size(struct symbol *sym)
1054 {
1055         int bytes;
1056         char buf[16];
1057
1058         if (!sym->ident)
1059                 return;
1060
1061         if (!(sym->ctype.modifiers & MOD_TOPLEVEL) ||
1062             sym->ctype.modifiers & MOD_STATIC)
1063                 return;
1064
1065         bytes = get_array_size_bytes(symbol_expression(sym));
1066         if (bytes <= 1)
1067                 return;
1068
1069         snprintf(buf, sizeof(buf), "%d", bytes);
1070         sql_insert_data_info_var_sym(sym->ident->name, sym, BUF_SIZE, buf);
1071 }
1072
1073 void register_buf_size(int id)
1074 {
1075         my_size_id = id;
1076
1077         set_dynamic_states(my_size_id);
1078
1079         add_unmatched_state_hook(my_size_id, &unmatched_size_state);
1080         add_merge_hook(my_size_id, &merge_estates);
1081
1082         select_caller_info_hook(set_param_buf_size, BUF_SIZE);
1083         select_return_states_hook(BUF_SIZE, &db_returns_buf_size);
1084         add_split_return_callback(print_returned_allocations);
1085
1086         allocation_funcs = create_function_hashtable(100);
1087         add_allocation_function("malloc", &match_alloc, 0);
1088         add_allocation_function("calloc", &match_calloc, 0);
1089         add_allocation_function("memdup", &match_alloc, 1);
1090         add_allocation_function("realloc", &match_alloc, 1);
1091         if (option_project == PROJ_KERNEL) {
1092                 add_allocation_function("kmalloc", &match_alloc, 0);
1093                 add_allocation_function("kmalloc_node", &match_alloc, 0);
1094                 add_allocation_function("kzalloc", &match_alloc, 0);
1095                 add_allocation_function("kzalloc_node", &match_alloc, 0);
1096                 add_allocation_function("vmalloc", &match_alloc, 0);
1097                 add_allocation_function("vzalloc", &match_alloc, 0);
1098                 add_allocation_function("__vmalloc", &match_alloc, 0);
1099                 add_allocation_function("kvmalloc", &match_alloc, 0);
1100                 add_allocation_function("kcalloc", &match_calloc, 0);
1101                 add_allocation_function("kvcalloc", &match_calloc, 0);
1102                 add_allocation_function("kmalloc_array", &match_calloc, 0);
1103                 add_allocation_function("devm_kmalloc_array", &match_calloc, 1);
1104                 add_allocation_function("sock_kmalloc", &match_alloc, 1);
1105                 add_allocation_function("kmemdup", &match_alloc, 1);
1106                 add_allocation_function("memdup_user", &match_alloc, 1);
1107                 add_allocation_function("dma_alloc_attrs", &match_alloc, 1);
1108                 add_allocation_function("devm_kmalloc", &match_alloc, 1);
1109                 add_allocation_function("devm_kzalloc", &match_alloc, 1);
1110                 add_allocation_function("krealloc", &match_alloc, 1);
1111                 add_allocation_function("__alloc_bootmem", &match_alloc, 0);
1112                 add_allocation_function("alloc_bootmem", &match_alloc, 0);
1113                 add_allocation_function("kmap", &match_page, 0);
1114                 add_allocation_function("kmap_atomic", &match_page, 0);
1115                 add_allocation_function("get_zeroed_page", &match_page, 0);
1116                 add_allocation_function("alloc_page", &match_page, 0);
1117                 add_allocation_function("alloc_pages", &match_alloc_pages, 1);
1118                 add_allocation_function("alloc_pages_current", &match_alloc_pages, 1);
1119                 add_allocation_function("__get_free_pages", &match_alloc_pages, 1);
1120                 add_allocation_function("dma_alloc_contiguous", &match_alloc, 1);
1121                 add_allocation_function("dma_alloc_coherent", &match_alloc, 1);
1122                 add_allocation_function("bitmap_alloc", &match_bitmap_alloc, 0);
1123                 add_allocation_function("bitmap_alloc_node", &match_bitmap_alloc, 0);
1124                 add_allocation_function("bitmap_zalloc", &match_bitmap_alloc, 0);
1125                 add_allocation_function("devm_bitmap_alloc", &match_bitmap_alloc, 1);
1126                 add_allocation_function("devm_bitmap_zalloc", &match_bitmap_alloc, 1);
1127         }
1128
1129         add_allocation_function("strndup", match_strndup, 0);
1130         if (option_project == PROJ_KERNEL)
1131                 add_allocation_function("kstrndup", match_strndup, 0);
1132
1133         add_modification_hook(my_size_id, &set_size_undefined);
1134
1135         add_merge_hook(my_size_id, &merge_size_func);
1136
1137         if (option_info)
1138                 add_hook(record_global_size, BASE_HOOK);
1139
1140         add_hook(&match_clear_cache, AFTER_FUNC_HOOK);
1141 }
1142
1143 void register_buf_size_late(int id)
1144 {
1145         /* has to happen after match_alloc() */
1146         add_hook(&match_array_assignment, ASSIGNMENT_HOOK);
1147
1148         add_hook(&match_call, FUNCTION_CALL_HOOK);
1149         add_member_info_callback(my_size_id, struct_member_callback);
1150 }