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
|
/* Copyright (c) 2010-2018 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "array.h"
#include "ostream.h"
#include "doveadm-print-private.h"
struct doveadm_print_flow_header {
const char *title;
enum doveadm_print_header_flags flags;
};
struct doveadm_print_flow_context {
pool_t pool;
ARRAY(struct doveadm_print_flow_header) headers;
unsigned int header_idx;
bool streaming:1;
};
static struct doveadm_print_flow_context *ctx;
static void
doveadm_print_flow_header(const struct doveadm_print_header *hdr)
{
struct doveadm_print_flow_header *fhdr;
fhdr = array_append_space(&ctx->headers);
fhdr->title = p_strdup(ctx->pool, hdr->title);
fhdr->flags = hdr->flags;
}
static void flow_next_hdr(void)
{
if (++ctx->header_idx < array_count(&ctx->headers))
o_stream_nsend(doveadm_print_ostream, " ", 1);
else {
ctx->header_idx = 0;
o_stream_nsend(doveadm_print_ostream, "\n", 1);
}
}
static void doveadm_print_flow_print_header(const struct doveadm_print_flow_header *hdr)
{
if ((hdr->flags & DOVEADM_PRINT_HEADER_FLAG_HIDE_TITLE) == 0) {
o_stream_nsend_str(doveadm_print_ostream, hdr->title);
o_stream_nsend(doveadm_print_ostream, "=", 1);
}
}
static void doveadm_print_flow_print(const char *value)
{
const struct doveadm_print_flow_header *hdr =
array_idx(&ctx->headers, ctx->header_idx);
doveadm_print_flow_print_header(hdr);
o_stream_nsend_str(doveadm_print_ostream, value);
flow_next_hdr();
}
static void
doveadm_print_flow_print_stream(const unsigned char *value, size_t size)
{
const struct doveadm_print_flow_header *hdr =
array_idx(&ctx->headers, ctx->header_idx);
if (!ctx->streaming) {
ctx->streaming = TRUE;
doveadm_print_flow_print_header(hdr);
}
o_stream_nsend(doveadm_print_ostream, value, size);
if (size == 0) {
flow_next_hdr();
ctx->streaming = FALSE;
}
}
static void doveadm_print_flow_init(void)
{
pool_t pool;
pool = pool_alloconly_create("doveadm print flow", 1024);
ctx = p_new(pool, struct doveadm_print_flow_context, 1);
ctx->pool = pool;
p_array_init(&ctx->headers, pool, 16);
}
static void doveadm_print_flow_flush(void)
{
if (ctx->header_idx != 0) {
o_stream_nsend(doveadm_print_ostream, "\n", 1);
ctx->header_idx = 0;
}
}
static void doveadm_print_flow_deinit(void)
{
pool_unref(&ctx->pool);
ctx = NULL;
}
struct doveadm_print_vfuncs doveadm_print_flow_vfuncs = {
"flow",
doveadm_print_flow_init,
doveadm_print_flow_deinit,
doveadm_print_flow_header,
doveadm_print_flow_print,
doveadm_print_flow_print_stream,
doveadm_print_flow_flush
};
|
