linux/net/rxrpc/rxperf.c

// SPDX-License-Identifier: GPL-2.0-or-later
/* In-kernel rxperf server for testing purposes.
 *
 * Copyright (C) 2022 Red Hat, Inc. All Rights Reserved.
 * Written by David Howells ([email protected])
 */

#define pr_fmt(fmt) …
#include <linux/module.h>
#include <linux/slab.h>
#include <net/sock.h>
#include <net/af_rxrpc.h>
#define RXRPC_TRACE_ONLY_DEFINE_ENUMS
#include <trace/events/rxrpc.h>

MODULE_DESCRIPTION(…) …;
MODULE_AUTHOR(…) …;
MODULE_LICENSE(…) …;

#define RXPERF_PORT …
#define RX_PERF_SERVICE …
#define RX_PERF_VERSION …
#define RX_PERF_SEND …
#define RX_PERF_RECV …
#define RX_PERF_RPC …
#define RX_PERF_FILE …
#define RX_PERF_MAGIC_COOKIE …

struct rxperf_proto_params { … } __packed;

static const u8 rxperf_magic_cookie[] = …;
static const u8 secret[8] = …;

enum rxperf_call_state { … };

struct rxperf_call { … };

static struct socket *rxperf_socket;
static struct key *rxperf_sec_keyring;	/* Ring of security/crypto keys */
static struct workqueue_struct *rxperf_workqueue;

static void rxperf_deliver_to_call(struct work_struct *work);
static int rxperf_deliver_param_block(struct rxperf_call *call);
static int rxperf_deliver_request(struct rxperf_call *call);
static int rxperf_process_call(struct rxperf_call *call);
static void rxperf_charge_preallocation(struct work_struct *work);

static DECLARE_WORK(rxperf_charge_preallocation_work,
		    rxperf_charge_preallocation);

static inline void rxperf_set_call_state(struct rxperf_call *call,
					 enum rxperf_call_state to)
{ … }

static inline void rxperf_set_call_complete(struct rxperf_call *call,
					    int error, s32 remote_abort)
{ … }

static void rxperf_rx_discard_new_call(struct rxrpc_call *rxcall,
				       unsigned long user_call_ID)
{ … }

static void rxperf_rx_new_call(struct sock *sk, struct rxrpc_call *rxcall,
			       unsigned long user_call_ID)
{ … }

static void rxperf_queue_call_work(struct rxperf_call *call)
{ … }

static void rxperf_notify_rx(struct sock *sk, struct rxrpc_call *rxcall,
			     unsigned long call_user_ID)
{ … }

static void rxperf_rx_attach(struct rxrpc_call *rxcall, unsigned long user_call_ID)
{ … }

static void rxperf_notify_end_reply_tx(struct sock *sock,
				       struct rxrpc_call *rxcall,
				       unsigned long call_user_ID)
{ … }

/*
 * Charge the incoming call preallocation.
 */
static void rxperf_charge_preallocation(struct work_struct *work)
{ … }

/*
 * Open an rxrpc socket and bind it to be a server for callback notifications
 * - the socket is left in blocking mode and non-blocking ops use MSG_DONTWAIT
 */
static int rxperf_open_socket(void)
{ … }

/*
 * close the rxrpc socket rxperf was using
 */
static void rxperf_close_socket(void)
{ … }

/*
 * Log remote abort codes that indicate that we have a protocol disagreement
 * with the server.
 */
static void rxperf_log_error(struct rxperf_call *call, s32 remote_abort)
{ … }

/*
 * deliver messages to a call
 */
static void rxperf_deliver_to_call(struct work_struct *work)
{ … }

/*
 * Extract a piece of data from the received data socket buffers.
 */
static int rxperf_extract_data(struct rxperf_call *call, bool want_more)
{ … }

/*
 * Grab the operation ID from an incoming manager call.
 */
static int rxperf_deliver_param_block(struct rxperf_call *call)
{ … }

/*
 * Deliver the request data.
 */
static int rxperf_deliver_request(struct rxperf_call *call)
{ … }

/*
 * Process a call for which we've received the request.
 */
static int rxperf_process_call(struct rxperf_call *call)
{ … }

/*
 * Add a key to the security keyring.
 */
static int rxperf_add_key(struct key *keyring)
{ … }

/*
 * Initialise the rxperf server.
 */
static int __init rxperf_init(void)
{ … }
late_initcall(rxperf_init); /* Must be called after net/ to create socket */

static void __exit rxperf_exit(void)
{ … }
module_exit(rxperf_exit);