client.h

   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         int error;
  51         int signal;
  52         int ipv6;
  53         int sent_job;
  54         int did_stat;
  55         uint32_t type;
  56
  57         uint32_t thread_number;
  58         uint32_t groupid;
  59
  60         struct flist_head eta_list;
  61         struct client_eta *eta_in_flight;
  62
  63         struct flist_head cmd_list;
  64
  65         uint16_t argc;
  66         char **argv;
  67
  68         struct client_ops *ops;
  69         void *client_data;
  70
  71         struct client_file *files;
  72         unsigned int nr_files;
  73 };
  74
  75 struct cmd_iolog_pdu;
  76 typedef void (client_cmd_op)(struct fio_client *, struct fio_net_cmd *);
  77 typedef void (client_eta_op)(struct jobs_eta *je);
  78 typedef void (client_timed_out_op)(struct fio_client *);
  79 typedef void (client_jobs_eta_op)(struct fio_client *client, struct jobs_eta *je);
  80 typedef void (client_iolog_op)(struct fio_client *client, struct cmd_iolog_pdu *);
  81
  82 struct client_ops {
  83         client_cmd_op           *text;
  84         client_cmd_op           *disk_util;
  85         client_cmd_op           *thread_status;
  86         client_cmd_op           *group_stats;
  87         client_jobs_eta_op      *jobs_eta;
  88         client_eta_op           *eta;
  89         client_cmd_op           *probe;
  90         client_cmd_op           *quit;
  91         client_cmd_op           *add_job;
  92         client_cmd_op           *update_job;
  93         client_timed_out_op     *timed_out;
  94         client_cmd_op           *stop;
  95         client_cmd_op           *start;
  96         client_cmd_op           *job_start;
  97         client_iolog_op         *iolog;
  98         client_timed_out_op     *removed;
  99
 100         unsigned int eta_msec;
 101         int stay_connected;
 102         uint32_t client_type;
 103 };
 104
 105 extern struct client_ops fio_client_ops;
 106
 107 struct client_eta {
 108         unsigned int pending;
 109         struct jobs_eta eta;
 110 };
 111
 112 extern int fio_handle_client(struct fio_client *);
 113 extern void fio_client_dec_jobs_eta(struct client_eta *eta, client_eta_op fn);
 114 extern void fio_client_sum_jobs_eta(struct jobs_eta *dst, struct jobs_eta *je);
 115
 116 enum {
 117         Fio_client_ipv4 = 1,
 118         Fio_client_ipv6,
 119         Fio_client_socket,
 120 };
 121
 122 extern int fio_client_connect(struct fio_client *);
 123 extern int fio_clients_connect(void);
 124 extern int fio_start_client(struct fio_client *);
 125 extern int fio_start_all_clients(void);
 126 extern int fio_clients_send_ini(const char *);
 127 extern int fio_client_send_ini(struct fio_client *, const char *, int);
 128 extern int fio_handle_clients(struct client_ops *);
 129 extern int fio_client_add(struct client_ops *, const char *, void **);
 130 extern struct fio_client *fio_client_add_explicit(struct client_ops *, const char *, int, int);
 131 extern void fio_client_add_cmd_option(void *, const char *);
 132 extern int fio_client_add_ini_file(void *, const char *, int);
 133 extern int fio_client_terminate(struct fio_client *);
 134 extern void fio_clients_terminate(void);
 135 extern struct fio_client *fio_get_client(struct fio_client *);
 136 extern void fio_put_client(struct fio_client *);
 137 extern int fio_client_update_options(struct fio_client *, struct thread_options *, uint64_t *);
 138 extern int fio_client_wait_for_reply(struct fio_client *, uint64_t);
 139
 140 #define FIO_CLIENT_DEF_ETA_MSEC         900
 141
 142 enum {
 143         FIO_CLIENT_TYPE_CLI             = 1,
 144         FIO_CLIENT_TYPE_GUI             = 2,
 145 };
 146
 147 #endif
 148