1

#ifndef SOCK_TEST_H

     2

#define SOCK_TEST_H

     3

     4

/**

     5

 * @file

     6

 *

     7

 * Dummy sock_stream implemention for local testing.

     8

 */

     9

#include "sock_internal.h"

    10

    11

/**

    12

 * IO vector

    13

 */

    14

struct io_vec {

    15

    /** The buffer */

    16

    char *buf;

    17

    18

    /** Buffer size */

    19

    size_t len;

    20

};

    21

    22

/**

    23

 * Vectored IO-buffer

    24

 */

    25

struct io_buf {

    26

    /** The array of buffer-vectors, {NULL}-terminated */

    27

    struct io_vec *vecs;

    28

    29

    /** The number of io_vecs */

    30

    size_t count;

    31

    32

    /** Current vector */

    33

    struct io_vec *vec;

    34

    35

    /** Offset into current vector */

    36

    size_t off;

    37

};

    38

    39

/**

    40

 * The per-socket state

    41

 */

    42

struct sock_test {

    43

    /** The base struct for sock_stream_* functions */

    44

    struct sock_stream base;

    45

    46

    /** The send/recieve buffers */

    47

    struct io_buf send_buf, recv_buf;

    48

};

    49

    50

/**

    51

 * Get a sock_stream pointer from a sock_tcp pointer

    52

 */

    53

#define SOCK_TEST_BASE(sock_ptr) (&(sock_ptr)->base)

    54

    55

/**

    56

 * A dummy stream socket intended for testing purposes.

    57

 */

    58

struct sock_test* sock_test_create (void);

    59

    60

/**

    61

 * Set the recieve buffer contents.

    62

 *

    63

 * The data is not copied, but the vectors are stored as-is.

    64

 */

    65

void sock_test_set_recv_buffer (struct sock_test *sock, struct io_vec *vecs, size_t count);

    66

    67

/**

    68

 * Get the send buffer contents as a single string, free() after use if you care about that

    69

 */

    70

void sock_test_get_send_data (struct sock_test *sock, char **buf, size_t *len);

    71

    72

#endif