Commit | Line | Data |
---|---|---|
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 | */ | |
18 | struct 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 | 33 | struct 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 | 40 | enum { |
2099539d | 41 | FIO_SERVER_VER = 47, |
132159a5 | 42 | |
b9d2f30a | 43 | FIO_SERVER_MAX_FRAGMENT_PDU = 1024, |
dfbf1f6f | 44 | FIO_SERVER_MAX_CMD_MB = 2048, |
132159a5 | 45 | |
5d7793aa JA |
46 | FIO_NET_CMD_QUIT = 1, |
47 | FIO_NET_CMD_EXIT = 2, | |
48 | FIO_NET_CMD_JOB = 3, | |
49 | FIO_NET_CMD_JOBLINE = 4, | |
50 | FIO_NET_CMD_TEXT = 5, | |
51 | FIO_NET_CMD_TS = 6, | |
52 | FIO_NET_CMD_GS = 7, | |
53 | FIO_NET_CMD_SEND_ETA = 8, | |
54 | FIO_NET_CMD_ETA = 9, | |
55 | FIO_NET_CMD_PROBE = 10, | |
56 | FIO_NET_CMD_START = 11, | |
57 | FIO_NET_CMD_STOP = 12, | |
58 | FIO_NET_CMD_DU = 13, | |
59 | FIO_NET_CMD_SERVER_START = 14, | |
60 | FIO_NET_CMD_ADD_JOB = 15, | |
b9d2f30a | 61 | FIO_NET_CMD_RUN = 16, |
1b42725f | 62 | FIO_NET_CMD_IOLOG = 17, |
f58bd2a4 | 63 | FIO_NET_CMD_UPDATE_JOB = 18, |
323255cc | 64 | FIO_NET_CMD_LOAD_FILE = 19, |
ca09be4b JA |
65 | FIO_NET_CMD_VTRIGGER = 20, |
66 | FIO_NET_CMD_SENDFILE = 21, | |
67 | FIO_NET_CMD_NR = 22, | |
794d69ca | 68 | |
5d7793aa | 69 | FIO_NET_CMD_F_MORE = 1UL << 0, |
fcee5ff6 JA |
70 | |
71 | /* crc does not include the crc fields */ | |
5d7793aa JA |
72 | FIO_NET_CMD_CRC_SZ = sizeof(struct fio_net_cmd) - |
73 | 2 * sizeof(uint16_t), | |
89c1707c | 74 | |
1b42725f JA |
75 | FIO_NET_NAME_MAX = 256, |
76 | ||
125451c0 | 77 | FIO_NET_CLIENT_TIMEOUT = 30000, |
3989b143 JA |
78 | |
79 | FIO_PROBE_FLAG_ZLIB = 1UL << 0, | |
132159a5 JA |
80 | }; |
81 | ||
ca09be4b JA |
82 | struct cmd_sendfile { |
83 | uint8_t path[FIO_NET_NAME_MAX]; | |
84 | }; | |
85 | ||
86 | struct cmd_sendfile_reply { | |
87 | uint32_t size; | |
88 | uint32_t error; | |
89 | uint8_t data[0]; | |
90 | }; | |
91 | ||
92 | /* | |
93 | * Client sends this to server on VTRIGGER, server sends back a full | |
94 | * all_io_list structure. | |
95 | */ | |
96 | struct cmd_vtrigger_pdu { | |
97 | uint16_t len; | |
98 | uint8_t cmd[]; | |
99 | }; | |
100 | ||
323255cc JA |
101 | struct cmd_load_file_pdu { |
102 | uint16_t name_len; | |
103 | uint16_t client_type; | |
104 | uint8_t file[]; | |
105 | }; | |
106 | ||
a64e88da JA |
107 | struct cmd_ts_pdu { |
108 | struct thread_stat ts; | |
109 | struct group_run_stats rs; | |
110 | }; | |
111 | ||
d09a64a0 JA |
112 | struct cmd_du_pdu { |
113 | struct disk_util_stat dus; | |
114 | struct disk_util_agg agg; | |
115 | }; | |
116 | ||
3989b143 JA |
117 | struct cmd_client_probe_pdu { |
118 | uint64_t flags; | |
ca09be4b | 119 | uint8_t server[128]; |
3989b143 JA |
120 | }; |
121 | ||
122 | struct cmd_probe_reply_pdu { | |
c28e8e8c | 123 | uint8_t hostname[64]; |
6eb24791 | 124 | uint8_t bigendian; |
750db473 | 125 | uint8_t fio_version[32]; |
cca84643 JA |
126 | uint8_t os; |
127 | uint8_t arch; | |
38fdef22 | 128 | uint8_t bpp; |
d31e26d0 JA |
129 | uint32_t cpus; |
130 | uint64_t flags; | |
c28e8e8c JA |
131 | }; |
132 | ||
fa2ea806 JA |
133 | struct cmd_single_line_pdu { |
134 | uint16_t len; | |
372aecb9 | 135 | uint8_t text[]; |
fa2ea806 JA |
136 | }; |
137 | ||
81179eec | 138 | struct cmd_line_pdu { |
fa2ea806 | 139 | uint16_t lines; |
46bcd498 | 140 | uint16_t client_type; |
372aecb9 | 141 | struct cmd_single_line_pdu options[]; |
81179eec JA |
142 | }; |
143 | ||
46bcd498 JA |
144 | struct cmd_job_pdu { |
145 | uint32_t buf_len; | |
146 | uint32_t client_type; | |
147 | uint8_t buf[0]; | |
148 | }; | |
149 | ||
11e950bd JA |
150 | struct cmd_start_pdu { |
151 | uint32_t jobs; | |
108fea77 | 152 | uint32_t stat_outputs; |
11e950bd JA |
153 | }; |
154 | ||
155 | struct cmd_end_pdu { | |
156 | uint32_t error; | |
122c7725 | 157 | uint32_t signal; |
11e950bd JA |
158 | }; |
159 | ||
807f9971 | 160 | struct cmd_add_job_pdu { |
2f122b13 JA |
161 | uint32_t thread_number; |
162 | uint32_t groupid; | |
dcaeb606 | 163 | struct thread_options_pack top; |
807f9971 JA |
164 | }; |
165 | ||
084d1c6f JA |
166 | struct cmd_text_pdu { |
167 | uint32_t level; | |
168 | uint32_t buf_len; | |
169 | uint64_t log_sec; | |
170 | uint64_t log_usec; | |
171 | uint8_t buf[0]; | |
172 | }; | |
173 | ||
1b42725f | 174 | struct cmd_iolog_pdu { |
ae588852 | 175 | uint64_t nr_samples; |
2f122b13 | 176 | uint32_t thread_number; |
1b42725f | 177 | uint32_t log_type; |
3989b143 | 178 | uint32_t compressed; |
ae588852 | 179 | uint32_t log_offset; |
1b42725f JA |
180 | uint8_t name[FIO_NET_NAME_MAX]; |
181 | struct io_sample samples[0]; | |
182 | }; | |
183 | ||
402668f3 | 184 | extern int fio_start_server(char *); |
084d1c6f | 185 | extern int fio_server_text_output(int, const char *, size_t); |
40c60516 | 186 | extern int fio_net_send_cmd(int, uint16_t, const void *, off_t, uint64_t *, struct flist_head *); |
89c1707c | 187 | extern int fio_net_send_simple_cmd(int, uint16_t, uint64_t, struct flist_head *); |
bebe6398 | 188 | extern void fio_server_set_arg(const char *); |
811826be | 189 | extern int fio_server_parse_string(const char *, char **, int *, int *, struct in_addr *, struct in6_addr *, int *); |
3aa3ceeb | 190 | extern int fio_server_parse_host(const char *, int, struct in_addr *, struct in6_addr *); |
89c1707c | 191 | extern const char *fio_server_op(unsigned int); |
7b821684 | 192 | extern void fio_server_got_signal(int); |
a64e88da JA |
193 | |
194 | struct thread_stat; | |
195 | struct group_run_stats; | |
196 | extern void fio_server_send_ts(struct thread_stat *, struct group_run_stats *); | |
197 | extern void fio_server_send_gs(struct group_run_stats *); | |
d09a64a0 | 198 | extern void fio_server_send_du(void); |
cc0df00a | 199 | extern void fio_server_idle_loop(void); |
c3546b53 | 200 | extern int fio_server_get_verify_state(const char *, int, void **, int *); |
132159a5 | 201 | |
132159a5 JA |
202 | extern int fio_recv_data(int sk, void *p, unsigned int len); |
203 | extern int fio_send_data(int sk, const void *p, unsigned int len); | |
204 | extern void fio_net_cmd_crc(struct fio_net_cmd *); | |
53bd8dbc | 205 | extern void fio_net_cmd_crc_pdu(struct fio_net_cmd *, const void *); |
e951bdc4 | 206 | extern struct fio_net_cmd *fio_net_recv_cmd(int sk); |
132159a5 | 207 | |
1b42725f | 208 | extern int fio_send_iolog(struct thread_data *, struct io_log *, const char *); |
2f122b13 | 209 | extern void fio_server_send_add_job(struct thread_data *); |
122c7725 JA |
210 | extern void fio_server_send_start(struct thread_data *); |
211 | extern int fio_net_send_stop(int sk, int error, int signal); | |
212 | extern int fio_net_send_quit(int sk); | |
807f9971 | 213 | |
009b1be4 | 214 | extern int exit_backend; |
132159a5 JA |
215 | extern int fio_net_port; |
216 | ||
1b42725f JA |
217 | static inline void __fio_init_net_cmd(struct fio_net_cmd *cmd, uint16_t opcode, |
218 | uint32_t pdu_len, uint64_t tag) | |
132159a5 JA |
219 | { |
220 | memset(cmd, 0, sizeof(*cmd)); | |
4d8f8780 | 221 | |
fa2ea806 | 222 | cmd->version = __cpu_to_le16(FIO_SERVER_VER); |
82fa6b21 | 223 | cmd->opcode = cpu_to_le16(opcode); |
af9c9fb3 | 224 | cmd->tag = cpu_to_le64(tag); |
1b42725f JA |
225 | cmd->pdu_len = cpu_to_le32(pdu_len); |
226 | } | |
227 | ||
228 | ||
229 | static inline void fio_init_net_cmd(struct fio_net_cmd *cmd, uint16_t opcode, | |
230 | const void *pdu, uint32_t pdu_len, | |
231 | uint64_t tag) | |
232 | { | |
233 | __fio_init_net_cmd(cmd, opcode, pdu_len, tag); | |
4d8f8780 | 234 | |
1b42725f | 235 | if (pdu) |
4d8f8780 | 236 | memcpy(&cmd->payload, pdu, pdu_len); |
132159a5 | 237 | } |
50d16976 JA |
238 | |
239 | #endif |