如何更改此代码以使其不容易受到算术溢出的影响？[关闭]

问：

Computer Systems： a Programmer's Perspective说道：

1 /* Illustration of code vulnerability similar to that found in
2 * Sun’s XDR library.
3 */
4 void* copy_elements(void *ele_src[], int ele_cnt, size_t ele_size) {
5 /*
6 * Allocate buffer for ele_cnt objects, each of ele_size bytes
7 * and copy from locations designated by ele_src
8 */
9 void *result = malloc(ele_cnt * ele_size);
10 if (result == NULL)
11 /* malloc failed */
12 return NULL;
13 void *next = result;
14 int i;
15 for (i = 0; i < ele_cnt; i++) {
16 /* Copy object i to destination */
17 memcpy(next, ele_src[i], ele_size);
18 /* Move pointer to next memory region */
19 next += ele_size;
20 }
21 return result;
22 }

copy_elements功能旨在复制ele_cnt数据 结构，每个结构体由 ele_ 大小的字节组成，分配到一个缓冲区中 按第 9 行的函数。所需的字节数计算如下： ele_cnt * ele_size。

但是，想象一下，恶意程序员使用 ele_cnt 是 1,048,577 （2^20 + 1），ele_size是 4,096 （2^12） 程序编译为 32 位。然后在线乘法 9 将溢出，导致只分配 4,096 个字节，而不是 保存这么多数据所需的 4,294,971,392 字节。循环 从第 15 行开始将尝试复制所有这些字节， 溢出已分配缓冲区的末尾，因此损坏 其他数据结构。这可能会导致程序崩溃或 否则行为不端。

我想知道如何更改代码以使其没有由于算术溢出而导致的漏洞？

谢谢。