c - 确保作为参数传递的字符串不会导致溢出

Question

我必须确保作为参数传递的字符串不会导致溢出。我是通过使用 strncpy 来做到这一点的，但是结尾 '\0'、分配适量的内存等等给我带来了一些麻烦...

我的解决方案是这样的:

l = strlen(argv[optind]);
if(l<MAX_LENGTH) {
    msg = malloc((l+1) * sizeof(char));
    msg = strcpy(msg, argv[optind]);
} else {
    msg = malloc((MAX_LENGTH+1) * sizeof(char));
    msg = strncpy(msg, argv[optind], MAX_LENGTH);
    msg[MAX_LENGTH+1] = '\0';
} 

它有效，但我想知道它是否真的正确，是否有更紧凑的解决方案？

Answer 1

我想这是最简单的了:

size_t l;
char* msg;
...
l = strlen(argv[optind]);
if (l > MAX_LENGTH) l = MAX_LENGTH;

msg = malloc(l + 1);
if (msg == NULL) /* handle the error as appropriate*/;

memcpy(msg, argv[optind], l);
msg[l] = '\0';