1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*
* Copyright (C) 2006-2007 Andre Noll <maan@systemlinux.org>
*
* Licensed under the GPL v2. For licencing details see COPYING.
*/
/** \file file_write.c simple output plugin for testing purposes */
#include "para.h"
#include "list.h"
#include "sched.h"
#include "write.h"
#include "string.h"
#include "fd.h"
#include "file_write.cmdline.h"
#include "error.h"
/** data specific to the file writer */
struct private_file_write_data {
/** the file descriptor of the output file */
int fd;
/** non-zero if \a fd was added to the write fd set */
int check_fd;
};
static int file_write_open(struct writer_node *wn)
{
struct private_file_write_data *pfwd = para_calloc(
sizeof(struct private_file_write_data));
struct file_write_args_info *conf = wn->conf;
char *filename;
if (conf->filename_given)
filename = conf->filename_arg;
else {
char *tmp = para_tmpname(), *home = para_homedir();
filename = make_message("%s/.paraslash/%s", home, tmp);
free(home);
free(tmp);
}
wn->private_data = pfwd;
pfwd->fd = open(filename, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR);
if (!conf->filename_given)
free(filename);
if (pfwd->fd >= 0)
return 8192;
free(pfwd);
return -E_FW_OPEN;
}
static int file_write_pre_select(struct sched *s, struct writer_node *wn)
{
struct private_file_write_data *pfwd = wn->private_data;
struct writer_node_group *wng = wn->wng;
pfwd->check_fd = 0;
if (pfwd->fd <= 0)
return -E_FW_NO_FILE;
if (!*wng->loaded)
return 1;
para_fd_set(pfwd->fd, &s->wfds, &s->max_fileno);
pfwd->check_fd = 1;
return 1;
}
static int file_write_post_select(struct sched *s, struct writer_node *wn)
{
struct private_file_write_data *pfwd = wn->private_data;
struct writer_node_group *wng = wn->wng;
int ret;
if (!pfwd->check_fd)
return 1;
if (*wng->loaded <= wn->written)
return 1;
if (!FD_ISSET(pfwd->fd, &s->wfds))
return 1;
// PARA_INFO_LOG("writing %zd\n", *wng->loaded);
ret = write(pfwd->fd, wng->buf + wn->written,
*wng->loaded - wn->written);
if (ret < 0)
return -E_FW_WRITE;
wn->written += ret;
return 1;
}
static void file_write_close(struct writer_node *wn)
{
struct private_file_write_data *pfwd = wn->private_data;
close(pfwd->fd);
free(pfwd);
}
__malloc static void *file_write_parse_config(const char *options)
{
PARA_INFO_LOG("options: %s\n", options);
struct file_write_args_info *conf
= para_calloc(sizeof(struct file_write_args_info));
int ret = file_cmdline_parser_string(options, conf, "file_write");
PARA_INFO_LOG("conf->filename_given: %d\n", conf->filename_given);
if (!ret)
return conf;
free(conf);
return NULL;
}
/** the init function of the file writer */
void file_write_init(struct writer *w)
{
w->open = file_write_open;
w->pre_select = file_write_pre_select;
w->post_select = file_write_post_select;
w->parse_config = file_write_parse_config;
w->close = file_write_close;
w->shutdown = NULL; /* nothing to do */
}
|
