client: unify cmd handling ops
[fio.git] / client.h
CommitLineData
dd366728
SC
1#ifndef CLIENT_H
2#define CLIENT_H
3
0f92bd20
JA
4#include <sys/socket.h>
5#include <sys/un.h>
6#include <netinet/in.h>
7#include <arpa/inet.h>
8
3e47bd25
JA
9#include "stat.h"
10
dd366728 11struct fio_net_cmd;
a5276616 12struct client_ops;
dd366728 13
b9d2f30a
JA
14enum {
15 Client_created = 0,
16 Client_connected = 1,
17 Client_started = 2,
18 Client_running = 3,
19 Client_stopped = 4,
20 Client_exited = 5,
21};
22
0f92bd20
JA
23struct fio_client {
24 struct flist_head list;
25 struct flist_head hash_list;
26 struct flist_head arg_list;
27 union {
28 struct sockaddr_in addr;
29 struct sockaddr_in6 addr6;
30 struct sockaddr_un addr_un;
31 };
32 char *hostname;
33 int port;
34 int fd;
5121a9aa 35 unsigned int refs;
0f92bd20
JA
36
37 char *name;
38
39 int state;
40
41 int skip_newline;
42 int is_sock;
43 int disk_stats_shown;
44 unsigned int jobs;
45 int error;
46 int ipv6;
47 int sent_job;
48
49 struct flist_head eta_list;
50 struct client_eta *eta_in_flight;
51
52 struct flist_head cmd_list;
53
54 uint16_t argc;
55 char **argv;
3ec62ec4 56
a5276616 57 struct client_ops *ops;
3ec62ec4 58 void *client_data;
0f92bd20
JA
59};
60
35c0ba7f
JA
61typedef void (client_cmd_op)(struct fio_client *, struct fio_net_cmd *);
62typedef void (client_eta_op)(struct jobs_eta *je);
63typedef void (client_timed_out_op)(struct fio_client *);
64typedef void (client_jobs_eta_op)(struct fio_client *client, struct jobs_eta *je);
3ec62ec4 65
dd366728 66struct client_ops {
35c0ba7f
JA
67 client_cmd_op *text;
68 client_cmd_op *disk_util;
69 client_cmd_op *thread_status;
70 client_cmd_op *group_stats;
71 client_jobs_eta_op *jobs_eta;
72 client_eta_op *eta;
73 client_cmd_op *probe;
74 client_cmd_op *quit;
75 client_cmd_op *add_job;
76 client_timed_out_op *timed_out;
77 client_cmd_op *stop;
78 client_cmd_op *start;
79 client_cmd_op *job_start;
80
6433ee05 81 unsigned int eta_msec;
3ec62ec4 82 int stay_connected;
dd366728
SC
83};
84
85extern struct client_ops fio_client_ops;
86
3e47bd25
JA
87struct client_eta {
88 struct jobs_eta eta;
89 unsigned int pending;
90};
91
a5276616
JA
92extern int fio_handle_client(struct fio_client *);
93extern void fio_client_dec_jobs_eta(struct client_eta *eta, client_eta_op fn);
3e47bd25 94extern void fio_client_sum_jobs_eta(struct jobs_eta *dst, struct jobs_eta *je);
3e47bd25 95
3ec62ec4
JA
96enum {
97 Fio_client_ipv4 = 1,
98 Fio_client_ipv6,
99 Fio_client_socket,
100};
101
2f99deb0 102extern int fio_client_connect(struct fio_client *);
3ec62ec4 103extern int fio_clients_connect(void);
b9d2f30a
JA
104extern int fio_start_client(struct fio_client *);
105extern int fio_start_all_clients(void);
9988ca70 106extern int fio_client_send_ini(struct fio_client *, const char *);
3ec62ec4 107extern int fio_clients_send_ini(const char *);
a5276616
JA
108extern int fio_handle_clients(struct client_ops *);
109extern int fio_client_add(struct client_ops *, const char *, void **);
110extern struct fio_client *fio_client_add_explicit(struct client_ops *, const char *, int, int);
3ec62ec4 111extern void fio_client_add_cmd_option(void *, const char *);
2f99deb0 112extern void fio_client_terminate(struct fio_client *);
df06f220 113extern void fio_clients_terminate(void);
343cb4a9
JA
114extern struct fio_client *fio_get_client(struct fio_client *);
115extern void fio_put_client(struct fio_client *);
df06f220 116
6433ee05
JA
117#define FIO_CLIENT_DEF_ETA_MSEC 900
118
dd366728
SC
119#endif
120