tools/testing/vsock/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 */
   9 enum test_mode {
  10         TEST_MODE_UNSET,
  11         TEST_MODE_CLIENT,
  12         TEST_MODE_SERVER
  13 };
  14
  15 /* Test runner options */
  16 struct 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  */
  24 struct 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
  36 void init_signals(void);
  37 unsigned int parse_cid(const char *str);
  38 int vsock_stream_connect(unsigned int cid, unsigned int port);
  39 int vsock_bind_connect(unsigned int cid, unsigned int port,
  40                        unsigned int bind_port, int type);
  41 int vsock_seqpacket_connect(unsigned int cid, unsigned int port);
  42 int vsock_stream_accept(unsigned int cid, unsigned int port,
  43                         struct sockaddr_vm *clientaddrp);
  44 int vsock_stream_listen(unsigned int cid, unsigned int port);
  45 int vsock_seqpacket_accept(unsigned int cid, unsigned int port,
  46                            struct sockaddr_vm *clientaddrp);
  47 void vsock_wait_remote_close(int fd);
  48 void send_buf(int fd, const void *buf, size_t len, int flags,
  49               ssize_t expected_ret);
  50 void recv_buf(int fd, void *buf, size_t len, int flags, ssize_t expected_ret);
  51 void send_byte(int fd, int expected_ret, int flags);
  52 void recv_byte(int fd, int expected_ret, int flags);
  53 void run_tests(const struct test_case *test_cases,
  54                const struct test_opts *opts);
  55 void list_tests(const struct test_case *test_cases);
  56 void skip_test(struct test_case *test_cases, size_t test_cases_len,
  57                const char *test_id_str);
  58 unsigned long hash_djb2(const void *data, size_t len);
  59 size_t iovec_bytes(const struct iovec *iov, size_t iovnum);
  60 unsigned long iovec_hash_djb2(const struct iovec *iov, size_t iovnum);
  61 struct iovec *alloc_test_iovec(const struct iovec *test_iovec, int iovnum);
  62 void free_test_iovec(const struct iovec *test_iovec,
  63                      struct iovec *iovec, int iovnum);
  64 #endif /* UTIL_H */