使用臨時(shí)文件要考慮幾個(gè)問(wèn)題：

保證臨時(shí)文件間的文件名不互助沖突。

保證臨時(shí)文件中內(nèi)容不被其他用戶(hù)或者黑客偷看、刪除和修改。

Linux中提供了mkstemp 和 tmpfile 函數(shù)來(lái)處理臨時(shí)文件。

mkstemp函數(shù)

int mkstemp(char *template);

mkstemp函數(shù)在系統(tǒng)中以唯一的文件名創(chuàng)建一個(gè)文件并打開(kāi)，而且只有當(dāng)前用戶(hù)才能訪問(wèn)這個(gè)臨時(shí)文件，并進(jìn)行讀、寫(xiě)操作。mkstemp函數(shù)只有一個(gè)參數(shù)，這個(gè)參數(shù)是個(gè)以“XXXXXX”結(jié)尾的非空字符串。mkstemp函數(shù)會(huì)用隨機(jī)產(chǎn)生的字符串替換“XXXXXX”，保證了文件名的唯一性。 函數(shù)返回一個(gè)文件描述符，如果執(zhí)行失敗返回-1。在glibc 2.0.6 以及更早的glibc庫(kù)中這個(gè)文件的訪問(wèn)權(quán)限是0666，glibc 2.0.7以后的庫(kù)這個(gè)文件的訪問(wèn)權(quán)限是0600。

臨時(shí)文件使用完成后應(yīng)及時(shí)刪除，否則臨時(shí)文件目錄會(huì)塞滿(mǎn)垃圾。由于mkstemp函數(shù)創(chuàng)建的臨時(shí)文件不能自動(dòng)刪除，所以執(zhí)行完mkstemp函數(shù)后要調(diào)用unlink函數(shù)，unlink函數(shù)刪除文件的目錄入口，但臨時(shí)文件還可以通過(guò)文件描述符進(jìn)行訪問(wèn)，直到最后一個(gè)打開(kāi)的進(jìn)程關(guān)閉文件操作符，或者程序退出后臨時(shí)文件被自動(dòng)徹底地刪除。

[cpp] view plaincopy在CODE上查看代碼片派生到我的代碼片  
<span style="font-size:18px;">#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;  
}</span>

tmpfile函數(shù)

如果您使用C library I/O函數(shù)，并且并沒(méi)有另一個(gè)程序使用這個(gè)臨時(shí)文件，有個(gè)更簡(jiǎn)潔的函數(shù)——tmpfile。tmpfile函數(shù)創(chuàng)建并打開(kāi)一個(gè)臨時(shí)文件，并且自動(dòng)執(zhí)行了unlink。tmpfile函數(shù)返回一個(gè)文件描述符，如果執(zhí)行失敗返回NULL。當(dāng)程序執(zhí)行了fclose或者退出時(shí)，資源被釋放。

另外，linux系統(tǒng)中還提供mktemp、 tmpnam、 和tempnam等函數(shù)，但是由于健壯性和安全性的問(wèn)題，不建議使用。

最新評(píng)論