You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.3 KiB
60 lines
1.3 KiB
2 months ago
|
/*
|
||
|
|
* UCW Library -- Temporary Fastbufs
|
||
|
|
*
|
||
|
|
* (c) 2002--2008 Martin Mares <mj@ucw.cz>
|
||
|
|
* (c) 2008 Michal Vaner <vorner@ucw.cz>
|
||
|
|
*
|
||
|
|
* This software may be freely distributed and used according to the terms
|
||
|
|
* of the GNU Lesser General Public License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ucw/lib.h>
|
||
|
|
#include <ucw/fastbuf.h>
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
|
||
|
|
struct fastbuf *
|
||
|
|
bopen_tmp_file(struct fb_params *params)
|
||
|
|
{
|
||
|
|
char name[TEMP_FILE_NAME_LEN];
|
||
|
|
int fd = open_tmp(name, O_RDWR | O_CREAT | O_TRUNC, 0600);
|
||
|
|
struct fastbuf *fb = bopen_fd_name(fd, params, name);
|
||
|
|
bconfig(fb, BCONFIG_IS_TEMP_FILE, 1);
|
||
|
|
return fb;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct fastbuf *
|
||
|
|
bopen_tmp(uint buflen)
|
||
|
|
{
|
||
|
|
return bopen_tmp_file(&(struct fb_params){ .type = FB_STD, .buffer_size = buflen });
|
||
|
|
}
|
||
|
|
|
||
|
|
void bfix_tmp_file(struct fastbuf *fb, const char *name)
|
||
|
|
{
|
||
|
|
int was_temp = bconfig(fb, BCONFIG_IS_TEMP_FILE, 0);
|
||
|
|
ASSERT(was_temp == 1);
|
||
|
|
if (rename(fb->name, name))
|
||
|
|
bthrow(fb, "tmp", "Cannot rename %s to %s: %m", fb->name, name);
|
||
|
|
bclose(fb);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef TEST
|
||
|
|
|
||
|
|
#include <ucw/getopt.h>
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
log_init(NULL);
|
||
|
|
if (cf_getopt(argc, argv, CF_SHORT_OPTS, CF_NO_LONG_OPTS, NULL) >= 0)
|
||
|
|
die("Hey, whaddya want?");
|
||
|
|
|
||
|
|
struct fastbuf *f = bopen_tmp(65536);
|
||
|
|
ASSERT(f && f->name);
|
||
|
|
bputsn(f, "Hello, world!");
|
||
|
|
bclose(f);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|