genfio: Splitting gen_template in 2 parts
[fio.git] / server.h
CommitLineData
50d16976
JA
1#ifndef FIO_SERVER_H
2#define FIO_SERVER_H
3
132159a5 4#include <inttypes.h>
142575e6 5#include <string.h>
89c1707c 6#include <sys/time.h>
811826be 7#include <netinet/in.h>
132159a5 8
a64e88da 9#include "stat.h"
610a730c 10#include "os/os.h"
d09a64a0 11#include "diskutil.h"
a64e88da 12
5adc2447
SC
13#define FIO_NET_PORT 8765
14
132159a5
JA
15/*
16 * On-wire encoding is little endian
17 */
18struct fio_net_cmd {
19 uint16_t version; /* protocol version */
20 uint16_t opcode; /* command opcode */
21 uint32_t flags; /* modifier flags */
af9c9fb3 22 uint64_t tag; /* passed back on reply */
132159a5 23 uint32_t pdu_len; /* length of post-cmd layload */
a64e88da
JA
24 /*
25 * These must be immediately before the payload, anything before
26 * these fields are checksummed.
27 */
fcee5ff6
JA
28 uint16_t cmd_crc16; /* cmd checksum */
29 uint16_t pdu_crc16; /* payload checksum */
372aecb9 30 uint8_t payload[]; /* payload */
132159a5
JA
31};
32
40c60516 33struct fio_net_cmd_reply {
89c1707c
JA
34 struct flist_head list;
35 struct timeval tv;
36 uint64_t saved_tag;
40c60516 37 uint16_t opcode;
89c1707c
JA
38};
39
132159a5 40enum {
714e85f3 41 FIO_SERVER_VER = 23,
132159a5 42
b9d2f30a 43 FIO_SERVER_MAX_FRAGMENT_PDU = 1024,
132159a5 44
5d7793aa
JA
45 FIO_NET_CMD_QUIT = 1,
46 FIO_NET_CMD_EXIT = 2,
47 FIO_NET_CMD_JOB = 3,
48 FIO_NET_CMD_JOBLINE = 4,
49 FIO_NET_CMD_TEXT = 5,
50 FIO_NET_CMD_TS = 6,
51 FIO_NET_CMD_GS = 7,
52 FIO_NET_CMD_SEND_ETA = 8,
53 FIO_NET_CMD_ETA = 9,
54 FIO_NET_CMD_PROBE = 10,
55 FIO_NET_CMD_START = 11,
56 FIO_NET_CMD_STOP = 12,
57 FIO_NET_CMD_DU = 13,
58 FIO_NET_CMD_SERVER_START = 14,
59 FIO_NET_CMD_ADD_JOB = 15,
b9d2f30a 60 FIO_NET_CMD_RUN = 16,
1b42725f 61 FIO_NET_CMD_IOLOG = 17,
f58bd2a4
JA
62 FIO_NET_CMD_UPDATE_JOB = 18,
63 FIO_NET_CMD_NR = 19,
794d69ca 64
5d7793aa 65 FIO_NET_CMD_F_MORE = 1UL << 0,
fcee5ff6
JA
66
67 /* crc does not include the crc fields */
5d7793aa
JA
68 FIO_NET_CMD_CRC_SZ = sizeof(struct fio_net_cmd) -
69 2 * sizeof(uint16_t),
89c1707c 70
1b42725f
JA
71 FIO_NET_NAME_MAX = 256,
72
5d7793aa 73 FIO_NET_CLIENT_TIMEOUT = 5000,
3989b143
JA
74
75 FIO_PROBE_FLAG_ZLIB = 1UL << 0,
132159a5
JA
76};
77
a64e88da
JA
78struct cmd_ts_pdu {
79 struct thread_stat ts;
80 struct group_run_stats rs;
81};
82
d09a64a0
JA
83struct cmd_du_pdu {
84 struct disk_util_stat dus;
85 struct disk_util_agg agg;
86};
87
3989b143
JA
88struct cmd_client_probe_pdu {
89 uint64_t flags;
90};
91
92struct cmd_probe_reply_pdu {
c28e8e8c 93 uint8_t hostname[64];
6eb24791 94 uint8_t bigendian;
750db473 95 uint8_t fio_version[32];
cca84643
JA
96 uint8_t os;
97 uint8_t arch;
38fdef22 98 uint8_t bpp;
d31e26d0
JA
99 uint32_t cpus;
100 uint64_t flags;
c28e8e8c
JA
101};
102
fa2ea806
JA
103struct cmd_single_line_pdu {
104 uint16_t len;
372aecb9 105 uint8_t text[];
fa2ea806
JA
106};
107
81179eec 108struct cmd_line_pdu {
fa2ea806 109 uint16_t lines;
46bcd498 110 uint16_t client_type;
372aecb9 111 struct cmd_single_line_pdu options[];
81179eec
JA
112};
113
46bcd498
JA
114struct cmd_job_pdu {
115 uint32_t buf_len;
116 uint32_t client_type;
117 uint8_t buf[0];
118};
119
11e950bd
JA
120struct cmd_start_pdu {
121 uint32_t jobs;
108fea77 122 uint32_t stat_outputs;
11e950bd
JA
123};
124
125struct cmd_end_pdu {
126 uint32_t error;
122c7725 127 uint32_t signal;
11e950bd
JA
128};
129
807f9971 130struct cmd_add_job_pdu {
2f122b13
JA
131 uint32_t thread_number;
132 uint32_t groupid;
dcaeb606 133 struct thread_options_pack top;
807f9971
JA
134};
135
084d1c6f
JA
136struct cmd_text_pdu {
137 uint32_t level;
138 uint32_t buf_len;
139 uint64_t log_sec;
140 uint64_t log_usec;
141 uint8_t buf[0];
142};
143
1b42725f 144struct cmd_iolog_pdu {
2f122b13 145 uint32_t thread_number;
1b42725f
JA
146 uint32_t nr_samples;
147 uint32_t log_type;
3989b143 148 uint32_t compressed;
1b42725f
JA
149 uint8_t name[FIO_NET_NAME_MAX];
150 struct io_sample samples[0];
151};
152
402668f3 153extern int fio_start_server(char *);
084d1c6f 154extern int fio_server_text_output(int, const char *, size_t);
40c60516 155extern int fio_net_send_cmd(int, uint16_t, const void *, off_t, uint64_t *, struct flist_head *);
89c1707c 156extern int fio_net_send_simple_cmd(int, uint16_t, uint64_t, struct flist_head *);
bebe6398 157extern void fio_server_set_arg(const char *);
811826be 158extern int fio_server_parse_string(const char *, char **, int *, int *, struct in_addr *, struct in6_addr *, int *);
3ec62ec4 159extern int fio_server_parse_host(const char *, int *, struct in_addr *, struct in6_addr *);
89c1707c 160extern const char *fio_server_op(unsigned int);
7b821684 161extern void fio_server_got_signal(int);
a64e88da
JA
162
163struct thread_stat;
164struct group_run_stats;
165extern void fio_server_send_ts(struct thread_stat *, struct group_run_stats *);
166extern void fio_server_send_gs(struct group_run_stats *);
d09a64a0 167extern void fio_server_send_du(void);
cc0df00a 168extern void fio_server_idle_loop(void);
132159a5 169
a37f69b7
JA
170extern int fio_clients_connect(void);
171extern int fio_clients_send_ini(const char *);
fa2ea806 172extern void fio_client_add_cmd_option(void *, const char *);
bd023601 173extern void fio_client_add_ini_file(void *, const char *);
132159a5 174
132159a5
JA
175extern int fio_recv_data(int sk, void *p, unsigned int len);
176extern int fio_send_data(int sk, const void *p, unsigned int len);
177extern void fio_net_cmd_crc(struct fio_net_cmd *);
53bd8dbc 178extern void fio_net_cmd_crc_pdu(struct fio_net_cmd *, const void *);
e951bdc4 179extern struct fio_net_cmd *fio_net_recv_cmd(int sk);
132159a5 180
1b42725f 181extern int fio_send_iolog(struct thread_data *, struct io_log *, const char *);
2f122b13 182extern void fio_server_send_add_job(struct thread_data *);
122c7725
JA
183extern void fio_server_send_start(struct thread_data *);
184extern int fio_net_send_stop(int sk, int error, int signal);
185extern int fio_net_send_quit(int sk);
807f9971 186
009b1be4 187extern int exit_backend;
132159a5
JA
188extern int fio_net_port;
189
1b42725f
JA
190static inline void __fio_init_net_cmd(struct fio_net_cmd *cmd, uint16_t opcode,
191 uint32_t pdu_len, uint64_t tag)
132159a5
JA
192{
193 memset(cmd, 0, sizeof(*cmd));
4d8f8780 194
fa2ea806 195 cmd->version = __cpu_to_le16(FIO_SERVER_VER);
82fa6b21 196 cmd->opcode = cpu_to_le16(opcode);
af9c9fb3 197 cmd->tag = cpu_to_le64(tag);
1b42725f
JA
198 cmd->pdu_len = cpu_to_le32(pdu_len);
199}
200
201
202static inline void fio_init_net_cmd(struct fio_net_cmd *cmd, uint16_t opcode,
203 const void *pdu, uint32_t pdu_len,
204 uint64_t tag)
205{
206 __fio_init_net_cmd(cmd, opcode, pdu_len, tag);
4d8f8780 207
1b42725f 208 if (pdu)
4d8f8780 209 memcpy(&cmd->payload, pdu, pdu_len);
132159a5 210}
50d16976
JA
211
212#endif