1/* SPDX-License-Identifier: GPL-2.0-only */
 2#ifndef UTIL_H
 3#define UTIL_H
 4
 5#include <sys/socket.h>
 6#include <linux/vm_sockets.h>
 7
 8/* Tests can either run as the client or the server */
 9enum test_mode {
10	TEST_MODE_UNSET,
11	TEST_MODE_CLIENT,
12	TEST_MODE_SERVER
13};
14
15/* Test runner options */
16struct test_opts {
17	enum test_mode mode;
18	unsigned int peer_cid;
19};
20
21/* A test case definition.  Test functions must print failures to stderr and
22 * terminate with exit(EXIT_FAILURE).
23 */
24struct test_case {
25	const char *name; /* human-readable name */
26
27	/* Called when test mode is TEST_MODE_CLIENT */
28	void (*run_client)(const struct test_opts *opts);
29
30	/* Called when test mode is TEST_MODE_SERVER */
31	void (*run_server)(const struct test_opts *opts);
32
33	bool skip;
34};
35
36void init_signals(void);
37unsigned int parse_cid(const char *str);
38int vsock_stream_connect(unsigned int cid, unsigned int port);
 
 
39int vsock_seqpacket_connect(unsigned int cid, unsigned int port);
40int vsock_stream_accept(unsigned int cid, unsigned int port,
41			struct sockaddr_vm *clientaddrp);
 
42int vsock_seqpacket_accept(unsigned int cid, unsigned int port,
43			   struct sockaddr_vm *clientaddrp);
44void vsock_wait_remote_close(int fd);
 
 
 
45void send_byte(int fd, int expected_ret, int flags);
46void recv_byte(int fd, int expected_ret, int flags);
47void run_tests(const struct test_case *test_cases,
48	       const struct test_opts *opts);
49void list_tests(const struct test_case *test_cases);
50void skip_test(struct test_case *test_cases, size_t test_cases_len,
51	       const char *test_id_str);
 
 
 
 
 
 
52#endif /* UTIL_H */

 1/* SPDX-License-Identifier: GPL-2.0-only */
 2#ifndef UTIL_H
 3#define UTIL_H
 4
 5#include <sys/socket.h>
 6#include <linux/vm_sockets.h>
 7
 8/* Tests can either run as the client or the server */
 9enum test_mode {
10	TEST_MODE_UNSET,
11	TEST_MODE_CLIENT,
12	TEST_MODE_SERVER
13};
14
15/* Test runner options */
16struct test_opts {
17	enum test_mode mode;
18	unsigned int peer_cid;
19};
20
21/* A test case definition.  Test functions must print failures to stderr and
22 * terminate with exit(EXIT_FAILURE).
23 */
24struct test_case {
25	const char *name; /* human-readable name */
26
27	/* Called when test mode is TEST_MODE_CLIENT */
28	void (*run_client)(const struct test_opts *opts);
29
30	/* Called when test mode is TEST_MODE_SERVER */
31	void (*run_server)(const struct test_opts *opts);
32
33	bool skip;
34};
35
36void init_signals(void);
37unsigned int parse_cid(const char *str);
38int vsock_stream_connect(unsigned int cid, unsigned int port);
39int vsock_bind_connect(unsigned int cid, unsigned int port,
40		       unsigned int bind_port, int type);
41int vsock_seqpacket_connect(unsigned int cid, unsigned int port);
42int vsock_stream_accept(unsigned int cid, unsigned int port,
43			struct sockaddr_vm *clientaddrp);
44int vsock_stream_listen(unsigned int cid, unsigned int port);
45int vsock_seqpacket_accept(unsigned int cid, unsigned int port,
46			   struct sockaddr_vm *clientaddrp);
47void vsock_wait_remote_close(int fd);
48void send_buf(int fd, const void *buf, size_t len, int flags,
49	      ssize_t expected_ret);
50void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret);
51void send_byte(int fd, int expected_ret, int flags);
52void recv_byte(int fd, int expected_ret, int flags);
53void run_tests(const struct test_case *test_cases,
54	       const struct test_opts *opts);
55void list_tests(const struct test_case *test_cases);
56void skip_test(struct test_case *test_cases, size_t test_cases_len,
57	       const char *test_id_str);
58unsigned long hash_djb2(const void *data, size_t len);
59size_t iovec_bytes(const struct iovec *iov, size_t iovnum);
60unsigned long iovec_hash_djb2(const struct iovec *iov, size_t iovnum);
61struct iovec *alloc_test_iovec(const struct iovec *test_iovec, int iovnum);
62void free_test_iovec(const struct iovec *test_iovec,
63		     struct iovec *iovec, int iovnum);
64#endif /* UTIL_H */