使用臨時文件要考慮幾個問題:
保證臨時文件間的文件名不互助沖突。
保證臨時文件中內(nèi)容不被其他用戶或者黑客偷看、刪除和修改。
Linux中提供了mkstemp 和 tmpfile 函數(shù)來處理臨時文件。
mkstemp函數(shù)
int mkstemp(char *template);
mkstemp函數(shù)在系統(tǒng)中以唯一的文件名創(chuàng)建一個文件并打開,而且只有當(dāng)前用戶才能訪問這個臨時文件,并進行讀、寫操作。mkstemp函數(shù)只有一個參數(shù),這個參數(shù)是個以“XXXXXX”結(jié)尾的非空字符串。mkstemp函數(shù)會用隨機產(chǎn)生的字符串替換“XXXXXX”,保證了文件名的唯一性。 函數(shù)返回一個文件描述符,如果執(zhí)行失敗返回-1。在glibc 2.0.6 以及更早的glibc庫中這個文件的訪問權(quán)限是0666,glibc 2.0.7以后的庫這個文件的訪問權(quán)限是0600。
臨時文件使用完成后應(yīng)及時刪除,否則臨時文件目錄會塞滿垃圾。由于mkstemp函數(shù)創(chuàng)建的臨時文件不能自動刪除,所以執(zhí)行完mkstemp函數(shù)后要調(diào)用unlink函數(shù),unlink函數(shù)刪除文件的目錄入口,但臨時文件還可以通過文件描述符進行訪問,直到最后一個打開的進程關(guān)閉文件操作符,或者程序退出后臨時文件被自動徹底地刪除。
[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ù),并且并沒有另一個程序使用這個臨時文件,有個更簡潔的函數(shù)——tmpfile。tmpfile函數(shù)創(chuàng)建并打開一個臨時文件,并且自動執(zhí)行了unlink。tmpfile函數(shù)返回一個文件描述符,如果執(zhí)行失敗返回NULL。當(dāng)程序執(zhí)行了fclose或者退出時,資源被釋放。
另外,linux系統(tǒng)中還提供mktemp、 tmpnam、 和tempnam等函數(shù),但是由于健壯性和安全性的問題,不建議使用。