萬盛學電腦網

 萬盛學電腦網 >> 服務器教程 >> Linux下對臨時文件的操作之mkstemp

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

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

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

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

  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

  #include

  #include

  #include

  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:%sn",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:%dnContent:%sn",len,result);

  free(result);

  return 0;

  }

  tmpfile函數

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

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

copyright © 萬盛學電腦網 all rights reserved