/*
- * Sherlock Library -- Temporary Fastbufs
+ * UCW Library -- Temporary Fastbufs
*
- * (c) 2002 Martin Mares <mj@ucw.cz>
+ * (c) 2002--2006 Martin Mares <mj@ucw.cz>
*
* This software may be freely distributed and used according to the terms
* of the GNU Lesser General Public License.
#include "lib/lib.h"
#include "lib/conf.h"
#include "lib/fastbuf.h"
+#include "lib/threads.h"
+#include <stdio.h>
#include <unistd.h>
#include <sys/fcntl.h>
-static byte *temp_template = "/tmp/temp%d.%d";
+static byte *temp_prefix = "/tmp/temp";
-static struct cfitem temp_config[] = {
- { "Tempfiles", CT_SECTION, NULL },
- { "Template", CT_STRING, &temp_template },
- { NULL, CT_STOP, NULL }
+static struct cf_section temp_config = {
+ CF_ITEMS {
+ CF_STRING("Prefix", &temp_prefix),
+ CF_END
+ }
};
-static void CONSTRUCTOR temp_init_config(void)
+static void CONSTRUCTOR temp_global_init(void)
{
- cf_register(temp_config);
+ cf_declare_section("Tempfiles", &temp_config, 0);
+}
+
+void
+temp_file_name(byte *buf)
+{
+ struct ucwlib_context *ctx = ucwlib_thread_context();
+ int cnt = ++ctx->temp_counter;
+ int pid = getpid();
+ if (ctx->thread_id == pid)
+ sprintf(buf, "%s%d-%d", temp_prefix, pid, cnt);
+ else
+ sprintf(buf, "%s%d-%d-%d", temp_prefix, pid, ctx->thread_id, cnt);
}
struct fastbuf *
-bopen_tmp(uns bufsize)
+bopen_tmp(uns buflen)
{
- byte buf[256];
+ byte buf[TEMP_FILE_NAME_LEN];
struct fastbuf *f;
- static uns temp_counter;
- sprintf(buf, temp_template, (int) getpid(), temp_counter++);
- f = bopen(buf, O_RDWR | O_CREAT | O_EXCL, bufsize);
- f->is_temp_file = 1;
+ temp_file_name(buf);
+ f = bopen(buf, O_RDWR | O_CREAT | O_TRUNC, buflen);
+ bconfig(f, BCONFIG_IS_TEMP_FILE, 1);
return f;
}
+
+#ifdef TEST
+
+#include "lib/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);
+ bputsn(f, "Hello, world!");
+ bclose(f);
+ return 0;
+}
+
+#endif