1 #ifndef CLIENT_H
2 #define CLIENT_H
3 
4 #include <sys/socket.h>
5 #include <sys/un.h>
6 #include <netinet/in.h>
7 #include <arpa/inet.h>
8 
9 #include "stat.h"
10 
11 struct fio_net_cmd;
12 struct client_ops;
13 
14 enum {
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 
23 struct client_file {
24 	char *file;
25 	int remote;
26 };
27 
28 struct fio_client {
29 	struct flist_head list;
30 	struct flist_head hash_list;
31 	struct flist_head arg_list;
32 	union {
33 		struct sockaddr_in addr;
34 		struct sockaddr_in6 addr6;
35 		struct sockaddr_un addr_un;
36 	};
37 	char *hostname;
38 	int port;
39 	int fd;
40 	unsigned int refs;
41 
42 	char *name;
43 
44 	int state;
45 
46 	int skip_newline;
47 	int is_sock;
48 	int disk_stats_shown;
49 	unsigned int jobs;
50 	unsigned int nr_stat;
51 	int error;
52 	int signal;
53 	int ipv6;
54 	int sent_job;
55 	int did_stat;
56 	uint32_t type;
57 
58 	uint32_t thread_number;
59 	uint32_t groupid;
60 
61 	struct flist_head eta_list;
62 	struct client_eta *eta_in_flight;
63 
64 	struct flist_head cmd_list;
65 
66 	uint16_t argc;
67 	char **argv;
68 
69 	struct client_ops *ops;
70 	void *client_data;
71 
72 	struct client_file *files;
73 	unsigned int nr_files;
74 };
75 
76 struct cmd_iolog_pdu;
77 typedef void (client_cmd_op)(struct fio_client *, struct fio_net_cmd *);
78 typedef void (client_eta_op)(struct jobs_eta *je);
79 typedef void (client_timed_out_op)(struct fio_client *);
80 typedef void (client_jobs_eta_op)(struct fio_client *client, struct jobs_eta *je);
81 typedef void (client_iolog_op)(struct fio_client *client, struct cmd_iolog_pdu *);
82 
83 struct client_ops {
84 	client_cmd_op		*text;
85 	client_cmd_op		*disk_util;
86 	client_cmd_op		*thread_status;
87 	client_cmd_op		*group_stats;
88 	client_jobs_eta_op	*jobs_eta;
89 	client_eta_op		*eta;
90 	client_cmd_op		*probe;
91 	client_cmd_op		*quit;
92 	client_cmd_op		*add_job;
93 	client_cmd_op		*update_job;
94 	client_timed_out_op	*timed_out;
95 	client_cmd_op		*stop;
96 	client_cmd_op		*start;
97 	client_cmd_op		*job_start;
98 	client_iolog_op		*iolog;
99 	client_timed_out_op	*removed;
100 
101 	unsigned int eta_msec;
102 	int stay_connected;
103 	uint32_t client_type;
104 };
105 
106 extern struct client_ops fio_client_ops;
107 
108 struct client_eta {
109 	unsigned int pending;
110 	struct jobs_eta eta;
111 };
112 
113 extern int fio_handle_client(struct fio_client *);
114 extern void fio_client_dec_jobs_eta(struct client_eta *eta, client_eta_op fn);
115 extern void fio_client_sum_jobs_eta(struct jobs_eta *dst, struct jobs_eta *je);
116 
117 enum {
118 	Fio_client_ipv4 = 1,
119 	Fio_client_ipv6,
120 	Fio_client_socket,
121 };
122 
123 extern int fio_client_connect(struct fio_client *);
124 extern int fio_clients_connect(void);
125 extern int fio_start_client(struct fio_client *);
126 extern int fio_start_all_clients(void);
127 extern int fio_clients_send_ini(const char *);
128 extern int fio_client_send_ini(struct fio_client *, const char *, int);
129 extern int fio_handle_clients(struct client_ops *);
130 extern int fio_client_add(struct client_ops *, const char *, void **);
131 extern struct fio_client *fio_client_add_explicit(struct client_ops *, const char *, int, int);
132 extern void fio_client_add_cmd_option(void *, const char *);
133 extern int fio_client_add_ini_file(void *, const char *, int);
134 extern int fio_client_terminate(struct fio_client *);
135 extern void fio_clients_terminate(void);
136 extern struct fio_client *fio_get_client(struct fio_client *);
137 extern void fio_put_client(struct fio_client *);
138 extern int fio_client_update_options(struct fio_client *, struct thread_options *, uint64_t *);
139 extern int fio_client_wait_for_reply(struct fio_client *, uint64_t);
140 extern int fio_clients_send_trigger(const char *);
141 
142 #define FIO_CLIENT_DEF_ETA_MSEC		900
143 
144 enum {
145 	FIO_CLIENT_TYPE_CLI		= 1,
146 	FIO_CLIENT_TYPE_GUI		= 2,
147 };
148 
149 #endif
150 
151