在 C 语言中使用文件重定向

Jinku Hu 2021年2月28日
在 C 语言中使用文件重定向

本文将介绍几种在 C 语言中使用文件重定向的方法。

使用 < 操作符重定向标准输入

文件重定向在基于 UNIX 的系统上一般称为 I/O 重定向,它允许用户重新定义标准输入来自哪里,或标准输出去哪里。< 运算符用来改变标准输入的来源。这种方法对于从文件内容中获取用户输入并将其存储在程序缓冲区中很有用。在这种情况下,我们利用 fgets 函数来读取文件的内容,直到遇到一个新行字符。当读取到一个新的行字符时,它也会被存储在给定的缓冲区中,之后会存储字符串终止的空字节。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>

#define SIZE 1000

int main(int argc, char *argv[]) {

    char buf[SIZE];
    printf("Write input text: ");
    fgets(buf, SIZE , stdin);
    printf("%s\n", buf);

    exit(EXIT_SUCCESS);
}

示例命令:

./program < input.txt

输出:

The first line from the input.txt file

在前面的示例代码中,我们分配了一个带有硬编码值的缓冲区。不过,用户提供的输入文件可能需要更多的内存来存储,我们应该实现一种自适应的方法来适应这种情况。为此,我们使用 malloc 分配动态内存,并将文件总大小作为存储文件中第一行所需的最大数量传递给它。如果给定的文件太大,而第一行又实在太短,这种方法可能会浪费太多内存。但是,它可以保证只要 malloc 不会失败,系统内存不会耗尽,代码就能读进最长的一行。

注意,我们使用 stat 系统调用来检索文件大小,但它需要输入文件的路径名,如果不作为命令行参数明确粘贴,就无法检索。不过要注意的是,每一次函数调用都需要检查是否成功返回,以保证 fgets 不会试图写入未初始化的内存区域,导致程序以失败告终。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>

int main(int argc, char *argv[]) {
    char *buffer = NULL;
    const char *filename = NULL;
    struct stat sb;

    if (argc != 2) {
        printf("Usage: ./program filename < filename\n");
        exit(EXIT_FAILURE);
    }

    filename = argv[1];

    if (stat(filename, &sb) == -1) {
        perror("stat");
        exit(EXIT_FAILURE);
    }

    buffer = malloc(sb.st_size);
    if (!buffer) {
        perror("malloc");
        exit(EXIT_FAILURE);
    }

    printf("Write input text: ");
    if (fgets(buffer, sb.st_size , stdin) == NULL) {
        perror("fgets");
        exit(EXIT_FAILURE);
    }
    printf("%s\n", buffer);


    exit(EXIT_SUCCESS);
}

示例命令:

./program input.txt < input.txt

输出:

The first line from the input.txt file
Author: Jinku Hu
Jinku Hu avatar Jinku Hu avatar

Founder of DelftStack.com. Jinku has worked in the robotics and automotive industries for over 8 years. He sharpened his coding skills when he needed to do the automatic testing, data collection from remote servers and report creation from the endurance test. He is from an electrical/electronics engineering background but has expanded his interest to embedded electronics, embedded programming and front-/back-end programming.

LinkedIn

相关文章 - C File