From 78c896b3b3127515478090c19447e27dc406427e Mon Sep 17 00:00:00 2001 From: Jianfeng Tan Date: Mon, 18 Nov 2019 06:59:50 +0000 Subject: TLDKv2 Signed-off-by: Jianfeng Tan Signed-off-by: Jielong Zhou Signed-off-by: Jian Zhang Signed-off-by: Chen Zhao Change-Id: I55c39de4c6cd30f991f35631eb507f770230f08e --- test/packetdrill/wire_conn.h | 88 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 test/packetdrill/wire_conn.h (limited to 'test/packetdrill/wire_conn.h') diff --git a/test/packetdrill/wire_conn.h b/test/packetdrill/wire_conn.h new file mode 100644 index 0000000..d8b85fe --- /dev/null +++ b/test/packetdrill/wire_conn.h @@ -0,0 +1,88 @@ +/* + * Copyright 2013 Google Inc. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + */ +/* + * Author: ncardwell@google.com (Neal Cardwell) + * + * TCP connection handling for remote on-the-wire testing using a real NIC. + */ + +#ifndef __WIRE_CONN_H__ +#define __WIRE_CONN_H__ + +#include "types.h" + +#include "ip_address.h" +#include "wire_protocol.h" + +struct config; + +/* Buffer holding input or output data for a TCP socket. */ +struct wire_conn_buffer { + char *buf; /* malloc-allocated buffer */ + int buf_space; /* bytes allocated in malloc-allocated "buf" buffer */ + int used; /* bytes of actual data at the start of "buf" */ +}; + +/* A TCP socket used for client<->server communication for doing + * remote on-the-wire testing using a real NIC. + */ +struct wire_conn { + int fd; /* socket for TCP connection (or -1) */ + struct wire_conn_buffer in; /* data read in last wire_conn_read() */ +}; + +/* Create a wire_conn. Note that a struct wire_conn shouldn't be + * stack-allocated and should always use wire_conn_new() and + * wire_conn_free(). + */ +struct wire_conn *wire_conn_new(void); + +/* Free a wire_conn. */ +void wire_conn_free(struct wire_conn *conn); + +/* Blocking connect. */ +void wire_conn_connect(struct wire_conn *conn, + const struct ip_address *ip, u16 port, + enum ip_version_t ip_version); + +/* Blocking bind and listen. */ +void wire_conn_bind_listen(struct wire_conn *listen_conn, u16 port, + enum ip_version_t ip_version); + +/* Blocking accept. */ +void wire_conn_accept(struct wire_conn *listen_conn, + struct wire_conn **accepted_conn); + +/* Blocking write of a single message. */ +int wire_conn_write(struct wire_conn *conn, + enum wire_op_t op, + const void *buf, int buf_len); + +/* Blocking read of a single message. Changes *buf to point to the + * wire_conn_buffer of this connection, which is guaranteed to be big + * enough to hold the whole *buf_len bytes returned. The wire_conn + * owns this memory, not the caller. The returned buffer can be + * invalidated (freed or re-written) by the next call to + * wire_conn_read(). + */ +int wire_conn_read(struct wire_conn *conn, + enum wire_op_t *op, + void **buf, int *buf_len); + +#endif /* __WIRE_CONN_H__ */ -- cgit 1.2.3-korg