unix编程创建前缀固定的临时文件代码分享
程序员文章站
2023-11-27 20:52:40
参数：pathname，存储临时文件的路径文件名，需要手动free()掉。dir，临时文件的路径，如果tmpdir环境变量不为空，则此参数被忽略，转而使用环境变量。pfx，...
参数：
pathname，存储临时文件的路径文件名，需要手动free()掉。
dir，临时文件的路径，如果tmpdir环境变量不为空，则此参数被忽略，转而使用环境变量。
pfx，临时文件名的前缀，只使用前5个字符。
注：
创建的临时文件需要手动unlink()掉。
创建临时文件的函数
复制代码 代码如下:
int make_temp_file(char **pathname,const char *dir,const char *pfx){
char *ptr,*tmp;
size_t len;
int fd;
debug_assert("invalid pointer","make_temp_file()",pathname);
/*前缀只能是多于5字符*/
if(pfx && (len=strlen(pfx))>0){
tmp=(char*)malloc((len>5?5:len)+1);
strncpy(tmp,pfx,len>5?5:len);
}
else
tmp=null;
ptr=tempnam(dir,tmp);
if(tmp)free(tmp);
len=strlen(ptr);
tmp=(char*)malloc(len+6+1);
if(snprintf(tmp,len+6+1,"%sxxxxxx",ptr)==-1)
err_sys(errno,"snprintf() error");
free(ptr);
fd=mkstemp(tmp);
*pathname=tmp;
return fd;
}
测试程序
复制代码 代码如下:
#include "wrap_ext.h"
int main(int argc,char **argv){
int fd;
char *path;
if(argc!=3)
err_quit(-1,"usage %s <dir> <prefix>",argv[0]);
fd=make_temp_file(&path,argv[1][0]==' '?null:argv[1],argv[2][0]==' '?null:argv[2]);
err_msg("temporary file path:%s",path);
close(fd);
unlink(path);
free(path);
return exit_success;
}
测试结果
复制代码 代码如下:
root@u-server:/home/apu/sysinfo# ./tmpfile " " " "
temporary file path:/tmp/fileq55hof8swffa
root@u-server:/home/apu/sysinfo# ll /tmp/fileq55hof8swffa
ls: cannot access /tmp/fileq55hof8swffa: no such file or directory
root@u-server:/home/apu/sysinfo# ./tmpfile " " tmp_
temporary file path:/tmp/tmp_0rzhqozlthxw
root@u-server:/home/apu/sysinfo# ./tmpfile /home tmp_
temporary file path:/home/tmp_phzxvrrp33ol