CentOS下对临时文件的操作之mkstemp

浏览次数:2007 关键词 ( 临时文件  操作  CentOS  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函数删除文件的目录入口,但临时文件还可以通过文件描述符进行访问,直到最后一个打开的进程关闭文件操作符,或者程序退出后临时文件被自动彻底地删除。

[cpp] view plaincopy在CODE上查看代码片派生到我的代码片  
  1. <span style="font-size:18px;">#include <stdio.h>  
  2. #include <string.h>  
  3. #include <unistd.h>  
  4. #include <stdlib.h>  
  5.   
  6. int write_temp_file(char* buffer,size_t length) {  
  7.     int len=length;  
  8.     char filename_template[]="/tmp/temp_file.XXXXXX";  
  9.     int fd=mkstemp(filename_template);  
  10.     unlink(filename_template);//Unlink the file, so it'll be removed when close  
  11.     printf("Template file name:%s\n",filename_template);  
  12.     write(fd,&len,sizeof(len));  
  13.     write(fd,buffer,len);  
  14.     return fd;  
  15. }  
  16.   
  17. char* read_temp_file(int fd, size_t* length) {  
  18.     char* buffer;  
  19.     lseek(fd,0,SEEK_SET);  
  20.     read(fd,length,sizeof(size_t));  
  21.     buffer=(char*)malloc(*length);  
  22.     read(fd,buffer,*length);  
  23.     close(fd); // Temp file will be deleted  
  24.     return buffer;  
  25. }  
  26.   
  27. int main(int argc, char** argv) {  
  28.     char buffer[]="Test template files";  
  29.     int fd=write_temp_file(buffer,strlen(buffer));  
  30.     int len=0;  
  31.     char* result=read_temp_file(fd,&len);  
  32.     printf("Len:%d\nContent:%s\n",len,result);  
  33.     free(result);  
  34.     return 0;  
  35. }</span>  

 

 
tmpfile函数
如果您使用C library I/O函数,并且并没有另一个程序使用这个临时文件,有个更简洁的函数——tmpfile。tmpfile函数创建并打开一个临时文件,并且自动执行了unlink。tmpfile函数返回一个文件描述符,如果执行失败返回NULL。当程序执行了fclose或者退出时,资源被释放。
另外,linux系统中还提供mktemp、 tmpnam、 和tempnam等函数,但是由于健壮性和安全性的问题,不建议使用。