struct annotation *notes = symbol__annotation(sym);
        const size_t size = symbol__size(sym);
-       size_t sizeof_sym_hist = (sizeof(struct sym_hist) + size * sizeof(u64));
+       size_t sizeof_sym_hist;
+
+       /* Check for overflow when calculating sizeof_sym_hist */
+       if (size > (SIZE_MAX / sizeof(u64) - sizeof(struct sym_hist)))
+               return -1;
+
+       sizeof_sym_hist = (sizeof(struct sym_hist) + size * sizeof(u64));
+
+       /* Check for overflow in zalloc argument */
+       if (sizeof_sym_hist > (SIZE_MAX / symbol_conf.nr_events
+                               - sizeof(*notes->src)))
+               return -1;

        notes->src = zalloc(sizeof(*notes->src) + symbol_conf.nr_events * 
sizeof_sym_hist);
        if (notes->src == NULL)


Actually, I don't think this is correct either (subtraction seems to occur in the wrong spot).

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majord...@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to