c - 读取不同流的独特功能

Question

我正在寻找一种使用 C 中独特的 void 函数来读取 file 和 stdin 流的方法>。我正在尝试使用此功能:

#define ENTER 10 //'\n' ASCII code

........

void read(FILE *stream, char *string) {
    char c;
    int counter = 0;

    do {
        c = fgetc(stream);
        string = realloc(string, (counter+1) * sizeof(char));
        string[counter++] = c;
    } while(c != ENTER && !feof(stream));

    string[counter-1] = '\0';
}

但它只适用于 stdin 流。当我使用文本文件时，文件内容在函数外是不可见的。我这样调用这个函数:

read(stdin, inputString);
read(inputFile, fileContent);

并且只在第一种情况中有效。

P.S.: 最初，inputString 和 fileContent 被声明为

char *inputString = malloc(sizeof(char));
char *fileContent = malloc(sizeof(char));

而且我知道 fgetc 返回 int，通过 char 重新分配的 char 很昂贵(但我只需要使用必要的内存)并且 EOF 或 '\n' 是存储在字符串中(但之后被 0 终止符替换)。

Answer 1

它不能正常工作，因为您重新分配了字符串但没有返回指向新分配的指针，并且因为您对 fgetc() 的处理不当。

您可能需要:

void read(FILE *stream, char **string)
{
    int c;
    int counter = 0;
    int available = 0;

    while ((c = fgetc(stream)) != EOF && c != ENTER)
    {
        if (counter >= available)
        {
            int new_size = (available + 2) * 2;
            void *space = realloc(*string, new_size);
            if (space == 0)
                // Handle out of memory reporting?
                return;
            *string = space;
            available = new_size;
        }
        (*string)[counter++] = c;
    }
    (*string)[counter] = '\0';
}

你会这样调用它:

read(stdin, &inputString);
read(inputFile, &fileContent);

这里有很多更正；仍然可以而且应该进行改进。

1. 将指针传递给指针，以便更改可以反射(reflect)在调用代码中。在原始代码中，如果 realloc() 调用可以增加当前 block ，您可能会逃脱，但当它必须移动内存时，移动的位置在调用函数中不可用。

2. fgetc()的返回类型是int，不是char。

3. 您应该更喜欢while 循环而不是do ... while 循环。如果您的代码遇到 EOF，它会尝试将存储的字符值放入数组中。

4. (仍然是一个问题)string 的初始分配在很大程度上被忽略了；它被视为零长度分配。

5. 您一次增加一个字符存储的空间——这是低效的。修改后的代码每次大约使可用空间增加一倍。它还为终止空字节分配了足够的空间。

6. 未固定:名称 read() 未由标准 C 保留，但如果您使用任何 POSIX 函数，则有效保留。

你(仍然)没有办法报告错误。我实际上更喜欢这样的功能:

char *read_line(FILE *stream)
{
    int c;
    int counter = 0;
    int available = 2;
    char *string = malloc(available);

    if (string == 0)
        return string;

    while ((c = fgetc(stream)) != EOF && c != ENTER)
    {
        if (counter >= available)
        {
            int new_size = (available + 2) * 2;
            void *space = realloc(string, new_size);
            if (space == 0)
            {
                free(string);
                return 0;
            }
            string = space;
            available = new_size;
        }
        string[counter++] = c;
    }
    string[counter] = '\0';
    return string;
}

您还应该查看 POSIX getline() 功能。