정수 오버플로우

취약한 코드 (c)

// 취약한 코드 - 정수 오버플로우
int main() {
    int size = atoi(argv[1]);  // 사용자 입력
    int total = size * sizeof(int);  // 오버플로우 가능
    char* buf = malloc(total);  // 작은 버퍼 할당 → 오버플로우
    memcpy(buf, data, size * sizeof(int));
}

안전한 코드 (c)

// 안전한 코드 - 범위 체크 후 안전한 연산
#include <stdint.h>
int main() {
    long long size = atoll(argv[1]);
    // 양수 범위 및 최대값 체크
    if (size <= 0 || size > MAX_SAFE_SIZE) {
        return -1;
    }
    // SIZE_MAX 초과 여부 확인
    if ((uint64_t)size > SIZE_MAX / sizeof(int)) {
        return -1;
    }
    char* buf = malloc(size * sizeof(int));
}