如何实现仅识别纯数字或单个q的C语言输入验证函数？

原代码存在的问题 #

• 内存泄漏：遇到非数字字符时直接将string设为NULL返回，但之前malloc的内存未释放，导致内存泄漏
• 无法处理'q'输入：当前逻辑仅允许数字输入，遇到'q'会直接判定无效返回NULL
• 缓冲区越界风险：当输入刚好占满nr个字符时，执行string[i]='\0'会超出malloc(nr)分配的内存范围
• 未处理EOF情况：若输入为EOF（如Ctrl+D），循环会持续执行导致异常
• 数字转换不安全：atoi不处理溢出，当输入数字超出int范围时会触发未定义行为

优化实现方案 #

调整逻辑为：先读取完整一行输入，再校验是否符合「纯数字（长度≥1）」或「单个字符'q'」的要求。这种方式逻辑更清晰，也更易维护。

修改后的代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include <limits.h>

// 读取整行输入，自动处理内存分配
char* readLine() {
    char* buffer = NULL;
    size_t bufSize = 0;
    // getline为POSIX标准，Windows环境可替换为_getline
    ssize_t len = getline(&buffer, &bufSize, stdin);
    
    if (len == -1) {
        free(buffer);
        return NULL;
    }
    
    // 移除末尾换行符
    if (len > 0 && buffer[len-1] == '\n') {
        buffer[len-1] = '\0';
    }
    
    return buffer;
}

// 校验输入是否符合要求：纯数字 或 单个'q'
int isValidInput(const char* input) {
    if (input == NULL || *input == '\0') {
        return 0;
    }
    
    // 判断是否为单个'q'
    if (strcmp(input, "q") == 0) {
        return 1;
    }
    
    // 判断是否全为数字
    const char* ptr = input;
    while (*ptr != '\0') {
        if (!isdigit((unsigned char)*ptr)) {
            return 0;
        }
        ptr++;
    }
    return 1;
}

int main(void) {
    int number;
    char* input = NULL;
    
    printf("Enter number or 'q' to quit: ");
    input = readLine();
    
    if (input == NULL) {
        printf("Invalid input or read error!\n");
        free(input);
        return 1;
    }
    
    if (!isValidInput(input)) {
        printf("Invalid input!\n");
    } else {
        if (strcmp(input, "q") == 0) {
            printf("You entered 'q', exiting.\n");
        } else {
            // 用strtol安全转换数字，处理溢出
            char* endPtr;
            long num = strtol(input, &endPtr, 10);
            if (num < INT_MIN || num > INT_MAX) {
                printf("Number out of int range!\n");
            } else {
                number = (int)num;
                printf("Number as int is %d\n", number);
            }
        }
    }
    
    free(input);
    return 0;
}

关键优化点说明 #

• 安全读取整行：用getline自动分配内存，避免手动设置缓冲区大小的限制，同时自动处理换行符与EOF情况
• 分离校验逻辑：将输入校验单独封装为isValidInput函数，职责单一，代码更易读和维护
• 内存安全：所有动态分配的内存均确保被释放，避免泄漏
• 安全数字转换：用strtol替代atoi，可检测数字是否超出int范围，避免溢出风险
• 清晰分支处理：分别处理'q'与纯数字的逻辑，流程更直观

测试案例验证 #

• 输入12345 → 输出转换后的数字，判定有效
• 输入q → 提示退出，判定有效
• 输入12q34/qq/5435q → 提示无效输入
• 输入空行（直接回车）→ 提示无效输入

内容的提问来源于stack exchange，提问作者Alexander Jonsson