Commit | Line | Data |
---|---|---|
ff1f3280 SC |
1 | /* |
2 | * gfio - gui front end for fio - the flexible io tester | |
3 | * | |
4 | * Copyright (C) 2012 Stephen M. Cameron <stephenmcameron@gmail.com> | |
5 | * | |
6 | * The license below covers all files distributed with fio unless otherwise | |
7 | * noted in the file itself. | |
8 | * | |
9 | * This program is free software; you can redistribute it and/or modify | |
10 | * it under the terms of the GNU General Public License version 2 as | |
11 | * published by the Free Software Foundation. | |
12 | * | |
13 | * This program is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
16 | * GNU General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU General Public License | |
19 | * along with this program; if not, write to the Free Software | |
20 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
21 | * | |
22 | */ | |
8232e285 | 23 | #include <locale.h> |
60f6b330 | 24 | #include <malloc.h> |
8232e285 | 25 | |
5b7573ab | 26 | #include <glib.h> |
ff1f3280 SC |
27 | #include <gtk/gtk.h> |
28 | ||
8232e285 SC |
29 | #include "fio_initialization.h" |
30 | #include "fio.h" | |
31 | ||
f3074008 SC |
32 | #define ARRAYSIZE(x) (sizeof((x)) / (sizeof((x)[0]))) |
33 | ||
34 | typedef void (*clickfunction)(GtkWidget *widget, gpointer data); | |
35 | ||
36 | static void quit_clicked(GtkWidget *widget, gpointer data); | |
37 | static void start_job_clicked(GtkWidget *widget, gpointer data); | |
38 | ||
39 | static struct button_spec { | |
40 | const char *buttontext; | |
41 | clickfunction f; | |
42 | const char *tooltiptext; | |
43 | } buttonspeclist[] = { | |
44 | #define START_JOB_BUTTON 0 | |
45 | { "Start Job", | |
46 | start_job_clicked, | |
47 | "Send current fio job to fio server to be executed" }, | |
48 | #define QUIT_BUTTON 1 | |
49 | { "Quit", quit_clicked, "Quit gfio" }, | |
50 | }; | |
51 | ||
ff1f3280 | 52 | struct gui { |
60f6b330 SC |
53 | int argc; |
54 | char **argv; | |
ff1f3280 | 55 | GtkWidget *window; |
5b7573ab | 56 | GtkWidget *vbox; |
04cc6b77 | 57 | GtkWidget *thread_status_pb; |
f3074008 SC |
58 | GtkWidget *buttonbox; |
59 | GtkWidget *button[ARRAYSIZE(buttonspeclist)]; | |
45032dd8 SC |
60 | GtkWidget *hostname_hbox; |
61 | GtkWidget *hostname_label; | |
62 | GtkWidget *hostname_entry; | |
63 | GtkWidget *port_label; | |
64 | GtkWidget *port_entry; | |
65 | GtkWidget *hostname_combo_box; /* ipv4, ipv6 or socket */ | |
47066341 SC |
66 | GtkWidget *jobfile_hbox; |
67 | GtkWidget *jobfile_label; | |
68 | GtkWidget *jobfile_entry; | |
736f2dff SC |
69 | GtkWidget *scrolled_window; |
70 | GtkWidget *textview; | |
71 | GtkTextBuffer *text; | |
25927259 | 72 | pthread_t t; |
5b7573ab | 73 | } ui; |
ff1f3280 | 74 | |
a1820207 SC |
75 | static void gfio_text_op(struct fio_client *client, |
76 | FILE *f, __u16 pdu_len, const char *buf) | |
77 | { | |
736f2dff SC |
78 | GtkTextBuffer *buffer; |
79 | GtkTextIter end; | |
80 | ||
81 | buffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui.textview)); | |
82 | gdk_threads_enter(); | |
83 | gtk_text_buffer_get_end_iter(buffer, &end); | |
84 | gtk_text_buffer_insert(buffer, &end, buf, -1); | |
85 | gdk_threads_leave(); | |
86 | gtk_text_view_scroll_to_iter(GTK_TEXT_VIEW(ui.textview), | |
87 | &end, 0.0, FALSE, 0.0,0.0); | |
a1820207 SC |
88 | } |
89 | ||
90 | static void gfio_disk_util_op(struct fio_client *client, struct fio_net_cmd *cmd) | |
91 | { | |
92 | printf("gfio_disk_util_op called\n"); | |
93 | fio_client_ops.disk_util(client, cmd); | |
94 | } | |
95 | ||
96 | static void gfio_thread_status_op(struct fio_net_cmd *cmd) | |
97 | { | |
98 | printf("gfio_thread_status_op called\n"); | |
99 | fio_client_ops.thread_status(cmd); | |
100 | } | |
101 | ||
102 | static void gfio_group_stats_op(struct fio_net_cmd *cmd) | |
103 | { | |
104 | printf("gfio_group_stats_op called\n"); | |
105 | fio_client_ops.group_stats(cmd); | |
106 | } | |
107 | ||
108 | static void gfio_eta_op(struct fio_client *client, struct fio_net_cmd *cmd) | |
109 | { | |
a1820207 SC |
110 | fio_client_ops.eta(client, cmd); |
111 | } | |
112 | ||
113 | static void gfio_probe_op(struct fio_client *client, struct fio_net_cmd *cmd) | |
114 | { | |
115 | printf("gfio_probe_op called\n"); | |
116 | fio_client_ops.probe(client, cmd); | |
117 | } | |
118 | ||
04cc6b77 | 119 | static void gfio_update_thread_status(char *status_message, double perc) |
5b7573ab SC |
120 | { |
121 | static char message[100]; | |
122 | const char *m = message; | |
123 | ||
124 | strncpy(message, status_message, sizeof(message) - 1); | |
04cc6b77 SC |
125 | gtk_progress_bar_set_text( |
126 | GTK_PROGRESS_BAR(ui.thread_status_pb), m); | |
127 | gtk_progress_bar_set_fraction( | |
128 | GTK_PROGRESS_BAR(ui.thread_status_pb), perc / 100.0); | |
5b7573ab SC |
129 | gdk_threads_enter(); |
130 | gtk_widget_queue_draw(ui.window); | |
131 | gdk_threads_leave(); | |
132 | } | |
133 | ||
a1820207 SC |
134 | struct client_ops gfio_client_ops = { |
135 | gfio_text_op, | |
136 | gfio_disk_util_op, | |
137 | gfio_thread_status_op, | |
138 | gfio_group_stats_op, | |
139 | gfio_eta_op, | |
140 | gfio_probe_op, | |
5b7573ab | 141 | gfio_update_thread_status, |
a1820207 SC |
142 | }; |
143 | ||
ff1f3280 SC |
144 | static void quit_clicked(__attribute__((unused)) GtkWidget *widget, |
145 | __attribute__((unused)) gpointer data) | |
146 | { | |
147 | gtk_main_quit(); | |
148 | } | |
149 | ||
60f6b330 SC |
150 | static void add_arg(char **argv, int index, const char *value) |
151 | { | |
152 | argv[index] = malloc(strlen(value) + 1); | |
153 | strcpy(argv[index], value); | |
154 | } | |
155 | ||
156 | static void free_args(int argc, char **argv) | |
157 | { | |
158 | int i; | |
159 | ||
160 | for (i = 0; i < argc; i++) | |
161 | free(argv[i]); | |
162 | free(argv); | |
163 | } | |
164 | ||
25927259 SC |
165 | static void *job_thread(void *arg) |
166 | { | |
167 | struct gui *ui = arg; | |
168 | ||
169 | fio_handle_clients(&gfio_client_ops); | |
170 | gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1); | |
60f6b330 | 171 | free_args(ui->argc, ui->argv); |
25927259 SC |
172 | return NULL; |
173 | } | |
174 | ||
60f6b330 SC |
175 | static void construct_options(struct gui *ui, int *argc, char ***argv) |
176 | { | |
177 | const char *hostname, *hostname_type, *port, *jobfile; | |
178 | char newarg[200]; | |
179 | ||
180 | hostname_type = gtk_entry_get_text(GTK_ENTRY(GTK_COMBO(ui->hostname_combo_box)->entry)); | |
181 | hostname = gtk_entry_get_text(GTK_ENTRY(ui->hostname_entry)); | |
182 | port = gtk_entry_get_text(GTK_ENTRY(ui->port_entry)); | |
183 | jobfile = gtk_entry_get_text(GTK_ENTRY(ui->jobfile_entry)); | |
184 | ||
185 | *argc = 3; | |
186 | *argv = malloc(*argc * sizeof(**argv)); | |
187 | add_arg(*argv, 0, "gfio"); | |
188 | snprintf(newarg, sizeof(newarg) - 1, "--client=%s", hostname); | |
189 | add_arg(*argv, 1, newarg); | |
190 | add_arg(*argv, 2, jobfile); | |
191 | } | |
192 | ||
25927259 SC |
193 | static void start_job_thread(pthread_t *t, struct gui *ui) |
194 | { | |
60f6b330 SC |
195 | construct_options(ui, &ui->argc, &ui->argv); |
196 | if (parse_options(ui->argc, ui->argv)) { | |
197 | printf("Yeah, I didn't really like those options too much.\n"); | |
198 | free_args(ui->argc, ui->argv); | |
199 | gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 1); | |
200 | return; | |
201 | } | |
25927259 SC |
202 | pthread_create(t, NULL, job_thread, ui); |
203 | } | |
204 | ||
f3074008 | 205 | static void start_job_clicked(__attribute__((unused)) GtkWidget *widget, |
25927259 | 206 | gpointer data) |
f3074008 | 207 | { |
25927259 SC |
208 | struct gui *ui = data; |
209 | ||
f3074008 | 210 | printf("Start job button was clicked.\n"); |
25927259 SC |
211 | gtk_widget_set_sensitive(ui->button[START_JOB_BUTTON], 0); |
212 | start_job_thread(&ui->t, ui); | |
f3074008 SC |
213 | } |
214 | ||
215 | static void add_button(struct gui *ui, int i, GtkWidget *buttonbox, | |
216 | struct button_spec *buttonspec) | |
217 | { | |
218 | ui->button[i] = gtk_button_new_with_label(buttonspec->buttontext); | |
219 | g_signal_connect(ui->button[i], "clicked", G_CALLBACK (buttonspec->f), ui); | |
220 | gtk_box_pack_start(GTK_BOX (ui->buttonbox), ui->button[i], TRUE, TRUE, 0); | |
221 | gtk_widget_set_tooltip_text(ui->button[i], buttonspeclist[i].tooltiptext); | |
222 | } | |
223 | ||
224 | static void add_buttons(struct gui *ui, | |
225 | struct button_spec *buttonlist, | |
226 | int nbuttons) | |
227 | { | |
228 | int i; | |
229 | ||
230 | ui->buttonbox = gtk_hbox_new(FALSE, 0); | |
5b7573ab | 231 | gtk_container_add(GTK_CONTAINER (ui->vbox), ui->buttonbox); |
f3074008 SC |
232 | for (i = 0; i < nbuttons; i++) |
233 | add_button(ui, i, ui->buttonbox, &buttonlist[i]); | |
234 | } | |
235 | ||
ff1f3280 SC |
236 | static void init_ui(int *argc, char **argv[], struct gui *ui) |
237 | { | |
45032dd8 SC |
238 | GList *hostname_type_list = NULL; |
239 | char portnum[20]; | |
240 | ||
2839f0c6 | 241 | /* Magical g*thread incantation, you just need this thread stuff. |
04cc6b77 | 242 | * Without it, the update that happens in gfio_update_thread_status |
2839f0c6 SC |
243 | * doesn't really happen in a timely fashion, you need expose events |
244 | */ | |
245 | if (!g_thread_supported ()) | |
246 | g_thread_init(NULL); | |
247 | gdk_threads_init(); | |
248 | ||
ff1f3280 SC |
249 | gtk_init(argc, argv); |
250 | ||
251 | ui->window = gtk_window_new(GTK_WINDOW_TOPLEVEL); | |
252 | gtk_window_set_title(GTK_WINDOW(ui->window), "fio"); | |
253 | gtk_window_set_default_size(GTK_WINDOW(ui->window), 700, 500); | |
254 | ||
255 | g_signal_connect(ui->window, "delete-event", G_CALLBACK (quit_clicked), NULL); | |
256 | g_signal_connect(ui->window, "destroy", G_CALLBACK (quit_clicked), NULL); | |
257 | ||
5b7573ab SC |
258 | ui->vbox = gtk_vbox_new(FALSE, 0); |
259 | gtk_container_add(GTK_CONTAINER (ui->window), ui->vbox); | |
04cc6b77 | 260 | |
45032dd8 SC |
261 | /* |
262 | * Set up hostname label + entry, port label + entry, | |
263 | */ | |
264 | ui->hostname_hbox = gtk_hbox_new(FALSE, 0); | |
265 | ui->hostname_label = gtk_label_new("Host:"); | |
266 | ui->hostname_entry = gtk_entry_new(); | |
267 | gtk_entry_set_text(GTK_ENTRY(ui->hostname_entry), "localhost"); | |
268 | ui->port_label = gtk_label_new("Port:"); | |
269 | ui->port_entry = gtk_entry_new(); | |
270 | snprintf(portnum, sizeof(portnum) - 1, "%d", FIO_NET_PORT); | |
271 | gtk_entry_set_text(GTK_ENTRY(ui->port_entry), (gchar *) portnum); | |
272 | ||
273 | /* | |
274 | * Set up combo box for address type | |
275 | */ | |
276 | ui->hostname_combo_box = gtk_combo_new(); | |
277 | gtk_entry_set_text(GTK_ENTRY (GTK_COMBO(ui->hostname_combo_box)->entry), "IPv4"); | |
278 | hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv4"); | |
279 | hostname_type_list = g_list_append(hostname_type_list, (gpointer) "local socket"); | |
280 | hostname_type_list = g_list_append(hostname_type_list, (gpointer) "IPv6"); | |
281 | gtk_combo_set_popdown_strings (GTK_COMBO (ui->hostname_combo_box), hostname_type_list); | |
282 | g_list_free(hostname_type_list); | |
283 | ||
284 | gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_label); | |
285 | gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_entry); | |
286 | gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_label); | |
287 | gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->port_entry); | |
288 | gtk_container_add(GTK_CONTAINER (ui->hostname_hbox), ui->hostname_combo_box); | |
289 | gtk_container_add(GTK_CONTAINER (ui->vbox), ui->hostname_hbox); | |
290 | ||
47066341 SC |
291 | /* |
292 | * Set up jobfile text entry (temporary until gui really works) | |
293 | */ | |
294 | ui->jobfile_hbox = gtk_hbox_new(FALSE, 0); | |
295 | ui->jobfile_label = gtk_label_new("Job file:"); | |
296 | ui->jobfile_entry = gtk_entry_new(); | |
297 | gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_label); | |
298 | gtk_container_add(GTK_CONTAINER (ui->jobfile_hbox), ui->jobfile_entry); | |
299 | gtk_container_add(GTK_CONTAINER (ui->vbox), ui->jobfile_hbox); | |
300 | ||
04cc6b77 SC |
301 | /* |
302 | * Set up thread status progress bar | |
303 | */ | |
304 | ui->thread_status_pb = gtk_progress_bar_new(); | |
305 | gtk_progress_bar_set_fraction( | |
306 | GTK_PROGRESS_BAR(ui->thread_status_pb), 0.0); | |
307 | gtk_progress_bar_set_text( | |
308 | GTK_PROGRESS_BAR(ui->thread_status_pb), "No jobs running"); | |
309 | gtk_container_add(GTK_CONTAINER (ui->vbox), ui->thread_status_pb); | |
5b7573ab | 310 | |
736f2dff SC |
311 | /* |
312 | * Add a text box for text op messages | |
313 | */ | |
314 | ui->textview = gtk_text_view_new(); | |
315 | ui->text = gtk_text_view_get_buffer(GTK_TEXT_VIEW(ui->textview)); | |
316 | gtk_text_buffer_set_text(ui->text, "", -1); | |
317 | gtk_text_view_set_editable(GTK_TEXT_VIEW(ui->textview), FALSE); | |
318 | gtk_text_view_set_cursor_visible(GTK_TEXT_VIEW(ui->textview), FALSE); | |
319 | ui->scrolled_window = gtk_scrolled_window_new(NULL, NULL); | |
320 | gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(ui->scrolled_window), | |
321 | GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC); | |
322 | gtk_container_add(GTK_CONTAINER(ui->scrolled_window), ui->textview); | |
323 | gtk_container_add(GTK_CONTAINER(ui->vbox), ui->scrolled_window); | |
324 | ||
f3074008 | 325 | add_buttons(ui, buttonspeclist, ARRAYSIZE(buttonspeclist)); |
ff1f3280 SC |
326 | gtk_widget_show_all(ui->window); |
327 | } | |
328 | ||
8232e285 | 329 | int main(int argc, char *argv[], char *envp[]) |
ff1f3280 | 330 | { |
8232e285 SC |
331 | if (initialize_fio(envp)) |
332 | return 1; | |
a1820207 | 333 | |
ff1f3280 | 334 | init_ui(&argc, &argv, &ui); |
5b7573ab | 335 | |
2839f0c6 | 336 | gdk_threads_enter(); |
ff1f3280 | 337 | gtk_main(); |
2839f0c6 | 338 | gdk_threads_leave(); |
ff1f3280 SC |
339 | return 0; |
340 | } |