[代码实例][C语言]写入文本文件

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

#define BUF_SIZE    1024
char buf[BUF_SIZE + 1];

int main(int argc, char * argv[])
{
    if(argc != 2)
    {
        printf("Usage : %s <write_file>\n", argv[0]);
        return EXIT_FAILURE;
    }

    FILE * fp;
    char * file_name = argv[1];
    size_t count;

    if((fp = fopen(file_name, "w")) == NULL)
    {
        perror("fopen");
        return EXIT_FAILURE;
    }

    char ch;
    int index = 0;
    while((ch = getchar()) != EOF)
    {
        buf[index++] = ch;
    }
    buf[index] = '\0';

    if(fwrite(buf, 1, index, fp) != index)
    {
        perror("fwrite");
        return EXIT_FAILURE;
    }

    if(fclose(fp) != 0)
    {
        perror("fclose");
        return EXIT_FAILURE;
    }

    return EXIT_SUCCESS;
}