1 /*
2  * QEMU I/O channel test helpers
3  *
4  * Copyright (c) 2015 Red Hat, Inc.
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18  *
19  */
20 
21 #include "qemu/osdep.h"
22 #include "io-channel-helpers.h"
23 #include "qemu/iov.h"
24 
25 struct QIOChannelTest {
26     QIOChannel *src;
27     QIOChannel *dst;
28     bool blocking;
29     size_t len;
30     size_t niov;
31     char *input;
32     struct iovec *inputv;
33     char *output;
34     struct iovec *outputv;
35     Error *writeerr;
36     Error *readerr;
37 };
38 
39 
40 /* This thread sends all data using iovecs */
41 static gpointer test_io_thread_writer(gpointer opaque)
42 {
43     QIOChannelTest *data = opaque;
44 
45     qio_channel_set_blocking(data->src, data->blocking, NULL);
46 
47     qio_channel_writev_all(data->src,
48                            data->inputv,
49                            data->niov,
50                            &data->writeerr);
51 
52     return NULL;
53 }
54 
55 
56 /* This thread receives all data using iovecs */
57 static gpointer test_io_thread_reader(gpointer opaque)
58 {
59     QIOChannelTest *data = opaque;
60 
61     qio_channel_set_blocking(data->dst, data->blocking, NULL);
62 
63     qio_channel_readv_all(data->dst,
64                           data->outputv,
65                           data->niov,
66                           &data->readerr);
67 
68     return NULL;
69 }
70 
71 
72 QIOChannelTest *qio_channel_test_new(void)
73 {
74     QIOChannelTest *data = g_new0(QIOChannelTest, 1);
75     size_t i;
76     size_t offset;
77 
78 
79     /* We'll send 1 MB of data */
80 #define CHUNK_COUNT 250
81 #define CHUNK_LEN 4194
82 
83     data->len = CHUNK_COUNT * CHUNK_LEN;
84     data->input = g_new0(char, data->len);
85     data->output = g_new0(gchar, data->len);
86 
87     /* Fill input with a pattern */
88     for (i = 0; i < data->len; i += CHUNK_LEN) {
89         memset(data->input + i, (i / CHUNK_LEN), CHUNK_LEN);
90     }
91 
92     /* We'll split the data across a bunch of IO vecs */
93     data->niov = CHUNK_COUNT;
94     data->inputv = g_new0(struct iovec, data->niov);
95     data->outputv = g_new0(struct iovec, data->niov);
96 
97     for (i = 0, offset = 0; i < data->niov; i++, offset += CHUNK_LEN) {
98         data->inputv[i].iov_base = data->input + offset;
99         data->outputv[i].iov_base = data->output + offset;
100         data->inputv[i].iov_len = CHUNK_LEN;
101         data->outputv[i].iov_len = CHUNK_LEN;
102     }
103 
104     return data;
105 }
106 
107 void qio_channel_test_run_threads(QIOChannelTest *test,
108                                   bool blocking,
109                                   QIOChannel *src,
110                                   QIOChannel *dst)
111 {
112     GThread *reader, *writer;
113 
114     test->src = src;
115     test->dst = dst;
116     test->blocking = blocking;
117 
118     reader = g_thread_new("reader",
119                           test_io_thread_reader,
120                           test);
121     writer = g_thread_new("writer",
122                           test_io_thread_writer,
123                           test);
124 
125     g_thread_join(reader);
126     g_thread_join(writer);
127 
128     test->dst = test->src = NULL;
129 }
130 
131 
132 void qio_channel_test_run_writer(QIOChannelTest *test,
133                                  QIOChannel *src)
134 {
135     test->src = src;
136     test_io_thread_writer(test);
137     test->src = NULL;
138 }
139 
140 
141 void qio_channel_test_run_reader(QIOChannelTest *test,
142                                  QIOChannel *dst)
143 {
144     test->dst = dst;
145     test_io_thread_reader(test);
146     test->dst = NULL;
147 }
148 
149 
150 void qio_channel_test_validate(QIOChannelTest *test)
151 {
152     g_assert(test->readerr == NULL);
153     g_assert(test->writeerr == NULL);
154     g_assert_cmpint(memcmp(test->input,
155                            test->output,
156                            test->len), ==, 0);
157 
158     g_free(test->inputv);
159     g_free(test->outputv);
160     g_free(test->input);
161     g_free(test->output);
162     g_free(test);
163 }
164