Home | History | Annotate | Download | only in fio
      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