Fix io_u->buf calculation overflow
[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>
132159a5 7
a64e88da 8#include "stat.h"
610a730c 9#include "os/os.h"
d09a64a0 10#include "diskutil.h"
a64e88da 11
132159a5
JA
12/*
13 * On-wire encoding is little endian
14 */
15struct fio_net_cmd {
16 uint16_t version; /* protocol version */
17 uint16_t opcode; /* command opcode */
18 uint32_t flags; /* modifier flags */
af9c9fb3 19 uint64_t tag; /* passed back on reply */
132159a5 20 uint32_t pdu_len; /* length of post-cmd layload */
a64e88da
JA
21 /*
22 * These must be immediately before the payload, anything before
23 * these fields are checksummed.
24 */
fcee5ff6
JA
25 uint16_t cmd_crc16; /* cmd checksum */
26 uint16_t pdu_crc16; /* payload checksum */
132159a5
JA
27 uint8_t payload[0]; /* payload */
28};
29
89c1707c
JA
30struct fio_net_int_cmd {
31 struct fio_net_cmd cmd;
32 struct flist_head list;
33 struct timeval tv;
34 uint64_t saved_tag;
35};
36
132159a5 37enum {
01be038e 38 FIO_SERVER_VER = 6,
132159a5 39
5cf6be78 40 FIO_SERVER_MAX_PDU = 1024,
132159a5
JA
41
42 FIO_NET_CMD_QUIT = 1,
d7959186
JA
43 FIO_NET_CMD_EXIT = 2,
44 FIO_NET_CMD_JOB = 3,
81179eec
JA
45 FIO_NET_CMD_JOBLINE = 4,
46 FIO_NET_CMD_TEXT = 5,
47 FIO_NET_CMD_TS = 6,
48 FIO_NET_CMD_GS = 7,
af9c9fb3
JA
49 FIO_NET_CMD_SEND_ETA = 8,
50 FIO_NET_CMD_ETA = 9,
51 FIO_NET_CMD_PROBE = 10,
52 FIO_NET_CMD_START = 11,
53 FIO_NET_CMD_STOP = 12,
d09a64a0 54 FIO_NET_CMD_DU = 13,
01be038e
JA
55 FIO_NET_CMD_RUN = 14,
56 FIO_NET_CMD_NR = 15,
794d69ca 57
d7959186 58 FIO_NET_CMD_F_MORE = 1UL << 0,
fcee5ff6
JA
59
60 /* crc does not include the crc fields */
61 FIO_NET_CMD_CRC_SZ = sizeof(struct fio_net_cmd) -
62 2 * sizeof(uint16_t),
89c1707c
JA
63
64 FIO_NET_CLIENT_TIMEOUT = 5000,
132159a5
JA
65};
66
a64e88da
JA
67struct cmd_ts_pdu {
68 struct thread_stat ts;
69 struct group_run_stats rs;
70};
71
d09a64a0
JA
72struct cmd_du_pdu {
73 struct disk_util_stat dus;
74 struct disk_util_agg agg;
75};
76
c28e8e8c
JA
77struct cmd_probe_pdu {
78 uint8_t hostname[64];
6eb24791 79 uint8_t bigendian;
c28e8e8c
JA
80 uint8_t fio_major;
81 uint8_t fio_minor;
82 uint8_t fio_patch;
cca84643
JA
83 uint8_t os;
84 uint8_t arch;
38fdef22 85 uint8_t bpp;
c28e8e8c
JA
86};
87
fa2ea806
JA
88struct cmd_single_line_pdu {
89 uint16_t len;
90 uint8_t text[0];
91};
92
81179eec 93struct cmd_line_pdu {
fa2ea806
JA
94 uint16_t lines;
95 struct cmd_single_line_pdu options[0];
81179eec
JA
96};
97
11e950bd
JA
98struct cmd_start_pdu {
99 uint32_t jobs;
100};
101
102struct cmd_end_pdu {
103 uint32_t error;
104};
105
402668f3 106extern int fio_start_server(char *);
13755d94 107extern int fio_server_text_output(const char *, size_t);
142575e6 108extern int fio_server_log(const char *format, ...);
af9c9fb3 109extern int fio_net_send_cmd(int, uint16_t, const void *, off_t, uint64_t);
89c1707c 110extern int fio_net_send_simple_cmd(int, uint16_t, uint64_t, struct flist_head *);
bebe6398
JA
111extern void fio_server_set_arg(const char *);
112extern int fio_server_parse_string(const char *, char **, int *, int *, struct in_addr *);
89c1707c 113extern const char *fio_server_op(unsigned int);
7b821684 114extern void fio_server_got_signal(int);
a64e88da
JA
115
116struct thread_stat;
117struct group_run_stats;
118extern void fio_server_send_ts(struct thread_stat *, struct group_run_stats *);
119extern void fio_server_send_gs(struct group_run_stats *);
d09a64a0 120extern void fio_server_send_du(void);
cc0df00a 121extern void fio_server_idle_loop(void);
132159a5 122
a37f69b7
JA
123extern int fio_clients_connect(void);
124extern int fio_clients_send_ini(const char *);
37db14fe 125extern int fio_handle_clients(void);
bebe6398 126extern int fio_client_add(const char *, void **);
fa2ea806 127extern void fio_client_add_cmd_option(void *, const char *);
132159a5
JA
128
129extern int fio_recv_data(int sk, void *p, unsigned int len);
130extern int fio_send_data(int sk, const void *p, unsigned int len);
131extern void fio_net_cmd_crc(struct fio_net_cmd *);
e951bdc4 132extern struct fio_net_cmd *fio_net_recv_cmd(int sk);
132159a5 133
009b1be4 134extern int exit_backend;
132159a5
JA
135extern int fio_net_port;
136
4d8f8780 137static inline void fio_init_net_cmd(struct fio_net_cmd *cmd, uint16_t opcode,
af9c9fb3
JA
138 const void *pdu, uint32_t pdu_len,
139 uint64_t tag)
132159a5
JA
140{
141 memset(cmd, 0, sizeof(*cmd));
4d8f8780 142
fa2ea806 143 cmd->version = __cpu_to_le16(FIO_SERVER_VER);
82fa6b21 144 cmd->opcode = cpu_to_le16(opcode);
af9c9fb3 145 cmd->tag = cpu_to_le64(tag);
4d8f8780
JA
146
147 if (pdu) {
148 cmd->pdu_len = cpu_to_le32(pdu_len);
149 memcpy(&cmd->payload, pdu, pdu_len);
150 }
132159a5 151}
50d16976
JA
152
153#endif