C/C++ 临时文件

一、背景

　　有时程序需要存储很大量的数据，或者在几个进程间交换数据，这时您可能考虑到使用临时文件。使用临时文件要考虑几个问题：
　　1、保证临时文件间的文件名不互助冲突。
　　2、保证临时文件中内容不被其他用户或者黑客偷看、删除和修改。
　　在linux下有专门处理临时文件的函数，先简单接收两个函数。

二、简单实用的两个函数：mkstemp和tempfile

　　2.1 mkstemp函数

　　头文件：#include <stdlib.h>

　　函数原型：int mkstemp(char * template);

　　执行：mkstemp函数在系统中以唯一的文件名创建一个文件并打开，而且只有当前用户才能访问这个临时文件，并进行读、写操作。mkstemp函数只有一个参数，这个参数是个以“XXXXXX”结尾的非空字符串。mkstemp函数会用随机产生的字符串替换“XXXXXX”，保证了文件名的唯一性。

　　结果： 函数返回一个文件描述符，如果执行失败返回-1。在glibc 2.0.6 以及更早的glibc库中这个文件的访问权限是0666，glibc 2.0.7以后的库这个文件的访问权限是0600。

　　备注：临时文件使用完成后应及时删除，否则临时文件目录会塞满垃圾。由于mkstemp函数创建的临时文件不能自动删除，所以执行完mkstemp函数后要调用unlink函数，unlink函数删除文件的目录入口，但临时文件还可以通过文件描述符进行访问，直到最后一个打开的进程关闭文件操作符，或者程序退出后临时文件被自动彻底地删除。

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

int write_temp_file(char* buffer,size_t length) {
    int len=length;
    char filename_template[]="/tmp/temp_file.XXXXXX";
    int fd=mkstemp(filename_template);
    unlink(filename_template);//Unlink the file, so it'll be removed when close
    printf("Template file name:%s\n",filename_template);
    write(fd,&len,sizeof(len));
    write(fd,buffer,len);
    return fd;
}

char* read_temp_file(int fd, size_t* length) {
    char* buffer;
    lseek(fd,0,SEEK_SET);
    read(fd,length,sizeof(size_t));
    buffer=(char*)malloc(*length);
    read(fd,buffer,*length);
    close(fd); // Temp file will be deleted
    return buffer;
}

int main(int argc, char** argv) {
    char buffer[]="Test template files";
    int fd=write_temp_file(buffer,strlen(buffer));
    int len=0;
    char* result=read_temp_file(fd,&len);
    printf("Len:%d\nContent:%s\n",len,result);
    free(result);
    return 0;
}

　　2.2 tmpfile函数

　　头文件：#include <stdio.h>

　　函数原型：FILE *tmpfile(void);

　　执行：tmpfile函数创建并打开一个临时文件（二进制，在 /tmp文件夹中），并且自动执行了unlink。tmpfile函数返回一个文件描述符，如果执行失败返回NULL。当程序执行了fclose或者退出时，资源被释放。

　　结果：如果成功，该函数返回一个流指针创建的临时文件。如果不能创建文件，则返回NULL。

#include <stdio.h>
int main()
{
    FILE *tempfp;
    char line[256];
    tempfp=tmpfile();
    if(tempfp==NULL)
    {
        printf("tmpfile error!\n");
        return 1;
    }
    printf("Opened a temporary file OK!\n");
    fputs("One line of output \n",tempfp);
    rewind(tempfp);
    if(fgets(line, sizeof(line),tempfp)==NULL)
    {
        printf("fgets error!\n");
        return 2;
    }
    fputs(line, stdout);
    return 0;
}