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;
}