歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux基礎 >> Linux教程 >> Linux下對臨時文件的操作之mkstemp

Linux下對臨時文件的操作之mkstemp

日期:2017/2/28 14:35:10   编辑:Linux教程

使用臨時文件要考慮幾個問題:

保證臨時文件間的文件名不互助沖突。

保證臨時文件中內容不被其他用戶或者黑客偷看、刪除和修改。

Linux中提供了mkstemp 和 tmpfile 函數來處理臨時文件。

mkstemp函數

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

tmpfile函數

如果您使用C library I/O函數,並且並沒有另一個程序使用這個臨時文件,有個更簡潔的函數——tmpfile。tmpfile函數創建並打開一個臨時文件,並且自動執行了unlink。tmpfile函數返回一個文件描述符,如果執行失敗返回NULL。當程序執行了fclose或者退出時,資源被釋放。

另外,linux系統中還提供mktemp、 tmpnam、 和tempnam等函數,但是由於健壯性和安全性的問題,不建議使用。

Copyright © Linux教程網 All Rights Reserved