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
